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

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * linux/fs/lockd/mon.c
  3.  *
  4.  * The kernel statd client.
  5.  *
  6.  * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
  7.  */
  8. #include <linux/types.h>
  9. #include <linux/utsname.h>
  10. #include <linux/kernel.h>
  11. #include <linux/sunrpc/clnt.h>
  12. #include <linux/sunrpc/svc.h>
  13. #include <linux/lockd/lockd.h>
  14. #include <linux/lockd/sm_inter.h>
  15. #define NLMDBG_FACILITY NLMDBG_MONITOR
  16. static struct rpc_clnt * nsm_create(void);
  17. extern struct rpc_program nsm_program;
  18. /*
  19.  * Local NSM state
  20.  */
  21. u32 nsm_local_state;
  22. /*
  23.  * Common procedure for SM_MON/SM_UNMON calls
  24.  */
  25. static int
  26. nsm_mon_unmon(struct nlm_host *host, u32 proc, struct nsm_res *res)
  27. {
  28. struct rpc_clnt *clnt;
  29. int status;
  30. struct nsm_args args;
  31. status = -EACCES;
  32. clnt = nsm_create();
  33. if (!clnt)
  34. goto out;
  35. args.addr = host->h_addr.sin_addr.s_addr;
  36. args.prog = NLM_PROGRAM;
  37. args.vers = host->h_version;
  38. args.proc = NLMPROC_NSM_NOTIFY;
  39. memset(res, 0, sizeof(*res));
  40. status = rpc_call(clnt, proc, &args, res, 0);
  41. if (status < 0)
  42. printk(KERN_DEBUG "nsm_mon_unmon: rpc failed, status=%dn",
  43. status);
  44. else
  45. status = 0;
  46.  out:
  47. return status;
  48. }
  49. /*
  50.  * Set up monitoring of a remote host
  51.  */
  52. int
  53. nsm_monitor(struct nlm_host *host)
  54. {
  55. struct nsm_res res;
  56. int status;
  57. dprintk("lockd: nsm_monitor(%s)n", host->h_name);
  58. status = nsm_mon_unmon(host, SM_MON, &res);
  59. if (status < 0 || res.status != 0)
  60. printk(KERN_NOTICE "lockd: cannot monitor %sn", host->h_name);
  61. else
  62. host->h_monitored = 1;
  63. return status;
  64. }
  65. /*
  66.  * Cease to monitor remote host
  67.  */
  68. int
  69. nsm_unmonitor(struct nlm_host *host)
  70. {
  71. struct nsm_res res;
  72. int status;
  73. dprintk("lockd: nsm_unmonitor(%s)n", host->h_name);
  74. status = nsm_mon_unmon(host, SM_UNMON, &res);
  75. if (status < 0)
  76. printk(KERN_NOTICE "lockd: cannot unmonitor %sn", host->h_name);
  77. else
  78. host->h_monitored = 0;
  79. return status;
  80. }
  81. /*
  82.  * Create NSM client for the local host
  83.  */
  84. static struct rpc_clnt *
  85. nsm_create(void)
  86. {
  87. struct rpc_xprt *xprt;
  88. struct rpc_clnt *clnt = NULL;
  89. struct sockaddr_in sin;
  90. sin.sin_family = AF_INET;
  91. sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
  92. sin.sin_port = 0;
  93. xprt = xprt_create_proto(IPPROTO_UDP, &sin, NULL);
  94. if (!xprt)
  95. goto out;
  96. clnt = rpc_create_client(xprt, "localhost",
  97. &nsm_program, SM_VERSION,
  98. RPC_AUTH_NULL);
  99. if (!clnt)
  100. goto out_destroy;
  101. clnt->cl_softrtry = 1;
  102. clnt->cl_chatty   = 1;
  103. clnt->cl_oneshot  = 1;
  104. out:
  105. return clnt;
  106. out_destroy:
  107. xprt_destroy(xprt);
  108. goto out;
  109. }
  110. /*
  111.  * XDR functions for NSM.
  112.  */
  113. static int
  114. xdr_error(struct rpc_rqst *rqstp, u32 *p, void *dummy)
  115. {
  116. return -EACCES;
  117. }
  118. static int
  119. xdr_encode_mon(struct rpc_rqst *rqstp, u32 *p, struct nsm_args *argp)
  120. {
  121. char buffer[20];
  122. u32 addr = ntohl(argp->addr);
  123. dprintk("nsm: xdr_encode_mon(%08x, %d, %d, %d)n",
  124. htonl(argp->addr), htonl(argp->prog),
  125. htonl(argp->vers), htonl(argp->proc));
  126. /*
  127.  * Use the dotted-quad IP address of the remote host as
  128.  * identifier. Linux statd always looks up the canonical
  129.  * hostname first for whatever remote hostname it receives,
  130.  * so this works alright.
  131.  */
  132. sprintf(buffer, "%d.%d.%d.%d", (addr>>24) & 0xff, (addr>>16) & 0xff,
  133.   (addr>>8) & 0xff,  (addr) & 0xff);
  134. if (!(p = xdr_encode_string(p, buffer))
  135.  || !(p = xdr_encode_string(p, system_utsname.nodename)))
  136. return -EIO;
  137. *p++ = htonl(argp->prog);
  138. *p++ = htonl(argp->vers);
  139. *p++ = htonl(argp->proc);
  140. /* This is the private part. Needed only for SM_MON call */
  141. if (rqstp->rq_task->tk_msg.rpc_proc == SM_MON) {
  142. *p++ = argp->addr;
  143. *p++ = 0;
  144. *p++ = 0;
  145. *p++ = 0;
  146. }
  147. rqstp->rq_slen = xdr_adjust_iovec(rqstp->rq_svec, p);
  148. return 0;
  149. }
  150. static int
  151. xdr_decode_stat_res(struct rpc_rqst *rqstp, u32 *p, struct nsm_res *resp)
  152. {
  153. resp->status = ntohl(*p++);
  154. resp->state = ntohl(*p++);
  155. dprintk("nsm: xdr_decode_stat_res status %d state %dn",
  156. resp->status, resp->state);
  157. return 0;
  158. }
  159. static int
  160. xdr_decode_stat(struct rpc_rqst *rqstp, u32 *p, struct nsm_res *resp)
  161. {
  162. resp->state = ntohl(*p++);
  163. return 0;
  164. }
  165. #define SM_my_name_sz (1+XDR_QUADLEN(SM_MAXSTRLEN))
  166. #define SM_my_id_sz (3+1+SM_my_name_sz)
  167. #define SM_mon_id_sz (1+XDR_QUADLEN(20)+SM_my_id_sz)
  168. #define SM_mon_sz (SM_mon_id_sz+4)
  169. #define SM_monres_sz 2
  170. #define SM_unmonres_sz 1
  171. #ifndef MAX
  172. # define MAX(a, b) (((a) > (b))? (a) : (b))
  173. #endif
  174. static struct rpc_procinfo nsm_procedures[] = {
  175.         { "sm_null",
  176. (kxdrproc_t) xdr_error,
  177. (kxdrproc_t) xdr_error, 0, 0 },
  178.         { "sm_stat",
  179. (kxdrproc_t) xdr_error,
  180. (kxdrproc_t) xdr_error, 0, 0 },
  181.         { "sm_mon",
  182. (kxdrproc_t) xdr_encode_mon,
  183. (kxdrproc_t) xdr_decode_stat_res, MAX(SM_mon_sz, SM_monres_sz) << 2, 0 },
  184.         { "sm_unmon",
  185. (kxdrproc_t) xdr_encode_mon,
  186. (kxdrproc_t) xdr_decode_stat, MAX(SM_mon_id_sz, SM_unmonres_sz) << 2, 0 },
  187.         { "sm_unmon_all",
  188. (kxdrproc_t) xdr_error,
  189. (kxdrproc_t) xdr_error, 0, 0 },
  190.         { "sm_simu_crash",
  191. (kxdrproc_t) xdr_error,
  192. (kxdrproc_t) xdr_error, 0, 0 },
  193.         { "sm_notify",
  194. (kxdrproc_t) xdr_error,
  195. (kxdrproc_t) xdr_error, 0, 0 },
  196. };
  197. static struct rpc_version nsm_version1 = {
  198. 1, 
  199. sizeof(nsm_procedures)/sizeof(nsm_procedures[0]),
  200. nsm_procedures
  201. };
  202. static struct rpc_version * nsm_version[] = {
  203. NULL,
  204. &nsm_version1,
  205. };
  206. static struct rpc_stat nsm_stats;
  207. struct rpc_program nsm_program = {
  208. "statd",
  209. SM_PROGRAM,
  210. sizeof(nsm_version)/sizeof(nsm_version[0]),
  211. nsm_version,
  212. &nsm_stats
  213. };