diff -rup qemu-kvm-1.0.1/block/vvfat.c foo/block/vvfat.c --- qemu-kvm-1.0.1/block/vvfat.c 2012-04-16 22:15:17.000000000 -0400 +++ foo/block/vvfat.c 2012-07-29 20:00:15.515321504 -0400 @@ -2799,7 +2799,12 @@ static int enable_write_target(BDRVVVFAT array_init(&(s->commits), sizeof(commit_t)); s->qcow_filename = g_malloc(1024); - get_tmp_filename(s->qcow_filename, 1024); + ret = get_tmp_filename(s->qcow_filename, 1024); + if (ret < 0) { + g_free(s->qcow_filename); + s->qcow_filename = NULL; + return ret; + } bdrv_qcow = bdrv_find_format("qcow"); options = parse_option_parameters("", bdrv_qcow->create_options, NULL); diff -rup qemu-kvm-1.0.1/block.c foo/block.c --- qemu-kvm-1.0.1/block.c 2012-04-16 22:15:17.000000000 -0400 +++ foo/block.c 2012-07-29 20:00:15.513321760 -0400 @@ -272,28 +272,36 @@ int bdrv_create_file(const char* filenam return bdrv_create(drv, filename, options); } -#ifdef _WIN32 -void get_tmp_filename(char *filename, int size) +/* + * Create a uniquely-named empty temporary file. + * Return 0 upon success, otherwise a negative errno value. + */ +int get_tmp_filename(char *filename, int size) { +#ifdef _WIN32 char temp_dir[MAX_PATH]; - - GetTempPath(MAX_PATH, temp_dir); - GetTempFileName(temp_dir, "qem", 0, filename); -} + /* GetTempFileName requires that its output buffer (4th param) + have length MAX_PATH or greater. */ + assert(size >= MAX_PATH); + return (GetTempPath(MAX_PATH, temp_dir) + && GetTempFileName(temp_dir, "qem", 0, filename) + ? 0 : -GetLastError()); #else -void get_tmp_filename(char *filename, int size) -{ int fd; const char *tmpdir; - /* XXX: race condition possible */ tmpdir = getenv("TMPDIR"); if (!tmpdir) tmpdir = "/tmp"; - snprintf(filename, size, "%s/vl.XXXXXX", tmpdir); + if (snprintf(filename, size, "%s/vl.XXXXXX", tmpdir) >= size) { + return -EOVERFLOW; + } fd = mkstemp(filename); - close(fd); -} + if (fd < 0 || close(fd)) { + return -errno; + } + return 0; #endif +} /* * Detect host devices. By convention, /dev/cdrom[N] is always @@ -601,7 +609,10 @@ int bdrv_open(BlockDriverState *bs, cons bdrv_delete(bs1); - get_tmp_filename(tmp_filename, sizeof(tmp_filename)); + ret = get_tmp_filename(tmp_filename, sizeof(tmp_filename)); + if (ret < 0) { + return ret; + } /* Real path is meaningless for protocols */ if (is_protocol) diff -rup qemu-kvm-1.0.1/block_int.h foo/block_int.h --- qemu-kvm-1.0.1/block_int.h 2012-04-16 22:15:17.000000000 -0400 +++ foo/block_int.h 2012-07-29 20:00:15.515321504 -0400 @@ -238,7 +238,7 @@ struct BlockDriverAIOCB { BlockDriverAIOCB *next; }; -void get_tmp_filename(char *filename, int size); +int get_tmp_filename(char *filename, int size); void *qemu_aio_get(AIOPool *pool, BlockDriverState *bs, BlockDriverCompletionFunc *cb, void *opaque);