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

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  * linux/fs/lockd/xdr4.c
  3.  *
  4.  * XDR support for lockd and the lock client.
  5.  *
  6.  * Copyright (C) 1995, 1996 Olaf Kirch <okir@monad.swb.de>
  7.  * Copyright (C) 1999, Trond Myklebust <trond.myklebust@fys.uio.no>
  8.  */
  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. s64_to_loff_t(__s64 offset)
  22. {
  23. return (loff_t)offset;
  24. }
  25. static inline s64
  26. loff_t_to_s64(loff_t offset)
  27. {
  28. s64 res;
  29. if (offset > NLM4_OFFSET_MAX)
  30. res = NLM4_OFFSET_MAX;
  31. else if (offset < -NLM4_OFFSET_MAX)
  32. res = -NLM4_OFFSET_MAX;
  33. else
  34. res = offset;
  35. return res;
  36. }
  37. /*
  38.  * XDR functions for basic NLM types
  39.  */
  40. static u32 *
  41. nlm4_decode_cookie(u32 *p, struct nlm_cookie *c)
  42. {
  43. unsigned int len;
  44. len = ntohl(*p++);
  45. if(len==0)
  46. {
  47. c->len=4;
  48. memset(c->data, 0, 4); /* hockeypux brain damage */
  49. }
  50. else if(len<=8)
  51. {
  52. c->len=len;
  53. memcpy(c->data, p, len);
  54. p+=(len+3)>>2;
  55. }
  56. else 
  57. {
  58. printk(KERN_NOTICE
  59. "lockd: bad cookie size %d (only cookies under 8 bytes are supported.)n", len);
  60. return NULL;
  61. }
  62. return p;
  63. }
  64. static u32 *
  65. nlm4_encode_cookie(u32 *p, struct nlm_cookie *c)
  66. {
  67. *p++ = htonl(c->len);
  68. memcpy(p, c->data, c->len);
  69. p+=(c->len+3)>>2;
  70. return p;
  71. }
  72. static u32 *
  73. nlm4_decode_fh(u32 *p, struct nfs_fh *f)
  74. {
  75. memset(f->data, 0, sizeof(f->data));
  76. f->size = ntohl(*p++);
  77. if (f->size > NFS_MAXFHSIZE) {
  78. printk(KERN_NOTICE
  79. "lockd: bad fhandle size %d (should be <=%d)n",
  80. f->size, NFS_MAXFHSIZE);
  81. return NULL;
  82. }
  83.        memcpy(f->data, p, f->size);
  84. return p + XDR_QUADLEN(f->size);
  85. }
  86. static u32 *
  87. nlm4_encode_fh(u32 *p, struct nfs_fh *f)
  88. {
  89. *p++ = htonl(f->size);
  90. if (f->size) p[XDR_QUADLEN(f->size)-1] = 0; /* don't leak anything */
  91. memcpy(p, f->data, f->size);
  92. return p + XDR_QUADLEN(f->size);
  93. }
  94. /*
  95.  * Encode and decode owner handle
  96.  */
  97. static u32 *
  98. nlm4_decode_oh(u32 *p, struct xdr_netobj *oh)
  99. {
  100. return xdr_decode_netobj(p, oh);
  101. }
  102. static u32 *
  103. nlm4_encode_oh(u32 *p, struct xdr_netobj *oh)
  104. {
  105. return xdr_encode_netobj(p, oh);
  106. }
  107. static u32 *
  108. nlm4_decode_lock(u32 *p, struct nlm_lock *lock)
  109. {
  110. struct file_lock *fl = &lock->fl;
  111. __s64 len, start, end;
  112. if (!(p = xdr_decode_string_inplace(p, &lock->caller,
  113.     &lock->len, NLM_MAXSTRLEN))
  114.  || !(p = nlm4_decode_fh(p, &lock->fh))
  115.  || !(p = nlm4_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. p = xdr_decode_hyper(p, &start);
  123. p = xdr_decode_hyper(p, &len);
  124. end = start + len - 1;
  125. fl->fl_start = s64_to_loff_t(start);
  126. if (len == 0 || end < 0)
  127. fl->fl_end = OFFSET_MAX;
  128. else
  129. fl->fl_end = s64_to_loff_t(end);
  130. return p;
  131. }
  132. /*
  133.  * Encode a lock as part of an NLM call
  134.  */
  135. static u32 *
  136. nlm4_encode_lock(u32 *p, struct nlm_lock *lock)
  137. {
  138. struct file_lock *fl = &lock->fl;
  139. __s64 start, len;
  140. if (!(p = xdr_encode_string(p, lock->caller))
  141.  || !(p = nlm4_encode_fh(p, &lock->fh))
  142.  || !(p = nlm4_encode_oh(p, &lock->oh)))
  143. return NULL;
  144. if (fl->fl_start > NLM4_OFFSET_MAX
  145.  || (fl->fl_end > NLM4_OFFSET_MAX && fl->fl_end != OFFSET_MAX))
  146. return NULL;
  147. *p++ = htonl(fl->fl_pid);
  148. start = loff_t_to_s64(fl->fl_start);
  149. if (fl->fl_end == OFFSET_MAX)
  150. len = 0;
  151. else
  152. len = loff_t_to_s64(fl->fl_end - fl->fl_start + 1);
  153. p = xdr_encode_hyper(p, start);
  154. p = xdr_encode_hyper(p, len);
  155. return p;
  156. }
  157. /*
  158.  * Encode result of a TEST/TEST_MSG call
  159.  */
  160. static u32 *
  161. nlm4_encode_testres(u32 *p, struct nlm_res *resp)
  162. {
  163. s64 start, len;
  164. dprintk("xdr: before encode_testres (p %p resp %p)n", p, resp);
  165. if (!(p = nlm4_encode_cookie(p, &resp->cookie)))
  166. return 0;
  167. *p++ = resp->status;
  168. if (resp->status == nlm_lck_denied) {
  169. struct file_lock *fl = &resp->lock.fl;
  170. *p++ = (fl->fl_type == F_RDLCK)? xdr_zero : xdr_one;
  171. *p++ = htonl(fl->fl_pid);
  172. /* Encode owner handle. */
  173. if (!(p = xdr_encode_netobj(p, &resp->lock.oh)))
  174. return 0;
  175. start = loff_t_to_s64(fl->fl_start);
  176. if (fl->fl_end == OFFSET_MAX)
  177. len = 0;
  178. else
  179. len = loff_t_to_s64(fl->fl_end - fl->fl_start + 1);
  180. p = xdr_encode_hyper(p, start);
  181. p = xdr_encode_hyper(p, len);
  182. dprintk("xdr: encode_testres (status %d pid %d type %d start %Ld end %Ld)n",
  183. resp->status, fl->fl_pid, fl->fl_type,
  184. (long long)fl->fl_start,  (long long)fl->fl_end);
  185. }
  186. dprintk("xdr: after encode_testres (p %p resp %p)n", p, resp);
  187. return p;
  188. }
  189. /*
  190.  * Check buffer bounds after decoding arguments
  191.  */
  192. static int
  193. xdr_argsize_check(struct svc_rqst *rqstp, u32 *p)
  194. {
  195. struct svc_buf *buf = &rqstp->rq_argbuf;
  196. return p - buf->base <= buf->buflen;
  197. }
  198. static int
  199. xdr_ressize_check(struct svc_rqst *rqstp, u32 *p)
  200. {
  201. struct svc_buf *buf = &rqstp->rq_resbuf;
  202. buf->len = p - buf->base;
  203. return (buf->len <= buf->buflen);
  204. }
  205. /*
  206.  * First, the server side XDR functions
  207.  */
  208. int
  209. nlm4svc_decode_testargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  210. {
  211. u32 exclusive;
  212. if (!(p = nlm4_decode_cookie(p, &argp->cookie)))
  213. return 0;
  214. exclusive = ntohl(*p++);
  215. if (!(p = nlm4_decode_lock(p, &argp->lock)))
  216. return 0;
  217. if (exclusive)
  218. argp->lock.fl.fl_type = F_WRLCK;
  219. return xdr_argsize_check(rqstp, p);
  220. }
  221. int
  222. nlm4svc_encode_testres(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  223. {
  224. if (!(p = nlm4_encode_testres(p, resp)))
  225. return 0;
  226. return xdr_ressize_check(rqstp, p);
  227. }
  228. int
  229. nlm4svc_decode_lockargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  230. {
  231. u32 exclusive;
  232. if (!(p = nlm4_decode_cookie(p, &argp->cookie)))
  233. return 0;
  234. argp->block  = ntohl(*p++);
  235. exclusive    = ntohl(*p++);
  236. if (!(p = nlm4_decode_lock(p, &argp->lock)))
  237. return 0;
  238. if (exclusive)
  239. argp->lock.fl.fl_type = F_WRLCK;
  240. argp->reclaim = ntohl(*p++);
  241. argp->state   = ntohl(*p++);
  242. argp->monitor = 1; /* monitor client by default */
  243. return xdr_argsize_check(rqstp, p);
  244. }
  245. int
  246. nlm4svc_decode_cancargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  247. {
  248. u32 exclusive;
  249. if (!(p = nlm4_decode_cookie(p, &argp->cookie)))
  250. return 0;
  251. argp->block = ntohl(*p++);
  252. exclusive = ntohl(*p++);
  253. if (!(p = nlm4_decode_lock(p, &argp->lock)))
  254. return 0;
  255. if (exclusive)
  256. argp->lock.fl.fl_type = F_WRLCK;
  257. return xdr_argsize_check(rqstp, p);
  258. }
  259. int
  260. nlm4svc_decode_unlockargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  261. {
  262. if (!(p = nlm4_decode_cookie(p, &argp->cookie))
  263.  || !(p = nlm4_decode_lock(p, &argp->lock)))
  264. return 0;
  265. argp->lock.fl.fl_type = F_UNLCK;
  266. return xdr_argsize_check(rqstp, p);
  267. }
  268. int
  269. nlm4svc_decode_shareargs(struct svc_rqst *rqstp, u32 *p, nlm_args *argp)
  270. {
  271. struct nlm_lock *lock = &argp->lock;
  272. memset(lock, 0, sizeof(*lock));
  273. locks_init_lock(&lock->fl);
  274. lock->fl.fl_pid = ~(u32) 0;
  275. if (!(p = nlm4_decode_cookie(p, &argp->cookie))
  276.  || !(p = xdr_decode_string_inplace(p, &lock->caller,
  277.     &lock->len, NLM_MAXSTRLEN))
  278.  || !(p = nlm4_decode_fh(p, &lock->fh))
  279.  || !(p = nlm4_decode_oh(p, &lock->oh)))
  280. return 0;
  281. argp->fsm_mode = ntohl(*p++);
  282. argp->fsm_access = ntohl(*p++);
  283. return xdr_argsize_check(rqstp, p);
  284. }
  285. int
  286. nlm4svc_encode_shareres(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  287. {
  288. if (!(p = nlm4_encode_cookie(p, &resp->cookie)))
  289. return 0;
  290. *p++ = resp->status;
  291. *p++ = xdr_zero; /* sequence argument */
  292. return xdr_ressize_check(rqstp, p);
  293. }
  294. int
  295. nlm4svc_encode_res(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  296. {
  297. if (!(p = nlm4_encode_cookie(p, &resp->cookie)))
  298. return 0;
  299. *p++ = resp->status;
  300. return xdr_ressize_check(rqstp, p);
  301. }
  302. int
  303. nlm4svc_decode_notify(struct svc_rqst *rqstp, u32 *p, struct nlm_args *argp)
  304. {
  305. struct nlm_lock *lock = &argp->lock;
  306. if (!(p = xdr_decode_string_inplace(p, &lock->caller,
  307.     &lock->len, NLM_MAXSTRLEN)))
  308. return 0;
  309. argp->state = ntohl(*p++);
  310. return xdr_argsize_check(rqstp, p);
  311. }
  312. int
  313. nlm4svc_decode_reboot(struct svc_rqst *rqstp, u32 *p, struct nlm_reboot *argp)
  314. {
  315. if (!(p = xdr_decode_string_inplace(p, &argp->mon, &argp->len, SM_MAXSTRLEN)))
  316. return 0;
  317. argp->state = ntohl(*p++);
  318. /* Preserve the address in network byte order */
  319. argp->addr = *p++;
  320. return xdr_argsize_check(rqstp, p);
  321. }
  322. int
  323. nlm4svc_decode_res(struct svc_rqst *rqstp, u32 *p, struct nlm_res *resp)
  324. {
  325. if (!(p = nlm4_decode_cookie(p, &resp->cookie)))
  326. return 0;
  327. resp->status = ntohl(*p++);
  328. return xdr_argsize_check(rqstp, p);
  329. }
  330. int
  331. nlm4svc_decode_void(struct svc_rqst *rqstp, u32 *p, void *dummy)
  332. {
  333. return xdr_argsize_check(rqstp, p);
  334. }
  335. int
  336. nlm4svc_encode_void(struct svc_rqst *rqstp, u32 *p, void *dummy)
  337. {
  338. return xdr_ressize_check(rqstp, p);
  339. }
  340. /*
  341.  * Now, the client side XDR functions
  342.  */
  343. static int
  344. nlm4clt_encode_void(struct rpc_rqst *req, u32 *p, void *ptr)
  345. {
  346. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  347. return 0;
  348. }
  349. static int
  350. nlm4clt_decode_void(struct rpc_rqst *req, u32 *p, void *ptr)
  351. {
  352. return 0;
  353. }
  354. static int
  355. nlm4clt_encode_testargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  356. {
  357. struct nlm_lock *lock = &argp->lock;
  358. if (!(p = nlm4_encode_cookie(p, &argp->cookie)))
  359. return -EIO;
  360. *p++ = (lock->fl.fl_type == F_WRLCK)? xdr_one : xdr_zero;
  361. if (!(p = nlm4_encode_lock(p, lock)))
  362. return -EIO;
  363. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  364. return 0;
  365. }
  366. static int
  367. nlm4clt_decode_testres(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  368. {
  369. if (!(p = nlm4_decode_cookie(p, &resp->cookie)))
  370. return -EIO;
  371. resp->status = ntohl(*p++);
  372. if (resp->status == NLM_LCK_DENIED) {
  373. struct file_lock *fl = &resp->lock.fl;
  374. u32 excl;
  375. s64 start, end, len;
  376. memset(&resp->lock, 0, sizeof(resp->lock));
  377. locks_init_lock(fl);
  378. excl = ntohl(*p++);
  379. fl->fl_pid = ntohl(*p++);
  380. if (!(p = nlm4_decode_oh(p, &resp->lock.oh)))
  381. return -EIO;
  382. fl->fl_flags = FL_POSIX;
  383. fl->fl_type  = excl? F_WRLCK : F_RDLCK;
  384. p = xdr_decode_hyper(p, &start);
  385. p = xdr_decode_hyper(p, &len);
  386. end = start + len - 1;
  387. fl->fl_start = s64_to_loff_t(start);
  388. if (len == 0 || end < 0)
  389. fl->fl_end = OFFSET_MAX;
  390. else
  391. fl->fl_end = s64_to_loff_t(end);
  392. }
  393. return 0;
  394. }
  395. static int
  396. nlm4clt_encode_lockargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  397. {
  398. struct nlm_lock *lock = &argp->lock;
  399. if (!(p = nlm4_encode_cookie(p, &argp->cookie)))
  400. return -EIO;
  401. *p++ = argp->block? xdr_one : xdr_zero;
  402. *p++ = (lock->fl.fl_type == F_WRLCK)? xdr_one : xdr_zero;
  403. if (!(p = nlm4_encode_lock(p, lock)))
  404. return -EIO;
  405. *p++ = argp->reclaim? xdr_one : xdr_zero;
  406. *p++ = htonl(argp->state);
  407. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  408. return 0;
  409. }
  410. static int
  411. nlm4clt_encode_cancargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  412. {
  413. struct nlm_lock *lock = &argp->lock;
  414. if (!(p = nlm4_encode_cookie(p, &argp->cookie)))
  415. return -EIO;
  416. *p++ = argp->block? xdr_one : xdr_zero;
  417. *p++ = (lock->fl.fl_type == F_WRLCK)? xdr_one : xdr_zero;
  418. if (!(p = nlm4_encode_lock(p, lock)))
  419. return -EIO;
  420. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  421. return 0;
  422. }
  423. static int
  424. nlm4clt_encode_unlockargs(struct rpc_rqst *req, u32 *p, nlm_args *argp)
  425. {
  426. struct nlm_lock *lock = &argp->lock;
  427. if (!(p = nlm4_encode_cookie(p, &argp->cookie)))
  428. return -EIO;
  429. if (!(p = nlm4_encode_lock(p, lock)))
  430. return -EIO;
  431. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  432. return 0;
  433. }
  434. static int
  435. nlm4clt_encode_res(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  436. {
  437. if (!(p = nlm4_encode_cookie(p, &resp->cookie)))
  438. return -EIO;
  439. *p++ = resp->status;
  440. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  441. return 0;
  442. }
  443. static int
  444. nlm4clt_encode_testres(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  445. {
  446. if (!(p = nlm4_encode_testres(p, resp)))
  447. return -EIO;
  448. req->rq_slen = xdr_adjust_iovec(req->rq_svec, p);
  449. return 0;
  450. }
  451. static int
  452. nlm4clt_decode_res(struct rpc_rqst *req, u32 *p, struct nlm_res *resp)
  453. {
  454. if (!(p = nlm4_decode_cookie(p, &resp->cookie)))
  455. return -EIO;
  456. resp->status = ntohl(*p++);
  457. return 0;
  458. }
  459. /*
  460.  * Buffer requirements for NLM
  461.  */
  462. #define NLM4_void_sz 0
  463. #define NLM4_cookie_sz 3 /* 1 len , 2 data */
  464. #define NLM4_caller_sz 1+XDR_QUADLEN(NLM_MAXSTRLEN)
  465. #define NLM4_netobj_sz 1+XDR_QUADLEN(XDR_MAX_NETOBJ)
  466. /* #define NLM4_owner_sz 1+XDR_QUADLEN(NLM4_MAXOWNER) */
  467. #define NLM4_fhandle_sz 1+XDR_QUADLEN(NFS3_FHSIZE)
  468. #define NLM4_lock_sz 5+NLM4_caller_sz+NLM4_netobj_sz+NLM4_fhandle_sz
  469. #define NLM4_holder_sz 6+NLM4_netobj_sz
  470. #define NLM4_testargs_sz NLM4_cookie_sz+1+NLM4_lock_sz
  471. #define NLM4_lockargs_sz NLM4_cookie_sz+4+NLM4_lock_sz
  472. #define NLM4_cancargs_sz NLM4_cookie_sz+2+NLM4_lock_sz
  473. #define NLM4_unlockargs_sz NLM4_cookie_sz+NLM4_lock_sz
  474. #define NLM4_testres_sz NLM4_cookie_sz+1+NLM4_holder_sz
  475. #define NLM4_res_sz NLM4_cookie_sz+1
  476. #define NLM4_norep_sz 0
  477. #ifndef MAX
  478. # define MAX(a,b) (((a) > (b))? (a) : (b))
  479. #endif
  480. /*
  481.  * For NLM, a void procedure really returns nothing
  482.  */
  483. #define nlm4clt_decode_norep NULL
  484. #define PROC(proc, argtype, restype)
  485.     { "nlm4_" #proc,
  486.       (kxdrproc_t) nlm4clt_encode_##argtype,
  487.       (kxdrproc_t) nlm4clt_decode_##restype,
  488.       MAX(NLM4_##argtype##_sz, NLM4_##restype##_sz) << 2,
  489.       0
  490.     }
  491. static struct rpc_procinfo nlm4_procedures[] = {
  492.     PROC(null, void, void),
  493.     PROC(test, testargs, testres),
  494.     PROC(lock, lockargs, res),
  495.     PROC(canc, cancargs, res),
  496.     PROC(unlock, unlockargs, res),
  497.     PROC(granted, testargs, res),
  498.     PROC(test_msg, testargs, norep),
  499.     PROC(lock_msg, lockargs, norep),
  500.     PROC(canc_msg, cancargs, norep),
  501.     PROC(unlock_msg, unlockargs, norep),
  502.     PROC(granted_msg, testargs, norep),
  503.     PROC(test_res, testres, norep),
  504.     PROC(lock_res, res, norep),
  505.     PROC(canc_res, res, norep),
  506.     PROC(unlock_res, res, norep),
  507.     PROC(granted_res, res, norep),
  508.     PROC(undef, void, void),
  509.     PROC(undef, void, void),
  510.     PROC(undef, void, void),
  511.     PROC(undef, void, void),
  512. #ifdef NLMCLNT_SUPPORT_SHARES
  513.     PROC(share, shareargs, shareres),
  514.     PROC(unshare, shareargs, shareres),
  515.     PROC(nm_lock, lockargs, res),
  516.     PROC(free_all, notify, void),
  517. #else
  518.     PROC(undef, void, void),
  519.     PROC(undef, void, void),
  520.     PROC(undef, void, void),
  521.     PROC(undef, void, void),
  522. #endif
  523. };
  524. struct rpc_version nlm_version4 = {
  525. 4, 24, nlm4_procedures,
  526. };