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

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  *  linux/net/sunrpc/rpcclnt.c
  3.  *
  4.  *  This file contains the high-level RPC interface.
  5.  *  It is modeled as a finite state machine to support both synchronous
  6.  *  and asynchronous requests.
  7.  *
  8.  *  - RPC header generation and argument serialization.
  9.  *  - Credential refresh.
  10.  *  - TCP reconnect handling (when finished).
  11.  *  - Retry of operation when it is suspected the operation failed because
  12.  * of uid squashing on the server, or when the credentials were stale
  13.  * and need to be refreshed, or when a packet was damaged in transit.
  14.  * This may be have to be moved to the VFS layer.
  15.  *
  16.  *  NB: BSD uses a more intelligent approach to guessing when a request
  17.  *  or reply has been lost by keeping the RTO estimate for each procedure.
  18.  *  We currently make do with a constant timeout value.
  19.  *
  20.  *  Copyright (C) 1992,1993 Rick Sladkey <jrs@world.std.com>
  21.  *  Copyright (C) 1995,1996 Olaf Kirch <okir@monad.swb.de>
  22.  */
  23. #include <asm/system.h>
  24. #include <linux/types.h>
  25. #include <linux/mm.h>
  26. #include <linux/slab.h>
  27. #include <linux/in.h>
  28. #include <linux/utsname.h>
  29. #include <linux/sunrpc/clnt.h>
  30. #include <linux/nfs.h>
  31. #define RPC_SLACK_SPACE 512 /* total overkill */
  32. #ifdef RPC_DEBUG
  33. # define RPCDBG_FACILITY RPCDBG_CALL
  34. #endif
  35. static DECLARE_WAIT_QUEUE_HEAD(destroy_wait);
  36. static void call_reserve(struct rpc_task *task);
  37. static void call_reserveresult(struct rpc_task *task);
  38. static void call_allocate(struct rpc_task *task);
  39. static void call_encode(struct rpc_task *task);
  40. static void call_decode(struct rpc_task *task);
  41. static void call_bind(struct rpc_task *task);
  42. static void call_transmit(struct rpc_task *task);
  43. static void call_status(struct rpc_task *task);
  44. static void call_refresh(struct rpc_task *task);
  45. static void call_refreshresult(struct rpc_task *task);
  46. static void call_timeout(struct rpc_task *task);
  47. static void call_reconnect(struct rpc_task *task);
  48. static void child_reconnect(struct rpc_task *);
  49. static void child_reconnect_status(struct rpc_task *);
  50. static u32 * call_header(struct rpc_task *task);
  51. static u32 * call_verify(struct rpc_task *task);
  52. /*
  53.  * Create an RPC client
  54.  * FIXME: This should also take a flags argument (as in task->tk_flags).
  55.  * It's called (among others) from pmap_create_client, which may in
  56.  * turn be called by an async task. In this case, rpciod should not be
  57.  * made to sleep too long.
  58.  */
  59. struct rpc_clnt *
  60. rpc_create_client(struct rpc_xprt *xprt, char *servname,
  61.   struct rpc_program *program, u32 vers, int flavor)
  62. {
  63. struct rpc_version *version;
  64. struct rpc_clnt *clnt = NULL;
  65. dprintk("RPC: creating %s client for %s (xprt %p)n",
  66. program->name, servname, xprt);
  67. #ifdef RPC_DEBUG
  68. rpc_register_sysctl();
  69. #endif
  70. if (!xprt)
  71. goto out;
  72. if (vers >= program->nrvers || !(version = program->version[vers]))
  73. goto out;
  74. clnt = (struct rpc_clnt *) rpc_allocate(0, sizeof(*clnt));
  75. if (!clnt)
  76. goto out_no_clnt;
  77. memset(clnt, 0, sizeof(*clnt));
  78. atomic_set(&clnt->cl_users, 0);
  79. clnt->cl_xprt     = xprt;
  80. clnt->cl_procinfo = version->procs;
  81. clnt->cl_maxproc  = version->nrprocs;
  82. clnt->cl_server   = servname;
  83. clnt->cl_protname = program->name;
  84. clnt->cl_port     = xprt->addr.sin_port;
  85. clnt->cl_prog     = program->number;
  86. clnt->cl_vers     = version->number;
  87. clnt->cl_prot     = xprt->prot;
  88. clnt->cl_stats    = program->stats;
  89. clnt->cl_bindwait = RPC_INIT_WAITQ("bindwait");
  90. if (!clnt->cl_port)
  91. clnt->cl_autobind = 1;
  92. if (!rpcauth_create(flavor, clnt))
  93. goto out_no_auth;
  94. /* save the nodename */
  95. clnt->cl_nodelen = strlen(system_utsname.nodename);
  96. if (clnt->cl_nodelen > UNX_MAXNODENAME)
  97. clnt->cl_nodelen = UNX_MAXNODENAME;
  98. memcpy(clnt->cl_nodename, system_utsname.nodename, clnt->cl_nodelen);
  99. out:
  100. return clnt;
  101. out_no_clnt:
  102. printk(KERN_INFO "RPC: out of memory in rpc_create_clientn");
  103. goto out;
  104. out_no_auth:
  105. printk(KERN_INFO "RPC: Couldn't create auth handle (flavor %d)n",
  106. flavor);
  107. rpc_free(clnt);
  108. clnt = NULL;
  109. goto out;
  110. }
  111. /*
  112.  * Properly shut down an RPC client, terminating all outstanding
  113.  * requests. Note that we must be certain that cl_oneshot and
  114.  * cl_dead are cleared, or else the client would be destroyed
  115.  * when the last task releases it.
  116.  */
  117. int
  118. rpc_shutdown_client(struct rpc_clnt *clnt)
  119. {
  120. dprintk("RPC: shutting down %s client for %sn",
  121. clnt->cl_protname, clnt->cl_server);
  122. while (atomic_read(&clnt->cl_users)) {
  123. #ifdef RPC_DEBUG
  124. dprintk("RPC: rpc_shutdown_client: client %s, tasks=%dn",
  125. clnt->cl_protname, atomic_read(&clnt->cl_users));
  126. #endif
  127. /* Don't let rpc_release_client destroy us */
  128. clnt->cl_oneshot = 0;
  129. clnt->cl_dead = 0;
  130. rpc_killall_tasks(clnt);
  131. sleep_on_timeout(&destroy_wait, 1*HZ);
  132. }
  133. return rpc_destroy_client(clnt);
  134. }
  135. /*
  136.  * Delete an RPC client
  137.  */
  138. int
  139. rpc_destroy_client(struct rpc_clnt *clnt)
  140. {
  141. dprintk("RPC: destroying %s client for %sn",
  142. clnt->cl_protname, clnt->cl_server);
  143. if (clnt->cl_auth) {
  144. rpcauth_destroy(clnt->cl_auth);
  145. clnt->cl_auth = NULL;
  146. }
  147. if (clnt->cl_xprt) {
  148. xprt_destroy(clnt->cl_xprt);
  149. clnt->cl_xprt = NULL;
  150. }
  151. rpc_free(clnt);
  152. return 0;
  153. }
  154. /*
  155.  * Release an RPC client
  156.  */
  157. void
  158. rpc_release_client(struct rpc_clnt *clnt)
  159. {
  160. dprintk("RPC:      rpc_release_client(%p, %d)n",
  161. clnt, atomic_read(&clnt->cl_users));
  162. if (!atomic_dec_and_test(&clnt->cl_users))
  163. return;
  164. wake_up(&destroy_wait);
  165. if (clnt->cl_oneshot || clnt->cl_dead)
  166. rpc_destroy_client(clnt);
  167. }
  168. /*
  169.  * Default callback for async RPC calls
  170.  */
  171. static void
  172. rpc_default_callback(struct rpc_task *task)
  173. {
  174. }
  175. /*
  176.  * Export the signal mask handling for aysnchronous code that
  177.  * sleeps on RPC calls
  178.  */
  179.  
  180. void rpc_clnt_sigmask(struct rpc_clnt *clnt, sigset_t *oldset)
  181. {
  182. unsigned long sigallow = sigmask(SIGKILL);
  183. unsigned long irqflags;
  184. /* Turn off various signals */
  185. if (clnt->cl_intr) {
  186. struct k_sigaction *action = current->sig->action;
  187. if (action[SIGINT-1].sa.sa_handler == SIG_DFL)
  188. sigallow |= sigmask(SIGINT);
  189. if (action[SIGQUIT-1].sa.sa_handler == SIG_DFL)
  190. sigallow |= sigmask(SIGQUIT);
  191. }
  192. spin_lock_irqsave(&current->sigmask_lock, irqflags);
  193. *oldset = current->blocked;
  194. siginitsetinv(&current->blocked, sigallow & ~oldset->sig[0]);
  195. recalc_sigpending(current);
  196. spin_unlock_irqrestore(&current->sigmask_lock, irqflags);
  197. }
  198. void rpc_clnt_sigunmask(struct rpc_clnt *clnt, sigset_t *oldset)
  199. {
  200. unsigned long irqflags;
  201. spin_lock_irqsave(&current->sigmask_lock, irqflags);
  202. current->blocked = *oldset;
  203. recalc_sigpending(current);
  204. spin_unlock_irqrestore(&current->sigmask_lock, irqflags);
  205. }
  206. /*
  207.  * New rpc_call implementation
  208.  */
  209. int rpc_call_sync(struct rpc_clnt *clnt, struct rpc_message *msg, int flags)
  210. {
  211. struct rpc_task my_task, *task = &my_task;
  212. sigset_t oldset;
  213. int status;
  214. /* If this client is slain all further I/O fails */
  215. if (clnt->cl_dead) 
  216. return -EIO;
  217. if (flags & RPC_TASK_ASYNC) {
  218. printk("rpc_call_sync: Illegal flag combination for synchronous taskn");
  219. flags &= ~RPC_TASK_ASYNC;
  220. }
  221. rpc_clnt_sigmask(clnt, &oldset);
  222. /* Create/initialize a new RPC task */
  223. rpc_init_task(task, clnt, NULL, flags);
  224. rpc_call_setup(task, msg, 0);
  225. /* Set up the call info struct and execute the task */
  226. if (task->tk_status == 0)
  227. status = rpc_execute(task);
  228. else {
  229. status = task->tk_status;
  230. rpc_release_task(task);
  231. }
  232. rpc_clnt_sigunmask(clnt, &oldset);
  233. return status;
  234. }
  235. /*
  236.  * New rpc_call implementation
  237.  */
  238. int
  239. rpc_call_async(struct rpc_clnt *clnt, struct rpc_message *msg, int flags,
  240.        rpc_action callback, void *data)
  241. {
  242. struct rpc_task *task;
  243. sigset_t oldset;
  244. int status;
  245. /* If this client is slain all further I/O fails */
  246. if (clnt->cl_dead) 
  247. return -EIO;
  248. flags |= RPC_TASK_ASYNC;
  249. rpc_clnt_sigmask(clnt, &oldset);
  250. /* Create/initialize a new RPC task */
  251. if (!callback)
  252. callback = rpc_default_callback;
  253. status = -ENOMEM;
  254. if (!(task = rpc_new_task(clnt, callback, flags)))
  255. goto out;
  256. task->tk_calldata = data;
  257. rpc_call_setup(task, msg, 0);
  258. /* Set up the call info struct and execute the task */
  259. if (task->tk_status == 0)
  260. status = rpc_execute(task);
  261. else {
  262. status = task->tk_status;
  263. rpc_release_task(task);
  264. }
  265. out:
  266. rpc_clnt_sigunmask(clnt, &oldset);
  267. return status;
  268. }
  269. void
  270. rpc_call_setup(struct rpc_task *task, struct rpc_message *msg, int flags)
  271. {
  272. task->tk_msg   = *msg;
  273. task->tk_flags |= flags;
  274. /* Bind the user cred */
  275. if (task->tk_msg.rpc_cred != NULL) {
  276. rpcauth_holdcred(task);
  277. } else
  278. rpcauth_bindcred(task);
  279. if (task->tk_status == 0)
  280. task->tk_action = call_reserve;
  281. else
  282. task->tk_action = NULL;
  283. /* Increment call count */
  284. if (task->tk_msg.rpc_proc < task->tk_client->cl_maxproc)
  285. rpcproc_count(task->tk_client, task->tk_msg.rpc_proc)++;
  286. }
  287. /*
  288.  * Restart an (async) RPC call. Usually called from within the
  289.  * exit handler.
  290.  */
  291. void
  292. rpc_restart_call(struct rpc_task *task)
  293. {
  294. if (RPC_ASSASSINATED(task))
  295. return;
  296. task->tk_action = call_reserve;
  297. rpcproc_count(task->tk_client, task->tk_msg.rpc_proc)++;
  298. }
  299. /*
  300.  * 1. Reserve an RPC call slot
  301.  */
  302. static void
  303. call_reserve(struct rpc_task *task)
  304. {
  305. struct rpc_clnt *clnt = task->tk_client;
  306. if (task->tk_msg.rpc_proc > clnt->cl_maxproc) {
  307. printk(KERN_WARNING "%s (vers %d): bad procedure number %dn",
  308. clnt->cl_protname, clnt->cl_vers, task->tk_msg.rpc_proc);
  309. rpc_exit(task, -EIO);
  310. return;
  311. }
  312. dprintk("RPC: %4d call_reserven", task->tk_pid);
  313. if (!rpcauth_uptodatecred(task)) {
  314. task->tk_action = call_refresh;
  315. return;
  316. }
  317. task->tk_status  = 0;
  318. task->tk_action  = call_reserveresult;
  319. task->tk_timeout = clnt->cl_timeout.to_resrvval;
  320. clnt->cl_stats->rpccnt++;
  321. xprt_reserve(task);
  322. }
  323. /*
  324.  * 1b. Grok the result of xprt_reserve()
  325.  */
  326. static void
  327. call_reserveresult(struct rpc_task *task)
  328. {
  329. int status = task->tk_status;
  330. dprintk("RPC: %4d call_reserveresult (status %d)n",
  331. task->tk_pid, task->tk_status);
  332. /*
  333.  * After a call to xprt_reserve(), we must have either
  334.  * a request slot or else an error status.
  335.  */
  336. if ((task->tk_status >= 0 && !task->tk_rqstp) ||
  337.     (task->tk_status < 0 && task->tk_rqstp))
  338. printk(KERN_ERR "call_reserveresult: status=%d, request=%p??n",
  339.  task->tk_status, task->tk_rqstp);
  340. if (task->tk_status >= 0) {
  341. task->tk_action = call_allocate;
  342. return;
  343. }
  344. task->tk_status = 0;
  345. switch (status) {
  346. case -EAGAIN:
  347. case -ENOBUFS:
  348. task->tk_timeout = task->tk_client->cl_timeout.to_resrvval;
  349. task->tk_action = call_reserve;
  350. break;
  351. case -ETIMEDOUT:
  352. dprintk("RPC: task timed outn");
  353. task->tk_action = call_timeout;
  354. break;
  355. default:
  356. if (!task->tk_rqstp) {
  357. printk(KERN_INFO "RPC: task has no request, exit EIOn");
  358. rpc_exit(task, -EIO);
  359. } else
  360. rpc_exit(task, status);
  361. }
  362. }
  363. /*
  364.  * 2. Allocate the buffer. For details, see sched.c:rpc_malloc.
  365.  * (Note: buffer memory is freed in rpc_task_release).
  366.  */
  367. static void
  368. call_allocate(struct rpc_task *task)
  369. {
  370. struct rpc_clnt *clnt = task->tk_client;
  371. unsigned int bufsiz;
  372. dprintk("RPC: %4d call_allocate (status %d)n", 
  373. task->tk_pid, task->tk_status);
  374. task->tk_action = call_encode;
  375. if (task->tk_buffer)
  376. return;
  377. /* FIXME: compute buffer requirements more exactly using
  378.  * auth->au_wslack */
  379. bufsiz = rpcproc_bufsiz(clnt, task->tk_msg.rpc_proc) + RPC_SLACK_SPACE;
  380. if ((task->tk_buffer = rpc_malloc(task, bufsiz << 1)) != NULL)
  381. return;
  382. printk(KERN_INFO "RPC: buffer allocation failed for task %pn", task); 
  383. if (RPC_IS_ASYNC(task) || !(task->tk_client->cl_intr && signalled())) {
  384. xprt_release(task);
  385. task->tk_action = call_reserve;
  386. rpc_delay(task, HZ>>4);
  387. return;
  388. }
  389. rpc_exit(task, -ERESTARTSYS);
  390. }
  391. /*
  392.  * 3. Encode arguments of an RPC call
  393.  */
  394. static void
  395. call_encode(struct rpc_task *task)
  396. {
  397. struct rpc_clnt *clnt = task->tk_client;
  398. struct rpc_rqst *req = task->tk_rqstp;
  399. unsigned int bufsiz;
  400. kxdrproc_t encode;
  401. int status;
  402. u32 *p;
  403. dprintk("RPC: %4d call_encode (status %d)n", 
  404. task->tk_pid, task->tk_status);
  405. task->tk_action = call_bind;
  406. /* Default buffer setup */
  407. bufsiz = rpcproc_bufsiz(clnt, task->tk_msg.rpc_proc)+RPC_SLACK_SPACE;
  408. req->rq_svec[0].iov_base = (void *)task->tk_buffer;
  409. req->rq_svec[0].iov_len  = bufsiz;
  410. req->rq_slen  = 0;
  411. req->rq_snr  = 1;
  412. req->rq_rvec[0].iov_base = (void *)((char *)task->tk_buffer + bufsiz);
  413. req->rq_rvec[0].iov_len  = bufsiz;
  414. req->rq_rlen  = bufsiz;
  415. req->rq_rnr  = 1;
  416. /* Zero buffer so we have automatic zero-padding of opaque & string */
  417. memset(task->tk_buffer, 0, bufsiz);
  418. /* Encode header and provided arguments */
  419. encode = rpcproc_encode(clnt, task->tk_msg.rpc_proc);
  420. if (!(p = call_header(task))) {
  421. printk(KERN_INFO "RPC: call_header failed, exit EIOn");
  422. rpc_exit(task, -EIO);
  423. } else
  424. if (encode && (status = encode(req, p, task->tk_msg.rpc_argp)) < 0) {
  425. printk(KERN_WARNING "%s: can't encode arguments: %dn",
  426. clnt->cl_protname, -status);
  427. rpc_exit(task, status);
  428. }
  429. }
  430. /*
  431.  * 4. Get the server port number if not yet set
  432.  */
  433. static void
  434. call_bind(struct rpc_task *task)
  435. {
  436. struct rpc_clnt *clnt = task->tk_client;
  437. struct rpc_xprt *xprt = clnt->cl_xprt;
  438. task->tk_action = (xprt_connected(xprt)) ? call_transmit : call_reconnect;
  439. if (!clnt->cl_port) {
  440. task->tk_action = call_reconnect;
  441. task->tk_timeout = clnt->cl_timeout.to_maxval;
  442. rpc_getport(task, clnt);
  443. }
  444. }
  445. /*
  446.  * 4a. Reconnect to the RPC server (TCP case)
  447.  */
  448. static void
  449. call_reconnect(struct rpc_task *task)
  450. {
  451. struct rpc_clnt *clnt = task->tk_client;
  452. struct rpc_task *child;
  453. dprintk("RPC: %4d call_reconnect status %dn",
  454. task->tk_pid, task->tk_status);
  455. task->tk_action = call_transmit;
  456. if (task->tk_status < 0 || !clnt->cl_xprt->stream)
  457. return;
  458. /* Run as a child to ensure it runs as an rpciod task */
  459. child = rpc_new_child(clnt, task);
  460. if (child) {
  461. child->tk_action = child_reconnect;
  462. rpc_run_child(task, child, NULL);
  463. }
  464. }
  465. static void child_reconnect(struct rpc_task *task)
  466. {
  467. task->tk_client->cl_stats->netreconn++;
  468. task->tk_status = 0;
  469. task->tk_action = child_reconnect_status;
  470. xprt_reconnect(task);
  471. }
  472. static void child_reconnect_status(struct rpc_task *task)
  473. {
  474. if (task->tk_status == -EAGAIN)
  475. task->tk_action = child_reconnect;
  476. else
  477. task->tk_action = NULL;
  478. }
  479. /*
  480.  * 5. Transmit the RPC request, and wait for reply
  481.  */
  482. static void
  483. call_transmit(struct rpc_task *task)
  484. {
  485. struct rpc_clnt *clnt = task->tk_client;
  486. dprintk("RPC: %4d call_transmit (status %d)n", 
  487. task->tk_pid, task->tk_status);
  488. task->tk_action = call_status;
  489. if (task->tk_status < 0)
  490. return;
  491. xprt_transmit(task);
  492. if (!rpcproc_decode(clnt, task->tk_msg.rpc_proc)) {
  493. task->tk_action = NULL;
  494. rpc_wake_up_task(task);
  495. }
  496. }
  497. /*
  498.  * 6. Sort out the RPC call status
  499.  */
  500. static void
  501. call_status(struct rpc_task *task)
  502. {
  503. struct rpc_clnt *clnt = task->tk_client;
  504. struct rpc_xprt *xprt = clnt->cl_xprt;
  505. struct rpc_rqst *req;
  506. int status = task->tk_status;
  507. dprintk("RPC: %4d call_status (status %d)n", 
  508. task->tk_pid, task->tk_status);
  509. if (status >= 0) {
  510. task->tk_action = call_decode;
  511. return;
  512. }
  513. task->tk_status = 0;
  514. req = task->tk_rqstp;
  515. switch(status) {
  516. case -ETIMEDOUT:
  517. task->tk_action = call_timeout;
  518. break;
  519. case -ECONNREFUSED:
  520. case -ENOTCONN:
  521. req->rq_bytes_sent = 0;
  522. if (clnt->cl_autobind || !clnt->cl_port) {
  523. clnt->cl_port = 0;
  524. task->tk_action = call_bind;
  525. break;
  526. }
  527. if (xprt->stream) {
  528. task->tk_action = call_reconnect;
  529. break;
  530. }
  531. /*
  532.  * Sleep and dream of an open connection
  533.  */
  534. task->tk_timeout = 5 * HZ;
  535. rpc_sleep_on(&xprt->sending, task, NULL, NULL);
  536. case -ENOMEM:
  537. case -EAGAIN:
  538. task->tk_action = call_transmit;
  539. clnt->cl_stats->rpcretrans++;
  540. break;
  541. default:
  542. if (clnt->cl_chatty)
  543. printk("%s: RPC call returned error %dn",
  544.        clnt->cl_protname, -status);
  545. rpc_exit(task, status);
  546. }
  547. }
  548. /*
  549.  * 6a. Handle RPC timeout
  550.  *  We do not release the request slot, so we keep using the
  551.  * same XID for all retransmits.
  552.  */
  553. static void
  554. call_timeout(struct rpc_task *task)
  555. {
  556. struct rpc_clnt *clnt = task->tk_client;
  557. struct rpc_rqst *req = task->tk_rqstp;
  558. if (req) {
  559. struct rpc_timeout *to = &req->rq_timeout;
  560. if (xprt_adjust_timeout(to)) {
  561. dprintk("RPC: %4d call_timeout (minor timeo)n",
  562. task->tk_pid);
  563. goto minor_timeout;
  564. }
  565. to->to_retries = clnt->cl_timeout.to_retries;
  566. }
  567. dprintk("RPC: %4d call_timeout (major timeo)n", task->tk_pid);
  568. if (clnt->cl_softrtry) {
  569. if (clnt->cl_chatty && !task->tk_exit)
  570. printk(KERN_NOTICE "%s: server %s not responding, timed outn",
  571. clnt->cl_protname, clnt->cl_server);
  572. rpc_exit(task, -EIO);
  573. return;
  574. }
  575. if (clnt->cl_chatty && !(task->tk_flags & RPC_CALL_MAJORSEEN)) {
  576. task->tk_flags |= RPC_CALL_MAJORSEEN;
  577. if (req)
  578. printk(KERN_NOTICE "%s: server %s not responding, still tryingn",
  579. clnt->cl_protname, clnt->cl_server);
  580. #ifdef RPC_DEBUG
  581. else
  582. printk(KERN_NOTICE "%s: task %d can't get a request slotn",
  583. clnt->cl_protname, task->tk_pid);
  584. #endif
  585. }
  586. if (clnt->cl_autobind)
  587. clnt->cl_port = 0;
  588. minor_timeout:
  589. if (!req)
  590. task->tk_action = call_reserve;
  591. else if (!clnt->cl_port) {
  592. task->tk_action = call_bind;
  593. clnt->cl_stats->rpcretrans++;
  594. } else if (!xprt_connected(clnt->cl_xprt)) {
  595. task->tk_action = call_reconnect;
  596. clnt->cl_stats->rpcretrans++;
  597. } else {
  598. task->tk_action = call_transmit;
  599. clnt->cl_stats->rpcretrans++;
  600. }
  601. task->tk_status = 0;
  602. }
  603. /*
  604.  * 7. Decode the RPC reply
  605.  */
  606. static void
  607. call_decode(struct rpc_task *task)
  608. {
  609. struct rpc_clnt *clnt = task->tk_client;
  610. struct rpc_rqst *req = task->tk_rqstp;
  611. kxdrproc_t decode = rpcproc_decode(clnt, task->tk_msg.rpc_proc);
  612. u32 *p;
  613. dprintk("RPC: %4d call_decode (status %d)n", 
  614. task->tk_pid, task->tk_status);
  615. if (clnt->cl_chatty && (task->tk_flags & RPC_CALL_MAJORSEEN)) {
  616. printk(KERN_NOTICE "%s: server %s OKn",
  617. clnt->cl_protname, clnt->cl_server);
  618. task->tk_flags &= ~RPC_CALL_MAJORSEEN;
  619. }
  620. if (task->tk_status < 12) {
  621. if (!clnt->cl_softrtry) {
  622. task->tk_action = call_transmit;
  623. clnt->cl_stats->rpcretrans++;
  624. } else {
  625. printk(KERN_WARNING "%s: too small RPC reply size (%d bytes)n",
  626. clnt->cl_protname, task->tk_status);
  627. rpc_exit(task, -EIO);
  628. }
  629. return;
  630. }
  631. /* Verify the RPC header */
  632. if (!(p = call_verify(task)))
  633. return;
  634. /*
  635.  * The following is an NFS-specific hack to cater for setuid
  636.  * processes whose uid is mapped to nobody on the server.
  637.  */
  638. if (task->tk_client->cl_droppriv && 
  639.             (ntohl(*p) == NFSERR_ACCES || ntohl(*p) == NFSERR_PERM)) {
  640. if (RPC_IS_SETUID(task) && task->tk_suid_retry) {
  641. dprintk("RPC: %4d retry squashed uidn", task->tk_pid);
  642. task->tk_flags ^= RPC_CALL_REALUID;
  643. task->tk_action = call_encode;
  644. task->tk_suid_retry--;
  645. return;
  646. }
  647. }
  648. task->tk_action = NULL;
  649. if (decode)
  650. task->tk_status = decode(req, p, task->tk_msg.rpc_resp);
  651. dprintk("RPC: %4d call_decode result %dn", task->tk_pid,
  652. task->tk_status);
  653. }
  654. /*
  655.  * 8. Refresh the credentials if rejected by the server
  656.  */
  657. static void
  658. call_refresh(struct rpc_task *task)
  659. {
  660. dprintk("RPC: %4d call_refreshn", task->tk_pid);
  661. xprt_release(task); /* Must do to obtain new XID */
  662. task->tk_action = call_refreshresult;
  663. task->tk_status = 0;
  664. task->tk_client->cl_stats->rpcauthrefresh++;
  665. rpcauth_refreshcred(task);
  666. }
  667. /*
  668.  * 8a. Process the results of a credential refresh
  669.  */
  670. static void
  671. call_refreshresult(struct rpc_task *task)
  672. {
  673. dprintk("RPC: %4d call_refreshresult (status %d)n", 
  674. task->tk_pid, task->tk_status);
  675. if (task->tk_status < 0)
  676. rpc_exit(task, -EACCES);
  677. else
  678. task->tk_action = call_reserve;
  679. }
  680. /*
  681.  * Call header serialization
  682.  */
  683. static u32 *
  684. call_header(struct rpc_task *task)
  685. {
  686. struct rpc_clnt *clnt = task->tk_client;
  687. struct rpc_xprt *xprt = clnt->cl_xprt;
  688. struct rpc_rqst *req = task->tk_rqstp;
  689. u32 *p = req->rq_svec[0].iov_base;
  690. /* FIXME: check buffer size? */
  691. if (xprt->stream)
  692. *p++ = 0; /* fill in later */
  693. *p++ = req->rq_xid; /* XID */
  694. *p++ = htonl(RPC_CALL); /* CALL */
  695. *p++ = htonl(RPC_VERSION); /* RPC version */
  696. *p++ = htonl(clnt->cl_prog); /* program number */
  697. *p++ = htonl(clnt->cl_vers); /* program version */
  698. *p++ = htonl(task->tk_msg.rpc_proc); /* procedure */
  699. return rpcauth_marshcred(task, p);
  700. }
  701. /*
  702.  * Reply header verification
  703.  */
  704. static u32 *
  705. call_verify(struct rpc_task *task)
  706. {
  707. u32 *p = task->tk_rqstp->rq_rvec[0].iov_base, n;
  708. p += 1; /* skip XID */
  709. if ((n = ntohl(*p++)) != RPC_REPLY) {
  710. printk(KERN_WARNING "call_verify: not an RPC reply: %xn", n);
  711. goto garbage;
  712. }
  713. if ((n = ntohl(*p++)) != RPC_MSG_ACCEPTED) {
  714. int error = -EACCES;
  715. if ((n = ntohl(*p++)) != RPC_AUTH_ERROR) {
  716. printk(KERN_WARNING "call_verify: RPC call rejected: %xn", n);
  717. } else
  718. switch ((n = ntohl(*p++))) {
  719. case RPC_AUTH_REJECTEDCRED:
  720. case RPC_AUTH_REJECTEDVERF:
  721. if (!task->tk_cred_retry)
  722. break;
  723. task->tk_cred_retry--;
  724. dprintk("RPC: %4d call_verify: retry stale credsn",
  725. task->tk_pid);
  726. rpcauth_invalcred(task);
  727. task->tk_action = call_refresh;
  728. return NULL;
  729. case RPC_AUTH_BADCRED:
  730. case RPC_AUTH_BADVERF:
  731. /* possibly garbled cred/verf? */
  732. if (!task->tk_garb_retry)
  733. break;
  734. task->tk_garb_retry--;
  735. dprintk("RPC: %4d call_verify: retry garbled credsn",
  736. task->tk_pid);
  737. task->tk_action = call_encode;
  738. return NULL;
  739. case RPC_AUTH_TOOWEAK:
  740. printk(KERN_NOTICE "call_verify: server requires stronger "
  741.        "authentication.n");
  742. break;
  743. default:
  744. printk(KERN_WARNING "call_verify: unknown auth error: %xn", n);
  745. error = -EIO;
  746. }
  747. dprintk("RPC: %4d call_verify: call rejected %dn",
  748. task->tk_pid, n);
  749. rpc_exit(task, error);
  750. return NULL;
  751. }
  752. if (!(p = rpcauth_checkverf(task, p))) {
  753. printk(KERN_WARNING "call_verify: auth check failedn");
  754. goto garbage; /* bad verifier, retry */
  755. }
  756. switch ((n = ntohl(*p++))) {
  757. case RPC_SUCCESS:
  758. return p;
  759. case RPC_GARBAGE_ARGS:
  760. break; /* retry */
  761. default:
  762. printk(KERN_WARNING "call_verify: server accept status: %xn", n);
  763. /* Also retry */
  764. }
  765. garbage:
  766. dprintk("RPC: %4d call_verify: server saw garbagen", task->tk_pid);
  767. task->tk_client->cl_stats->rpcgarbage++;
  768. if (task->tk_garb_retry) {
  769. task->tk_garb_retry--;
  770. dprintk(KERN_WARNING "RPC: garbage, retrying %4dn", task->tk_pid);
  771. task->tk_action = call_encode;
  772. return NULL;
  773. }
  774. printk(KERN_WARNING "RPC: garbage, exit EIOn");
  775. rpc_exit(task, -EIO);
  776. return NULL;
  777. }