lowcomms.c 41.6 KB
Newer Older
1
// SPDX-License-Identifier: GPL-2.0-only
2 3 4 5
/******************************************************************************
*******************************************************************************
**
**  Copyright (C) Sistina Software, Inc.  1997-2003  All rights reserved.
6
**  Copyright (C) 2004-2009 Red Hat, Inc.  All rights reserved.
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
**
**
*******************************************************************************
******************************************************************************/

/*
 * lowcomms.c
 *
 * This is the "low-level" comms layer.
 *
 * It is responsible for sending/receiving messages
 * from other nodes in the cluster.
 *
 * Cluster nodes are referred to by their nodeids. nodeids are
 * simply 32 bit numbers to the locking module - if they need to
J
Joe Perches 已提交
22
 * be expanded for the cluster infrastructure then that is its
23 24 25 26 27 28 29 30 31 32 33 34 35 36
 * responsibility. It is this layer's
 * responsibility to resolve these into IP address or
 * whatever it needs for inter-node communication.
 *
 * The comms level is two kernel threads that deal mainly with
 * the receiving of messages from other nodes and passing them
 * up to the mid-level comms layer (which understands the
 * message format) for execution by the locking core, and
 * a send thread which does all the setting up of connections
 * to remote nodes and the sending of data. Threads are not allowed
 * to send their own data because it may cause them to wait in times
 * of high load. Also, this way, the sending thread can collect together
 * messages bound for one node and send them in one block.
 *
J
Joe Perches 已提交
37
 * lowcomms will choose to use either TCP or SCTP as its transport layer
38
 * depending on the configuration variable 'protocol'. This should be set
J
Joe Perches 已提交
39
 * to 0 (default) for TCP or 1 for SCTP. It should be configured using a
40 41
 * cluster-wide mechanism as it must be the same on all nodes of the cluster
 * for the DLM to function.
42 43 44 45 46 47 48
 *
 */

#include <asm/ioctls.h>
#include <net/sock.h>
#include <net/tcp.h>
#include <linux/pagemap.h>
49
#include <linux/file.h>
50
#include <linux/mutex.h>
51
#include <linux/sctp.h>
52
#include <linux/slab.h>
53
#include <net/sctp/sctp.h>
J
Joe Perches 已提交
54
#include <net/ipv6.h>
55 56 57 58 59 60

#include "dlm_internal.h"
#include "lowcomms.h"
#include "midcomms.h"
#include "config.h"

61
#define NEEDED_RMEM (4*1024*1024)
62
#define CONN_HASH_SIZE 32
63

64 65
/* Number of messages to send before rescheduling */
#define MAX_SEND_MSG_COUNT 25
66
#define DLM_SHUTDOWN_WAIT_TIMEOUT msecs_to_jiffies(10000)
67

68 69 70
struct connection {
	struct socket *sock;	/* NULL if not connected */
	uint32_t nodeid;	/* So we know who we are in the list */
71
	struct mutex sock_mutex;
72
	unsigned long flags;
73
#define CF_READ_PENDING 1
74
#define CF_WRITE_PENDING 2
75 76
#define CF_INIT_PENDING 4
#define CF_IS_OTHERCON 5
77
#define CF_CLOSE 6
78
#define CF_APP_LIMITED 7
79
#define CF_CLOSING 8
80
#define CF_SHUTDOWN 9
81
#define CF_CONNECTED 10
P
Patrick Caulfield 已提交
82
	struct list_head writequeue;  /* List of outgoing writequeue_entries */
83
	spinlock_t writequeue_lock;
84
	void (*connect_action) (struct connection *);	/* What to do to connect */
85
	void (*shutdown_action)(struct connection *con); /* What to do to shutdown */
86 87
	int retries;
#define MAX_CONNECT_RETRIES 3
88
	struct hlist_node list;
89
	struct connection *othercon;
90 91
	struct work_struct rwork; /* Receive workqueue */
	struct work_struct swork; /* Send workqueue */
92
	wait_queue_head_t shutdown_wait; /* wait for graceful shutdown */
93 94 95
	unsigned char *rx_buf;
	int rx_buflen;
	int rx_leftover;
96
	struct rcu_head rcu;
97 98 99
};
#define sock2con(x) ((struct connection *)(x)->sk_user_data)

100 101 102 103 104
struct listen_connection {
	struct socket *sock;
	struct work_struct rwork;
};

105 106 107 108 109 110 111 112 113 114 115
/* An entry waiting to be sent */
struct writequeue_entry {
	struct list_head list;
	struct page *page;
	int offset;
	int len;
	int end;
	int users;
	struct connection *con;
};

116 117 118 119
struct dlm_node_addr {
	struct list_head list;
	int nodeid;
	int addr_count;
120
	int curr_addr_index;
121 122 123
	struct sockaddr_storage *addr[DLM_MAX_ADDR_COUNT];
};

B
Bob Peterson 已提交
124 125 126 127 128 129 130
static struct listen_sock_callbacks {
	void (*sk_error_report)(struct sock *);
	void (*sk_data_ready)(struct sock *);
	void (*sk_state_change)(struct sock *);
	void (*sk_write_space)(struct sock *);
} listen_sock;

131 132 133
static LIST_HEAD(dlm_node_addrs);
static DEFINE_SPINLOCK(dlm_node_addrs_spin);

134
static struct listen_connection listen_con;
135 136
static struct sockaddr_storage *dlm_local_addr[DLM_MAX_ADDR_COUNT];
static int dlm_local_count;
137
static int dlm_allow_conn;
138

139 140 141
/* Work queues */
static struct workqueue_struct *recv_workqueue;
static struct workqueue_struct *send_workqueue;
142

143
static struct hlist_head connection_hash[CONN_HASH_SIZE];
144 145
static DEFINE_SPINLOCK(connections_lock);
DEFINE_STATIC_SRCU(connections_srcu);
146

147 148
static void process_recv_sockets(struct work_struct *work);
static void process_send_sockets(struct work_struct *work);
149

150 151
static void sctp_connect_to_sock(struct connection *con);
static void tcp_connect_to_sock(struct connection *con);
152
static void dlm_tcp_shutdown(struct connection *con);
153 154 155 156 157 158 159 160 161 162 163

/* This is deliberately very simple because most clusters have simple
   sequential nodeids, so we should be able to go straight to a connection
   struct in the array */
static inline int nodeid_hash(int nodeid)
{
	return nodeid & (CONN_HASH_SIZE-1);
}

static struct connection *__find_con(int nodeid)
{
164
	int r, idx;
165 166 167 168
	struct connection *con;

	r = nodeid_hash(nodeid);

169 170 171 172
	idx = srcu_read_lock(&connections_srcu);
	hlist_for_each_entry_rcu(con, &connection_hash[r], list) {
		if (con->nodeid == nodeid) {
			srcu_read_unlock(&connections_srcu, idx);
173
			return con;
174
		}
175
	}
176 177
	srcu_read_unlock(&connections_srcu, idx);

178 179 180
	return NULL;
}

181
static int dlm_con_init(struct connection *con, int nodeid)
182
{
183 184
	con->rx_buflen = dlm_config.ci_buffer_size;
	con->rx_buf = kmalloc(con->rx_buflen, GFP_NOFS);
185 186
	if (!con->rx_buf)
		return -ENOMEM;
187

188 189 190 191 192 193
	con->nodeid = nodeid;
	mutex_init(&con->sock_mutex);
	INIT_LIST_HEAD(&con->writequeue);
	spin_lock_init(&con->writequeue_lock);
	INIT_WORK(&con->swork, process_send_sockets);
	INIT_WORK(&con->rwork, process_recv_sockets);
194
	init_waitqueue_head(&con->shutdown_wait);
195

196
	if (dlm_config.ci_protocol == 0) {
197
		con->connect_action = tcp_connect_to_sock;
198 199
		con->shutdown_action = dlm_tcp_shutdown;
	} else {
200
		con->connect_action = sctp_connect_to_sock;
201
	}
202

203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228
	return 0;
}

