lowcomms.c 40.5 KB
Newer Older
1 2 3 4
/******************************************************************************
*******************************************************************************
**
**  Copyright (C) Sistina Software, Inc.  1997-2003  All rights reserved.
5
**  Copyright (C) 2004-2009 Red Hat, Inc.  All rights reserved.
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
**
**  This copyrighted material is made available to anyone wishing to use,
**  modify, copy, or redistribute it subject to the terms and conditions
**  of the GNU General Public License v.2.
**
*******************************************************************************
******************************************************************************/

/*
 * 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 已提交
24
 * be expanded for the cluster infrastructure then that is its
25 26 27 28 29 30 31 32 33 34 35 36 37 38
 * 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 已提交
39
 * lowcomms will choose to use either TCP or SCTP as its transport layer
40
 * depending on the configuration variable 'protocol'. This should be set
J
Joe Perches 已提交
41
 * to 0 (default) for TCP or 1 for SCTP. It should be configured using a
42 43
 * cluster-wide mechanism as it must be the same on all nodes of the cluster
 * for the DLM to function.
44 45 46 47 48 49 50
 *
 */

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

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

63
#define NEEDED_RMEM (4*1024*1024)
64
#define CONN_HASH_SIZE 32
65

66 67 68
/* Number of messages to send before rescheduling */
#define MAX_SEND_MSG_COUNT 25

69
struct cbuf {
P
Patrick Caulfield 已提交
70 71 72
	unsigned int base;
	unsigned int len;
	unsigned int mask;
73 74
};

P
Patrick Caulfield 已提交
75 76 77 78
static void cbuf_add(struct cbuf *cb, int n)
{
	cb->len += n;
}
79

P
Patrick Caulfield 已提交
80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101
static int cbuf_data(struct cbuf *cb)
{
	return ((cb->base + cb->len) & cb->mask);
}

static void cbuf_init(struct cbuf *cb, int size)
{
	cb->base = cb->len = 0;
	cb->mask = size-1;
}

static void cbuf_eat(struct cbuf *cb, int n)
{
	cb->len  -= n;
	cb->base += n;
	cb->base &= cb->mask;
}

static bool cbuf_empty(struct cbuf *cb)
{
	return cb->len == 0;
}
102 103 104 105

struct connection {
	struct socket *sock;	/* NULL if not connected */
	uint32_t nodeid;	/* So we know who we are in the list */
106
	struct mutex sock_mutex;
107
	unsigned long flags;
108 109 110
#define CF_READ_PENDING 1
#define CF_WRITE_PENDING 2
#define CF_CONNECT_PENDING 3
111 112
#define CF_INIT_PENDING 4
#define CF_IS_OTHERCON 5
113
#define CF_CLOSE 6
114
#define CF_APP_LIMITED 7
P
Patrick Caulfield 已提交
115
	struct list_head writequeue;  /* List of outgoing writequeue_entries */
116 117
	spinlock_t writequeue_lock;
	int (*rx_action) (struct connection *);	/* What to do when active */
118
	void (*connect_action) (struct connection *);	/* What to do to connect */
119 120 121 122
	struct page *rx_page;
	struct cbuf cb;
	int retries;
#define MAX_CONNECT_RETRIES 3
123
	struct hlist_node list;
124
	struct connection *othercon;
125 126
	struct work_struct rwork; /* Receive workqueue */
	struct work_struct swork; /* Send workqueue */
127
	void (*orig_error_report)(struct sock *sk);
128 129 130 131 132 133 134 135 136 137 138 139 140 141
};
#define sock2con(x) ((struct connection *)(x)->sk_user_data)

/* 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;
};

142 143 144 145
struct dlm_node_addr {
	struct list_head list;
	int nodeid;
	int addr_count;
146
	int curr_addr_index;
147 148 149 150 151 152
	struct sockaddr_storage *addr[DLM_MAX_ADDR_COUNT];
};

static LIST_HEAD(dlm_node_addrs);
static DEFINE_SPINLOCK(dlm_node_addrs_spin);

153 154
static struct sockaddr_storage *dlm_local_addr[DLM_MAX_ADDR_COUNT];
static int dlm_local_count;
155
static int dlm_allow_conn;
156

157 158 159
/* Work queues */
static struct workqueue_struct *recv_workqueue;
static struct workqueue_struct *send_workqueue;
160

161
static struct hlist_head connection_hash[CONN_HASH_SIZE];
162
static DEFINE_MUTEX(connections_lock);
P
Patrick Caulfield 已提交
163
static struct kmem_cache *con_cache;
164

165 166
static void process_recv_sockets(struct work_struct *work);
static void process_send_sockets(struct work_struct *work);
167

168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183

/* 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)
{
	int r;
	struct connection *con;

	r = nodeid_hash(nodeid);

184
	hlist_for_each_entry(con, &connection_hash[r], list) {
185 186 187 188 189 190
		if (con->nodeid == nodeid)
			return con;
	}
	return NULL;
}

191 192 193 194 195
/*
 * 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)
196 197
{
	struct connection *con = NULL;
198
	int r;
199

200
	con = __find_con(nodeid);
201 202
	if (con || !alloc)
		return con;
203

204 205 206
	con = kmem_cache_zalloc(con_cache, alloc);
	if (!con)
		return NULL;
207

208 209
	r = nodeid_hash(nodeid);
	hlist_add_head(&con->list, &connection_hash[r]);
210

211 212 213 214 215 216
	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);
217

218 219
	/* Setup action pointers for child sockets */
	if (con->nodeid) {
220
		struct connection *zerocon = __find_con(0);
221

222 223 224
		con->connect_action = zerocon->connect_action;
		if (!con->rx_action)
			con->rx_action = zerocon->rx_action;
225 226
	}

227 228 229
	return con;
}

