94 lines
3.0 KiB
Diff
94 lines
3.0 KiB
Diff
|
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);
|