]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - sys/nfsclient/nfs_bio.c
- The VI_DOOMED flag now signals the end of a vnode's relationship with
[FreeBSD/FreeBSD.git] / sys / nfsclient / nfs_bio.c
1 /*-
2  * Copyright (c) 1989, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * This code is derived from software contributed to Berkeley by
6  * Rick Macklem at The University of Guelph.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 4. Neither the name of the University nor the names of its 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 REGENTS 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 REGENTS 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  *      @(#)nfs_bio.c   8.9 (Berkeley) 3/30/95
33  */
34
35 #include <sys/cdefs.h>
36 __FBSDID("$FreeBSD$");
37
38 #include <sys/param.h>
39 #include <sys/systm.h>
40 #include <sys/bio.h>
41 #include <sys/buf.h>
42 #include <sys/kernel.h>
43 #include <sys/mount.h>
44 #include <sys/proc.h>
45 #include <sys/resourcevar.h>
46 #include <sys/signalvar.h>
47 #include <sys/vmmeter.h>
48 #include <sys/vnode.h>
49
50 #include <vm/vm.h>
51 #include <vm/vm_extern.h>
52 #include <vm/vm_page.h>
53 #include <vm/vm_object.h>
54 #include <vm/vm_pager.h>
55 #include <vm/vnode_pager.h>
56
57 #include <rpc/rpcclnt.h>
58
59 #include <nfs/rpcv2.h>
60 #include <nfs/nfsproto.h>
61 #include <nfsclient/nfs.h>
62 #include <nfsclient/nfsmount.h>
63 #include <nfsclient/nfsnode.h>
64
65 #include <nfs4client/nfs4.h>
66
67 static struct buf *nfs_getcacheblk(struct vnode *vp, daddr_t bn, int size,
68                     struct thread *td);
69 static int nfs_directio_write(struct vnode *vp, struct uio *uiop, 
70                               struct ucred *cred, int ioflag);
71
72 extern int nfs_directio_enable;
73 extern int nfs_directio_allow_mmap;
74 /*
75  * Vnode op for VM getpages.
76  */
77 int
78 nfs_getpages(struct vop_getpages_args *ap)
79 {
80         int i, error, nextoff, size, toff, count, npages;
81         struct uio uio;
82         struct iovec iov;
83         vm_offset_t kva;
84         struct buf *bp;
85         struct vnode *vp;
86         struct thread *td;
87         struct ucred *cred;
88         struct nfsmount *nmp;
89         vm_object_t object;
90         vm_page_t *pages;
91         struct nfsnode *np;
92
93         GIANT_REQUIRED;
94
95         vp = ap->a_vp;
96         np = VTONFS(vp);
97         td = curthread;                         /* XXX */
98         cred = curthread->td_ucred;             /* XXX */
99         nmp = VFSTONFS(vp->v_mount);
100         pages = ap->a_m;
101         count = ap->a_count;
102
103         if ((object = vp->v_object) == NULL) {
104                 printf("nfs_getpages: called with non-merged cache vnode??\n");
105                 return VM_PAGER_ERROR;
106         }
107
108         if (!nfs_directio_allow_mmap && (np->n_flag & NNONCACHE) && 
109             (vp->v_type == VREG)) {
110                 printf("nfs_getpages: called on non-cacheable vnode??\n");
111                 return VM_PAGER_ERROR;
112         }
113
114         if ((nmp->nm_flag & NFSMNT_NFSV3) != 0 &&
115             (nmp->nm_state & NFSSTA_GOTFSINFO) == 0) {
116                 /* We'll never get here for v4, because we always have fsinfo */
117                 (void)nfs_fsinfo(nmp, vp, cred, td);
118         }
119
120         npages = btoc(count);
121
122         /*
123          * If the requested page is partially valid, just return it and
124          * allow the pager to zero-out the blanks.  Partially valid pages
125          * can only occur at the file EOF.
126          */
127
128         {
129                 vm_page_t m = pages[ap->a_reqpage];
130
131                 VM_OBJECT_LOCK(object);
132                 vm_page_lock_queues();
133                 if (m->valid != 0) {
134                         /* handled by vm_fault now        */
135                         /* vm_page_zero_invalid(m, TRUE); */
136                         for (i = 0; i < npages; ++i) {
137                                 if (i != ap->a_reqpage)
138                                         vm_page_free(pages[i]);
139                         }
140                         vm_page_unlock_queues();
141                         VM_OBJECT_UNLOCK(object);
142                         return(0);
143                 }
144                 vm_page_unlock_queues();
145                 VM_OBJECT_UNLOCK(object);
146         }
147
148         /*
149          * We use only the kva address for the buffer, but this is extremely
150          * convienient and fast.
151          */
152         bp = getpbuf(&nfs_pbuf_freecnt);
153
154         kva = (vm_offset_t) bp->b_data;
155         pmap_qenter(kva, pages, npages);
156         cnt.v_vnodein++;
157         cnt.v_vnodepgsin += npages;
158
159         iov.iov_base = (caddr_t) kva;
160         iov.iov_len = count;
161         uio.uio_iov = &iov;
162         uio.uio_iovcnt = 1;
163         uio.uio_offset = IDX_TO_OFF(pages[0]->pindex);
164         uio.uio_resid = count;
165         uio.uio_segflg = UIO_SYSSPACE;
166         uio.uio_rw = UIO_READ;
167         uio.uio_td = td;
168
169         error = (nmp->nm_rpcops->nr_readrpc)(vp, &uio, cred);
170         pmap_qremove(kva, npages);
171
172         relpbuf(bp, &nfs_pbuf_freecnt);
173
174         if (error && (uio.uio_resid == count)) {
175                 printf("nfs_getpages: error %d\n", error);
176                 VM_OBJECT_LOCK(object);
177                 vm_page_lock_queues();
178                 for (i = 0; i < npages; ++i) {
179                         if (i != ap->a_reqpage)
180                                 vm_page_free(pages[i]);
181                 }
182                 vm_page_unlock_queues();
183                 VM_OBJECT_UNLOCK(object);
184                 return VM_PAGER_ERROR;
185         }
186
187         /*
188          * Calculate the number of bytes read and validate only that number
189          * of bytes.  Note that due to pending writes, size may be 0.  This
190          * does not mean that the remaining data is invalid!
191          */
192
193         size = count - uio.uio_resid;
194         VM_OBJECT_LOCK(object);
195         vm_page_lock_queues();
196         for (i = 0, toff = 0; i < npages; i++, toff = nextoff) {
197                 vm_page_t m;
198                 nextoff = toff + PAGE_SIZE;
199                 m = pages[i];
200
201                 if (nextoff <= size) {
202                         /*
203                          * Read operation filled an entire page
204                          */
205                         m->valid = VM_PAGE_BITS_ALL;
206                         vm_page_undirty(m);
207                 } else if (size > toff) {
208                         /*
209                          * Read operation filled a partial page.
210                          */
211                         m->valid = 0;
212                         vm_page_set_validclean(m, 0, size - toff);
213                         /* handled by vm_fault now        */
214                         /* vm_page_zero_invalid(m, TRUE); */
215                 } else {
216                         /*
217                          * Read operation was short.  If no error occured
218                          * we may have hit a zero-fill section.   We simply
219                          * leave valid set to 0.
220                          */
221                         ;
222                 }
223                 if (i != ap->a_reqpage) {
224                         /*
225                          * Whether or not to leave the page activated is up in
226                          * the air, but we should put the page on a page queue
227                          * somewhere (it already is in the object).  Result:
228                          * It appears that emperical results show that
229                          * deactivating pages is best.
230                          */
231
232                         /*
233                          * Just in case someone was asking for this page we
234                          * now tell them that it is ok to use.
235                          */
236                         if (!error) {
237                                 if (m->flags & PG_WANTED)
238                                         vm_page_activate(m);
239                                 else
240                                         vm_page_deactivate(m);
241                                 vm_page_wakeup(m);
242                         } else {
243                                 vm_page_free(m);
244                         }
245                 }
246         }
247         vm_page_unlock_queues();
248         VM_OBJECT_UNLOCK(object);
249         return 0;
250 }
251
252 /*
253  * Vnode op for VM putpages.
254  */
255 int
256 nfs_putpages(struct vop_putpages_args *ap)
257 {
258         struct uio uio;
259         struct iovec iov;
260         vm_offset_t kva;
261         struct buf *bp;
262         int iomode, must_commit, i, error, npages, count;
263         off_t offset;
264         int *rtvals;
265         struct vnode *vp;
266         struct thread *td;
267         struct ucred *cred;
268         struct nfsmount *nmp;
269         struct nfsnode *np;
270         vm_page_t *pages;
271
272         GIANT_REQUIRED;
273
274         vp = ap->a_vp;
275         np = VTONFS(vp);
276         td = curthread;                         /* XXX */
277         cred = curthread->td_ucred;             /* XXX */
278         nmp = VFSTONFS(vp->v_mount);
279         pages = ap->a_m;
280         count = ap->a_count;
281         rtvals = ap->a_rtvals;
282         npages = btoc(count);
283         offset = IDX_TO_OFF(pages[0]->pindex);
284
285         if ((nmp->nm_flag & NFSMNT_NFSV3) != 0 &&
286             (nmp->nm_state & NFSSTA_GOTFSINFO) == 0) {
287                 (void)nfs_fsinfo(nmp, vp, cred, td);
288         }
289
290         if (!nfs_directio_allow_mmap && (np->n_flag & NNONCACHE) && 
291             (vp->v_type == VREG))
292                 printf("nfs_putpages: called on noncache-able vnode??\n");
293
294         for (i = 0; i < npages; i++)
295                 rtvals[i] = VM_PAGER_AGAIN;
296
297         /*
298          * When putting pages, do not extend file past EOF.
299          */
300
301         if (offset + count > np->n_size) {
302                 count = np->n_size - offset;
303                 if (count < 0)
304                         count = 0;
305         }
306
307         /*
308          * We use only the kva address for the buffer, but this is extremely
309          * convienient and fast.
310          */
311         bp = getpbuf(&nfs_pbuf_freecnt);
312
313         kva = (vm_offset_t) bp->b_data;
314         pmap_qenter(kva, pages, npages);
315         cnt.v_vnodeout++;
316         cnt.v_vnodepgsout += count;
317
318         iov.iov_base = (caddr_t) kva;
319         iov.iov_len = count;
320         uio.uio_iov = &iov;
321         uio.uio_iovcnt = 1;
322         uio.uio_offset = offset;
323         uio.uio_resid = count;
324         uio.uio_segflg = UIO_SYSSPACE;
325         uio.uio_rw = UIO_WRITE;
326         uio.uio_td = td;
327
328         if ((ap->a_sync & VM_PAGER_PUT_SYNC) == 0)
329             iomode = NFSV3WRITE_UNSTABLE;
330         else
331             iomode = NFSV3WRITE_FILESYNC;
332
333         error = (nmp->nm_rpcops->nr_writerpc)(vp, &uio, cred, &iomode, &must_commit);
334
335         pmap_qremove(kva, npages);
336         relpbuf(bp, &nfs_pbuf_freecnt);
337
338         if (!error) {
339                 int nwritten = round_page(count - uio.uio_resid) / PAGE_SIZE;
340                 for (i = 0; i < nwritten; i++) {
341                         rtvals[i] = VM_PAGER_OK;
342                         vm_page_undirty(pages[i]);
343                 }
344                 if (must_commit) {
345                         nfs_clearcommit(vp->v_mount);
346                 }
347         }
348         return rtvals[0];
349 }
350
351 /*
352  * Vnode op for read using bio
353  */
354 int
355 nfs_bioread(struct vnode *vp, struct uio *uio, int ioflag, struct ucred *cred)
356 {
357         struct nfsnode *np = VTONFS(vp);
358         int biosize, i;
359         struct buf *bp = 0, *rabp;
360         struct vattr vattr;
361         struct thread *td;
362         struct nfsmount *nmp = VFSTONFS(vp->v_mount);
363         daddr_t lbn, rabn;
364         int bcount;
365         int seqcount;
366         int nra, error = 0, n = 0, on = 0;
367
368 #ifdef DIAGNOSTIC
369         if (uio->uio_rw != UIO_READ)
370                 panic("nfs_read mode");
371 #endif
372         if (uio->uio_resid == 0)
373                 return (0);
374         if (uio->uio_offset < 0)        /* XXX VDIR cookies can be negative */
375                 return (EINVAL);
376         td = uio->uio_td;
377
378         if ((nmp->nm_flag & NFSMNT_NFSV3) != 0 &&
379             (nmp->nm_state & NFSSTA_GOTFSINFO) == 0)
380                 (void)nfs_fsinfo(nmp, vp, cred, td);
381         if (vp->v_type != VDIR &&
382             (uio->uio_offset + uio->uio_resid) > nmp->nm_maxfilesize)
383                 return (EFBIG);
384
385         if (nfs_directio_enable && (ioflag & IO_DIRECT) && (vp->v_type == VREG))
386                 /* No caching/ no readaheads. Just read data into the user buffer */
387                 return nfs_readrpc(vp, uio, cred);
388
389         biosize = vp->v_mount->mnt_stat.f_iosize;
390         seqcount = (int)((off_t)(ioflag >> IO_SEQSHIFT) * biosize / BKVASIZE);
391         /*
392          * For nfs, cache consistency can only be maintained approximately.
393          * Although RFC1094 does not specify the criteria, the following is
394          * believed to be compatible with the reference port.
395          * For nfs:
396          * If the file's modify time on the server has changed since the
397          * last read rpc or you have written to the file,
398          * you may have lost data cache consistency with the
399          * server, so flush all of the file's data out of the cache.
400          * Then force a getattr rpc to ensure that you have up to date
401          * attributes.
402          * NB: This implies that cache data can be read when up to
403          * NFS_ATTRTIMEO seconds out of date. If you find that you need current
404          * attributes this could be forced by setting n_attrstamp to 0 before
405          * the VOP_GETATTR() call.
406          */
407         if (np->n_flag & NMODIFIED) {
408                 if (vp->v_type != VREG) {
409                         if (vp->v_type != VDIR)
410                                 panic("nfs: bioread, not dir");
411                         (nmp->nm_rpcops->nr_invaldir)(vp);
412                         error = nfs_vinvalbuf(vp, V_SAVE, td, 1);
413                         if (error)
414                                 return (error);
415                 }
416                 np->n_attrstamp = 0;
417                 error = VOP_GETATTR(vp, &vattr, cred, td);
418                 if (error)
419                         return (error);
420                 np->n_mtime = vattr.va_mtime;
421         } else {
422                 error = VOP_GETATTR(vp, &vattr, cred, td);
423                 if (error)
424                         return (error);
425                 if ((np->n_flag & NSIZECHANGED)
426                     || (NFS_TIMESPEC_COMPARE(&np->n_mtime, &vattr.va_mtime))) {
427                         if (vp->v_type == VDIR)
428                                 (nmp->nm_rpcops->nr_invaldir)(vp);
429                         error = nfs_vinvalbuf(vp, V_SAVE, td, 1);
430                         if (error)
431                                 return (error);
432                         np->n_mtime = vattr.va_mtime;
433                         np->n_flag &= ~NSIZECHANGED;
434                 }
435         }
436         do {
437             switch (vp->v_type) {
438             case VREG:
439                 nfsstats.biocache_reads++;
440                 lbn = uio->uio_offset / biosize;
441                 on = uio->uio_offset & (biosize - 1);
442
443                 /*
444                  * Start the read ahead(s), as required.
445                  * The readahead is kicked off only if sequential access
446                  * is detected, based on the readahead hint (ra_expect_lbn).
447                  */
448                 if (nmp->nm_readahead > 0 && np->ra_expect_lbn == lbn) {
449                     for (nra = 0; nra < nmp->nm_readahead && nra < seqcount &&
450                         (off_t)(lbn + 1 + nra) * biosize < np->n_size; nra++) {
451                         rabn = lbn + 1 + nra;
452                         if (incore(&vp->v_bufobj, rabn) == NULL) {
453                             rabp = nfs_getcacheblk(vp, rabn, biosize, td);
454                             if (!rabp) {
455                                 error = nfs_sigintr(nmp, NULL, td);
456                                 return (error ? error : EINTR);
457                             }
458                             if ((rabp->b_flags & (B_CACHE|B_DELWRI)) == 0) {
459                                 rabp->b_flags |= B_ASYNC;
460                                 rabp->b_iocmd = BIO_READ;
461                                 vfs_busy_pages(rabp, 0);
462                                 if (nfs_asyncio(nmp, rabp, cred, td)) {
463                                     rabp->b_flags |= B_INVAL;
464                                     rabp->b_ioflags |= BIO_ERROR;
465                                     vfs_unbusy_pages(rabp);
466                                     brelse(rabp);
467                                     break;
468                                 }
469                             } else {
470                                 brelse(rabp);
471                             }
472                         }
473                     }
474                     np->ra_expect_lbn = lbn + 1;
475                 }
476
477                 /*
478                  * Obtain the buffer cache block.  Figure out the buffer size
479                  * when we are at EOF.  If we are modifying the size of the
480                  * buffer based on an EOF condition we need to hold
481                  * nfs_rslock() through obtaining the buffer to prevent
482                  * a potential writer-appender from messing with n_size.
483                  * Otherwise we may accidently truncate the buffer and
484                  * lose dirty data.
485                  *
486                  * Note that bcount is *not* DEV_BSIZE aligned.
487                  */
488
489 again:
490                 bcount = biosize;
491                 if ((off_t)lbn * biosize >= np->n_size) {
492                         bcount = 0;
493                 } else if ((off_t)(lbn + 1) * biosize > np->n_size) {
494                         bcount = np->n_size - (off_t)lbn * biosize;
495                 }
496                 if (bcount != biosize) {
497                         switch(nfs_rslock(np, td)) {
498                         case ENOLCK:
499                                 goto again;
500                                 /* not reached */
501                         case EIO:
502                                 return (EIO);
503                         case EINTR:
504                         case ERESTART:
505                                 return(EINTR);
506                                 /* not reached */
507                         default:
508                                 break;
509                         }
510                 }
511
512                 bp = nfs_getcacheblk(vp, lbn, bcount, td);
513
514                 if (bcount != biosize)
515                         nfs_rsunlock(np, td);
516                 if (!bp) {
517                         error = nfs_sigintr(nmp, NULL, td);
518                         return (error ? error : EINTR);
519                 }
520
521                 /*
522                  * If B_CACHE is not set, we must issue the read.  If this
523                  * fails, we return an error.
524                  */
525
526                 if ((bp->b_flags & B_CACHE) == 0) {
527                     bp->b_iocmd = BIO_READ;
528                     vfs_busy_pages(bp, 0);
529                     error = nfs_doio(vp, bp, cred, td);
530                     if (error) {
531                         brelse(bp);
532                         return (error);
533                     }
534                 }
535
536                 /*
537                  * on is the offset into the current bp.  Figure out how many
538                  * bytes we can copy out of the bp.  Note that bcount is
539                  * NOT DEV_BSIZE aligned.
540                  *
541                  * Then figure out how many bytes we can copy into the uio.
542                  */
543
544                 n = 0;
545                 if (on < bcount)
546                         n = min((unsigned)(bcount - on), uio->uio_resid);
547                 break;
548             case VLNK:
549                 nfsstats.biocache_readlinks++;
550                 bp = nfs_getcacheblk(vp, (daddr_t)0, NFS_MAXPATHLEN, td);
551                 if (!bp) {
552                         error = nfs_sigintr(nmp, NULL, td);
553                         return (error ? error : EINTR);
554                 }
555                 if ((bp->b_flags & B_CACHE) == 0) {
556                     bp->b_iocmd = BIO_READ;
557                     vfs_busy_pages(bp, 0);
558                     error = nfs_doio(vp, bp, cred, td);
559                     if (error) {
560                         bp->b_ioflags |= BIO_ERROR;
561                         brelse(bp);
562                         return (error);
563                     }
564                 }
565                 n = min(uio->uio_resid, NFS_MAXPATHLEN - bp->b_resid);
566                 on = 0;
567                 break;
568             case VDIR:
569                 nfsstats.biocache_readdirs++;
570                 if (np->n_direofoffset
571                     && uio->uio_offset >= np->n_direofoffset) {
572                     return (0);
573                 }
574                 lbn = (uoff_t)uio->uio_offset / NFS_DIRBLKSIZ;
575                 on = uio->uio_offset & (NFS_DIRBLKSIZ - 1);
576                 bp = nfs_getcacheblk(vp, lbn, NFS_DIRBLKSIZ, td);
577                 if (!bp) {
578                     error = nfs_sigintr(nmp, NULL, td);
579                     return (error ? error : EINTR);
580                 }
581                 if ((bp->b_flags & B_CACHE) == 0) {
582                     bp->b_iocmd = BIO_READ;
583                     vfs_busy_pages(bp, 0);
584                     error = nfs_doio(vp, bp, cred, td);
585                     if (error) {
586                             brelse(bp);
587                     }
588                     while (error == NFSERR_BAD_COOKIE) {
589                         (nmp->nm_rpcops->nr_invaldir)(vp);
590                         error = nfs_vinvalbuf(vp, 0, td, 1);
591                         /*
592                          * Yuck! The directory has been modified on the
593                          * server. The only way to get the block is by
594                          * reading from the beginning to get all the
595                          * offset cookies.
596                          *
597                          * Leave the last bp intact unless there is an error.
598                          * Loop back up to the while if the error is another
599                          * NFSERR_BAD_COOKIE (double yuch!).
600                          */
601                         for (i = 0; i <= lbn && !error; i++) {
602                             if (np->n_direofoffset
603                                 && (i * NFS_DIRBLKSIZ) >= np->n_direofoffset)
604                                     return (0);
605                             bp = nfs_getcacheblk(vp, i, NFS_DIRBLKSIZ, td);
606                             if (!bp) {
607                                 error = nfs_sigintr(nmp, NULL, td);
608                                 return (error ? error : EINTR);
609                             }
610                             if ((bp->b_flags & B_CACHE) == 0) {
611                                     bp->b_iocmd = BIO_READ;
612                                     vfs_busy_pages(bp, 0);
613                                     error = nfs_doio(vp, bp, cred, td);
614                                     /*
615                                      * no error + B_INVAL == directory EOF,
616                                      * use the block.
617                                      */
618                                     if (error == 0 && (bp->b_flags & B_INVAL))
619                                             break;
620                             }
621                             /*
622                              * An error will throw away the block and the
623                              * for loop will break out.  If no error and this
624                              * is not the block we want, we throw away the
625                              * block and go for the next one via the for loop.
626                              */
627                             if (error || i < lbn)
628                                     brelse(bp);
629                         }
630                     }
631                     /*
632                      * The above while is repeated if we hit another cookie
633                      * error.  If we hit an error and it wasn't a cookie error,
634                      * we give up.
635                      */
636                     if (error)
637                             return (error);
638                 }
639
640                 /*
641                  * If not eof and read aheads are enabled, start one.
642                  * (You need the current block first, so that you have the
643                  *  directory offset cookie of the next block.)
644                  */
645                 if (nmp->nm_readahead > 0 &&
646                     (bp->b_flags & B_INVAL) == 0 &&
647                     (np->n_direofoffset == 0 ||
648                     (lbn + 1) * NFS_DIRBLKSIZ < np->n_direofoffset) &&
649                     incore(&vp->v_bufobj, lbn + 1) == NULL) {
650                         rabp = nfs_getcacheblk(vp, lbn + 1, NFS_DIRBLKSIZ, td);
651                         if (rabp) {
652                             if ((rabp->b_flags & (B_CACHE|B_DELWRI)) == 0) {
653                                 rabp->b_flags |= B_ASYNC;
654                                 rabp->b_iocmd = BIO_READ;
655                                 vfs_busy_pages(rabp, 0);
656                                 if (nfs_asyncio(nmp, rabp, cred, td)) {
657                                     rabp->b_flags |= B_INVAL;
658                                     rabp->b_ioflags |= BIO_ERROR;
659                                     vfs_unbusy_pages(rabp);
660                                     brelse(rabp);
661                                 }
662                             } else {
663                                 brelse(rabp);
664                             }
665                         }
666                 }
667                 /*
668                  * Unlike VREG files, whos buffer size ( bp->b_bcount ) is
669                  * chopped for the EOF condition, we cannot tell how large
670                  * NFS directories are going to be until we hit EOF.  So
671                  * an NFS directory buffer is *not* chopped to its EOF.  Now,
672                  * it just so happens that b_resid will effectively chop it
673                  * to EOF.  *BUT* this information is lost if the buffer goes
674                  * away and is reconstituted into a B_CACHE state ( due to
675                  * being VMIO ) later.  So we keep track of the directory eof
676                  * in np->n_direofoffset and chop it off as an extra step
677                  * right here.
678                  */
679                 n = lmin(uio->uio_resid, NFS_DIRBLKSIZ - bp->b_resid - on);
680                 if (np->n_direofoffset && n > np->n_direofoffset - uio->uio_offset)
681                         n = np->n_direofoffset - uio->uio_offset;
682                 break;
683             default:
684                 printf(" nfs_bioread: type %x unexpected\n", vp->v_type);
685                 break;
686             };
687
688             if (n > 0) {
689                     error = uiomove(bp->b_data + on, (int)n, uio);
690             }
691             switch (vp->v_type) {
692             case VREG:
693                 break;
694             case VLNK:
695                 n = 0;
696                 break;
697             case VDIR:
698                 break;
699             default:
700                 printf(" nfs_bioread: type %x unexpected\n", vp->v_type);
701             }
702             brelse(bp);
703         } while (error == 0 && uio->uio_resid > 0 && n > 0);
704         return (error);
705 }
706
707 /*
708  * The NFS write path cannot handle iovecs with len > 1. So we need to 
709  * break up iovecs accordingly (restricting them to wsize).
710  * For the SYNC case, we can do this with 1 copy (user buffer -> mbuf). 
711  * For the ASYNC case, 2 copies are needed. The first a copy from the 
712  * user buffer to a staging buffer and then a second copy from the staging
713  * buffer to mbufs. This can be optimized by copying from the user buffer
714  * directly into mbufs and passing the chain down, but that requires a 
715  * fair amount of re-working of the relevant codepaths (and can be done
716  * later).
717  */
718 static int
719 nfs_directio_write(vp, uiop, cred, ioflag)
720         struct vnode *vp;
721         struct uio *uiop;
722         struct ucred *cred;
723         int ioflag;
724 {
725         int error;
726         struct nfsmount *nmp = VFSTONFS(vp->v_mount);
727         struct thread *td = uiop->uio_td;
728         int size;
729
730         if (ioflag & IO_SYNC) {
731                 int iomode, must_commit;
732                 struct uio uio;
733                 struct iovec iov;
734 do_sync:
735                 while (uiop->uio_resid > 0) {
736                         size = min(uiop->uio_resid, nmp->nm_wsize);
737                         size = min(uiop->uio_iov->iov_len, size);
738                         iov.iov_base = uiop->uio_iov->iov_base;
739                         iov.iov_len = size;
740                         uio.uio_iov = &iov;
741                         uio.uio_iovcnt = 1;
742                         uio.uio_offset = uiop->uio_offset;
743                         uio.uio_resid = size;
744                         uio.uio_segflg = UIO_USERSPACE;
745                         uio.uio_rw = UIO_WRITE;
746                         uio.uio_td = td;
747                         iomode = NFSV3WRITE_FILESYNC;
748                         error = (nmp->nm_rpcops->nr_writerpc)(vp, &uio, cred, 
749                                                       &iomode, &must_commit);
750                         KASSERT((must_commit == 0), 
751                                 ("nfs_directio_write: Did not commit write"));
752                         if (error)
753                                 return (error);
754                         uiop->uio_offset += size;
755                         uiop->uio_resid -= size;
756                         if (uiop->uio_iov->iov_len <= size) {
757                                 uiop->uio_iovcnt--;
758                                 uiop->uio_iov++;
759                         } else {
760                                 uiop->uio_iov->iov_base = 
761                                         (char *)uiop->uio_iov->iov_base + size;
762                                 uiop->uio_iov->iov_len -= size;
763                         }
764                 }
765         } else {
766                 struct uio *t_uio;
767                 struct iovec *t_iov;
768                 struct buf *bp;
769                 
770                 /*
771                  * Break up the write into blocksize chunks and hand these
772                  * over to nfsiod's for write back.
773                  * Unfortunately, this incurs a copy of the data. Since 
774                  * the user could modify the buffer before the write is 
775                  * initiated.
776                  * 
777                  * The obvious optimization here is that one of the 2 copies
778                  * in the async write path can be eliminated by copying the
779                  * data here directly into mbufs and passing the mbuf chain
780                  * down. But that will require a fair amount of re-working
781                  * of the code and can be done if there's enough interest
782                  * in NFS directio access.
783                  */
784                 while (uiop->uio_resid > 0) {
785                         size = min(uiop->uio_resid, nmp->nm_wsize);
786                         size = min(uiop->uio_iov->iov_len, size);
787                         bp = getpbuf(&nfs_pbuf_freecnt);
788                         t_uio = malloc(sizeof(struct uio), M_NFSDIRECTIO, M_WAITOK);
789                         t_iov = malloc(sizeof(struct iovec), M_NFSDIRECTIO, M_WAITOK);
790                         t_iov->iov_base = malloc(size, M_NFSDIRECTIO, M_WAITOK);
791                         t_iov->iov_len = size;
792                         t_uio->uio_iov = t_iov;
793                         t_uio->uio_iovcnt = 1;
794                         t_uio->uio_offset = uiop->uio_offset;
795                         t_uio->uio_resid = size;
796                         t_uio->uio_segflg = UIO_SYSSPACE;
797                         t_uio->uio_rw = UIO_WRITE;
798                         t_uio->uio_td = td;
799                         bcopy(uiop->uio_iov->iov_base, t_iov->iov_base, size);
800                         bp->b_flags |= B_DIRECT;
801                         bp->b_iocmd = BIO_WRITE;
802                         if (cred != NOCRED) {
803                                 crhold(cred);
804                                 bp->b_wcred = cred;
805                         } else 
806                                 bp->b_wcred = NOCRED;                   
807                         bp->b_caller1 = (void *)t_uio;
808                         bp->b_vp = vp;
809                         vhold(vp);
810                         error = nfs_asyncio(nmp, bp, NOCRED, td);
811                         if (error) {
812                                 free(t_iov->iov_base, M_NFSDIRECTIO);
813                                 free(t_iov, M_NFSDIRECTIO);
814                                 free(t_uio, M_NFSDIRECTIO);
815                                 vdrop(bp->b_vp);
816                                 bp->b_vp = NULL;
817                                 relpbuf(bp, &nfs_pbuf_freecnt);
818                                 if (error == EINTR)
819                                         return (error);
820                                 goto do_sync;
821                         }
822                         uiop->uio_offset += size;
823                         uiop->uio_resid -= size;
824                         if (uiop->uio_iov->iov_len <= size) {
825                                 uiop->uio_iovcnt--;
826                                 uiop->uio_iov++;
827                         } else {
828                                 uiop->uio_iov->iov_base = 
829                                         (char *)uiop->uio_iov->iov_base + size;
830                                 uiop->uio_iov->iov_len -= size;
831                         }
832                 }
833         }
834         return (0);
835 }
836
837 /*
838  * Vnode op for write using bio
839  */
840 int
841 nfs_write(struct vop_write_args *ap)
842 {
843         int biosize;
844         struct uio *uio = ap->a_uio;
845         struct thread *td = uio->uio_td;
846         struct vnode *vp = ap->a_vp;
847         struct nfsnode *np = VTONFS(vp);
848         struct ucred *cred = ap->a_cred;
849         int ioflag = ap->a_ioflag;
850         struct buf *bp;
851         struct vattr vattr;
852         struct nfsmount *nmp = VFSTONFS(vp->v_mount);
853         daddr_t lbn;
854         int bcount;
855         int n, on, error = 0;
856         int haverslock = 0;
857         struct proc *p = td?td->td_proc:NULL;
858
859         GIANT_REQUIRED;
860
861 #ifdef DIAGNOSTIC
862         if (uio->uio_rw != UIO_WRITE)
863                 panic("nfs_write mode");
864         if (uio->uio_segflg == UIO_USERSPACE && uio->uio_td != curthread)
865                 panic("nfs_write proc");
866 #endif
867         if (vp->v_type != VREG)
868                 return (EIO);
869         if (np->n_flag & NWRITEERR) {
870                 np->n_flag &= ~NWRITEERR;
871                 return (np->n_error);
872         }
873         if ((nmp->nm_flag & NFSMNT_NFSV3) != 0 &&
874             (nmp->nm_state & NFSSTA_GOTFSINFO) == 0)
875                 (void)nfs_fsinfo(nmp, vp, cred, td);
876
877         /*
878          * Synchronously flush pending buffers if we are in synchronous
879          * mode or if we are appending.
880          */
881         if (ioflag & (IO_APPEND | IO_SYNC)) {
882                 if (np->n_flag & NMODIFIED) {
883                         np->n_attrstamp = 0;
884                         error = nfs_vinvalbuf(vp, V_SAVE, td, 1);
885                         if (error)
886                                 return (error);
887                 }
888         }
889
890         /*
891          * If IO_APPEND then load uio_offset.  We restart here if we cannot
892          * get the append lock.
893          */
894 restart:
895         if (ioflag & IO_APPEND) {
896                 np->n_attrstamp = 0;
897                 error = VOP_GETATTR(vp, &vattr, cred, td);
898                 if (error)
899                         return (error);
900                 uio->uio_offset = np->n_size;
901         }
902
903         if (uio->uio_offset < 0)
904                 return (EINVAL);
905         if ((uio->uio_offset + uio->uio_resid) > nmp->nm_maxfilesize)
906                 return (EFBIG);
907         if (uio->uio_resid == 0)
908                 return (0);
909
910         if (nfs_directio_enable && (ioflag & IO_DIRECT) && vp->v_type == VREG)
911                 return nfs_directio_write(vp, uio, cred, ioflag);
912
913         /*
914          * We need to obtain the rslock if we intend to modify np->n_size
915          * in order to guarentee the append point with multiple contending
916          * writers, to guarentee that no other appenders modify n_size
917          * while we are trying to obtain a truncated buffer (i.e. to avoid
918          * accidently truncating data written by another appender due to
919          * the race), and to ensure that the buffer is populated prior to
920          * our extending of the file.  We hold rslock through the entire
921          * operation.
922          *
923          * Note that we do not synchronize the case where someone truncates
924          * the file while we are appending to it because attempting to lock
925          * this case may deadlock other parts of the system unexpectedly.
926          */
927         if ((ioflag & IO_APPEND) ||
928             uio->uio_offset + uio->uio_resid > np->n_size) {
929                 switch(nfs_rslock(np, td)) {
930                 case ENOLCK:
931                         goto restart;
932                         /* not reached */
933                 case EIO:
934                         return (EIO);
935                 case EINTR:
936                 case ERESTART:
937                         return(EINTR);
938                         /* not reached */
939                 default:
940                         break;
941                 }
942                 haverslock = 1;
943         }
944
945         /*
946          * Maybe this should be above the vnode op call, but so long as
947          * file servers have no limits, i don't think it matters
948          */
949         if (p != NULL) {
950                 PROC_LOCK(p);
951                 if (uio->uio_offset + uio->uio_resid >
952                     lim_cur(p, RLIMIT_FSIZE)) {
953                         psignal(p, SIGXFSZ);
954                         PROC_UNLOCK(p);
955                         if (haverslock)
956                                 nfs_rsunlock(np, td);
957                         return (EFBIG);
958                 }
959                 PROC_UNLOCK(p);
960         }
961
962         biosize = vp->v_mount->mnt_stat.f_iosize;
963
964         do {
965                 nfsstats.biocache_writes++;
966                 lbn = uio->uio_offset / biosize;
967                 on = uio->uio_offset & (biosize-1);
968                 n = min((unsigned)(biosize - on), uio->uio_resid);
969 again:
970                 /*
971                  * Handle direct append and file extension cases, calculate
972                  * unaligned buffer size.
973                  */
974
975                 if (uio->uio_offset == np->n_size && n) {
976                         /*
977                          * Get the buffer (in its pre-append state to maintain
978                          * B_CACHE if it was previously set).  Resize the
979                          * nfsnode after we have locked the buffer to prevent
980                          * readers from reading garbage.
981                          */
982                         bcount = on;
983                         bp = nfs_getcacheblk(vp, lbn, bcount, td);
984
985                         if (bp != NULL) {
986                                 long save;
987
988                                 np->n_size = uio->uio_offset + n;
989                                 np->n_flag |= NMODIFIED;
990                                 vnode_pager_setsize(vp, np->n_size);
991
992                                 save = bp->b_flags & B_CACHE;
993                                 bcount += n;
994                                 allocbuf(bp, bcount);
995                                 bp->b_flags |= save;
996                         }
997                 } else {
998                         /*
999                          * Obtain the locked cache block first, and then
1000                          * adjust the file's size as appropriate.
1001                          */
1002                         bcount = on + n;
1003                         if ((off_t)lbn * biosize + bcount < np->n_size) {
1004                                 if ((off_t)(lbn + 1) * biosize < np->n_size)
1005                                         bcount = biosize;
1006                                 else
1007                                         bcount = np->n_size - (off_t)lbn * biosize;
1008                         }
1009                         bp = nfs_getcacheblk(vp, lbn, bcount, td);
1010                         if (uio->uio_offset + n > np->n_size) {
1011                                 np->n_size = uio->uio_offset + n;
1012                                 np->n_flag |= NMODIFIED;
1013                                 vnode_pager_setsize(vp, np->n_size);
1014                         }
1015                 }
1016
1017                 if (!bp) {
1018                         error = nfs_sigintr(nmp, NULL, td);
1019                         if (!error)
1020                                 error = EINTR;
1021                         break;
1022                 }
1023
1024                 /*
1025                  * Issue a READ if B_CACHE is not set.  In special-append
1026                  * mode, B_CACHE is based on the buffer prior to the write
1027                  * op and is typically set, avoiding the read.  If a read
1028                  * is required in special append mode, the server will
1029                  * probably send us a short-read since we extended the file
1030                  * on our end, resulting in b_resid == 0 and, thusly,
1031                  * B_CACHE getting set.
1032                  *
1033                  * We can also avoid issuing the read if the write covers
1034                  * the entire buffer.  We have to make sure the buffer state
1035                  * is reasonable in this case since we will not be initiating
1036                  * I/O.  See the comments in kern/vfs_bio.c's getblk() for
1037                  * more information.
1038                  *
1039                  * B_CACHE may also be set due to the buffer being cached
1040                  * normally.
1041                  */
1042
1043                 if (on == 0 && n == bcount) {
1044                         bp->b_flags |= B_CACHE;
1045                         bp->b_flags &= ~B_INVAL;
1046                         bp->b_ioflags &= ~BIO_ERROR;
1047                 }
1048
1049                 if ((bp->b_flags & B_CACHE) == 0) {
1050                         bp->b_iocmd = BIO_READ;
1051                         vfs_busy_pages(bp, 0);
1052                         error = nfs_doio(vp, bp, cred, td);
1053                         if (error) {
1054                                 brelse(bp);
1055                                 break;
1056                         }
1057                 }
1058                 if (!bp) {
1059                         error = nfs_sigintr(nmp, NULL, td);
1060                         if (!error)
1061                                 error = EINTR;
1062                         break;
1063                 }
1064                 if (bp->b_wcred == NOCRED)
1065                         bp->b_wcred = crhold(cred);
1066                 np->n_flag |= NMODIFIED;
1067
1068                 /*
1069                  * If dirtyend exceeds file size, chop it down.  This should
1070                  * not normally occur but there is an append race where it
1071                  * might occur XXX, so we log it.
1072                  *
1073                  * If the chopping creates a reverse-indexed or degenerate
1074                  * situation with dirtyoff/end, we 0 both of them.
1075                  */
1076
1077                 if (bp->b_dirtyend > bcount) {
1078                         printf("NFS append race @%lx:%d\n",
1079                             (long)bp->b_blkno * DEV_BSIZE,
1080                             bp->b_dirtyend - bcount);
1081                         bp->b_dirtyend = bcount;
1082                 }
1083
1084                 if (bp->b_dirtyoff >= bp->b_dirtyend)
1085                         bp->b_dirtyoff = bp->b_dirtyend = 0;
1086
1087                 /*
1088                  * If the new write will leave a contiguous dirty
1089                  * area, just update the b_dirtyoff and b_dirtyend,
1090                  * otherwise force a write rpc of the old dirty area.
1091                  *
1092                  * While it is possible to merge discontiguous writes due to
1093                  * our having a B_CACHE buffer ( and thus valid read data
1094                  * for the hole), we don't because it could lead to
1095                  * significant cache coherency problems with multiple clients,
1096                  * especially if locking is implemented later on.
1097                  *
1098                  * as an optimization we could theoretically maintain
1099                  * a linked list of discontinuous areas, but we would still
1100                  * have to commit them separately so there isn't much
1101                  * advantage to it except perhaps a bit of asynchronization.
1102                  */
1103
1104                 if (bp->b_dirtyend > 0 &&
1105                     (on > bp->b_dirtyend || (on + n) < bp->b_dirtyoff)) {
1106                         if (bwrite(bp) == EINTR) {
1107                                 error = EINTR;
1108                                 break;
1109                         }
1110                         goto again;
1111                 }
1112
1113                 error = uiomove((char *)bp->b_data + on, n, uio);
1114
1115                 /*
1116                  * Since this block is being modified, it must be written
1117                  * again and not just committed.  Since write clustering does
1118                  * not work for the stage 1 data write, only the stage 2
1119                  * commit rpc, we have to clear B_CLUSTEROK as well.
1120                  */
1121                 bp->b_flags &= ~(B_NEEDCOMMIT | B_CLUSTEROK);
1122
1123                 if (error) {
1124                         bp->b_ioflags |= BIO_ERROR;
1125                         brelse(bp);
1126                         break;
1127                 }
1128
1129                 /*
1130                  * Only update dirtyoff/dirtyend if not a degenerate
1131                  * condition.
1132                  */
1133                 if (n) {
1134                         if (bp->b_dirtyend > 0) {
1135                                 bp->b_dirtyoff = min(on, bp->b_dirtyoff);
1136                                 bp->b_dirtyend = max((on + n), bp->b_dirtyend);
1137                         } else {
1138                                 bp->b_dirtyoff = on;
1139                                 bp->b_dirtyend = on + n;
1140                         }
1141                         vfs_bio_set_validclean(bp, on, n);
1142                 }
1143
1144                 /*
1145                  * If IO_SYNC do bwrite().
1146                  *
1147                  * IO_INVAL appears to be unused.  The idea appears to be
1148                  * to turn off caching in this case.  Very odd.  XXX
1149                  */
1150                 if ((ioflag & IO_SYNC)) {
1151                         if (ioflag & IO_INVAL)
1152                                 bp->b_flags |= B_NOCACHE;
1153                         error = bwrite(bp);
1154                         if (error)
1155                                 break;
1156                 } else if ((n + on) == biosize) {
1157                         bp->b_flags |= B_ASYNC;
1158                         (void) (nmp->nm_rpcops->nr_writebp)(bp, 0, 0);
1159                 } else {
1160                         bdwrite(bp);
1161                 }
1162         } while (uio->uio_resid > 0 && n > 0);
1163
1164         if (haverslock)
1165                 nfs_rsunlock(np, td);
1166
1167         return (error);
1168 }
1169
1170 /*
1171  * Get an nfs cache block.
1172  *
1173  * Allocate a new one if the block isn't currently in the cache
1174  * and return the block marked busy. If the calling process is
1175  * interrupted by a signal for an interruptible mount point, return
1176  * NULL.
1177  *
1178  * The caller must carefully deal with the possible B_INVAL state of
1179  * the buffer.  nfs_doio() clears B_INVAL (and nfs_asyncio() clears it
1180  * indirectly), so synchronous reads can be issued without worrying about
1181  * the B_INVAL state.  We have to be a little more careful when dealing
1182  * with writes (see comments in nfs_write()) when extending a file past
1183  * its EOF.
1184  */
1185 static struct buf *
1186 nfs_getcacheblk(struct vnode *vp, daddr_t bn, int size, struct thread *td)
1187 {
1188         struct buf *bp;
1189         struct mount *mp;
1190         struct nfsmount *nmp;
1191
1192         mp = vp->v_mount;
1193         nmp = VFSTONFS(mp);
1194
1195         if (nmp->nm_flag & NFSMNT_INT) {
1196                 sigset_t oldset;
1197
1198                 nfs_set_sigmask(td, &oldset);
1199                 bp = getblk(vp, bn, size, PCATCH, 0, 0);
1200                 nfs_restore_sigmask(td, &oldset);
1201                 while (bp == NULL) {
1202                         if (nfs_sigintr(nmp, NULL, td))
1203                                 return (NULL);
1204                         bp = getblk(vp, bn, size, 0, 2 * hz, 0);
1205                 }
1206         } else {
1207                 bp = getblk(vp, bn, size, 0, 0, 0);
1208         }
1209
1210         if (vp->v_type == VREG) {
1211                 int biosize;
1212
1213                 biosize = mp->mnt_stat.f_iosize;
1214                 bp->b_blkno = bn * (biosize / DEV_BSIZE);
1215         }
1216         return (bp);
1217 }
1218
1219 /*
1220  * Flush and invalidate all dirty buffers. If another process is already
1221  * doing the flush, just wait for completion.
1222  */
1223 int
1224 nfs_vinvalbuf(struct vnode *vp, int flags, struct thread *td, int intrflg)
1225 {
1226         struct nfsnode *np = VTONFS(vp);
1227         struct nfsmount *nmp = VFSTONFS(vp->v_mount);
1228         int error = 0, slpflag, slptimeo;
1229         int old_lock = 0;
1230
1231         ASSERT_VOP_LOCKED(vp, "nfs_vinvalbuf");
1232
1233         /*
1234          * XXX This check stops us from needlessly doing a vinvalbuf when
1235          * being called through vclean().  It is not clear that this is
1236          * unsafe.
1237          */
1238         if (vp->v_iflag & VI_DOOMED)
1239                 return (0);
1240
1241         if ((nmp->nm_flag & NFSMNT_INT) == 0)
1242                 intrflg = 0;
1243         if (intrflg) {
1244                 slpflag = PCATCH;
1245                 slptimeo = 2 * hz;
1246         } else {
1247                 slpflag = 0;
1248                 slptimeo = 0;
1249         }
1250
1251         if ((old_lock = VOP_ISLOCKED(vp, td)) != LK_EXCLUSIVE) {
1252                 if (old_lock == LK_SHARED) {
1253                         /* Upgrade to exclusive lock, this might block */
1254                         vn_lock(vp, LK_UPGRADE | LK_RETRY, td);
1255                 } else {
1256                         vn_lock(vp, LK_EXCLUSIVE | LK_RETRY, td);
1257                 }
1258         }
1259
1260         /*
1261          * Now, flush as required.
1262          */
1263         error = vinvalbuf(vp, flags, td, slpflag, 0);
1264         while (error) {
1265                 if (intrflg && (error = nfs_sigintr(nmp, NULL, td)))
1266                         goto out;
1267                 error = vinvalbuf(vp, flags, td, 0, slptimeo);
1268         }
1269         np->n_flag &= ~NMODIFIED;
1270 out:
1271         if (old_lock != LK_EXCLUSIVE) {
1272                 if (old_lock == LK_SHARED) {
1273                         /* Downgrade from exclusive lock, this might block */
1274                         vn_lock(vp, LK_DOWNGRADE, td);
1275                 } else {
1276                         VOP_UNLOCK(vp, 0, td);
1277                 }
1278         }
1279         return error;
1280 }
1281
1282 /*
1283  * Initiate asynchronous I/O. Return an error if no nfsiods are available.
1284  * This is mainly to avoid queueing async I/O requests when the nfsiods
1285  * are all hung on a dead server.
1286  *
1287  * Note: nfs_asyncio() does not clear (BIO_ERROR|B_INVAL) but when the bp
1288  * is eventually dequeued by the async daemon, nfs_doio() *will*.
1289  */
1290 int
1291 nfs_asyncio(struct nfsmount *nmp, struct buf *bp, struct ucred *cred, struct thread *td)
1292 {
1293         int iod;
1294         int gotiod;
1295         int slpflag = 0;
1296         int slptimeo = 0;
1297         int error, error2;
1298
1299         /*
1300          * Commits are usually short and sweet so lets save some cpu and
1301          * leave the async daemons for more important rpc's (such as reads
1302          * and writes).
1303          */
1304         if (bp->b_iocmd == BIO_WRITE && (bp->b_flags & B_NEEDCOMMIT) &&
1305             (nmp->nm_bufqiods > nfs_numasync / 2)) {
1306                 return(EIO);
1307         }
1308
1309 again:
1310         if (nmp->nm_flag & NFSMNT_INT)
1311                 slpflag = PCATCH;
1312         gotiod = FALSE;
1313
1314         /*
1315          * Find a free iod to process this request.
1316          */
1317         for (iod = 0; iod < nfs_numasync; iod++)
1318                 if (nfs_iodwant[iod]) {
1319                         gotiod = TRUE;
1320                         break;
1321                 }
1322
1323         /*
1324          * Try to create one if none are free.
1325          */
1326         if (!gotiod) {
1327                 iod = nfs_nfsiodnew();
1328                 if (iod != -1)
1329                         gotiod = TRUE;
1330         }
1331
1332         if (gotiod) {
1333                 /*
1334                  * Found one, so wake it up and tell it which
1335                  * mount to process.
1336                  */
1337                 NFS_DPF(ASYNCIO, ("nfs_asyncio: waking iod %d for mount %p\n",
1338                     iod, nmp));
1339                 nfs_iodwant[iod] = NULL;
1340                 nfs_iodmount[iod] = nmp;
1341                 nmp->nm_bufqiods++;
1342                 wakeup(&nfs_iodwant[iod]);
1343         }
1344
1345         /*
1346          * If none are free, we may already have an iod working on this mount
1347          * point.  If so, it will process our request.
1348          */
1349         if (!gotiod) {
1350                 if (nmp->nm_bufqiods > 0) {
1351                         NFS_DPF(ASYNCIO,
1352                                 ("nfs_asyncio: %d iods are already processing mount %p\n",
1353                                  nmp->nm_bufqiods, nmp));
1354                         gotiod = TRUE;
1355                 }
1356         }
1357
1358         /*
1359          * If we have an iod which can process the request, then queue
1360          * the buffer.
1361          */
1362         if (gotiod) {
1363                 /*
1364                  * Ensure that the queue never grows too large.  We still want
1365                  * to asynchronize so we block rather then return EIO.
1366                  */
1367                 while (nmp->nm_bufqlen >= 2*nfs_numasync) {
1368                         NFS_DPF(ASYNCIO,
1369                                 ("nfs_asyncio: waiting for mount %p queue to drain\n", nmp));
1370                         nmp->nm_bufqwant = TRUE;
1371                         error = nfs_tsleep(td, &nmp->nm_bufq, slpflag | PRIBIO,
1372                                            "nfsaio", slptimeo);
1373                         if (error) {
1374                                 error2 = nfs_sigintr(nmp, NULL, td);
1375                                 if (error2)
1376                                         return (error2);
1377                                 if (slpflag == PCATCH) {
1378                                         slpflag = 0;
1379                                         slptimeo = 2 * hz;
1380                                 }
1381                         }
1382                         /*
1383                          * We might have lost our iod while sleeping,
1384                          * so check and loop if nescessary.
1385                          */
1386                         if (nmp->nm_bufqiods == 0) {
1387                                 NFS_DPF(ASYNCIO,
1388                                         ("nfs_asyncio: no iods after mount %p queue was drained, looping\n", nmp));
1389                                 goto again;
1390                         }
1391                 }
1392
1393                 if (bp->b_iocmd == BIO_READ) {
1394                         if (bp->b_rcred == NOCRED && cred != NOCRED)
1395                                 bp->b_rcred = crhold(cred);
1396                 } else {
1397                         if (bp->b_wcred == NOCRED && cred != NOCRED)
1398                                 bp->b_wcred = crhold(cred);
1399                 }
1400
1401                 if (bp->b_flags & B_REMFREE)
1402                         bremfreef(bp);
1403                 BUF_KERNPROC(bp);
1404                 TAILQ_INSERT_TAIL(&nmp->nm_bufq, bp, b_freelist);
1405                 nmp->nm_bufqlen++;
1406                 return (0);
1407         }
1408
1409         /*
1410          * All the iods are busy on other mounts, so return EIO to
1411          * force the caller to process the i/o synchronously.
1412          */
1413         NFS_DPF(ASYNCIO, ("nfs_asyncio: no iods available, i/o is synchronous\n"));
1414         return (EIO);
1415 }
1416
1417 void
1418 nfs_doio_directwrite(struct buf *bp)
1419 {
1420         int iomode, must_commit;
1421         struct uio *uiop = (struct uio *)bp->b_caller1;
1422         char *iov_base = uiop->uio_iov->iov_base;
1423         struct nfsmount *nmp = VFSTONFS(bp->b_vp->v_mount);
1424         
1425         iomode = NFSV3WRITE_FILESYNC;
1426         uiop->uio_td = NULL; /* NULL since we're in nfsiod */
1427         (nmp->nm_rpcops->nr_writerpc)(bp->b_vp, uiop, bp->b_wcred, &iomode, &must_commit);
1428         KASSERT((must_commit == 0), ("nfs_doio_directwrite: Did not commit write"));
1429         free(iov_base, M_NFSDIRECTIO);
1430         free(uiop->uio_iov, M_NFSDIRECTIO);
1431         free(uiop, M_NFSDIRECTIO);
1432         vdrop(bp->b_vp);
1433         bp->b_vp = NULL;
1434         relpbuf(bp, &nfs_pbuf_freecnt);
1435 }
1436
1437 /*
1438  * Do an I/O operation to/from a cache block. This may be called
1439  * synchronously or from an nfsiod.
1440  */
1441 int
1442 nfs_doio(struct vnode *vp, struct buf *bp, struct ucred *cr, struct thread *td)
1443 {
1444         struct uio *uiop;
1445         struct nfsnode *np;
1446         struct nfsmount *nmp;
1447         int error = 0, iomode, must_commit = 0;
1448         struct uio uio;
1449         struct iovec io;
1450         struct proc *p = td ? td->td_proc : NULL;
1451
1452         np = VTONFS(vp);
1453         nmp = VFSTONFS(vp->v_mount);
1454         uiop = &uio;
1455         uiop->uio_iov = &io;
1456         uiop->uio_iovcnt = 1;
1457         uiop->uio_segflg = UIO_SYSSPACE;
1458         uiop->uio_td = td;
1459
1460         /*
1461          * clear BIO_ERROR and B_INVAL state prior to initiating the I/O.  We
1462          * do this here so we do not have to do it in all the code that
1463          * calls us.
1464          */
1465         bp->b_flags &= ~B_INVAL;
1466         bp->b_ioflags &= ~BIO_ERROR;
1467
1468         KASSERT(!(bp->b_flags & B_DONE), ("nfs_doio: bp %p already marked done", bp));
1469
1470         if (bp->b_iocmd == BIO_READ) {
1471             io.iov_len = uiop->uio_resid = bp->b_bcount;
1472             io.iov_base = bp->b_data;
1473             uiop->uio_rw = UIO_READ;
1474
1475             switch (vp->v_type) {
1476             case VREG:
1477                 uiop->uio_offset = ((off_t)bp->b_blkno) * DEV_BSIZE;
1478                 nfsstats.read_bios++;
1479                 error = (nmp->nm_rpcops->nr_readrpc)(vp, uiop, cr);
1480
1481                 if (!error) {
1482                     if (uiop->uio_resid) {
1483                         /*
1484                          * If we had a short read with no error, we must have
1485                          * hit a file hole.  We should zero-fill the remainder.
1486                          * This can also occur if the server hits the file EOF.
1487                          *
1488                          * Holes used to be able to occur due to pending
1489                          * writes, but that is not possible any longer.
1490                          */
1491                         int nread = bp->b_bcount - uiop->uio_resid;
1492                         int left  = uiop->uio_resid;
1493
1494                         if (left > 0)
1495                                 bzero((char *)bp->b_data + nread, left);
1496                         uiop->uio_resid = 0;
1497                     }
1498                 }
1499                 /* ASSERT_VOP_LOCKED(vp, "nfs_doio"); */
1500                 if (p && (vp->v_vflag & VV_TEXT) &&
1501                     (NFS_TIMESPEC_COMPARE(&np->n_mtime, &np->n_vattr.va_mtime))) {
1502                         PROC_LOCK(p);
1503                         killproc(p, "text file modification");
1504                         PROC_UNLOCK(p);
1505                 }
1506                 break;
1507             case VLNK:
1508                 uiop->uio_offset = (off_t)0;
1509                 nfsstats.readlink_bios++;
1510                 error = (nmp->nm_rpcops->nr_readlinkrpc)(vp, uiop, cr);
1511                 break;
1512             case VDIR:
1513                 nfsstats.readdir_bios++;
1514                 uiop->uio_offset = ((u_quad_t)bp->b_lblkno) * NFS_DIRBLKSIZ;
1515                 if ((nmp->nm_flag & NFSMNT_NFSV4) != 0)
1516                         error = nfs4_readdirrpc(vp, uiop, cr);
1517                 else {
1518                         if ((nmp->nm_flag & NFSMNT_RDIRPLUS) != 0) {
1519                                 error = nfs_readdirplusrpc(vp, uiop, cr);
1520                                 if (error == NFSERR_NOTSUPP)
1521                                         nmp->nm_flag &= ~NFSMNT_RDIRPLUS;
1522                         }
1523                         if ((nmp->nm_flag & NFSMNT_RDIRPLUS) == 0)
1524                                 error = nfs_readdirrpc(vp, uiop, cr);
1525                 }
1526                 /*
1527                  * end-of-directory sets B_INVAL but does not generate an
1528                  * error.
1529                  */
1530                 if (error == 0 && uiop->uio_resid == bp->b_bcount)
1531                         bp->b_flags |= B_INVAL;
1532                 break;
1533             default:
1534                 printf("nfs_doio:  type %x unexpected\n", vp->v_type);
1535                 break;
1536             };
1537             if (error) {
1538                 bp->b_ioflags |= BIO_ERROR;
1539                 bp->b_error = error;
1540             }
1541         } else {
1542             /*
1543              * If we only need to commit, try to commit
1544              */
1545             if (bp->b_flags & B_NEEDCOMMIT) {
1546                     int retv;
1547                     off_t off;
1548
1549                     off = ((u_quad_t)bp->b_blkno) * DEV_BSIZE + bp->b_dirtyoff;
1550                     retv = (nmp->nm_rpcops->nr_commit)(
1551                                 vp, off, bp->b_dirtyend-bp->b_dirtyoff,
1552                                 bp->b_wcred, td);
1553                     if (retv == 0) {
1554                             bp->b_dirtyoff = bp->b_dirtyend = 0;
1555                             bp->b_flags &= ~(B_NEEDCOMMIT | B_CLUSTEROK);
1556                             bp->b_resid = 0;
1557                             bufdone(bp);
1558                             return (0);
1559                     }
1560                     if (retv == NFSERR_STALEWRITEVERF) {
1561                             nfs_clearcommit(vp->v_mount);
1562                     }
1563             }
1564
1565             /*
1566              * Setup for actual write
1567              */
1568
1569             if ((off_t)bp->b_blkno * DEV_BSIZE + bp->b_dirtyend > np->n_size)
1570                 bp->b_dirtyend = np->n_size - (off_t)bp->b_blkno * DEV_BSIZE;
1571
1572             if (bp->b_dirtyend > bp->b_dirtyoff) {
1573                 io.iov_len = uiop->uio_resid = bp->b_dirtyend
1574                     - bp->b_dirtyoff;
1575                 uiop->uio_offset = (off_t)bp->b_blkno * DEV_BSIZE
1576                     + bp->b_dirtyoff;
1577                 io.iov_base = (char *)bp->b_data + bp->b_dirtyoff;
1578                 uiop->uio_rw = UIO_WRITE;
1579                 nfsstats.write_bios++;
1580
1581                 if ((bp->b_flags & (B_ASYNC | B_NEEDCOMMIT | B_NOCACHE | B_CLUSTER)) == B_ASYNC)
1582                     iomode = NFSV3WRITE_UNSTABLE;
1583                 else
1584                     iomode = NFSV3WRITE_FILESYNC;
1585
1586                 error = (nmp->nm_rpcops->nr_writerpc)(vp, uiop, cr, &iomode, &must_commit);
1587
1588                 /*
1589                  * When setting B_NEEDCOMMIT also set B_CLUSTEROK to try
1590                  * to cluster the buffers needing commit.  This will allow
1591                  * the system to submit a single commit rpc for the whole
1592                  * cluster.  We can do this even if the buffer is not 100%
1593                  * dirty (relative to the NFS blocksize), so we optimize the
1594                  * append-to-file-case.
1595                  *
1596                  * (when clearing B_NEEDCOMMIT, B_CLUSTEROK must also be
1597                  * cleared because write clustering only works for commit
1598                  * rpc's, not for the data portion of the write).
1599                  */
1600
1601                 if (!error && iomode == NFSV3WRITE_UNSTABLE) {
1602                     bp->b_flags |= B_NEEDCOMMIT;
1603                     if (bp->b_dirtyoff == 0
1604                         && bp->b_dirtyend == bp->b_bcount)
1605                         bp->b_flags |= B_CLUSTEROK;
1606                 } else {
1607                     bp->b_flags &= ~(B_NEEDCOMMIT | B_CLUSTEROK);
1608                 }
1609
1610                 /*
1611                  * For an interrupted write, the buffer is still valid
1612                  * and the write hasn't been pushed to the server yet,
1613                  * so we can't set BIO_ERROR and report the interruption
1614                  * by setting B_EINTR. For the B_ASYNC case, B_EINTR
1615                  * is not relevant, so the rpc attempt is essentially
1616                  * a noop.  For the case of a V3 write rpc not being
1617                  * committed to stable storage, the block is still
1618                  * dirty and requires either a commit rpc or another
1619                  * write rpc with iomode == NFSV3WRITE_FILESYNC before
1620                  * the block is reused. This is indicated by setting
1621                  * the B_DELWRI and B_NEEDCOMMIT flags.
1622                  *
1623                  * If the buffer is marked B_PAGING, it does not reside on
1624                  * the vp's paging queues so we cannot call bdirty().  The
1625                  * bp in this case is not an NFS cache block so we should
1626                  * be safe. XXX
1627                  */
1628                 if (error == EINTR || error == EIO
1629                     || (!error && (bp->b_flags & B_NEEDCOMMIT))) {
1630                         int s;
1631
1632                         s = splbio();
1633                         bp->b_flags &= ~(B_INVAL|B_NOCACHE);
1634                         if ((bp->b_flags & B_PAGING) == 0) {
1635                             bdirty(bp);
1636                             bp->b_flags &= ~B_DONE;
1637                         }
1638                         if (error && (bp->b_flags & B_ASYNC) == 0)
1639                             bp->b_flags |= B_EINTR;
1640                         splx(s);
1641                 } else {
1642                     if (error) {
1643                         bp->b_ioflags |= BIO_ERROR;
1644                         bp->b_error = np->n_error = error;
1645                         np->n_flag |= NWRITEERR;
1646                     }
1647                     bp->b_dirtyoff = bp->b_dirtyend = 0;
1648                 }
1649             } else {
1650                 bp->b_resid = 0;
1651                 bufdone(bp);
1652                 return (0);
1653             }
1654         }
1655         bp->b_resid = uiop->uio_resid;
1656         if (must_commit)
1657             nfs_clearcommit(vp->v_mount);
1658         bufdone(bp);
1659         return (error);
1660 }
1661
1662 /*
1663  * Used to aid in handling ftruncate() operations on the NFS client side.
1664  * Truncation creates a number of special problems for NFS.  We have to
1665  * throw away VM pages and buffer cache buffers that are beyond EOF, and
1666  * we have to properly handle VM pages or (potentially dirty) buffers
1667  * that straddle the truncation point.
1668  */
1669
1670 int
1671 nfs_meta_setsize(struct vnode *vp, struct ucred *cred, struct thread *td, u_quad_t nsize)
1672 {
1673         struct nfsnode *np = VTONFS(vp);
1674         u_quad_t tsize = np->n_size;
1675         int biosize = vp->v_mount->mnt_stat.f_iosize;
1676         int error = 0;
1677
1678         np->n_size = nsize;
1679
1680         if (np->n_size < tsize) {
1681                 struct buf *bp;
1682                 daddr_t lbn;
1683                 int bufsize;
1684
1685                 /*
1686                  * vtruncbuf() doesn't get the buffer overlapping the 
1687                  * truncation point.  We may have a B_DELWRI and/or B_CACHE
1688                  * buffer that now needs to be truncated.
1689                  */
1690                 error = vtruncbuf(vp, cred, td, nsize, biosize);
1691                 lbn = nsize / biosize;
1692                 bufsize = nsize & (biosize - 1);
1693                 bp = nfs_getcacheblk(vp, lbn, bufsize, td);
1694                 if (!bp)
1695                         return EINTR;
1696                 if (bp->b_dirtyoff > bp->b_bcount)
1697                         bp->b_dirtyoff = bp->b_bcount;
1698                 if (bp->b_dirtyend > bp->b_bcount)
1699                         bp->b_dirtyend = bp->b_bcount;
1700                 bp->b_flags |= B_RELBUF;  /* don't leave garbage around */
1701                 brelse(bp);
1702         } else {
1703                 vnode_pager_setsize(vp, nsize);
1704         }
1705         return(error);
1706 }
1707