230 231 232 233
/* Loop round all connections */
static void foreach_conn(void (*conn_func)(struct connection *c))
{
	int i;
234
	struct hlist_node *n;
235 236 237
	struct connection *con;

	for (i = 0; i < CONN_HASH_SIZE; i++) {
238
		hlist_for_each_entry_safe(con, n, &connection_hash[i], list)
239 240 241 242
			conn_func(con);
	}
}

243 244 245 246
static struct connection *nodeid2con(int nodeid, gfp_t allocation)
{
	struct connection *con;

247
	mutex_lock(&connections_lock);
248
	con = __nodeid2con(nodeid, allocation);
249
	mutex_unlock(&connections_lock);
250

251 252 253
	return con;
}

254 255 256 257 258 259 260 261 262 263 264 265
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;
}

static int addr_compare(struct sockaddr_storage *x, struct sockaddr_storage *y)
266
{
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
	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,
293
			  struct sockaddr *sa_out, bool try_new_addr)
294 295 296
{
	struct sockaddr_storage sas;
	struct dlm_node_addr *na;
297 298 299 300

	if (!dlm_local_count)
		return -1;

301 302
	spin_lock(&dlm_node_addrs_spin);
	na = find_node_addr(nodeid);
303
	if (na && na->addr_count) {
304 305 306
		memcpy(&sas, na->addr[na->curr_addr_index],
		       sizeof(struct sockaddr_storage));

307 308 309 310 311 312
		if (try_new_addr) {
			na->curr_addr_index++;
			if (na->curr_addr_index == na->addr_count)
				na->curr_addr_index = 0;
		}
	}
313 314 315 316 317 318 319 320 321 322 323 324 325
	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;
326 327

	if (dlm_local_addr[0]->ss_family == AF_INET) {
328 329
		struct sockaddr_in *in4  = (struct sockaddr_in *) &sas;
		struct sockaddr_in *ret4 = (struct sockaddr_in *) sa_out;
330 331
		ret4->sin_addr.s_addr = in4->sin_addr.s_addr;
	} else {
332 333
		struct sockaddr_in6 *in6  = (struct sockaddr_in6 *) &sas;
		struct sockaddr_in6 *ret6 = (struct sockaddr_in6 *) sa_out;
A
Alexey Dobriyan 已提交
334
		ret6->sin6_addr = in6->sin6_addr;
335 336 337 338 339
	}

	return 0;
}

340 341 342 343
static int addr_to_nodeid(struct sockaddr_storage *addr, int *nodeid)
{
	struct dlm_node_addr *na;
	int rv = -EEXIST;
344
	int addr_i;
345 346 347 348 349 350

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

351 352 353 354 355 356 357
		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;
			}
		}
358
	}
359
unlock:
360 361 362 363 364 365 366 367 368 369 370 371 372 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
	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;
}

405
/* Data available on socket or listen socket received a connect */
406
static void lowcomms_data_ready(struct sock *sk)
407 408
{
	struct connection *con = sock2con(sk);
P
Patrick Caulfield 已提交
409
	if (con && !test_and_set_bit(CF_READ_PENDING, &con->flags))
410
		queue_work(recv_workqueue, &con->rwork);
411 412 413 414 415 416
}

static void lowcomms_write_space(struct sock *sk)
{
	struct connection *con = sock2con(sk);

417 418 419 420 421 422 423
	if (!con)
		return;

	clear_bit(SOCK_NOSPACE, &con->sock->flags);

	if (test_and_clear_bit(CF_APP_LIMITED, &con->flags)) {
		con->sock->sk->sk_write_pending--;
424
		clear_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags);
425 426 427
	}

	if (!test_and_set_bit(CF_WRITE_PENDING, &con->flags))
428
		queue_work(send_workqueue, &con->swork);
429 430 431 432
}

static inline void lowcomms_connect_sock(struct connection *con)
{
433 434
	if (test_bit(CF_CLOSE, &con->flags))
		return;
435 436
	if (!test_and_set_bit(CF_CONNECT_PENDING, &con->flags))
		queue_work(send_workqueue, &con->swork);
437 438 439 440
}

static void lowcomms_state_change(struct sock *sk)
{
441 442 443 444 445 446 447 448 449
	/* 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) {
450
		lowcomms_write_space(sk);
451
	}
452 453
}

454 455 456 457 458 459 460 461 462 463 464 465 466 467
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;
}

468 469 470 471
static void lowcomms_error_report(struct sock *sk)
{
	struct connection *con = sock2con(sk);
	struct sockaddr_storage saddr;
472
	int buflen;
473

474 475
	if (con->sock == NULL ||
	    kernel_getpeername(con->sock, (struct sockaddr *)&saddr, &buflen)) {
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506
		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);
		return;
	} 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);
	}
	con->orig_error_report(sk);
}

507
/* Make a socket active */
508
static void add_sock(struct socket *sock, struct connection *con)
509 510 511 512 513 514 515
{
	con->sock = sock;

	/* Install a data_ready callback */
	con->sock->sk->sk_data_ready = lowcomms_data_ready;
	con->sock->sk->sk_write_space = lowcomms_write_space;
	con->sock->sk->sk_state_change = lowcomms_state_change;
516
	con->sock->sk->sk_user_data = con;
517
	con->sock->sk->sk_allocation = GFP_NOFS;
518 519
	con->orig_error_report = con->sock->sk->sk_error_report;
	con->sock->sk->sk_error_report = lowcomms_error_report;
520 521
}

522
/* Add the port number to an IPv6 or 4 sockaddr and return the address
523 524 525 526
   length */
static void make_sockaddr(struct sockaddr_storage *saddr, uint16_t port,
			  int *addr_len)
{
527
	saddr->ss_family =  dlm_local_addr[0]->ss_family;
P
Patrick Caulfield 已提交
528
	if (saddr->ss_family == AF_INET) {
529 530 531
		struct sockaddr_in *in4_addr = (struct sockaddr_in *)saddr;
		in4_addr->sin_port = cpu_to_be16(port);
		*addr_len = sizeof(struct sockaddr_in);
532
		memset(&in4_addr->sin_zero, 0, sizeof(in4_addr->sin_zero));
P
Patrick Caulfield 已提交
533
	} else {
534 535 536 537
		struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)saddr;
		in6_addr->sin6_port = cpu_to_be16(port);
		*addr_len = sizeof(struct sockaddr_in6);
	}
538
	memset((char *)saddr + *addr_len, 0, sizeof(struct sockaddr_storage) - *addr_len);
539 540 541
}