/*
 * If 'allocation' is zero then we don't attempt to create a new
 * connection structure for this node.
 */
static struct connection *nodeid2con(int nodeid, gfp_t alloc)
{
	struct connection *con, *tmp;
	int r, ret;

	con = __find_con(nodeid);
	if (con || !alloc)
		return con;

	con = kzalloc(sizeof(*con), alloc);
	if (!con)
		return NULL;

	ret = dlm_con_init(con, nodeid);
	if (ret) {
		kfree(con);
		return NULL;
	}

229 230 231
	r = nodeid_hash(nodeid);

	spin_lock(&connections_lock);
A
Alexander Aring 已提交
232 233 234 235 236 237 238 239 240 241 242 243 244 245
	/* Because multiple workqueues/threads calls this function it can
	 * race on multiple cpu's. Instead of locking hot path __find_con()
	 * we just check in rare cases of recently added nodes again
	 * under protection of connections_lock. If this is the case we
	 * abort our connection creation and return the existing connection.
	 */
	tmp = __find_con(nodeid);
	if (tmp) {
		spin_unlock(&connections_lock);
		kfree(con->rx_buf);
		kfree(con);
		return tmp;
	}

246 247 248
	hlist_add_head_rcu(&con->list, &connection_hash[r]);
	spin_unlock(&connections_lock);

249 250 251
	return con;
}

252 253 254
/* Loop round all connections */
static void foreach_conn(void (*conn_func)(struct connection *c))
{
255
	int i, idx;
256 257
	struct connection *con;

258
	idx = srcu_read_lock(&connections_srcu);
259
	for (i = 0; i < CONN_HASH_SIZE; i++) {
260
		hlist_for_each_entry_rcu(con, &connection_hash[i], list)
261 262
			conn_func(con);
	}
263
	srcu_read_unlock(&connections_srcu, idx);
264 265
}

266 267 268 269 270 271 272 273 274 275 276
static struct dlm_node_addr *find_node_addr(int nodeid)
{
	struct dlm_node_addr *na;

	list_for_each_entry(na, &dlm_node_addrs, list) {
		if (na->nodeid == nodeid)
			return na;
	}
	return NULL;
}

A
Alexander Aring 已提交
277 278
static int addr_compare(const struct sockaddr_storage *x,
			const struct sockaddr_storage *y)
279
{
280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305
	switch (x->ss_family) {
	case AF_INET: {
		struct sockaddr_in *sinx = (struct sockaddr_in *)x;
		struct sockaddr_in *siny = (struct sockaddr_in *)y;
		if (sinx->sin_addr.s_addr != siny->sin_addr.s_addr)
			return 0;
		if (sinx->sin_port != siny->sin_port)
			return 0;
		break;
	}
	case AF_INET6: {
		struct sockaddr_in6 *sinx = (struct sockaddr_in6 *)x;
		struct sockaddr_in6 *siny = (struct sockaddr_in6 *)y;
		if (!ipv6_addr_equal(&sinx->sin6_addr, &siny->sin6_addr))
			return 0;
		if (sinx->sin6_port != siny->sin6_port)
			return 0;
		break;
	}
	default:
		return 0;
	}
	return 1;
}

static int nodeid_to_addr(int nodeid, struct sockaddr_storage *sas_out,
306
			  struct sockaddr *sa_out, bool try_new_addr)
307 308 309
{
	struct sockaddr_storage sas;
	struct dlm_node_addr *na;
310 311 312 313

	if (!dlm_local_count)
		return -1;

314 315
	spin_lock(&dlm_node_addrs_spin);
	na = find_node_addr(nodeid);
316
	if (na && na->addr_count) {
317 318 319
		memcpy(&sas, na->addr[na->curr_addr_index],
		       sizeof(struct sockaddr_storage));

320 321 322 323 324 325
		if (try_new_addr) {
			na->curr_addr_index++;
			if (na->curr_addr_index == na->addr_count)
				na->curr_addr_index = 0;
		}
	}
326 327 328 329 330 331 332 333 334 335 336 337 338
	spin_unlock(&dlm_node_addrs_spin);

	if (!na)
		return -EEXIST;

	if (!na->addr_count)
		return -ENOENT;

	if (sas_out)
		memcpy(sas_out, &sas, sizeof(struct sockaddr_storage));

	if (!sa_out)
		return 0;
339 340

	if (dlm_local_addr[0]->ss_family == AF_INET) {
341 342
		struct sockaddr_in *in4  = (struct sockaddr_in *) &sas;
		struct sockaddr_in *ret4 = (struct sockaddr_in *) sa_out;
343 344
		ret4->sin_addr.s_addr = in4->sin_addr.s_addr;
	} else {
345 346
		struct sockaddr_in6 *in6  = (struct sockaddr_in6 *) &sas;
		struct sockaddr_in6 *ret6 = (struct sockaddr_in6 *) sa_out;
A
Alexey Dobriyan 已提交
347
		ret6->sin6_addr = in6->sin6_addr;
348 349 350 351 352
	}

	return 0;
}

353 354 355 356
static int addr_to_nodeid(struct sockaddr_storage *addr, int *nodeid)
{
	struct dlm_node_addr *na;
	int rv = -EEXIST;
357
	int addr_i;
358 359 360 361 362 363

	spin_lock(&dlm_node_addrs_spin);
	list_for_each_entry(na, &dlm_node_addrs, list) {
		if (!na->addr_count)
			continue;

364 365 366 367 368 369 370
		for (addr_i = 0; addr_i < na->addr_count; addr_i++) {
			if (addr_compare(na->addr[addr_i], addr)) {
				*nodeid = na->nodeid;
				rv = 0;
				goto unlock;
			}
		}
371
	}
372
unlock:
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417
	spin_unlock(&dlm_node_addrs_spin);
	return rv;
}

int dlm_lowcomms_addr(int nodeid, struct sockaddr_storage *addr, int len)
{
	struct sockaddr_storage *new_addr;
	struct dlm_node_addr *new_node, *na;

	new_node = kzalloc(sizeof(struct dlm_node_addr), GFP_NOFS);
	if (!new_node)
		return -ENOMEM;

	new_addr = kzalloc(sizeof(struct sockaddr_storage), GFP_NOFS);
	if (!new_addr) {
		kfree(new_node);
		return -ENOMEM;
	}

	memcpy(new_addr, addr, len);

	spin_lock(&dlm_node_addrs_spin);
	na = find_node_addr(nodeid);
	if (!na) {
		new_node->nodeid = nodeid;
		new_node->addr[0] = new_addr;
		new_node->addr_count = 1;
		list_add(&new_node->list, &dlm_node_addrs);
		spin_unlock(&dlm_node_addrs_spin);
		return 0;
	}

	if (na->addr_count >= DLM_MAX_ADDR_COUNT) {
		spin_unlock(&dlm_node_addrs_spin);
		kfree(new_addr);
		kfree(new_node);
		return -ENOSPC;
	}

	na->addr[na->addr_count++] = new_addr;
	spin_unlock(&dlm_node_addrs_spin);
	kfree(new_node);
	return 0;
}

418
/* Data available on socket or listen socket received a connect */
419
static void lowcomms_data_ready(struct sock *sk)
420
{
421 422 423 424
	struct connection *con;

	read_lock_bh(&sk->sk_callback_lock);
	con = sock2con(sk);
P
Patrick Caulfield 已提交
425
	if (con && !test_and_set_bit(CF_READ_PENDING, &con->flags))
426
		queue_work(recv_workqueue, &con->rwork);
427
	read_unlock_bh(&sk->sk_callback_lock);
428 429
}

