1*4882a593Smuzhiyun // SPDX-License-Identifier: GPL-2.0-only
2*4882a593Smuzhiyun /******************************************************************************
3*4882a593Smuzhiyun *******************************************************************************
4*4882a593Smuzhiyun **
5*4882a593Smuzhiyun ** Copyright (C) Sistina Software, Inc. 1997-2003 All rights reserved.
6*4882a593Smuzhiyun ** Copyright (C) 2004-2009 Red Hat, Inc. All rights reserved.
7*4882a593Smuzhiyun **
8*4882a593Smuzhiyun **
9*4882a593Smuzhiyun *******************************************************************************
10*4882a593Smuzhiyun ******************************************************************************/
11*4882a593Smuzhiyun
12*4882a593Smuzhiyun /*
13*4882a593Smuzhiyun * lowcomms.c
14*4882a593Smuzhiyun *
15*4882a593Smuzhiyun * This is the "low-level" comms layer.
16*4882a593Smuzhiyun *
17*4882a593Smuzhiyun * It is responsible for sending/receiving messages
18*4882a593Smuzhiyun * from other nodes in the cluster.
19*4882a593Smuzhiyun *
20*4882a593Smuzhiyun * Cluster nodes are referred to by their nodeids. nodeids are
21*4882a593Smuzhiyun * simply 32 bit numbers to the locking module - if they need to
22*4882a593Smuzhiyun * be expanded for the cluster infrastructure then that is its
23*4882a593Smuzhiyun * responsibility. It is this layer's
24*4882a593Smuzhiyun * responsibility to resolve these into IP address or
25*4882a593Smuzhiyun * whatever it needs for inter-node communication.
26*4882a593Smuzhiyun *
27*4882a593Smuzhiyun * The comms level is two kernel threads that deal mainly with
28*4882a593Smuzhiyun * the receiving of messages from other nodes and passing them
29*4882a593Smuzhiyun * up to the mid-level comms layer (which understands the
30*4882a593Smuzhiyun * message format) for execution by the locking core, and
31*4882a593Smuzhiyun * a send thread which does all the setting up of connections
32*4882a593Smuzhiyun * to remote nodes and the sending of data. Threads are not allowed
33*4882a593Smuzhiyun * to send their own data because it may cause them to wait in times
34*4882a593Smuzhiyun * of high load. Also, this way, the sending thread can collect together
35*4882a593Smuzhiyun * messages bound for one node and send them in one block.
36*4882a593Smuzhiyun *
37*4882a593Smuzhiyun * lowcomms will choose to use either TCP or SCTP as its transport layer
38*4882a593Smuzhiyun * depending on the configuration variable 'protocol'. This should be set
39*4882a593Smuzhiyun * to 0 (default) for TCP or 1 for SCTP. It should be configured using a
40*4882a593Smuzhiyun * cluster-wide mechanism as it must be the same on all nodes of the cluster
41*4882a593Smuzhiyun * for the DLM to function.
42*4882a593Smuzhiyun *
43*4882a593Smuzhiyun */
44*4882a593Smuzhiyun
45*4882a593Smuzhiyun #include <asm/ioctls.h>
46*4882a593Smuzhiyun #include <net/sock.h>
47*4882a593Smuzhiyun #include <net/tcp.h>
48*4882a593Smuzhiyun #include <linux/pagemap.h>
49*4882a593Smuzhiyun #include <linux/file.h>
50*4882a593Smuzhiyun #include <linux/mutex.h>
51*4882a593Smuzhiyun #include <linux/sctp.h>
52*4882a593Smuzhiyun #include <linux/slab.h>
53*4882a593Smuzhiyun #include <net/sctp/sctp.h>
54*4882a593Smuzhiyun #include <net/ipv6.h>
55*4882a593Smuzhiyun
56*4882a593Smuzhiyun #include "dlm_internal.h"
57*4882a593Smuzhiyun #include "lowcomms.h"
58*4882a593Smuzhiyun #include "midcomms.h"
59*4882a593Smuzhiyun #include "config.h"
60*4882a593Smuzhiyun
61*4882a593Smuzhiyun #define NEEDED_RMEM (4*1024*1024)
62*4882a593Smuzhiyun #define CONN_HASH_SIZE 32
63*4882a593Smuzhiyun
64*4882a593Smuzhiyun /* Number of messages to send before rescheduling */
65*4882a593Smuzhiyun #define MAX_SEND_MSG_COUNT 25
66*4882a593Smuzhiyun #define DLM_SHUTDOWN_WAIT_TIMEOUT msecs_to_jiffies(10000)
67*4882a593Smuzhiyun
68*4882a593Smuzhiyun struct connection {
69*4882a593Smuzhiyun struct socket *sock; /* NULL if not connected */
70*4882a593Smuzhiyun uint32_t nodeid; /* So we know who we are in the list */
71*4882a593Smuzhiyun struct mutex sock_mutex;
72*4882a593Smuzhiyun unsigned long flags;
73*4882a593Smuzhiyun #define CF_READ_PENDING 1
74*4882a593Smuzhiyun #define CF_WRITE_PENDING 2
75*4882a593Smuzhiyun #define CF_INIT_PENDING 4
76*4882a593Smuzhiyun #define CF_IS_OTHERCON 5
77*4882a593Smuzhiyun #define CF_CLOSE 6
78*4882a593Smuzhiyun #define CF_APP_LIMITED 7
79*4882a593Smuzhiyun #define CF_CLOSING 8
80*4882a593Smuzhiyun #define CF_SHUTDOWN 9
81*4882a593Smuzhiyun struct list_head writequeue; /* List of outgoing writequeue_entries */
82*4882a593Smuzhiyun spinlock_t writequeue_lock;
83*4882a593Smuzhiyun int (*rx_action) (struct connection *); /* What to do when active */
84*4882a593Smuzhiyun void (*connect_action) (struct connection *); /* What to do to connect */
85*4882a593Smuzhiyun void (*shutdown_action)(struct connection *con); /* What to do to shutdown */
86*4882a593Smuzhiyun int retries;
87*4882a593Smuzhiyun #define MAX_CONNECT_RETRIES 3
88*4882a593Smuzhiyun struct hlist_node list;
89*4882a593Smuzhiyun struct connection *othercon;
90*4882a593Smuzhiyun struct work_struct rwork; /* Receive workqueue */
91*4882a593Smuzhiyun struct work_struct swork; /* Send workqueue */
92*4882a593Smuzhiyun wait_queue_head_t shutdown_wait; /* wait for graceful shutdown */
93*4882a593Smuzhiyun unsigned char *rx_buf;
94*4882a593Smuzhiyun int rx_buflen;
95*4882a593Smuzhiyun int rx_leftover;
96*4882a593Smuzhiyun struct rcu_head rcu;
97*4882a593Smuzhiyun };
98*4882a593Smuzhiyun #define sock2con(x) ((struct connection *)(x)->sk_user_data)
99*4882a593Smuzhiyun
100*4882a593Smuzhiyun /* An entry waiting to be sent */
101*4882a593Smuzhiyun struct writequeue_entry {
102*4882a593Smuzhiyun struct list_head list;
103*4882a593Smuzhiyun struct page *page;
104*4882a593Smuzhiyun int offset;
105*4882a593Smuzhiyun int len;
106*4882a593Smuzhiyun int end;
107*4882a593Smuzhiyun int users;
108*4882a593Smuzhiyun struct connection *con;
109*4882a593Smuzhiyun };
110*4882a593Smuzhiyun
111*4882a593Smuzhiyun struct dlm_node_addr {
112*4882a593Smuzhiyun struct list_head list;
113*4882a593Smuzhiyun int nodeid;
114*4882a593Smuzhiyun int addr_count;
115*4882a593Smuzhiyun int curr_addr_index;
116*4882a593Smuzhiyun struct sockaddr_storage *addr[DLM_MAX_ADDR_COUNT];
117*4882a593Smuzhiyun };
118*4882a593Smuzhiyun
119*4882a593Smuzhiyun static struct listen_sock_callbacks {
120*4882a593Smuzhiyun void (*sk_error_report)(struct sock *);
121*4882a593Smuzhiyun void (*sk_data_ready)(struct sock *);
122*4882a593Smuzhiyun void (*sk_state_change)(struct sock *);
123*4882a593Smuzhiyun void (*sk_write_space)(struct sock *);
124*4882a593Smuzhiyun } listen_sock;
125*4882a593Smuzhiyun
126*4882a593Smuzhiyun static LIST_HEAD(dlm_node_addrs);
127*4882a593Smuzhiyun static DEFINE_SPINLOCK(dlm_node_addrs_spin);
128*4882a593Smuzhiyun
129*4882a593Smuzhiyun static struct sockaddr_storage *dlm_local_addr[DLM_MAX_ADDR_COUNT];
130*4882a593Smuzhiyun static int dlm_local_count;
131*4882a593Smuzhiyun static int dlm_allow_conn;
132*4882a593Smuzhiyun
133*4882a593Smuzhiyun /* Work queues */
134*4882a593Smuzhiyun static struct workqueue_struct *recv_workqueue;
135*4882a593Smuzhiyun static struct workqueue_struct *send_workqueue;
136*4882a593Smuzhiyun
137*4882a593Smuzhiyun static struct hlist_head connection_hash[CONN_HASH_SIZE];
138*4882a593Smuzhiyun static DEFINE_SPINLOCK(connections_lock);
139*4882a593Smuzhiyun DEFINE_STATIC_SRCU(connections_srcu);
140*4882a593Smuzhiyun
141*4882a593Smuzhiyun static void process_recv_sockets(struct work_struct *work);
142*4882a593Smuzhiyun static void process_send_sockets(struct work_struct *work);
143*4882a593Smuzhiyun
144*4882a593Smuzhiyun
145*4882a593Smuzhiyun /* This is deliberately very simple because most clusters have simple
146*4882a593Smuzhiyun sequential nodeids, so we should be able to go straight to a connection
147*4882a593Smuzhiyun struct in the array */
nodeid_hash(int nodeid)148*4882a593Smuzhiyun static inline int nodeid_hash(int nodeid)
149*4882a593Smuzhiyun {
150*4882a593Smuzhiyun return nodeid & (CONN_HASH_SIZE-1);
151*4882a593Smuzhiyun }
152*4882a593Smuzhiyun
__find_con(int nodeid)153*4882a593Smuzhiyun static struct connection *__find_con(int nodeid)
154*4882a593Smuzhiyun {
155*4882a593Smuzhiyun int r, idx;
156*4882a593Smuzhiyun struct connection *con;
157*4882a593Smuzhiyun
158*4882a593Smuzhiyun r = nodeid_hash(nodeid);
159*4882a593Smuzhiyun
160*4882a593Smuzhiyun idx = srcu_read_lock(&connections_srcu);
161*4882a593Smuzhiyun hlist_for_each_entry_rcu(con, &connection_hash[r], list) {
162*4882a593Smuzhiyun if (con->nodeid == nodeid) {
163*4882a593Smuzhiyun srcu_read_unlock(&connections_srcu, idx);
164*4882a593Smuzhiyun return con;
165*4882a593Smuzhiyun }
166*4882a593Smuzhiyun }
167*4882a593Smuzhiyun srcu_read_unlock(&connections_srcu, idx);
168*4882a593Smuzhiyun
169*4882a593Smuzhiyun return NULL;
170*4882a593Smuzhiyun }
171*4882a593Smuzhiyun
172*4882a593Smuzhiyun /*
173*4882a593Smuzhiyun * If 'allocation' is zero then we don't attempt to create a new
174*4882a593Smuzhiyun * connection structure for this node.
175*4882a593Smuzhiyun */
nodeid2con(int nodeid,gfp_t alloc)176*4882a593Smuzhiyun static struct connection *nodeid2con(int nodeid, gfp_t alloc)
177*4882a593Smuzhiyun {
178*4882a593Smuzhiyun struct connection *con, *tmp;
179*4882a593Smuzhiyun int r;
180*4882a593Smuzhiyun
181*4882a593Smuzhiyun con = __find_con(nodeid);
182*4882a593Smuzhiyun if (con || !alloc)
183*4882a593Smuzhiyun return con;
184*4882a593Smuzhiyun
185*4882a593Smuzhiyun con = kzalloc(sizeof(*con), alloc);
186*4882a593Smuzhiyun if (!con)
187*4882a593Smuzhiyun return NULL;
188*4882a593Smuzhiyun
189*4882a593Smuzhiyun con->rx_buflen = dlm_config.ci_buffer_size;
190*4882a593Smuzhiyun con->rx_buf = kmalloc(con->rx_buflen, GFP_NOFS);
191*4882a593Smuzhiyun if (!con->rx_buf) {
192*4882a593Smuzhiyun kfree(con);
193*4882a593Smuzhiyun return NULL;
194*4882a593Smuzhiyun }
195*4882a593Smuzhiyun
196*4882a593Smuzhiyun con->nodeid = nodeid;
197*4882a593Smuzhiyun mutex_init(&con->sock_mutex);
198*4882a593Smuzhiyun INIT_LIST_HEAD(&con->writequeue);
199*4882a593Smuzhiyun spin_lock_init(&con->writequeue_lock);
200*4882a593Smuzhiyun INIT_WORK(&con->swork, process_send_sockets);
201*4882a593Smuzhiyun INIT_WORK(&con->rwork, process_recv_sockets);
202*4882a593Smuzhiyun init_waitqueue_head(&con->shutdown_wait);
203*4882a593Smuzhiyun
204*4882a593Smuzhiyun /* Setup action pointers for child sockets */
205*4882a593Smuzhiyun if (con->nodeid) {
206*4882a593Smuzhiyun struct connection *zerocon = __find_con(0);
207*4882a593Smuzhiyun
208*4882a593Smuzhiyun con->connect_action = zerocon->connect_action;
209*4882a593Smuzhiyun if (!con->rx_action)
210*4882a593Smuzhiyun con->rx_action = zerocon->rx_action;
211*4882a593Smuzhiyun }
212*4882a593Smuzhiyun
213*4882a593Smuzhiyun r = nodeid_hash(nodeid);
214*4882a593Smuzhiyun
215*4882a593Smuzhiyun spin_lock(&connections_lock);
216*4882a593Smuzhiyun /* Because multiple workqueues/threads calls this function it can
217*4882a593Smuzhiyun * race on multiple cpu's. Instead of locking hot path __find_con()
218*4882a593Smuzhiyun * we just check in rare cases of recently added nodes again
219*4882a593Smuzhiyun * under protection of connections_lock. If this is the case we
220*4882a593Smuzhiyun * abort our connection creation and return the existing connection.
221*4882a593Smuzhiyun */
222*4882a593Smuzhiyun tmp = __find_con(nodeid);
223*4882a593Smuzhiyun if (tmp) {
224*4882a593Smuzhiyun spin_unlock(&connections_lock);
225*4882a593Smuzhiyun kfree(con->rx_buf);
226*4882a593Smuzhiyun kfree(con);
227*4882a593Smuzhiyun return tmp;
228*4882a593Smuzhiyun }
229*4882a593Smuzhiyun
230*4882a593Smuzhiyun hlist_add_head_rcu(&con->list, &connection_hash[r]);
231*4882a593Smuzhiyun spin_unlock(&connections_lock);
232*4882a593Smuzhiyun
233*4882a593Smuzhiyun return con;
234*4882a593Smuzhiyun }
235*4882a593Smuzhiyun
236*4882a593Smuzhiyun /* Loop round all connections */
foreach_conn(void (* conn_func)(struct connection * c))237*4882a593Smuzhiyun static void foreach_conn(void (*conn_func)(struct connection *c))
238*4882a593Smuzhiyun {
239*4882a593Smuzhiyun int i, idx;
240*4882a593Smuzhiyun struct connection *con;
241*4882a593Smuzhiyun
242*4882a593Smuzhiyun idx = srcu_read_lock(&connections_srcu);
243*4882a593Smuzhiyun for (i = 0; i < CONN_HASH_SIZE; i++) {
244*4882a593Smuzhiyun hlist_for_each_entry_rcu(con, &connection_hash[i], list)
245*4882a593Smuzhiyun conn_func(con);
246*4882a593Smuzhiyun }
247*4882a593Smuzhiyun srcu_read_unlock(&connections_srcu, idx);
248*4882a593Smuzhiyun }
249*4882a593Smuzhiyun
find_node_addr(int nodeid)250*4882a593Smuzhiyun static struct dlm_node_addr *find_node_addr(int nodeid)
251*4882a593Smuzhiyun {
252*4882a593Smuzhiyun struct dlm_node_addr *na;
253*4882a593Smuzhiyun
254*4882a593Smuzhiyun list_for_each_entry(na, &dlm_node_addrs, list) {
255*4882a593Smuzhiyun if (na->nodeid == nodeid)
256*4882a593Smuzhiyun return na;
257*4882a593Smuzhiyun }
258*4882a593Smuzhiyun return NULL;
259*4882a593Smuzhiyun }
260*4882a593Smuzhiyun
addr_compare(struct sockaddr_storage * x,struct sockaddr_storage * y)261*4882a593Smuzhiyun static int addr_compare(struct sockaddr_storage *x, struct sockaddr_storage *y)
262*4882a593Smuzhiyun {
263*4882a593Smuzhiyun switch (x->ss_family) {
264*4882a593Smuzhiyun case AF_INET: {
265*4882a593Smuzhiyun struct sockaddr_in *sinx = (struct sockaddr_in *)x;
266*4882a593Smuzhiyun struct sockaddr_in *siny = (struct sockaddr_in *)y;
267*4882a593Smuzhiyun if (sinx->sin_addr.s_addr != siny->sin_addr.s_addr)
268*4882a593Smuzhiyun return 0;
269*4882a593Smuzhiyun if (sinx->sin_port != siny->sin_port)
270*4882a593Smuzhiyun return 0;
271*4882a593Smuzhiyun break;
272*4882a593Smuzhiyun }
273*4882a593Smuzhiyun case AF_INET6: {
274*4882a593Smuzhiyun struct sockaddr_in6 *sinx = (struct sockaddr_in6 *)x;
275*4882a593Smuzhiyun struct sockaddr_in6 *siny = (struct sockaddr_in6 *)y;
276*4882a593Smuzhiyun if (!ipv6_addr_equal(&sinx->sin6_addr, &siny->sin6_addr))
277*4882a593Smuzhiyun return 0;
278*4882a593Smuzhiyun if (sinx->sin6_port != siny->sin6_port)
279*4882a593Smuzhiyun return 0;
280*4882a593Smuzhiyun break;
281*4882a593Smuzhiyun }
282*4882a593Smuzhiyun default:
283*4882a593Smuzhiyun return 0;
284*4882a593Smuzhiyun }
285*4882a593Smuzhiyun return 1;
286*4882a593Smuzhiyun }
287*4882a593Smuzhiyun
nodeid_to_addr(int nodeid,struct sockaddr_storage * sas_out,struct sockaddr * sa_out,bool try_new_addr)288*4882a593Smuzhiyun static int nodeid_to_addr(int nodeid, struct sockaddr_storage *sas_out,
289*4882a593Smuzhiyun struct sockaddr *sa_out, bool try_new_addr)
290*4882a593Smuzhiyun {
291*4882a593Smuzhiyun struct sockaddr_storage sas;
292*4882a593Smuzhiyun struct dlm_node_addr *na;
293*4882a593Smuzhiyun
294*4882a593Smuzhiyun if (!dlm_local_count)
295*4882a593Smuzhiyun return -1;
296*4882a593Smuzhiyun
297*4882a593Smuzhiyun spin_lock(&dlm_node_addrs_spin);
298*4882a593Smuzhiyun na = find_node_addr(nodeid);
299*4882a593Smuzhiyun if (na && na->addr_count) {
300*4882a593Smuzhiyun memcpy(&sas, na->addr[na->curr_addr_index],
301*4882a593Smuzhiyun sizeof(struct sockaddr_storage));
302*4882a593Smuzhiyun
303*4882a593Smuzhiyun if (try_new_addr) {
304*4882a593Smuzhiyun na->curr_addr_index++;
305*4882a593Smuzhiyun if (na->curr_addr_index == na->addr_count)
306*4882a593Smuzhiyun na->curr_addr_index = 0;
307*4882a593Smuzhiyun }
308*4882a593Smuzhiyun }
309*4882a593Smuzhiyun spin_unlock(&dlm_node_addrs_spin);
310*4882a593Smuzhiyun
311*4882a593Smuzhiyun if (!na)
312*4882a593Smuzhiyun return -EEXIST;
313*4882a593Smuzhiyun
314*4882a593Smuzhiyun if (!na->addr_count)
315*4882a593Smuzhiyun return -ENOENT;
316*4882a593Smuzhiyun
317*4882a593Smuzhiyun if (sas_out)
318*4882a593Smuzhiyun memcpy(sas_out, &sas, sizeof(struct sockaddr_storage));
319*4882a593Smuzhiyun
320*4882a593Smuzhiyun if (!sa_out)
321*4882a593Smuzhiyun return 0;
322*4882a593Smuzhiyun
323*4882a593Smuzhiyun if (dlm_local_addr[0]->ss_family == AF_INET) {
324*4882a593Smuzhiyun struct sockaddr_in *in4 = (struct sockaddr_in *) &sas;
325*4882a593Smuzhiyun struct sockaddr_in *ret4 = (struct sockaddr_in *) sa_out;
326*4882a593Smuzhiyun ret4->sin_addr.s_addr = in4->sin_addr.s_addr;
327*4882a593Smuzhiyun } else {
328*4882a593Smuzhiyun struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) &sas;
329*4882a593Smuzhiyun struct sockaddr_in6 *ret6 = (struct sockaddr_in6 *) sa_out;
330*4882a593Smuzhiyun ret6->sin6_addr = in6->sin6_addr;
331*4882a593Smuzhiyun }
332*4882a593Smuzhiyun
333*4882a593Smuzhiyun return 0;
334*4882a593Smuzhiyun }
335*4882a593Smuzhiyun
addr_to_nodeid(struct sockaddr_storage * addr,int * nodeid)336*4882a593Smuzhiyun static int addr_to_nodeid(struct sockaddr_storage *addr, int *nodeid)
337*4882a593Smuzhiyun {
338*4882a593Smuzhiyun struct dlm_node_addr *na;
339*4882a593Smuzhiyun int rv = -EEXIST;
340*4882a593Smuzhiyun int addr_i;
341*4882a593Smuzhiyun
342*4882a593Smuzhiyun spin_lock(&dlm_node_addrs_spin);
343*4882a593Smuzhiyun list_for_each_entry(na, &dlm_node_addrs, list) {
344*4882a593Smuzhiyun if (!na->addr_count)
345*4882a593Smuzhiyun continue;
346*4882a593Smuzhiyun
347*4882a593Smuzhiyun for (addr_i = 0; addr_i < na->addr_count; addr_i++) {
348*4882a593Smuzhiyun if (addr_compare(na->addr[addr_i], addr)) {
349*4882a593Smuzhiyun *nodeid = na->nodeid;
350*4882a593Smuzhiyun rv = 0;
351*4882a593Smuzhiyun goto unlock;
352*4882a593Smuzhiyun }
353*4882a593Smuzhiyun }
354*4882a593Smuzhiyun }
355*4882a593Smuzhiyun unlock:
356*4882a593Smuzhiyun spin_unlock(&dlm_node_addrs_spin);
357*4882a593Smuzhiyun return rv;
358*4882a593Smuzhiyun }
359*4882a593Smuzhiyun
dlm_lowcomms_addr(int nodeid,struct sockaddr_storage * addr,int len)360*4882a593Smuzhiyun int dlm_lowcomms_addr(int nodeid, struct sockaddr_storage *addr, int len)
361*4882a593Smuzhiyun {
362*4882a593Smuzhiyun struct sockaddr_storage *new_addr;
363*4882a593Smuzhiyun struct dlm_node_addr *new_node, *na;
364*4882a593Smuzhiyun
365*4882a593Smuzhiyun new_node = kzalloc(sizeof(struct dlm_node_addr), GFP_NOFS);
366*4882a593Smuzhiyun if (!new_node)
367*4882a593Smuzhiyun return -ENOMEM;
368*4882a593Smuzhiyun
369*4882a593Smuzhiyun new_addr = kzalloc(sizeof(struct sockaddr_storage), GFP_NOFS);
370*4882a593Smuzhiyun if (!new_addr) {
371*4882a593Smuzhiyun kfree(new_node);
372*4882a593Smuzhiyun return -ENOMEM;
373*4882a593Smuzhiyun }
374*4882a593Smuzhiyun
375*4882a593Smuzhiyun memcpy(new_addr, addr, len);
376*4882a593Smuzhiyun
377*4882a593Smuzhiyun spin_lock(&dlm_node_addrs_spin);
378*4882a593Smuzhiyun na = find_node_addr(nodeid);
379*4882a593Smuzhiyun if (!na) {
380*4882a593Smuzhiyun new_node->nodeid = nodeid;
381*4882a593Smuzhiyun new_node->addr[0] = new_addr;
382*4882a593Smuzhiyun new_node->addr_count = 1;
383*4882a593Smuzhiyun list_add(&new_node->list, &dlm_node_addrs);
384*4882a593Smuzhiyun spin_unlock(&dlm_node_addrs_spin);
385*4882a593Smuzhiyun return 0;
386*4882a593Smuzhiyun }
387*4882a593Smuzhiyun
388*4882a593Smuzhiyun if (na->addr_count >= DLM_MAX_ADDR_COUNT) {
389*4882a593Smuzhiyun spin_unlock(&dlm_node_addrs_spin);
390*4882a593Smuzhiyun kfree(new_addr);
391*4882a593Smuzhiyun kfree(new_node);
392*4882a593Smuzhiyun return -ENOSPC;
393*4882a593Smuzhiyun }
394*4882a593Smuzhiyun
395*4882a593Smuzhiyun na->addr[na->addr_count++] = new_addr;
396*4882a593Smuzhiyun spin_unlock(&dlm_node_addrs_spin);
397*4882a593Smuzhiyun kfree(new_node);
398*4882a593Smuzhiyun return 0;
399*4882a593Smuzhiyun }
400*4882a593Smuzhiyun
401*4882a593Smuzhiyun /* Data available on socket or listen socket received a connect */
lowcomms_data_ready(struct sock * sk)402*4882a593Smuzhiyun static void lowcomms_data_ready(struct sock *sk)
403*4882a593Smuzhiyun {
404*4882a593Smuzhiyun struct connection *con;
405*4882a593Smuzhiyun
406*4882a593Smuzhiyun read_lock_bh(&sk->sk_callback_lock);
407*4882a593Smuzhiyun con = sock2con(sk);
408*4882a593Smuzhiyun if (con && !test_and_set_bit(CF_READ_PENDING, &con->flags))
409*4882a593Smuzhiyun queue_work(recv_workqueue, &con->rwork);
410*4882a593Smuzhiyun read_unlock_bh(&sk->sk_callback_lock);
411*4882a593Smuzhiyun }
412*4882a593Smuzhiyun
lowcomms_write_space(struct sock * sk)413*4882a593Smuzhiyun static void lowcomms_write_space(struct sock *sk)
414*4882a593Smuzhiyun {
415*4882a593Smuzhiyun struct connection *con;
416*4882a593Smuzhiyun
417*4882a593Smuzhiyun read_lock_bh(&sk->sk_callback_lock);
418*4882a593Smuzhiyun con = sock2con(sk);
419*4882a593Smuzhiyun if (!con)
420*4882a593Smuzhiyun goto out;
421*4882a593Smuzhiyun
422*4882a593Smuzhiyun clear_bit(SOCK_NOSPACE, &con->sock->flags);
423*4882a593Smuzhiyun
424*4882a593Smuzhiyun if (test_and_clear_bit(CF_APP_LIMITED, &con->flags)) {
425*4882a593Smuzhiyun con->sock->sk->sk_write_pending--;
426*4882a593Smuzhiyun clear_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags);
427*4882a593Smuzhiyun }
428*4882a593Smuzhiyun
429*4882a593Smuzhiyun queue_work(send_workqueue, &con->swork);
430*4882a593Smuzhiyun out:
431*4882a593Smuzhiyun read_unlock_bh(&sk->sk_callback_lock);
432*4882a593Smuzhiyun }
433*4882a593Smuzhiyun
lowcomms_connect_sock(struct connection * con)434*4882a593Smuzhiyun static inline void lowcomms_connect_sock(struct connection *con)
435*4882a593Smuzhiyun {
436*4882a593Smuzhiyun if (test_bit(CF_CLOSE, &con->flags))
437*4882a593Smuzhiyun return;
438*4882a593Smuzhiyun queue_work(send_workqueue, &con->swork);
439*4882a593Smuzhiyun cond_resched();
440*4882a593Smuzhiyun }
441*4882a593Smuzhiyun
lowcomms_state_change(struct sock * sk)442*4882a593Smuzhiyun static void lowcomms_state_change(struct sock *sk)
443*4882a593Smuzhiyun {
444*4882a593Smuzhiyun /* SCTP layer is not calling sk_data_ready when the connection
445*4882a593Smuzhiyun * is done, so we catch the signal through here. Also, it
446*4882a593Smuzhiyun * doesn't switch socket state when entering shutdown, so we
447*4882a593Smuzhiyun * skip the write in that case.
448*4882a593Smuzhiyun */
449*4882a593Smuzhiyun if (sk->sk_shutdown) {
450*4882a593Smuzhiyun if (sk->sk_shutdown == RCV_SHUTDOWN)
451*4882a593Smuzhiyun lowcomms_data_ready(sk);
452*4882a593Smuzhiyun } else if (sk->sk_state == TCP_ESTABLISHED) {
453*4882a593Smuzhiyun lowcomms_write_space(sk);
454*4882a593Smuzhiyun }
455*4882a593Smuzhiyun }
456*4882a593Smuzhiyun
dlm_lowcomms_connect_node(int nodeid)457*4882a593Smuzhiyun int dlm_lowcomms_connect_node(int nodeid)
458*4882a593Smuzhiyun {
459*4882a593Smuzhiyun struct connection *con;
460*4882a593Smuzhiyun
461*4882a593Smuzhiyun if (nodeid == dlm_our_nodeid())
462*4882a593Smuzhiyun return 0;
463*4882a593Smuzhiyun
464*4882a593Smuzhiyun con = nodeid2con(nodeid, GFP_NOFS);
465*4882a593Smuzhiyun if (!con)
466*4882a593Smuzhiyun return -ENOMEM;
467*4882a593Smuzhiyun lowcomms_connect_sock(con);
468*4882a593Smuzhiyun return 0;
469*4882a593Smuzhiyun }
470*4882a593Smuzhiyun
lowcomms_error_report(struct sock * sk)471*4882a593Smuzhiyun static void lowcomms_error_report(struct sock *sk)
472*4882a593Smuzhiyun {
473*4882a593Smuzhiyun struct connection *con;
474*4882a593Smuzhiyun void (*orig_report)(struct sock *) = NULL;
475*4882a593Smuzhiyun struct inet_sock *inet;
476*4882a593Smuzhiyun
477*4882a593Smuzhiyun read_lock_bh(&sk->sk_callback_lock);
478*4882a593Smuzhiyun con = sock2con(sk);
479*4882a593Smuzhiyun if (con == NULL)
480*4882a593Smuzhiyun goto out;
481*4882a593Smuzhiyun
482*4882a593Smuzhiyun orig_report = listen_sock.sk_error_report;
483*4882a593Smuzhiyun
484*4882a593Smuzhiyun inet = inet_sk(sk);
485*4882a593Smuzhiyun switch (sk->sk_family) {
486*4882a593Smuzhiyun case AF_INET:
487*4882a593Smuzhiyun printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
488*4882a593Smuzhiyun "sending to node %d at %pI4, dport %d, "
489*4882a593Smuzhiyun "sk_err=%d/%d\n", dlm_our_nodeid(),
490*4882a593Smuzhiyun con->nodeid, &inet->inet_daddr,
491*4882a593Smuzhiyun ntohs(inet->inet_dport), sk->sk_err,
492*4882a593Smuzhiyun sk->sk_err_soft);
493*4882a593Smuzhiyun break;
494*4882a593Smuzhiyun #if IS_ENABLED(CONFIG_IPV6)
495*4882a593Smuzhiyun case AF_INET6:
496*4882a593Smuzhiyun printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
497*4882a593Smuzhiyun "sending to node %d at %pI6c, "
498*4882a593Smuzhiyun "dport %d, sk_err=%d/%d\n", dlm_our_nodeid(),
499*4882a593Smuzhiyun con->nodeid, &sk->sk_v6_daddr,
500*4882a593Smuzhiyun ntohs(inet->inet_dport), sk->sk_err,
501*4882a593Smuzhiyun sk->sk_err_soft);
502*4882a593Smuzhiyun break;
503*4882a593Smuzhiyun #endif
504*4882a593Smuzhiyun default:
505*4882a593Smuzhiyun printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
506*4882a593Smuzhiyun "invalid socket family %d set, "
507*4882a593Smuzhiyun "sk_err=%d/%d\n", dlm_our_nodeid(),
508*4882a593Smuzhiyun sk->sk_family, sk->sk_err, sk->sk_err_soft);
509*4882a593Smuzhiyun goto out;
510*4882a593Smuzhiyun }
511*4882a593Smuzhiyun out:
512*4882a593Smuzhiyun read_unlock_bh(&sk->sk_callback_lock);
513*4882a593Smuzhiyun if (orig_report)
514*4882a593Smuzhiyun orig_report(sk);
515*4882a593Smuzhiyun }
516*4882a593Smuzhiyun
517*4882a593Smuzhiyun /* Note: sk_callback_lock must be locked before calling this function. */
save_listen_callbacks(struct socket * sock)518*4882a593Smuzhiyun static void save_listen_callbacks(struct socket *sock)
519*4882a593Smuzhiyun {
520*4882a593Smuzhiyun struct sock *sk = sock->sk;
521*4882a593Smuzhiyun
522*4882a593Smuzhiyun listen_sock.sk_data_ready = sk->sk_data_ready;
523*4882a593Smuzhiyun listen_sock.sk_state_change = sk->sk_state_change;
524*4882a593Smuzhiyun listen_sock.sk_write_space = sk->sk_write_space;
525*4882a593Smuzhiyun listen_sock.sk_error_report = sk->sk_error_report;
526*4882a593Smuzhiyun }
527*4882a593Smuzhiyun
restore_callbacks(struct socket * sock)528*4882a593Smuzhiyun static void restore_callbacks(struct socket *sock)
529*4882a593Smuzhiyun {
530*4882a593Smuzhiyun struct sock *sk = sock->sk;
531*4882a593Smuzhiyun
532*4882a593Smuzhiyun write_lock_bh(&sk->sk_callback_lock);
533*4882a593Smuzhiyun sk->sk_user_data = NULL;
534*4882a593Smuzhiyun sk->sk_data_ready = listen_sock.sk_data_ready;
535*4882a593Smuzhiyun sk->sk_state_change = listen_sock.sk_state_change;
536*4882a593Smuzhiyun sk->sk_write_space = listen_sock.sk_write_space;
537*4882a593Smuzhiyun sk->sk_error_report = listen_sock.sk_error_report;
538*4882a593Smuzhiyun write_unlock_bh(&sk->sk_callback_lock);
539*4882a593Smuzhiyun }
540*4882a593Smuzhiyun
541*4882a593Smuzhiyun /* Make a socket active */
add_sock(struct socket * sock,struct connection * con)542*4882a593Smuzhiyun static void add_sock(struct socket *sock, struct connection *con)
543*4882a593Smuzhiyun {
544*4882a593Smuzhiyun struct sock *sk = sock->sk;
545*4882a593Smuzhiyun
546*4882a593Smuzhiyun write_lock_bh(&sk->sk_callback_lock);
547*4882a593Smuzhiyun con->sock = sock;
548*4882a593Smuzhiyun
549*4882a593Smuzhiyun sk->sk_user_data = con;
550*4882a593Smuzhiyun /* Install a data_ready callback */
551*4882a593Smuzhiyun sk->sk_data_ready = lowcomms_data_ready;
552*4882a593Smuzhiyun sk->sk_write_space = lowcomms_write_space;
553*4882a593Smuzhiyun sk->sk_state_change = lowcomms_state_change;
554*4882a593Smuzhiyun sk->sk_allocation = GFP_NOFS;
555*4882a593Smuzhiyun sk->sk_error_report = lowcomms_error_report;
556*4882a593Smuzhiyun write_unlock_bh(&sk->sk_callback_lock);
557*4882a593Smuzhiyun }
558*4882a593Smuzhiyun
559*4882a593Smuzhiyun /* Add the port number to an IPv6 or 4 sockaddr and return the address
560*4882a593Smuzhiyun length */
make_sockaddr(struct sockaddr_storage * saddr,uint16_t port,int * addr_len)561*4882a593Smuzhiyun static void make_sockaddr(struct sockaddr_storage *saddr, uint16_t port,
562*4882a593Smuzhiyun int *addr_len)
563*4882a593Smuzhiyun {
564*4882a593Smuzhiyun saddr->ss_family = dlm_local_addr[0]->ss_family;
565*4882a593Smuzhiyun if (saddr->ss_family == AF_INET) {
566*4882a593Smuzhiyun struct sockaddr_in *in4_addr = (struct sockaddr_in *)saddr;
567*4882a593Smuzhiyun in4_addr->sin_port = cpu_to_be16(port);
568*4882a593Smuzhiyun *addr_len = sizeof(struct sockaddr_in);
569*4882a593Smuzhiyun memset(&in4_addr->sin_zero, 0, sizeof(in4_addr->sin_zero));
570*4882a593Smuzhiyun } else {
571*4882a593Smuzhiyun struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)saddr;
572*4882a593Smuzhiyun in6_addr->sin6_port = cpu_to_be16(port);
573*4882a593Smuzhiyun *addr_len = sizeof(struct sockaddr_in6);
574*4882a593Smuzhiyun }
575*4882a593Smuzhiyun memset((char *)saddr + *addr_len, 0, sizeof(struct sockaddr_storage) - *addr_len);
576*4882a593Smuzhiyun }
577*4882a593Smuzhiyun
578*4882a593Smuzhiyun /* Close a remote connection and tidy up */
close_connection(struct connection * con,bool and_other,bool tx,bool rx)579*4882a593Smuzhiyun static void close_connection(struct connection *con, bool and_other,
580*4882a593Smuzhiyun bool tx, bool rx)
581*4882a593Smuzhiyun {
582*4882a593Smuzhiyun bool closing = test_and_set_bit(CF_CLOSING, &con->flags);
583*4882a593Smuzhiyun
584*4882a593Smuzhiyun if (tx && !closing && cancel_work_sync(&con->swork)) {
585*4882a593Smuzhiyun log_print("canceled swork for node %d", con->nodeid);
586*4882a593Smuzhiyun clear_bit(CF_WRITE_PENDING, &con->flags);
587*4882a593Smuzhiyun }
588*4882a593Smuzhiyun if (rx && !closing && cancel_work_sync(&con->rwork)) {
589*4882a593Smuzhiyun log_print("canceled rwork for node %d", con->nodeid);
590*4882a593Smuzhiyun clear_bit(CF_READ_PENDING, &con->flags);
591*4882a593Smuzhiyun }
592*4882a593Smuzhiyun
593*4882a593Smuzhiyun mutex_lock(&con->sock_mutex);
594*4882a593Smuzhiyun if (con->sock) {
595*4882a593Smuzhiyun restore_callbacks(con->sock);
596*4882a593Smuzhiyun sock_release(con->sock);
597*4882a593Smuzhiyun con->sock = NULL;
598*4882a593Smuzhiyun }
599*4882a593Smuzhiyun if (con->othercon && and_other) {
600*4882a593Smuzhiyun /* Will only re-enter once. */
601*4882a593Smuzhiyun close_connection(con->othercon, false, tx, rx);
602*4882a593Smuzhiyun }
603*4882a593Smuzhiyun
604*4882a593Smuzhiyun con->rx_leftover = 0;
605*4882a593Smuzhiyun con->retries = 0;
606*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
607*4882a593Smuzhiyun clear_bit(CF_CLOSING, &con->flags);
608*4882a593Smuzhiyun }
609*4882a593Smuzhiyun
shutdown_connection(struct connection * con)610*4882a593Smuzhiyun static void shutdown_connection(struct connection *con)
611*4882a593Smuzhiyun {
612*4882a593Smuzhiyun int ret;
613*4882a593Smuzhiyun
614*4882a593Smuzhiyun flush_work(&con->swork);
615*4882a593Smuzhiyun
616*4882a593Smuzhiyun mutex_lock(&con->sock_mutex);
617*4882a593Smuzhiyun /* nothing to shutdown */
618*4882a593Smuzhiyun if (!con->sock) {
619*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
620*4882a593Smuzhiyun return;
621*4882a593Smuzhiyun }
622*4882a593Smuzhiyun
623*4882a593Smuzhiyun set_bit(CF_SHUTDOWN, &con->flags);
624*4882a593Smuzhiyun ret = kernel_sock_shutdown(con->sock, SHUT_WR);
625*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
626*4882a593Smuzhiyun if (ret) {
627*4882a593Smuzhiyun log_print("Connection %p failed to shutdown: %d will force close",
628*4882a593Smuzhiyun con, ret);
629*4882a593Smuzhiyun goto force_close;
630*4882a593Smuzhiyun } else {
631*4882a593Smuzhiyun ret = wait_event_timeout(con->shutdown_wait,
632*4882a593Smuzhiyun !test_bit(CF_SHUTDOWN, &con->flags),
633*4882a593Smuzhiyun DLM_SHUTDOWN_WAIT_TIMEOUT);
634*4882a593Smuzhiyun if (ret == 0) {
635*4882a593Smuzhiyun log_print("Connection %p shutdown timed out, will force close",
636*4882a593Smuzhiyun con);
637*4882a593Smuzhiyun goto force_close;
638*4882a593Smuzhiyun }
639*4882a593Smuzhiyun }
640*4882a593Smuzhiyun
641*4882a593Smuzhiyun return;
642*4882a593Smuzhiyun
643*4882a593Smuzhiyun force_close:
644*4882a593Smuzhiyun clear_bit(CF_SHUTDOWN, &con->flags);
645*4882a593Smuzhiyun close_connection(con, false, true, true);
646*4882a593Smuzhiyun }
647*4882a593Smuzhiyun
dlm_tcp_shutdown(struct connection * con)648*4882a593Smuzhiyun static void dlm_tcp_shutdown(struct connection *con)
649*4882a593Smuzhiyun {
650*4882a593Smuzhiyun if (con->othercon)
651*4882a593Smuzhiyun shutdown_connection(con->othercon);
652*4882a593Smuzhiyun shutdown_connection(con);
653*4882a593Smuzhiyun }
654*4882a593Smuzhiyun
con_realloc_receive_buf(struct connection * con,int newlen)655*4882a593Smuzhiyun static int con_realloc_receive_buf(struct connection *con, int newlen)
656*4882a593Smuzhiyun {
657*4882a593Smuzhiyun unsigned char *newbuf;
658*4882a593Smuzhiyun
659*4882a593Smuzhiyun newbuf = kmalloc(newlen, GFP_NOFS);
660*4882a593Smuzhiyun if (!newbuf)
661*4882a593Smuzhiyun return -ENOMEM;
662*4882a593Smuzhiyun
663*4882a593Smuzhiyun /* copy any leftover from last receive */
664*4882a593Smuzhiyun if (con->rx_leftover)
665*4882a593Smuzhiyun memmove(newbuf, con->rx_buf, con->rx_leftover);
666*4882a593Smuzhiyun
667*4882a593Smuzhiyun /* swap to new buffer space */
668*4882a593Smuzhiyun kfree(con->rx_buf);
669*4882a593Smuzhiyun con->rx_buflen = newlen;
670*4882a593Smuzhiyun con->rx_buf = newbuf;
671*4882a593Smuzhiyun
672*4882a593Smuzhiyun return 0;
673*4882a593Smuzhiyun }
674*4882a593Smuzhiyun
675*4882a593Smuzhiyun /* Data received from remote end */
receive_from_sock(struct connection * con)676*4882a593Smuzhiyun static int receive_from_sock(struct connection *con)
677*4882a593Smuzhiyun {
678*4882a593Smuzhiyun int call_again_soon = 0;
679*4882a593Smuzhiyun struct msghdr msg;
680*4882a593Smuzhiyun struct kvec iov;
681*4882a593Smuzhiyun int ret, buflen;
682*4882a593Smuzhiyun
683*4882a593Smuzhiyun mutex_lock(&con->sock_mutex);
684*4882a593Smuzhiyun
685*4882a593Smuzhiyun if (con->sock == NULL) {
686*4882a593Smuzhiyun ret = -EAGAIN;
687*4882a593Smuzhiyun goto out_close;
688*4882a593Smuzhiyun }
689*4882a593Smuzhiyun
690*4882a593Smuzhiyun if (con->nodeid == 0) {
691*4882a593Smuzhiyun ret = -EINVAL;
692*4882a593Smuzhiyun goto out_close;
693*4882a593Smuzhiyun }
694*4882a593Smuzhiyun
695*4882a593Smuzhiyun /* realloc if we get new buffer size to read out */
696*4882a593Smuzhiyun buflen = dlm_config.ci_buffer_size;
697*4882a593Smuzhiyun if (con->rx_buflen != buflen && con->rx_leftover <= buflen) {
698*4882a593Smuzhiyun ret = con_realloc_receive_buf(con, buflen);
699*4882a593Smuzhiyun if (ret < 0)
700*4882a593Smuzhiyun goto out_resched;
701*4882a593Smuzhiyun }
702*4882a593Smuzhiyun
703*4882a593Smuzhiyun /* calculate new buffer parameter regarding last receive and
704*4882a593Smuzhiyun * possible leftover bytes
705*4882a593Smuzhiyun */
706*4882a593Smuzhiyun iov.iov_base = con->rx_buf + con->rx_leftover;
707*4882a593Smuzhiyun iov.iov_len = con->rx_buflen - con->rx_leftover;
708*4882a593Smuzhiyun
709*4882a593Smuzhiyun memset(&msg, 0, sizeof(msg));
710*4882a593Smuzhiyun msg.msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL;
711*4882a593Smuzhiyun ret = kernel_recvmsg(con->sock, &msg, &iov, 1, iov.iov_len,
712*4882a593Smuzhiyun msg.msg_flags);
713*4882a593Smuzhiyun if (ret <= 0)
714*4882a593Smuzhiyun goto out_close;
715*4882a593Smuzhiyun else if (ret == iov.iov_len)
716*4882a593Smuzhiyun call_again_soon = 1;
717*4882a593Smuzhiyun
718*4882a593Smuzhiyun /* new buflen according readed bytes and leftover from last receive */
719*4882a593Smuzhiyun buflen = ret + con->rx_leftover;
720*4882a593Smuzhiyun ret = dlm_process_incoming_buffer(con->nodeid, con->rx_buf, buflen);
721*4882a593Smuzhiyun if (ret < 0)
722*4882a593Smuzhiyun goto out_close;
723*4882a593Smuzhiyun
724*4882a593Smuzhiyun /* calculate leftover bytes from process and put it into begin of
725*4882a593Smuzhiyun * the receive buffer, so next receive we have the full message
726*4882a593Smuzhiyun * at the start address of the receive buffer.
727*4882a593Smuzhiyun */
728*4882a593Smuzhiyun con->rx_leftover = buflen - ret;
729*4882a593Smuzhiyun if (con->rx_leftover) {
730*4882a593Smuzhiyun memmove(con->rx_buf, con->rx_buf + ret,
731*4882a593Smuzhiyun con->rx_leftover);
732*4882a593Smuzhiyun call_again_soon = true;
733*4882a593Smuzhiyun }
734*4882a593Smuzhiyun
735*4882a593Smuzhiyun if (call_again_soon)
736*4882a593Smuzhiyun goto out_resched;
737*4882a593Smuzhiyun
738*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
739*4882a593Smuzhiyun return 0;
740*4882a593Smuzhiyun
741*4882a593Smuzhiyun out_resched:
742*4882a593Smuzhiyun if (!test_and_set_bit(CF_READ_PENDING, &con->flags))
743*4882a593Smuzhiyun queue_work(recv_workqueue, &con->rwork);
744*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
745*4882a593Smuzhiyun return -EAGAIN;
746*4882a593Smuzhiyun
747*4882a593Smuzhiyun out_close:
748*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
749*4882a593Smuzhiyun if (ret != -EAGAIN) {
750*4882a593Smuzhiyun /* Reconnect when there is something to send */
751*4882a593Smuzhiyun close_connection(con, false, true, false);
752*4882a593Smuzhiyun if (ret == 0) {
753*4882a593Smuzhiyun log_print("connection %p got EOF from %d",
754*4882a593Smuzhiyun con, con->nodeid);
755*4882a593Smuzhiyun /* handling for tcp shutdown */
756*4882a593Smuzhiyun clear_bit(CF_SHUTDOWN, &con->flags);
757*4882a593Smuzhiyun wake_up(&con->shutdown_wait);
758*4882a593Smuzhiyun /* signal to breaking receive worker */
759*4882a593Smuzhiyun ret = -1;
760*4882a593Smuzhiyun }
761*4882a593Smuzhiyun }
762*4882a593Smuzhiyun return ret;
763*4882a593Smuzhiyun }
764*4882a593Smuzhiyun
765*4882a593Smuzhiyun /* Listening socket is busy, accept a connection */
accept_from_sock(struct connection * con)766*4882a593Smuzhiyun static int accept_from_sock(struct connection *con)
767*4882a593Smuzhiyun {
768*4882a593Smuzhiyun int result;
769*4882a593Smuzhiyun struct sockaddr_storage peeraddr;
770*4882a593Smuzhiyun struct socket *newsock;
771*4882a593Smuzhiyun int len;
772*4882a593Smuzhiyun int nodeid;
773*4882a593Smuzhiyun struct connection *newcon;
774*4882a593Smuzhiyun struct connection *addcon;
775*4882a593Smuzhiyun unsigned int mark;
776*4882a593Smuzhiyun
777*4882a593Smuzhiyun if (!dlm_allow_conn) {
778*4882a593Smuzhiyun return -1;
779*4882a593Smuzhiyun }
780*4882a593Smuzhiyun
781*4882a593Smuzhiyun mutex_lock_nested(&con->sock_mutex, 0);
782*4882a593Smuzhiyun
783*4882a593Smuzhiyun if (!con->sock) {
784*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
785*4882a593Smuzhiyun return -ENOTCONN;
786*4882a593Smuzhiyun }
787*4882a593Smuzhiyun
788*4882a593Smuzhiyun result = kernel_accept(con->sock, &newsock, O_NONBLOCK);
789*4882a593Smuzhiyun if (result < 0)
790*4882a593Smuzhiyun goto accept_err;
791*4882a593Smuzhiyun
792*4882a593Smuzhiyun /* Get the connected socket's peer */
793*4882a593Smuzhiyun memset(&peeraddr, 0, sizeof(peeraddr));
794*4882a593Smuzhiyun len = newsock->ops->getname(newsock, (struct sockaddr *)&peeraddr, 2);
795*4882a593Smuzhiyun if (len < 0) {
796*4882a593Smuzhiyun result = -ECONNABORTED;
797*4882a593Smuzhiyun goto accept_err;
798*4882a593Smuzhiyun }
799*4882a593Smuzhiyun
800*4882a593Smuzhiyun /* Get the new node's NODEID */
801*4882a593Smuzhiyun make_sockaddr(&peeraddr, 0, &len);
802*4882a593Smuzhiyun if (addr_to_nodeid(&peeraddr, &nodeid)) {
803*4882a593Smuzhiyun unsigned char *b=(unsigned char *)&peeraddr;
804*4882a593Smuzhiyun log_print("connect from non cluster node");
805*4882a593Smuzhiyun print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE,
806*4882a593Smuzhiyun b, sizeof(struct sockaddr_storage));
807*4882a593Smuzhiyun sock_release(newsock);
808*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
809*4882a593Smuzhiyun return -1;
810*4882a593Smuzhiyun }
811*4882a593Smuzhiyun
812*4882a593Smuzhiyun dlm_comm_mark(nodeid, &mark);
813*4882a593Smuzhiyun sock_set_mark(newsock->sk, mark);
814*4882a593Smuzhiyun
815*4882a593Smuzhiyun log_print("got connection from %d", nodeid);
816*4882a593Smuzhiyun
817*4882a593Smuzhiyun /* Check to see if we already have a connection to this node. This
818*4882a593Smuzhiyun * could happen if the two nodes initiate a connection at roughly
819*4882a593Smuzhiyun * the same time and the connections cross on the wire.
820*4882a593Smuzhiyun * In this case we store the incoming one in "othercon"
821*4882a593Smuzhiyun */
822*4882a593Smuzhiyun newcon = nodeid2con(nodeid, GFP_NOFS);
823*4882a593Smuzhiyun if (!newcon) {
824*4882a593Smuzhiyun result = -ENOMEM;
825*4882a593Smuzhiyun goto accept_err;
826*4882a593Smuzhiyun }
827*4882a593Smuzhiyun mutex_lock_nested(&newcon->sock_mutex, 1);
828*4882a593Smuzhiyun if (newcon->sock) {
829*4882a593Smuzhiyun struct connection *othercon = newcon->othercon;
830*4882a593Smuzhiyun
831*4882a593Smuzhiyun if (!othercon) {
832*4882a593Smuzhiyun othercon = kzalloc(sizeof(*othercon), GFP_NOFS);
833*4882a593Smuzhiyun if (!othercon) {
834*4882a593Smuzhiyun log_print("failed to allocate incoming socket");
835*4882a593Smuzhiyun mutex_unlock(&newcon->sock_mutex);
836*4882a593Smuzhiyun result = -ENOMEM;
837*4882a593Smuzhiyun goto accept_err;
838*4882a593Smuzhiyun }
839*4882a593Smuzhiyun
840*4882a593Smuzhiyun othercon->rx_buflen = dlm_config.ci_buffer_size;
841*4882a593Smuzhiyun othercon->rx_buf = kmalloc(othercon->rx_buflen, GFP_NOFS);
842*4882a593Smuzhiyun if (!othercon->rx_buf) {
843*4882a593Smuzhiyun mutex_unlock(&newcon->sock_mutex);
844*4882a593Smuzhiyun kfree(othercon);
845*4882a593Smuzhiyun log_print("failed to allocate incoming socket receive buffer");
846*4882a593Smuzhiyun result = -ENOMEM;
847*4882a593Smuzhiyun goto accept_err;
848*4882a593Smuzhiyun }
849*4882a593Smuzhiyun
850*4882a593Smuzhiyun othercon->nodeid = nodeid;
851*4882a593Smuzhiyun othercon->rx_action = receive_from_sock;
852*4882a593Smuzhiyun mutex_init(&othercon->sock_mutex);
853*4882a593Smuzhiyun INIT_LIST_HEAD(&othercon->writequeue);
854*4882a593Smuzhiyun spin_lock_init(&othercon->writequeue_lock);
855*4882a593Smuzhiyun INIT_WORK(&othercon->swork, process_send_sockets);
856*4882a593Smuzhiyun INIT_WORK(&othercon->rwork, process_recv_sockets);
857*4882a593Smuzhiyun init_waitqueue_head(&othercon->shutdown_wait);
858*4882a593Smuzhiyun set_bit(CF_IS_OTHERCON, &othercon->flags);
859*4882a593Smuzhiyun } else {
860*4882a593Smuzhiyun /* close other sock con if we have something new */
861*4882a593Smuzhiyun close_connection(othercon, false, true, false);
862*4882a593Smuzhiyun }
863*4882a593Smuzhiyun
864*4882a593Smuzhiyun mutex_lock_nested(&othercon->sock_mutex, 2);
865*4882a593Smuzhiyun newcon->othercon = othercon;
866*4882a593Smuzhiyun add_sock(newsock, othercon);
867*4882a593Smuzhiyun addcon = othercon;
868*4882a593Smuzhiyun mutex_unlock(&othercon->sock_mutex);
869*4882a593Smuzhiyun }
870*4882a593Smuzhiyun else {
871*4882a593Smuzhiyun newcon->rx_action = receive_from_sock;
872*4882a593Smuzhiyun /* accept copies the sk after we've saved the callbacks, so we
873*4882a593Smuzhiyun don't want to save them a second time or comm errors will
874*4882a593Smuzhiyun result in calling sk_error_report recursively. */
875*4882a593Smuzhiyun add_sock(newsock, newcon);
876*4882a593Smuzhiyun addcon = newcon;
877*4882a593Smuzhiyun }
878*4882a593Smuzhiyun
879*4882a593Smuzhiyun mutex_unlock(&newcon->sock_mutex);
880*4882a593Smuzhiyun
881*4882a593Smuzhiyun /*
882*4882a593Smuzhiyun * Add it to the active queue in case we got data
883*4882a593Smuzhiyun * between processing the accept adding the socket
884*4882a593Smuzhiyun * to the read_sockets list
885*4882a593Smuzhiyun */
886*4882a593Smuzhiyun if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags))
887*4882a593Smuzhiyun queue_work(recv_workqueue, &addcon->rwork);
888*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
889*4882a593Smuzhiyun
890*4882a593Smuzhiyun return 0;
891*4882a593Smuzhiyun
892*4882a593Smuzhiyun accept_err:
893*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
894*4882a593Smuzhiyun if (newsock)
895*4882a593Smuzhiyun sock_release(newsock);
896*4882a593Smuzhiyun
897*4882a593Smuzhiyun if (result != -EAGAIN)
898*4882a593Smuzhiyun log_print("error accepting connection from node: %d", result);
899*4882a593Smuzhiyun return result;
900*4882a593Smuzhiyun }
901*4882a593Smuzhiyun
free_entry(struct writequeue_entry * e)902*4882a593Smuzhiyun static void free_entry(struct writequeue_entry *e)
903*4882a593Smuzhiyun {
904*4882a593Smuzhiyun __free_page(e->page);
905*4882a593Smuzhiyun kfree(e);
906*4882a593Smuzhiyun }
907*4882a593Smuzhiyun
908*4882a593Smuzhiyun /*
909*4882a593Smuzhiyun * writequeue_entry_complete - try to delete and free write queue entry
910*4882a593Smuzhiyun * @e: write queue entry to try to delete
911*4882a593Smuzhiyun * @completed: bytes completed
912*4882a593Smuzhiyun *
913*4882a593Smuzhiyun * writequeue_lock must be held.
914*4882a593Smuzhiyun */
writequeue_entry_complete(struct writequeue_entry * e,int completed)915*4882a593Smuzhiyun static void writequeue_entry_complete(struct writequeue_entry *e, int completed)
916*4882a593Smuzhiyun {
917*4882a593Smuzhiyun e->offset += completed;
918*4882a593Smuzhiyun e->len -= completed;
919*4882a593Smuzhiyun
920*4882a593Smuzhiyun if (e->len == 0 && e->users == 0) {
921*4882a593Smuzhiyun list_del(&e->list);
922*4882a593Smuzhiyun free_entry(e);
923*4882a593Smuzhiyun }
924*4882a593Smuzhiyun }
925*4882a593Smuzhiyun
926*4882a593Smuzhiyun /*
927*4882a593Smuzhiyun * sctp_bind_addrs - bind a SCTP socket to all our addresses
928*4882a593Smuzhiyun */
sctp_bind_addrs(struct connection * con,uint16_t port)929*4882a593Smuzhiyun static int sctp_bind_addrs(struct connection *con, uint16_t port)
930*4882a593Smuzhiyun {
931*4882a593Smuzhiyun struct sockaddr_storage localaddr;
932*4882a593Smuzhiyun struct sockaddr *addr = (struct sockaddr *)&localaddr;
933*4882a593Smuzhiyun int i, addr_len, result = 0;
934*4882a593Smuzhiyun
935*4882a593Smuzhiyun for (i = 0; i < dlm_local_count; i++) {
936*4882a593Smuzhiyun memcpy(&localaddr, dlm_local_addr[i], sizeof(localaddr));
937*4882a593Smuzhiyun make_sockaddr(&localaddr, port, &addr_len);
938*4882a593Smuzhiyun
939*4882a593Smuzhiyun if (!i)
940*4882a593Smuzhiyun result = kernel_bind(con->sock, addr, addr_len);
941*4882a593Smuzhiyun else
942*4882a593Smuzhiyun result = sock_bind_add(con->sock->sk, addr, addr_len);
943*4882a593Smuzhiyun
944*4882a593Smuzhiyun if (result < 0) {
945*4882a593Smuzhiyun log_print("Can't bind to %d addr number %d, %d.\n",
946*4882a593Smuzhiyun port, i + 1, result);
947*4882a593Smuzhiyun break;
948*4882a593Smuzhiyun }
949*4882a593Smuzhiyun }
950*4882a593Smuzhiyun return result;
951*4882a593Smuzhiyun }
952*4882a593Smuzhiyun
953*4882a593Smuzhiyun /* Initiate an SCTP association.
954*4882a593Smuzhiyun This is a special case of send_to_sock() in that we don't yet have a
955*4882a593Smuzhiyun peeled-off socket for this association, so we use the listening socket
956*4882a593Smuzhiyun and add the primary IP address of the remote node.
957*4882a593Smuzhiyun */
sctp_connect_to_sock(struct connection * con)958*4882a593Smuzhiyun static void sctp_connect_to_sock(struct connection *con)
959*4882a593Smuzhiyun {
960*4882a593Smuzhiyun struct sockaddr_storage daddr;
961*4882a593Smuzhiyun int result;
962*4882a593Smuzhiyun int addr_len;
963*4882a593Smuzhiyun struct socket *sock;
964*4882a593Smuzhiyun unsigned int mark;
965*4882a593Smuzhiyun
966*4882a593Smuzhiyun if (con->nodeid == 0) {
967*4882a593Smuzhiyun log_print("attempt to connect sock 0 foiled");
968*4882a593Smuzhiyun return;
969*4882a593Smuzhiyun }
970*4882a593Smuzhiyun
971*4882a593Smuzhiyun dlm_comm_mark(con->nodeid, &mark);
972*4882a593Smuzhiyun
973*4882a593Smuzhiyun mutex_lock(&con->sock_mutex);
974*4882a593Smuzhiyun
975*4882a593Smuzhiyun /* Some odd races can cause double-connects, ignore them */
976*4882a593Smuzhiyun if (con->retries++ > MAX_CONNECT_RETRIES)
977*4882a593Smuzhiyun goto out;
978*4882a593Smuzhiyun
979*4882a593Smuzhiyun if (con->sock) {
980*4882a593Smuzhiyun log_print("node %d already connected.", con->nodeid);
981*4882a593Smuzhiyun goto out;
982*4882a593Smuzhiyun }
983*4882a593Smuzhiyun
984*4882a593Smuzhiyun memset(&daddr, 0, sizeof(daddr));
985*4882a593Smuzhiyun result = nodeid_to_addr(con->nodeid, &daddr, NULL, true);
986*4882a593Smuzhiyun if (result < 0) {
987*4882a593Smuzhiyun log_print("no address for nodeid %d", con->nodeid);
988*4882a593Smuzhiyun goto out;
989*4882a593Smuzhiyun }
990*4882a593Smuzhiyun
991*4882a593Smuzhiyun /* Create a socket to communicate with */
992*4882a593Smuzhiyun result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
993*4882a593Smuzhiyun SOCK_STREAM, IPPROTO_SCTP, &sock);
994*4882a593Smuzhiyun if (result < 0)
995*4882a593Smuzhiyun goto socket_err;
996*4882a593Smuzhiyun
997*4882a593Smuzhiyun sock_set_mark(sock->sk, mark);
998*4882a593Smuzhiyun
999*4882a593Smuzhiyun con->rx_action = receive_from_sock;
1000*4882a593Smuzhiyun con->connect_action = sctp_connect_to_sock;
1001*4882a593Smuzhiyun add_sock(sock, con);
1002*4882a593Smuzhiyun
1003*4882a593Smuzhiyun /* Bind to all addresses. */
1004*4882a593Smuzhiyun if (sctp_bind_addrs(con, 0))
1005*4882a593Smuzhiyun goto bind_err;
1006*4882a593Smuzhiyun
1007*4882a593Smuzhiyun make_sockaddr(&daddr, dlm_config.ci_tcp_port, &addr_len);
1008*4882a593Smuzhiyun
1009*4882a593Smuzhiyun log_print("connecting to %d", con->nodeid);
1010*4882a593Smuzhiyun
1011*4882a593Smuzhiyun /* Turn off Nagle's algorithm */
1012*4882a593Smuzhiyun sctp_sock_set_nodelay(sock->sk);
1013*4882a593Smuzhiyun
1014*4882a593Smuzhiyun /*
1015*4882a593Smuzhiyun * Make sock->ops->connect() function return in specified time,
1016*4882a593Smuzhiyun * since O_NONBLOCK argument in connect() function does not work here,
1017*4882a593Smuzhiyun * then, we should restore the default value of this attribute.
1018*4882a593Smuzhiyun */
1019*4882a593Smuzhiyun sock_set_sndtimeo(sock->sk, 5);
1020*4882a593Smuzhiyun result = sock->ops->connect(sock, (struct sockaddr *)&daddr, addr_len,
1021*4882a593Smuzhiyun 0);
1022*4882a593Smuzhiyun sock_set_sndtimeo(sock->sk, 0);
1023*4882a593Smuzhiyun
1024*4882a593Smuzhiyun if (result == -EINPROGRESS)
1025*4882a593Smuzhiyun result = 0;
1026*4882a593Smuzhiyun if (result == 0)
1027*4882a593Smuzhiyun goto out;
1028*4882a593Smuzhiyun
1029*4882a593Smuzhiyun bind_err:
1030*4882a593Smuzhiyun con->sock = NULL;
1031*4882a593Smuzhiyun sock_release(sock);
1032*4882a593Smuzhiyun
1033*4882a593Smuzhiyun socket_err:
1034*4882a593Smuzhiyun /*
1035*4882a593Smuzhiyun * Some errors are fatal and this list might need adjusting. For other
1036*4882a593Smuzhiyun * errors we try again until the max number of retries is reached.
1037*4882a593Smuzhiyun */
1038*4882a593Smuzhiyun if (result != -EHOSTUNREACH &&
1039*4882a593Smuzhiyun result != -ENETUNREACH &&
1040*4882a593Smuzhiyun result != -ENETDOWN &&
1041*4882a593Smuzhiyun result != -EINVAL &&
1042*4882a593Smuzhiyun result != -EPROTONOSUPPORT) {
1043*4882a593Smuzhiyun log_print("connect %d try %d error %d", con->nodeid,
1044*4882a593Smuzhiyun con->retries, result);
1045*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1046*4882a593Smuzhiyun msleep(1000);
1047*4882a593Smuzhiyun lowcomms_connect_sock(con);
1048*4882a593Smuzhiyun return;
1049*4882a593Smuzhiyun }
1050*4882a593Smuzhiyun
1051*4882a593Smuzhiyun out:
1052*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1053*4882a593Smuzhiyun }
1054*4882a593Smuzhiyun
1055*4882a593Smuzhiyun /* Connect a new socket to its peer */
tcp_connect_to_sock(struct connection * con)1056*4882a593Smuzhiyun static void tcp_connect_to_sock(struct connection *con)
1057*4882a593Smuzhiyun {
1058*4882a593Smuzhiyun struct sockaddr_storage saddr, src_addr;
1059*4882a593Smuzhiyun int addr_len;
1060*4882a593Smuzhiyun struct socket *sock = NULL;
1061*4882a593Smuzhiyun unsigned int mark;
1062*4882a593Smuzhiyun int result;
1063*4882a593Smuzhiyun
1064*4882a593Smuzhiyun if (con->nodeid == 0) {
1065*4882a593Smuzhiyun log_print("attempt to connect sock 0 foiled");
1066*4882a593Smuzhiyun return;
1067*4882a593Smuzhiyun }
1068*4882a593Smuzhiyun
1069*4882a593Smuzhiyun dlm_comm_mark(con->nodeid, &mark);
1070*4882a593Smuzhiyun
1071*4882a593Smuzhiyun mutex_lock(&con->sock_mutex);
1072*4882a593Smuzhiyun if (con->retries++ > MAX_CONNECT_RETRIES)
1073*4882a593Smuzhiyun goto out;
1074*4882a593Smuzhiyun
1075*4882a593Smuzhiyun /* Some odd races can cause double-connects, ignore them */
1076*4882a593Smuzhiyun if (con->sock)
1077*4882a593Smuzhiyun goto out;
1078*4882a593Smuzhiyun
1079*4882a593Smuzhiyun /* Create a socket to communicate with */
1080*4882a593Smuzhiyun result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
1081*4882a593Smuzhiyun SOCK_STREAM, IPPROTO_TCP, &sock);
1082*4882a593Smuzhiyun if (result < 0)
1083*4882a593Smuzhiyun goto out_err;
1084*4882a593Smuzhiyun
1085*4882a593Smuzhiyun sock_set_mark(sock->sk, mark);
1086*4882a593Smuzhiyun
1087*4882a593Smuzhiyun memset(&saddr, 0, sizeof(saddr));
1088*4882a593Smuzhiyun result = nodeid_to_addr(con->nodeid, &saddr, NULL, false);
1089*4882a593Smuzhiyun if (result < 0) {
1090*4882a593Smuzhiyun log_print("no address for nodeid %d", con->nodeid);
1091*4882a593Smuzhiyun goto out_err;
1092*4882a593Smuzhiyun }
1093*4882a593Smuzhiyun
1094*4882a593Smuzhiyun con->rx_action = receive_from_sock;
1095*4882a593Smuzhiyun con->connect_action = tcp_connect_to_sock;
1096*4882a593Smuzhiyun con->shutdown_action = dlm_tcp_shutdown;
1097*4882a593Smuzhiyun add_sock(sock, con);
1098*4882a593Smuzhiyun
1099*4882a593Smuzhiyun /* Bind to our cluster-known address connecting to avoid
1100*4882a593Smuzhiyun routing problems */
1101*4882a593Smuzhiyun memcpy(&src_addr, dlm_local_addr[0], sizeof(src_addr));
1102*4882a593Smuzhiyun make_sockaddr(&src_addr, 0, &addr_len);
1103*4882a593Smuzhiyun result = sock->ops->bind(sock, (struct sockaddr *) &src_addr,
1104*4882a593Smuzhiyun addr_len);
1105*4882a593Smuzhiyun if (result < 0) {
1106*4882a593Smuzhiyun log_print("could not bind for connect: %d", result);
1107*4882a593Smuzhiyun /* This *may* not indicate a critical error */
1108*4882a593Smuzhiyun }
1109*4882a593Smuzhiyun
1110*4882a593Smuzhiyun make_sockaddr(&saddr, dlm_config.ci_tcp_port, &addr_len);
1111*4882a593Smuzhiyun
1112*4882a593Smuzhiyun log_print("connecting to %d", con->nodeid);
1113*4882a593Smuzhiyun
1114*4882a593Smuzhiyun /* Turn off Nagle's algorithm */
1115*4882a593Smuzhiyun tcp_sock_set_nodelay(sock->sk);
1116*4882a593Smuzhiyun
1117*4882a593Smuzhiyun result = sock->ops->connect(sock, (struct sockaddr *)&saddr, addr_len,
1118*4882a593Smuzhiyun O_NONBLOCK);
1119*4882a593Smuzhiyun if (result == -EINPROGRESS)
1120*4882a593Smuzhiyun result = 0;
1121*4882a593Smuzhiyun if (result == 0)
1122*4882a593Smuzhiyun goto out;
1123*4882a593Smuzhiyun
1124*4882a593Smuzhiyun out_err:
1125*4882a593Smuzhiyun if (con->sock) {
1126*4882a593Smuzhiyun sock_release(con->sock);
1127*4882a593Smuzhiyun con->sock = NULL;
1128*4882a593Smuzhiyun } else if (sock) {
1129*4882a593Smuzhiyun sock_release(sock);
1130*4882a593Smuzhiyun }
1131*4882a593Smuzhiyun /*
1132*4882a593Smuzhiyun * Some errors are fatal and this list might need adjusting. For other
1133*4882a593Smuzhiyun * errors we try again until the max number of retries is reached.
1134*4882a593Smuzhiyun */
1135*4882a593Smuzhiyun if (result != -EHOSTUNREACH &&
1136*4882a593Smuzhiyun result != -ENETUNREACH &&
1137*4882a593Smuzhiyun result != -ENETDOWN &&
1138*4882a593Smuzhiyun result != -EINVAL &&
1139*4882a593Smuzhiyun result != -EPROTONOSUPPORT) {
1140*4882a593Smuzhiyun log_print("connect %d try %d error %d", con->nodeid,
1141*4882a593Smuzhiyun con->retries, result);
1142*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1143*4882a593Smuzhiyun msleep(1000);
1144*4882a593Smuzhiyun lowcomms_connect_sock(con);
1145*4882a593Smuzhiyun return;
1146*4882a593Smuzhiyun }
1147*4882a593Smuzhiyun out:
1148*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1149*4882a593Smuzhiyun return;
1150*4882a593Smuzhiyun }
1151*4882a593Smuzhiyun
tcp_create_listen_sock(struct connection * con,struct sockaddr_storage * saddr)1152*4882a593Smuzhiyun static struct socket *tcp_create_listen_sock(struct connection *con,
1153*4882a593Smuzhiyun struct sockaddr_storage *saddr)
1154*4882a593Smuzhiyun {
1155*4882a593Smuzhiyun struct socket *sock = NULL;
1156*4882a593Smuzhiyun int result = 0;
1157*4882a593Smuzhiyun int addr_len;
1158*4882a593Smuzhiyun
1159*4882a593Smuzhiyun if (dlm_local_addr[0]->ss_family == AF_INET)
1160*4882a593Smuzhiyun addr_len = sizeof(struct sockaddr_in);
1161*4882a593Smuzhiyun else
1162*4882a593Smuzhiyun addr_len = sizeof(struct sockaddr_in6);
1163*4882a593Smuzhiyun
1164*4882a593Smuzhiyun /* Create a socket to communicate with */
1165*4882a593Smuzhiyun result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
1166*4882a593Smuzhiyun SOCK_STREAM, IPPROTO_TCP, &sock);
1167*4882a593Smuzhiyun if (result < 0) {
1168*4882a593Smuzhiyun log_print("Can't create listening comms socket");
1169*4882a593Smuzhiyun goto create_out;
1170*4882a593Smuzhiyun }
1171*4882a593Smuzhiyun
1172*4882a593Smuzhiyun sock_set_mark(sock->sk, dlm_config.ci_mark);
1173*4882a593Smuzhiyun
1174*4882a593Smuzhiyun /* Turn off Nagle's algorithm */
1175*4882a593Smuzhiyun tcp_sock_set_nodelay(sock->sk);
1176*4882a593Smuzhiyun
1177*4882a593Smuzhiyun sock_set_reuseaddr(sock->sk);
1178*4882a593Smuzhiyun
1179*4882a593Smuzhiyun write_lock_bh(&sock->sk->sk_callback_lock);
1180*4882a593Smuzhiyun sock->sk->sk_user_data = con;
1181*4882a593Smuzhiyun save_listen_callbacks(sock);
1182*4882a593Smuzhiyun con->rx_action = accept_from_sock;
1183*4882a593Smuzhiyun con->connect_action = tcp_connect_to_sock;
1184*4882a593Smuzhiyun write_unlock_bh(&sock->sk->sk_callback_lock);
1185*4882a593Smuzhiyun
1186*4882a593Smuzhiyun /* Bind to our port */
1187*4882a593Smuzhiyun make_sockaddr(saddr, dlm_config.ci_tcp_port, &addr_len);
1188*4882a593Smuzhiyun result = sock->ops->bind(sock, (struct sockaddr *) saddr, addr_len);
1189*4882a593Smuzhiyun if (result < 0) {
1190*4882a593Smuzhiyun log_print("Can't bind to port %d", dlm_config.ci_tcp_port);
1191*4882a593Smuzhiyun sock_release(sock);
1192*4882a593Smuzhiyun sock = NULL;
1193*4882a593Smuzhiyun con->sock = NULL;
1194*4882a593Smuzhiyun goto create_out;
1195*4882a593Smuzhiyun }
1196*4882a593Smuzhiyun sock_set_keepalive(sock->sk);
1197*4882a593Smuzhiyun
1198*4882a593Smuzhiyun result = sock->ops->listen(sock, 5);
1199*4882a593Smuzhiyun if (result < 0) {
1200*4882a593Smuzhiyun log_print("Can't listen on port %d", dlm_config.ci_tcp_port);
1201*4882a593Smuzhiyun sock_release(sock);
1202*4882a593Smuzhiyun sock = NULL;
1203*4882a593Smuzhiyun goto create_out;
1204*4882a593Smuzhiyun }
1205*4882a593Smuzhiyun
1206*4882a593Smuzhiyun create_out:
1207*4882a593Smuzhiyun return sock;
1208*4882a593Smuzhiyun }
1209*4882a593Smuzhiyun
1210*4882a593Smuzhiyun /* Get local addresses */
init_local(void)1211*4882a593Smuzhiyun static void init_local(void)
1212*4882a593Smuzhiyun {
1213*4882a593Smuzhiyun struct sockaddr_storage sas, *addr;
1214*4882a593Smuzhiyun int i;
1215*4882a593Smuzhiyun
1216*4882a593Smuzhiyun dlm_local_count = 0;
1217*4882a593Smuzhiyun for (i = 0; i < DLM_MAX_ADDR_COUNT; i++) {
1218*4882a593Smuzhiyun if (dlm_our_addr(&sas, i))
1219*4882a593Smuzhiyun break;
1220*4882a593Smuzhiyun
1221*4882a593Smuzhiyun addr = kmemdup(&sas, sizeof(*addr), GFP_NOFS);
1222*4882a593Smuzhiyun if (!addr)
1223*4882a593Smuzhiyun break;
1224*4882a593Smuzhiyun dlm_local_addr[dlm_local_count++] = addr;
1225*4882a593Smuzhiyun }
1226*4882a593Smuzhiyun }
1227*4882a593Smuzhiyun
deinit_local(void)1228*4882a593Smuzhiyun static void deinit_local(void)
1229*4882a593Smuzhiyun {
1230*4882a593Smuzhiyun int i;
1231*4882a593Smuzhiyun
1232*4882a593Smuzhiyun for (i = 0; i < dlm_local_count; i++)
1233*4882a593Smuzhiyun kfree(dlm_local_addr[i]);
1234*4882a593Smuzhiyun }
1235*4882a593Smuzhiyun
1236*4882a593Smuzhiyun /* Initialise SCTP socket and bind to all interfaces */
sctp_listen_for_all(void)1237*4882a593Smuzhiyun static int sctp_listen_for_all(void)
1238*4882a593Smuzhiyun {
1239*4882a593Smuzhiyun struct socket *sock = NULL;
1240*4882a593Smuzhiyun int result = -EINVAL;
1241*4882a593Smuzhiyun struct connection *con = nodeid2con(0, GFP_NOFS);
1242*4882a593Smuzhiyun
1243*4882a593Smuzhiyun if (!con)
1244*4882a593Smuzhiyun return -ENOMEM;
1245*4882a593Smuzhiyun
1246*4882a593Smuzhiyun log_print("Using SCTP for communications");
1247*4882a593Smuzhiyun
1248*4882a593Smuzhiyun result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
1249*4882a593Smuzhiyun SOCK_STREAM, IPPROTO_SCTP, &sock);
1250*4882a593Smuzhiyun if (result < 0) {
1251*4882a593Smuzhiyun log_print("Can't create comms socket, check SCTP is loaded");
1252*4882a593Smuzhiyun goto out;
1253*4882a593Smuzhiyun }
1254*4882a593Smuzhiyun
1255*4882a593Smuzhiyun sock_set_rcvbuf(sock->sk, NEEDED_RMEM);
1256*4882a593Smuzhiyun sock_set_mark(sock->sk, dlm_config.ci_mark);
1257*4882a593Smuzhiyun sctp_sock_set_nodelay(sock->sk);
1258*4882a593Smuzhiyun
1259*4882a593Smuzhiyun write_lock_bh(&sock->sk->sk_callback_lock);
1260*4882a593Smuzhiyun /* Init con struct */
1261*4882a593Smuzhiyun sock->sk->sk_user_data = con;
1262*4882a593Smuzhiyun save_listen_callbacks(sock);
1263*4882a593Smuzhiyun con->sock = sock;
1264*4882a593Smuzhiyun con->sock->sk->sk_data_ready = lowcomms_data_ready;
1265*4882a593Smuzhiyun con->rx_action = accept_from_sock;
1266*4882a593Smuzhiyun con->connect_action = sctp_connect_to_sock;
1267*4882a593Smuzhiyun
1268*4882a593Smuzhiyun write_unlock_bh(&sock->sk->sk_callback_lock);
1269*4882a593Smuzhiyun
1270*4882a593Smuzhiyun /* Bind to all addresses. */
1271*4882a593Smuzhiyun if (sctp_bind_addrs(con, dlm_config.ci_tcp_port))
1272*4882a593Smuzhiyun goto create_delsock;
1273*4882a593Smuzhiyun
1274*4882a593Smuzhiyun result = sock->ops->listen(sock, 5);
1275*4882a593Smuzhiyun if (result < 0) {
1276*4882a593Smuzhiyun log_print("Can't set socket listening");
1277*4882a593Smuzhiyun goto create_delsock;
1278*4882a593Smuzhiyun }
1279*4882a593Smuzhiyun
1280*4882a593Smuzhiyun return 0;
1281*4882a593Smuzhiyun
1282*4882a593Smuzhiyun create_delsock:
1283*4882a593Smuzhiyun sock_release(sock);
1284*4882a593Smuzhiyun con->sock = NULL;
1285*4882a593Smuzhiyun out:
1286*4882a593Smuzhiyun return result;
1287*4882a593Smuzhiyun }
1288*4882a593Smuzhiyun
tcp_listen_for_all(void)1289*4882a593Smuzhiyun static int tcp_listen_for_all(void)
1290*4882a593Smuzhiyun {
1291*4882a593Smuzhiyun struct socket *sock = NULL;
1292*4882a593Smuzhiyun struct connection *con = nodeid2con(0, GFP_NOFS);
1293*4882a593Smuzhiyun int result = -EINVAL;
1294*4882a593Smuzhiyun
1295*4882a593Smuzhiyun if (!con)
1296*4882a593Smuzhiyun return -ENOMEM;
1297*4882a593Smuzhiyun
1298*4882a593Smuzhiyun /* We don't support multi-homed hosts */
1299*4882a593Smuzhiyun if (dlm_local_addr[1] != NULL) {
1300*4882a593Smuzhiyun log_print("TCP protocol can't handle multi-homed hosts, "
1301*4882a593Smuzhiyun "try SCTP");
1302*4882a593Smuzhiyun return -EINVAL;
1303*4882a593Smuzhiyun }
1304*4882a593Smuzhiyun
1305*4882a593Smuzhiyun log_print("Using TCP for communications");
1306*4882a593Smuzhiyun
1307*4882a593Smuzhiyun sock = tcp_create_listen_sock(con, dlm_local_addr[0]);
1308*4882a593Smuzhiyun if (sock) {
1309*4882a593Smuzhiyun add_sock(sock, con);
1310*4882a593Smuzhiyun result = 0;
1311*4882a593Smuzhiyun }
1312*4882a593Smuzhiyun else {
1313*4882a593Smuzhiyun result = -EADDRINUSE;
1314*4882a593Smuzhiyun }
1315*4882a593Smuzhiyun
1316*4882a593Smuzhiyun return result;
1317*4882a593Smuzhiyun }
1318*4882a593Smuzhiyun
1319*4882a593Smuzhiyun
1320*4882a593Smuzhiyun
new_writequeue_entry(struct connection * con,gfp_t allocation)1321*4882a593Smuzhiyun static struct writequeue_entry *new_writequeue_entry(struct connection *con,
1322*4882a593Smuzhiyun gfp_t allocation)
1323*4882a593Smuzhiyun {
1324*4882a593Smuzhiyun struct writequeue_entry *entry;
1325*4882a593Smuzhiyun
1326*4882a593Smuzhiyun entry = kmalloc(sizeof(struct writequeue_entry), allocation);
1327*4882a593Smuzhiyun if (!entry)
1328*4882a593Smuzhiyun return NULL;
1329*4882a593Smuzhiyun
1330*4882a593Smuzhiyun entry->page = alloc_page(allocation);
1331*4882a593Smuzhiyun if (!entry->page) {
1332*4882a593Smuzhiyun kfree(entry);
1333*4882a593Smuzhiyun return NULL;
1334*4882a593Smuzhiyun }
1335*4882a593Smuzhiyun
1336*4882a593Smuzhiyun entry->offset = 0;
1337*4882a593Smuzhiyun entry->len = 0;
1338*4882a593Smuzhiyun entry->end = 0;
1339*4882a593Smuzhiyun entry->users = 0;
1340*4882a593Smuzhiyun entry->con = con;
1341*4882a593Smuzhiyun
1342*4882a593Smuzhiyun return entry;
1343*4882a593Smuzhiyun }
1344*4882a593Smuzhiyun
dlm_lowcomms_get_buffer(int nodeid,int len,gfp_t allocation,char ** ppc)1345*4882a593Smuzhiyun void *dlm_lowcomms_get_buffer(int nodeid, int len, gfp_t allocation, char **ppc)
1346*4882a593Smuzhiyun {
1347*4882a593Smuzhiyun struct connection *con;
1348*4882a593Smuzhiyun struct writequeue_entry *e;
1349*4882a593Smuzhiyun int offset = 0;
1350*4882a593Smuzhiyun
1351*4882a593Smuzhiyun con = nodeid2con(nodeid, allocation);
1352*4882a593Smuzhiyun if (!con)
1353*4882a593Smuzhiyun return NULL;
1354*4882a593Smuzhiyun
1355*4882a593Smuzhiyun spin_lock(&con->writequeue_lock);
1356*4882a593Smuzhiyun e = list_entry(con->writequeue.prev, struct writequeue_entry, list);
1357*4882a593Smuzhiyun if ((&e->list == &con->writequeue) ||
1358*4882a593Smuzhiyun (PAGE_SIZE - e->end < len)) {
1359*4882a593Smuzhiyun e = NULL;
1360*4882a593Smuzhiyun } else {
1361*4882a593Smuzhiyun offset = e->end;
1362*4882a593Smuzhiyun e->end += len;
1363*4882a593Smuzhiyun e->users++;
1364*4882a593Smuzhiyun }
1365*4882a593Smuzhiyun spin_unlock(&con->writequeue_lock);
1366*4882a593Smuzhiyun
1367*4882a593Smuzhiyun if (e) {
1368*4882a593Smuzhiyun got_one:
1369*4882a593Smuzhiyun *ppc = page_address(e->page) + offset;
1370*4882a593Smuzhiyun return e;
1371*4882a593Smuzhiyun }
1372*4882a593Smuzhiyun
1373*4882a593Smuzhiyun e = new_writequeue_entry(con, allocation);
1374*4882a593Smuzhiyun if (e) {
1375*4882a593Smuzhiyun spin_lock(&con->writequeue_lock);
1376*4882a593Smuzhiyun offset = e->end;
1377*4882a593Smuzhiyun e->end += len;
1378*4882a593Smuzhiyun e->users++;
1379*4882a593Smuzhiyun list_add_tail(&e->list, &con->writequeue);
1380*4882a593Smuzhiyun spin_unlock(&con->writequeue_lock);
1381*4882a593Smuzhiyun goto got_one;
1382*4882a593Smuzhiyun }
1383*4882a593Smuzhiyun return NULL;
1384*4882a593Smuzhiyun }
1385*4882a593Smuzhiyun
dlm_lowcomms_commit_buffer(void * mh)1386*4882a593Smuzhiyun void dlm_lowcomms_commit_buffer(void *mh)
1387*4882a593Smuzhiyun {
1388*4882a593Smuzhiyun struct writequeue_entry *e = (struct writequeue_entry *)mh;
1389*4882a593Smuzhiyun struct connection *con = e->con;
1390*4882a593Smuzhiyun int users;
1391*4882a593Smuzhiyun
1392*4882a593Smuzhiyun spin_lock(&con->writequeue_lock);
1393*4882a593Smuzhiyun users = --e->users;
1394*4882a593Smuzhiyun if (users)
1395*4882a593Smuzhiyun goto out;
1396*4882a593Smuzhiyun e->len = e->end - e->offset;
1397*4882a593Smuzhiyun spin_unlock(&con->writequeue_lock);
1398*4882a593Smuzhiyun
1399*4882a593Smuzhiyun queue_work(send_workqueue, &con->swork);
1400*4882a593Smuzhiyun return;
1401*4882a593Smuzhiyun
1402*4882a593Smuzhiyun out:
1403*4882a593Smuzhiyun spin_unlock(&con->writequeue_lock);
1404*4882a593Smuzhiyun return;
1405*4882a593Smuzhiyun }
1406*4882a593Smuzhiyun
1407*4882a593Smuzhiyun /* Send a message */
send_to_sock(struct connection * con)1408*4882a593Smuzhiyun static void send_to_sock(struct connection *con)
1409*4882a593Smuzhiyun {
1410*4882a593Smuzhiyun int ret = 0;
1411*4882a593Smuzhiyun const int msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL;
1412*4882a593Smuzhiyun struct writequeue_entry *e;
1413*4882a593Smuzhiyun int len, offset;
1414*4882a593Smuzhiyun int count = 0;
1415*4882a593Smuzhiyun
1416*4882a593Smuzhiyun mutex_lock(&con->sock_mutex);
1417*4882a593Smuzhiyun if (con->sock == NULL)
1418*4882a593Smuzhiyun goto out_connect;
1419*4882a593Smuzhiyun
1420*4882a593Smuzhiyun spin_lock(&con->writequeue_lock);
1421*4882a593Smuzhiyun for (;;) {
1422*4882a593Smuzhiyun e = list_entry(con->writequeue.next, struct writequeue_entry,
1423*4882a593Smuzhiyun list);
1424*4882a593Smuzhiyun if ((struct list_head *) e == &con->writequeue)
1425*4882a593Smuzhiyun break;
1426*4882a593Smuzhiyun
1427*4882a593Smuzhiyun len = e->len;
1428*4882a593Smuzhiyun offset = e->offset;
1429*4882a593Smuzhiyun BUG_ON(len == 0 && e->users == 0);
1430*4882a593Smuzhiyun spin_unlock(&con->writequeue_lock);
1431*4882a593Smuzhiyun
1432*4882a593Smuzhiyun ret = 0;
1433*4882a593Smuzhiyun if (len) {
1434*4882a593Smuzhiyun ret = kernel_sendpage(con->sock, e->page, offset, len,
1435*4882a593Smuzhiyun msg_flags);
1436*4882a593Smuzhiyun if (ret == -EAGAIN || ret == 0) {
1437*4882a593Smuzhiyun if (ret == -EAGAIN &&
1438*4882a593Smuzhiyun test_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags) &&
1439*4882a593Smuzhiyun !test_and_set_bit(CF_APP_LIMITED, &con->flags)) {
1440*4882a593Smuzhiyun /* Notify TCP that we're limited by the
1441*4882a593Smuzhiyun * application window size.
1442*4882a593Smuzhiyun */
1443*4882a593Smuzhiyun set_bit(SOCK_NOSPACE, &con->sock->flags);
1444*4882a593Smuzhiyun con->sock->sk->sk_write_pending++;
1445*4882a593Smuzhiyun }
1446*4882a593Smuzhiyun cond_resched();
1447*4882a593Smuzhiyun goto out;
1448*4882a593Smuzhiyun } else if (ret < 0)
1449*4882a593Smuzhiyun goto send_error;
1450*4882a593Smuzhiyun }
1451*4882a593Smuzhiyun
1452*4882a593Smuzhiyun /* Don't starve people filling buffers */
1453*4882a593Smuzhiyun if (++count >= MAX_SEND_MSG_COUNT) {
1454*4882a593Smuzhiyun cond_resched();
1455*4882a593Smuzhiyun count = 0;
1456*4882a593Smuzhiyun }
1457*4882a593Smuzhiyun
1458*4882a593Smuzhiyun spin_lock(&con->writequeue_lock);
1459*4882a593Smuzhiyun writequeue_entry_complete(e, ret);
1460*4882a593Smuzhiyun }
1461*4882a593Smuzhiyun spin_unlock(&con->writequeue_lock);
1462*4882a593Smuzhiyun out:
1463*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1464*4882a593Smuzhiyun return;
1465*4882a593Smuzhiyun
1466*4882a593Smuzhiyun send_error:
1467*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1468*4882a593Smuzhiyun close_connection(con, false, false, true);
1469*4882a593Smuzhiyun /* Requeue the send work. When the work daemon runs again, it will try
1470*4882a593Smuzhiyun a new connection, then call this function again. */
1471*4882a593Smuzhiyun queue_work(send_workqueue, &con->swork);
1472*4882a593Smuzhiyun return;
1473*4882a593Smuzhiyun
1474*4882a593Smuzhiyun out_connect:
1475*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1476*4882a593Smuzhiyun queue_work(send_workqueue, &con->swork);
1477*4882a593Smuzhiyun cond_resched();
1478*4882a593Smuzhiyun }
1479*4882a593Smuzhiyun
clean_one_writequeue(struct connection * con)1480*4882a593Smuzhiyun static void clean_one_writequeue(struct connection *con)
1481*4882a593Smuzhiyun {
1482*4882a593Smuzhiyun struct writequeue_entry *e, *safe;
1483*4882a593Smuzhiyun
1484*4882a593Smuzhiyun spin_lock(&con->writequeue_lock);
1485*4882a593Smuzhiyun list_for_each_entry_safe(e, safe, &con->writequeue, list) {
1486*4882a593Smuzhiyun list_del(&e->list);
1487*4882a593Smuzhiyun free_entry(e);
1488*4882a593Smuzhiyun }
1489*4882a593Smuzhiyun spin_unlock(&con->writequeue_lock);
1490*4882a593Smuzhiyun }
1491*4882a593Smuzhiyun
1492*4882a593Smuzhiyun /* Called from recovery when it knows that a node has
1493*4882a593Smuzhiyun left the cluster */
dlm_lowcomms_close(int nodeid)1494*4882a593Smuzhiyun int dlm_lowcomms_close(int nodeid)
1495*4882a593Smuzhiyun {
1496*4882a593Smuzhiyun struct connection *con;
1497*4882a593Smuzhiyun struct dlm_node_addr *na;
1498*4882a593Smuzhiyun
1499*4882a593Smuzhiyun log_print("closing connection to node %d", nodeid);
1500*4882a593Smuzhiyun con = nodeid2con(nodeid, 0);
1501*4882a593Smuzhiyun if (con) {
1502*4882a593Smuzhiyun set_bit(CF_CLOSE, &con->flags);
1503*4882a593Smuzhiyun close_connection(con, true, true, true);
1504*4882a593Smuzhiyun clean_one_writequeue(con);
1505*4882a593Smuzhiyun }
1506*4882a593Smuzhiyun
1507*4882a593Smuzhiyun spin_lock(&dlm_node_addrs_spin);
1508*4882a593Smuzhiyun na = find_node_addr(nodeid);
1509*4882a593Smuzhiyun if (na) {
1510*4882a593Smuzhiyun list_del(&na->list);
1511*4882a593Smuzhiyun while (na->addr_count--)
1512*4882a593Smuzhiyun kfree(na->addr[na->addr_count]);
1513*4882a593Smuzhiyun kfree(na);
1514*4882a593Smuzhiyun }
1515*4882a593Smuzhiyun spin_unlock(&dlm_node_addrs_spin);
1516*4882a593Smuzhiyun
1517*4882a593Smuzhiyun return 0;
1518*4882a593Smuzhiyun }
1519*4882a593Smuzhiyun
1520*4882a593Smuzhiyun /* Receive workqueue function */
process_recv_sockets(struct work_struct * work)1521*4882a593Smuzhiyun static void process_recv_sockets(struct work_struct *work)
1522*4882a593Smuzhiyun {
1523*4882a593Smuzhiyun struct connection *con = container_of(work, struct connection, rwork);
1524*4882a593Smuzhiyun int err;
1525*4882a593Smuzhiyun
1526*4882a593Smuzhiyun clear_bit(CF_READ_PENDING, &con->flags);
1527*4882a593Smuzhiyun do {
1528*4882a593Smuzhiyun err = con->rx_action(con);
1529*4882a593Smuzhiyun } while (!err);
1530*4882a593Smuzhiyun }
1531*4882a593Smuzhiyun
1532*4882a593Smuzhiyun /* Send workqueue function */
process_send_sockets(struct work_struct * work)1533*4882a593Smuzhiyun static void process_send_sockets(struct work_struct *work)
1534*4882a593Smuzhiyun {
1535*4882a593Smuzhiyun struct connection *con = container_of(work, struct connection, swork);
1536*4882a593Smuzhiyun
1537*4882a593Smuzhiyun clear_bit(CF_WRITE_PENDING, &con->flags);
1538*4882a593Smuzhiyun if (con->sock == NULL) /* not mutex protected so check it inside too */
1539*4882a593Smuzhiyun con->connect_action(con);
1540*4882a593Smuzhiyun if (!list_empty(&con->writequeue))
1541*4882a593Smuzhiyun send_to_sock(con);
1542*4882a593Smuzhiyun }
1543*4882a593Smuzhiyun
work_stop(void)1544*4882a593Smuzhiyun static void work_stop(void)
1545*4882a593Smuzhiyun {
1546*4882a593Smuzhiyun if (recv_workqueue)
1547*4882a593Smuzhiyun destroy_workqueue(recv_workqueue);
1548*4882a593Smuzhiyun if (send_workqueue)
1549*4882a593Smuzhiyun destroy_workqueue(send_workqueue);
1550*4882a593Smuzhiyun }
1551*4882a593Smuzhiyun
work_start(void)1552*4882a593Smuzhiyun static int work_start(void)
1553*4882a593Smuzhiyun {
1554*4882a593Smuzhiyun recv_workqueue = alloc_workqueue("dlm_recv",
1555*4882a593Smuzhiyun WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
1556*4882a593Smuzhiyun if (!recv_workqueue) {
1557*4882a593Smuzhiyun log_print("can't start dlm_recv");
1558*4882a593Smuzhiyun return -ENOMEM;
1559*4882a593Smuzhiyun }
1560*4882a593Smuzhiyun
1561*4882a593Smuzhiyun send_workqueue = alloc_workqueue("dlm_send",
1562*4882a593Smuzhiyun WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
1563*4882a593Smuzhiyun if (!send_workqueue) {
1564*4882a593Smuzhiyun log_print("can't start dlm_send");
1565*4882a593Smuzhiyun destroy_workqueue(recv_workqueue);
1566*4882a593Smuzhiyun return -ENOMEM;
1567*4882a593Smuzhiyun }
1568*4882a593Smuzhiyun
1569*4882a593Smuzhiyun return 0;
1570*4882a593Smuzhiyun }
1571*4882a593Smuzhiyun
_stop_conn(struct connection * con,bool and_other)1572*4882a593Smuzhiyun static void _stop_conn(struct connection *con, bool and_other)
1573*4882a593Smuzhiyun {
1574*4882a593Smuzhiyun mutex_lock(&con->sock_mutex);
1575*4882a593Smuzhiyun set_bit(CF_CLOSE, &con->flags);
1576*4882a593Smuzhiyun set_bit(CF_READ_PENDING, &con->flags);
1577*4882a593Smuzhiyun set_bit(CF_WRITE_PENDING, &con->flags);
1578*4882a593Smuzhiyun if (con->sock && con->sock->sk) {
1579*4882a593Smuzhiyun write_lock_bh(&con->sock->sk->sk_callback_lock);
1580*4882a593Smuzhiyun con->sock->sk->sk_user_data = NULL;
1581*4882a593Smuzhiyun write_unlock_bh(&con->sock->sk->sk_callback_lock);
1582*4882a593Smuzhiyun }
1583*4882a593Smuzhiyun if (con->othercon && and_other)
1584*4882a593Smuzhiyun _stop_conn(con->othercon, false);
1585*4882a593Smuzhiyun mutex_unlock(&con->sock_mutex);
1586*4882a593Smuzhiyun }
1587*4882a593Smuzhiyun
stop_conn(struct connection * con)1588*4882a593Smuzhiyun static void stop_conn(struct connection *con)
1589*4882a593Smuzhiyun {
1590*4882a593Smuzhiyun _stop_conn(con, true);
1591*4882a593Smuzhiyun }
1592*4882a593Smuzhiyun
shutdown_conn(struct connection * con)1593*4882a593Smuzhiyun static void shutdown_conn(struct connection *con)
1594*4882a593Smuzhiyun {
1595*4882a593Smuzhiyun if (con->shutdown_action)
1596*4882a593Smuzhiyun con->shutdown_action(con);
1597*4882a593Smuzhiyun }
1598*4882a593Smuzhiyun
connection_release(struct rcu_head * rcu)1599*4882a593Smuzhiyun static void connection_release(struct rcu_head *rcu)
1600*4882a593Smuzhiyun {
1601*4882a593Smuzhiyun struct connection *con = container_of(rcu, struct connection, rcu);
1602*4882a593Smuzhiyun
1603*4882a593Smuzhiyun kfree(con->rx_buf);
1604*4882a593Smuzhiyun kfree(con);
1605*4882a593Smuzhiyun }
1606*4882a593Smuzhiyun
free_conn(struct connection * con)1607*4882a593Smuzhiyun static void free_conn(struct connection *con)
1608*4882a593Smuzhiyun {
1609*4882a593Smuzhiyun close_connection(con, true, true, true);
1610*4882a593Smuzhiyun spin_lock(&connections_lock);
1611*4882a593Smuzhiyun hlist_del_rcu(&con->list);
1612*4882a593Smuzhiyun spin_unlock(&connections_lock);
1613*4882a593Smuzhiyun if (con->othercon) {
1614*4882a593Smuzhiyun clean_one_writequeue(con->othercon);
1615*4882a593Smuzhiyun call_rcu(&con->othercon->rcu, connection_release);
1616*4882a593Smuzhiyun }
1617*4882a593Smuzhiyun clean_one_writequeue(con);
1618*4882a593Smuzhiyun call_rcu(&con->rcu, connection_release);
1619*4882a593Smuzhiyun }
1620*4882a593Smuzhiyun
work_flush(void)1621*4882a593Smuzhiyun static void work_flush(void)
1622*4882a593Smuzhiyun {
1623*4882a593Smuzhiyun int ok, idx;
1624*4882a593Smuzhiyun int i;
1625*4882a593Smuzhiyun struct connection *con;
1626*4882a593Smuzhiyun
1627*4882a593Smuzhiyun do {
1628*4882a593Smuzhiyun ok = 1;
1629*4882a593Smuzhiyun foreach_conn(stop_conn);
1630*4882a593Smuzhiyun if (recv_workqueue)
1631*4882a593Smuzhiyun flush_workqueue(recv_workqueue);
1632*4882a593Smuzhiyun if (send_workqueue)
1633*4882a593Smuzhiyun flush_workqueue(send_workqueue);
1634*4882a593Smuzhiyun idx = srcu_read_lock(&connections_srcu);
1635*4882a593Smuzhiyun for (i = 0; i < CONN_HASH_SIZE && ok; i++) {
1636*4882a593Smuzhiyun hlist_for_each_entry_rcu(con, &connection_hash[i],
1637*4882a593Smuzhiyun list) {
1638*4882a593Smuzhiyun ok &= test_bit(CF_READ_PENDING, &con->flags);
1639*4882a593Smuzhiyun ok &= test_bit(CF_WRITE_PENDING, &con->flags);
1640*4882a593Smuzhiyun if (con->othercon) {
1641*4882a593Smuzhiyun ok &= test_bit(CF_READ_PENDING,
1642*4882a593Smuzhiyun &con->othercon->flags);
1643*4882a593Smuzhiyun ok &= test_bit(CF_WRITE_PENDING,
1644*4882a593Smuzhiyun &con->othercon->flags);
1645*4882a593Smuzhiyun }
1646*4882a593Smuzhiyun }
1647*4882a593Smuzhiyun }
1648*4882a593Smuzhiyun srcu_read_unlock(&connections_srcu, idx);
1649*4882a593Smuzhiyun } while (!ok);
1650*4882a593Smuzhiyun }
1651*4882a593Smuzhiyun
dlm_lowcomms_stop(void)1652*4882a593Smuzhiyun void dlm_lowcomms_stop(void)
1653*4882a593Smuzhiyun {
1654*4882a593Smuzhiyun /* Set all the flags to prevent any
1655*4882a593Smuzhiyun socket activity.
1656*4882a593Smuzhiyun */
1657*4882a593Smuzhiyun dlm_allow_conn = 0;
1658*4882a593Smuzhiyun
1659*4882a593Smuzhiyun if (recv_workqueue)
1660*4882a593Smuzhiyun flush_workqueue(recv_workqueue);
1661*4882a593Smuzhiyun if (send_workqueue)
1662*4882a593Smuzhiyun flush_workqueue(send_workqueue);
1663*4882a593Smuzhiyun
1664*4882a593Smuzhiyun foreach_conn(shutdown_conn);
1665*4882a593Smuzhiyun work_flush();
1666*4882a593Smuzhiyun foreach_conn(free_conn);
1667*4882a593Smuzhiyun work_stop();
1668*4882a593Smuzhiyun deinit_local();
1669*4882a593Smuzhiyun }
1670*4882a593Smuzhiyun
dlm_lowcomms_start(void)1671*4882a593Smuzhiyun int dlm_lowcomms_start(void)
1672*4882a593Smuzhiyun {
1673*4882a593Smuzhiyun int error = -EINVAL;
1674*4882a593Smuzhiyun struct connection *con;
1675*4882a593Smuzhiyun int i;
1676*4882a593Smuzhiyun
1677*4882a593Smuzhiyun for (i = 0; i < CONN_HASH_SIZE; i++)
1678*4882a593Smuzhiyun INIT_HLIST_HEAD(&connection_hash[i]);
1679*4882a593Smuzhiyun
1680*4882a593Smuzhiyun init_local();
1681*4882a593Smuzhiyun if (!dlm_local_count) {
1682*4882a593Smuzhiyun error = -ENOTCONN;
1683*4882a593Smuzhiyun log_print("no local IP address has been set");
1684*4882a593Smuzhiyun goto fail;
1685*4882a593Smuzhiyun }
1686*4882a593Smuzhiyun
1687*4882a593Smuzhiyun error = work_start();
1688*4882a593Smuzhiyun if (error)
1689*4882a593Smuzhiyun goto fail;
1690*4882a593Smuzhiyun
1691*4882a593Smuzhiyun dlm_allow_conn = 1;
1692*4882a593Smuzhiyun
1693*4882a593Smuzhiyun /* Start listening */
1694*4882a593Smuzhiyun if (dlm_config.ci_protocol == 0)
1695*4882a593Smuzhiyun error = tcp_listen_for_all();
1696*4882a593Smuzhiyun else
1697*4882a593Smuzhiyun error = sctp_listen_for_all();
1698*4882a593Smuzhiyun if (error)
1699*4882a593Smuzhiyun goto fail_unlisten;
1700*4882a593Smuzhiyun
1701*4882a593Smuzhiyun return 0;
1702*4882a593Smuzhiyun
1703*4882a593Smuzhiyun fail_unlisten:
1704*4882a593Smuzhiyun dlm_allow_conn = 0;
1705*4882a593Smuzhiyun con = nodeid2con(0,0);
1706*4882a593Smuzhiyun if (con)
1707*4882a593Smuzhiyun free_conn(con);
1708*4882a593Smuzhiyun fail:
1709*4882a593Smuzhiyun return error;
1710*4882a593Smuzhiyun }
1711*4882a593Smuzhiyun
dlm_lowcomms_exit(void)1712*4882a593Smuzhiyun void dlm_lowcomms_exit(void)
1713*4882a593Smuzhiyun {
1714*4882a593Smuzhiyun struct dlm_node_addr *na, *safe;
1715*4882a593Smuzhiyun
1716*4882a593Smuzhiyun spin_lock(&dlm_node_addrs_spin);
1717*4882a593Smuzhiyun list_for_each_entry_safe(na, safe, &dlm_node_addrs, list) {
1718*4882a593Smuzhiyun list_del(&na->list);
1719*4882a593Smuzhiyun while (na->addr_count--)
1720*4882a593Smuzhiyun kfree(na->addr[na->addr_count]);
1721*4882a593Smuzhiyun kfree(na);
1722*4882a593Smuzhiyun }
1723*4882a593Smuzhiyun spin_unlock(&dlm_node_addrs_spin);
1724*4882a593Smuzhiyun }
1725