Home | History | Annotate | Line # | Download | only in mount_psshfs
psshfs.h revision 1.9
      1 /*	$NetBSD: psshfs.h,v 1.9 2007/04/12 20:42:46 pooka Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 2006  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  * 3. The name of the company nor the name of the author may be used to
     15  *    endorse or promote products derived from this software without specific
     16  *    prior written permission.
     17  *
     18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
     19  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     20  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     21  * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
     24  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     28  * SUCH DAMAGE.
     29  */
     30 
     31 #ifndef PSSHFS_H_
     32 #define PSSHFS_H_
     33 
     34 #include <sys/queue.h>
     35 
     36 #include <puffs.h>
     37 
     38 /*
     39  * Later protocol versions would have some advantages (such as link count
     40  * supported directly as a part of stat), but since proto version 3 seems
     41  * to be the only widely available version, let's not try to jump through
     42  * too many hoops to be compatible with all versions.
     43  */
     44 #define SFTP_PROTOVERSION 3
     45 
     46 /*
     47  * Refresh directories every n seconds as indicated by the following macro.
     48  * Note that local changes will still be visible immediately.
     49  */
     50 #define PSSHFS_REFRESHIVAL 30
     51 
     52 /* warm getattr cache in readdir */
     53 #define SUPERREADDIR
     54 
     55 PUFFSOP_PROTOS(psshfs);
     56 
     57 #define NEXTREQ(pctx) ((pctx->nextreq)++)
     58 #define PSSHFSAUTOVAR(pcc)						\
     59 	struct puffs_usermount *pu = puffs_cc_getusermount(pcc);	\
     60 	struct psshfs_ctx *pctx = puffs_getspecific(pu);		\
     61 	uint32_t reqid = NEXTREQ(pctx);					\
     62 	struct psbuf *pb = psbuf_make(PSB_OUT);				\
     63 	int rv = 0
     64 
     65 #define PSSHFSRETURN(rv)						\
     66 	psbuf_destroy(pb);						\
     67 	return (rv)
     68 
     69 struct psshfs_dir {
     70 	int valid;
     71 	struct puffs_node *entry;
     72 
     73 	char *entryname;
     74 	struct vattr va;
     75 	time_t attrread;
     76 };
     77 
     78 struct psshfs_fid {
     79 	time_t mounttime;
     80 	struct puffs_node *node;
     81 };
     82 
     83 struct psshfs_node {
     84 	struct puffs_node *parent;
     85 
     86 	struct psshfs_dir *dir;	/* only valid if we're of type VDIR */
     87 	size_t denttot;
     88 	size_t dentnext;
     89 	time_t dentread;
     90 	int childcount;
     91 	int reclaimed;
     92 	int hasfh;
     93 
     94 	time_t attrread;
     95 };
     96 
     97 struct psshfs_ctx;
     98 /*
     99  * XXX: urgh
    100  *
    101  * This is slightly messy / abusatory structure.  It is used for multiple
    102  * things.  Typical life cycle: create output buffer, append to output
    103  * queue (pcc included) .  Once the buffer has been sent, the buffer is
    104  * freed and the structure is appended to reqqueue as psreq.  It is kept
    105  * there until matching network data is read.  Once this happens, the
    106  * data from the received buffer is copied to buffer stored on the queue.
    107  * This should be rewritten, clearly.
    108  */
    109 #define PSDEFALLOC 0x2000
    110 #define PSBUFMAX 0x40000
    111 #define PSB_OUT 0
    112 #define PSB_IN 1
    113 struct psbuf {
    114 	struct psreq {
    115 		uint32_t reqid;
    116 
    117 		/* either ... */
    118 		struct puffs_cc *pcc;
    119 
    120 		/* ... or */
    121 		void (*func)(struct psshfs_ctx *, struct psbuf *, void *);
    122 		void *arg;
    123 		/* no union, we'd need a "which" flag */
    124 
    125 		TAILQ_ENTRY(psbuf) entries;
    126 	} psr;
    127 
    128 	/* in / out */
    129 	uint32_t len;
    130 	uint32_t remain;
    131 	uint32_t offset;
    132 	uint8_t *buf;
    133 
    134 	int state;
    135 
    136 	/* helpers for in */
    137 	uint8_t type;		/* buf[0] */
    138 	uint32_t reqid;		/* buf[1-4] */
    139 };
    140 #define PSBUF_PUT 0
    141 #define PSBUF_PUTDONE 1
    142 #define PSBUF_GETLEN 2
    143 #define PSBUF_GETDATA 3
    144 #define PSBUF_GETREADY 4
    145 
    146 struct psshfs_ctx {
    147 	int sshfd;
    148 	pid_t sshpid;
    149 	const char *mountpath;
    150 
    151 	int protover;
    152 	uint32_t nextreq;
    153 
    154 	struct psbuf *curpb;
    155 
    156 	struct psshfs_node psn_root;
    157 	ino_t nextino;
    158 
    159 	int canexport;
    160 	time_t mounttime;
    161 
    162 	TAILQ_HEAD(, psbuf) outbufq;
    163 	TAILQ_HEAD(, psbuf) req_queue;
    164 };
    165 
    166 int	psshfs_domount(struct puffs_usermount *);
    167 
    168 struct psbuf 	*psbuf_make(int);
    169 void		psbuf_destroy(struct psbuf *);
    170 void		psbuf_recycle(struct psbuf *, int);
    171 
    172 int		psbuf_read(struct psshfs_ctx *, struct psbuf *);
    173 int		psbuf_write(struct psshfs_ctx *, struct psbuf *);
    174 
    175 void		pssh_outbuf_enqueue(struct psshfs_ctx *, struct psbuf *,
    176 				    struct puffs_cc *, uint32_t);
    177 void		pssh_outbuf_enqueue_nocc(struct psshfs_ctx *, struct psbuf *,
    178 				    void (*f)(struct psshfs_ctx *,
    179 					      struct psbuf *, void *),
    180 				    void *, uint32_t);
    181 struct psbuf	*psshreq_get(struct psshfs_ctx *, uint32_t);
    182 
    183 
    184 int	psbuf_put_1(struct psbuf *, uint8_t);
    185 int	psbuf_put_2(struct psbuf *, uint16_t);
    186 int	psbuf_put_4(struct psbuf *, uint32_t);
    187 int	psbuf_put_8(struct psbuf *, uint64_t);
    188 int	psbuf_put_str(struct psbuf *, const char *);
    189 int	psbuf_put_data(struct psbuf *, const void *, uint32_t);
    190 int	psbuf_put_vattr(struct psbuf *, const struct vattr *);
    191 
    192 int	psbuf_get_1(struct psbuf *, uint8_t *);
    193 int	psbuf_get_2(struct psbuf *, uint16_t *);
    194 int	psbuf_get_4(struct psbuf *, uint32_t *);
    195 int	psbuf_get_8(struct psbuf *, uint64_t *);
    196 int	psbuf_get_str(struct psbuf *, char **, uint32_t *);
    197 int	psbuf_get_vattr(struct psbuf *, struct vattr *);
    198 
    199 int	psbuf_expect_status(struct psbuf *);
    200 int	psbuf_expect_handle(struct psbuf *, char **, uint32_t *);
    201 int	psbuf_expect_name(struct psbuf *, uint32_t *);
    202 int	psbuf_expect_attrs(struct psbuf *, struct vattr *);
    203 
    204 int	psbuf_do_data(struct psbuf *, uint8_t *, uint32_t *);
    205 
    206 int	psbuf_req_data(struct psbuf *, int, uint32_t, const void *, uint32_t);
    207 int	psbuf_req_str(struct psbuf *, int, uint32_t, const char *);
    208 
    209 
    210 int	sftp_readdir(struct puffs_cc *, struct psshfs_ctx *,
    211 		     struct puffs_node *);
    212 
    213 struct psshfs_dir *lookup(struct psshfs_dir *, size_t, const char *);
    214 struct puffs_node *makenode(struct puffs_usermount *, struct puffs_node *,
    215 			    struct psshfs_dir *, const struct vattr *);
    216 struct puffs_node *allocnode(struct puffs_usermount *, struct puffs_node *,
    217 			    const char *, const struct vattr *);
    218 struct psshfs_dir *direnter(struct puffs_node *, const char *);
    219 void nukenode(struct puffs_node *, const char *, int);
    220 void doreclaim(struct puffs_node *);
    221 int getpathattr(struct puffs_cc *, const char *, struct vattr *);
    222 int getnodeattr(struct puffs_cc *, struct puffs_node *);
    223 
    224 #endif /* PSSHFS_H_ */
    225