tests/test-offset.c

   1 /* nbdkit
   2  * Copyright (C) 2018 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 <unistd.h>
  41 #include <fcntl.h>
  42 #include <errno.h>
  43 #include <sys/types.h>
  44 #include <sys/stat.h>
  45
  46 #include <guestfs.h>
  47
  48 #include "test.h"
  49
  50 static uint8_t buf[1024*1024];
  51
  52 static void
  53 check_buf (void)
  54 {
  55   size_t i;
  56
  57   for (i = 0; i < sizeof buf; i += 2) {
  58     if (buf[i] != 0x55 || buf[i+1] != 0xAA) {
  59       fprintf (stderr, "%s FAILED: file overwritten outside offset/range\n",
  60                program_name);
  61       exit (EXIT_FAILURE);
  62     }
  63   }
  64 }
  65
  66 int
  67 main (int argc, char *argv[])
  68 {
  69   guestfs_h *g;
  70   int r, fd;
  71   char *data;
  72
  73   /* offset-data (created by tests/Makefile.am) is a 10 MB file
  74    * containing test pattern data 0x55AA repeated.  We use the middle
  75    * 8 MB to create a partition table and filesystem, and check
  76    * afterwards that the test pattern in the first and final megabyte
  77    * has not been overwritten.
  78    */
  79   if (test_start_nbdkit ("--filter", "offset",
  80                          "file", "offset-data",
  81                          "offset=1M", "range=8M",
  82                          NULL) == -1)
  83     exit (EXIT_FAILURE);
  84
  85   g = guestfs_create ();
  86   if (g == NULL) {
  87     perror ("guestfs_create");
  88     exit (EXIT_FAILURE);
  89   }
  90
  91   r = guestfs_add_drive_opts (g, "",
  92                               GUESTFS_ADD_DRIVE_OPTS_FORMAT, "raw",
  93                               GUESTFS_ADD_DRIVE_OPTS_PROTOCOL, "nbd",
  94                               GUESTFS_ADD_DRIVE_OPTS_SERVER, server,
  95                               -1);
  96   if (r == -1)
  97     exit (EXIT_FAILURE);
  98
  99   if (guestfs_launch (g) == -1)
 100     exit (EXIT_FAILURE);
 101
 102   if (guestfs_part_disk (g, "/dev/sda", "mbr") == -1)
 103     exit (EXIT_FAILURE);
 104   if (guestfs_mkfs (g, "ext2", "/dev/sda1") == -1)
 105     exit (EXIT_FAILURE);
 106
 107   if (guestfs_mount (g, "/dev/sda1", "/") == -1)
 108     exit (EXIT_FAILURE);
 109
 110 #define filename "/hello.txt"
 111 #define content "hello, people of the world"
 112
 113   if (guestfs_write (g, filename, content, strlen (content)) == -1)
 114     exit (EXIT_FAILURE);
 115
 116   data = guestfs_cat (g, filename);
 117   if (!data)
 118     exit (EXIT_FAILURE);
 119
 120   if (strcmp (data, content) != 0) {
 121     fprintf (stderr,
 122              "%s FAILED: unexpected content of %s file "
 123              "(actual: %s, expected: %s)\n",
 124              program_name, filename, data, content);
 125     exit (EXIT_FAILURE);
 126   }
 127
 128   if (guestfs_fill_dir (g, "/", 1000) == -1)
 129     exit (EXIT_FAILURE);
 130
 131   if (guestfs_shutdown (g) == -1)
 132     exit (EXIT_FAILURE);
 133
 134   guestfs_close (g);
 135
 136   /* Check the first and final megabyte of test patterns has not been
 137    * overwritten in the underlying file.
 138    */
 139   fd = open ("offset-data", O_RDONLY|O_CLOEXEC);
 140   if (fd == -1) {
 141     perror ("open: offset-data");
 142     exit (EXIT_FAILURE);
 143   }
 144   if (pread (fd, buf, sizeof buf, 0) != sizeof buf) {
 145     fprintf (stderr, "pread: short read or error (%d)\n", errno);
 146     exit (EXIT_FAILURE);
 147   }
 148   check_buf ();
 149   if (pread (fd, buf, sizeof buf, 9*1024*1024) != sizeof buf) {
 150     fprintf (stderr, "pread: short read or error (%d)\n", errno);
 151     exit (EXIT_FAILURE);
 152   }
 153   check_buf ();
 154   close (fd);
 155
 156   exit (EXIT_SUCCESS);
 157 }