commands: Fix memory allocation in assign

Also use the safer snprintf since we just computed the size.

Signed-off-by: Quentin Glidic <sardemff7+git@sardemff7.net>

1 files changed, 3 insertions, 3 deletions

diff --git a/sway/commands.c b/sway/commands.c
index fc3444ee..9f6e5032 100644
--- a/sway/commands.c
+++ b/sway/commands.c
@@ -177,10 +177,10 @@ static struct cmd_results *cmd_assign(int argc, char **argv) {
 	}
 
 	char *movecmd = "move container to workspace ";
-	int arglen = strlen(*argv);
-	char *cmdlist = calloc(1, sizeof(movecmd) + arglen);
+	int arglen = strlen(movecmd) + strlen(*argv) + 1;
+	char *cmdlist = calloc(1, arglen);
 
-	sprintf(cmdlist, "%s%s", movecmd, *argv);
+	snprintf(cmdlist, arglen, "%s%s", movecmd, *argv);
 
 	struct criteria *crit = malloc(sizeof(struct criteria));
 	crit->crit_raw = strdup(criteria);