diff options
| -rw-r--r-- | types/wlr_screenshooter.c | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/types/wlr_screenshooter.c b/types/wlr_screenshooter.c index f2d0937b..b16c1ba9 100644 --- a/types/wlr_screenshooter.c +++ b/types/wlr_screenshooter.c @@ -43,7 +43,7 @@ static void output_frame_notify(struct wl_listener *listener, void *_data) { orbital_screenshot_send_done(state->screenshot->resource); - // TODO: free(state) + free(state); } static void screenshooter_shoot(struct wl_client *client, @@ -69,8 +69,10 @@ static void screenshooter_shoot(struct wl_client *client, return; } - struct wlr_screenshot *screenshot; - if (!(screenshot = calloc(1, sizeof(struct wlr_screenshot)))) { + struct wlr_screenshot *screenshot = + calloc(1, sizeof(struct wlr_screenshot)); + if (!screenshot) { + wl_client_post_no_memory(client); return; } screenshot->output = _output; |