430 431 432 433 434
static void lowcomms_listen_data_ready(struct sock *sk)
{
	queue_work(recv_workqueue, &listen_con.rwork);
}

435 436
static void lowcomms_write_space(struct sock *sk)
{
437
	struct connection *con;
438

439 440
	read_lock_bh(&sk->sk_callback_lock);
	con = sock2con(sk);
441
	if (!con)
442
		goto out;
443

444 445 446 447 448 449
	if (!test_and_set_bit(CF_CONNECTED, &con->flags)) {
		log_print("successful connected to node %d", con->nodeid);
		queue_work(send_workqueue, &con->swork);
		goto out;
	}

450 451 452 453
	clear_bit(SOCK_NOSPACE, &con->sock->flags);

	if (test_and_clear_bit(CF_APP_LIMITED, &con->flags)) {
		con->sock->sk->sk_write_pending--;
454
		clear_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags);
455 456
	}

457
	queue_work(send_workqueue, &con->swork);
458 459
out:
	read_unlock_bh(&sk->sk_callback_lock);
460 461 462 463
}

static inline void lowcomms_connect_sock(struct connection *con)
{
464 465
	if (test_bit(CF_CLOSE, &con->flags))
		return;
466 467
	queue_work(send_workqueue, &con->swork);
	cond_resched();
468 469 470 471
}

static void lowcomms_state_change(struct sock *sk)
{
472 473 474 475 476 477 478 479 480
	/* SCTP layer is not calling sk_data_ready when the connection
	 * is done, so we catch the signal through here. Also, it
	 * doesn't switch socket state when entering shutdown, so we
	 * skip the write in that case.
	 */
	if (sk->sk_shutdown) {
		if (sk->sk_shutdown == RCV_SHUTDOWN)
			lowcomms_data_ready(sk);
	} else if (sk->sk_state == TCP_ESTABLISHED) {
481
		lowcomms_write_space(sk);
482
	}
483 484
}

485 486 487 488 489 490 491 492 493 494 495 496 497 498
int dlm_lowcomms_connect_node(int nodeid)
{
	struct connection *con;

	if (nodeid == dlm_our_nodeid())
		return 0;

	con = nodeid2con(nodeid, GFP_NOFS);
	if (!con)
		return -ENOMEM;
	lowcomms_connect_sock(con);
	return 0;
}

499 500
static void lowcomms_error_report(struct sock *sk)
{
501
	struct connection *con;
502
	struct sockaddr_storage saddr;
503
	void (*orig_report)(struct sock *) = NULL;
504

505 506 507 508 509
	read_lock_bh(&sk->sk_callback_lock);
	con = sock2con(sk);
	if (con == NULL)
		goto out;

B
Bob Peterson 已提交
510
	orig_report = listen_sock.sk_error_report;
511
	if (con->sock == NULL ||
512
	    kernel_getpeername(con->sock, (struct sockaddr *)&saddr) < 0) {
513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539
		printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
				   "sending to node %d, port %d, "
				   "sk_err=%d/%d\n", dlm_our_nodeid(),
				   con->nodeid, dlm_config.ci_tcp_port,
				   sk->sk_err, sk->sk_err_soft);
	} else if (saddr.ss_family == AF_INET) {
		struct sockaddr_in *sin4 = (struct sockaddr_in *)&saddr;

		printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
				   "sending to node %d at %pI4, port %d, "
				   "sk_err=%d/%d\n", dlm_our_nodeid(),
				   con->nodeid, &sin4->sin_addr.s_addr,
				   dlm_config.ci_tcp_port, sk->sk_err,
				   sk->sk_err_soft);
	} else {
		struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&saddr;

		printk_ratelimited(KERN_ERR "dlm: node %d: socket error "
				   "sending to node %d at %u.%u.%u.%u, "
				   "port %d, sk_err=%d/%d\n", dlm_our_nodeid(),
				   con->nodeid, sin6->sin6_addr.s6_addr32[0],
				   sin6->sin6_addr.s6_addr32[1],
				   sin6->sin6_addr.s6_addr32[2],
				   sin6->sin6_addr.s6_addr32[3],
				   dlm_config.ci_tcp_port, sk->sk_err,
				   sk->sk_err_soft);
	}
540 541 542 543 544 545 546
out:
	read_unlock_bh(&sk->sk_callback_lock);
	if (orig_report)
		orig_report(sk);
}

/* Note: sk_callback_lock must be locked before calling this function. */
B
Bob Peterson 已提交
547
static void save_listen_callbacks(struct socket *sock)
548
{
B
Bob Peterson 已提交
549 550 551 552 553 554
	struct sock *sk = sock->sk;

	listen_sock.sk_data_ready = sk->sk_data_ready;
	listen_sock.sk_state_change = sk->sk_state_change;
	listen_sock.sk_write_space = sk->sk_write_space;
	listen_sock.sk_error_report = sk->sk_error_report;
555 556
}

B
Bob Peterson 已提交
557
static void restore_callbacks(struct socket *sock)
558
{
B
Bob Peterson 已提交
559 560
	struct sock *sk = sock->sk;

561 562
	write_lock_bh(&sk->sk_callback_lock);
	sk->sk_user_data = NULL;
B
Bob Peterson 已提交
563 564 565 566
	sk->sk_data_ready = listen_sock.sk_data_ready;
	sk->sk_state_change = listen_sock.sk_state_change;
	sk->sk_write_space = listen_sock.sk_write_space;
	sk->sk_error_report = listen_sock.sk_error_report;
567
	write_unlock_bh(&sk->sk_callback_lock);
568 569
}

570 571 572 573 574 575 576 577 578 579 580 581 582 583 584
static void add_listen_sock(struct socket *sock, struct listen_connection *con)
{
	struct sock *sk = sock->sk;

	write_lock_bh(&sk->sk_callback_lock);
	save_listen_callbacks(sock);
	con->sock = sock;

	sk->sk_user_data = con;
	sk->sk_allocation = GFP_NOFS;
	/* Install a data_ready callback */
	sk->sk_data_ready = lowcomms_listen_data_ready;
	write_unlock_bh(&sk->sk_callback_lock);
}

585
/* Make a socket active */
586
static void add_sock(struct socket *sock, struct connection *con)
587
{
588 589 590
	struct sock *sk = sock->sk;

	write_lock_bh(&sk->sk_callback_lock);
591 592
	con->sock = sock;

593
	sk->sk_user_data = con;
594
	/* Install a data_ready callback */
595 596 597 598 599 600
	sk->sk_data_ready = lowcomms_data_ready;
	sk->sk_write_space = lowcomms_write_space;
	sk->sk_state_change = lowcomms_state_change;
	sk->sk_allocation = GFP_NOFS;
	sk->sk_error_report = lowcomms_error_report;
	write_unlock_bh(&sk->sk_callback_lock);
601 602
}

603
/* Add the port number to an IPv6 or 4 sockaddr and return the address
604 605 606 607
   length */
static void make_sockaddr(struct sockaddr_storage *saddr, uint16_t port,
			  int *addr_len)
{
608
	saddr->ss_family =  dlm_local_addr[0]->ss_family;
P
Patrick Caulfield 已提交
609
	if (saddr->ss_family == AF_INET) {
610 611 612
		struct sockaddr_in *in4_addr = (struct sockaddr_in *)saddr;
		in4_addr->sin_port = cpu_to_be16(port);
		*addr_len = sizeof(struct sockaddr_in);
613
		memset(&in4_addr->sin_zero, 0, sizeof(in4_addr->sin_zero));
P
Patrick Caulfield 已提交
614
	} else {
615 616 617 618
		struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)saddr;
		in6_addr->sin6_port = cpu_to_be16(port);
		*addr_len = sizeof(struct sockaddr_in6);
	}
