tests/test-streaming.c

   1 /* nbdkit
   2  * Copyright (C) 2014 Red Hat Inc.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions are
   6  * met:
   7  *
   8  * * Redistributions of source code must retain the above copyright
   9  * notice, this list of conditions and the following disclaimer.
  10  *
  11  * * Redistributions in binary form must reproduce the above copyright
  12  * notice, this list of conditions and the following disclaimer in the
  13  * documentation and/or other materials provided with the distribution.
  14  *
  15  * * Neither the name of Red Hat nor the names of its contributors may be
  16  * used to endorse or promote products derived from this software without
  17  * specific prior written permission.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY RED HAT AND CONTRIBUTORS ''AS IS'' AND
  20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  21  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  22  * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RED HAT OR
  23  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  24  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  25  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  26  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  27  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  28  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  29  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  30  * SUCH DAMAGE.
  31  */
  32
  33 #include <config.h>
  34
  35 #include <stdio.h>
  36 #include <stdlib.h>
  37 #include <stdint.h>
  38 #include <inttypes.h>
  39 #include <string.h>
  40 #include <fcntl.h>
  41 #include <unistd.h>
  42 #include <sys/types.h>
  43 #include <sys/stat.h>
  44 #include <sys/wait.h>
  45
  46 #include <guestfs.h>
  47
  48 #include "test.h"
  49
  50 static char data[65536];
  51
  52 int
  53 main (int argc, char *argv[])
  54 {
  55   pid_t md5pid;
  56   guestfs_h *g;
  57   int r;
  58   size_t i;
  59   int pipefd[2];
  60   char md5[33];
  61   int status;
  62
  63   unlink ("streaming.fifo");
  64   if (mknod ("streaming.fifo", S_IFIFO | 0600, 0) == -1) {
  65     perror ("streaming.fifo");
  66     exit (EXIT_FAILURE);
  67   }
  68
  69   if (test_start_nbdkit ("streaming", "pipe=streaming.fifo", "size=640k",
  70                          NULL) == -1)
  71     exit (EXIT_FAILURE);
  72
  73   /* Fork to run a second process which reads from streaming.fifo
  74    * and checks that the content is correct.
  75    */
  76   if (pipe (pipefd) == -1) {
  77     perror ("pipe");
  78     exit (EXIT_FAILURE);
  79   }
  80
  81   md5pid = fork ();
  82   if (md5pid == -1) {
  83     perror ("fork");
  84     exit (EXIT_FAILURE);
  85   }
  86
  87   if (md5pid == 0) {
  88     /* Child: run md5sum on the pipe. */
  89     char *argv[] = { "md5sum", NULL };
  90
  91     close (0);
  92     open ("streaming.fifo", O_RDONLY);
  93     close (1);
  94     dup2 (pipefd[1], 1);
  95     close (pipefd[0]);
  96
  97     execvp ("md5sum", argv);
  98     perror ("md5sum");
  99     _exit (EXIT_FAILURE);
 100   }
 101
 102   close (pipefd[1]);
 103
 104   g = guestfs_create ();
 105   if (g == NULL) {
 106     perror ("guestfs_create");
 107     exit (EXIT_FAILURE);
 108   }
 109
 110   r = guestfs_add_drive_opts (g, "",
 111                               GUESTFS_ADD_DRIVE_OPTS_FORMAT, "raw",
 112                               GUESTFS_ADD_DRIVE_OPTS_PROTOCOL, "nbd",
 113                               GUESTFS_ADD_DRIVE_OPTS_SERVER, server,
 114                               -1);
 115   if (r == -1)
 116     exit (EXIT_FAILURE);
 117
 118   if (guestfs_launch (g) == -1)
 119     exit (EXIT_FAILURE);
 120
 121   /* Write linearly to the virtual disk. */
 122   memset (data, 1, sizeof data);
 123   for (i = 0; i < 10; ++i) {
 124     guestfs_pwrite_device (g, "/dev/sda", data, sizeof data,
 125                            i * sizeof data);
 126   }
 127
 128   if (guestfs_shutdown (g) == -1)
 129     exit (EXIT_FAILURE);
 130
 131   guestfs_close (g);
 132
 133   /* We have to explicitly kill nbdkit here so that it closes the
 134    * pipe.
 135    */
 136   kill (pid, SIGINT);
 137
 138   /* Check the hash computed by the child process. */
 139   if (read (pipefd[0], md5, 32) != 32) {
 140     perror ("read");
 141     exit (EXIT_FAILURE);
 142   }
 143   md5[32] = '\0';
 144
 145   if (strcmp (md5, "2c7f81c580f7f9eb52c1bd2f6f493b6f") != 0) {
 146     fprintf (stderr, "unexpected hash: %s\n", md5);
 147     exit (EXIT_FAILURE);
 148   }
 149
 150   if (waitpid (md5pid, &status, 0) == -1) {
 151     perror ("waitpid");
 152     exit (EXIT_FAILURE);
 153   }
 154   if (status != 0) {
 155     fprintf (stderr, "md5sum subprocess failed\n");
 156     exit (EXIT_FAILURE);
 157   }
 158
 159   exit (EXIT_SUCCESS);
 160 }