LINUX: Sometimes let dentry_open handle refcounts
[openafs.git] / src / afs / LINUX / osi_file.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  * 
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 #include <afsconfig.h>
11 #include "afs/param.h"
12
13
14 #include <linux/module.h> /* early to avoid printf->printk mapping */
15 #include "afs/sysincludes.h"    /* Standard vendor system headers */
16 #include "afsincludes.h"        /* Afs-based standard headers */
17 #include "afs/afs_stats.h"      /* afs statistics */
18 #include <linux/namei.h>
19
20 #if defined(HAVE_LINUX_EXPORTFS_H)
21 #include <linux/exportfs.h>
22 #endif
23 #include "osi_compat.h"
24
25 int cache_fh_type = -1;
26 int cache_fh_len = -1;
27
28 afs_lock_t afs_xosi;            /* lock is for tvattr */
29 extern struct osi_dev cacheDev;
30 extern struct vfsmount *afs_cacheMnt;
31 extern struct super_block *afs_cacheSBp;
32 #if defined(STRUCT_TASK_STRUCT_HAS_CRED)
33 extern struct cred *cache_creds;
34 #endif
35
36 /* Old export ops: decode_fh will call back here. Accept any dentry it suggests */
37 int
38 afs_fh_acceptable(void *context, struct dentry *dp)
39 {
40     return 1;
41 }
42
43 struct file *
44 afs_linux_raw_open(afs_dcache_id_t *ainode)
45 {
46     struct inode *tip = NULL;
47     struct dentry *dp = NULL;
48     struct file* filp;
49
50     dp = afs_get_dentry_from_fh(afs_cacheSBp, ainode, cache_fh_len, cache_fh_type,
51                 afs_fh_acceptable);
52     if ((!dp) || IS_ERR(dp))
53            osi_Panic("Can't get dentry\n");
54     tip = dp->d_inode;
55     tip->i_flags |= S_NOATIME;  /* Disable updating access times. */
56
57 #if defined(STRUCT_TASK_STRUCT_HAS_CRED)
58     /* Use stashed credentials - prevent selinux/apparmor problems  */
59     filp = afs_dentry_open(dp, afs_cacheMnt, O_RDWR, cache_creds);
60     if (IS_ERR(filp))
61         filp = afs_dentry_open(dp, afs_cacheMnt, O_RDWR, current_cred());
62 #else
63     filp = dentry_open(dget(dp), mntget(afs_cacheMnt), O_RDWR);
64 #endif
65     if (IS_ERR(filp))
66         osi_Panic("Can't open file: %d\n", (int) PTR_ERR(filp));
67
68     dput(dp);
69
70     return filp;
71 }
72
73 void *
74 osi_UFSOpen(afs_dcache_id_t *ainode)
75 {
76     struct osi_file *afile = NULL;
77     extern int cacheDiskType;
78
79     AFS_STATCNT(osi_UFSOpen);
80     if (cacheDiskType != AFS_FCACHE_TYPE_UFS) {
81         osi_Panic("UFSOpen called for non-UFS cache\n");
82     }
83     if (!afs_osicred_initialized) {
84         /* valid for alpha_osf, SunOS, Ultrix */
85         memset(&afs_osi_cred, 0, sizeof(afs_ucred_t));
86         crhold(&afs_osi_cred);  /* don't let it evaporate, since it is static */
87         afs_osicred_initialized = 1;
88     }
89     AFS_GUNLOCK();
90     afile = kmalloc(sizeof(struct osi_file), GFP_NOFS);
91     if (!afile) {
92         osi_Panic("osi_UFSOpen: Failed to allocate %d bytes for osi_file.\n",
93                   (int)sizeof(struct osi_file));
94     }
95     memset(afile, 0, sizeof(struct osi_file));
96
97     afile->filp = afs_linux_raw_open(ainode);
98     afile->size = i_size_read(FILE_INODE(afile->filp));
99     AFS_GLOCK();
100     afile->offset = 0;
101     afile->proc = (int (*)())0;
102     return (void *)afile;
103 }
104
105 /*
106  * Given a dentry, return the file handle as encoded by the filesystem.
107  * We can't assume anything about the length (words, not bytes).
108  * The cache has to live on a single filesystem with uniform file 
109  * handles, otherwise we panic.
110  */
111 void osi_get_fh(struct dentry *dp, afs_ufs_dcache_id_t *ainode) {
112     int max_len;
113     int type;
114
115     if (cache_fh_len > 0)
116         max_len = cache_fh_len;
117     else
118         max_len = MAX_FH_LEN;
119     type = afs_get_fh_from_dentry(dp, ainode, &max_len);
120     if (type == 255) {
121         osi_Panic("File handle encoding failed\n");
122     }
123     if (cache_fh_type < 0)
124         cache_fh_type = type;
125     if (cache_fh_len < 0) {
126         cache_fh_len = max_len;
127     }
128     if (type != cache_fh_type || max_len != cache_fh_len) {
129         osi_Panic("Inconsistent file handles within cache\n");
130     }
131 }
132
133 int
134 afs_osi_Stat(struct osi_file *afile, struct osi_stat *astat)
135 {
136     AFS_STATCNT(osi_Stat);
137     ObtainWriteLock(&afs_xosi, 320);
138     astat->size = i_size_read(OSIFILE_INODE(afile));
139     astat->mtime = OSIFILE_INODE(afile)->i_mtime.tv_sec;
140     astat->atime = OSIFILE_INODE(afile)->i_atime.tv_sec;
141
142     ReleaseWriteLock(&afs_xosi);
143     return 0;
144 }
145
146 int
147 osi_UFSClose(struct osi_file *afile)
148 {
149     AFS_STATCNT(osi_Close);
150     if (afile) {
151         if (OSIFILE_INODE(afile)) {
152             filp_close(afile->filp, NULL);
153         }
154     }
155     kfree(afile);
156     return 0;
157 }
158
159 int
160 osi_UFSTruncate(struct osi_file *afile, afs_int32 asize)
161 {
162     afs_int32 code;
163     struct osi_stat tstat;
164     struct iattr newattrs;
165     struct inode *inode = OSIFILE_INODE(afile);
166     AFS_STATCNT(osi_Truncate);
167
168     /* This routine only shrinks files, and most systems
169      * have very slow truncates, even when the file is already
170      * small enough.  Check now and save some time.
171      */
172     code = afs_osi_Stat(afile, &tstat);
173     if (code || tstat.size <= asize)
174         return code;
175     ObtainWriteLock(&afs_xosi, 321);
176     AFS_GUNLOCK();
177     afs_linux_lock_inode(inode);
178 #ifdef STRUCT_INODE_HAS_I_ALLOC_SEM
179     down_write(&inode->i_alloc_sem);
180 #endif
181     newattrs.ia_size = asize;
182     newattrs.ia_valid = ATTR_SIZE | ATTR_CTIME;
183     newattrs.ia_ctime = CURRENT_TIME;
184
185     /* avoid notify_change() since it wants to update dentry->d_parent */
186     code = inode_change_ok(inode, &newattrs);
187     if (!code)
188         code = afs_inode_setattr(afile, &newattrs);
189     if (!code)
190         truncate_inode_pages(&inode->i_data, asize);
191     code = -code;
192 #ifdef STRUCT_INODE_HAS_I_ALLOC_SEM
193     up_write(&inode->i_alloc_sem);
194 #endif
195     afs_linux_unlock_inode(inode);
196     AFS_GLOCK();
197     ReleaseWriteLock(&afs_xosi);
198     return code;
199 }
200
201
202 /* Generic read interface */
203 int
204 afs_osi_Read(struct osi_file *afile, int offset, void *aptr,
205              afs_int32 asize)
206 {
207     struct uio auio;
208     struct iovec iov;
209     afs_int32 code;
210
211     AFS_STATCNT(osi_Read);
212
213     /*
214      * If the osi_file passed in is NULL, panic only if AFS is not shutting
215      * down. No point in crashing when we are already shutting down
216      */
217     if (!afile) {
218         if (!afs_shuttingdown)
219             osi_Panic("osi_Read called with null param");
220         else
221             return -EIO;
222     }
223
224     if (offset != -1)
225         afile->offset = offset;
226     setup_uio(&auio, &iov, aptr, afile->offset, asize, UIO_READ, AFS_UIOSYS);
227     AFS_GUNLOCK();
228     code = osi_rdwr(afile, &auio, UIO_READ);
229     AFS_GLOCK();
230     if (code == 0) {
231         code = asize - auio.uio_resid;
232         afile->offset += code;
233     } else {
234         afs_Trace2(afs_iclSetp, CM_TRACE_READFAILED, ICL_TYPE_INT32, auio.uio_resid,
235                    ICL_TYPE_INT32, code);
236         if (code > 0) {
237             code = -code;
238         }
239     }
240     return code;
241 }
242
243 /* Generic write interface */
244 int
245 afs_osi_Write(struct osi_file *afile, afs_int32 offset, void *aptr,
246               afs_int32 asize)
247 {
248     struct uio auio;
249     struct iovec iov;
250     afs_int32 code;
251
252     AFS_STATCNT(osi_Write);
253
254     if (!afile) {
255         if (!afs_shuttingdown)
256             osi_Panic("afs_osi_Write called with null param");
257         else
258             return -EIO;
259     }
260
261     if (offset != -1)
262         afile->offset = offset;
263     setup_uio(&auio, &iov, aptr, afile->offset, asize, UIO_WRITE, AFS_UIOSYS);
264     AFS_GUNLOCK();
265     code = osi_rdwr(afile, &auio, UIO_WRITE);
266     AFS_GLOCK();
267     if (code == 0) {
268         code = asize - auio.uio_resid;
269         afile->offset += code;
270     } else {
271         if (code == ENOSPC)
272             afs_warnuser
273                 ("\n\n\n*** Cache partition is FULL - Decrease cachesize!!! ***\n\n");
274         if (code > 0) {
275             code = -code;
276         }
277     }
278
279     if (afile->proc)
280         (*afile->proc)(afile, code);
281
282     return code;
283 }
284
285
286 /*  This work should be handled by physstrat in ca/machdep.c.
287     This routine written from the RT NFS port strategy routine.
288     It has been generalized a bit, but should still be pretty clear. */
289 int
290 afs_osi_MapStrategy(int (*aproc) (struct buf * bp), struct buf *bp)
291 {
292     afs_int32 returnCode;
293
294     AFS_STATCNT(osi_MapStrategy);
295     returnCode = (*aproc) (bp);
296
297     return returnCode;
298 }
299
300 void
301 shutdown_osifile(void)
302 {
303     AFS_STATCNT(shutdown_osifile);
304     if (afs_cold_shutdown) {
305         afs_osicred_initialized = 0;
306     }
307 }
308
309 /* Intialize cache device info and fragment size for disk cache partition. */
310 int
311 osi_InitCacheInfo(char *aname)
312 {
313     int code;
314     extern afs_dcache_id_t cacheInode;
315     struct dentry *dp;
316     extern struct osi_dev cacheDev;
317     extern afs_int32 afs_fsfragsize;
318     extern struct super_block *afs_cacheSBp;
319     extern struct vfsmount *afs_cacheMnt;
320     code = osi_lookupname_internal(aname, 1, &afs_cacheMnt, &dp);
321     if (code)
322         return ENOENT;
323
324     osi_get_fh(dp, &cacheInode.ufs);
325     cacheDev.dev = dp->d_inode->i_sb->s_dev;
326     afs_fsfragsize = dp->d_inode->i_sb->s_blocksize - 1;
327     afs_cacheSBp = dp->d_inode->i_sb;
328
329     dput(dp);
330
331     afs_init_sb_export_ops(afs_cacheSBp);
332
333     return 0;
334 }
335
336
337 /* osi_rdwr
338  * seek, then read or write to an open inode. addrp points to data in
339  * kernel space.
340  */
341 int
342 osi_rdwr(struct osi_file *osifile, struct uio *uiop, int rw)
343 {
344     struct file *filp = osifile->filp;
345     mm_segment_t old_fs = {0};
346     int code = 0;
347     struct iovec *iov;
348     size_t count;
349     unsigned long savelim;
350     loff_t pos;
351
352     savelim = current->TASK_STRUCT_RLIM[RLIMIT_FSIZE].rlim_cur;
353     current->TASK_STRUCT_RLIM[RLIMIT_FSIZE].rlim_cur = RLIM_INFINITY;
354
355     if (uiop->uio_seg == AFS_UIOSYS) {
356         /* Switch into user space */
357         old_fs = get_fs();
358         set_fs(get_ds());
359     }
360
361     while (code == 0 && uiop->uio_resid > 0 && uiop->uio_iovcnt > 0) {
362         iov = uiop->uio_iov;
363         count = iov->iov_len;
364         if (count == 0) {
365             uiop->uio_iov++;
366             uiop->uio_iovcnt--;
367             continue;
368         }
369
370         pos = uiop->uio_offset;
371         if (rw == UIO_READ)
372             code = filp->f_op->read(filp, iov->iov_base, count, &pos);
373         else
374             code = filp->f_op->write(filp, iov->iov_base, count, &pos);
375
376         if (code < 0) {
377             code = -code;
378             break;
379         } else if (code == 0) {
380             /*
381              * This is bad -- we can't read any more data from the
382              * file, but we have no good way of signaling a partial
383              * read either.
384              */
385             code = EIO;
386             break;
387         }
388
389         iov->iov_base += code;
390         iov->iov_len -= code;
391         uiop->uio_resid -= code;
392         uiop->uio_offset += code;
393         code = 0;
394     }
395
396     if (uiop->uio_seg == AFS_UIOSYS) {
397         /* Switch back into kernel space */
398         set_fs(old_fs);
399     }
400
401     current->TASK_STRUCT_RLIM[RLIMIT_FSIZE].rlim_cur = savelim;
402
403     return code;
404 }
405
406 /* setup_uio 
407  * Setup a uio struct.
408  */
409 void
410 setup_uio(struct uio *uiop, struct iovec *iovecp, const char *buf, afs_offs_t pos,
411           int count, uio_flag_t flag, uio_seg_t seg)
412 {
413     iovecp->iov_base = (char *)buf;
414     iovecp->iov_len = count;
415     uiop->uio_iov = iovecp;
416     uiop->uio_iovcnt = 1;
417     uiop->uio_offset = pos;
418     uiop->uio_seg = seg;
419     uiop->uio_resid = count;
420     uiop->uio_flag = flag;
421 }
422
423
424 /* uiomove
425  * UIO_READ : dp -> uio
426  * UIO_WRITE : uio -> dp
427  */
428 int
429 uiomove(char *dp, int length, uio_flag_t rw, struct uio *uiop)
430 {
431     int count;
432     struct iovec *iov;
433     int code;
434
435     while (length > 0 && uiop->uio_resid > 0 && uiop->uio_iovcnt > 0) {
436         iov = uiop->uio_iov;
437         count = iov->iov_len;
438
439         if (!count) {
440             uiop->uio_iov++;
441             uiop->uio_iovcnt--;
442             continue;
443         }
444
445         if (count > length)
446             count = length;
447
448         switch (uiop->uio_seg) {
449         case AFS_UIOSYS:
450             switch (rw) {
451             case UIO_READ:
452                 memcpy(iov->iov_base, dp, count);
453                 break;
454             case UIO_WRITE:
455                 memcpy(dp, iov->iov_base, count);
456                 break;
457             default:
458                 printf("uiomove: Bad rw = %d\n", rw);
459                 return -EINVAL;
460             }
461             break;
462         case AFS_UIOUSER:
463             switch (rw) {
464             case UIO_READ:
465                 AFS_COPYOUT(dp, iov->iov_base, count, code);
466                 break;
467             case UIO_WRITE:
468                 AFS_COPYIN(iov->iov_base, dp, count, code);
469                 break;
470             default:
471                 printf("uiomove: Bad rw = %d\n", rw);
472                 return -EINVAL;
473             }
474             break;
475         default:
476             printf("uiomove: Bad seg = %d\n", uiop->uio_seg);
477             return -EINVAL;
478         }
479
480         dp += count;
481         length -= count;
482         iov->iov_base += count;
483         iov->iov_len -= count;
484         uiop->uio_offset += count;
485         uiop->uio_resid -= count;
486     }
487     return 0;
488 }
489