tests/fs/common/fstest_zfs.c

   1 /*      $NetBSD: fstest_zfs.c,v 1.1 2012/08/20 16:37:35 pooka Exp $     */
   2
   3 /*-
   4  * Copyright (c) 2010, 2011  The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Nicolas Joly.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31
  32 #include <sys/mount.h>
  33 #include <sys/stat.h>
  34
  35 #include <atf-c.h>
  36 #include <stdio.h>
  37 #include <stdlib.h>
  38 #include <string.h>
  39 #include <unistd.h>
  40
  41 #include <rump/rump.h>
  42 #include <rump/rump_syscalls.h>
  43
  44 #include "h_fsmacros.h"
  45
  46 #define SRVPATH "zfssurvo"
  47 #define SRVURL "unix://" SRVPATH
  48 #define ZFSDEV "/zfsdev"
  49
  50 int
  51 zfs_fstest_newfs(const atf_tc_t *tc, void **buf, const char *image,
  52         off_t size, void *fspriv)
  53 {
  54         int res;
  55         int fd;
  56
  57         /* XXX: hardcoded zfs minimum size */
  58         size = MAX(64*1024*1024, size);
  59
  60         res = rump_init();
  61         if (res != 0) {
  62                 errno = res;
  63                 return -1;
  64         }
  65
  66         /* create backing image, sparse file is enough */
  67         if ((fd = open(image, O_RDWR | O_CREAT, 0777)) == -1)
  68                 return -1;
  69         if (ftruncate(fd, size) == -1) {
  70                 close(fd);
  71                 return -1;
  72         }
  73         close(fd);
  74
  75         res = rump_pub_etfs_register(ZFSDEV, image, RUMP_ETFS_BLK);
  76         if (res != 0) {
  77                 errno = res;
  78                 return -1;
  79         }
  80
  81         res = rump_init_server(SRVURL);
  82         if (res != 0) {
  83                 errno = res;
  84                 return -1;
  85         }
  86         *buf = NULL;
  87
  88         return 0;
  89 }
  90
  91 int
  92 zfs_fstest_delfs(const atf_tc_t *tc, void *buf)
  93 {
  94
  95         unlink(SRVPATH);
  96         return 0;
  97 }
  98
  99 int
 100 zfs_fstest_mount(const atf_tc_t *tc, void *buf, const char *path, int flags)
 101 {
 102         char tmpbuf[128];
 103         int error;
 104
 105         /* set up the hijack env for running zpool */
 106         setenv("RUMP_SERVER", SRVURL, 1);
 107         snprintf(tmpbuf, sizeof(tmpbuf)-1, "blanket=/dev/zfs:%s:%s",
 108             ZFSDEV, path);
 109         setenv("RUMPHIJACK", tmpbuf, 1);
 110         setenv("LD_PRELOAD", "/usr/lib/librumphijack.so", 1);
 111
 112         while (*path == '/')
 113                 path++;
 114
 115         /* run zpool create */
 116         snprintf(tmpbuf, sizeof(tmpbuf)-1, "zpool create %s %s",
 117             path, ZFSDEV);
 118         if ((error = system(tmpbuf)) != 0) {
 119                 errno = error;
 120                 return -1;
 121         }
 122
 123         return 0;
 124 }
 125
 126 int
 127 zfs_fstest_unmount(const atf_tc_t *tc, const char *path, int flags)
 128 {
 129
 130         unmount(path, flags);
 131         unlink(SRVPATH);
 132
 133         return 0;
 134 }