Home | History | Annotate | Line # | Download | only in mount_psshfs
      1 /*	$NetBSD: fs.c,v 1.25 2019/09/23 12:00:58 christos Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 2006-2009  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 #include <sys/cdefs.h>
     29 #ifndef lint
     30 __RCSID("$NetBSD: fs.c,v 1.25 2019/09/23 12:00:58 christos Exp $");
     31 #endif /* !lint */
     32 
     33 #include <err.h>
     34 #include <errno.h>
     35 #include <puffs.h>
     36 #include <signal.h>
     37 #include <stdio.h>
     38 #include <stdlib.h>
     39 #include <unistd.h>
     40 
     41 #include "psshfs.h"
     42 #include "sftp_proto.h"
     43 
     44 #define DO_IO(fname, a1, a2, a3, a4, rv)				\
     45 do {									\
     46 	puffs_framebuf_seekset(a2, 0);					\
     47 	*(a4) = 0;							\
     48 	rv = fname(a1, a2, a3, a4);					\
     49 	if (rv) {							\
     50 		return rv ? rv : EPROTO;				\
     51 	}								\
     52 } while (/*CONSTCOND*/0)
     53 
     54 #define reterr(str, rv)							\
     55 do {									\
     56 	fprintf str;							\
     57 	return rv;							\
     58 } while (/*CONSTCOND*/0)
     59 
     60 /* openssh extensions */
     61 static const struct extunit {
     62 	const char *ext;
     63 	const char *val;
     64 	int extflag;
     65 } exttable[] = {
     66 {
     67 	"posix-rename (at) openssh.com",
     68 	"1",
     69 	SFTP_EXT_POSIX_RENAME,
     70 },{
     71 	"statvfs (at) openssh.com",
     72 	"2",
     73 	SFTP_EXT_STATVFS,
     74 },{
     75 	"fstatvfs (at) openssh.com",
     76 	"2",
     77 	SFTP_EXT_FSTATVFS,
     78 },{
     79 	NULL,
     80 	NULL,
     81 	0
     82 }};
     83 
     84 int
     85 psshfs_handshake(struct puffs_usermount *pu, int fd)
     86 {
     87 	struct psshfs_ctx *pctx = puffs_getspecific(pu);
     88 	struct puffs_framebuf *pb;
     89 	struct puffs_pathobj *po_root;
     90 	struct puffs_node *pn_root;
     91 	struct vattr va, *rva;
     92 	const struct extunit *extu;
     93 	char *rootpath;
     94 	char *ext, *val;
     95 	uint32_t count;
     96 	int rv, done;
     97 
     98 	pb = psbuf_makeout();
     99 	psbuf_put_1(pb, SSH_FXP_INIT);
    100 	psbuf_put_4(pb, SFTP_PROTOVERSION);
    101 	DO_IO(psbuf_write, pu, pb, fd, &done, rv);
    102 
    103 	puffs_framebuf_recycle(pb);
    104 	DO_IO(psbuf_read, pu, pb, fd, &done, rv);
    105 	if (psbuf_get_type(pb) != SSH_FXP_VERSION)
    106 		reterr((stderr, "invalid server response: %d",
    107 		    psbuf_get_type(pb)), EPROTO);
    108 	pctx->protover = psbuf_get_reqid(pb);
    109 
    110 	/*
    111 	 * Check out which extensions are available.  Currently
    112 	 * we are only interested in the openssh statvfs extension.
    113 	 */
    114 	for (;;) {
    115 		if (psbuf_get_str(pb, &ext, NULL) != 0)
    116 			break;
    117 		if (psbuf_get_str(pb, &val, NULL) != 0)
    118 			break;
    119 
    120 		for (extu = exttable; extu->ext; extu++)
    121 			if (strcmp(ext, extu->ext) == 0
    122 			    && strcmp(val, extu->val) == 0)
    123 				pctx->extensions |= extu->extflag;
    124 	}
    125 
    126 	/* scope out our rootpath */
    127 	psbuf_recycleout(pb);
    128 	psbuf_put_1(pb, SSH_FXP_REALPATH);
    129 	psbuf_put_4(pb, NEXTREQ(pctx));
    130 	psbuf_put_str(pb, pctx->mountpath);
    131 	DO_IO(psbuf_write, pu, pb, fd, &done, rv);
    132 
    133 	puffs_framebuf_recycle(pb);
    134 	DO_IO(psbuf_read, pu, pb, fd, &done, rv);
    135 	if (psbuf_get_type(pb) != SSH_FXP_NAME)
    136 		reterr((stderr, "invalid server realpath response for \"%s\"",
    137 		    pctx->mountpath), EPROTO);
    138 	if (psbuf_get_4(pb, &count) == -1)
    139 		reterr((stderr, "invalid realpath response: count"), EPROTO);
    140 	if (psbuf_get_str(pb, &rootpath, NULL) == -1)
    141 		reterr((stderr, "invalid realpath response: rootpath"), EPROTO);
    142 
    143 	/* stat the rootdir so that we know it's a dir */
    144 	psbuf_recycleout(pb);
    145 	psbuf_req_str(pb, SSH_FXP_LSTAT, NEXTREQ(pctx), rootpath);
    146 	DO_IO(psbuf_write, pu, pb, fd, &done, rv);
    147 
    148 	puffs_framebuf_recycle(pb);
    149 	DO_IO(psbuf_read, pu, pb, fd, &done, rv);
    150 
    151 	rv = psbuf_expect_attrs(pb, &va);
    152 	if (rv)
    153 		reterr((stderr, "couldn't stat rootpath"), rv);
    154 	puffs_framebuf_destroy(pb);
    155 
    156 	if (puffs_mode2vt(va.va_mode) != VDIR)
    157 		reterr((stderr, "remote path (%s) not a directory", rootpath),
    158 		    ENOTDIR);
    159 
    160 	pn_root = puffs_getroot(pu);
    161 	rva = &pn_root->pn_va;
    162 	puffs_setvattr(rva, &va);
    163 
    164 	po_root = puffs_getrootpathobj(pu);
    165 	if (po_root == NULL)
    166 		err(1, "getrootpathobj");
    167 	po_root->po_path = rootpath;
    168 	po_root->po_len = strlen(rootpath);
    169 
    170 	return 0;
    171 }
    172 
    173 int
    174 psshfs_fs_statvfs(struct puffs_usermount *pu, struct puffs_statvfs *sbp)
    175 {
    176 	PSSHFSAUTOVAR(pu);
    177 	uint64_t tmpval;
    178 	uint8_t type;
    179 
    180 	memset(sbp, 0, sizeof(*sbp));
    181 	sbp->f_bsize = sbp->f_frsize = sbp->f_iosize = 512;
    182 
    183 	if ((pctx->extensions & SFTP_EXT_STATVFS) == 0)
    184 		goto out;
    185 
    186 	psbuf_req_str(pb, SSH_FXP_EXTENDED, reqid, "statvfs (at) openssh.com");
    187 	psbuf_put_str(pb, pctx->mountpath);
    188 	GETRESPONSE(pb, pctx->sshfd);
    189 
    190 	type = psbuf_get_type(pb);
    191 	if (type != SSH_FXP_EXTENDED_REPLY) {
    192 		/* use the default */
    193 		goto out;
    194 	}
    195 
    196 	psbuf_get_8(pb, &tmpval);
    197 	sbp->f_bsize = tmpval;
    198 	psbuf_get_8(pb, &tmpval);
    199 	sbp->f_frsize = tmpval;
    200 	psbuf_get_8(pb, &sbp->f_blocks);
    201 	psbuf_get_8(pb, &sbp->f_bfree);
    202 	psbuf_get_8(pb, &sbp->f_bavail);
    203 	psbuf_get_8(pb, &sbp->f_files);
    204 	psbuf_get_8(pb, &sbp->f_ffree);
    205 	psbuf_get_8(pb, &sbp->f_favail);
    206 
    207 	psbuf_get_8(pb, &tmpval); /* fsid */
    208 	psbuf_get_8(pb, &tmpval); /* flag */
    209 	psbuf_get_8(pb, &tmpval);
    210 	sbp->f_namemax = tmpval;
    211 
    212 	sbp->f_bresvd = sbp->f_bfree - sbp->f_bavail;
    213 	sbp->f_fresvd = sbp->f_ffree - sbp->f_favail;
    214 
    215  out:
    216 	PSSHFSRETURN(rv);
    217 }
    218 
    219 int
    220 psshfs_fs_unmount(struct puffs_usermount *pu, int flags)
    221 {
    222 	struct psshfs_ctx *pctx = puffs_getspecific(pu);
    223 
    224 	kill(pctx->sshpid, SIGTERM);
    225 	close(pctx->sshfd);
    226 	if (pctx->numconnections == 2) {
    227 		kill(pctx->sshpid_data, SIGTERM);
    228 		close(pctx->sshfd_data);
    229 	}
    230 
    231 	return 0;
    232 }
    233 
    234 int
    235 psshfs_fs_nodetofh(struct puffs_usermount *pu, puffs_cookie_t cookie,
    236 	void *fid, size_t *fidsize)
    237 {
    238 	struct psshfs_ctx *pctx = puffs_getspecific(pu);
    239 	struct puffs_node *pn = cookie;
    240 	struct psshfs_node *psn = pn->pn_data;
    241 	struct psshfs_fid *pf = fid;
    242 
    243 	pf->mounttime = pctx->mounttime;
    244 	pf->node = pn;
    245 
    246 	psn->stat |= PSN_HASFH;
    247 
    248 	return 0;
    249 }
    250 
    251 int
    252 psshfs_fs_fhtonode(struct puffs_usermount *pu, void *fid, size_t fidsize,
    253 	struct puffs_newinfo *pni)
    254 {
    255 	struct psshfs_ctx *pctx = puffs_getspecific(pu);
    256 	struct psshfs_fid *pf = fid;
    257 	struct puffs_node *pn = pf->node;
    258 	struct psshfs_node *psn;
    259 	int rv;
    260 
    261 	if (pf->mounttime != pctx->mounttime)
    262 		return EINVAL;
    263 	if (pn == 0)
    264 		return EINVAL;
    265 	psn = pn->pn_data;
    266 	if ((psn->stat & PSN_HASFH) == 0)
    267 		return EINVAL;
    268 
    269 	/* update node attributes */
    270 	rv = getnodeattr(pu, pn, NULL);
    271 	if (rv)
    272 		return EINVAL;
    273 
    274 	puffs_newinfo_setcookie(pni, pn);
    275 	puffs_newinfo_setvtype(pni, pn->pn_va.va_type);
    276 	puffs_newinfo_setsize(pni, pn->pn_va.va_size);
    277 
    278 	return 0;
    279 }
    280