zero: Use $(NULL) in Makefile
[nbdkit/ericb.git] / tests / test-file-block.c
blob0c26f7b0ee63c74a7807e6c74d85b9e7f2cc0f8c
1 /* nbdkit
2 * Copyright (C) 2013-2018 Red Hat Inc.
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>
41 #include <sys/types.h>
43 #include <guestfs.h>
45 #include "test.h"
47 static char *loopdev; /* Name of the loop device. */
48 static void detach_loopdev (void);
50 int
51 main (int argc, char *argv[])
53 guestfs_h *g;
54 int r;
55 int fd;
56 char cmd[64], buf[64];
57 char disk[] = LARGE_TMPDIR "/diskXXXXXX"; /* Backing disk. */
58 FILE *pp;
59 char *data;
60 size_t len;
61 char *s;
63 /* This test can only be run as root, and will be skipped otherwise. */
64 if (geteuid () != 0) {
65 fprintf (stderr, "%s: this test has to be run as root.\n",
66 program_name);
67 exit (77);
70 /* losetup must be available. */
71 r = system ("losetup --version");
72 if (r != 0) {
73 fprintf (stderr, "%s: losetup program must be installed.\n",
74 program_name);
75 exit (77);
78 /* Create the temporary backing disk. */
79 fd = mkstemp (disk);
80 if (fd == -1) {
81 perror ("mkstemp");
82 exit (EXIT_FAILURE);
84 if (ftruncate (fd, 100 * 1024 * 1024) == -1) {
85 perror ("ftruncate");
86 unlink (disk);
87 exit (EXIT_FAILURE);
90 /* Create the loopback device. */
91 snprintf (cmd, sizeof cmd, "losetup -f --show %s", disk);
92 pp = popen (cmd, "r");
93 if (pp == NULL) {
94 perror ("popen: losetup");
95 unlink (disk);
96 exit (EXIT_FAILURE);
98 if (fgets (buf, sizeof buf, pp) == NULL) {
99 fprintf (stderr, "%s: could not read loop device name from losetup\n",
100 program_name);
101 unlink (disk);
102 exit (EXIT_FAILURE);
104 len = strlen (buf);
105 if (len > 0 && buf[len-1] == '\n') {
106 buf[len-1] = '\0';
107 len--;
109 pclose (pp);
111 /* We can delete the backing disk. The loop device will hold it open. */
112 unlink (disk);
114 /* If we get to this point, set up an atexit handler to detach the
115 * loop device.
117 loopdev = malloc (len+1);
118 if (loopdev == NULL) {
119 perror ("malloc");
120 exit (EXIT_FAILURE);
122 strcpy (loopdev, buf);
123 atexit (detach_loopdev);
125 /* Start nbdkit. */
126 if (test_start_nbdkit ("-D", "file.zero=1",
127 "file", loopdev, NULL) == -1)
128 exit (EXIT_FAILURE);
130 g = guestfs_create ();
131 if (g == NULL) {
132 perror ("guestfs_create");
133 exit (EXIT_FAILURE);
136 r = guestfs_add_drive_opts (g, "",
137 GUESTFS_ADD_DRIVE_OPTS_FORMAT, "raw",
138 GUESTFS_ADD_DRIVE_OPTS_PROTOCOL, "nbd",
139 GUESTFS_ADD_DRIVE_OPTS_SERVER, server,
140 -1);
141 if (r == -1)
142 exit (EXIT_FAILURE);
144 if (guestfs_launch (g) == -1)
145 exit (EXIT_FAILURE);
147 /* Partition the disk. */
148 if (guestfs_part_disk (g, "/dev/sda", "mbr") == -1)
149 exit (EXIT_FAILURE);
150 if (guestfs_mkfs (g, "ext4", "/dev/sda1") == -1)
151 exit (EXIT_FAILURE);
153 if (guestfs_mount_options (g, "discard", "/dev/sda1", "/") == -1)
154 exit (EXIT_FAILURE);
156 #define filename "/hello.txt"
157 #define content "hello, people of the world"
159 if (guestfs_write (g, filename, content, strlen (content)) == -1)
160 exit (EXIT_FAILURE);
162 data = guestfs_cat (g, filename);
163 if (!data)
164 exit (EXIT_FAILURE);
166 if (strcmp (data, content) != 0) {
167 fprintf (stderr,
168 "%s FAILED: unexpected content of %s file "
169 "(actual: %s, expected: %s)\n",
170 program_name, filename, data, content);
171 exit (EXIT_FAILURE);
174 /* Run sync to test flush path. */
175 if (guestfs_sync (g) == -1)
176 exit (EXIT_FAILURE);
178 /* Run fstrim to test trim path. However only recent versions of
179 * libguestfs have this, and it probably only works in recent
180 * versions of qemu.
182 #ifdef GUESTFS_HAVE_FSTRIM
183 if (guestfs_fstrim (g, "/", -1) == -1)
184 exit (EXIT_FAILURE);
185 #endif
187 /* Run fallocate(1) on the device to test zero path. */
188 if (guestfs_umount (g, "/") == -1)
189 exit (EXIT_FAILURE);
190 const char *sh[] = { "fallocate", "-nzl", "64k", "/dev/sda", NULL };
191 s = guestfs_debug (g, "sh", (char **) sh);
192 free (s);
194 if (guestfs_shutdown (g) == -1)
195 exit (EXIT_FAILURE);
197 guestfs_close (g);
199 /* The atexit handler should detach the loop device and clean up
200 * the backing disk.
202 exit (EXIT_SUCCESS);
205 /* atexit handler. */
206 static void
207 detach_loopdev (void)
209 char cmd[64];
211 if (loopdev == NULL)
212 return;
214 snprintf (cmd, sizeof cmd, "losetup -d %s", loopdev);
215 #pragma GCC diagnostic push
216 #pragma GCC diagnostic ignored "-Wunused-result"
217 system (cmd);
218 #pragma GCC diagnostic pop
219 free (loopdev);
220 loopdev = NULL;