Updated testlib.h patch to new version

cip999 · Jan 4, 2023 · 7785d40 · 7785d40
1 parent 3359d23
commit 7785d40
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/p2d/generate_testlib_for_domjudge.py b/p2d/generate_testlib_for_domjudge.py
@@ -23,6 +23,7 @@
 NEW_REGISTER_INTERACTION = '''\
 void registerInteraction(int argc, char *argv[]) {
     __testlib_ensuresPreconditions();
+    TestlibFinalizeGuard::registered = true;
 
     testlibMode = _interactor;
     __testlib_set_binary(stdin);
@@ -57,6 +58,7 @@
 NEW_REGISTER_TESTLIB_CMD = '''\
 void registerTestlibCmd(int argc, char *argv[]) {
     __testlib_ensuresPreconditions();
+    TestlibFinalizeGuard::registered = true;
 
     testlibMode = _checker;
     __testlib_set_binary(stdin);