ppm-save: pass opened fd

This will allow to pre-open the file before running the async finish
handler and avoid potential monitor fdset races.

(note: this is preliminary work for asynchronous screendump support)

Signed-off-by: Marc-André Lureau <marcandre.lureau@redhat.com>
Reviewed-by: Daniel P. Berrangé <berrange@redhat.com>
master
Marc-André Lureau 2019-11-08 15:23:15 +04:00
parent 4d6316218b
commit 46e5841cd2
2 changed files with 23 additions and 24 deletions

View File

@ -193,6 +193,7 @@ static void dpy_refresh(DisplayState *s);
static DisplayState *get_alloc_displaystate(void); static DisplayState *get_alloc_displaystate(void);
static void text_console_update_cursor_timer(void); static void text_console_update_cursor_timer(void);
static void text_console_update_cursor(void *opaque); static void text_console_update_cursor(void *opaque);
static bool ppm_save(int fd, DisplaySurface *ds, Error **errp);
static void gui_update(void *opaque) static void gui_update(void *opaque)
{ {
@ -308,29 +309,22 @@ void graphic_hw_invalidate(QemuConsole *con)
} }
} }
static void ppm_save(const char *filename, DisplaySurface *ds, static bool ppm_save(int fd, DisplaySurface *ds, Error **errp)
Error **errp)
{ {
int width = pixman_image_get_width(ds->image); int width = pixman_image_get_width(ds->image);
int height = pixman_image_get_height(ds->image); int height = pixman_image_get_height(ds->image);
int fd;
FILE *f; FILE *f;
int y; int y;
int ret; int ret;
pixman_image_t *linebuf; pixman_image_t *linebuf;
bool success = false;
trace_ppm_save(filename, ds); trace_ppm_save(fd, ds);
fd = qemu_open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666);
if (fd == -1) {
error_setg(errp, "failed to open file '%s': %s", filename,
strerror(errno));
return;
}
f = fdopen(fd, "wb"); f = fdopen(fd, "wb");
ret = fprintf(f, "P6\n%d %d\n%d\n", width, height, 255); ret = fprintf(f, "P6\n%d %d\n%d\n", width, height, 255);
if (ret < 0) { if (ret < 0) {
linebuf = NULL; linebuf = NULL;
goto write_err; goto end;
} }
linebuf = qemu_pixman_linebuf_create(PIXMAN_BE_r8g8b8, width); linebuf = qemu_pixman_linebuf_create(PIXMAN_BE_r8g8b8, width);
for (y = 0; y < height; y++) { for (y = 0; y < height; y++) {
@ -339,21 +333,16 @@ static void ppm_save(const char *filename, DisplaySurface *ds,
ret = fwrite(pixman_image_get_data(linebuf), 1, ret = fwrite(pixman_image_get_data(linebuf), 1,
pixman_image_get_stride(linebuf), f); pixman_image_get_stride(linebuf), f);
(void)ret; (void)ret;
if (ferror(f)) { success = !ferror(f);
goto write_err;
}
} }
out: end:
if (!success) {
error_setg(errp, "failed to write to PPM file: %s", strerror(errno));
}
qemu_pixman_image_unref(linebuf); qemu_pixman_image_unref(linebuf);
fclose(f); fclose(f);
return; return success;
write_err:
error_setg(errp, "failed to write to file '%s': %s", filename,
strerror(errno));
unlink(filename);
goto out;
} }
void qmp_screendump(const char *filename, bool has_device, const char *device, void qmp_screendump(const char *filename, bool has_device, const char *device,
@ -361,6 +350,7 @@ void qmp_screendump(const char *filename, bool has_device, const char *device,
{ {
QemuConsole *con; QemuConsole *con;
DisplaySurface *surface; DisplaySurface *surface;
int fd;
if (has_device) { if (has_device) {
con = qemu_console_lookup_by_device_name(device, has_head ? head : 0, con = qemu_console_lookup_by_device_name(device, has_head ? head : 0,
@ -387,7 +377,16 @@ void qmp_screendump(const char *filename, bool has_device, const char *device,
return; return;
} }
ppm_save(filename, surface, errp); fd = qemu_open(filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY, 0666);
if (fd == -1) {
error_setg(errp, "failed to open file '%s': %s", filename,
strerror(errno));
return;
}
if (!ppm_save(fd, surface, errp)) {
unlink(filename);
}
} }
void graphic_hw_text_update(QemuConsole *con, console_ch_t *chardata) void graphic_hw_text_update(QemuConsole *con, console_ch_t *chardata)

View File

@ -15,7 +15,7 @@ displaysurface_create_pixman(void *display_surface) "surface=%p"
displaysurface_free(void *display_surface) "surface=%p" displaysurface_free(void *display_surface) "surface=%p"
displaychangelistener_register(void *dcl, const char *name) "%p [ %s ]" displaychangelistener_register(void *dcl, const char *name) "%p [ %s ]"
displaychangelistener_unregister(void *dcl, const char *name) "%p [ %s ]" displaychangelistener_unregister(void *dcl, const char *name) "%p [ %s ]"
ppm_save(const char *filename, void *display_surface) "%s surface=%p" ppm_save(int fd, void *display_surface) "fd=%d surface=%p"
# gtk.c # gtk.c
# gtk-gl-area.c # gtk-gl-area.c