619
	memset((char *)saddr + *addr_len, 0, sizeof(struct sockaddr_storage) - *addr_len);
620 621
}

622 623 624 625 626 627 628 629 630
static void dlm_close_sock(struct socket **sock)
{
	if (*sock) {
		restore_callbacks(*sock);
		sock_release(*sock);
		*sock = NULL;
	}
}

631
/* Close a remote connection and tidy up */
632 633
static void close_connection(struct connection *con, bool and_other,
			     bool tx, bool rx)
634
{
635 636
	bool closing = test_and_set_bit(CF_CLOSING, &con->flags);

637
	if (tx && !closing && cancel_work_sync(&con->swork)) {
638
		log_print("canceled swork for node %d", con->nodeid);
639 640 641
		clear_bit(CF_WRITE_PENDING, &con->flags);
	}
	if (rx && !closing && cancel_work_sync(&con->rwork)) {
642
		log_print("canceled rwork for node %d", con->nodeid);
643 644
		clear_bit(CF_READ_PENDING, &con->flags);
	}
645

646
	mutex_lock(&con->sock_mutex);
647 648
	dlm_close_sock(&con->sock);

649
	if (con->othercon && and_other) {
P
Patrick Caulfield 已提交
650
		/* Will only re-enter once. */
651
		close_connection(con->othercon, false, true, true);
652
	}
P
Patrick Caulfield 已提交
653

654
	con->rx_leftover = 0;
655
	con->retries = 0;
656
	clear_bit(CF_CONNECTED, &con->flags);
657
	mutex_unlock(&con->sock_mutex);
658
	clear_bit(CF_CLOSING, &con->flags);
659 660
}

661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708
static void shutdown_connection(struct connection *con)
{
	int ret;

	if (cancel_work_sync(&con->swork)) {
		log_print("canceled swork for node %d", con->nodeid);
		clear_bit(CF_WRITE_PENDING, &con->flags);
	}

	mutex_lock(&con->sock_mutex);
	/* nothing to shutdown */
	if (!con->sock) {
		mutex_unlock(&con->sock_mutex);
		return;
	}

	set_bit(CF_SHUTDOWN, &con->flags);
	ret = kernel_sock_shutdown(con->sock, SHUT_WR);
	mutex_unlock(&con->sock_mutex);
	if (ret) {
		log_print("Connection %p failed to shutdown: %d will force close",
			  con, ret);
		goto force_close;
	} else {
		ret = wait_event_timeout(con->shutdown_wait,
					 !test_bit(CF_SHUTDOWN, &con->flags),
					 DLM_SHUTDOWN_WAIT_TIMEOUT);
		if (ret == 0) {
			log_print("Connection %p shutdown timed out, will force close",
				  con);
			goto force_close;
		}
	}

	return;

force_close:
	clear_bit(CF_SHUTDOWN, &con->flags);
	close_connection(con, false, true, true);
}

static void dlm_tcp_shutdown(struct connection *con)
{
	if (con->othercon)
		shutdown_connection(con->othercon);
	shutdown_connection(con);
}

709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728
static int con_realloc_receive_buf(struct connection *con, int newlen)
{
	unsigned char *newbuf;

	newbuf = kmalloc(newlen, GFP_NOFS);
	if (!newbuf)
		return -ENOMEM;

	/* copy any leftover from last receive */
	if (con->rx_leftover)
		memmove(newbuf, con->rx_buf, con->rx_leftover);

	/* swap to new buffer space */
	kfree(con->rx_buf);
	con->rx_buflen = newlen;
	con->rx_buf = newbuf;

	return 0;
}

729 730 731 732
/* Data received from remote end */
static int receive_from_sock(struct connection *con)
{
	int call_again_soon = 0;
733 734 735
	struct msghdr msg;
	struct kvec iov;
	int ret, buflen;
736

737
	mutex_lock(&con->sock_mutex);
738

739 740 741 742
	if (con->sock == NULL) {
		ret = -EAGAIN;
		goto out_close;
	}
743 744 745 746 747 748

	/* realloc if we get new buffer size to read out */
	buflen = dlm_config.ci_buffer_size;
	if (con->rx_buflen != buflen && con->rx_leftover <= buflen) {
		ret = con_realloc_receive_buf(con, buflen);
		if (ret < 0)
749 750 751
			goto out_resched;
	}

752 753
	/* calculate new buffer parameter regarding last receive and
	 * possible leftover bytes
754
	 */
755 756
	iov.iov_base = con->rx_buf + con->rx_leftover;
	iov.iov_len = con->rx_buflen - con->rx_leftover;
757

758 759 760 761
	memset(&msg, 0, sizeof(msg));
	msg.msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL;
	ret = kernel_recvmsg(con->sock, &msg, &iov, 1, iov.iov_len,
			     msg.msg_flags);
762 763
	if (ret <= 0)
		goto out_close;
764
	else if (ret == iov.iov_len)
765
		call_again_soon = 1;
P
Patrick Caulfield 已提交
766

767 768 769 770 771
	/* new buflen according readed bytes and leftover from last receive */
	buflen = ret + con->rx_leftover;
	ret = dlm_process_incoming_buffer(con->nodeid, con->rx_buf, buflen);
	if (ret < 0)
		goto out_close;
772

773 774 775 776 777 778 779 780 781
	/* calculate leftover bytes from process and put it into begin of
	 * the receive buffer, so next receive we have the full message
	 * at the start address of the receive buffer.
	 */
	con->rx_leftover = buflen - ret;
	if (con->rx_leftover) {
		memmove(con->rx_buf, con->rx_buf + ret,
			con->rx_leftover);
		call_again_soon = true;
782 783 784 785
	}

	if (call_again_soon)
		goto out_resched;
786

787
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
788
	return 0;
789

P
Patrick Caulfield 已提交
790
out_resched:
791 792
	if (!test_and_set_bit(CF_READ_PENDING, &con->flags))
		queue_work(recv_workqueue, &con->rwork);
793
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
794
	return -EAGAIN;
795

P
Patrick Caulfield 已提交
796
out_close:
797
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
798
	if (ret != -EAGAIN) {
799
		/* Reconnect when there is something to send */
800 801 802 803 804 805 806 807 808 809
		close_connection(con, false, true, false);
		if (ret == 0) {
			log_print("connection %p got EOF from %d",
				  con, con->nodeid);
			/* handling for tcp shutdown */
			clear_bit(CF_SHUTDOWN, &con->flags);
			wake_up(&con->shutdown_wait);
			/* signal to breaking receive worker */
			ret = -1;
		}
810 811 812 813 814
	}
	return ret;
}

/* Listening socket is busy, accept a connection */
815
static int accept_from_sock(struct listen_connection *con)
816 817 818 819 820 821 822
{
	int result;
	struct sockaddr_storage peeraddr;
	struct socket *newsock;
	int len;
	int nodeid;
	struct connection *newcon;
P
Patrick Caulfield 已提交
823
	struct connection *addcon;
824
	unsigned int mark;
825

826 827 828 829
	if (!dlm_allow_conn) {
		return -1;
	}

830
	if (!con->sock)
831
		return -ENOTCONN;
832

833
	result = kernel_accept(con->sock, &newsock, O_NONBLOCK);
834 835 836 837 838
	if (result < 0)
		goto accept_err;

	/* Get the connected socket's peer */
	memset(&peeraddr, 0, sizeof(peeraddr));
839 840
	len = newsock->ops->getname(newsock, (struct sockaddr *)&peeraddr, 2);
	if (len < 0) {
841 842 843 844 845 846
		result = -ECONNABORTED;
		goto accept_err;
	}

	/* Get the new node's NODEID */
	make_sockaddr(&peeraddr, 0, &len);
847
	if (addr_to_nodeid(&peeraddr, &nodeid)) {
848
		unsigned char *b=(unsigned char *)&peeraddr;
D
David Teigland 已提交
849
		log_print("connect from non cluster node");
850 851
		print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE, 
				     b, sizeof(struct sockaddr_storage));
852 853 854 855
		sock_release(newsock);
		return -1;
	}

