xref: /dragonfly/sys/vfs/smbfs/smbfs_node.c (revision 28c7b939)
1 /*
2  * Copyright (c) 2000-2001 Boris Popov
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *    This product includes software developed by Boris Popov.
16  * 4. Neither the name of the author nor the names of any co-contributors
17  *    may be used to endorse or promote products derived from this software
18  *    without specific prior written permission.
19  *
20  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
21  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
24  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30  * SUCH DAMAGE.
31  *
32  * $FreeBSD: src/sys/fs/smbfs/smbfs_node.c,v 1.2.2.3 2003/01/17 08:20:26 tjr Exp $
33  * $DragonFly: src/sys/vfs/smbfs/smbfs_node.c,v 1.8 2003/11/10 06:12:17 dillon Exp $
34  */
35 #include <sys/param.h>
36 #include <sys/systm.h>
37 #include <sys/kernel.h>
38 #include <sys/time.h>
39 #include <sys/proc.h>
40 #include <sys/mount.h>
41 #include <sys/vnode.h>
42 #include <sys/malloc.h>
43 #include <sys/sysctl.h>
44 #include <vm/vm.h>
45 #include <vm/vm_extern.h>
46 /*#include <vm/vm_page.h>
47 #include <vm/vm_object.h>*/
48 #include <sys/queue.h>
49 
50 #include <netproto/smb/smb.h>
51 #include <netproto/smb/smb_conn.h>
52 #include <netproto/smb/smb_subr.h>
53 
54 #include "smbfs.h"
55 #include "smbfs_node.h"
56 #include "smbfs_subr.h"
57 
58 #define	SMBFS_NOHASH(smp, hval)	(&(smp)->sm_hash[(hval) & (smp)->sm_hashlen])
59 #define	smbfs_hash_lock(smp, td)	lockmgr(&smp->sm_hashlock, LK_EXCLUSIVE, NULL, td)
60 #define	smbfs_hash_unlock(smp, td)	lockmgr(&smp->sm_hashlock, LK_RELEASE, NULL, td)
61 
62 
63 extern vop_t **smbfs_vnodeop_p;
64 
65 MALLOC_DEFINE(M_SMBNODE, "SMBFS node", "SMBFS vnode private part");
66 static MALLOC_DEFINE(M_SMBNODENAME, "SMBFS nname", "SMBFS node name");
67 
68 int smbfs_hashprint(struct mount *mp);
69 
70 #if 0
71 #ifdef SYSCTL_DECL
72 SYSCTL_DECL(_vfs_smbfs);
73 #endif
74 SYSCTL_PROC(_vfs_smbfs, OID_AUTO, vnprint, CTLFLAG_WR|CTLTYPE_OPAQUE,
75 	    NULL, 0, smbfs_hashprint, "S,vnlist", "vnode hash");
76 #endif
77 
78 #define	FNV_32_PRIME ((u_int32_t) 0x01000193UL)
79 #define	FNV1_32_INIT ((u_int32_t) 33554467UL)
80 
81 u_int32_t
82 smbfs_hash(const u_char *name, int nmlen)
83 {
84 	u_int32_t v;
85 
86 	for (v = FNV1_32_INIT; nmlen; name++, nmlen--) {
87 		v *= FNV_32_PRIME;
88 		v ^= (u_int32_t)*name;
89 	}
90 	return v;
91 }
92 
93 int
94 smbfs_hashprint(struct mount *mp)
95 {
96 	struct smbmount *smp = VFSTOSMBFS(mp);
97 	struct smbnode_hashhead *nhpp;
98 	struct smbnode *np;
99 	int i;
100 
101 	for(i = 0; i <= smp->sm_hashlen; i++) {
102 		nhpp = &smp->sm_hash[i];
103 		LIST_FOREACH(np, nhpp, n_hash)
104 			vprint(NULL, SMBTOV(np));
105 	}
106 	return 0;
107 }
108 
109 static char *
110 smbfs_name_alloc(const u_char *name, int nmlen)
111 {
112 	u_char *cp;
113 
114 	nmlen++;
115 #ifdef SMBFS_NAME_DEBUG
116 	cp = malloc(nmlen + 2 + sizeof(int), M_SMBNODENAME, M_WAITOK);
117 	*(int*)cp = nmlen;
118 	cp += sizeof(int);
119 	cp[0] = 0xfc;
120 	cp++;
121 	bcopy(name, cp, nmlen - 1);
122 	cp[nmlen] = 0xfe;
123 #else
124 	cp = malloc(nmlen, M_SMBNODENAME, M_WAITOK);
125 	bcopy(name, cp, nmlen - 1);
126 #endif
127 	cp[nmlen - 1] = 0;
128 	return cp;
129 }
130 
131 static void
132 smbfs_name_free(u_char *name)
133 {
134 #ifdef SMBFS_NAME_DEBUG
135 	int nmlen, slen;
136 	u_char *cp;
137 
138 	cp = name;
139 	cp--;
140 	if (*cp != 0xfc) {
141 		printf("First byte of name entry '%s' corrupted\n", name);
142 		Debugger("ditto");
143 	}
144 	cp -= sizeof(int);
145 	nmlen = *(int*)cp;
146 	slen = strlen(name) + 1;
147 	if (nmlen != slen) {
148 		printf("Name length mismatch: was %d, now %d name '%s'\n",
149 		    nmlen, slen, name);
150 		Debugger("ditto");
151 	}
152 	if (name[nmlen] != 0xfe) {
153 		printf("Last byte of name entry '%s' corrupted\n", name);
154 		Debugger("ditto");
155 	}
156 	free(cp, M_SMBNODENAME);
157 #else
158 	free(name, M_SMBNODENAME);
159 #endif
160 }
161 
162 static int
163 smbfs_node_alloc(struct mount *mp, struct vnode *dvp,
164 	const char *name, int nmlen, struct smbfattr *fap, struct vnode **vpp)
165 {
166 	struct thread *td = curthread;	/* XXX */
167 	struct smbmount *smp = VFSTOSMBFS(mp);
168 	struct smbnode_hashhead *nhpp;
169 	struct smbnode *np, *np2, *dnp;
170 	struct vnode *vp;
171 	u_long hashval;
172 	int error;
173 
174 	*vpp = NULL;
175 	if (smp->sm_root != NULL && dvp == NULL) {
176 		SMBERROR("do not allocate root vnode twice!\n");
177 		return EINVAL;
178 	}
179 	if (nmlen == 2 && bcmp(name, "..", 2) == 0) {
180 		if (dvp == NULL)
181 			return EINVAL;
182 		vp = VTOSMB(VTOSMB(dvp)->n_parent)->n_vnode;
183 		error = vget(vp, LK_EXCLUSIVE, td);
184 		if (error == 0)
185 			*vpp = vp;
186 		return error;
187 	} else if (nmlen == 1 && name[0] == '.') {
188 		SMBERROR("do not call me with dot!\n");
189 		return EINVAL;
190 	}
191 	dnp = dvp ? VTOSMB(dvp) : NULL;
192 	if (dnp == NULL && dvp != NULL) {
193 		vprint("smbfs_node_alloc: dead parent vnode", dvp);
194 		return EINVAL;
195 	}
196 	hashval = smbfs_hash(name, nmlen);
197 retry:
198 	smbfs_hash_lock(smp, td);
199 loop:
200 	nhpp = SMBFS_NOHASH(smp, hashval);
201 	LIST_FOREACH(np, nhpp, n_hash) {
202 		vp = SMBTOV(np);
203 		if (np->n_parent != dvp ||
204 		    np->n_nmlen != nmlen || bcmp(name, np->n_name, nmlen) != 0)
205 			continue;
206 		VI_LOCK(vp);
207 		smbfs_hash_unlock(smp, td);
208 		if (vget(vp, LK_EXCLUSIVE | LK_INTERLOCK, td) != 0)
209 			goto retry;
210 		*vpp = vp;
211 		return 0;
212 	}
213 	smbfs_hash_unlock(smp, td);
214 	/*
215 	 * If we don't have node attributes, then it is an explicit lookup
216 	 * for an existing vnode.
217 	 */
218 	if (fap == NULL)
219 		return ENOENT;
220 
221 	MALLOC(np, struct smbnode *, sizeof *np, M_SMBNODE, M_WAITOK);
222 	error = getnewvnode(VT_SMBFS, mp, smbfs_vnodeop_p, &vp);
223 	if (error) {
224 		FREE(np, M_SMBNODE);
225 		return error;
226 	}
227 	vp->v_type = fap->fa_attr & SMB_FA_DIR ? VDIR : VREG;
228 	bzero(np, sizeof(*np));
229 	vp->v_data = np;
230 	np->n_vnode = vp;
231 	np->n_mount = VFSTOSMBFS(mp);
232 	np->n_nmlen = nmlen;
233 	np->n_name = smbfs_name_alloc(name, nmlen);
234 	np->n_ino = fap->fa_ino;
235 
236 	if (dvp) {
237 		np->n_parent = dvp;
238 		if (/*vp->v_type == VDIR &&*/ (dvp->v_flag & VROOT) == 0) {
239 			vref(dvp);
240 			np->n_flag |= NREFPARENT;
241 		}
242 	} else if (vp->v_type == VREG)
243 		SMBERROR("new vnode '%s' born without parent ?\n", np->n_name);
244 
245 	lockinit(&np->n_lock, 0, "smbnode", VLKTIMEOUT, LK_CANRECURSE);
246 	vn_lock(vp, LK_EXCLUSIVE | LK_RETRY, td);
247 
248 	smbfs_hash_lock(smp, td);
249 	LIST_FOREACH(np2, nhpp, n_hash) {
250 		if (np2->n_parent != dvp ||
251 		    np2->n_nmlen != nmlen || bcmp(name, np2->n_name, nmlen) != 0)
252 			continue;
253 		vput(vp);
254 /*		smb_name_free(np->n_name);
255 		FREE(np, M_SMBNODE);*/
256 		goto loop;
257 	}
258 	LIST_INSERT_HEAD(nhpp, np, n_hash);
259 	smbfs_hash_unlock(smp, td);
260 	*vpp = vp;
261 	return 0;
262 }
263 
264 int
265 smbfs_nget(struct mount *mp, struct vnode *dvp, const char *name, int nmlen,
266 	struct smbfattr *fap, struct vnode **vpp)
267 {
268 	struct smbnode *np;
269 	struct vnode *vp;
270 	int error;
271 
272 	*vpp = NULL;
273 	error = smbfs_node_alloc(mp, dvp, name, nmlen, fap, &vp);
274 	if (error)
275 		return error;
276 	np = VTOSMB(vp);
277 	if (fap)
278 		smbfs_attr_cacheenter(vp, fap);
279 	*vpp = vp;
280 	return 0;
281 }
282 
283 /*
284  * Free smbnode, and give vnode back to system
285  */
286 int
287 smbfs_reclaim(ap)
288         struct vop_reclaim_args /* {
289 		struct vnode *a_vp;
290 		struct thread *a_td;
291         } */ *ap;
292 {
293 	struct vnode *vp = ap->a_vp;
294 	struct thread *td = ap->a_td;
295 	struct vnode *dvp;
296 	struct smbnode *np = VTOSMB(vp);
297 	struct smbmount *smp = VTOSMBFS(vp);
298 
299 	SMBVDEBUG("%s,%d\n", np->n_name, vp->v_usecount);
300 
301 	smbfs_hash_lock(smp, td);
302 
303 	dvp = (np->n_parent && (np->n_flag & NREFPARENT)) ?
304 	    np->n_parent : NULL;
305 
306 	if (np->n_hash.le_prev)
307 		LIST_REMOVE(np, n_hash);
308 	cache_purge(vp);
309 	if (smp->sm_root == np) {
310 		SMBVDEBUG("root vnode\n");
311 		smp->sm_root = NULL;
312 	}
313 	vp->v_data = NULL;
314 	smbfs_hash_unlock(smp, td);
315 	if (np->n_name)
316 		smbfs_name_free(np->n_name);
317 	FREE(np, M_SMBNODE);
318 	if (dvp) {
319 		VI_LOCK(dvp);
320 		if (dvp->v_usecount >= 1) {
321 			VI_UNLOCK(dvp);
322 			vrele(dvp);
323 			/*
324 			 * Indicate that we released something; see comment
325 			 * in smbfs_unmount().
326 			 */
327 			smp->sm_didrele = 1;
328 		} else {
329 			VI_UNLOCK(dvp);
330 			SMBERROR("BUG: negative use count for parent!\n");
331 		}
332 	}
333 	return 0;
334 }
335 
336 int
337 smbfs_inactive(ap)
338 	struct vop_inactive_args /* {
339 		struct vnode *a_vp;
340 		struct thread *a_td;
341 	} */ *ap;
342 {
343 	struct thread *td = ap->a_td;
344 	struct ucred *cred;
345 	struct vnode *vp = ap->a_vp;
346 	struct smbnode *np = VTOSMB(vp);
347 	struct smb_cred scred;
348 	int error;
349 
350 	KKASSERT(td->td_proc);
351 	cred = td->td_proc->p_ucred;
352 
353 	SMBVDEBUG("%s: %d\n", VTOSMB(vp)->n_name, vp->v_usecount);
354 	if (np->n_opencount) {
355 		error = smbfs_vinvalbuf(vp, V_SAVE, td, 1);
356 		smb_makescred(&scred, td, cred);
357 		error = smbfs_smb_close(np->n_mount->sm_share, np->n_fid,
358 		   &np->n_mtime, &scred);
359 		np->n_opencount = 0;
360 	}
361 	VOP_UNLOCK(vp, 0, td);
362 	return (0);
363 }
364 /*
365  * routines to maintain vnode attributes cache
366  * smbfs_attr_cacheenter: unpack np.i to vattr structure
367  */
368 void
369 smbfs_attr_cacheenter(struct vnode *vp, struct smbfattr *fap)
370 {
371 	struct smbnode *np = VTOSMB(vp);
372 
373 	if (vp->v_type == VREG) {
374 		if (np->n_size != fap->fa_size) {
375 			np->n_size = fap->fa_size;
376 			vnode_pager_setsize(vp, np->n_size);
377 		}
378 	} else if (vp->v_type == VDIR) {
379 		np->n_size = 16384; 		/* should be a better way ... */
380 	} else
381 		return;
382 	np->n_mtime = fap->fa_mtime;
383 	np->n_dosattr = fap->fa_attr;
384 	np->n_attrage = time_second;
385 	return;
386 }
387 
388 int
389 smbfs_attr_cachelookup(struct vnode *vp, struct vattr *va)
390 {
391 	struct smbnode *np = VTOSMB(vp);
392 	struct smbmount *smp = VTOSMBFS(vp);
393 	int diff;
394 
395 	diff = time_second - np->n_attrage;
396 	if (diff > 2)	/* XXX should be configurable */
397 		return ENOENT;
398 	va->va_type = vp->v_type;		/* vnode type (for create) */
399 	if (vp->v_type == VREG) {
400 		va->va_mode = smp->sm_args.file_mode;	/* files access mode and type */
401 	} else if (vp->v_type == VDIR) {
402 		va->va_mode = smp->sm_args.dir_mode;	/* files access mode and type */
403 	} else
404 		return EINVAL;
405 	va->va_size = np->n_size;
406 	va->va_nlink = 1;		/* number of references to file */
407 	va->va_uid = smp->sm_args.uid;	/* owner user id */
408 	va->va_gid = smp->sm_args.gid;	/* owner group id */
409 	va->va_fsid = vp->v_mount->mnt_stat.f_fsid.val[0];
410 	va->va_fileid = np->n_ino;	/* file id */
411 	if (va->va_fileid == 0)
412 		va->va_fileid = 2;
413 	va->va_blocksize = SSTOVC(smp->sm_share)->vc_txmax;
414 	va->va_mtime = np->n_mtime;
415 	va->va_atime = va->va_ctime = va->va_mtime;	/* time file changed */
416 	va->va_gen = VNOVAL;		/* generation number of file */
417 	va->va_flags = 0;		/* flags defined for file */
418 	va->va_rdev = VNOVAL;		/* device the special file represents */
419 	va->va_bytes = va->va_size;	/* bytes of disk space held by file */
420 	va->va_filerev = 0;		/* file modification number */
421 	va->va_vaflags = 0;		/* operations flags */
422 	return 0;
423 }
424