svc.c
上传用户:lgb322
上传日期:2013-02-24
资源大小:30529k
文件大小:9k
源码类别:

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * linux/fs/lockd/svc.c
  3.  *
  4.  * This is the central lockd service.
  5.  *
  6.  * FIXME: Separate the lockd NFS server functionality from the lockd NFS
  7.  *    client functionality. Oh why didn't Sun create two separate
  8.  *   services in the first place?
  9.  *
  10.  * Authors: Olaf Kirch (okir@monad.swb.de)
  11.  *
  12.  * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de>
  13.  */
  14. #define __KERNEL_SYSCALLS__
  15. #include <linux/config.h>
  16. #include <linux/module.h>
  17. #include <linux/init.h>
  18. #include <linux/sched.h>
  19. #include <linux/errno.h>
  20. #include <linux/in.h>
  21. #include <linux/uio.h>
  22. #include <linux/version.h>
  23. #include <linux/unistd.h>
  24. #include <linux/slab.h>
  25. #include <linux/smp.h>
  26. #include <linux/smp_lock.h>
  27. #include <linux/sunrpc/types.h>
  28. #include <linux/sunrpc/stats.h>
  29. #include <linux/sunrpc/clnt.h>
  30. #include <linux/sunrpc/svc.h>
  31. #include <linux/sunrpc/svcsock.h>
  32. #include <linux/lockd/lockd.h>
  33. #include <linux/nfs.h>
  34. #define NLMDBG_FACILITY NLMDBG_SVC
  35. #define LOCKD_BUFSIZE (1024 + NLMSSVC_XDRSIZE)
  36. #define ALLOWED_SIGS (sigmask(SIGKILL))
  37. extern struct svc_program nlmsvc_program;
  38. struct nlmsvc_binding * nlmsvc_ops;
  39. static DECLARE_MUTEX(nlmsvc_sema);
  40. static unsigned int nlmsvc_users;
  41. static pid_t nlmsvc_pid;
  42. int nlmsvc_grace_period;
  43. unsigned long nlmsvc_timeout;
  44. static DECLARE_MUTEX_LOCKED(lockd_start);
  45. static DECLARE_WAIT_QUEUE_HEAD(lockd_exit);
  46. /*
  47.  * Currently the following can be set only at insmod time.
  48.  * Ideally, they would be accessible through the sysctl interface.
  49.  */
  50. unsigned long nlm_grace_period;
  51. unsigned long nlm_timeout = LOCKD_DFLT_TIMEO;
  52. unsigned long nlm_udpport, nlm_tcpport;
  53. static unsigned long set_grace_period(void)
  54. {
  55. unsigned long grace_period;
  56. /* Note: nlm_timeout should always be nonzero */
  57. if (nlm_grace_period)
  58. grace_period = ((nlm_grace_period + nlm_timeout - 1)
  59. / nlm_timeout) * nlm_timeout * HZ;
  60. else
  61. grace_period = nlm_timeout * 5 * HZ;
  62. nlmsvc_grace_period = 1;
  63. return grace_period + jiffies;
  64. }
  65. static inline void clear_grace_period(void)
  66. {
  67. nlmsvc_grace_period = 0;
  68. }
  69. /*
  70.  * This is the lockd kernel thread
  71.  */
  72. static void
  73. lockd(struct svc_rqst *rqstp)
  74. {
  75. struct svc_serv *serv = rqstp->rq_server;
  76. int err = 0;
  77. unsigned long grace_period_expire;
  78. /* Lock module and set up kernel thread */
  79. MOD_INC_USE_COUNT;
  80. lock_kernel();
  81. /*
  82.  * Let our maker know we're running.
  83.  */
  84. nlmsvc_pid = current->pid;
  85. up(&lockd_start);
  86. daemonize();
  87. reparent_to_init();
  88. sprintf(current->comm, "lockd");
  89. /* Process request with signals blocked.  */
  90. spin_lock_irq(&current->sigmask_lock);
  91. siginitsetinv(&current->blocked, sigmask(SIGKILL));
  92. recalc_sigpending(current);
  93. spin_unlock_irq(&current->sigmask_lock);
  94. /* kick rpciod */
  95. rpciod_up();
  96. dprintk("NFS locking service started (ver " LOCKD_VERSION ").n");
  97. if (!nlm_timeout)
  98. nlm_timeout = LOCKD_DFLT_TIMEO;
  99. nlmsvc_timeout = nlm_timeout * HZ;
  100. grace_period_expire = set_grace_period();
  101. /*
  102.  * The main request loop. We don't terminate until the last
  103.  * NFS mount or NFS daemon has gone away, and we've been sent a
  104.  * signal, or else another process has taken over our job.
  105.  */
  106. while ((nlmsvc_users || !signalled()) && nlmsvc_pid == current->pid)
  107. {
  108. long timeout = MAX_SCHEDULE_TIMEOUT;
  109. if (signalled()) {
  110. spin_lock_irq(&current->sigmask_lock);
  111. flush_signals(current);
  112. spin_unlock_irq(&current->sigmask_lock);
  113. if (nlmsvc_ops) {
  114. nlmsvc_ops->detach();
  115. grace_period_expire = set_grace_period();
  116. }
  117. }
  118. /*
  119.  * Retry any blocked locks that have been notified by
  120.  * the VFS. Don't do this during grace period.
  121.  * (Theoretically, there shouldn't even be blocked locks
  122.  * during grace period).
  123.  */
  124. if (!nlmsvc_grace_period) {
  125. timeout = nlmsvc_retry_blocked();
  126. } else if (time_before(grace_period_expire, jiffies))
  127. clear_grace_period();
  128. /*
  129.  * Find a socket with data available and call its
  130.  * recvfrom routine.
  131.  */
  132. err = svc_recv(serv, rqstp, timeout);
  133. if (err == -EAGAIN || err == -EINTR)
  134. continue;
  135. if (err < 0) {
  136. printk(KERN_WARNING
  137.        "lockd: terminating on error %dn",
  138.        -err);
  139. break;
  140. }
  141. dprintk("lockd: request from %08xn",
  142. (unsigned)ntohl(rqstp->rq_addr.sin_addr.s_addr));
  143. /*
  144.  * Look up the NFS client handle. The handle is needed for
  145.  * all but the GRANTED callback RPCs.
  146.  */
  147. rqstp->rq_client = NULL;
  148. if (nlmsvc_ops) {
  149. nlmsvc_ops->exp_readlock();
  150. rqstp->rq_client =
  151. nlmsvc_ops->exp_getclient(&rqstp->rq_addr);
  152. }
  153. svc_process(serv, rqstp);
  154. /* Unlock export hash tables */
  155. if (nlmsvc_ops)
  156. nlmsvc_ops->exp_unlock();
  157. }
  158. /*
  159.  * Check whether there's a new lockd process before
  160.  * shutting down the hosts and clearing the slot.
  161.  */
  162. if (!nlmsvc_pid || current->pid == nlmsvc_pid) {
  163. if (nlmsvc_ops)
  164. nlmsvc_ops->detach();
  165. nlm_shutdown_hosts();
  166. nlmsvc_pid = 0;
  167. } else
  168. printk(KERN_DEBUG
  169. "lockd: new process, skipping host shutdownn");
  170. wake_up(&lockd_exit);
  171. /* Exit the RPC thread */
  172. svc_exit_thread(rqstp);
  173. /* release rpciod */
  174. rpciod_down();
  175. /* Release module */
  176. MOD_DEC_USE_COUNT;
  177. }
  178. /*
  179.  * Bring up the lockd process if it's not already up.
  180.  */
  181. int
  182. lockd_up(void)
  183. {
  184. static int warned = 0; 
  185. struct svc_serv * serv;
  186. int error = 0;
  187. down(&nlmsvc_sema);
  188. /*
  189.  * Unconditionally increment the user count ... this is
  190.  * the number of clients who _want_ a lockd process.
  191.  */
  192. nlmsvc_users++; 
  193. /*
  194.  * Check whether we're already up and running.
  195.  */
  196. if (nlmsvc_pid)
  197. goto out;
  198. /*
  199.  * Sanity check: if there's no pid,
  200.  * we should be the first user ...
  201.  */
  202. if (nlmsvc_users > 1)
  203. printk(KERN_WARNING
  204. "lockd_up: no pid, %d users??n", nlmsvc_users);
  205. error = -ENOMEM;
  206. serv = svc_create(&nlmsvc_program, 0, NLMSVC_XDRSIZE);
  207. if (!serv) {
  208. printk(KERN_WARNING "lockd_up: create service failedn");
  209. goto out;
  210. }
  211. if ((error = svc_makesock(serv, IPPROTO_UDP, nlm_udpport)) < 0 
  212. #ifdef CONFIG_NFSD_TCP
  213.  || (error = svc_makesock(serv, IPPROTO_TCP, nlm_tcpport)) < 0
  214. #endif
  215. ) {
  216. if (warned++ == 0) 
  217. printk(KERN_WARNING
  218. "lockd_up: makesock failed, error=%dn", error);
  219. goto destroy_and_out;
  220. warned = 0;
  221. /*
  222.  * Create the kernel thread and wait for it to start.
  223.  */
  224. error = svc_create_thread(lockd, serv);
  225. if (error) {
  226. printk(KERN_WARNING
  227. "lockd_up: create thread failed, error=%dn", error);
  228. goto destroy_and_out;
  229. }
  230. down(&lockd_start);
  231. /*
  232.  * Note: svc_serv structures have an initial use count of 1,
  233.  * so we exit through here on both success and failure.
  234.  */
  235. destroy_and_out:
  236. svc_destroy(serv);
  237. out:
  238. up(&nlmsvc_sema);
  239. return error;
  240. }
  241. /*
  242.  * Decrement the user count and bring down lockd if we're the last.
  243.  */
  244. void
  245. lockd_down(void)
  246. {
  247. static int warned = 0; 
  248. down(&nlmsvc_sema);
  249. if (nlmsvc_users) {
  250. if (--nlmsvc_users)
  251. goto out;
  252. } else
  253. printk(KERN_WARNING "lockd_down: no users! pid=%dn", nlmsvc_pid);
  254. if (!nlmsvc_pid) {
  255. if (warned++ == 0)
  256. printk(KERN_WARNING "lockd_down: no lockd running.n"); 
  257. goto out;
  258. }
  259. warned = 0;
  260. kill_proc(nlmsvc_pid, SIGKILL, 1);
  261. /*
  262.  * Wait for the lockd process to exit, but since we're holding
  263.  * the lockd semaphore, we can't wait around forever ...
  264.  */
  265. current->sigpending = 0;
  266. interruptible_sleep_on_timeout(&lockd_exit, HZ);
  267. if (nlmsvc_pid) {
  268. printk(KERN_WARNING 
  269. "lockd_down: lockd failed to exit, clearing pidn");
  270. nlmsvc_pid = 0;
  271. }
  272. spin_lock_irq(&current->sigmask_lock);
  273. recalc_sigpending(current);
  274. spin_unlock_irq(&current->sigmask_lock);
  275. out:
  276. up(&nlmsvc_sema);
  277. }
  278. #ifdef MODULE
  279. /* New module support in 2.1.18 */
  280. MODULE_AUTHOR("Olaf Kirch <okir@monad.swb.de>");
  281. MODULE_DESCRIPTION("NFS file locking service version " LOCKD_VERSION ".");
  282. MODULE_LICENSE("GPL");
  283. MODULE_PARM(nlm_grace_period, "10-240l");
  284. MODULE_PARM(nlm_timeout, "3-20l");
  285. MODULE_PARM(nlm_udpport, "0-65535l");
  286. MODULE_PARM(nlm_tcpport, "0-65535l");
  287. int
  288. init_module(void)
  289. {
  290. /* Init the static variables */
  291. init_MUTEX(&nlmsvc_sema);
  292. nlmsvc_users = 0;
  293. nlmsvc_pid = 0;
  294. return 0;
  295. }
  296. void
  297. cleanup_module(void)
  298. {
  299. /* FIXME: delete all NLM clients */
  300. nlm_shutdown_hosts();
  301. }
  302. #else
  303. /* not a module, so process bootargs
  304.  * lockd.udpport and lockd.tcpport
  305.  */
  306. static int __init udpport_set(char *str)
  307. {
  308. nlm_udpport = simple_strtoul(str, NULL, 0);
  309. return 1;
  310. }
  311. static int __init tcpport_set(char *str)
  312. {
  313. nlm_tcpport = simple_strtoul(str, NULL, 0);
  314. return 1;
  315. }
  316. __setup("lockd.udpport=", udpport_set);
  317. __setup("lockd.tcpport=", tcpport_set);
  318. #endif
  319. /*
  320.  * Define NLM program and procedures
  321.  */
  322. static struct svc_version nlmsvc_version1 = {
  323. 1, 17, nlmsvc_procedures, NULL
  324. };
  325. static struct svc_version nlmsvc_version3 = {
  326. 3, 24, nlmsvc_procedures, NULL
  327. };
  328. #ifdef CONFIG_LOCKD_V4
  329. static struct svc_version nlmsvc_version4 = {
  330. 4, 24, nlmsvc_procedures4, NULL
  331. };
  332. #endif
  333. static struct svc_version * nlmsvc_version[] = {
  334. NULL,
  335. &nlmsvc_version1,
  336. NULL,
  337. &nlmsvc_version3,
  338. #ifdef CONFIG_LOCKD_V4
  339. &nlmsvc_version4,
  340. #endif
  341. };
  342. static struct svc_stat nlmsvc_stats;
  343. #define NLM_NRVERS (sizeof(nlmsvc_version)/sizeof(nlmsvc_version[0]))
  344. struct svc_program nlmsvc_program = {
  345. NLM_PROGRAM, /* program number */
  346. 1, NLM_NRVERS-1, /* version range */
  347. NLM_NRVERS, /* number of entries in nlmsvc_version */
  348. nlmsvc_version, /* version table */
  349. "lockd", /* service name */
  350. &nlmsvc_stats, /* stats table */
  351. };