856 857 858
	dlm_comm_mark(nodeid, &mark);
	sock_set_mark(newsock->sk, mark);

859 860 861 862 863 864 865
	log_print("got connection from %d", nodeid);

	/*  Check to see if we already have a connection to this node. This
	 *  could happen if the two nodes initiate a connection at roughly
	 *  the same time and the connections cross on the wire.
	 *  In this case we store the incoming one in "othercon"
	 */
D
David Teigland 已提交
866
	newcon = nodeid2con(nodeid, GFP_NOFS);
867 868 869 870
	if (!newcon) {
		result = -ENOMEM;
		goto accept_err;
	}
871 872

	mutex_lock(&newcon->sock_mutex);
873
	if (newcon->sock) {
P
Patrick Caulfield 已提交
874
		struct connection *othercon = newcon->othercon;
875 876

		if (!othercon) {
877
			othercon = kzalloc(sizeof(*othercon), GFP_NOFS);
878
			if (!othercon) {
D
David Teigland 已提交
879
				log_print("failed to allocate incoming socket");
880
				mutex_unlock(&newcon->sock_mutex);
881 882 883
				result = -ENOMEM;
				goto accept_err;
			}
884

885 886
			result = dlm_con_init(othercon, nodeid);
			if (result < 0) {
887 888 889 890
				kfree(othercon);
				goto accept_err;
			}

891
			newcon->othercon = othercon;
892 893 894
		} else {
			/* close other sock con if we have something new */
			close_connection(othercon, false, true, false);
895
		}
896

897
		mutex_lock_nested(&othercon->sock_mutex, 1);
898 899 900
		add_sock(newsock, othercon);
		addcon = othercon;
		mutex_unlock(&othercon->sock_mutex);
901 902
	}
	else {
903 904 905
		/* accept copies the sk after we've saved the callbacks, so we
		   don't want to save them a second time or comm errors will
		   result in calling sk_error_report recursively. */
906
		add_sock(newsock, newcon);
P
Patrick Caulfield 已提交
907
		addcon = newcon;
908 909
	}

910
	mutex_unlock(&newcon->sock_mutex);
911 912 913

	/*
	 * Add it to the active queue in case we got data
L
Lucas De Marchi 已提交
914
	 * between processing the accept adding the socket
915 916
	 * to the read_sockets list
	 */
P
Patrick Caulfield 已提交
917 918
	if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags))
		queue_work(recv_workqueue, &addcon->rwork);
919 920 921

	return 0;

P
Patrick Caulfield 已提交
922
accept_err:
923 924
	if (newsock)
		sock_release(newsock);
925 926

	if (result != -EAGAIN)
D
David Teigland 已提交
927
		log_print("error accepting connection from node: %d", result);
928 929 930
	return result;
}

931 932 933 934 935 936
static void free_entry(struct writequeue_entry *e)
{
	__free_page(e->page);
	kfree(e);
}

M
Mike Christie 已提交
937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954
/*
 * writequeue_entry_complete - try to delete and free write queue entry
 * @e: write queue entry to try to delete
 * @completed: bytes completed
 *
 * writequeue_lock must be held.
 */
static void writequeue_entry_complete(struct writequeue_entry *e, int completed)
{
	e->offset += completed;
	e->len -= completed;

	if (e->len == 0 && e->users == 0) {
		list_del(&e->list);
		free_entry(e);
	}
}

955 956 957
/*
 * sctp_bind_addrs - bind a SCTP socket to all our addresses
 */
958
static int sctp_bind_addrs(struct socket *sock, uint16_t port)
959 960
{
	struct sockaddr_storage localaddr;
961
	struct sockaddr *addr = (struct sockaddr *)&localaddr;
962 963 964 965 966 967 968
	int i, addr_len, result = 0;

	for (i = 0; i < dlm_local_count; i++) {
		memcpy(&localaddr, dlm_local_addr[i], sizeof(localaddr));
		make_sockaddr(&localaddr, port, &addr_len);

		if (!i)
969
			result = kernel_bind(sock, addr, addr_len);
970
		else
971
			result = sock_bind_add(sock->sk, addr, addr_len);
972 973 974 975 976 977 978 979 980 981

		if (result < 0) {
			log_print("Can't bind to %d addr number %d, %d.\n",
				  port, i + 1, result);
			break;
		}
	}
	return result;
}

982 983 984 985 986
/* Initiate an SCTP association.
   This is a special case of send_to_sock() in that we don't yet have a
   peeled-off socket for this association, so we use the listening socket
   and add the primary IP address of the remote node.
 */
987
static void sctp_connect_to_sock(struct connection *con)
988
{
989 990 991 992
	struct sockaddr_storage daddr;
	int result;
	int addr_len;
	struct socket *sock;
993
	unsigned int mark;
994

995
	dlm_comm_mark(con->nodeid, &mark);
996

M
Mike Christie 已提交
997
	mutex_lock(&con->sock_mutex);
998

999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
	/* Some odd races can cause double-connects, ignore them */
	if (con->retries++ > MAX_CONNECT_RETRIES)
		goto out;

	if (con->sock) {
		log_print("node %d already connected.", con->nodeid);
		goto out;
	}

	memset(&daddr, 0, sizeof(daddr));
	result = nodeid_to_addr(con->nodeid, &daddr, NULL, true);
	if (result < 0) {
1011
		log_print("no address for nodeid %d", con->nodeid);
1012
		goto out;
1013 1014
	}

1015 1016 1017 1018 1019
	/* Create a socket to communicate with */
	result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
				  SOCK_STREAM, IPPROTO_SCTP, &sock);
	if (result < 0)
		goto socket_err;
1020

1021 1022
	sock_set_mark(sock->sk, mark);

1023
	add_sock(sock, con);
1024

1025
	/* Bind to all addresses. */
1026
	if (sctp_bind_addrs(con->sock, 0))
1027
		goto bind_err;
1028

1029
	make_sockaddr(&daddr, dlm_config.ci_tcp_port, &addr_len);
1030

1031
	log_print("connecting to %d", con->nodeid);
1032

1033
	/* Turn off Nagle's algorithm */
1034
	sctp_sock_set_nodelay(sock->sk);
1035

1036 1037 1038 1039 1040
	/*
	 * Make sock->ops->connect() function return in specified time,
	 * since O_NONBLOCK argument in connect() function does not work here,
	 * then, we should restore the default value of this attribute.
	 */
C
Christoph Hellwig 已提交
1041
	sock_set_sndtimeo(sock->sk, 5);
1042
	result = sock->ops->connect(sock, (struct sockaddr *)&daddr, addr_len,
1043
				   0);
C
Christoph Hellwig 已提交
1044
	sock_set_sndtimeo(sock->sk, 0);
1045

1046 1047
	if (result == -EINPROGRESS)
		result = 0;
1048 1049 1050
	if (result == 0) {
		if (!test_and_set_bit(CF_CONNECTED, &con->flags))
			log_print("successful connected to node %d", con->nodeid);
1051
		goto out;
1052
	}
1053

1054 1055 1056
bind_err:
	con->sock = NULL;
	sock_release(sock);
1057

1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073
socket_err:
	/*
	 * Some errors are fatal and this list might need adjusting. For other
	 * errors we try again until the max number of retries is reached.
	 */
	if (result != -EHOSTUNREACH &&
	    result != -ENETUNREACH &&
	    result != -ENETDOWN &&
	    result != -EINVAL &&
	    result != -EPROTONOSUPPORT) {
		log_print("connect %d try %d error %d", con->nodeid,
			  con->retries, result);
		mutex_unlock(&con->sock_mutex);
		msleep(1000);
		lowcomms_connect_sock(con);
		return;
1074
	}