/* Close a remote connection and tidy up */
542 543
static void close_connection(struct connection *con, bool and_other,
			     bool tx, bool rx)
544
{
545 546 547 548 549 550
	clear_bit(CF_CONNECT_PENDING, &con->flags);
	clear_bit(CF_WRITE_PENDING, &con->flags);
	if (tx && cancel_work_sync(&con->swork))
		log_print("canceled swork for node %d", con->nodeid);
	if (rx && cancel_work_sync(&con->rwork))
		log_print("canceled rwork for node %d", con->nodeid);
551

552
	mutex_lock(&con->sock_mutex);
553 554 555 556 557
	if (con->sock) {
		sock_release(con->sock);
		con->sock = NULL;
	}
	if (con->othercon && and_other) {
P
Patrick Caulfield 已提交
558
		/* Will only re-enter once. */
559
		close_connection(con->othercon, false, true, true);
560 561 562 563 564
	}
	if (con->rx_page) {
		__free_page(con->rx_page);
		con->rx_page = NULL;
	}
P
Patrick Caulfield 已提交
565

566 567
	con->retries = 0;
	mutex_unlock(&con->sock_mutex);
568 569 570 571 572 573
}

/* Data received from remote end */
static int receive_from_sock(struct connection *con)
{
	int ret = 0;
A
Al Viro 已提交
574 575
	struct msghdr msg = {};
	struct kvec iov[2];
576 577 578
	unsigned len;
	int r;
	int call_again_soon = 0;
A
Al Viro 已提交
579
	int nvec;
580

581
	mutex_lock(&con->sock_mutex);
582

583 584 585 586
	if (con->sock == NULL) {
		ret = -EAGAIN;
		goto out_close;
	}
587 588 589 590
	if (con->nodeid == 0) {
		ret = -EINVAL;
		goto out_close;
	}
591

592 593 594 595 596 597 598 599
	if (con->rx_page == NULL) {
		/*
		 * This doesn't need to be atomic, but I think it should
		 * improve performance if it is.
		 */
		con->rx_page = alloc_page(GFP_ATOMIC);
		if (con->rx_page == NULL)
			goto out_resched;
P
Patrick Caulfield 已提交
600
		cbuf_init(&con->cb, PAGE_CACHE_SIZE);
601 602 603 604 605 606
	}

	/*
	 * iov[0] is the bit of the circular buffer between the current end
	 * point (cb.base + cb.len) and the end of the buffer.
	 */
P
Patrick Caulfield 已提交
607 608
	iov[0].iov_len = con->cb.base - cbuf_data(&con->cb);
	iov[0].iov_base = page_address(con->rx_page) + cbuf_data(&con->cb);
609
	iov[1].iov_len = 0;
A
Al Viro 已提交
610
	nvec = 1;
611 612 613 614 615

	/*
	 * iov[1] is the bit of the circular buffer between the start of the
	 * buffer and the start of the currently used section (cb.base)
	 */
P
Patrick Caulfield 已提交
616 617
	if (cbuf_data(&con->cb) >= con->cb.base) {
		iov[0].iov_len = PAGE_CACHE_SIZE - cbuf_data(&con->cb);
618 619
		iov[1].iov_len = con->cb.base;
		iov[1].iov_base = page_address(con->rx_page);
A
Al Viro 已提交
620
		nvec = 2;
621 622 623
	}
	len = iov[0].iov_len + iov[1].iov_len;

A
Al Viro 已提交
624
	r = ret = kernel_recvmsg(con->sock, &msg, iov, nvec, len,
625 626 627
			       MSG_DONTWAIT | MSG_NOSIGNAL);
	if (ret <= 0)
		goto out_close;
628 629
	else if (ret == len)
		call_again_soon = 1;
P
Patrick Caulfield 已提交
630

P
Patrick Caulfield 已提交
631
	cbuf_add(&con->cb, ret);
632 633 634 635 636
	ret = dlm_process_incoming_buffer(con->nodeid,
					  page_address(con->rx_page),
					  con->cb.base, con->cb.len,
					  PAGE_CACHE_SIZE);
	if (ret == -EBADMSG) {
637 638 639
		log_print("lowcomms: addr=%p, base=%u, len=%u, read=%d",
			  page_address(con->rx_page), con->cb.base,
			  con->cb.len, r);
640 641 642
	}
	if (ret < 0)
		goto out_close;
P
Patrick Caulfield 已提交
643
	cbuf_eat(&con->cb, ret);
644

P
Patrick Caulfield 已提交
645
	if (cbuf_empty(&con->cb) && !call_again_soon) {
646 647 648 649 650 651
		__free_page(con->rx_page);
		con->rx_page = NULL;
	}

	if (call_again_soon)
		goto out_resched;
652
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
653
	return 0;
654

P
Patrick Caulfield 已提交
655
out_resched:
656 657
	if (!test_and_set_bit(CF_READ_PENDING, &con->flags))
		queue_work(recv_workqueue, &con->rwork);
658
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
659
	return -EAGAIN;
660

P
Patrick Caulfield 已提交
661
out_close:
662
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
663
	if (ret != -EAGAIN) {
664
		close_connection(con, false, true, false);
665 666
		/* Reconnect when there is something to send */
	}
667 668 669
	/* Don't return success if we really got EOF */
	if (ret == 0)
		ret = -EAGAIN;
670 671 672 673 674

	return ret;
}

