sway

commit d10e72318315fa8ad0772f685f24cf2c9e2b5cc2
parent 66d3777c1443207c6590fce63def93cb102a05a8
Author: Drew DeVault <sir@cmpwn.com>
Date:   Sun, 22 Oct 2017 19:50:43 -0400

Merge pull request #1424 from ggreer/swaygrab-json

swaygrab: Prevent segfault if IPC response can't be parsed.
Diffstat:
M
swaygrab/json.c
 | 
11
++++++++++-

1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/swaygrab/json.c b/swaygrab/json.c
@@ -4,6 +4,7 @@
 #include <stdbool.h>
 #include <stdlib.h>
 #include <unistd.h>
+#include "log.h"
 #include "ipc-client.h"
 #include "swaygrab/json.h"
 
@@ -12,7 +13,15 @@ static json_object *tree;
 void init_json_tree(int socketfd) {
 	uint32_t len = 0;
 	char *res = ipc_single_command(socketfd, IPC_GET_TREE, NULL, &len);
-	tree = json_tokener_parse(res);
+	struct json_tokener *tok = json_tokener_new_ex(256);
+	if (!tok) {
+		sway_abort("Unable to get json tokener.");
+	}
+	tree = json_tokener_parse_ex(tok, res, len);
+	if (!tree || tok->err != json_tokener_success) {
+		sway_abort("Unable to parse IPC response as JSON: %s", json_tokener_error_desc(tok->err));
+	}
+	json_tokener_free(tok);
 }
 
 void free_json_tree() {