M
Mike Christie 已提交
1075

1076
out:
M
Mike Christie 已提交
1077
	mutex_unlock(&con->sock_mutex);
1078 1079
}

1080
/* Connect a new socket to its peer */
1081
static void tcp_connect_to_sock(struct connection *con)
1082
{
1083
	struct sockaddr_storage saddr, src_addr;
1084
	int addr_len;
1085
	struct socket *sock = NULL;
1086
	unsigned int mark;
1087
	int result;
1088

1089
	dlm_comm_mark(con->nodeid, &mark);
1090

1091
	mutex_lock(&con->sock_mutex);
1092 1093 1094 1095
	if (con->retries++ > MAX_CONNECT_RETRIES)
		goto out;

	/* Some odd races can cause double-connects, ignore them */
1096
	if (con->sock)
1097 1098 1099
		goto out;

	/* Create a socket to communicate with */
1100 1101
	result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
				  SOCK_STREAM, IPPROTO_TCP, &sock);
1102 1103 1104
	if (result < 0)
		goto out_err;

1105 1106
	sock_set_mark(sock->sk, mark);

1107
	memset(&saddr, 0, sizeof(saddr));
1108
	result = nodeid_to_addr(con->nodeid, &saddr, NULL, false);
1109 1110
	if (result < 0) {
		log_print("no address for nodeid %d", con->nodeid);
P
Patrick Caulfield 已提交
1111
		goto out_err;
1112
	}
1113

1114
	add_sock(sock, con);
1115

1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
	/* Bind to our cluster-known address connecting to avoid
	   routing problems */
	memcpy(&src_addr, dlm_local_addr[0], sizeof(src_addr));
	make_sockaddr(&src_addr, 0, &addr_len);
	result = sock->ops->bind(sock, (struct sockaddr *) &src_addr,
				 addr_len);
	if (result < 0) {
		log_print("could not bind for connect: %d", result);
		/* This *may* not indicate a critical error */
	}

1127
	make_sockaddr(&saddr, dlm_config.ci_tcp_port, &addr_len);
1128 1129

	log_print("connecting to %d", con->nodeid);
D
David Teigland 已提交
1130 1131

	/* Turn off Nagle's algorithm */
1132
	tcp_sock_set_nodelay(sock->sk);
D
David Teigland 已提交
1133

1134
	result = sock->ops->connect(sock, (struct sockaddr *)&saddr, addr_len,
P
Patrick Caulfield 已提交
1135
				   O_NONBLOCK);
1136 1137
	if (result == -EINPROGRESS)
		result = 0;
P
Patrick Caulfield 已提交
1138 1139
	if (result == 0)
		goto out;
1140

P
Patrick Caulfield 已提交
1141
out_err:
1142 1143 1144
	if (con->sock) {
		sock_release(con->sock);
		con->sock = NULL;
1145 1146
	} else if (sock) {
		sock_release(sock);
1147 1148 1149 1150 1151
	}
	/*
	 * Some errors are fatal and this list might need adjusting. For other
	 * errors we try again until the max number of retries is reached.
	 */
1152 1153 1154 1155 1156 1157 1158 1159 1160
	if (result != -EHOSTUNREACH &&
	    result != -ENETUNREACH &&
	    result != -ENETDOWN && 
	    result != -EINVAL &&
	    result != -EPROTONOSUPPORT) {
		log_print("connect %d try %d error %d", con->nodeid,
			  con->retries, result);
		mutex_unlock(&con->sock_mutex);
		msleep(1000);
1161
		lowcomms_connect_sock(con);
1162
		return;
1163
	}
P
Patrick Caulfield 已提交
1164
out:
1165
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
1166
	return;
1167 1168
}

1169 1170 1171 1172 1173
/* On error caller must run dlm_close_sock() for the
 * listen connection socket.
 */
static int tcp_create_listen_sock(struct listen_connection *con,
				  struct sockaddr_storage *saddr)
1174
{
P
Patrick Caulfield 已提交
1175
	struct socket *sock = NULL;
1176 1177 1178
	int result = 0;
	int addr_len;

1179
	if (dlm_local_addr[0]->ss_family == AF_INET)
1180 1181 1182 1183 1184
		addr_len = sizeof(struct sockaddr_in);
	else
		addr_len = sizeof(struct sockaddr_in6);

	/* Create a socket to communicate with */
1185 1186
	result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
				  SOCK_STREAM, IPPROTO_TCP, &sock);
1187
	if (result < 0) {
D
David Teigland 已提交
1188
		log_print("Can't create listening comms socket");
1189 1190 1191
		goto create_out;
	}

1192 1193
	sock_set_mark(sock->sk, dlm_config.ci_mark);

D
David Teigland 已提交
1194
	/* Turn off Nagle's algorithm */
1195
	tcp_sock_set_nodelay(sock->sk);
D
David Teigland 已提交
1196

C
Christoph Hellwig 已提交
1197
	sock_set_reuseaddr(sock->sk);
1198

1199
	add_listen_sock(sock, con);
1200 1201

	/* Bind to our port */
1202
	make_sockaddr(saddr, dlm_config.ci_tcp_port, &addr_len);
1203 1204
	result = sock->ops->bind(sock, (struct sockaddr *) saddr, addr_len);
	if (result < 0) {
D
David Teigland 已提交
1205
		log_print("Can't bind to port %d", dlm_config.ci_tcp_port);
1206 1207
		goto create_out;
	}
C
Christoph Hellwig 已提交
1208
	sock_set_keepalive(sock->sk);
1209 1210 1211

	result = sock->ops->listen(sock, 5);
	if (result < 0) {
D
David Teigland 已提交
1212
		log_print("Can't listen on port %d", dlm_config.ci_tcp_port);
1213 1214 1215
		goto create_out;
	}

1216 1217
	return 0;

P
Patrick Caulfield 已提交
1218
create_out:
1219
	return result;
1220 1221
}

1222 1223 1224 1225 1226 1227
/* Get local addresses */
static void init_local(void)
{
	struct sockaddr_storage sas, *addr;
	int i;

1228
	dlm_local_count = 0;
1229
	for (i = 0; i < DLM_MAX_ADDR_COUNT; i++) {
1230 1231 1232
		if (dlm_our_addr(&sas, i))
			break;

1233
		addr = kmemdup(&sas, sizeof(*addr), GFP_NOFS);
1234 1235 1236 1237 1238 1239
		if (!addr)
			break;
		dlm_local_addr[dlm_local_count++] = addr;
	}
}

1240 1241 1242 1243 1244 1245 1246 1247
static void deinit_local(void)
{
	int i;

	for (i = 0; i < dlm_local_count; i++)
		kfree(dlm_local_addr[i]);
}

1248 1249 1250 1251 1252
/* Initialise SCTP socket and bind to all interfaces
 * On error caller must run dlm_close_sock() for the
 * listen connection socket.
 */
static int sctp_listen_for_all(struct listen_connection *con)
1253 1254
{
	struct socket *sock = NULL;
1255
	int result = -EINVAL;
1256 1257 1258

	log_print("Using SCTP for communications");

1259
	result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
1260
				  SOCK_STREAM, IPPROTO_SCTP, &sock);
1261 1262 1263 1264 1265
	if (result < 0) {
		log_print("Can't create comms socket, check SCTP is loaded");
		goto out;
	}

C
Christoph Hellwig 已提交
1266
	sock_set_rcvbuf(sock->sk, NEEDED_RMEM);
1267
	sock_set_mark(sock->sk, dlm_config.ci_mark);