/* Listening socket is busy, accept a connection */
675
static int tcp_accept_from_sock(struct connection *con)
676 677 678 679 680 681 682
{
	int result;
	struct sockaddr_storage peeraddr;
	struct socket *newsock;
	int len;
	int nodeid;
	struct connection *newcon;
P
Patrick Caulfield 已提交
683
	struct connection *addcon;
684

685 686 687 688 689 690 691
	mutex_lock(&connections_lock);
	if (!dlm_allow_conn) {
		mutex_unlock(&connections_lock);
		return -1;
	}
	mutex_unlock(&connections_lock);

692
	memset(&peeraddr, 0, sizeof(peeraddr));
693 694
	result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
				  SOCK_STREAM, IPPROTO_TCP, &newsock);
695 696 697
	if (result < 0)
		return -ENOMEM;

698
	mutex_lock_nested(&con->sock_mutex, 0);
699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720

	result = -ENOTCONN;
	if (con->sock == NULL)
		goto accept_err;

	newsock->type = con->sock->type;
	newsock->ops = con->sock->ops;

	result = con->sock->ops->accept(con->sock, newsock, O_NONBLOCK);
	if (result < 0)
		goto accept_err;

	/* Get the connected socket's peer */
	memset(&peeraddr, 0, sizeof(peeraddr));
	if (newsock->ops->getname(newsock, (struct sockaddr *)&peeraddr,
				  &len, 2)) {
		result = -ECONNABORTED;
		goto accept_err;
	}

	/* Get the new node's NODEID */
	make_sockaddr(&peeraddr, 0, &len);
721
	if (addr_to_nodeid(&peeraddr, &nodeid)) {
722
		unsigned char *b=(unsigned char *)&peeraddr;
D
David Teigland 已提交
723
		log_print("connect from non cluster node");
724 725
		print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE, 
				     b, sizeof(struct sockaddr_storage));
726
		sock_release(newsock);
727
		mutex_unlock(&con->sock_mutex);
728 729 730 731 732 733 734 735 736 737
		return -1;
	}

	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 已提交
738
	newcon = nodeid2con(nodeid, GFP_NOFS);
739 740 741 742
	if (!newcon) {
		result = -ENOMEM;
		goto accept_err;
	}
743
	mutex_lock_nested(&newcon->sock_mutex, 1);
744
	if (newcon->sock) {
P
Patrick Caulfield 已提交
745
		struct connection *othercon = newcon->othercon;
746 747

		if (!othercon) {
D
David Teigland 已提交
748
			othercon = kmem_cache_zalloc(con_cache, GFP_NOFS);
749
			if (!othercon) {
D
David Teigland 已提交
750
				log_print("failed to allocate incoming socket");
751
				mutex_unlock(&newcon->sock_mutex);
752 753 754 755 756
				result = -ENOMEM;
				goto accept_err;
			}
			othercon->nodeid = nodeid;
			othercon->rx_action = receive_from_sock;
757
			mutex_init(&othercon->sock_mutex);
758 759
			INIT_WORK(&othercon->swork, process_send_sockets);
			INIT_WORK(&othercon->rwork, process_recv_sockets);
760
			set_bit(CF_IS_OTHERCON, &othercon->flags);
761 762
		}
		if (!othercon->sock) {
763
			newcon->othercon = othercon;
764 765 766 767 768 769 770 771
			othercon->sock = newsock;
			newsock->sk->sk_user_data = othercon;
			add_sock(newsock, othercon);
			addcon = othercon;
		}
		else {
			printk("Extra connection from node %d attempted\n", nodeid);
			result = -EAGAIN;
772
			mutex_unlock(&newcon->sock_mutex);
773
			goto accept_err;
774 775 776 777 778 779
		}
	}
	else {
		newsock->sk->sk_user_data = newcon;
		newcon->rx_action = receive_from_sock;
		add_sock(newsock, newcon);
P
Patrick Caulfield 已提交
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
L
Lucas De Marchi 已提交
787
	 * between processing the accept adding the socket
788 789
	 * to the read_sockets list
	 */
P
Patrick Caulfield 已提交
790 791
	if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags))
		queue_work(recv_workqueue, &addcon->rwork);
792
	mutex_unlock(&con->sock_mutex);
793 794 795

	return 0;

P
Patrick Caulfield 已提交
796
accept_err:
797
	mutex_unlock(&con->sock_mutex);
798 799 800
	sock_release(newsock);

	if (result != -EAGAIN)
D
David Teigland 已提交
801
		log_print("error accepting connection from node: %d", result);
802 803 804
	return result;
}

