ensure args count for incr/incrby

KDr2 · Jun 9, 2014 · c7e7426 · c7e7426
1 parent 1401e23
commit c7e7426
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/rl_kv.cpp b/src/rl_kv.cpp
@@ -31,6 +31,11 @@
 
 void RLRequest::rl_incr(){
 
+    if(args.size()!=1){
+        connection->write_error("ERR wrong number of arguments for 'incr' command");
+        return;
+    }
+
     std::string out;
     leveldb::Status status;
 
@@ -72,6 +77,11 @@ void RLRequest::rl_incr(){
 
 void RLRequest::rl_incrby(){
 
+    if(args.size()!=2){
+        connection->write_error("ERR wrong number of arguments for 'incrby' command");
+        return;
+    }
+
     std::string out;
     leveldb::Status status;
     status = connection->server->db[connection->db_index]->Get(connection->server->read_options,