/* libguestfs - the guestfsd daemon
- * Copyright (C) 2009 Red Hat Inc.
+ * Copyright (C) 2009-2011 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
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
+#include <fcntl.h>
+#include <sys/stat.h>
-#include "../src/guestfs_protocol.h"
+#include "guestfs_protocol.h"
#include "daemon.h"
#include "actions.h"
-char *
-do_checksum (const char *csumtype, const char *path)
+static const char *
+program_of_csum (const char *csumtype)
{
- const char *program;
- char *buf;
- char *out, *err;
- int r;
- int len;
-
if (STRCASEEQ (csumtype, "crc"))
- program = "cksum";
+ return "cksum";
else if (STRCASEEQ (csumtype, "md5"))
- program = "md5sum";
+ return "md5sum";
else if (STRCASEEQ (csumtype, "sha1"))
- program = "sha1sum";
+ return "sha1sum";
else if (STRCASEEQ (csumtype, "sha224"))
- program = "sha224sum";
+ return "sha224sum";
else if (STRCASEEQ (csumtype, "sha256"))
- program = "sha256sum";
+ return "sha256sum";
else if (STRCASEEQ (csumtype, "sha384"))
- program = "sha384sum";
+ return "sha384sum";
else if (STRCASEEQ (csumtype, "sha512"))
- program = "sha512sum";
+ return "sha512sum";
else {
reply_with_error ("unknown checksum type, expecting crc|md5|sha1|sha224|sha256|sha384|sha512");
return NULL;
}
+}
- /* Make the path relative to /sysroot. */
- buf = sysroot_path (path);
- if (!buf) {
- reply_with_perror ("malloc");
+static char *
+checksum (const char *csumtype, int fd)
+{
+ const char *program;
+ char *out, *err;
+ int flags, r;
+ int len;
+
+ program = program_of_csum (csumtype);
+ if (program == NULL) {
+ close (fd);
return NULL;
}
- r = command (&out, &err, program, buf, NULL);
- free (buf);
+ pulse_mode_start ();
+
+ flags = COMMAND_FLAG_CHROOT_COPY_FILE_TO_STDIN | fd;
+ r = commandf (&out, &err, flags, program, NULL);
if (r == -1) {
+ pulse_mode_cancel ();
reply_with_error ("%s: %s", program, err);
free (out);
free (err);
len = strcspn (out, " \t\n");
out[len] = '\0';
+ pulse_mode_end ();
+
return out; /* Caller frees. */
}
+
+char *
+do_checksum (const char *csumtype, const char *path)
+{
+ int fd;
+
+ CHROOT_IN;
+ fd = open (path, O_RDONLY);
+ CHROOT_OUT;
+
+ if (fd == -1) {
+ reply_with_perror ("%s", path);
+ return NULL;
+ }
+
+ return checksum (csumtype, fd);
+}
+
+char *
+do_checksum_device (const char *csumtype, const char *device)
+{
+ int fd;
+
+ fd = open (device, O_RDONLY);
+ if (fd == -1) {
+ reply_with_perror ("%s", device);
+ return NULL;
+ }
+
+ return checksum (csumtype, fd);
+}
+
+/* Has one FileOut parameter. */
+int
+do_checksums_out (const char *csumtype, const char *dir)
+{
+ struct stat statbuf;
+ int r;
+
+ const char *program = program_of_csum (csumtype);
+ if (program == NULL)
+ return -1;
+
+ char *sysrootdir = sysroot_path (dir);
+ if (!sysrootdir) {
+ reply_with_perror ("malloc");
+ return -1;
+ }
+
+ r = stat (sysrootdir, &statbuf);
+ if (r == -1) {
+ reply_with_perror ("%s", dir);
+ free (sysrootdir);
+ return -1;
+ }
+ if (!S_ISDIR (statbuf.st_mode)) {
+ reply_with_error ("%s: not a directory", dir);
+ free (sysrootdir);
+ return -1;
+ }
+
+ char *cmd;
+ if (asprintf_nowarn (&cmd, "cd %Q && find -type f -print0 | xargs -0 %s",
+ sysrootdir, program) == -1) {
+ reply_with_perror ("asprintf");
+ free (sysrootdir);
+ return -1;
+ }
+ free (sysrootdir);
+
+ if (verbose)
+ fprintf (stderr, "%s\n", cmd);
+
+ FILE *fp = popen (cmd, "r");
+ if (fp == NULL) {
+ reply_with_perror ("%s", cmd);
+ free (cmd);
+ return -1;
+ }
+ free (cmd);
+
+ /* 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);
+
+ char str[GUESTFS_MAX_CHUNK_SIZE];
+
+ while ((r = fread (str, 1, GUESTFS_MAX_CHUNK_SIZE, fp)) > 0) {
+ if (send_file_write (str, r) < 0) {
+ pclose (fp);
+ return -1;
+ }
+ }
+
+ if (ferror (fp)) {
+ perror (dir);
+ send_file_end (1); /* Cancel. */
+ pclose (fp);
+ return -1;
+ }
+
+ if (pclose (fp) != 0) {
+ perror (dir);
+ send_file_end (1); /* Cancel. */
+ return -1;
+ }
+
+ if (send_file_end (0)) /* Normal end of file. */
+ return -1;
+
+ return 0;
+}