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

嵌入式Linux

开发平台:

Unix_Linux

  1. /*
  2.  *  linux/fs/ncpfs/sock.c
  3.  *
  4.  *  Copyright (C) 1992, 1993  Rick Sladkey
  5.  *
  6.  *  Modified 1995, 1996 by Volker Lendecke to be usable for ncp
  7.  *  Modified 1997 Peter Waltenberg, Bill Hawes, David Woodhouse for 2.1 dcache
  8.  *
  9.  */
  10. #include <linux/config.h>
  11. #include <linux/sched.h>
  12. #include <linux/errno.h>
  13. #include <linux/socket.h>
  14. #include <linux/fcntl.h>
  15. #include <linux/stat.h>
  16. #include <asm/uaccess.h>
  17. #include <linux/in.h>
  18. #include <linux/net.h>
  19. #include <linux/mm.h>
  20. #include <linux/netdevice.h>
  21. #include <linux/signal.h>
  22. #include <net/scm.h>
  23. #include <net/sock.h>
  24. #include <linux/ipx.h>
  25. #include <linux/poll.h>
  26. #include <linux/file.h>
  27. #include <linux/ncp_fs.h>
  28. #ifdef CONFIG_NCPFS_PACKET_SIGNING
  29. #include "ncpsign_kernel.h"
  30. #endif
  31. static int _recv(struct socket *sock, unsigned char *ubuf, int size,
  32.  unsigned flags)
  33. {
  34. struct iovec iov;
  35. struct msghdr msg;
  36. struct scm_cookie scm;
  37. memset(&scm, 0, sizeof(scm));
  38. iov.iov_base = ubuf;
  39. iov.iov_len = size;
  40. msg.msg_name = NULL;
  41. msg.msg_namelen = 0;
  42. msg.msg_control = NULL;
  43. msg.msg_iov = &iov;
  44. msg.msg_iovlen = 1;
  45. return sock->ops->recvmsg(sock, &msg, size, flags, &scm);
  46. }
  47. static int _send(struct socket *sock, const void *buff, int len)
  48. {
  49. struct iovec iov;
  50. struct msghdr msg;
  51. struct scm_cookie scm;
  52. int err;
  53. iov.iov_base = (void *) buff;
  54. iov.iov_len = len;
  55. msg.msg_name = NULL;
  56. msg.msg_namelen = 0;
  57. msg.msg_control = NULL;
  58. msg.msg_iov = &iov;
  59. msg.msg_iovlen = 1;
  60. msg.msg_flags = 0;
  61. err = scm_send(sock, &msg, &scm);
  62. if (err < 0) {
  63. return err;
  64. }
  65. err = sock->ops->sendmsg(sock, &msg, len, &scm);
  66. scm_destroy(&scm);
  67. return err;
  68. }
  69. static int do_ncp_rpc_call(struct ncp_server *server, int size,
  70. struct ncp_reply_header* reply_buf, int max_reply_size)
  71. {
  72. struct file *file;
  73. struct socket *sock;
  74. int result;
  75. char *start = server->packet;
  76. poll_table wait_table;
  77. int init_timeout, max_timeout;
  78. int timeout;
  79. int retrans;
  80. int major_timeout_seen;
  81. int acknowledge_seen;
  82. int n;
  83. /* We have to check the result, so store the complete header */
  84. struct ncp_request_header request =
  85. *((struct ncp_request_header *) (server->packet));
  86. struct ncp_reply_header reply;
  87. file = server->ncp_filp;
  88. sock = &file->f_dentry->d_inode->u.socket_i;
  89. init_timeout = server->m.time_out;
  90. max_timeout = NCP_MAX_RPC_TIMEOUT;
  91. retrans = server->m.retry_count;
  92. major_timeout_seen = 0;
  93. acknowledge_seen = 0;
  94. for (n = 0, timeout = init_timeout;; n++, timeout <<= 1) {
  95. /*
  96. DDPRINTK("ncpfs: %08lX:%02X%02X%02X%02X%02X%02X:%04Xn",
  97.  htonl(server->m.serv_addr.sipx_network),
  98.  server->m.serv_addr.sipx_node[0],
  99.  server->m.serv_addr.sipx_node[1],
  100.  server->m.serv_addr.sipx_node[2],
  101.  server->m.serv_addr.sipx_node[3],
  102.  server->m.serv_addr.sipx_node[4],
  103.  server->m.serv_addr.sipx_node[5],
  104.  ntohs(server->m.serv_addr.sipx_port));
  105. */
  106. DDPRINTK("ncpfs: req.typ: %04X, con: %d, "
  107.  "seq: %d",
  108.  request.type,
  109.  (request.conn_high << 8) + request.conn_low,
  110.  request.sequence);
  111. DDPRINTK(" func: %dn",
  112.  request.function);
  113. result = _send(sock, (void *) start, size);
  114. if (result < 0) {
  115. printk(KERN_ERR "ncp_rpc_call: send error = %dn", result);
  116. break;
  117. }
  118.       re_select:
  119. poll_initwait(&wait_table);
  120. /* mb() is not necessary because ->poll() will serialize
  121.    instructions adding the wait_table waitqueues in the
  122.    waitqueue-head before going to calculate the mask-retval. */
  123. __set_current_state(TASK_INTERRUPTIBLE);
  124. if (!(sock->ops->poll(file, sock, &wait_table) & POLLIN)) {
  125. int timed_out;
  126. if (timeout > max_timeout) {
  127. /* JEJB/JSP 2/7/94
  128.  * This is useful to see if the system is
  129.  * hanging */
  130. if (acknowledge_seen == 0) {
  131. printk(KERN_WARNING "NCP max timeoutn");
  132. }
  133. timeout = max_timeout;
  134. }
  135. timed_out = !schedule_timeout(timeout);
  136. poll_freewait(&wait_table);
  137. current->state = TASK_RUNNING;
  138. if (signal_pending(current)) {
  139. result = -ERESTARTSYS;
  140. break;
  141. }
  142. if(wait_table.error) {
  143. result = wait_table.error;
  144. break;
  145. }
  146. if (timed_out) {
  147. if (n < retrans)
  148. continue;
  149. if (server->m.flags & NCP_MOUNT_SOFT) {
  150. printk(KERN_WARNING "NCP server not respondingn");
  151. result = -EIO;
  152. break;
  153. }
  154. n = 0;
  155. timeout = init_timeout;
  156. if (init_timeout < max_timeout)
  157. init_timeout <<= 1;
  158. if (!major_timeout_seen) {
  159. printk(KERN_WARNING "NCP server not respondingn");
  160. }
  161. major_timeout_seen = 1;
  162. continue;
  163. }
  164. } else {
  165. poll_freewait(&wait_table);
  166. }
  167. current->state = TASK_RUNNING;
  168. /* Get the header from the next packet using a peek, so keep it
  169.  * on the recv queue.  If it is wrong, it will be some reply
  170.  * we don't now need, so discard it */
  171. result = _recv(sock, (void *) &reply, sizeof(reply),
  172.        MSG_PEEK | MSG_DONTWAIT);
  173. if (result < 0) {
  174. if (result == -EAGAIN) {
  175. DDPRINTK("ncp_rpc_call: bad select readyn");
  176. goto re_select;
  177. }
  178. if (result == -ECONNREFUSED) {
  179. DPRINTK("ncp_rpc_call: server playing coyn");
  180. goto re_select;
  181. }
  182. if (result != -ERESTARTSYS) {
  183. printk(KERN_ERR "ncp_rpc_call: recv error = %dn",
  184.        -result);
  185. }
  186. break;
  187. }
  188. if ((result == sizeof(reply))
  189.     && (reply.type == NCP_POSITIVE_ACK)) {
  190. /* Throw away the packet */
  191. DPRINTK("ncp_rpc_call: got positive acknowledgen");
  192. _recv(sock, (void *) &reply, sizeof(reply),
  193.       MSG_DONTWAIT);
  194. n = 0;
  195. timeout = max_timeout;
  196. acknowledge_seen = 1;
  197. goto re_select;
  198. }
  199. DDPRINTK("ncpfs: rep.typ: %04X, con: %d, tsk: %d,"
  200.  "seq: %dn",
  201.  reply.type,
  202.  (reply.conn_high << 8) + reply.conn_low,
  203.  reply.task,
  204.  reply.sequence);
  205. if ((result >= sizeof(reply))
  206.     && (reply.type == NCP_REPLY)
  207.     && ((request.type == NCP_ALLOC_SLOT_REQUEST)
  208. || ((reply.sequence == request.sequence)
  209.     && (reply.conn_low == request.conn_low)
  210. /* seem to get wrong task from NW311 && (reply.task      == request.task) */
  211.     && (reply.conn_high == request.conn_high)))) {
  212. if (major_timeout_seen)
  213. printk(KERN_NOTICE "NCP server OKn");
  214. break;
  215. }
  216. /* JEJB/JSP 2/7/94
  217.  * we have xid mismatch, so discard the packet and start
  218.  * again.  What a hack! but I can't call recvfrom with
  219.  * a null buffer yet. */
  220. _recv(sock, (void *) &reply, sizeof(reply), MSG_DONTWAIT);
  221. DPRINTK("ncp_rpc_call: reply mismatchn");
  222. goto re_select;
  223. }
  224. /* 
  225.  * we have the correct reply, so read into the correct place and
  226.  * return it
  227.  */
  228. result = _recv(sock, (void *)reply_buf, max_reply_size, MSG_DONTWAIT);
  229. if (result < 0) {
  230. printk(KERN_WARNING "NCP: notice message: result=%dn", result);
  231. } else if (result < sizeof(struct ncp_reply_header)) {
  232. printk(KERN_ERR "NCP: just caught a too small read memory size..., "
  233.        "email to NET channeln");
  234. printk(KERN_ERR "NCP: result=%dn", result);
  235. result = -EIO;
  236. }
  237. return result;
  238. }
  239. static int do_tcp_rcv(struct ncp_server *server, void *buffer, size_t len) {
  240. poll_table wait_table;
  241. struct file *file;
  242. struct socket *sock;
  243. int init_timeout;
  244. size_t dataread;
  245. int result = 0;
  246. file = server->ncp_filp;
  247. sock = &file->f_dentry->d_inode->u.socket_i;
  248. dataread = 0;
  249. init_timeout = server->m.time_out * 20;
  250. /* hard-mounted volumes have no timeout, except connection close... */
  251. if (!(server->m.flags & NCP_MOUNT_SOFT))
  252. init_timeout = 0x7FFF0000;
  253. while (len) {
  254. poll_initwait(&wait_table);
  255. /* mb() is not necessary because ->poll() will serialize
  256.    instructions adding the wait_table waitqueues in the
  257.    waitqueue-head before going to calculate the mask-retval. */
  258. __set_current_state(TASK_INTERRUPTIBLE);
  259. if (!(sock->ops->poll(file, sock, &wait_table) & POLLIN)) {
  260. init_timeout = schedule_timeout(init_timeout);
  261. poll_freewait(&wait_table);
  262. current->state = TASK_RUNNING;
  263. if (signal_pending(current)) {
  264. return -ERESTARTSYS;
  265. }
  266. if (!init_timeout) {
  267. return -EIO;
  268. }
  269. if(wait_table.error) {
  270. return wait_table.error;
  271. }
  272. } else {
  273. poll_freewait(&wait_table);
  274. }
  275. current->state = TASK_RUNNING;
  276. result = _recv(sock, buffer, len, MSG_DONTWAIT);
  277. if (result < 0) {
  278. if (result == -EAGAIN) {
  279. DDPRINTK("ncpfs: tcp: bad select readyn");
  280. continue;
  281. }
  282. return result;
  283. }
  284. if (result == 0) {
  285. printk(KERN_ERR "ncpfs: tcp: EOF on socketn");
  286. return -EIO;
  287. }
  288. if (result > len) {
  289. printk(KERN_ERR "ncpfs: tcp: bug in recvmsgn");
  290. return -EIO;
  291. }
  292. dataread += result;
  293. buffer += result;
  294. len -= result;
  295. }
  296. return 0;
  297. }
  298. #define NCP_TCP_XMIT_MAGIC (0x446D6454)
  299. #define NCP_TCP_XMIT_VERSION (1)
  300. #define NCP_TCP_RCVD_MAGIC (0x744E6350)
  301. static int do_ncp_tcp_rpc_call(struct ncp_server *server, int size,
  302. struct ncp_reply_header* reply_buf, int max_reply_size)
  303. {
  304. struct file *file;
  305. struct socket *sock;
  306. int result;
  307. struct iovec iov[2];
  308. struct msghdr msg;
  309. struct scm_cookie scm;
  310. __u32 ncptcp_rcvd_hdr[2];
  311. __u32 ncptcp_xmit_hdr[4];
  312. int   datalen;
  313. /* We have to check the result, so store the complete header */
  314. struct ncp_request_header request =
  315. *((struct ncp_request_header *) (server->packet));
  316. file = server->ncp_filp;
  317. sock = &file->f_dentry->d_inode->u.socket_i;
  318. ncptcp_xmit_hdr[0] = htonl(NCP_TCP_XMIT_MAGIC);
  319. ncptcp_xmit_hdr[1] = htonl(size + 16);
  320. ncptcp_xmit_hdr[2] = htonl(NCP_TCP_XMIT_VERSION);
  321. ncptcp_xmit_hdr[3] = htonl(max_reply_size + 8);
  322. DDPRINTK("ncpfs: req.typ: %04X, con: %d, "
  323.  "seq: %d",
  324.  request.type,
  325.  (request.conn_high << 8) + request.conn_low,
  326.  request.sequence);
  327. DDPRINTK(" func: %dn",
  328.  request.function);
  329. iov[1].iov_base = (void *) server->packet;
  330. iov[1].iov_len = size;
  331. iov[0].iov_base = ncptcp_xmit_hdr;
  332. iov[0].iov_len = 16;
  333. msg.msg_name = NULL;
  334. msg.msg_namelen = 0;
  335. msg.msg_control = NULL;
  336. msg.msg_iov = iov;
  337. msg.msg_iovlen = 2;
  338. msg.msg_flags = MSG_NOSIGNAL;
  339. result = scm_send(sock, &msg, &scm);
  340. if (result < 0) {
  341. return result;
  342. }
  343. result = sock->ops->sendmsg(sock, &msg, size + 16, &scm);
  344. scm_destroy(&scm);
  345. if (result < 0) {
  346. printk(KERN_ERR "ncpfs: tcp: Send failed: %dn", result);
  347. return result;
  348. }
  349. rstrcv:
  350. result = do_tcp_rcv(server, ncptcp_rcvd_hdr, 8);
  351. if (result)
  352. return result;
  353. if (ncptcp_rcvd_hdr[0] != htonl(NCP_TCP_RCVD_MAGIC)) {
  354. printk(KERN_ERR "ncpfs: tcp: Unexpected reply type %08Xn", ntohl(ncptcp_rcvd_hdr[0]));
  355. return -EIO;
  356. }
  357. datalen = ntohl(ncptcp_rcvd_hdr[1]);
  358. if (datalen < 8 + sizeof(*reply_buf) || datalen > max_reply_size + 8) {
  359. printk(KERN_ERR "ncpfs: tcp: Unexpected reply len %dn", datalen);
  360. return -EIO;
  361. }
  362. datalen -= 8;
  363. result = do_tcp_rcv(server, reply_buf, datalen);
  364. if (result)
  365. return result;
  366. if (reply_buf->type != NCP_REPLY) {
  367. DDPRINTK("ncpfs: tcp: Unexpected NCP type %02Xn", reply_buf->type);
  368. goto rstrcv;
  369. }
  370. if (request.type == NCP_ALLOC_SLOT_REQUEST)
  371. return datalen;
  372. if (reply_buf->sequence != request.sequence) {
  373. printk(KERN_ERR "ncpfs: tcp: Bad sequence numbern");
  374. return -EIO;
  375. }
  376. if ((reply_buf->conn_low != request.conn_low) ||
  377.     (reply_buf->conn_high != request.conn_high)) {
  378. printk(KERN_ERR "ncpfs: tcp: Connection number mismatchn");
  379. return -EIO;
  380. }
  381. return datalen;
  382. }
  383. /*
  384.  * We need the server to be locked here, so check!
  385.  */
  386. static int ncp_do_request(struct ncp_server *server, int size,
  387. void* reply, int max_reply_size)
  388. {
  389. struct file *file;
  390. struct socket *sock;
  391. int result;
  392. if (server->lock == 0) {
  393. printk(KERN_ERR "ncpfs: Server not locked!n");
  394. return -EIO;
  395. }
  396. if (!ncp_conn_valid(server)) {
  397. return -EIO;
  398. }
  399. #ifdef CONFIG_NCPFS_PACKET_SIGNING
  400. if (server->sign_active)
  401. {
  402. sign_packet(server, &size);
  403. }
  404. #endif /* CONFIG_NCPFS_PACKET_SIGNING */
  405. file = server->ncp_filp;
  406. sock = &file->f_dentry->d_inode->u.socket_i;
  407. /* N.B. this isn't needed ... check socket type? */
  408. if (!sock) {
  409. printk(KERN_ERR "ncp_rpc_call: socki_lookup failedn");
  410. result = -EBADF;
  411. } else {
  412. mm_segment_t fs;
  413. sigset_t old_set;
  414. unsigned long mask, flags;
  415. spin_lock_irqsave(&current->sigmask_lock, flags);
  416. old_set = current->blocked;
  417. if (current->flags & PF_EXITING)
  418. mask = 0;
  419. else
  420. mask = sigmask(SIGKILL);
  421. if (server->m.flags & NCP_MOUNT_INTR) {
  422. /* FIXME: This doesn't seem right at all.  So, like,
  423.    we can't handle SIGINT and get whatever to stop?
  424.    What if we've blocked it ourselves?  What about
  425.    alarms?  Why, in fact, are we mucking with the
  426.    sigmask at all? -- r~ */
  427. if (current->sig->action[SIGINT - 1].sa.sa_handler == SIG_DFL)
  428. mask |= sigmask(SIGINT);
  429. if (current->sig->action[SIGQUIT - 1].sa.sa_handler == SIG_DFL)
  430. mask |= sigmask(SIGQUIT);
  431. }
  432. siginitsetinv(&current->blocked, mask);
  433. recalc_sigpending(current);
  434. spin_unlock_irqrestore(&current->sigmask_lock, flags);
  435. fs = get_fs();
  436. set_fs(get_ds());
  437. if (sock->type == SOCK_STREAM)
  438. result = do_ncp_tcp_rpc_call(server, size, reply, max_reply_size);
  439. else
  440. result = do_ncp_rpc_call(server, size, reply, max_reply_size);
  441. set_fs(fs);
  442. spin_lock_irqsave(&current->sigmask_lock, flags);
  443. current->blocked = old_set;
  444. recalc_sigpending(current);
  445. spin_unlock_irqrestore(&current->sigmask_lock, flags);
  446. }
  447. DDPRINTK("do_ncp_rpc_call returned %dn", result);
  448. if (result < 0) {
  449. /* There was a problem with I/O, so the connections is
  450.  * no longer usable. */
  451. ncp_invalidate_conn(server);
  452. }
  453. return result;
  454. }
  455. /* ncp_do_request assures that at least a complete reply header is
  456.  * received. It assumes that server->current_size contains the ncp
  457.  * request size
  458.  */
  459. int ncp_request2(struct ncp_server *server, int function, 
  460. void* rpl, int size)
  461. {
  462. struct ncp_request_header *h;
  463. struct ncp_reply_header* reply = rpl;
  464. int request_size = server->current_size
  465.  - sizeof(struct ncp_request_header);
  466. int result;
  467. h = (struct ncp_request_header *) (server->packet);
  468. if (server->has_subfunction != 0) {
  469. *(__u16 *) & (h->data[0]) = htons(request_size - 2);
  470. }
  471. h->type = NCP_REQUEST;
  472. server->sequence += 1;
  473. h->sequence = server->sequence;
  474. h->conn_low = (server->connection) & 0xff;
  475. h->conn_high = ((server->connection) & 0xff00) >> 8;
  476. /*
  477.  * The server shouldn't know or care what task is making a
  478.  * request, so we always use the same task number.
  479.  */
  480. h->task = 2; /* (current->pid) & 0xff; */
  481. h->function = function;
  482. result = ncp_do_request(server, request_size + sizeof(*h), reply, size);
  483. if (result < 0) {
  484. DPRINTK("ncp_request_error: %dn", result);
  485. goto out;
  486. }
  487. server->completion = reply->completion_code;
  488. server->conn_status = reply->connection_state;
  489. server->reply_size = result;
  490. server->ncp_reply_size = result - sizeof(struct ncp_reply_header);
  491. result = reply->completion_code;
  492. if (result != 0)
  493. PPRINTK("ncp_request: completion code=%xn", result);
  494. out:
  495. return result;
  496. }
  497. int ncp_connect(struct ncp_server *server)
  498. {
  499. struct ncp_request_header *h;
  500. int result;
  501. h = (struct ncp_request_header *) (server->packet);
  502. h->type = NCP_ALLOC_SLOT_REQUEST;
  503. server->sequence = 0;
  504. h->sequence = server->sequence;
  505. h->conn_low = 0xff;
  506. h->conn_high = 0xff;
  507. h->task = 2; /* see above */
  508. h->function = 0;
  509. result = ncp_do_request(server, sizeof(*h), server->packet, server->packet_size);
  510. if (result < 0)
  511. goto out;
  512. server->sequence = 0;
  513. server->connection = h->conn_low + (h->conn_high * 256);
  514. result = 0;
  515. out:
  516. return result;
  517. }
  518. int ncp_disconnect(struct ncp_server *server)
  519. {
  520. struct ncp_request_header *h;
  521. h = (struct ncp_request_header *) (server->packet);
  522. h->type = NCP_DEALLOC_SLOT_REQUEST;
  523. server->sequence += 1;
  524. h->sequence = server->sequence;
  525. h->conn_low = (server->connection) & 0xff;
  526. h->conn_high = ((server->connection) & 0xff00) >> 8;
  527. h->task = 2; /* see above */
  528. h->function = 0;
  529. return ncp_do_request(server, sizeof(*h), server->packet, server->packet_size);
  530. }
  531. void ncp_lock_server(struct ncp_server *server)
  532. {
  533. down(&server->sem);
  534. if (server->lock)
  535. printk(KERN_WARNING "ncp_lock_server: was locked!n");
  536. server->lock = 1;
  537. }
  538. void ncp_unlock_server(struct ncp_server *server)
  539. {
  540. if (!server->lock) {
  541. printk(KERN_WARNING "ncp_unlock_server: was not locked!n");
  542. return;
  543. }
  544. server->lock = 0;
  545. up(&server->sem);
  546. }