805
static int sctp_accept_from_sock(struct connection *con)
806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918
{
	/* Check that the new node is in the lockspace */
	struct sctp_prim prim;
	int nodeid;
	int prim_len, ret;
	int addr_len;
	struct connection *newcon;
	struct connection *addcon;
	struct socket *newsock;

	mutex_lock(&connections_lock);
	if (!dlm_allow_conn) {
		mutex_unlock(&connections_lock);
		return -1;
	}
	mutex_unlock(&connections_lock);

	mutex_lock_nested(&con->sock_mutex, 0);

	ret = kernel_accept(con->sock, &newsock, O_NONBLOCK);
	if (ret < 0)
		goto accept_err;

	memset(&prim, 0, sizeof(struct sctp_prim));
	prim_len = sizeof(struct sctp_prim);

	ret = kernel_getsockopt(newsock, IPPROTO_SCTP, SCTP_PRIMARY_ADDR,
				(char *)&prim, &prim_len);
	if (ret < 0) {
		log_print("getsockopt/sctp_primary_addr failed: %d", ret);
		goto accept_err;
	}

	make_sockaddr(&prim.ssp_addr, 0, &addr_len);
	if (addr_to_nodeid(&prim.ssp_addr, &nodeid)) {
		unsigned char *b = (unsigned char *)&prim.ssp_addr;

		log_print("reject connect from unknown addr");
		print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE,
				     b, sizeof(struct sockaddr_storage));
		goto accept_err;
	}

	newcon = nodeid2con(nodeid, GFP_NOFS);
	if (!newcon) {
		ret = -ENOMEM;
		goto accept_err;
	}

	mutex_lock_nested(&newcon->sock_mutex, 1);

	if (newcon->sock) {
		struct connection *othercon = newcon->othercon;

		if (!othercon) {
			othercon = kmem_cache_zalloc(con_cache, GFP_NOFS);
			if (!othercon) {
				log_print("failed to allocate incoming socket");
				mutex_unlock(&newcon->sock_mutex);
				ret = -ENOMEM;
				goto accept_err;
			}
			othercon->nodeid = nodeid;
			othercon->rx_action = receive_from_sock;
			mutex_init(&othercon->sock_mutex);
			INIT_WORK(&othercon->swork, process_send_sockets);
			INIT_WORK(&othercon->rwork, process_recv_sockets);
			set_bit(CF_IS_OTHERCON, &othercon->flags);
		}
		if (!othercon->sock) {
			newcon->othercon = othercon;
			othercon->sock = newsock;
			newsock->sk->sk_user_data = othercon;
			add_sock(newsock, othercon);
			addcon = othercon;
		} else {
			printk("Extra connection from node %d attempted\n", nodeid);
			ret = -EAGAIN;
			mutex_unlock(&newcon->sock_mutex);
			goto accept_err;
		}
	} else {
		newsock->sk->sk_user_data = newcon;
		newcon->rx_action = receive_from_sock;
		add_sock(newsock, newcon);
		addcon = newcon;
	}

	log_print("connected to %d", nodeid);

	mutex_unlock(&newcon->sock_mutex);

	/*
	 * Add it to the active queue in case we got data
	 * between processing the accept adding the socket
	 * to the read_sockets list
	 */
	if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags))
		queue_work(recv_workqueue, &addcon->rwork);
	mutex_unlock(&con->sock_mutex);

	return 0;

accept_err:
	mutex_unlock(&con->sock_mutex);
	if (newsock)
		sock_release(newsock);
	if (ret != -EAGAIN)
		log_print("error accepting connection from node: %d", ret);

	return ret;
}

919 920 921 922 923 924
static void free_entry(struct writequeue_entry *e)
{
	__free_page(e->page);
	kfree(e);
}

M
Mike Christie 已提交
925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942
/*
 * 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);
	}
}

943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972
/*
 * sctp_bind_addrs - bind a SCTP socket to all our addresses
 */
static int sctp_bind_addrs(struct connection *con, uint16_t port)
{
	struct sockaddr_storage localaddr;
	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)
			result = kernel_bind(con->sock,
					     (struct sockaddr *)&localaddr,
					     addr_len);
		else
			result = kernel_setsockopt(con->sock, SOL_SCTP,
						   SCTP_SOCKOPT_BINDX_ADD,
						   (char *)&localaddr, addr_len);

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

973 974 975 976 977
/* 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.
 */
978
static void sctp_connect_to_sock(struct connection *con)
979
{
980 981 982 983 984 985 986 987 988 989
	struct sockaddr_storage daddr;
	int one = 1;
	int result;
	int addr_len;
	struct socket *sock;

	if (con->nodeid == 0) {
		log_print("attempt to connect sock 0 foiled");
		return;
	}
990

M
Mike Christie 已提交
991
	mutex_lock(&con->sock_mutex);
992

993 994 995 996 997 998 999 1000 1001 1002 1003 1004
	/* 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) {
1005
		log_print("no address for nodeid %d", con->nodeid);
1006
		goto out;
1007 1008
	}

1009 1010 1011 1012 1013
	/* 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;
1014

1015 1016 1017 1018
	sock->sk->sk_user_data = con;
	con->rx_action = receive_from_sock;
	con->connect_action = sctp_connect_to_sock;
	add_sock(sock, con);
1019

1020 1021 1022
	/* Bind to all addresses. */
	if (sctp_bind_addrs(con, 0))
		goto bind_err;
1023

1024
	make_sockaddr(&daddr, dlm_config.ci_tcp_port, &addr_len);
1025

1026
	log_print("connecting to %d", con->nodeid);
1027

1028 1029 1030
	/* Turn off Nagle's algorithm */
	kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one,
			  sizeof(one));
1031

1032 1033 1034 1035 1036 1037
	result = sock->ops->connect(sock, (struct sockaddr *)&daddr, addr_len,
				   O_NONBLOCK);
	if (result == -EINPROGRESS)
		result = 0;
	if (result == 0)
		goto out;
1038

1039

1040 1041 1042
bind_err:
	con->sock = NULL;
	sock_release(sock);
1043

1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
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);
1058
		clear_bit(CF_CONNECT_PENDING, &con->flags);
1059 1060
		lowcomms_connect_sock(con);
		return;
1061
	}
M
Mike Christie 已提交
1062

1063
out:
M
Mike Christie 已提交
1064
	mutex_unlock(&con->sock_mutex);
1065
	set_bit(CF_WRITE_PENDING, &con->flags);
1066 1067
}

1068
/* Connect a new socket to its peer */
1069
static void tcp_connect_to_sock(struct connection *con)
1070
{
1071
	struct sockaddr_storage saddr, src_addr;
1072
	int addr_len;
1073
	struct socket *sock = NULL;
D
David Teigland 已提交
1074
	int one = 1;
1075
	int result;
1076 1077 1078

	if (con->nodeid == 0) {
		log_print("attempt to connect sock 0 foiled");
P
Patrick Caulfield 已提交
1079
		return;
1080 1081
	}

1082
	mutex_lock(&con->sock_mutex);
1083 1084 1085 1086
	if (con->retries++ > MAX_CONNECT_RETRIES)
		goto out;

	/* Some odd races can cause double-connects, ignore them */
1087
	if (con->sock)
1088 1089 1090
		goto out;

	/* Create a socket to communicate with */
1091 1092
	result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
				  SOCK_STREAM, IPPROTO_TCP, &sock);
