blob: d76c171f7b7e414b44a6143d5e8aad627cdee56d [file] [log] [blame]
Per Lidenb97bf3f2006-01-02 19:04:38 +01001/*
Jon Paul Maloy02c00c22014-06-09 11:08:18 -05002 * net/tipc/socket.c: TIPC socket API
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09003 *
Jon Paul Maloy3c724ac2015-02-05 08:36:43 -05004 * Copyright (c) 2001-2007, 2012-2015, Ericsson AB
Ying Xuec5fa7b32013-06-17 10:54:39 -04005 * Copyright (c) 2004-2008, 2010-2013, Wind River Systems
Per Lidenb97bf3f2006-01-02 19:04:38 +01006 * All rights reserved.
7 *
Per Liden9ea1fd32006-01-11 13:30:43 +01008 * Redistribution and use in source and binary forms, with or without
Per Lidenb97bf3f2006-01-02 19:04:38 +01009 * modification, are permitted provided that the following conditions are met:
10 *
Per Liden9ea1fd32006-01-11 13:30:43 +010011 * 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.
Per Lidenb97bf3f2006-01-02 19:04:38 +010019 *
Per Liden9ea1fd32006-01-11 13:30:43 +010020 * 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
Per Lidenb97bf3f2006-01-02 19:04:38 +010034 * POSSIBILITY OF SUCH DAMAGE.
35 */
36
Ying Xue07f6c4b2015-01-07 13:41:58 +080037#include <linux/rhashtable.h>
38#include <linux/jhash.h>
Per Lidenb97bf3f2006-01-02 19:04:38 +010039#include "core.h"
Jon Paul Maloye2dafe82014-06-25 20:41:37 -050040#include "name_table.h"
Erik Hugne78acb1f2014-04-24 16:26:47 +020041#include "node.h"
Jon Paul Maloye2dafe82014-06-25 20:41:37 -050042#include "link.h"
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -040043#include "config.h"
Jon Paul Maloyc637c102015-02-05 08:36:41 -050044#include "name_distr.h"
Jon Paul Maloy2e84c602014-08-22 18:09:18 -040045#include "socket.h"
Erik Hugne2cf8aa12012-06-29 00:16:37 -040046
Ying Xue07f6c4b2015-01-07 13:41:58 +080047#define SS_LISTENING -1 /* socket is listening */
48#define SS_READY -2 /* socket is connectionless */
Per Lidenb97bf3f2006-01-02 19:04:38 +010049
Ying Xue07f6c4b2015-01-07 13:41:58 +080050#define CONN_TIMEOUT_DEFAULT 8000 /* default connect timeout = 8s */
Ying Xue2f55c432015-01-09 15:27:00 +080051#define CONN_PROBING_INTERVAL msecs_to_jiffies(3600000) /* [ms] => 1 h */
Ying Xue07f6c4b2015-01-07 13:41:58 +080052#define TIPC_FWD_MSG 1
53#define TIPC_CONN_OK 0
54#define TIPC_CONN_PROBING 1
55#define TIPC_MAX_PORT 0xffffffff
56#define TIPC_MIN_PORT 1
Jon Paul Maloy301bae52014-08-22 18:09:20 -040057
58/**
59 * struct tipc_sock - TIPC socket structure
60 * @sk: socket - interacts with 'port' and with user via the socket API
61 * @connected: non-zero if port is currently connected to a peer port
62 * @conn_type: TIPC type used when connection was established
63 * @conn_instance: TIPC instance used when connection was established
64 * @published: non-zero if port has one or more associated names
65 * @max_pkt: maximum packet size "hint" used when building messages sent by port
Ying Xue07f6c4b2015-01-07 13:41:58 +080066 * @portid: unique port identity in TIPC socket hash table
Jon Paul Maloy301bae52014-08-22 18:09:20 -040067 * @phdr: preformatted message header used when sending messages
68 * @port_list: adjacent ports in TIPC's global list of ports
69 * @publications: list of publications for port
70 * @pub_count: total # of publications port has made during its lifetime
71 * @probing_state:
Ying Xue2f55c432015-01-09 15:27:00 +080072 * @probing_intv:
Jon Paul Maloy301bae52014-08-22 18:09:20 -040073 * @conn_timeout: the time we can wait for an unresponded setup request
74 * @dupl_rcvcnt: number of bytes counted twice, in both backlog and rcv queue
75 * @link_cong: non-zero if owner must sleep because of link congestion
76 * @sent_unacked: # messages sent by socket, and not yet acked by peer
77 * @rcv_unacked: # messages read by user, but not yet acked back to peer
Ying Xue07f6c4b2015-01-07 13:41:58 +080078 * @node: hash table node
79 * @rcu: rcu struct for tipc_sock
Jon Paul Maloy301bae52014-08-22 18:09:20 -040080 */
81struct tipc_sock {
82 struct sock sk;
83 int connected;
84 u32 conn_type;
85 u32 conn_instance;
86 int published;
87 u32 max_pkt;
Ying Xue07f6c4b2015-01-07 13:41:58 +080088 u32 portid;
Jon Paul Maloy301bae52014-08-22 18:09:20 -040089 struct tipc_msg phdr;
90 struct list_head sock_list;
91 struct list_head publications;
92 u32 pub_count;
93 u32 probing_state;
Ying Xue2f55c432015-01-09 15:27:00 +080094 unsigned long probing_intv;
Jon Paul Maloy301bae52014-08-22 18:09:20 -040095 uint conn_timeout;
96 atomic_t dupl_rcvcnt;
97 bool link_cong;
98 uint sent_unacked;
99 uint rcv_unacked;
Ying Xue07f6c4b2015-01-07 13:41:58 +0800100 struct rhash_head node;
101 struct rcu_head rcu;
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400102};
Per Lidenb97bf3f2006-01-02 19:04:38 +0100103
Jon Paul Maloy4f4482d2014-05-14 05:39:09 -0400104static int tipc_backlog_rcv(struct sock *sk, struct sk_buff *skb);
David S. Miller676d2362014-04-11 16:15:36 -0400105static void tipc_data_ready(struct sock *sk);
Ying Xuef288bef2012-08-21 11:16:57 +0800106static void tipc_write_space(struct sock *sk);
Ying Xue247f0f32014-02-18 16:06:46 +0800107static int tipc_release(struct socket *sock);
108static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags);
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400109static int tipc_wait_for_sndmsg(struct socket *sock, long *timeo_p);
Ying Xuef2f2a962015-01-09 15:27:02 +0800110static void tipc_sk_timeout(unsigned long data);
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400111static int tipc_sk_publish(struct tipc_sock *tsk, uint scope,
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400112 struct tipc_name_seq const *seq);
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400113static int tipc_sk_withdraw(struct tipc_sock *tsk, uint scope,
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400114 struct tipc_name_seq const *seq);
Ying Xuee05b31f2015-01-09 15:27:08 +0800115static struct tipc_sock *tipc_sk_lookup(struct net *net, u32 portid);
Ying Xue07f6c4b2015-01-07 13:41:58 +0800116static int tipc_sk_insert(struct tipc_sock *tsk);
117static void tipc_sk_remove(struct tipc_sock *tsk);
Per Lidenb97bf3f2006-01-02 19:04:38 +0100118
Florian Westphalbca65ea2008-02-07 18:18:01 -0800119static const struct proto_ops packet_ops;
120static const struct proto_ops stream_ops;
121static const struct proto_ops msg_ops;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100122
123static struct proto tipc_proto;
Ying Xuec5fa7b32013-06-17 10:54:39 -0400124static struct proto tipc_proto_kern;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100125
Richard Alpe1a1a1432014-11-20 10:29:11 +0100126static const struct nla_policy tipc_nl_sock_policy[TIPC_NLA_SOCK_MAX + 1] = {
127 [TIPC_NLA_SOCK_UNSPEC] = { .type = NLA_UNSPEC },
128 [TIPC_NLA_SOCK_ADDR] = { .type = NLA_U32 },
129 [TIPC_NLA_SOCK_REF] = { .type = NLA_U32 },
130 [TIPC_NLA_SOCK_CON] = { .type = NLA_NESTED },
131 [TIPC_NLA_SOCK_HAS_PUBL] = { .type = NLA_FLAG }
132};
133
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900134/*
Allan Stephens0c3141e2008-04-15 00:22:02 -0700135 * Revised TIPC socket locking policy:
136 *
137 * Most socket operations take the standard socket lock when they start
138 * and hold it until they finish (or until they need to sleep). Acquiring
139 * this lock grants the owner exclusive access to the fields of the socket
140 * data structures, with the exception of the backlog queue. A few socket
141 * operations can be done without taking the socket lock because they only
142 * read socket information that never changes during the life of the socket.
143 *
144 * Socket operations may acquire the lock for the associated TIPC port if they
145 * need to perform an operation on the port. If any routine needs to acquire
146 * both the socket lock and the port lock it must take the socket lock first
147 * to avoid the risk of deadlock.
148 *
149 * The dispatcher handling incoming messages cannot grab the socket lock in
150 * the standard fashion, since invoked it runs at the BH level and cannot block.
151 * Instead, it checks to see if the socket lock is currently owned by someone,
152 * and either handles the message itself or adds it to the socket's backlog
153 * queue; in the latter case the queued message is processed once the process
154 * owning the socket lock releases it.
155 *
156 * NOTE: Releasing the socket lock while an operation is sleeping overcomes
157 * the problem of a blocked socket operation preventing any other operations
158 * from occurring. However, applications must be careful if they have
159 * multiple threads trying to send (or receive) on the same socket, as these
160 * operations might interfere with each other. For example, doing a connect
161 * and a receive at the same time might allow the receive to consume the
162 * ACK message meant for the connect. While additional work could be done
163 * to try and overcome this, it doesn't seem to be worthwhile at the present.
164 *
165 * NOTE: Releasing the socket lock while an operation is sleeping also ensures
166 * that another operation that must be performed in a non-blocking manner is
167 * not delayed for very long because the lock has already been taken.
168 *
169 * NOTE: This code assumes that certain fields of a port/socket pair are
170 * constant over its lifetime; such fields can be examined without taking
171 * the socket lock and/or port lock, and do not need to be re-read even
172 * after resuming processing after waiting. These fields include:
173 * - socket type
174 * - pointer to socket sk structure (aka tipc_sock structure)
175 * - pointer to port structure
176 * - port reference
Per Lidenb97bf3f2006-01-02 19:04:38 +0100177 */
Per Lidenb97bf3f2006-01-02 19:04:38 +0100178
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500179static u32 tsk_own_node(struct tipc_sock *tsk)
180{
181 return msg_prevnode(&tsk->phdr);
182}
183
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400184static u32 tsk_peer_node(struct tipc_sock *tsk)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400185{
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400186 return msg_destnode(&tsk->phdr);
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400187}
188
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400189static u32 tsk_peer_port(struct tipc_sock *tsk)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400190{
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400191 return msg_destport(&tsk->phdr);
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400192}
193
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400194static bool tsk_unreliable(struct tipc_sock *tsk)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400195{
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400196 return msg_src_droppable(&tsk->phdr) != 0;
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400197}
198
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400199static void tsk_set_unreliable(struct tipc_sock *tsk, bool unreliable)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400200{
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400201 msg_set_src_droppable(&tsk->phdr, unreliable ? 1 : 0);
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400202}
203
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400204static bool tsk_unreturnable(struct tipc_sock *tsk)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400205{
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400206 return msg_dest_droppable(&tsk->phdr) != 0;
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400207}
208
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400209static void tsk_set_unreturnable(struct tipc_sock *tsk, bool unreturnable)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400210{
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400211 msg_set_dest_droppable(&tsk->phdr, unreturnable ? 1 : 0);
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400212}
213
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400214static int tsk_importance(struct tipc_sock *tsk)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400215{
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400216 return msg_importance(&tsk->phdr);
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400217}
218
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400219static int tsk_set_importance(struct tipc_sock *tsk, int imp)
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400220{
221 if (imp > TIPC_CRITICAL_IMPORTANCE)
222 return -EINVAL;
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400223 msg_set_importance(&tsk->phdr, (u32)imp);
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400224 return 0;
225}
Jon Paul Maloy8826cde2014-03-12 11:31:09 -0400226
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400227static struct tipc_sock *tipc_sk(const struct sock *sk)
228{
229 return container_of(sk, struct tipc_sock, sk);
230}
231
232static int tsk_conn_cong(struct tipc_sock *tsk)
233{
234 return tsk->sent_unacked >= TIPC_FLOWCTRL_WIN;
235}
236
Per Lidenb97bf3f2006-01-02 19:04:38 +0100237/**
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400238 * tsk_advance_rx_queue - discard first buffer in socket receive queue
Allan Stephens0c3141e2008-04-15 00:22:02 -0700239 *
240 * Caller must hold socket lock
Per Lidenb97bf3f2006-01-02 19:04:38 +0100241 */
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400242static void tsk_advance_rx_queue(struct sock *sk)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100243{
Allan Stephens5f6d9122011-11-04 13:24:29 -0400244 kfree_skb(__skb_dequeue(&sk->sk_receive_queue));
Per Lidenb97bf3f2006-01-02 19:04:38 +0100245}
246
247/**
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400248 * tsk_rej_rx_queue - reject all buffers in socket receive queue
Allan Stephens0c3141e2008-04-15 00:22:02 -0700249 *
250 * Caller must hold socket lock
251 */
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400252static void tsk_rej_rx_queue(struct sock *sk)
Allan Stephens0c3141e2008-04-15 00:22:02 -0700253{
Ying Xuea6ca1092014-11-26 11:41:55 +0800254 struct sk_buff *skb;
Jon Paul Maloy8db1bae2014-06-25 20:41:35 -0500255 u32 dnode;
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500256 u32 own_node = tsk_own_node(tipc_sk(sk));
Allan Stephens0c3141e2008-04-15 00:22:02 -0700257
Ying Xuea6ca1092014-11-26 11:41:55 +0800258 while ((skb = __skb_dequeue(&sk->sk_receive_queue))) {
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500259 if (tipc_msg_reverse(own_node, skb, &dnode, TIPC_ERR_NO_PORT))
260 tipc_link_xmit_skb(sock_net(sk), skb, dnode, 0);
Jon Paul Maloy8db1bae2014-06-25 20:41:35 -0500261 }
Allan Stephens0c3141e2008-04-15 00:22:02 -0700262}
263
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400264/* tsk_peer_msg - verify if message was sent by connected port's peer
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400265 *
266 * Handles cases where the node's network address has changed from
267 * the default of <0.0.0> to its configured setting.
268 */
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400269static bool tsk_peer_msg(struct tipc_sock *tsk, struct tipc_msg *msg)
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400270{
Ying Xue34747532015-01-09 15:27:10 +0800271 struct tipc_net *tn = net_generic(sock_net(&tsk->sk), tipc_net_id);
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400272 u32 peer_port = tsk_peer_port(tsk);
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400273 u32 orig_node;
274 u32 peer_node;
275
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400276 if (unlikely(!tsk->connected))
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400277 return false;
278
279 if (unlikely(msg_origport(msg) != peer_port))
280 return false;
281
282 orig_node = msg_orignode(msg);
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400283 peer_node = tsk_peer_node(tsk);
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400284
285 if (likely(orig_node == peer_node))
286 return true;
287
Ying Xue34747532015-01-09 15:27:10 +0800288 if (!orig_node && (peer_node == tn->own_addr))
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400289 return true;
290
Ying Xue34747532015-01-09 15:27:10 +0800291 if (!peer_node && (orig_node == tn->own_addr))
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -0400292 return true;
293
294 return false;
295}
296
Allan Stephens0c3141e2008-04-15 00:22:02 -0700297/**
Ying Xuec5fa7b32013-06-17 10:54:39 -0400298 * tipc_sk_create - create a TIPC socket
Allan Stephens0c3141e2008-04-15 00:22:02 -0700299 * @net: network namespace (must be default network)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100300 * @sock: pre-allocated socket structure
301 * @protocol: protocol indicator (must be 0)
Eric Paris3f378b62009-11-05 22:18:14 -0800302 * @kern: caused by kernel or by userspace?
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900303 *
Allan Stephens0c3141e2008-04-15 00:22:02 -0700304 * This routine creates additional data structures used by the TIPC socket,
305 * initializes them, and links them together.
Per Lidenb97bf3f2006-01-02 19:04:38 +0100306 *
307 * Returns 0 on success, errno otherwise
308 */
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400309static int tipc_sk_create(struct net *net, struct socket *sock,
310 int protocol, int kern)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100311{
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500312 struct tipc_net *tn;
Allan Stephens0c3141e2008-04-15 00:22:02 -0700313 const struct proto_ops *ops;
314 socket_state state;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100315 struct sock *sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400316 struct tipc_sock *tsk;
Jon Paul Maloy5b8fa7c2014-08-22 18:09:13 -0400317 struct tipc_msg *msg;
Allan Stephens0c3141e2008-04-15 00:22:02 -0700318
319 /* Validate arguments */
Per Lidenb97bf3f2006-01-02 19:04:38 +0100320 if (unlikely(protocol != 0))
321 return -EPROTONOSUPPORT;
322
Per Lidenb97bf3f2006-01-02 19:04:38 +0100323 switch (sock->type) {
324 case SOCK_STREAM:
Allan Stephens0c3141e2008-04-15 00:22:02 -0700325 ops = &stream_ops;
326 state = SS_UNCONNECTED;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100327 break;
328 case SOCK_SEQPACKET:
Allan Stephens0c3141e2008-04-15 00:22:02 -0700329 ops = &packet_ops;
330 state = SS_UNCONNECTED;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100331 break;
332 case SOCK_DGRAM:
Per Lidenb97bf3f2006-01-02 19:04:38 +0100333 case SOCK_RDM:
Allan Stephens0c3141e2008-04-15 00:22:02 -0700334 ops = &msg_ops;
335 state = SS_READY;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100336 break;
Allan Stephens49978652006-06-25 23:47:18 -0700337 default:
Allan Stephens49978652006-06-25 23:47:18 -0700338 return -EPROTOTYPE;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100339 }
340
Allan Stephens0c3141e2008-04-15 00:22:02 -0700341 /* Allocate socket's protocol area */
Ying Xuec5fa7b32013-06-17 10:54:39 -0400342 if (!kern)
343 sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto);
344 else
345 sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto_kern);
346
Allan Stephens0c3141e2008-04-15 00:22:02 -0700347 if (sk == NULL)
348 return -ENOMEM;
349
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400350 tsk = tipc_sk(sk);
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400351 tsk->max_pkt = MAX_PKT_DEFAULT;
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400352 INIT_LIST_HEAD(&tsk->publications);
353 msg = &tsk->phdr;
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500354 tn = net_generic(sock_net(sk), tipc_net_id);
355 tipc_msg_init(tn->own_addr, msg, TIPC_LOW_IMPORTANCE, TIPC_NAMED_MSG,
Jon Paul Maloy5b8fa7c2014-08-22 18:09:13 -0400356 NAMED_H_SIZE, 0);
Per Lidenb97bf3f2006-01-02 19:04:38 +0100357
Allan Stephens0c3141e2008-04-15 00:22:02 -0700358 /* Finish initializing socket data structures */
Allan Stephens0c3141e2008-04-15 00:22:02 -0700359 sock->ops = ops;
360 sock->state = state;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100361 sock_init_data(sock, sk);
Ying Xue07f6c4b2015-01-07 13:41:58 +0800362 if (tipc_sk_insert(tsk)) {
363 pr_warn("Socket create failed; port numbrer exhausted\n");
364 return -EINVAL;
365 }
366 msg_set_origport(msg, tsk->portid);
Ying Xue3721e9c2015-01-13 17:07:48 +0800367 setup_timer(&sk->sk_timer, tipc_sk_timeout, (unsigned long)tsk);
Jon Paul Maloy4f4482d2014-05-14 05:39:09 -0400368 sk->sk_backlog_rcv = tipc_backlog_rcv;
Ying Xuecc79dd12013-06-17 10:54:37 -0400369 sk->sk_rcvbuf = sysctl_tipc_rmem[1];
Ying Xuef288bef2012-08-21 11:16:57 +0800370 sk->sk_data_ready = tipc_data_ready;
371 sk->sk_write_space = tipc_write_space;
Jon Paul Maloy4f4482d2014-05-14 05:39:09 -0400372 tsk->conn_timeout = CONN_TIMEOUT_DEFAULT;
Jon Paul Maloy60120522014-06-25 20:41:42 -0500373 tsk->sent_unacked = 0;
Jon Paul Maloy4f4482d2014-05-14 05:39:09 -0400374 atomic_set(&tsk->dupl_rcvcnt, 0);
Allan Stephens7ef43eb2008-05-12 15:42:28 -0700375
Allan Stephens0c3141e2008-04-15 00:22:02 -0700376 if (sock->state == SS_READY) {
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400377 tsk_set_unreturnable(tsk, true);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700378 if (sock->type == SOCK_DGRAM)
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400379 tsk_set_unreliable(tsk, true);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700380 }
Per Lidenb97bf3f2006-01-02 19:04:38 +0100381 return 0;
382}
383
384/**
Ying Xuec5fa7b32013-06-17 10:54:39 -0400385 * tipc_sock_create_local - create TIPC socket from inside TIPC module
386 * @type: socket type - SOCK_RDM or SOCK_SEQPACKET
387 *
388 * We cannot use sock_creat_kern here because it bumps module user count.
389 * Since socket owner and creator is the same module we must make sure
390 * that module count remains zero for module local sockets, otherwise
391 * we cannot do rmmod.
392 *
393 * Returns 0 on success, errno otherwise
394 */
Ying Xuea62fbcc2015-01-09 15:27:11 +0800395int tipc_sock_create_local(struct net *net, int type, struct socket **res)
Ying Xuec5fa7b32013-06-17 10:54:39 -0400396{
397 int rc;
Ying Xuec5fa7b32013-06-17 10:54:39 -0400398
399 rc = sock_create_lite(AF_TIPC, type, 0, res);
400 if (rc < 0) {
401 pr_err("Failed to create kernel socket\n");
402 return rc;
403 }
Ying Xuea62fbcc2015-01-09 15:27:11 +0800404 tipc_sk_create(net, *res, 0, 1);
Ying Xuec5fa7b32013-06-17 10:54:39 -0400405
Ying Xuec5fa7b32013-06-17 10:54:39 -0400406 return 0;
407}
408
409/**
410 * tipc_sock_release_local - release socket created by tipc_sock_create_local
411 * @sock: the socket to be released.
412 *
413 * Module reference count is not incremented when such sockets are created,
414 * so we must keep it from being decremented when they are released.
415 */
416void tipc_sock_release_local(struct socket *sock)
417{
Ying Xue247f0f32014-02-18 16:06:46 +0800418 tipc_release(sock);
Ying Xuec5fa7b32013-06-17 10:54:39 -0400419 sock->ops = NULL;
420 sock_release(sock);
421}
422
423/**
424 * tipc_sock_accept_local - accept a connection on a socket created
425 * with tipc_sock_create_local. Use this function to avoid that
426 * module reference count is inadvertently incremented.
427 *
428 * @sock: the accepting socket
429 * @newsock: reference to the new socket to be created
430 * @flags: socket flags
431 */
432
433int tipc_sock_accept_local(struct socket *sock, struct socket **newsock,
Paul Gortmakerae8509c2013-06-17 10:54:47 -0400434 int flags)
Ying Xuec5fa7b32013-06-17 10:54:39 -0400435{
436 struct sock *sk = sock->sk;
437 int ret;
438
439 ret = sock_create_lite(sk->sk_family, sk->sk_type,
440 sk->sk_protocol, newsock);
441 if (ret < 0)
442 return ret;
443
Ying Xue247f0f32014-02-18 16:06:46 +0800444 ret = tipc_accept(sock, *newsock, flags);
Ying Xuec5fa7b32013-06-17 10:54:39 -0400445 if (ret < 0) {
446 sock_release(*newsock);
447 return ret;
448 }
449 (*newsock)->ops = sock->ops;
450 return ret;
451}
452
Ying Xue07f6c4b2015-01-07 13:41:58 +0800453static void tipc_sk_callback(struct rcu_head *head)
454{
455 struct tipc_sock *tsk = container_of(head, struct tipc_sock, rcu);
456
457 sock_put(&tsk->sk);
458}
459
Ying Xuec5fa7b32013-06-17 10:54:39 -0400460/**
Ying Xue247f0f32014-02-18 16:06:46 +0800461 * tipc_release - destroy a TIPC socket
Per Lidenb97bf3f2006-01-02 19:04:38 +0100462 * @sock: socket to destroy
463 *
464 * This routine cleans up any messages that are still queued on the socket.
465 * For DGRAM and RDM socket types, all queued messages are rejected.
466 * For SEQPACKET and STREAM socket types, the first message is rejected
467 * and any others are discarded. (If the first message on a STREAM socket
468 * is partially-read, it is discarded and the next one is rejected instead.)
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900469 *
Per Lidenb97bf3f2006-01-02 19:04:38 +0100470 * NOTE: Rejected messages are not necessarily returned to the sender! They
471 * are returned or discarded according to the "destination droppable" setting
472 * specified for the message by the sender.
473 *
474 * Returns 0 on success, errno otherwise
475 */
Ying Xue247f0f32014-02-18 16:06:46 +0800476static int tipc_release(struct socket *sock)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100477{
Per Lidenb97bf3f2006-01-02 19:04:38 +0100478 struct sock *sk = sock->sk;
Sasha Levin357c4772015-01-13 12:46:41 -0500479 struct net *net;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400480 struct tipc_sock *tsk;
Ying Xuea6ca1092014-11-26 11:41:55 +0800481 struct sk_buff *skb;
Ying Xuef2f2a962015-01-09 15:27:02 +0800482 u32 dnode, probing_state;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100483
Allan Stephens0c3141e2008-04-15 00:22:02 -0700484 /*
485 * Exit if socket isn't fully initialized (occurs when a failed accept()
486 * releases a pre-allocated child socket that was never used)
487 */
Allan Stephens0c3141e2008-04-15 00:22:02 -0700488 if (sk == NULL)
489 return 0;
490
Sasha Levin357c4772015-01-13 12:46:41 -0500491 net = sock_net(sk);
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400492 tsk = tipc_sk(sk);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700493 lock_sock(sk);
494
495 /*
496 * Reject all unreceived messages, except on an active connection
497 * (which disconnects locally & sends a 'FIN+' to peer)
498 */
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400499 dnode = tsk_peer_node(tsk);
Per Lidenb97bf3f2006-01-02 19:04:38 +0100500 while (sock->state != SS_DISCONNECTING) {
Ying Xuea6ca1092014-11-26 11:41:55 +0800501 skb = __skb_dequeue(&sk->sk_receive_queue);
502 if (skb == NULL)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100503 break;
Ying Xuea6ca1092014-11-26 11:41:55 +0800504 if (TIPC_SKB_CB(skb)->handle != NULL)
505 kfree_skb(skb);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700506 else {
507 if ((sock->state == SS_CONNECTING) ||
508 (sock->state == SS_CONNECTED)) {
509 sock->state = SS_DISCONNECTING;
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400510 tsk->connected = 0;
Ying Xuef2f98002015-01-09 15:27:05 +0800511 tipc_node_remove_conn(net, dnode, tsk->portid);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700512 }
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500513 if (tipc_msg_reverse(tsk_own_node(tsk), skb, &dnode,
Ying Xue34747532015-01-09 15:27:10 +0800514 TIPC_ERR_NO_PORT))
Ying Xuef2f98002015-01-09 15:27:05 +0800515 tipc_link_xmit_skb(net, skb, dnode, 0);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700516 }
Per Lidenb97bf3f2006-01-02 19:04:38 +0100517 }
518
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400519 tipc_sk_withdraw(tsk, 0, NULL);
Ying Xuef2f2a962015-01-09 15:27:02 +0800520 probing_state = tsk->probing_state;
Ying Xue3721e9c2015-01-13 17:07:48 +0800521 if (del_timer_sync(&sk->sk_timer) &&
522 probing_state != TIPC_CONN_PROBING)
Ying Xuef2f2a962015-01-09 15:27:02 +0800523 sock_put(sk);
Ying Xue07f6c4b2015-01-07 13:41:58 +0800524 tipc_sk_remove(tsk);
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400525 if (tsk->connected) {
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500526 skb = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE,
Ying Xue34747532015-01-09 15:27:10 +0800527 TIPC_CONN_MSG, SHORT_H_SIZE, 0, dnode,
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500528 tsk_own_node(tsk), tsk_peer_port(tsk),
Ying Xue07f6c4b2015-01-07 13:41:58 +0800529 tsk->portid, TIPC_ERR_NO_PORT);
Ying Xuea6ca1092014-11-26 11:41:55 +0800530 if (skb)
Ying Xuef2f98002015-01-09 15:27:05 +0800531 tipc_link_xmit_skb(net, skb, dnode, tsk->portid);
532 tipc_node_remove_conn(net, dnode, tsk->portid);
Jon Paul Maloy5b8fa7c2014-08-22 18:09:13 -0400533 }
Per Lidenb97bf3f2006-01-02 19:04:38 +0100534
Allan Stephens0c3141e2008-04-15 00:22:02 -0700535 /* Discard any remaining (connection-based) messages in receive queue */
Ying Xue57467e52013-01-20 23:30:08 +0100536 __skb_queue_purge(&sk->sk_receive_queue);
Per Lidenb97bf3f2006-01-02 19:04:38 +0100537
Allan Stephens0c3141e2008-04-15 00:22:02 -0700538 /* Reject any messages that accumulated in backlog queue */
Allan Stephens0c3141e2008-04-15 00:22:02 -0700539 sock->state = SS_DISCONNECTING;
540 release_sock(sk);
Ying Xue07f6c4b2015-01-07 13:41:58 +0800541
542 call_rcu(&tsk->rcu, tipc_sk_callback);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700543 sock->sk = NULL;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100544
Geert Uytterhoeven065d7e32014-04-06 15:56:14 +0200545 return 0;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100546}
547
548/**
Ying Xue247f0f32014-02-18 16:06:46 +0800549 * tipc_bind - associate or disassocate TIPC name(s) with a socket
Per Lidenb97bf3f2006-01-02 19:04:38 +0100550 * @sock: socket structure
551 * @uaddr: socket address describing name(s) and desired operation
552 * @uaddr_len: size of socket address data structure
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900553 *
Per Lidenb97bf3f2006-01-02 19:04:38 +0100554 * Name and name sequence binding is indicated using a positive scope value;
555 * a negative scope value unbinds the specified name. Specifying no name
556 * (i.e. a socket address length of 0) unbinds all names from the socket.
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900557 *
Per Lidenb97bf3f2006-01-02 19:04:38 +0100558 * Returns 0 on success, errno otherwise
Allan Stephens0c3141e2008-04-15 00:22:02 -0700559 *
560 * NOTE: This routine doesn't need to take the socket lock since it doesn't
561 * access any non-constant socket information.
Per Lidenb97bf3f2006-01-02 19:04:38 +0100562 */
Ying Xue247f0f32014-02-18 16:06:46 +0800563static int tipc_bind(struct socket *sock, struct sockaddr *uaddr,
564 int uaddr_len)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100565{
Ying Xue84602762013-12-27 10:18:28 +0800566 struct sock *sk = sock->sk;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100567 struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400568 struct tipc_sock *tsk = tipc_sk(sk);
Ying Xue84602762013-12-27 10:18:28 +0800569 int res = -EINVAL;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100570
Ying Xue84602762013-12-27 10:18:28 +0800571 lock_sock(sk);
572 if (unlikely(!uaddr_len)) {
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400573 res = tipc_sk_withdraw(tsk, 0, NULL);
Ying Xue84602762013-12-27 10:18:28 +0800574 goto exit;
575 }
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900576
Ying Xue84602762013-12-27 10:18:28 +0800577 if (uaddr_len < sizeof(struct sockaddr_tipc)) {
578 res = -EINVAL;
579 goto exit;
580 }
581 if (addr->family != AF_TIPC) {
582 res = -EAFNOSUPPORT;
583 goto exit;
584 }
Per Lidenb97bf3f2006-01-02 19:04:38 +0100585
Per Lidenb97bf3f2006-01-02 19:04:38 +0100586 if (addr->addrtype == TIPC_ADDR_NAME)
587 addr->addr.nameseq.upper = addr->addr.nameseq.lower;
Ying Xue84602762013-12-27 10:18:28 +0800588 else if (addr->addrtype != TIPC_ADDR_NAMESEQ) {
589 res = -EAFNOSUPPORT;
590 goto exit;
591 }
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900592
Ying Xue13a2e892013-06-17 10:54:40 -0400593 if ((addr->addr.nameseq.type < TIPC_RESERVED_TYPES) &&
Ying Xue7d0ab172013-06-17 10:54:41 -0400594 (addr->addr.nameseq.type != TIPC_TOP_SRV) &&
Ying Xue84602762013-12-27 10:18:28 +0800595 (addr->addr.nameseq.type != TIPC_CFG_SRV)) {
596 res = -EACCES;
597 goto exit;
598 }
Allan Stephensc422f1b2011-11-02 15:49:40 -0400599
Ying Xue84602762013-12-27 10:18:28 +0800600 res = (addr->scope > 0) ?
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400601 tipc_sk_publish(tsk, addr->scope, &addr->addr.nameseq) :
602 tipc_sk_withdraw(tsk, -addr->scope, &addr->addr.nameseq);
Ying Xue84602762013-12-27 10:18:28 +0800603exit:
604 release_sock(sk);
605 return res;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100606}
607
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900608/**
Ying Xue247f0f32014-02-18 16:06:46 +0800609 * tipc_getname - get port ID of socket or peer socket
Per Lidenb97bf3f2006-01-02 19:04:38 +0100610 * @sock: socket structure
611 * @uaddr: area for returned socket address
612 * @uaddr_len: area for returned length of socket address
Allan Stephens2da59912008-07-14 22:43:32 -0700613 * @peer: 0 = own ID, 1 = current peer ID, 2 = current/former peer ID
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900614 *
Per Lidenb97bf3f2006-01-02 19:04:38 +0100615 * Returns 0 on success, errno otherwise
Allan Stephens0c3141e2008-04-15 00:22:02 -0700616 *
Allan Stephens2da59912008-07-14 22:43:32 -0700617 * NOTE: This routine doesn't need to take the socket lock since it only
618 * accesses socket information that is unchanging (or which changes in
Allan Stephens0e659672010-12-31 18:59:32 +0000619 * a completely predictable manner).
Per Lidenb97bf3f2006-01-02 19:04:38 +0100620 */
Ying Xue247f0f32014-02-18 16:06:46 +0800621static int tipc_getname(struct socket *sock, struct sockaddr *uaddr,
622 int *uaddr_len, int peer)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100623{
Per Lidenb97bf3f2006-01-02 19:04:38 +0100624 struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400625 struct tipc_sock *tsk = tipc_sk(sock->sk);
Ying Xue34747532015-01-09 15:27:10 +0800626 struct tipc_net *tn = net_generic(sock_net(sock->sk), tipc_net_id);
Per Lidenb97bf3f2006-01-02 19:04:38 +0100627
Kulikov Vasiliy88f8a5e2010-10-31 07:10:32 +0000628 memset(addr, 0, sizeof(*addr));
Allan Stephens0c3141e2008-04-15 00:22:02 -0700629 if (peer) {
Allan Stephens2da59912008-07-14 22:43:32 -0700630 if ((sock->state != SS_CONNECTED) &&
631 ((peer != 2) || (sock->state != SS_DISCONNECTING)))
632 return -ENOTCONN;
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400633 addr->addr.id.ref = tsk_peer_port(tsk);
634 addr->addr.id.node = tsk_peer_node(tsk);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700635 } else {
Ying Xue07f6c4b2015-01-07 13:41:58 +0800636 addr->addr.id.ref = tsk->portid;
Ying Xue34747532015-01-09 15:27:10 +0800637 addr->addr.id.node = tn->own_addr;
Allan Stephens0c3141e2008-04-15 00:22:02 -0700638 }
Per Lidenb97bf3f2006-01-02 19:04:38 +0100639
640 *uaddr_len = sizeof(*addr);
641 addr->addrtype = TIPC_ADDR_ID;
642 addr->family = AF_TIPC;
643 addr->scope = 0;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100644 addr->addr.name.domain = 0;
645
Allan Stephens0c3141e2008-04-15 00:22:02 -0700646 return 0;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100647}
648
649/**
Ying Xue247f0f32014-02-18 16:06:46 +0800650 * tipc_poll - read and possibly block on pollmask
Per Lidenb97bf3f2006-01-02 19:04:38 +0100651 * @file: file structure associated with the socket
652 * @sock: socket for which to calculate the poll bits
653 * @wait: ???
654 *
Allan Stephens9b674e82008-03-26 16:48:21 -0700655 * Returns pollmask value
656 *
657 * COMMENTARY:
658 * It appears that the usual socket locking mechanisms are not useful here
659 * since the pollmask info is potentially out-of-date the moment this routine
660 * exits. TCP and other protocols seem to rely on higher level poll routines
661 * to handle any preventable race conditions, so TIPC will do the same ...
662 *
663 * TIPC sets the returned events as follows:
Allan Stephens9b674e82008-03-26 16:48:21 -0700664 *
Allan Stephensf662c072010-08-17 11:00:06 +0000665 * socket state flags set
666 * ------------ ---------
667 * unconnected no read flags
Erik Hugnec4fc2982012-10-16 16:47:06 +0200668 * POLLOUT if port is not congested
Allan Stephensf662c072010-08-17 11:00:06 +0000669 *
670 * connecting POLLIN/POLLRDNORM if ACK/NACK in rx queue
671 * no write flags
672 *
673 * connected POLLIN/POLLRDNORM if data in rx queue
674 * POLLOUT if port is not congested
675 *
676 * disconnecting POLLIN/POLLRDNORM/POLLHUP
677 * no write flags
678 *
679 * listening POLLIN if SYN in rx queue
680 * no write flags
681 *
682 * ready POLLIN/POLLRDNORM if data in rx queue
683 * [connectionless] POLLOUT (since port cannot be congested)
684 *
685 * IMPORTANT: The fact that a read or write operation is indicated does NOT
686 * imply that the operation will succeed, merely that it should be performed
687 * and will not block.
Per Lidenb97bf3f2006-01-02 19:04:38 +0100688 */
Ying Xue247f0f32014-02-18 16:06:46 +0800689static unsigned int tipc_poll(struct file *file, struct socket *sock,
690 poll_table *wait)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100691{
Allan Stephens9b674e82008-03-26 16:48:21 -0700692 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400693 struct tipc_sock *tsk = tipc_sk(sk);
Allan Stephensf662c072010-08-17 11:00:06 +0000694 u32 mask = 0;
Allan Stephens9b674e82008-03-26 16:48:21 -0700695
Ying Xuef288bef2012-08-21 11:16:57 +0800696 sock_poll_wait(file, sk_sleep(sk), wait);
Allan Stephens9b674e82008-03-26 16:48:21 -0700697
Allan Stephensf662c072010-08-17 11:00:06 +0000698 switch ((int)sock->state) {
Erik Hugnec4fc2982012-10-16 16:47:06 +0200699 case SS_UNCONNECTED:
Jon Paul Maloy60120522014-06-25 20:41:42 -0500700 if (!tsk->link_cong)
Erik Hugnec4fc2982012-10-16 16:47:06 +0200701 mask |= POLLOUT;
702 break;
Allan Stephensf662c072010-08-17 11:00:06 +0000703 case SS_READY:
704 case SS_CONNECTED:
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400705 if (!tsk->link_cong && !tsk_conn_cong(tsk))
Allan Stephensf662c072010-08-17 11:00:06 +0000706 mask |= POLLOUT;
707 /* fall thru' */
708 case SS_CONNECTING:
709 case SS_LISTENING:
710 if (!skb_queue_empty(&sk->sk_receive_queue))
711 mask |= (POLLIN | POLLRDNORM);
712 break;
713 case SS_DISCONNECTING:
714 mask = (POLLIN | POLLRDNORM | POLLHUP);
715 break;
716 }
Allan Stephens9b674e82008-03-26 16:48:21 -0700717
718 return mask;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100719}
720
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400721/**
722 * tipc_sendmcast - send multicast message
723 * @sock: socket structure
724 * @seq: destination address
Al Viro562640f2014-11-15 01:13:43 -0500725 * @msg: message to send
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400726 * @dsz: total length of message data
727 * @timeo: timeout to wait for wakeup
728 *
729 * Called from function tipc_sendmsg(), which has done all sanity checks
730 * Returns the number of bytes sent on success, or errno
731 */
732static int tipc_sendmcast(struct socket *sock, struct tipc_name_seq *seq,
Al Viro562640f2014-11-15 01:13:43 -0500733 struct msghdr *msg, size_t dsz, long timeo)
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400734{
735 struct sock *sk = sock->sk;
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500736 struct tipc_sock *tsk = tipc_sk(sk);
Ying Xuef2f98002015-01-09 15:27:05 +0800737 struct net *net = sock_net(sk);
Jon Paul Maloyc5898632015-02-05 08:36:36 -0500738 struct tipc_msg *mhdr = &tsk->phdr;
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500739 struct sk_buff_head *pktchain = &sk->sk_write_queue;
Al Virof25dcc72014-11-28 15:52:29 -0500740 struct iov_iter save = msg->msg_iter;
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400741 uint mtu;
742 int rc;
743
744 msg_set_type(mhdr, TIPC_MCAST_MSG);
745 msg_set_lookup_scope(mhdr, TIPC_CLUSTER_SCOPE);
746 msg_set_destport(mhdr, 0);
747 msg_set_destnode(mhdr, 0);
748 msg_set_nametype(mhdr, seq->type);
749 msg_set_namelower(mhdr, seq->lower);
750 msg_set_nameupper(mhdr, seq->upper);
751 msg_set_hdr_sz(mhdr, MCAST_H_SIZE);
752
753new_mtu:
754 mtu = tipc_bclink_get_mtu();
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500755 rc = tipc_msg_build(mhdr, msg, 0, dsz, mtu, pktchain);
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400756 if (unlikely(rc < 0))
757 return rc;
758
759 do {
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500760 rc = tipc_bclink_xmit(net, pktchain);
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400761 if (likely(rc >= 0)) {
762 rc = dsz;
763 break;
764 }
Al Virof25dcc72014-11-28 15:52:29 -0500765 if (rc == -EMSGSIZE) {
766 msg->msg_iter = save;
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400767 goto new_mtu;
Al Virof25dcc72014-11-28 15:52:29 -0500768 }
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400769 if (rc != -ELINKCONG)
770 break;
Jon Paul Maloy50100a52014-08-22 18:09:07 -0400771 tipc_sk(sk)->link_cong = 1;
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400772 rc = tipc_wait_for_sndmsg(sock, &timeo);
773 if (rc)
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500774 __skb_queue_purge(pktchain);
Jon Paul Maloy0abd8ff2014-07-16 20:41:01 -0400775 } while (!rc);
776 return rc;
777}
778
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500779/**
780 * tipc_sk_mcast_rcv - Deliver multicast messages to all destination sockets
781 * @arrvq: queue with arriving messages, to be cloned after destination lookup
782 * @inputq: queue with cloned messages, delivered to socket after dest lookup
783 *
784 * Multi-threaded: parallel calls with reference to same queues may occur
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400785 */
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500786void tipc_sk_mcast_rcv(struct net *net, struct sk_buff_head *arrvq,
787 struct sk_buff_head *inputq)
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400788{
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500789 struct tipc_msg *msg;
Jon Paul Maloy3c724ac2015-02-05 08:36:43 -0500790 struct tipc_plist dports;
Jon Paul Maloy3c724ac2015-02-05 08:36:43 -0500791 u32 portid;
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400792 u32 scope = TIPC_CLUSTER_SCOPE;
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500793 struct sk_buff_head tmpq;
794 uint hsz;
795 struct sk_buff *skb, *_skb;
Jon Paul Maloy3c724ac2015-02-05 08:36:43 -0500796
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500797 __skb_queue_head_init(&tmpq);
Jon Paul Maloy3c724ac2015-02-05 08:36:43 -0500798 tipc_plist_init(&dports);
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400799
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500800 skb = tipc_skb_peek(arrvq, &inputq->lock);
801 for (; skb; skb = tipc_skb_peek(arrvq, &inputq->lock)) {
802 msg = buf_msg(skb);
803 hsz = skb_headroom(skb) + msg_hdr_sz(msg);
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400804
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500805 if (in_own_node(net, msg_orignode(msg)))
806 scope = TIPC_NODE_SCOPE;
807
808 /* Create destination port list and message clones: */
809 tipc_nametbl_mc_translate(net,
810 msg_nametype(msg), msg_namelower(msg),
811 msg_nameupper(msg), scope, &dports);
812 portid = tipc_plist_pop(&dports);
813 for (; portid; portid = tipc_plist_pop(&dports)) {
814 _skb = __pskb_copy(skb, hsz, GFP_ATOMIC);
815 if (_skb) {
816 msg_set_destport(buf_msg(_skb), portid);
817 __skb_queue_tail(&tmpq, _skb);
818 continue;
819 }
820 pr_warn("Failed to clone mcast rcv buffer\n");
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400821 }
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500822 /* Append to inputq if not already done by other thread */
823 spin_lock_bh(&inputq->lock);
824 if (skb_peek(arrvq) == skb) {
825 skb_queue_splice_tail_init(&tmpq, inputq);
826 kfree_skb(__skb_dequeue(arrvq));
827 }
828 spin_unlock_bh(&inputq->lock);
829 __skb_queue_purge(&tmpq);
830 kfree_skb(skb);
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400831 }
Jon Paul Maloycb1b7282015-02-05 08:36:44 -0500832 tipc_sk_rcv(net, inputq);
Jon Paul Maloy078bec82014-07-16 20:41:00 -0400833}
834
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900835/**
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500836 * tipc_sk_proto_rcv - receive a connection mng protocol message
837 * @tsk: receiving socket
Jon Paul Maloy1186adf2015-02-05 08:36:37 -0500838 * @skb: pointer to message buffer. Set to NULL if buffer is consumed.
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500839 */
Jon Paul Maloy1186adf2015-02-05 08:36:37 -0500840static void tipc_sk_proto_rcv(struct tipc_sock *tsk, struct sk_buff **skb)
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500841{
Jon Paul Maloy1186adf2015-02-05 08:36:37 -0500842 struct tipc_msg *msg = buf_msg(*skb);
Jon Paul Maloy60120522014-06-25 20:41:42 -0500843 int conn_cong;
Jon Paul Maloy1186adf2015-02-05 08:36:37 -0500844 u32 dnode;
845 u32 own_node = tsk_own_node(tsk);
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500846 /* Ignore if connection cannot be validated: */
Jon Paul Maloy2e84c602014-08-22 18:09:18 -0400847 if (!tsk_peer_msg(tsk, msg))
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500848 goto exit;
849
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400850 tsk->probing_state = TIPC_CONN_OK;
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500851
852 if (msg_type(msg) == CONN_ACK) {
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400853 conn_cong = tsk_conn_cong(tsk);
Jon Paul Maloy60120522014-06-25 20:41:42 -0500854 tsk->sent_unacked -= msg_msgcnt(msg);
855 if (conn_cong)
Jon Paul Maloy50100a52014-08-22 18:09:07 -0400856 tsk->sk.sk_write_space(&tsk->sk);
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500857 } else if (msg_type(msg) == CONN_PROBE) {
Jon Paul Maloy1186adf2015-02-05 08:36:37 -0500858 if (tipc_msg_reverse(own_node, *skb, &dnode, TIPC_OK)) {
859 msg_set_type(msg, CONN_PROBE_REPLY);
860 return;
861 }
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500862 }
863 /* Do nothing if msg_type() == CONN_PROBE_REPLY */
864exit:
Jon Paul Maloy1186adf2015-02-05 08:36:37 -0500865 kfree_skb(*skb);
866 *skb = NULL;
Jon Paul Maloyac0074e2014-06-25 20:41:41 -0500867}
868
Ying Xue3f405042014-01-17 09:50:05 +0800869static int tipc_wait_for_sndmsg(struct socket *sock, long *timeo_p)
870{
871 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400872 struct tipc_sock *tsk = tipc_sk(sk);
Ying Xue3f405042014-01-17 09:50:05 +0800873 DEFINE_WAIT(wait);
874 int done;
875
876 do {
877 int err = sock_error(sk);
878 if (err)
879 return err;
880 if (sock->state == SS_DISCONNECTING)
881 return -EPIPE;
882 if (!*timeo_p)
883 return -EAGAIN;
884 if (signal_pending(current))
885 return sock_intr_errno(*timeo_p);
886
887 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Jon Paul Maloy60120522014-06-25 20:41:42 -0500888 done = sk_wait_event(sk, timeo_p, !tsk->link_cong);
Ying Xue3f405042014-01-17 09:50:05 +0800889 finish_wait(sk_sleep(sk), &wait);
890 } while (!done);
891 return 0;
892}
893
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500894/**
Ying Xue247f0f32014-02-18 16:06:46 +0800895 * tipc_sendmsg - send message in connectionless manner
Allan Stephens0c3141e2008-04-15 00:22:02 -0700896 * @iocb: if NULL, indicates that socket lock is already held
Per Lidenb97bf3f2006-01-02 19:04:38 +0100897 * @sock: socket structure
898 * @m: message to send
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500899 * @dsz: amount of user data to be sent
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900900 *
Per Lidenb97bf3f2006-01-02 19:04:38 +0100901 * Message must have an destination specified explicitly.
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900902 * Used for SOCK_RDM and SOCK_DGRAM messages,
Per Lidenb97bf3f2006-01-02 19:04:38 +0100903 * and for 'SYN' messages on SOCK_SEQPACKET and SOCK_STREAM connections.
904 * (Note: 'SYN+' is prohibited on SOCK_STREAM.)
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900905 *
Per Lidenb97bf3f2006-01-02 19:04:38 +0100906 * Returns the number of bytes sent on success, or errno otherwise
907 */
Ying Xue247f0f32014-02-18 16:06:46 +0800908static int tipc_sendmsg(struct kiocb *iocb, struct socket *sock,
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500909 struct msghdr *m, size_t dsz)
Per Lidenb97bf3f2006-01-02 19:04:38 +0100910{
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500911 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
Allan Stephens0c3141e2008-04-15 00:22:02 -0700912 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -0400913 struct tipc_sock *tsk = tipc_sk(sk);
Ying Xuef2f98002015-01-09 15:27:05 +0800914 struct net *net = sock_net(sk);
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400915 struct tipc_msg *mhdr = &tsk->phdr;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500916 u32 dnode, dport;
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500917 struct sk_buff_head *pktchain = &sk->sk_write_queue;
Ying Xuea6ca1092014-11-26 11:41:55 +0800918 struct sk_buff *skb;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500919 struct tipc_name_seq *seq = &dest->addr.nameseq;
Al Virof25dcc72014-11-28 15:52:29 -0500920 struct iov_iter save;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500921 u32 mtu;
Ying Xue3f405042014-01-17 09:50:05 +0800922 long timeo;
Erik Hugne88b17b62014-12-03 14:44:44 +0100923 int rc;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100924
925 if (unlikely(!dest))
926 return -EDESTADDRREQ;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500927
Allan Stephens51f9cc12006-06-25 23:49:06 -0700928 if (unlikely((m->msg_namelen < sizeof(*dest)) ||
929 (dest->family != AF_TIPC)))
Per Lidenb97bf3f2006-01-02 19:04:38 +0100930 return -EINVAL;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500931
932 if (dsz > TIPC_MAX_USER_MSG_SIZE)
Allan Stephensc29c3f72010-04-20 17:58:24 -0400933 return -EMSGSIZE;
Per Lidenb97bf3f2006-01-02 19:04:38 +0100934
Allan Stephens0c3141e2008-04-15 00:22:02 -0700935 if (iocb)
936 lock_sock(sk);
937
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500938 if (unlikely(sock->state != SS_READY)) {
Allan Stephens0c3141e2008-04-15 00:22:02 -0700939 if (sock->state == SS_LISTENING) {
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500940 rc = -EPIPE;
Allan Stephens0c3141e2008-04-15 00:22:02 -0700941 goto exit;
Allan Stephens33880072006-06-25 23:44:57 -0700942 }
Allan Stephens0c3141e2008-04-15 00:22:02 -0700943 if (sock->state != SS_UNCONNECTED) {
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500944 rc = -EISCONN;
Allan Stephens0c3141e2008-04-15 00:22:02 -0700945 goto exit;
946 }
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400947 if (tsk->published) {
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500948 rc = -EOPNOTSUPP;
Allan Stephens0c3141e2008-04-15 00:22:02 -0700949 goto exit;
950 }
951 if (dest->addrtype == TIPC_ADDR_NAME) {
Jon Paul Maloy301bae52014-08-22 18:09:20 -0400952 tsk->conn_type = dest->addr.name.name.type;
953 tsk->conn_instance = dest->addr.name.name.instance;
Allan Stephens0c3141e2008-04-15 00:22:02 -0700954 }
Per Lidenb97bf3f2006-01-02 19:04:38 +0100955 }
956
Ying Xue3f405042014-01-17 09:50:05 +0800957 timeo = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500958
959 if (dest->addrtype == TIPC_ADDR_MCAST) {
Al Viro562640f2014-11-15 01:13:43 -0500960 rc = tipc_sendmcast(sock, seq, m, dsz, timeo);
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500961 goto exit;
962 } else if (dest->addrtype == TIPC_ADDR_NAME) {
963 u32 type = dest->addr.name.name.type;
964 u32 inst = dest->addr.name.name.instance;
965 u32 domain = dest->addr.name.domain;
966
967 dnode = domain;
968 msg_set_type(mhdr, TIPC_NAMED_MSG);
969 msg_set_hdr_sz(mhdr, NAMED_H_SIZE);
970 msg_set_nametype(mhdr, type);
971 msg_set_nameinst(mhdr, inst);
972 msg_set_lookup_scope(mhdr, tipc_addr_scope(domain));
Ying Xue4ac1c8d2015-01-09 15:27:09 +0800973 dport = tipc_nametbl_translate(net, type, inst, &dnode);
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500974 msg_set_destnode(mhdr, dnode);
975 msg_set_destport(mhdr, dport);
976 if (unlikely(!dport && !dnode)) {
977 rc = -EHOSTUNREACH;
978 goto exit;
979 }
980 } else if (dest->addrtype == TIPC_ADDR_ID) {
981 dnode = dest->addr.id.node;
982 msg_set_type(mhdr, TIPC_DIRECT_MSG);
983 msg_set_lookup_scope(mhdr, 0);
984 msg_set_destnode(mhdr, dnode);
985 msg_set_destport(mhdr, dest->addr.id.ref);
986 msg_set_hdr_sz(mhdr, BASIC_H_SIZE);
987 }
988
Al Virof25dcc72014-11-28 15:52:29 -0500989 save = m->msg_iter;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500990new_mtu:
Ying Xuef2f98002015-01-09 15:27:05 +0800991 mtu = tipc_node_get_mtu(net, dnode, tsk->portid);
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500992 rc = tipc_msg_build(mhdr, m, 0, dsz, mtu, pktchain);
Jon Paul Maloye2dafe82014-06-25 20:41:37 -0500993 if (rc < 0)
994 goto exit;
995
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +0900996 do {
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500997 skb = skb_peek(pktchain);
Ying Xuea6ca1092014-11-26 11:41:55 +0800998 TIPC_SKB_CB(skb)->wakeup_pending = tsk->link_cong;
Jon Paul Maloy94153e32015-02-05 08:36:40 -0500999 rc = tipc_link_xmit(net, pktchain, dnode, tsk->portid);
Jon Paul Maloye2dafe82014-06-25 20:41:37 -05001000 if (likely(rc >= 0)) {
1001 if (sock->state != SS_READY)
Allan Stephens0c3141e2008-04-15 00:22:02 -07001002 sock->state = SS_CONNECTING;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -05001003 rc = dsz;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001004 break;
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001005 }
Al Virof25dcc72014-11-28 15:52:29 -05001006 if (rc == -EMSGSIZE) {
1007 m->msg_iter = save;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -05001008 goto new_mtu;
Al Virof25dcc72014-11-28 15:52:29 -05001009 }
Jon Paul Maloye2dafe82014-06-25 20:41:37 -05001010 if (rc != -ELINKCONG)
Allan Stephens0c3141e2008-04-15 00:22:02 -07001011 break;
Jon Paul Maloy50100a52014-08-22 18:09:07 -04001012 tsk->link_cong = 1;
Jon Paul Maloye2dafe82014-06-25 20:41:37 -05001013 rc = tipc_wait_for_sndmsg(sock, &timeo);
Erik Hugne70452dc2014-07-06 20:38:50 -04001014 if (rc)
Jon Paul Maloy94153e32015-02-05 08:36:40 -05001015 __skb_queue_purge(pktchain);
Jon Paul Maloye2dafe82014-06-25 20:41:37 -05001016 } while (!rc);
Allan Stephens0c3141e2008-04-15 00:22:02 -07001017exit:
1018 if (iocb)
1019 release_sock(sk);
Jon Paul Maloye2dafe82014-06-25 20:41:37 -05001020
1021 return rc;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001022}
1023
Ying Xue391a6dd2014-01-17 09:50:06 +08001024static int tipc_wait_for_sndpkt(struct socket *sock, long *timeo_p)
1025{
1026 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04001027 struct tipc_sock *tsk = tipc_sk(sk);
Ying Xue391a6dd2014-01-17 09:50:06 +08001028 DEFINE_WAIT(wait);
1029 int done;
1030
1031 do {
1032 int err = sock_error(sk);
1033 if (err)
1034 return err;
1035 if (sock->state == SS_DISCONNECTING)
1036 return -EPIPE;
1037 else if (sock->state != SS_CONNECTED)
1038 return -ENOTCONN;
1039 if (!*timeo_p)
1040 return -EAGAIN;
1041 if (signal_pending(current))
1042 return sock_intr_errno(*timeo_p);
1043
1044 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
1045 done = sk_wait_event(sk, timeo_p,
Jon Paul Maloy60120522014-06-25 20:41:42 -05001046 (!tsk->link_cong &&
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001047 !tsk_conn_cong(tsk)) ||
1048 !tsk->connected);
Ying Xue391a6dd2014-01-17 09:50:06 +08001049 finish_wait(sk_sleep(sk), &wait);
1050 } while (!done);
1051 return 0;
1052}
1053
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001054/**
Ying Xue247f0f32014-02-18 16:06:46 +08001055 * tipc_send_stream - send stream-oriented data
Per Lidenb97bf3f2006-01-02 19:04:38 +01001056 * @iocb: (unused)
1057 * @sock: socket structure
1058 * @m: data to send
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001059 * @dsz: total length of data to be transmitted
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001060 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01001061 * Used for SOCK_STREAM data.
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001062 *
1063 * Returns the number of bytes sent on success (or partial success),
Allan Stephens1303e8f2006-06-25 23:46:50 -07001064 * or errno if no data sent
Per Lidenb97bf3f2006-01-02 19:04:38 +01001065 */
Ying Xue247f0f32014-02-18 16:06:46 +08001066static int tipc_send_stream(struct kiocb *iocb, struct socket *sock,
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001067 struct msghdr *m, size_t dsz)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001068{
Allan Stephens0c3141e2008-04-15 00:22:02 -07001069 struct sock *sk = sock->sk;
Ying Xuef2f98002015-01-09 15:27:05 +08001070 struct net *net = sock_net(sk);
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04001071 struct tipc_sock *tsk = tipc_sk(sk);
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001072 struct tipc_msg *mhdr = &tsk->phdr;
Jon Paul Maloy94153e32015-02-05 08:36:40 -05001073 struct sk_buff_head *pktchain = &sk->sk_write_queue;
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001074 DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
Ying Xue07f6c4b2015-01-07 13:41:58 +08001075 u32 portid = tsk->portid;
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001076 int rc = -EINVAL;
1077 long timeo;
1078 u32 dnode;
1079 uint mtu, send, sent = 0;
Al Virof25dcc72014-11-28 15:52:29 -05001080 struct iov_iter save;
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001081
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001082 /* Handle implied connection establishment */
1083 if (unlikely(dest)) {
1084 rc = tipc_sendmsg(iocb, sock, m, dsz);
1085 if (dsz && (dsz == rc))
Jon Paul Maloy60120522014-06-25 20:41:42 -05001086 tsk->sent_unacked = 1;
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001087 return rc;
1088 }
1089 if (dsz > (uint)INT_MAX)
1090 return -EMSGSIZE;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001091
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001092 if (iocb)
1093 lock_sock(sk);
1094
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001095 if (unlikely(sock->state != SS_CONNECTED)) {
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001096 if (sock->state == SS_DISCONNECTING)
1097 rc = -EPIPE;
wangweidongb0555972013-12-27 10:09:39 +08001098 else
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001099 rc = -ENOTCONN;
wangweidong3b8401f2013-12-12 09:36:40 +08001100 goto exit;
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001101 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01001102
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001103 timeo = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001104 dnode = tsk_peer_node(tsk);
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001105
1106next:
Al Virof25dcc72014-11-28 15:52:29 -05001107 save = m->msg_iter;
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001108 mtu = tsk->max_pkt;
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001109 send = min_t(uint, dsz - sent, TIPC_MAX_USER_MSG_SIZE);
Jon Paul Maloy94153e32015-02-05 08:36:40 -05001110 rc = tipc_msg_build(mhdr, m, sent, send, mtu, pktchain);
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001111 if (unlikely(rc < 0))
Allan Stephens0c3141e2008-04-15 00:22:02 -07001112 goto exit;
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001113 do {
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001114 if (likely(!tsk_conn_cong(tsk))) {
Jon Paul Maloy94153e32015-02-05 08:36:40 -05001115 rc = tipc_link_xmit(net, pktchain, dnode, portid);
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001116 if (likely(!rc)) {
Jon Paul Maloy60120522014-06-25 20:41:42 -05001117 tsk->sent_unacked++;
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001118 sent += send;
1119 if (sent == dsz)
1120 break;
1121 goto next;
Allan Stephens1303e8f2006-06-25 23:46:50 -07001122 }
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001123 if (rc == -EMSGSIZE) {
Ying Xuef2f98002015-01-09 15:27:05 +08001124 tsk->max_pkt = tipc_node_get_mtu(net, dnode,
1125 portid);
Al Virof25dcc72014-11-28 15:52:29 -05001126 m->msg_iter = save;
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001127 goto next;
1128 }
1129 if (rc != -ELINKCONG)
1130 break;
Jon Paul Maloy50100a52014-08-22 18:09:07 -04001131 tsk->link_cong = 1;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001132 }
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001133 rc = tipc_wait_for_sndpkt(sock, &timeo);
Erik Hugne70452dc2014-07-06 20:38:50 -04001134 if (rc)
Jon Paul Maloy94153e32015-02-05 08:36:40 -05001135 __skb_queue_purge(pktchain);
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001136 } while (!rc);
Allan Stephens0c3141e2008-04-15 00:22:02 -07001137exit:
Jon Paul Maloy4ccfe5e2014-06-25 20:41:38 -05001138 if (iocb)
1139 release_sock(sk);
1140 return sent ? sent : rc;
1141}
1142
1143/**
1144 * tipc_send_packet - send a connection-oriented message
1145 * @iocb: if NULL, indicates that socket lock is already held
1146 * @sock: socket structure
1147 * @m: message to send
1148 * @dsz: length of data to be transmitted
1149 *
1150 * Used for SOCK_SEQPACKET messages.
1151 *
1152 * Returns the number of bytes sent on success, or errno otherwise
1153 */
1154static int tipc_send_packet(struct kiocb *iocb, struct socket *sock,
1155 struct msghdr *m, size_t dsz)
1156{
1157 if (dsz > TIPC_MAX_USER_MSG_SIZE)
1158 return -EMSGSIZE;
1159
1160 return tipc_send_stream(iocb, sock, m, dsz);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001161}
1162
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001163/* tipc_sk_finish_conn - complete the setup of a connection
Per Lidenb97bf3f2006-01-02 19:04:38 +01001164 */
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001165static void tipc_sk_finish_conn(struct tipc_sock *tsk, u32 peer_port,
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001166 u32 peer_node)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001167{
Ying Xue3721e9c2015-01-13 17:07:48 +08001168 struct sock *sk = &tsk->sk;
1169 struct net *net = sock_net(sk);
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001170 struct tipc_msg *msg = &tsk->phdr;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001171
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001172 msg_set_destnode(msg, peer_node);
1173 msg_set_destport(msg, peer_port);
1174 msg_set_type(msg, TIPC_CONN_MSG);
1175 msg_set_lookup_scope(msg, 0);
1176 msg_set_hdr_sz(msg, SHORT_H_SIZE);
Jon Paul Maloyf9fef182014-03-12 11:31:08 -04001177
Ying Xue2f55c432015-01-09 15:27:00 +08001178 tsk->probing_intv = CONN_PROBING_INTERVAL;
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001179 tsk->probing_state = TIPC_CONN_OK;
1180 tsk->connected = 1;
Ying Xue3721e9c2015-01-13 17:07:48 +08001181 sk_reset_timer(sk, &sk->sk_timer, jiffies + tsk->probing_intv);
Ying Xuef2f98002015-01-09 15:27:05 +08001182 tipc_node_add_conn(net, peer_node, tsk->portid, peer_port);
1183 tsk->max_pkt = tipc_node_get_mtu(net, peer_node, tsk->portid);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001184}
1185
1186/**
1187 * set_orig_addr - capture sender's address for received message
1188 * @m: descriptor for message info
1189 * @msg: received message header
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001190 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01001191 * Note: Address is not captured if not requested by receiver.
1192 */
Sam Ravnborg05790c62006-03-20 22:37:04 -08001193static void set_orig_addr(struct msghdr *m, struct tipc_msg *msg)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001194{
Steffen Hurrle342dfc32014-01-17 22:53:15 +01001195 DECLARE_SOCKADDR(struct sockaddr_tipc *, addr, m->msg_name);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001196
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001197 if (addr) {
Per Lidenb97bf3f2006-01-02 19:04:38 +01001198 addr->family = AF_TIPC;
1199 addr->addrtype = TIPC_ADDR_ID;
Mathias Krause60085c32013-04-07 01:52:00 +00001200 memset(&addr->addr, 0, sizeof(addr->addr));
Per Lidenb97bf3f2006-01-02 19:04:38 +01001201 addr->addr.id.ref = msg_origport(msg);
1202 addr->addr.id.node = msg_orignode(msg);
Allan Stephens0e659672010-12-31 18:59:32 +00001203 addr->addr.name.domain = 0; /* could leave uninitialized */
1204 addr->scope = 0; /* could leave uninitialized */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001205 m->msg_namelen = sizeof(struct sockaddr_tipc);
1206 }
1207}
1208
1209/**
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001210 * tipc_sk_anc_data_recv - optionally capture ancillary data for received message
Per Lidenb97bf3f2006-01-02 19:04:38 +01001211 * @m: descriptor for message info
1212 * @msg: received message header
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001213 * @tsk: TIPC port associated with message
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001214 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01001215 * Note: Ancillary data is not captured if not requested by receiver.
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001216 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01001217 * Returns 0 if successful, otherwise errno
1218 */
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001219static int tipc_sk_anc_data_recv(struct msghdr *m, struct tipc_msg *msg,
1220 struct tipc_sock *tsk)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001221{
1222 u32 anc_data[3];
1223 u32 err;
1224 u32 dest_type;
Allan Stephens3546c752006-06-25 23:45:24 -07001225 int has_name;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001226 int res;
1227
1228 if (likely(m->msg_controllen == 0))
1229 return 0;
1230
1231 /* Optionally capture errored message object(s) */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001232 err = msg ? msg_errcode(msg) : 0;
1233 if (unlikely(err)) {
1234 anc_data[0] = err;
1235 anc_data[1] = msg_data_sz(msg);
Allan Stephens2db99832010-12-31 18:59:33 +00001236 res = put_cmsg(m, SOL_TIPC, TIPC_ERRINFO, 8, anc_data);
1237 if (res)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001238 return res;
Allan Stephens2db99832010-12-31 18:59:33 +00001239 if (anc_data[1]) {
1240 res = put_cmsg(m, SOL_TIPC, TIPC_RETDATA, anc_data[1],
1241 msg_data(msg));
1242 if (res)
1243 return res;
1244 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01001245 }
1246
1247 /* Optionally capture message destination object */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001248 dest_type = msg ? msg_type(msg) : TIPC_DIRECT_MSG;
1249 switch (dest_type) {
1250 case TIPC_NAMED_MSG:
Allan Stephens3546c752006-06-25 23:45:24 -07001251 has_name = 1;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001252 anc_data[0] = msg_nametype(msg);
1253 anc_data[1] = msg_namelower(msg);
1254 anc_data[2] = msg_namelower(msg);
1255 break;
1256 case TIPC_MCAST_MSG:
Allan Stephens3546c752006-06-25 23:45:24 -07001257 has_name = 1;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001258 anc_data[0] = msg_nametype(msg);
1259 anc_data[1] = msg_namelower(msg);
1260 anc_data[2] = msg_nameupper(msg);
1261 break;
1262 case TIPC_CONN_MSG:
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001263 has_name = (tsk->conn_type != 0);
1264 anc_data[0] = tsk->conn_type;
1265 anc_data[1] = tsk->conn_instance;
1266 anc_data[2] = tsk->conn_instance;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001267 break;
1268 default:
Allan Stephens3546c752006-06-25 23:45:24 -07001269 has_name = 0;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001270 }
Allan Stephens2db99832010-12-31 18:59:33 +00001271 if (has_name) {
1272 res = put_cmsg(m, SOL_TIPC, TIPC_DESTNAME, 12, anc_data);
1273 if (res)
1274 return res;
1275 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01001276
1277 return 0;
1278}
1279
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001280static void tipc_sk_send_ack(struct tipc_sock *tsk, uint ack)
Jon Paul Maloy739f5e42014-08-22 18:09:12 -04001281{
Ying Xuef2f98002015-01-09 15:27:05 +08001282 struct net *net = sock_net(&tsk->sk);
Ying Xuea6ca1092014-11-26 11:41:55 +08001283 struct sk_buff *skb = NULL;
Jon Paul Maloy739f5e42014-08-22 18:09:12 -04001284 struct tipc_msg *msg;
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001285 u32 peer_port = tsk_peer_port(tsk);
1286 u32 dnode = tsk_peer_node(tsk);
Jon Paul Maloy739f5e42014-08-22 18:09:12 -04001287
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001288 if (!tsk->connected)
Jon Paul Maloy739f5e42014-08-22 18:09:12 -04001289 return;
Jon Paul Maloyc5898632015-02-05 08:36:36 -05001290 skb = tipc_msg_create(CONN_MANAGER, CONN_ACK, INT_H_SIZE, 0,
1291 dnode, tsk_own_node(tsk), peer_port,
1292 tsk->portid, TIPC_OK);
Ying Xuea6ca1092014-11-26 11:41:55 +08001293 if (!skb)
Jon Paul Maloy739f5e42014-08-22 18:09:12 -04001294 return;
Ying Xuea6ca1092014-11-26 11:41:55 +08001295 msg = buf_msg(skb);
Jon Paul Maloy739f5e42014-08-22 18:09:12 -04001296 msg_set_msgcnt(msg, ack);
Ying Xuef2f98002015-01-09 15:27:05 +08001297 tipc_link_xmit_skb(net, skb, dnode, msg_link_selector(msg));
Jon Paul Maloy739f5e42014-08-22 18:09:12 -04001298}
1299
Arnaldo Carvalho de Melo85d3fc92014-05-23 15:55:12 -04001300static int tipc_wait_for_rcvmsg(struct socket *sock, long *timeop)
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001301{
1302 struct sock *sk = sock->sk;
1303 DEFINE_WAIT(wait);
Arnaldo Carvalho de Melo85d3fc92014-05-23 15:55:12 -04001304 long timeo = *timeop;
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001305 int err;
1306
1307 for (;;) {
1308 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
Ying Xuefe8e4642014-03-06 14:40:18 +01001309 if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001310 if (sock->state == SS_DISCONNECTING) {
1311 err = -ENOTCONN;
1312 break;
1313 }
1314 release_sock(sk);
1315 timeo = schedule_timeout(timeo);
1316 lock_sock(sk);
1317 }
1318 err = 0;
1319 if (!skb_queue_empty(&sk->sk_receive_queue))
1320 break;
1321 err = sock_intr_errno(timeo);
1322 if (signal_pending(current))
1323 break;
1324 err = -EAGAIN;
1325 if (!timeo)
1326 break;
1327 }
1328 finish_wait(sk_sleep(sk), &wait);
Arnaldo Carvalho de Melo85d3fc92014-05-23 15:55:12 -04001329 *timeop = timeo;
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001330 return err;
1331}
1332
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001333/**
Ying Xue247f0f32014-02-18 16:06:46 +08001334 * tipc_recvmsg - receive packet-oriented message
Per Lidenb97bf3f2006-01-02 19:04:38 +01001335 * @iocb: (unused)
1336 * @m: descriptor for message info
1337 * @buf_len: total size of user buffer area
1338 * @flags: receive flags
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001339 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01001340 * Used for SOCK_DGRAM, SOCK_RDM, and SOCK_SEQPACKET messages.
1341 * If the complete message doesn't fit in user area, truncate it.
1342 *
1343 * Returns size of returned message data, errno otherwise
1344 */
Ying Xue247f0f32014-02-18 16:06:46 +08001345static int tipc_recvmsg(struct kiocb *iocb, struct socket *sock,
1346 struct msghdr *m, size_t buf_len, int flags)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001347{
Allan Stephens0c3141e2008-04-15 00:22:02 -07001348 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04001349 struct tipc_sock *tsk = tipc_sk(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001350 struct sk_buff *buf;
1351 struct tipc_msg *msg;
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001352 long timeo;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001353 unsigned int sz;
1354 u32 err;
1355 int res;
1356
Allan Stephens0c3141e2008-04-15 00:22:02 -07001357 /* Catch invalid receive requests */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001358 if (unlikely(!buf_len))
1359 return -EINVAL;
1360
Allan Stephens0c3141e2008-04-15 00:22:02 -07001361 lock_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001362
Allan Stephens0c3141e2008-04-15 00:22:02 -07001363 if (unlikely(sock->state == SS_UNCONNECTED)) {
Per Lidenb97bf3f2006-01-02 19:04:38 +01001364 res = -ENOTCONN;
1365 goto exit;
1366 }
1367
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001368 timeo = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
Allan Stephens0c3141e2008-04-15 00:22:02 -07001369restart:
Per Lidenb97bf3f2006-01-02 19:04:38 +01001370
Allan Stephens0c3141e2008-04-15 00:22:02 -07001371 /* Look for a message in receive queue; wait if necessary */
Arnaldo Carvalho de Melo85d3fc92014-05-23 15:55:12 -04001372 res = tipc_wait_for_rcvmsg(sock, &timeo);
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001373 if (res)
1374 goto exit;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001375
1376 /* Look at first message in receive queue */
Allan Stephens0c3141e2008-04-15 00:22:02 -07001377 buf = skb_peek(&sk->sk_receive_queue);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001378 msg = buf_msg(buf);
1379 sz = msg_data_sz(msg);
1380 err = msg_errcode(msg);
1381
Per Lidenb97bf3f2006-01-02 19:04:38 +01001382 /* Discard an empty non-errored message & try again */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001383 if ((!sz) && (!err)) {
Jon Paul Maloy2e84c602014-08-22 18:09:18 -04001384 tsk_advance_rx_queue(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001385 goto restart;
1386 }
1387
1388 /* Capture sender's address (optional) */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001389 set_orig_addr(m, msg);
1390
1391 /* Capture ancillary data (optional) */
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001392 res = tipc_sk_anc_data_recv(m, msg, tsk);
Allan Stephens0c3141e2008-04-15 00:22:02 -07001393 if (res)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001394 goto exit;
1395
1396 /* Capture message data (if valid) & compute return value (always) */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001397 if (!err) {
1398 if (unlikely(buf_len < sz)) {
1399 sz = buf_len;
1400 m->msg_flags |= MSG_TRUNC;
1401 }
David S. Miller51f3d022014-11-05 16:46:40 -05001402 res = skb_copy_datagram_msg(buf, msg_hdr_sz(msg), m, sz);
Allan Stephens0232fd02011-02-21 09:45:40 -05001403 if (res)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001404 goto exit;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001405 res = sz;
1406 } else {
1407 if ((sock->state == SS_READY) ||
1408 ((err == TIPC_CONN_SHUTDOWN) || m->msg_control))
1409 res = 0;
1410 else
1411 res = -ECONNRESET;
1412 }
1413
1414 /* Consume received message (optional) */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001415 if (likely(!(flags & MSG_PEEK))) {
Allan Stephens99009802008-04-15 00:06:12 -07001416 if ((sock->state != SS_READY) &&
Jon Paul Maloy60120522014-06-25 20:41:42 -05001417 (++tsk->rcv_unacked >= TIPC_CONNACK_INTV)) {
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001418 tipc_sk_send_ack(tsk, tsk->rcv_unacked);
Jon Paul Maloy60120522014-06-25 20:41:42 -05001419 tsk->rcv_unacked = 0;
1420 }
Jon Paul Maloy2e84c602014-08-22 18:09:18 -04001421 tsk_advance_rx_queue(sk);
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001422 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01001423exit:
Allan Stephens0c3141e2008-04-15 00:22:02 -07001424 release_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001425 return res;
1426}
1427
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001428/**
Ying Xue247f0f32014-02-18 16:06:46 +08001429 * tipc_recv_stream - receive stream-oriented data
Per Lidenb97bf3f2006-01-02 19:04:38 +01001430 * @iocb: (unused)
1431 * @m: descriptor for message info
1432 * @buf_len: total size of user buffer area
1433 * @flags: receive flags
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001434 *
1435 * Used for SOCK_STREAM messages only. If not enough data is available
Per Lidenb97bf3f2006-01-02 19:04:38 +01001436 * will optionally wait for more; never truncates data.
1437 *
1438 * Returns size of returned message data, errno otherwise
1439 */
Ying Xue247f0f32014-02-18 16:06:46 +08001440static int tipc_recv_stream(struct kiocb *iocb, struct socket *sock,
1441 struct msghdr *m, size_t buf_len, int flags)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001442{
Allan Stephens0c3141e2008-04-15 00:22:02 -07001443 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04001444 struct tipc_sock *tsk = tipc_sk(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001445 struct sk_buff *buf;
1446 struct tipc_msg *msg;
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001447 long timeo;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001448 unsigned int sz;
Florian Westphal3720d402010-08-17 11:00:04 +00001449 int sz_to_copy, target, needed;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001450 int sz_copied = 0;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001451 u32 err;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001452 int res = 0;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001453
1454 /* Catch invalid receive attempts */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001455 if (unlikely(!buf_len))
1456 return -EINVAL;
1457
Allan Stephens0c3141e2008-04-15 00:22:02 -07001458 lock_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001459
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001460 if (unlikely(sock->state == SS_UNCONNECTED)) {
Per Lidenb97bf3f2006-01-02 19:04:38 +01001461 res = -ENOTCONN;
1462 goto exit;
1463 }
1464
Florian Westphal3720d402010-08-17 11:00:04 +00001465 target = sock_rcvlowat(sk, flags & MSG_WAITALL, buf_len);
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001466 timeo = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
Paul Gortmaker617d3c72012-04-30 15:29:02 -04001467
Allan Stephens0c3141e2008-04-15 00:22:02 -07001468restart:
Allan Stephens0c3141e2008-04-15 00:22:02 -07001469 /* Look for a message in receive queue; wait if necessary */
Arnaldo Carvalho de Melo85d3fc92014-05-23 15:55:12 -04001470 res = tipc_wait_for_rcvmsg(sock, &timeo);
Ying Xue9bbb4ec2014-01-17 09:50:07 +08001471 if (res)
1472 goto exit;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001473
1474 /* Look at first message in receive queue */
Allan Stephens0c3141e2008-04-15 00:22:02 -07001475 buf = skb_peek(&sk->sk_receive_queue);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001476 msg = buf_msg(buf);
1477 sz = msg_data_sz(msg);
1478 err = msg_errcode(msg);
1479
1480 /* Discard an empty non-errored message & try again */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001481 if ((!sz) && (!err)) {
Jon Paul Maloy2e84c602014-08-22 18:09:18 -04001482 tsk_advance_rx_queue(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001483 goto restart;
1484 }
1485
1486 /* Optionally capture sender's address & ancillary data of first msg */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001487 if (sz_copied == 0) {
1488 set_orig_addr(m, msg);
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001489 res = tipc_sk_anc_data_recv(m, msg, tsk);
Allan Stephens0c3141e2008-04-15 00:22:02 -07001490 if (res)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001491 goto exit;
1492 }
1493
1494 /* Capture message data (if valid) & compute return value (always) */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001495 if (!err) {
Allan Stephens0232fd02011-02-21 09:45:40 -05001496 u32 offset = (u32)(unsigned long)(TIPC_SKB_CB(buf)->handle);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001497
Allan Stephens0232fd02011-02-21 09:45:40 -05001498 sz -= offset;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001499 needed = (buf_len - sz_copied);
1500 sz_to_copy = (sz <= needed) ? sz : needed;
Allan Stephens0232fd02011-02-21 09:45:40 -05001501
David S. Miller51f3d022014-11-05 16:46:40 -05001502 res = skb_copy_datagram_msg(buf, msg_hdr_sz(msg) + offset,
1503 m, sz_to_copy);
Allan Stephens0232fd02011-02-21 09:45:40 -05001504 if (res)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001505 goto exit;
Allan Stephens0232fd02011-02-21 09:45:40 -05001506
Per Lidenb97bf3f2006-01-02 19:04:38 +01001507 sz_copied += sz_to_copy;
1508
1509 if (sz_to_copy < sz) {
1510 if (!(flags & MSG_PEEK))
Allan Stephens0232fd02011-02-21 09:45:40 -05001511 TIPC_SKB_CB(buf)->handle =
1512 (void *)(unsigned long)(offset + sz_to_copy);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001513 goto exit;
1514 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01001515 } else {
1516 if (sz_copied != 0)
1517 goto exit; /* can't add error msg to valid data */
1518
1519 if ((err == TIPC_CONN_SHUTDOWN) || m->msg_control)
1520 res = 0;
1521 else
1522 res = -ECONNRESET;
1523 }
1524
1525 /* Consume received message (optional) */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001526 if (likely(!(flags & MSG_PEEK))) {
Jon Paul Maloy60120522014-06-25 20:41:42 -05001527 if (unlikely(++tsk->rcv_unacked >= TIPC_CONNACK_INTV)) {
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001528 tipc_sk_send_ack(tsk, tsk->rcv_unacked);
Jon Paul Maloy60120522014-06-25 20:41:42 -05001529 tsk->rcv_unacked = 0;
1530 }
Jon Paul Maloy2e84c602014-08-22 18:09:18 -04001531 tsk_advance_rx_queue(sk);
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001532 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01001533
1534 /* Loop around if more data is required */
Joe Perchesf64f9e72009-11-29 16:55:45 -08001535 if ((sz_copied < buf_len) && /* didn't get all requested data */
1536 (!skb_queue_empty(&sk->sk_receive_queue) ||
Florian Westphal3720d402010-08-17 11:00:04 +00001537 (sz_copied < target)) && /* and more is ready or required */
Joe Perchesf64f9e72009-11-29 16:55:45 -08001538 (!(flags & MSG_PEEK)) && /* and aren't just peeking at data */
1539 (!err)) /* and haven't reached a FIN */
Per Lidenb97bf3f2006-01-02 19:04:38 +01001540 goto restart;
1541
1542exit:
Allan Stephens0c3141e2008-04-15 00:22:02 -07001543 release_sock(sk);
Allan Stephensa3b0a5a2006-06-25 23:48:22 -07001544 return sz_copied ? sz_copied : res;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001545}
1546
1547/**
Ying Xuef288bef2012-08-21 11:16:57 +08001548 * tipc_write_space - wake up thread if port congestion is released
1549 * @sk: socket
1550 */
1551static void tipc_write_space(struct sock *sk)
1552{
1553 struct socket_wq *wq;
1554
1555 rcu_read_lock();
1556 wq = rcu_dereference(sk->sk_wq);
1557 if (wq_has_sleeper(wq))
1558 wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
1559 POLLWRNORM | POLLWRBAND);
1560 rcu_read_unlock();
1561}
1562
1563/**
1564 * tipc_data_ready - wake up threads to indicate messages have been received
1565 * @sk: socket
1566 * @len: the length of messages
1567 */
David S. Miller676d2362014-04-11 16:15:36 -04001568static void tipc_data_ready(struct sock *sk)
Ying Xuef288bef2012-08-21 11:16:57 +08001569{
1570 struct socket_wq *wq;
1571
1572 rcu_read_lock();
1573 wq = rcu_dereference(sk->sk_wq);
1574 if (wq_has_sleeper(wq))
1575 wake_up_interruptible_sync_poll(&wq->wait, POLLIN |
1576 POLLRDNORM | POLLRDBAND);
1577 rcu_read_unlock();
1578}
1579
1580/**
Ying Xue7e6c1312012-11-29 18:39:14 -05001581 * filter_connect - Handle all incoming messages for a connection-based socket
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04001582 * @tsk: TIPC socket
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001583 * @skb: pointer to message buffer. Set to NULL if buffer is consumed
Ying Xue7e6c1312012-11-29 18:39:14 -05001584 *
stephen hemmingerb2ad5e52014-10-29 22:58:51 -07001585 * Returns 0 (TIPC_OK) if everything ok, -TIPC_ERR_NO_PORT otherwise
Ying Xue7e6c1312012-11-29 18:39:14 -05001586 */
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001587static int filter_connect(struct tipc_sock *tsk, struct sk_buff **skb)
Ying Xue7e6c1312012-11-29 18:39:14 -05001588{
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04001589 struct sock *sk = &tsk->sk;
Ying Xuef2f98002015-01-09 15:27:05 +08001590 struct net *net = sock_net(sk);
Jon Paul Maloy8826cde2014-03-12 11:31:09 -04001591 struct socket *sock = sk->sk_socket;
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001592 struct tipc_msg *msg = buf_msg(*skb);
Jon Paul Maloye4de5fa2014-06-25 20:41:31 -05001593 int retval = -TIPC_ERR_NO_PORT;
Ying Xue7e6c1312012-11-29 18:39:14 -05001594
1595 if (msg_mcast(msg))
1596 return retval;
1597
1598 switch ((int)sock->state) {
1599 case SS_CONNECTED:
1600 /* Accept only connection-based messages sent by peer */
Jon Paul Maloy2e84c602014-08-22 18:09:18 -04001601 if (tsk_peer_msg(tsk, msg)) {
Ying Xue7e6c1312012-11-29 18:39:14 -05001602 if (unlikely(msg_errcode(msg))) {
1603 sock->state = SS_DISCONNECTING;
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001604 tsk->connected = 0;
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001605 /* let timer expire on it's own */
Ying Xuef2f98002015-01-09 15:27:05 +08001606 tipc_node_remove_conn(net, tsk_peer_node(tsk),
Ying Xue07f6c4b2015-01-07 13:41:58 +08001607 tsk->portid);
Ying Xue7e6c1312012-11-29 18:39:14 -05001608 }
1609 retval = TIPC_OK;
1610 }
1611 break;
1612 case SS_CONNECTING:
1613 /* Accept only ACK or NACK message */
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001614
1615 if (unlikely(!msg_connected(msg)))
1616 break;
1617
Ying Xue584d24b2012-11-29 18:51:19 -05001618 if (unlikely(msg_errcode(msg))) {
1619 sock->state = SS_DISCONNECTING;
Erik Hugne2c8d8512013-08-28 09:29:58 +02001620 sk->sk_err = ECONNREFUSED;
Ying Xue7e6c1312012-11-29 18:39:14 -05001621 retval = TIPC_OK;
Ying Xue584d24b2012-11-29 18:51:19 -05001622 break;
1623 }
1624
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001625 if (unlikely(msg_importance(msg) > TIPC_CRITICAL_IMPORTANCE)) {
Ying Xue584d24b2012-11-29 18:51:19 -05001626 sock->state = SS_DISCONNECTING;
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001627 sk->sk_err = EINVAL;
Ying Xue584d24b2012-11-29 18:51:19 -05001628 retval = TIPC_OK;
1629 break;
1630 }
1631
Jon Paul Maloy301bae52014-08-22 18:09:20 -04001632 tipc_sk_finish_conn(tsk, msg_origport(msg), msg_orignode(msg));
1633 msg_set_importance(&tsk->phdr, msg_importance(msg));
Jon Paul Maloydadebc02014-08-22 18:09:11 -04001634 sock->state = SS_CONNECTED;
1635
Ying Xue584d24b2012-11-29 18:51:19 -05001636 /* If an incoming message is an 'ACK-', it should be
1637 * discarded here because it doesn't contain useful
1638 * data. In addition, we should try to wake up
1639 * connect() routine if sleeping.
1640 */
1641 if (msg_data_sz(msg) == 0) {
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001642 kfree_skb(*skb);
1643 *skb = NULL;
Ying Xue584d24b2012-11-29 18:51:19 -05001644 if (waitqueue_active(sk_sleep(sk)))
1645 wake_up_interruptible(sk_sleep(sk));
1646 }
1647 retval = TIPC_OK;
Ying Xue7e6c1312012-11-29 18:39:14 -05001648 break;
1649 case SS_LISTENING:
1650 case SS_UNCONNECTED:
1651 /* Accept only SYN message */
1652 if (!msg_connected(msg) && !(msg_errcode(msg)))
1653 retval = TIPC_OK;
1654 break;
1655 case SS_DISCONNECTING:
1656 break;
1657 default:
1658 pr_err("Unknown socket state %u\n", sock->state);
1659 }
1660 return retval;
1661}
1662
1663/**
Ying Xueaba79f32013-01-20 23:30:09 +01001664 * rcvbuf_limit - get proper overload limit of socket receive queue
1665 * @sk: socket
1666 * @buf: message
1667 *
1668 * For all connection oriented messages, irrespective of importance,
1669 * the default overload value (i.e. 67MB) is set as limit.
1670 *
1671 * For all connectionless messages, by default new queue limits are
1672 * as belows:
1673 *
Ying Xuecc79dd12013-06-17 10:54:37 -04001674 * TIPC_LOW_IMPORTANCE (4 MB)
1675 * TIPC_MEDIUM_IMPORTANCE (8 MB)
1676 * TIPC_HIGH_IMPORTANCE (16 MB)
1677 * TIPC_CRITICAL_IMPORTANCE (32 MB)
Ying Xueaba79f32013-01-20 23:30:09 +01001678 *
1679 * Returns overload limit according to corresponding message importance
1680 */
1681static unsigned int rcvbuf_limit(struct sock *sk, struct sk_buff *buf)
1682{
1683 struct tipc_msg *msg = buf_msg(buf);
Ying Xueaba79f32013-01-20 23:30:09 +01001684
1685 if (msg_connected(msg))
wangweidong0cee6bb2013-12-12 09:36:39 +08001686 return sysctl_tipc_rmem[2];
1687
1688 return sk->sk_rcvbuf >> TIPC_CRITICAL_IMPORTANCE <<
1689 msg_importance(msg);
Ying Xueaba79f32013-01-20 23:30:09 +01001690}
1691
1692/**
Allan Stephens0c3141e2008-04-15 00:22:02 -07001693 * filter_rcv - validate incoming message
1694 * @sk: socket
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001695 * @skb: pointer to message. Set to NULL if buffer is consumed.
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001696 *
Allan Stephens0c3141e2008-04-15 00:22:02 -07001697 * Enqueues message on receive queue if acceptable; optionally handles
1698 * disconnect indication for a connected socket.
1699 *
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001700 * Called with socket lock already taken
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001701 *
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001702 * Returns 0 (TIPC_OK) if message was ok, -TIPC error code if rejected
Per Lidenb97bf3f2006-01-02 19:04:38 +01001703 */
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001704static int filter_rcv(struct sock *sk, struct sk_buff **skb)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001705{
Allan Stephens0c3141e2008-04-15 00:22:02 -07001706 struct socket *sock = sk->sk_socket;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04001707 struct tipc_sock *tsk = tipc_sk(sk);
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001708 struct tipc_msg *msg = buf_msg(*skb);
1709 unsigned int limit = rcvbuf_limit(sk, *skb);
Jon Paul Maloye4de5fa2014-06-25 20:41:31 -05001710 int rc = TIPC_OK;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001711
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001712 if (unlikely(msg_user(msg) == CONN_MANAGER)) {
1713 tipc_sk_proto_rcv(tsk, skb);
1714 return TIPC_OK;
1715 }
Jon Paul Maloyec8a2e52014-06-25 20:41:40 -05001716
Jon Paul Maloy50100a52014-08-22 18:09:07 -04001717 if (unlikely(msg_user(msg) == SOCK_WAKEUP)) {
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001718 kfree_skb(*skb);
Jon Paul Maloy50100a52014-08-22 18:09:07 -04001719 tsk->link_cong = 0;
1720 sk->sk_write_space(sk);
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001721 *skb = NULL;
Jon Paul Maloy50100a52014-08-22 18:09:07 -04001722 return TIPC_OK;
1723 }
1724
Per Lidenb97bf3f2006-01-02 19:04:38 +01001725 /* Reject message if it is wrong sort of message for socket */
Allan Stephensaad58542012-04-26 18:13:08 -04001726 if (msg_type(msg) > TIPC_DIRECT_MSG)
Jon Paul Maloye4de5fa2014-06-25 20:41:31 -05001727 return -TIPC_ERR_NO_PORT;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001728
Per Lidenb97bf3f2006-01-02 19:04:38 +01001729 if (sock->state == SS_READY) {
Allan Stephensb29f1422010-12-31 18:59:25 +00001730 if (msg_connected(msg))
Jon Paul Maloye4de5fa2014-06-25 20:41:31 -05001731 return -TIPC_ERR_NO_PORT;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001732 } else {
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001733 rc = filter_connect(tsk, skb);
1734 if (rc != TIPC_OK || !*skb)
Jon Paul Maloye4de5fa2014-06-25 20:41:31 -05001735 return rc;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001736 }
1737
1738 /* Reject message if there isn't room to queue it */
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001739 if (sk_rmem_alloc_get(sk) + (*skb)->truesize >= limit)
Jon Paul Maloye4de5fa2014-06-25 20:41:31 -05001740 return -TIPC_ERR_OVERLOAD;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001741
Ying Xueaba79f32013-01-20 23:30:09 +01001742 /* Enqueue message */
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001743 TIPC_SKB_CB(*skb)->handle = NULL;
1744 __skb_queue_tail(&sk->sk_receive_queue, *skb);
1745 skb_set_owner_r(*skb, sk);
Allan Stephens0c3141e2008-04-15 00:22:02 -07001746
David S. Miller676d2362014-04-11 16:15:36 -04001747 sk->sk_data_ready(sk);
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001748 *skb = NULL;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001749 return TIPC_OK;
1750}
1751
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001752/**
Jon Paul Maloy4f4482d2014-05-14 05:39:09 -04001753 * tipc_backlog_rcv - handle incoming message from backlog queue
Allan Stephens0c3141e2008-04-15 00:22:02 -07001754 * @sk: socket
Ying Xuea6ca1092014-11-26 11:41:55 +08001755 * @skb: message
Allan Stephens0c3141e2008-04-15 00:22:02 -07001756 *
Jon Paul Maloye3a77562015-02-05 08:36:39 -05001757 * Caller must hold socket lock
Allan Stephens0c3141e2008-04-15 00:22:02 -07001758 *
1759 * Returns 0
1760 */
Ying Xuea6ca1092014-11-26 11:41:55 +08001761static int tipc_backlog_rcv(struct sock *sk, struct sk_buff *skb)
Allan Stephens0c3141e2008-04-15 00:22:02 -07001762{
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001763 int err;
1764 atomic_t *dcnt;
1765 u32 dnode;
Jon Paul Maloy4f4482d2014-05-14 05:39:09 -04001766 struct tipc_sock *tsk = tipc_sk(sk);
Ying Xue34747532015-01-09 15:27:10 +08001767 struct net *net = sock_net(sk);
Ying Xuea6ca1092014-11-26 11:41:55 +08001768 uint truesize = skb->truesize;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001769
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001770 err = filter_rcv(sk, &skb);
1771 if (likely(!skb)) {
1772 dcnt = &tsk->dupl_rcvcnt;
1773 if (atomic_read(dcnt) < TIPC_CONN_OVERLOAD_LIMIT)
1774 atomic_add(truesize, dcnt);
Jon Paul Maloyac0074e2014-06-25 20:41:41 -05001775 return 0;
1776 }
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001777 if (!err || tipc_msg_reverse(tsk_own_node(tsk), skb, &dnode, -err))
1778 tipc_link_xmit_skb(net, skb, dnode, tsk->portid);
Allan Stephens0c3141e2008-04-15 00:22:02 -07001779 return 0;
1780}
1781
1782/**
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001783 * tipc_sk_enqueue - extract all buffers with destination 'dport' from
1784 * inputq and try adding them to socket or backlog queue
1785 * @inputq: list of incoming buffers with potentially different destinations
1786 * @sk: socket where the buffers should be enqueued
1787 * @dport: port number for the socket
1788 * @_skb: returned buffer to be forwarded or rejected, if applicable
Jon Paul Maloyd570d862015-02-05 08:36:38 -05001789 *
1790 * Caller must hold socket lock
1791 *
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001792 * Returns TIPC_OK if all buffers enqueued, otherwise -TIPC_ERR_OVERLOAD
1793 * or -TIPC_ERR_NO_PORT
Jon Paul Maloyd570d862015-02-05 08:36:38 -05001794 */
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001795static int tipc_sk_enqueue(struct sk_buff_head *inputq, struct sock *sk,
1796 u32 dport, struct sk_buff **_skb)
Jon Paul Maloyd570d862015-02-05 08:36:38 -05001797{
1798 unsigned int lim;
1799 atomic_t *dcnt;
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001800 int err;
1801 struct sk_buff *skb;
1802 unsigned long time_limit = jiffies + 2;
Jon Paul Maloyd570d862015-02-05 08:36:38 -05001803
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001804 while (skb_queue_len(inputq)) {
Jon Paul Maloy51a00da2015-02-08 11:10:50 -05001805 if (unlikely(time_after_eq(jiffies, time_limit)))
1806 return TIPC_OK;
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001807 skb = tipc_skb_dequeue(inputq, dport);
1808 if (unlikely(!skb))
1809 return TIPC_OK;
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001810 if (!sock_owned_by_user(sk)) {
1811 err = filter_rcv(sk, &skb);
1812 if (likely(!skb))
1813 continue;
1814 *_skb = skb;
1815 return err;
1816 }
1817 dcnt = &tipc_sk(sk)->dupl_rcvcnt;
1818 if (sk->sk_backlog.len)
1819 atomic_set(dcnt, 0);
1820 lim = rcvbuf_limit(sk, skb) + atomic_read(dcnt);
1821 if (likely(!sk_add_backlog(sk, skb, lim)))
1822 continue;
1823 *_skb = skb;
Jon Paul Maloyd570d862015-02-05 08:36:38 -05001824 return -TIPC_ERR_OVERLOAD;
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001825 }
Jon Paul Maloyd570d862015-02-05 08:36:38 -05001826 return TIPC_OK;
1827}
1828
1829/**
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001830 * tipc_sk_rcv - handle a chain of incoming buffers
1831 * @inputq: buffer list containing the buffers
1832 * Consumes all buffers in list until inputq is empty
1833 * Note: may be called in multiple threads referring to the same queue
1834 * Returns 0 if last buffer was accepted, otherwise -EHOSTUNREACH
1835 * Only node local calls check the return value, sending single-buffer queues
Allan Stephens0c3141e2008-04-15 00:22:02 -07001836 */
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001837int tipc_sk_rcv(struct net *net, struct sk_buff_head *inputq)
Allan Stephens0c3141e2008-04-15 00:22:02 -07001838{
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001839 u32 dnode, dport = 0;
1840 int err = -TIPC_ERR_NO_PORT;
1841 struct sk_buff *skb;
Jon Paul Maloy9816f062014-05-14 05:39:15 -04001842 struct tipc_sock *tsk;
Jon Paul Maloyc5898632015-02-05 08:36:36 -05001843 struct tipc_net *tn;
Jon Paul Maloy9816f062014-05-14 05:39:15 -04001844 struct sock *sk;
Jon Paul Maloy9816f062014-05-14 05:39:15 -04001845
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001846 while (skb_queue_len(inputq)) {
1847 skb = NULL;
1848 dport = tipc_skb_peek_port(inputq, dport);
1849 tsk = tipc_sk_lookup(net, dport);
1850 if (likely(tsk)) {
1851 sk = &tsk->sk;
1852 if (likely(spin_trylock_bh(&sk->sk_lock.slock))) {
1853 err = tipc_sk_enqueue(inputq, sk, dport, &skb);
1854 spin_unlock_bh(&sk->sk_lock.slock);
1855 dport = 0;
1856 }
1857 sock_put(sk);
1858 } else {
1859 skb = tipc_skb_dequeue(inputq, dport);
1860 }
1861 if (likely(!skb))
1862 continue;
1863 if (tipc_msg_lookup_dest(net, skb, &dnode, &err))
1864 goto xmit;
1865 if (!err) {
1866 dnode = msg_destnode(buf_msg(skb));
1867 goto xmit;
1868 }
1869 tn = net_generic(net, tipc_net_id);
1870 if (!tipc_msg_reverse(tn->own_addr, skb, &dnode, -err))
1871 continue;
Jon Paul Maloye3a77562015-02-05 08:36:39 -05001872xmit:
Jon Paul Maloyc637c102015-02-05 08:36:41 -05001873 tipc_link_xmit_skb(net, skb, dnode, dport);
1874 }
Jon Paul Maloy1186adf2015-02-05 08:36:37 -05001875 return err ? -EHOSTUNREACH : 0;
Allan Stephens0c3141e2008-04-15 00:22:02 -07001876}
1877
Ying Xue78eb3a52014-01-17 09:50:03 +08001878static int tipc_wait_for_connect(struct socket *sock, long *timeo_p)
1879{
1880 struct sock *sk = sock->sk;
1881 DEFINE_WAIT(wait);
1882 int done;
1883
1884 do {
1885 int err = sock_error(sk);
1886 if (err)
1887 return err;
1888 if (!*timeo_p)
1889 return -ETIMEDOUT;
1890 if (signal_pending(current))
1891 return sock_intr_errno(*timeo_p);
1892
1893 prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
1894 done = sk_wait_event(sk, timeo_p, sock->state != SS_CONNECTING);
1895 finish_wait(sk_sleep(sk), &wait);
1896 } while (!done);
1897 return 0;
1898}
1899
Per Lidenb97bf3f2006-01-02 19:04:38 +01001900/**
Ying Xue247f0f32014-02-18 16:06:46 +08001901 * tipc_connect - establish a connection to another TIPC port
Per Lidenb97bf3f2006-01-02 19:04:38 +01001902 * @sock: socket structure
1903 * @dest: socket address for destination port
1904 * @destlen: size of socket address data structure
Allan Stephens0c3141e2008-04-15 00:22:02 -07001905 * @flags: file-related flags associated with socket
Per Lidenb97bf3f2006-01-02 19:04:38 +01001906 *
1907 * Returns 0 on success, errno otherwise
1908 */
Ying Xue247f0f32014-02-18 16:06:46 +08001909static int tipc_connect(struct socket *sock, struct sockaddr *dest,
1910 int destlen, int flags)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001911{
Allan Stephens0c3141e2008-04-15 00:22:02 -07001912 struct sock *sk = sock->sk;
Allan Stephensb89741a2008-04-15 00:20:37 -07001913 struct sockaddr_tipc *dst = (struct sockaddr_tipc *)dest;
1914 struct msghdr m = {NULL,};
Ying Xue78eb3a52014-01-17 09:50:03 +08001915 long timeout = (flags & O_NONBLOCK) ? 0 : tipc_sk(sk)->conn_timeout;
1916 socket_state previous;
Allan Stephensb89741a2008-04-15 00:20:37 -07001917 int res;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001918
Allan Stephens0c3141e2008-04-15 00:22:02 -07001919 lock_sock(sk);
1920
Allan Stephensb89741a2008-04-15 00:20:37 -07001921 /* For now, TIPC does not allow use of connect() with DGRAM/RDM types */
Allan Stephens0c3141e2008-04-15 00:22:02 -07001922 if (sock->state == SS_READY) {
1923 res = -EOPNOTSUPP;
1924 goto exit;
1925 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01001926
Allan Stephensb89741a2008-04-15 00:20:37 -07001927 /*
1928 * Reject connection attempt using multicast address
1929 *
1930 * Note: send_msg() validates the rest of the address fields,
1931 * so there's no need to do it here
1932 */
Allan Stephens0c3141e2008-04-15 00:22:02 -07001933 if (dst->addrtype == TIPC_ADDR_MCAST) {
1934 res = -EINVAL;
1935 goto exit;
1936 }
1937
Ying Xue78eb3a52014-01-17 09:50:03 +08001938 previous = sock->state;
Ying Xue584d24b2012-11-29 18:51:19 -05001939 switch (sock->state) {
1940 case SS_UNCONNECTED:
1941 /* Send a 'SYN-' to destination */
1942 m.msg_name = dest;
1943 m.msg_namelen = destlen;
1944
1945 /* If connect is in non-blocking case, set MSG_DONTWAIT to
1946 * indicate send_msg() is never blocked.
1947 */
1948 if (!timeout)
1949 m.msg_flags = MSG_DONTWAIT;
1950
Ying Xue247f0f32014-02-18 16:06:46 +08001951 res = tipc_sendmsg(NULL, sock, &m, 0);
Ying Xue584d24b2012-11-29 18:51:19 -05001952 if ((res < 0) && (res != -EWOULDBLOCK))
1953 goto exit;
1954
1955 /* Just entered SS_CONNECTING state; the only
1956 * difference is that return value in non-blocking
1957 * case is EINPROGRESS, rather than EALREADY.
1958 */
1959 res = -EINPROGRESS;
Ying Xue584d24b2012-11-29 18:51:19 -05001960 case SS_CONNECTING:
Ying Xue78eb3a52014-01-17 09:50:03 +08001961 if (previous == SS_CONNECTING)
1962 res = -EALREADY;
1963 if (!timeout)
1964 goto exit;
1965 timeout = msecs_to_jiffies(timeout);
1966 /* Wait until an 'ACK' or 'RST' arrives, or a timeout occurs */
1967 res = tipc_wait_for_connect(sock, &timeout);
Ying Xue584d24b2012-11-29 18:51:19 -05001968 break;
1969 case SS_CONNECTED:
1970 res = -EISCONN;
1971 break;
1972 default:
1973 res = -EINVAL;
Ying Xue78eb3a52014-01-17 09:50:03 +08001974 break;
Allan Stephensb89741a2008-04-15 00:20:37 -07001975 }
Allan Stephens0c3141e2008-04-15 00:22:02 -07001976exit:
1977 release_sock(sk);
Allan Stephensb89741a2008-04-15 00:20:37 -07001978 return res;
Per Lidenb97bf3f2006-01-02 19:04:38 +01001979}
1980
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001981/**
Ying Xue247f0f32014-02-18 16:06:46 +08001982 * tipc_listen - allow socket to listen for incoming connections
Per Lidenb97bf3f2006-01-02 19:04:38 +01001983 * @sock: socket structure
1984 * @len: (unused)
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09001985 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01001986 * Returns 0 on success, errno otherwise
1987 */
Ying Xue247f0f32014-02-18 16:06:46 +08001988static int tipc_listen(struct socket *sock, int len)
Per Lidenb97bf3f2006-01-02 19:04:38 +01001989{
Allan Stephens0c3141e2008-04-15 00:22:02 -07001990 struct sock *sk = sock->sk;
1991 int res;
1992
1993 lock_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01001994
Ying Xue245f3d32011-07-06 06:01:13 -04001995 if (sock->state != SS_UNCONNECTED)
Allan Stephens0c3141e2008-04-15 00:22:02 -07001996 res = -EINVAL;
1997 else {
1998 sock->state = SS_LISTENING;
1999 res = 0;
2000 }
2001
2002 release_sock(sk);
2003 return res;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002004}
2005
Ying Xue6398e232014-01-17 09:50:04 +08002006static int tipc_wait_for_accept(struct socket *sock, long timeo)
2007{
2008 struct sock *sk = sock->sk;
2009 DEFINE_WAIT(wait);
2010 int err;
2011
2012 /* True wake-one mechanism for incoming connections: only
2013 * one process gets woken up, not the 'whole herd'.
2014 * Since we do not 'race & poll' for established sockets
2015 * anymore, the common case will execute the loop only once.
2016 */
2017 for (;;) {
2018 prepare_to_wait_exclusive(sk_sleep(sk), &wait,
2019 TASK_INTERRUPTIBLE);
Ying Xuefe8e4642014-03-06 14:40:18 +01002020 if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
Ying Xue6398e232014-01-17 09:50:04 +08002021 release_sock(sk);
2022 timeo = schedule_timeout(timeo);
2023 lock_sock(sk);
2024 }
2025 err = 0;
2026 if (!skb_queue_empty(&sk->sk_receive_queue))
2027 break;
2028 err = -EINVAL;
2029 if (sock->state != SS_LISTENING)
2030 break;
2031 err = sock_intr_errno(timeo);
2032 if (signal_pending(current))
2033 break;
2034 err = -EAGAIN;
2035 if (!timeo)
2036 break;
2037 }
2038 finish_wait(sk_sleep(sk), &wait);
2039 return err;
2040}
2041
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002042/**
Ying Xue247f0f32014-02-18 16:06:46 +08002043 * tipc_accept - wait for connection request
Per Lidenb97bf3f2006-01-02 19:04:38 +01002044 * @sock: listening socket
2045 * @newsock: new socket that is to be connected
2046 * @flags: file-related flags associated with socket
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002047 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01002048 * Returns 0 on success, errno otherwise
2049 */
Ying Xue247f0f32014-02-18 16:06:46 +08002050static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002051{
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002052 struct sock *new_sk, *sk = sock->sk;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002053 struct sk_buff *buf;
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002054 struct tipc_sock *new_tsock;
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002055 struct tipc_msg *msg;
Ying Xue6398e232014-01-17 09:50:04 +08002056 long timeo;
Allan Stephens0c3141e2008-04-15 00:22:02 -07002057 int res;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002058
Allan Stephens0c3141e2008-04-15 00:22:02 -07002059 lock_sock(sk);
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002060
Allan Stephens0c3141e2008-04-15 00:22:02 -07002061 if (sock->state != SS_LISTENING) {
2062 res = -EINVAL;
2063 goto exit;
2064 }
Ying Xue6398e232014-01-17 09:50:04 +08002065 timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
2066 res = tipc_wait_for_accept(sock, timeo);
2067 if (res)
2068 goto exit;
Allan Stephens0c3141e2008-04-15 00:22:02 -07002069
2070 buf = skb_peek(&sk->sk_receive_queue);
2071
Ying Xuec5fa7b32013-06-17 10:54:39 -04002072 res = tipc_sk_create(sock_net(sock->sk), new_sock, 0, 1);
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002073 if (res)
2074 goto exit;
Allan Stephens0c3141e2008-04-15 00:22:02 -07002075
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002076 new_sk = new_sock->sk;
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002077 new_tsock = tipc_sk(new_sk);
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002078 msg = buf_msg(buf);
Allan Stephens0c3141e2008-04-15 00:22:02 -07002079
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002080 /* we lock on new_sk; but lockdep sees the lock on sk */
2081 lock_sock_nested(new_sk, SINGLE_DEPTH_NESTING);
Allan Stephens0c3141e2008-04-15 00:22:02 -07002082
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002083 /*
2084 * Reject any stray messages received by new socket
2085 * before the socket lock was taken (very, very unlikely)
2086 */
Jon Paul Maloy2e84c602014-08-22 18:09:18 -04002087 tsk_rej_rx_queue(new_sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002088
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002089 /* Connect new socket to it's peer */
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002090 tipc_sk_finish_conn(new_tsock, msg_origport(msg), msg_orignode(msg));
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002091 new_sock->state = SS_CONNECTED;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002092
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002093 tsk_set_importance(new_tsock, msg_importance(msg));
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002094 if (msg_named(msg)) {
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002095 new_tsock->conn_type = msg_nametype(msg);
2096 new_tsock->conn_instance = msg_nameinst(msg);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002097 }
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002098
2099 /*
2100 * Respond to 'SYN-' by discarding it & returning 'ACK'-.
2101 * Respond to 'SYN+' by queuing it on new socket.
2102 */
2103 if (!msg_data_sz(msg)) {
2104 struct msghdr m = {NULL,};
2105
Jon Paul Maloy2e84c602014-08-22 18:09:18 -04002106 tsk_advance_rx_queue(sk);
Ying Xue247f0f32014-02-18 16:06:46 +08002107 tipc_send_packet(NULL, new_sock, &m, 0);
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002108 } else {
2109 __skb_dequeue(&sk->sk_receive_queue);
2110 __skb_queue_head(&new_sk->sk_receive_queue, buf);
Ying Xueaba79f32013-01-20 23:30:09 +01002111 skb_set_owner_r(buf, new_sk);
Paul Gortmaker0fef8f22012-12-04 11:01:55 -05002112 }
2113 release_sock(new_sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002114exit:
Allan Stephens0c3141e2008-04-15 00:22:02 -07002115 release_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002116 return res;
2117}
2118
2119/**
Ying Xue247f0f32014-02-18 16:06:46 +08002120 * tipc_shutdown - shutdown socket connection
Per Lidenb97bf3f2006-01-02 19:04:38 +01002121 * @sock: socket structure
Allan Stephense247a8f2008-03-06 15:05:38 -08002122 * @how: direction to close (must be SHUT_RDWR)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002123 *
2124 * Terminates connection (if necessary), then purges socket's receive queue.
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002125 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01002126 * Returns 0 on success, errno otherwise
2127 */
Ying Xue247f0f32014-02-18 16:06:46 +08002128static int tipc_shutdown(struct socket *sock, int how)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002129{
Allan Stephens0c3141e2008-04-15 00:22:02 -07002130 struct sock *sk = sock->sk;
Ying Xuef2f98002015-01-09 15:27:05 +08002131 struct net *net = sock_net(sk);
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04002132 struct tipc_sock *tsk = tipc_sk(sk);
Ying Xuea6ca1092014-11-26 11:41:55 +08002133 struct sk_buff *skb;
Jon Paul Maloy80e44c22014-08-22 18:09:10 -04002134 u32 dnode;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002135 int res;
2136
Allan Stephense247a8f2008-03-06 15:05:38 -08002137 if (how != SHUT_RDWR)
2138 return -EINVAL;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002139
Allan Stephens0c3141e2008-04-15 00:22:02 -07002140 lock_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002141
2142 switch (sock->state) {
Allan Stephens0c3141e2008-04-15 00:22:02 -07002143 case SS_CONNECTING:
Per Lidenb97bf3f2006-01-02 19:04:38 +01002144 case SS_CONNECTED:
2145
Per Lidenb97bf3f2006-01-02 19:04:38 +01002146restart:
Paul Gortmaker617d3c72012-04-30 15:29:02 -04002147 /* Disconnect and send a 'FIN+' or 'FIN-' message to peer */
Ying Xuea6ca1092014-11-26 11:41:55 +08002148 skb = __skb_dequeue(&sk->sk_receive_queue);
2149 if (skb) {
2150 if (TIPC_SKB_CB(skb)->handle != NULL) {
2151 kfree_skb(skb);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002152 goto restart;
2153 }
Jon Paul Maloyc5898632015-02-05 08:36:36 -05002154 if (tipc_msg_reverse(tsk_own_node(tsk), skb, &dnode,
Ying Xue34747532015-01-09 15:27:10 +08002155 TIPC_CONN_SHUTDOWN))
Ying Xuef2f98002015-01-09 15:27:05 +08002156 tipc_link_xmit_skb(net, skb, dnode,
2157 tsk->portid);
2158 tipc_node_remove_conn(net, dnode, tsk->portid);
Allan Stephens0c3141e2008-04-15 00:22:02 -07002159 } else {
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002160 dnode = tsk_peer_node(tsk);
Jon Paul Maloyc5898632015-02-05 08:36:36 -05002161
2162 skb = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE,
Jon Paul Maloy80e44c22014-08-22 18:09:10 -04002163 TIPC_CONN_MSG, SHORT_H_SIZE,
Jon Paul Maloyc5898632015-02-05 08:36:36 -05002164 0, dnode, tsk_own_node(tsk),
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002165 tsk_peer_port(tsk),
Ying Xue07f6c4b2015-01-07 13:41:58 +08002166 tsk->portid, TIPC_CONN_SHUTDOWN);
Ying Xuef2f98002015-01-09 15:27:05 +08002167 tipc_link_xmit_skb(net, skb, dnode, tsk->portid);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002168 }
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002169 tsk->connected = 0;
Allan Stephens0c3141e2008-04-15 00:22:02 -07002170 sock->state = SS_DISCONNECTING;
Ying Xuef2f98002015-01-09 15:27:05 +08002171 tipc_node_remove_conn(net, dnode, tsk->portid);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002172 /* fall through */
2173
2174 case SS_DISCONNECTING:
2175
Ying Xue75031152012-10-29 09:38:15 -04002176 /* Discard any unreceived messages */
Ying Xue57467e52013-01-20 23:30:08 +01002177 __skb_queue_purge(&sk->sk_receive_queue);
Ying Xue75031152012-10-29 09:38:15 -04002178
2179 /* Wake up anyone sleeping in poll */
2180 sk->sk_state_change(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002181 res = 0;
2182 break;
2183
2184 default:
2185 res = -ENOTCONN;
2186 }
2187
Allan Stephens0c3141e2008-04-15 00:22:02 -07002188 release_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002189 return res;
2190}
2191
Ying Xuef2f2a962015-01-09 15:27:02 +08002192static void tipc_sk_timeout(unsigned long data)
Jon Paul Maloy57289012014-08-22 18:09:09 -04002193{
Ying Xuef2f2a962015-01-09 15:27:02 +08002194 struct tipc_sock *tsk = (struct tipc_sock *)data;
2195 struct sock *sk = &tsk->sk;
Ying Xuea6ca1092014-11-26 11:41:55 +08002196 struct sk_buff *skb = NULL;
Jon Paul Maloy57289012014-08-22 18:09:09 -04002197 u32 peer_port, peer_node;
Jon Paul Maloyc5898632015-02-05 08:36:36 -05002198 u32 own_node = tsk_own_node(tsk);
Jon Paul Maloy57289012014-08-22 18:09:09 -04002199
Jon Paul Maloy57289012014-08-22 18:09:09 -04002200 bh_lock_sock(sk);
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002201 if (!tsk->connected) {
Jon Paul Maloy6c9808c2014-08-22 18:09:16 -04002202 bh_unlock_sock(sk);
2203 goto exit;
2204 }
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002205 peer_port = tsk_peer_port(tsk);
2206 peer_node = tsk_peer_node(tsk);
Jon Paul Maloy57289012014-08-22 18:09:09 -04002207
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002208 if (tsk->probing_state == TIPC_CONN_PROBING) {
Jon Paul Maloy57289012014-08-22 18:09:09 -04002209 /* Previous probe not answered -> self abort */
Jon Paul Maloyc5898632015-02-05 08:36:36 -05002210 skb = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE,
Ying Xue34747532015-01-09 15:27:10 +08002211 TIPC_CONN_MSG, SHORT_H_SIZE, 0,
Jon Paul Maloyc5898632015-02-05 08:36:36 -05002212 own_node, peer_node, tsk->portid,
Ying Xue34747532015-01-09 15:27:10 +08002213 peer_port, TIPC_ERR_NO_PORT);
Jon Paul Maloy57289012014-08-22 18:09:09 -04002214 } else {
Jon Paul Maloyc5898632015-02-05 08:36:36 -05002215 skb = tipc_msg_create(CONN_MANAGER, CONN_PROBE,
2216 INT_H_SIZE, 0, peer_node, own_node,
Ying Xuef2f2a962015-01-09 15:27:02 +08002217 peer_port, tsk->portid, TIPC_OK);
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002218 tsk->probing_state = TIPC_CONN_PROBING;
Ying Xue3721e9c2015-01-13 17:07:48 +08002219 sk_reset_timer(sk, &sk->sk_timer, jiffies + tsk->probing_intv);
Jon Paul Maloy57289012014-08-22 18:09:09 -04002220 }
2221 bh_unlock_sock(sk);
Ying Xuea6ca1092014-11-26 11:41:55 +08002222 if (skb)
Ying Xuef2f98002015-01-09 15:27:05 +08002223 tipc_link_xmit_skb(sock_net(sk), skb, peer_node, tsk->portid);
Jon Paul Maloy6c9808c2014-08-22 18:09:16 -04002224exit:
Ying Xue07f6c4b2015-01-07 13:41:58 +08002225 sock_put(sk);
Jon Paul Maloy57289012014-08-22 18:09:09 -04002226}
2227
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002228static int tipc_sk_publish(struct tipc_sock *tsk, uint scope,
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002229 struct tipc_name_seq const *seq)
2230{
Ying Xuef2f98002015-01-09 15:27:05 +08002231 struct net *net = sock_net(&tsk->sk);
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002232 struct publication *publ;
2233 u32 key;
2234
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002235 if (tsk->connected)
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002236 return -EINVAL;
Ying Xue07f6c4b2015-01-07 13:41:58 +08002237 key = tsk->portid + tsk->pub_count + 1;
2238 if (key == tsk->portid)
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002239 return -EADDRINUSE;
2240
Ying Xuef2f98002015-01-09 15:27:05 +08002241 publ = tipc_nametbl_publish(net, seq->type, seq->lower, seq->upper,
Ying Xue07f6c4b2015-01-07 13:41:58 +08002242 scope, tsk->portid, key);
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002243 if (unlikely(!publ))
2244 return -EINVAL;
2245
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002246 list_add(&publ->pport_list, &tsk->publications);
2247 tsk->pub_count++;
2248 tsk->published = 1;
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002249 return 0;
2250}
2251
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002252static int tipc_sk_withdraw(struct tipc_sock *tsk, uint scope,
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002253 struct tipc_name_seq const *seq)
2254{
Ying Xuef2f98002015-01-09 15:27:05 +08002255 struct net *net = sock_net(&tsk->sk);
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002256 struct publication *publ;
2257 struct publication *safe;
2258 int rc = -EINVAL;
2259
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002260 list_for_each_entry_safe(publ, safe, &tsk->publications, pport_list) {
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002261 if (seq) {
2262 if (publ->scope != scope)
2263 continue;
2264 if (publ->type != seq->type)
2265 continue;
2266 if (publ->lower != seq->lower)
2267 continue;
2268 if (publ->upper != seq->upper)
2269 break;
Ying Xuef2f98002015-01-09 15:27:05 +08002270 tipc_nametbl_withdraw(net, publ->type, publ->lower,
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002271 publ->ref, publ->key);
2272 rc = 0;
2273 break;
2274 }
Ying Xuef2f98002015-01-09 15:27:05 +08002275 tipc_nametbl_withdraw(net, publ->type, publ->lower,
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002276 publ->ref, publ->key);
2277 rc = 0;
2278 }
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002279 if (list_empty(&tsk->publications))
2280 tsk->published = 0;
Jon Paul Maloy0fc87aa2014-08-22 18:09:17 -04002281 return rc;
2282}
2283
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002284static int tipc_sk_show(struct tipc_sock *tsk, char *buf,
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002285 int len, int full_id)
2286{
Ying Xue34747532015-01-09 15:27:10 +08002287 struct net *net = sock_net(&tsk->sk);
2288 struct tipc_net *tn = net_generic(net, tipc_net_id);
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002289 struct publication *publ;
2290 int ret;
2291
2292 if (full_id)
2293 ret = tipc_snprintf(buf, len, "<%u.%u.%u:%u>:",
Ying Xue34747532015-01-09 15:27:10 +08002294 tipc_zone(tn->own_addr),
2295 tipc_cluster(tn->own_addr),
2296 tipc_node(tn->own_addr), tsk->portid);
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002297 else
Ying Xue07f6c4b2015-01-07 13:41:58 +08002298 ret = tipc_snprintf(buf, len, "%-10u:", tsk->portid);
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002299
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002300 if (tsk->connected) {
2301 u32 dport = tsk_peer_port(tsk);
2302 u32 destnode = tsk_peer_node(tsk);
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002303
2304 ret += tipc_snprintf(buf + ret, len - ret,
2305 " connected to <%u.%u.%u:%u>",
2306 tipc_zone(destnode),
2307 tipc_cluster(destnode),
2308 tipc_node(destnode), dport);
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002309 if (tsk->conn_type != 0)
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002310 ret += tipc_snprintf(buf + ret, len - ret,
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002311 " via {%u,%u}", tsk->conn_type,
2312 tsk->conn_instance);
2313 } else if (tsk->published) {
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002314 ret += tipc_snprintf(buf + ret, len - ret, " bound to");
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002315 list_for_each_entry(publ, &tsk->publications, pport_list) {
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002316 if (publ->lower == publ->upper)
2317 ret += tipc_snprintf(buf + ret, len - ret,
2318 " {%u,%u}", publ->type,
2319 publ->lower);
2320 else
2321 ret += tipc_snprintf(buf + ret, len - ret,
2322 " {%u,%u,%u}", publ->type,
2323 publ->lower, publ->upper);
2324 }
2325 }
2326 ret += tipc_snprintf(buf + ret, len - ret, "\n");
2327 return ret;
2328}
2329
Ying Xuee05b31f2015-01-09 15:27:08 +08002330struct sk_buff *tipc_sk_socks_show(struct net *net)
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002331{
Ying Xuee05b31f2015-01-09 15:27:08 +08002332 struct tipc_net *tn = net_generic(net, tipc_net_id);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002333 const struct bucket_table *tbl;
2334 struct rhash_head *pos;
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002335 struct sk_buff *buf;
2336 struct tlv_desc *rep_tlv;
2337 char *pb;
2338 int pb_len;
2339 struct tipc_sock *tsk;
2340 int str_len = 0;
Ying Xue07f6c4b2015-01-07 13:41:58 +08002341 int i;
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002342
2343 buf = tipc_cfg_reply_alloc(TLV_SPACE(ULTRA_STRING_MAX_LEN));
2344 if (!buf)
2345 return NULL;
2346 rep_tlv = (struct tlv_desc *)buf->data;
2347 pb = TLV_DATA(rep_tlv);
2348 pb_len = ULTRA_STRING_MAX_LEN;
2349
Ying Xue07f6c4b2015-01-07 13:41:58 +08002350 rcu_read_lock();
Ying Xuee05b31f2015-01-09 15:27:08 +08002351 tbl = rht_dereference_rcu((&tn->sk_rht)->tbl, &tn->sk_rht);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002352 for (i = 0; i < tbl->size; i++) {
2353 rht_for_each_entry_rcu(tsk, pos, tbl, i, node) {
2354 spin_lock_bh(&tsk->sk.sk_lock.slock);
2355 str_len += tipc_sk_show(tsk, pb + str_len,
2356 pb_len - str_len, 0);
2357 spin_unlock_bh(&tsk->sk.sk_lock.slock);
2358 }
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002359 }
Ying Xue07f6c4b2015-01-07 13:41:58 +08002360 rcu_read_unlock();
2361
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002362 str_len += 1; /* for "\0" */
2363 skb_put(buf, TLV_SPACE(str_len));
2364 TLV_SET(rep_tlv, TIPC_TLV_ULTRA_STRING, NULL, str_len);
2365
2366 return buf;
2367}
2368
2369/* tipc_sk_reinit: set non-zero address in all existing sockets
2370 * when we go from standalone to network mode.
2371 */
Ying Xuee05b31f2015-01-09 15:27:08 +08002372void tipc_sk_reinit(struct net *net)
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002373{
Ying Xuee05b31f2015-01-09 15:27:08 +08002374 struct tipc_net *tn = net_generic(net, tipc_net_id);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002375 const struct bucket_table *tbl;
2376 struct rhash_head *pos;
2377 struct tipc_sock *tsk;
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002378 struct tipc_msg *msg;
Ying Xue07f6c4b2015-01-07 13:41:58 +08002379 int i;
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002380
Ying Xue07f6c4b2015-01-07 13:41:58 +08002381 rcu_read_lock();
Ying Xuee05b31f2015-01-09 15:27:08 +08002382 tbl = rht_dereference_rcu((&tn->sk_rht)->tbl, &tn->sk_rht);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002383 for (i = 0; i < tbl->size; i++) {
2384 rht_for_each_entry_rcu(tsk, pos, tbl, i, node) {
2385 spin_lock_bh(&tsk->sk.sk_lock.slock);
2386 msg = &tsk->phdr;
Ying Xue34747532015-01-09 15:27:10 +08002387 msg_set_prevnode(msg, tn->own_addr);
2388 msg_set_orignode(msg, tn->own_addr);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002389 spin_unlock_bh(&tsk->sk.sk_lock.slock);
2390 }
2391 }
2392 rcu_read_unlock();
2393}
2394
Ying Xuee05b31f2015-01-09 15:27:08 +08002395static struct tipc_sock *tipc_sk_lookup(struct net *net, u32 portid)
Ying Xue07f6c4b2015-01-07 13:41:58 +08002396{
Ying Xuee05b31f2015-01-09 15:27:08 +08002397 struct tipc_net *tn = net_generic(net, tipc_net_id);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002398 struct tipc_sock *tsk;
2399
2400 rcu_read_lock();
Ying Xuee05b31f2015-01-09 15:27:08 +08002401 tsk = rhashtable_lookup(&tn->sk_rht, &portid);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002402 if (tsk)
2403 sock_hold(&tsk->sk);
2404 rcu_read_unlock();
2405
2406 return tsk;
2407}
2408
2409static int tipc_sk_insert(struct tipc_sock *tsk)
2410{
Ying Xuee05b31f2015-01-09 15:27:08 +08002411 struct sock *sk = &tsk->sk;
2412 struct net *net = sock_net(sk);
2413 struct tipc_net *tn = net_generic(net, tipc_net_id);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002414 u32 remaining = (TIPC_MAX_PORT - TIPC_MIN_PORT) + 1;
2415 u32 portid = prandom_u32() % remaining + TIPC_MIN_PORT;
2416
2417 while (remaining--) {
2418 portid++;
2419 if ((portid < TIPC_MIN_PORT) || (portid > TIPC_MAX_PORT))
2420 portid = TIPC_MIN_PORT;
2421 tsk->portid = portid;
2422 sock_hold(&tsk->sk);
Ying Xuee05b31f2015-01-09 15:27:08 +08002423 if (rhashtable_lookup_insert(&tn->sk_rht, &tsk->node))
Ying Xue07f6c4b2015-01-07 13:41:58 +08002424 return 0;
2425 sock_put(&tsk->sk);
2426 }
2427
2428 return -1;
2429}
2430
2431static void tipc_sk_remove(struct tipc_sock *tsk)
2432{
2433 struct sock *sk = &tsk->sk;
Ying Xuee05b31f2015-01-09 15:27:08 +08002434 struct tipc_net *tn = net_generic(sock_net(sk), tipc_net_id);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002435
Ying Xuee05b31f2015-01-09 15:27:08 +08002436 if (rhashtable_remove(&tn->sk_rht, &tsk->node)) {
Ying Xue07f6c4b2015-01-07 13:41:58 +08002437 WARN_ON(atomic_read(&sk->sk_refcnt) == 1);
2438 __sock_put(sk);
Jon Paul Maloy5a9ee0be2014-08-22 18:09:14 -04002439 }
2440}
2441
Ying Xuee05b31f2015-01-09 15:27:08 +08002442int tipc_sk_rht_init(struct net *net)
Jon Paul Maloy808d90f2014-08-22 18:09:19 -04002443{
Ying Xuee05b31f2015-01-09 15:27:08 +08002444 struct tipc_net *tn = net_generic(net, tipc_net_id);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002445 struct rhashtable_params rht_params = {
2446 .nelem_hint = 192,
2447 .head_offset = offsetof(struct tipc_sock, node),
2448 .key_offset = offsetof(struct tipc_sock, portid),
2449 .key_len = sizeof(u32), /* portid */
2450 .hashfn = jhash,
2451 .max_shift = 20, /* 1M */
2452 .min_shift = 8, /* 256 */
2453 .grow_decision = rht_grow_above_75,
2454 .shrink_decision = rht_shrink_below_30,
Jon Paul Maloy808d90f2014-08-22 18:09:19 -04002455 };
2456
Ying Xuee05b31f2015-01-09 15:27:08 +08002457 return rhashtable_init(&tn->sk_rht, &rht_params);
Jon Paul Maloy808d90f2014-08-22 18:09:19 -04002458}
2459
Ying Xuee05b31f2015-01-09 15:27:08 +08002460void tipc_sk_rht_destroy(struct net *net)
Jon Paul Maloy808d90f2014-08-22 18:09:19 -04002461{
Ying Xuee05b31f2015-01-09 15:27:08 +08002462 struct tipc_net *tn = net_generic(net, tipc_net_id);
2463
Ying Xue07f6c4b2015-01-07 13:41:58 +08002464 /* Wait for socket readers to complete */
2465 synchronize_net();
Jon Paul Maloy808d90f2014-08-22 18:09:19 -04002466
Ying Xuee05b31f2015-01-09 15:27:08 +08002467 rhashtable_destroy(&tn->sk_rht);
Jon Paul Maloy808d90f2014-08-22 18:09:19 -04002468}
2469
2470/**
Ying Xue247f0f32014-02-18 16:06:46 +08002471 * tipc_setsockopt - set socket option
Per Lidenb97bf3f2006-01-02 19:04:38 +01002472 * @sock: socket structure
2473 * @lvl: option level
2474 * @opt: option identifier
2475 * @ov: pointer to new option value
2476 * @ol: length of option value
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002477 *
2478 * For stream sockets only, accepts and ignores all IPPROTO_TCP options
Per Lidenb97bf3f2006-01-02 19:04:38 +01002479 * (to ease compatibility).
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002480 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01002481 * Returns 0 on success, errno otherwise
2482 */
Ying Xue247f0f32014-02-18 16:06:46 +08002483static int tipc_setsockopt(struct socket *sock, int lvl, int opt,
2484 char __user *ov, unsigned int ol)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002485{
Allan Stephens0c3141e2008-04-15 00:22:02 -07002486 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04002487 struct tipc_sock *tsk = tipc_sk(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002488 u32 value;
2489 int res;
2490
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002491 if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
2492 return 0;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002493 if (lvl != SOL_TIPC)
2494 return -ENOPROTOOPT;
2495 if (ol < sizeof(value))
2496 return -EINVAL;
Allan Stephens2db99832010-12-31 18:59:33 +00002497 res = get_user(value, (u32 __user *)ov);
2498 if (res)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002499 return res;
2500
Allan Stephens0c3141e2008-04-15 00:22:02 -07002501 lock_sock(sk);
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002502
Per Lidenb97bf3f2006-01-02 19:04:38 +01002503 switch (opt) {
2504 case TIPC_IMPORTANCE:
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002505 res = tsk_set_importance(tsk, value);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002506 break;
2507 case TIPC_SRC_DROPPABLE:
2508 if (sock->type != SOCK_STREAM)
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002509 tsk_set_unreliable(tsk, value);
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002510 else
Per Lidenb97bf3f2006-01-02 19:04:38 +01002511 res = -ENOPROTOOPT;
2512 break;
2513 case TIPC_DEST_DROPPABLE:
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002514 tsk_set_unreturnable(tsk, value);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002515 break;
2516 case TIPC_CONN_TIMEOUT:
Allan Stephensa0f40f02011-05-26 13:44:34 -04002517 tipc_sk(sk)->conn_timeout = value;
Allan Stephens0c3141e2008-04-15 00:22:02 -07002518 /* no need to set "res", since already 0 at this point */
Per Lidenb97bf3f2006-01-02 19:04:38 +01002519 break;
2520 default:
2521 res = -EINVAL;
2522 }
2523
Allan Stephens0c3141e2008-04-15 00:22:02 -07002524 release_sock(sk);
2525
Per Lidenb97bf3f2006-01-02 19:04:38 +01002526 return res;
2527}
2528
2529/**
Ying Xue247f0f32014-02-18 16:06:46 +08002530 * tipc_getsockopt - get socket option
Per Lidenb97bf3f2006-01-02 19:04:38 +01002531 * @sock: socket structure
2532 * @lvl: option level
2533 * @opt: option identifier
2534 * @ov: receptacle for option value
2535 * @ol: receptacle for length of option value
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002536 *
2537 * For stream sockets only, returns 0 length result for all IPPROTO_TCP options
Per Lidenb97bf3f2006-01-02 19:04:38 +01002538 * (to ease compatibility).
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002539 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01002540 * Returns 0 on success, errno otherwise
2541 */
Ying Xue247f0f32014-02-18 16:06:46 +08002542static int tipc_getsockopt(struct socket *sock, int lvl, int opt,
2543 char __user *ov, int __user *ol)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002544{
Allan Stephens0c3141e2008-04-15 00:22:02 -07002545 struct sock *sk = sock->sk;
Jon Paul Maloy58ed9442014-03-12 11:31:12 -04002546 struct tipc_sock *tsk = tipc_sk(sk);
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002547 int len;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002548 u32 value;
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002549 int res;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002550
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002551 if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
2552 return put_user(0, ol);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002553 if (lvl != SOL_TIPC)
2554 return -ENOPROTOOPT;
Allan Stephens2db99832010-12-31 18:59:33 +00002555 res = get_user(len, ol);
2556 if (res)
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002557 return res;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002558
Allan Stephens0c3141e2008-04-15 00:22:02 -07002559 lock_sock(sk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002560
2561 switch (opt) {
2562 case TIPC_IMPORTANCE:
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002563 value = tsk_importance(tsk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002564 break;
2565 case TIPC_SRC_DROPPABLE:
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002566 value = tsk_unreliable(tsk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002567 break;
2568 case TIPC_DEST_DROPPABLE:
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002569 value = tsk_unreturnable(tsk);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002570 break;
2571 case TIPC_CONN_TIMEOUT:
Jon Paul Maloy301bae52014-08-22 18:09:20 -04002572 value = tsk->conn_timeout;
Allan Stephens0c3141e2008-04-15 00:22:02 -07002573 /* no need to set "res", since already 0 at this point */
Per Lidenb97bf3f2006-01-02 19:04:38 +01002574 break;
Allan Stephens0e659672010-12-31 18:59:32 +00002575 case TIPC_NODE_RECVQ_DEPTH:
Ying Xue9da3d472012-11-27 06:15:27 -05002576 value = 0; /* was tipc_queue_size, now obsolete */
oscar.medina@motorola.com66506132009-06-30 03:25:39 +00002577 break;
Allan Stephens0e659672010-12-31 18:59:32 +00002578 case TIPC_SOCK_RECVQ_DEPTH:
oscar.medina@motorola.com66506132009-06-30 03:25:39 +00002579 value = skb_queue_len(&sk->sk_receive_queue);
2580 break;
Per Lidenb97bf3f2006-01-02 19:04:38 +01002581 default:
2582 res = -EINVAL;
2583 }
2584
Allan Stephens0c3141e2008-04-15 00:22:02 -07002585 release_sock(sk);
2586
Paul Gortmaker25860c32010-12-31 18:59:31 +00002587 if (res)
2588 return res; /* "get" failed */
Per Lidenb97bf3f2006-01-02 19:04:38 +01002589
Paul Gortmaker25860c32010-12-31 18:59:31 +00002590 if (len < sizeof(value))
2591 return -EINVAL;
2592
2593 if (copy_to_user(ov, &value, sizeof(value)))
2594 return -EFAULT;
2595
2596 return put_user(sizeof(value), ol);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002597}
2598
Ying Xuef2f98002015-01-09 15:27:05 +08002599static int tipc_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg)
Erik Hugne78acb1f2014-04-24 16:26:47 +02002600{
Ying Xuef2f98002015-01-09 15:27:05 +08002601 struct sock *sk = sock->sk;
Erik Hugne78acb1f2014-04-24 16:26:47 +02002602 struct tipc_sioc_ln_req lnr;
2603 void __user *argp = (void __user *)arg;
2604
2605 switch (cmd) {
2606 case SIOCGETLINKNAME:
2607 if (copy_from_user(&lnr, argp, sizeof(lnr)))
2608 return -EFAULT;
Ying Xuef2f98002015-01-09 15:27:05 +08002609 if (!tipc_node_get_linkname(sock_net(sk),
2610 lnr.bearer_id & 0xffff, lnr.peer,
Erik Hugne78acb1f2014-04-24 16:26:47 +02002611 lnr.linkname, TIPC_MAX_LINK_NAME)) {
2612 if (copy_to_user(argp, &lnr, sizeof(lnr)))
2613 return -EFAULT;
2614 return 0;
2615 }
2616 return -EADDRNOTAVAIL;
Erik Hugne78acb1f2014-04-24 16:26:47 +02002617 default:
2618 return -ENOIOCTLCMD;
2619 }
2620}
2621
Ben Hutchingsae86b9e2012-07-10 10:55:35 +00002622/* Protocol switches for the various types of TIPC sockets */
2623
Florian Westphalbca65ea2008-02-07 18:18:01 -08002624static const struct proto_ops msg_ops = {
Allan Stephens0e659672010-12-31 18:59:32 +00002625 .owner = THIS_MODULE,
Per Lidenb97bf3f2006-01-02 19:04:38 +01002626 .family = AF_TIPC,
Ying Xue247f0f32014-02-18 16:06:46 +08002627 .release = tipc_release,
2628 .bind = tipc_bind,
2629 .connect = tipc_connect,
Allan Stephens5eee6a62007-06-10 17:24:55 -07002630 .socketpair = sock_no_socketpair,
Ying Xue245f3d32011-07-06 06:01:13 -04002631 .accept = sock_no_accept,
Ying Xue247f0f32014-02-18 16:06:46 +08002632 .getname = tipc_getname,
2633 .poll = tipc_poll,
Erik Hugne78acb1f2014-04-24 16:26:47 +02002634 .ioctl = tipc_ioctl,
Ying Xue245f3d32011-07-06 06:01:13 -04002635 .listen = sock_no_listen,
Ying Xue247f0f32014-02-18 16:06:46 +08002636 .shutdown = tipc_shutdown,
2637 .setsockopt = tipc_setsockopt,
2638 .getsockopt = tipc_getsockopt,
2639 .sendmsg = tipc_sendmsg,
2640 .recvmsg = tipc_recvmsg,
YOSHIFUJI Hideaki82387452007-07-19 10:44:56 +09002641 .mmap = sock_no_mmap,
2642 .sendpage = sock_no_sendpage
Per Lidenb97bf3f2006-01-02 19:04:38 +01002643};
2644
Florian Westphalbca65ea2008-02-07 18:18:01 -08002645static const struct proto_ops packet_ops = {
Allan Stephens0e659672010-12-31 18:59:32 +00002646 .owner = THIS_MODULE,
Per Lidenb97bf3f2006-01-02 19:04:38 +01002647 .family = AF_TIPC,
Ying Xue247f0f32014-02-18 16:06:46 +08002648 .release = tipc_release,
2649 .bind = tipc_bind,
2650 .connect = tipc_connect,
Allan Stephens5eee6a62007-06-10 17:24:55 -07002651 .socketpair = sock_no_socketpair,
Ying Xue247f0f32014-02-18 16:06:46 +08002652 .accept = tipc_accept,
2653 .getname = tipc_getname,
2654 .poll = tipc_poll,
Erik Hugne78acb1f2014-04-24 16:26:47 +02002655 .ioctl = tipc_ioctl,
Ying Xue247f0f32014-02-18 16:06:46 +08002656 .listen = tipc_listen,
2657 .shutdown = tipc_shutdown,
2658 .setsockopt = tipc_setsockopt,
2659 .getsockopt = tipc_getsockopt,
2660 .sendmsg = tipc_send_packet,
2661 .recvmsg = tipc_recvmsg,
YOSHIFUJI Hideaki82387452007-07-19 10:44:56 +09002662 .mmap = sock_no_mmap,
2663 .sendpage = sock_no_sendpage
Per Lidenb97bf3f2006-01-02 19:04:38 +01002664};
2665
Florian Westphalbca65ea2008-02-07 18:18:01 -08002666static const struct proto_ops stream_ops = {
Allan Stephens0e659672010-12-31 18:59:32 +00002667 .owner = THIS_MODULE,
Per Lidenb97bf3f2006-01-02 19:04:38 +01002668 .family = AF_TIPC,
Ying Xue247f0f32014-02-18 16:06:46 +08002669 .release = tipc_release,
2670 .bind = tipc_bind,
2671 .connect = tipc_connect,
Allan Stephens5eee6a62007-06-10 17:24:55 -07002672 .socketpair = sock_no_socketpair,
Ying Xue247f0f32014-02-18 16:06:46 +08002673 .accept = tipc_accept,
2674 .getname = tipc_getname,
2675 .poll = tipc_poll,
Erik Hugne78acb1f2014-04-24 16:26:47 +02002676 .ioctl = tipc_ioctl,
Ying Xue247f0f32014-02-18 16:06:46 +08002677 .listen = tipc_listen,
2678 .shutdown = tipc_shutdown,
2679 .setsockopt = tipc_setsockopt,
2680 .getsockopt = tipc_getsockopt,
2681 .sendmsg = tipc_send_stream,
2682 .recvmsg = tipc_recv_stream,
YOSHIFUJI Hideaki82387452007-07-19 10:44:56 +09002683 .mmap = sock_no_mmap,
2684 .sendpage = sock_no_sendpage
Per Lidenb97bf3f2006-01-02 19:04:38 +01002685};
2686
Florian Westphalbca65ea2008-02-07 18:18:01 -08002687static const struct net_proto_family tipc_family_ops = {
Allan Stephens0e659672010-12-31 18:59:32 +00002688 .owner = THIS_MODULE,
Per Lidenb97bf3f2006-01-02 19:04:38 +01002689 .family = AF_TIPC,
Ying Xuec5fa7b32013-06-17 10:54:39 -04002690 .create = tipc_sk_create
Per Lidenb97bf3f2006-01-02 19:04:38 +01002691};
2692
2693static struct proto tipc_proto = {
2694 .name = "TIPC",
2695 .owner = THIS_MODULE,
Ying Xuecc79dd12013-06-17 10:54:37 -04002696 .obj_size = sizeof(struct tipc_sock),
2697 .sysctl_rmem = sysctl_tipc_rmem
Per Lidenb97bf3f2006-01-02 19:04:38 +01002698};
2699
Ying Xuec5fa7b32013-06-17 10:54:39 -04002700static struct proto tipc_proto_kern = {
2701 .name = "TIPC",
2702 .obj_size = sizeof(struct tipc_sock),
2703 .sysctl_rmem = sysctl_tipc_rmem
2704};
2705
Per Lidenb97bf3f2006-01-02 19:04:38 +01002706/**
Per Liden4323add2006-01-18 00:38:21 +01002707 * tipc_socket_init - initialize TIPC socket interface
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002708 *
Per Lidenb97bf3f2006-01-02 19:04:38 +01002709 * Returns 0 on success, errno otherwise
2710 */
Per Liden4323add2006-01-18 00:38:21 +01002711int tipc_socket_init(void)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002712{
2713 int res;
2714
YOSHIFUJI Hideakic4307282007-02-09 23:25:21 +09002715 res = proto_register(&tipc_proto, 1);
Per Lidenb97bf3f2006-01-02 19:04:38 +01002716 if (res) {
Erik Hugne2cf8aa12012-06-29 00:16:37 -04002717 pr_err("Failed to register TIPC protocol type\n");
Per Lidenb97bf3f2006-01-02 19:04:38 +01002718 goto out;
2719 }
2720
2721 res = sock_register(&tipc_family_ops);
2722 if (res) {
Erik Hugne2cf8aa12012-06-29 00:16:37 -04002723 pr_err("Failed to register TIPC socket type\n");
Per Lidenb97bf3f2006-01-02 19:04:38 +01002724 proto_unregister(&tipc_proto);
2725 goto out;
2726 }
Per Lidenb97bf3f2006-01-02 19:04:38 +01002727 out:
2728 return res;
2729}
2730
2731/**
Per Liden4323add2006-01-18 00:38:21 +01002732 * tipc_socket_stop - stop TIPC socket interface
Per Lidenb97bf3f2006-01-02 19:04:38 +01002733 */
Per Liden4323add2006-01-18 00:38:21 +01002734void tipc_socket_stop(void)
Per Lidenb97bf3f2006-01-02 19:04:38 +01002735{
Per Lidenb97bf3f2006-01-02 19:04:38 +01002736 sock_unregister(tipc_family_ops.family);
2737 proto_unregister(&tipc_proto);
2738}
Richard Alpe34b78a122014-11-20 10:29:10 +01002739
2740/* Caller should hold socket lock for the passed tipc socket. */
Richard Alped8182802014-11-24 11:10:29 +01002741static int __tipc_nl_add_sk_con(struct sk_buff *skb, struct tipc_sock *tsk)
Richard Alpe34b78a122014-11-20 10:29:10 +01002742{
2743 u32 peer_node;
2744 u32 peer_port;
2745 struct nlattr *nest;
2746
2747 peer_node = tsk_peer_node(tsk);
2748 peer_port = tsk_peer_port(tsk);
2749
2750 nest = nla_nest_start(skb, TIPC_NLA_SOCK_CON);
2751
2752 if (nla_put_u32(skb, TIPC_NLA_CON_NODE, peer_node))
2753 goto msg_full;
2754 if (nla_put_u32(skb, TIPC_NLA_CON_SOCK, peer_port))
2755 goto msg_full;
2756
2757 if (tsk->conn_type != 0) {
2758 if (nla_put_flag(skb, TIPC_NLA_CON_FLAG))
2759 goto msg_full;
2760 if (nla_put_u32(skb, TIPC_NLA_CON_TYPE, tsk->conn_type))
2761 goto msg_full;
2762 if (nla_put_u32(skb, TIPC_NLA_CON_INST, tsk->conn_instance))
2763 goto msg_full;
2764 }
2765 nla_nest_end(skb, nest);
2766
2767 return 0;
2768
2769msg_full:
2770 nla_nest_cancel(skb, nest);
2771
2772 return -EMSGSIZE;
2773}
2774
2775/* Caller should hold socket lock for the passed tipc socket. */
Richard Alped8182802014-11-24 11:10:29 +01002776static int __tipc_nl_add_sk(struct sk_buff *skb, struct netlink_callback *cb,
2777 struct tipc_sock *tsk)
Richard Alpe34b78a122014-11-20 10:29:10 +01002778{
2779 int err;
2780 void *hdr;
2781 struct nlattr *attrs;
Ying Xue34747532015-01-09 15:27:10 +08002782 struct net *net = sock_net(skb->sk);
2783 struct tipc_net *tn = net_generic(net, tipc_net_id);
Richard Alpe34b78a122014-11-20 10:29:10 +01002784
2785 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
Richard Alpebfb3e5d2015-02-09 09:50:03 +01002786 &tipc_genl_family, NLM_F_MULTI, TIPC_NL_SOCK_GET);
Richard Alpe34b78a122014-11-20 10:29:10 +01002787 if (!hdr)
2788 goto msg_cancel;
2789
2790 attrs = nla_nest_start(skb, TIPC_NLA_SOCK);
2791 if (!attrs)
2792 goto genlmsg_cancel;
Ying Xue07f6c4b2015-01-07 13:41:58 +08002793 if (nla_put_u32(skb, TIPC_NLA_SOCK_REF, tsk->portid))
Richard Alpe34b78a122014-11-20 10:29:10 +01002794 goto attr_msg_cancel;
Ying Xue34747532015-01-09 15:27:10 +08002795 if (nla_put_u32(skb, TIPC_NLA_SOCK_ADDR, tn->own_addr))
Richard Alpe34b78a122014-11-20 10:29:10 +01002796 goto attr_msg_cancel;
2797
2798 if (tsk->connected) {
2799 err = __tipc_nl_add_sk_con(skb, tsk);
2800 if (err)
2801 goto attr_msg_cancel;
2802 } else if (!list_empty(&tsk->publications)) {
2803 if (nla_put_flag(skb, TIPC_NLA_SOCK_HAS_PUBL))
2804 goto attr_msg_cancel;
2805 }
2806 nla_nest_end(skb, attrs);
2807 genlmsg_end(skb, hdr);
2808
2809 return 0;
2810
2811attr_msg_cancel:
2812 nla_nest_cancel(skb, attrs);
2813genlmsg_cancel:
2814 genlmsg_cancel(skb, hdr);
2815msg_cancel:
2816 return -EMSGSIZE;
2817}
2818
2819int tipc_nl_sk_dump(struct sk_buff *skb, struct netlink_callback *cb)
2820{
2821 int err;
2822 struct tipc_sock *tsk;
Ying Xue07f6c4b2015-01-07 13:41:58 +08002823 const struct bucket_table *tbl;
2824 struct rhash_head *pos;
Ying Xuee05b31f2015-01-09 15:27:08 +08002825 struct net *net = sock_net(skb->sk);
2826 struct tipc_net *tn = net_generic(net, tipc_net_id);
Richard Alped6e164e2015-01-16 12:30:40 +01002827 u32 tbl_id = cb->args[0];
2828 u32 prev_portid = cb->args[1];
Richard Alpe34b78a122014-11-20 10:29:10 +01002829
Ying Xue07f6c4b2015-01-07 13:41:58 +08002830 rcu_read_lock();
Ying Xuee05b31f2015-01-09 15:27:08 +08002831 tbl = rht_dereference_rcu((&tn->sk_rht)->tbl, &tn->sk_rht);
Richard Alped6e164e2015-01-16 12:30:40 +01002832 for (; tbl_id < tbl->size; tbl_id++) {
2833 rht_for_each_entry_rcu(tsk, pos, tbl, tbl_id, node) {
Ying Xue07f6c4b2015-01-07 13:41:58 +08002834 spin_lock_bh(&tsk->sk.sk_lock.slock);
Richard Alped6e164e2015-01-16 12:30:40 +01002835 if (prev_portid && prev_portid != tsk->portid) {
2836 spin_unlock_bh(&tsk->sk.sk_lock.slock);
2837 continue;
2838 }
Richard Alpe34b78a122014-11-20 10:29:10 +01002839
Richard Alped6e164e2015-01-16 12:30:40 +01002840 err = __tipc_nl_add_sk(skb, cb, tsk);
2841 if (err) {
2842 prev_portid = tsk->portid;
2843 spin_unlock_bh(&tsk->sk.sk_lock.slock);
2844 goto out;
2845 }
2846 prev_portid = 0;
2847 spin_unlock_bh(&tsk->sk.sk_lock.slock);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002848 }
Richard Alpe34b78a122014-11-20 10:29:10 +01002849 }
Richard Alped6e164e2015-01-16 12:30:40 +01002850out:
Ying Xue07f6c4b2015-01-07 13:41:58 +08002851 rcu_read_unlock();
Richard Alped6e164e2015-01-16 12:30:40 +01002852 cb->args[0] = tbl_id;
2853 cb->args[1] = prev_portid;
Richard Alpe34b78a122014-11-20 10:29:10 +01002854
2855 return skb->len;
2856}
Richard Alpe1a1a1432014-11-20 10:29:11 +01002857
2858/* Caller should hold socket lock for the passed tipc socket. */
Richard Alped8182802014-11-24 11:10:29 +01002859static int __tipc_nl_add_sk_publ(struct sk_buff *skb,
2860 struct netlink_callback *cb,
2861 struct publication *publ)
Richard Alpe1a1a1432014-11-20 10:29:11 +01002862{
2863 void *hdr;
2864 struct nlattr *attrs;
2865
2866 hdr = genlmsg_put(skb, NETLINK_CB(cb->skb).portid, cb->nlh->nlmsg_seq,
Richard Alpebfb3e5d2015-02-09 09:50:03 +01002867 &tipc_genl_family, NLM_F_MULTI, TIPC_NL_PUBL_GET);
Richard Alpe1a1a1432014-11-20 10:29:11 +01002868 if (!hdr)
2869 goto msg_cancel;
2870
2871 attrs = nla_nest_start(skb, TIPC_NLA_PUBL);
2872 if (!attrs)
2873 goto genlmsg_cancel;
2874
2875 if (nla_put_u32(skb, TIPC_NLA_PUBL_KEY, publ->key))
2876 goto attr_msg_cancel;
2877 if (nla_put_u32(skb, TIPC_NLA_PUBL_TYPE, publ->type))
2878 goto attr_msg_cancel;
2879 if (nla_put_u32(skb, TIPC_NLA_PUBL_LOWER, publ->lower))
2880 goto attr_msg_cancel;
2881 if (nla_put_u32(skb, TIPC_NLA_PUBL_UPPER, publ->upper))
2882 goto attr_msg_cancel;
2883
2884 nla_nest_end(skb, attrs);
2885 genlmsg_end(skb, hdr);
2886
2887 return 0;
2888
2889attr_msg_cancel:
2890 nla_nest_cancel(skb, attrs);
2891genlmsg_cancel:
2892 genlmsg_cancel(skb, hdr);
2893msg_cancel:
2894 return -EMSGSIZE;
2895}
2896
2897/* Caller should hold socket lock for the passed tipc socket. */
Richard Alped8182802014-11-24 11:10:29 +01002898static int __tipc_nl_list_sk_publ(struct sk_buff *skb,
2899 struct netlink_callback *cb,
2900 struct tipc_sock *tsk, u32 *last_publ)
Richard Alpe1a1a1432014-11-20 10:29:11 +01002901{
2902 int err;
2903 struct publication *p;
2904
2905 if (*last_publ) {
2906 list_for_each_entry(p, &tsk->publications, pport_list) {
2907 if (p->key == *last_publ)
2908 break;
2909 }
2910 if (p->key != *last_publ) {
2911 /* We never set seq or call nl_dump_check_consistent()
2912 * this means that setting prev_seq here will cause the
2913 * consistence check to fail in the netlink callback
2914 * handler. Resulting in the last NLMSG_DONE message
2915 * having the NLM_F_DUMP_INTR flag set.
2916 */
2917 cb->prev_seq = 1;
2918 *last_publ = 0;
2919 return -EPIPE;
2920 }
2921 } else {
2922 p = list_first_entry(&tsk->publications, struct publication,
2923 pport_list);
2924 }
2925
2926 list_for_each_entry_from(p, &tsk->publications, pport_list) {
2927 err = __tipc_nl_add_sk_publ(skb, cb, p);
2928 if (err) {
2929 *last_publ = p->key;
2930 return err;
2931 }
2932 }
2933 *last_publ = 0;
2934
2935 return 0;
2936}
2937
2938int tipc_nl_publ_dump(struct sk_buff *skb, struct netlink_callback *cb)
2939{
2940 int err;
Ying Xue07f6c4b2015-01-07 13:41:58 +08002941 u32 tsk_portid = cb->args[0];
Richard Alpe1a1a1432014-11-20 10:29:11 +01002942 u32 last_publ = cb->args[1];
2943 u32 done = cb->args[2];
Ying Xuee05b31f2015-01-09 15:27:08 +08002944 struct net *net = sock_net(skb->sk);
Richard Alpe1a1a1432014-11-20 10:29:11 +01002945 struct tipc_sock *tsk;
2946
Ying Xue07f6c4b2015-01-07 13:41:58 +08002947 if (!tsk_portid) {
Richard Alpe1a1a1432014-11-20 10:29:11 +01002948 struct nlattr **attrs;
2949 struct nlattr *sock[TIPC_NLA_SOCK_MAX + 1];
2950
2951 err = tipc_nlmsg_parse(cb->nlh, &attrs);
2952 if (err)
2953 return err;
2954
2955 err = nla_parse_nested(sock, TIPC_NLA_SOCK_MAX,
2956 attrs[TIPC_NLA_SOCK],
2957 tipc_nl_sock_policy);
2958 if (err)
2959 return err;
2960
2961 if (!sock[TIPC_NLA_SOCK_REF])
2962 return -EINVAL;
2963
Ying Xue07f6c4b2015-01-07 13:41:58 +08002964 tsk_portid = nla_get_u32(sock[TIPC_NLA_SOCK_REF]);
Richard Alpe1a1a1432014-11-20 10:29:11 +01002965 }
2966
2967 if (done)
2968 return 0;
2969
Ying Xuee05b31f2015-01-09 15:27:08 +08002970 tsk = tipc_sk_lookup(net, tsk_portid);
Richard Alpe1a1a1432014-11-20 10:29:11 +01002971 if (!tsk)
2972 return -EINVAL;
2973
2974 lock_sock(&tsk->sk);
2975 err = __tipc_nl_list_sk_publ(skb, cb, tsk, &last_publ);
2976 if (!err)
2977 done = 1;
2978 release_sock(&tsk->sk);
Ying Xue07f6c4b2015-01-07 13:41:58 +08002979 sock_put(&tsk->sk);
Richard Alpe1a1a1432014-11-20 10:29:11 +01002980
Ying Xue07f6c4b2015-01-07 13:41:58 +08002981 cb->args[0] = tsk_portid;
Richard Alpe1a1a1432014-11-20 10:29:11 +01002982 cb->args[1] = last_publ;
2983 cb->args[2] = done;
2984
2985 return skb->len;
2986}