subr.c revision 1.13 1 1.13 pooka /* $NetBSD: subr.c,v 1.13 2007/04/12 15:09:02 pooka Exp $ */
2 1.1 pooka
3 1.1 pooka /*
4 1.1 pooka * Copyright (c) 2006 Antti Kantee. All Rights Reserved.
5 1.1 pooka *
6 1.1 pooka * Redistribution and use in source and binary forms, with or without
7 1.1 pooka * modification, are permitted provided that the following conditions
8 1.1 pooka * are met:
9 1.1 pooka * 1. Redistributions of source code must retain the above copyright
10 1.1 pooka * notice, this list of conditions and the following disclaimer.
11 1.1 pooka * 2. Redistributions in binary form must reproduce the above copyright
12 1.1 pooka * notice, this list of conditions and the following disclaimer in the
13 1.1 pooka * documentation and/or other materials provided with the distribution.
14 1.1 pooka * 3. The name of the company nor the name of the author may be used to
15 1.1 pooka * endorse or promote products derived from this software without specific
16 1.1 pooka * prior written permission.
17 1.1 pooka *
18 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
19 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
21 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
24 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28 1.1 pooka * SUCH DAMAGE.
29 1.1 pooka */
30 1.1 pooka
31 1.1 pooka #include <sys/cdefs.h>
32 1.1 pooka #ifndef lint
33 1.13 pooka __RCSID("$NetBSD: subr.c,v 1.13 2007/04/12 15:09:02 pooka Exp $");
34 1.1 pooka #endif /* !lint */
35 1.1 pooka
36 1.1 pooka #include <assert.h>
37 1.1 pooka #include <errno.h>
38 1.1 pooka #include <puffs.h>
39 1.1 pooka #include <stdlib.h>
40 1.1 pooka #include <util.h>
41 1.1 pooka
42 1.1 pooka #include "psshfs.h"
43 1.1 pooka #include "sftp_proto.h"
44 1.1 pooka
45 1.1 pooka static void
46 1.1 pooka freedircache(struct psshfs_dir *base, size_t count)
47 1.1 pooka {
48 1.1 pooka int i;
49 1.1 pooka
50 1.1 pooka for (i = 0; i < count; i++) {
51 1.1 pooka free(base[i].entryname);
52 1.1 pooka base[i].entryname = NULL;
53 1.1 pooka }
54 1.1 pooka
55 1.1 pooka free(base);
56 1.1 pooka }
57 1.1 pooka
58 1.1 pooka #define ENTRYCHUNK 16
59 1.1 pooka static void
60 1.1 pooka allocdirs(struct psshfs_node *psn)
61 1.1 pooka {
62 1.1 pooka size_t oldtot = psn->denttot;
63 1.1 pooka
64 1.1 pooka psn->denttot += ENTRYCHUNK;
65 1.1 pooka psn->dir = erealloc(psn->dir,
66 1.1 pooka psn->denttot * sizeof(struct psshfs_dir));
67 1.1 pooka memset(psn->dir + oldtot, 0, ENTRYCHUNK * sizeof(struct psshfs_dir));
68 1.1 pooka }
69 1.1 pooka
70 1.1 pooka struct psshfs_dir *
71 1.9 pooka lookup(struct psshfs_dir *bdir, size_t ndir, const char *name)
72 1.1 pooka {
73 1.9 pooka struct psshfs_dir *test;
74 1.1 pooka int i;
75 1.1 pooka
76 1.1 pooka for (i = 0; i < ndir; i++) {
77 1.9 pooka test = &bdir[i];
78 1.9 pooka if (test->valid != 1)
79 1.1 pooka continue;
80 1.9 pooka if (strcmp(test->entryname, name) == 0)
81 1.9 pooka return test;
82 1.9 pooka }
83 1.9 pooka
84 1.9 pooka return NULL;
85 1.9 pooka }
86 1.9 pooka
87 1.9 pooka static struct psshfs_dir *
88 1.9 pooka lookup_by_entry(struct psshfs_dir *bdir, size_t ndir, struct puffs_node *entry)
89 1.9 pooka {
90 1.9 pooka struct psshfs_dir *test;
91 1.9 pooka int i;
92 1.9 pooka
93 1.9 pooka for (i = 0; i < ndir; i++) {
94 1.9 pooka test = &bdir[i];
95 1.9 pooka if (test->valid != 1)
96 1.9 pooka continue;
97 1.9 pooka if (test->entry == entry)
98 1.9 pooka return test;
99 1.1 pooka }
100 1.1 pooka
101 1.1 pooka return NULL;
102 1.1 pooka }
103 1.1 pooka
104 1.8 pooka #ifdef SUPERREADDIR
105 1.8 pooka struct readdirattr {
106 1.8 pooka struct psshfs_node *psn;
107 1.8 pooka int idx;
108 1.8 pooka char entryname[MAXPATHLEN+1];
109 1.8 pooka };
110 1.8 pooka
111 1.8 pooka static void
112 1.8 pooka readdir_getattr_resp(struct psshfs_ctx *pctx, struct psbuf *pb, void *arg)
113 1.8 pooka {
114 1.8 pooka struct readdirattr *rda = arg;
115 1.8 pooka struct psshfs_node *psn = rda->psn;
116 1.8 pooka struct psshfs_dir *pdir;
117 1.8 pooka struct vattr va;
118 1.8 pooka
119 1.8 pooka pdir = lookup(psn->dir, psn->denttot, rda->entryname);
120 1.8 pooka if (!pdir)
121 1.8 pooka goto out;
122 1.8 pooka
123 1.8 pooka if (psbuf_expect_attrs(pb, &va))
124 1.8 pooka goto out;
125 1.8 pooka
126 1.8 pooka if (pdir->entry) {
127 1.8 pooka struct psshfs_node *psn_targ;
128 1.8 pooka psn_targ = pdir->entry->pn_data;
129 1.8 pooka
130 1.8 pooka puffs_setvattr(&pdir->entry->pn_va, &va);
131 1.8 pooka psn_targ->attrread = time(NULL);
132 1.8 pooka } else {
133 1.8 pooka puffs_setvattr(&pdir->va, &va);
134 1.8 pooka pdir->attrread = time(NULL);
135 1.8 pooka }
136 1.8 pooka
137 1.8 pooka out:
138 1.8 pooka free(rda);
139 1.8 pooka psbuf_destroy(pb);
140 1.8 pooka }
141 1.8 pooka
142 1.8 pooka static void
143 1.8 pooka readdir_getattr(struct psshfs_ctx *pctx, struct psshfs_node *psn,
144 1.8 pooka const char *basepath, int idx)
145 1.8 pooka {
146 1.8 pooka char path[MAXPATHLEN+1];
147 1.8 pooka struct psshfs_dir *pdir = psn->dir;
148 1.8 pooka struct psbuf *pb;
149 1.8 pooka struct readdirattr *rda;
150 1.8 pooka const char *entryname = pdir[idx].entryname;
151 1.8 pooka uint32_t reqid = NEXTREQ(pctx);
152 1.8 pooka
153 1.8 pooka rda = emalloc(sizeof(struct readdirattr));
154 1.8 pooka rda->psn = psn;
155 1.8 pooka rda->idx = idx;
156 1.8 pooka strlcpy(rda->entryname, entryname, sizeof(rda->entryname));
157 1.8 pooka
158 1.8 pooka strcpy(path, basepath);
159 1.8 pooka strcat(path, "/");
160 1.8 pooka strlcat(path, entryname, sizeof(path));
161 1.8 pooka
162 1.8 pooka pb = psbuf_make(PSB_OUT);
163 1.8 pooka psbuf_req_str(pb, SSH_FXP_LSTAT, reqid, path);
164 1.8 pooka pssh_outbuf_enqueue_nocc(pctx, pb, readdir_getattr_resp, rda, reqid);
165 1.8 pooka }
166 1.8 pooka #endif
167 1.8 pooka
168 1.1 pooka int
169 1.12 pooka getpathattr(struct puffs_cc *pcc, const char *path, struct vattr *vap)
170 1.12 pooka {
171 1.12 pooka PSSHFSAUTOVAR(pcc);
172 1.12 pooka
173 1.12 pooka psbuf_req_str(pb, SSH_FXP_LSTAT, reqid, path);
174 1.12 pooka pssh_outbuf_enqueue(pctx, pb, pcc, reqid);
175 1.12 pooka puffs_cc_yield(pcc);
176 1.12 pooka
177 1.12 pooka rv = psbuf_expect_attrs(pb, vap);
178 1.12 pooka
179 1.12 pooka PSSHFSRETURN(rv);
180 1.12 pooka }
181 1.12 pooka
182 1.12 pooka int
183 1.12 pooka getnodeattr(struct puffs_cc *pcc, struct puffs_node *pn)
184 1.12 pooka {
185 1.12 pooka struct puffs_usermount *pu = puffs_cc_getusermount(pcc);
186 1.12 pooka struct psshfs_node *psn = pn->pn_data;
187 1.12 pooka struct vattr va;
188 1.12 pooka int rv;
189 1.12 pooka
190 1.12 pooka if ((time(NULL) - psn->attrread) >= PSSHFS_REFRESHIVAL) {
191 1.12 pooka rv = getpathattr(pcc, PNPATH(pn), &va);
192 1.12 pooka if (rv)
193 1.12 pooka return rv;
194 1.12 pooka
195 1.12 pooka /*
196 1.12 pooka * Check if the file was modified from below us. If
197 1.12 pooka * so, invalidate page cache. This is the only sensible
198 1.12 pooka * place we can do this in.
199 1.12 pooka */
200 1.12 pooka if (psn->attrread)
201 1.12 pooka if (pn->pn_va.va_mtime.tv_sec != va.va_mtime.tv_sec)
202 1.12 pooka puffs_inval_pagecache_node(pu, pn);
203 1.12 pooka
204 1.12 pooka puffs_setvattr(&pn->pn_va, &va);
205 1.12 pooka psn->attrread = time(NULL);
206 1.12 pooka }
207 1.12 pooka
208 1.12 pooka return 0;
209 1.12 pooka }
210 1.12 pooka
211 1.12 pooka int
212 1.1 pooka sftp_readdir(struct puffs_cc *pcc, struct psshfs_ctx *pctx,
213 1.1 pooka struct puffs_node *pn)
214 1.1 pooka {
215 1.1 pooka struct psshfs_node *psn = pn->pn_data;
216 1.1 pooka struct psshfs_dir *olddir, *testd;
217 1.1 pooka struct psbuf *pb;
218 1.1 pooka uint32_t reqid = NEXTREQ(pctx);
219 1.10 pooka uint32_t count, dhandlen;
220 1.1 pooka char *dhand = NULL;
221 1.10 pooka size_t nent;
222 1.1 pooka char *longname;
223 1.1 pooka int idx, rv;
224 1.1 pooka
225 1.1 pooka assert(pn->pn_va.va_type == VDIR);
226 1.1 pooka
227 1.1 pooka if (psn->dir && (time(NULL) - psn->dentread) < PSSHFS_REFRESHIVAL)
228 1.1 pooka return 0;
229 1.1 pooka
230 1.5 pooka puffs_inval_namecache_dir(puffs_cc_getusermount(pcc), pn);
231 1.5 pooka
232 1.1 pooka pb = psbuf_make(PSB_OUT);
233 1.7 pooka psbuf_req_str(pb, SSH_FXP_OPENDIR, reqid, PNPATH(pn));
234 1.1 pooka pssh_outbuf_enqueue(pctx, pb, pcc, reqid);
235 1.1 pooka
236 1.1 pooka puffs_cc_yield(pcc);
237 1.1 pooka
238 1.1 pooka rv = psbuf_expect_handle(pb, &dhand, &dhandlen);
239 1.1 pooka if (rv)
240 1.1 pooka goto wayout;
241 1.1 pooka
242 1.1 pooka /*
243 1.1 pooka * Well, the following is O(n^2), so feel free to improve if it
244 1.1 pooka * gets too taxing on your system.
245 1.1 pooka */
246 1.1 pooka olddir = psn->dir;
247 1.1 pooka nent = psn->dentnext;
248 1.1 pooka
249 1.8 pooka /*
250 1.8 pooka * note: for the "getattr in batch" to work, this must be before
251 1.8 pooka * the attribute-getting. Otherwise times for first entries in
252 1.8 pooka * large directories might expire before the directory itself and
253 1.8 pooka * result in one-by-one attribute fetching.
254 1.8 pooka */
255 1.8 pooka psn->dentread = time(NULL);
256 1.8 pooka
257 1.1 pooka psn->dentnext = 0;
258 1.1 pooka psn->denttot = 0;
259 1.1 pooka psn->dir = NULL;
260 1.1 pooka idx = 0;
261 1.1 pooka
262 1.1 pooka for (;;) {
263 1.1 pooka reqid = NEXTREQ(pctx);
264 1.1 pooka psbuf_recycle(pb, PSB_OUT);
265 1.1 pooka psbuf_req_data(pb, SSH_FXP_READDIR, reqid, dhand, dhandlen);
266 1.1 pooka pssh_outbuf_enqueue(pctx, pb, pcc, reqid);
267 1.1 pooka
268 1.1 pooka puffs_cc_yield(pcc);
269 1.1 pooka
270 1.1 pooka /* check for EOF */
271 1.1 pooka if (pb->type == SSH_FXP_STATUS) {
272 1.1 pooka rv = psbuf_expect_status(pb);
273 1.1 pooka goto out;
274 1.1 pooka }
275 1.1 pooka rv = psbuf_expect_name(pb, &count);
276 1.1 pooka if (rv)
277 1.1 pooka goto out;
278 1.1 pooka
279 1.1 pooka for (; count--; idx++) {
280 1.1 pooka if (idx == psn->denttot)
281 1.1 pooka allocdirs(psn);
282 1.1 pooka if (!psbuf_get_str(pb, &psn->dir[idx].entryname,
283 1.1 pooka NULL)) {
284 1.1 pooka rv = EPROTO;
285 1.1 pooka goto out;
286 1.1 pooka }
287 1.1 pooka if (!psbuf_get_str(pb, &longname, NULL)) {
288 1.1 pooka rv = EPROTO;
289 1.1 pooka goto out;
290 1.1 pooka }
291 1.1 pooka if (!psbuf_get_vattr(pb, &psn->dir[idx].va)) {
292 1.1 pooka rv = EPROTO;
293 1.1 pooka goto out;
294 1.1 pooka }
295 1.1 pooka if (sscanf(longname, "%*s%d",
296 1.1 pooka &psn->dir[idx].va.va_nlink) != 1) {
297 1.1 pooka rv = EPROTO;
298 1.1 pooka goto out;
299 1.1 pooka }
300 1.1 pooka free(longname);
301 1.1 pooka
302 1.1 pooka testd = lookup(olddir, nent, psn->dir[idx].entryname);
303 1.1 pooka if (testd) {
304 1.1 pooka psn->dir[idx].entry = testd->entry;
305 1.8 pooka psn->dir[idx].va = testd->va;
306 1.1 pooka } else {
307 1.1 pooka psn->dir[idx].entry = NULL;
308 1.1 pooka psn->dir[idx].va.va_fileid = pctx->nextino++;
309 1.1 pooka }
310 1.8 pooka #ifdef SUPERREADDIR
311 1.8 pooka /*
312 1.8 pooka * XXX: there's a dangling pointer race here if
313 1.8 pooka * the server responds to our queries out-of-order.
314 1.8 pooka * fixxxme some day
315 1.8 pooka */
316 1.8 pooka readdir_getattr(pctx, psn, PNPATH(pn), idx);
317 1.8 pooka #endif
318 1.1 pooka psn->dir[idx].valid = 1;
319 1.1 pooka }
320 1.1 pooka }
321 1.1 pooka
322 1.1 pooka out:
323 1.1 pooka /* XXX: rv */
324 1.1 pooka psn->dentnext = idx;
325 1.1 pooka freedircache(olddir, nent);
326 1.1 pooka
327 1.1 pooka reqid = NEXTREQ(pctx);
328 1.1 pooka psbuf_recycle(pb, PSB_OUT);
329 1.1 pooka psbuf_req_data(pb, SSH_FXP_CLOSE, reqid, dhand, dhandlen);
330 1.1 pooka
331 1.11 pooka pssh_outbuf_enqueue_nocc(pctx, pb, NULL, NULL, reqid);
332 1.11 pooka free(dhand);
333 1.12 pooka return rv;
334 1.1 pooka
335 1.1 pooka wayout:
336 1.1 pooka free(dhand);
337 1.11 pooka PSSHFSRETURN(rv);
338 1.1 pooka }
339 1.1 pooka
340 1.1 pooka struct puffs_node *
341 1.1 pooka makenode(struct puffs_usermount *pu, struct puffs_node *parent,
342 1.1 pooka struct psshfs_dir *pd, const struct vattr *vap)
343 1.1 pooka {
344 1.1 pooka struct psshfs_node *psn_parent = parent->pn_data;
345 1.1 pooka struct psshfs_node *psn;
346 1.1 pooka struct puffs_node *pn;
347 1.1 pooka
348 1.1 pooka psn = emalloc(sizeof(struct psshfs_node));
349 1.1 pooka memset(psn, 0, sizeof(struct psshfs_node));
350 1.1 pooka
351 1.1 pooka pn = puffs_pn_new(pu, psn);
352 1.1 pooka if (!pn) {
353 1.1 pooka free(psn);
354 1.1 pooka return NULL;
355 1.1 pooka }
356 1.1 pooka puffs_setvattr(&pn->pn_va, &pd->va);
357 1.8 pooka psn->attrread = pd->attrread;
358 1.1 pooka puffs_setvattr(&pn->pn_va, vap);
359 1.1 pooka
360 1.1 pooka pd->entry = pn;
361 1.1 pooka psn->parent = parent;
362 1.1 pooka psn_parent->childcount++;
363 1.1 pooka
364 1.1 pooka return pn;
365 1.1 pooka }
366 1.1 pooka
367 1.1 pooka struct puffs_node *
368 1.1 pooka allocnode(struct puffs_usermount *pu, struct puffs_node *parent,
369 1.1 pooka const char *entryname, const struct vattr *vap)
370 1.1 pooka {
371 1.13 pooka struct psshfs_ctx *pctx = puffs_getspecific(pu);
372 1.1 pooka struct psshfs_dir *pd;
373 1.3 pooka struct puffs_node *pn;
374 1.1 pooka
375 1.1 pooka pd = direnter(parent, entryname);
376 1.1 pooka
377 1.1 pooka pd->va.va_fileid = pctx->nextino++;
378 1.2 pooka if (vap->va_type == VDIR) {
379 1.1 pooka pd->va.va_nlink = 2;
380 1.2 pooka parent->pn_va.va_nlink++;
381 1.2 pooka } else {
382 1.1 pooka pd->va.va_nlink = 1;
383 1.2 pooka }
384 1.1 pooka
385 1.3 pooka pn = makenode(pu, parent, pd, vap);
386 1.3 pooka if (pn)
387 1.3 pooka pd->va.va_fileid = pn->pn_va.va_fileid;
388 1.3 pooka
389 1.3 pooka return pn;
390 1.1 pooka }
391 1.1 pooka
392 1.1 pooka struct psshfs_dir *
393 1.1 pooka direnter(struct puffs_node *parent, const char *entryname)
394 1.1 pooka {
395 1.1 pooka struct psshfs_node *psn_parent = parent->pn_data;
396 1.1 pooka struct psshfs_dir *pd;
397 1.1 pooka int i;
398 1.1 pooka
399 1.1 pooka /* create directory entry */
400 1.1 pooka if (psn_parent->denttot == psn_parent->dentnext)
401 1.1 pooka allocdirs(psn_parent);
402 1.1 pooka
403 1.1 pooka i = psn_parent->dentnext;
404 1.1 pooka pd = &psn_parent->dir[i];
405 1.1 pooka pd->entryname = estrdup(entryname);
406 1.1 pooka pd->valid = 1;
407 1.8 pooka pd->attrread = 0;
408 1.4 pooka puffs_vattr_null(&pd->va);
409 1.1 pooka psn_parent->dentnext++;
410 1.1 pooka
411 1.1 pooka return pd;
412 1.1 pooka }
413 1.1 pooka
414 1.1 pooka void
415 1.9 pooka doreclaim(struct puffs_node *pn)
416 1.9 pooka {
417 1.9 pooka struct psshfs_node *psn = pn->pn_data;
418 1.9 pooka struct psshfs_node *psn_parent;
419 1.9 pooka struct psshfs_dir *dent;
420 1.9 pooka
421 1.9 pooka psn_parent = psn->parent->pn_data;
422 1.9 pooka psn_parent->childcount--;
423 1.9 pooka
424 1.11 pooka /*
425 1.11 pooka * Null out entry from directory. Do not treat a missing entry
426 1.11 pooka * as an invariant error, since the node might be removed from
427 1.11 pooka * under us, and we might do a readdir before the reclaim resulting
428 1.11 pooka * in no directory entry in the parent directory.
429 1.11 pooka */
430 1.9 pooka dent = lookup_by_entry(psn_parent->dir, psn_parent->dentnext, pn);
431 1.11 pooka if (dent)
432 1.11 pooka dent->entry = NULL;
433 1.9 pooka
434 1.9 pooka if (pn->pn_va.va_type == VDIR)
435 1.9 pooka freedircache(psn->dir, psn->dentnext);
436 1.9 pooka
437 1.9 pooka puffs_pn_put(pn);
438 1.9 pooka }
439 1.9 pooka
440 1.9 pooka void
441 1.9 pooka nukenode(struct puffs_node *node, const char *entryname, int reclaim)
442 1.1 pooka {
443 1.1 pooka struct psshfs_node *psn, *psn_parent;
444 1.1 pooka struct psshfs_dir *pd;
445 1.1 pooka
446 1.1 pooka psn = node->pn_data;
447 1.1 pooka psn_parent = psn->parent->pn_data;
448 1.1 pooka pd = lookup(psn_parent->dir, psn_parent->dentnext, entryname);
449 1.1 pooka assert(pd != NULL);
450 1.1 pooka pd->valid = 0;
451 1.1 pooka free(pd->entryname);
452 1.1 pooka pd->entryname = NULL;
453 1.1 pooka
454 1.9 pooka if (node->pn_va.va_type == VDIR)
455 1.1 pooka psn->parent->pn_va.va_nlink--;
456 1.6 pooka
457 1.9 pooka if (reclaim)
458 1.9 pooka doreclaim(node);
459 1.1 pooka }
460