1093 1094 1095 1096
	if (result < 0)
		goto out_err;

	memset(&saddr, 0, sizeof(saddr));
1097
	result = nodeid_to_addr(con->nodeid, &saddr, NULL, false);
1098 1099
	if (result < 0) {
		log_print("no address for nodeid %d", con->nodeid);
P
Patrick Caulfield 已提交
1100
		goto out_err;
1101
	}
1102 1103 1104

	sock->sk->sk_user_data = con;
	con->rx_action = receive_from_sock;
1105 1106
	con->connect_action = tcp_connect_to_sock;
	add_sock(sock, con);
1107

1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118
	/* 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 */
	}

1119
	make_sockaddr(&saddr, dlm_config.ci_tcp_port, &addr_len);
1120 1121

	log_print("connecting to %d", con->nodeid);
D
David Teigland 已提交
1122 1123 1124 1125 1126

	/* Turn off Nagle's algorithm */
	kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one,
			  sizeof(one));

1127
	result = sock->ops->connect(sock, (struct sockaddr *)&saddr, addr_len,
P
Patrick Caulfield 已提交
1128
				   O_NONBLOCK);
1129 1130
	if (result == -EINPROGRESS)
		result = 0;
P
Patrick Caulfield 已提交
1131 1132
	if (result == 0)
		goto out;
1133

P
Patrick Caulfield 已提交
1134
out_err:
1135 1136 1137
	if (con->sock) {
		sock_release(con->sock);
		con->sock = NULL;
1138 1139
	} else if (sock) {
		sock_release(sock);
1140 1141 1142 1143 1144
	}
	/*
	 * Some errors are fatal and this list might need adjusting. For other
	 * errors we try again until the max number of retries is reached.
	 */
1145 1146 1147 1148 1149 1150 1151 1152 1153
	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);
1154
		clear_bit(CF_CONNECT_PENDING, &con->flags);
1155
		lowcomms_connect_sock(con);
1156
		return;
1157
	}
P
Patrick Caulfield 已提交
1158
out:
1159
	mutex_unlock(&con->sock_mutex);
1160
	set_bit(CF_WRITE_PENDING, &con->flags);
P
Patrick Caulfield 已提交
1161
	return;
1162 1163
}

1164 1165
static struct socket *tcp_create_listen_sock(struct connection *con,
					     struct sockaddr_storage *saddr)
1166
{
P
Patrick Caulfield 已提交
1167
	struct socket *sock = NULL;
1168 1169 1170 1171
	int result = 0;
	int one = 1;
	int addr_len;

1172
	if (dlm_local_addr[0]->ss_family == AF_INET)
1173 1174 1175 1176 1177
		addr_len = sizeof(struct sockaddr_in);
	else
		addr_len = sizeof(struct sockaddr_in6);

	/* Create a socket to communicate with */
1178 1179
	result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family,
				  SOCK_STREAM, IPPROTO_TCP, &sock);
1180
	if (result < 0) {
D
David Teigland 已提交
1181
		log_print("Can't create listening comms socket");
1182 1183 1184
		goto create_out;
	}

D
David Teigland 已提交
1185 1186 1187 1188
	/* Turn off Nagle's algorithm */
	kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one,
			  sizeof(one));

1189 1190 1191
	result = kernel_setsockopt(sock, SOL_SOCKET, SO_REUSEADDR,
				   (char *)&one, sizeof(one));

1192
	if (result < 0) {
D
David Teigland 已提交
1193
		log_print("Failed to set SO_REUSEADDR on socket: %d", result);
1194
	}
1195 1196
	con->rx_action = tcp_accept_from_sock;
	con->connect_action = tcp_connect_to_sock;
1197 1198

	/* Bind to our port */
1199
	make_sockaddr(saddr, dlm_config.ci_tcp_port, &addr_len);
1200 1201
	result = sock->ops->bind(sock, (struct sockaddr *) saddr, addr_len);
	if (result < 0) {
D
David Teigland 已提交
1202
		log_print("Can't bind to port %d", dlm_config.ci_tcp_port);
1203 1204 1205 1206 1207
		sock_release(sock);
		sock = NULL;
		con->sock = NULL;
		goto create_out;
	}
1208
	result = kernel_setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE,
P
Patrick Caulfield 已提交
1209
				 (char *)&one, sizeof(one));
1210
	if (result < 0) {
D
David Teigland 已提交
1211
		log_print("Set keepalive failed: %d", result);
1212 1213 1214 1215
	}

	result = sock->ops->listen(sock, 5);
	if (result < 0) {
D
David Teigland 已提交
1216
		log_print("Can't listen on port %d", dlm_config.ci_tcp_port);
1217 1218 1219 1220 1221
		sock_release(sock);
		sock = NULL;
		goto create_out;
	}

P
Patrick Caulfield 已提交
1222
create_out:
1223 1224 1225
	return sock;
}

1226 1227 1228 1229 1230 1231
/* Get local addresses */
static void init_local(void)
{
	struct sockaddr_storage sas, *addr;
	int i;

1232
	dlm_local_count = 0;
1233
	for (i = 0; i < DLM_MAX_ADDR_COUNT; i++) {
1234 1235 1236
		if (dlm_our_addr(&sas, i))
			break;

D
David Teigland 已提交
1237
		addr = kmalloc(sizeof(*addr), GFP_NOFS);
1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248
		if (!addr)
			break;
		memcpy(addr, &sas, sizeof(*addr));
		dlm_local_addr[dlm_local_count++] = addr;
	}
}

