[PATCH] knfsd: Correctly handle error condition from lockd_up
[safe/jmp/linux-2.6] / fs / nfsd / nfssvc.c
1 /*
2  * linux/fs/nfsd/nfssvc.c
3  *
4  * Central processing for nfsd.
5  *
6  * Authors:     Olaf Kirch (okir@monad.swb.de)
7  *
8  * Copyright (C) 1995, 1996, 1997 Olaf Kirch <okir@monad.swb.de>
9  */
10
11 #include <linux/module.h>
12
13 #include <linux/time.h>
14 #include <linux/errno.h>
15 #include <linux/nfs.h>
16 #include <linux/in.h>
17 #include <linux/uio.h>
18 #include <linux/unistd.h>
19 #include <linux/slab.h>
20 #include <linux/smp.h>
21 #include <linux/smp_lock.h>
22 #include <linux/fs_struct.h>
23
24 #include <linux/sunrpc/types.h>
25 #include <linux/sunrpc/stats.h>
26 #include <linux/sunrpc/svc.h>
27 #include <linux/sunrpc/svcsock.h>
28 #include <linux/sunrpc/cache.h>
29 #include <linux/nfsd/nfsd.h>
30 #include <linux/nfsd/stats.h>
31 #include <linux/nfsd/cache.h>
32 #include <linux/nfsd/syscall.h>
33 #include <linux/lockd/bind.h>
34 #include <linux/nfsacl.h>
35
36 #define NFSDDBG_FACILITY        NFSDDBG_SVC
37
38 /* these signals will be delivered to an nfsd thread 
39  * when handling a request
40  */
41 #define ALLOWED_SIGS    (sigmask(SIGKILL))
42 /* these signals will be delivered to an nfsd thread
43  * when not handling a request. i.e. when waiting
44  */
45 #define SHUTDOWN_SIGS   (sigmask(SIGKILL) | sigmask(SIGHUP) | sigmask(SIGINT) | sigmask(SIGQUIT))
46 /* if the last thread dies with SIGHUP, then the exports table is
47  * left unchanged ( like 2.4-{0-9} ).  Any other signal will clear
48  * the exports table (like 2.2).
49  */
50 #define SIG_NOCLEAN     SIGHUP
51
52 extern struct svc_program       nfsd_program;
53 static void                     nfsd(struct svc_rqst *rqstp);
54 struct timeval                  nfssvc_boot;
55        struct svc_serv          *nfsd_serv;
56 static atomic_t                 nfsd_busy;
57 static unsigned long            nfsd_last_call;
58 static DEFINE_SPINLOCK(nfsd_call_lock);
59
60 struct nfsd_list {
61         struct list_head        list;
62         struct task_struct      *task;
63 };
64 static struct list_head nfsd_list = LIST_HEAD_INIT(nfsd_list);
65
66 #if defined(CONFIG_NFSD_V2_ACL) || defined(CONFIG_NFSD_V3_ACL)
67 static struct svc_stat  nfsd_acl_svcstats;
68 static struct svc_version *     nfsd_acl_version[] = {
69         [2] = &nfsd_acl_version2,
70         [3] = &nfsd_acl_version3,
71 };
72
73 #define NFSD_ACL_MINVERS            2
74 #define NFSD_ACL_NRVERS         ARRAY_SIZE(nfsd_acl_version)
75 static struct svc_version *nfsd_acl_versions[NFSD_ACL_NRVERS];
76
77 static struct svc_program       nfsd_acl_program = {
78         .pg_prog                = NFS_ACL_PROGRAM,
79         .pg_nvers               = NFSD_ACL_NRVERS,
80         .pg_vers                = nfsd_acl_versions,
81         .pg_name                = "nfsd",
82         .pg_class               = "nfsd",
83         .pg_stats               = &nfsd_acl_svcstats,
84         .pg_authenticate        = &svc_set_client,
85 };
86
87 static struct svc_stat  nfsd_acl_svcstats = {
88         .program        = &nfsd_acl_program,
89 };
90 #endif /* defined(CONFIG_NFSD_V2_ACL) || defined(CONFIG_NFSD_V3_ACL) */
91
92 static struct svc_version *     nfsd_version[] = {
93         [2] = &nfsd_version2,
94 #if defined(CONFIG_NFSD_V3)
95         [3] = &nfsd_version3,
96 #endif
97 #if defined(CONFIG_NFSD_V4)
98         [4] = &nfsd_version4,
99 #endif
100 };
101
102 #define NFSD_MINVERS            2
103 #define NFSD_NRVERS             ARRAY_SIZE(nfsd_version)
104 static struct svc_version *nfsd_versions[NFSD_NRVERS];
105
106 struct svc_program              nfsd_program = {
107 #if defined(CONFIG_NFSD_V2_ACL) || defined(CONFIG_NFSD_V3_ACL)
108         .pg_next                = &nfsd_acl_program,
109 #endif
110         .pg_prog                = NFS_PROGRAM,          /* program number */
111         .pg_nvers               = NFSD_NRVERS,          /* nr of entries in nfsd_version */
112         .pg_vers                = nfsd_versions,        /* version table */
113         .pg_name                = "nfsd",               /* program name */
114         .pg_class               = "nfsd",               /* authentication class */
115         .pg_stats               = &nfsd_svcstats,       /* version table */
116         .pg_authenticate        = &svc_set_client,      /* export authentication */
117
118 };
119
120 int nfsd_vers(int vers, enum vers_op change)
121 {
122         if (vers < NFSD_MINVERS || vers >= NFSD_NRVERS)
123                 return -1;
124         switch(change) {
125         case NFSD_SET:
126                 nfsd_versions[vers] = nfsd_version[vers];
127                 break;
128 #if defined(CONFIG_NFSD_V2_ACL) || defined(CONFIG_NFSD_V3_ACL)
129                 if (vers < NFSD_ACL_NRVERS)
130                         nfsd_acl_version[vers] = nfsd_acl_version[vers];
131 #endif
132         case NFSD_CLEAR:
133                 nfsd_versions[vers] = NULL;
134 #if defined(CONFIG_NFSD_V2_ACL) || defined(CONFIG_NFSD_V3_ACL)
135                 if (vers < NFSD_ACL_NRVERS)
136                         nfsd_acl_version[vers] = NULL;
137 #endif
138                 break;
139         case NFSD_TEST:
140                 return nfsd_versions[vers] != NULL;
141         case NFSD_AVAIL:
142                 return nfsd_version[vers] != NULL;
143         }
144         return 0;
145 }
146 /*
147  * Maximum number of nfsd processes
148  */
149 #define NFSD_MAXSERVS           8192
150
151 int nfsd_nrthreads(void)
152 {
153         if (nfsd_serv == NULL)
154                 return 0;
155         else
156                 return nfsd_serv->sv_nrthreads;
157 }
158
159 static int killsig;     /* signal that was used to kill last nfsd */
160 static void nfsd_last_thread(struct svc_serv *serv)
161 {
162         /* When last nfsd thread exits we need to do some clean-up */
163         struct svc_sock *svsk;
164         list_for_each_entry(svsk, &serv->sv_permsocks, sk_list)
165                 lockd_down();
166         nfsd_serv = NULL;
167         nfsd_racache_shutdown();
168         nfs4_state_shutdown();
169
170         printk(KERN_WARNING "nfsd: last server has exited\n");
171         if (killsig != SIG_NOCLEAN) {
172                 printk(KERN_WARNING "nfsd: unexporting all filesystems\n");
173                 nfsd_export_flush();
174         }
175 }
176
177 void nfsd_reset_versions(void)
178 {
179         int found_one = 0;
180         int i;
181
182         for (i = NFSD_MINVERS; i < NFSD_NRVERS; i++) {
183                 if (nfsd_program.pg_vers[i])
184                         found_one = 1;
185         }
186
187         if (!found_one) {
188                 for (i = NFSD_MINVERS; i < NFSD_NRVERS; i++)
189                         nfsd_program.pg_vers[i] = nfsd_version[i];
190 #if defined(CONFIG_NFSD_V2_ACL) || defined(CONFIG_NFSD_V3_ACL)
191                 for (i = NFSD_ACL_MINVERS; i < NFSD_ACL_NRVERS; i++)
192                         nfsd_acl_program.pg_vers[i] =
193                                 nfsd_acl_version[i];
194 #endif
195         }
196 }
197
198 int nfsd_create_serv(void)
199 {
200         int err = 0;
201         lock_kernel();
202         if (nfsd_serv) {
203                 nfsd_serv->sv_nrthreads++;
204                 unlock_kernel();
205                 return 0;
206         }
207
208         atomic_set(&nfsd_busy, 0);
209         nfsd_serv = svc_create(&nfsd_program, NFSD_BUFSIZE,
210                                nfsd_last_thread);
211         if (nfsd_serv == NULL)
212                 err = -ENOMEM;
213         unlock_kernel();
214         do_gettimeofday(&nfssvc_boot);          /* record boot time */
215         return err;
216 }
217
218 static int nfsd_init_socks(int port)
219 {
220         int error;
221         if (!list_empty(&nfsd_serv->sv_permsocks))
222                 return 0;
223
224         error = lockd_up(IPPROTO_UDP);
225         if (error >= 0) {
226                 error = svc_makesock(nfsd_serv, IPPROTO_UDP, port);
227                 if (error < 0)
228                         lockd_down();
229         }
230         if (error < 0)
231                 return error;
232
233 #ifdef CONFIG_NFSD_TCP
234         error = lockd_up(IPPROTO_TCP);
235         if (error >= 0) {
236                 error = svc_makesock(nfsd_serv, IPPROTO_TCP, port);
237                 if (error < 0)
238                         lockd_down();
239         }
240         if (error < 0)
241                 return error;
242 #endif
243         return 0;
244 }
245
246 int
247 nfsd_svc(unsigned short port, int nrservs)
248 {
249         int     error;
250         struct list_head *victim;
251         
252         lock_kernel();
253         dprintk("nfsd: creating service\n");
254         error = -EINVAL;
255         if (nrservs <= 0)
256                 nrservs = 0;
257         if (nrservs > NFSD_MAXSERVS)
258                 nrservs = NFSD_MAXSERVS;
259         
260         /* Readahead param cache - will no-op if it already exists */
261         error = nfsd_racache_init(2*nrservs);
262         if (error<0)
263                 goto out;
264         error = nfs4_state_start();
265         if (error<0)
266                 goto out;
267
268         nfsd_reset_versions();
269
270         error = nfsd_create_serv();
271
272         if (error)
273                 goto out;
274         error = nfsd_init_socks(port);
275         if (error)
276                 goto failure;
277
278         nrservs -= (nfsd_serv->sv_nrthreads-1);
279         while (nrservs > 0) {
280                 nrservs--;
281                 __module_get(THIS_MODULE);
282                 error = svc_create_thread(nfsd, nfsd_serv);
283                 if (error < 0) {
284                         module_put(THIS_MODULE);
285                         break;
286                 }
287         }
288         victim = nfsd_list.next;
289         while (nrservs < 0 && victim != &nfsd_list) {
290                 struct nfsd_list *nl =
291                         list_entry(victim,struct nfsd_list, list);
292                 victim = victim->next;
293                 send_sig(SIG_NOCLEAN, nl->task, 1);
294                 nrservs++;
295         }
296  failure:
297         svc_destroy(nfsd_serv);         /* Release server */
298  out:
299         unlock_kernel();
300         return error;
301 }
302
303 static inline void
304 update_thread_usage(int busy_threads)
305 {
306         unsigned long prev_call;
307         unsigned long diff;
308         int decile;
309
310         spin_lock(&nfsd_call_lock);
311         prev_call = nfsd_last_call;
312         nfsd_last_call = jiffies;
313         decile = busy_threads*10/nfsdstats.th_cnt;
314         if (decile>0 && decile <= 10) {
315                 diff = nfsd_last_call - prev_call;
316                 if ( (nfsdstats.th_usage[decile-1] += diff) >= NFSD_USAGE_WRAP)
317                         nfsdstats.th_usage[decile-1] -= NFSD_USAGE_WRAP;
318                 if (decile == 10)
319                         nfsdstats.th_fullcnt++;
320         }
321         spin_unlock(&nfsd_call_lock);
322 }
323
324 /*
325  * This is the NFS server kernel thread
326  */
327 static void
328 nfsd(struct svc_rqst *rqstp)
329 {
330         struct fs_struct *fsp;
331         int             err;
332         struct nfsd_list me;
333         sigset_t shutdown_mask, allowed_mask;
334
335         /* Lock module and set up kernel thread */
336         lock_kernel();
337         daemonize("nfsd");
338
339         /* After daemonize() this kernel thread shares current->fs
340          * with the init process. We need to create files with a
341          * umask of 0 instead of init's umask. */
342         fsp = copy_fs_struct(current->fs);
343         if (!fsp) {
344                 printk("Unable to start nfsd thread: out of memory\n");
345                 goto out;
346         }
347         exit_fs(current);
348         current->fs = fsp;
349         current->fs->umask = 0;
350
351         siginitsetinv(&shutdown_mask, SHUTDOWN_SIGS);
352         siginitsetinv(&allowed_mask, ALLOWED_SIGS);
353
354         nfsdstats.th_cnt++;
355
356         me.task = current;
357         list_add(&me.list, &nfsd_list);
358
359         unlock_kernel();
360
361         /*
362          * We want less throttling in balance_dirty_pages() so that nfs to
363          * localhost doesn't cause nfsd to lock up due to all the client's
364          * dirty pages.
365          */
366         current->flags |= PF_LESS_THROTTLE;
367
368         /*
369          * The main request loop
370          */
371         for (;;) {
372                 /* Block all but the shutdown signals */
373                 sigprocmask(SIG_SETMASK, &shutdown_mask, NULL);
374
375                 /*
376                  * Find a socket with data available and call its
377                  * recvfrom routine.
378                  */
379                 while ((err = svc_recv(rqstp, 60*60*HZ)) == -EAGAIN)
380                         ;
381                 if (err < 0)
382                         break;
383                 update_thread_usage(atomic_read(&nfsd_busy));
384                 atomic_inc(&nfsd_busy);
385
386                 /* Lock the export hash tables for reading. */
387                 exp_readlock();
388
389                 /* Process request with signals blocked.  */
390                 sigprocmask(SIG_SETMASK, &allowed_mask, NULL);
391
392                 svc_process(rqstp);
393
394                 /* Unlock export hash tables */
395                 exp_readunlock();
396                 update_thread_usage(atomic_read(&nfsd_busy));
397                 atomic_dec(&nfsd_busy);
398         }
399
400         if (err != -EINTR) {
401                 printk(KERN_WARNING "nfsd: terminating on error %d\n", -err);
402         } else {
403                 unsigned int    signo;
404
405                 for (signo = 1; signo <= _NSIG; signo++)
406                         if (sigismember(&current->pending.signal, signo) &&
407                             !sigismember(&current->blocked, signo))
408                                 break;
409                 killsig = signo;
410         }
411         /* Clear signals before calling svc_exit_thread() */
412         flush_signals(current);
413
414         lock_kernel();
415
416         list_del(&me.list);
417         nfsdstats.th_cnt --;
418
419 out:
420         /* Release the thread */
421         svc_exit_thread(rqstp);
422
423         /* Release module */
424         unlock_kernel();
425         module_put_and_exit(0);
426 }
427
428 int
429 nfsd_dispatch(struct svc_rqst *rqstp, u32 *statp)
430 {
431         struct svc_procedure    *proc;
432         kxdrproc_t              xdr;
433         u32                     nfserr;
434         u32                     *nfserrp;
435
436         dprintk("nfsd_dispatch: vers %d proc %d\n",
437                                 rqstp->rq_vers, rqstp->rq_proc);
438         proc = rqstp->rq_procinfo;
439
440         /* Check whether we have this call in the cache. */
441         switch (nfsd_cache_lookup(rqstp, proc->pc_cachetype)) {
442         case RC_INTR:
443         case RC_DROPIT:
444                 return 0;
445         case RC_REPLY:
446                 return 1;
447         case RC_DOIT:;
448                 /* do it */
449         }
450
451         /* Decode arguments */
452         xdr = proc->pc_decode;
453         if (xdr && !xdr(rqstp, (u32*)rqstp->rq_arg.head[0].iov_base,
454                         rqstp->rq_argp)) {
455                 dprintk("nfsd: failed to decode arguments!\n");
456                 nfsd_cache_update(rqstp, RC_NOCACHE, NULL);
457                 *statp = rpc_garbage_args;
458                 return 1;
459         }
460
461         /* need to grab the location to store the status, as
462          * nfsv4 does some encoding while processing 
463          */
464         nfserrp = rqstp->rq_res.head[0].iov_base
465                 + rqstp->rq_res.head[0].iov_len;
466         rqstp->rq_res.head[0].iov_len += sizeof(u32);
467
468         /* Now call the procedure handler, and encode NFS status. */
469         nfserr = proc->pc_func(rqstp, rqstp->rq_argp, rqstp->rq_resp);
470         if (nfserr == nfserr_jukebox && rqstp->rq_vers == 2)
471                 nfserr = nfserr_dropit;
472         if (nfserr == nfserr_dropit) {
473                 dprintk("nfsd: Dropping request due to malloc failure!\n");
474                 nfsd_cache_update(rqstp, RC_NOCACHE, NULL);
475                 return 0;
476         }
477
478         if (rqstp->rq_proc != 0)
479                 *nfserrp++ = nfserr;
480
481         /* Encode result.
482          * For NFSv2, additional info is never returned in case of an error.
483          */
484         if (!(nfserr && rqstp->rq_vers == 2)) {
485                 xdr = proc->pc_encode;
486                 if (xdr && !xdr(rqstp, nfserrp,
487                                 rqstp->rq_resp)) {
488                         /* Failed to encode result. Release cache entry */
489                         dprintk("nfsd: failed to encode result!\n");
490                         nfsd_cache_update(rqstp, RC_NOCACHE, NULL);
491                         *statp = rpc_system_err;
492                         return 1;
493                 }
494         }
495
496         /* Store reply in cache. */
497         nfsd_cache_update(rqstp, proc->pc_cachetype, statp + 1);
498         return 1;
499 }