configure: --disable-prgname-prefix

gc00 · gc00 · commit dbf404a4ab1a · 2025-11-13T14:39:58.000-05:00
* This creates PRGNAME_PREFIX; The 'DMTCP:' prefix will be prepended
    to the comm field in 'ps' on restarted processes _only_ if this is
    enabled (default).
  * This is needed for the CUDA ckpt API, which uses the 'comm' field
    on restart
diff --git a/configure b/configure
@@ -846,6 +846,7 @@ enable_test_suite
 enable_experts_only_space1
 enable_experts_only
 enable_experts_only_after
+enable_prgname_prefix
 enable_fsgsbase_override
 enable_m32
 enable_multilib
@@ -1535,6 +1536,10 @@ Optional Features:
                  === **** NOTE:  EXPERTS ONLY BELOW HERE **** ===
                           (Use at your own risk!!)
 
+  --disable-prgname-prefix
+                          Disables 'DMTCP:' prefix of comm field in ps to
+                          avoid incompatibility with CUDA ckp API
+                          (Alternative: set DISABLE_PRGNAME_PREFIX env var)
   --enable-fsgsbase-override
                           Defines HAS_FSGSBASE for cross-configure; The target
                           computer must have the FSGSBASE kernel patch.
@@ -8073,6 +8078,20 @@ then :
 fi
 
 
+# Check whether --enable-prgname-prefix was given.
+if test ${enable_prgname_prefix+y}
+then :
+  enableval=$enable_prgname_prefix; use_prgname_prefix=$enableval
+else $as_nop
+  use_prgname_prefix=yes
+fi
+
+if test "$use_prgname_prefix" = "yes"; then
+
+printf "%s\n" "#define ENABLE_PRGNAME_PREFIX 1" >>confdefs.h
+
+fi
+
 # Check whether --enable-fsgsbase-override was given.
 if test ${enable_fsgsbase_override+y}
 then :
diff --git a/configure.ac b/configure.ac
@@ -229,6 +229,19 @@ AC_ARG_ENABLE([experts_only],
 	    [], [])
 AC_ARG_ENABLE([experts_only_after], [AS_HELP_STRING([ ], [])], [], [])
 
+AC_ARG_ENABLE([prgname-prefix],
+            [AS_HELP_STRING([--disable-prgname-prefix],
+                            [Disables 'DMTCP:' prefix of comm field in ps
+                             to avoid incompatibility with CUDA ckp API
+                             (Alternative:  set DISABLE_PRGNAME_PREFIX
+                             env var)])],
+            [use_prgname_prefix=$enableval],
+            [use_prgname_prefix=yes])
+if test "$use_prgname_prefix" = "yes"; then
+  AC_DEFINE([ENABLE_PRGNAME_PREFIX],[1],[Enable 'DMTCP:' prefix in comm
+                                         field of 'ps' (for CUDA ckpt)])
+fi
+
 AC_ARG_ENABLE([fsgsbase-override],
             [AS_HELP_STRING([--enable-fsgsbase-override],
                             [Defines HAS_FSGSBASE for cross-configure; The
diff --git a/include/config.h.in b/include/config.h.in
@@ -21,6 +21,9 @@
 /* Override to use FSGSBASE kernel patch */
 #undef ENABLE_FSGSBASE_OVERRIDE
 
+/* Enable 'DMTCP:' prefix in comm field of 'ps' (for CUDA ckpt) */
+#undef ENABLE_PRGNAME_PREFIX
+
 /* Child process does checkpointing */
 #undef FORKED_CHECKPOINTING
 
@@ -51,7 +54,7 @@
 /* Define to 1 if you have the <inttypes.h> header file. */
 #undef HAVE_INTTYPES_H
 
-/* Define to 1 if you have the 'atomic' library (-latomic). */
+/* Define to 1 if you have the `atomic' library (-latomic). */
 #undef HAVE_LIBATOMIC
 
 /* Define to 1 if you have the <linux/version.h> header file. */
@@ -132,7 +135,7 @@
 /* No output, not even NOTE and WARNING */
 #undef QUIET
 
-/* Define to 1 if all of the C89 standard headers exist (not just the ones
+/* Define to 1 if all of the C90 standard headers exist (not just the ones
    required in a freestanding environment). This macro is provided for
    backward compatibility; new code need not use it. */
 #undef STDC_HEADERS
diff --git a/src/threadlist.cpp b/src/threadlist.cpp
@@ -659,6 +659,7 @@ stopthisthread(int signum)
       JASSERT(DmtcpRWLockUnlock(&threadResumeLock) == 0);
     } else {
 #if LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 11)
+# ifdef PRGNAME_PREFIX
       if (!Util::strStartsWith(curThread->procname, DMTCP_PRGNAME_PREFIX)) {
         // Add the "DMTCP:" prefix.
         string newName = string(DMTCP_PRGNAME_PREFIX) + curThread->procname;
@@ -669,6 +670,7 @@ stopthisthread(int signum)
         // Add a NULL at the end to make sure the string terminates in all cases
         curThread->procname[sizeof(curThread->procname) - 1] = '\0';
       }
+# endif
       JASSERT(prctl(PR_SET_NAME, curThread->procname) != -1 || errno == EINVAL)
         (curThread->procname) (JASSERT_ERRNO)
       .Text("prctl(PR_SET_NAME, ...) failed");
