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