/* Initialise SCTP socket and bind to all interfaces */
static int sctp_listen_for_all(void)
{
	struct socket *sock = NULL;
1249
	int result = -EINVAL;
D
David Teigland 已提交
1250
	struct connection *con = nodeid2con(0, GFP_NOFS);
1251
	int bufsize = NEEDED_RMEM;
M
Mike Christie 已提交
1252
	int one = 1;
1253 1254 1255 1256 1257 1258

	if (!con)
		return -ENOMEM;

	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;
	}

1266
	result = kernel_setsockopt(sock, SOL_SOCKET, SO_RCVBUFFORCE,
1267 1268
				 (char *)&bufsize, sizeof(bufsize));
	if (result)
D
David Teigland 已提交
1269
		log_print("Error increasing buffer space on socket %d", result);
1270

M
Mike Christie 已提交
1271 1272 1273 1274 1275
	result = kernel_setsockopt(sock, SOL_SCTP, SCTP_NODELAY, (char *)&one,
				   sizeof(one));
	if (result < 0)
		log_print("Could not set SCTP NODELAY error %d\n", result);

1276 1277 1278 1279
	/* Init con struct */
	sock->sk->sk_user_data = con;
	con->sock = sock;
	con->sock->sk->sk_data_ready = lowcomms_data_ready;
1280 1281
	con->rx_action = sctp_accept_from_sock;
	con->connect_action = sctp_connect_to_sock;
1282

1283 1284 1285
	/* Bind to all addresses. */
	if (sctp_bind_addrs(con, dlm_config.ci_tcp_port))
		goto create_delsock;
1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302

	result = sock->ops->listen(sock, 5);
	if (result < 0) {
		log_print("Can't set socket listening");
		goto create_delsock;
	}

	return 0;

create_delsock:
	sock_release(sock);
	con->sock = NULL;
out:
	return result;
}

static int tcp_listen_for_all(void)
1303 1304
{
	struct socket *sock = NULL;
D
David Teigland 已提交
1305
	struct connection *con = nodeid2con(0, GFP_NOFS);
1306 1307
	int result = -EINVAL;

1308 1309 1310
	if (!con)
		return -ENOMEM;

1311
	/* We don't support multi-homed hosts */
1312
	if (dlm_local_addr[1] != NULL) {
D
David Teigland 已提交
1313 1314
		log_print("TCP protocol can't handle multi-homed hosts, "
			  "try SCTP");
1315 1316 1317 1318 1319 1320
		return -EINVAL;
	}

	log_print("Using TCP for communications");

	sock = tcp_create_listen_sock(con, dlm_local_addr[0]);
1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357
	if (sock) {
		add_sock(sock, con);
		result = 0;
	}
	else {
		result = -EADDRINUSE;
	}

	return result;
}



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 已提交
1358
void *dlm_lowcomms_get_buffer(int nodeid, int len, gfp_t allocation, char **ppc)
1359 1360 1361 1362 1363 1364 1365 1366 1367
{
	struct connection *con;
	struct writequeue_entry *e;
	int offset = 0;

	con = nodeid2con(nodeid, allocation);
	if (!con)
		return NULL;

1368
	spin_lock(&con->writequeue_lock);
1369
	e = list_entry(con->writequeue.prev, struct writequeue_entry, list);
P
Patrick Caulfield 已提交
1370
	if ((&e->list == &con->writequeue) ||
1371 1372 1373 1374 1375
	    (PAGE_CACHE_SIZE - e->end < len)) {
		e = NULL;
	} else {
		offset = e->end;
		e->end += len;
1376
		e->users++;
1377 1378 1379 1380
	}
	spin_unlock(&con->writequeue_lock);

	if (e) {
P
Patrick Caulfield 已提交
1381
	got_one:
1382 1383 1384 1385 1386 1387 1388 1389 1390
		*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;
1391
		e->users++;
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404
		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;

1405
	spin_lock(&con->writequeue_lock);
1406 1407 1408 1409 1410 1411
	users = --e->users;
	if (users)
		goto out;
	e->len = e->end - e->offset;
	spin_unlock(&con->writequeue_lock);

1412 1413
	if (!test_and_set_bit(CF_WRITE_PENDING, &con->flags)) {
		queue_work(send_workqueue, &con->swork);
1414 1415 1416
	}
	return;

P
Patrick Caulfield 已提交
1417
out:
1418 1419 1420 1421 1422
	spin_unlock(&con->writequeue_lock);
	return;
}

/* Send a message */
P
Patrick Caulfield 已提交
1423
static void send_to_sock(struct connection *con)
1424 1425 1426 1427 1428
{
	int ret = 0;
	const int msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL;
	struct writequeue_entry *e;
	int len, offset;
1429
	int count = 0;
1430

1431
	mutex_lock(&con->sock_mutex);
1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448
	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 已提交
1449 1450
			ret = kernel_sendpage(con->sock, e->page, offset, len,
					      msg_flags);
1451
			if (ret == -EAGAIN || ret == 0) {
1452
				if (ret == -EAGAIN &&
1453
				    test_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags) &&
1454 1455 1456 1457 1458 1459 1460
				    !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++;
				}
1461
				cond_resched();
1462
				goto out;
Y
Ying Xue 已提交
1463
			} else if (ret < 0)
1464
				goto send_error;
1465
		}
1466 1467 1468

		/* Don't starve people filling buffers */
		if (++count >= MAX_SEND_MSG_COUNT) {
P
Patrick Caulfield 已提交
1469
			cond_resched();
1470 1471
			count = 0;
		}
1472 1473

		spin_lock(&con->writequeue_lock);
M
Mike Christie 已提交
1474
		writequeue_entry_complete(e, ret);
1475 1476
	}
	spin_unlock(&con->writequeue_lock);
P
Patrick Caulfield 已提交
1477
out:
1478
	mutex_unlock(&con->sock_mutex);
P
Patrick Caulfield 已提交
1479
	return;
1480

