server/crypto.c: Improve error messages in crypto_send
[nbdkit.git] / tests / test-nbd.c
blobc0c008099616e4500fdbfd2ce491d2e3ca2f9e89
1 /* nbdkit
2 * Copyright Red Hat
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are
6 * met:
8 * * Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
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.
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.
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.
33 #include <config.h>
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <stdint.h>
38 #include <inttypes.h>
39 #include <string.h>
40 #include <unistd.h>
42 #include <guestfs.h>
44 #include "test.h"
46 int
47 main (int argc, char *argv[])
49 guestfs_h *g;
50 int r;
51 char *data;
52 char *sockarg = NULL;
53 size_t i, size;
55 /* Run an oldstyle nbdkit server. */
56 if (test_start_nbdkit ("-o", "file", "file-data", NULL) == -1)
57 exit (EXIT_FAILURE);
59 /* Now run a second (newstyle) server connecting to the oldstyle
60 * server above. Add shared=true on this handle to test that
61 * feature.
63 if (asprintf (&sockarg, "socket=%s", sock) < 0) {
64 perror ("asprintf");
65 exit (EXIT_FAILURE);
67 if (test_start_nbdkit ("-e", "wrap", "nbd", sockarg,
68 "shared=true", NULL) == -1)
69 exit (EXIT_FAILURE);
70 free (sockarg);
72 /* And now run another oldstyle server connecting to the newstyle
73 * server above, for a total of 3 nbdkit instances.
75 if (asprintf (&sockarg, "socket=%s", sock) < 0) {
76 perror ("asprintf");
77 exit (EXIT_FAILURE);
79 if (test_start_nbdkit ("-o", "nbd", sockarg, "export=wrap", NULL) == -1)
80 exit (EXIT_FAILURE);
81 free (sockarg);
83 /* Now connect to the front (oldstyle) server using libguestfs. */
84 g = guestfs_create ();
85 if (g == NULL) {
86 perror ("guestfs_create");
87 exit (EXIT_FAILURE);
90 r = guestfs_add_drive_opts (g, "",
91 GUESTFS_ADD_DRIVE_OPTS_FORMAT, "raw",
92 GUESTFS_ADD_DRIVE_OPTS_PROTOCOL, "nbd",
93 GUESTFS_ADD_DRIVE_OPTS_SERVER, server,
94 -1);
95 if (r == -1)
96 exit (EXIT_FAILURE);
98 if (guestfs_launch (g) == -1)
99 exit (EXIT_FAILURE);
101 /* Check the data in the file is \x01-\x08 repeated 512 times. */
102 data = guestfs_pread_device (g, "/dev/sda", 8 * 512, 0, &size);
103 if (!data)
104 exit (EXIT_FAILURE);
105 if (size != 8 * 512) {
106 fprintf (stderr,
107 "%s FAILED: unexpected size (actual: %zu, expected: 512)\n",
108 program_name, size);
109 exit (EXIT_FAILURE);
112 for (i = 0; i < 512 * 8; i += 8) {
113 if (data[i] != 1 || data[i+1] != 2 ||
114 data[i+2] != 3 || data[i+3] != 4 ||
115 data[i+4] != 5 || data[i+5] != 6 ||
116 data[i+6] != 7 || data[i+7] != 8) {
117 fprintf (stderr, "%s FAILED: unexpected data returned at offset %zu\n",
118 program_name, i);
119 exit (EXIT_FAILURE);
123 free (data);
125 guestfs_close (g);
126 exit (EXIT_SUCCESS);