Bläddra i källkod

add argument bound check

Denys 5 år sedan
förälder
incheckning
d831b56bf4
1 ändrade filer med 10 tillägg och 3 borttagningar
  1. 10 3
      cli/src/cmdman.cpp

+ 10 - 3
cli/src/cmdman.cpp

@@ -260,10 +260,17 @@ CmdMan::CmdRet CmdMan::cmdDeleteme(vector<string> args) {
   CmdRet retval;
   DEBUGPRINT(string(__PRETTY_FUNCTION__) + " begin");
   Json::Value root;
-
+  
   root["command"] = "deleteme";
-  root["pass"] = args[0];
-  retval.type = send;
+  
+  if(args.size() < 1) {
+    retval.type = error;
+    root["accept"] = false;
+    root["error"] = "not enough arguments, at least 1 argument required";
+  } else {  
+    retval.type = send;
+    root["pass"] = args[0];
+  }
   retval.msg = root;
 
   return retval;