tests/test-pause.c

   1 /* nbdkit
   2  * Copyright Red Hat
   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 <stdbool.h>
  38 #include <stdint.h>
  39 #include <inttypes.h>
  40 #include <string.h>
  41 #include <unistd.h>
  42 #include <time.h>
  43 #undef NDEBUG /* Keep test strong even for nbdkit built without assertions */
  44 #include <assert.h>
  45 #include <sys/types.h>
  46 #include <sys/socket.h>
  47 #include <sys/un.h>
  48
  49 #include <libnbd.h>
  50
  51 #define SOCKET "pause.sock"
  52
  53 static bool command1_completed = false;
  54 static bool command2_completed = false;
  55
  56 static int
  57 callback (void *vp, int *err)
  58 {
  59   bool *flag = vp;
  60   *flag = true;
  61   return 1;
  62 }
  63
  64 int
  65 main (int argc, char *argv[])
  66 {
  67   struct nbd_handle *nbd;
  68   int ctrlsock;
  69   struct sockaddr_un addr;
  70   char buf[512];
  71   char c;
  72   int64_t cookie;
  73   time_t start_t, t;
  74
  75   nbd = nbd_create ();
  76   if (nbd == NULL) {
  77     fprintf (stderr, "%s\n", nbd_get_error ());
  78     exit (EXIT_FAILURE);
  79   }
  80
  81   if (nbd_connect_command (nbd,
  82                            (char *[]) {
  83                              "nbdkit", "-s", "--exit-with-parent",
  84                              "--filter", "pause",
  85                              "example1", "pause-control=" SOCKET,
  86                              NULL }) == -1) {
  87     fprintf (stderr, "%s\n", nbd_get_error ());
  88     exit (EXIT_FAILURE);
  89   }
  90
  91   /* Connect separately to the pause control socket. */
  92   ctrlsock = socket (AF_UNIX, SOCK_STREAM, 0);
  93   if (ctrlsock == -1) {
  94     perror ("socket");
  95     exit (EXIT_FAILURE);
  96   }
  97   addr.sun_family = AF_UNIX;
  98   strcpy (addr.sun_path, SOCKET);
  99
 100   if (connect (ctrlsock, (struct sockaddr *) &addr, sizeof addr) == -1) {
 101     perror (SOCKET);
 102     exit (EXIT_FAILURE);
 103   }
 104
 105   /* To start with, we should be able to read synchronously normally. */
 106   if (nbd_pread (nbd, buf, sizeof buf, 0, 0) == -1) {
 107     fprintf (stderr, "%s\n", nbd_get_error ());
 108     exit (EXIT_FAILURE);
 109   }
 110
 111   /* Pause the connection. */
 112   fprintf (stderr, "pausing the connection\n");
 113   c = 'p';
 114   if (write (ctrlsock, &c, 1) != 1) {
 115     perror ("write: ctrlsock: pause");
 116     exit (EXIT_FAILURE);
 117   }
 118   if (read (ctrlsock, &c, 1) != 1) {
 119     perror ("read: ctrlsock: response to pause");
 120     exit (EXIT_FAILURE);
 121   }
 122   assert (c == 'P');
 123
 124   /* Issue some asynchronous commands.  These should hang. */
 125   cookie = nbd_aio_pread (nbd, buf, sizeof buf, 0,
 126                           (nbd_completion_callback) {
 127                             .callback = callback,
 128                             .user_data = &command1_completed,
 129                           }, 0);
 130   if (cookie == -1) {
 131     fprintf (stderr, "%s\n", nbd_get_error ());
 132     exit (EXIT_FAILURE);
 133   }
 134   cookie = nbd_aio_pread (nbd, buf, sizeof buf, 0,
 135                           (nbd_completion_callback) {
 136                             .callback = callback,
 137                             .user_data = &command2_completed,
 138                           }, 0);
 139   if (cookie == -1) {
 140     fprintf (stderr, "%s\n", nbd_get_error ());
 141     exit (EXIT_FAILURE);
 142   }
 143
 144   /* Wait a bit to check they don't complete. */
 145   time (&start_t);
 146   while (time (&t) <= start_t + 5)
 147     nbd_poll (nbd, 1000);
 148   assert (!command1_completed);
 149   assert (!command2_completed);
 150
 151   /* Resume the connection. */
 152   fprintf (stderr, "resuming the connection\n");
 153   c = 'r';
 154   if (write (ctrlsock, &c, 1) != 1) {
 155     perror ("write: ctrlsock: resume");
 156     exit (EXIT_FAILURE);
 157   }
 158   if (read (ctrlsock, &c, 1) != 1) {
 159     perror ("read: ctrlsock: response to resume");
 160     exit (EXIT_FAILURE);
 161   }
 162   assert (c == 'R');
 163
 164   /* Now at least one of the commands should complete. */
 165   time (&start_t);
 166   while (!command1_completed && !command2_completed &&
 167          time (&t) <= start_t + 60)
 168     nbd_poll (nbd, 1000);
 169   assert (command1_completed || command2_completed);
 170
 171   close (ctrlsock);
 172   nbd_shutdown (nbd, 0);
 173   nbd_close (nbd);
 174   exit (EXIT_SUCCESS);
 175 }