PageRenderTime 53ms CodeModel.GetById 17ms RepoModel.GetById 0ms app.codeStats 1ms

/net/tipc/socket.c

https://bitbucket.org/emiliolopez/linux
C | 3359 lines | 2373 code | 409 blank | 577 comment | 495 complexity | a0b933d91793237934454f97aea48547 MD5 | raw file
Possible License(s): GPL-2.0, LGPL-2.0, AGPL-1.0

Large files files are truncated, but you can click here to view the full file

  1. /*
  2. * net/tipc/socket.c: TIPC socket API
  3. *
  4. * Copyright (c) 2001-2007, 2012-2017, Ericsson AB
  5. * Copyright (c) 2004-2008, 2010-2013, Wind River Systems
  6. * All rights reserved.
  7. *
  8. * Redistribution and use in source and binary forms, with or without
  9. * modification, are permitted provided that the following conditions are met:
  10. *
  11. * 1. Redistributions of source code must retain the above copyright
  12. * notice, this list of conditions and the following disclaimer.
  13. * 2. Redistributions in binary form must reproduce the above copyright
  14. * notice, this list of conditions and the following disclaimer in the
  15. * documentation and/or other materials provided with the distribution.
  16. * 3. Neither the names of the copyright holders nor the names of its
  17. * contributors may be used to endorse or promote products derived from
  18. * this software without specific prior written permission.
  19. *
  20. * Alternatively, this software may be distributed under the terms of the
  21. * GNU General Public License ("GPL") version 2 as published by the Free
  22. * Software Foundation.
  23. *
  24. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  25. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  28. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  29. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  30. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  31. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  32. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  33. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  34. * POSSIBILITY OF SUCH DAMAGE.
  35. */
  36. #include <linux/rhashtable.h>
  37. #include <linux/sched/signal.h>
  38. #include "core.h"
  39. #include "name_table.h"
  40. #include "node.h"
  41. #include "link.h"
  42. #include "name_distr.h"
  43. #include "socket.h"
  44. #include "bcast.h"
  45. #include "netlink.h"
  46. #include "group.h"
  47. #define CONN_TIMEOUT_DEFAULT 8000 /* default connect timeout = 8s */
  48. #define CONN_PROBING_INTV msecs_to_jiffies(3600000) /* [ms] => 1 h */
  49. #define TIPC_FWD_MSG 1
  50. #define TIPC_MAX_PORT 0xffffffff
  51. #define TIPC_MIN_PORT 1
  52. #define TIPC_ACK_RATE 4 /* ACK at 1/4 of of rcv window size */
  53. enum {
  54. TIPC_LISTEN = TCP_LISTEN,
  55. TIPC_ESTABLISHED = TCP_ESTABLISHED,
  56. TIPC_OPEN = TCP_CLOSE,
  57. TIPC_DISCONNECTING = TCP_CLOSE_WAIT,
  58. TIPC_CONNECTING = TCP_SYN_SENT,
  59. };
  60. struct sockaddr_pair {
  61. struct sockaddr_tipc sock;
  62. struct sockaddr_tipc member;
  63. };
  64. /**
  65. * struct tipc_sock - TIPC socket structure
  66. * @sk: socket - interacts with 'port' and with user via the socket API
  67. * @conn_type: TIPC type used when connection was established
  68. * @conn_instance: TIPC instance used when connection was established
  69. * @published: non-zero if port has one or more associated names
  70. * @max_pkt: maximum packet size "hint" used when building messages sent by port
  71. * @portid: unique port identity in TIPC socket hash table
  72. * @phdr: preformatted message header used when sending messages
  73. * #cong_links: list of congested links
  74. * @publications: list of publications for port
  75. * @blocking_link: address of the congested link we are currently sleeping on
  76. * @pub_count: total # of publications port has made during its lifetime
  77. * @probing_state:
  78. * @conn_timeout: the time we can wait for an unresponded setup request
  79. * @dupl_rcvcnt: number of bytes counted twice, in both backlog and rcv queue
  80. * @cong_link_cnt: number of congested links
  81. * @snt_unacked: # messages sent by socket, and not yet acked by peer
  82. * @rcv_unacked: # messages read by user, but not yet acked back to peer
  83. * @peer: 'connected' peer for dgram/rdm
  84. * @node: hash table node
  85. * @mc_method: cookie for use between socket and broadcast layer
  86. * @rcu: rcu struct for tipc_sock
  87. */
  88. struct tipc_sock {
  89. struct sock sk;
  90. u32 conn_type;
  91. u32 conn_instance;
  92. int published;
  93. u32 max_pkt;
  94. u32 portid;
  95. struct tipc_msg phdr;
  96. struct list_head cong_links;
  97. struct list_head publications;
  98. u32 pub_count;
  99. uint conn_timeout;
  100. atomic_t dupl_rcvcnt;
  101. bool probe_unacked;
  102. u16 cong_link_cnt;
  103. u16 snt_unacked;
  104. u16 snd_win;
  105. u16 peer_caps;
  106. u16 rcv_unacked;
  107. u16 rcv_win;
  108. struct sockaddr_tipc peer;
  109. struct rhash_head node;
  110. struct tipc_mc_method mc_method;
  111. struct rcu_head rcu;
  112. struct tipc_group *group;
  113. };
  114. static int tipc_sk_backlog_rcv(struct sock *sk, struct sk_buff *skb);
  115. static void tipc_data_ready(struct sock *sk);
  116. static void tipc_write_space(struct sock *sk);
  117. static void tipc_sock_destruct(struct sock *sk);
  118. static int tipc_release(struct socket *sock);
  119. static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags,
  120. bool kern);
  121. static void tipc_sk_timeout(struct timer_list *t);
  122. static int tipc_sk_publish(struct tipc_sock *tsk, uint scope,
  123. struct tipc_name_seq const *seq);
  124. static int tipc_sk_withdraw(struct tipc_sock *tsk, uint scope,
  125. struct tipc_name_seq const *seq);
  126. static int tipc_sk_leave(struct tipc_sock *tsk);
  127. static struct tipc_sock *tipc_sk_lookup(struct net *net, u32 portid);
  128. static int tipc_sk_insert(struct tipc_sock *tsk);
  129. static void tipc_sk_remove(struct tipc_sock *tsk);
  130. static int __tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dsz);
  131. static int __tipc_sendmsg(struct socket *sock, struct msghdr *m, size_t dsz);
  132. static const struct proto_ops packet_ops;
  133. static const struct proto_ops stream_ops;
  134. static const struct proto_ops msg_ops;
  135. static struct proto tipc_proto;
  136. static const struct rhashtable_params tsk_rht_params;
  137. static u32 tsk_own_node(struct tipc_sock *tsk)
  138. {
  139. return msg_prevnode(&tsk->phdr);
  140. }
  141. static u32 tsk_peer_node(struct tipc_sock *tsk)
  142. {
  143. return msg_destnode(&tsk->phdr);
  144. }
  145. static u32 tsk_peer_port(struct tipc_sock *tsk)
  146. {
  147. return msg_destport(&tsk->phdr);
  148. }
  149. static bool tsk_unreliable(struct tipc_sock *tsk)
  150. {
  151. return msg_src_droppable(&tsk->phdr) != 0;
  152. }
  153. static void tsk_set_unreliable(struct tipc_sock *tsk, bool unreliable)
  154. {
  155. msg_set_src_droppable(&tsk->phdr, unreliable ? 1 : 0);
  156. }
  157. static bool tsk_unreturnable(struct tipc_sock *tsk)
  158. {
  159. return msg_dest_droppable(&tsk->phdr) != 0;
  160. }
  161. static void tsk_set_unreturnable(struct tipc_sock *tsk, bool unreturnable)
  162. {
  163. msg_set_dest_droppable(&tsk->phdr, unreturnable ? 1 : 0);
  164. }
  165. static int tsk_importance(struct tipc_sock *tsk)
  166. {
  167. return msg_importance(&tsk->phdr);
  168. }
  169. static int tsk_set_importance(struct tipc_sock *tsk, int imp)
  170. {
  171. if (imp > TIPC_CRITICAL_IMPORTANCE)
  172. return -EINVAL;
  173. msg_set_importance(&tsk->phdr, (u32)imp);
  174. return 0;
  175. }
  176. static struct tipc_sock *tipc_sk(const struct sock *sk)
  177. {
  178. return container_of(sk, struct tipc_sock, sk);
  179. }
  180. static bool tsk_conn_cong(struct tipc_sock *tsk)
  181. {
  182. return tsk->snt_unacked > tsk->snd_win;
  183. }
  184. static u16 tsk_blocks(int len)
  185. {
  186. return ((len / FLOWCTL_BLK_SZ) + 1);
  187. }
  188. /* tsk_blocks(): translate a buffer size in bytes to number of
  189. * advertisable blocks, taking into account the ratio truesize(len)/len
  190. * We can trust that this ratio is always < 4 for len >= FLOWCTL_BLK_SZ
  191. */
  192. static u16 tsk_adv_blocks(int len)
  193. {
  194. return len / FLOWCTL_BLK_SZ / 4;
  195. }
  196. /* tsk_inc(): increment counter for sent or received data
  197. * - If block based flow control is not supported by peer we
  198. * fall back to message based ditto, incrementing the counter
  199. */
  200. static u16 tsk_inc(struct tipc_sock *tsk, int msglen)
  201. {
  202. if (likely(tsk->peer_caps & TIPC_BLOCK_FLOWCTL))
  203. return ((msglen / FLOWCTL_BLK_SZ) + 1);
  204. return 1;
  205. }
  206. /**
  207. * tsk_advance_rx_queue - discard first buffer in socket receive queue
  208. *
  209. * Caller must hold socket lock
  210. */
  211. static void tsk_advance_rx_queue(struct sock *sk)
  212. {
  213. kfree_skb(__skb_dequeue(&sk->sk_receive_queue));
  214. }
  215. /* tipc_sk_respond() : send response message back to sender
  216. */
  217. static void tipc_sk_respond(struct sock *sk, struct sk_buff *skb, int err)
  218. {
  219. u32 selector;
  220. u32 dnode;
  221. u32 onode = tipc_own_addr(sock_net(sk));
  222. if (!tipc_msg_reverse(onode, &skb, err))
  223. return;
  224. dnode = msg_destnode(buf_msg(skb));
  225. selector = msg_origport(buf_msg(skb));
  226. tipc_node_xmit_skb(sock_net(sk), skb, dnode, selector);
  227. }
  228. /**
  229. * tsk_rej_rx_queue - reject all buffers in socket receive queue
  230. *
  231. * Caller must hold socket lock
  232. */
  233. static void tsk_rej_rx_queue(struct sock *sk)
  234. {
  235. struct sk_buff *skb;
  236. while ((skb = __skb_dequeue(&sk->sk_receive_queue)))
  237. tipc_sk_respond(sk, skb, TIPC_ERR_NO_PORT);
  238. }
  239. static bool tipc_sk_connected(struct sock *sk)
  240. {
  241. return sk->sk_state == TIPC_ESTABLISHED;
  242. }
  243. /* tipc_sk_type_connectionless - check if the socket is datagram socket
  244. * @sk: socket
  245. *
  246. * Returns true if connection less, false otherwise
  247. */
  248. static bool tipc_sk_type_connectionless(struct sock *sk)
  249. {
  250. return sk->sk_type == SOCK_RDM || sk->sk_type == SOCK_DGRAM;
  251. }
  252. /* tsk_peer_msg - verify if message was sent by connected port's peer
  253. *
  254. * Handles cases where the node's network address has changed from
  255. * the default of <0.0.0> to its configured setting.
  256. */
  257. static bool tsk_peer_msg(struct tipc_sock *tsk, struct tipc_msg *msg)
  258. {
  259. struct sock *sk = &tsk->sk;
  260. struct tipc_net *tn = net_generic(sock_net(sk), tipc_net_id);
  261. u32 peer_port = tsk_peer_port(tsk);
  262. u32 orig_node;
  263. u32 peer_node;
  264. if (unlikely(!tipc_sk_connected(sk)))
  265. return false;
  266. if (unlikely(msg_origport(msg) != peer_port))
  267. return false;
  268. orig_node = msg_orignode(msg);
  269. peer_node = tsk_peer_node(tsk);
  270. if (likely(orig_node == peer_node))
  271. return true;
  272. if (!orig_node && (peer_node == tn->own_addr))
  273. return true;
  274. if (!peer_node && (orig_node == tn->own_addr))
  275. return true;
  276. return false;
  277. }
  278. /* tipc_set_sk_state - set the sk_state of the socket
  279. * @sk: socket
  280. *
  281. * Caller must hold socket lock
  282. *
  283. * Returns 0 on success, errno otherwise
  284. */
  285. static int tipc_set_sk_state(struct sock *sk, int state)
  286. {
  287. int oldsk_state = sk->sk_state;
  288. int res = -EINVAL;
  289. switch (state) {
  290. case TIPC_OPEN:
  291. res = 0;
  292. break;
  293. case TIPC_LISTEN:
  294. case TIPC_CONNECTING:
  295. if (oldsk_state == TIPC_OPEN)
  296. res = 0;
  297. break;
  298. case TIPC_ESTABLISHED:
  299. if (oldsk_state == TIPC_CONNECTING ||
  300. oldsk_state == TIPC_OPEN)
  301. res = 0;
  302. break;
  303. case TIPC_DISCONNECTING:
  304. if (oldsk_state == TIPC_CONNECTING ||
  305. oldsk_state == TIPC_ESTABLISHED)
  306. res = 0;
  307. break;
  308. }
  309. if (!res)
  310. sk->sk_state = state;
  311. return res;
  312. }
  313. static int tipc_sk_sock_err(struct socket *sock, long *timeout)
  314. {
  315. struct sock *sk = sock->sk;
  316. int err = sock_error(sk);
  317. int typ = sock->type;
  318. if (err)
  319. return err;
  320. if (typ == SOCK_STREAM || typ == SOCK_SEQPACKET) {
  321. if (sk->sk_state == TIPC_DISCONNECTING)
  322. return -EPIPE;
  323. else if (!tipc_sk_connected(sk))
  324. return -ENOTCONN;
  325. }
  326. if (!*timeout)
  327. return -EAGAIN;
  328. if (signal_pending(current))
  329. return sock_intr_errno(*timeout);
  330. return 0;
  331. }
  332. #define tipc_wait_for_cond(sock_, timeo_, condition_) \
  333. ({ \
  334. struct sock *sk_; \
  335. int rc_; \
  336. \
  337. while ((rc_ = !(condition_))) { \
  338. DEFINE_WAIT_FUNC(wait_, woken_wake_function); \
  339. sk_ = (sock_)->sk; \
  340. rc_ = tipc_sk_sock_err((sock_), timeo_); \
  341. if (rc_) \
  342. break; \
  343. prepare_to_wait(sk_sleep(sk_), &wait_, TASK_INTERRUPTIBLE); \
  344. release_sock(sk_); \
  345. *(timeo_) = wait_woken(&wait_, TASK_INTERRUPTIBLE, *(timeo_)); \
  346. sched_annotate_sleep(); \
  347. lock_sock(sk_); \
  348. remove_wait_queue(sk_sleep(sk_), &wait_); \
  349. } \
  350. rc_; \
  351. })
  352. /**
  353. * tipc_sk_create - create a TIPC socket
  354. * @net: network namespace (must be default network)
  355. * @sock: pre-allocated socket structure
  356. * @protocol: protocol indicator (must be 0)
  357. * @kern: caused by kernel or by userspace?
  358. *
  359. * This routine creates additional data structures used by the TIPC socket,
  360. * initializes them, and links them together.
  361. *
  362. * Returns 0 on success, errno otherwise
  363. */
  364. static int tipc_sk_create(struct net *net, struct socket *sock,
  365. int protocol, int kern)
  366. {
  367. struct tipc_net *tn;
  368. const struct proto_ops *ops;
  369. struct sock *sk;
  370. struct tipc_sock *tsk;
  371. struct tipc_msg *msg;
  372. /* Validate arguments */
  373. if (unlikely(protocol != 0))
  374. return -EPROTONOSUPPORT;
  375. switch (sock->type) {
  376. case SOCK_STREAM:
  377. ops = &stream_ops;
  378. break;
  379. case SOCK_SEQPACKET:
  380. ops = &packet_ops;
  381. break;
  382. case SOCK_DGRAM:
  383. case SOCK_RDM:
  384. ops = &msg_ops;
  385. break;
  386. default:
  387. return -EPROTOTYPE;
  388. }
  389. /* Allocate socket's protocol area */
  390. sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto, kern);
  391. if (sk == NULL)
  392. return -ENOMEM;
  393. tsk = tipc_sk(sk);
  394. tsk->max_pkt = MAX_PKT_DEFAULT;
  395. INIT_LIST_HEAD(&tsk->publications);
  396. INIT_LIST_HEAD(&tsk->cong_links);
  397. msg = &tsk->phdr;
  398. tn = net_generic(sock_net(sk), tipc_net_id);
  399. /* Finish initializing socket data structures */
  400. sock->ops = ops;
  401. sock_init_data(sock, sk);
  402. tipc_set_sk_state(sk, TIPC_OPEN);
  403. if (tipc_sk_insert(tsk)) {
  404. pr_warn("Socket create failed; port number exhausted\n");
  405. return -EINVAL;
  406. }
  407. /* Ensure tsk is visible before we read own_addr. */
  408. smp_mb();
  409. tipc_msg_init(tn->own_addr, msg, TIPC_LOW_IMPORTANCE, TIPC_NAMED_MSG,
  410. NAMED_H_SIZE, 0);
  411. msg_set_origport(msg, tsk->portid);
  412. timer_setup(&sk->sk_timer, tipc_sk_timeout, 0);
  413. sk->sk_shutdown = 0;
  414. sk->sk_backlog_rcv = tipc_sk_backlog_rcv;
  415. sk->sk_rcvbuf = sysctl_tipc_rmem[1];
  416. sk->sk_data_ready = tipc_data_ready;
  417. sk->sk_write_space = tipc_write_space;
  418. sk->sk_destruct = tipc_sock_destruct;
  419. tsk->conn_timeout = CONN_TIMEOUT_DEFAULT;
  420. atomic_set(&tsk->dupl_rcvcnt, 0);
  421. /* Start out with safe limits until we receive an advertised window */
  422. tsk->snd_win = tsk_adv_blocks(RCVBUF_MIN);
  423. tsk->rcv_win = tsk->snd_win;
  424. if (tipc_sk_type_connectionless(sk)) {
  425. tsk_set_unreturnable(tsk, true);
  426. if (sock->type == SOCK_DGRAM)
  427. tsk_set_unreliable(tsk, true);
  428. }
  429. return 0;
  430. }
  431. static void tipc_sk_callback(struct rcu_head *head)
  432. {
  433. struct tipc_sock *tsk = container_of(head, struct tipc_sock, rcu);
  434. sock_put(&tsk->sk);
  435. }
  436. /* Caller should hold socket lock for the socket. */
  437. static void __tipc_shutdown(struct socket *sock, int error)
  438. {
  439. struct sock *sk = sock->sk;
  440. struct tipc_sock *tsk = tipc_sk(sk);
  441. struct net *net = sock_net(sk);
  442. long timeout = CONN_TIMEOUT_DEFAULT;
  443. u32 dnode = tsk_peer_node(tsk);
  444. struct sk_buff *skb;
  445. /* Avoid that hi-prio shutdown msgs bypass msgs in link wakeup queue */
  446. tipc_wait_for_cond(sock, &timeout, (!tsk->cong_link_cnt &&
  447. !tsk_conn_cong(tsk)));
  448. /* Reject all unreceived messages, except on an active connection
  449. * (which disconnects locally & sends a 'FIN+' to peer).
  450. */
  451. while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
  452. if (TIPC_SKB_CB(skb)->bytes_read) {
  453. kfree_skb(skb);
  454. continue;
  455. }
  456. if (!tipc_sk_type_connectionless(sk) &&
  457. sk->sk_state != TIPC_DISCONNECTING) {
  458. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  459. tipc_node_remove_conn(net, dnode, tsk->portid);
  460. }
  461. tipc_sk_respond(sk, skb, error);
  462. }
  463. if (tipc_sk_type_connectionless(sk))
  464. return;
  465. if (sk->sk_state != TIPC_DISCONNECTING) {
  466. skb = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE,
  467. TIPC_CONN_MSG, SHORT_H_SIZE, 0, dnode,
  468. tsk_own_node(tsk), tsk_peer_port(tsk),
  469. tsk->portid, error);
  470. if (skb)
  471. tipc_node_xmit_skb(net, skb, dnode, tsk->portid);
  472. tipc_node_remove_conn(net, dnode, tsk->portid);
  473. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  474. }
  475. }
  476. /**
  477. * tipc_release - destroy a TIPC socket
  478. * @sock: socket to destroy
  479. *
  480. * This routine cleans up any messages that are still queued on the socket.
  481. * For DGRAM and RDM socket types, all queued messages are rejected.
  482. * For SEQPACKET and STREAM socket types, the first message is rejected
  483. * and any others are discarded. (If the first message on a STREAM socket
  484. * is partially-read, it is discarded and the next one is rejected instead.)
  485. *
  486. * NOTE: Rejected messages are not necessarily returned to the sender! They
  487. * are returned or discarded according to the "destination droppable" setting
  488. * specified for the message by the sender.
  489. *
  490. * Returns 0 on success, errno otherwise
  491. */
  492. static int tipc_release(struct socket *sock)
  493. {
  494. struct sock *sk = sock->sk;
  495. struct tipc_sock *tsk;
  496. /*
  497. * Exit if socket isn't fully initialized (occurs when a failed accept()
  498. * releases a pre-allocated child socket that was never used)
  499. */
  500. if (sk == NULL)
  501. return 0;
  502. tsk = tipc_sk(sk);
  503. lock_sock(sk);
  504. __tipc_shutdown(sock, TIPC_ERR_NO_PORT);
  505. sk->sk_shutdown = SHUTDOWN_MASK;
  506. tipc_sk_leave(tsk);
  507. tipc_sk_withdraw(tsk, 0, NULL);
  508. sk_stop_timer(sk, &sk->sk_timer);
  509. tipc_sk_remove(tsk);
  510. /* Reject any messages that accumulated in backlog queue */
  511. release_sock(sk);
  512. tipc_dest_list_purge(&tsk->cong_links);
  513. tsk->cong_link_cnt = 0;
  514. call_rcu(&tsk->rcu, tipc_sk_callback);
  515. sock->sk = NULL;
  516. return 0;
  517. }
  518. /**
  519. * tipc_bind - associate or disassocate TIPC name(s) with a socket
  520. * @sock: socket structure
  521. * @uaddr: socket address describing name(s) and desired operation
  522. * @uaddr_len: size of socket address data structure
  523. *
  524. * Name and name sequence binding is indicated using a positive scope value;
  525. * a negative scope value unbinds the specified name. Specifying no name
  526. * (i.e. a socket address length of 0) unbinds all names from the socket.
  527. *
  528. * Returns 0 on success, errno otherwise
  529. *
  530. * NOTE: This routine doesn't need to take the socket lock since it doesn't
  531. * access any non-constant socket information.
  532. */
  533. static int tipc_bind(struct socket *sock, struct sockaddr *uaddr,
  534. int uaddr_len)
  535. {
  536. struct sock *sk = sock->sk;
  537. struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
  538. struct tipc_sock *tsk = tipc_sk(sk);
  539. int res = -EINVAL;
  540. lock_sock(sk);
  541. if (unlikely(!uaddr_len)) {
  542. res = tipc_sk_withdraw(tsk, 0, NULL);
  543. goto exit;
  544. }
  545. if (tsk->group) {
  546. res = -EACCES;
  547. goto exit;
  548. }
  549. if (uaddr_len < sizeof(struct sockaddr_tipc)) {
  550. res = -EINVAL;
  551. goto exit;
  552. }
  553. if (addr->family != AF_TIPC) {
  554. res = -EAFNOSUPPORT;
  555. goto exit;
  556. }
  557. if (addr->addrtype == TIPC_ADDR_NAME)
  558. addr->addr.nameseq.upper = addr->addr.nameseq.lower;
  559. else if (addr->addrtype != TIPC_ADDR_NAMESEQ) {
  560. res = -EAFNOSUPPORT;
  561. goto exit;
  562. }
  563. if ((addr->addr.nameseq.type < TIPC_RESERVED_TYPES) &&
  564. (addr->addr.nameseq.type != TIPC_TOP_SRV) &&
  565. (addr->addr.nameseq.type != TIPC_CFG_SRV)) {
  566. res = -EACCES;
  567. goto exit;
  568. }
  569. res = (addr->scope > 0) ?
  570. tipc_sk_publish(tsk, addr->scope, &addr->addr.nameseq) :
  571. tipc_sk_withdraw(tsk, -addr->scope, &addr->addr.nameseq);
  572. exit:
  573. release_sock(sk);
  574. return res;
  575. }
  576. /**
  577. * tipc_getname - get port ID of socket or peer socket
  578. * @sock: socket structure
  579. * @uaddr: area for returned socket address
  580. * @uaddr_len: area for returned length of socket address
  581. * @peer: 0 = own ID, 1 = current peer ID, 2 = current/former peer ID
  582. *
  583. * Returns 0 on success, errno otherwise
  584. *
  585. * NOTE: This routine doesn't need to take the socket lock since it only
  586. * accesses socket information that is unchanging (or which changes in
  587. * a completely predictable manner).
  588. */
  589. static int tipc_getname(struct socket *sock, struct sockaddr *uaddr,
  590. int *uaddr_len, int peer)
  591. {
  592. struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
  593. struct sock *sk = sock->sk;
  594. struct tipc_sock *tsk = tipc_sk(sk);
  595. struct tipc_net *tn = net_generic(sock_net(sock->sk), tipc_net_id);
  596. memset(addr, 0, sizeof(*addr));
  597. if (peer) {
  598. if ((!tipc_sk_connected(sk)) &&
  599. ((peer != 2) || (sk->sk_state != TIPC_DISCONNECTING)))
  600. return -ENOTCONN;
  601. addr->addr.id.ref = tsk_peer_port(tsk);
  602. addr->addr.id.node = tsk_peer_node(tsk);
  603. } else {
  604. addr->addr.id.ref = tsk->portid;
  605. addr->addr.id.node = tn->own_addr;
  606. }
  607. *uaddr_len = sizeof(*addr);
  608. addr->addrtype = TIPC_ADDR_ID;
  609. addr->family = AF_TIPC;
  610. addr->scope = 0;
  611. addr->addr.name.domain = 0;
  612. return 0;
  613. }
  614. /**
  615. * tipc_poll - read and possibly block on pollmask
  616. * @file: file structure associated with the socket
  617. * @sock: socket for which to calculate the poll bits
  618. * @wait: ???
  619. *
  620. * Returns pollmask value
  621. *
  622. * COMMENTARY:
  623. * It appears that the usual socket locking mechanisms are not useful here
  624. * since the pollmask info is potentially out-of-date the moment this routine
  625. * exits. TCP and other protocols seem to rely on higher level poll routines
  626. * to handle any preventable race conditions, so TIPC will do the same ...
  627. *
  628. * IMPORTANT: The fact that a read or write operation is indicated does NOT
  629. * imply that the operation will succeed, merely that it should be performed
  630. * and will not block.
  631. */
  632. static unsigned int tipc_poll(struct file *file, struct socket *sock,
  633. poll_table *wait)
  634. {
  635. struct sock *sk = sock->sk;
  636. struct tipc_sock *tsk = tipc_sk(sk);
  637. struct tipc_group *grp = tsk->group;
  638. u32 revents = 0;
  639. sock_poll_wait(file, sk_sleep(sk), wait);
  640. if (sk->sk_shutdown & RCV_SHUTDOWN)
  641. revents |= POLLRDHUP | POLLIN | POLLRDNORM;
  642. if (sk->sk_shutdown == SHUTDOWN_MASK)
  643. revents |= POLLHUP;
  644. switch (sk->sk_state) {
  645. case TIPC_ESTABLISHED:
  646. if (!tsk->cong_link_cnt && !tsk_conn_cong(tsk))
  647. revents |= POLLOUT;
  648. /* fall thru' */
  649. case TIPC_LISTEN:
  650. case TIPC_CONNECTING:
  651. if (!skb_queue_empty(&sk->sk_receive_queue))
  652. revents |= POLLIN | POLLRDNORM;
  653. break;
  654. case TIPC_OPEN:
  655. if (!grp || tipc_group_size(grp))
  656. if (!tsk->cong_link_cnt)
  657. revents |= POLLOUT;
  658. if (!tipc_sk_type_connectionless(sk))
  659. break;
  660. if (skb_queue_empty(&sk->sk_receive_queue))
  661. break;
  662. revents |= POLLIN | POLLRDNORM;
  663. break;
  664. case TIPC_DISCONNECTING:
  665. revents = POLLIN | POLLRDNORM | POLLHUP;
  666. break;
  667. }
  668. return revents;
  669. }
  670. /**
  671. * tipc_sendmcast - send multicast message
  672. * @sock: socket structure
  673. * @seq: destination address
  674. * @msg: message to send
  675. * @dlen: length of data to send
  676. * @timeout: timeout to wait for wakeup
  677. *
  678. * Called from function tipc_sendmsg(), which has done all sanity checks
  679. * Returns the number of bytes sent on success, or errno
  680. */
  681. static int tipc_sendmcast(struct socket *sock, struct tipc_name_seq *seq,
  682. struct msghdr *msg, size_t dlen, long timeout)
  683. {
  684. struct sock *sk = sock->sk;
  685. struct tipc_sock *tsk = tipc_sk(sk);
  686. struct tipc_msg *hdr = &tsk->phdr;
  687. struct net *net = sock_net(sk);
  688. int mtu = tipc_bcast_get_mtu(net);
  689. struct tipc_mc_method *method = &tsk->mc_method;
  690. u32 domain = addr_domain(net, TIPC_CLUSTER_SCOPE);
  691. struct sk_buff_head pkts;
  692. struct tipc_nlist dsts;
  693. int rc;
  694. if (tsk->group)
  695. return -EACCES;
  696. /* Block or return if any destination link is congested */
  697. rc = tipc_wait_for_cond(sock, &timeout, !tsk->cong_link_cnt);
  698. if (unlikely(rc))
  699. return rc;
  700. /* Lookup destination nodes */
  701. tipc_nlist_init(&dsts, tipc_own_addr(net));
  702. tipc_nametbl_lookup_dst_nodes(net, seq->type, seq->lower,
  703. seq->upper, domain, &dsts);
  704. if (!dsts.local && !dsts.remote)
  705. return -EHOSTUNREACH;
  706. /* Build message header */
  707. msg_set_type(hdr, TIPC_MCAST_MSG);
  708. msg_set_hdr_sz(hdr, MCAST_H_SIZE);
  709. msg_set_lookup_scope(hdr, TIPC_CLUSTER_SCOPE);
  710. msg_set_destport(hdr, 0);
  711. msg_set_destnode(hdr, 0);
  712. msg_set_nametype(hdr, seq->type);
  713. msg_set_namelower(hdr, seq->lower);
  714. msg_set_nameupper(hdr, seq->upper);
  715. /* Build message as chain of buffers */
  716. skb_queue_head_init(&pkts);
  717. rc = tipc_msg_build(hdr, msg, 0, dlen, mtu, &pkts);
  718. /* Send message if build was successful */
  719. if (unlikely(rc == dlen))
  720. rc = tipc_mcast_xmit(net, &pkts, method, &dsts,
  721. &tsk->cong_link_cnt);
  722. tipc_nlist_purge(&dsts);
  723. return rc ? rc : dlen;
  724. }
  725. /**
  726. * tipc_send_group_msg - send a message to a member in the group
  727. * @net: network namespace
  728. * @m: message to send
  729. * @mb: group member
  730. * @dnode: destination node
  731. * @dport: destination port
  732. * @dlen: total length of message data
  733. */
  734. static int tipc_send_group_msg(struct net *net, struct tipc_sock *tsk,
  735. struct msghdr *m, struct tipc_member *mb,
  736. u32 dnode, u32 dport, int dlen)
  737. {
  738. u16 bc_snd_nxt = tipc_group_bc_snd_nxt(tsk->group);
  739. struct tipc_mc_method *method = &tsk->mc_method;
  740. int blks = tsk_blocks(GROUP_H_SIZE + dlen);
  741. struct tipc_msg *hdr = &tsk->phdr;
  742. struct sk_buff_head pkts;
  743. int mtu, rc;
  744. /* Complete message header */
  745. msg_set_type(hdr, TIPC_GRP_UCAST_MSG);
  746. msg_set_hdr_sz(hdr, GROUP_H_SIZE);
  747. msg_set_destport(hdr, dport);
  748. msg_set_destnode(hdr, dnode);
  749. msg_set_grp_bc_seqno(hdr, bc_snd_nxt);
  750. /* Build message as chain of buffers */
  751. skb_queue_head_init(&pkts);
  752. mtu = tipc_node_get_mtu(net, dnode, tsk->portid);
  753. rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
  754. if (unlikely(rc != dlen))
  755. return rc;
  756. /* Send message */
  757. rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
  758. if (unlikely(rc == -ELINKCONG)) {
  759. tipc_dest_push(&tsk->cong_links, dnode, 0);
  760. tsk->cong_link_cnt++;
  761. }
  762. /* Update send window */
  763. tipc_group_update_member(mb, blks);
  764. /* A broadcast sent within next EXPIRE period must follow same path */
  765. method->rcast = true;
  766. method->mandatory = true;
  767. return dlen;
  768. }
  769. /**
  770. * tipc_send_group_unicast - send message to a member in the group
  771. * @sock: socket structure
  772. * @m: message to send
  773. * @dlen: total length of message data
  774. * @timeout: timeout to wait for wakeup
  775. *
  776. * Called from function tipc_sendmsg(), which has done all sanity checks
  777. * Returns the number of bytes sent on success, or errno
  778. */
  779. static int tipc_send_group_unicast(struct socket *sock, struct msghdr *m,
  780. int dlen, long timeout)
  781. {
  782. struct sock *sk = sock->sk;
  783. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  784. int blks = tsk_blocks(GROUP_H_SIZE + dlen);
  785. struct tipc_sock *tsk = tipc_sk(sk);
  786. struct tipc_group *grp = tsk->group;
  787. struct net *net = sock_net(sk);
  788. struct tipc_member *mb = NULL;
  789. u32 node, port;
  790. int rc;
  791. node = dest->addr.id.node;
  792. port = dest->addr.id.ref;
  793. if (!port && !node)
  794. return -EHOSTUNREACH;
  795. /* Block or return if destination link or member is congested */
  796. rc = tipc_wait_for_cond(sock, &timeout,
  797. !tipc_dest_find(&tsk->cong_links, node, 0) &&
  798. !tipc_group_cong(grp, node, port, blks, &mb));
  799. if (unlikely(rc))
  800. return rc;
  801. if (unlikely(!mb))
  802. return -EHOSTUNREACH;
  803. rc = tipc_send_group_msg(net, tsk, m, mb, node, port, dlen);
  804. return rc ? rc : dlen;
  805. }
  806. /**
  807. * tipc_send_group_anycast - send message to any member with given identity
  808. * @sock: socket structure
  809. * @m: message to send
  810. * @dlen: total length of message data
  811. * @timeout: timeout to wait for wakeup
  812. *
  813. * Called from function tipc_sendmsg(), which has done all sanity checks
  814. * Returns the number of bytes sent on success, or errno
  815. */
  816. static int tipc_send_group_anycast(struct socket *sock, struct msghdr *m,
  817. int dlen, long timeout)
  818. {
  819. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  820. struct sock *sk = sock->sk;
  821. struct tipc_sock *tsk = tipc_sk(sk);
  822. struct list_head *cong_links = &tsk->cong_links;
  823. int blks = tsk_blocks(GROUP_H_SIZE + dlen);
  824. struct tipc_group *grp = tsk->group;
  825. struct tipc_member *first = NULL;
  826. struct tipc_member *mbr = NULL;
  827. struct net *net = sock_net(sk);
  828. u32 node, port, exclude;
  829. u32 type, inst, domain;
  830. struct list_head dsts;
  831. int lookups = 0;
  832. int dstcnt, rc;
  833. bool cong;
  834. INIT_LIST_HEAD(&dsts);
  835. type = dest->addr.name.name.type;
  836. inst = dest->addr.name.name.instance;
  837. domain = addr_domain(net, dest->scope);
  838. exclude = tipc_group_exclude(grp);
  839. while (++lookups < 4) {
  840. first = NULL;
  841. /* Look for a non-congested destination member, if any */
  842. while (1) {
  843. if (!tipc_nametbl_lookup(net, type, inst, domain, &dsts,
  844. &dstcnt, exclude, false))
  845. return -EHOSTUNREACH;
  846. tipc_dest_pop(&dsts, &node, &port);
  847. cong = tipc_group_cong(grp, node, port, blks, &mbr);
  848. if (!cong)
  849. break;
  850. if (mbr == first)
  851. break;
  852. if (!first)
  853. first = mbr;
  854. }
  855. /* Start over if destination was not in member list */
  856. if (unlikely(!mbr))
  857. continue;
  858. if (likely(!cong && !tipc_dest_find(cong_links, node, 0)))
  859. break;
  860. /* Block or return if destination link or member is congested */
  861. rc = tipc_wait_for_cond(sock, &timeout,
  862. !tipc_dest_find(cong_links, node, 0) &&
  863. !tipc_group_cong(grp, node, port,
  864. blks, &mbr));
  865. if (unlikely(rc))
  866. return rc;
  867. /* Send, unless destination disappeared while waiting */
  868. if (likely(mbr))
  869. break;
  870. }
  871. if (unlikely(lookups >= 4))
  872. return -EHOSTUNREACH;
  873. rc = tipc_send_group_msg(net, tsk, m, mbr, node, port, dlen);
  874. return rc ? rc : dlen;
  875. }
  876. /**
  877. * tipc_send_group_bcast - send message to all members in communication group
  878. * @sk: socket structure
  879. * @m: message to send
  880. * @dlen: total length of message data
  881. * @timeout: timeout to wait for wakeup
  882. *
  883. * Called from function tipc_sendmsg(), which has done all sanity checks
  884. * Returns the number of bytes sent on success, or errno
  885. */
  886. static int tipc_send_group_bcast(struct socket *sock, struct msghdr *m,
  887. int dlen, long timeout)
  888. {
  889. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  890. struct sock *sk = sock->sk;
  891. struct net *net = sock_net(sk);
  892. struct tipc_sock *tsk = tipc_sk(sk);
  893. struct tipc_group *grp = tsk->group;
  894. struct tipc_nlist *dsts = tipc_group_dests(grp);
  895. struct tipc_mc_method *method = &tsk->mc_method;
  896. bool ack = method->mandatory && method->rcast;
  897. int blks = tsk_blocks(MCAST_H_SIZE + dlen);
  898. struct tipc_msg *hdr = &tsk->phdr;
  899. int mtu = tipc_bcast_get_mtu(net);
  900. struct sk_buff_head pkts;
  901. int rc = -EHOSTUNREACH;
  902. if (!dsts->local && !dsts->remote)
  903. return -EHOSTUNREACH;
  904. /* Block or return if any destination link or member is congested */
  905. rc = tipc_wait_for_cond(sock, &timeout, !tsk->cong_link_cnt &&
  906. !tipc_group_bc_cong(grp, blks));
  907. if (unlikely(rc))
  908. return rc;
  909. /* Complete message header */
  910. if (dest) {
  911. msg_set_type(hdr, TIPC_GRP_MCAST_MSG);
  912. msg_set_nameinst(hdr, dest->addr.name.name.instance);
  913. } else {
  914. msg_set_type(hdr, TIPC_GRP_BCAST_MSG);
  915. msg_set_nameinst(hdr, 0);
  916. }
  917. msg_set_hdr_sz(hdr, GROUP_H_SIZE);
  918. msg_set_destport(hdr, 0);
  919. msg_set_destnode(hdr, 0);
  920. msg_set_grp_bc_seqno(hdr, tipc_group_bc_snd_nxt(grp));
  921. /* Avoid getting stuck with repeated forced replicasts */
  922. msg_set_grp_bc_ack_req(hdr, ack);
  923. /* Build message as chain of buffers */
  924. skb_queue_head_init(&pkts);
  925. rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
  926. if (unlikely(rc != dlen))
  927. return rc;
  928. /* Send message */
  929. rc = tipc_mcast_xmit(net, &pkts, method, dsts, &tsk->cong_link_cnt);
  930. if (unlikely(rc))
  931. return rc;
  932. /* Update broadcast sequence number and send windows */
  933. tipc_group_update_bc_members(tsk->group, blks, ack);
  934. /* Broadcast link is now free to choose method for next broadcast */
  935. method->mandatory = false;
  936. method->expires = jiffies;
  937. return dlen;
  938. }
  939. /**
  940. * tipc_send_group_mcast - send message to all members with given identity
  941. * @sock: socket structure
  942. * @m: message to send
  943. * @dlen: total length of message data
  944. * @timeout: timeout to wait for wakeup
  945. *
  946. * Called from function tipc_sendmsg(), which has done all sanity checks
  947. * Returns the number of bytes sent on success, or errno
  948. */
  949. static int tipc_send_group_mcast(struct socket *sock, struct msghdr *m,
  950. int dlen, long timeout)
  951. {
  952. struct sock *sk = sock->sk;
  953. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  954. struct tipc_name_seq *seq = &dest->addr.nameseq;
  955. struct tipc_sock *tsk = tipc_sk(sk);
  956. struct tipc_group *grp = tsk->group;
  957. struct net *net = sock_net(sk);
  958. u32 domain, exclude, dstcnt;
  959. struct list_head dsts;
  960. INIT_LIST_HEAD(&dsts);
  961. if (seq->lower != seq->upper)
  962. return -ENOTSUPP;
  963. domain = addr_domain(net, dest->scope);
  964. exclude = tipc_group_exclude(grp);
  965. if (!tipc_nametbl_lookup(net, seq->type, seq->lower, domain,
  966. &dsts, &dstcnt, exclude, true))
  967. return -EHOSTUNREACH;
  968. if (dstcnt == 1) {
  969. tipc_dest_pop(&dsts, &dest->addr.id.node, &dest->addr.id.ref);
  970. return tipc_send_group_unicast(sock, m, dlen, timeout);
  971. }
  972. tipc_dest_list_purge(&dsts);
  973. return tipc_send_group_bcast(sock, m, dlen, timeout);
  974. }
  975. /**
  976. * tipc_sk_mcast_rcv - Deliver multicast messages to all destination sockets
  977. * @arrvq: queue with arriving messages, to be cloned after destination lookup
  978. * @inputq: queue with cloned messages, delivered to socket after dest lookup
  979. *
  980. * Multi-threaded: parallel calls with reference to same queues may occur
  981. */
  982. void tipc_sk_mcast_rcv(struct net *net, struct sk_buff_head *arrvq,
  983. struct sk_buff_head *inputq)
  984. {
  985. u32 scope = TIPC_CLUSTER_SCOPE;
  986. u32 self = tipc_own_addr(net);
  987. struct sk_buff *skb, *_skb;
  988. u32 lower = 0, upper = ~0;
  989. struct sk_buff_head tmpq;
  990. u32 portid, oport, onode;
  991. struct list_head dports;
  992. struct tipc_msg *msg;
  993. int user, mtyp, hsz;
  994. __skb_queue_head_init(&tmpq);
  995. INIT_LIST_HEAD(&dports);
  996. skb = tipc_skb_peek(arrvq, &inputq->lock);
  997. for (; skb; skb = tipc_skb_peek(arrvq, &inputq->lock)) {
  998. msg = buf_msg(skb);
  999. user = msg_user(msg);
  1000. mtyp = msg_type(msg);
  1001. if (mtyp == TIPC_GRP_UCAST_MSG || user == GROUP_PROTOCOL) {
  1002. spin_lock_bh(&inputq->lock);
  1003. if (skb_peek(arrvq) == skb) {
  1004. __skb_dequeue(arrvq);
  1005. __skb_queue_tail(inputq, skb);
  1006. }
  1007. refcount_dec(&skb->users);
  1008. spin_unlock_bh(&inputq->lock);
  1009. continue;
  1010. }
  1011. hsz = skb_headroom(skb) + msg_hdr_sz(msg);
  1012. oport = msg_origport(msg);
  1013. onode = msg_orignode(msg);
  1014. if (onode == self)
  1015. scope = TIPC_NODE_SCOPE;
  1016. /* Create destination port list and message clones: */
  1017. if (!msg_in_group(msg)) {
  1018. lower = msg_namelower(msg);
  1019. upper = msg_nameupper(msg);
  1020. }
  1021. tipc_nametbl_mc_translate(net, msg_nametype(msg), lower, upper,
  1022. scope, &dports);
  1023. while (tipc_dest_pop(&dports, NULL, &portid)) {
  1024. _skb = __pskb_copy(skb, hsz, GFP_ATOMIC);
  1025. if (_skb) {
  1026. msg_set_destport(buf_msg(_skb), portid);
  1027. __skb_queue_tail(&tmpq, _skb);
  1028. continue;
  1029. }
  1030. pr_warn("Failed to clone mcast rcv buffer\n");
  1031. }
  1032. /* Append to inputq if not already done by other thread */
  1033. spin_lock_bh(&inputq->lock);
  1034. if (skb_peek(arrvq) == skb) {
  1035. skb_queue_splice_tail_init(&tmpq, inputq);
  1036. kfree_skb(__skb_dequeue(arrvq));
  1037. }
  1038. spin_unlock_bh(&inputq->lock);
  1039. __skb_queue_purge(&tmpq);
  1040. kfree_skb(skb);
  1041. }
  1042. tipc_sk_rcv(net, inputq);
  1043. }
  1044. /**
  1045. * tipc_sk_conn_proto_rcv - receive a connection mng protocol message
  1046. * @tsk: receiving socket
  1047. * @skb: pointer to message buffer.
  1048. */
  1049. static void tipc_sk_conn_proto_rcv(struct tipc_sock *tsk, struct sk_buff *skb,
  1050. struct sk_buff_head *xmitq)
  1051. {
  1052. struct tipc_msg *hdr = buf_msg(skb);
  1053. u32 onode = tsk_own_node(tsk);
  1054. struct sock *sk = &tsk->sk;
  1055. int mtyp = msg_type(hdr);
  1056. bool conn_cong;
  1057. /* Ignore if connection cannot be validated: */
  1058. if (!tsk_peer_msg(tsk, hdr))
  1059. goto exit;
  1060. if (unlikely(msg_errcode(hdr))) {
  1061. tipc_set_sk_state(sk, TIPC_DISCONNECTING);
  1062. tipc_node_remove_conn(sock_net(sk), tsk_peer_node(tsk),
  1063. tsk_peer_port(tsk));
  1064. sk->sk_state_change(sk);
  1065. goto exit;
  1066. }
  1067. tsk->probe_unacked = false;
  1068. if (mtyp == CONN_PROBE) {
  1069. msg_set_type(hdr, CONN_PROBE_REPLY);
  1070. if (tipc_msg_reverse(onode, &skb, TIPC_OK))
  1071. __skb_queue_tail(xmitq, skb);
  1072. return;
  1073. } else if (mtyp == CONN_ACK) {
  1074. conn_cong = tsk_conn_cong(tsk);
  1075. tsk->snt_unacked -= msg_conn_ack(hdr);
  1076. if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL)
  1077. tsk->snd_win = msg_adv_win(hdr);
  1078. if (conn_cong)
  1079. sk->sk_write_space(sk);
  1080. } else if (mtyp != CONN_PROBE_REPLY) {
  1081. pr_warn("Received unknown CONN_PROTO msg\n");
  1082. }
  1083. exit:
  1084. kfree_skb(skb);
  1085. }
  1086. /**
  1087. * tipc_sendmsg - send message in connectionless manner
  1088. * @sock: socket structure
  1089. * @m: message to send
  1090. * @dsz: amount of user data to be sent
  1091. *
  1092. * Message must have an destination specified explicitly.
  1093. * Used for SOCK_RDM and SOCK_DGRAM messages,
  1094. * and for 'SYN' messages on SOCK_SEQPACKET and SOCK_STREAM connections.
  1095. * (Note: 'SYN+' is prohibited on SOCK_STREAM.)
  1096. *
  1097. * Returns the number of bytes sent on success, or errno otherwise
  1098. */
  1099. static int tipc_sendmsg(struct socket *sock,
  1100. struct msghdr *m, size_t dsz)
  1101. {
  1102. struct sock *sk = sock->sk;
  1103. int ret;
  1104. lock_sock(sk);
  1105. ret = __tipc_sendmsg(sock, m, dsz);
  1106. release_sock(sk);
  1107. return ret;
  1108. }
  1109. static int __tipc_sendmsg(struct socket *sock, struct msghdr *m, size_t dlen)
  1110. {
  1111. struct sock *sk = sock->sk;
  1112. struct net *net = sock_net(sk);
  1113. struct tipc_sock *tsk = tipc_sk(sk);
  1114. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  1115. long timeout = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
  1116. struct list_head *clinks = &tsk->cong_links;
  1117. bool syn = !tipc_sk_type_connectionless(sk);
  1118. struct tipc_group *grp = tsk->group;
  1119. struct tipc_msg *hdr = &tsk->phdr;
  1120. struct tipc_name_seq *seq;
  1121. struct sk_buff_head pkts;
  1122. u32 type, inst, domain;
  1123. u32 dnode, dport;
  1124. int mtu, rc;
  1125. if (unlikely(dlen > TIPC_MAX_USER_MSG_SIZE))
  1126. return -EMSGSIZE;
  1127. if (likely(dest)) {
  1128. if (unlikely(m->msg_namelen < sizeof(*dest)))
  1129. return -EINVAL;
  1130. if (unlikely(dest->family != AF_TIPC))
  1131. return -EINVAL;
  1132. }
  1133. if (grp) {
  1134. if (!dest)
  1135. return tipc_send_group_bcast(sock, m, dlen, timeout);
  1136. if (dest->addrtype == TIPC_ADDR_NAME)
  1137. return tipc_send_group_anycast(sock, m, dlen, timeout);
  1138. if (dest->addrtype == TIPC_ADDR_ID)
  1139. return tipc_send_group_unicast(sock, m, dlen, timeout);
  1140. if (dest->addrtype == TIPC_ADDR_MCAST)
  1141. return tipc_send_group_mcast(sock, m, dlen, timeout);
  1142. return -EINVAL;
  1143. }
  1144. if (unlikely(!dest)) {
  1145. dest = &tsk->peer;
  1146. if (!syn || dest->family != AF_TIPC)
  1147. return -EDESTADDRREQ;
  1148. }
  1149. if (unlikely(syn)) {
  1150. if (sk->sk_state == TIPC_LISTEN)
  1151. return -EPIPE;
  1152. if (sk->sk_state != TIPC_OPEN)
  1153. return -EISCONN;
  1154. if (tsk->published)
  1155. return -EOPNOTSUPP;
  1156. if (dest->addrtype == TIPC_ADDR_NAME) {
  1157. tsk->conn_type = dest->addr.name.name.type;
  1158. tsk->conn_instance = dest->addr.name.name.instance;
  1159. }
  1160. }
  1161. seq = &dest->addr.nameseq;
  1162. if (dest->addrtype == TIPC_ADDR_MCAST)
  1163. return tipc_sendmcast(sock, seq, m, dlen, timeout);
  1164. if (dest->addrtype == TIPC_ADDR_NAME) {
  1165. type = dest->addr.name.name.type;
  1166. inst = dest->addr.name.name.instance;
  1167. domain = dest->addr.name.domain;
  1168. dnode = domain;
  1169. msg_set_type(hdr, TIPC_NAMED_MSG);
  1170. msg_set_hdr_sz(hdr, NAMED_H_SIZE);
  1171. msg_set_nametype(hdr, type);
  1172. msg_set_nameinst(hdr, inst);
  1173. msg_set_lookup_scope(hdr, tipc_addr_scope(domain));
  1174. dport = tipc_nametbl_translate(net, type, inst, &dnode);
  1175. msg_set_destnode(hdr, dnode);
  1176. msg_set_destport(hdr, dport);
  1177. if (unlikely(!dport && !dnode))
  1178. return -EHOSTUNREACH;
  1179. } else if (dest->addrtype == TIPC_ADDR_ID) {
  1180. dnode = dest->addr.id.node;
  1181. msg_set_type(hdr, TIPC_DIRECT_MSG);
  1182. msg_set_lookup_scope(hdr, 0);
  1183. msg_set_destnode(hdr, dnode);
  1184. msg_set_destport(hdr, dest->addr.id.ref);
  1185. msg_set_hdr_sz(hdr, BASIC_H_SIZE);
  1186. }
  1187. /* Block or return if destination link is congested */
  1188. rc = tipc_wait_for_cond(sock, &timeout,
  1189. !tipc_dest_find(clinks, dnode, 0));
  1190. if (unlikely(rc))
  1191. return rc;
  1192. skb_queue_head_init(&pkts);
  1193. mtu = tipc_node_get_mtu(net, dnode, tsk->portid);
  1194. rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
  1195. if (unlikely(rc != dlen))
  1196. return rc;
  1197. rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
  1198. if (unlikely(rc == -ELINKCONG)) {
  1199. tipc_dest_push(clinks, dnode, 0);
  1200. tsk->cong_link_cnt++;
  1201. rc = 0;
  1202. }
  1203. if (unlikely(syn && !rc))
  1204. tipc_set_sk_state(sk, TIPC_CONNECTING);
  1205. return rc ? rc : dlen;
  1206. }
  1207. /**
  1208. * tipc_sendstream - send stream-oriented data
  1209. * @sock: socket structure
  1210. * @m: data to send
  1211. * @dsz: total length of data to be transmitted
  1212. *
  1213. * Used for SOCK_STREAM data.
  1214. *
  1215. * Returns the number of bytes sent on success (or partial success),
  1216. * or errno if no data sent
  1217. */
  1218. static int tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dsz)
  1219. {
  1220. struct sock *sk = sock->sk;
  1221. int ret;
  1222. lock_sock(sk);
  1223. ret = __tipc_sendstream(sock, m, dsz);
  1224. release_sock(sk);
  1225. return ret;
  1226. }
  1227. static int __tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dlen)
  1228. {
  1229. struct sock *sk = sock->sk;
  1230. DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
  1231. long timeout = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
  1232. struct tipc_sock *tsk = tipc_sk(sk);
  1233. struct tipc_msg *hdr = &tsk->phdr;
  1234. struct net *net = sock_net(sk);
  1235. struct sk_buff_head pkts;
  1236. u32 dnode = tsk_peer_node(tsk);
  1237. int send, sent = 0;
  1238. int rc = 0;
  1239. skb_queue_head_init(&pkts);
  1240. if (unlikely(dlen > INT_MAX))
  1241. return -EMSGSIZE;
  1242. /* Handle implicit connection setup */
  1243. if (unlikely(dest)) {
  1244. rc = __tipc_sendmsg(sock, m, dlen);
  1245. if (dlen && (dlen == rc))
  1246. tsk->snt_unacked = tsk_inc(tsk, dlen + msg_hdr_sz(hdr));
  1247. return rc;
  1248. }
  1249. do {
  1250. rc = tipc_wait_for_cond(sock, &timeout,
  1251. (!tsk->cong_link_cnt &&
  1252. !tsk_conn_cong(tsk) &&
  1253. tipc_sk_connected(sk)));
  1254. if (unlikely(rc))
  1255. break;
  1256. send = min_t(size_t, dlen - sent, TIPC_MAX_USER_MSG_SIZE);
  1257. rc = tipc_msg_build(hdr, m, sent, send, tsk->max_pkt, &pkts);
  1258. if (unlikely(rc != send))
  1259. break;
  1260. rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
  1261. if (unlikely(rc == -ELINKCONG)) {
  1262. tsk->cong_link_cnt = 1;
  1263. rc = 0;
  1264. }
  1265. if (likely(!rc)) {
  1266. tsk->snt_unacked += tsk_inc(tsk, send + MIN_H_SIZE);
  1267. sent += send;
  1268. }
  1269. } while (sent < dlen && !rc);
  1270. return sent ? sent : rc;
  1271. }
  1272. /**
  1273. * tipc_send_packet - send a connection-oriented message
  1274. * @sock: socket structure
  1275. * @m: message to send
  1276. * @dsz: length of data to be transmitted
  1277. *
  1278. * Used for SOCK_SEQPACKET messages.
  1279. *
  1280. * Returns the number of bytes sent on success, or errno otherwise
  1281. */
  1282. static int tipc_send_packet(struct socket *sock, struct msghdr *m, size_t dsz)
  1283. {
  1284. if (dsz > TIPC_MAX_USER_MSG_SIZE)
  1285. return -EMSGSIZE;
  1286. return tipc_sendstream(sock, m, dsz);
  1287. }
  1288. /* tipc_sk_finish_conn - complete the setup of a connection
  1289. */
  1290. static void tipc_sk_finish_conn(struct tipc_sock *tsk, u32 peer_port,
  1291. u32 peer_node)
  1292. {
  1293. struct sock *sk = &tsk->sk;
  1294. struct net *net = sock_net(sk);
  1295. struct tipc_msg *msg = &tsk->phdr;
  1296. msg_set_destnode(msg, peer_node);
  1297. msg_set_destport(msg, peer_port);
  1298. msg_set_type(msg, TIPC_CONN_MSG);
  1299. msg_set_lookup_scope(msg, 0);
  1300. msg_set_hdr_sz(msg, SHORT_H_SIZE);
  1301. sk_reset_timer(sk, &sk->sk_timer, jiffies + CONN_PROBING_INTV);
  1302. tipc_set_sk_state(sk, TIPC_ESTABLISHED);
  1303. tipc_node_add_conn(net, peer_node, tsk->portid, peer_port);
  1304. tsk->max_pkt = tipc_node_get_mtu(net, peer_node, tsk->portid);
  1305. tsk->peer_caps = tipc_node_get_capabilities(net, peer_node);
  1306. if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL)
  1307. return;
  1308. /* Fall back to message based flow control */
  1309. tsk->rcv_win = FLOWCTL_MSG_WIN;
  1310. tsk->snd_win = FLOWCTL_MSG_WIN;
  1311. }
  1312. /**
  1313. * tipc_sk_set_orig_addr - capture sender's address for received message
  1314. * @m: descriptor for message info
  1315. * @hdr: received message header
  1316. *
  1317. * Note: Address is not captured if not requested by receiver.
  1318. */
  1319. static void tipc_sk_set_orig_addr(struct msghdr *m, struct sk_buff *skb)
  1320. {
  1321. DECLARE_SOCKADDR(struct sockaddr_pair *, srcaddr, m->msg_name);
  1322. struct tipc_msg *hdr = buf_msg(skb);
  1323. if (!srcaddr)
  1324. return;
  1325. srcaddr->sock.family = AF_TIPC;
  1326. srcaddr->sock.addrtype = TIPC_ADDR_ID;
  1327. srcaddr->sock.addr.id.ref = msg_origport(hdr);
  1328. srcaddr->sock.addr.id.node = msg_orignode(hdr);
  1329. srcaddr->sock.addr.name.domain = 0;
  1330. srcaddr->sock.scope = 0;
  1331. m->msg_namelen = sizeof(struct sockaddr_tipc);
  1332. if (!msg_in_group(hdr))
  1333. return;
  1334. /* Group message users may also want to know sending member's id */
  1335. srcaddr->member.family = AF_TIPC;
  1336. srcaddr->member.addrtype = TIPC_ADDR_NAME;
  1337. srcaddr->member.addr.name.name.type = msg_nametype(hdr);
  1338. srcaddr->member.addr.name.name.instance = TIPC_SKB_CB(skb)->orig_member;
  1339. srcaddr->member.addr.name.domain = 0;
  1340. m->msg_namelen = sizeof(*srcaddr);
  1341. }
  1342. /**
  1343. * tipc_sk_anc_data_recv - optionally capture ancillary data for received message
  1344. * @m: descriptor for message info
  1345. * @msg: received message header
  1346. * @tsk: TIPC port associated with message
  1347. *
  1348. * Note: Ancillary data is not captured if not requested by receiver.
  1349. *
  1350. * Returns 0 if successful, otherwise errno
  1351. */
  1352. static int tipc_sk_anc_data_recv(struct msghdr *m, struct tipc_msg *msg,
  1353. struct tipc_sock *tsk)
  1354. {
  1355. u32 anc_data[3];
  1356. u32 err;
  1357. u32 dest_type;
  1358. int has_name;
  1359. int res;
  1360. if (likely(m->msg_controllen == 0))
  1361. return 0;
  1362. /* Optionally capture errored message object(s) */
  1363. err = msg ? msg_errcode(msg) : 0;
  1364. if (unlikely(err)) {
  1365. anc_data[0] = err;
  1366. anc_data[1] = msg_data_sz(msg);
  1367. res = put_cmsg(m, SOL_TIPC, TIPC_ERRINFO, 8, anc_data);
  1368. if (res)
  1369. return res;
  1370. if (anc_data[1]) {
  1371. res = put_cmsg(m, SOL_TIPC, TIPC_RETDATA, anc_data[1],
  1372. msg_data(msg));
  1373. if (res)
  1374. return res;
  1375. }
  1376. }
  1377. /* Optionally capture message destination object */
  1378. dest_type = msg ? msg_type(msg) : TIPC_DIRECT_MSG;
  1379. switch (dest_type) {
  1380. case TIPC_NAMED_MSG:
  1381. has_name = 1;
  1382. anc_data[0] = msg_nametype(msg);
  1383. anc_data[1] = msg_namelower(msg);
  1384. anc_data[2] = msg_namelower(msg);
  1385. break;
  1386. case TIPC_MCAST_MSG:
  1387. has_name = 1;
  1388. anc_data[0] = msg_nametype(msg);
  1389. anc_data[1] = msg_namelower(msg);
  1390. anc_data[2] = msg_nameupper(msg);
  1391. break;
  1392. case TIPC_CONN_MSG:
  1393. has_name = (tsk->conn_type != 0);
  1394. anc_data[0] = tsk->conn_type;
  1395. anc_data[1] = tsk->conn_instance;
  1396. anc_data[2] = tsk->conn_instance;
  1397. break;
  1398. default:
  1399. has_name = 0;
  1400. }
  1401. if (has_name) {
  1402. res = put_cmsg(m, SOL_TIPC, TIPC_DESTNAME, 12, anc_data);
  1403. if (res)
  1404. return res;
  1405. }
  1406. return 0;
  1407. }
  1408. static void tipc_sk_send_ack(struct tipc_sock *tsk)
  1409. {
  1410. struct sock *sk = &tsk->sk;
  1411. struct net *net = sock_net(sk);
  1412. struct sk_buff *skb = NULL;
  1413. struct tipc_msg *msg;
  1414. u32 peer_port = tsk_peer_port(tsk);
  1415. u32 dnode = tsk_peer_node(tsk);
  1416. if (!tipc_sk_connected(sk))
  1417. return;
  1418. skb = tipc_msg_create(CONN_MANAGER, CONN_ACK, INT_H_SIZE, 0,
  1419. dnode, tsk_own_node(tsk), peer_port,
  1420. tsk->portid, TIPC_OK);
  1421. if (!skb)
  1422. return;
  1423. msg = buf_msg(skb);
  1424. msg_set_conn_ack(msg, tsk->rcv_unacked);
  1425. tsk->rcv_unacked = 0;
  1426. /* Adjust to and advertize the correct window limit */
  1427. if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL) {
  1428. tsk->rcv_win = tsk_adv_blocks(tsk->sk.sk_rcvbuf);
  1429. msg_set_adv_win(msg, tsk->rcv_win);
  1430. }
  1431. tipc_node_xmit_skb(net, skb, dnode, msg_link_selector(msg));
  1432. }
  1433. static int tipc_wait_for_rcvmsg(struct socket *sock, long *timeop)
  1434. {
  1435. struct sock *sk = sock->sk;
  1436. DEFINE_WAIT(wait);
  1437. long timeo = *timeop;
  1438. int err = sock_error(sk);
  1439. if (err)
  1440. return err;
  1441. for (;;) {
  1442. prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
  1443. if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
  1444. if (sk->sk_shutdown & RCV_SHUTDOWN) {
  1445. err = -ENOTCONN;
  1446. break;
  1447. }
  1448. release_sock(sk);
  1449. timeo = schedule_timeout(timeo);
  1450. lock_sock(sk);
  1451. }
  1452. err = 0;
  1453. if (!skb_queue_empty(&sk->sk_receive_queue))
  1454. break;
  1455. err = -EAGAIN;
  1456. if (!timeo)
  1457. break;
  1458. err = sock_intr_errno(timeo);
  1459. if (signal_pending(current))
  1460. break;
  1461. err = sock_error(sk);
  1462. if (err)
  1463. break;
  1464. }
  1465. finish_wait(sk_sleep(sk), &wait);
  1466. *timeop = timeo;
  1467. return err;
  1468. }
  1469. /**
  1470. * tipc_recvmsg - receive packet-oriented message
  1471. * @m: descriptor for message info
  1472. * @buflen: length of user buffer area
  1473. * @flags: receive flags
  1474. *
  1475. * Used for SOCK_DGRAM, SOCK_RDM, and SOCK_SEQPACKET messages.
  1476. * If the complete message doesn't fit in user area, truncate it.
  1477. *
  1478. * Returns size of returned message data, errno otherwise
  1479. */
  1480. static int tipc_recvmsg(struct socket *sock, struct msghdr *m,
  1481. size_t buflen, int flags)
  1482. {
  1483. struct sock *sk = sock->sk;
  1484. bool connected = !tipc_sk_type_connectionless(sk);
  1485. struct tipc_sock *tsk = tipc_sk(sk);
  1486. int rc, err, hlen, dlen, copy;
  1487. struct sk_buff_head xmitq;
  1488. struct tipc_msg *hdr;
  1489. struct sk_buff *skb;
  1490. bool grp_evt;
  1491. long timeout;
  1492. /* Catch invalid receive requests */
  1493. if (unlikely(!buflen))
  1494. return -EINVAL;
  1495. lock_sock(sk);
  1496. if (unlikely(connected && sk->sk_state == TIPC_OPEN)) {
  1497. rc = -ENOTCONN;
  1498. goto exit;
  1499. }
  1500. timeout = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
  1501. /* Step rcv queue to first msg with data or error; wait if necessary */
  1502. do {
  1503. rc = tipc_wait_for_rcvmsg(sock, &timeout);
  1504. if (unlikely(rc))
  1505. goto exit;
  1506. skb = skb_peek(&sk->sk_receive_queue);
  1507. hdr = buf_msg(skb);
  1508. dlen = msg_data_sz(hdr);
  1509. hlen = msg_hdr_sz(hdr);
  1510. err = msg_errcode(hdr);
  1511. grp_evt = msg_is_grp_evt(hdr);
  1512. if (likely(dlen || err))
  1513. break;
  1514. tsk_advance_rx_queue(sk);
  1515. } while (1);
  1516. /* Collect msg meta data, including error code and rejected data */
  1517. tipc_sk_set_orig_addr(m, skb);
  1518. rc = tipc_sk_anc_data_recv(m, hdr, tsk);
  1519. if (unlikely(rc))
  1520. goto exit;
  1521. /* Capture data if non-error msg, otherwise just set return value */
  1522. if (likely(!err)) {
  1523. copy = min_t(int, dlen, buflen);
  1524. if (unlikely(copy != dlen))
  1525. m->msg_flags |= MSG_TRUNC;
  1526. rc = skb_copy_datagram_msg(skb, hlen, m, copy);
  1527. } else {
  1528. copy = 0;
  1529. rc = 0;
  1530. if (err != TIPC_CONN_SHUTDOWN && connected && !m->msg_control)
  1531. rc = -ECONNRESET;
  1532. }
  1533. if (unlikely(rc))
  1534. goto exit;
  1535. /* Mark message as group event if applicable */
  1536. if (unlikely(grp_evt)) {
  1537. if (msg_grp_evt(hdr) == TIPC_WITHDRAWN)
  1538. m->msg_flags |= MSG_EOR;
  1539. m->msg_flags |= MSG_OOB;
  1540. copy = 0;
  1541. }
  1542. /* Caption of data or error code/rejected data was successful */
  1543. if (unlikely(flags & MSG_PEEK))
  1544. goto exit;
  1545. /* Send group flow control advertisement when applicable */
  1546. if (tsk->group && msg_in_group(hdr) && !grp_evt) {
  1547. skb_queue_head_init(&xmitq);
  1548. tipc_group_update_rcv_win(tsk->group, tsk_blocks(hlen + dlen),
  1549. msg_orignode(hdr), msg_origport(hdr),
  1550. &xmitq);
  1551. tipc_node_distr_xmit(sock_net(sk), &xmitq);
  1552. }
  1553. tsk_advance_rx_queue(sk);
  1554. if (likely(!connected))
  1555. goto exit;
  1556. /* Send connection flow control advertisement when applicable */
  1557. tsk->rcv_unacked += tsk_inc(tsk, hlen + dlen);
  1558. if (tsk->rcv_unacked >= tsk->rcv_win / TIPC_ACK_RATE)
  1559. tipc_sk_send_ack(tsk);
  1560. exit:
  1561. release_sock(sk);
  1562. return rc ? rc : copy;
  1563. }
  1564. /**
  1565. * tipc_recvstream - receive stream-oriented data
  1566. * @m: descriptor for message info
  1567. * @buflen: total size of user buffer area
  1568. * @flags: receive flags
  1569. *
  1570. * Used for SOCK_STREAM messages only. If not enough data is available
  1571. * will optionally wait for more; never truncates data.
  1572. *
  1573. * Returns size of returned message data, errno otherwise
  1574. */
  1575. static int tipc_recvstream(struct socket *sock, struct msghdr *m,
  1576. siz

Large files files are truncated, but you can click here to view the full file