Fix code bug

OpenAtomFoundation · Aug 25, 2024 · b69ee36 · b69ee36
1 parent 025fdb6
commit b69ee36
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/pikiwidb.cc b/src/pikiwidb.cc
@@ -126,8 +126,18 @@ bool PikiwiDB::ParseArgs(int argc, char* argv[]) {
         break;
       }
       case 's': {
-        master_.resize(strlen(optarg));
-        sscanf(optarg, "%s:%d", master_, master_port_);
+        unsigned int optarg_long = strlen(optarg);
+        char *str = calloc(optarg_long, sizeof(char*));
+        if (str)
+        {
+          sscanf(optarg, "%s:%d", str, master_port_);
+          master_ = str;
+          free(str);
+        }
+        else
+        {
+          ERROR("Memory alloc failed.");
+        }
         break;
       }
       case '?': {