blob: dc9ae6d670dc792b2ad31b848b008a6a79f46a0f [file] [log] [blame]
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001/******************************************************************************
2*******************************************************************************
3**
4** Copyright (C) Sistina Software, Inc. 1997-2003 All rights reserved.
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06005** Copyright (C) 2004-2009 Red Hat, Inc. All rights reserved.
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01006**
7** This copyrighted material is made available to anyone wishing to use,
8** modify, copy, or redistribute it subject to the terms and conditions
9** of the GNU General Public License v.2.
10**
11*******************************************************************************
12******************************************************************************/
13
14/*
15 * lowcomms.c
16 *
17 * This is the "low-level" comms layer.
18 *
19 * It is responsible for sending/receiving messages
20 * from other nodes in the cluster.
21 *
22 * Cluster nodes are referred to by their nodeids. nodeids are
23 * simply 32 bit numbers to the locking module - if they need to
Joe Perches2cf12c02009-01-22 13:26:47 -080024 * be expanded for the cluster infrastructure then that is its
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010025 * responsibility. It is this layer's
26 * responsibility to resolve these into IP address or
27 * whatever it needs for inter-node communication.
28 *
29 * The comms level is two kernel threads that deal mainly with
30 * the receiving of messages from other nodes and passing them
31 * up to the mid-level comms layer (which understands the
32 * message format) for execution by the locking core, and
33 * a send thread which does all the setting up of connections
34 * to remote nodes and the sending of data. Threads are not allowed
35 * to send their own data because it may cause them to wait in times
36 * of high load. Also, this way, the sending thread can collect together
37 * messages bound for one node and send them in one block.
38 *
Joe Perches2cf12c02009-01-22 13:26:47 -080039 * lowcomms will choose to use either TCP or SCTP as its transport layer
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010040 * depending on the configuration variable 'protocol'. This should be set
Joe Perches2cf12c02009-01-22 13:26:47 -080041 * to 0 (default) for TCP or 1 for SCTP. It should be configured using a
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010042 * cluster-wide mechanism as it must be the same on all nodes of the cluster
43 * for the DLM to function.
44 *
45 */
46
47#include <asm/ioctls.h>
48#include <net/sock.h>
49#include <net/tcp.h>
50#include <linux/pagemap.h>
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010051#include <linux/file.h>
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -050052#include <linux/mutex.h>
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010053#include <linux/sctp.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090054#include <linux/slab.h>
Benjamin Poirier2f2d76c2012-03-08 05:55:59 +000055#include <net/sctp/sctp.h>
Joe Perches44ad5322009-01-22 13:24:49 -080056#include <net/ipv6.h>
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010057
58#include "dlm_internal.h"
59#include "lowcomms.h"
60#include "midcomms.h"
61#include "config.h"
62
63#define NEEDED_RMEM (4*1024*1024)
Christine Caulfield5e9ccc32009-01-28 12:57:40 -060064#define CONN_HASH_SIZE 32
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010065
Bob Petersonf92c8dd2010-11-12 11:15:20 -060066/* Number of messages to send before rescheduling */
67#define MAX_SEND_MSG_COUNT 25
68
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +010069struct cbuf {
70 unsigned int base;
71 unsigned int len;
72 unsigned int mask;
73};
74
75static void cbuf_add(struct cbuf *cb, int n)
76{
77 cb->len += n;
78}
79
80static int cbuf_data(struct cbuf *cb)
81{
82 return ((cb->base + cb->len) & cb->mask);
83}
84
85static void cbuf_init(struct cbuf *cb, int size)
86{
87 cb->base = cb->len = 0;
88 cb->mask = size-1;
89}
90
91static void cbuf_eat(struct cbuf *cb, int n)
92{
93 cb->len -= n;
94 cb->base += n;
95 cb->base &= cb->mask;
96}
97
98static bool cbuf_empty(struct cbuf *cb)
99{
100 return cb->len == 0;
101}
102
103struct connection {
104 struct socket *sock; /* NULL if not connected */
105 uint32_t nodeid; /* So we know who we are in the list */
106 struct mutex sock_mutex;
107 unsigned long flags;
108#define CF_READ_PENDING 1
109#define CF_WRITE_PENDING 2
110#define CF_CONNECT_PENDING 3
111#define CF_INIT_PENDING 4
112#define CF_IS_OTHERCON 5
Lars Marowsky-Bree063c4c92009-08-11 16:18:23 -0500113#define CF_CLOSE 6
David Millerb36930d2010-11-10 21:56:39 -0800114#define CF_APP_LIMITED 7
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100115 struct list_head writequeue; /* List of outgoing writequeue_entries */
116 spinlock_t writequeue_lock;
117 int (*rx_action) (struct connection *); /* What to do when active */
118 void (*connect_action) (struct connection *); /* What to do to connect */
119 struct page *rx_page;
120 struct cbuf cb;
121 int retries;
122#define MAX_CONNECT_RETRIES 3
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600123 struct hlist_node list;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100124 struct connection *othercon;
125 struct work_struct rwork; /* Receive workqueue */
126 struct work_struct swork; /* Send workqueue */
Bob Petersonb3a5bbf2015-08-27 09:34:47 -0500127 void (*orig_error_report)(struct sock *sk);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100128};
129#define sock2con(x) ((struct connection *)(x)->sk_user_data)
130
131/* An entry waiting to be sent */
132struct writequeue_entry {
133 struct list_head list;
134 struct page *page;
135 int offset;
136 int len;
137 int end;
138 int users;
139 struct connection *con;
140};
141
David Teigland36b71a82012-07-26 12:44:30 -0500142struct dlm_node_addr {
143 struct list_head list;
144 int nodeid;
145 int addr_count;
Mike Christie98e1b602013-06-14 04:56:12 -0500146 int curr_addr_index;
David Teigland36b71a82012-07-26 12:44:30 -0500147 struct sockaddr_storage *addr[DLM_MAX_ADDR_COUNT];
148};
149
150static LIST_HEAD(dlm_node_addrs);
151static DEFINE_SPINLOCK(dlm_node_addrs_spin);
152
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100153static struct sockaddr_storage *dlm_local_addr[DLM_MAX_ADDR_COUNT];
154static int dlm_local_count;
David Teigland513ef592012-03-30 11:46:08 -0500155static int dlm_allow_conn;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100156
157/* Work queues */
158static struct workqueue_struct *recv_workqueue;
159static struct workqueue_struct *send_workqueue;
160
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600161static struct hlist_head connection_hash[CONN_HASH_SIZE];
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -0500162static DEFINE_MUTEX(connections_lock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100163static struct kmem_cache *con_cache;
164
165static void process_recv_sockets(struct work_struct *work);
166static void process_send_sockets(struct work_struct *work);
167
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600168
169/* This is deliberately very simple because most clusters have simple
170 sequential nodeids, so we should be able to go straight to a connection
171 struct in the array */
172static inline int nodeid_hash(int nodeid)
173{
174 return nodeid & (CONN_HASH_SIZE-1);
175}
176
177static struct connection *__find_con(int nodeid)
178{
179 int r;
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600180 struct connection *con;
181
182 r = nodeid_hash(nodeid);
183
Sasha Levinb67bfe02013-02-27 17:06:00 -0800184 hlist_for_each_entry(con, &connection_hash[r], list) {
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600185 if (con->nodeid == nodeid)
186 return con;
187 }
188 return NULL;
189}
190
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100191/*
192 * If 'allocation' is zero then we don't attempt to create a new
193 * connection structure for this node.
194 */
195static struct connection *__nodeid2con(int nodeid, gfp_t alloc)
196{
197 struct connection *con = NULL;
198 int r;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100199
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600200 con = __find_con(nodeid);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100201 if (con || !alloc)
202 return con;
203
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100204 con = kmem_cache_zalloc(con_cache, alloc);
205 if (!con)
206 return NULL;
207
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600208 r = nodeid_hash(nodeid);
209 hlist_add_head(&con->list, &connection_hash[r]);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100210
211 con->nodeid = nodeid;
212 mutex_init(&con->sock_mutex);
213 INIT_LIST_HEAD(&con->writequeue);
214 spin_lock_init(&con->writequeue_lock);
215 INIT_WORK(&con->swork, process_send_sockets);
216 INIT_WORK(&con->rwork, process_recv_sockets);
217
218 /* Setup action pointers for child sockets */
219 if (con->nodeid) {
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600220 struct connection *zerocon = __find_con(0);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100221
222 con->connect_action = zerocon->connect_action;
223 if (!con->rx_action)
224 con->rx_action = zerocon->rx_action;
225 }
226
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100227 return con;
228}
229
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600230/* Loop round all connections */
231static void foreach_conn(void (*conn_func)(struct connection *c))
232{
233 int i;
Sasha Levinb67bfe02013-02-27 17:06:00 -0800234 struct hlist_node *n;
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600235 struct connection *con;
236
237 for (i = 0; i < CONN_HASH_SIZE; i++) {
Sasha Levinb67bfe02013-02-27 17:06:00 -0800238 hlist_for_each_entry_safe(con, n, &connection_hash[i], list)
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600239 conn_func(con);
Christine Caulfield5e9ccc32009-01-28 12:57:40 -0600240 }
241}
242
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100243static struct connection *nodeid2con(int nodeid, gfp_t allocation)
244{
245 struct connection *con;
246
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -0500247 mutex_lock(&connections_lock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100248 con = __nodeid2con(nodeid, allocation);
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -0500249 mutex_unlock(&connections_lock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100250
251 return con;
252}
253
David Teigland36b71a82012-07-26 12:44:30 -0500254static struct dlm_node_addr *find_node_addr(int nodeid)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100255{
David Teigland36b71a82012-07-26 12:44:30 -0500256 struct dlm_node_addr *na;
257
258 list_for_each_entry(na, &dlm_node_addrs, list) {
259 if (na->nodeid == nodeid)
260 return na;
261 }
262 return NULL;
263}
264
265static int addr_compare(struct sockaddr_storage *x, struct sockaddr_storage *y)
266{
267 switch (x->ss_family) {
268 case AF_INET: {
269 struct sockaddr_in *sinx = (struct sockaddr_in *)x;
270 struct sockaddr_in *siny = (struct sockaddr_in *)y;
271 if (sinx->sin_addr.s_addr != siny->sin_addr.s_addr)
272 return 0;
273 if (sinx->sin_port != siny->sin_port)
274 return 0;
275 break;
276 }
277 case AF_INET6: {
278 struct sockaddr_in6 *sinx = (struct sockaddr_in6 *)x;
279 struct sockaddr_in6 *siny = (struct sockaddr_in6 *)y;
280 if (!ipv6_addr_equal(&sinx->sin6_addr, &siny->sin6_addr))
281 return 0;
282 if (sinx->sin6_port != siny->sin6_port)
283 return 0;
284 break;
285 }
286 default:
287 return 0;
288 }
289 return 1;
290}
291
292static int nodeid_to_addr(int nodeid, struct sockaddr_storage *sas_out,
Mike Christie98e1b602013-06-14 04:56:12 -0500293 struct sockaddr *sa_out, bool try_new_addr)
David Teigland36b71a82012-07-26 12:44:30 -0500294{
295 struct sockaddr_storage sas;
296 struct dlm_node_addr *na;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100297
298 if (!dlm_local_count)
299 return -1;
300
David Teigland36b71a82012-07-26 12:44:30 -0500301 spin_lock(&dlm_node_addrs_spin);
302 na = find_node_addr(nodeid);
Mike Christie98e1b602013-06-14 04:56:12 -0500303 if (na && na->addr_count) {
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300304 memcpy(&sas, na->addr[na->curr_addr_index],
305 sizeof(struct sockaddr_storage));
306
Mike Christie98e1b602013-06-14 04:56:12 -0500307 if (try_new_addr) {
308 na->curr_addr_index++;
309 if (na->curr_addr_index == na->addr_count)
310 na->curr_addr_index = 0;
311 }
Mike Christie98e1b602013-06-14 04:56:12 -0500312 }
David Teigland36b71a82012-07-26 12:44:30 -0500313 spin_unlock(&dlm_node_addrs_spin);
314
315 if (!na)
316 return -EEXIST;
317
318 if (!na->addr_count)
319 return -ENOENT;
320
321 if (sas_out)
322 memcpy(sas_out, &sas, sizeof(struct sockaddr_storage));
323
324 if (!sa_out)
325 return 0;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100326
327 if (dlm_local_addr[0]->ss_family == AF_INET) {
David Teigland36b71a82012-07-26 12:44:30 -0500328 struct sockaddr_in *in4 = (struct sockaddr_in *) &sas;
329 struct sockaddr_in *ret4 = (struct sockaddr_in *) sa_out;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100330 ret4->sin_addr.s_addr = in4->sin_addr.s_addr;
331 } else {
David Teigland36b71a82012-07-26 12:44:30 -0500332 struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) &sas;
333 struct sockaddr_in6 *ret6 = (struct sockaddr_in6 *) sa_out;
Alexey Dobriyan4e3fd7a2011-11-21 03:39:03 +0000334 ret6->sin6_addr = in6->sin6_addr;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100335 }
336
337 return 0;
338}
339
David Teigland36b71a82012-07-26 12:44:30 -0500340static int addr_to_nodeid(struct sockaddr_storage *addr, int *nodeid)
341{
342 struct dlm_node_addr *na;
343 int rv = -EEXIST;
Mike Christie98e1b602013-06-14 04:56:12 -0500344 int addr_i;
David Teigland36b71a82012-07-26 12:44:30 -0500345
346 spin_lock(&dlm_node_addrs_spin);
347 list_for_each_entry(na, &dlm_node_addrs, list) {
348 if (!na->addr_count)
349 continue;
350
Mike Christie98e1b602013-06-14 04:56:12 -0500351 for (addr_i = 0; addr_i < na->addr_count; addr_i++) {
352 if (addr_compare(na->addr[addr_i], addr)) {
353 *nodeid = na->nodeid;
354 rv = 0;
355 goto unlock;
356 }
357 }
David Teigland36b71a82012-07-26 12:44:30 -0500358 }
Mike Christie98e1b602013-06-14 04:56:12 -0500359unlock:
David Teigland36b71a82012-07-26 12:44:30 -0500360 spin_unlock(&dlm_node_addrs_spin);
361 return rv;
362}
363
364int dlm_lowcomms_addr(int nodeid, struct sockaddr_storage *addr, int len)
365{
366 struct sockaddr_storage *new_addr;
367 struct dlm_node_addr *new_node, *na;
368
369 new_node = kzalloc(sizeof(struct dlm_node_addr), GFP_NOFS);
370 if (!new_node)
371 return -ENOMEM;
372
373 new_addr = kzalloc(sizeof(struct sockaddr_storage), GFP_NOFS);
374 if (!new_addr) {
375 kfree(new_node);
376 return -ENOMEM;
377 }
378
379 memcpy(new_addr, addr, len);
380
381 spin_lock(&dlm_node_addrs_spin);
382 na = find_node_addr(nodeid);
383 if (!na) {
384 new_node->nodeid = nodeid;
385 new_node->addr[0] = new_addr;
386 new_node->addr_count = 1;
387 list_add(&new_node->list, &dlm_node_addrs);
388 spin_unlock(&dlm_node_addrs_spin);
389 return 0;
390 }
391
392 if (na->addr_count >= DLM_MAX_ADDR_COUNT) {
393 spin_unlock(&dlm_node_addrs_spin);
394 kfree(new_addr);
395 kfree(new_node);
396 return -ENOSPC;
397 }
398
399 na->addr[na->addr_count++] = new_addr;
400 spin_unlock(&dlm_node_addrs_spin);
401 kfree(new_node);
402 return 0;
403}
404
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100405/* Data available on socket or listen socket received a connect */
David S. Miller676d2362014-04-11 16:15:36 -0400406static void lowcomms_data_ready(struct sock *sk)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100407{
408 struct connection *con = sock2con(sk);
Patrick Caulfieldafb853f2007-06-01 10:07:26 -0500409 if (con && !test_and_set_bit(CF_READ_PENDING, &con->flags))
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100410 queue_work(recv_workqueue, &con->rwork);
411}
412
413static void lowcomms_write_space(struct sock *sk)
414{
415 struct connection *con = sock2con(sk);
416
David Millerb36930d2010-11-10 21:56:39 -0800417 if (!con)
418 return;
419
420 clear_bit(SOCK_NOSPACE, &con->sock->flags);
421
422 if (test_and_clear_bit(CF_APP_LIMITED, &con->flags)) {
423 con->sock->sk->sk_write_pending--;
Eric Dumazet9cd3e072015-11-29 20:03:10 -0800424 clear_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags);
David Millerb36930d2010-11-10 21:56:39 -0800425 }
426
427 if (!test_and_set_bit(CF_WRITE_PENDING, &con->flags))
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100428 queue_work(send_workqueue, &con->swork);
429}
430
431static inline void lowcomms_connect_sock(struct connection *con)
432{
Lars Marowsky-Bree063c4c92009-08-11 16:18:23 -0500433 if (test_bit(CF_CLOSE, &con->flags))
434 return;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100435 if (!test_and_set_bit(CF_CONNECT_PENDING, &con->flags))
436 queue_work(send_workqueue, &con->swork);
437}
438
439static void lowcomms_state_change(struct sock *sk)
440{
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300441 /* SCTP layer is not calling sk_data_ready when the connection
442 * is done, so we catch the signal through here. Also, it
443 * doesn't switch socket state when entering shutdown, so we
444 * skip the write in that case.
445 */
446 if (sk->sk_shutdown) {
447 if (sk->sk_shutdown == RCV_SHUTDOWN)
448 lowcomms_data_ready(sk);
449 } else if (sk->sk_state == TCP_ESTABLISHED) {
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100450 lowcomms_write_space(sk);
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300451 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100452}
453
Christine Caulfield391fbdc2009-05-07 10:54:16 -0500454int dlm_lowcomms_connect_node(int nodeid)
455{
456 struct connection *con;
457
458 if (nodeid == dlm_our_nodeid())
459 return 0;
460
461 con = nodeid2con(nodeid, GFP_NOFS);
462 if (!con)
463 return -ENOMEM;
464 lowcomms_connect_sock(con);
465 return 0;
466}
467
Bob Petersonb3a5bbf2015-08-27 09:34:47 -0500468static void lowcomms_error_report(struct sock *sk)
469{
470 struct connection *con = sock2con(sk);
471 struct sockaddr_storage saddr;
Bob Peterson1a318332016-01-18 12:29:15 -0500472 int buflen;
Bob Petersonb3a5bbf2015-08-27 09:34:47 -0500473
Bob Peterson1a318332016-01-18 12:29:15 -0500474 if (con->sock == NULL ||
475 kernel_getpeername(con->sock, (struct sockaddr *)&saddr, &buflen)) {
Bob Petersonb3a5bbf2015-08-27 09:34:47 -0500476 printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
477 "sending to node %d, port %d, "
478 "sk_err=%d/%d\n", dlm_our_nodeid(),
479 con->nodeid, dlm_config.ci_tcp_port,
480 sk->sk_err, sk->sk_err_soft);
481 return;
482 } else if (saddr.ss_family == AF_INET) {
483 struct sockaddr_in *sin4 = (struct sockaddr_in *)&saddr;
484
485 printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
486 "sending to node %d at %pI4, port %d, "
487 "sk_err=%d/%d\n", dlm_our_nodeid(),
488 con->nodeid, &sin4->sin_addr.s_addr,
489 dlm_config.ci_tcp_port, sk->sk_err,
490 sk->sk_err_soft);
491 } else {
492 struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&saddr;
493
494 printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
495 "sending to node %d at %u.%u.%u.%u, "
496 "port %d, sk_err=%d/%d\n", dlm_our_nodeid(),
497 con->nodeid, sin6->sin6_addr.s6_addr32[0],
498 sin6->sin6_addr.s6_addr32[1],
499 sin6->sin6_addr.s6_addr32[2],
500 sin6->sin6_addr.s6_addr32[3],
501 dlm_config.ci_tcp_port, sk->sk_err,
502 sk->sk_err_soft);
503 }
504 con->orig_error_report(sk);
505}
506
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100507/* Make a socket active */
Ying Xue4dd40f02012-08-10 14:58:42 +0800508static void add_sock(struct socket *sock, struct connection *con)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100509{
510 con->sock = sock;
511
512 /* Install a data_ready callback */
513 con->sock->sk->sk_data_ready = lowcomms_data_ready;
514 con->sock->sk->sk_write_space = lowcomms_write_space;
515 con->sock->sk->sk_state_change = lowcomms_state_change;
516 con->sock->sk->sk_user_data = con;
Steven Whitehoused6d7b702008-11-12 16:49:48 -0600517 con->sock->sk->sk_allocation = GFP_NOFS;
Bob Petersonb3a5bbf2015-08-27 09:34:47 -0500518 con->orig_error_report = con->sock->sk->sk_error_report;
519 con->sock->sk->sk_error_report = lowcomms_error_report;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100520}
521
522/* Add the port number to an IPv6 or 4 sockaddr and return the address
523 length */
524static void make_sockaddr(struct sockaddr_storage *saddr, uint16_t port,
525 int *addr_len)
526{
527 saddr->ss_family = dlm_local_addr[0]->ss_family;
528 if (saddr->ss_family == AF_INET) {
529 struct sockaddr_in *in4_addr = (struct sockaddr_in *)saddr;
530 in4_addr->sin_port = cpu_to_be16(port);
531 *addr_len = sizeof(struct sockaddr_in);
532 memset(&in4_addr->sin_zero, 0, sizeof(in4_addr->sin_zero));
533 } else {
534 struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)saddr;
535 in6_addr->sin6_port = cpu_to_be16(port);
536 *addr_len = sizeof(struct sockaddr_in6);
537 }
Patrick Caulfield01c8cab2007-07-17 16:53:15 +0100538 memset((char *)saddr + *addr_len, 0, sizeof(struct sockaddr_storage) - *addr_len);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100539}
540
541/* Close a remote connection and tidy up */
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -0300542static void close_connection(struct connection *con, bool and_other,
543 bool tx, bool rx)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100544{
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -0300545 clear_bit(CF_CONNECT_PENDING, &con->flags);
546 clear_bit(CF_WRITE_PENDING, &con->flags);
547 if (tx && cancel_work_sync(&con->swork))
548 log_print("canceled swork for node %d", con->nodeid);
549 if (rx && cancel_work_sync(&con->rwork))
550 log_print("canceled rwork for node %d", con->nodeid);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100551
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -0300552 mutex_lock(&con->sock_mutex);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100553 if (con->sock) {
554 sock_release(con->sock);
555 con->sock = NULL;
556 }
557 if (con->othercon && and_other) {
558 /* Will only re-enter once. */
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -0300559 close_connection(con->othercon, false, true, true);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100560 }
561 if (con->rx_page) {
562 __free_page(con->rx_page);
563 con->rx_page = NULL;
564 }
Patrick Caulfield9e5f2822007-08-02 14:58:14 +0100565
Patrick Caulfield61d96be2007-08-20 15:13:38 +0100566 con->retries = 0;
567 mutex_unlock(&con->sock_mutex);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100568}
569
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100570/* Data received from remote end */
571static int receive_from_sock(struct connection *con)
572{
573 int ret = 0;
574 struct msghdr msg = {};
575 struct kvec iov[2];
576 unsigned len;
577 int r;
578 int call_again_soon = 0;
579 int nvec;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100580
581 mutex_lock(&con->sock_mutex);
582
583 if (con->sock == NULL) {
584 ret = -EAGAIN;
585 goto out_close;
586 }
Marcelo Ricardo Leitneracee4e52015-08-11 19:22:24 -0300587 if (con->nodeid == 0) {
588 ret = -EINVAL;
589 goto out_close;
590 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100591
592 if (con->rx_page == NULL) {
593 /*
594 * This doesn't need to be atomic, but I think it should
595 * improve performance if it is.
596 */
597 con->rx_page = alloc_page(GFP_ATOMIC);
598 if (con->rx_page == NULL)
599 goto out_resched;
600 cbuf_init(&con->cb, PAGE_CACHE_SIZE);
601 }
602
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100603 /*
604 * iov[0] is the bit of the circular buffer between the current end
605 * point (cb.base + cb.len) and the end of the buffer.
606 */
607 iov[0].iov_len = con->cb.base - cbuf_data(&con->cb);
608 iov[0].iov_base = page_address(con->rx_page) + cbuf_data(&con->cb);
609 iov[1].iov_len = 0;
610 nvec = 1;
611
612 /*
613 * iov[1] is the bit of the circular buffer between the start of the
614 * buffer and the start of the currently used section (cb.base)
615 */
616 if (cbuf_data(&con->cb) >= con->cb.base) {
617 iov[0].iov_len = PAGE_CACHE_SIZE - cbuf_data(&con->cb);
618 iov[1].iov_len = con->cb.base;
619 iov[1].iov_base = page_address(con->rx_page);
620 nvec = 2;
621 }
622 len = iov[0].iov_len + iov[1].iov_len;
623
624 r = ret = kernel_recvmsg(con->sock, &msg, iov, nvec, len,
625 MSG_DONTWAIT | MSG_NOSIGNAL);
626 if (ret <= 0)
627 goto out_close;
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300628 else if (ret == len)
629 call_again_soon = 1;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100630
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100631 cbuf_add(&con->cb, ret);
632 ret = dlm_process_incoming_buffer(con->nodeid,
633 page_address(con->rx_page),
634 con->cb.base, con->cb.len,
635 PAGE_CACHE_SIZE);
636 if (ret == -EBADMSG) {
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300637 log_print("lowcomms: addr=%p, base=%u, len=%u, read=%d",
638 page_address(con->rx_page), con->cb.base,
639 con->cb.len, r);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100640 }
641 if (ret < 0)
642 goto out_close;
643 cbuf_eat(&con->cb, ret);
644
645 if (cbuf_empty(&con->cb) && !call_again_soon) {
646 __free_page(con->rx_page);
647 con->rx_page = NULL;
648 }
649
650 if (call_again_soon)
651 goto out_resched;
652 mutex_unlock(&con->sock_mutex);
653 return 0;
654
655out_resched:
656 if (!test_and_set_bit(CF_READ_PENDING, &con->flags))
657 queue_work(recv_workqueue, &con->rwork);
658 mutex_unlock(&con->sock_mutex);
659 return -EAGAIN;
660
661out_close:
662 mutex_unlock(&con->sock_mutex);
Patrick Caulfield9e5f2822007-08-02 14:58:14 +0100663 if (ret != -EAGAIN) {
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -0300664 close_connection(con, false, true, false);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100665 /* Reconnect when there is something to send */
666 }
667 /* Don't return success if we really got EOF */
668 if (ret == 0)
669 ret = -EAGAIN;
670
671 return ret;
672}
673
674/* Listening socket is busy, accept a connection */
675static int tcp_accept_from_sock(struct connection *con)
676{
677 int result;
678 struct sockaddr_storage peeraddr;
679 struct socket *newsock;
680 int len;
681 int nodeid;
682 struct connection *newcon;
683 struct connection *addcon;
684
David Teigland513ef592012-03-30 11:46:08 -0500685 mutex_lock(&connections_lock);
686 if (!dlm_allow_conn) {
687 mutex_unlock(&connections_lock);
688 return -1;
689 }
690 mutex_unlock(&connections_lock);
691
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100692 memset(&peeraddr, 0, sizeof(peeraddr));
Eric W. Biedermaneeb1bd52015-05-08 21:08:05 -0500693 result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
694 SOCK_STREAM, IPPROTO_TCP, &newsock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100695 if (result < 0)
696 return -ENOMEM;
697
698 mutex_lock_nested(&con->sock_mutex, 0);
699
700 result = -ENOTCONN;
701 if (con->sock == NULL)
702 goto accept_err;
703
704 newsock->type = con->sock->type;
705 newsock->ops = con->sock->ops;
706
707 result = con->sock->ops->accept(con->sock, newsock, O_NONBLOCK);
708 if (result < 0)
709 goto accept_err;
710
711 /* Get the connected socket's peer */
712 memset(&peeraddr, 0, sizeof(peeraddr));
713 if (newsock->ops->getname(newsock, (struct sockaddr *)&peeraddr,
714 &len, 2)) {
715 result = -ECONNABORTED;
716 goto accept_err;
717 }
718
719 /* Get the new node's NODEID */
720 make_sockaddr(&peeraddr, 0, &len);
David Teigland36b71a82012-07-26 12:44:30 -0500721 if (addr_to_nodeid(&peeraddr, &nodeid)) {
Masatake YAMATObcaadf52011-07-04 12:25:51 +0900722 unsigned char *b=(unsigned char *)&peeraddr;
David Teigland617e82e2007-04-26 13:46:49 -0500723 log_print("connect from non cluster node");
Masatake YAMATObcaadf52011-07-04 12:25:51 +0900724 print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE,
725 b, sizeof(struct sockaddr_storage));
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100726 sock_release(newsock);
727 mutex_unlock(&con->sock_mutex);
728 return -1;
729 }
730
731 log_print("got connection from %d", nodeid);
732
733 /* Check to see if we already have a connection to this node. This
734 * could happen if the two nodes initiate a connection at roughly
735 * the same time and the connections cross on the wire.
736 * In this case we store the incoming one in "othercon"
737 */
David Teigland748285c2009-05-15 10:50:57 -0500738 newcon = nodeid2con(nodeid, GFP_NOFS);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100739 if (!newcon) {
740 result = -ENOMEM;
741 goto accept_err;
742 }
743 mutex_lock_nested(&newcon->sock_mutex, 1);
744 if (newcon->sock) {
745 struct connection *othercon = newcon->othercon;
746
747 if (!othercon) {
David Teigland748285c2009-05-15 10:50:57 -0500748 othercon = kmem_cache_zalloc(con_cache, GFP_NOFS);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100749 if (!othercon) {
David Teigland617e82e2007-04-26 13:46:49 -0500750 log_print("failed to allocate incoming socket");
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100751 mutex_unlock(&newcon->sock_mutex);
752 result = -ENOMEM;
753 goto accept_err;
754 }
755 othercon->nodeid = nodeid;
756 othercon->rx_action = receive_from_sock;
757 mutex_init(&othercon->sock_mutex);
758 INIT_WORK(&othercon->swork, process_send_sockets);
759 INIT_WORK(&othercon->rwork, process_recv_sockets);
760 set_bit(CF_IS_OTHERCON, &othercon->flags);
Patrick Caulfield61d96be2007-08-20 15:13:38 +0100761 }
762 if (!othercon->sock) {
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100763 newcon->othercon = othercon;
Patrick Caulfield97d84832007-06-27 11:36:23 +0100764 othercon->sock = newsock;
765 newsock->sk->sk_user_data = othercon;
766 add_sock(newsock, othercon);
767 addcon = othercon;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100768 }
Patrick Caulfield97d84832007-06-27 11:36:23 +0100769 else {
770 printk("Extra connection from node %d attempted\n", nodeid);
771 result = -EAGAIN;
akpm@linux-foundation.orgf4fadb23c2007-06-27 14:43:37 -0700772 mutex_unlock(&newcon->sock_mutex);
Patrick Caulfield97d84832007-06-27 11:36:23 +0100773 goto accept_err;
774 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100775 }
776 else {
777 newsock->sk->sk_user_data = newcon;
778 newcon->rx_action = receive_from_sock;
779 add_sock(newsock, newcon);
780 addcon = newcon;
781 }
782
783 mutex_unlock(&newcon->sock_mutex);
784
785 /*
786 * Add it to the active queue in case we got data
Lucas De Marchi25985ed2011-03-30 22:57:33 -0300787 * between processing the accept adding the socket
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100788 * to the read_sockets list
789 */
790 if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags))
791 queue_work(recv_workqueue, &addcon->rwork);
792 mutex_unlock(&con->sock_mutex);
793
794 return 0;
795
796accept_err:
797 mutex_unlock(&con->sock_mutex);
798 sock_release(newsock);
799
800 if (result != -EAGAIN)
David Teigland617e82e2007-04-26 13:46:49 -0500801 log_print("error accepting connection from node: %d", result);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100802 return result;
803}
804
kbuild test robot18df8a82015-08-18 04:50:36 +0800805static int sctp_accept_from_sock(struct connection *con)
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300806{
807 /* Check that the new node is in the lockspace */
808 struct sctp_prim prim;
809 int nodeid;
810 int prim_len, ret;
811 int addr_len;
812 struct connection *newcon;
813 struct connection *addcon;
814 struct socket *newsock;
815
816 mutex_lock(&connections_lock);
817 if (!dlm_allow_conn) {
818 mutex_unlock(&connections_lock);
819 return -1;
820 }
821 mutex_unlock(&connections_lock);
822
823 mutex_lock_nested(&con->sock_mutex, 0);
824
825 ret = kernel_accept(con->sock, &newsock, O_NONBLOCK);
826 if (ret < 0)
827 goto accept_err;
828
829 memset(&prim, 0, sizeof(struct sctp_prim));
830 prim_len = sizeof(struct sctp_prim);
831
832 ret = kernel_getsockopt(newsock, IPPROTO_SCTP, SCTP_PRIMARY_ADDR,
833 (char *)&prim, &prim_len);
834 if (ret < 0) {
835 log_print("getsockopt/sctp_primary_addr failed: %d", ret);
836 goto accept_err;
837 }
838
839 make_sockaddr(&prim.ssp_addr, 0, &addr_len);
840 if (addr_to_nodeid(&prim.ssp_addr, &nodeid)) {
841 unsigned char *b = (unsigned char *)&prim.ssp_addr;
842
843 log_print("reject connect from unknown addr");
844 print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE,
845 b, sizeof(struct sockaddr_storage));
846 goto accept_err;
847 }
848
849 newcon = nodeid2con(nodeid, GFP_NOFS);
850 if (!newcon) {
851 ret = -ENOMEM;
852 goto accept_err;
853 }
854
855 mutex_lock_nested(&newcon->sock_mutex, 1);
856
857 if (newcon->sock) {
858 struct connection *othercon = newcon->othercon;
859
860 if (!othercon) {
861 othercon = kmem_cache_zalloc(con_cache, GFP_NOFS);
862 if (!othercon) {
863 log_print("failed to allocate incoming socket");
864 mutex_unlock(&newcon->sock_mutex);
865 ret = -ENOMEM;
866 goto accept_err;
867 }
868 othercon->nodeid = nodeid;
869 othercon->rx_action = receive_from_sock;
870 mutex_init(&othercon->sock_mutex);
871 INIT_WORK(&othercon->swork, process_send_sockets);
872 INIT_WORK(&othercon->rwork, process_recv_sockets);
873 set_bit(CF_IS_OTHERCON, &othercon->flags);
874 }
875 if (!othercon->sock) {
876 newcon->othercon = othercon;
877 othercon->sock = newsock;
878 newsock->sk->sk_user_data = othercon;
879 add_sock(newsock, othercon);
880 addcon = othercon;
881 } else {
882 printk("Extra connection from node %d attempted\n", nodeid);
883 ret = -EAGAIN;
884 mutex_unlock(&newcon->sock_mutex);
885 goto accept_err;
886 }
887 } else {
888 newsock->sk->sk_user_data = newcon;
889 newcon->rx_action = receive_from_sock;
890 add_sock(newsock, newcon);
891 addcon = newcon;
892 }
893
894 log_print("connected to %d", nodeid);
895
896 mutex_unlock(&newcon->sock_mutex);
897
898 /*
899 * Add it to the active queue in case we got data
900 * between processing the accept adding the socket
901 * to the read_sockets list
902 */
903 if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags))
904 queue_work(recv_workqueue, &addcon->rwork);
905 mutex_unlock(&con->sock_mutex);
906
907 return 0;
908
909accept_err:
910 mutex_unlock(&con->sock_mutex);
911 if (newsock)
912 sock_release(newsock);
913 if (ret != -EAGAIN)
914 log_print("error accepting connection from node: %d", ret);
915
916 return ret;
917}
918
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100919static void free_entry(struct writequeue_entry *e)
920{
921 __free_page(e->page);
922 kfree(e);
923}
924
Mike Christie5d689872013-06-14 04:56:13 -0500925/*
926 * writequeue_entry_complete - try to delete and free write queue entry
927 * @e: write queue entry to try to delete
928 * @completed: bytes completed
929 *
930 * writequeue_lock must be held.
931 */
932static void writequeue_entry_complete(struct writequeue_entry *e, int completed)
933{
934 e->offset += completed;
935 e->len -= completed;
936
937 if (e->len == 0 && e->users == 0) {
938 list_del(&e->list);
939 free_entry(e);
940 }
941}
942
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300943/*
944 * sctp_bind_addrs - bind a SCTP socket to all our addresses
945 */
946static int sctp_bind_addrs(struct connection *con, uint16_t port)
947{
948 struct sockaddr_storage localaddr;
949 int i, addr_len, result = 0;
950
951 for (i = 0; i < dlm_local_count; i++) {
952 memcpy(&localaddr, dlm_local_addr[i], sizeof(localaddr));
953 make_sockaddr(&localaddr, port, &addr_len);
954
955 if (!i)
956 result = kernel_bind(con->sock,
957 (struct sockaddr *)&localaddr,
958 addr_len);
959 else
960 result = kernel_setsockopt(con->sock, SOL_SCTP,
961 SCTP_SOCKOPT_BINDX_ADD,
962 (char *)&localaddr, addr_len);
963
964 if (result < 0) {
965 log_print("Can't bind to %d addr number %d, %d.\n",
966 port, i + 1, result);
967 break;
968 }
969 }
970 return result;
971}
972
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100973/* Initiate an SCTP association.
974 This is a special case of send_to_sock() in that we don't yet have a
975 peeled-off socket for this association, so we use the listening socket
976 and add the primary IP address of the remote node.
977 */
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300978static void sctp_connect_to_sock(struct connection *con)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100979{
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300980 struct sockaddr_storage daddr;
981 int one = 1;
982 int result;
983 int addr_len;
984 struct socket *sock;
985
986 if (con->nodeid == 0) {
987 log_print("attempt to connect sock 0 foiled");
988 return;
989 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100990
Mike Christie5d689872013-06-14 04:56:13 -0500991 mutex_lock(&con->sock_mutex);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +0100992
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -0300993 /* Some odd races can cause double-connects, ignore them */
994 if (con->retries++ > MAX_CONNECT_RETRIES)
995 goto out;
996
997 if (con->sock) {
998 log_print("node %d already connected.", con->nodeid);
999 goto out;
1000 }
1001
1002 memset(&daddr, 0, sizeof(daddr));
1003 result = nodeid_to_addr(con->nodeid, &daddr, NULL, true);
1004 if (result < 0) {
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001005 log_print("no address for nodeid %d", con->nodeid);
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001006 goto out;
David Teigland04bedd72009-09-18 14:31:47 -05001007 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001008
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001009 /* Create a socket to communicate with */
1010 result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
1011 SOCK_STREAM, IPPROTO_SCTP, &sock);
1012 if (result < 0)
1013 goto socket_err;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001014
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001015 sock->sk->sk_user_data = con;
1016 con->rx_action = receive_from_sock;
1017 con->connect_action = sctp_connect_to_sock;
1018 add_sock(sock, con);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001019
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001020 /* Bind to all addresses. */
1021 if (sctp_bind_addrs(con, 0))
1022 goto bind_err;
Mike Christie98e1b602013-06-14 04:56:12 -05001023
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001024 make_sockaddr(&daddr, dlm_config.ci_tcp_port, &addr_len);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001025
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001026 log_print("connecting to %d", con->nodeid);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001027
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001028 /* Turn off Nagle's algorithm */
1029 kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one,
1030 sizeof(one));
1031
1032 result = sock->ops->connect(sock, (struct sockaddr *)&daddr, addr_len,
1033 O_NONBLOCK);
1034 if (result == -EINPROGRESS)
1035 result = 0;
1036 if (result == 0)
1037 goto out;
1038
1039
1040bind_err:
1041 con->sock = NULL;
1042 sock_release(sock);
1043
1044socket_err:
1045 /*
1046 * Some errors are fatal and this list might need adjusting. For other
1047 * errors we try again until the max number of retries is reached.
1048 */
1049 if (result != -EHOSTUNREACH &&
1050 result != -ENETUNREACH &&
1051 result != -ENETDOWN &&
1052 result != -EINVAL &&
1053 result != -EPROTONOSUPPORT) {
1054 log_print("connect %d try %d error %d", con->nodeid,
1055 con->retries, result);
1056 mutex_unlock(&con->sock_mutex);
1057 msleep(1000);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001058 clear_bit(CF_CONNECT_PENDING, &con->flags);
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001059 lowcomms_connect_sock(con);
1060 return;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001061 }
Mike Christie5d689872013-06-14 04:56:13 -05001062
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001063out:
Mike Christie5d689872013-06-14 04:56:13 -05001064 mutex_unlock(&con->sock_mutex);
Marcelo Ricardo Leitner00dcffa2015-08-11 19:22:25 -03001065 set_bit(CF_WRITE_PENDING, &con->flags);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001066}
1067
1068/* Connect a new socket to its peer */
1069static void tcp_connect_to_sock(struct connection *con)
1070{
Lon Hohberger6bd8fed2007-10-25 18:51:54 -04001071 struct sockaddr_storage saddr, src_addr;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001072 int addr_len;
Casey Dahlina89d63a2009-07-14 12:17:51 -05001073 struct socket *sock = NULL;
David Teiglandcb2d45d2010-11-12 11:12:55 -06001074 int one = 1;
David Teigland36b71a82012-07-26 12:44:30 -05001075 int result;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001076
1077 if (con->nodeid == 0) {
1078 log_print("attempt to connect sock 0 foiled");
1079 return;
1080 }
1081
1082 mutex_lock(&con->sock_mutex);
1083 if (con->retries++ > MAX_CONNECT_RETRIES)
1084 goto out;
1085
1086 /* Some odd races can cause double-connects, ignore them */
David Teigland36b71a82012-07-26 12:44:30 -05001087 if (con->sock)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001088 goto out;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001089
1090 /* Create a socket to communicate with */
Eric W. Biedermaneeb1bd52015-05-08 21:08:05 -05001091 result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
1092 SOCK_STREAM, IPPROTO_TCP, &sock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001093 if (result < 0)
1094 goto out_err;
1095
1096 memset(&saddr, 0, sizeof(saddr));
Mike Christie98e1b602013-06-14 04:56:12 -05001097 result = nodeid_to_addr(con->nodeid, &saddr, NULL, false);
David Teigland36b71a82012-07-26 12:44:30 -05001098 if (result < 0) {
1099 log_print("no address for nodeid %d", con->nodeid);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001100 goto out_err;
David Teigland36b71a82012-07-26 12:44:30 -05001101 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001102
1103 sock->sk->sk_user_data = con;
1104 con->rx_action = receive_from_sock;
1105 con->connect_action = tcp_connect_to_sock;
1106 add_sock(sock, con);
1107
Lon Hohberger6bd8fed2007-10-25 18:51:54 -04001108 /* Bind to our cluster-known address connecting to avoid
1109 routing problems */
1110 memcpy(&src_addr, dlm_local_addr[0], sizeof(src_addr));
1111 make_sockaddr(&src_addr, 0, &addr_len);
1112 result = sock->ops->bind(sock, (struct sockaddr *) &src_addr,
1113 addr_len);
1114 if (result < 0) {
1115 log_print("could not bind for connect: %d", result);
1116 /* This *may* not indicate a critical error */
1117 }
1118
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001119 make_sockaddr(&saddr, dlm_config.ci_tcp_port, &addr_len);
1120
1121 log_print("connecting to %d", con->nodeid);
David Teiglandcb2d45d2010-11-12 11:12:55 -06001122
1123 /* Turn off Nagle's algorithm */
1124 kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one,
1125 sizeof(one));
1126
David Teigland36b71a82012-07-26 12:44:30 -05001127 result = sock->ops->connect(sock, (struct sockaddr *)&saddr, addr_len,
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001128 O_NONBLOCK);
1129 if (result == -EINPROGRESS)
1130 result = 0;
1131 if (result == 0)
1132 goto out;
1133
1134out_err:
1135 if (con->sock) {
1136 sock_release(con->sock);
1137 con->sock = NULL;
Casey Dahlina89d63a2009-07-14 12:17:51 -05001138 } else if (sock) {
1139 sock_release(sock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001140 }
1141 /*
1142 * Some errors are fatal and this list might need adjusting. For other
1143 * errors we try again until the max number of retries is reached.
1144 */
David Teigland36b71a82012-07-26 12:44:30 -05001145 if (result != -EHOSTUNREACH &&
1146 result != -ENETUNREACH &&
1147 result != -ENETDOWN &&
1148 result != -EINVAL &&
1149 result != -EPROTONOSUPPORT) {
1150 log_print("connect %d try %d error %d", con->nodeid,
1151 con->retries, result);
1152 mutex_unlock(&con->sock_mutex);
1153 msleep(1000);
Marcelo Ricardo Leitner356344c2015-08-11 19:22:22 -03001154 clear_bit(CF_CONNECT_PENDING, &con->flags);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001155 lowcomms_connect_sock(con);
David Teigland36b71a82012-07-26 12:44:30 -05001156 return;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001157 }
1158out:
1159 mutex_unlock(&con->sock_mutex);
Marcelo Ricardo Leitner00dcffa2015-08-11 19:22:25 -03001160 set_bit(CF_WRITE_PENDING, &con->flags);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001161 return;
1162}
1163
1164static struct socket *tcp_create_listen_sock(struct connection *con,
1165 struct sockaddr_storage *saddr)
1166{
1167 struct socket *sock = NULL;
1168 int result = 0;
1169 int one = 1;
1170 int addr_len;
1171
1172 if (dlm_local_addr[0]->ss_family == AF_INET)
1173 addr_len = sizeof(struct sockaddr_in);
1174 else
1175 addr_len = sizeof(struct sockaddr_in6);
1176
1177 /* Create a socket to communicate with */
Eric W. Biedermaneeb1bd52015-05-08 21:08:05 -05001178 result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
1179 SOCK_STREAM, IPPROTO_TCP, &sock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001180 if (result < 0) {
David Teigland617e82e2007-04-26 13:46:49 -05001181 log_print("Can't create listening comms socket");
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001182 goto create_out;
1183 }
1184
David Teiglandcb2d45d2010-11-12 11:12:55 -06001185 /* Turn off Nagle's algorithm */
1186 kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one,
1187 sizeof(one));
1188
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001189 result = kernel_setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
1190 (char *)&one, sizeof(one));
1191
1192 if (result < 0) {
David Teigland617e82e2007-04-26 13:46:49 -05001193 log_print("Failed to set SO_REUSEADDR on socket: %d", result);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001194 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001195 con->rx_action = tcp_accept_from_sock;
1196 con->connect_action = tcp_connect_to_sock;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001197
1198 /* Bind to our port */
1199 make_sockaddr(saddr, dlm_config.ci_tcp_port, &addr_len);
1200 result = sock->ops->bind(sock, (struct sockaddr *) saddr, addr_len);
1201 if (result < 0) {
David Teigland617e82e2007-04-26 13:46:49 -05001202 log_print("Can't bind to port %d", dlm_config.ci_tcp_port);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001203 sock_release(sock);
1204 sock = NULL;
1205 con->sock = NULL;
1206 goto create_out;
1207 }
1208 result = kernel_setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE,
1209 (char *)&one, sizeof(one));
1210 if (result < 0) {
David Teigland617e82e2007-04-26 13:46:49 -05001211 log_print("Set keepalive failed: %d", result);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001212 }
1213
1214 result = sock->ops->listen(sock, 5);
1215 if (result < 0) {
David Teigland617e82e2007-04-26 13:46:49 -05001216 log_print("Can't listen on port %d", dlm_config.ci_tcp_port);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001217 sock_release(sock);
1218 sock = NULL;
1219 goto create_out;
1220 }
1221
1222create_out:
1223 return sock;
1224}
1225
1226/* Get local addresses */
1227static void init_local(void)
1228{
1229 struct sockaddr_storage sas, *addr;
1230 int i;
1231
Patrick Caulfield30d3a232007-04-23 16:26:21 +01001232 dlm_local_count = 0;
David Teigland1b189b82012-03-21 09:18:34 -05001233 for (i = 0; i < DLM_MAX_ADDR_COUNT; i++) {
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001234 if (dlm_our_addr(&sas, i))
1235 break;
1236
David Teigland573c24c2009-11-30 16:34:43 -06001237 addr = kmalloc(sizeof(*addr), GFP_NOFS);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001238 if (!addr)
1239 break;
1240 memcpy(addr, &sas, sizeof(*addr));
1241 dlm_local_addr[dlm_local_count++] = addr;
1242 }
1243}
1244
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001245/* Initialise SCTP socket and bind to all interfaces */
1246static int sctp_listen_for_all(void)
1247{
1248 struct socket *sock = NULL;
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001249 int result = -EINVAL;
David Teigland573c24c2009-11-30 16:34:43 -06001250 struct connection *con = nodeid2con(0, GFP_NOFS);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001251 int bufsize = NEEDED_RMEM;
Mike Christie86e92ad2013-06-14 04:56:14 -05001252 int one = 1;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001253
1254 if (!con)
1255 return -ENOMEM;
1256
1257 log_print("Using SCTP for communications");
1258
Eric W. Biedermaneeb1bd52015-05-08 21:08:05 -05001259 result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001260 SOCK_STREAM, IPPROTO_SCTP, &sock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001261 if (result < 0) {
1262 log_print("Can't create comms socket, check SCTP is loaded");
1263 goto out;
1264 }
1265
David S. Millerdf61c952007-11-06 23:48:57 -08001266 result = kernel_setsockopt(sock, SOL_SOCKET, SO_RCVBUFFORCE,
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001267 (char *)&bufsize, sizeof(bufsize));
1268 if (result)
David Teigland617e82e2007-04-26 13:46:49 -05001269 log_print("Error increasing buffer space on socket %d", result);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001270
Mike Christie86e92ad2013-06-14 04:56:14 -05001271 result = kernel_setsockopt(sock, SOL_SCTP, SCTP_NODELAY, (char *)&one,
1272 sizeof(one));
1273 if (result < 0)
1274 log_print("Could not set SCTP NODELAY error %d\n", result);
1275
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001276 /* Init con struct */
1277 sock->sk->sk_user_data = con;
1278 con->sock = sock;
1279 con->sock->sk->sk_data_ready = lowcomms_data_ready;
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001280 con->rx_action = sctp_accept_from_sock;
1281 con->connect_action = sctp_connect_to_sock;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001282
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001283 /* Bind to all addresses. */
1284 if (sctp_bind_addrs(con, dlm_config.ci_tcp_port))
1285 goto create_delsock;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001286
1287 result = sock->ops->listen(sock, 5);
1288 if (result < 0) {
1289 log_print("Can't set socket listening");
1290 goto create_delsock;
1291 }
1292
1293 return 0;
1294
1295create_delsock:
1296 sock_release(sock);
1297 con->sock = NULL;
1298out:
1299 return result;
1300}
1301
1302static int tcp_listen_for_all(void)
1303{
1304 struct socket *sock = NULL;
David Teigland573c24c2009-11-30 16:34:43 -06001305 struct connection *con = nodeid2con(0, GFP_NOFS);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001306 int result = -EINVAL;
1307
1308 if (!con)
1309 return -ENOMEM;
1310
1311 /* We don't support multi-homed hosts */
1312 if (dlm_local_addr[1] != NULL) {
David Teigland617e82e2007-04-26 13:46:49 -05001313 log_print("TCP protocol can't handle multi-homed hosts, "
1314 "try SCTP");
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001315 return -EINVAL;
1316 }
1317
1318 log_print("Using TCP for communications");
1319
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001320 sock = tcp_create_listen_sock(con, dlm_local_addr[0]);
1321 if (sock) {
1322 add_sock(sock, con);
1323 result = 0;
1324 }
1325 else {
1326 result = -EADDRINUSE;
1327 }
1328
1329 return result;
1330}
1331
1332
1333
1334static struct writequeue_entry *new_writequeue_entry(struct connection *con,
1335 gfp_t allocation)
1336{
1337 struct writequeue_entry *entry;
1338
1339 entry = kmalloc(sizeof(struct writequeue_entry), allocation);
1340 if (!entry)
1341 return NULL;
1342
1343 entry->page = alloc_page(allocation);
1344 if (!entry->page) {
1345 kfree(entry);
1346 return NULL;
1347 }
1348
1349 entry->offset = 0;
1350 entry->len = 0;
1351 entry->end = 0;
1352 entry->users = 0;
1353 entry->con = con;
1354
1355 return entry;
1356}
1357
David Teigland617e82e2007-04-26 13:46:49 -05001358void *dlm_lowcomms_get_buffer(int nodeid, int len, gfp_t allocation, char **ppc)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001359{
1360 struct connection *con;
1361 struct writequeue_entry *e;
1362 int offset = 0;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001363
1364 con = nodeid2con(nodeid, allocation);
1365 if (!con)
1366 return NULL;
1367
1368 spin_lock(&con->writequeue_lock);
1369 e = list_entry(con->writequeue.prev, struct writequeue_entry, list);
1370 if ((&e->list == &con->writequeue) ||
1371 (PAGE_CACHE_SIZE - e->end < len)) {
1372 e = NULL;
1373 } else {
1374 offset = e->end;
1375 e->end += len;
Wei Yongjuneeee2b52012-10-18 22:57:19 +08001376 e->users++;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001377 }
1378 spin_unlock(&con->writequeue_lock);
1379
1380 if (e) {
1381 got_one:
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001382 *ppc = page_address(e->page) + offset;
1383 return e;
1384 }
1385
1386 e = new_writequeue_entry(con, allocation);
1387 if (e) {
1388 spin_lock(&con->writequeue_lock);
1389 offset = e->end;
1390 e->end += len;
Wei Yongjuneeee2b52012-10-18 22:57:19 +08001391 e->users++;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001392 list_add_tail(&e->list, &con->writequeue);
1393 spin_unlock(&con->writequeue_lock);
1394 goto got_one;
1395 }
1396 return NULL;
1397}
1398
1399void dlm_lowcomms_commit_buffer(void *mh)
1400{
1401 struct writequeue_entry *e = (struct writequeue_entry *)mh;
1402 struct connection *con = e->con;
1403 int users;
1404
1405 spin_lock(&con->writequeue_lock);
1406 users = --e->users;
1407 if (users)
1408 goto out;
1409 e->len = e->end - e->offset;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001410 spin_unlock(&con->writequeue_lock);
1411
1412 if (!test_and_set_bit(CF_WRITE_PENDING, &con->flags)) {
1413 queue_work(send_workqueue, &con->swork);
1414 }
1415 return;
1416
1417out:
1418 spin_unlock(&con->writequeue_lock);
1419 return;
1420}
1421
1422/* Send a message */
1423static void send_to_sock(struct connection *con)
1424{
1425 int ret = 0;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001426 const int msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL;
1427 struct writequeue_entry *e;
1428 int len, offset;
Bob Petersonf92c8dd2010-11-12 11:15:20 -06001429 int count = 0;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001430
1431 mutex_lock(&con->sock_mutex);
1432 if (con->sock == NULL)
1433 goto out_connect;
1434
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001435 spin_lock(&con->writequeue_lock);
1436 for (;;) {
1437 e = list_entry(con->writequeue.next, struct writequeue_entry,
1438 list);
1439 if ((struct list_head *) e == &con->writequeue)
1440 break;
1441
1442 len = e->len;
1443 offset = e->offset;
1444 BUG_ON(len == 0 && e->users == 0);
1445 spin_unlock(&con->writequeue_lock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001446
1447 ret = 0;
1448 if (len) {
Paolo Bonzini1329e3f2009-08-24 13:18:04 -05001449 ret = kernel_sendpage(con->sock, e->page, offset, len,
1450 msg_flags);
Patrick Caulfieldd66f8272007-09-14 08:49:21 +01001451 if (ret == -EAGAIN || ret == 0) {
David Millerb36930d2010-11-10 21:56:39 -08001452 if (ret == -EAGAIN &&
Eric Dumazet9cd3e072015-11-29 20:03:10 -08001453 test_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags) &&
David Millerb36930d2010-11-10 21:56:39 -08001454 !test_and_set_bit(CF_APP_LIMITED, &con->flags)) {
1455 /* Notify TCP that we're limited by the
1456 * application window size.
1457 */
1458 set_bit(SOCK_NOSPACE, &con->sock->flags);
1459 con->sock->sk->sk_write_pending++;
1460 }
Patrick Caulfieldd66f8272007-09-14 08:49:21 +01001461 cond_resched();
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001462 goto out;
Ying Xue9c5bef52012-08-13 14:29:55 +08001463 } else if (ret < 0)
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001464 goto send_error;
Patrick Caulfieldd66f8272007-09-14 08:49:21 +01001465 }
Bob Petersonf92c8dd2010-11-12 11:15:20 -06001466
1467 /* Don't starve people filling buffers */
1468 if (++count >= MAX_SEND_MSG_COUNT) {
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001469 cond_resched();
Bob Petersonf92c8dd2010-11-12 11:15:20 -06001470 count = 0;
1471 }
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001472
1473 spin_lock(&con->writequeue_lock);
Mike Christie5d689872013-06-14 04:56:13 -05001474 writequeue_entry_complete(e, ret);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001475 }
1476 spin_unlock(&con->writequeue_lock);
1477out:
1478 mutex_unlock(&con->sock_mutex);
1479 return;
1480
1481send_error:
1482 mutex_unlock(&con->sock_mutex);
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -03001483 close_connection(con, false, false, true);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001484 lowcomms_connect_sock(con);
1485 return;
1486
1487out_connect:
1488 mutex_unlock(&con->sock_mutex);
Marcelo Ricardo Leitneree44b4b2015-08-11 19:22:23 -03001489 lowcomms_connect_sock(con);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001490}
1491
1492static void clean_one_writequeue(struct connection *con)
1493{
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001494 struct writequeue_entry *e, *safe;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001495
1496 spin_lock(&con->writequeue_lock);
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001497 list_for_each_entry_safe(e, safe, &con->writequeue, list) {
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001498 list_del(&e->list);
1499 free_entry(e);
1500 }
1501 spin_unlock(&con->writequeue_lock);
1502}
1503
1504/* Called from recovery when it knows that a node has
1505 left the cluster */
1506int dlm_lowcomms_close(int nodeid)
1507{
1508 struct connection *con;
David Teigland36b71a82012-07-26 12:44:30 -05001509 struct dlm_node_addr *na;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001510
1511 log_print("closing connection to node %d", nodeid);
1512 con = nodeid2con(nodeid, 0);
1513 if (con) {
Lars Marowsky-Bree063c4c92009-08-11 16:18:23 -05001514 set_bit(CF_CLOSE, &con->flags);
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -03001515 close_connection(con, true, true, true);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001516 clean_one_writequeue(con);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001517 }
David Teigland36b71a82012-07-26 12:44:30 -05001518
1519 spin_lock(&dlm_node_addrs_spin);
1520 na = find_node_addr(nodeid);
1521 if (na) {
1522 list_del(&na->list);
1523 while (na->addr_count--)
1524 kfree(na->addr[na->addr_count]);
1525 kfree(na);
1526 }
1527 spin_unlock(&dlm_node_addrs_spin);
1528
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001529 return 0;
1530}
1531
1532/* Receive workqueue function */
1533static void process_recv_sockets(struct work_struct *work)
1534{
1535 struct connection *con = container_of(work, struct connection, rwork);
1536 int err;
1537
1538 clear_bit(CF_READ_PENDING, &con->flags);
1539 do {
1540 err = con->rx_action(con);
1541 } while (!err);
1542}
1543
1544/* Send workqueue function */
1545static void process_send_sockets(struct work_struct *work)
1546{
1547 struct connection *con = container_of(work, struct connection, swork);
1548
Marcelo Ricardo Leitner00dcffa2015-08-11 19:22:25 -03001549 if (test_and_clear_bit(CF_CONNECT_PENDING, &con->flags))
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001550 con->connect_action(con);
Lars Marowsky-Bree063c4c92009-08-11 16:18:23 -05001551 if (test_and_clear_bit(CF_WRITE_PENDING, &con->flags))
1552 send_to_sock(con);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001553}
1554
1555
1556/* Discard all entries on the write queues */
1557static void clean_writequeues(void)
1558{
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001559 foreach_conn(clean_one_writequeue);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001560}
1561
1562static void work_stop(void)
1563{
1564 destroy_workqueue(recv_workqueue);
1565 destroy_workqueue(send_workqueue);
1566}
1567
1568static int work_start(void)
1569{
David Teiglande43f0552011-03-10 13:22:34 -06001570 recv_workqueue = alloc_workqueue("dlm_recv",
1571 WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
Namhyung Kimb9d41052010-12-13 13:42:24 -06001572 if (!recv_workqueue) {
1573 log_print("can't start dlm_recv");
1574 return -ENOMEM;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001575 }
1576
David Teiglande43f0552011-03-10 13:22:34 -06001577 send_workqueue = alloc_workqueue("dlm_send",
1578 WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
Namhyung Kimb9d41052010-12-13 13:42:24 -06001579 if (!send_workqueue) {
1580 log_print("can't start dlm_send");
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001581 destroy_workqueue(recv_workqueue);
Namhyung Kimb9d41052010-12-13 13:42:24 -06001582 return -ENOMEM;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001583 }
1584
1585 return 0;
1586}
1587
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001588static void stop_conn(struct connection *con)
1589{
1590 con->flags |= 0x0F;
Christine Caulfield391fbdc2009-05-07 10:54:16 -05001591 if (con->sock && con->sock->sk)
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001592 con->sock->sk->sk_user_data = NULL;
1593}
1594
1595static void free_conn(struct connection *con)
1596{
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -03001597 close_connection(con, true, true, true);
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001598 if (con->othercon)
1599 kmem_cache_free(con_cache, con->othercon);
1600 hlist_del(&con->list);
1601 kmem_cache_free(con_cache, con);
1602}
1603
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001604void dlm_lowcomms_stop(void)
1605{
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001606 /* Set all the flags to prevent any
1607 socket activity.
1608 */
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -05001609 mutex_lock(&connections_lock);
David Teigland513ef592012-03-30 11:46:08 -05001610 dlm_allow_conn = 0;
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001611 foreach_conn(stop_conn);
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -05001612 mutex_unlock(&connections_lock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001613
1614 work_stop();
1615
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -05001616 mutex_lock(&connections_lock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001617 clean_writequeues();
1618
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001619 foreach_conn(free_conn);
1620
Matthias Kaehlcke7a936ce2008-05-12 10:04:51 -05001621 mutex_unlock(&connections_lock);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001622 kmem_cache_destroy(con_cache);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001623}
1624
1625int dlm_lowcomms_start(void)
1626{
1627 int error = -EINVAL;
1628 struct connection *con;
Christine Caulfield5e9ccc32009-01-28 12:57:40 -06001629 int i;
1630
1631 for (i = 0; i < CONN_HASH_SIZE; i++)
1632 INIT_HLIST_HEAD(&connection_hash[i]);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001633
1634 init_local();
1635 if (!dlm_local_count) {
David Teigland617e82e2007-04-26 13:46:49 -05001636 error = -ENOTCONN;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001637 log_print("no local IP address has been set");
David Teigland513ef592012-03-30 11:46:08 -05001638 goto fail;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001639 }
1640
1641 error = -ENOMEM;
1642 con_cache = kmem_cache_create("dlm_conn", sizeof(struct connection),
1643 __alignof__(struct connection), 0,
Paul Mundt20c2df82007-07-20 10:11:58 +09001644 NULL);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001645 if (!con_cache)
David Teigland513ef592012-03-30 11:46:08 -05001646 goto fail;
1647
1648 error = work_start();
1649 if (error)
1650 goto fail_destroy;
1651
1652 dlm_allow_conn = 1;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001653
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001654 /* Start listening */
1655 if (dlm_config.ci_protocol == 0)
1656 error = tcp_listen_for_all();
1657 else
1658 error = sctp_listen_for_all();
1659 if (error)
1660 goto fail_unlisten;
1661
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001662 return 0;
1663
1664fail_unlisten:
David Teigland513ef592012-03-30 11:46:08 -05001665 dlm_allow_conn = 0;
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001666 con = nodeid2con(0,0);
1667 if (con) {
Marcelo Ricardo Leitner0d737a82015-08-11 19:22:21 -03001668 close_connection(con, false, true, true);
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001669 kmem_cache_free(con_cache, con);
1670 }
David Teigland513ef592012-03-30 11:46:08 -05001671fail_destroy:
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001672 kmem_cache_destroy(con_cache);
David Teigland513ef592012-03-30 11:46:08 -05001673fail:
Patrick Caulfield6ed7257b2007-04-17 15:39:57 +01001674 return error;
1675}
David Teigland36b71a82012-07-26 12:44:30 -05001676
1677void dlm_lowcomms_exit(void)
1678{
1679 struct dlm_node_addr *na, *safe;
1680
1681 spin_lock(&dlm_node_addrs_spin);
1682 list_for_each_entry_safe(na, safe, &dlm_node_addrs, list) {
1683 list_del(&na->list);
1684 while (na->addr_count--)
1685 kfree(na->addr[na->addr_count]);
1686 kfree(na);
1687 }
1688 spin_unlock(&dlm_node_addrs_spin);
1689}