X-Git-Url: http://git.annexia.org/?p=libguestfs.git;a=blobdiff_plain;f=daemon%2Fupload.c;h=41b1bf2dd5430af766ced4b723ff790dd86608b0;hp=d2feb028394800e9f8f21eecc34c3bc6ae476332;hb=aed0fa2c015e56a882fd6d4b759c82df08fc40d7;hpb=1765330e07a48dc6f7bdef7007f69ebe606fa731

diff --git a/daemon/upload.c b/daemon/upload.c
index d2feb02..41b1bf2 100644
--- a/daemon/upload.c
+++ b/daemon/upload.c
@@ -27,14 +27,110 @@
 #include "daemon.h"
 #include "actions.h"
 
+static int
+write_cb (void *fd_ptr, const void *buf, int len)
+{
+  int fd = *(int *)fd_ptr;
+  return xwrite (fd, buf, len);
+}
+
+/* Has one FileIn parameter. */
 int
-do_upload (const char *remote_filename)
+do_upload (const char *filename)
 {
-  XXX_NOT_IMPL (-1);
+  int err, fd, r, is_dev;
+
+  is_dev = strncmp (filename, "/dev/", 5) == 0;
+  if (!is_dev) {
+    if (!root_mounted || filename[0] != '/') {
+      cancel_receive ();
+      reply_with_error ("upload: root must be mounted and path must be absolute");
+      return -1;
+    }
+  }
+
+  if (!is_dev) CHROOT_IN;
+  fd = open (filename, O_WRONLY|O_CREAT|O_TRUNC|O_NOCTTY, 0666);
+  if (!is_dev) CHROOT_OUT;
+  if (fd == -1) {
+    err = errno;
+    cancel_receive ();
+    errno = err;
+    reply_with_perror ("%s", filename);
+    return -1;
+  }
+
+  r = receive_file (write_cb, &fd);
+  if (r == -1) {		/* write error */
+    err = errno;
+    cancel_receive ();
+    errno = err;
+    reply_with_perror ("write: %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;
+    cancel_receive ();
+    errno = err;
+    reply_with_perror ("close: %s", filename);
+    return -1;
+  }
+
+  return 0;
 }
 
+/* Has one FileOut parameter. */
 int
-do_download (const char *remote_filename)
+do_download (const char *filename)
 {
-  XXX_NOT_IMPL (-1);
+  int fd, r, is_dev;
+  char buf[GUESTFS_MAX_CHUNK_SIZE];
+
+  NEED_ROOT_OR_IS_DEVICE (filename, -1);
+
+  is_dev = strncmp (filename, "/dev/", 5) == 0;
+
+  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;
+  }
+
+  /* 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;
+    }
+  }
+
+  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;
+  }
+
+  send_file_end (0);		/* Normal end of file. */
+  return 0;
 }