feat(cmd): memory allocation errors to SET command

Author: aloima

docs/COMMANDS.md

@@ -823,11 +823,12 @@ RENAME name user_name
 **Since**: `0.1.0`  
 **Time complexity**: `O(1)`  
 **Permissions**: (`P_READ` if used `GET` argument) and `P_WRITE`  
-**Returns**: `OK` or a value or `ERROR`  
+**Returns**: `OK` or a value or an error message  
 **Behavior**:
 * If the key is exist, new value will be overwritten.
 * If value type is not specified using `AS type` argument, type will be determined by value itself. For example, `boolean` type for `true` value.
 * If value type is specified using `AS type` and value does not match the type, throws an error.
+* If a memory allocation is failed, throws an error.
 
 **Arguments**:
 - **NX**: Only set if the key does not exist.

src/commands/kv/set.c

@@ -164,7 +164,11 @@ static string_t run(struct CommandEntry *entry) {
         break;
 
       case TELLY_STR:
-        take_as_string(&value, entry->args->data[1]);
+        if (take_as_string(&value, entry->args->data[1]) == -1) {
+          PASS_NO_CLIENT(entry->client);
+          return RESP_ERROR_MESSAGE("Out of memory");
+        }
+
         break;
 
       default:
@@ -184,7 +188,11 @@ static string_t run(struct CommandEntry *entry) {
         break;
 
       case TELLY_STR:
-        take_as_string(&value, entry->args->data[1]);
+        if (take_as_string(&value, entry->args->data[1]) == -1) {
+          PASS_NO_CLIENT(entry->client);
+          return RESP_ERROR_MESSAGE("Out of memory");
+        }
+
         break;
 
       default:
@@ -198,12 +206,20 @@ static string_t run(struct CommandEntry *entry) {
       return RESP_ERROR_MESSAGE("The type must be string for this value");
     }
 
-    take_as_string(&value, entry->args->data[1]);
+    if (take_as_string(&value, entry->args->data[1]) == -1) {
+      PASS_NO_CLIENT(entry->client);
+      return RESP_ERROR_MESSAGE("Out of memory");
+    }
   }
 
   if (get) {
     if (entry->password->permissions & P_READ) {
-      set_data(entry->database, res, key, value, type);
+      const bool success = (set_data(entry->database, res, key, value, type) != NULL);
+
+      if (!success) {
+        PASS_NO_CLIENT(entry->client);
+        return RESP_ERROR_MESSAGE("Out of memory");
+      }
 
       if (res) {
         if (entry->client) {
@@ -224,7 +240,7 @@ static string_t run(struct CommandEntry *entry) {
     if (success) {
       return RESP_OK();
     } else {
-      return RESP_ERROR();
+      return RESP_ERROR_MESSAGE("Out of memory");
     }
   }
 }