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