uvm32/apps/maze/maze.c
Stéphane Travostino cdfae7d453 apps/maze: fix possible stack overflow
The original code this app is inspired from uses rand(), which returns
a number >= 0.
mulberry32() on the other hand can return a negative number, in which
case will overflow the accesses in the dirs array. Fix that by forcing
the result to be unsigned.
2025-12-14 10:15:04 +01:00

76 lines
1.6 KiB
C

#include "uvm32_target.h"
// This example generates a random maze using the recursive backtracking algorithm.
// https://github.com/ccattuto/riscv-python/blob/main/tests/test_newlib_maze.c
#define WIDTH 79 // must be odd
#define HEIGHT 31 // must be odd
char maze[HEIGHT * WIDTH];
void* my_memset(void* buf, int c, int len) {
uint8_t* b = (uint8_t*)buf;
while (len--) {
*(b++) = c;
}
return buf;
}
int dx[] = {0, 1, 0, -1};
int dy[] = {-1, 0, 1, 0};
uint32_t state = 0x1;
int32_t mulberry32(void) {
uint32_t z = (state += 0x6D2B79F5UL);
z = (z ^ (z >> 15)) * (z | 1UL);
z ^= z + (z ^ (z >> 7)) * (z | 61UL);
return z ^ (z >> 14);
}
void init_maze() {
my_memset(maze, '#', sizeof(maze));
}
int in_bounds(int x, int y) {
return x > 0 && y > 0 && x < WIDTH - 1 && y < HEIGHT - 1;
}
void carve(int x, int y) {
maze[y * WIDTH + x] = ' ';
int dirs[] = {0, 1, 2, 3};
// Fisher-Yates shuffle
for (int i = 3; i > 0; i--) {
int j = (unsigned)mulberry32() % (i + 1);
int tmp = dirs[i];
dirs[i] = dirs[j];
dirs[j] = tmp;
}
for (int i = 0; i < 4; i++) {
int nx = x + dx[dirs[i]] * 2;
int ny = y + dy[dirs[i]] * 2;
if (in_bounds(nx, ny) && maze[ny * WIDTH + nx] == '#') {
maze[(y + dy[dirs[i]]) * WIDTH + (x + dx[dirs[i]])] = ' ';
carve(nx, ny);
}
}
}
void print_maze() {
for (int y = 0; y < HEIGHT; y++) {
for (int x = 0; x < WIDTH; x++) {
putc(maze[y * WIDTH + x]);
}
putc('\n');
}
}
void main(void) {
init_maze();
carve(1, 1);
print_maze();
}