diff options
Diffstat (limited to 'src/cmd/rio/main.c')
| -rw-r--r-- | src/cmd/rio/main.c | 4 |
1 files changed, 4 insertions, 0 deletions
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); |