X-Git-Url: http://git.annexia.org/?p=libguestfs.git;a=blobdiff_plain;f=daemon%2Fupload.c;h=c026af8c5bdfdf8e8d37e841ffec7dccf63bf66c;hp=d2feb028394800e9f8f21eecc34c3bc6ae476332;hb=9ff99418361cd1b56b00e4ffef52444021e60c72;hpb=1765330e07a48dc6f7bdef7007f69ebe606fa731 diff --git a/daemon/upload.c b/daemon/upload.c index d2feb02..c026af8 100644 --- a/daemon/upload.c +++ b/daemon/upload.c @@ -1,5 +1,5 @@ /* libguestfs - the guestfsd daemon - * Copyright (C) 2009 Red Hat Inc. + * Copyright (C) 2009 Red Hat Inc. * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by @@ -20,21 +20,248 @@ #include #include +#include #include #include +#include +#include +#include -#include "../src/guestfs_protocol.h" +#include "guestfs_protocol.h" #include "daemon.h" #include "actions.h" +static int +write_cb (void *fd_ptr, const void *buf, size_t len) +{ + int fd = *(int *)fd_ptr; + return xwrite (fd, buf, len); +} + +/* Has one FileIn parameter. */ +static int +upload (const char *filename, int flags, int64_t offset) +{ + int err, fd, r, is_dev; + + is_dev = STRPREFIX (filename, "/dev/"); + + if (!is_dev) CHROOT_IN; + fd = open (filename, flags, 0666); + if (!is_dev) CHROOT_OUT; + if (fd == -1) { + err = errno; + r = cancel_receive (); + errno = err; + if (r != -2) reply_with_perror ("%s", filename); + return -1; + } + + if (offset) { + if (lseek (fd, offset, SEEK_SET) == -1) { + err = errno; + r = cancel_receive (); + errno = err; + if (r != -2) reply_with_perror ("lseek: %s", filename); + return -1; + } + } + + r = receive_file (write_cb, &fd); + if (r == -1) { /* write error */ + err = errno; + r = cancel_receive (); + errno = err; + if (r != -2) reply_with_error ("write error: %s", filename); + close (fd); + return -1; + } + if (r == -2) { /* cancellation from library */ + close (fd); + /* Do NOT send any error. */ + return -1; + } + + if (close (fd) == -1) { + err = errno; + if (r == -1) /* if r == 0, file transfer ended already */ + r = cancel_receive (); + errno = err; + if (r != -2) + reply_with_perror ("close: %s", filename); + return -1; + } + + return 0; +} + +/* Has one FileIn parameter. */ int -do_upload (const char *remote_filename) +do_upload (const char *filename) { - XXX_NOT_IMPL (-1); + return upload (filename, O_WRONLY|O_CREAT|O_TRUNC|O_NOCTTY, 0); } +/* Has one FileIn parameter. */ int -do_download (const char *remote_filename) +do_upload_offset (const char *filename, int64_t offset) { - XXX_NOT_IMPL (-1); + if (offset < 0) { + reply_with_perror ("%s: offset in file is negative", filename); + return -1; + } + + return upload (filename, O_WRONLY|O_CREAT|O_NOCTTY, offset); +} + +/* Has one FileOut parameter. */ +int +do_download (const char *filename) +{ + int fd, r, is_dev; + char buf[GUESTFS_MAX_CHUNK_SIZE]; + + is_dev = STRPREFIX (filename, "/dev/"); + + if (!is_dev) CHROOT_IN; + fd = open (filename, O_RDONLY); + if (!is_dev) CHROOT_OUT; + if (fd == -1) { + reply_with_perror ("%s", filename); + return -1; + } + + /* Calculate the size of the file or device for notification messages. */ + uint64_t total, sent = 0; + if (!is_dev) { + struct stat statbuf; + if (fstat (fd, &statbuf) == -1) { + reply_with_perror ("%s", filename); + close (fd); + return -1; + } + total = statbuf.st_size; + } else { + int64_t size = do_blockdev_getsize64 (filename); + if (size == -1) { + /* do_blockdev_getsize64 has already sent a reply. */ + close (fd); + return -1; + } + total = (uint64_t) size; + } + + /* Now we must send the reply message, before the file contents. After + * this there is no opportunity in the protocol to send any error + * message back. Instead we can only cancel the transfer. + */ + reply (NULL, NULL); + + while ((r = read (fd, buf, sizeof buf)) > 0) { + if (send_file_write (buf, r) < 0) { + close (fd); + return -1; + } + + sent += r; + notify_progress (sent, total); + } + + if (r == -1) { + perror (filename); + send_file_end (1); /* Cancel. */ + close (fd); + return -1; + } + + if (close (fd) == -1) { + perror (filename); + send_file_end (1); /* Cancel. */ + return -1; + } + + if (send_file_end (0)) /* Normal end of file. */ + return -1; + + return 0; +} + +/* Has one FileOut parameter. */ +int +do_download_offset (const char *filename, int64_t offset, int64_t size) +{ + int fd, r, is_dev; + char buf[GUESTFS_MAX_CHUNK_SIZE]; + + if (offset < 0) { + reply_with_perror ("%s: offset in file is negative", filename); + return -1; + } + + if (size < 0) { + reply_with_perror ("%s: size is negative", filename); + return -1; + } + uint64_t usize = (uint64_t) size; + + is_dev = STRPREFIX (filename, "/dev/"); + + if (!is_dev) CHROOT_IN; + fd = open (filename, O_RDONLY); + if (!is_dev) CHROOT_OUT; + if (fd == -1) { + reply_with_perror ("%s", filename); + return -1; + } + + if (offset) { + if (lseek (fd, offset, SEEK_SET) == -1) { + reply_with_perror ("lseek: %s", filename); + return -1; + } + } + + uint64_t total = usize, sent = 0; + + /* Now we must send the reply message, before the file contents. After + * this there is no opportunity in the protocol to send any error + * message back. Instead we can only cancel the transfer. + */ + reply (NULL, NULL); + + while (usize > 0) { + r = read (fd, buf, usize > sizeof buf ? sizeof buf : usize); + if (r == -1) { + perror (filename); + send_file_end (1); /* Cancel. */ + close (fd); + return -1; + } + + if (r == 0) + /* The documentation leaves this case undefined. Currently we + * just read fewer bytes than requested. + */ + break; + + if (send_file_write (buf, r) < 0) { + close (fd); + return -1; + } + + sent += r; + usize -= r; + notify_progress (sent, total); + } + + if (close (fd) == -1) { + perror (filename); + send_file_end (1); /* Cancel. */ + return -1; + } + + if (send_file_end (0)) /* Normal end of file. */ + return -1; + + return 0; }