Home | History | Annotate | Line # | Download | only in pgfs
mount.c revision 1.2
      1 /*	$NetBSD: mount.c,v 1.2 2012/04/11 14:25:54 yamt Exp $	*/
      2 
      3 /*-
      4  * Copyright (c)2010,2011 YAMAMOTO Takashi,
      5  * All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  *
     16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
     17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     19  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     20  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     21  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     22  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     23  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     24  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     25  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     26  * SUCH DAMAGE.
     27  */
     28 
     29 #include <sys/cdefs.h>
     30 #ifndef lint
     31 __RCSID("$NetBSD: mount.c,v 1.2 2012/04/11 14:25:54 yamt Exp $");
     32 #endif /* not lint */
     33 
     34 #include <err.h>
     35 #include <errno.h>
     36 #include <mntopts.h>
     37 #include <paths.h>
     38 #include <puffs.h>
     39 #include <stdbool.h>
     40 #include <stdlib.h>
     41 #include <unistd.h>
     42 
     43 #include "pgfs.h"
     44 #include "pgfs_db.h"
     45 
     46 #define	PGFS_MNT_ALT_DUMMY	1
     47 #define	PGFS_MNT_ALT_DEBUG	2
     48 
     49 int
     50 main(int argc, char *argv[])
     51 {
     52 	extern char *optarg;
     53 	extern int optind;
     54 	mntoptparse_t mp;
     55 	struct puffs_usermount *pu;
     56 	struct puffs_ops *pops;
     57 	int mntflags;
     58 	int altmntflags;
     59 	int ch;
     60 	int error;
     61 	const char *dbname = NULL;
     62 	const char *dbuser = NULL;
     63 	static const struct mntopt mopts[] = {
     64 		MOPT_STDOPTS,
     65 		MOPT_SYNC,
     66 		{ .m_option = "dbname", .m_inverse = 0,
     67 		  .m_flag = PGFS_MNT_ALT_DUMMY, .m_altloc = 1, },
     68 		{ .m_option = "dbuser", .m_inverse = 0,
     69 		  .m_flag = PGFS_MNT_ALT_DUMMY, .m_altloc = 1, },
     70 		{ .m_option = "debug", .m_inverse = 0,
     71 		  .m_flag = PGFS_MNT_ALT_DEBUG, .m_altloc = 1, },
     72 		{ .m_option = "nconn", .m_inverse = 0,
     73 		  .m_flag = PGFS_MNT_ALT_DUMMY, .m_altloc = 1, },
     74 		MOPT_NULL,
     75 	};
     76 	uint32_t pflags = PUFFS_KFLAG_IAONDEMAND;
     77 	unsigned int nconn = 8;
     78 	bool debug = false;
     79 	bool dosync;
     80 
     81 	mntflags = 0;
     82 	altmntflags = 0;
     83 	while ((ch = getopt(argc, argv, "o:")) != -1) {
     84 		long v;
     85 
     86 		switch (ch) {
     87 		case 'o':
     88 			mp = getmntopts(optarg, mopts, &mntflags,
     89 			    &altmntflags);
     90 			if (mp == NULL) {
     91 				err(EXIT_FAILURE, "getmntopts");
     92 			}
     93 			getmnt_silent = 1; /* XXX silly api */
     94 			dbname = getmntoptstr(mp, "dbname");
     95 			dbuser = getmntoptstr(mp, "dbuser");
     96 			v = getmntoptnum(mp, "nconn");
     97 			getmnt_silent = 0;
     98 			if (v != -1) {
     99 				nconn = v;
    100 			}
    101 			if ((altmntflags & PGFS_MNT_ALT_DEBUG) != 0) {
    102 				debug = true;
    103 			}
    104 			freemntopts(mp);
    105 			break;
    106 		}
    107 	}
    108 	argc -= optind;
    109 	argv += optind;
    110 
    111 	PUFFSOP_INIT(pops);
    112 	PUFFSOP_SETFSNOP(pops, unmount);
    113 	PUFFSOP_SETFSNOP(pops, sync);
    114 	PUFFSOP_SET(pops, pgfs, fs, statvfs);
    115 	PUFFSOP_SET(pops, pgfs, node, readdir);
    116 	PUFFSOP_SET(pops, pgfs, node, getattr);
    117 	PUFFSOP_SET(pops, pgfs, node, lookup);
    118 	PUFFSOP_SET(pops, pgfs, node, mkdir);
    119 	PUFFSOP_SET(pops, pgfs, node, create);
    120 	PUFFSOP_SET(pops, pgfs, node, read);
    121 	PUFFSOP_SET(pops, pgfs, node, write);
    122 	PUFFSOP_SET(pops, pgfs, node, link);
    123 	PUFFSOP_SET(pops, pgfs, node, remove);
    124 	PUFFSOP_SET(pops, pgfs, node, rmdir);
    125 	PUFFSOP_SET(pops, pgfs, node, inactive);
    126 	PUFFSOP_SET(pops, pgfs, node, setattr);
    127 	PUFFSOP_SET(pops, pgfs, node, rename);
    128 	PUFFSOP_SET(pops, pgfs, node, symlink);
    129 	PUFFSOP_SET(pops, pgfs, node, readlink);
    130 	PUFFSOP_SET(pops, pgfs, node, access);
    131 	dosync = (mntflags & MNT_SYNCHRONOUS) != 0;
    132 	if (!dosync) {
    133 		PUFFSOP_SET(pops, pgfs, node, fsync);
    134 	}
    135 	if (debug) {
    136 		pflags |= PUFFS_FLAG_OPDUMP;
    137 	}
    138 	pu = puffs_init(pops, _PATH_PUFFS, "pgfs", NULL, pflags);
    139 	if (pu == NULL) {
    140 		err(EXIT_FAILURE, "puffs_init");
    141 	}
    142 	error = pgfs_connectdb(pu, dbname, dbuser, debug, dosync, nconn);
    143 	if (error != 0) {
    144 		errno = error;
    145 		err(EXIT_FAILURE, "pgfs_connectdb");
    146 	}
    147 	if (!debug) {
    148 		if (puffs_daemon(pu, 1, 1)) {
    149 			err(EXIT_FAILURE, "puffs_daemon");
    150 		}
    151 	}
    152 	if (puffs_mount(pu, argv[1], mntflags, pgfs_root_cookie()) == -1) {
    153 		err(EXIT_FAILURE, "puffs_mount");
    154 	}
    155 	if (!debug) {
    156 		char tmpl[] = "/tmp/pgfs.XXXXXXXX";
    157 		const char *path;
    158 		int fd;
    159 		int ret;
    160 
    161 		path = mkdtemp(tmpl);
    162 		if (path == NULL) {
    163 			err(EXIT_FAILURE, "mkdtemp");
    164 		}
    165 		fd = open(path, O_RDONLY | O_DIRECTORY);
    166 		if (fd == -1) {
    167 			err(EXIT_FAILURE, "open %s", path);
    168 		}
    169 		ret = rmdir(path);
    170 		if (ret != 0) {
    171 			err(EXIT_FAILURE, "rmdir %s", path);
    172 		}
    173 		ret = fchroot(fd);
    174 		if (ret != 0) {
    175 			err(EXIT_FAILURE, "fchroot");
    176 		}
    177 		ret = close(fd);
    178 		if (ret != 0) {
    179 			err(EXIT_FAILURE, "close");
    180 		}
    181 	}
    182 	if (puffs_mainloop(pu) == -1) {
    183 		err(EXIT_FAILURE, "puffs_mainloop");
    184 	}
    185 	exit(EXIT_SUCCESS);
    186 }
    187