2 * Copyright (c) 2000-2001, Boris Popov
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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
32 * $FreeBSD: src/sys/fs/smbfs/smbfs_io.c,v 1.3.2.3 2003/01/17 08:20:26 tjr Exp $
33 * $DragonFly: src/sys/vfs/smbfs/smbfs_io.c,v 1.10 2004/03/01 06:33:22 dillon Exp $
36 #include <sys/param.h>
37 #include <sys/systm.h>
38 #include <sys/resourcevar.h> /* defines plimit structure in proc struct */
39 #include <sys/kernel.h>
41 #include <sys/fcntl.h>
42 #include <sys/mount.h>
43 #include <sys/namei.h>
44 #include <sys/vnode.h>
45 #include <sys/dirent.h>
46 #include <sys/signalvar.h>
47 #include <sys/sysctl.h>
50 #include <vm/vm_page.h>
51 #include <vm/vm_extern.h>
52 #include <vm/vm_object.h>
53 #include <vm/vm_pager.h>
54 #include <vm/vnode_pager.h>
56 #include <sys/ioccom.h>
58 #include <netproto/smb/smb.h>
59 #include <netproto/smb/smb_conn.h>
60 #include <netproto/smb/smb_subr.h>
63 #include "smbfs_node.h"
64 #include "smbfs_subr.h"
68 /*#define SMBFS_RWGENERIC*/
70 extern int smbfs_pbuf_freecnt;
72 static int smbfs_fastlookup = 1;
74 SYSCTL_DECL(_vfs_smbfs);
75 SYSCTL_INT(_vfs_smbfs, OID_AUTO, fastlookup, CTLFLAG_RW, &smbfs_fastlookup, 0, "");
78 #define DE_SIZE (sizeof(struct dirent))
81 smbfs_readvdir(struct vnode *vp, struct uio *uio, struct ucred *cred)
84 struct componentname cn;
85 struct smb_cred scred;
86 struct smbfs_fctx *ctx;
88 struct smbnode *np = VTOSMB(vp);
89 int error/*, *eofflag = ap->a_eofflag*/;
93 SMBVDEBUG("dirname='%s'\n", np->n_name);
94 smb_makescred(&scred, uio->uio_td, cred);
95 offset = uio->uio_offset / DE_SIZE; /* offset in the directory */
96 limit = uio->uio_resid / DE_SIZE;
97 if (uio->uio_resid < DE_SIZE || uio->uio_offset < 0)
99 while (limit && offset < 2) {
101 bzero((caddr_t)&de, DE_SIZE);
102 de.d_reclen = DE_SIZE;
103 de.d_fileno = (offset == 0) ? np->n_ino :
104 (np->n_parent ? VTOSMB(np->n_parent)->n_ino : 2);
105 if (de.d_fileno == 0)
106 de.d_fileno = 0x7ffffffd + offset;
107 de.d_namlen = offset + 1;
110 de.d_name[offset + 1] = '\0';
112 error = uiomove((caddr_t)&de, DE_SIZE, uio);
116 uio->uio_offset += DE_SIZE;
120 if (offset != np->n_dirofs || np->n_dirseq == NULL) {
121 SMBVDEBUG("Reopening search %ld:%ld\n", offset, np->n_dirofs);
123 smbfs_findclose(np->n_dirseq, &scred);
127 error = smbfs_findopen(np, "*", 1,
128 SMB_FA_SYSTEM | SMB_FA_HIDDEN | SMB_FA_DIR,
131 SMBVDEBUG("can not open search, error = %d", error);
137 while (np->n_dirofs < offset) {
138 error = smbfs_findnext(ctx, offset - np->n_dirofs++, &scred);
140 smbfs_findclose(np->n_dirseq, &scred);
142 return error == ENOENT ? 0 : error;
146 for (; limit; limit--, offset++) {
147 error = smbfs_findnext(ctx, limit, &scred);
151 bzero((caddr_t)&de, DE_SIZE);
152 de.d_reclen = DE_SIZE;
153 de.d_fileno = ctx->f_attr.fa_ino;
154 de.d_type = (ctx->f_attr.fa_attr & SMB_FA_DIR) ? DT_DIR : DT_REG;
155 de.d_namlen = ctx->f_nmlen;
156 bcopy(ctx->f_name, de.d_name, de.d_namlen);
157 de.d_name[de.d_namlen] = '\0';
158 if (smbfs_fastlookup) {
159 error = smbfs_nget(vp->v_mount, vp, ctx->f_name,
160 ctx->f_nmlen, &ctx->f_attr, &newvp);
162 cn.cn_nameptr = de.d_name;
163 cn.cn_namelen = de.d_namlen;
164 cache_enter(vp, NCPNULL, newvp, &cn);
168 error = uiomove((caddr_t)&de, DE_SIZE, uio);
174 uio->uio_offset = offset * DE_SIZE;
179 smbfs_readvnode(struct vnode *vp, struct uio *uiop, struct ucred *cred)
182 struct smbmount *smp = VFSTOSMBFS(vp->v_mount);
183 struct smbnode *np = VTOSMB(vp);
185 struct smb_cred scred;
189 * Protect against method which is not supported for now
191 if (uiop->uio_segflg == UIO_NOCOPY)
194 if (vp->v_type != VREG && vp->v_type != VDIR) {
195 SMBFSERR("vn types other than VREG or VDIR are unsupported !\n");
198 if (uiop->uio_resid == 0)
200 if (uiop->uio_offset < 0)
202 /* if (uiop->uio_offset + uiop->uio_resid > smp->nm_maxfilesize)
205 if (vp->v_type == VDIR) {
206 lks = LK_EXCLUSIVE;/*lockstatus(&vp->v_lock, td);*/
207 if (lks == LK_SHARED)
208 vn_lock(vp, NULL, LK_UPGRADE | LK_RETRY, td);
209 error = smbfs_readvdir(vp, uiop, cred);
210 if (lks == LK_SHARED)
211 vn_lock(vp, NULL, LK_DOWNGRADE | LK_RETRY, td);
215 /* biosize = SSTOCN(smp->sm_share)->sc_txmax;*/
216 if (np->n_flag & NMODIFIED) {
217 smbfs_attr_cacheremove(vp);
218 error = VOP_GETATTR(vp, &vattr, td);
221 np->n_mtime.tv_sec = vattr.va_mtime.tv_sec;
223 error = VOP_GETATTR(vp, &vattr, td);
226 if (np->n_mtime.tv_sec != vattr.va_mtime.tv_sec) {
227 error = smbfs_vinvalbuf(vp, V_SAVE, td, 1);
230 np->n_mtime.tv_sec = vattr.va_mtime.tv_sec;
233 smb_makescred(&scred, td, cred);
234 return smb_read(smp->sm_share, np->n_fid, uiop, &scred);
238 smbfs_writevnode(struct vnode *vp, struct uio *uiop,
239 struct ucred *cred, int ioflag)
242 struct smbmount *smp = VTOSMBFS(vp);
243 struct smbnode *np = VTOSMB(vp);
244 struct smb_cred scred;
247 if (vp->v_type != VREG) {
248 SMBERROR("vn types other than VREG unsupported !\n");
251 SMBVDEBUG("ofs=%d,resid=%d\n",(int)uiop->uio_offset, uiop->uio_resid);
252 if (uiop->uio_offset < 0)
254 /* if (uiop->uio_offset + uiop->uio_resid > smp->nm_maxfilesize)
257 if (ioflag & (IO_APPEND | IO_SYNC)) {
258 if (np->n_flag & NMODIFIED) {
259 smbfs_attr_cacheremove(vp);
260 error = smbfs_vinvalbuf(vp, V_SAVE, td, 1);
264 if (ioflag & IO_APPEND) {
267 * File size can be changed by another client
269 smbfs_attr_cacheremove(vp);
270 error = VOP_GETATTR(vp, &vattr, td);
271 if (error) return (error);
273 uiop->uio_offset = np->n_size;
276 if (uiop->uio_resid == 0)
279 uiop->uio_offset + uiop->uio_resid >
280 td->td_proc->p_rlimit[RLIMIT_FSIZE].rlim_cur) {
281 psignal(td->td_proc, SIGXFSZ);
284 smb_makescred(&scred, td, cred);
285 error = smb_write(smp->sm_share, np->n_fid, uiop, &scred);
286 SMBVDEBUG("after: ofs=%d,resid=%d\n",(int)uiop->uio_offset, uiop->uio_resid);
288 if (uiop->uio_offset > np->n_size) {
289 np->n_size = uiop->uio_offset;
290 vnode_pager_setsize(vp, np->n_size);
297 * Do an I/O operation to/from a cache block.
300 smbfs_doio(struct buf *bp, struct ucred *cr, struct thread *td)
302 struct vnode *vp = bp->b_vp;
303 struct smbmount *smp = VFSTOSMBFS(vp->v_mount);
304 struct smbnode *np = VTOSMB(vp);
305 struct uio uio, *uiop = &uio;
307 struct smb_cred scred;
311 uiop->uio_iovcnt = 1;
312 uiop->uio_segflg = UIO_SYSSPACE;
315 smb_makescred(&scred, td, cr);
317 if (bp->b_flags & B_READ) {
318 io.iov_len = uiop->uio_resid = bp->b_bcount;
319 io.iov_base = bp->b_data;
320 uiop->uio_rw = UIO_READ;
321 switch (vp->v_type) {
323 uiop->uio_offset = ((off_t)bp->b_blkno) * DEV_BSIZE;
324 error = smb_read(smp->sm_share, np->n_fid, uiop, &scred);
327 if (uiop->uio_resid) {
328 int left = uiop->uio_resid;
329 int nread = bp->b_bcount - left;
331 bzero((char *)bp->b_data + nread, left);
335 printf("smbfs_doio: type %x unexpected\n",vp->v_type);
340 bp->b_flags |= B_ERROR;
343 if (((bp->b_blkno * DEV_BSIZE) + bp->b_dirtyend) > np->n_size)
344 bp->b_dirtyend = np->n_size - (bp->b_blkno * DEV_BSIZE);
346 if (bp->b_dirtyend > bp->b_dirtyoff) {
347 io.iov_len = uiop->uio_resid = bp->b_dirtyend - bp->b_dirtyoff;
348 uiop->uio_offset = ((off_t)bp->b_blkno) * DEV_BSIZE + bp->b_dirtyoff;
349 io.iov_base = (char *)bp->b_data + bp->b_dirtyoff;
350 uiop->uio_rw = UIO_WRITE;
351 bp->b_flags |= B_WRITEINPROG;
352 error = smb_write(smp->sm_share, np->n_fid, uiop, &scred);
353 bp->b_flags &= ~B_WRITEINPROG;
356 * For an interrupted write, the buffer is still valid
357 * and the write hasn't been pushed to the server yet,
358 * so we can't set BIO_ERROR and report the interruption
359 * by setting B_EINTR. For the B_ASYNC case, B_EINTR
360 * is not relevant, so the rpc attempt is essentially
361 * a noop. For the case of a V3 write rpc not being
362 * committed to stable storage, the block is still
363 * dirty and requires either a commit rpc or another
364 * write rpc with iomode == NFSV3WRITE_FILESYNC before
365 * the block is reused. This is indicated by setting
366 * the B_DELWRI and B_NEEDCOMMIT flags.
369 || (!error && (bp->b_flags & B_NEEDCOMMIT))) {
373 bp->b_flags &= ~(B_INVAL|B_NOCACHE);
374 if ((bp->b_flags & B_ASYNC) == 0)
375 bp->b_flags |= B_EINTR;
376 if ((bp->b_flags & B_PAGING) == 0) {
378 bp->b_flags &= ~B_DONE;
380 if ((bp->b_flags & B_ASYNC) == 0)
381 bp->b_flags |= B_EINTR;
385 bp->b_flags |= B_ERROR;
388 bp->b_dirtyoff = bp->b_dirtyend = 0;
396 bp->b_resid = uiop->uio_resid;
402 * Vnode op for VM getpages.
403 * Wish wish .... get rid from multiple IO routines
407 struct vop_getpages_args /* {
412 vm_ooffset_t a_offset;
415 #ifdef SMBFS_RWGENERIC
416 return vnode_pager_generic_getpages(ap->a_vp, ap->a_m, ap->a_count,
419 int i, error, nextoff, size, toff, npages, count;
425 struct thread *td = curthread; /* XXX */
427 struct smbmount *smp;
429 struct smb_cred scred;
432 KKASSERT(td->td_proc);
435 cred = td->td_proc->p_ucred;
437 smp = VFSTOSMBFS(vp->v_mount);
441 if (vp->v_object == NULL) {
442 printf("smbfs_getpages: called with non-merged cache vnode??\n");
443 return VM_PAGER_ERROR;
445 smb_makescred(&scred, td, cred);
447 bp = getpbuf(&smbfs_pbuf_freecnt);
448 npages = btoc(count);
449 kva = (vm_offset_t) bp->b_data;
450 pmap_qenter(kva, pages, npages);
452 iov.iov_base = (caddr_t) kva;
456 uio.uio_offset = IDX_TO_OFF(pages[0]->pindex);
457 uio.uio_resid = count;
458 uio.uio_segflg = UIO_SYSSPACE;
459 uio.uio_rw = UIO_READ;
462 error = smb_read(smp->sm_share, np->n_fid, &uio, &scred);
463 pmap_qremove(kva, npages);
465 relpbuf(bp, &smbfs_pbuf_freecnt);
467 if (error && (uio.uio_resid == count)) {
468 printf("smbfs_getpages: error %d\n",error);
469 for (i = 0; i < npages; i++) {
470 if (ap->a_reqpage != i)
471 vnode_pager_freepage(pages[i]);
473 return VM_PAGER_ERROR;
476 size = count - uio.uio_resid;
478 for (i = 0, toff = 0; i < npages; i++, toff = nextoff) {
480 nextoff = toff + PAGE_SIZE;
483 m->flags &= ~PG_ZERO;
485 if (nextoff <= size) {
486 m->valid = VM_PAGE_BITS_ALL;
489 int nvalid = ((size + DEV_BSIZE - 1) - toff) & ~(DEV_BSIZE - 1);
490 vm_page_set_validclean(m, 0, nvalid);
493 if (i != ap->a_reqpage) {
495 * Whether or not to leave the page activated is up in
496 * the air, but we should put the page on a page queue
497 * somewhere (it already is in the object). Result:
498 * It appears that emperical results show that
499 * deactivating pages is best.
503 * Just in case someone was asking for this page we
504 * now tell them that it is ok to use.
507 if (m->flags & PG_WANTED)
510 vm_page_deactivate(m);
513 vnode_pager_freepage(m);
518 #endif /* SMBFS_RWGENERIC */
522 * Vnode op for VM putpages.
523 * possible bug: all IO done in sync mode
524 * Note that vop_close always invalidate pages before close, so it's
525 * not necessary to open vnode.
529 struct vop_putpages_args /* {
535 vm_ooffset_t a_offset;
539 struct vnode *vp = ap->a_vp;
540 struct thread *td = curthread; /* XXX */
543 #ifdef SMBFS_RWGENERIC
544 KKASSERT(td->td_proc);
545 cred = td->td_proc->p_ucred;
546 VOP_OPEN(vp, FWRITE, cred, td);
547 error = vnode_pager_generic_putpages(ap->a_vp, ap->a_m, ap->a_count,
548 ap->a_sync, ap->a_rtvals);
549 VOP_CLOSE(vp, FWRITE, cred, td);
556 int i, npages, count;
558 struct smbmount *smp;
560 struct smb_cred scred;
563 KKASSERT(td->td_proc);
564 cred = td->td_proc->p_ucred;
565 /* VOP_OPEN(vp, FWRITE, cred, td);*/
567 smp = VFSTOSMBFS(vp->v_mount);
570 rtvals = ap->a_rtvals;
571 npages = btoc(count);
573 for (i = 0; i < npages; i++) {
574 rtvals[i] = VM_PAGER_AGAIN;
577 bp = getpbuf(&smbfs_pbuf_freecnt);
578 kva = (vm_offset_t) bp->b_data;
579 pmap_qenter(kva, pages, npages);
581 iov.iov_base = (caddr_t) kva;
585 uio.uio_offset = IDX_TO_OFF(pages[0]->pindex);
586 uio.uio_resid = count;
587 uio.uio_segflg = UIO_SYSSPACE;
588 uio.uio_rw = UIO_WRITE;
590 SMBVDEBUG("ofs=%d,resid=%d\n",(int)uio.uio_offset, uio.uio_resid);
592 smb_makescred(&scred, td, cred);
593 error = smb_write(smp->sm_share, np->n_fid, &uio, &scred);
594 /* VOP_CLOSE(vp, FWRITE, cred, td);*/
595 SMBVDEBUG("paged write done: %d\n", error);
597 pmap_qremove(kva, npages);
598 relpbuf(bp, &smbfs_pbuf_freecnt);
601 int nwritten = round_page(count - uio.uio_resid) / PAGE_SIZE;
602 for (i = 0; i < nwritten; i++) {
603 rtvals[i] = VM_PAGER_OK;
608 #endif /* SMBFS_RWGENERIC */
612 * Flush and invalidate all dirty buffers. If another process is already
613 * doing the flush, just wait for completion.
616 smbfs_vinvalbuf(vp, flags, td, intrflg)
622 struct smbnode *np = VTOSMB(vp);
623 int error = 0, slpflag, slptimeo;
625 if (vp->v_flag & VXLOCK)
634 while (np->n_flag & NFLUSHINPROG) {
635 np->n_flag |= NFLUSHWANT;
636 error = tsleep((caddr_t)&np->n_flag, 0, "smfsvinv", slptimeo);
637 error = smb_proc_intr(td);
638 if (error == EINTR && intrflg)
641 np->n_flag |= NFLUSHINPROG;
642 error = vinvalbuf(vp, flags, td, slpflag, 0);
644 if (intrflg && (error == ERESTART || error == EINTR)) {
645 np->n_flag &= ~NFLUSHINPROG;
646 if (np->n_flag & NFLUSHWANT) {
647 np->n_flag &= ~NFLUSHWANT;
648 wakeup((caddr_t)&np->n_flag);
652 error = vinvalbuf(vp, flags, td, slpflag, 0);
654 np->n_flag &= ~(NMODIFIED | NFLUSHINPROG);
655 if (np->n_flag & NFLUSHWANT) {
656 np->n_flag &= ~NFLUSHWANT;
657 wakeup((caddr_t)&np->n_flag);