From ffbdd1aa20c8a20a8e9dcd3cec644b6dfa3c6acb Mon Sep 17 00:00:00 2001 From: Tom Schwindl Date: Sat, 10 Sep 2022 15:18:31 +0200 Subject: rio: check the return value of malloc(3) --- src/cmd/rio/main.c | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/cmd/rio/main.c') diff --git a/src/cmd/rio/main.c b/src/cmd/rio/main.c index 1e0b3797..09c1316e 100644 --- a/src/cmd/rio/main.c +++ b/src/cmd/rio/main.c @@ -213,6 +213,10 @@ main(int argc, char *argv[]) num_screens = ScreenCount(dpy); screens = (ScreenInfo *)malloc(sizeof(ScreenInfo) * num_screens); + if (!screens){ + fprintf(stderr, "rio: Failed to allocate memory\n"); + return 1; + } for(i = 0; i < num_screens; i++) initscreen(&screens[i], i, background); -- cgit v1.2.3