1/*	$NetBSD: smbfs_io.c,v 1.33 2009/06/22 21:13:50 njoly Exp $	*/
2
3/*
4 * Copyright (c) 2000-2001, Boris Popov
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. All advertising materials mentioning features or use of this software
16 *    must display the following acknowledgement:
17 *    This product includes software developed by Boris Popov.
18 * 4. Neither the name of the author nor the names of any co-contributors
19 *    may be used to endorse or promote products derived from this software
20 *    without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 *
34 * FreeBSD: src/sys/fs/smbfs/smbfs_io.c,v 1.7 2001/12/02 08:56:58 bp Exp
35 *
36 */
37
38#include <sys/cdefs.h>
39__KERNEL_RCSID(0, "$NetBSD: smbfs_io.c,v 1.33 2009/06/22 21:13:50 njoly Exp $");
40
41#include <sys/param.h>
42#include <sys/systm.h>
43#include <sys/resourcevar.h>	/* defines plimit structure in proc struct */
44#include <sys/kernel.h>
45#include <sys/proc.h>
46#include <sys/fcntl.h>
47#include <sys/buf.h>
48#include <sys/mount.h>
49#include <sys/malloc.h>
50#include <sys/namei.h>
51#include <sys/vnode.h>
52#include <sys/dirent.h>
53#include <sys/signalvar.h>
54#include <sys/sysctl.h>
55#include <sys/vmmeter.h>
56#include <sys/kauth.h>
57
58#ifndef __NetBSD__
59#include <vm/vm.h>
60#if __FreeBSD_version < 400000
61#include <vm/vm_prot.h>
62#endif
63#include <vm/vm_page.h>
64#include <vm/vm_extern.h>
65#include <vm/vm_object.h>
66#include <vm/vm_pager.h>
67#include <vm/vnode_pager.h>
68#else /* __NetBSD__ */
69#include <uvm/uvm.h>
70#include <uvm/uvm_extern.h>
71#endif /* __NetBSD__ */
72
73/*
74#include <sys/ioccom.h>
75*/
76#include <netsmb/smb.h>
77#include <netsmb/smb_conn.h>
78#include <netsmb/smb_subr.h>
79
80#include <fs/smbfs/smbfs.h>
81#include <fs/smbfs/smbfs_node.h>
82#include <fs/smbfs/smbfs_subr.h>
83
84/*#define SMBFS_RWGENERIC*/
85
86#define DE_SIZE	(sizeof(struct dirent))
87
88static int
89smbfs_readvdir(struct vnode *vp, struct uio *uio, kauth_cred_t cred)
90{
91	struct dirent *de;
92	struct smb_cred scred;
93	struct smbfs_fctx *ctx;
94	struct smbnode *np = VTOSMB(vp);
95	int error = 0/*, *eofflag = ap->a_eofflag*/;
96	long offset, limit;
97
98	KASSERT(vp->v_type == VDIR);
99
100	if (uio->uio_resid < DE_SIZE || uio->uio_offset < 0)
101		return EINVAL;
102
103	de = malloc(DE_SIZE, M_SMBFSDATA, M_WAITOK | M_ZERO);
104
105	SMBVDEBUG("dirname='%.*s'\n", (int) np->n_nmlen, np->n_name);
106	smb_makescred(&scred, curlwp, cred);
107	offset = uio->uio_offset / DE_SIZE; 	/* offset in the directory */
108	limit = uio->uio_resid / DE_SIZE;
109
110	/* Simulate . */
111	if (offset < 1) {
112		memset(de, 0, DE_SIZE);
113		de->d_fileno = np->n_ino;
114		de->d_reclen = DE_SIZE;
115		de->d_type = DT_DIR;
116		de->d_namlen = 1;
117		strncpy(de->d_name, ".", 2);
118		error = uiomove(de, DE_SIZE, uio);
119		if (error)
120			goto out;
121		limit--;
122		offset++;
123	}
124	/* Simulate .. */
125	if (limit > 0 && offset < 2) {
126		memset(de, 0, DE_SIZE);
127		de->d_fileno = (np->n_parent ? VTOSMB(np->n_parent)->n_ino : 2);
128		de->d_reclen = DE_SIZE;
129		de->d_type = DT_DIR;
130		de->d_namlen = 2;
131		strncpy(de->d_name, "..", 3);
132		error = uiomove(de, DE_SIZE, uio);
133		if (error)
134			goto out;
135		limit--;
136		offset++;
137	}
138
139	if (limit == 0)
140		goto out;
141
142	if (offset != np->n_dirofs || np->n_dirseq == NULL) {
143		SMBVDEBUG("Reopening search %ld:%ld\n", offset, np->n_dirofs);
144		if (np->n_dirseq) {
145			smbfs_findclose(np->n_dirseq, &scred);
146			np->n_dirseq = NULL;
147		}
148		np->n_dirofs = 2;
149		error = smbfs_findopen(np, "*", 1,
150		    SMB_FA_SYSTEM | SMB_FA_HIDDEN | SMB_FA_DIR,
151		    &scred, &ctx);
152		if (error) {
153			SMBVDEBUG("can not open search, error = %d", error);
154			goto out;
155		}
156		np->n_dirseq = ctx;
157	} else
158		ctx = np->n_dirseq;
159
160	/* skip entries before offset */
161	while (np->n_dirofs < offset) {
162		error = smbfs_findnext(ctx, offset - np->n_dirofs++, &scred);
163		if (error) {
164			smbfs_findclose(np->n_dirseq, &scred);
165			np->n_dirseq = NULL;
166			error = (error == ENOENT) ? 0 : error;
167			goto out;
168		}
169	}
170
171	for (; limit; limit--, offset++) {
172		error = smbfs_findnext(ctx, limit, &scred);
173		if (error) {
174			if (error == ENOENT)
175				error = 0;
176			break;
177		}
178		np->n_dirofs++;
179		memset(de, 0, DE_SIZE);
180		de->d_reclen = DE_SIZE;
181		de->d_fileno = ctx->f_attr.fa_ino;
182		de->d_type = (ctx->f_attr.fa_attr & SMB_FA_DIR) ?
183		    DT_DIR : DT_REG;
184		de->d_namlen = ctx->f_nmlen;
185		memcpy(de->d_name, ctx->f_name, de->d_namlen);
186		de->d_name[de->d_namlen] = '\0';
187		error = uiomove(de, DE_SIZE, uio);
188		if (error)
189			break;
190	}
191
192out:
193	free(de, M_SMBFSDATA);
194	return (error);
195}
196
197int
198smbfs_readvnode(struct vnode *vp, struct uio *uiop, kauth_cred_t cred)
199{
200	struct smbmount *smp = VFSTOSMBFS(vp->v_mount);
201	struct smbnode *np = VTOSMB(vp);
202	struct lwp *l = curlwp;
203	struct vattr vattr;
204	struct smb_cred scred;
205	int error;
206
207	KASSERT(vp->v_type == VREG || vp->v_type == VDIR);
208
209	if (uiop->uio_resid == 0)
210		return 0;
211	if (uiop->uio_offset < 0)
212		return EINVAL;
213/*	if (uiop->uio_offset + uiop->uio_resid > smp->nm_maxfilesize)
214		return EFBIG;*/
215	if (vp->v_type == VDIR) {
216		error = smbfs_readvdir(vp, uiop, cred);
217		return error;
218	}
219
220	if (np->n_flag & NMODIFIED) {
221		smbfs_attr_cacheremove(vp);
222		error = VOP_GETATTR(vp, &vattr, cred);
223		if (error)
224			return error;
225		np->n_mtime.tv_sec = vattr.va_mtime.tv_sec;
226	} else {
227		error = VOP_GETATTR(vp, &vattr, cred);
228		if (error)
229			return error;
230		if (np->n_mtime.tv_sec != vattr.va_mtime.tv_sec) {
231			error = smbfs_vinvalbuf(vp, V_SAVE, cred, l, 1);
232			if (error)
233				return error;
234			np->n_mtime.tv_sec = vattr.va_mtime.tv_sec;
235		}
236	}
237	smb_makescred(&scred, l, cred);
238	return smb_read(smp->sm_share, np->n_fid, uiop, &scred);
239}
240
241int
242smbfs_writevnode(struct vnode *vp, struct uio *uiop,
243	kauth_cred_t cred, int ioflag)
244{
245	struct smbmount *smp = VTOSMBFS(vp);
246	struct smbnode *np = VTOSMB(vp);
247	struct lwp *l = curlwp;
248	struct smb_cred scred;
249	int error = 0;
250	int extended = 0;
251	size_t resid = uiop->uio_resid;
252
253	/* vn types other than VREG unsupported */
254	KASSERT(vp->v_type == VREG);
255
256	SMBVDEBUG("ofs=%lld,resid=%zu\n",
257		(long long int) uiop->uio_offset,
258		uiop->uio_resid);
259	if (uiop->uio_offset < 0)
260		return EINVAL;
261/*	if (uiop->uio_offset + uiop->uio_resid > smp->nm_maxfilesize)
262		return (EFBIG);*/
263	if (ioflag & (IO_APPEND | IO_SYNC)) {
264		if (np->n_flag & NMODIFIED) {
265			smbfs_attr_cacheremove(vp);
266			error = smbfs_vinvalbuf(vp, V_SAVE, cred, l, 1);
267			if (error)
268				return error;
269		}
270		if (ioflag & IO_APPEND) {
271#if notyet
272			struct proc *p = curproc;
273
274			/*
275			 * File size can be changed by another client
276			 */
277			smbfs_attr_cacheremove(vp);
278			error = VOP_GETATTR(vp, &vattr, cred, td);
279			if (error)
280				return (error);
281			if (np->n_size + uiop->uio_resid >
282			    p->p_rlimit[RLIMIT_FSIZE].rlim_cur) {
283				mutex_enter(proc_lock);
284				psignal(p, SIGXFSZ);
285				mutex_exit(proc_lock);
286				return EFBIG;
287			}
288#endif
289			uiop->uio_offset = np->n_size;
290		}
291	}
292	if (uiop->uio_resid == 0)
293		return 0;
294	smb_makescred(&scred, l, cred);
295	error = smb_write(smp->sm_share, np->n_fid, uiop, &scred);
296	SMBVDEBUG("after: ofs=%lld,resid=%zu,err=%d\n",
297	    (long long int)uiop->uio_offset, uiop->uio_resid, error);
298	if (!error) {
299		if (uiop->uio_offset > np->n_size) {
300			np->n_size = uiop->uio_offset;
301			uvm_vnp_setsize(vp, np->n_size);
302			extended = 1;
303		}
304
305	}
306	if (resid > uiop->uio_resid)
307		VN_KNOTE(vp, NOTE_WRITE | (extended ? NOTE_EXTEND : 0));
308	return error;
309}
310
311/*
312 * Do an I/O operation to/from a cache block.
313 */
314int
315smbfs_doio(struct buf *bp, kauth_cred_t cr, struct lwp *l)
316{
317	struct vnode *vp = bp->b_vp;
318	struct smbmount *smp = VFSTOSMBFS(vp->v_mount);
319	struct smbnode *np = VTOSMB(vp);
320	struct uio uio, *uiop = &uio;
321	struct iovec io;
322	struct smb_cred scred;
323	int error = 0;
324
325	uiop->uio_iov = &io;
326	uiop->uio_iovcnt = 1;
327	UIO_SETUP_SYSSPACE(uiop);
328
329	smb_makescred(&scred, l, cr);
330
331	if (bp->b_flags == B_READ) {
332	    io.iov_len = uiop->uio_resid = bp->b_bcount;
333	    io.iov_base = bp->b_data;
334	    uiop->uio_rw = UIO_READ;
335	    switch (vp->v_type) {
336	      case VREG:
337		uiop->uio_offset = ((off_t)bp->b_blkno) * DEV_BSIZE;
338		error = smb_read(smp->sm_share, np->n_fid, uiop, &scred);
339		if (error)
340			break;
341		if (uiop->uio_resid) {
342			int left = uiop->uio_resid;
343			int nread = bp->b_bcount - left;
344			if (left > 0)
345			    memset((char *)bp->b_data + nread, 0, left);
346		}
347		break;
348	    default:
349		printf("smbfs_doio:  type %x unexpected\n",vp->v_type);
350		break;
351	    };
352	    if (error) {
353		bp->b_error = error;
354	    }
355	} else { /* write */
356		io.iov_len = uiop->uio_resid = bp->b_bcount;
357		uiop->uio_offset = ((off_t)bp->b_blkno) << DEV_BSHIFT;
358		io.iov_base = bp->b_data;
359		uiop->uio_rw = UIO_WRITE;
360		bp->b_cflags |= BC_BUSY;
361		error = smb_write(smp->sm_share, np->n_fid, uiop, &scred);
362		bp->b_cflags &= ~BC_BUSY;
363
364
365#ifndef __NetBSD__ /* XXX */
366		/*
367		 * For an interrupted write, the buffer is still valid
368		 * and the write hasn't been pushed to the server yet,
369		 * so we can't set BIO_ERROR and report the interruption
370		 * by setting B_EINTR. For the B_ASYNC case, B_EINTR
371		 * is not relevant, so the rpc attempt is essentially
372		 * a noop.  For the case of a V3 write rpc not being
373		 * committed to stable storage, the block is still
374		 * dirty and requires either a commit rpc or another
375		 * write rpc with iomode == NFSV3WRITE_FILESYNC before
376		 * the block is reused. This is indicated by setting
377		 * the B_DELWRI and B_NEEDCOMMIT flags.
378		 */
379    		if (error == EINTR
380		    || (!error && (bp->b_flags & B_NEEDCOMMIT))) {
381			int s;
382
383			s = splbio();
384			bp->b_flags &= ~(B_INVAL|B_NOCACHE);
385			if ((bp->b_flags & B_ASYNC) == 0)
386			    bp->b_flags |= B_EINTR;
387			if ((bp->b_flags & B_PAGING) == 0) {
388			    bdirty(bp);
389			    bp->b_flags &= ~B_DONE;
390			}
391			if ((bp->b_flags & B_ASYNC) == 0)
392			    bp->b_flags |= B_EINTR;
393			splx(s);
394	    	} else {
395			if (error) {
396				bp->b_ioflags |= BIO_ERROR;
397				bp->b_error = error;
398			}
399			bp->b_dirtyoff = bp->b_dirtyend = 0;
400		}
401#endif /* !__NetBSD__ */
402	}
403	bp->b_resid = uiop->uio_resid;
404	biodone(bp);
405	return error;
406}
407
408/*
409 * Flush and invalidate all dirty buffers. If another process is already
410 * doing the flush, just wait for completion.
411 */
412int
413smbfs_vinvalbuf(struct vnode *vp, int flags, kauth_cred_t cred, struct lwp *l, int intrflg)
414{
415	struct smbnode *np = VTOSMB(vp);
416	int error = 0, slpflag;
417
418	if (intrflg)
419		slpflag = PCATCH;
420	else
421		slpflag = 0;
422
423	while (np->n_flag & NFLUSHINPROG) {
424		np->n_flag |= NFLUSHWANT;
425		error = tsleep(&np->n_flag,
426			(PRIBIO + 2) | slpflag, "smfsvinv", 0);
427		if (error)
428			return (error);
429	}
430	np->n_flag |= NFLUSHINPROG;
431	for(;;) {
432		if ((error = vinvalbuf(vp, flags, cred, l, slpflag, 0)) == 0)
433			break;
434
435		if (intrflg && (error == ERESTART || error == EINTR)) {
436			np->n_flag &= ~NFLUSHINPROG;
437			if (np->n_flag & NFLUSHWANT) {
438				np->n_flag &= ~NFLUSHWANT;
439				wakeup(&np->n_flag);
440			}
441			return (error);
442		}
443	}
444	np->n_flag &= ~(NMODIFIED | NFLUSHINPROG);
445	if (np->n_flag & NFLUSHWANT) {
446		np->n_flag &= ~NFLUSHWANT;
447		wakeup(&np->n_flag);
448	}
449	return (error);
450}
451