X-Git-Url: https://git.distorted.org.uk/~mdw/sgt/puzzles/blobdiff_plain/ac9f41c4c8ccbf9364a5d351e4cb134812a9926b..08d7c4ba7266c626d28273750218446d3c693805:/loopy.c

diff --git a/loopy.c b/loopy.c
index 9f54c8a..ce49c73 100644
--- a/loopy.c
+++ b/loopy.c
@@ -683,13 +683,12 @@ static void print_board(const game_params *params, const char *board)
 #endif
 }
 
-static char *new_fullyclued_board(game_params *params, random_state *rs)
+static void add_full_clues(game_state *state, game_params *params,
+			   random_state *rs)
 {
     char *clues;
     char *board;
     int i, j, a, b, c;
-    game_state s;
-    game_state *state = &s;
     int board_area = SQUARE_COUNT(params);
     int t;
 
@@ -763,11 +762,7 @@ static char *new_fullyclued_board(game_params *params, random_state *rs)
 #define SCORE_DISTANCE 1
 
     board = snewn(board_area, char);
-    clues = snewn(board_area, char);
-
-    state->h = params->h;
-    state->w = params->w;
-    state->clues = clues;
+    clues = state->clues;
 
     /* Make a board */
     memset(board, SQUARE_UNLIT, board_area);
@@ -932,7 +927,6 @@ static char *new_fullyclued_board(game_params *params, random_state *rs)
     }
 
     sfree(board);
-    return clues;
 }
 
 static solver_state *solve_game_rec(const solver_state *sstate, int diff);
@@ -1004,6 +998,7 @@ static char *new_game_desc(game_params *params, random_state *rs,
     state->h = params->h;
     state->w = params->w;
 
+    state->clues = snewn(SQUARE_COUNT(params), char);
     state->hl = snewn(HL_COUNT(params), char);
     state->vl = snewn(VL_COUNT(params), char);
 
@@ -1019,7 +1014,7 @@ newboard_please:
      * preventing games smaller than 4x4 seems to stop this happening */
 
     do {
-        state->clues = new_fullyclued_board(params, rs);
+        add_full_clues(state, params, rs);
     } while (!game_has_unique_soln(state, params->diff));
 
     state_new = remove_clues(state, rs, params->diff);
@@ -1210,8 +1205,14 @@ static char *encode_solve_move(const game_state *state)
         }
     }
 
-    /* No point in doing sums like that if they're going to be wrong */
-    assert(strlen(ret) == (size_t)len);
+    /*
+     * Ensure we haven't overrun the buffer we allocated (which we
+     * really shouldn't have, since we computed its maximum size).
+     * Note that this assert is <= rather than ==, because the
+     * solver is permitted to produce an incomplete solution in
+     * which case the buffer will be only partially used.
+     */
+    assert(strlen(ret) <= (size_t)len);
     return ret;
 }
 
@@ -1282,7 +1283,7 @@ static int dot_setall_dlines(solver_state *sstate, enum dline dl, int i, int j,
             break;                                                          
         case DLINE_DL:                                                  
         case DLINE_DR:                                                  
-            if (j <= (state)->h && BELOW_DOT(state, i, j) == line_old) {
+            if (j < (state)->h && BELOW_DOT(state, i, j) == line_old) {
                 LV_BELOW_DOT(state, i, j) = line_new;                   
                 retval = TRUE;
             }
@@ -1307,13 +1308,13 @@ static int dot_setall_dlines(solver_state *sstate, enum dline dl, int i, int j,
         case DLINE_UR:                                                  
         case DLINE_DR:                                                  
         case DLINE_HORIZ:                                                  
-            if (i <= (state)->w && RIGHTOF_DOT(state, i, j) == line_old) {
+            if (i < (state)->w && RIGHTOF_DOT(state, i, j) == line_old) {
                 LV_RIGHTOF_DOT(state, i, j) = line_new;                 
                 retval = TRUE;
             }
             break;                                                          
         case DLINE_VERT:                                                  
-            if (j <= (state)->h && BELOW_DOT(state, i, j) == line_old) {
+            if (j < (state)->h && BELOW_DOT(state, i, j) == line_old) {
                 LV_BELOW_DOT(state, i, j) = line_new;                   
                 retval = TRUE;
             }