Merge pull request #88 from katahiromz/copilot/fix-crossword-generation-race-condition

Fix race conditions in crossword generation completion/cancellation

Author: katahiromz

XG_CancelFromWordsDialog.hpp

@@ -28,7 +28,10 @@ class XG_CancelFromWordsDialog : public XG_Dialog
         // タイマーを解除する。
         ::KillTimer(hwnd, uTimerID);
         // キャンセルしてスレッドを待つ。
-        crossword_generation::s_canceled = true;
+        // 生成完了していなければキャンセル
+        if (!crossword_generation::s_generated) {
+            crossword_generation::s_canceled = true;
+        }
         ::Sleep(SLEEP);
     }
 

XG_CancelGenBlacksDialog.hpp

@@ -22,7 +22,10 @@ class XG_CancelGenBlacksDialog : public XG_Dialog
         ::KillTimer(hwnd, uTimerID);
         // キャンセルしてスレッドを待つ。
         ::EnterCriticalSection(&xg_csLock);
-        xg_bCancelled = true;
+        // 生成完了していなければキャンセル
+        if (!xg_bSolved) {
+            xg_bCancelled = true;
+        }
         m_cancellation_manager.SetCompleted();
         ::LeaveCriticalSection(&xg_csLock);
         // スレッドを待つ（タイムアウト付き）。

XG_CancelSmartSolveDialog.hpp

@@ -38,7 +38,10 @@ class XG_CancelSmartSolveDialog : public XG_Dialog
         ::KillTimer(hwnd, uTimerID);
         // キャンセルしてスレッドを待つ。
         ::EnterCriticalSection(&xg_csLock);
-        xg_bCancelled = true;
+        // 生成完了していなければキャンセル
+        if (!xg_bSolved) {
+            xg_bCancelled = true;
+        }
         m_cancellation_manager.SetCompleted();
         ::LeaveCriticalSection(&xg_csLock);
         // スレッドを待つ（タイムアウト付き）。
@@ -60,7 +63,10 @@ class XG_CancelSmartSolveDialog : public XG_Dialog
         ::KillTimer(hwnd, uTimerID);
         // キャンセルしてスレッドを待つ。
         ::EnterCriticalSection(&xg_csLock);
-        xg_bCancelled = true;
+        // 生成完了していなければキャンセル
+        if (!xg_bSolved) {
+            xg_bCancelled = true;
+        }
         m_cancellation_manager.SetCompleted();
         ::LeaveCriticalSection(&xg_csLock);
         // スレッドを待つ（タイムアウト付き）。

XG_CancelSolveDialog.hpp

@@ -37,7 +37,10 @@ class XG_CancelSolveDialog : public XG_Dialog
         ::KillTimer(hwnd, uTimerID);
         // キャンセルしてスレッドを待つ。
         ::EnterCriticalSection(&xg_csLock);
-        xg_bCancelled = true;
+        // 生成完了していなければキャンセル
+        if (!xg_bSolved) {
+            xg_bCancelled = true;
+        }
         m_cancellation_manager.SetCompleted();
         ::LeaveCriticalSection(&xg_csLock);
         // スレッドを待つ（タイムアウト付き）。
@@ -60,7 +63,10 @@ class XG_CancelSolveDialog : public XG_Dialog
 
         // キャンセルしてスレッドを待つ。
         ::EnterCriticalSection(&xg_csLock);
-        xg_bCancelled = true;
+        // 生成完了していなければキャンセル
+        if (!xg_bSolved) {
+            xg_bCancelled = true;
+        }
         m_cancellation_manager.SetCompleted();
         ::LeaveCriticalSection(&xg_csLock);
         // スレッドを待つ（タイムアウト付き）。

XG_CancelSolveNoAddBlackDialog.hpp

@@ -37,7 +37,10 @@ class XG_CancelSolveNoAddBlackDialog : public XG_Dialog
         ::KillTimer(hwnd, uTimerID);
         // キャンセルしてスレッドを待つ。
         ::EnterCriticalSection(&xg_csLock);
-        xg_bCancelled = true;
+        // 生成完了していなければキャンセル
+        if (!xg_bSolved) {
+            xg_bCancelled = true;
+        }
         m_cancellation_manager.SetCompleted();
         ::LeaveCriticalSection(&xg_csLock);
         // スレッドを待つ（タイムアウト付き）。

XG_CancellationManager.hpp

@@ -48,9 +48,12 @@ class XG_CancellationManager
     // Notify completion (on successful processing)
     void SetCompleted() noexcept
     {
-        if (m_hCompletionEvent != nullptr) {
+        ::EnterCriticalSection(&xg_csLock);
+        // キャンセル済みでなければ完了を通知
+        if (!xg_bCancelled && m_hCompletionEvent != nullptr) {
             ::SetEvent(m_hCompletionEvent);
         }
+        ::LeaveCriticalSection(&xg_csLock);
     }
 
     // 完了またはキャンセルを待機（タイムアウト付き）

crossword_generation.hpp

@@ -14,6 +14,7 @@
 #include <queue>                // 待ち行列（std::queue）
 #include <thread>               // std::thread
 #include <mutex>                // std::mutex
+#include <atomic>               // std::atomic
 #include <algorithm>            // std::shuffle
 #include <utility>              // ????
 #include <random>               // 新しい乱数生成。
@@ -60,8 +61,8 @@ namespace std {
 // クロスワード生成用の名前空間。現状は「単語群から自動生成」のみ実装。
 // 将来的にはその他の生成方法もこのようなモダンな方向に移行する。
 namespace crossword_generation {
-inline static bool s_generated = false;     // 生成済みか？
-inline static bool s_canceled = false;      // キャンセルされたか？
+inline static std::atomic<bool> s_generated{false};     // 生成済みか？
+inline static std::atomic<bool> s_canceled{false};      // キャンセルされたか？
 inline static std::mutex s_mutex;           // ミューテックス（排他処理用）。
 
 // ルールを表すフラグ群。
@@ -1546,10 +1547,12 @@ struct from_words_t {
                 board0.replace('?', '#');
                 if (is_solution(board0)) { // 盤面が解ならば
                     std::lock_guard<std::mutex> lock(s_mutex); // 排他制御しながら
-                    // 解をセットして
-                    s_generated = true;
-                    s_solution = board0;
-                    return true; // 成功。
+                    // キャンセルされていなければ解をセット
+                    if (!s_canceled) {
+                        s_generated = true;
+                        s_solution = board0;
+                    }
+                    return !s_canceled; // キャンセルされていなければ成功。
                 }
             }
             return s_generated; // 生成済みなら成功。
@@ -1930,9 +1933,12 @@ struct non_add_block_t {
         // 盤面が解ならば成功。
         if (is_solution(m_board)) {
             std::lock_guard<std::mutex> lock(s_mutex);
-            s_generated = true;
-            s_solution = m_board;
-            return true;
+            // キャンセルされていなければ解をセット
+            if (!s_canceled) {
+                s_generated = true;
+                s_solution = m_board;
+            }
+            return !s_canceled; // キャンセルされていなければ成功。
         }
 
         return s_generated; // 生成済みなら成功。