node.c revision 1.58 1 1.58 pooka /* $NetBSD: node.c,v 1.58 2009/05/20 14:08:21 pooka Exp $ */
2 1.1 pooka
3 1.1 pooka /*
4 1.58 pooka * Copyright (c) 2006-2009 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 *
15 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 1.1 pooka * SUCH DAMAGE.
26 1.1 pooka */
27 1.1 pooka
28 1.1 pooka #include <sys/cdefs.h>
29 1.1 pooka #ifndef lint
30 1.58 pooka __RCSID("$NetBSD: node.c,v 1.58 2009/05/20 14:08:21 pooka Exp $");
31 1.1 pooka #endif /* !lint */
32 1.1 pooka
33 1.1 pooka #include <assert.h>
34 1.1 pooka #include <errno.h>
35 1.1 pooka #include <stdio.h>
36 1.1 pooka #include <stdlib.h>
37 1.1 pooka
38 1.1 pooka #include "psshfs.h"
39 1.1 pooka #include "sftp_proto.h"
40 1.1 pooka
41 1.1 pooka int
42 1.54 pooka psshfs_node_lookup(struct puffs_usermount *pu, puffs_cookie_t opc,
43 1.48 pooka struct puffs_newinfo *pni, const struct puffs_cn *pcn)
44 1.1 pooka {
45 1.16 pooka struct psshfs_ctx *pctx = puffs_getspecific(pu);
46 1.1 pooka struct puffs_node *pn_dir = opc;
47 1.8 pooka struct psshfs_node *psn, *psn_dir = pn_dir->pn_data;
48 1.1 pooka struct puffs_node *pn;
49 1.1 pooka struct psshfs_dir *pd;
50 1.11 pooka struct vattr va;
51 1.1 pooka int rv;
52 1.1 pooka
53 1.8 pooka if (PCNISDOTDOT(pcn)) {
54 1.8 pooka psn = psn_dir->parent->pn_data;
55 1.28 pooka psn->stat &= ~PSN_RECLAIMED;
56 1.8 pooka
57 1.32 pooka puffs_newinfo_setcookie(pni, psn_dir->parent);
58 1.32 pooka puffs_newinfo_setvtype(pni, VDIR);
59 1.1 pooka return 0;
60 1.1 pooka }
61 1.1 pooka
62 1.52 pooka rv = sftp_readdir(pu, pctx, pn_dir);
63 1.1 pooka if (rv) {
64 1.11 pooka if (rv != EPERM)
65 1.11 pooka return rv;
66 1.11 pooka
67 1.11 pooka /*
68 1.11 pooka * Can't read the directory. We still might be
69 1.11 pooka * able to find the node with getattr in -r+x dirs
70 1.11 pooka */
71 1.48 pooka rv = getpathattr(pu, PCNPATH(pcn), &va);
72 1.11 pooka if (rv)
73 1.11 pooka return rv;
74 1.11 pooka
75 1.11 pooka /* guess */
76 1.11 pooka if (va.va_type == VDIR)
77 1.11 pooka va.va_nlink = 2;
78 1.11 pooka else
79 1.11 pooka va.va_nlink = 1;
80 1.11 pooka
81 1.11 pooka pn = allocnode(pu, pn_dir, pcn->pcn_name, &va);
82 1.12 pooka psn = pn->pn_data;
83 1.12 pooka psn->attrread = time(NULL);
84 1.11 pooka } else {
85 1.11 pooka pd = lookup(psn_dir->dir, psn_dir->dentnext, pcn->pcn_name);
86 1.11 pooka if (!pd) {
87 1.11 pooka return ENOENT;
88 1.11 pooka }
89 1.1 pooka
90 1.11 pooka if (pd->entry)
91 1.11 pooka pn = pd->entry;
92 1.11 pooka else
93 1.11 pooka pn = makenode(pu, pn_dir, pd, &pd->va);
94 1.12 pooka psn = pn->pn_data;
95 1.6 pooka }
96 1.1 pooka
97 1.28 pooka psn->stat &= ~PSN_RECLAIMED;
98 1.8 pooka
99 1.32 pooka puffs_newinfo_setcookie(pni, pn);
100 1.32 pooka puffs_newinfo_setvtype(pni, pn->pn_va.va_type);
101 1.32 pooka puffs_newinfo_setsize(pni, pn->pn_va.va_size);
102 1.1 pooka
103 1.1 pooka return 0;
104 1.1 pooka }
105 1.1 pooka
106 1.1 pooka int
107 1.54 pooka psshfs_node_getattr(struct puffs_usermount *pu, puffs_cookie_t opc,
108 1.54 pooka struct vattr *vap, const struct puffs_cred *pcr)
109 1.1 pooka {
110 1.1 pooka struct puffs_node *pn = opc;
111 1.11 pooka int rv;
112 1.1 pooka
113 1.48 pooka rv = getnodeattr(pu, pn);
114 1.11 pooka if (rv)
115 1.11 pooka return rv;
116 1.1 pooka
117 1.1 pooka memcpy(vap, &pn->pn_va, sizeof(struct vattr));
118 1.1 pooka
119 1.11 pooka return 0;
120 1.1 pooka }
121 1.1 pooka
122 1.1 pooka int
123 1.54 pooka psshfs_node_setattr(struct puffs_usermount *pu, puffs_cookie_t opc,
124 1.46 pooka const struct vattr *va, const struct puffs_cred *pcr)
125 1.1 pooka {
126 1.48 pooka PSSHFSAUTOVAR(pu);
127 1.3 pooka struct vattr kludgeva;
128 1.1 pooka struct puffs_node *pn = opc;
129 1.1 pooka
130 1.5 pooka psbuf_req_str(pb, SSH_FXP_SETSTAT, reqid, PNPATH(pn));
131 1.3 pooka
132 1.3 pooka memcpy(&kludgeva, va, sizeof(struct vattr));
133 1.3 pooka
134 1.3 pooka /* XXX: kludge due to openssh server implementation */
135 1.3 pooka if (va->va_atime.tv_sec != PUFFS_VNOVAL
136 1.3 pooka && va->va_mtime.tv_sec == PUFFS_VNOVAL) {
137 1.3 pooka if (pn->pn_va.va_mtime.tv_sec != PUFFS_VNOVAL)
138 1.3 pooka kludgeva.va_mtime.tv_sec = pn->pn_va.va_mtime.tv_sec;
139 1.3 pooka else
140 1.3 pooka kludgeva.va_mtime.tv_sec = va->va_atime.tv_sec;
141 1.3 pooka }
142 1.3 pooka if (va->va_mtime.tv_sec != PUFFS_VNOVAL
143 1.3 pooka && va->va_atime.tv_sec == PUFFS_VNOVAL) {
144 1.3 pooka if (pn->pn_va.va_atime.tv_sec != PUFFS_VNOVAL)
145 1.3 pooka kludgeva.va_atime.tv_sec = pn->pn_va.va_atime.tv_sec;
146 1.3 pooka else
147 1.3 pooka kludgeva.va_atime.tv_sec = va->va_mtime.tv_sec;
148 1.3 pooka }
149 1.3 pooka
150 1.3 pooka psbuf_put_vattr(pb, &kludgeva);
151 1.57 pooka GETRESPONSE(pb, pctx->sshfd);
152 1.1 pooka
153 1.1 pooka rv = psbuf_expect_status(pb);
154 1.1 pooka if (rv == 0)
155 1.3 pooka puffs_setvattr(&pn->pn_va, &kludgeva);
156 1.1 pooka
157 1.27 pooka out:
158 1.1 pooka PSSHFSRETURN(rv);
159 1.1 pooka }
160 1.1 pooka
161 1.1 pooka int
162 1.54 pooka psshfs_node_create(struct puffs_usermount *pu, puffs_cookie_t opc,
163 1.53 pooka struct puffs_newinfo *pni, const struct puffs_cn *pcn,
164 1.53 pooka const struct vattr *va)
165 1.1 pooka {
166 1.48 pooka PSSHFSAUTOVAR(pu);
167 1.1 pooka struct puffs_node *pn = opc;
168 1.1 pooka struct puffs_node *pn_new;
169 1.1 pooka char *fhand = NULL;
170 1.9 pooka uint32_t fhandlen;
171 1.1 pooka
172 1.53 pooka /* Create node on server first */
173 1.53 pooka psbuf_req_str(pb, SSH_FXP_OPEN, reqid, PCNPATH(pcn));
174 1.53 pooka psbuf_put_4(pb, SSH_FXF_WRITE | SSH_FXF_CREAT | SSH_FXF_TRUNC);
175 1.53 pooka psbuf_put_vattr(pb, va);
176 1.57 pooka GETRESPONSE(pb, pctx->sshfd);
177 1.53 pooka rv = psbuf_expect_handle(pb, &fhand, &fhandlen);
178 1.53 pooka if (rv)
179 1.53 pooka goto out;
180 1.53 pooka
181 1.53 pooka /*
182 1.53 pooka * Do *not* create the local node before getting a response
183 1.53 pooka * from the server. Otherwise we might screw up consistency,
184 1.53 pooka * namely that the node can be looked up before create has
185 1.53 pooka * returned (mind you, the kernel will unlock the directory
186 1.53 pooka * before the create call from userspace returns).
187 1.53 pooka */
188 1.1 pooka pn_new = allocnode(pu, pn, pcn->pcn_name, va);
189 1.13 pooka if (!pn_new) {
190 1.53 pooka struct puffs_framebuf *pb2 = psbuf_makeout();
191 1.53 pooka reqid = NEXTREQ(pctx);
192 1.53 pooka psbuf_req_str(pb2, SSH_FXP_REMOVE, reqid, PCNPATH(pcn));
193 1.57 pooka JUSTSEND(pb2, pctx->sshfd);
194 1.1 pooka rv = ENOMEM;
195 1.1 pooka }
196 1.1 pooka
197 1.53 pooka if (pn_new)
198 1.32 pooka puffs_newinfo_setcookie(pni, pn_new);
199 1.1 pooka
200 1.1 pooka reqid = NEXTREQ(pctx);
201 1.22 pooka psbuf_recycleout(pb);
202 1.1 pooka psbuf_req_data(pb, SSH_FXP_CLOSE, reqid, fhand, fhandlen);
203 1.57 pooka JUSTSEND(pb, pctx->sshfd);
204 1.23 pooka free(fhand);
205 1.53 pooka return rv;
206 1.1 pooka
207 1.1 pooka out:
208 1.1 pooka free(fhand);
209 1.1 pooka PSSHFSRETURN(rv);
210 1.1 pooka }
211 1.1 pooka
212 1.47 pooka /*
213 1.47 pooka * Open a file handle. This is used for read and write. We do not
214 1.47 pooka * wait here for the success or failure of this operation. This is
215 1.47 pooka * because otherwise opening and closing file handles would block
216 1.47 pooka * reading potentially cached information. Rather, we defer the wait
217 1.47 pooka * to read/write and therefore allow cached access without a wait.
218 1.47 pooka *
219 1.47 pooka * If we have not yet succesfully opened a type of handle, we do wait
220 1.47 pooka * here. Also, if a lazy open fails, we revert back to the same
221 1.47 pooka * state of waiting.
222 1.47 pooka */
223 1.36 pooka int
224 1.54 pooka psshfs_node_open(struct puffs_usermount *pu, puffs_cookie_t opc, int mode,
225 1.46 pooka const struct puffs_cred *pcr)
226 1.24 pooka {
227 1.48 pooka struct puffs_cc *pcc = puffs_cc_getcc(pu);
228 1.48 pooka struct psshfs_ctx *pctx = puffs_getspecific(pu);
229 1.47 pooka struct puffs_framebuf *pb, *pb2;
230 1.24 pooka struct vattr va;
231 1.24 pooka struct puffs_node *pn = opc;
232 1.24 pooka struct psshfs_node *psn = pn->pn_data;
233 1.47 pooka uint32_t reqid;
234 1.47 pooka int didread, didwrite;
235 1.47 pooka int rv = 0;
236 1.24 pooka
237 1.24 pooka if (pn->pn_va.va_type == VDIR)
238 1.47 pooka return 0;
239 1.24 pooka
240 1.25 pooka puffs_setback(pcc, PUFFS_SETBACK_INACT_N1);
241 1.24 pooka puffs_vattr_null(&va);
242 1.47 pooka didread = didwrite = 0;
243 1.47 pooka if (mode & FREAD && psn->fhand_r == NULL && psn->lazyopen_r == NULL) {
244 1.47 pooka pb = psbuf_makeout();
245 1.47 pooka
246 1.47 pooka reqid = NEXTREQ(pctx);
247 1.24 pooka psbuf_req_str(pb, SSH_FXP_OPEN, reqid, PNPATH(pn));
248 1.24 pooka psbuf_put_4(pb, SSH_FXF_READ);
249 1.24 pooka psbuf_put_vattr(pb, &va);
250 1.24 pooka
251 1.57 pooka if (puffs_framev_enqueue_cb(pu, pctx->sshfd_data, pb,
252 1.47 pooka lazyopen_rresp, psn, 0) == -1) {
253 1.56 pooka rv = errno;
254 1.47 pooka puffs_framebuf_destroy(pb);
255 1.24 pooka goto out;
256 1.47 pooka }
257 1.47 pooka
258 1.47 pooka psn->lazyopen_r = pb;
259 1.47 pooka didread = 1;
260 1.24 pooka }
261 1.47 pooka if (mode & FWRITE && psn->fhand_w == NULL && psn->lazyopen_w == NULL) {
262 1.47 pooka pb2 = psbuf_makeout();
263 1.24 pooka
264 1.47 pooka reqid = NEXTREQ(pctx);
265 1.47 pooka psbuf_req_str(pb2, SSH_FXP_OPEN, reqid, PNPATH(pn));
266 1.47 pooka psbuf_put_4(pb2, SSH_FXF_WRITE);
267 1.47 pooka psbuf_put_vattr(pb2, &va);
268 1.47 pooka
269 1.57 pooka if (puffs_framev_enqueue_cb(pu, pctx->sshfd_data, pb2,
270 1.47 pooka lazyopen_wresp, psn, 0) == -1) {
271 1.56 pooka rv = errno;
272 1.47 pooka puffs_framebuf_destroy(pb2);
273 1.24 pooka goto out;
274 1.47 pooka }
275 1.47 pooka
276 1.47 pooka psn->lazyopen_w = pb2;
277 1.47 pooka didwrite = 1;
278 1.24 pooka }
279 1.47 pooka psn->stat &= ~PSN_HANDLECLOSE;
280 1.24 pooka
281 1.24 pooka out:
282 1.47 pooka /* wait? */
283 1.47 pooka if (didread && (psn->stat & PSN_DOLAZY_R) == 0) {
284 1.47 pooka assert(psn->lazyopen_r);
285 1.47 pooka
286 1.47 pooka rv = puffs_framev_framebuf_ccpromote(psn->lazyopen_r, pcc);
287 1.47 pooka lazyopen_rresp(pu, psn->lazyopen_r, psn, rv);
288 1.47 pooka if (psn->fhand_r) {
289 1.47 pooka psn->stat |= PSN_DOLAZY_R;
290 1.47 pooka } else {
291 1.47 pooka if (psn->lazyopen_err_r)
292 1.47 pooka return psn->lazyopen_err_r;
293 1.47 pooka return EINVAL;
294 1.47 pooka }
295 1.47 pooka }
296 1.24 pooka
297 1.47 pooka /* wait? */
298 1.47 pooka if (didwrite && (psn->stat & PSN_DOLAZY_W) == 0) {
299 1.47 pooka assert(psn->lazyopen_w);
300 1.47 pooka
301 1.47 pooka rv = puffs_framev_framebuf_ccpromote(psn->lazyopen_w, pcc);
302 1.47 pooka lazyopen_wresp(pu, psn->lazyopen_w, psn, rv);
303 1.47 pooka if (psn->fhand_w) {
304 1.47 pooka psn->stat |= PSN_DOLAZY_W;
305 1.47 pooka } else {
306 1.47 pooka if (psn->lazyopen_err_w)
307 1.47 pooka return psn->lazyopen_err_w;
308 1.47 pooka return EINVAL;
309 1.47 pooka }
310 1.24 pooka }
311 1.24 pooka
312 1.47 pooka return rv;
313 1.36 pooka }
314 1.36 pooka
315 1.36 pooka int
316 1.54 pooka psshfs_node_inactive(struct puffs_usermount *pu, puffs_cookie_t opc)
317 1.36 pooka {
318 1.36 pooka struct puffs_node *pn = opc;
319 1.36 pooka
320 1.48 pooka closehandles(pu, pn->pn_data, HANDLE_READ | HANDLE_WRITE);
321 1.24 pooka return 0;
322 1.24 pooka }
323 1.24 pooka
324 1.24 pooka int
325 1.54 pooka psshfs_node_readdir(struct puffs_usermount *pu, puffs_cookie_t opc,
326 1.54 pooka struct dirent *dent, off_t *readoff, size_t *reslen,
327 1.54 pooka const struct puffs_cred *pcr, int *eofflag,
328 1.54 pooka off_t *cookies, size_t *ncookies)
329 1.1 pooka {
330 1.48 pooka struct puffs_cc *pcc = puffs_cc_getcc(pu);
331 1.48 pooka struct psshfs_ctx *pctx = puffs_getspecific(pu);
332 1.1 pooka struct puffs_node *pn = opc;
333 1.1 pooka struct psshfs_node *psn = pn->pn_data;
334 1.1 pooka struct psshfs_dir *pd;
335 1.38 pooka int i, rv, set_readdir;
336 1.38 pooka
337 1.47 pooka restart:
338 1.38 pooka if (psn->stat & PSN_READDIR) {
339 1.41 pooka struct psshfs_wait pw;
340 1.38 pooka
341 1.38 pooka set_readdir = 0;
342 1.41 pooka pw.pw_cc = pcc;
343 1.47 pooka pw.pw_type = PWTYPE_READDIR;
344 1.41 pooka TAILQ_INSERT_TAIL(&psn->pw, &pw, pw_entries);
345 1.38 pooka puffs_cc_yield(pcc);
346 1.47 pooka goto restart;
347 1.38 pooka } else {
348 1.38 pooka psn->stat |= PSN_READDIR;
349 1.38 pooka set_readdir = 1;
350 1.38 pooka }
351 1.1 pooka
352 1.17 pooka *ncookies = 0;
353 1.52 pooka rv = sftp_readdir(pu, pctx, pn);
354 1.1 pooka if (rv)
355 1.38 pooka goto out;
356 1.1 pooka
357 1.35 pooka /* find next dirent */
358 1.35 pooka for (i = *readoff;;i++) {
359 1.38 pooka if (i >= psn->dentnext)
360 1.35 pooka goto out;
361 1.1 pooka pd = &psn->dir[i];
362 1.35 pooka if (pd->valid)
363 1.35 pooka break;
364 1.35 pooka }
365 1.35 pooka
366 1.35 pooka for (;;) {
367 1.35 pooka *readoff = i;
368 1.1 pooka if (!puffs_nextdent(&dent, pd->entryname,
369 1.38 pooka pd->va.va_fileid, puffs_vtype2dt(pd->va.va_type), reslen)) {
370 1.38 pooka rv = 0;
371 1.38 pooka goto out;
372 1.38 pooka }
373 1.35 pooka
374 1.35 pooka /* find next entry, store possible nfs key */
375 1.35 pooka do {
376 1.38 pooka if (++i >= psn->dentnext)
377 1.35 pooka goto out;
378 1.35 pooka pd = &psn->dir[i];
379 1.35 pooka } while (pd->valid == 0);
380 1.19 pooka PUFFS_STORE_DCOOKIE(cookies, ncookies, (off_t)i);
381 1.1 pooka }
382 1.35 pooka
383 1.35 pooka out:
384 1.38 pooka if (rv == 0) {
385 1.38 pooka if (i >= psn->dentnext)
386 1.38 pooka *eofflag = 1;
387 1.38 pooka
388 1.38 pooka *readoff = i;
389 1.38 pooka }
390 1.38 pooka
391 1.38 pooka if (set_readdir) {
392 1.41 pooka struct psshfs_wait *pw;
393 1.1 pooka
394 1.42 pooka /* all will likely run to completion because of cache */
395 1.47 pooka TAILQ_FOREACH(pw, &psn->pw, pw_entries) {
396 1.47 pooka assert(pw->pw_type == PWTYPE_READDIR);
397 1.42 pooka puffs_cc_schedule(pw->pw_cc);
398 1.49 pooka TAILQ_REMOVE(&psn->pw, pw, pw_entries);
399 1.47 pooka }
400 1.38 pooka
401 1.38 pooka psn->stat &= ~PSN_READDIR;
402 1.38 pooka }
403 1.38 pooka
404 1.38 pooka return rv;
405 1.1 pooka }
406 1.1 pooka
407 1.1 pooka int
408 1.54 pooka psshfs_node_read(struct puffs_usermount *pu, puffs_cookie_t opc, uint8_t *buf,
409 1.1 pooka off_t offset, size_t *resid, const struct puffs_cred *pcr,
410 1.1 pooka int ioflag)
411 1.1 pooka {
412 1.48 pooka PSSHFSAUTOVAR(pu);
413 1.1 pooka struct puffs_node *pn = opc;
414 1.24 pooka struct psshfs_node *psn = pn->pn_data;
415 1.47 pooka struct psshfs_wait *pwp;
416 1.24 pooka uint32_t readlen;
417 1.1 pooka
418 1.42 pooka if (pn->pn_va.va_type == VDIR) {
419 1.42 pooka rv = EISDIR;
420 1.42 pooka goto farout;
421 1.42 pooka }
422 1.42 pooka
423 1.47 pooka /* check that a lazyopen didn't fail */
424 1.47 pooka if (!psn->fhand_r && !psn->lazyopen_r) {
425 1.47 pooka rv = psn->lazyopen_err_r;
426 1.47 pooka goto farout;
427 1.47 pooka }
428 1.47 pooka
429 1.47 pooka /* if someone is already waiting for the lazyopen, "just" wait */
430 1.47 pooka if (psn->stat & PSN_LAZYWAIT_R) {
431 1.47 pooka struct psshfs_wait pw;
432 1.47 pooka
433 1.47 pooka assert(psn->lazyopen_r);
434 1.47 pooka
435 1.47 pooka pw.pw_cc = pcc;
436 1.47 pooka pw.pw_type = PWTYPE_READ1;
437 1.47 pooka TAILQ_INSERT_TAIL(&psn->pw, &pw, pw_entries);
438 1.47 pooka puffs_cc_yield(pcc);
439 1.47 pooka }
440 1.47 pooka
441 1.47 pooka /* if lazyopening, wait for the result */
442 1.47 pooka if (psn->lazyopen_r) {
443 1.47 pooka psn->stat |= PSN_LAZYWAIT_R;
444 1.47 pooka rv = puffs_framev_framebuf_ccpromote(psn->lazyopen_r, pcc);
445 1.47 pooka lazyopen_rresp(pu, psn->lazyopen_r, psn, rv);
446 1.47 pooka
447 1.47 pooka /* schedule extra waiters */
448 1.47 pooka TAILQ_FOREACH(pwp, &psn->pw, pw_entries)
449 1.49 pooka if (pwp->pw_type == PWTYPE_READ1) {
450 1.47 pooka puffs_cc_schedule(pwp->pw_cc);
451 1.49 pooka TAILQ_REMOVE(&psn->pw, pwp, pw_entries);
452 1.49 pooka }
453 1.47 pooka psn->stat &= ~PSN_LAZYWAIT_R;
454 1.47 pooka
455 1.47 pooka if ((rv = psn->lazyopen_err_r) != 0)
456 1.47 pooka goto farout;
457 1.47 pooka }
458 1.47 pooka
459 1.47 pooka /* if there is still no handle, just refuse to live with this */
460 1.47 pooka if (!psn->fhand_r) {
461 1.47 pooka rv = EINVAL;
462 1.47 pooka goto farout;
463 1.47 pooka }
464 1.47 pooka
465 1.41 pooka readlen = *resid;
466 1.41 pooka psbuf_req_data(pb, SSH_FXP_READ, reqid, psn->fhand_r, psn->fhand_r_len);
467 1.41 pooka psbuf_put_8(pb, offset);
468 1.41 pooka psbuf_put_4(pb, readlen);
469 1.41 pooka
470 1.41 pooka /*
471 1.41 pooka * Do this *after* accessing the file, the handle might not
472 1.41 pooka * exist after blocking.
473 1.41 pooka */
474 1.41 pooka if (max_reads && ++psn->readcount > max_reads) {
475 1.41 pooka struct psshfs_wait pw;
476 1.41 pooka
477 1.41 pooka pw.pw_cc = pcc;
478 1.47 pooka pw.pw_type = PWTYPE_READ2;
479 1.41 pooka TAILQ_INSERT_TAIL(&psn->pw, &pw, pw_entries);
480 1.41 pooka puffs_cc_yield(pcc);
481 1.1 pooka }
482 1.1 pooka
483 1.57 pooka GETRESPONSE(pb, pctx->sshfd_data);
484 1.1 pooka
485 1.1 pooka rv = psbuf_do_data(pb, buf, &readlen);
486 1.1 pooka if (rv == 0)
487 1.1 pooka *resid -= readlen;
488 1.1 pooka
489 1.27 pooka out:
490 1.41 pooka if (max_reads && --psn->readcount >= max_reads) {
491 1.47 pooka TAILQ_FOREACH(pwp, &psn->pw, pw_entries)
492 1.47 pooka if (pwp->pw_type == PWTYPE_READ2)
493 1.47 pooka break;
494 1.47 pooka assert(pwp != NULL);
495 1.49 pooka puffs_cc_schedule(pwp->pw_cc);
496 1.47 pooka TAILQ_REMOVE(&psn->pw, pwp, pw_entries);
497 1.41 pooka }
498 1.41 pooka
499 1.42 pooka farout:
500 1.47 pooka /* check if we need a lazyclose */
501 1.47 pooka if (psn->stat & PSN_HANDLECLOSE && psn->fhand_r) {
502 1.47 pooka TAILQ_FOREACH(pwp, &psn->pw, pw_entries)
503 1.47 pooka if (pwp->pw_type == PWTYPE_READ1)
504 1.47 pooka break;
505 1.47 pooka if (pwp == NULL)
506 1.47 pooka closehandles(pu, psn, HANDLE_READ);
507 1.47 pooka }
508 1.1 pooka PSSHFSRETURN(rv);
509 1.1 pooka }
510 1.1 pooka
511 1.1 pooka /* XXX: we should getattr for size */
512 1.1 pooka int
513 1.54 pooka psshfs_node_write(struct puffs_usermount *pu, puffs_cookie_t opc, uint8_t *buf,
514 1.1 pooka off_t offset, size_t *resid, const struct puffs_cred *cred,
515 1.1 pooka int ioflag)
516 1.1 pooka {
517 1.48 pooka PSSHFSAUTOVAR(pu);
518 1.47 pooka struct psshfs_wait *pwp;
519 1.1 pooka struct puffs_node *pn = opc;
520 1.24 pooka struct psshfs_node *psn = pn->pn_data;
521 1.24 pooka uint32_t writelen;
522 1.1 pooka
523 1.1 pooka if (pn->pn_va.va_type == VDIR) {
524 1.1 pooka rv = EISDIR;
525 1.27 pooka goto out;
526 1.1 pooka }
527 1.1 pooka
528 1.47 pooka /* check that a lazyopen didn't fail */
529 1.47 pooka if (!psn->fhand_w && !psn->lazyopen_w) {
530 1.47 pooka rv = psn->lazyopen_err_w;
531 1.47 pooka goto out;
532 1.47 pooka }
533 1.47 pooka
534 1.47 pooka if (psn->stat & PSN_LAZYWAIT_W) {
535 1.47 pooka struct psshfs_wait pw;
536 1.47 pooka
537 1.47 pooka assert(psn->lazyopen_w);
538 1.47 pooka
539 1.47 pooka pw.pw_cc = pcc;
540 1.47 pooka pw.pw_type = PWTYPE_WRITE;
541 1.47 pooka TAILQ_INSERT_TAIL(&psn->pw, &pw, pw_entries);
542 1.47 pooka puffs_cc_yield(pcc);
543 1.47 pooka }
544 1.47 pooka
545 1.47 pooka /*
546 1.47 pooka * If lazyopening, wait for the result.
547 1.47 pooka * There can still be more than oen writer at a time in case
548 1.47 pooka * the kernel issues write FAFs.
549 1.47 pooka */
550 1.47 pooka if (psn->lazyopen_w) {
551 1.47 pooka psn->stat |= PSN_LAZYWAIT_W;
552 1.47 pooka rv = puffs_framev_framebuf_ccpromote(psn->lazyopen_w, pcc);
553 1.47 pooka lazyopen_wresp(pu, psn->lazyopen_w, psn, rv);
554 1.47 pooka
555 1.47 pooka /* schedule extra waiters */
556 1.47 pooka TAILQ_FOREACH(pwp, &psn->pw, pw_entries)
557 1.49 pooka if (pwp->pw_type == PWTYPE_WRITE) {
558 1.47 pooka puffs_cc_schedule(pwp->pw_cc);
559 1.49 pooka TAILQ_REMOVE(&psn->pw, pwp, pw_entries);
560 1.49 pooka }
561 1.47 pooka psn->stat &= ~PSN_LAZYWAIT_W;
562 1.47 pooka
563 1.47 pooka if ((rv = psn->lazyopen_err_w) != 0)
564 1.47 pooka goto out;
565 1.47 pooka }
566 1.47 pooka
567 1.47 pooka if (!psn->fhand_w) {
568 1.47 pooka abort();
569 1.47 pooka rv = EINVAL;
570 1.47 pooka goto out;
571 1.47 pooka }
572 1.47 pooka
573 1.1 pooka writelen = *resid;
574 1.24 pooka psbuf_req_data(pb, SSH_FXP_WRITE, reqid, psn->fhand_w,psn->fhand_w_len);
575 1.1 pooka psbuf_put_8(pb, offset);
576 1.1 pooka psbuf_put_data(pb, buf, writelen);
577 1.57 pooka GETRESPONSE(pb, pctx->sshfd_data);
578 1.1 pooka
579 1.1 pooka rv = psbuf_expect_status(pb);
580 1.1 pooka if (rv == 0)
581 1.1 pooka *resid = 0;
582 1.1 pooka
583 1.1 pooka if (pn->pn_va.va_size < offset + writelen)
584 1.1 pooka pn->pn_va.va_size = offset + writelen;
585 1.1 pooka
586 1.27 pooka out:
587 1.47 pooka /* check if we need a lazyclose */
588 1.47 pooka if (psn->stat & PSN_HANDLECLOSE && psn->fhand_w) {
589 1.47 pooka TAILQ_FOREACH(pwp, &psn->pw, pw_entries)
590 1.47 pooka if (pwp->pw_type == PWTYPE_WRITE)
591 1.47 pooka break;
592 1.47 pooka if (pwp == NULL)
593 1.47 pooka closehandles(pu, psn, HANDLE_WRITE);
594 1.47 pooka }
595 1.1 pooka PSSHFSRETURN(rv);
596 1.1 pooka }
597 1.1 pooka
598 1.1 pooka int
599 1.54 pooka psshfs_node_readlink(struct puffs_usermount *pu, puffs_cookie_t opc,
600 1.1 pooka const struct puffs_cred *cred, char *linkvalue, size_t *linklen)
601 1.1 pooka {
602 1.48 pooka PSSHFSAUTOVAR(pu);
603 1.1 pooka struct puffs_node *pn = opc;
604 1.43 pooka struct psshfs_node *psn = pn->pn_data;
605 1.1 pooka uint32_t count;
606 1.1 pooka
607 1.1 pooka if (pctx->protover < 3) {
608 1.1 pooka rv = EOPNOTSUPP;
609 1.1 pooka goto out;
610 1.1 pooka }
611 1.1 pooka
612 1.44 pooka /*
613 1.44 pooka * check if we can use a cached version
614 1.44 pooka *
615 1.44 pooka * XXX: we might end up reading the same link multiple times
616 1.44 pooka * from the server if we get many requests at once, but that's
617 1.44 pooka * quite harmless as this routine is reentrant.
618 1.44 pooka */
619 1.44 pooka if (psn->symlink && !REFRESHTIMEOUT(pctx, time(NULL) - psn->slread))
620 1.43 pooka goto copy;
621 1.43 pooka
622 1.44 pooka if (psn->symlink) {
623 1.44 pooka free(psn->symlink);
624 1.44 pooka psn->symlink = NULL;
625 1.44 pooka psn->slread = 0;
626 1.44 pooka }
627 1.44 pooka
628 1.5 pooka psbuf_req_str(pb, SSH_FXP_READLINK, reqid, PNPATH(pn));
629 1.57 pooka GETRESPONSE(pb, pctx->sshfd);
630 1.1 pooka
631 1.1 pooka rv = psbuf_expect_name(pb, &count);
632 1.22 pooka if (rv)
633 1.22 pooka goto out;
634 1.1 pooka if (count != 1) {
635 1.1 pooka rv = EPROTO;
636 1.1 pooka goto out;
637 1.1 pooka }
638 1.22 pooka
639 1.43 pooka rv = psbuf_get_str(pb, &psn->symlink, NULL);
640 1.1 pooka if (rv)
641 1.1 pooka goto out;
642 1.44 pooka psn->slread = time(NULL);
643 1.43 pooka
644 1.43 pooka copy:
645 1.43 pooka *linklen = strlen(psn->symlink);
646 1.43 pooka (void) memcpy(linkvalue, psn->symlink, *linklen);
647 1.1 pooka
648 1.1 pooka out:
649 1.1 pooka PSSHFSRETURN(rv);
650 1.1 pooka }
651 1.1 pooka
652 1.34 pooka static int
653 1.48 pooka doremove(struct puffs_usermount *pu, struct puffs_node *pn_dir,
654 1.34 pooka struct puffs_node *pn, const char *name)
655 1.34 pooka {
656 1.48 pooka PSSHFSAUTOVAR(pu);
657 1.34 pooka int op;
658 1.34 pooka
659 1.34 pooka if (pn->pn_va.va_type == VDIR)
660 1.34 pooka op = SSH_FXP_RMDIR;
661 1.34 pooka else
662 1.34 pooka op = SSH_FXP_REMOVE;
663 1.34 pooka
664 1.34 pooka psbuf_req_str(pb, op, reqid, PNPATH(pn));
665 1.57 pooka GETRESPONSE(pb, pctx->sshfd);
666 1.34 pooka
667 1.34 pooka rv = psbuf_expect_status(pb);
668 1.34 pooka if (rv == 0)
669 1.34 pooka nukenode(pn, name, 0);
670 1.34 pooka
671 1.34 pooka out:
672 1.34 pooka PSSHFSRETURN(rv);
673 1.34 pooka }
674 1.34 pooka
675 1.1 pooka int
676 1.54 pooka psshfs_node_remove(struct puffs_usermount *pu, puffs_cookie_t opc,
677 1.54 pooka puffs_cookie_t targ, const struct puffs_cn *pcn)
678 1.1 pooka {
679 1.1 pooka struct puffs_node *pn_targ = targ;
680 1.34 pooka int rv;
681 1.34 pooka
682 1.34 pooka assert(pn_targ->pn_va.va_type != VDIR);
683 1.34 pooka
684 1.48 pooka rv = doremove(pu, opc, targ, pcn->pcn_name);
685 1.34 pooka if (rv == 0)
686 1.48 pooka puffs_setback(puffs_cc_getcc(pu), PUFFS_SETBACK_NOREF_N2);
687 1.1 pooka
688 1.34 pooka return rv;
689 1.34 pooka }
690 1.1 pooka
691 1.34 pooka int
692 1.54 pooka psshfs_node_rmdir(struct puffs_usermount *pu, puffs_cookie_t opc,
693 1.54 pooka puffs_cookie_t targ, const struct puffs_cn *pcn)
694 1.34 pooka {
695 1.34 pooka struct puffs_node *pn_targ = targ;
696 1.34 pooka int rv;
697 1.1 pooka
698 1.34 pooka assert(pn_targ->pn_va.va_type == VDIR);
699 1.1 pooka
700 1.48 pooka rv = doremove(pu, opc, targ, pcn->pcn_name);
701 1.34 pooka if (rv == 0)
702 1.48 pooka puffs_setback(puffs_cc_getcc(pu), PUFFS_SETBACK_NOREF_N2);
703 1.1 pooka
704 1.34 pooka return rv;
705 1.1 pooka }
706 1.1 pooka
707 1.1 pooka int
708 1.54 pooka psshfs_node_mkdir(struct puffs_usermount *pu, puffs_cookie_t opc,
709 1.53 pooka struct puffs_newinfo *pni, const struct puffs_cn *pcn,
710 1.53 pooka const struct vattr *va)
711 1.1 pooka {
712 1.48 pooka PSSHFSAUTOVAR(pu);
713 1.1 pooka struct puffs_node *pn = opc;
714 1.1 pooka struct puffs_node *pn_new;
715 1.1 pooka
716 1.5 pooka psbuf_req_str(pb, SSH_FXP_MKDIR, reqid, PCNPATH(pcn));
717 1.1 pooka psbuf_put_vattr(pb, va);
718 1.57 pooka GETRESPONSE(pb, pctx->sshfd);
719 1.1 pooka
720 1.1 pooka rv = psbuf_expect_status(pb);
721 1.53 pooka if (rv)
722 1.53 pooka goto out;
723 1.1 pooka
724 1.53 pooka pn_new = allocnode(pu, pn, pcn->pcn_name, va);
725 1.53 pooka if (pn_new) {
726 1.32 pooka puffs_newinfo_setcookie(pni, pn_new);
727 1.53 pooka } else {
728 1.53 pooka struct puffs_framebuf *pb2 = psbuf_makeout();
729 1.53 pooka reqid = NEXTREQ(pctx);
730 1.53 pooka psbuf_recycleout(pb2);
731 1.53 pooka psbuf_req_str(pb2, SSH_FXP_RMDIR, reqid, PCNPATH(pcn));
732 1.57 pooka JUSTSEND(pb2, pctx->sshfd);
733 1.53 pooka rv = ENOMEM;
734 1.53 pooka }
735 1.1 pooka
736 1.1 pooka out:
737 1.1 pooka PSSHFSRETURN(rv);
738 1.1 pooka }
739 1.1 pooka
740 1.1 pooka int
741 1.54 pooka psshfs_node_symlink(struct puffs_usermount *pu, puffs_cookie_t opc,
742 1.53 pooka struct puffs_newinfo *pni, const struct puffs_cn *pcn,
743 1.53 pooka const struct vattr *va, const char *link_target)
744 1.1 pooka {
745 1.48 pooka PSSHFSAUTOVAR(pu);
746 1.1 pooka struct puffs_node *pn = opc;
747 1.1 pooka struct puffs_node *pn_new;
748 1.1 pooka
749 1.1 pooka if (pctx->protover < 3) {
750 1.1 pooka rv = EOPNOTSUPP;
751 1.1 pooka goto out;
752 1.1 pooka }
753 1.1 pooka
754 1.1 pooka /*
755 1.1 pooka * XXX: ietf says: source, target. openssh says: ietf who?
756 1.1 pooka * Let's go with openssh and build quirk tables later if we care
757 1.1 pooka */
758 1.1 pooka psbuf_req_str(pb, SSH_FXP_SYMLINK, reqid, link_target);
759 1.5 pooka psbuf_put_str(pb, PCNPATH(pcn));
760 1.57 pooka GETRESPONSE(pb, pctx->sshfd);
761 1.1 pooka
762 1.1 pooka rv = psbuf_expect_status(pb);
763 1.53 pooka if (rv)
764 1.53 pooka goto out;
765 1.53 pooka
766 1.53 pooka pn_new = allocnode(pu, pn, pcn->pcn_name, va);
767 1.53 pooka if (pn_new) {
768 1.32 pooka puffs_newinfo_setcookie(pni, pn_new);
769 1.53 pooka } else {
770 1.53 pooka struct puffs_framebuf *pb2 = psbuf_makeout();
771 1.53 pooka reqid = NEXTREQ(pctx);
772 1.53 pooka psbuf_recycleout(pb2);
773 1.53 pooka psbuf_req_str(pb2, SSH_FXP_REMOVE, reqid, PCNPATH(pcn));
774 1.57 pooka JUSTSEND(pb2, pctx->sshfd);
775 1.53 pooka rv = ENOMEM;
776 1.53 pooka }
777 1.1 pooka
778 1.1 pooka out:
779 1.1 pooka PSSHFSRETURN(rv);
780 1.1 pooka }
781 1.1 pooka
782 1.1 pooka int
783 1.54 pooka psshfs_node_rename(struct puffs_usermount *pu, puffs_cookie_t opc,
784 1.54 pooka puffs_cookie_t src, const struct puffs_cn *pcn_src,
785 1.54 pooka puffs_cookie_t targ_dir, puffs_cookie_t targ,
786 1.1 pooka const struct puffs_cn *pcn_targ)
787 1.1 pooka {
788 1.48 pooka PSSHFSAUTOVAR(pu);
789 1.1 pooka struct puffs_node *pn_sf = src;
790 1.1 pooka struct puffs_node *pn_td = targ_dir, *pn_tf = targ;
791 1.55 pooka struct psshfs_node *psn_src = pn_sf->pn_data;
792 1.1 pooka struct psshfs_node *psn_targdir = pn_td->pn_data;
793 1.1 pooka
794 1.1 pooka if (pctx->protover < 2) {
795 1.1 pooka rv = EOPNOTSUPP;
796 1.1 pooka goto out;
797 1.1 pooka }
798 1.1 pooka
799 1.1 pooka if (pn_tf) {
800 1.48 pooka rv = doremove(pu, targ_dir, pn_tf, pcn_targ->pcn_name);
801 1.1 pooka if (rv)
802 1.1 pooka goto out;
803 1.1 pooka }
804 1.1 pooka
805 1.5 pooka psbuf_req_str(pb, SSH_FXP_RENAME, reqid, PCNPATH(pcn_src));
806 1.5 pooka psbuf_put_str(pb, PCNPATH(pcn_targ));
807 1.57 pooka GETRESPONSE(pb, pctx->sshfd);
808 1.1 pooka
809 1.1 pooka rv = psbuf_expect_status(pb);
810 1.1 pooka if (rv == 0) {
811 1.1 pooka struct psshfs_dir *pd;
812 1.1 pooka
813 1.1 pooka /*
814 1.1 pooka * XXX: interfaces didn't quite work with rename..
815 1.1 pooka * the song remains the same. go figure .. ;)
816 1.1 pooka */
817 1.4 pooka nukenode(pn_sf, pcn_src->pcn_name, 0);
818 1.1 pooka pd = direnter(pn_td, pcn_targ->pcn_name);
819 1.1 pooka pd->entry = pn_sf;
820 1.1 pooka puffs_setvattr(&pd->va, &pn_sf->pn_va);
821 1.1 pooka
822 1.1 pooka if (opc != targ_dir) {
823 1.1 pooka psn_targdir->childcount++;
824 1.55 pooka psn_src->parent = pn_td;
825 1.1 pooka if (pn_sf->pn_va.va_type == VDIR)
826 1.1 pooka pn_td->pn_va.va_nlink++;
827 1.1 pooka }
828 1.1 pooka }
829 1.1 pooka
830 1.1 pooka out:
831 1.1 pooka PSSHFSRETURN(rv);
832 1.1 pooka }
833 1.8 pooka
834 1.8 pooka /*
835 1.8 pooka * So this file system happened to be written in such a way that
836 1.8 pooka * lookup for ".." is hard if we lose the in-memory node. We'd
837 1.8 pooka * need to recreate the entire directory structure from the root
838 1.8 pooka * node up to the ".." node we're looking up.
839 1.8 pooka *
840 1.8 pooka * And since our entire fs structure is purely fictional (i.e. it's
841 1.8 pooka * only in-memory, not fetchable from the server), the easiest way
842 1.8 pooka * to deal with it is to not allow nodes with children to be
843 1.8 pooka * reclaimed.
844 1.8 pooka *
845 1.8 pooka * If a node with children is being attempted to be reclaimed, we
846 1.8 pooka * just mark it "reclaimed" but leave it as is until all its children
847 1.8 pooka * have been reclaimed. If a lookup for that node is done meanwhile,
848 1.8 pooka * it will be found by lookup() and we just remove the "reclaimed"
849 1.8 pooka * bit.
850 1.8 pooka */
851 1.8 pooka int
852 1.54 pooka psshfs_node_reclaim(struct puffs_usermount *pu, puffs_cookie_t opc)
853 1.8 pooka {
854 1.16 pooka struct puffs_node *pn = opc, *pn_next, *pn_root;
855 1.8 pooka struct psshfs_node *psn = pn->pn_data;
856 1.8 pooka
857 1.17 pooka /*
858 1.17 pooka * don't reclaim if we have file handle issued, otherwise
859 1.17 pooka * we can't do fhtonode
860 1.17 pooka */
861 1.28 pooka if (psn->stat & PSN_HASFH)
862 1.17 pooka return 0;
863 1.17 pooka
864 1.28 pooka psn->stat |= PSN_RECLAIMED;
865 1.16 pooka pn_root = puffs_getroot(pu);
866 1.16 pooka for (; pn != pn_root; pn = pn_next) {
867 1.8 pooka psn = pn->pn_data;
868 1.28 pooka if ((psn->stat & PSN_RECLAIMED) == 0 || psn->childcount != 0)
869 1.8 pooka break;
870 1.8 pooka
871 1.8 pooka pn_next = psn->parent;
872 1.8 pooka doreclaim(pn);
873 1.8 pooka }
874 1.8 pooka
875 1.8 pooka return 0;
876 1.8 pooka }
877