usr.sbin/puffs/mount_9p/ninepuffs.c

   1 /*      $NetBSD: ninepuffs.c,v 1.22 2007/11/16 18:39:01 pooka Exp $     */
   2
   3 /*
   4  * Copyright (c) 2007  Antti Kantee.  All Rights Reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  * 2. 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  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
  16  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  17  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18  * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  21  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  25  * SUCH DAMAGE.
  26  */
  27
  28 /*
  29  * 9puffs: access a 9P file server as a vfs via puffs
  30  */
  31
  32 #include <sys/cdefs.h>
  33 #ifndef lint
  34 __RCSID("$NetBSD: ninepuffs.c,v 1.22 2007/11/16 18:39:01 pooka Exp $");
  35 #endif /* !lint */
  36
  37 #include <sys/types.h>
  38 #include <sys/socket.h>
  39 #include <sys/poll.h>
  40
  41 #include <netinet/in.h>
  42
  43 #include <assert.h>
  44 #include <err.h>
  45 #include <netdb.h>
  46 #include <pwd.h>
  47 #include <puffs.h>
  48 #include <stdio.h>
  49 #include <stdlib.h>
  50 #include <unistd.h>
  51
  52 #include "ninepuffs.h"
  53
  54 #define DEFPORT_9P 564
  55
  56 static void
  57 usage(void)
  58 {
  59
  60         fprintf(stderr, "usage: %s [-s] [-o mntopts] [-p port] "
  61             "[user@]server[:path] mountpoint\n", getprogname());
  62         exit(1);
  63 }
  64
  65 /*
  66  * TCPv4 connection to 9P file server, forget IL and IPv6 for now.
  67  * Return connected socket or exit with error.
  68  */
  69 static int
  70 serverconnect(const char *addr, unsigned short port)
  71 {
  72         struct sockaddr_in mysin;
  73         struct hostent *he;
  74         int s;
  75
  76         he = gethostbyname2(addr, AF_INET);
  77         if (he == NULL) {
  78                 herror("gethostbyname");
  79                 exit(1);
  80         }
  81
  82         s = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
  83         if (s == -1)
  84                 err(1, "socket");
  85
  86         memset(&mysin, 0, sizeof(struct sockaddr_in));
  87         mysin.sin_family = AF_INET;
  88         mysin.sin_port = htons(port);
  89         memcpy(&mysin.sin_addr, he->h_addr, sizeof(struct in_addr));
  90
  91         if (connect(s, (struct sockaddr *)&mysin, sizeof(mysin)) == -1)
  92                 err(1, "connect");
  93
  94         return s;
  95 }
  96
  97 int
  98 main(int argc, char *argv[])
  99 {
 100         struct puffs9p p9p;
 101         struct puffs_usermount *pu;
 102         struct puffs_ops *pops;
 103         struct puffs_node *pn_root;
 104         mntoptparse_t mp;
 105         const char *user, *srvhost, *srvpath;
 106         char *p;
 107         unsigned short port;
 108         int mntflags, pflags, ch;
 109         int detach;
 110
 111         setprogname(argv[0]);
 112
 113         if (argc < 2)
 114                 usage();
 115
 116         mntflags = pflags = 0;
 117         detach = 1;
 118         port = DEFPORT_9P;
 119
 120         while ((ch = getopt(argc, argv, "o:p:s")) != -1) {
 121                 switch (ch) {
 122                 case 'o':
 123                         mp = getmntopts(optarg, puffsmopts, &mntflags, &pflags);
 124                         if (mp == NULL)
 125                                 err(1, "getmntopts");
 126                         freemntopts(mp);
 127                         break;
 128                 case 'p':
 129                         port = atoi(optarg);
 130                         break;
 131                 case 's':
 132                         detach = 0;
 133                         break;
 134                 default:
 135                         usage();
 136                         /*NOTREACHED*/
 137                 }
 138         }
 139         argc -= optind;
 140         argv += optind;
 141
 142         if (argc != 2)
 143                 usage();
 144
 145         if (pflags & PUFFS_FLAG_OPDUMP)
 146                 detach = 0;
 147         pflags |= PUFFS_KFLAG_WTCACHE | PUFFS_KFLAG_IAONDEMAND;
 148
 149         PUFFSOP_INIT(pops);
 150
 151         PUFFSOP_SET(pops, puffs9p, fs, unmount);
 152         PUFFSOP_SETFSNOP(pops, sync);
 153         PUFFSOP_SETFSNOP(pops, statvfs);
 154
 155         PUFFSOP_SET(pops, puffs9p, node, lookup);
 156         PUFFSOP_SET(pops, puffs9p, node, readdir);
 157         PUFFSOP_SET(pops, puffs9p, node, getattr);
 158         PUFFSOP_SET(pops, puffs9p, node, setattr);
 159         PUFFSOP_SET(pops, puffs9p, node, create);
 160         PUFFSOP_SET(pops, puffs9p, node, open);
 161         PUFFSOP_SET(pops, puffs9p, node, mkdir);
 162         PUFFSOP_SET(pops, puffs9p, node, remove);
 163         PUFFSOP_SET(pops, puffs9p, node, rmdir);
 164         PUFFSOP_SET(pops, puffs9p, node, read);
 165         PUFFSOP_SET(pops, puffs9p, node, write);
 166         PUFFSOP_SET(pops, puffs9p, node, rename);
 167         PUFFSOP_SET(pops, puffs9p, node, inactive);
 168         PUFFSOP_SET(pops, puffs9p, node, reclaim);
 169 #if 0
 170         PUFFSOP_SET(pops, puffs9p, node, mknod);
 171 #endif
 172
 173         pu = puffs_init(pops, argv[0], "9p", &p9p, pflags);
 174         if (pu == NULL)
 175                 err(1, "puffs_init");
 176
 177         memset(&p9p, 0, sizeof(p9p));
 178         p9p.maxreq = P9P_DEFREQLEN;
 179         p9p.nextfid = 1;
 180
 181         /* user@ */
 182         if ((p = strchr(argv[0], '@')) != NULL) {
 183                 *p = '\0';
 184                 srvhost = p+1;
 185                 user = argv[0];
 186         } else {
 187                 struct passwd *pw;
 188
 189                 srvhost = argv[0];
 190                 pw = getpwuid(getuid());
 191                 if (pw == NULL)
 192                         err(1, "getpwuid");
 193                 user = pw->pw_name;
 194         }
 195
 196         /* :/mountpath */
 197         if ((p = strchr(srvhost, ':')) != NULL) {
 198                 *p = '\0';
 199                 srvpath = p+1;
 200                 if (*srvpath != '/')
 201                         errx(1, "%s is not an absolute path", srvpath);
 202         } else {
 203                 srvpath = "/";
 204         }
 205
 206         p9p.servsock = serverconnect(srvhost, port);
 207         if ((pn_root = p9p_handshake(pu, user, srvpath)) == NULL) {
 208                 puffs_exit(pu, 1);
 209                 exit(1);
 210         }
 211
 212         puffs_framev_init(pu, p9pbuf_read, p9pbuf_write, p9pbuf_cmp, NULL,
 213             puffs_framev_unmountonclose);
 214         if (puffs_framev_addfd(pu, p9p.servsock,
 215             PUFFS_FBIO_READ | PUFFS_FBIO_WRITE) == -1)
 216                 err(1, "puffs_framebuf_addfd");
 217
 218         if (detach)
 219                 if (puffs_daemon(pu, 1, 1) == -1)
 220                         err(1, "puffs_daemon");
 221
 222         if (puffs_mount(pu, argv[1], mntflags, pn_root) == -1)
 223                 err(1, "puffs_mount");
 224         if (puffs_setblockingmode(pu, PUFFSDEV_NONBLOCK) == -1)
 225                 err(1, "setblockingmode");
 226
 227         if (puffs_mainloop(pu) == -1)
 228                 err(1, "mainloop");
 229
 230         return 0;
 231 }