1268
	sctp_sock_set_nodelay(sock->sk);
M
Mike Christie 已提交
1269

1270
	add_listen_sock(sock, con);
1271

1272
	/* Bind to all addresses. */
1273 1274 1275
	result = sctp_bind_addrs(con->sock, dlm_config.ci_tcp_port);
	if (result < 0)
		goto out;
1276 1277 1278 1279

	result = sock->ops->listen(sock, 5);
	if (result < 0) {
		log_print("Can't set socket listening");
1280
		goto out;
1281 1282 1283 1284 1285 1286 1287 1288 1289
	}

	return 0;

out:
	return result;
}

static int tcp_listen_for_all(void)
1290 1291
{
	/* We don't support multi-homed hosts */
1292
	if (dlm_local_count > 1) {
D
David Teigland 已提交
1293 1294
		log_print("TCP protocol can't handle multi-homed hosts, "
			  "try SCTP");
1295 1296 1297 1298 1299
		return -EINVAL;
	}

	log_print("Using TCP for communications");

1300
	return tcp_create_listen_sock(&listen_con, dlm_local_addr[0]);
1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328
}



static struct writequeue_entry *new_writequeue_entry(struct connection *con,
						     gfp_t allocation)
{
	struct writequeue_entry *entry;

	entry = kmalloc(sizeof(struct writequeue_entry), allocation);
	if (!entry)
		return NULL;

	entry->page = alloc_page(allocation);
	if (!entry->page) {
		kfree(entry);
		return NULL;
	}

	entry->offset = 0;
	entry->len = 0;
	entry->end = 0;
	entry->users = 0;
	entry->con = con;

	return entry;
}

D
David Teigland 已提交
1329
void *dlm_lowcomms_get_buffer(int nodeid, int len, gfp_t allocation, char **ppc)
1330 1331 1332 1333 1334
{
	struct connection *con;
	struct writequeue_entry *e;
	int offset = 0;

1335 1336 1337 1338 1339 1340
	if (len > LOWCOMMS_MAX_TX_BUFFER_LEN) {
		BUILD_BUG_ON(PAGE_SIZE < LOWCOMMS_MAX_TX_BUFFER_LEN);
		log_print("failed to allocate a buffer of size %d", len);
		return NULL;
	}

1341 1342 1343 1344
	con = nodeid2con(nodeid, allocation);
	if (!con)
		return NULL;

1345
	spin_lock(&con->writequeue_lock);
1346
	e = list_entry(con->writequeue.prev, struct writequeue_entry, list);
P
Patrick Caulfield 已提交
1347
	if ((&e->list == &con->writequeue) ||
1348
	    (PAGE_SIZE - e->end < len)) {
1349 1350 1351 1352
		e = NULL;
	} else {
		offset = e->end;
		e->end += len;
1353
		e->users++;
1354 1355 1356 1357
	}
	spin_unlock(&con->writequeue_lock);

	if (e) {
P
Patrick Caulfield 已提交
1358
	got_one:
1359 1360 1361 1362 1363 1364 1365 1366 1367
		*ppc = page_address(e->page) + offset;
		return e;
	}

	e = new_writequeue_entry(con, allocation);
	if (e) {
		spin_lock(&con->writequeue_lock);
		offset = e->end;
		e->end += len;
1368
		e->users++;
1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381
		list_add_tail(&e->list, &con->writequeue);
		spin_unlock(&con->writequeue_lock);
		goto got_one;
	}
	return NULL;
}

void dlm_lowcomms_commit_buffer(void *mh)
{
	struct writequeue_entry *e = (struct writequeue_entry *)mh;
	struct connection *con = e->con;
	int users;

1382
	spin_lock(&con->writequeue_lock);
1383 1384 1385 1386 1387 1388
	users = --e->users;
	if (users)
		goto out;
	e->len = e->end - e->offset;
	spin_unlock(&con->writequeue_lock);

1389
	queue_work(send_workqueue, &con->swork);
1390 1391
	return;

P
Patrick Caulfield 已提交
1392
out:
1393 1394 1395 1396 1397
	spin_unlock(&con->writequeue_lock);
	return;
}

/* Send a message */
P
Patrick Caulfield 已提交
1398
static void send_to_sock(struct connection *con)
1399 1400 1401 1402 1403
{
	int ret = 0;
	const int msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL;
	struct writequeue_entry *e;
	int len, offset;
1404
	int count = 0;
1405

1406
	mutex_lock(&con->sock_mutex);
1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423
	if (con->sock == NULL)
		goto out_connect;

	spin_lock(&con->writequeue_lock);
	for (;;) {
		e = list_entry(con->writequeue.next, struct writequeue_entry,
			       list);
		if ((struct list_head *) e == &con->writequeue)
			break;

		len = e->len;
		offset = e->offset;
		BUG_ON(len == 0 && e->users == 0);
		spin_unlock(&con->writequeue_lock);

		ret = 0;
		if (len) {
P
Paolo Bonzini 已提交
1424 1425
			ret = kernel_sendpage(con->sock, e->page, offset, len,
					      msg_flags);
1426
			if (ret == -EAGAIN || ret == 0) {
1427
				if (ret == -EAGAIN &&
1428
				    test_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags) &&
1429 1430 1431 1432 1433 1434 1435
				    !test_and_set_bit(CF_APP_LIMITED, &con->flags)) {
					/* Notify TCP that we're limited by the
					 * application window size.
					 */
					set_bit(SOCK_NOSPACE, &con->sock->flags);
					con->sock->sk->sk_write_pending++;
				}
1436
				cond_resched();
1437
				goto out;
Y
Ying Xue 已提交
1438
			} else if (ret < 0)
1439
				goto send_error;
1440
		}
1441 1442 1443

		/* Don't starve people filling buffers */
		if (++count >= MAX_SEND_MSG_COUNT) {
P
Patrick Caulfield 已提交
1444
			cond_resched();
1445 1446
			count = 0;
		}
1447 1448

		spin_lock(&con->writequeue_lock);
M
Mike Christie 已提交
1449
		writequeue_entry_complete(e, ret);
1450 1451
	}
	spin_unlock(&con->writequeue_lock);
P
Patrick Caulfield 已提交
1452
out:
1453
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
1454
	return;
1455

P
Patrick Caulfield 已提交
1456
send_error:
1457
	mutex_unlock(&con->sock_mutex);
1458
	close_connection(con, false, false, true);
1459 1460 1461
	/* Requeue the send work. When the work daemon runs again, it will try
	   a new connection, then call this function again. */
	queue_work(send_workqueue, &con->swork);
P
Patrick Caulfield 已提交
1462
	return;
1463

P
Patrick Caulfield 已提交
1464
out_connect:
1465
	mutex_unlock(&con->sock_mutex);
1466 1467
	queue_work(send_workqueue, &con->swork);
	cond_resched();
1468 1469 1470 1471
}

static void clean_one_writequeue(struct connection *con)
{
1472
	struct writequeue_entry *e, *safe;
1473 1474

	spin_lock(&con->writequeue_lock);
1475
	list_for_each_entry_safe(e, safe, &con->writequeue, list) {
1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486
		list_del(&e->list);
		free_entry(e);
	}
	spin_unlock(&con->writequeue_lock);
}

/* Called from recovery when it knows that a node has
   left the cluster */
int dlm_lowcomms_close(int nodeid)
{
	struct connection *con;
1487
	struct dlm_node_addr *na;
1488 1489 1490 1491

	log_print("closing connection to node %d", nodeid);
	con = nodeid2con(nodeid, 0);
	if (con) {
1492
		set_bit(CF_CLOSE, &con->flags);
1493
		close_connection(con, true, true, true);
1494
		clean_one_writequeue(con);
A
Alexander Aring 已提交
1495 1496
		if (con->othercon)
			clean_one_writequeue(con->othercon);
1497
	}
1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508

	spin_lock(&dlm_node_addrs_spin);
	na = find_node_addr(nodeid);
	if (na) {
		list_del(&na->list);
		while (na->addr_count--)
			kfree(na->addr[na->addr_count]);
		kfree(na);
	}
	spin_unlock(&dlm_node_addrs_spin);

1509 1510 1511
	return 0;
}

