Handle allocation failure in commands

1 files changed, 3 insertions, 0 deletions

diff --git a/sway/commands/assign.c b/sway/commands/assign.c
index 53c599ca..1824692b 100644
--- a/sway/commands/assign.c
+++ b/sway/commands/assign.c
@@ -27,6 +27,9 @@ struct cmd_results *cmd_assign(int argc, char **argv) {
         snprintf(cmdlist, arglen, "%s%s", movecmd, *argv);
 
         struct criteria *crit = malloc(sizeof(struct criteria));
+        if (!crit) {
+                return cmd_results_new(CMD_FAILURE, "assign", "Unable to allocate criteria");
+        }
         crit->crit_raw = strdup(criteria);
         crit->cmdlist = cmdlist;
         crit->tokens = create_list();