X-Git-Url: http://git.annexia.org/?a=blobdiff_plain;f=daemon%2Fupload.c;h=fdb86549b3b7277d12418269344bb45a31c913da;hb=582b64a7b18d846e03d5ed2d24036d566b964124;hp=5c1adaf65ba0b38593a81bc5068be93a68d40028;hpb=fc266e5ab8c4f013d0e473029c5e0b4cd32cca92;p=libguestfs.git diff --git a/daemon/upload.c b/daemon/upload.c index 5c1adaf..fdb8654 100644 --- a/daemon/upload.c +++ b/daemon/upload.c @@ -36,11 +36,11 @@ write_cb (void *fd_ptr, const void *buf, int len) /* Has one FileIn parameter. */ int -do_upload (char *filename) +do_upload (const char *filename) { int err, fd, r, is_dev; - is_dev = strncmp (filename, "/dev/", 5) == 0; + is_dev = STRPREFIX (filename, "/dev/"); if (!is_dev) { if (!root_mounted || filename[0] != '/') { cancel_receive (); @@ -88,14 +88,12 @@ do_upload (char *filename) /* Has one FileOut parameter. */ int -do_download (char *filename) +do_download (const char *filename) { int fd, r, is_dev; char buf[GUESTFS_MAX_CHUNK_SIZE]; - REQUIRE_ROOT_OR_RESOLVE_DEVICE (filename, return -1); - - is_dev = strncmp (filename, "/dev/", 5) == 0; + is_dev = STRPREFIX (filename, "/dev/"); if (!is_dev) CHROOT_IN; fd = open (filename, O_RDONLY); @@ -131,6 +129,8 @@ do_download (char *filename) return -1; } - send_file_end (0); /* Normal end of file. */ + if (send_file_end (0)) /* Normal end of file. */ + return -1; + return 0; }