yutani: pex read() can return error code

klange · klange · commit abe66fb45ba6 · 2022-08-21T19:01:05.000+09:00
diff --git a/lib/yutani.c b/lib/yutani.c
@@ -120,10 +120,11 @@ yutani_msg_t * yutani_poll(yutani_t * y) {
 		return out;
 	}
 
-	size_t size;
+	ssize_t size;
 	{
 		char tmp[MAX_PACKET_SIZE];
 		size = pex_recv(y->sock, tmp);
+		if (size <= 0) return NULL;
 		out = malloc(size);
 		memcpy(out, tmp, size);
 	}

Original file line number	Diff line number	Diff line change
`@@ -120,10 +120,11 @@ yutani_msg_t * yutani_poll(yutani_t * y) {`
`120`	`120`	`return out;`
`121`	`121`	`}`
`122`	`122`
`123`		`- size_t size;`
	`123`	`+ ssize_t size;`
`124`	`124`	`{`
`125`	`125`	`char tmp[MAX_PACKET_SIZE];`
`126`	`126`	`size = pex_recv(y->sock, tmp);`
	`127`	`+ if (size <= 0) return NULL;`
`127`	`128`	`out = malloc(size);`
`128`	`129`	`memcpy(out, tmp, size);`
`129`	`130`	`}`