1512
/* Receive workqueue function */
1513
static void process_recv_sockets(struct work_struct *work)
1514
{
1515 1516
	struct connection *con = container_of(work, struct connection, rwork);
	int err;
1517

1518 1519
	clear_bit(CF_READ_PENDING, &con->flags);
	do {
1520
		err = receive_from_sock(con);
1521
	} while (!err);
1522 1523
}

1524 1525 1526 1527 1528
static void process_listen_recv_socket(struct work_struct *work)
{
	accept_from_sock(&listen_con);
}

1529
/* Send workqueue function */
1530
static void process_send_sockets(struct work_struct *work)
1531
{
1532
	struct connection *con = container_of(work, struct connection, swork);
1533

1534
	clear_bit(CF_WRITE_PENDING, &con->flags);
1535
	if (con->sock == NULL) /* not mutex protected so check it inside too */
1536
		con->connect_action(con);
1537
	if (!list_empty(&con->writequeue))
1538
		send_to_sock(con);
1539 1540
}

1541
static void work_stop(void)
1542
{
1543 1544 1545 1546
	if (recv_workqueue)
		destroy_workqueue(recv_workqueue);
	if (send_workqueue)
		destroy_workqueue(send_workqueue);
1547 1548
}

1549
static int work_start(void)
1550
{
1551 1552
	recv_workqueue = alloc_workqueue("dlm_recv",
					 WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
1553 1554 1555
	if (!recv_workqueue) {
		log_print("can't start dlm_recv");
		return -ENOMEM;
1556 1557
	}

1558 1559
	send_workqueue = alloc_workqueue("dlm_send",
					 WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
1560 1561
	if (!send_workqueue) {
		log_print("can't start dlm_send");
1562
		destroy_workqueue(recv_workqueue);
1563
		return -ENOMEM;
1564 1565 1566 1567 1568
	}

	return 0;
}

1569
static void _stop_conn(struct connection *con, bool and_other)
1570
{
1571
	mutex_lock(&con->sock_mutex);
1572
	set_bit(CF_CLOSE, &con->flags);
1573
	set_bit(CF_READ_PENDING, &con->flags);
1574
	set_bit(CF_WRITE_PENDING, &con->flags);
1575 1576
	if (con->sock && con->sock->sk) {
		write_lock_bh(&con->sock->sk->sk_callback_lock);
1577
		con->sock->sk->sk_user_data = NULL;
1578 1579
		write_unlock_bh(&con->sock->sk->sk_callback_lock);
	}
1580 1581 1582 1583 1584 1585 1586 1587
	if (con->othercon && and_other)
		_stop_conn(con->othercon, false);
	mutex_unlock(&con->sock_mutex);
}

static void stop_conn(struct connection *con)
{
	_stop_conn(con, true);
1588
}
1589

1590 1591 1592 1593 1594 1595
static void shutdown_conn(struct connection *con)
{
	if (con->shutdown_action)
		con->shutdown_action(con);
}

1596 1597 1598 1599 1600 1601 1602 1603
static void connection_release(struct rcu_head *rcu)
{
	struct connection *con = container_of(rcu, struct connection, rcu);

	kfree(con->rx_buf);
	kfree(con);
}

1604 1605
static void free_conn(struct connection *con)
{
1606
	close_connection(con, true, true, true);
1607 1608 1609
	spin_lock(&connections_lock);
	hlist_del_rcu(&con->list);
	spin_unlock(&connections_lock);
1610 1611
	if (con->othercon) {
		clean_one_writequeue(con->othercon);
1612 1613
		call_srcu(&connections_srcu, &con->othercon->rcu,
			  connection_release);
1614
	}
1615
	clean_one_writequeue(con);
1616
	call_srcu(&connections_srcu, &con->rcu, connection_release);
1617 1618
}

1619 1620
static void work_flush(void)
{
1621
	int ok, idx;
1622 1623 1624 1625 1626 1627
	int i;
	struct connection *con;

	do {
		ok = 1;
		foreach_conn(stop_conn);
1628 1629 1630 1631
		if (recv_workqueue)
			flush_workqueue(recv_workqueue);
		if (send_workqueue)
			flush_workqueue(send_workqueue);
1632
		idx = srcu_read_lock(&connections_srcu);
1633
		for (i = 0; i < CONN_HASH_SIZE && ok; i++) {
1634 1635
			hlist_for_each_entry_rcu(con, &connection_hash[i],
						 list) {
1636
				ok &= test_bit(CF_READ_PENDING, &con->flags);
1637 1638
				ok &= test_bit(CF_WRITE_PENDING, &con->flags);
				if (con->othercon) {
1639 1640
					ok &= test_bit(CF_READ_PENDING,
						       &con->othercon->flags);
1641 1642 1643
					ok &= test_bit(CF_WRITE_PENDING,
						       &con->othercon->flags);
				}
1644 1645
			}
		}
1646
		srcu_read_unlock(&connections_srcu, idx);
1647 1648 1649
	} while (!ok);
}

1650 1651
void dlm_lowcomms_stop(void)
{
P
Patrick Caulfield 已提交
1652
	/* Set all the flags to prevent any
1653 1654
	   socket activity.
	*/
1655
	dlm_allow_conn = 0;
1656 1657 1658 1659 1660 1661

	if (recv_workqueue)
		flush_workqueue(recv_workqueue);
	if (send_workqueue)
		flush_workqueue(send_workqueue);

1662 1663
	dlm_close_sock(&listen_con.sock);

1664
	foreach_conn(shutdown_conn);
1665
	work_flush();
1666
	foreach_conn(free_conn);
1667
	work_stop();
1668
	deinit_local();
1669 1670 1671 1672
}

int dlm_lowcomms_start(void)
{
1673
	int error = -EINVAL;
1674 1675 1676 1677
	int i;

	for (i = 0; i < CONN_HASH_SIZE; i++)
		INIT_HLIST_HEAD(&connection_hash[i]);
1678

1679 1680
	init_local();
	if (!dlm_local_count) {
D
David Teigland 已提交
1681
		error = -ENOTCONN;
1682
		log_print("no local IP address has been set");
1683
		goto fail;
1684 1685
	}

1686 1687
	INIT_WORK(&listen_con.rwork, process_listen_recv_socket);

1688 1689
	error = work_start();
	if (error)
1690
		goto fail;
1691 1692

	dlm_allow_conn = 1;
1693 1694

	/* Start listening */
1695 1696 1697
	if (dlm_config.ci_protocol == 0)
		error = tcp_listen_for_all();
	else
1698
		error = sctp_listen_for_all(&listen_con);
1699 1700 1701 1702 1703
	if (error)
		goto fail_unlisten;

	return 0;

P
Patrick Caulfield 已提交
1704
fail_unlisten:
1705
	dlm_allow_conn = 0;
1706
	dlm_close_sock(&listen_con.sock);
1707
fail:
1708 1709
	return error;
}
1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723

void dlm_lowcomms_exit(void)
{
	struct dlm_node_addr *na, *safe;

	spin_lock(&dlm_node_addrs_spin);
	list_for_each_entry_safe(na, safe, &dlm_node_addrs, list) {
		list_del(&na->list);
		while (na->addr_count--)
			kfree(na->addr[na->addr_count]);
		kfree(na);
	}
	spin_unlock(&dlm_node_addrs_spin);
}