]> git.proxmox.com Git - mirror_ubuntu-jammy-kernel.git/blame - net/tipc/socket.c
Merge branch 's390-fixes'
[mirror_ubuntu-jammy-kernel.git] / net / tipc / socket.c
CommitLineData
b97bf3fd 1/*
02c00c2a 2 * net/tipc/socket.c: TIPC socket API
c4307285 3 *
75da2163 4 * Copyright (c) 2001-2007, 2012-2017, Ericsson AB
c5fa7b3c 5 * Copyright (c) 2004-2008, 2010-2013, Wind River Systems
b97bf3fd
PL
6 * All rights reserved.
7 *
9ea1fd3c 8 * Redistribution and use in source and binary forms, with or without
b97bf3fd
PL
9 * modification, are permitted provided that the following conditions are met:
10 *
9ea1fd3c
PL
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.
b97bf3fd 19 *
9ea1fd3c
PL
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
b97bf3fd
PL
34 * POSSIBILITY OF SUCH DAMAGE.
35 */
36
07f6c4bc 37#include <linux/rhashtable.h>
174cd4b1
IM
38#include <linux/sched/signal.h>
39
b97bf3fd 40#include "core.h"
e2dafe87 41#include "name_table.h"
78acb1f9 42#include "node.h"
e2dafe87 43#include "link.h"
c637c103 44#include "name_distr.h"
2e84c60b 45#include "socket.h"
a6bf70f7 46#include "bcast.h"
49cc66ea 47#include "netlink.h"
75da2163 48#include "group.h"
b4b9771b 49#include "trace.h"
2cf8aa19 50
67879274 51#define CONN_TIMEOUT_DEFAULT 8000 /* default connect timeout = 8s */
0d5fcebf 52#define CONN_PROBING_INTV msecs_to_jiffies(3600000) /* [ms] => 1 h */
07f6c4bc 53#define TIPC_FWD_MSG 1
07f6c4bc
YX
54#define TIPC_MAX_PORT 0xffffffff
55#define TIPC_MIN_PORT 1
e9f8b101 56#define TIPC_ACK_RATE 4 /* ACK at 1/4 of of rcv window size */
301bae56 57
0c288c86
PB
58enum {
59 TIPC_LISTEN = TCP_LISTEN,
8ea642ee 60 TIPC_ESTABLISHED = TCP_ESTABLISHED,
438adcaf 61 TIPC_OPEN = TCP_CLOSE,
9fd4b070 62 TIPC_DISCONNECTING = TCP_CLOSE_WAIT,
99a20889 63 TIPC_CONNECTING = TCP_SYN_SENT,
0c288c86
PB
64};
65
31c82a2d
JM
66struct sockaddr_pair {
67 struct sockaddr_tipc sock;
68 struct sockaddr_tipc member;
69};
70
301bae56
JPM
71/**
72 * struct tipc_sock - TIPC socket structure
73 * @sk: socket - interacts with 'port' and with user via the socket API
301bae56
JPM
74 * @conn_type: TIPC type used when connection was established
75 * @conn_instance: TIPC instance used when connection was established
76 * @published: non-zero if port has one or more associated names
77 * @max_pkt: maximum packet size "hint" used when building messages sent by port
c0bceb97 78 * @maxnagle: maximum size of msg which can be subject to nagle
07f6c4bc 79 * @portid: unique port identity in TIPC socket hash table
301bae56 80 * @phdr: preformatted message header used when sending messages
365ad353 81 * #cong_links: list of congested links
301bae56 82 * @publications: list of publications for port
365ad353 83 * @blocking_link: address of the congested link we are currently sleeping on
301bae56 84 * @pub_count: total # of publications port has made during its lifetime
301bae56
JPM
85 * @conn_timeout: the time we can wait for an unresponded setup request
86 * @dupl_rcvcnt: number of bytes counted twice, in both backlog and rcv queue
365ad353 87 * @cong_link_cnt: number of congested links
75da2163 88 * @snt_unacked: # messages sent by socket, and not yet acked by peer
301bae56 89 * @rcv_unacked: # messages read by user, but not yet acked back to peer
aeda16b6 90 * @peer: 'connected' peer for dgram/rdm
07f6c4bc 91 * @node: hash table node
01fd12bb 92 * @mc_method: cookie for use between socket and broadcast layer
07f6c4bc 93 * @rcu: rcu struct for tipc_sock
301bae56
JPM
94 */
95struct tipc_sock {
96 struct sock sk;
301bae56
JPM
97 u32 conn_type;
98 u32 conn_instance;
99 int published;
100 u32 max_pkt;
c0bceb97 101 u32 maxnagle;
07f6c4bc 102 u32 portid;
301bae56 103 struct tipc_msg phdr;
365ad353 104 struct list_head cong_links;
301bae56
JPM
105 struct list_head publications;
106 u32 pub_count;
301bae56 107 atomic_t dupl_rcvcnt;
67879274 108 u16 conn_timeout;
8ea642ee 109 bool probe_unacked;
365ad353 110 u16 cong_link_cnt;
10724cc7
JPM
111 u16 snt_unacked;
112 u16 snd_win;
60020e18 113 u16 peer_caps;
10724cc7
JPM
114 u16 rcv_unacked;
115 u16 rcv_win;
aeda16b6 116 struct sockaddr_tipc peer;
07f6c4bc 117 struct rhash_head node;
01fd12bb 118 struct tipc_mc_method mc_method;
07f6c4bc 119 struct rcu_head rcu;
75da2163 120 struct tipc_group *group;
c0bceb97
JM
121 u32 oneway;
122 u16 snd_backlog;
123 bool expect_ack;
124 bool nodelay;
60c25306 125 bool group_is_open;
301bae56 126};
b97bf3fd 127
64ac5f59 128static int tipc_sk_backlog_rcv(struct sock *sk, struct sk_buff *skb);
676d2369 129static void tipc_data_ready(struct sock *sk);
f288bef4 130static void tipc_write_space(struct sock *sk);
f4195d1e 131static void tipc_sock_destruct(struct sock *sk);
247f0f3c 132static int tipc_release(struct socket *sock);
cdfbabfb
DH
133static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags,
134 bool kern);
31b102bb 135static void tipc_sk_timeout(struct timer_list *t);
301bae56 136static int tipc_sk_publish(struct tipc_sock *tsk, uint scope,
0fc87aae 137 struct tipc_name_seq const *seq);
301bae56 138static int tipc_sk_withdraw(struct tipc_sock *tsk, uint scope,
0fc87aae 139 struct tipc_name_seq const *seq);
75da2163 140static int tipc_sk_leave(struct tipc_sock *tsk);
e05b31f4 141static struct tipc_sock *tipc_sk_lookup(struct net *net, u32 portid);
07f6c4bc
YX
142static int tipc_sk_insert(struct tipc_sock *tsk);
143static void tipc_sk_remove(struct tipc_sock *tsk);
365ad353 144static int __tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dsz);
39a0295f 145static int __tipc_sendmsg(struct socket *sock, struct msghdr *m, size_t dsz);
c0bceb97 146static void tipc_sk_push_backlog(struct tipc_sock *tsk);
b97bf3fd 147
bca65eae
FW
148static const struct proto_ops packet_ops;
149static const struct proto_ops stream_ops;
150static const struct proto_ops msg_ops;
b97bf3fd 151static struct proto tipc_proto;
6cca7289
HX
152static const struct rhashtable_params tsk_rht_params;
153
c5898636
JPM
154static u32 tsk_own_node(struct tipc_sock *tsk)
155{
156 return msg_prevnode(&tsk->phdr);
157}
158
301bae56 159static u32 tsk_peer_node(struct tipc_sock *tsk)
2e84c60b 160{
301bae56 161 return msg_destnode(&tsk->phdr);
2e84c60b
JPM
162}
163
301bae56 164static u32 tsk_peer_port(struct tipc_sock *tsk)
2e84c60b 165{
301bae56 166 return msg_destport(&tsk->phdr);
2e84c60b
JPM
167}
168
301bae56 169static bool tsk_unreliable(struct tipc_sock *tsk)
2e84c60b 170{
301bae56 171 return msg_src_droppable(&tsk->phdr) != 0;
2e84c60b
JPM
172}
173
301bae56 174static void tsk_set_unreliable(struct tipc_sock *tsk, bool unreliable)
2e84c60b 175{
301bae56 176 msg_set_src_droppable(&tsk->phdr, unreliable ? 1 : 0);
2e84c60b
JPM
177}
178
301bae56 179static bool tsk_unreturnable(struct tipc_sock *tsk)
2e84c60b 180{
301bae56 181 return msg_dest_droppable(&tsk->phdr) != 0;
2e84c60b
JPM
182}
183
301bae56 184static void tsk_set_unreturnable(struct tipc_sock *tsk, bool unreturnable)
2e84c60b 185{
301bae56 186 msg_set_dest_droppable(&tsk->phdr, unreturnable ? 1 : 0);
2e84c60b
JPM
187}
188
301bae56 189static int tsk_importance(struct tipc_sock *tsk)
2e84c60b 190{
301bae56 191 return msg_importance(&tsk->phdr);
2e84c60b
JPM
192}
193
301bae56 194static int tsk_set_importance(struct tipc_sock *tsk, int imp)
2e84c60b
JPM
195{
196 if (imp > TIPC_CRITICAL_IMPORTANCE)
197 return -EINVAL;
301bae56 198 msg_set_importance(&tsk->phdr, (u32)imp);
2e84c60b
JPM
199 return 0;
200}
8826cde6 201
301bae56
JPM
202static struct tipc_sock *tipc_sk(const struct sock *sk)
203{
204 return container_of(sk, struct tipc_sock, sk);
205}
206
10724cc7 207static bool tsk_conn_cong(struct tipc_sock *tsk)
301bae56 208{
6998cc6e 209 return tsk->snt_unacked > tsk->snd_win;
10724cc7
JPM
210}
211
b7d42635
JM
212static u16 tsk_blocks(int len)
213{
214 return ((len / FLOWCTL_BLK_SZ) + 1);
215}
216
10724cc7
JPM
217/* tsk_blocks(): translate a buffer size in bytes to number of
218 * advertisable blocks, taking into account the ratio truesize(len)/len
219 * We can trust that this ratio is always < 4 for len >= FLOWCTL_BLK_SZ
220 */
221static u16 tsk_adv_blocks(int len)
222{
223 return len / FLOWCTL_BLK_SZ / 4;
224}
225
226/* tsk_inc(): increment counter for sent or received data
227 * - If block based flow control is not supported by peer we
228 * fall back to message based ditto, incrementing the counter
229 */
230static u16 tsk_inc(struct tipc_sock *tsk, int msglen)
231{
232 if (likely(tsk->peer_caps & TIPC_BLOCK_FLOWCTL))
233 return ((msglen / FLOWCTL_BLK_SZ) + 1);
234 return 1;
301bae56
JPM
235}
236
c0bceb97
JM
237/* tsk_set_nagle - enable/disable nagle property by manipulating maxnagle
238 */
239static void tsk_set_nagle(struct tipc_sock *tsk)
240{
241 struct sock *sk = &tsk->sk;
242
243 tsk->maxnagle = 0;
244 if (sk->sk_type != SOCK_STREAM)
245 return;
246 if (tsk->nodelay)
247 return;
248 if (!(tsk->peer_caps & TIPC_NAGLE))
249 return;
250 /* Limit node local buffer size to avoid receive queue overflow */
251 if (tsk->max_pkt == MAX_MSG_SIZE)
252 tsk->maxnagle = 1500;
253 else
254 tsk->maxnagle = tsk->max_pkt;
255}
256
0c3141e9 257/**
2e84c60b 258 * tsk_advance_rx_queue - discard first buffer in socket receive queue
0c3141e9
AS
259 *
260 * Caller must hold socket lock
b97bf3fd 261 */
2e84c60b 262static void tsk_advance_rx_queue(struct sock *sk)
b97bf3fd 263{
01e661eb 264 trace_tipc_sk_advance_rx(sk, NULL, TIPC_DUMP_SK_RCVQ, " ");
5f6d9123 265 kfree_skb(__skb_dequeue(&sk->sk_receive_queue));
b97bf3fd
PL
266}
267
bcd3ffd4
JPM
268/* tipc_sk_respond() : send response message back to sender
269 */
270static void tipc_sk_respond(struct sock *sk, struct sk_buff *skb, int err)
271{
272 u32 selector;
273 u32 dnode;
274 u32 onode = tipc_own_addr(sock_net(sk));
275
276 if (!tipc_msg_reverse(onode, &skb, err))
277 return;
278
01e661eb 279 trace_tipc_sk_rej_msg(sk, skb, TIPC_DUMP_NONE, "@sk_respond!");
bcd3ffd4
JPM
280 dnode = msg_destnode(buf_msg(skb));
281 selector = msg_origport(buf_msg(skb));
282 tipc_node_xmit_skb(sock_net(sk), skb, dnode, selector);
283}
284
b97bf3fd 285/**
2e84c60b 286 * tsk_rej_rx_queue - reject all buffers in socket receive queue
0c3141e9
AS
287 *
288 * Caller must hold socket lock
b97bf3fd 289 */
49afb806 290static void tsk_rej_rx_queue(struct sock *sk, int error)
b97bf3fd 291{
a6ca1094 292 struct sk_buff *skb;
0c3141e9 293
bcd3ffd4 294 while ((skb = __skb_dequeue(&sk->sk_receive_queue)))
49afb806 295 tipc_sk_respond(sk, skb, error);
b97bf3fd
PL
296}
297
d6fb7e9c
PB
298static bool tipc_sk_connected(struct sock *sk)
299{
f40acbaf 300 return sk->sk_state == TIPC_ESTABLISHED;
d6fb7e9c
PB
301}
302
c752023a
PB
303/* tipc_sk_type_connectionless - check if the socket is datagram socket
304 * @sk: socket
305 *
306 * Returns true if connection less, false otherwise
307 */
308static bool tipc_sk_type_connectionless(struct sock *sk)
309{
310 return sk->sk_type == SOCK_RDM || sk->sk_type == SOCK_DGRAM;
311}
312
2e84c60b 313/* tsk_peer_msg - verify if message was sent by connected port's peer
0fc87aae
JPM
314 *
315 * Handles cases where the node's network address has changed from
316 * the default of <0.0.0> to its configured setting.
317 */
2e84c60b 318static bool tsk_peer_msg(struct tipc_sock *tsk, struct tipc_msg *msg)
0fc87aae 319{
d6fb7e9c 320 struct sock *sk = &tsk->sk;
23fd3eac 321 u32 self = tipc_own_addr(sock_net(sk));
301bae56 322 u32 peer_port = tsk_peer_port(tsk);
23fd3eac 323 u32 orig_node, peer_node;
0fc87aae 324
d6fb7e9c 325 if (unlikely(!tipc_sk_connected(sk)))
0fc87aae
JPM
326 return false;
327
328 if (unlikely(msg_origport(msg) != peer_port))
329 return false;
330
331 orig_node = msg_orignode(msg);
301bae56 332 peer_node = tsk_peer_node(tsk);
0fc87aae
JPM
333
334 if (likely(orig_node == peer_node))
335 return true;
336
23fd3eac 337 if (!orig_node && peer_node == self)
0fc87aae
JPM
338 return true;
339
23fd3eac 340 if (!peer_node && orig_node == self)
0fc87aae
JPM
341 return true;
342
343 return false;
344}
345
0c288c86
PB
346/* tipc_set_sk_state - set the sk_state of the socket
347 * @sk: socket
348 *
349 * Caller must hold socket lock
350 *
351 * Returns 0 on success, errno otherwise
352 */
353static int tipc_set_sk_state(struct sock *sk, int state)
354{
438adcaf 355 int oldsk_state = sk->sk_state;
0c288c86
PB
356 int res = -EINVAL;
357
358 switch (state) {
438adcaf
PB
359 case TIPC_OPEN:
360 res = 0;
361 break;
0c288c86 362 case TIPC_LISTEN:
99a20889 363 case TIPC_CONNECTING:
438adcaf 364 if (oldsk_state == TIPC_OPEN)
0c288c86
PB
365 res = 0;
366 break;
8ea642ee 367 case TIPC_ESTABLISHED:
99a20889 368 if (oldsk_state == TIPC_CONNECTING ||
438adcaf 369 oldsk_state == TIPC_OPEN)
8ea642ee
PB
370 res = 0;
371 break;
9fd4b070 372 case TIPC_DISCONNECTING:
99a20889 373 if (oldsk_state == TIPC_CONNECTING ||
9fd4b070
PB
374 oldsk_state == TIPC_ESTABLISHED)
375 res = 0;
376 break;
0c288c86
PB
377 }
378
379 if (!res)
380 sk->sk_state = state;
381
382 return res;
383}
384
8c44e1af
JPM
385static int tipc_sk_sock_err(struct socket *sock, long *timeout)
386{
387 struct sock *sk = sock->sk;
388 int err = sock_error(sk);
389 int typ = sock->type;
390
391 if (err)
392 return err;
393 if (typ == SOCK_STREAM || typ == SOCK_SEQPACKET) {
394 if (sk->sk_state == TIPC_DISCONNECTING)
395 return -EPIPE;
396 else if (!tipc_sk_connected(sk))
397 return -ENOTCONN;
398 }
399 if (!*timeout)
400 return -EAGAIN;
401 if (signal_pending(current))
402 return sock_intr_errno(*timeout);
403
404 return 0;
405}
406
844cf763
JPM
407#define tipc_wait_for_cond(sock_, timeo_, condition_) \
408({ \
bfd07f3d 409 DEFINE_WAIT_FUNC(wait_, woken_wake_function); \
844cf763
JPM
410 struct sock *sk_; \
411 int rc_; \
412 \
413 while ((rc_ = !(condition_))) { \
bfd07f3d
TN
414 /* coupled with smp_wmb() in tipc_sk_proto_rcv() */ \
415 smp_rmb(); \
844cf763
JPM
416 sk_ = (sock_)->sk; \
417 rc_ = tipc_sk_sock_err((sock_), timeo_); \
418 if (rc_) \
419 break; \
223b7329 420 add_wait_queue(sk_sleep(sk_), &wait_); \
844cf763
JPM
421 release_sock(sk_); \
422 *(timeo_) = wait_woken(&wait_, TASK_INTERRUPTIBLE, *(timeo_)); \
423 sched_annotate_sleep(); \
424 lock_sock(sk_); \
425 remove_wait_queue(sk_sleep(sk_), &wait_); \
426 } \
427 rc_; \
8c44e1af
JPM
428})
429
b97bf3fd 430/**
c5fa7b3c 431 * tipc_sk_create - create a TIPC socket
0c3141e9 432 * @net: network namespace (must be default network)
b97bf3fd
PL
433 * @sock: pre-allocated socket structure
434 * @protocol: protocol indicator (must be 0)
3f378b68 435 * @kern: caused by kernel or by userspace?
c4307285 436 *
0c3141e9
AS
437 * This routine creates additional data structures used by the TIPC socket,
438 * initializes them, and links them together.
b97bf3fd
PL
439 *
440 * Returns 0 on success, errno otherwise
441 */
58ed9442
JPM
442static int tipc_sk_create(struct net *net, struct socket *sock,
443 int protocol, int kern)
b97bf3fd 444{
0c3141e9 445 const struct proto_ops *ops;
b97bf3fd 446 struct sock *sk;
58ed9442 447 struct tipc_sock *tsk;
5b8fa7ce 448 struct tipc_msg *msg;
0c3141e9
AS
449
450 /* Validate arguments */
b97bf3fd
PL
451 if (unlikely(protocol != 0))
452 return -EPROTONOSUPPORT;
453
b97bf3fd
PL
454 switch (sock->type) {
455 case SOCK_STREAM:
0c3141e9 456 ops = &stream_ops;
b97bf3fd
PL
457 break;
458 case SOCK_SEQPACKET:
0c3141e9 459 ops = &packet_ops;
b97bf3fd
PL
460 break;
461 case SOCK_DGRAM:
b97bf3fd 462 case SOCK_RDM:
0c3141e9 463 ops = &msg_ops;
b97bf3fd 464 break;
49978651 465 default:
49978651 466 return -EPROTOTYPE;
b97bf3fd
PL
467 }
468
0c3141e9 469 /* Allocate socket's protocol area */
11aa9c28 470 sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto, kern);
0c3141e9 471 if (sk == NULL)
b97bf3fd 472 return -ENOMEM;
b97bf3fd 473
58ed9442 474 tsk = tipc_sk(sk);
301bae56 475 tsk->max_pkt = MAX_PKT_DEFAULT;
c0bceb97 476 tsk->maxnagle = 0;
301bae56 477 INIT_LIST_HEAD(&tsk->publications);
365ad353 478 INIT_LIST_HEAD(&tsk->cong_links);
301bae56 479 msg = &tsk->phdr;
b97bf3fd 480
0c3141e9 481 /* Finish initializing socket data structures */
0c3141e9 482 sock->ops = ops;
0c3141e9 483 sock_init_data(sock, sk);
438adcaf 484 tipc_set_sk_state(sk, TIPC_OPEN);
07f6c4bc 485 if (tipc_sk_insert(tsk)) {
c19ca6cb 486 pr_warn("Socket create failed; port number exhausted\n");
07f6c4bc
YX
487 return -EINVAL;
488 }
40f9f439
HX
489
490 /* Ensure tsk is visible before we read own_addr. */
491 smp_mb();
492
23fd3eac
JM
493 tipc_msg_init(tipc_own_addr(net), msg, TIPC_LOW_IMPORTANCE,
494 TIPC_NAMED_MSG, NAMED_H_SIZE, 0);
40f9f439 495
07f6c4bc 496 msg_set_origport(msg, tsk->portid);
31b102bb 497 timer_setup(&sk->sk_timer, tipc_sk_timeout, 0);
6f00089c 498 sk->sk_shutdown = 0;
64ac5f59 499 sk->sk_backlog_rcv = tipc_sk_backlog_rcv;
cc79dd1b 500 sk->sk_rcvbuf = sysctl_tipc_rmem[1];
f288bef4
YX
501 sk->sk_data_ready = tipc_data_ready;
502 sk->sk_write_space = tipc_write_space;
f4195d1e 503 sk->sk_destruct = tipc_sock_destruct;
4f4482dc 504 tsk->conn_timeout = CONN_TIMEOUT_DEFAULT;
1b22bcad 505 tsk->group_is_open = true;
4f4482dc 506 atomic_set(&tsk->dupl_rcvcnt, 0);
7ef43eba 507
10724cc7
JPM
508 /* Start out with safe limits until we receive an advertised window */
509 tsk->snd_win = tsk_adv_blocks(RCVBUF_MIN);
510 tsk->rcv_win = tsk->snd_win;
511
c752023a 512 if (tipc_sk_type_connectionless(sk)) {
301bae56 513 tsk_set_unreturnable(tsk, true);
0c3141e9 514 if (sock->type == SOCK_DGRAM)
301bae56 515 tsk_set_unreliable(tsk, true);
0c3141e9 516 }
2948a1fc 517 __skb_queue_head_init(&tsk->mc_method.deferredq);
01e661eb 518 trace_tipc_sk_create(sk, NULL, TIPC_DUMP_NONE, " ");
b97bf3fd
PL
519 return 0;
520}
521
07f6c4bc
YX
522static void tipc_sk_callback(struct rcu_head *head)
523{
524 struct tipc_sock *tsk = container_of(head, struct tipc_sock, rcu);
525
526 sock_put(&tsk->sk);
527}
528
6f00089c
PB
529/* Caller should hold socket lock for the socket. */
530static void __tipc_shutdown(struct socket *sock, int error)
531{
532 struct sock *sk = sock->sk;
533 struct tipc_sock *tsk = tipc_sk(sk);
534 struct net *net = sock_net(sk);
12db3c80 535 long timeout = msecs_to_jiffies(CONN_TIMEOUT_DEFAULT);
6f00089c
PB
536 u32 dnode = tsk_peer_node(tsk);
537 struct sk_buff *skb;
538
365ad353
JPM
539 /* Avoid that hi-prio shutdown msgs bypass msgs in link wakeup queue */
540 tipc_wait_for_cond(sock, &timeout, (!tsk->cong_link_cnt &&
541 !tsk_conn_cong(tsk)));
542
d34910e1
TN
543 /* Push out delayed messages if in Nagle mode */
544 tipc_sk_push_backlog(tsk);
545 /* Remove pending SYN */
546 __skb_queue_purge(&sk->sk_write_queue);
67879274 547
49afb806
TL
548 /* Remove partially received buffer if any */
549 skb = skb_peek(&sk->sk_receive_queue);
550 if (skb && TIPC_SKB_CB(skb)->bytes_read) {
551 __skb_unlink(skb, &sk->sk_receive_queue);
552 kfree_skb(skb);
6f00089c 553 }
693c5649 554
49afb806
TL
555 /* Reject all unreceived messages if connectionless */
556 if (tipc_sk_type_connectionless(sk)) {
557 tsk_rej_rx_queue(sk, error);
693c5649 558 return;
49afb806 559 }
693c5649 560
49afb806
TL
561 switch (sk->sk_state) {
562 case TIPC_CONNECTING:
563 case TIPC_ESTABLISHED:
564 tipc_set_sk_state(sk, TIPC_DISCONNECTING);
565 tipc_node_remove_conn(net, dnode, tsk->portid);
566 /* Send a FIN+/- to its peer */
567 skb = __skb_dequeue(&sk->sk_receive_queue);
568 if (skb) {
569 __skb_queue_purge(&sk->sk_receive_queue);
570 tipc_sk_respond(sk, skb, error);
571 break;
572 }
6f00089c
PB
573 skb = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE,
574 TIPC_CONN_MSG, SHORT_H_SIZE, 0, dnode,
575 tsk_own_node(tsk), tsk_peer_port(tsk),
576 tsk->portid, error);
577 if (skb)
578 tipc_node_xmit_skb(net, skb, dnode, tsk->portid);
49afb806
TL
579 break;
580 case TIPC_LISTEN:
581 /* Reject all SYN messages */
582 tsk_rej_rx_queue(sk, error);
583 break;
584 default:
585 __skb_queue_purge(&sk->sk_receive_queue);
586 break;
6f00089c
PB
587 }
588}
589
b97bf3fd 590/**
247f0f3c 591 * tipc_release - destroy a TIPC socket
b97bf3fd
PL
592 * @sock: socket to destroy
593 *
594 * This routine cleans up any messages that are still queued on the socket.
595 * For DGRAM and RDM socket types, all queued messages are rejected.
596 * For SEQPACKET and STREAM socket types, the first message is rejected
597 * and any others are discarded. (If the first message on a STREAM socket
598 * is partially-read, it is discarded and the next one is rejected instead.)
c4307285 599 *
b97bf3fd
PL
600 * NOTE: Rejected messages are not necessarily returned to the sender! They
601 * are returned or discarded according to the "destination droppable" setting
602 * specified for the message by the sender.
603 *
604 * Returns 0 on success, errno otherwise
605 */
247f0f3c 606static int tipc_release(struct socket *sock)
b97bf3fd 607{
b97bf3fd 608 struct sock *sk = sock->sk;
58ed9442 609 struct tipc_sock *tsk;
b97bf3fd 610
0c3141e9
AS
611 /*
612 * Exit if socket isn't fully initialized (occurs when a failed accept()
613 * releases a pre-allocated child socket that was never used)
614 */
0c3141e9 615 if (sk == NULL)
b97bf3fd 616 return 0;
c4307285 617
58ed9442 618 tsk = tipc_sk(sk);
0c3141e9
AS
619 lock_sock(sk);
620
01e661eb 621 trace_tipc_sk_release(sk, NULL, TIPC_DUMP_ALL, " ");
6f00089c
PB
622 __tipc_shutdown(sock, TIPC_ERR_NO_PORT);
623 sk->sk_shutdown = SHUTDOWN_MASK;
75da2163 624 tipc_sk_leave(tsk);
301bae56 625 tipc_sk_withdraw(tsk, 0, NULL);
c55c8eda 626 __skb_queue_purge(&tsk->mc_method.deferredq);
1ea23a21 627 sk_stop_timer(sk, &sk->sk_timer);
07f6c4bc 628 tipc_sk_remove(tsk);
b97bf3fd 629
0a3b8b2b 630 sock_orphan(sk);
0c3141e9 631 /* Reject any messages that accumulated in backlog queue */
0c3141e9 632 release_sock(sk);
a80ae530 633 tipc_dest_list_purge(&tsk->cong_links);
365ad353 634 tsk->cong_link_cnt = 0;
07f6c4bc 635 call_rcu(&tsk->rcu, tipc_sk_callback);
0c3141e9 636 sock->sk = NULL;
b97bf3fd 637
065d7e39 638 return 0;
b97bf3fd
PL
639}
640
641/**
247f0f3c 642 * tipc_bind - associate or disassocate TIPC name(s) with a socket
b97bf3fd
PL
643 * @sock: socket structure
644 * @uaddr: socket address describing name(s) and desired operation
645 * @uaddr_len: size of socket address data structure
c4307285 646 *
b97bf3fd
PL
647 * Name and name sequence binding is indicated using a positive scope value;
648 * a negative scope value unbinds the specified name. Specifying no name
649 * (i.e. a socket address length of 0) unbinds all names from the socket.
c4307285 650 *
b97bf3fd 651 * Returns 0 on success, errno otherwise
0c3141e9
AS
652 *
653 * NOTE: This routine doesn't need to take the socket lock since it doesn't
654 * access any non-constant socket information.
b97bf3fd 655 */
247f0f3c
YX
656static int tipc_bind(struct socket *sock, struct sockaddr *uaddr,
657 int uaddr_len)
b97bf3fd 658{
84602761 659 struct sock *sk = sock->sk;
b97bf3fd 660 struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
58ed9442 661 struct tipc_sock *tsk = tipc_sk(sk);
84602761 662 int res = -EINVAL;
b97bf3fd 663
84602761
YX
664 lock_sock(sk);
665 if (unlikely(!uaddr_len)) {
301bae56 666 res = tipc_sk_withdraw(tsk, 0, NULL);
84602761
YX
667 goto exit;
668 }
75da2163
JM
669 if (tsk->group) {
670 res = -EACCES;
671 goto exit;
672 }
84602761
YX
673 if (uaddr_len < sizeof(struct sockaddr_tipc)) {
674 res = -EINVAL;
675 goto exit;
676 }
677 if (addr->family != AF_TIPC) {
678 res = -EAFNOSUPPORT;
679 goto exit;
680 }
b97bf3fd 681
b97bf3fd
PL
682 if (addr->addrtype == TIPC_ADDR_NAME)
683 addr->addr.nameseq.upper = addr->addr.nameseq.lower;
84602761
YX
684 else if (addr->addrtype != TIPC_ADDR_NAMESEQ) {
685 res = -EAFNOSUPPORT;
686 goto exit;
687 }
c4307285 688
13a2e898 689 if ((addr->addr.nameseq.type < TIPC_RESERVED_TYPES) &&
7d0ab17b 690 (addr->addr.nameseq.type != TIPC_TOP_SRV) &&
84602761
YX
691 (addr->addr.nameseq.type != TIPC_CFG_SRV)) {
692 res = -EACCES;
693 goto exit;
694 }
c422f1bd 695
928df188 696 res = (addr->scope >= 0) ?
301bae56
JPM
697 tipc_sk_publish(tsk, addr->scope, &addr->addr.nameseq) :
698 tipc_sk_withdraw(tsk, -addr->scope, &addr->addr.nameseq);
84602761
YX
699exit:
700 release_sock(sk);
701 return res;
b97bf3fd
PL
702}
703
c4307285 704/**
247f0f3c 705 * tipc_getname - get port ID of socket or peer socket
b97bf3fd
PL
706 * @sock: socket structure
707 * @uaddr: area for returned socket address
708 * @uaddr_len: area for returned length of socket address
2da59918 709 * @peer: 0 = own ID, 1 = current peer ID, 2 = current/former peer ID
c4307285 710 *
b97bf3fd 711 * Returns 0 on success, errno otherwise
0c3141e9 712 *
2da59918
AS
713 * NOTE: This routine doesn't need to take the socket lock since it only
714 * accesses socket information that is unchanging (or which changes in
0e65967e 715 * a completely predictable manner).
b97bf3fd 716 */
247f0f3c 717static int tipc_getname(struct socket *sock, struct sockaddr *uaddr,
9b2c45d4 718 int peer)
b97bf3fd 719{
b97bf3fd 720 struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
9fd4b070
PB
721 struct sock *sk = sock->sk;
722 struct tipc_sock *tsk = tipc_sk(sk);
b97bf3fd 723
88f8a5e3 724 memset(addr, 0, sizeof(*addr));
0c3141e9 725 if (peer) {
f40acbaf 726 if ((!tipc_sk_connected(sk)) &&
9fd4b070 727 ((peer != 2) || (sk->sk_state != TIPC_DISCONNECTING)))
2da59918 728 return -ENOTCONN;
301bae56
JPM
729 addr->addr.id.ref = tsk_peer_port(tsk);
730 addr->addr.id.node = tsk_peer_node(tsk);
0c3141e9 731 } else {
07f6c4bc 732 addr->addr.id.ref = tsk->portid;
23fd3eac 733 addr->addr.id.node = tipc_own_addr(sock_net(sk));
0c3141e9 734 }
b97bf3fd 735
b97bf3fd
PL
736 addr->addrtype = TIPC_ADDR_ID;
737 addr->family = AF_TIPC;
738 addr->scope = 0;
b97bf3fd
PL
739 addr->addr.name.domain = 0;
740
9b2c45d4 741 return sizeof(*addr);
b97bf3fd
PL
742}
743
744/**
a11e1d43 745 * tipc_poll - read and possibly block on pollmask
b97bf3fd
PL
746 * @file: file structure associated with the socket
747 * @sock: socket for which to calculate the poll bits
a11e1d43 748 * @wait: ???
b97bf3fd 749 *
9b674e82
AS
750 * Returns pollmask value
751 *
752 * COMMENTARY:
753 * It appears that the usual socket locking mechanisms are not useful here
754 * since the pollmask info is potentially out-of-date the moment this routine
755 * exits. TCP and other protocols seem to rely on higher level poll routines
756 * to handle any preventable race conditions, so TIPC will do the same ...
757 *
f662c070
AS
758 * IMPORTANT: The fact that a read or write operation is indicated does NOT
759 * imply that the operation will succeed, merely that it should be performed
760 * and will not block.
b97bf3fd 761 */
a11e1d43
LT
762static __poll_t tipc_poll(struct file *file, struct socket *sock,
763 poll_table *wait)
b97bf3fd 764{
9b674e82 765 struct sock *sk = sock->sk;
58ed9442 766 struct tipc_sock *tsk = tipc_sk(sk);
ade994f4 767 __poll_t revents = 0;
9b674e82 768
89ab066d 769 sock_poll_wait(file, sock, wait);
01e661eb 770 trace_tipc_sk_poll(sk, NULL, TIPC_DUMP_ALL, " ");
a11e1d43 771
6f00089c 772 if (sk->sk_shutdown & RCV_SHUTDOWN)
a9a08845 773 revents |= EPOLLRDHUP | EPOLLIN | EPOLLRDNORM;
6f00089c 774 if (sk->sk_shutdown == SHUTDOWN_MASK)
a9a08845 775 revents |= EPOLLHUP;
6f00089c 776
f40acbaf
PB
777 switch (sk->sk_state) {
778 case TIPC_ESTABLISHED:
365ad353 779 if (!tsk->cong_link_cnt && !tsk_conn_cong(tsk))
a9a08845 780 revents |= EPOLLOUT;
f79e3365 781 /* fall through */
f40acbaf 782 case TIPC_LISTEN:
ff946833 783 case TIPC_CONNECTING:
3ef7cf57 784 if (!skb_queue_empty_lockless(&sk->sk_receive_queue))
a9a08845 785 revents |= EPOLLIN | EPOLLRDNORM;
f40acbaf
PB
786 break;
787 case TIPC_OPEN:
60c25306 788 if (tsk->group_is_open && !tsk->cong_link_cnt)
a9a08845 789 revents |= EPOLLOUT;
ae236fb2
JM
790 if (!tipc_sk_type_connectionless(sk))
791 break;
3ef7cf57 792 if (skb_queue_empty_lockless(&sk->sk_receive_queue))
ae236fb2 793 break;
a9a08845 794 revents |= EPOLLIN | EPOLLRDNORM;
f40acbaf
PB
795 break;
796 case TIPC_DISCONNECTING:
a9a08845 797 revents = EPOLLIN | EPOLLRDNORM | EPOLLHUP;
f40acbaf 798 break;
f662c070 799 }
ae236fb2 800 return revents;
b97bf3fd
PL
801}
802
0abd8ff2
JPM
803/**
804 * tipc_sendmcast - send multicast message
805 * @sock: socket structure
806 * @seq: destination address
562640f3 807 * @msg: message to send
365ad353
JPM
808 * @dlen: length of data to send
809 * @timeout: timeout to wait for wakeup
0abd8ff2
JPM
810 *
811 * Called from function tipc_sendmsg(), which has done all sanity checks
812 * Returns the number of bytes sent on success, or errno
813 */
814static int tipc_sendmcast(struct socket *sock, struct tipc_name_seq *seq,
365ad353 815 struct msghdr *msg, size_t dlen, long timeout)
0abd8ff2
JPM
816{
817 struct sock *sk = sock->sk;
c5898636 818 struct tipc_sock *tsk = tipc_sk(sk);
365ad353 819 struct tipc_msg *hdr = &tsk->phdr;
f2f9800d 820 struct net *net = sock_net(sk);
365ad353 821 int mtu = tipc_bcast_get_mtu(net);
01fd12bb 822 struct tipc_mc_method *method = &tsk->mc_method;
365ad353 823 struct sk_buff_head pkts;
a853e4c6 824 struct tipc_nlist dsts;
0abd8ff2
JPM
825 int rc;
826
75da2163
JM
827 if (tsk->group)
828 return -EACCES;
829
a853e4c6 830 /* Block or return if any destination link is congested */
365ad353
JPM
831 rc = tipc_wait_for_cond(sock, &timeout, !tsk->cong_link_cnt);
832 if (unlikely(rc))
833 return rc;
f214fc40 834
a853e4c6
JPM
835 /* Lookup destination nodes */
836 tipc_nlist_init(&dsts, tipc_own_addr(net));
837 tipc_nametbl_lookup_dst_nodes(net, seq->type, seq->lower,
e9a03445 838 seq->upper, &dsts);
a853e4c6
JPM
839 if (!dsts.local && !dsts.remote)
840 return -EHOSTUNREACH;
841
842 /* Build message header */
365ad353 843 msg_set_type(hdr, TIPC_MCAST_MSG);
a853e4c6 844 msg_set_hdr_sz(hdr, MCAST_H_SIZE);
365ad353
JPM
845 msg_set_lookup_scope(hdr, TIPC_CLUSTER_SCOPE);
846 msg_set_destport(hdr, 0);
847 msg_set_destnode(hdr, 0);
848 msg_set_nametype(hdr, seq->type);
849 msg_set_namelower(hdr, seq->lower);
850 msg_set_nameupper(hdr, seq->upper);
365ad353 851
a853e4c6 852 /* Build message as chain of buffers */
e654f9f5 853 __skb_queue_head_init(&pkts);
365ad353 854 rc = tipc_msg_build(hdr, msg, 0, dlen, mtu, &pkts);
0abd8ff2 855
a853e4c6 856 /* Send message if build was successful */
01e661eb
TL
857 if (unlikely(rc == dlen)) {
858 trace_tipc_sk_sendmcast(sk, skb_peek(&pkts),
859 TIPC_DUMP_SK_SNDQ, " ");
01fd12bb 860 rc = tipc_mcast_xmit(net, &pkts, method, &dsts,
a853e4c6 861 &tsk->cong_link_cnt);
01e661eb 862 }
a853e4c6
JPM
863
864 tipc_nlist_purge(&dsts);
365ad353
JPM
865
866 return rc ? rc : dlen;
0abd8ff2
JPM
867}
868
27bd9ec0
JM
869/**
870 * tipc_send_group_msg - send a message to a member in the group
871 * @net: network namespace
872 * @m: message to send
873 * @mb: group member
874 * @dnode: destination node
875 * @dport: destination port
876 * @dlen: total length of message data
877 */
878static int tipc_send_group_msg(struct net *net, struct tipc_sock *tsk,
879 struct msghdr *m, struct tipc_member *mb,
880 u32 dnode, u32 dport, int dlen)
881{
b87a5ea3 882 u16 bc_snd_nxt = tipc_group_bc_snd_nxt(tsk->group);
2f487712 883 struct tipc_mc_method *method = &tsk->mc_method;
27bd9ec0
JM
884 int blks = tsk_blocks(GROUP_H_SIZE + dlen);
885 struct tipc_msg *hdr = &tsk->phdr;
886 struct sk_buff_head pkts;
887 int mtu, rc;
888
889 /* Complete message header */
890 msg_set_type(hdr, TIPC_GRP_UCAST_MSG);
891 msg_set_hdr_sz(hdr, GROUP_H_SIZE);
892 msg_set_destport(hdr, dport);
893 msg_set_destnode(hdr, dnode);
b87a5ea3 894 msg_set_grp_bc_seqno(hdr, bc_snd_nxt);
27bd9ec0
JM
895
896 /* Build message as chain of buffers */
e654f9f5 897 __skb_queue_head_init(&pkts);
f73b1281 898 mtu = tipc_node_get_mtu(net, dnode, tsk->portid, false);
27bd9ec0
JM
899 rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
900 if (unlikely(rc != dlen))
901 return rc;
902
903 /* Send message */
904 rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
905 if (unlikely(rc == -ELINKCONG)) {
906 tipc_dest_push(&tsk->cong_links, dnode, 0);
907 tsk->cong_link_cnt++;
908 }
909
2f487712 910 /* Update send window */
27bd9ec0
JM
911 tipc_group_update_member(mb, blks);
912
2f487712
JM
913 /* A broadcast sent within next EXPIRE period must follow same path */
914 method->rcast = true;
915 method->mandatory = true;
27bd9ec0
JM
916 return dlen;
917}
918
919/**
920 * tipc_send_group_unicast - send message to a member in the group
921 * @sock: socket structure
922 * @m: message to send
923 * @dlen: total length of message data
924 * @timeout: timeout to wait for wakeup
925 *
926 * Called from function tipc_sendmsg(), which has done all sanity checks
927 * Returns the number of bytes sent on success, or errno
928 */
929static int tipc_send_group_unicast(struct socket *sock, struct msghdr *m,
930 int dlen, long timeout)
931{
932 struct sock *sk = sock->sk;
933 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
934 int blks = tsk_blocks(GROUP_H_SIZE + dlen);
935 struct tipc_sock *tsk = tipc_sk(sk);
27bd9ec0
JM
936 struct net *net = sock_net(sk);
937 struct tipc_member *mb = NULL;
938 u32 node, port;
939 int rc;
940
941 node = dest->addr.id.node;
942 port = dest->addr.id.ref;
943 if (!port && !node)
944 return -EHOSTUNREACH;
945
946 /* Block or return if destination link or member is congested */
947 rc = tipc_wait_for_cond(sock, &timeout,
948 !tipc_dest_find(&tsk->cong_links, node, 0) &&
143ece65
CW
949 tsk->group &&
950 !tipc_group_cong(tsk->group, node, port, blks,
951 &mb));
27bd9ec0
JM
952 if (unlikely(rc))
953 return rc;
954
955 if (unlikely(!mb))
956 return -EHOSTUNREACH;
957
958 rc = tipc_send_group_msg(net, tsk, m, mb, node, port, dlen);
959
960 return rc ? rc : dlen;
961}
962
ee106d7f
JM
963/**
964 * tipc_send_group_anycast - send message to any member with given identity
965 * @sock: socket structure
966 * @m: message to send
967 * @dlen: total length of message data
968 * @timeout: timeout to wait for wakeup
969 *
970 * Called from function tipc_sendmsg(), which has done all sanity checks
971 * Returns the number of bytes sent on success, or errno
972 */
973static int tipc_send_group_anycast(struct socket *sock, struct msghdr *m,
974 int dlen, long timeout)
975{
976 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
977 struct sock *sk = sock->sk;
978 struct tipc_sock *tsk = tipc_sk(sk);
979 struct list_head *cong_links = &tsk->cong_links;
980 int blks = tsk_blocks(GROUP_H_SIZE + dlen);
232d07b7 981 struct tipc_msg *hdr = &tsk->phdr;
ee106d7f
JM
982 struct tipc_member *first = NULL;
983 struct tipc_member *mbr = NULL;
984 struct net *net = sock_net(sk);
985 u32 node, port, exclude;
ee106d7f 986 struct list_head dsts;
232d07b7 987 u32 type, inst, scope;
ee106d7f
JM
988 int lookups = 0;
989 int dstcnt, rc;
990 bool cong;
991
992 INIT_LIST_HEAD(&dsts);
993
232d07b7 994 type = msg_nametype(hdr);
ee106d7f 995 inst = dest->addr.name.name.instance;
232d07b7 996 scope = msg_lookup_scope(hdr);
ee106d7f
JM
997
998 while (++lookups < 4) {
143ece65
CW
999 exclude = tipc_group_exclude(tsk->group);
1000
ee106d7f
JM
1001 first = NULL;
1002
1003 /* Look for a non-congested destination member, if any */
1004 while (1) {
232d07b7 1005 if (!tipc_nametbl_lookup(net, type, inst, scope, &dsts,
ee106d7f
JM
1006 &dstcnt, exclude, false))
1007 return -EHOSTUNREACH;
1008 tipc_dest_pop(&dsts, &node, &port);
143ece65
CW
1009 cong = tipc_group_cong(tsk->group, node, port, blks,
1010 &mbr);
ee106d7f
JM
1011 if (!cong)
1012 break;
1013 if (mbr == first)
1014 break;
1015 if (!first)
1016 first = mbr;
1017 }
1018
1019 /* Start over if destination was not in member list */
1020 if (unlikely(!mbr))
1021 continue;
1022
1023 if (likely(!cong && !tipc_dest_find(cong_links, node, 0)))
1024 break;
1025
1026 /* Block or return if destination link or member is congested */
1027 rc = tipc_wait_for_cond(sock, &timeout,
1028 !tipc_dest_find(cong_links, node, 0) &&
143ece65
CW
1029 tsk->group &&
1030 !tipc_group_cong(tsk->group, node, port,
ee106d7f
JM
1031 blks, &mbr));
1032 if (unlikely(rc))
1033 return rc;
1034
1035 /* Send, unless destination disappeared while waiting */
1036 if (likely(mbr))
1037 break;
1038 }
1039
1040 if (unlikely(lookups >= 4))
1041 return -EHOSTUNREACH;
1042
1043 rc = tipc_send_group_msg(net, tsk, m, mbr, node, port, dlen);
1044
1045 return rc ? rc : dlen;
1046}
1047
75da2163
JM
1048/**
1049 * tipc_send_group_bcast - send message to all members in communication group
1050 * @sk: socket structure
1051 * @m: message to send
1052 * @dlen: total length of message data
1053 * @timeout: timeout to wait for wakeup
1054 *
1055 * Called from function tipc_sendmsg(), which has done all sanity checks
1056 * Returns the number of bytes sent on success, or errno
1057 */
1058static int tipc_send_group_bcast(struct socket *sock, struct msghdr *m,
1059 int dlen, long timeout)
1060{
5b8dddb6 1061 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
75da2163
JM
1062 struct sock *sk = sock->sk;
1063 struct net *net = sock_net(sk);
1064 struct tipc_sock *tsk = tipc_sk(sk);
3c6306d4 1065 struct tipc_nlist *dsts;
75da2163 1066 struct tipc_mc_method *method = &tsk->mc_method;
2f487712 1067 bool ack = method->mandatory && method->rcast;
b7d42635 1068 int blks = tsk_blocks(MCAST_H_SIZE + dlen);
75da2163
JM
1069 struct tipc_msg *hdr = &tsk->phdr;
1070 int mtu = tipc_bcast_get_mtu(net);
1071 struct sk_buff_head pkts;
1072 int rc = -EHOSTUNREACH;
1073
b7d42635 1074 /* Block or return if any destination link or member is congested */
143ece65
CW
1075 rc = tipc_wait_for_cond(sock, &timeout,
1076 !tsk->cong_link_cnt && tsk->group &&
1077 !tipc_group_bc_cong(tsk->group, blks));
75da2163
JM
1078 if (unlikely(rc))
1079 return rc;
1080
3c6306d4
CW
1081 dsts = tipc_group_dests(tsk->group);
1082 if (!dsts->local && !dsts->remote)
1083 return -EHOSTUNREACH;
1084
75da2163 1085 /* Complete message header */
5b8dddb6
JM
1086 if (dest) {
1087 msg_set_type(hdr, TIPC_GRP_MCAST_MSG);
1088 msg_set_nameinst(hdr, dest->addr.name.name.instance);
1089 } else {
1090 msg_set_type(hdr, TIPC_GRP_BCAST_MSG);
1091 msg_set_nameinst(hdr, 0);
1092 }
b7d42635 1093 msg_set_hdr_sz(hdr, GROUP_H_SIZE);
75da2163
JM
1094 msg_set_destport(hdr, 0);
1095 msg_set_destnode(hdr, 0);
143ece65 1096 msg_set_grp_bc_seqno(hdr, tipc_group_bc_snd_nxt(tsk->group));
75da2163 1097
2f487712
JM
1098 /* Avoid getting stuck with repeated forced replicasts */
1099 msg_set_grp_bc_ack_req(hdr, ack);
1100
75da2163 1101 /* Build message as chain of buffers */
e654f9f5 1102 __skb_queue_head_init(&pkts);
75da2163
JM
1103 rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
1104 if (unlikely(rc != dlen))
1105 return rc;
1106
1107 /* Send message */
2f487712 1108 rc = tipc_mcast_xmit(net, &pkts, method, dsts, &tsk->cong_link_cnt);
75da2163
JM
1109 if (unlikely(rc))
1110 return rc;
1111
b7d42635 1112 /* Update broadcast sequence number and send windows */
2f487712
JM
1113 tipc_group_update_bc_members(tsk->group, blks, ack);
1114
1115 /* Broadcast link is now free to choose method for next broadcast */
1116 method->mandatory = false;
1117 method->expires = jiffies;
1118
75da2163
JM
1119 return dlen;
1120}
1121
5b8dddb6
JM
1122/**
1123 * tipc_send_group_mcast - send message to all members with given identity
1124 * @sock: socket structure
1125 * @m: message to send
1126 * @dlen: total length of message data
1127 * @timeout: timeout to wait for wakeup
1128 *
1129 * Called from function tipc_sendmsg(), which has done all sanity checks
1130 * Returns the number of bytes sent on success, or errno
1131 */
1132static int tipc_send_group_mcast(struct socket *sock, struct msghdr *m,
1133 int dlen, long timeout)
1134{
1135 struct sock *sk = sock->sk;
1136 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
5b8dddb6
JM
1137 struct tipc_sock *tsk = tipc_sk(sk);
1138 struct tipc_group *grp = tsk->group;
232d07b7 1139 struct tipc_msg *hdr = &tsk->phdr;
5b8dddb6 1140 struct net *net = sock_net(sk);
232d07b7 1141 u32 type, inst, scope, exclude;
5b8dddb6 1142 struct list_head dsts;
232d07b7 1143 u32 dstcnt;
5b8dddb6
JM
1144
1145 INIT_LIST_HEAD(&dsts);
1146
232d07b7
JM
1147 type = msg_nametype(hdr);
1148 inst = dest->addr.name.name.instance;
1149 scope = msg_lookup_scope(hdr);
5b8dddb6 1150 exclude = tipc_group_exclude(grp);
232d07b7
JM
1151
1152 if (!tipc_nametbl_lookup(net, type, inst, scope, &dsts,
1153 &dstcnt, exclude, true))
5b8dddb6
JM
1154 return -EHOSTUNREACH;
1155
1156 if (dstcnt == 1) {
1157 tipc_dest_pop(&dsts, &dest->addr.id.node, &dest->addr.id.ref);
1158 return tipc_send_group_unicast(sock, m, dlen, timeout);
1159 }
1160
1161 tipc_dest_list_purge(&dsts);
1162 return tipc_send_group_bcast(sock, m, dlen, timeout);
1163}
1164
cb1b7280
JPM
1165/**
1166 * tipc_sk_mcast_rcv - Deliver multicast messages to all destination sockets
1167 * @arrvq: queue with arriving messages, to be cloned after destination lookup
1168 * @inputq: queue with cloned messages, delivered to socket after dest lookup
1169 *
1170 * Multi-threaded: parallel calls with reference to same queues may occur
078bec82 1171 */
cb1b7280
JPM
1172void tipc_sk_mcast_rcv(struct net *net, struct sk_buff_head *arrvq,
1173 struct sk_buff_head *inputq)
078bec82 1174{
75da2163 1175 u32 self = tipc_own_addr(net);
232d07b7 1176 u32 type, lower, upper, scope;
cb1b7280 1177 struct sk_buff *skb, *_skb;
b053fcc4 1178 u32 portid, onode;
232d07b7 1179 struct sk_buff_head tmpq;
75da2163 1180 struct list_head dports;
232d07b7
JM
1181 struct tipc_msg *hdr;
1182 int user, mtyp, hlen;
1183 bool exact;
3c724acd 1184
cb1b7280 1185 __skb_queue_head_init(&tmpq);
4d8642d8 1186 INIT_LIST_HEAD(&dports);
078bec82 1187
cb1b7280
JPM
1188 skb = tipc_skb_peek(arrvq, &inputq->lock);
1189 for (; skb; skb = tipc_skb_peek(arrvq, &inputq->lock)) {
232d07b7
JM
1190 hdr = buf_msg(skb);
1191 user = msg_user(hdr);
1192 mtyp = msg_type(hdr);
1193 hlen = skb_headroom(skb) + msg_hdr_sz(hdr);
232d07b7
JM
1194 onode = msg_orignode(hdr);
1195 type = msg_nametype(hdr);
1196
2f487712
JM
1197 if (mtyp == TIPC_GRP_UCAST_MSG || user == GROUP_PROTOCOL) {
1198 spin_lock_bh(&inputq->lock);
1199 if (skb_peek(arrvq) == skb) {
1200 __skb_dequeue(arrvq);
1201 __skb_queue_tail(inputq, skb);
1202 }
c545a945 1203 kfree_skb(skb);
2f487712
JM
1204 spin_unlock_bh(&inputq->lock);
1205 continue;
1206 }
232d07b7
JM
1207
1208 /* Group messages require exact scope match */
1209 if (msg_in_group(hdr)) {
1210 lower = 0;
1211 upper = ~0;
1212 scope = msg_lookup_scope(hdr);
1213 exact = true;
1214 } else {
1215 /* TIPC_NODE_SCOPE means "any scope" in this context */
1216 if (onode == self)
1217 scope = TIPC_NODE_SCOPE;
1218 else
1219 scope = TIPC_CLUSTER_SCOPE;
1220 exact = false;
1221 lower = msg_namelower(hdr);
1222 upper = msg_nameupper(hdr);
75da2163 1223 }
232d07b7
JM
1224
1225 /* Create destination port list: */
1226 tipc_nametbl_mc_lookup(net, type, lower, upper,
1227 scope, exact, &dports);
1228
1229 /* Clone message per destination */
a80ae530 1230 while (tipc_dest_pop(&dports, NULL, &portid)) {
232d07b7 1231 _skb = __pskb_copy(skb, hlen, GFP_ATOMIC);
cb1b7280
JPM
1232 if (_skb) {
1233 msg_set_destport(buf_msg(_skb), portid);
1234 __skb_queue_tail(&tmpq, _skb);
1235 continue;
1236 }
1237 pr_warn("Failed to clone mcast rcv buffer\n");
078bec82 1238 }
cb1b7280
JPM
1239 /* Append to inputq if not already done by other thread */
1240 spin_lock_bh(&inputq->lock);
1241 if (skb_peek(arrvq) == skb) {
1242 skb_queue_splice_tail_init(&tmpq, inputq);
1243 kfree_skb(__skb_dequeue(arrvq));
1244 }
1245 spin_unlock_bh(&inputq->lock);
1246 __skb_queue_purge(&tmpq);
1247 kfree_skb(skb);
078bec82 1248 }
cb1b7280 1249 tipc_sk_rcv(net, inputq);
078bec82
JPM
1250}
1251
c0bceb97
JM
1252/* tipc_sk_push_backlog(): send accumulated buffers in socket write queue
1253 * when socket is in Nagle mode
1254 */
1255static void tipc_sk_push_backlog(struct tipc_sock *tsk)
1256{
1257 struct sk_buff_head *txq = &tsk->sk.sk_write_queue;
1258 struct net *net = sock_net(&tsk->sk);
1259 u32 dnode = tsk_peer_node(tsk);
d34910e1 1260 struct sk_buff *skb = skb_peek(txq);
c0bceb97
JM
1261 int rc;
1262
d34910e1
TN
1263 if (!skb || tsk->cong_link_cnt)
1264 return;
1265
1266 /* Do not send SYN again after congestion */
1267 if (msg_is_syn(buf_msg(skb)))
c0bceb97
JM
1268 return;
1269
1270 tsk->snt_unacked += tsk->snd_backlog;
1271 tsk->snd_backlog = 0;
1272 tsk->expect_ack = true;
1273 rc = tipc_node_xmit(net, txq, dnode, tsk->portid);
1274 if (rc == -ELINKCONG)
1275 tsk->cong_link_cnt = 1;
1276}
1277
ac0074ee 1278/**
64ac5f59 1279 * tipc_sk_conn_proto_rcv - receive a connection mng protocol message
ac0074ee 1280 * @tsk: receiving socket
bcd3ffd4 1281 * @skb: pointer to message buffer.
ac0074ee 1282 */
64ac5f59 1283static void tipc_sk_conn_proto_rcv(struct tipc_sock *tsk, struct sk_buff *skb,
e7eb0582 1284 struct sk_buff_head *inputq,
64ac5f59 1285 struct sk_buff_head *xmitq)
ac0074ee 1286{
bcd3ffd4 1287 struct tipc_msg *hdr = buf_msg(skb);
64ac5f59
JM
1288 u32 onode = tsk_own_node(tsk);
1289 struct sock *sk = &tsk->sk;
bcd3ffd4 1290 int mtyp = msg_type(hdr);
c0bceb97 1291 bool was_cong;
bcd3ffd4 1292
ac0074ee 1293 /* Ignore if connection cannot be validated: */
01e661eb
TL
1294 if (!tsk_peer_msg(tsk, hdr)) {
1295 trace_tipc_sk_drop_msg(sk, skb, TIPC_DUMP_NONE, "@proto_rcv!");
ac0074ee 1296 goto exit;
01e661eb 1297 }
ac0074ee 1298
c1be7756
PB
1299 if (unlikely(msg_errcode(hdr))) {
1300 tipc_set_sk_state(sk, TIPC_DISCONNECTING);
1301 tipc_node_remove_conn(sock_net(sk), tsk_peer_node(tsk),
1302 tsk_peer_port(tsk));
1303 sk->sk_state_change(sk);
e7eb0582
PB
1304
1305 /* State change is ignored if socket already awake,
1306 * - convert msg to abort msg and add to inqueue
1307 */
1308 msg_set_user(hdr, TIPC_CRITICAL_IMPORTANCE);
1309 msg_set_type(hdr, TIPC_CONN_MSG);
1310 msg_set_size(hdr, BASIC_H_SIZE);
1311 msg_set_hdr_sz(hdr, BASIC_H_SIZE);
1312 __skb_queue_tail(inputq, skb);
1313 return;
c1be7756
PB
1314 }
1315
8ea642ee 1316 tsk->probe_unacked = false;
ac0074ee 1317
bcd3ffd4
JPM
1318 if (mtyp == CONN_PROBE) {
1319 msg_set_type(hdr, CONN_PROBE_REPLY);
f1d048f2
JPM
1320 if (tipc_msg_reverse(onode, &skb, TIPC_OK))
1321 __skb_queue_tail(xmitq, skb);
bcd3ffd4
JPM
1322 return;
1323 } else if (mtyp == CONN_ACK) {
c0bceb97
JM
1324 was_cong = tsk_conn_cong(tsk);
1325 tsk->expect_ack = false;
1326 tipc_sk_push_backlog(tsk);
10724cc7
JPM
1327 tsk->snt_unacked -= msg_conn_ack(hdr);
1328 if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL)
1329 tsk->snd_win = msg_adv_win(hdr);
c0bceb97 1330 if (was_cong && !tsk_conn_cong(tsk))
bcd3ffd4
JPM
1331 sk->sk_write_space(sk);
1332 } else if (mtyp != CONN_PROBE_REPLY) {
1333 pr_warn("Received unknown CONN_PROTO msg\n");
ac0074ee 1334 }
ac0074ee 1335exit:
bcd3ffd4 1336 kfree_skb(skb);
ac0074ee
JPM
1337}
1338
b97bf3fd 1339/**
247f0f3c 1340 * tipc_sendmsg - send message in connectionless manner
b97bf3fd
PL
1341 * @sock: socket structure
1342 * @m: message to send
e2dafe87 1343 * @dsz: amount of user data to be sent
c4307285 1344 *
b97bf3fd 1345 * Message must have an destination specified explicitly.
c4307285 1346 * Used for SOCK_RDM and SOCK_DGRAM messages,
b97bf3fd
PL
1347 * and for 'SYN' messages on SOCK_SEQPACKET and SOCK_STREAM connections.
1348 * (Note: 'SYN+' is prohibited on SOCK_STREAM.)
c4307285 1349 *
b97bf3fd
PL
1350 * Returns the number of bytes sent on success, or errno otherwise
1351 */
1b784140 1352static int tipc_sendmsg(struct socket *sock,
e2dafe87 1353 struct msghdr *m, size_t dsz)
39a0295f
YX
1354{
1355 struct sock *sk = sock->sk;
1356 int ret;
1357
1358 lock_sock(sk);
1359 ret = __tipc_sendmsg(sock, m, dsz);
1360 release_sock(sk);
1361
1362 return ret;
1363}
1364
365ad353 1365static int __tipc_sendmsg(struct socket *sock, struct msghdr *m, size_t dlen)
b97bf3fd 1366{
0c3141e9 1367 struct sock *sk = sock->sk;
f2f9800d 1368 struct net *net = sock_net(sk);
365ad353
JPM
1369 struct tipc_sock *tsk = tipc_sk(sk);
1370 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
1371 long timeout = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
1372 struct list_head *clinks = &tsk->cong_links;
1373 bool syn = !tipc_sk_type_connectionless(sk);
75da2163 1374 struct tipc_group *grp = tsk->group;
365ad353 1375 struct tipc_msg *hdr = &tsk->phdr;
f2f8036e 1376 struct tipc_name_seq *seq;
365ad353 1377 struct sk_buff_head pkts;
abc9b4e0
TL
1378 u32 dport = 0, dnode = 0;
1379 u32 type = 0, inst = 0;
365ad353 1380 int mtu, rc;
b97bf3fd 1381
365ad353 1382 if (unlikely(dlen > TIPC_MAX_USER_MSG_SIZE))
c29c3f70 1383 return -EMSGSIZE;
365ad353 1384
27bd9ec0
JM
1385 if (likely(dest)) {
1386 if (unlikely(m->msg_namelen < sizeof(*dest)))
1387 return -EINVAL;
1388 if (unlikely(dest->family != AF_TIPC))
1389 return -EINVAL;
1390 }
1391
1392 if (grp) {
1393 if (!dest)
1394 return tipc_send_group_bcast(sock, m, dlen, timeout);
ee106d7f
JM
1395 if (dest->addrtype == TIPC_ADDR_NAME)
1396 return tipc_send_group_anycast(sock, m, dlen, timeout);
27bd9ec0
JM
1397 if (dest->addrtype == TIPC_ADDR_ID)
1398 return tipc_send_group_unicast(sock, m, dlen, timeout);
5b8dddb6
JM
1399 if (dest->addrtype == TIPC_ADDR_MCAST)
1400 return tipc_send_group_mcast(sock, m, dlen, timeout);
27bd9ec0
JM
1401 return -EINVAL;
1402 }
75da2163 1403
f2f8036e 1404 if (unlikely(!dest)) {
365ad353 1405 dest = &tsk->peer;
0e632089 1406 if (!syn && dest->family != AF_TIPC)
f2f8036e 1407 return -EDESTADDRREQ;
f2f8036e 1408 }
365ad353 1409
365ad353 1410 if (unlikely(syn)) {
0c288c86 1411 if (sk->sk_state == TIPC_LISTEN)
39a0295f 1412 return -EPIPE;
438adcaf 1413 if (sk->sk_state != TIPC_OPEN)
39a0295f
YX
1414 return -EISCONN;
1415 if (tsk->published)
1416 return -EOPNOTSUPP;
3388007b 1417 if (dest->addrtype == TIPC_ADDR_NAME) {
301bae56
JPM
1418 tsk->conn_type = dest->addr.name.name.type;
1419 tsk->conn_instance = dest->addr.name.name.instance;
3388007b 1420 }
25b9221b 1421 msg_set_syn(hdr, 1);
b97bf3fd 1422 }
e2dafe87 1423
365ad353
JPM
1424 seq = &dest->addr.nameseq;
1425 if (dest->addrtype == TIPC_ADDR_MCAST)
1426 return tipc_sendmcast(sock, seq, m, dlen, timeout);
e2dafe87 1427
365ad353
JPM
1428 if (dest->addrtype == TIPC_ADDR_NAME) {
1429 type = dest->addr.name.name.type;
1430 inst = dest->addr.name.name.instance;
928df188 1431 dnode = dest->addr.name.domain;
4ac1c8d0 1432 dport = tipc_nametbl_translate(net, type, inst, &dnode);
39a0295f
YX
1433 if (unlikely(!dport && !dnode))
1434 return -EHOSTUNREACH;
e2dafe87
JPM
1435 } else if (dest->addrtype == TIPC_ADDR_ID) {
1436 dnode = dest->addr.id.node;
335b929b
JM
1437 } else {
1438 return -EINVAL;
e2dafe87
JPM
1439 }
1440
365ad353 1441 /* Block or return if destination link is congested */
a80ae530
JM
1442 rc = tipc_wait_for_cond(sock, &timeout,
1443 !tipc_dest_find(clinks, dnode, 0));
365ad353
JPM
1444 if (unlikely(rc))
1445 return rc;
1446
abc9b4e0
TL
1447 if (dest->addrtype == TIPC_ADDR_NAME) {
1448 msg_set_type(hdr, TIPC_NAMED_MSG);
1449 msg_set_hdr_sz(hdr, NAMED_H_SIZE);
1450 msg_set_nametype(hdr, type);
1451 msg_set_nameinst(hdr, inst);
1452 msg_set_lookup_scope(hdr, tipc_node2scope(dnode));
1453 msg_set_destnode(hdr, dnode);
1454 msg_set_destport(hdr, dport);
1455 } else { /* TIPC_ADDR_ID */
1456 msg_set_type(hdr, TIPC_DIRECT_MSG);
1457 msg_set_lookup_scope(hdr, 0);
1458 msg_set_destnode(hdr, dnode);
1459 msg_set_destport(hdr, dest->addr.id.ref);
1460 msg_set_hdr_sz(hdr, BASIC_H_SIZE);
1461 }
1462
e654f9f5 1463 __skb_queue_head_init(&pkts);
8b1e5b0a 1464 mtu = tipc_node_get_mtu(net, dnode, tsk->portid, true);
365ad353
JPM
1465 rc = tipc_msg_build(hdr, m, 0, dlen, mtu, &pkts);
1466 if (unlikely(rc != dlen))
39a0295f 1467 return rc;
2fe97a57
TN
1468 if (unlikely(syn && !tipc_msg_skb_clone(&pkts, &sk->sk_write_queue))) {
1469 __skb_queue_purge(&pkts);
67879274 1470 return -ENOMEM;
2fe97a57 1471 }
e2dafe87 1472
01e661eb 1473 trace_tipc_sk_sendmsg(sk, skb_peek(&pkts), TIPC_DUMP_SK_SNDQ, " ");
365ad353
JPM
1474 rc = tipc_node_xmit(net, &pkts, dnode, tsk->portid);
1475 if (unlikely(rc == -ELINKCONG)) {
a80ae530 1476 tipc_dest_push(clinks, dnode, 0);
365ad353
JPM
1477 tsk->cong_link_cnt++;
1478 rc = 0;
1479 }
e2dafe87 1480
365ad353
JPM
1481 if (unlikely(syn && !rc))
1482 tipc_set_sk_state(sk, TIPC_CONNECTING);
1483
1484 return rc ? rc : dlen;
b97bf3fd
PL
1485}
1486
c4307285 1487/**
365ad353 1488 * tipc_sendstream - send stream-oriented data
b97bf3fd 1489 * @sock: socket structure
4ccfe5e0
JPM
1490 * @m: data to send
1491 * @dsz: total length of data to be transmitted
c4307285 1492 *
4ccfe5e0 1493 * Used for SOCK_STREAM data.
c4307285 1494 *
4ccfe5e0
JPM
1495 * Returns the number of bytes sent on success (or partial success),
1496 * or errno if no data sent
b97bf3fd 1497 */
365ad353 1498static int tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dsz)
39a0295f
YX
1499{
1500 struct sock *sk = sock->sk;
1501 int ret;
1502
1503 lock_sock(sk);
365ad353 1504 ret = __tipc_sendstream(sock, m, dsz);
39a0295f
YX
1505 release_sock(sk);
1506
1507 return ret;
1508}
1509
365ad353 1510static int __tipc_sendstream(struct socket *sock, struct msghdr *m, size_t dlen)
b97bf3fd 1511{
0c3141e9 1512 struct sock *sk = sock->sk;
342dfc30 1513 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
365ad353 1514 long timeout = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
c0bceb97 1515 struct sk_buff_head *txq = &sk->sk_write_queue;
365ad353
JPM
1516 struct tipc_sock *tsk = tipc_sk(sk);
1517 struct tipc_msg *hdr = &tsk->phdr;
1518 struct net *net = sock_net(sk);
365ad353 1519 u32 dnode = tsk_peer_node(tsk);
c0bceb97
JM
1520 int maxnagle = tsk->maxnagle;
1521 int maxpkt = tsk->max_pkt;
365ad353 1522 int send, sent = 0;
c0bceb97 1523 int blocks, rc = 0;
7cf87fa2 1524
365ad353
JPM
1525 if (unlikely(dlen > INT_MAX))
1526 return -EMSGSIZE;
4ccfe5e0 1527
365ad353
JPM
1528 /* Handle implicit connection setup */
1529 if (unlikely(dest)) {
1530 rc = __tipc_sendmsg(sock, m, dlen);
92ef12b3
PB
1531 if (dlen && dlen == rc) {
1532 tsk->peer_caps = tipc_node_get_capabilities(net, dnode);
365ad353 1533 tsk->snt_unacked = tsk_inc(tsk, dlen + msg_hdr_sz(hdr));
92ef12b3 1534 }
39a0295f 1535 return rc;
365ad353 1536 }
f214fc40 1537
c4307285 1538 do {
365ad353
JPM
1539 rc = tipc_wait_for_cond(sock, &timeout,
1540 (!tsk->cong_link_cnt &&
8c44e1af
JPM
1541 !tsk_conn_cong(tsk) &&
1542 tipc_sk_connected(sk)));
365ad353
JPM
1543 if (unlikely(rc))
1544 break;
365ad353 1545 send = min_t(size_t, dlen - sent, TIPC_MAX_USER_MSG_SIZE);
c0bceb97
JM
1546 blocks = tsk->snd_backlog;
1547 if (tsk->oneway++ >= 4 && send <= maxnagle) {
1548 rc = tipc_msg_append(hdr, m, send, maxnagle, txq);
1549 if (unlikely(rc < 0))
1550 break;
1551 blocks += rc;
1552 if (blocks <= 64 && tsk->expect_ack) {
1553 tsk->snd_backlog = blocks;
1554 sent += send;
1555 break;
1556 }
1557 tsk->expect_ack = true;
1558 } else {
1559 rc = tipc_msg_build(hdr, m, sent, send, maxpkt, txq);
1560 if (unlikely(rc != send))
1561 break;
1562 blocks += tsk_inc(tsk, send + MIN_H_SIZE);
1563 }
1564 trace_tipc_sk_sendstream(sk, skb_peek(txq),
01e661eb 1565 TIPC_DUMP_SK_SNDQ, " ");
c0bceb97 1566 rc = tipc_node_xmit(net, txq, dnode, tsk->portid);
365ad353
JPM
1567 if (unlikely(rc == -ELINKCONG)) {
1568 tsk->cong_link_cnt = 1;
1569 rc = 0;
1570 }
1571 if (likely(!rc)) {
c0bceb97
JM
1572 tsk->snt_unacked += blocks;
1573 tsk->snd_backlog = 0;
365ad353
JPM
1574 sent += send;
1575 }
1576 } while (sent < dlen && !rc);
39a0295f 1577
3364d61c 1578 return sent ? sent : rc;
b97bf3fd
PL
1579}
1580
c4307285 1581/**
4ccfe5e0 1582 * tipc_send_packet - send a connection-oriented message
b97bf3fd 1583 * @sock: socket structure
4ccfe5e0
JPM
1584 * @m: message to send
1585 * @dsz: length of data to be transmitted
c4307285 1586 *
4ccfe5e0 1587 * Used for SOCK_SEQPACKET messages.
c4307285 1588 *
4ccfe5e0 1589 * Returns the number of bytes sent on success, or errno otherwise
b97bf3fd 1590 */
1b784140 1591static int tipc_send_packet(struct socket *sock, struct msghdr *m, size_t dsz)
b97bf3fd 1592{
4ccfe5e0
JPM
1593 if (dsz > TIPC_MAX_USER_MSG_SIZE)
1594 return -EMSGSIZE;
b97bf3fd 1595
365ad353 1596 return tipc_sendstream(sock, m, dsz);
b97bf3fd
PL
1597}
1598
dadebc00 1599/* tipc_sk_finish_conn - complete the setup of a connection
b97bf3fd 1600 */
301bae56 1601static void tipc_sk_finish_conn(struct tipc_sock *tsk, u32 peer_port,
dadebc00 1602 u32 peer_node)
b97bf3fd 1603{
3721e9c7
YX
1604 struct sock *sk = &tsk->sk;
1605 struct net *net = sock_net(sk);
301bae56 1606 struct tipc_msg *msg = &tsk->phdr;
b97bf3fd 1607
25b9221b 1608 msg_set_syn(msg, 0);
dadebc00
JPM
1609 msg_set_destnode(msg, peer_node);
1610 msg_set_destport(msg, peer_port);
1611 msg_set_type(msg, TIPC_CONN_MSG);
1612 msg_set_lookup_scope(msg, 0);
1613 msg_set_hdr_sz(msg, SHORT_H_SIZE);
584d24b3 1614
0d5fcebf 1615 sk_reset_timer(sk, &sk->sk_timer, jiffies + CONN_PROBING_INTV);
8ea642ee 1616 tipc_set_sk_state(sk, TIPC_ESTABLISHED);
f2f9800d 1617 tipc_node_add_conn(net, peer_node, tsk->portid, peer_port);
f73b1281 1618 tsk->max_pkt = tipc_node_get_mtu(net, peer_node, tsk->portid, true);
60020e18 1619 tsk->peer_caps = tipc_node_get_capabilities(net, peer_node);
c0bceb97 1620 tsk_set_nagle(tsk);
67879274 1621 __skb_queue_purge(&sk->sk_write_queue);
10724cc7
JPM
1622 if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL)
1623 return;
1624
1625 /* Fall back to message based flow control */
1626 tsk->rcv_win = FLOWCTL_MSG_WIN;
1627 tsk->snd_win = FLOWCTL_MSG_WIN;
b97bf3fd
PL
1628}
1629
1630/**
31c82a2d 1631 * tipc_sk_set_orig_addr - capture sender's address for received message
b97bf3fd 1632 * @m: descriptor for message info
31c82a2d 1633 * @hdr: received message header
c4307285 1634 *
b97bf3fd
PL
1635 * Note: Address is not captured if not requested by receiver.
1636 */
31c82a2d 1637static void tipc_sk_set_orig_addr(struct msghdr *m, struct sk_buff *skb)
b97bf3fd 1638{
31c82a2d
JM
1639 DECLARE_SOCKADDR(struct sockaddr_pair *, srcaddr, m->msg_name);
1640 struct tipc_msg *hdr = buf_msg(skb);
1641
1642 if (!srcaddr)
1643 return;
1644
1645 srcaddr->sock.family = AF_TIPC;
1646 srcaddr->sock.addrtype = TIPC_ADDR_ID;
09c8b971 1647 srcaddr->sock.scope = 0;
31c82a2d
JM
1648 srcaddr->sock.addr.id.ref = msg_origport(hdr);
1649 srcaddr->sock.addr.id.node = msg_orignode(hdr);
1650 srcaddr->sock.addr.name.domain = 0;
31c82a2d
JM
1651 m->msg_namelen = sizeof(struct sockaddr_tipc);
1652
1653 if (!msg_in_group(hdr))
1654 return;
1655
1656 /* Group message users may also want to know sending member's id */
1657 srcaddr->member.family = AF_TIPC;
1658 srcaddr->member.addrtype = TIPC_ADDR_NAME;
09c8b971 1659 srcaddr->member.scope = 0;
31c82a2d
JM
1660 srcaddr->member.addr.name.name.type = msg_nametype(hdr);
1661 srcaddr->member.addr.name.name.instance = TIPC_SKB_CB(skb)->orig_member;
1662 srcaddr->member.addr.name.domain = 0;
1663 m->msg_namelen = sizeof(*srcaddr);
b97bf3fd
PL
1664}
1665
1666/**
301bae56 1667 * tipc_sk_anc_data_recv - optionally capture ancillary data for received message
b97bf3fd 1668 * @m: descriptor for message info
1c1274a5 1669 * @skb: received message buffer
301bae56 1670 * @tsk: TIPC port associated with message
c4307285 1671 *
b97bf3fd 1672 * Note: Ancillary data is not captured if not requested by receiver.
c4307285 1673 *
b97bf3fd
PL
1674 * Returns 0 if successful, otherwise errno
1675 */
1c1274a5 1676static int tipc_sk_anc_data_recv(struct msghdr *m, struct sk_buff *skb,
301bae56 1677 struct tipc_sock *tsk)
b97bf3fd 1678{
1c1274a5 1679 struct tipc_msg *msg;
b97bf3fd
PL
1680 u32 anc_data[3];
1681 u32 err;
1682 u32 dest_type;
3546c750 1683 int has_name;
b97bf3fd
PL
1684 int res;
1685
1686 if (likely(m->msg_controllen == 0))
1687 return 0;
1c1274a5 1688 msg = buf_msg(skb);
b97bf3fd
PL
1689
1690 /* Optionally capture errored message object(s) */
b97bf3fd
PL
1691 err = msg ? msg_errcode(msg) : 0;
1692 if (unlikely(err)) {
1693 anc_data[0] = err;
1694 anc_data[1] = msg_data_sz(msg);
2db9983a
AS
1695 res = put_cmsg(m, SOL_TIPC, TIPC_ERRINFO, 8, anc_data);
1696 if (res)
b97bf3fd 1697 return res;
2db9983a 1698 if (anc_data[1]) {
1c1274a5
JM
1699 if (skb_linearize(skb))
1700 return -ENOMEM;
1701 msg = buf_msg(skb);
2db9983a
AS
1702 res = put_cmsg(m, SOL_TIPC, TIPC_RETDATA, anc_data[1],
1703 msg_data(msg));
1704 if (res)
1705 return res;
1706 }
b97bf3fd
PL
1707 }
1708
1709 /* Optionally capture message destination object */
b97bf3fd
PL
1710 dest_type = msg ? msg_type(msg) : TIPC_DIRECT_MSG;
1711 switch (dest_type) {
1712 case TIPC_NAMED_MSG:
3546c750 1713 has_name = 1;
b97bf3fd
PL
1714 anc_data[0] = msg_nametype(msg);
1715 anc_data[1] = msg_namelower(msg);
1716 anc_data[2] = msg_namelower(msg);
1717 break;
1718 case TIPC_MCAST_MSG:
3546c750 1719 has_name = 1;
b97bf3fd
PL
1720 anc_data[0] = msg_nametype(msg);
1721 anc_data[1] = msg_namelower(msg);
1722 anc_data[2] = msg_nameupper(msg);
1723 break;
1724 case TIPC_CONN_MSG:
301bae56
JPM
1725 has_name = (tsk->conn_type != 0);
1726 anc_data[0] = tsk->conn_type;
1727 anc_data[1] = tsk->conn_instance;
1728 anc_data[2] = tsk->conn_instance;
b97bf3fd
PL
1729 break;
1730 default:
3546c750 1731 has_name = 0;
b97bf3fd 1732 }
2db9983a
AS
1733 if (has_name) {
1734 res = put_cmsg(m, SOL_TIPC, TIPC_DESTNAME, 12, anc_data);
1735 if (res)
1736 return res;
1737 }
b97bf3fd
PL
1738
1739 return 0;
1740}
1741
10724cc7 1742static void tipc_sk_send_ack(struct tipc_sock *tsk)
739f5e4e 1743{
d6fb7e9c
PB
1744 struct sock *sk = &tsk->sk;
1745 struct net *net = sock_net(sk);
a6ca1094 1746 struct sk_buff *skb = NULL;
739f5e4e 1747 struct tipc_msg *msg;
301bae56
JPM
1748 u32 peer_port = tsk_peer_port(tsk);
1749 u32 dnode = tsk_peer_node(tsk);
739f5e4e 1750
d6fb7e9c 1751 if (!tipc_sk_connected(sk))
739f5e4e 1752 return;
c5898636
JPM
1753 skb = tipc_msg_create(CONN_MANAGER, CONN_ACK, INT_H_SIZE, 0,
1754 dnode, tsk_own_node(tsk), peer_port,
1755 tsk->portid, TIPC_OK);
a6ca1094 1756 if (!skb)
739f5e4e 1757 return;
a6ca1094 1758 msg = buf_msg(skb);
10724cc7
JPM
1759 msg_set_conn_ack(msg, tsk->rcv_unacked);
1760 tsk->rcv_unacked = 0;
1761
1762 /* Adjust to and advertize the correct window limit */
1763 if (tsk->peer_caps & TIPC_BLOCK_FLOWCTL) {
1764 tsk->rcv_win = tsk_adv_blocks(tsk->sk.sk_rcvbuf);
1765 msg_set_adv_win(msg, tsk->rcv_win);
1766 }
af9b028e 1767 tipc_node_xmit_skb(net, skb, dnode, msg_link_selector(msg));
739f5e4e
JPM
1768}
1769
85d3fc94 1770static int tipc_wait_for_rcvmsg(struct socket *sock, long *timeop)
9bbb4ecc
YX
1771{
1772 struct sock *sk = sock->sk;
48766a58 1773 DEFINE_WAIT_FUNC(wait, woken_wake_function);
85d3fc94 1774 long timeo = *timeop;
4e0df495
PB
1775 int err = sock_error(sk);
1776
1777 if (err)
1778 return err;
9bbb4ecc
YX
1779
1780 for (;;) {
fe8e4649 1781 if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
6f00089c 1782 if (sk->sk_shutdown & RCV_SHUTDOWN) {
9bbb4ecc
YX
1783 err = -ENOTCONN;
1784 break;
1785 }
48766a58 1786 add_wait_queue(sk_sleep(sk), &wait);
9bbb4ecc 1787 release_sock(sk);
48766a58
TN
1788 timeo = wait_woken(&wait, TASK_INTERRUPTIBLE, timeo);
1789 sched_annotate_sleep();
9bbb4ecc 1790 lock_sock(sk);
48766a58 1791 remove_wait_queue(sk_sleep(sk), &wait);
9bbb4ecc
YX
1792 }
1793 err = 0;
1794 if (!skb_queue_empty(&sk->sk_receive_queue))
1795 break;
9bbb4ecc
YX
1796 err = -EAGAIN;
1797 if (!timeo)
1798 break;
143fe22f
EH
1799 err = sock_intr_errno(timeo);
1800 if (signal_pending(current))
1801 break;
4e0df495
PB
1802
1803 err = sock_error(sk);
1804 if (err)
1805 break;
9bbb4ecc 1806 }
85d3fc94 1807 *timeop = timeo;
9bbb4ecc
YX
1808 return err;
1809}
1810
c4307285 1811/**
247f0f3c 1812 * tipc_recvmsg - receive packet-oriented message
b97bf3fd 1813 * @m: descriptor for message info
e9f8b101 1814 * @buflen: length of user buffer area
b97bf3fd 1815 * @flags: receive flags
c4307285 1816 *
b97bf3fd
PL
1817 * Used for SOCK_DGRAM, SOCK_RDM, and SOCK_SEQPACKET messages.
1818 * If the complete message doesn't fit in user area, truncate it.
1819 *
1820 * Returns size of returned message data, errno otherwise
1821 */
e9f8b101
JPM
1822static int tipc_recvmsg(struct socket *sock, struct msghdr *m,
1823 size_t buflen, int flags)
b97bf3fd 1824{
0c3141e9 1825 struct sock *sk = sock->sk;
e9f8b101 1826 bool connected = !tipc_sk_type_connectionless(sk);
ae236fb2 1827 struct tipc_sock *tsk = tipc_sk(sk);
e9f8b101 1828 int rc, err, hlen, dlen, copy;
b7d42635 1829 struct sk_buff_head xmitq;
ae236fb2
JM
1830 struct tipc_msg *hdr;
1831 struct sk_buff *skb;
1832 bool grp_evt;
e9f8b101 1833 long timeout;
b97bf3fd 1834
0c3141e9 1835 /* Catch invalid receive requests */
e9f8b101 1836 if (unlikely(!buflen))
b97bf3fd
PL
1837 return -EINVAL;
1838
0c3141e9 1839 lock_sock(sk);
e9f8b101
JPM
1840 if (unlikely(connected && sk->sk_state == TIPC_OPEN)) {
1841 rc = -ENOTCONN;
b97bf3fd
PL
1842 goto exit;
1843 }
e9f8b101 1844 timeout = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
b97bf3fd 1845
b7d42635 1846 /* Step rcv queue to first msg with data or error; wait if necessary */
e9f8b101 1847 do {
e9f8b101
JPM
1848 rc = tipc_wait_for_rcvmsg(sock, &timeout);
1849 if (unlikely(rc))
1850 goto exit;
1851 skb = skb_peek(&sk->sk_receive_queue);
1852 hdr = buf_msg(skb);
1853 dlen = msg_data_sz(hdr);
1854 hlen = msg_hdr_sz(hdr);
1855 err = msg_errcode(hdr);
ae236fb2 1856 grp_evt = msg_is_grp_evt(hdr);
e9f8b101
JPM
1857 if (likely(dlen || err))
1858 break;
2e84c60b 1859 tsk_advance_rx_queue(sk);
e9f8b101 1860 } while (1);
b97bf3fd 1861
e9f8b101 1862 /* Collect msg meta data, including error code and rejected data */
31c82a2d 1863 tipc_sk_set_orig_addr(m, skb);
1c1274a5 1864 rc = tipc_sk_anc_data_recv(m, skb, tsk);
e9f8b101 1865 if (unlikely(rc))
b97bf3fd 1866 goto exit;
1c1274a5 1867 hdr = buf_msg(skb);
b97bf3fd 1868
e9f8b101
JPM
1869 /* Capture data if non-error msg, otherwise just set return value */
1870 if (likely(!err)) {
1871 copy = min_t(int, dlen, buflen);
1872 if (unlikely(copy != dlen))
b97bf3fd 1873 m->msg_flags |= MSG_TRUNC;
e9f8b101 1874 rc = skb_copy_datagram_msg(skb, hlen, m, copy);
b97bf3fd 1875 } else {
e9f8b101
JPM
1876 copy = 0;
1877 rc = 0;
1878 if (err != TIPC_CONN_SHUTDOWN && connected && !m->msg_control)
1879 rc = -ECONNRESET;
b97bf3fd 1880 }
e9f8b101
JPM
1881 if (unlikely(rc))
1882 goto exit;
b97bf3fd 1883
ae236fb2
JM
1884 /* Mark message as group event if applicable */
1885 if (unlikely(grp_evt)) {
1886 if (msg_grp_evt(hdr) == TIPC_WITHDRAWN)
1887 m->msg_flags |= MSG_EOR;
1888 m->msg_flags |= MSG_OOB;
1889 copy = 0;
1890 }
1891
e9f8b101 1892 /* Caption of data or error code/rejected data was successful */
10724cc7
JPM
1893 if (unlikely(flags & MSG_PEEK))
1894 goto exit;
1895
b7d42635
JM
1896 /* Send group flow control advertisement when applicable */
1897 if (tsk->group && msg_in_group(hdr) && !grp_evt) {
e654f9f5 1898 __skb_queue_head_init(&xmitq);
b7d42635
JM
1899 tipc_group_update_rcv_win(tsk->group, tsk_blocks(hlen + dlen),
1900 msg_orignode(hdr), msg_origport(hdr),
1901 &xmitq);
1902 tipc_node_distr_xmit(sock_net(sk), &xmitq);
1903 }
1904
10724cc7 1905 tsk_advance_rx_queue(sk);
ae236fb2 1906
e9f8b101
JPM
1907 if (likely(!connected))
1908 goto exit;
1909
b7d42635 1910 /* Send connection flow control advertisement when applicable */
e9f8b101
JPM
1911 tsk->rcv_unacked += tsk_inc(tsk, hlen + dlen);
1912 if (tsk->rcv_unacked >= tsk->rcv_win / TIPC_ACK_RATE)
1913 tipc_sk_send_ack(tsk);
b97bf3fd 1914exit:
0c3141e9 1915 release_sock(sk);
e9f8b101 1916 return rc ? rc : copy;
b97bf3fd
PL
1917}
1918
c4307285 1919/**
ec8a09fb 1920 * tipc_recvstream - receive stream-oriented data
b97bf3fd 1921 * @m: descriptor for message info
ec8a09fb 1922 * @buflen: total size of user buffer area
b97bf3fd 1923 * @flags: receive flags
c4307285
YH
1924 *
1925 * Used for SOCK_STREAM messages only. If not enough data is available
b97bf3fd
PL
1926 * will optionally wait for more; never truncates data.
1927 *
1928 * Returns size of returned message data, errno otherwise
1929 */
ec8a09fb
JPM
1930static int tipc_recvstream(struct socket *sock, struct msghdr *m,
1931 size_t buflen, int flags)
b97bf3fd 1932{
0c3141e9 1933 struct sock *sk = sock->sk;
58ed9442 1934 struct tipc_sock *tsk = tipc_sk(sk);
ec8a09fb
JPM
1935 struct sk_buff *skb;
1936 struct tipc_msg *hdr;
1937 struct tipc_skb_cb *skb_cb;
1938 bool peek = flags & MSG_PEEK;
1939 int offset, required, copy, copied = 0;
1940 int hlen, dlen, err, rc;
c0bceb97 1941 bool ack = false;
ec8a09fb 1942 long timeout;
b97bf3fd 1943
0c3141e9 1944 /* Catch invalid receive attempts */
ec8a09fb 1945 if (unlikely(!buflen))
b97bf3fd
PL
1946 return -EINVAL;
1947
0c3141e9 1948 lock_sock(sk);
b97bf3fd 1949
438adcaf 1950 if (unlikely(sk->sk_state == TIPC_OPEN)) {
ec8a09fb 1951 rc = -ENOTCONN;
9bbb4ecc 1952 goto exit;
b97bf3fd 1953 }
ec8a09fb
JPM
1954 required = sock_rcvlowat(sk, flags & MSG_WAITALL, buflen);
1955 timeout = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
b97bf3fd 1956
ec8a09fb
JPM
1957 do {
1958 /* Look at first msg in receive queue; wait if necessary */
1959 rc = tipc_wait_for_rcvmsg(sock, &timeout);
1960 if (unlikely(rc))
1961 break;
1962 skb = skb_peek(&sk->sk_receive_queue);
1963 skb_cb = TIPC_SKB_CB(skb);
1964 hdr = buf_msg(skb);
1965 dlen = msg_data_sz(hdr);
1966 hlen = msg_hdr_sz(hdr);
1967 err = msg_errcode(hdr);
0232fd0a 1968
ec8a09fb
JPM
1969 /* Discard any empty non-errored (SYN-) message */
1970 if (unlikely(!dlen && !err)) {
1971 tsk_advance_rx_queue(sk);
1972 continue;
1973 }
0232fd0a 1974
ec8a09fb
JPM
1975 /* Collect msg meta data, incl. error code and rejected data */
1976 if (!copied) {
31c82a2d 1977 tipc_sk_set_orig_addr(m, skb);
1c1274a5 1978 rc = tipc_sk_anc_data_recv(m, skb, tsk);
ec8a09fb
JPM
1979 if (rc)
1980 break;
1c1274a5 1981 hdr = buf_msg(skb);
ec8a09fb 1982 }
b97bf3fd 1983
ec8a09fb
JPM
1984 /* Copy data if msg ok, otherwise return error/partial data */
1985 if (likely(!err)) {
c0bceb97 1986 ack = msg_ack_required(hdr);
ec8a09fb
JPM
1987 offset = skb_cb->bytes_read;
1988 copy = min_t(int, dlen - offset, buflen - copied);
1989 rc = skb_copy_datagram_msg(skb, hlen + offset, m, copy);
1990 if (unlikely(rc))
1991 break;
1992 copied += copy;
1993 offset += copy;
1994 if (unlikely(offset < dlen)) {
1995 if (!peek)
1996 skb_cb->bytes_read = offset;
1997 break;
1998 }
1999 } else {
2000 rc = 0;
2001 if ((err != TIPC_CONN_SHUTDOWN) && !m->msg_control)
2002 rc = -ECONNRESET;
2003 if (copied || rc)
2004 break;
b97bf3fd 2005 }
b97bf3fd 2006
ec8a09fb
JPM
2007 if (unlikely(peek))
2008 break;
b97bf3fd 2009
ec8a09fb 2010 tsk_advance_rx_queue(sk);
10724cc7 2011
ec8a09fb
JPM
2012 /* Send connection flow control advertisement when applicable */
2013 tsk->rcv_unacked += tsk_inc(tsk, hlen + dlen);
c0bceb97 2014 if (ack || tsk->rcv_unacked >= tsk->rcv_win / TIPC_ACK_RATE)
ec8a09fb 2015 tipc_sk_send_ack(tsk);
b97bf3fd 2016
ec8a09fb
JPM
2017 /* Exit if all requested data or FIN/error received */
2018 if (copied == buflen || err)
2019 break;
b97bf3fd 2020
ec8a09fb 2021 } while (!skb_queue_empty(&sk->sk_receive_queue) || copied < required);
b97bf3fd 2022exit:
0c3141e9 2023 release_sock(sk);
ec8a09fb 2024 return copied ? copied : rc;
b97bf3fd
PL
2025}
2026
f288bef4
YX
2027/**
2028 * tipc_write_space - wake up thread if port congestion is released
2029 * @sk: socket
2030 */
2031static void tipc_write_space(struct sock *sk)
2032{
2033 struct socket_wq *wq;
2034
2035 rcu_read_lock();
2036 wq = rcu_dereference(sk->sk_wq);
1ce0bf50 2037 if (skwq_has_sleeper(wq))
a9a08845
LT
2038 wake_up_interruptible_sync_poll(&wq->wait, EPOLLOUT |
2039 EPOLLWRNORM | EPOLLWRBAND);
f288bef4
YX
2040 rcu_read_unlock();
2041}
2042
2043/**
2044 * tipc_data_ready - wake up threads to indicate messages have been received
2045 * @sk: socket
2046 * @len: the length of messages
2047 */
676d2369 2048static void tipc_data_ready(struct sock *sk)
f288bef4
YX
2049{
2050 struct socket_wq *wq;
2051
2052 rcu_read_lock();
2053 wq = rcu_dereference(sk->sk_wq);
1ce0bf50 2054 if (skwq_has_sleeper(wq))
a9a08845
LT
2055 wake_up_interruptible_sync_poll(&wq->wait, EPOLLIN |
2056 EPOLLRDNORM | EPOLLRDBAND);
f288bef4
YX
2057 rcu_read_unlock();
2058}
2059
f4195d1e
YX
2060static void tipc_sock_destruct(struct sock *sk)
2061{
2062 __skb_queue_purge(&sk->sk_receive_queue);
2063}
2064
64ac5f59
JM
2065static void tipc_sk_proto_rcv(struct sock *sk,
2066 struct sk_buff_head *inputq,
2067 struct sk_buff_head *xmitq)
2068{
2069 struct sk_buff *skb = __skb_dequeue(inputq);
2070 struct tipc_sock *tsk = tipc_sk(sk);
2071 struct tipc_msg *hdr = buf_msg(skb);
75da2163 2072 struct tipc_group *grp = tsk->group;
b7d42635 2073 bool wakeup = false;
64ac5f59
JM
2074
2075 switch (msg_user(hdr)) {
2076 case CONN_MANAGER:
e7eb0582 2077 tipc_sk_conn_proto_rcv(tsk, skb, inputq, xmitq);
64ac5f59
JM
2078 return;
2079 case SOCK_WAKEUP:
a80ae530 2080 tipc_dest_del(&tsk->cong_links, msg_orignode(hdr), 0);
bfd07f3d
TN
2081 /* coupled with smp_rmb() in tipc_wait_for_cond() */
2082 smp_wmb();
64ac5f59 2083 tsk->cong_link_cnt--;
b7d42635 2084 wakeup = true;
c0bceb97 2085 tipc_sk_push_backlog(tsk);
64ac5f59 2086 break;
75da2163 2087 case GROUP_PROTOCOL:
b7d42635 2088 tipc_group_proto_rcv(grp, &wakeup, hdr, inputq, xmitq);
75da2163 2089 break;
64ac5f59 2090 case TOP_SRV:
b7d42635 2091 tipc_group_member_evt(tsk->group, &wakeup, &sk->sk_rcvbuf,
7ad32bcb 2092 hdr, inputq, xmitq);
64ac5f59
JM
2093 break;
2094 default:
2095 break;
2096 }
2097
b7d42635
JM
2098 if (wakeup)
2099 sk->sk_write_space(sk);
2100
64ac5f59
JM
2101 kfree_skb(skb);
2102}
2103
7e6c131e 2104/**
39fdc9c7 2105 * tipc_sk_filter_connect - check incoming message for a connection-based socket
58ed9442 2106 * @tsk: TIPC socket
39fdc9c7
JM
2107 * @skb: pointer to message buffer.
2108 * Returns true if message should be added to receive queue, false otherwise
7e6c131e 2109 */
64ac5f59 2110static bool tipc_sk_filter_connect(struct tipc_sock *tsk, struct sk_buff *skb)
7e6c131e 2111{
58ed9442 2112 struct sock *sk = &tsk->sk;
f2f9800d 2113 struct net *net = sock_net(sk);
cda3696d 2114 struct tipc_msg *hdr = buf_msg(skb);
39fdc9c7
JM
2115 bool con_msg = msg_connected(hdr);
2116 u32 pport = tsk_peer_port(tsk);
2117 u32 pnode = tsk_peer_node(tsk);
2118 u32 oport = msg_origport(hdr);
2119 u32 onode = msg_orignode(hdr);
2120 int err = msg_errcode(hdr);
67879274 2121 unsigned long delay;
7e6c131e 2122
cda3696d
JPM
2123 if (unlikely(msg_mcast(hdr)))
2124 return false;
c0bceb97 2125 tsk->oneway = 0;
7e6c131e 2126
99a20889
PB
2127 switch (sk->sk_state) {
2128 case TIPC_CONNECTING:
39fdc9c7
JM
2129 /* Setup ACK */
2130 if (likely(con_msg)) {
2131 if (err)
2132 break;
2133 tipc_sk_finish_conn(tsk, oport, onode);
2134 msg_set_importance(&tsk->phdr, msg_importance(hdr));
2135 /* ACK+ message with data is added to receive queue */
2136 if (msg_data_sz(hdr))
2137 return true;
2138 /* Empty ACK-, - wake up sleeping connect() and drop */
ff946833 2139 sk->sk_state_change(sk);
39fdc9c7
JM
2140 msg_set_dest_droppable(hdr, 1);
2141 return false;
584d24b3 2142 }
39fdc9c7
JM
2143 /* Ignore connectionless message if not from listening socket */
2144 if (oport != pport || onode != pnode)
2145 return false;
584d24b3 2146
67879274
TN
2147 /* Rejected SYN */
2148 if (err != TIPC_ERR_OVERLOAD)
2149 break;
2150
2151 /* Prepare for new setup attempt if we have a SYN clone */
2152 if (skb_queue_empty(&sk->sk_write_queue))
2153 break;
2154 get_random_bytes(&delay, 2);
2155 delay %= (tsk->conn_timeout / 4);
2156 delay = msecs_to_jiffies(delay + 100);
2157 sk_reset_timer(sk, &sk->sk_timer, jiffies + delay);
2158 return false;
438adcaf 2159 case TIPC_OPEN:
9fd4b070 2160 case TIPC_DISCONNECTING:
39fdc9c7 2161 return false;
438adcaf 2162 case TIPC_LISTEN:
7e6c131e 2163 /* Accept only SYN message */
25b9221b
JM
2164 if (!msg_is_syn(hdr) &&
2165 tipc_node_get_capabilities(net, onode) & TIPC_SYN_BIT)
2166 return false;
39fdc9c7 2167 if (!con_msg && !err)
cda3696d 2168 return true;
39fdc9c7 2169 return false;
f40acbaf 2170 case TIPC_ESTABLISHED:
c0bceb97
JM
2171 if (!skb_queue_empty(&sk->sk_write_queue))
2172 tipc_sk_push_backlog(tsk);
f40acbaf 2173 /* Accept only connection-based messages sent by peer */
39fdc9c7
JM
2174 if (likely(con_msg && !err && pport == oport && pnode == onode))
2175 return true;
2176 if (!tsk_peer_msg(tsk, hdr))
f40acbaf 2177 return false;
39fdc9c7
JM
2178 if (!err)
2179 return true;
2180 tipc_set_sk_state(sk, TIPC_DISCONNECTING);
2181 tipc_node_remove_conn(net, pnode, tsk->portid);
2182 sk->sk_state_change(sk);
f40acbaf 2183 return true;
7e6c131e 2184 default:
438adcaf 2185 pr_err("Unknown sk_state %u\n", sk->sk_state);
7e6c131e 2186 }
39fdc9c7
JM
2187 /* Abort connection setup attempt */
2188 tipc_set_sk_state(sk, TIPC_DISCONNECTING);
2189 sk->sk_err = ECONNREFUSED;
2190 sk->sk_state_change(sk);
2191 return true;
7e6c131e
YX
2192}
2193
aba79f33
YX
2194/**
2195 * rcvbuf_limit - get proper overload limit of socket receive queue
2196 * @sk: socket
10724cc7 2197 * @skb: message
aba79f33 2198 *
10724cc7
JPM
2199 * For connection oriented messages, irrespective of importance,
2200 * default queue limit is 2 MB.
aba79f33 2201 *
10724cc7
JPM
2202 * For connectionless messages, queue limits are based on message
2203 * importance as follows:
aba79f33 2204 *
10724cc7
JPM
2205 * TIPC_LOW_IMPORTANCE (2 MB)
2206 * TIPC_MEDIUM_IMPORTANCE (4 MB)
2207 * TIPC_HIGH_IMPORTANCE (8 MB)
2208 * TIPC_CRITICAL_IMPORTANCE (16 MB)
aba79f33
YX
2209 *
2210 * Returns overload limit according to corresponding message importance
2211 */
10724cc7 2212static unsigned int rcvbuf_limit(struct sock *sk, struct sk_buff *skb)
aba79f33 2213{
10724cc7
JPM
2214 struct tipc_sock *tsk = tipc_sk(sk);
2215 struct tipc_msg *hdr = buf_msg(skb);
2216
b7d42635 2217 if (unlikely(msg_in_group(hdr)))
8265792b 2218 return READ_ONCE(sk->sk_rcvbuf);
b7d42635 2219
10724cc7 2220 if (unlikely(!msg_connected(hdr)))
8265792b 2221 return READ_ONCE(sk->sk_rcvbuf) << msg_importance(hdr);
aba79f33 2222
10724cc7 2223 if (likely(tsk->peer_caps & TIPC_BLOCK_FLOWCTL))
8265792b 2224 return READ_ONCE(sk->sk_rcvbuf);
0cee6bbe 2225
10724cc7 2226 return FLOWCTL_MSG_LIM;
aba79f33
YX
2227}
2228
c4307285 2229/**
64ac5f59 2230 * tipc_sk_filter_rcv - validate incoming message
0c3141e9 2231 * @sk: socket
cda3696d 2232 * @skb: pointer to message.
c4307285 2233 *
0c3141e9
AS
2234 * Enqueues message on receive queue if acceptable; optionally handles
2235 * disconnect indication for a connected socket.
2236 *
1186adf7 2237 * Called with socket lock already taken
c4307285 2238 *
b97bf3fd 2239 */
64ac5f59
JM
2240static void tipc_sk_filter_rcv(struct sock *sk, struct sk_buff *skb,
2241 struct sk_buff_head *xmitq)
b97bf3fd 2242{
64ac5f59 2243 bool sk_conn = !tipc_sk_type_connectionless(sk);
58ed9442 2244 struct tipc_sock *tsk = tipc_sk(sk);
75da2163 2245 struct tipc_group *grp = tsk->group;
cda3696d 2246 struct tipc_msg *hdr = buf_msg(skb);
64ac5f59
JM
2247 struct net *net = sock_net(sk);
2248 struct sk_buff_head inputq;
77d5ad40 2249 int mtyp = msg_type(hdr);
64ac5f59 2250 int limit, err = TIPC_OK;
ec8a2e56 2251
01e661eb 2252 trace_tipc_sk_filter_rcv(sk, skb, TIPC_DUMP_ALL, " ");
64ac5f59
JM
2253 TIPC_SKB_CB(skb)->bytes_read = 0;
2254 __skb_queue_head_init(&inputq);
2255 __skb_queue_tail(&inputq, skb);
50100a5e 2256
64ac5f59
JM
2257 if (unlikely(!msg_isdata(hdr)))
2258 tipc_sk_proto_rcv(sk, &inputq, xmitq);
0c3141e9 2259
75da2163
JM
2260 if (unlikely(grp))
2261 tipc_group_filter_msg(grp, &inputq, xmitq);
2262
77d5ad40 2263 if (unlikely(!grp) && mtyp == TIPC_MCAST_MSG)
08e046c8 2264 tipc_mcast_filter_msg(net, &tsk->mc_method.deferredq, &inputq);
c55c8eda 2265
64ac5f59
JM
2266 /* Validate and add to receive buffer if there is space */
2267 while ((skb = __skb_dequeue(&inputq))) {
2268 hdr = buf_msg(skb);
2269 limit = rcvbuf_limit(sk, skb);
2270 if ((sk_conn && !tipc_sk_filter_connect(tsk, skb)) ||
75da2163
JM
2271 (!sk_conn && msg_connected(hdr)) ||
2272 (!grp && msg_in_group(hdr)))
cda3696d 2273 err = TIPC_ERR_NO_PORT;
872619d8 2274 else if (sk_rmem_alloc_get(sk) + skb->truesize >= limit) {
01e661eb
TL
2275 trace_tipc_sk_dump(sk, skb, TIPC_DUMP_ALL,
2276 "err_overload2!");
872619d8 2277 atomic_inc(&sk->sk_drops);
64ac5f59 2278 err = TIPC_ERR_OVERLOAD;
872619d8 2279 }
b97bf3fd 2280
64ac5f59 2281 if (unlikely(err)) {
01e661eb
TL
2282 if (tipc_msg_reverse(tipc_own_addr(net), &skb, err)) {
2283 trace_tipc_sk_rej_msg(sk, skb, TIPC_DUMP_NONE,
2284 "@filter_rcv!");
2285 __skb_queue_tail(xmitq, skb);
2286 }
64ac5f59
JM
2287 err = TIPC_OK;
2288 continue;
2289 }
2290 __skb_queue_tail(&sk->sk_receive_queue, skb);
2291 skb_set_owner_r(skb, sk);
01e661eb
TL
2292 trace_tipc_sk_overlimit2(sk, skb, TIPC_DUMP_ALL,
2293 "rcvq >90% allocated!");
64ac5f59 2294 sk->sk_data_ready(sk);
cda3696d 2295 }
0c3141e9 2296}
b97bf3fd 2297
0c3141e9 2298/**
64ac5f59 2299 * tipc_sk_backlog_rcv - handle incoming message from backlog queue
0c3141e9 2300 * @sk: socket
a6ca1094 2301 * @skb: message
0c3141e9 2302 *
e3a77561 2303 * Caller must hold socket lock
0c3141e9 2304 */
64ac5f59 2305static int tipc_sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
0c3141e9 2306{
64ac5f59 2307 unsigned int before = sk_rmem_alloc_get(sk);
f1d048f2 2308 struct sk_buff_head xmitq;
64ac5f59 2309 unsigned int added;
0c3141e9 2310
f1d048f2
JPM
2311 __skb_queue_head_init(&xmitq);
2312
64ac5f59
JM
2313 tipc_sk_filter_rcv(sk, skb, &xmitq);
2314 added = sk_rmem_alloc_get(sk) - before;
2315 atomic_add(added, &tipc_sk(sk)->dupl_rcvcnt);
f1d048f2 2316
64ac5f59 2317 /* Send pending response/rejected messages, if any */
f70d37b7 2318 tipc_node_distr_xmit(sock_net(sk), &xmitq);
0c3141e9
AS
2319 return 0;
2320}
2321
d570d864 2322/**
c637c103
JPM
2323 * tipc_sk_enqueue - extract all buffers with destination 'dport' from
2324 * inputq and try adding them to socket or backlog queue
2325 * @inputq: list of incoming buffers with potentially different destinations
2326 * @sk: socket where the buffers should be enqueued
2327 * @dport: port number for the socket
d570d864
JPM
2328 *
2329 * Caller must hold socket lock
d570d864 2330 */
cda3696d 2331static void tipc_sk_enqueue(struct sk_buff_head *inputq, struct sock *sk,
f1d048f2 2332 u32 dport, struct sk_buff_head *xmitq)
d570d864 2333{
f1d048f2
JPM
2334 unsigned long time_limit = jiffies + 2;
2335 struct sk_buff *skb;
d570d864
JPM
2336 unsigned int lim;
2337 atomic_t *dcnt;
f1d048f2 2338 u32 onode;
c637c103
JPM
2339
2340 while (skb_queue_len(inputq)) {
51a00daf 2341 if (unlikely(time_after_eq(jiffies, time_limit)))
cda3696d
JPM
2342 return;
2343
c637c103
JPM
2344 skb = tipc_skb_dequeue(inputq, dport);
2345 if (unlikely(!skb))
cda3696d
JPM
2346 return;
2347
2348 /* Add message directly to receive queue if possible */
c637c103 2349 if (!sock_owned_by_user(sk)) {
64ac5f59 2350 tipc_sk_filter_rcv(sk, skb, xmitq);
cda3696d 2351 continue;
c637c103 2352 }
cda3696d
JPM
2353
2354 /* Try backlog, compensating for double-counted bytes */
c637c103 2355 dcnt = &tipc_sk(sk)->dupl_rcvcnt;
7c8bcfb1 2356 if (!sk->sk_backlog.len)
c637c103
JPM
2357 atomic_set(dcnt, 0);
2358 lim = rcvbuf_limit(sk, skb) + atomic_read(dcnt);
01e661eb
TL
2359 if (likely(!sk_add_backlog(sk, skb, lim))) {
2360 trace_tipc_sk_overlimit1(sk, skb, TIPC_DUMP_ALL,
2361 "bklg & rcvq >90% allocated!");
c637c103 2362 continue;
01e661eb 2363 }
cda3696d 2364
01e661eb 2365 trace_tipc_sk_dump(sk, skb, TIPC_DUMP_ALL, "err_overload!");
cda3696d 2366 /* Overload => reject message back to sender */
f1d048f2 2367 onode = tipc_own_addr(sock_net(sk));
872619d8 2368 atomic_inc(&sk->sk_drops);
01e661eb
TL
2369 if (tipc_msg_reverse(onode, &skb, TIPC_ERR_OVERLOAD)) {
2370 trace_tipc_sk_rej_msg(sk, skb, TIPC_DUMP_ALL,
2371 "@sk_enqueue!");
f1d048f2 2372 __skb_queue_tail(xmitq, skb);
01e661eb 2373 }
cda3696d 2374 break;
c637c103 2375 }
d570d864
JPM
2376}
2377
0c3141e9 2378/**
c637c103
JPM
2379 * tipc_sk_rcv - handle a chain of incoming buffers
2380 * @inputq: buffer list containing the buffers
2381 * Consumes all buffers in list until inputq is empty
2382 * Note: may be called in multiple threads referring to the same queue
0c3141e9 2383 */
cda3696d 2384void tipc_sk_rcv(struct net *net, struct sk_buff_head *inputq)
0c3141e9 2385{
f1d048f2 2386 struct sk_buff_head xmitq;
c637c103 2387 u32 dnode, dport = 0;
9871b27f 2388 int err;
9816f061 2389 struct tipc_sock *tsk;
9816f061 2390 struct sock *sk;
cda3696d 2391 struct sk_buff *skb;
9816f061 2392
f1d048f2 2393 __skb_queue_head_init(&xmitq);
c637c103 2394 while (skb_queue_len(inputq)) {
c637c103
JPM
2395 dport = tipc_skb_peek_port(inputq, dport);
2396 tsk = tipc_sk_lookup(net, dport);
cda3696d 2397
c637c103
JPM
2398 if (likely(tsk)) {
2399 sk = &tsk->sk;
2400 if (likely(spin_trylock_bh(&sk->sk_lock.slock))) {
f1d048f2 2401 tipc_sk_enqueue(inputq, sk, dport, &xmitq);
c637c103 2402 spin_unlock_bh(&sk->sk_lock.slock);
c637c103 2403 }
f1d048f2 2404 /* Send pending response/rejected messages, if any */
f70d37b7 2405 tipc_node_distr_xmit(sock_net(sk), &xmitq);
c637c103 2406 sock_put(sk);
c637c103 2407 continue;
c637c103 2408 }
cda3696d
JPM
2409 /* No destination socket => dequeue skb if still there */
2410 skb = tipc_skb_dequeue(inputq, dport);
2411 if (!skb)
2412 return;
2413
2414 /* Try secondary lookup if unresolved named message */
2415 err = TIPC_ERR_NO_PORT;
2416 if (tipc_msg_lookup_dest(net, skb, &err))
2417 goto xmit;
2418
2419 /* Prepare for message rejection */
2420 if (!tipc_msg_reverse(tipc_own_addr(net), &skb, err))
c637c103 2421 continue;
01e661eb
TL
2422
2423 trace_tipc_sk_rej_msg(NULL, skb, TIPC_DUMP_NONE, "@sk_rcv!");
e3a77561 2424xmit:
cda3696d 2425 dnode = msg_destnode(buf_msg(skb));
af9b028e 2426 tipc_node_xmit_skb(net, skb, dnode, dport);
c637c103 2427 }
b97bf3fd
PL
2428}
2429
78eb3a53
YX
2430static int tipc_wait_for_connect(struct socket *sock, long *timeo_p)
2431{
d9dc8b0f 2432 DEFINE_WAIT_FUNC(wait, woken_wake_function);
78eb3a53 2433 struct sock *sk = sock->sk;
78eb3a53
YX
2434 int done;
2435
2436 do {
2437 int err = sock_error(sk);
2438 if (err)
2439 return err;
2440 if (!*timeo_p)
2441 return -ETIMEDOUT;
2442 if (signal_pending(current))
2443 return sock_intr_errno(*timeo_p);
5391a877
TL
2444 if (sk->sk_state == TIPC_DISCONNECTING)
2445 break;
78eb3a53 2446
d9dc8b0f 2447 add_wait_queue(sk_sleep(sk), &wait);
9546a0b7
TL
2448 done = sk_wait_event(sk, timeo_p, tipc_sk_connected(sk),
2449 &wait);
d9dc8b0f 2450 remove_wait_queue(sk_sleep(sk), &wait);
78eb3a53
YX
2451 } while (!done);
2452 return 0;
2453}
2454
ea239314
EH
2455static bool tipc_sockaddr_is_sane(struct sockaddr_tipc *addr)
2456{
2457 if (addr->family != AF_TIPC)
2458 return false;
2459 if (addr->addrtype == TIPC_SERVICE_RANGE)
2460 return (addr->addr.nameseq.lower <= addr->addr.nameseq.upper);
2461 return (addr->addrtype == TIPC_SERVICE_ADDR ||
2462 addr->addrtype == TIPC_SOCKET_ADDR);
2463}
2464
b97bf3fd 2465/**
247f0f3c 2466 * tipc_connect - establish a connection to another TIPC port
b97bf3fd
PL
2467 * @sock: socket structure
2468 * @dest: socket address for destination port
2469 * @destlen: size of socket address data structure
0c3141e9 2470 * @flags: file-related flags associated with socket
b97bf3fd
PL
2471 *
2472 * Returns 0 on success, errno otherwise
2473 */
247f0f3c
YX
2474static int tipc_connect(struct socket *sock, struct sockaddr *dest,
2475 int destlen, int flags)
b97bf3fd 2476{
0c3141e9 2477 struct sock *sk = sock->sk;
f2f8036e 2478 struct tipc_sock *tsk = tipc_sk(sk);
b89741a0
AS
2479 struct sockaddr_tipc *dst = (struct sockaddr_tipc *)dest;
2480 struct msghdr m = {NULL,};
f2f8036e 2481 long timeout = (flags & O_NONBLOCK) ? 0 : tsk->conn_timeout;
99a20889 2482 int previous;
f2f8036e 2483 int res = 0;
b89741a0 2484
23998835
JM
2485 if (destlen != sizeof(struct sockaddr_tipc))
2486 return -EINVAL;
2487
0c3141e9
AS
2488 lock_sock(sk);
2489
75da2163
JM
2490 if (tsk->group) {
2491 res = -EINVAL;
2492 goto exit;
2493 }
2494
23998835
JM
2495 if (dst->family == AF_UNSPEC) {
2496 memset(&tsk->peer, 0, sizeof(struct sockaddr_tipc));
2497 if (!tipc_sk_type_connectionless(sk))
610600c8 2498 res = -EINVAL;
0c3141e9
AS
2499 goto exit;
2500 }
ea239314 2501 if (!tipc_sockaddr_is_sane(dst)) {
0c3141e9 2502 res = -EINVAL;
23998835 2503 goto exit;
ea239314 2504 }
23998835
JM
2505 /* DGRAM/RDM connect(), just save the destaddr */
2506 if (tipc_sk_type_connectionless(sk)) {
2507 memcpy(&tsk->peer, dest, destlen);
0c3141e9 2508 goto exit;
ea239314
EH
2509 } else if (dst->addrtype == TIPC_SERVICE_RANGE) {
2510 res = -EINVAL;
2511 goto exit;
0c3141e9
AS
2512 }
2513
99a20889 2514 previous = sk->sk_state;
438adcaf
PB
2515
2516 switch (sk->sk_state) {
2517 case TIPC_OPEN:
584d24b3
YX
2518 /* Send a 'SYN-' to destination */
2519 m.msg_name = dest;
2520 m.msg_namelen = destlen;
2521
2522 /* If connect is in non-blocking case, set MSG_DONTWAIT to
2523 * indicate send_msg() is never blocked.
2524 */
2525 if (!timeout)
2526 m.msg_flags = MSG_DONTWAIT;
2527
39a0295f 2528 res = __tipc_sendmsg(sock, &m, 0);
584d24b3
YX
2529 if ((res < 0) && (res != -EWOULDBLOCK))
2530 goto exit;
2531
99a20889 2532 /* Just entered TIPC_CONNECTING state; the only
584d24b3
YX
2533 * difference is that return value in non-blocking
2534 * case is EINPROGRESS, rather than EALREADY.
2535 */
2536 res = -EINPROGRESS;
f79e3365 2537 /* fall through */
99a20889
PB
2538 case TIPC_CONNECTING:
2539 if (!timeout) {
2540 if (previous == TIPC_CONNECTING)
2541 res = -EALREADY;
78eb3a53 2542 goto exit;
99a20889 2543 }
78eb3a53
YX
2544 timeout = msecs_to_jiffies(timeout);
2545 /* Wait until an 'ACK' or 'RST' arrives, or a timeout occurs */
2546 res = tipc_wait_for_connect(sock, &timeout);
f40acbaf
PB
2547 break;
2548 case TIPC_ESTABLISHED:
584d24b3 2549 res = -EISCONN;
f40acbaf
PB
2550 break;
2551 default:
584d24b3 2552 res = -EINVAL;
f40acbaf 2553 }
99a20889 2554
0c3141e9
AS
2555exit:
2556 release_sock(sk);
b89741a0 2557 return res;
b97bf3fd
PL
2558}
2559
c4307285 2560/**
247f0f3c 2561 * tipc_listen - allow socket to listen for incoming connections
b97bf3fd
PL
2562 * @sock: socket structure
2563 * @len: (unused)
c4307285 2564 *
b97bf3fd
PL
2565 * Returns 0 on success, errno otherwise
2566 */
247f0f3c 2567static int tipc_listen(struct socket *sock, int len)
b97bf3fd 2568{
0c3141e9
AS
2569 struct sock *sk = sock->sk;
2570 int res;
2571
2572 lock_sock(sk);
0c288c86 2573 res = tipc_set_sk_state(sk, TIPC_LISTEN);
0c3141e9 2574 release_sock(sk);
0c288c86 2575
0c3141e9 2576 return res;
b97bf3fd
PL
2577}
2578
6398e23c
YX
2579static int tipc_wait_for_accept(struct socket *sock, long timeo)
2580{
2581 struct sock *sk = sock->sk;
2582 DEFINE_WAIT(wait);
2583 int err;
2584
2585 /* True wake-one mechanism for incoming connections: only
2586 * one process gets woken up, not the 'whole herd'.
2587 * Since we do not 'race & poll' for established sockets
2588 * anymore, the common case will execute the loop only once.
2589 */
2590 for (;;) {
2591 prepare_to_wait_exclusive(sk_sleep(sk), &wait,
2592 TASK_INTERRUPTIBLE);
fe8e4649 2593 if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
6398e23c
YX
2594 release_sock(sk);
2595 timeo = schedule_timeout(timeo);
2596 lock_sock(sk);
2597 }
2598 err = 0;
2599 if (!skb_queue_empty(&sk->sk_receive_queue))
2600 break;
6398e23c
YX
2601 err = -EAGAIN;
2602 if (!timeo)
2603 break;
143fe22f
EH
2604 err = sock_intr_errno(timeo);
2605 if (signal_pending(current))
2606 break;
6398e23c
YX
2607 }
2608 finish_wait(sk_sleep(sk), &wait);
2609 return err;
2610}
2611
c4307285 2612/**
247f0f3c 2613 * tipc_accept - wait for connection request
b97bf3fd
PL
2614 * @sock: listening socket
2615 * @newsock: new socket that is to be connected
2616 * @flags: file-related flags associated with socket
c4307285 2617 *
b97bf3fd
PL
2618 * Returns 0 on success, errno otherwise
2619 */
cdfbabfb
DH
2620static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags,
2621 bool kern)
b97bf3fd 2622{
0fef8f20 2623 struct sock *new_sk, *sk = sock->sk;
b97bf3fd 2624 struct sk_buff *buf;
301bae56 2625 struct tipc_sock *new_tsock;
0fef8f20 2626 struct tipc_msg *msg;
6398e23c 2627 long timeo;
0c3141e9 2628 int res;
b97bf3fd 2629
0c3141e9 2630 lock_sock(sk);
b97bf3fd 2631
0c288c86 2632 if (sk->sk_state != TIPC_LISTEN) {
0c3141e9 2633 res = -EINVAL;
b97bf3fd
PL
2634 goto exit;
2635 }
6398e23c
YX
2636 timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
2637 res = tipc_wait_for_accept(sock, timeo);
2638 if (res)
2639 goto exit;
0c3141e9
AS
2640
2641 buf = skb_peek(&sk->sk_receive_queue);
2642
cdfbabfb 2643 res = tipc_sk_create(sock_net(sock->sk), new_sock, 0, kern);
0fef8f20
PG
2644 if (res)
2645 goto exit;
fdd75ea8 2646 security_sk_clone(sock->sk, new_sock->sk);
b97bf3fd 2647
0fef8f20 2648 new_sk = new_sock->sk;
301bae56 2649 new_tsock = tipc_sk(new_sk);
0fef8f20 2650 msg = buf_msg(buf);
b97bf3fd 2651
0fef8f20
PG
2652 /* we lock on new_sk; but lockdep sees the lock on sk */
2653 lock_sock_nested(new_sk, SINGLE_DEPTH_NESTING);
2654
2655 /*
2656 * Reject any stray messages received by new socket
2657 * before the socket lock was taken (very, very unlikely)
2658 */
49afb806 2659 tsk_rej_rx_queue(new_sk, TIPC_ERR_NO_PORT);
0fef8f20
PG
2660
2661 /* Connect new socket to it's peer */
301bae56 2662 tipc_sk_finish_conn(new_tsock, msg_origport(msg), msg_orignode(msg));
0fef8f20 2663
301bae56 2664 tsk_set_importance(new_tsock, msg_importance(msg));
0fef8f20 2665 if (msg_named(msg)) {
301bae56
JPM
2666 new_tsock->conn_type = msg_nametype(msg);
2667 new_tsock->conn_instance = msg_nameinst(msg);
b97bf3fd 2668 }
0fef8f20
PG
2669
2670 /*
2671 * Respond to 'SYN-' by discarding it & returning 'ACK'-.
2672 * Respond to 'SYN+' by queuing it on new socket.
2673 */
2674 if (!msg_data_sz(msg)) {
2675 struct msghdr m = {NULL,};
2676
2e84c60b 2677 tsk_advance_rx_queue(sk);
365ad353 2678 __tipc_sendstream(new_sock, &m, 0);
0fef8f20
PG
2679 } else {
2680 __skb_dequeue(&sk->sk_receive_queue);
2681 __skb_queue_head(&new_sk->sk_receive_queue, buf);
aba79f33 2682 skb_set_owner_r(buf, new_sk);
0fef8f20
PG
2683 }
2684 release_sock(new_sk);
b97bf3fd 2685exit:
0c3141e9 2686 release_sock(sk);
b97bf3fd
PL
2687 return res;
2688}
2689
2690/**
247f0f3c 2691 * tipc_shutdown - shutdown socket connection
b97bf3fd 2692 * @sock: socket structure
e247a8f5 2693 * @how: direction to close (must be SHUT_RDWR)
b97bf3fd
PL
2694 *
2695 * Terminates connection (if necessary), then purges socket's receive queue.
c4307285 2696 *
b97bf3fd
PL
2697 * Returns 0 on success, errno otherwise
2698 */
247f0f3c 2699static int tipc_shutdown(struct socket *sock, int how)
b97bf3fd 2700{
0c3141e9 2701 struct sock *sk = sock->sk;
b97bf3fd
PL
2702 int res;
2703
e247a8f5
AS
2704 if (how != SHUT_RDWR)
2705 return -EINVAL;
b97bf3fd 2706
0c3141e9 2707 lock_sock(sk);
b97bf3fd 2708
01e661eb 2709 trace_tipc_sk_shutdown(sk, NULL, TIPC_DUMP_ALL, " ");
6f00089c
PB
2710 __tipc_shutdown(sock, TIPC_CONN_SHUTDOWN);
2711 sk->sk_shutdown = SEND_SHUTDOWN;
b97bf3fd 2712
6f00089c 2713 if (sk->sk_state == TIPC_DISCONNECTING) {
75031151 2714 /* Discard any unreceived messages */
57467e56 2715 __skb_queue_purge(&sk->sk_receive_queue);
75031151
YX
2716
2717 /* Wake up anyone sleeping in poll */
2718 sk->sk_state_change(sk);
b97bf3fd 2719 res = 0;
6f00089c 2720 } else {
b97bf3fd
PL
2721 res = -ENOTCONN;
2722 }
2723
0c3141e9 2724 release_sock(sk);
b97bf3fd
PL
2725 return res;
2726}
2727
afe8792f
JM
2728static void tipc_sk_check_probing_state(struct sock *sk,
2729 struct sk_buff_head *list)
2730{
2731 struct tipc_sock *tsk = tipc_sk(sk);
2732 u32 pnode = tsk_peer_node(tsk);
2733 u32 pport = tsk_peer_port(tsk);
2734 u32 self = tsk_own_node(tsk);
2735 u32 oport = tsk->portid;
2736 struct sk_buff *skb;
2737
2738 if (tsk->probe_unacked) {
2739 tipc_set_sk_state(sk, TIPC_DISCONNECTING);
2740 sk->sk_err = ECONNABORTED;
2741 tipc_node_remove_conn(sock_net(sk), pnode, pport);
2742 sk->sk_state_change(sk);
2743 return;
2744 }
2745 /* Prepare new probe */
2746 skb = tipc_msg_create(CONN_MANAGER, CONN_PROBE, INT_H_SIZE, 0,
2747 pnode, self, pport, oport, TIPC_OK);
2748 if (skb)
2749 __skb_queue_tail(list, skb);
2750 tsk->probe_unacked = true;
2751 sk_reset_timer(sk, &sk->sk_timer, jiffies + CONN_PROBING_INTV);
2752}
2753
67879274
TN
2754static void tipc_sk_retry_connect(struct sock *sk, struct sk_buff_head *list)
2755{
2756 struct tipc_sock *tsk = tipc_sk(sk);
2757
2758 /* Try again later if dest link is congested */
2759 if (tsk->cong_link_cnt) {
2760 sk_reset_timer(sk, &sk->sk_timer, msecs_to_jiffies(100));
2761 return;
2762 }
2763 /* Prepare SYN for retransmit */
2764 tipc_msg_skb_clone(&sk->sk_write_queue, list);
2765}
2766
31b102bb 2767static void tipc_sk_timeout(struct timer_list *t)
57289015 2768{
31b102bb
KC
2769 struct sock *sk = from_timer(sk, t, sk_timer);
2770 struct tipc_sock *tsk = tipc_sk(sk);
afe8792f
JM
2771 u32 pnode = tsk_peer_node(tsk);
2772 struct sk_buff_head list;
67879274 2773 int rc = 0;
57289015 2774
e654f9f5 2775 __skb_queue_head_init(&list);
6c9808ce 2776 bh_lock_sock(sk);
0d5fcebf
JM
2777
2778 /* Try again later if socket is busy */
2779 if (sock_owned_by_user(sk)) {
2780 sk_reset_timer(sk, &sk->sk_timer, jiffies + HZ / 20);
afe8792f 2781 bh_unlock_sock(sk);
91a4a3eb 2782 sock_put(sk);
afe8792f 2783 return;
57289015 2784 }
57289015 2785
afe8792f
JM
2786 if (sk->sk_state == TIPC_ESTABLISHED)
2787 tipc_sk_check_probing_state(sk, &list);
67879274
TN
2788 else if (sk->sk_state == TIPC_CONNECTING)
2789 tipc_sk_retry_connect(sk, &list);
afe8792f 2790
57289015 2791 bh_unlock_sock(sk);
afe8792f
JM
2792
2793 if (!skb_queue_empty(&list))
67879274 2794 rc = tipc_node_xmit(sock_net(sk), &list, pnode, tsk->portid);
afe8792f 2795
67879274
TN
2796 /* SYN messages may cause link congestion */
2797 if (rc == -ELINKCONG) {
2798 tipc_dest_push(&tsk->cong_links, pnode, 0);
2799 tsk->cong_link_cnt = 1;
2800 }
07f6c4bc 2801 sock_put(sk);
57289015
JPM
2802}
2803
301bae56 2804static int tipc_sk_publish(struct tipc_sock *tsk, uint scope,
0fc87aae
JPM
2805 struct tipc_name_seq const *seq)
2806{
d6fb7e9c
PB
2807 struct sock *sk = &tsk->sk;
2808 struct net *net = sock_net(sk);
0fc87aae
JPM
2809 struct publication *publ;
2810 u32 key;
2811
928df188
JM
2812 if (scope != TIPC_NODE_SCOPE)
2813 scope = TIPC_CLUSTER_SCOPE;
2814
d6fb7e9c 2815 if (tipc_sk_connected(sk))
0fc87aae 2816 return -EINVAL;
07f6c4bc
YX
2817 key = tsk->portid + tsk->pub_count + 1;
2818 if (key == tsk->portid)
0fc87aae
JPM
2819 return -EADDRINUSE;
2820
f2f9800d 2821 publ = tipc_nametbl_publish(net, seq->type, seq->lower, seq->upper,
07f6c4bc 2822 scope, tsk->portid, key);
0fc87aae
JPM
2823 if (unlikely(!publ))
2824 return -EINVAL;
2825
e50e73e1 2826 list_add(&publ->binding_sock, &tsk->publications);
301bae56
JPM
2827 tsk->pub_count++;
2828 tsk->published = 1;
0fc87aae
JPM
2829 return 0;
2830}
2831
301bae56 2832static int tipc_sk_withdraw(struct tipc_sock *tsk, uint scope,
0fc87aae
JPM
2833 struct tipc_name_seq const *seq)
2834{
f2f9800d 2835 struct net *net = sock_net(&tsk->sk);
0fc87aae
JPM
2836 struct publication *publ;
2837 struct publication *safe;
2838 int rc = -EINVAL;
2839
928df188
JM
2840 if (scope != TIPC_NODE_SCOPE)
2841 scope = TIPC_CLUSTER_SCOPE;
2842
e50e73e1 2843 list_for_each_entry_safe(publ, safe, &tsk->publications, binding_sock) {
0fc87aae
JPM
2844 if (seq) {
2845 if (publ->scope != scope)
2846 continue;
2847 if (publ->type != seq->type)
2848 continue;
2849 if (publ->lower != seq->lower)
2850 continue;
2851 if (publ->upper != seq->upper)
2852 break;
f2f9800d 2853 tipc_nametbl_withdraw(net, publ->type, publ->lower,
37922ea4 2854 publ->upper, publ->key);
0fc87aae
JPM
2855 rc = 0;
2856 break;
2857 }
f2f9800d 2858 tipc_nametbl_withdraw(net, publ->type, publ->lower,
37922ea4 2859 publ->upper, publ->key);
0fc87aae
JPM
2860 rc = 0;
2861 }
301bae56
JPM
2862 if (list_empty(&tsk->publications))
2863 tsk->published = 0;
0fc87aae
JPM
2864 return rc;
2865}
2866
5a9ee0be
JPM
2867/* tipc_sk_reinit: set non-zero address in all existing sockets
2868 * when we go from standalone to network mode.
2869 */
e05b31f4 2870void tipc_sk_reinit(struct net *net)
5a9ee0be 2871{
e05b31f4 2872 struct tipc_net *tn = net_generic(net, tipc_net_id);
40f9f439 2873 struct rhashtable_iter iter;
07f6c4bc 2874 struct tipc_sock *tsk;
5a9ee0be 2875 struct tipc_msg *msg;
5a9ee0be 2876
40f9f439
HX
2877 rhashtable_walk_enter(&tn->sk_rht, &iter);
2878
2879 do {
97a6ec4a 2880 rhashtable_walk_start(&iter);
40f9f439
HX
2881
2882 while ((tsk = rhashtable_walk_next(&iter)) && !IS_ERR(tsk)) {
15ef70e2
CW
2883 sock_hold(&tsk->sk);
2884 rhashtable_walk_stop(&iter);
2885 lock_sock(&tsk->sk);
07f6c4bc 2886 msg = &tsk->phdr;
23fd3eac
JM
2887 msg_set_prevnode(msg, tipc_own_addr(net));
2888 msg_set_orignode(msg, tipc_own_addr(net));
15ef70e2
CW
2889 release_sock(&tsk->sk);
2890 rhashtable_walk_start(&iter);
2891 sock_put(&tsk->sk);
07f6c4bc 2892 }
97a6ec4a 2893
40f9f439
HX
2894 rhashtable_walk_stop(&iter);
2895 } while (tsk == ERR_PTR(-EAGAIN));
bd583fe3
CW
2896
2897 rhashtable_walk_exit(&iter);
5a9ee0be
JPM
2898}
2899
e05b31f4 2900static struct tipc_sock *tipc_sk_lookup(struct net *net, u32 portid)
808d90f9 2901{
e05b31f4 2902 struct tipc_net *tn = net_generic(net, tipc_net_id);
07f6c4bc 2903 struct tipc_sock *tsk;
808d90f9 2904
07f6c4bc 2905 rcu_read_lock();
ab818362 2906 tsk = rhashtable_lookup(&tn->sk_rht, &portid, tsk_rht_params);
07f6c4bc
YX
2907 if (tsk)
2908 sock_hold(&tsk->sk);
2909 rcu_read_unlock();
808d90f9 2910
07f6c4bc 2911 return tsk;
808d90f9
JPM
2912}
2913
07f6c4bc 2914static int tipc_sk_insert(struct tipc_sock *tsk)
808d90f9 2915{
e05b31f4
YX
2916 struct sock *sk = &tsk->sk;
2917 struct net *net = sock_net(sk);
2918 struct tipc_net *tn = net_generic(net, tipc_net_id);
07f6c4bc
YX
2919 u32 remaining = (TIPC_MAX_PORT - TIPC_MIN_PORT) + 1;
2920 u32 portid = prandom_u32() % remaining + TIPC_MIN_PORT;
808d90f9 2921
07f6c4bc
YX
2922 while (remaining--) {
2923 portid++;
2924 if ((portid < TIPC_MIN_PORT) || (portid > TIPC_MAX_PORT))
2925 portid = TIPC_MIN_PORT;
2926 tsk->portid = portid;
2927 sock_hold(&tsk->sk);
6cca7289
HX
2928 if (!rhashtable_lookup_insert_fast(&tn->sk_rht, &tsk->node,
2929 tsk_rht_params))
07f6c4bc
YX
2930 return 0;
2931 sock_put(&tsk->sk);
808d90f9
JPM
2932 }
2933
07f6c4bc 2934 return -1;
808d90f9
JPM
2935}
2936
07f6c4bc 2937static void tipc_sk_remove(struct tipc_sock *tsk)
808d90f9 2938{
07f6c4bc 2939 struct sock *sk = &tsk->sk;
e05b31f4 2940 struct tipc_net *tn = net_generic(sock_net(sk), tipc_net_id);
808d90f9 2941
6cca7289 2942 if (!rhashtable_remove_fast(&tn->sk_rht, &tsk->node, tsk_rht_params)) {
41c6d650 2943 WARN_ON(refcount_read(&sk->sk_refcnt) == 1);
07f6c4bc 2944 __sock_put(sk);
808d90f9 2945 }
808d90f9
JPM
2946}
2947
6cca7289
HX
2948static const struct rhashtable_params tsk_rht_params = {
2949 .nelem_hint = 192,
2950 .head_offset = offsetof(struct tipc_sock, node),
2951 .key_offset = offsetof(struct tipc_sock, portid),
2952 .key_len = sizeof(u32), /* portid */
6cca7289
HX
2953 .max_size = 1048576,
2954 .min_size = 256,
b5e2c150 2955 .automatic_shrinking = true,
6cca7289
HX
2956};
2957
e05b31f4 2958int tipc_sk_rht_init(struct net *net)
808d90f9 2959{
e05b31f4 2960 struct tipc_net *tn = net_generic(net, tipc_net_id);
6cca7289
HX
2961
2962 return rhashtable_init(&tn->sk_rht, &tsk_rht_params);
808d90f9
JPM
2963}
2964
e05b31f4 2965void tipc_sk_rht_destroy(struct net *net)
808d90f9 2966{
e05b31f4
YX
2967 struct tipc_net *tn = net_generic(net, tipc_net_id);
2968
07f6c4bc
YX
2969 /* Wait for socket readers to complete */
2970 synchronize_net();
808d90f9 2971
e05b31f4 2972 rhashtable_destroy(&tn->sk_rht);
808d90f9
JPM
2973}
2974
75da2163
JM
2975static int tipc_sk_join(struct tipc_sock *tsk, struct tipc_group_req *mreq)
2976{
2977 struct net *net = sock_net(&tsk->sk);
75da2163
JM
2978 struct tipc_group *grp = tsk->group;
2979 struct tipc_msg *hdr = &tsk->phdr;
2980 struct tipc_name_seq seq;
2981 int rc;
2982
2983 if (mreq->type < TIPC_RESERVED_TYPES)
2984 return -EACCES;
232d07b7
JM
2985 if (mreq->scope > TIPC_NODE_SCOPE)
2986 return -EINVAL;
75da2163
JM
2987 if (grp)
2988 return -EACCES;
60c25306 2989 grp = tipc_group_create(net, tsk->portid, mreq, &tsk->group_is_open);
75da2163
JM
2990 if (!grp)
2991 return -ENOMEM;
2992 tsk->group = grp;
2993 msg_set_lookup_scope(hdr, mreq->scope);
2994 msg_set_nametype(hdr, mreq->type);
2995 msg_set_dest_droppable(hdr, true);
2996 seq.type = mreq->type;
2997 seq.lower = mreq->instance;
2998 seq.upper = seq.lower;
232d07b7 2999 tipc_nametbl_build_group(net, grp, mreq->type, mreq->scope);
75da2163 3000 rc = tipc_sk_publish(tsk, mreq->scope, &seq);
e233df01 3001 if (rc) {
75da2163 3002 tipc_group_delete(net, grp);
e233df01 3003 tsk->group = NULL;
febafc84 3004 return rc;
e233df01 3005 }
d12d2e12 3006 /* Eliminate any risk that a broadcast overtakes sent JOINs */
399574d4
JM
3007 tsk->mc_method.rcast = true;
3008 tsk->mc_method.mandatory = true;
d12d2e12 3009 tipc_group_join(net, grp, &tsk->sk.sk_rcvbuf);
75da2163
JM
3010 return rc;
3011}
3012
3013static int tipc_sk_leave(struct tipc_sock *tsk)
3014{
3015 struct net *net = sock_net(&tsk->sk);
3016 struct tipc_group *grp = tsk->group;
3017 struct tipc_name_seq seq;
3018 int scope;
3019
3020 if (!grp)
3021 return -EINVAL;
3022 tipc_group_self(grp, &seq, &scope);
3023 tipc_group_delete(net, grp);
3024 tsk->group = NULL;
3025 tipc_sk_withdraw(tsk, scope, &seq);
3026 return 0;
3027}
3028
b97bf3fd 3029/**
247f0f3c 3030 * tipc_setsockopt - set socket option
b97bf3fd
PL
3031 * @sock: socket structure
3032 * @lvl: option level
3033 * @opt: option identifier
3034 * @ov: pointer to new option value
3035 * @ol: length of option value
c4307285
YH
3036 *
3037 * For stream sockets only, accepts and ignores all IPPROTO_TCP options
b97bf3fd 3038 * (to ease compatibility).
c4307285 3039 *
b97bf3fd
PL
3040 * Returns 0 on success, errno otherwise
3041 */
247f0f3c
YX
3042static int tipc_setsockopt(struct socket *sock, int lvl, int opt,
3043 char __user *ov, unsigned int ol)
b97bf3fd 3044{
0c3141e9 3045 struct sock *sk = sock->sk;
58ed9442 3046 struct tipc_sock *tsk = tipc_sk(sk);
75da2163 3047 struct tipc_group_req mreq;
01fd12bb 3048 u32 value = 0;
a08ef476 3049 int res = 0;
b97bf3fd 3050
c4307285
YH
3051 if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
3052 return 0;
b97bf3fd
PL
3053 if (lvl != SOL_TIPC)
3054 return -ENOPROTOOPT;
01fd12bb
JPM
3055
3056 switch (opt) {
3057 case TIPC_IMPORTANCE:
3058 case TIPC_SRC_DROPPABLE:
3059 case TIPC_DEST_DROPPABLE:
3060 case TIPC_CONN_TIMEOUT:
c0bceb97 3061 case TIPC_NODELAY:
01fd12bb
JPM
3062 if (ol < sizeof(value))
3063 return -EINVAL;
75da2163
JM
3064 if (get_user(value, (u32 __user *)ov))
3065 return -EFAULT;
3066 break;
3067 case TIPC_GROUP_JOIN:
3068 if (ol < sizeof(mreq))
3069 return -EINVAL;
3070 if (copy_from_user(&mreq, ov, sizeof(mreq)))
3071 return -EFAULT;
01fd12bb
JPM
3072 break;
3073 default:
3074 if (ov || ol)
3075 return -EINVAL;
3076 }
b97bf3fd 3077
0c3141e9 3078 lock_sock(sk);
c4307285 3079
b97bf3fd
PL
3080 switch (opt) {
3081 case TIPC_IMPORTANCE:
301bae56 3082 res = tsk_set_importance(tsk, value);
b97bf3fd
PL
3083 break;
3084 case TIPC_SRC_DROPPABLE:
3085 if (sock->type != SOCK_STREAM)
301bae56 3086 tsk_set_unreliable(tsk, value);
c4307285 3087 else
b97bf3fd
PL
3088 res = -ENOPROTOOPT;
3089 break;
3090 case TIPC_DEST_DROPPABLE:
301bae56 3091 tsk_set_unreturnable(tsk, value);
b97bf3fd
PL
3092 break;
3093 case TIPC_CONN_TIMEOUT:
a0f40f02 3094 tipc_sk(sk)->conn_timeout = value;
b97bf3fd 3095 break;
01fd12bb
JPM
3096 case TIPC_MCAST_BROADCAST:
3097 tsk->mc_method.rcast = false;
3098 tsk->mc_method.mandatory = true;
3099 break;
3100 case TIPC_MCAST_REPLICAST:
3101 tsk->mc_method.rcast = true;
3102 tsk->mc_method.mandatory = true;
3103 break;
75da2163
JM
3104 case TIPC_GROUP_JOIN:
3105 res = tipc_sk_join(tsk, &mreq);
3106 break;
3107 case TIPC_GROUP_LEAVE:
3108 res = tipc_sk_leave(tsk);
3109 break;
c0bceb97
JM
3110 case TIPC_NODELAY:
3111 tsk->nodelay = !!value;
3112 tsk_set_nagle(tsk);
3113 break;
b97bf3fd
PL
3114 default:
3115 res = -EINVAL;
3116 }
3117
0c3141e9
AS
3118 release_sock(sk);
3119
b97bf3fd
PL
3120 return res;
3121}
3122
3123/**
247f0f3c 3124 * tipc_getsockopt - get socket option
b97bf3fd
PL
3125 * @sock: socket structure
3126 * @lvl: option level
3127 * @opt: option identifier
3128 * @ov: receptacle for option value
3129 * @ol: receptacle for length of option value
c4307285
YH
3130 *
3131 * For stream sockets only, returns 0 length result for all IPPROTO_TCP options
b97bf3fd 3132 * (to ease compatibility).
c4307285 3133 *
b97bf3fd
PL
3134 * Returns 0 on success, errno otherwise
3135 */
247f0f3c
YX
3136static int tipc_getsockopt(struct socket *sock, int lvl, int opt,
3137 char __user *ov, int __user *ol)
b97bf3fd 3138{
0c3141e9 3139 struct sock *sk = sock->sk;
58ed9442 3140 struct tipc_sock *tsk = tipc_sk(sk);
75da2163
JM
3141 struct tipc_name_seq seq;
3142 int len, scope;
b97bf3fd 3143 u32 value;
c4307285 3144 int res;
b97bf3fd 3145
c4307285
YH
3146 if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
3147 return put_user(0, ol);
b97bf3fd
PL
3148 if (lvl != SOL_TIPC)
3149 return -ENOPROTOOPT;
2db9983a
AS
3150 res = get_user(len, ol);
3151 if (res)
c4307285 3152 return res;
b97bf3fd 3153
0c3141e9 3154 lock_sock(sk);
b97bf3fd
PL
3155
3156 switch (opt) {
3157 case TIPC_IMPORTANCE:
301bae56 3158 value = tsk_importance(tsk);
b97bf3fd
PL
3159 break;
3160 case TIPC_SRC_DROPPABLE:
301bae56 3161 value = tsk_unreliable(tsk);
b97bf3fd
PL
3162 break;
3163 case TIPC_DEST_DROPPABLE:
301bae56 3164 value = tsk_unreturnable(tsk);
b97bf3fd
PL
3165 break;
3166 case TIPC_CONN_TIMEOUT:
301bae56 3167 value = tsk->conn_timeout;
0c3141e9 3168 /* no need to set "res", since already 0 at this point */
b97bf3fd 3169 break;
0e65967e 3170 case TIPC_NODE_RECVQ_DEPTH:
9da3d475 3171 value = 0; /* was tipc_queue_size, now obsolete */
6650613d 3172 break;
0e65967e 3173 case TIPC_SOCK_RECVQ_DEPTH:
6650613d 3174 value = skb_queue_len(&sk->sk_receive_queue);
3175 break;
42e5425a
TN
3176 case TIPC_SOCK_RECVQ_USED:
3177 value = sk_rmem_alloc_get(sk);
3178 break;
75da2163
JM
3179 case TIPC_GROUP_JOIN:
3180 seq.type = 0;
3181 if (tsk->group)
3182 tipc_group_self(tsk->group, &seq, &scope);
3183 value = seq.type;
3184 break;
b97bf3fd
PL
3185 default:
3186 res = -EINVAL;
3187 }
3188
0c3141e9
AS
3189 release_sock(sk);
3190
25860c3b
PG
3191 if (res)
3192 return res; /* "get" failed */
b97bf3fd 3193
25860c3b
PG
3194 if (len < sizeof(value))
3195 return -EINVAL;
3196
3197 if (copy_to_user(ov, &value, sizeof(value)))
3198 return -EFAULT;
3199
3200 return put_user(sizeof(value), ol);
b97bf3fd
PL
3201}
3202
f2f9800d 3203static int tipc_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
78acb1f9 3204{
3e5cf362
JM
3205 struct net *net = sock_net(sock->sk);
3206 struct tipc_sioc_nodeid_req nr = {0};
78acb1f9
EH
3207 struct tipc_sioc_ln_req lnr;
3208 void __user *argp = (void __user *)arg;
3209
3210 switch (cmd) {
3211 case SIOCGETLINKNAME:
3212 if (copy_from_user(&lnr, argp, sizeof(lnr)))
3213 return -EFAULT;
3e5cf362 3214 if (!tipc_node_get_linkname(net,
f2f9800d 3215 lnr.bearer_id & 0xffff, lnr.peer,
78acb1f9
EH
3216 lnr.linkname, TIPC_MAX_LINK_NAME)) {
3217 if (copy_to_user(argp, &lnr, sizeof(lnr)))
3218 return -EFAULT;
3219 return 0;
3220 }
3221 return -EADDRNOTAVAIL;
3e5cf362
JM
3222 case SIOCGETNODEID:
3223 if (copy_from_user(&nr, argp, sizeof(nr)))
3224 return -EFAULT;
3225 if (!tipc_node_get_id(net, nr.peer, nr.node_id))
3226 return -EADDRNOTAVAIL;
3227 if (copy_to_user(argp, &nr, sizeof(nr)))
3228 return -EFAULT;
3229 return 0;
78acb1f9
EH
3230 default:
3231 return -ENOIOCTLCMD;
3232 }
3233}
3234
70b03759
EH
3235static int tipc_socketpair(struct socket *sock1, struct socket *sock2)
3236{
3237 struct tipc_sock *tsk2 = tipc_sk(sock2->sk);
3238 struct tipc_sock *tsk1 = tipc_sk(sock1->sk);
66bc1e8d
EH
3239 u32 onode = tipc_own_addr(sock_net(sock1->sk));
3240
3241 tsk1->peer.family = AF_TIPC;
3242 tsk1->peer.addrtype = TIPC_ADDR_ID;
3243 tsk1->peer.scope = TIPC_NODE_SCOPE;
3244 tsk1->peer.addr.id.ref = tsk2->portid;
3245 tsk1->peer.addr.id.node = onode;
3246 tsk2->peer.family = AF_TIPC;
3247 tsk2->peer.addrtype = TIPC_ADDR_ID;
3248 tsk2->peer.scope = TIPC_NODE_SCOPE;
3249 tsk2->peer.addr.id.ref = tsk1->portid;
3250 tsk2->peer.addr.id.node = onode;
3251
3252 tipc_sk_finish_conn(tsk1, tsk2->portid, onode);
3253 tipc_sk_finish_conn(tsk2, tsk1->portid, onode);
70b03759
EH
3254 return 0;
3255}
3256
ae86b9e3
BH
3257/* Protocol switches for the various types of TIPC sockets */
3258
bca65eae 3259static const struct proto_ops msg_ops = {
0e65967e 3260 .owner = THIS_MODULE,
b97bf3fd 3261 .family = AF_TIPC,
247f0f3c
YX
3262 .release = tipc_release,
3263 .bind = tipc_bind,
3264 .connect = tipc_connect,
66bc1e8d 3265 .socketpair = tipc_socketpair,
245f3d34 3266 .accept = sock_no_accept,
247f0f3c 3267 .getname = tipc_getname,
a11e1d43 3268 .poll = tipc_poll,
78acb1f9 3269 .ioctl = tipc_ioctl,
245f3d34 3270 .listen = sock_no_listen,
247f0f3c
YX
3271 .shutdown = tipc_shutdown,
3272 .setsockopt = tipc_setsockopt,
3273 .getsockopt = tipc_getsockopt,
3274 .sendmsg = tipc_sendmsg,
3275 .recvmsg = tipc_recvmsg,
8238745a
YH
3276 .mmap = sock_no_mmap,
3277 .sendpage = sock_no_sendpage
b97bf3fd
PL
3278};
3279
bca65eae 3280static const struct proto_ops packet_ops = {
0e65967e 3281 .owner = THIS_MODULE,
b97bf3fd 3282 .family = AF_TIPC,
247f0f3c
YX
3283 .release = tipc_release,
3284 .bind = tipc_bind,
3285 .connect = tipc_connect,
70b03759 3286 .socketpair = tipc_socketpair,
247f0f3c
YX
3287 .accept = tipc_accept,
3288 .getname = tipc_getname,
a11e1d43 3289 .poll = tipc_poll,
78acb1f9 3290 .ioctl = tipc_ioctl,
247f0f3c
YX
3291 .listen = tipc_listen,
3292 .shutdown = tipc_shutdown,
3293 .setsockopt = tipc_setsockopt,
3294 .getsockopt = tipc_getsockopt,
3295 .sendmsg = tipc_send_packet,
3296 .recvmsg = tipc_recvmsg,
8238745a
YH
3297 .mmap = sock_no_mmap,
3298 .sendpage = sock_no_sendpage
b97bf3fd
PL
3299};
3300
bca65eae 3301static const struct proto_ops stream_ops = {
0e65967e 3302 .owner = THIS_MODULE,
b97bf3fd 3303 .family = AF_TIPC,
247f0f3c
YX
3304 .release = tipc_release,
3305 .bind = tipc_bind,
3306 .connect = tipc_connect,
70b03759 3307 .socketpair = tipc_socketpair,
247f0f3c
YX
3308 .accept = tipc_accept,
3309 .getname = tipc_getname,
a11e1d43 3310 .poll = tipc_poll,
78acb1f9 3311 .ioctl = tipc_ioctl,
247f0f3c
YX
3312 .listen = tipc_listen,
3313 .shutdown = tipc_shutdown,
3314 .setsockopt = tipc_setsockopt,
3315 .getsockopt = tipc_getsockopt,
365ad353 3316 .sendmsg = tipc_sendstream,
ec8a09fb 3317 .recvmsg = tipc_recvstream,
8238745a
YH
3318 .mmap = sock_no_mmap,
3319 .sendpage = sock_no_sendpage
b97bf3fd
PL
3320};
3321
bca65eae 3322static const struct net_proto_family tipc_family_ops = {
0e65967e 3323 .owner = THIS_MODULE,
b97bf3fd 3324 .family = AF_TIPC,
c5fa7b3c 3325 .create = tipc_sk_create
b97bf3fd
PL
3326};
3327
3328static struct proto tipc_proto = {
3329 .name = "TIPC",
3330 .owner = THIS_MODULE,
cc79dd1b
YX
3331 .obj_size = sizeof(struct tipc_sock),
3332 .sysctl_rmem = sysctl_tipc_rmem
b97bf3fd
PL
3333};
3334
3335/**
4323add6 3336 * tipc_socket_init - initialize TIPC socket interface
c4307285 3337 *
b97bf3fd
PL
3338 * Returns 0 on success, errno otherwise
3339 */
4323add6 3340int tipc_socket_init(void)
b97bf3fd
PL
3341{
3342 int res;
3343
c4307285 3344 res = proto_register(&tipc_proto, 1);
b97bf3fd 3345 if (res) {
2cf8aa19 3346 pr_err("Failed to register TIPC protocol type\n");
b97bf3fd
PL
3347 goto out;
3348 }
3349
3350 res = sock_register(&tipc_family_ops);
3351 if (res) {
2cf8aa19 3352 pr_err("Failed to register TIPC socket type\n");
b97bf3fd
PL
3353 proto_unregister(&tipc_proto);
3354 goto out;
3355 }
b97bf3fd
PL
3356 out:
3357 return res;
3358}
3359
3360/**
4323add6 3361 * tipc_socket_stop - stop TIPC socket interface
b97bf3fd 3362 */
4323add6 3363void tipc_socket_stop(void)
b97bf3fd 3364{
b97bf3fd
PL
3365 sock_unregister(tipc_family_ops.family);
3366 proto_unregister(&tipc_proto);
3367}
34b78a12
RA
3368
3369/* Caller should hold socket lock for the passed tipc socket. */
d8182804 3370static int __tipc_nl_add_sk_con(struct sk_buff *skb, struct tipc_sock *tsk)
34b78a12
RA
3371{
3372 u32 peer_node;
3373 u32 peer_port;
3374 struct nlattr *nest;
3375
3376 peer_node = tsk_peer_node(tsk);
3377 peer_port = tsk_peer_port(tsk);
3378
ae0be8de 3379 nest = nla_nest_start_noflag(skb, TIPC_NLA_SOCK_CON);
517ccc2a
KL
3380 if (!nest)
3381 return -EMSGSIZE;
34b78a12
RA
3382
3383 if (nla_put_u32(skb, TIPC_NLA_CON_NODE, peer_node))
3384 goto msg_full;
3385 if (nla_put_u32(skb, TIPC_NLA_CON_SOCK, peer_port))
3386 goto msg_full;
3387
3388 if (tsk->conn_type != 0) {
3389 if (nla_put_flag(skb, TIPC_NLA_CON_FLAG))
3390 goto msg_full;
3391 if (nla_put_u32(skb, TIPC_NLA_CON_TYPE, tsk->conn_type))
3392 goto msg_full;
3393 if (nla_put_u32(skb, TIPC_NLA_CON_INST, tsk->conn_instance))
3394 goto msg_full;
3395 }
3396 nla_nest_end(skb, nest);
3397
3398 return 0;
3399
3400msg_full:
3401 nla_nest_cancel(skb, nest);
3402
3403 return -EMSGSIZE;
3404}
3405
dfde331e
GM
3406static int __tipc_nl_add_sk_info(struct sk_buff *skb, struct tipc_sock
3407 *tsk)
3408{
3409 struct net *net = sock_net(skb->sk);
dfde331e
GM
3410 struct sock *sk = &tsk->sk;
3411
3412 if (nla_put_u32(skb, TIPC_NLA_SOCK_REF, tsk->portid) ||
23fd3eac 3413 nla_put_u32(skb, TIPC_NLA_SOCK_ADDR, tipc_own_addr(net)))
dfde331e
GM
3414 return -EMSGSIZE;
3415
3416 if (tipc_sk_connected(sk)) {
3417 if (__tipc_nl_add_sk_con(skb, tsk))
3418 return -EMSGSIZE;
3419 } else if (!list_empty(&tsk->publications)) {
3420 if (nla_put_flag(skb, TIPC_NLA_SOCK_HAS_PUBL))
3421 return -EMSGSIZE;
3422 }
3423 return 0;
3424}
3425
34b78a12 3426/* Caller should hold socket lock for the passed tipc socket. */
d8182804
RA
3427static int __tipc_nl_add_sk(struct sk_buff *skb, struct netlink_callback *cb,
3428 struct tipc_sock *tsk)
34b78a12 3429{
34b78a12 3430 struct nlattr *attrs;
dfde331e 3431 void *hdr;
34b78a12
RA
3432
3433 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
bfb3e5dd 3434 &tipc_genl_family, NLM_F_MULTI, TIPC_NL_SOCK_GET);
34b78a12
RA
3435 if (!hdr)
3436 goto msg_cancel;
3437
ae0be8de 3438 attrs = nla_nest_start_noflag(skb, TIPC_NLA_SOCK);
34b78a12
RA
3439 if (!attrs)
3440 goto genlmsg_cancel;
dfde331e
GM
3441
3442 if (__tipc_nl_add_sk_info(skb, tsk))
34b78a12
RA
3443 goto attr_msg_cancel;
3444
34b78a12
RA
3445 nla_nest_end(skb, attrs);
3446 genlmsg_end(skb, hdr);
3447
3448 return 0;
3449
3450attr_msg_cancel:
3451 nla_nest_cancel(skb, attrs);
3452genlmsg_cancel:
3453 genlmsg_cancel(skb, hdr);
3454msg_cancel:
3455 return -EMSGSIZE;
3456}
3457
c30b70de
GM
3458int tipc_nl_sk_walk(struct sk_buff *skb, struct netlink_callback *cb,
3459 int (*skb_handler)(struct sk_buff *skb,
3460 struct netlink_callback *cb,
3461 struct tipc_sock *tsk))
34b78a12 3462{
8f5c5fcf 3463 struct rhashtable_iter *iter = (void *)cb->args[4];
dfde331e
GM
3464 struct tipc_sock *tsk;
3465 int err;
34b78a12 3466
9a07efa9
CW
3467 rhashtable_walk_start(iter);
3468 while ((tsk = rhashtable_walk_next(iter)) != NULL) {
3469 if (IS_ERR(tsk)) {
3470 err = PTR_ERR(tsk);
3471 if (err == -EAGAIN) {
3472 err = 0;
d6e164e3
RA
3473 continue;
3474 }
9a07efa9
CW
3475 break;
3476 }
d6e164e3 3477
9a07efa9
CW
3478 sock_hold(&tsk->sk);
3479 rhashtable_walk_stop(iter);
3480 lock_sock(&tsk->sk);
3481 err = skb_handler(skb, cb, tsk);
3482 if (err) {
3483 release_sock(&tsk->sk);
3484 sock_put(&tsk->sk);
3485 goto out;
07f6c4bc 3486 }
9a07efa9
CW
3487 release_sock(&tsk->sk);
3488 rhashtable_walk_start(iter);
3489 sock_put(&tsk->sk);
34b78a12 3490 }
9a07efa9 3491 rhashtable_walk_stop(iter);
d6e164e3 3492out:
34b78a12
RA
3493 return skb->len;
3494}
c30b70de
GM
3495EXPORT_SYMBOL(tipc_nl_sk_walk);
3496
9a07efa9
CW
3497int tipc_dump_start(struct netlink_callback *cb)
3498{
8f5c5fcf
CW
3499 return __tipc_dump_start(cb, sock_net(cb->skb->sk));
3500}
3501EXPORT_SYMBOL(tipc_dump_start);
3502
3503int __tipc_dump_start(struct netlink_callback *cb, struct net *net)
3504{
3505 /* tipc_nl_name_table_dump() uses cb->args[0...3]. */
3506 struct rhashtable_iter *iter = (void *)cb->args[4];
9a07efa9
CW
3507 struct tipc_net *tn = tipc_net(net);
3508
3509 if (!iter) {
3510 iter = kmalloc(sizeof(*iter), GFP_KERNEL);
3511 if (!iter)
3512 return -ENOMEM;
3513
8f5c5fcf 3514 cb->args[4] = (long)iter;
9a07efa9
CW
3515 }
3516
3517 rhashtable_walk_enter(&tn->sk_rht, iter);
3518 return 0;
3519}
9a07efa9
CW
3520
3521int tipc_dump_done(struct netlink_callback *cb)
3522{
8f5c5fcf 3523 struct rhashtable_iter *hti = (void *)cb->args[4];
9a07efa9
CW
3524
3525 rhashtable_walk_exit(hti);
3526 kfree(hti);
3527 return 0;
3528}
3529EXPORT_SYMBOL(tipc_dump_done);
3530
e41f0548
CW
3531int tipc_sk_fill_sock_diag(struct sk_buff *skb, struct netlink_callback *cb,
3532 struct tipc_sock *tsk, u32 sk_filter_state,
c30b70de
GM
3533 u64 (*tipc_diag_gen_cookie)(struct sock *sk))
3534{
3535 struct sock *sk = &tsk->sk;
3536 struct nlattr *attrs;
3537 struct nlattr *stat;
3538
3539 /*filter response w.r.t sk_state*/
3540 if (!(sk_filter_state & (1 << sk->sk_state)))
3541 return 0;
3542
ae0be8de 3543 attrs = nla_nest_start_noflag(skb, TIPC_NLA_SOCK);
c30b70de
GM
3544 if (!attrs)
3545 goto msg_cancel;
3546
3547 if (__tipc_nl_add_sk_info(skb, tsk))
3548 goto attr_msg_cancel;
3549
3550 if (nla_put_u32(skb, TIPC_NLA_SOCK_TYPE, (u32)sk->sk_type) ||
3551 nla_put_u32(skb, TIPC_NLA_SOCK_TIPC_STATE, (u32)sk->sk_state) ||
3552 nla_put_u32(skb, TIPC_NLA_SOCK_INO, sock_i_ino(sk)) ||
3553 nla_put_u32(skb, TIPC_NLA_SOCK_UID,
e41f0548 3554 from_kuid_munged(sk_user_ns(NETLINK_CB(cb->skb).sk),
4b2e6877 3555 sock_i_uid(sk))) ||
c30b70de
GM
3556 nla_put_u64_64bit(skb, TIPC_NLA_SOCK_COOKIE,
3557 tipc_diag_gen_cookie(sk),
3558 TIPC_NLA_SOCK_PAD))
3559 goto attr_msg_cancel;
3560
ae0be8de 3561 stat = nla_nest_start_noflag(skb, TIPC_NLA_SOCK_STAT);
c30b70de
GM
3562 if (!stat)
3563 goto attr_msg_cancel;
3564
3565 if (nla_put_u32(skb, TIPC_NLA_SOCK_STAT_RCVQ,
3566 skb_queue_len(&sk->sk_receive_queue)) ||
3567 nla_put_u32(skb, TIPC_NLA_SOCK_STAT_SENDQ,
872619d8
GM
3568 skb_queue_len(&sk->sk_write_queue)) ||
3569 nla_put_u32(skb, TIPC_NLA_SOCK_STAT_DROP,
3570 atomic_read(&sk->sk_drops)))
c30b70de
GM
3571 goto stat_msg_cancel;
3572
3573 if (tsk->cong_link_cnt &&
3574 nla_put_flag(skb, TIPC_NLA_SOCK_STAT_LINK_CONG))
3575 goto stat_msg_cancel;
3576
3577 if (tsk_conn_cong(tsk) &&
3578 nla_put_flag(skb, TIPC_NLA_SOCK_STAT_CONN_CONG))
3579 goto stat_msg_cancel;
3580
3581 nla_nest_end(skb, stat);
a1be5a20
GM
3582
3583 if (tsk->group)
3584 if (tipc_group_fill_sock_diag(tsk->group, skb))
3585 goto stat_msg_cancel;
3586
c30b70de
GM
3587 nla_nest_end(skb, attrs);
3588
3589 return 0;
3590
3591stat_msg_cancel:
3592 nla_nest_cancel(skb, stat);
3593attr_msg_cancel:
3594 nla_nest_cancel(skb, attrs);
3595msg_cancel:
3596 return -EMSGSIZE;
3597}
3598EXPORT_SYMBOL(tipc_sk_fill_sock_diag);
1a1a143d 3599
dfde331e
GM
3600int tipc_nl_sk_dump(struct sk_buff *skb, struct netlink_callback *cb)
3601{
c30b70de 3602 return tipc_nl_sk_walk(skb, cb, __tipc_nl_add_sk);
dfde331e
GM
3603}
3604
1a1a143d 3605/* Caller should hold socket lock for the passed tipc socket. */
d8182804
RA
3606static int __tipc_nl_add_sk_publ(struct sk_buff *skb,
3607 struct netlink_callback *cb,
3608 struct publication *publ)
1a1a143d
RA
3609{
3610 void *hdr;
3611 struct nlattr *attrs;
3612
3613 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
bfb3e5dd 3614 &tipc_genl_family, NLM_F_MULTI, TIPC_NL_PUBL_GET);
1a1a143d
RA
3615 if (!hdr)
3616 goto msg_cancel;
3617
ae0be8de 3618 attrs = nla_nest_start_noflag(skb, TIPC_NLA_PUBL);
1a1a143d
RA
3619 if (!attrs)
3620 goto genlmsg_cancel;
3621
3622 if (nla_put_u32(skb, TIPC_NLA_PUBL_KEY, publ->key))
3623 goto attr_msg_cancel;
3624 if (nla_put_u32(skb, TIPC_NLA_PUBL_TYPE, publ->type))
3625 goto attr_msg_cancel;
3626 if (nla_put_u32(skb, TIPC_NLA_PUBL_LOWER, publ->lower))
3627 goto attr_msg_cancel;
3628 if (nla_put_u32(skb, TIPC_NLA_PUBL_UPPER, publ->upper))
3629 goto attr_msg_cancel;
3630
3631 nla_nest_end(skb, attrs);
3632 genlmsg_end(skb, hdr);
3633
3634 return 0;
3635
3636attr_msg_cancel:
3637 nla_nest_cancel(skb, attrs);
3638genlmsg_cancel:
3639 genlmsg_cancel(skb, hdr);
3640msg_cancel:
3641 return -EMSGSIZE;
3642}
3643
3644/* Caller should hold socket lock for the passed tipc socket. */
d8182804
RA
3645static int __tipc_nl_list_sk_publ(struct sk_buff *skb,
3646 struct netlink_callback *cb,
3647 struct tipc_sock *tsk, u32 *last_publ)
1a1a143d
RA
3648{
3649 int err;
3650 struct publication *p;
3651
3652 if (*last_publ) {
e50e73e1 3653 list_for_each_entry(p, &tsk->publications, binding_sock) {
1a1a143d
RA
3654 if (p->key == *last_publ)
3655 break;
3656 }
3657 if (p->key != *last_publ) {
3658 /* We never set seq or call nl_dump_check_consistent()
3659 * this means that setting prev_seq here will cause the
3660 * consistence check to fail in the netlink callback
3661 * handler. Resulting in the last NLMSG_DONE message
3662 * having the NLM_F_DUMP_INTR flag set.
3663 */
3664 cb->prev_seq = 1;
3665 *last_publ = 0;
3666 return -EPIPE;
3667 }
3668 } else {
3669 p = list_first_entry(&tsk->publications, struct publication,
e50e73e1 3670 binding_sock);
1a1a143d
RA
3671 }
3672
e50e73e1 3673 list_for_each_entry_from(p, &tsk->publications, binding_sock) {
1a1a143d
RA
3674 err = __tipc_nl_add_sk_publ(skb, cb, p);
3675 if (err) {
3676 *last_publ = p->key;
3677 return err;
3678 }
3679 }
3680 *last_publ = 0;
3681
3682 return 0;
3683}
3684
3685int tipc_nl_publ_dump(struct sk_buff *skb, struct netlink_callback *cb)
3686{
3687 int err;
07f6c4bc 3688 u32 tsk_portid = cb->args[0];
1a1a143d
RA
3689 u32 last_publ = cb->args[1];
3690 u32 done = cb->args[2];
e05b31f4 3691 struct net *net = sock_net(skb->sk);
1a1a143d
RA
3692 struct tipc_sock *tsk;
3693
07f6c4bc 3694 if (!tsk_portid) {
057af707 3695 struct nlattr **attrs = genl_dumpit_info(cb)->attrs;
1a1a143d
RA
3696 struct nlattr *sock[TIPC_NLA_SOCK_MAX + 1];
3697
45e093ae
RA
3698 if (!attrs[TIPC_NLA_SOCK])
3699 return -EINVAL;
3700
8cb08174
JB
3701 err = nla_parse_nested_deprecated(sock, TIPC_NLA_SOCK_MAX,
3702 attrs[TIPC_NLA_SOCK],
3703 tipc_nl_sock_policy, NULL);
1a1a143d
RA
3704 if (err)
3705 return err;
3706
3707 if (!sock[TIPC_NLA_SOCK_REF])
3708 return -EINVAL;
3709
07f6c4bc 3710 tsk_portid = nla_get_u32(sock[TIPC_NLA_SOCK_REF]);
1a1a143d
RA
3711 }
3712
3713 if (done)
3714 return 0;
3715
e05b31f4 3716 tsk = tipc_sk_lookup(net, tsk_portid);
1a1a143d
RA
3717 if (!tsk)
3718 return -EINVAL;
3719
3720 lock_sock(&tsk->sk);
3721 err = __tipc_nl_list_sk_publ(skb, cb, tsk, &last_publ);
3722 if (!err)
3723 done = 1;
3724 release_sock(&tsk->sk);
07f6c4bc 3725 sock_put(&tsk->sk);
1a1a143d 3726
07f6c4bc 3727 cb->args[0] = tsk_portid;
1a1a143d
RA
3728 cb->args[1] = last_publ;
3729 cb->args[2] = done;
3730
3731 return skb->len;
3732}
b4b9771b 3733
01e661eb
TL
3734/**
3735 * tipc_sk_filtering - check if a socket should be traced
3736 * @sk: the socket to be examined
3737 * @sysctl_tipc_sk_filter[]: the socket tuple for filtering,
3738 * (portid, sock type, name type, name lower, name upper)
3739 *
3740 * Returns true if the socket meets the socket tuple data
3741 * (value 0 = 'any') or when there is no tuple set (all = 0),
3742 * otherwise false
3743 */
3744bool tipc_sk_filtering(struct sock *sk)
3745{
3746 struct tipc_sock *tsk;
3747 struct publication *p;
3748 u32 _port, _sktype, _type, _lower, _upper;
3749 u32 type = 0, lower = 0, upper = 0;
3750
3751 if (!sk)
3752 return true;
3753
3754 tsk = tipc_sk(sk);
3755
3756 _port = sysctl_tipc_sk_filter[0];
3757 _sktype = sysctl_tipc_sk_filter[1];
3758 _type = sysctl_tipc_sk_filter[2];
3759 _lower = sysctl_tipc_sk_filter[3];
3760 _upper = sysctl_tipc_sk_filter[4];
3761
3762 if (!_port && !_sktype && !_type && !_lower && !_upper)
3763 return true;
3764
3765 if (_port)
3766 return (_port == tsk->portid);
3767
3768 if (_sktype && _sktype != sk->sk_type)
3769 return false;
3770
3771 if (tsk->published) {
3772 p = list_first_entry_or_null(&tsk->publications,
3773 struct publication, binding_sock);
3774 if (p) {
3775 type = p->type;
3776 lower = p->lower;
3777 upper = p->upper;
3778 }
3779 }
3780
3781 if (!tipc_sk_type_connectionless(sk)) {
3782 type = tsk->conn_type;
3783 lower = tsk->conn_instance;
3784 upper = tsk->conn_instance;
3785 }
3786
3787 if ((_type && _type != type) || (_lower && _lower != lower) ||
3788 (_upper && _upper != upper))
3789 return false;
3790
3791 return true;
3792}
3793
b4b9771b
TL
3794u32 tipc_sock_get_portid(struct sock *sk)
3795{
3796 return (sk) ? (tipc_sk(sk))->portid : 0;
3797}
3798
01e661eb
TL
3799/**
3800 * tipc_sk_overlimit1 - check if socket rx queue is about to be overloaded,
3801 * both the rcv and backlog queues are considered
3802 * @sk: tipc sk to be checked
3803 * @skb: tipc msg to be checked
3804 *
3805 * Returns true if the socket rx queue allocation is > 90%, otherwise false
3806 */
3807
3808bool tipc_sk_overlimit1(struct sock *sk, struct sk_buff *skb)
3809{
3810 atomic_t *dcnt = &tipc_sk(sk)->dupl_rcvcnt;
3811 unsigned int lim = rcvbuf_limit(sk, skb) + atomic_read(dcnt);
3812 unsigned int qsize = sk->sk_backlog.len + sk_rmem_alloc_get(sk);
3813
3814 return (qsize > lim * 90 / 100);
3815}
3816
3817/**
3818 * tipc_sk_overlimit2 - check if socket rx queue is about to be overloaded,
3819 * only the rcv queue is considered
3820 * @sk: tipc sk to be checked
3821 * @skb: tipc msg to be checked
3822 *
3823 * Returns true if the socket rx queue allocation is > 90%, otherwise false
3824 */
3825
3826bool tipc_sk_overlimit2(struct sock *sk, struct sk_buff *skb)
3827{
3828 unsigned int lim = rcvbuf_limit(sk, skb);
3829 unsigned int qsize = sk_rmem_alloc_get(sk);
3830
3831 return (qsize > lim * 90 / 100);
3832}
3833
b4b9771b
TL
3834/**
3835 * tipc_sk_dump - dump TIPC socket
3836 * @sk: tipc sk to be dumped
3837 * @dqueues: bitmask to decide if any socket queue to be dumped?
3838 * - TIPC_DUMP_NONE: don't dump socket queues
3839 * - TIPC_DUMP_SK_SNDQ: dump socket send queue
3840 * - TIPC_DUMP_SK_RCVQ: dump socket rcv queue
3841 * - TIPC_DUMP_SK_BKLGQ: dump socket backlog queue
3842 * - TIPC_DUMP_ALL: dump all the socket queues above
3843 * @buf: returned buffer of dump data in format
3844 */
3845int tipc_sk_dump(struct sock *sk, u16 dqueues, char *buf)
3846{
3847 int i = 0;
3848 size_t sz = (dqueues) ? SK_LMAX : SK_LMIN;
3849 struct tipc_sock *tsk;
3850 struct publication *p;
3851 bool tsk_connected;
3852
3853 if (!sk) {
3854 i += scnprintf(buf, sz, "sk data: (null)\n");
3855 return i;
3856 }
3857
3858 tsk = tipc_sk(sk);
3859 tsk_connected = !tipc_sk_type_connectionless(sk);
3860
3861 i += scnprintf(buf, sz, "sk data: %u", sk->sk_type);
3862 i += scnprintf(buf + i, sz - i, " %d", sk->sk_state);
3863 i += scnprintf(buf + i, sz - i, " %x", tsk_own_node(tsk));
3864 i += scnprintf(buf + i, sz - i, " %u", tsk->portid);
3865 i += scnprintf(buf + i, sz - i, " | %u", tsk_connected);
3866 if (tsk_connected) {
3867 i += scnprintf(buf + i, sz - i, " %x", tsk_peer_node(tsk));
3868 i += scnprintf(buf + i, sz - i, " %u", tsk_peer_port(tsk));
3869 i += scnprintf(buf + i, sz - i, " %u", tsk->conn_type);
3870 i += scnprintf(buf + i, sz - i, " %u", tsk->conn_instance);
3871 }
3872 i += scnprintf(buf + i, sz - i, " | %u", tsk->published);
3873 if (tsk->published) {
3874 p = list_first_entry_or_null(&tsk->publications,
3875 struct publication, binding_sock);
3876 i += scnprintf(buf + i, sz - i, " %u", (p) ? p->type : 0);
3877 i += scnprintf(buf + i, sz - i, " %u", (p) ? p->lower : 0);
3878 i += scnprintf(buf + i, sz - i, " %u", (p) ? p->upper : 0);
3879 }
3880 i += scnprintf(buf + i, sz - i, " | %u", tsk->snd_win);
3881 i += scnprintf(buf + i, sz - i, " %u", tsk->rcv_win);
3882 i += scnprintf(buf + i, sz - i, " %u", tsk->max_pkt);
3883 i += scnprintf(buf + i, sz - i, " %x", tsk->peer_caps);
3884 i += scnprintf(buf + i, sz - i, " %u", tsk->cong_link_cnt);
3885 i += scnprintf(buf + i, sz - i, " %u", tsk->snt_unacked);
3886 i += scnprintf(buf + i, sz - i, " %u", tsk->rcv_unacked);
3887 i += scnprintf(buf + i, sz - i, " %u", atomic_read(&tsk->dupl_rcvcnt));
3888 i += scnprintf(buf + i, sz - i, " %u", sk->sk_shutdown);
3889 i += scnprintf(buf + i, sz - i, " | %d", sk_wmem_alloc_get(sk));
3890 i += scnprintf(buf + i, sz - i, " %d", sk->sk_sndbuf);
3891 i += scnprintf(buf + i, sz - i, " | %d", sk_rmem_alloc_get(sk));
3892 i += scnprintf(buf + i, sz - i, " %d", sk->sk_rcvbuf);
70c26558 3893 i += scnprintf(buf + i, sz - i, " | %d\n", READ_ONCE(sk->sk_backlog.len));
b4b9771b
TL
3894
3895 if (dqueues & TIPC_DUMP_SK_SNDQ) {
3896 i += scnprintf(buf + i, sz - i, "sk_write_queue: ");
3897 i += tipc_list_dump(&sk->sk_write_queue, false, buf + i);
3898 }
3899
3900 if (dqueues & TIPC_DUMP_SK_RCVQ) {
3901 i += scnprintf(buf + i, sz - i, "sk_receive_queue: ");
3902 i += tipc_list_dump(&sk->sk_receive_queue, false, buf + i);
3903 }
3904
3905 if (dqueues & TIPC_DUMP_SK_BKLGQ) {
3906 i += scnprintf(buf + i, sz - i, "sk_backlog:\n head ");
3907 i += tipc_skb_dump(sk->sk_backlog.head, false, buf + i);
3908 if (sk->sk_backlog.tail != sk->sk_backlog.head) {
3909 i += scnprintf(buf + i, sz - i, " tail ");
3910 i += tipc_skb_dump(sk->sk_backlog.tail, false,
3911 buf + i);
3912 }
3913 }
3914
3915 return i;
3916}