P
Patrick Caulfield 已提交
1481
send_error:
1482
	mutex_unlock(&con->sock_mutex);
1483
	close_connection(con, false, false, true);
1484
	lowcomms_connect_sock(con);
P
Patrick Caulfield 已提交
1485
	return;
1486

P
Patrick Caulfield 已提交
1487
out_connect:
1488
	mutex_unlock(&con->sock_mutex);
1489
	lowcomms_connect_sock(con);
1490 1491 1492 1493
}

static void clean_one_writequeue(struct connection *con)
{
1494
	struct writequeue_entry *e, *safe;
1495 1496

	spin_lock(&con->writequeue_lock);
1497
	list_for_each_entry_safe(e, safe, &con->writequeue, list) {
1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508
		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;
1509
	struct dlm_node_addr *na;
1510 1511 1512 1513

	log_print("closing connection to node %d", nodeid);
	con = nodeid2con(nodeid, 0);
	if (con) {
1514
		set_bit(CF_CLOSE, &con->flags);
1515
		close_connection(con, true, true, true);
1516 1517
		clean_one_writequeue(con);
	}
1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528

	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);

1529 1530 1531
	return 0;
}

1532
/* Receive workqueue function */
1533
static void process_recv_sockets(struct work_struct *work)
1534
{
1535 1536
	struct connection *con = container_of(work, struct connection, rwork);
	int err;
1537

1538 1539 1540 1541
	clear_bit(CF_READ_PENDING, &con->flags);
	do {
		err = con->rx_action(con);
	} while (!err);
1542 1543
}

1544
/* Send workqueue function */
1545
static void process_send_sockets(struct work_struct *work)
1546
{
1547
	struct connection *con = container_of(work, struct connection, swork);
1548

1549
	if (test_and_clear_bit(CF_CONNECT_PENDING, &con->flags))
1550
		con->connect_action(con);
1551 1552
	if (test_and_clear_bit(CF_WRITE_PENDING, &con->flags))
		send_to_sock(con);
1553 1554 1555 1556 1557 1558
}


/* Discard all entries on the write queues */
static void clean_writequeues(void)
{
1559
	foreach_conn(clean_one_writequeue);
1560 1561
}

1562
static void work_stop(void)
1563
{
1564 1565
	destroy_workqueue(recv_workqueue);
	destroy_workqueue(send_workqueue);
1566 1567
}

1568
static int work_start(void)
1569
{
1570 1571
	recv_workqueue = alloc_workqueue("dlm_recv",
					 WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
1572 1573 1574
	if (!recv_workqueue) {
		log_print("can't start dlm_recv");
		return -ENOMEM;
1575 1576
	}

1577 1578
	send_workqueue = alloc_workqueue("dlm_send",
					 WQ_UNBOUND | WQ_MEM_RECLAIM, 1);
1579 1580
	if (!send_workqueue) {
		log_print("can't start dlm_send");
1581
		destroy_workqueue(recv_workqueue);
1582
		return -ENOMEM;
1583 1584 1585 1586 1587
	}

	return 0;
}

1588
static void stop_conn(struct connection *con)
1589
{
1590
	con->flags |= 0x0F;
1591
	if (con->sock && con->sock->sk)
1592 1593
		con->sock->sk->sk_user_data = NULL;
}
1594

1595 1596
static void free_conn(struct connection *con)
{
1597
	close_connection(con, true, true, true);
1598 1599 1600 1601 1602 1603 1604 1605
	if (con->othercon)
		kmem_cache_free(con_cache, con->othercon);
	hlist_del(&con->list);
	kmem_cache_free(con_cache, con);
}

void dlm_lowcomms_stop(void)
{
P
Patrick Caulfield 已提交
1606
	/* Set all the flags to prevent any
1607 1608
	   socket activity.
	*/
1609
	mutex_lock(&connections_lock);
1610
	dlm_allow_conn = 0;
1611
	foreach_conn(stop_conn);
1612
	mutex_unlock(&connections_lock);
P
Patrick Caulfield 已提交
1613

1614
	work_stop();
1615

1616
	mutex_lock(&connections_lock);
1617 1618
	clean_writequeues();

1619 1620
	foreach_conn(free_conn);

1621
	mutex_unlock(&connections_lock);
1622 1623 1624 1625 1626
	kmem_cache_destroy(con_cache);
}

int dlm_lowcomms_start(void)
{
1627 1628
	int error = -EINVAL;
	struct connection *con;
1629 1630 1631 1632
	int i;

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

1634 1635
	init_local();
	if (!dlm_local_count) {
D
David Teigland 已提交
1636
		error = -ENOTCONN;
1637
		log_print("no local IP address has been set");
1638
		goto fail;
1639 1640
	}

1641
	error = -ENOMEM;
1642
	con_cache = kmem_cache_create("dlm_conn", sizeof(struct connection),
P
Patrick Caulfield 已提交
1643
				      __alignof__(struct connection), 0,
1644
				      NULL);
1645
	if (!con_cache)
1646 1647 1648 1649 1650 1651 1652
		goto fail;

	error = work_start();
	if (error)
		goto fail_destroy;

	dlm_allow_conn = 1;
1653 1654

	/* Start listening */
1655 1656 1657 1658
	if (dlm_config.ci_protocol == 0)
		error = tcp_listen_for_all();
	else
		error = sctp_listen_for_all();
1659 1660 1661 1662 1663
	if (error)
		goto fail_unlisten;

	return 0;

P
Patrick Caulfield 已提交
1664
fail_unlisten:
1665
	dlm_allow_conn = 0;
1666 1667
	con = nodeid2con(0,0);
	if (con) {
1668
		close_connection(con, false, true, true);
1669 1670
		kmem_cache_free(con_cache, con);
	}
1671
fail_destroy:
1672
	kmem_cache_destroy(con_cache);
1673
fail:
1674 1675
	return error;
}
1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689

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);
}