1/*	$NetBSD: fs.c,v 1.8 2008/08/24 18:26:05 pooka Exp $	*/
2
3/*
4 * Copyright (c) 2007  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.8 2008/08/24 18:26:05 pooka Exp $");
31#endif /* !lint */
32
33#include <assert.h>
34#include <err.h>
35#include <errno.h>
36#include <puffs.h>
37#include <signal.h>
38#include <stdio.h>
39#include <stdlib.h>
40#include <unistd.h>
41
42#include "ninepuffs.h"
43#include "nineproto.h"
44
45#define DO_IO(fname, a1, a2, a3, a4, rv)				\
46	puffs_framebuf_seekset(a2, 0);					\
47	*(a4) = 0;							\
48	rv = fname(a1, a2, a3, a4);					\
49	if (rv) errx(1, "p9p_handshake io failed %d, %d", rv, *a4)
50
51struct puffs_node *
52p9p_handshake(struct puffs_usermount *pu,
53	const char *username, const char *path)
54{
55	struct puffs9p *p9p = puffs_getspecific(pu);
56	struct puffs_framebuf *pb;
57	struct puffs_node *pn;
58	struct vattr rootva;
59	uint32_t maxreq;
60	uint16_t dummy;
61	p9ptag_t tagid, rtagid;
62	p9pfid_t curfid;
63	const char *p;
64	uint8_t type;
65	int rv, done, x = 1, ncomp;
66
67	/* send initial handshake */
68	pb = p9pbuf_makeout();
69	p9pbuf_put_1(pb, P9PROTO_T_VERSION);
70	p9pbuf_put_2(pb, P9PROTO_NOTAG);
71	p9pbuf_put_4(pb, p9p->maxreq);
72	p9pbuf_put_str(pb, P9PROTO_VERSION);
73	DO_IO(p9pbuf_write, pu, pb, p9p->servsock, &done, rv);
74
75	puffs_framebuf_recycle(pb);
76	DO_IO(p9pbuf_read, pu, pb, p9p->servsock, &done, rv);
77
78	if ((type = p9pbuf_get_type(pb)) != P9PROTO_R_VERSION)
79		errx(1, "server invalid response to Tversion: %d", type);
80	if ((rtagid = p9pbuf_get_tag(pb)) != P9PROTO_NOTAG) {
81		errx(1, "server invalid tag: %d vs. %d",
82		    P9PROTO_NOTAG, rtagid);
83		return NULL;
84	}
85	if (p9pbuf_get_4(pb, &maxreq))
86		errx(1, "server invalid response: no request length");
87	if (maxreq < P9P_MINREQLEN)
88		errx(1, "server request length below minimum accepted: "
89		    "%d vs. %d", P9P_MINREQLEN, maxreq);
90	p9p->maxreq = maxreq;
91
92	/* tell the server we don't support authentication */
93	p9pbuf_recycleout(pb);
94	tagid = NEXTTAG(p9p);
95	p9pbuf_put_1(pb, P9PROTO_T_AUTH);
96	p9pbuf_put_2(pb, tagid);
97	p9pbuf_put_4(pb, P9PROTO_NOFID);
98	p9pbuf_put_str(pb, username);
99	p9pbuf_put_str(pb, "");
100	DO_IO(p9pbuf_write, pu, pb, p9p->servsock, &done, rv);
101
102	puffs_framebuf_recycle(pb);
103	DO_IO(p9pbuf_read, pu, pb, p9p->servsock, &done, rv);
104
105	/* assume all Rerror is "no auth" */
106	if (p9pbuf_get_type(pb) != P9PROTO_R_ERROR)
107		errx(1, "mount_9p supports only NO auth");
108	if ((rtagid = p9pbuf_get_tag(pb)) != tagid)
109		errx(1, "server invalid tag: %d vs. %d", tagid, rtagid);
110
111	/* build attach message */
112	p9pbuf_recycleout(pb);
113	tagid = NEXTTAG(p9p);
114	p9pbuf_put_1(pb, P9PROTO_T_ATTACH);
115	p9pbuf_put_2(pb, tagid);
116	p9pbuf_put_4(pb, P9P_ROOTFID);
117	p9pbuf_put_4(pb, P9PROTO_NOFID);
118	p9pbuf_put_str(pb, username);
119	p9pbuf_put_str(pb, "");
120	DO_IO(p9pbuf_write, pu, pb, p9p->servsock, &done, rv);
121
122	puffs_framebuf_recycle(pb);
123	DO_IO(p9pbuf_read, pu, pb, p9p->servsock, &done, rv);
124
125	if ((type = p9pbuf_get_type(pb)) != P9PROTO_R_ATTACH)
126		errx(1, "Rattach not received, got %d", type);
127	if ((rtagid = p9pbuf_get_tag(pb)) != tagid)
128		errx(1, "server invalid tag: %d vs. %d", tagid, rtagid);
129
130	/* just walk away rootfid, you won't see me follow you back home */
131
132#define EATSLASH(p) while (*(p) == '/') p++
133	assert(*path == '/');
134	p = path;
135	EATSLASH(p);
136	for (ncomp = 0; p && *p; ncomp++) {
137		EATSLASH(p);
138		if (!*p)
139			break;
140		p = strchr(p, '/');
141	}
142
143	if (ncomp == 0) {
144		curfid = P9P_ROOTFID;
145	} else {
146		uint16_t walked;
147
148		p9pbuf_recycleout(pb);
149		tagid = NEXTTAG(p9p);
150		curfid = NEXTFID(p9p);
151		p9pbuf_put_1(pb, P9PROTO_T_WALK);
152		p9pbuf_put_2(pb, tagid);
153		p9pbuf_put_4(pb, P9P_ROOTFID);
154		p9pbuf_put_4(pb, curfid);
155		p9pbuf_put_2(pb, ncomp);
156
157		p = path;
158		while (p && *p) {
159			char *p2;
160
161			EATSLASH(p);
162			if (!*p)
163				break;
164			if ((p2 = strchr(p, '/')) == NULL)
165				p2 = strchr(p, '\0');
166			p9pbuf_put_data(pb, p, p2-p);
167			p = p2;
168		}
169
170		DO_IO(p9pbuf_write, pu, pb, p9p->servsock, &done, rv);
171
172		puffs_framebuf_recycle(pb);
173		DO_IO(p9pbuf_read, pu, pb, p9p->servsock, &done, rv);
174
175		if ((type = p9pbuf_get_type(pb)) != P9PROTO_R_WALK)
176			errx(1, "Rwalk not received for rnode, got %d", type);
177		if ((rtagid = p9pbuf_get_tag(pb)) != tagid)
178			errx(1, "server invalid tag: %d vs. %d",
179			    tagid, rtagid);
180		if (p9pbuf_get_2(pb, &walked) == -1)
181			errx(1, "can't get number of walked qids");
182		if (walked != ncomp)
183			errx(1, "can't locate rootpath %s, only %d/%d "
184			    "components found", path, walked, ncomp);
185
186		/* curfid is alive, clunk P9P_ROOTFID */
187		p9pbuf_recycleout(pb);
188		tagid = NEXTTAG(p9p);
189		p9pbuf_put_1(pb, P9PROTO_T_CLUNK);
190		p9pbuf_put_2(pb, tagid);
191		p9pbuf_put_4(pb, P9P_ROOTFID);
192
193		DO_IO(p9pbuf_write, pu, pb, p9p->servsock, &done, rv);
194		puffs_framebuf_recycle(pb);
195		DO_IO(p9pbuf_read, pu, pb, p9p->servsock, &done, rv);
196		/* wedontcare */
197	}
198
199	/* finally, stat the node */
200	p9pbuf_recycleout(pb);
201	tagid = NEXTTAG(p9p);
202	p9pbuf_put_1(pb, P9PROTO_T_STAT);
203	p9pbuf_put_2(pb, tagid);
204	p9pbuf_put_4(pb, curfid);
205	DO_IO(p9pbuf_write, pu, pb, p9p->servsock, &done, rv);
206
207	puffs_framebuf_recycle(pb);
208	DO_IO(p9pbuf_read, pu, pb, p9p->servsock, &done, rv);
209
210	if ((type = p9pbuf_get_type(pb)) != P9PROTO_R_STAT)
211		errx(1, "Rstat not received, got %d", type);
212	if ((rtagid = p9pbuf_get_tag(pb)) != tagid)
213		errx(1, "server invalid tag: %d vs. %d", tagid, rtagid);
214	if (p9pbuf_get_2(pb, &dummy))
215		errx(1, "couldn't get stat len parameter");
216	if (proto_getstat(pb, &rootva, NULL, NULL))
217		errx(1, "could not parse root attributes");
218	puffs_framebuf_destroy(pb);
219
220	rootva.va_nlink = 0156; /* guess, will be fixed with first readdir */
221	pn = newp9pnode_va(pu, &rootva, curfid);
222
223	if (ioctl(p9p->servsock, FIONBIO, &x) == -1)
224		err(1, "cannot set socket in nonblocking mode");
225
226	return pn;
227}
228
229int
230puffs9p_fs_unmount(struct puffs_usermount *pu, int flags)
231{
232	struct puffs9p *p9p = puffs_getspecific(pu);
233
234	close(p9p->servsock);
235	return 0;
236}
237