Support restore with -XX:CRaCRestoreFrom=PATH

Signed-off-by: Jason Feng <fengj@ca.ibm.com>

Author: JasonFengJ9

closed/src/java.base/share/native/libjli/criuhelpers.c

@@ -22,9 +22,16 @@
  * ===========================================================================
  */
 
-#include "criuhelpers.h"
+#include "j9cfg.h"
 
 #if defined(J9VM_OPT_CRAC_SUPPORT)
+
+#include "java.h"
+#include <ctype.h>
+#include <sys/wait.h>
+
+void handleCRaCRestore(int argc, char **argv);
+
 /**
  * Get the value of a command line option in the array of command line arguments.
  * @param[in] optionName The name of the command line option to search for
@@ -37,7 +44,7 @@ static const char *
 getCommandLineOptionValue(const char *optionName, int argc, char **argv, int *error)
 {
     const char *value = NULL;
-    int i = argc - 1;
+    int i = 0;
     size_t optionNameLength = strlen(optionName);
     *error = -1;
     for (i = argc - 1; i >= 0; i--) {
@@ -94,8 +101,7 @@ static int
 getLogLevel(int argc, char **argv, int *error)
 {
     int logLevelValue = 2; /* default */
-    const char *logLevelPropertyValue = NULL;
-    logLevelPropertyValue = getCommandLineOptionValue("-Dopenj9.internal.criu.logLevel", argc, argv, error);
+    const char *logLevelPropertyValue = getCommandLineOptionValue("-Dopenj9.internal.criu.logLevel", argc, argv, error);
     if (0 == *error) {
         const char *c = logLevelPropertyValue;
         if (NULL == c) {
@@ -116,7 +122,9 @@ getLogLevel(int argc, char **argv, int *error)
         goto done;
     }
 setLogLevelOptionValueNotValidError:
-    JLI_ReportErrorMessage("The value of the command line option is not valid, option=-Dopenj9.internal.criu.logLevel, value=%s.", logLevelPropertyValue);
+    JLI_ReportErrorMessage(
+        "The option '-Dopenj9.internal.criu.logLevel=%s' is not valid.",
+        logLevelPropertyValue);
     *error = -2;
 done:
     return logLevelValue;
@@ -138,7 +146,9 @@ isUnprivilegedModeOn(int argc, char **argv, int *error)
         if (NULL == unprivilegedModePropertyValue) {
             isUnprivilegedModeOn = JNI_TRUE;
         } else {
-            JLI_ReportErrorMessage("The value of the command line option is not expected, option=-Dopenj9.internal.criu.unprivilegedMode, value=%s.", unprivilegedModePropertyValue);
+            JLI_ReportErrorMessage(
+                    "The option '-Dopenj9.internal.criu.unprivilegedMode=%s' does not accept a value.",
+                    unprivilegedModePropertyValue);
             *error = -2;
         }
     }
@@ -161,7 +171,7 @@ getLogFile(int argc, char **argv, int *error)
         if (NULL != logFilePropertyValue) {
             logFile = logFilePropertyValue;
         } else {
-            JLI_ReportErrorMessage("The value of the command line option -Dopenj9.internal.criu.logFile was not found.");
+            JLI_ReportErrorMessage("The option -Dopenj9.internal.criu.logFile requires a value.");
             *error = -2;
         }
     }
@@ -179,7 +189,6 @@ getLogFile(int argc, char **argv, int *error)
 static int
 restoreFromCheckpoint(const char *checkpointDirectory, int logLevel, jboolean unprivilegedModeOn, const char *logFile)
 {
-    int restoreStatus = 0;
     int length = -1;
     char logLevelOption[4] = { 0 };
     char *logFileOption = NULL;
@@ -192,45 +201,35 @@ restoreFromCheckpoint(const char *checkpointDirectory, int logLevel, jboolean un
     length = snprintf(NULL, 0, "-v%d", logLevel);
     if (length < 0) {
         JLI_ReportErrorMessage("Failed to calculate the length of option '-v%d'.", logLevel);
-        restoreStatus = -1;
-        goto done;
-    }
-    if (snprintf(logLevelOption, sizeof(logLevelOption), "-v%d", logLevel) < 0) {
-        JLI_ReportErrorMessage("Failed to format the log level option '-v%d'.", logLevel);
-        restoreStatus = -1;
-        goto done;
+        goto fail;
     }
+    snprintf(logLevelOption, sizeof(logLevelOption), "-v%d", logLevel);
     argv[argc++] = logLevelOption;
     argv[argc++] = "--shell-job";
     if (unprivilegedModeOn) {
         argv[argc++] = "--unprivileged";
     }
     if (NULL != logFile) {
         length = strlen(logFile) + sizeof("--log-file=%s") - 1;
-        logFileOption = (char *)JLI_MemAlloc(length - 1);
+        logFileOption = (char *)JLI_MemAlloc(length + 1);
         if (NULL == logFileOption) {
             JLI_ReportErrorMessage("Failed to allocate memory for option '--log-file=%s'.", logFile);
-            restoreStatus = -1;
-            goto done;
+            goto fail;
         }
-        if (snprintf(logFileOption, length - 1, "--log-file=%s", logFile) < 0) {
+        if (snprintf(logFileOption, length + 1, "--log-file=%s", logFile) < 0) {
             JLI_ReportErrorMessage("Failed to format option '--log-file=%s'.", logFile);
-            restoreStatus = -1;
-            goto done;
+            goto fail;
         }
         argv[argc++] = logFileOption;
     }
     argv[argc] = NULL;
     execvp(argv[0], (char * const *)argv);
     /* If execvp returns, there was an error. */
-    restoreStatus = -1;
-freeLogFileOption:
     if (NULL != logFileOption) {
         JLI_MemFree((void *)logFileOption);
-        logFileOption = NULL;
     }
-done:
-    return restoreStatus;
+fail:
+    return -1;
 }
 
 /**
@@ -252,12 +251,12 @@ handleCRaCRestore(int argc, char **argv)
     int childProcessPidStatus = 0;
     int childProcessPidExitStatus = 0;
     checkpointDirectory = getCheckpointDirectory(argc, argv, &error);
-    /* We ignore restore if the restore option -XX:CRaCRestoreFrom is not specified. */
     if (-1 == error) {
+        /* Option -XX:CRaCRestoreFrom not specified. */
         return;
     }
     if (-2 == error) {
-        JLI_ReportErrorMessage("Failed to get the CRIU checkpoint directory, error=%d.", error);
+        JLI_ReportErrorMessage("Failed to get the CRIU checkpoint directory, error=-2.");
         parentProcessExitStatus = EXIT_FAILURE;
         goto doneParentProcess;
     }

closed/src/java.base/share/native/libjli/criuhelpers.h

@@ -59,9 +59,9 @@
 
 #include <assert.h>
 
-#include "criuhelpers.h"
 #include "java.h"
 #include "jni.h"
+#include "j9cfg.h"
 
 /*
  * A NOTE TO DEVELOPERS: For performance reasons it is important that
@@ -567,12 +567,12 @@ JavaMain(void* _args)
     jfieldID noArgMainField;
     jboolean noArgMain;
 
-    RegisterThread();
-
 #if defined(J9VM_OPT_CRAC_SUPPORT)
     handleCRaCRestore(argc, argv);
 #endif /* defined(J9VM_OPT_CRAC_SUPPORT) */
 
+    RegisterThread();
+
     /* Initialize the virtual machine */
     start = CurrentTimeMicros();
     if (!InitializeJVM(&vm, &env, &ifn)) {