xdr.c
上传用户:jlfgdled
上传日期:2013-04-10
资源大小:33168k
文件大小:14k
源码类别:

Linux/Unix编程

开发平台:

Unix_Linux

  1. /*
  2.  * linux/fs/lockd/xdr.c
  3.  *
  4.  * XDR support for lockd and the lock client.
  5.  *
  6.  * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de>
  7.  */
  8. #include <linux/config.h>
  9. #include <linux/types.h>
  10. #include <linux/sched.h>
  11. #include <linux/utsname.h>
  12. #include <linux/nfs.h>
  13. #include <linux/sunrpc/xdr.h>
  14. #include <linux/sunrpc/clnt.h>
  15. #include <linux/sunrpc/svc.h>
  16. #include <linux/sunrpc/stats.h>
  17. #include <linux/lockd/lockd.h>
  18. #include <linux/lockd/sm_inter.h>
  19. #define NLMDBG_FACILITY NLMDBG_XDR
  20. static inline loff_t
  21. s32_to_loff_t(__s32 offset)
  22. {
  23. return (loff_t)offset;
  24. }
  25. static inline __s32
  26. loff_t_to_s32(loff_t offset)
  27. {
  28. __s32 res;
  29. if (offset >= NLM_OFFSET_MAX)
  30. res = NLM_OFFSET_MAX;
  31. else if (offset <= -NLM_OFFSET_MAX)
  32. res = -NLM_OFFSET_MAX;
  33. else
  34. res = offset;
  35. return res;
  36. }
  37. /*
  38.  * XDR functions for basic NLM types
  39.  */
  40. static inline u32 *nlm_decode_cookie(u32 *p, struct nlm_cookie *c)
  41. {
  42. unsigned int len;
  43. len = ntohl(*p++);
  44. if(len==0)
  45. {
  46. c->len=4;
  47. memset(c->data, 0, 4); /* hockeypux brain damage */
  48. }
  49. else if(len<=8)
  50. {
  51. c->len=len;
  52. memcpy(c->data, p, len);
  53. p+=(len+3)>>2;
  54. }
  55. else 
  56. {
  57. printk(KERN_NOTICE
  58. "lockd: bad cookie size %d (only cookies under 8 bytes are supported.)n", len);
  59. return NULL;
  60. }
  61. return p;
  62. }
  63. static inline u32 *
  64. nlm_encode_cookie(u32 *p, struct nlm_cookie *c)
  65. {
  66. *p++ = htonl(c->len);
  67. memcpy(p, c->data, c->len);
  68. p+=(c->len+3)>>2;
  69. return p;
  70. }
  71. static inline u32 *
  72. nlm_decode_fh(u32 *p, struct nfs_fh *f)
  73. {
  74. unsigned int len;
  75. if ((len = ntohl(*p++)) != NFS2_FHSIZE) {
  76. printk(KERN_NOTICE
  77. "lockd: bad fhandle size %x (should be %d)n",
  78. len, NFS2_FHSIZE);
  79. return NULL;
  80. }
  81. f->size = NFS2_FHSIZE;
  82. memset(f->data, 0, sizeof(f->data));
  83. memcpy(f->data, p, NFS2_FHSIZE);
  84. return p + XDR_QUADLEN(NFS2_FHSIZE);
  85. }
  86. static inline u32 *
  87. nlm_encode_fh(u32 *p, struct nfs_fh *f)
  88. {
  89. *p++ = htonl(NFS2_FHSIZE);
  90. memcpy(p, f->data, NFS2_FHSIZE);
  91. return p + XDR_QUADLEN(NFS2_FHSIZE);
  92. }
  93. /*
  94.  * Encode and decode owner handle
  95.  */
  96. static inline u32 *
  97. nlm_decode_oh(u32 *p, struct xdr_netobj *oh)
  98. {
  99. return xdr_decode_netobj(p, oh);
  100. }
  101. static inline u32 *
  102. nlm_encode_oh(u32 *p, struct xdr_netobj *oh)
  103. {
  104. return xdr_encode_netobj(p, oh);
  105. }
  106. static inline u32 *
  107. nlm_decode_lock(u32 *p, struct nlm_lock *lock)
  108. {
  109. struct file_lock *fl = &lock->fl;
  110. s32 start, len, end;
  111. if (!(p = xdr_decode_string_inplace(p, &lock->caller,
  112.     &lock->len,
  113.     NLM_MAXSTRLEN))
  114.  || !(p = nlm_decode_fh(p, &lock->fh))
  115.  || !(p = nlm_decode_oh(p, &lock->oh)))
  116. return NULL;
  117. locks_init_lock(fl);
  118. fl->fl_owner = current->files;
  119. fl->fl_pid   = ntohl(*p++);
  120. fl->fl_flags = FL_POSIX;
  121. fl->fl_type  = F_RDLCK; /* as good as anything else */
  122. start = ntohl(*p++);
  123. len = ntohl(*p++);
  124. end = start + len - 1;
  125. fl->fl_start = s32_to_loff_t(start);
  126. if (len == 0 || end < 0)
  127. fl->fl_end = OFFSET_MAX;
  128. else
  129. fl->fl_end = s32_to_loff_t(end);
  130. return p;
  131. }
  132. /*
  133.  * Encode a lock as part of an NLM call
  134.  */
  135. static u32 *
  136. nlm_encode_lock(u32 *p, struct nlm_lock *lock)
  137. {
  138. struct file_lock *fl = &lock->fl;
  139. __s32 start, len;
  140. if (!(p = xdr_encode_string(p, lock->caller))
  141.  || !(p = nlm_encode_fh(p, &lock->fh))
  142.  || !(p = nlm_encode_oh(p, &lock->oh)))
  143. return NULL;
  144. if (fl->fl_start > NLM_OFFSET_MAX
  145.  || (fl->fl_end > NLM_OFFSET_MAX && fl->fl_end != OFFSET_MAX))
  146. return NULL;
  147. start = loff_t_to_s32(fl->fl_start);
  148. if (fl->fl_end == OFFSET_MAX)
  149. len = 0;
  150. else
  151. len = loff_t_to_s32(fl->fl_end - fl->fl_start + 1);
  152. *p++ = htonl(fl->fl_pid);
  153. *p++ = htonl(start);
  154. *p++ = htonl(len);
  155. return p;
  156. }
  157. /*
  158.  * Encode result of a TEST/TEST_MSG call
  159.  */
  160. static u32 *
  161. nlm_encode_testres(u32 *p, struct nlm_res *resp)
  162. {
  163. s32 start, len;
  164. if (!(p = nlm_encode_cookie(p, &resp->cookie)))
  165. return 0;
  166. *p++ = resp->status;
  167. if (resp->status == nlm_lck_denied) {
  168. struct file_lock *fl = &resp->lock.fl;
  169. *p++ = (fl->fl_type == F_RDLCK)? xdr_zero : xdr_one;
  170. *p++ = htonl(fl->fl_pid);
  171. /* Encode owner handle. */
  172. if (!(p = xdr_encode_netobj(p, &resp->lock.oh)))
  173. return 0;
  174. start = loff_t_to_s32(fl->fl_start);
  175. if (fl->fl_end == OFFSET_MAX)
  176. len = 0;
  177. else
  178. len = loff_t_to_s32(fl->fl_end - fl->fl_start + 1);
  179. *p++ = htonl(start);
  180. *p++ = htonl(len);
  181. }
  182. return p;
  183. }
  184. /*
  185.  * Check buffer bounds after decoding arguments
  186.  */
  187. static inline int
  188. xdr_argsize_check(struct svc_rqst *rqstp, u32 *p)
  189. {
  190. struct svc_buf *buf = &rqstp->rq_argbuf;
  191. return p - buf->base <= buf->buflen;
  192. }
  193. static inline int
  194. xdr_ressize_check(struct svc_rqst *rqstp, u32 *p)
  195. {
  196. struct svc_buf *buf = &rqstp->rq_resbuf;
  197. buf->len = p - buf->base;
  198. return (buf->len <= buf->buflen);
  199. }
  200. /*
  201.  * First, the server side XDR functions
  202.  */
  203. int
  204. nlmsvc_decode_testargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  205. {
  206. u32 exclusive;
  207. if (!(p = nlm_decode_cookie(p, &argp->cookie)))
  208. return 0;
  209. exclusive = ntohl(*p++);
  210. if (!(p = nlm_decode_lock(p, &argp->lock)))
  211. return 0;
  212. if (exclusive)
  213. argp->lock.fl.fl_type = F_WRLCK;
  214. return xdr_argsize_check(rqstp, p);
  215. }
  216. int
  217. nlmsvc_encode_testres(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  218. {
  219. if (!(p = nlm_encode_testres(p, resp)))
  220. return 0;
  221. return xdr_ressize_check(rqstp, p);
  222. }
  223. int
  224. nlmsvc_decode_lockargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  225. {
  226. u32 exclusive;
  227. if (!(p = nlm_decode_cookie(p, &argp->cookie)))
  228. return 0;
  229. argp->block  = ntohl(*p++);
  230. exclusive    = ntohl(*p++);
  231. if (!(p = nlm_decode_lock(p, &argp->lock)))
  232. return 0;
  233. if (exclusive)
  234. argp->lock.fl.fl_type = F_WRLCK;
  235. argp->reclaim = ntohl(*p++);
  236. argp->state   = ntohl(*p++);
  237. argp->monitor = 1; /* monitor client by default */
  238. return xdr_argsize_check(rqstp, p);
  239. }
  240. int
  241. nlmsvc_decode_cancargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  242. {
  243. u32 exclusive;
  244. if (!(p = nlm_decode_cookie(p, &argp->cookie)))
  245. return 0;
  246. argp->block = ntohl(*p++);
  247. exclusive = ntohl(*p++);
  248. if (!(p = nlm_decode_lock(p, &argp->lock)))
  249. return 0;
  250. if (exclusive)
  251. argp->lock.fl.fl_type = F_WRLCK;
  252. return xdr_argsize_check(rqstp, p);
  253. }
  254. int
  255. nlmsvc_decode_unlockargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  256. {
  257. if (!(p = nlm_decode_cookie(p, &argp->cookie))
  258.  || !(p = nlm_decode_lock(p, &argp->lock)))
  259. return 0;
  260. argp->lock.fl.fl_type = F_UNLCK;
  261. return xdr_argsize_check(rqstp, p);
  262. }
  263. int
  264. nlmsvc_decode_shareargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  265. {
  266. struct nlm_lock *lock = &argp->lock;
  267. memset(lock, 0, sizeof(*lock));
  268. locks_init_lock(&lock->fl);
  269. lock->fl.fl_pid = ~(u32) 0;
  270. if (!(p = nlm_decode_cookie(p, &argp->cookie))
  271.  || !(p = xdr_decode_string_inplace(p, &lock->caller,
  272.     &lock->len, NLM_MAXSTRLEN))
  273.  || !(p = nlm_decode_fh(p, &lock->fh))
  274.  || !(p = nlm_decode_oh(p, &lock->oh)))
  275. return 0;
  276. argp->fsm_mode = ntohl(*p++);
  277. argp->fsm_access = ntohl(*p++);
  278. return xdr_argsize_check(rqstp, p);
  279. }
  280. int
  281. nlmsvc_encode_shareres(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  282. {
  283. if (!(p = nlm_encode_cookie(p, &resp->cookie)))
  284. return 0;
  285. *p++ = resp->status;
  286. *p++ = xdr_zero; /* sequence argument */
  287. return xdr_ressize_check(rqstp, p);
  288. }
  289. int
  290. nlmsvc_encode_res(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  291. {
  292. if (!(p = nlm_encode_cookie(p, &resp->cookie)))
  293. return 0;
  294. *p++ = resp->status;
  295. return xdr_ressize_check(rqstp, p);
  296. }
  297. int
  298. nlmsvc_decode_notify(struct svc_rqst *rqstp, u32 *p, struct nlm_args *argp)
  299. {
  300. struct nlm_lock *lock = &argp->lock;
  301. if (!(p = xdr_decode_string_inplace(p, &lock->caller,
  302.     &lock->len, NLM_MAXSTRLEN)))
  303. return 0;
  304. argp->state = ntohl(*p++);
  305. return xdr_argsize_check(rqstp, p);
  306. }
  307. int
  308. nlmsvc_decode_reboot(struct svc_rqst *rqstp, u32 *p, struct nlm_reboot *argp)
  309. {
  310. if (!(p = xdr_decode_string_inplace(p, &argp->mon, &argp->len, SM_MAXSTRLEN)))
  311. return 0;
  312. argp->state = ntohl(*p++);
  313. /* Preserve the address in network byte order */
  314. argp->addr = *p++;
  315. return xdr_argsize_check(rqstp, p);
  316. }
  317. int
  318. nlmsvc_decode_res(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  319. {
  320. if (!(p = nlm_decode_cookie(p, &resp->cookie)))
  321. return 0;
  322. resp->status = ntohl(*p++);
  323. return xdr_argsize_check(rqstp, p);
  324. }
  325. int
  326. nlmsvc_decode_void(struct svc_rqst *rqstp, u32 *p, void *dummy)
  327. {
  328. return xdr_argsize_check(rqstp, p);
  329. }
  330. int
  331. nlmsvc_encode_void(struct svc_rqst *rqstp, u32 *p, void *dummy)
  332. {
  333. return xdr_ressize_check(rqstp, p);
  334. }
  335. /*
  336.  * Now, the client side XDR functions
  337.  */
  338. static int
  339. nlmclt_encode_void(struct rpc_rqst *req, u32 *p, void *ptr)
  340. {
  341. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  342. return 0;
  343. }
  344. static int
  345. nlmclt_decode_void(struct rpc_rqst *req, u32 *p, void *ptr)
  346. {
  347. return 0;
  348. }
  349. static int
  350. nlmclt_encode_testargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  351. {
  352. struct nlm_lock *lock = &argp->lock;
  353. if (!(p = nlm_encode_cookie(p, &argp->cookie)))
  354. return -EIO;
  355. *p++ = (lock->fl.fl_type == F_WRLCK)? xdr_one : xdr_zero;
  356. if (!(p = nlm_encode_lock(p, lock)))
  357. return -EIO;
  358. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  359. return 0;
  360. }
  361. static int
  362. nlmclt_decode_testres(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  363. {
  364. if (!(p = nlm_decode_cookie(p, &resp->cookie)))
  365. return -EIO;
  366. resp->status = ntohl(*p++);
  367. if (resp->status == NLM_LCK_DENIED) {
  368. struct file_lock *fl = &resp->lock.fl;
  369. u32 excl;
  370. s32 start, len, end;
  371. memset(&resp->lock, 0, sizeof(resp->lock));
  372. locks_init_lock(fl);
  373. excl = ntohl(*p++);
  374. fl->fl_pid = ntohl(*p++);
  375. if (!(p = nlm_decode_oh(p, &resp->lock.oh)))
  376. return -EIO;
  377. fl->fl_flags = FL_POSIX;
  378. fl->fl_type  = excl? F_WRLCK : F_RDLCK;
  379. start = ntohl(*p++);
  380. len = ntohl(*p++);
  381. end = start + len - 1;
  382. fl->fl_start = s32_to_loff_t(start);
  383. if (len == 0 || end < 0)
  384. fl->fl_end = OFFSET_MAX;
  385. else
  386. fl->fl_end = s32_to_loff_t(end);
  387. }
  388. return 0;
  389. }
  390. static int
  391. nlmclt_encode_lockargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  392. {
  393. struct nlm_lock *lock = &argp->lock;
  394. if (!(p = nlm_encode_cookie(p, &argp->cookie)))
  395. return -EIO;
  396. *p++ = argp->block? xdr_one : xdr_zero;
  397. *p++ = (lock->fl.fl_type == F_WRLCK)? xdr_one : xdr_zero;
  398. if (!(p = nlm_encode_lock(p, lock)))
  399. return -EIO;
  400. *p++ = argp->reclaim? xdr_one : xdr_zero;
  401. *p++ = htonl(argp->state);
  402. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  403. return 0;
  404. }
  405. static int
  406. nlmclt_encode_cancargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  407. {
  408. struct nlm_lock *lock = &argp->lock;
  409. if (!(p = nlm_encode_cookie(p, &argp->cookie)))
  410. return -EIO;
  411. *p++ = argp->block? xdr_one : xdr_zero;
  412. *p++ = (lock->fl.fl_type == F_WRLCK)? xdr_one : xdr_zero;
  413. if (!(p = nlm_encode_lock(p, lock)))
  414. return -EIO;
  415. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  416. return 0;
  417. }
  418. static int
  419. nlmclt_encode_unlockargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  420. {
  421. struct nlm_lock *lock = &argp->lock;
  422. if (!(p = nlm_encode_cookie(p, &argp->cookie)))
  423. return -EIO;
  424. if (!(p = nlm_encode_lock(p, lock)))
  425. return -EIO;
  426. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  427. return 0;
  428. }
  429. static int
  430. nlmclt_encode_res(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  431. {
  432. if (!(p = nlm_encode_cookie(p, &resp->cookie)))
  433. return -EIO;
  434. *p++ = resp->status;
  435. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  436. return 0;
  437. }
  438. static int
  439. nlmclt_encode_testres(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  440. {
  441. if (!(p = nlm_encode_testres(p, resp)))
  442. return -EIO;
  443. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  444. return 0;
  445. }
  446. static int
  447. nlmclt_decode_res(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  448. {
  449. if (!(p = nlm_decode_cookie(p, &resp->cookie)))
  450. return -EIO;
  451. resp->status = ntohl(*p++);
  452. return 0;
  453. }
  454. /*
  455.  * Buffer requirements for NLM
  456.  */
  457. #define NLM_void_sz 0
  458. #define NLM_cookie_sz 3 /* 1 len , 2 data */
  459. #define NLM_caller_sz 1+QUADLEN(sizeof(system_utsname.nodename))
  460. #define NLM_netobj_sz 1+QUADLEN(XDR_MAX_NETOBJ)
  461. /* #define NLM_owner_sz 1+QUADLEN(NLM_MAXOWNER) */
  462. #define NLM_fhandle_sz 1+QUADLEN(NFS2_FHSIZE)
  463. #define NLM_lock_sz 3+NLM_caller_sz+NLM_netobj_sz+NLM_fhandle_sz
  464. #define NLM_holder_sz 4+NLM_netobj_sz
  465. #define NLM_testargs_sz NLM_cookie_sz+1+NLM_lock_sz
  466. #define NLM_lockargs_sz NLM_cookie_sz+4+NLM_lock_sz
  467. #define NLM_cancargs_sz NLM_cookie_sz+2+NLM_lock_sz
  468. #define NLM_unlockargs_sz NLM_cookie_sz+NLM_lock_sz
  469. #define NLM_testres_sz NLM_cookie_sz+1+NLM_holder_sz
  470. #define NLM_res_sz NLM_cookie_sz+1
  471. #define NLM_norep_sz 0
  472. #ifndef MAX
  473. # define MAX(a, b) (((a) > (b))? (a) : (b))
  474. #endif
  475. /*
  476.  * For NLM, a void procedure really returns nothing
  477.  */
  478. #define nlmclt_decode_norep NULL
  479. #define PROC(proc, argtype, restype)
  480.     { .p_procname  = "nlm_" #proc,
  481.       .p_encode    = (kxdrproc_t) nlmclt_encode_##argtype,
  482.       .p_decode    = (kxdrproc_t) nlmclt_decode_##restype,
  483.       .p_bufsiz    = MAX(NLM_##argtype##_sz, NLM_##restype##_sz) << 2
  484.     }
  485. static struct rpc_procinfo nlm_procedures[] = {
  486.     PROC(null, void, void),
  487.     PROC(test, testargs, testres),
  488.     PROC(lock, lockargs, res),
  489.     PROC(canc, cancargs, res),
  490.     PROC(unlock, unlockargs, res),
  491.     PROC(granted, testargs, res),
  492.     PROC(test_msg, testargs, norep),
  493.     PROC(lock_msg, lockargs, norep),
  494.     PROC(canc_msg, cancargs, norep),
  495.     PROC(unlock_msg, unlockargs, norep),
  496.     PROC(granted_msg, testargs, norep),
  497.     PROC(test_res, testres, norep),
  498.     PROC(lock_res, res, norep),
  499.     PROC(canc_res, res, norep),
  500.     PROC(unlock_res, res, norep),
  501.     PROC(granted_res, res, norep),
  502.     PROC(undef, void, void),
  503.     PROC(undef, void, void),
  504.     PROC(undef, void, void),
  505.     PROC(undef, void, void),
  506. #ifdef NLMCLNT_SUPPORT_SHARES
  507.     PROC(share, shareargs, shareres),
  508.     PROC(unshare, shareargs, shareres),
  509.     PROC(nm_lock, lockargs, res),
  510.     PROC(free_all, notify, void),
  511. #else
  512.     PROC(undef, void, void),
  513.     PROC(undef, void, void),
  514.     PROC(undef, void, void),
  515.     PROC(undef, void, void),
  516. #endif
  517. };
  518. static struct rpc_version nlm_version1 = {
  519. 1, 16, nlm_procedures,
  520. };
  521. static struct rpc_version nlm_version3 = {
  522. 3, 24, nlm_procedures,
  523. };
  524. #ifdef  CONFIG_LOCKD_V4
  525. extern struct rpc_version nlm_version4;
  526. #endif
  527. static struct rpc_version * nlm_versions[] = {
  528. NULL,
  529. &nlm_version1,
  530. NULL,
  531. &nlm_version3,
  532. #ifdef  CONFIG_LOCKD_V4
  533. &nlm_version4,
  534. #endif
  535. };
  536. static struct rpc_stat nlm_stats;
  537. struct rpc_program nlm_program = {
  538. "lockd",
  539. NLM_PROGRAM,
  540. sizeof(nlm_versions) / sizeof(nlm_versions[0]),
  541. nlm_versions,
  542. &nlm_stats,
  543. };
  544. #ifdef LOCKD_DEBUG
  545. char *
  546. nlm_procname(u32 proc)
  547. {
  548. if (proc < sizeof(nlm_procedures)/sizeof(nlm_procedures[0]))
  549. return nlm_procedures[proc].p_procname;
  550. return "unknown";
  551. }
  552. #endif