16ed7257bSPatrick Caulfield /****************************************************************************** 26ed7257bSPatrick Caulfield ******************************************************************************* 36ed7257bSPatrick Caulfield ** 46ed7257bSPatrick Caulfield ** Copyright (C) Sistina Software, Inc. 1997-2003 All rights reserved. 55e9ccc37SChristine Caulfield ** Copyright (C) 2004-2009 Red Hat, Inc. All rights reserved. 66ed7257bSPatrick Caulfield ** 76ed7257bSPatrick Caulfield ** This copyrighted material is made available to anyone wishing to use, 86ed7257bSPatrick Caulfield ** modify, copy, or redistribute it subject to the terms and conditions 96ed7257bSPatrick Caulfield ** of the GNU General Public License v.2. 106ed7257bSPatrick Caulfield ** 116ed7257bSPatrick Caulfield ******************************************************************************* 126ed7257bSPatrick Caulfield ******************************************************************************/ 136ed7257bSPatrick Caulfield 146ed7257bSPatrick Caulfield /* 156ed7257bSPatrick Caulfield * lowcomms.c 166ed7257bSPatrick Caulfield * 176ed7257bSPatrick Caulfield * This is the "low-level" comms layer. 186ed7257bSPatrick Caulfield * 196ed7257bSPatrick Caulfield * It is responsible for sending/receiving messages 206ed7257bSPatrick Caulfield * from other nodes in the cluster. 216ed7257bSPatrick Caulfield * 226ed7257bSPatrick Caulfield * Cluster nodes are referred to by their nodeids. nodeids are 236ed7257bSPatrick Caulfield * simply 32 bit numbers to the locking module - if they need to 242cf12c0bSJoe Perches * be expanded for the cluster infrastructure then that is its 256ed7257bSPatrick Caulfield * responsibility. It is this layer's 266ed7257bSPatrick Caulfield * responsibility to resolve these into IP address or 276ed7257bSPatrick Caulfield * whatever it needs for inter-node communication. 286ed7257bSPatrick Caulfield * 296ed7257bSPatrick Caulfield * The comms level is two kernel threads that deal mainly with 306ed7257bSPatrick Caulfield * the receiving of messages from other nodes and passing them 316ed7257bSPatrick Caulfield * up to the mid-level comms layer (which understands the 326ed7257bSPatrick Caulfield * message format) for execution by the locking core, and 336ed7257bSPatrick Caulfield * a send thread which does all the setting up of connections 346ed7257bSPatrick Caulfield * to remote nodes and the sending of data. Threads are not allowed 356ed7257bSPatrick Caulfield * to send their own data because it may cause them to wait in times 366ed7257bSPatrick Caulfield * of high load. Also, this way, the sending thread can collect together 376ed7257bSPatrick Caulfield * messages bound for one node and send them in one block. 386ed7257bSPatrick Caulfield * 392cf12c0bSJoe Perches * lowcomms will choose to use either TCP or SCTP as its transport layer 406ed7257bSPatrick Caulfield * depending on the configuration variable 'protocol'. This should be set 416ed7257bSPatrick Caulfield * to 0 (default) for TCP or 1 for SCTP. It should be configured using a 426ed7257bSPatrick Caulfield * cluster-wide mechanism as it must be the same on all nodes of the cluster 436ed7257bSPatrick Caulfield * for the DLM to function. 446ed7257bSPatrick Caulfield * 456ed7257bSPatrick Caulfield */ 466ed7257bSPatrick Caulfield 476ed7257bSPatrick Caulfield #include <asm/ioctls.h> 486ed7257bSPatrick Caulfield #include <net/sock.h> 496ed7257bSPatrick Caulfield #include <net/tcp.h> 506ed7257bSPatrick Caulfield #include <linux/pagemap.h> 516ed7257bSPatrick Caulfield #include <linux/file.h> 527a936ce7SMatthias Kaehlcke #include <linux/mutex.h> 536ed7257bSPatrick Caulfield #include <linux/sctp.h> 545a0e3ad6STejun Heo #include <linux/slab.h> 552f2d76ccSBenjamin Poirier #include <net/sctp/sctp.h> 566ed7257bSPatrick Caulfield #include <net/sctp/user.h> 5744ad532bSJoe Perches #include <net/ipv6.h> 586ed7257bSPatrick Caulfield 596ed7257bSPatrick Caulfield #include "dlm_internal.h" 606ed7257bSPatrick Caulfield #include "lowcomms.h" 616ed7257bSPatrick Caulfield #include "midcomms.h" 626ed7257bSPatrick Caulfield #include "config.h" 636ed7257bSPatrick Caulfield 646ed7257bSPatrick Caulfield #define NEEDED_RMEM (4*1024*1024) 655e9ccc37SChristine Caulfield #define CONN_HASH_SIZE 32 666ed7257bSPatrick Caulfield 67f92c8dd7SBob Peterson /* Number of messages to send before rescheduling */ 68f92c8dd7SBob Peterson #define MAX_SEND_MSG_COUNT 25 69f92c8dd7SBob Peterson 706ed7257bSPatrick Caulfield struct cbuf { 716ed7257bSPatrick Caulfield unsigned int base; 726ed7257bSPatrick Caulfield unsigned int len; 736ed7257bSPatrick Caulfield unsigned int mask; 746ed7257bSPatrick Caulfield }; 756ed7257bSPatrick Caulfield 766ed7257bSPatrick Caulfield static void cbuf_add(struct cbuf *cb, int n) 776ed7257bSPatrick Caulfield { 786ed7257bSPatrick Caulfield cb->len += n; 796ed7257bSPatrick Caulfield } 806ed7257bSPatrick Caulfield 816ed7257bSPatrick Caulfield static int cbuf_data(struct cbuf *cb) 826ed7257bSPatrick Caulfield { 836ed7257bSPatrick Caulfield return ((cb->base + cb->len) & cb->mask); 846ed7257bSPatrick Caulfield } 856ed7257bSPatrick Caulfield 866ed7257bSPatrick Caulfield static void cbuf_init(struct cbuf *cb, int size) 876ed7257bSPatrick Caulfield { 886ed7257bSPatrick Caulfield cb->base = cb->len = 0; 896ed7257bSPatrick Caulfield cb->mask = size-1; 906ed7257bSPatrick Caulfield } 916ed7257bSPatrick Caulfield 926ed7257bSPatrick Caulfield static void cbuf_eat(struct cbuf *cb, int n) 936ed7257bSPatrick Caulfield { 946ed7257bSPatrick Caulfield cb->len -= n; 956ed7257bSPatrick Caulfield cb->base += n; 966ed7257bSPatrick Caulfield cb->base &= cb->mask; 976ed7257bSPatrick Caulfield } 986ed7257bSPatrick Caulfield 996ed7257bSPatrick Caulfield static bool cbuf_empty(struct cbuf *cb) 1006ed7257bSPatrick Caulfield { 1016ed7257bSPatrick Caulfield return cb->len == 0; 1026ed7257bSPatrick Caulfield } 1036ed7257bSPatrick Caulfield 1046ed7257bSPatrick Caulfield struct connection { 1056ed7257bSPatrick Caulfield struct socket *sock; /* NULL if not connected */ 1066ed7257bSPatrick Caulfield uint32_t nodeid; /* So we know who we are in the list */ 1076ed7257bSPatrick Caulfield struct mutex sock_mutex; 1086ed7257bSPatrick Caulfield unsigned long flags; 1096ed7257bSPatrick Caulfield #define CF_READ_PENDING 1 1106ed7257bSPatrick Caulfield #define CF_WRITE_PENDING 2 1116ed7257bSPatrick Caulfield #define CF_CONNECT_PENDING 3 1126ed7257bSPatrick Caulfield #define CF_INIT_PENDING 4 1136ed7257bSPatrick Caulfield #define CF_IS_OTHERCON 5 114063c4c99SLars Marowsky-Bree #define CF_CLOSE 6 115b36930ddSDavid Miller #define CF_APP_LIMITED 7 1166ed7257bSPatrick Caulfield struct list_head writequeue; /* List of outgoing writequeue_entries */ 1176ed7257bSPatrick Caulfield spinlock_t writequeue_lock; 1186ed7257bSPatrick Caulfield int (*rx_action) (struct connection *); /* What to do when active */ 1196ed7257bSPatrick Caulfield void (*connect_action) (struct connection *); /* What to do to connect */ 1206ed7257bSPatrick Caulfield struct page *rx_page; 1216ed7257bSPatrick Caulfield struct cbuf cb; 1226ed7257bSPatrick Caulfield int retries; 1236ed7257bSPatrick Caulfield #define MAX_CONNECT_RETRIES 3 1246ed7257bSPatrick Caulfield int sctp_assoc; 1255e9ccc37SChristine Caulfield struct hlist_node list; 1266ed7257bSPatrick Caulfield struct connection *othercon; 1276ed7257bSPatrick Caulfield struct work_struct rwork; /* Receive workqueue */ 1286ed7257bSPatrick Caulfield struct work_struct swork; /* Send workqueue */ 1296ed7257bSPatrick Caulfield }; 1306ed7257bSPatrick Caulfield #define sock2con(x) ((struct connection *)(x)->sk_user_data) 1316ed7257bSPatrick Caulfield 1326ed7257bSPatrick Caulfield /* An entry waiting to be sent */ 1336ed7257bSPatrick Caulfield struct writequeue_entry { 1346ed7257bSPatrick Caulfield struct list_head list; 1356ed7257bSPatrick Caulfield struct page *page; 1366ed7257bSPatrick Caulfield int offset; 1376ed7257bSPatrick Caulfield int len; 1386ed7257bSPatrick Caulfield int end; 1396ed7257bSPatrick Caulfield int users; 1406ed7257bSPatrick Caulfield struct connection *con; 1416ed7257bSPatrick Caulfield }; 1426ed7257bSPatrick Caulfield 14336b71a8bSDavid Teigland struct dlm_node_addr { 14436b71a8bSDavid Teigland struct list_head list; 14536b71a8bSDavid Teigland int nodeid; 14636b71a8bSDavid Teigland int addr_count; 14736b71a8bSDavid Teigland struct sockaddr_storage *addr[DLM_MAX_ADDR_COUNT]; 14836b71a8bSDavid Teigland }; 14936b71a8bSDavid Teigland 15036b71a8bSDavid Teigland static LIST_HEAD(dlm_node_addrs); 15136b71a8bSDavid Teigland static DEFINE_SPINLOCK(dlm_node_addrs_spin); 15236b71a8bSDavid Teigland 1536ed7257bSPatrick Caulfield static struct sockaddr_storage *dlm_local_addr[DLM_MAX_ADDR_COUNT]; 1546ed7257bSPatrick Caulfield static int dlm_local_count; 155513ef596SDavid Teigland static int dlm_allow_conn; 1566ed7257bSPatrick Caulfield 1576ed7257bSPatrick Caulfield /* Work queues */ 1586ed7257bSPatrick Caulfield static struct workqueue_struct *recv_workqueue; 1596ed7257bSPatrick Caulfield static struct workqueue_struct *send_workqueue; 1606ed7257bSPatrick Caulfield 1615e9ccc37SChristine Caulfield static struct hlist_head connection_hash[CONN_HASH_SIZE]; 1627a936ce7SMatthias Kaehlcke static DEFINE_MUTEX(connections_lock); 1636ed7257bSPatrick Caulfield static struct kmem_cache *con_cache; 1646ed7257bSPatrick Caulfield 1656ed7257bSPatrick Caulfield static void process_recv_sockets(struct work_struct *work); 1666ed7257bSPatrick Caulfield static void process_send_sockets(struct work_struct *work); 1676ed7257bSPatrick Caulfield 1685e9ccc37SChristine Caulfield 1695e9ccc37SChristine Caulfield /* This is deliberately very simple because most clusters have simple 1705e9ccc37SChristine Caulfield sequential nodeids, so we should be able to go straight to a connection 1715e9ccc37SChristine Caulfield struct in the array */ 1725e9ccc37SChristine Caulfield static inline int nodeid_hash(int nodeid) 1735e9ccc37SChristine Caulfield { 1745e9ccc37SChristine Caulfield return nodeid & (CONN_HASH_SIZE-1); 1755e9ccc37SChristine Caulfield } 1765e9ccc37SChristine Caulfield 1775e9ccc37SChristine Caulfield static struct connection *__find_con(int nodeid) 1785e9ccc37SChristine Caulfield { 1795e9ccc37SChristine Caulfield int r; 1805e9ccc37SChristine Caulfield struct hlist_node *h; 1815e9ccc37SChristine Caulfield struct connection *con; 1825e9ccc37SChristine Caulfield 1835e9ccc37SChristine Caulfield r = nodeid_hash(nodeid); 1845e9ccc37SChristine Caulfield 1855e9ccc37SChristine Caulfield hlist_for_each_entry(con, h, &connection_hash[r], list) { 1865e9ccc37SChristine Caulfield if (con->nodeid == nodeid) 1875e9ccc37SChristine Caulfield return con; 1885e9ccc37SChristine Caulfield } 1895e9ccc37SChristine Caulfield return NULL; 1905e9ccc37SChristine Caulfield } 1915e9ccc37SChristine Caulfield 1926ed7257bSPatrick Caulfield /* 1936ed7257bSPatrick Caulfield * If 'allocation' is zero then we don't attempt to create a new 1946ed7257bSPatrick Caulfield * connection structure for this node. 1956ed7257bSPatrick Caulfield */ 1966ed7257bSPatrick Caulfield static struct connection *__nodeid2con(int nodeid, gfp_t alloc) 1976ed7257bSPatrick Caulfield { 1986ed7257bSPatrick Caulfield struct connection *con = NULL; 1996ed7257bSPatrick Caulfield int r; 2006ed7257bSPatrick Caulfield 2015e9ccc37SChristine Caulfield con = __find_con(nodeid); 2026ed7257bSPatrick Caulfield if (con || !alloc) 2036ed7257bSPatrick Caulfield return con; 2046ed7257bSPatrick Caulfield 2056ed7257bSPatrick Caulfield con = kmem_cache_zalloc(con_cache, alloc); 2066ed7257bSPatrick Caulfield if (!con) 2076ed7257bSPatrick Caulfield return NULL; 2086ed7257bSPatrick Caulfield 2095e9ccc37SChristine Caulfield r = nodeid_hash(nodeid); 2105e9ccc37SChristine Caulfield hlist_add_head(&con->list, &connection_hash[r]); 2116ed7257bSPatrick Caulfield 2126ed7257bSPatrick Caulfield con->nodeid = nodeid; 2136ed7257bSPatrick Caulfield mutex_init(&con->sock_mutex); 2146ed7257bSPatrick Caulfield INIT_LIST_HEAD(&con->writequeue); 2156ed7257bSPatrick Caulfield spin_lock_init(&con->writequeue_lock); 2166ed7257bSPatrick Caulfield INIT_WORK(&con->swork, process_send_sockets); 2176ed7257bSPatrick Caulfield INIT_WORK(&con->rwork, process_recv_sockets); 2186ed7257bSPatrick Caulfield 2196ed7257bSPatrick Caulfield /* Setup action pointers for child sockets */ 2206ed7257bSPatrick Caulfield if (con->nodeid) { 2215e9ccc37SChristine Caulfield struct connection *zerocon = __find_con(0); 2226ed7257bSPatrick Caulfield 2236ed7257bSPatrick Caulfield con->connect_action = zerocon->connect_action; 2246ed7257bSPatrick Caulfield if (!con->rx_action) 2256ed7257bSPatrick Caulfield con->rx_action = zerocon->rx_action; 2266ed7257bSPatrick Caulfield } 2276ed7257bSPatrick Caulfield 2286ed7257bSPatrick Caulfield return con; 2296ed7257bSPatrick Caulfield } 2306ed7257bSPatrick Caulfield 2315e9ccc37SChristine Caulfield /* Loop round all connections */ 2325e9ccc37SChristine Caulfield static void foreach_conn(void (*conn_func)(struct connection *c)) 2335e9ccc37SChristine Caulfield { 2345e9ccc37SChristine Caulfield int i; 2355e9ccc37SChristine Caulfield struct hlist_node *h, *n; 2365e9ccc37SChristine Caulfield struct connection *con; 2375e9ccc37SChristine Caulfield 2385e9ccc37SChristine Caulfield for (i = 0; i < CONN_HASH_SIZE; i++) { 2395e9ccc37SChristine Caulfield hlist_for_each_entry_safe(con, h, n, &connection_hash[i], list){ 2405e9ccc37SChristine Caulfield conn_func(con); 2415e9ccc37SChristine Caulfield } 2425e9ccc37SChristine Caulfield } 2435e9ccc37SChristine Caulfield } 2445e9ccc37SChristine Caulfield 2456ed7257bSPatrick Caulfield static struct connection *nodeid2con(int nodeid, gfp_t allocation) 2466ed7257bSPatrick Caulfield { 2476ed7257bSPatrick Caulfield struct connection *con; 2486ed7257bSPatrick Caulfield 2497a936ce7SMatthias Kaehlcke mutex_lock(&connections_lock); 2506ed7257bSPatrick Caulfield con = __nodeid2con(nodeid, allocation); 2517a936ce7SMatthias Kaehlcke mutex_unlock(&connections_lock); 2526ed7257bSPatrick Caulfield 2536ed7257bSPatrick Caulfield return con; 2546ed7257bSPatrick Caulfield } 2556ed7257bSPatrick Caulfield 2566ed7257bSPatrick Caulfield /* This is a bit drastic, but only called when things go wrong */ 2576ed7257bSPatrick Caulfield static struct connection *assoc2con(int assoc_id) 2586ed7257bSPatrick Caulfield { 2596ed7257bSPatrick Caulfield int i; 2605e9ccc37SChristine Caulfield struct hlist_node *h; 2616ed7257bSPatrick Caulfield struct connection *con; 2626ed7257bSPatrick Caulfield 2637a936ce7SMatthias Kaehlcke mutex_lock(&connections_lock); 2645e9ccc37SChristine Caulfield 2655e9ccc37SChristine Caulfield for (i = 0 ; i < CONN_HASH_SIZE; i++) { 2665e9ccc37SChristine Caulfield hlist_for_each_entry(con, h, &connection_hash[i], list) { 267f70cb33bSJulia Lawall if (con->sctp_assoc == assoc_id) { 2687a936ce7SMatthias Kaehlcke mutex_unlock(&connections_lock); 2696ed7257bSPatrick Caulfield return con; 2706ed7257bSPatrick Caulfield } 2716ed7257bSPatrick Caulfield } 2725e9ccc37SChristine Caulfield } 2737a936ce7SMatthias Kaehlcke mutex_unlock(&connections_lock); 2746ed7257bSPatrick Caulfield return NULL; 2756ed7257bSPatrick Caulfield } 2766ed7257bSPatrick Caulfield 27736b71a8bSDavid Teigland static struct dlm_node_addr *find_node_addr(int nodeid) 2786ed7257bSPatrick Caulfield { 27936b71a8bSDavid Teigland struct dlm_node_addr *na; 28036b71a8bSDavid Teigland 28136b71a8bSDavid Teigland list_for_each_entry(na, &dlm_node_addrs, list) { 28236b71a8bSDavid Teigland if (na->nodeid == nodeid) 28336b71a8bSDavid Teigland return na; 28436b71a8bSDavid Teigland } 28536b71a8bSDavid Teigland return NULL; 28636b71a8bSDavid Teigland } 28736b71a8bSDavid Teigland 28836b71a8bSDavid Teigland static int addr_compare(struct sockaddr_storage *x, struct sockaddr_storage *y) 28936b71a8bSDavid Teigland { 29036b71a8bSDavid Teigland switch (x->ss_family) { 29136b71a8bSDavid Teigland case AF_INET: { 29236b71a8bSDavid Teigland struct sockaddr_in *sinx = (struct sockaddr_in *)x; 29336b71a8bSDavid Teigland struct sockaddr_in *siny = (struct sockaddr_in *)y; 29436b71a8bSDavid Teigland if (sinx->sin_addr.s_addr != siny->sin_addr.s_addr) 29536b71a8bSDavid Teigland return 0; 29636b71a8bSDavid Teigland if (sinx->sin_port != siny->sin_port) 29736b71a8bSDavid Teigland return 0; 29836b71a8bSDavid Teigland break; 29936b71a8bSDavid Teigland } 30036b71a8bSDavid Teigland case AF_INET6: { 30136b71a8bSDavid Teigland struct sockaddr_in6 *sinx = (struct sockaddr_in6 *)x; 30236b71a8bSDavid Teigland struct sockaddr_in6 *siny = (struct sockaddr_in6 *)y; 30336b71a8bSDavid Teigland if (!ipv6_addr_equal(&sinx->sin6_addr, &siny->sin6_addr)) 30436b71a8bSDavid Teigland return 0; 30536b71a8bSDavid Teigland if (sinx->sin6_port != siny->sin6_port) 30636b71a8bSDavid Teigland return 0; 30736b71a8bSDavid Teigland break; 30836b71a8bSDavid Teigland } 30936b71a8bSDavid Teigland default: 31036b71a8bSDavid Teigland return 0; 31136b71a8bSDavid Teigland } 31236b71a8bSDavid Teigland return 1; 31336b71a8bSDavid Teigland } 31436b71a8bSDavid Teigland 31536b71a8bSDavid Teigland static int nodeid_to_addr(int nodeid, struct sockaddr_storage *sas_out, 31636b71a8bSDavid Teigland struct sockaddr *sa_out) 31736b71a8bSDavid Teigland { 31836b71a8bSDavid Teigland struct sockaddr_storage sas; 31936b71a8bSDavid Teigland struct dlm_node_addr *na; 3206ed7257bSPatrick Caulfield 3216ed7257bSPatrick Caulfield if (!dlm_local_count) 3226ed7257bSPatrick Caulfield return -1; 3236ed7257bSPatrick Caulfield 32436b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 32536b71a8bSDavid Teigland na = find_node_addr(nodeid); 32636b71a8bSDavid Teigland if (na && na->addr_count) 32736b71a8bSDavid Teigland memcpy(&sas, na->addr[0], sizeof(struct sockaddr_storage)); 32836b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 32936b71a8bSDavid Teigland 33036b71a8bSDavid Teigland if (!na) 33136b71a8bSDavid Teigland return -EEXIST; 33236b71a8bSDavid Teigland 33336b71a8bSDavid Teigland if (!na->addr_count) 33436b71a8bSDavid Teigland return -ENOENT; 33536b71a8bSDavid Teigland 33636b71a8bSDavid Teigland if (sas_out) 33736b71a8bSDavid Teigland memcpy(sas_out, &sas, sizeof(struct sockaddr_storage)); 33836b71a8bSDavid Teigland 33936b71a8bSDavid Teigland if (!sa_out) 34036b71a8bSDavid Teigland return 0; 3416ed7257bSPatrick Caulfield 3426ed7257bSPatrick Caulfield if (dlm_local_addr[0]->ss_family == AF_INET) { 34336b71a8bSDavid Teigland struct sockaddr_in *in4 = (struct sockaddr_in *) &sas; 34436b71a8bSDavid Teigland struct sockaddr_in *ret4 = (struct sockaddr_in *) sa_out; 3456ed7257bSPatrick Caulfield ret4->sin_addr.s_addr = in4->sin_addr.s_addr; 3466ed7257bSPatrick Caulfield } else { 34736b71a8bSDavid Teigland struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) &sas; 34836b71a8bSDavid Teigland struct sockaddr_in6 *ret6 = (struct sockaddr_in6 *) sa_out; 3494e3fd7a0SAlexey Dobriyan ret6->sin6_addr = in6->sin6_addr; 3506ed7257bSPatrick Caulfield } 3516ed7257bSPatrick Caulfield 3526ed7257bSPatrick Caulfield return 0; 3536ed7257bSPatrick Caulfield } 3546ed7257bSPatrick Caulfield 35536b71a8bSDavid Teigland static int addr_to_nodeid(struct sockaddr_storage *addr, int *nodeid) 35636b71a8bSDavid Teigland { 35736b71a8bSDavid Teigland struct dlm_node_addr *na; 35836b71a8bSDavid Teigland int rv = -EEXIST; 35936b71a8bSDavid Teigland 36036b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 36136b71a8bSDavid Teigland list_for_each_entry(na, &dlm_node_addrs, list) { 36236b71a8bSDavid Teigland if (!na->addr_count) 36336b71a8bSDavid Teigland continue; 36436b71a8bSDavid Teigland 36536b71a8bSDavid Teigland if (!addr_compare(na->addr[0], addr)) 36636b71a8bSDavid Teigland continue; 36736b71a8bSDavid Teigland 36836b71a8bSDavid Teigland *nodeid = na->nodeid; 36936b71a8bSDavid Teigland rv = 0; 37036b71a8bSDavid Teigland break; 37136b71a8bSDavid Teigland } 37236b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 37336b71a8bSDavid Teigland return rv; 37436b71a8bSDavid Teigland } 37536b71a8bSDavid Teigland 37636b71a8bSDavid Teigland int dlm_lowcomms_addr(int nodeid, struct sockaddr_storage *addr, int len) 37736b71a8bSDavid Teigland { 37836b71a8bSDavid Teigland struct sockaddr_storage *new_addr; 37936b71a8bSDavid Teigland struct dlm_node_addr *new_node, *na; 38036b71a8bSDavid Teigland 38136b71a8bSDavid Teigland new_node = kzalloc(sizeof(struct dlm_node_addr), GFP_NOFS); 38236b71a8bSDavid Teigland if (!new_node) 38336b71a8bSDavid Teigland return -ENOMEM; 38436b71a8bSDavid Teigland 38536b71a8bSDavid Teigland new_addr = kzalloc(sizeof(struct sockaddr_storage), GFP_NOFS); 38636b71a8bSDavid Teigland if (!new_addr) { 38736b71a8bSDavid Teigland kfree(new_node); 38836b71a8bSDavid Teigland return -ENOMEM; 38936b71a8bSDavid Teigland } 39036b71a8bSDavid Teigland 39136b71a8bSDavid Teigland memcpy(new_addr, addr, len); 39236b71a8bSDavid Teigland 39336b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 39436b71a8bSDavid Teigland na = find_node_addr(nodeid); 39536b71a8bSDavid Teigland if (!na) { 39636b71a8bSDavid Teigland new_node->nodeid = nodeid; 39736b71a8bSDavid Teigland new_node->addr[0] = new_addr; 39836b71a8bSDavid Teigland new_node->addr_count = 1; 39936b71a8bSDavid Teigland list_add(&new_node->list, &dlm_node_addrs); 40036b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 40136b71a8bSDavid Teigland return 0; 40236b71a8bSDavid Teigland } 40336b71a8bSDavid Teigland 40436b71a8bSDavid Teigland if (na->addr_count >= DLM_MAX_ADDR_COUNT) { 40536b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 40636b71a8bSDavid Teigland kfree(new_addr); 40736b71a8bSDavid Teigland kfree(new_node); 40836b71a8bSDavid Teigland return -ENOSPC; 40936b71a8bSDavid Teigland } 41036b71a8bSDavid Teigland 41136b71a8bSDavid Teigland na->addr[na->addr_count++] = new_addr; 41236b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 41336b71a8bSDavid Teigland kfree(new_node); 41436b71a8bSDavid Teigland return 0; 41536b71a8bSDavid Teigland } 41636b71a8bSDavid Teigland 4176ed7257bSPatrick Caulfield /* Data available on socket or listen socket received a connect */ 4186ed7257bSPatrick Caulfield static void lowcomms_data_ready(struct sock *sk, int count_unused) 4196ed7257bSPatrick Caulfield { 4206ed7257bSPatrick Caulfield struct connection *con = sock2con(sk); 421afb853fbSPatrick Caulfield if (con && !test_and_set_bit(CF_READ_PENDING, &con->flags)) 4226ed7257bSPatrick Caulfield queue_work(recv_workqueue, &con->rwork); 4236ed7257bSPatrick Caulfield } 4246ed7257bSPatrick Caulfield 4256ed7257bSPatrick Caulfield static void lowcomms_write_space(struct sock *sk) 4266ed7257bSPatrick Caulfield { 4276ed7257bSPatrick Caulfield struct connection *con = sock2con(sk); 4286ed7257bSPatrick Caulfield 429b36930ddSDavid Miller if (!con) 430b36930ddSDavid Miller return; 431b36930ddSDavid Miller 432b36930ddSDavid Miller clear_bit(SOCK_NOSPACE, &con->sock->flags); 433b36930ddSDavid Miller 434b36930ddSDavid Miller if (test_and_clear_bit(CF_APP_LIMITED, &con->flags)) { 435b36930ddSDavid Miller con->sock->sk->sk_write_pending--; 436b36930ddSDavid Miller clear_bit(SOCK_ASYNC_NOSPACE, &con->sock->flags); 437b36930ddSDavid Miller } 438b36930ddSDavid Miller 439b36930ddSDavid Miller if (!test_and_set_bit(CF_WRITE_PENDING, &con->flags)) 4406ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 4416ed7257bSPatrick Caulfield } 4426ed7257bSPatrick Caulfield 4436ed7257bSPatrick Caulfield static inline void lowcomms_connect_sock(struct connection *con) 4446ed7257bSPatrick Caulfield { 445063c4c99SLars Marowsky-Bree if (test_bit(CF_CLOSE, &con->flags)) 446063c4c99SLars Marowsky-Bree return; 4476ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_CONNECT_PENDING, &con->flags)) 4486ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 4496ed7257bSPatrick Caulfield } 4506ed7257bSPatrick Caulfield 4516ed7257bSPatrick Caulfield static void lowcomms_state_change(struct sock *sk) 4526ed7257bSPatrick Caulfield { 4536ed7257bSPatrick Caulfield if (sk->sk_state == TCP_ESTABLISHED) 4546ed7257bSPatrick Caulfield lowcomms_write_space(sk); 4556ed7257bSPatrick Caulfield } 4566ed7257bSPatrick Caulfield 457391fbdc5SChristine Caulfield int dlm_lowcomms_connect_node(int nodeid) 458391fbdc5SChristine Caulfield { 459391fbdc5SChristine Caulfield struct connection *con; 460391fbdc5SChristine Caulfield 46104bedd79SDavid Teigland /* with sctp there's no connecting without sending */ 46204bedd79SDavid Teigland if (dlm_config.ci_protocol != 0) 46304bedd79SDavid Teigland return 0; 46404bedd79SDavid Teigland 465391fbdc5SChristine Caulfield if (nodeid == dlm_our_nodeid()) 466391fbdc5SChristine Caulfield return 0; 467391fbdc5SChristine Caulfield 468391fbdc5SChristine Caulfield con = nodeid2con(nodeid, GFP_NOFS); 469391fbdc5SChristine Caulfield if (!con) 470391fbdc5SChristine Caulfield return -ENOMEM; 471391fbdc5SChristine Caulfield lowcomms_connect_sock(con); 472391fbdc5SChristine Caulfield return 0; 473391fbdc5SChristine Caulfield } 474391fbdc5SChristine Caulfield 4756ed7257bSPatrick Caulfield /* Make a socket active */ 4766ed7257bSPatrick Caulfield static int add_sock(struct socket *sock, struct connection *con) 4776ed7257bSPatrick Caulfield { 4786ed7257bSPatrick Caulfield con->sock = sock; 4796ed7257bSPatrick Caulfield 4806ed7257bSPatrick Caulfield /* Install a data_ready callback */ 4816ed7257bSPatrick Caulfield con->sock->sk->sk_data_ready = lowcomms_data_ready; 4826ed7257bSPatrick Caulfield con->sock->sk->sk_write_space = lowcomms_write_space; 4836ed7257bSPatrick Caulfield con->sock->sk->sk_state_change = lowcomms_state_change; 4846ed7257bSPatrick Caulfield con->sock->sk->sk_user_data = con; 485d6d7b702SSteven Whitehouse con->sock->sk->sk_allocation = GFP_NOFS; 4866ed7257bSPatrick Caulfield return 0; 4876ed7257bSPatrick Caulfield } 4886ed7257bSPatrick Caulfield 4896ed7257bSPatrick Caulfield /* Add the port number to an IPv6 or 4 sockaddr and return the address 4906ed7257bSPatrick Caulfield length */ 4916ed7257bSPatrick Caulfield static void make_sockaddr(struct sockaddr_storage *saddr, uint16_t port, 4926ed7257bSPatrick Caulfield int *addr_len) 4936ed7257bSPatrick Caulfield { 4946ed7257bSPatrick Caulfield saddr->ss_family = dlm_local_addr[0]->ss_family; 4956ed7257bSPatrick Caulfield if (saddr->ss_family == AF_INET) { 4966ed7257bSPatrick Caulfield struct sockaddr_in *in4_addr = (struct sockaddr_in *)saddr; 4976ed7257bSPatrick Caulfield in4_addr->sin_port = cpu_to_be16(port); 4986ed7257bSPatrick Caulfield *addr_len = sizeof(struct sockaddr_in); 4996ed7257bSPatrick Caulfield memset(&in4_addr->sin_zero, 0, sizeof(in4_addr->sin_zero)); 5006ed7257bSPatrick Caulfield } else { 5016ed7257bSPatrick Caulfield struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)saddr; 5026ed7257bSPatrick Caulfield in6_addr->sin6_port = cpu_to_be16(port); 5036ed7257bSPatrick Caulfield *addr_len = sizeof(struct sockaddr_in6); 5046ed7257bSPatrick Caulfield } 50501c8cab2SPatrick Caulfield memset((char *)saddr + *addr_len, 0, sizeof(struct sockaddr_storage) - *addr_len); 5066ed7257bSPatrick Caulfield } 5076ed7257bSPatrick Caulfield 5086ed7257bSPatrick Caulfield /* Close a remote connection and tidy up */ 5096ed7257bSPatrick Caulfield static void close_connection(struct connection *con, bool and_other) 5106ed7257bSPatrick Caulfield { 5116ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 5126ed7257bSPatrick Caulfield 5136ed7257bSPatrick Caulfield if (con->sock) { 5146ed7257bSPatrick Caulfield sock_release(con->sock); 5156ed7257bSPatrick Caulfield con->sock = NULL; 5166ed7257bSPatrick Caulfield } 5176ed7257bSPatrick Caulfield if (con->othercon && and_other) { 5186ed7257bSPatrick Caulfield /* Will only re-enter once. */ 5196ed7257bSPatrick Caulfield close_connection(con->othercon, false); 5206ed7257bSPatrick Caulfield } 5216ed7257bSPatrick Caulfield if (con->rx_page) { 5226ed7257bSPatrick Caulfield __free_page(con->rx_page); 5236ed7257bSPatrick Caulfield con->rx_page = NULL; 5246ed7257bSPatrick Caulfield } 5259e5f2825SPatrick Caulfield 5266ed7257bSPatrick Caulfield con->retries = 0; 5276ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 5286ed7257bSPatrick Caulfield } 5296ed7257bSPatrick Caulfield 5306ed7257bSPatrick Caulfield /* We only send shutdown messages to nodes that are not part of the cluster */ 5316ed7257bSPatrick Caulfield static void sctp_send_shutdown(sctp_assoc_t associd) 5326ed7257bSPatrick Caulfield { 5336ed7257bSPatrick Caulfield static char outcmsg[CMSG_SPACE(sizeof(struct sctp_sndrcvinfo))]; 5346ed7257bSPatrick Caulfield struct msghdr outmessage; 5356ed7257bSPatrick Caulfield struct cmsghdr *cmsg; 5366ed7257bSPatrick Caulfield struct sctp_sndrcvinfo *sinfo; 5376ed7257bSPatrick Caulfield int ret; 5386ed7257bSPatrick Caulfield struct connection *con; 5396ed7257bSPatrick Caulfield 5406ed7257bSPatrick Caulfield con = nodeid2con(0,0); 5416ed7257bSPatrick Caulfield BUG_ON(con == NULL); 5426ed7257bSPatrick Caulfield 5436ed7257bSPatrick Caulfield outmessage.msg_name = NULL; 5446ed7257bSPatrick Caulfield outmessage.msg_namelen = 0; 5456ed7257bSPatrick Caulfield outmessage.msg_control = outcmsg; 5466ed7257bSPatrick Caulfield outmessage.msg_controllen = sizeof(outcmsg); 5476ed7257bSPatrick Caulfield outmessage.msg_flags = MSG_EOR; 5486ed7257bSPatrick Caulfield 5496ed7257bSPatrick Caulfield cmsg = CMSG_FIRSTHDR(&outmessage); 5506ed7257bSPatrick Caulfield cmsg->cmsg_level = IPPROTO_SCTP; 5516ed7257bSPatrick Caulfield cmsg->cmsg_type = SCTP_SNDRCV; 5526ed7257bSPatrick Caulfield cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo)); 5536ed7257bSPatrick Caulfield outmessage.msg_controllen = cmsg->cmsg_len; 5546ed7257bSPatrick Caulfield sinfo = CMSG_DATA(cmsg); 5556ed7257bSPatrick Caulfield memset(sinfo, 0x00, sizeof(struct sctp_sndrcvinfo)); 5566ed7257bSPatrick Caulfield 5576ed7257bSPatrick Caulfield sinfo->sinfo_flags |= MSG_EOF; 5586ed7257bSPatrick Caulfield sinfo->sinfo_assoc_id = associd; 5596ed7257bSPatrick Caulfield 5606ed7257bSPatrick Caulfield ret = kernel_sendmsg(con->sock, &outmessage, NULL, 0, 0); 5616ed7257bSPatrick Caulfield 5626ed7257bSPatrick Caulfield if (ret != 0) 5636ed7257bSPatrick Caulfield log_print("send EOF to node failed: %d", ret); 5646ed7257bSPatrick Caulfield } 5656ed7257bSPatrick Caulfield 5665e9ccc37SChristine Caulfield static void sctp_init_failed_foreach(struct connection *con) 5675e9ccc37SChristine Caulfield { 5685e9ccc37SChristine Caulfield con->sctp_assoc = 0; 5695e9ccc37SChristine Caulfield if (test_and_clear_bit(CF_CONNECT_PENDING, &con->flags)) { 5705e9ccc37SChristine Caulfield if (!test_and_set_bit(CF_WRITE_PENDING, &con->flags)) 5715e9ccc37SChristine Caulfield queue_work(send_workqueue, &con->swork); 5725e9ccc37SChristine Caulfield } 5735e9ccc37SChristine Caulfield } 5745e9ccc37SChristine Caulfield 5756ed7257bSPatrick Caulfield /* INIT failed but we don't know which node... 5766ed7257bSPatrick Caulfield restart INIT on all pending nodes */ 5776ed7257bSPatrick Caulfield static void sctp_init_failed(void) 5786ed7257bSPatrick Caulfield { 5797a936ce7SMatthias Kaehlcke mutex_lock(&connections_lock); 5805e9ccc37SChristine Caulfield 5815e9ccc37SChristine Caulfield foreach_conn(sctp_init_failed_foreach); 5825e9ccc37SChristine Caulfield 5837a936ce7SMatthias Kaehlcke mutex_unlock(&connections_lock); 5846ed7257bSPatrick Caulfield } 5856ed7257bSPatrick Caulfield 5866ed7257bSPatrick Caulfield /* Something happened to an association */ 587617e82e1SDavid Teigland static void process_sctp_notification(struct connection *con, 588617e82e1SDavid Teigland struct msghdr *msg, char *buf) 5896ed7257bSPatrick Caulfield { 5906ed7257bSPatrick Caulfield union sctp_notification *sn = (union sctp_notification *)buf; 5916ed7257bSPatrick Caulfield 5926ed7257bSPatrick Caulfield if (sn->sn_header.sn_type == SCTP_ASSOC_CHANGE) { 5936ed7257bSPatrick Caulfield switch (sn->sn_assoc_change.sac_state) { 5946ed7257bSPatrick Caulfield 5956ed7257bSPatrick Caulfield case SCTP_COMM_UP: 5966ed7257bSPatrick Caulfield case SCTP_RESTART: 5976ed7257bSPatrick Caulfield { 5986ed7257bSPatrick Caulfield /* Check that the new node is in the lockspace */ 5996ed7257bSPatrick Caulfield struct sctp_prim prim; 6006ed7257bSPatrick Caulfield int nodeid; 6016ed7257bSPatrick Caulfield int prim_len, ret; 6026ed7257bSPatrick Caulfield int addr_len; 6036ed7257bSPatrick Caulfield struct connection *new_con; 6046ed7257bSPatrick Caulfield 6056ed7257bSPatrick Caulfield /* 6066ed7257bSPatrick Caulfield * We get this before any data for an association. 6076ed7257bSPatrick Caulfield * We verify that the node is in the cluster and 6086ed7257bSPatrick Caulfield * then peel off a socket for it. 6096ed7257bSPatrick Caulfield */ 6106ed7257bSPatrick Caulfield if ((int)sn->sn_assoc_change.sac_assoc_id <= 0) { 6116ed7257bSPatrick Caulfield log_print("COMM_UP for invalid assoc ID %d", 6126ed7257bSPatrick Caulfield (int)sn->sn_assoc_change.sac_assoc_id); 6136ed7257bSPatrick Caulfield sctp_init_failed(); 6146ed7257bSPatrick Caulfield return; 6156ed7257bSPatrick Caulfield } 6166ed7257bSPatrick Caulfield memset(&prim, 0, sizeof(struct sctp_prim)); 6176ed7257bSPatrick Caulfield prim_len = sizeof(struct sctp_prim); 6186ed7257bSPatrick Caulfield prim.ssp_assoc_id = sn->sn_assoc_change.sac_assoc_id; 6196ed7257bSPatrick Caulfield 6206ed7257bSPatrick Caulfield ret = kernel_getsockopt(con->sock, 6216ed7257bSPatrick Caulfield IPPROTO_SCTP, 6226ed7257bSPatrick Caulfield SCTP_PRIMARY_ADDR, 6236ed7257bSPatrick Caulfield (char*)&prim, 6246ed7257bSPatrick Caulfield &prim_len); 6256ed7257bSPatrick Caulfield if (ret < 0) { 6266ed7257bSPatrick Caulfield log_print("getsockopt/sctp_primary_addr on " 6276ed7257bSPatrick Caulfield "new assoc %d failed : %d", 6286ed7257bSPatrick Caulfield (int)sn->sn_assoc_change.sac_assoc_id, 6296ed7257bSPatrick Caulfield ret); 6306ed7257bSPatrick Caulfield 6316ed7257bSPatrick Caulfield /* Retry INIT later */ 6326ed7257bSPatrick Caulfield new_con = assoc2con(sn->sn_assoc_change.sac_assoc_id); 6336ed7257bSPatrick Caulfield if (new_con) 6346ed7257bSPatrick Caulfield clear_bit(CF_CONNECT_PENDING, &con->flags); 6356ed7257bSPatrick Caulfield return; 6366ed7257bSPatrick Caulfield } 6376ed7257bSPatrick Caulfield make_sockaddr(&prim.ssp_addr, 0, &addr_len); 63836b71a8bSDavid Teigland if (addr_to_nodeid(&prim.ssp_addr, &nodeid)) { 6396ed7257bSPatrick Caulfield unsigned char *b=(unsigned char *)&prim.ssp_addr; 6406ed7257bSPatrick Caulfield log_print("reject connect from unknown addr"); 641bcaadf5cSMasatake YAMATO print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE, 642bcaadf5cSMasatake YAMATO b, sizeof(struct sockaddr_storage)); 6436ed7257bSPatrick Caulfield sctp_send_shutdown(prim.ssp_assoc_id); 6446ed7257bSPatrick Caulfield return; 6456ed7257bSPatrick Caulfield } 6466ed7257bSPatrick Caulfield 647748285ccSDavid Teigland new_con = nodeid2con(nodeid, GFP_NOFS); 6486ed7257bSPatrick Caulfield if (!new_con) 6496ed7257bSPatrick Caulfield return; 6506ed7257bSPatrick Caulfield 6516ed7257bSPatrick Caulfield /* Peel off a new sock */ 6522f2d76ccSBenjamin Poirier sctp_lock_sock(con->sock->sk); 6532f2d76ccSBenjamin Poirier ret = sctp_do_peeloff(con->sock->sk, 6542f2d76ccSBenjamin Poirier sn->sn_assoc_change.sac_assoc_id, 6552f2d76ccSBenjamin Poirier &new_con->sock); 6562f2d76ccSBenjamin Poirier sctp_release_sock(con->sock->sk); 6576861f350SDavid Teigland if (ret < 0) { 658617e82e1SDavid Teigland log_print("Can't peel off a socket for " 6596861f350SDavid Teigland "connection %d to node %d: err=%d", 6602f2d76ccSBenjamin Poirier (int)sn->sn_assoc_change.sac_assoc_id, 6612f2d76ccSBenjamin Poirier nodeid, ret); 6626861f350SDavid Teigland return; 6636861f350SDavid Teigland } 6646ed7257bSPatrick Caulfield add_sock(new_con->sock, new_con); 6656ed7257bSPatrick Caulfield 6666861f350SDavid Teigland log_print("connecting to %d sctp association %d", 6676861f350SDavid Teigland nodeid, (int)sn->sn_assoc_change.sac_assoc_id); 6686ed7257bSPatrick Caulfield 6696ed7257bSPatrick Caulfield /* Send any pending writes */ 6706ed7257bSPatrick Caulfield clear_bit(CF_CONNECT_PENDING, &new_con->flags); 6716ed7257bSPatrick Caulfield clear_bit(CF_INIT_PENDING, &con->flags); 6726ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_WRITE_PENDING, &new_con->flags)) { 6736ed7257bSPatrick Caulfield queue_work(send_workqueue, &new_con->swork); 6746ed7257bSPatrick Caulfield } 6756ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_READ_PENDING, &new_con->flags)) 6766ed7257bSPatrick Caulfield queue_work(recv_workqueue, &new_con->rwork); 6776ed7257bSPatrick Caulfield } 6786ed7257bSPatrick Caulfield break; 6796ed7257bSPatrick Caulfield 6806ed7257bSPatrick Caulfield case SCTP_COMM_LOST: 6816ed7257bSPatrick Caulfield case SCTP_SHUTDOWN_COMP: 6826ed7257bSPatrick Caulfield { 6836ed7257bSPatrick Caulfield con = assoc2con(sn->sn_assoc_change.sac_assoc_id); 6846ed7257bSPatrick Caulfield if (con) { 6856ed7257bSPatrick Caulfield con->sctp_assoc = 0; 6866ed7257bSPatrick Caulfield } 6876ed7257bSPatrick Caulfield } 6886ed7257bSPatrick Caulfield break; 6896ed7257bSPatrick Caulfield 6906ed7257bSPatrick Caulfield /* We don't know which INIT failed, so clear the PENDING flags 6916ed7257bSPatrick Caulfield * on them all. if assoc_id is zero then it will then try 6926ed7257bSPatrick Caulfield * again */ 6936ed7257bSPatrick Caulfield 6946ed7257bSPatrick Caulfield case SCTP_CANT_STR_ASSOC: 6956ed7257bSPatrick Caulfield { 6966ed7257bSPatrick Caulfield log_print("Can't start SCTP association - retrying"); 6976ed7257bSPatrick Caulfield sctp_init_failed(); 6986ed7257bSPatrick Caulfield } 6996ed7257bSPatrick Caulfield break; 7006ed7257bSPatrick Caulfield 7016ed7257bSPatrick Caulfield default: 7026ed7257bSPatrick Caulfield log_print("unexpected SCTP assoc change id=%d state=%d", 7036ed7257bSPatrick Caulfield (int)sn->sn_assoc_change.sac_assoc_id, 7046ed7257bSPatrick Caulfield sn->sn_assoc_change.sac_state); 7056ed7257bSPatrick Caulfield } 7066ed7257bSPatrick Caulfield } 7076ed7257bSPatrick Caulfield } 7086ed7257bSPatrick Caulfield 7096ed7257bSPatrick Caulfield /* Data received from remote end */ 7106ed7257bSPatrick Caulfield static int receive_from_sock(struct connection *con) 7116ed7257bSPatrick Caulfield { 7126ed7257bSPatrick Caulfield int ret = 0; 7136ed7257bSPatrick Caulfield struct msghdr msg = {}; 7146ed7257bSPatrick Caulfield struct kvec iov[2]; 7156ed7257bSPatrick Caulfield unsigned len; 7166ed7257bSPatrick Caulfield int r; 7176ed7257bSPatrick Caulfield int call_again_soon = 0; 7186ed7257bSPatrick Caulfield int nvec; 7196ed7257bSPatrick Caulfield char incmsg[CMSG_SPACE(sizeof(struct sctp_sndrcvinfo))]; 7206ed7257bSPatrick Caulfield 7216ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 7226ed7257bSPatrick Caulfield 7236ed7257bSPatrick Caulfield if (con->sock == NULL) { 7246ed7257bSPatrick Caulfield ret = -EAGAIN; 7256ed7257bSPatrick Caulfield goto out_close; 7266ed7257bSPatrick Caulfield } 7276ed7257bSPatrick Caulfield 7286ed7257bSPatrick Caulfield if (con->rx_page == NULL) { 7296ed7257bSPatrick Caulfield /* 7306ed7257bSPatrick Caulfield * This doesn't need to be atomic, but I think it should 7316ed7257bSPatrick Caulfield * improve performance if it is. 7326ed7257bSPatrick Caulfield */ 7336ed7257bSPatrick Caulfield con->rx_page = alloc_page(GFP_ATOMIC); 7346ed7257bSPatrick Caulfield if (con->rx_page == NULL) 7356ed7257bSPatrick Caulfield goto out_resched; 7366ed7257bSPatrick Caulfield cbuf_init(&con->cb, PAGE_CACHE_SIZE); 7376ed7257bSPatrick Caulfield } 7386ed7257bSPatrick Caulfield 7396ed7257bSPatrick Caulfield /* Only SCTP needs these really */ 7406ed7257bSPatrick Caulfield memset(&incmsg, 0, sizeof(incmsg)); 7416ed7257bSPatrick Caulfield msg.msg_control = incmsg; 7426ed7257bSPatrick Caulfield msg.msg_controllen = sizeof(incmsg); 7436ed7257bSPatrick Caulfield 7446ed7257bSPatrick Caulfield /* 7456ed7257bSPatrick Caulfield * iov[0] is the bit of the circular buffer between the current end 7466ed7257bSPatrick Caulfield * point (cb.base + cb.len) and the end of the buffer. 7476ed7257bSPatrick Caulfield */ 7486ed7257bSPatrick Caulfield iov[0].iov_len = con->cb.base - cbuf_data(&con->cb); 7496ed7257bSPatrick Caulfield iov[0].iov_base = page_address(con->rx_page) + cbuf_data(&con->cb); 7506ed7257bSPatrick Caulfield iov[1].iov_len = 0; 7516ed7257bSPatrick Caulfield nvec = 1; 7526ed7257bSPatrick Caulfield 7536ed7257bSPatrick Caulfield /* 7546ed7257bSPatrick Caulfield * iov[1] is the bit of the circular buffer between the start of the 7556ed7257bSPatrick Caulfield * buffer and the start of the currently used section (cb.base) 7566ed7257bSPatrick Caulfield */ 7576ed7257bSPatrick Caulfield if (cbuf_data(&con->cb) >= con->cb.base) { 7586ed7257bSPatrick Caulfield iov[0].iov_len = PAGE_CACHE_SIZE - cbuf_data(&con->cb); 7596ed7257bSPatrick Caulfield iov[1].iov_len = con->cb.base; 7606ed7257bSPatrick Caulfield iov[1].iov_base = page_address(con->rx_page); 7616ed7257bSPatrick Caulfield nvec = 2; 7626ed7257bSPatrick Caulfield } 7636ed7257bSPatrick Caulfield len = iov[0].iov_len + iov[1].iov_len; 7646ed7257bSPatrick Caulfield 7656ed7257bSPatrick Caulfield r = ret = kernel_recvmsg(con->sock, &msg, iov, nvec, len, 7666ed7257bSPatrick Caulfield MSG_DONTWAIT | MSG_NOSIGNAL); 7676ed7257bSPatrick Caulfield if (ret <= 0) 7686ed7257bSPatrick Caulfield goto out_close; 7696ed7257bSPatrick Caulfield 7706ed7257bSPatrick Caulfield /* Process SCTP notifications */ 7716ed7257bSPatrick Caulfield if (msg.msg_flags & MSG_NOTIFICATION) { 7726ed7257bSPatrick Caulfield msg.msg_control = incmsg; 7736ed7257bSPatrick Caulfield msg.msg_controllen = sizeof(incmsg); 7746ed7257bSPatrick Caulfield 7756ed7257bSPatrick Caulfield process_sctp_notification(con, &msg, 7766ed7257bSPatrick Caulfield page_address(con->rx_page) + con->cb.base); 7776ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 7786ed7257bSPatrick Caulfield return 0; 7796ed7257bSPatrick Caulfield } 7806ed7257bSPatrick Caulfield BUG_ON(con->nodeid == 0); 7816ed7257bSPatrick Caulfield 7826ed7257bSPatrick Caulfield if (ret == len) 7836ed7257bSPatrick Caulfield call_again_soon = 1; 7846ed7257bSPatrick Caulfield cbuf_add(&con->cb, ret); 7856ed7257bSPatrick Caulfield ret = dlm_process_incoming_buffer(con->nodeid, 7866ed7257bSPatrick Caulfield page_address(con->rx_page), 7876ed7257bSPatrick Caulfield con->cb.base, con->cb.len, 7886ed7257bSPatrick Caulfield PAGE_CACHE_SIZE); 7896ed7257bSPatrick Caulfield if (ret == -EBADMSG) { 790617e82e1SDavid Teigland log_print("lowcomms: addr=%p, base=%u, len=%u, " 791617e82e1SDavid Teigland "iov_len=%u, iov_base[0]=%p, read=%d", 7926ed7257bSPatrick Caulfield page_address(con->rx_page), con->cb.base, con->cb.len, 7936ed7257bSPatrick Caulfield len, iov[0].iov_base, r); 7946ed7257bSPatrick Caulfield } 7956ed7257bSPatrick Caulfield if (ret < 0) 7966ed7257bSPatrick Caulfield goto out_close; 7976ed7257bSPatrick Caulfield cbuf_eat(&con->cb, ret); 7986ed7257bSPatrick Caulfield 7996ed7257bSPatrick Caulfield if (cbuf_empty(&con->cb) && !call_again_soon) { 8006ed7257bSPatrick Caulfield __free_page(con->rx_page); 8016ed7257bSPatrick Caulfield con->rx_page = NULL; 8026ed7257bSPatrick Caulfield } 8036ed7257bSPatrick Caulfield 8046ed7257bSPatrick Caulfield if (call_again_soon) 8056ed7257bSPatrick Caulfield goto out_resched; 8066ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 8076ed7257bSPatrick Caulfield return 0; 8086ed7257bSPatrick Caulfield 8096ed7257bSPatrick Caulfield out_resched: 8106ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_READ_PENDING, &con->flags)) 8116ed7257bSPatrick Caulfield queue_work(recv_workqueue, &con->rwork); 8126ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 8136ed7257bSPatrick Caulfield return -EAGAIN; 8146ed7257bSPatrick Caulfield 8156ed7257bSPatrick Caulfield out_close: 8166ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 8179e5f2825SPatrick Caulfield if (ret != -EAGAIN) { 8186ed7257bSPatrick Caulfield close_connection(con, false); 8196ed7257bSPatrick Caulfield /* Reconnect when there is something to send */ 8206ed7257bSPatrick Caulfield } 8216ed7257bSPatrick Caulfield /* Don't return success if we really got EOF */ 8226ed7257bSPatrick Caulfield if (ret == 0) 8236ed7257bSPatrick Caulfield ret = -EAGAIN; 8246ed7257bSPatrick Caulfield 8256ed7257bSPatrick Caulfield return ret; 8266ed7257bSPatrick Caulfield } 8276ed7257bSPatrick Caulfield 8286ed7257bSPatrick Caulfield /* Listening socket is busy, accept a connection */ 8296ed7257bSPatrick Caulfield static int tcp_accept_from_sock(struct connection *con) 8306ed7257bSPatrick Caulfield { 8316ed7257bSPatrick Caulfield int result; 8326ed7257bSPatrick Caulfield struct sockaddr_storage peeraddr; 8336ed7257bSPatrick Caulfield struct socket *newsock; 8346ed7257bSPatrick Caulfield int len; 8356ed7257bSPatrick Caulfield int nodeid; 8366ed7257bSPatrick Caulfield struct connection *newcon; 8376ed7257bSPatrick Caulfield struct connection *addcon; 8386ed7257bSPatrick Caulfield 839513ef596SDavid Teigland mutex_lock(&connections_lock); 840513ef596SDavid Teigland if (!dlm_allow_conn) { 841513ef596SDavid Teigland mutex_unlock(&connections_lock); 842513ef596SDavid Teigland return -1; 843513ef596SDavid Teigland } 844513ef596SDavid Teigland mutex_unlock(&connections_lock); 845513ef596SDavid Teigland 8466ed7257bSPatrick Caulfield memset(&peeraddr, 0, sizeof(peeraddr)); 8476ed7257bSPatrick Caulfield result = sock_create_kern(dlm_local_addr[0]->ss_family, SOCK_STREAM, 8486ed7257bSPatrick Caulfield IPPROTO_TCP, &newsock); 8496ed7257bSPatrick Caulfield if (result < 0) 8506ed7257bSPatrick Caulfield return -ENOMEM; 8516ed7257bSPatrick Caulfield 8526ed7257bSPatrick Caulfield mutex_lock_nested(&con->sock_mutex, 0); 8536ed7257bSPatrick Caulfield 8546ed7257bSPatrick Caulfield result = -ENOTCONN; 8556ed7257bSPatrick Caulfield if (con->sock == NULL) 8566ed7257bSPatrick Caulfield goto accept_err; 8576ed7257bSPatrick Caulfield 8586ed7257bSPatrick Caulfield newsock->type = con->sock->type; 8596ed7257bSPatrick Caulfield newsock->ops = con->sock->ops; 8606ed7257bSPatrick Caulfield 8616ed7257bSPatrick Caulfield result = con->sock->ops->accept(con->sock, newsock, O_NONBLOCK); 8626ed7257bSPatrick Caulfield if (result < 0) 8636ed7257bSPatrick Caulfield goto accept_err; 8646ed7257bSPatrick Caulfield 8656ed7257bSPatrick Caulfield /* Get the connected socket's peer */ 8666ed7257bSPatrick Caulfield memset(&peeraddr, 0, sizeof(peeraddr)); 8676ed7257bSPatrick Caulfield if (newsock->ops->getname(newsock, (struct sockaddr *)&peeraddr, 8686ed7257bSPatrick Caulfield &len, 2)) { 8696ed7257bSPatrick Caulfield result = -ECONNABORTED; 8706ed7257bSPatrick Caulfield goto accept_err; 8716ed7257bSPatrick Caulfield } 8726ed7257bSPatrick Caulfield 8736ed7257bSPatrick Caulfield /* Get the new node's NODEID */ 8746ed7257bSPatrick Caulfield make_sockaddr(&peeraddr, 0, &len); 87536b71a8bSDavid Teigland if (addr_to_nodeid(&peeraddr, &nodeid)) { 876bcaadf5cSMasatake YAMATO unsigned char *b=(unsigned char *)&peeraddr; 877617e82e1SDavid Teigland log_print("connect from non cluster node"); 878bcaadf5cSMasatake YAMATO print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE, 879bcaadf5cSMasatake YAMATO b, sizeof(struct sockaddr_storage)); 8806ed7257bSPatrick Caulfield sock_release(newsock); 8816ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 8826ed7257bSPatrick Caulfield return -1; 8836ed7257bSPatrick Caulfield } 8846ed7257bSPatrick Caulfield 8856ed7257bSPatrick Caulfield log_print("got connection from %d", nodeid); 8866ed7257bSPatrick Caulfield 8876ed7257bSPatrick Caulfield /* Check to see if we already have a connection to this node. This 8886ed7257bSPatrick Caulfield * could happen if the two nodes initiate a connection at roughly 8896ed7257bSPatrick Caulfield * the same time and the connections cross on the wire. 8906ed7257bSPatrick Caulfield * In this case we store the incoming one in "othercon" 8916ed7257bSPatrick Caulfield */ 892748285ccSDavid Teigland newcon = nodeid2con(nodeid, GFP_NOFS); 8936ed7257bSPatrick Caulfield if (!newcon) { 8946ed7257bSPatrick Caulfield result = -ENOMEM; 8956ed7257bSPatrick Caulfield goto accept_err; 8966ed7257bSPatrick Caulfield } 8976ed7257bSPatrick Caulfield mutex_lock_nested(&newcon->sock_mutex, 1); 8986ed7257bSPatrick Caulfield if (newcon->sock) { 8996ed7257bSPatrick Caulfield struct connection *othercon = newcon->othercon; 9006ed7257bSPatrick Caulfield 9016ed7257bSPatrick Caulfield if (!othercon) { 902748285ccSDavid Teigland othercon = kmem_cache_zalloc(con_cache, GFP_NOFS); 9036ed7257bSPatrick Caulfield if (!othercon) { 904617e82e1SDavid Teigland log_print("failed to allocate incoming socket"); 9056ed7257bSPatrick Caulfield mutex_unlock(&newcon->sock_mutex); 9066ed7257bSPatrick Caulfield result = -ENOMEM; 9076ed7257bSPatrick Caulfield goto accept_err; 9086ed7257bSPatrick Caulfield } 9096ed7257bSPatrick Caulfield othercon->nodeid = nodeid; 9106ed7257bSPatrick Caulfield othercon->rx_action = receive_from_sock; 9116ed7257bSPatrick Caulfield mutex_init(&othercon->sock_mutex); 9126ed7257bSPatrick Caulfield INIT_WORK(&othercon->swork, process_send_sockets); 9136ed7257bSPatrick Caulfield INIT_WORK(&othercon->rwork, process_recv_sockets); 9146ed7257bSPatrick Caulfield set_bit(CF_IS_OTHERCON, &othercon->flags); 91561d96be0SPatrick Caulfield } 91661d96be0SPatrick Caulfield if (!othercon->sock) { 9176ed7257bSPatrick Caulfield newcon->othercon = othercon; 9186ed7257bSPatrick Caulfield othercon->sock = newsock; 9196ed7257bSPatrick Caulfield newsock->sk->sk_user_data = othercon; 9206ed7257bSPatrick Caulfield add_sock(newsock, othercon); 9216ed7257bSPatrick Caulfield addcon = othercon; 9226ed7257bSPatrick Caulfield } 9236ed7257bSPatrick Caulfield else { 92497d84836SPatrick Caulfield printk("Extra connection from node %d attempted\n", nodeid); 92597d84836SPatrick Caulfield result = -EAGAIN; 926f4fadb23Sakpm@linux-foundation.org mutex_unlock(&newcon->sock_mutex); 92797d84836SPatrick Caulfield goto accept_err; 92897d84836SPatrick Caulfield } 92997d84836SPatrick Caulfield } 93097d84836SPatrick Caulfield else { 9316ed7257bSPatrick Caulfield newsock->sk->sk_user_data = newcon; 9326ed7257bSPatrick Caulfield newcon->rx_action = receive_from_sock; 9336ed7257bSPatrick Caulfield add_sock(newsock, newcon); 9346ed7257bSPatrick Caulfield addcon = newcon; 9356ed7257bSPatrick Caulfield } 9366ed7257bSPatrick Caulfield 9376ed7257bSPatrick Caulfield mutex_unlock(&newcon->sock_mutex); 9386ed7257bSPatrick Caulfield 9396ed7257bSPatrick Caulfield /* 9406ed7257bSPatrick Caulfield * Add it to the active queue in case we got data 94125985edcSLucas De Marchi * between processing the accept adding the socket 9426ed7257bSPatrick Caulfield * to the read_sockets list 9436ed7257bSPatrick Caulfield */ 9446ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags)) 9456ed7257bSPatrick Caulfield queue_work(recv_workqueue, &addcon->rwork); 9466ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 9476ed7257bSPatrick Caulfield 9486ed7257bSPatrick Caulfield return 0; 9496ed7257bSPatrick Caulfield 9506ed7257bSPatrick Caulfield accept_err: 9516ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 9526ed7257bSPatrick Caulfield sock_release(newsock); 9536ed7257bSPatrick Caulfield 9546ed7257bSPatrick Caulfield if (result != -EAGAIN) 955617e82e1SDavid Teigland log_print("error accepting connection from node: %d", result); 9566ed7257bSPatrick Caulfield return result; 9576ed7257bSPatrick Caulfield } 9586ed7257bSPatrick Caulfield 9596ed7257bSPatrick Caulfield static void free_entry(struct writequeue_entry *e) 9606ed7257bSPatrick Caulfield { 9616ed7257bSPatrick Caulfield __free_page(e->page); 9626ed7257bSPatrick Caulfield kfree(e); 9636ed7257bSPatrick Caulfield } 9646ed7257bSPatrick Caulfield 9656ed7257bSPatrick Caulfield /* Initiate an SCTP association. 9666ed7257bSPatrick Caulfield This is a special case of send_to_sock() in that we don't yet have a 9676ed7257bSPatrick Caulfield peeled-off socket for this association, so we use the listening socket 9686ed7257bSPatrick Caulfield and add the primary IP address of the remote node. 9696ed7257bSPatrick Caulfield */ 9706ed7257bSPatrick Caulfield static void sctp_init_assoc(struct connection *con) 9716ed7257bSPatrick Caulfield { 9726ed7257bSPatrick Caulfield struct sockaddr_storage rem_addr; 9736ed7257bSPatrick Caulfield char outcmsg[CMSG_SPACE(sizeof(struct sctp_sndrcvinfo))]; 9746ed7257bSPatrick Caulfield struct msghdr outmessage; 9756ed7257bSPatrick Caulfield struct cmsghdr *cmsg; 9766ed7257bSPatrick Caulfield struct sctp_sndrcvinfo *sinfo; 9776ed7257bSPatrick Caulfield struct connection *base_con; 9786ed7257bSPatrick Caulfield struct writequeue_entry *e; 9796ed7257bSPatrick Caulfield int len, offset; 9806ed7257bSPatrick Caulfield int ret; 9816ed7257bSPatrick Caulfield int addrlen; 9826ed7257bSPatrick Caulfield struct kvec iov[1]; 9836ed7257bSPatrick Caulfield 9846ed7257bSPatrick Caulfield if (test_and_set_bit(CF_INIT_PENDING, &con->flags)) 9856ed7257bSPatrick Caulfield return; 9866ed7257bSPatrick Caulfield 9876ed7257bSPatrick Caulfield if (con->retries++ > MAX_CONNECT_RETRIES) 9886ed7257bSPatrick Caulfield return; 9896ed7257bSPatrick Caulfield 99036b71a8bSDavid Teigland if (nodeid_to_addr(con->nodeid, NULL, (struct sockaddr *)&rem_addr)) { 9916ed7257bSPatrick Caulfield log_print("no address for nodeid %d", con->nodeid); 9926ed7257bSPatrick Caulfield return; 9936ed7257bSPatrick Caulfield } 9946ed7257bSPatrick Caulfield base_con = nodeid2con(0, 0); 9956ed7257bSPatrick Caulfield BUG_ON(base_con == NULL); 9966ed7257bSPatrick Caulfield 9976ed7257bSPatrick Caulfield make_sockaddr(&rem_addr, dlm_config.ci_tcp_port, &addrlen); 9986ed7257bSPatrick Caulfield 9996ed7257bSPatrick Caulfield outmessage.msg_name = &rem_addr; 10006ed7257bSPatrick Caulfield outmessage.msg_namelen = addrlen; 10016ed7257bSPatrick Caulfield outmessage.msg_control = outcmsg; 10026ed7257bSPatrick Caulfield outmessage.msg_controllen = sizeof(outcmsg); 10036ed7257bSPatrick Caulfield outmessage.msg_flags = MSG_EOR; 10046ed7257bSPatrick Caulfield 10056ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 10066ed7257bSPatrick Caulfield 100704bedd79SDavid Teigland if (list_empty(&con->writequeue)) { 100804bedd79SDavid Teigland spin_unlock(&con->writequeue_lock); 100904bedd79SDavid Teigland log_print("writequeue empty for nodeid %d", con->nodeid); 101004bedd79SDavid Teigland return; 101104bedd79SDavid Teigland } 10126ed7257bSPatrick Caulfield 101304bedd79SDavid Teigland e = list_first_entry(&con->writequeue, struct writequeue_entry, list); 10146ed7257bSPatrick Caulfield len = e->len; 10156ed7257bSPatrick Caulfield offset = e->offset; 10166ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 10176ed7257bSPatrick Caulfield 10186ed7257bSPatrick Caulfield /* Send the first block off the write queue */ 10196ed7257bSPatrick Caulfield iov[0].iov_base = page_address(e->page)+offset; 10206ed7257bSPatrick Caulfield iov[0].iov_len = len; 10216ed7257bSPatrick Caulfield 10226ed7257bSPatrick Caulfield cmsg = CMSG_FIRSTHDR(&outmessage); 10236ed7257bSPatrick Caulfield cmsg->cmsg_level = IPPROTO_SCTP; 10246ed7257bSPatrick Caulfield cmsg->cmsg_type = SCTP_SNDRCV; 10256ed7257bSPatrick Caulfield cmsg->cmsg_len = CMSG_LEN(sizeof(struct sctp_sndrcvinfo)); 10266ed7257bSPatrick Caulfield sinfo = CMSG_DATA(cmsg); 10276ed7257bSPatrick Caulfield memset(sinfo, 0x00, sizeof(struct sctp_sndrcvinfo)); 10286ed7257bSPatrick Caulfield sinfo->sinfo_ppid = cpu_to_le32(dlm_our_nodeid()); 10296ed7257bSPatrick Caulfield outmessage.msg_controllen = cmsg->cmsg_len; 10306ed7257bSPatrick Caulfield 10316ed7257bSPatrick Caulfield ret = kernel_sendmsg(base_con->sock, &outmessage, iov, 1, len); 10326ed7257bSPatrick Caulfield if (ret < 0) { 1033617e82e1SDavid Teigland log_print("Send first packet to node %d failed: %d", 1034617e82e1SDavid Teigland con->nodeid, ret); 10356ed7257bSPatrick Caulfield 10366ed7257bSPatrick Caulfield /* Try again later */ 10376ed7257bSPatrick Caulfield clear_bit(CF_CONNECT_PENDING, &con->flags); 10386ed7257bSPatrick Caulfield clear_bit(CF_INIT_PENDING, &con->flags); 10396ed7257bSPatrick Caulfield } 10406ed7257bSPatrick Caulfield else { 10416ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 10426ed7257bSPatrick Caulfield e->offset += ret; 10436ed7257bSPatrick Caulfield e->len -= ret; 10446ed7257bSPatrick Caulfield 10456ed7257bSPatrick Caulfield if (e->len == 0 && e->users == 0) { 10466ed7257bSPatrick Caulfield list_del(&e->list); 10476ed7257bSPatrick Caulfield free_entry(e); 10486ed7257bSPatrick Caulfield } 10496ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 10506ed7257bSPatrick Caulfield } 10516ed7257bSPatrick Caulfield } 10526ed7257bSPatrick Caulfield 10536ed7257bSPatrick Caulfield /* Connect a new socket to its peer */ 10546ed7257bSPatrick Caulfield static void tcp_connect_to_sock(struct connection *con) 10556ed7257bSPatrick Caulfield { 10566bd8fedaSLon Hohberger struct sockaddr_storage saddr, src_addr; 10576ed7257bSPatrick Caulfield int addr_len; 1058a89d63a1SCasey Dahlin struct socket *sock = NULL; 1059cb2d45daSDavid Teigland int one = 1; 106036b71a8bSDavid Teigland int result; 10616ed7257bSPatrick Caulfield 10626ed7257bSPatrick Caulfield if (con->nodeid == 0) { 10636ed7257bSPatrick Caulfield log_print("attempt to connect sock 0 foiled"); 10646ed7257bSPatrick Caulfield return; 10656ed7257bSPatrick Caulfield } 10666ed7257bSPatrick Caulfield 10676ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 10686ed7257bSPatrick Caulfield if (con->retries++ > MAX_CONNECT_RETRIES) 10696ed7257bSPatrick Caulfield goto out; 10706ed7257bSPatrick Caulfield 10716ed7257bSPatrick Caulfield /* Some odd races can cause double-connects, ignore them */ 107236b71a8bSDavid Teigland if (con->sock) 10736ed7257bSPatrick Caulfield goto out; 10746ed7257bSPatrick Caulfield 10756ed7257bSPatrick Caulfield /* Create a socket to communicate with */ 10766ed7257bSPatrick Caulfield result = sock_create_kern(dlm_local_addr[0]->ss_family, SOCK_STREAM, 10776ed7257bSPatrick Caulfield IPPROTO_TCP, &sock); 10786ed7257bSPatrick Caulfield if (result < 0) 10796ed7257bSPatrick Caulfield goto out_err; 10806ed7257bSPatrick Caulfield 10816ed7257bSPatrick Caulfield memset(&saddr, 0, sizeof(saddr)); 108236b71a8bSDavid Teigland result = nodeid_to_addr(con->nodeid, &saddr, NULL); 108336b71a8bSDavid Teigland if (result < 0) { 108436b71a8bSDavid Teigland log_print("no address for nodeid %d", con->nodeid); 10856ed7257bSPatrick Caulfield goto out_err; 108636b71a8bSDavid Teigland } 10876ed7257bSPatrick Caulfield 10886ed7257bSPatrick Caulfield sock->sk->sk_user_data = con; 10896ed7257bSPatrick Caulfield con->rx_action = receive_from_sock; 10906ed7257bSPatrick Caulfield con->connect_action = tcp_connect_to_sock; 10916ed7257bSPatrick Caulfield add_sock(sock, con); 10926ed7257bSPatrick Caulfield 10936bd8fedaSLon Hohberger /* Bind to our cluster-known address connecting to avoid 10946bd8fedaSLon Hohberger routing problems */ 10956bd8fedaSLon Hohberger memcpy(&src_addr, dlm_local_addr[0], sizeof(src_addr)); 10966bd8fedaSLon Hohberger make_sockaddr(&src_addr, 0, &addr_len); 10976bd8fedaSLon Hohberger result = sock->ops->bind(sock, (struct sockaddr *) &src_addr, 10986bd8fedaSLon Hohberger addr_len); 10996bd8fedaSLon Hohberger if (result < 0) { 11006bd8fedaSLon Hohberger log_print("could not bind for connect: %d", result); 11016bd8fedaSLon Hohberger /* This *may* not indicate a critical error */ 11026bd8fedaSLon Hohberger } 11036bd8fedaSLon Hohberger 11046ed7257bSPatrick Caulfield make_sockaddr(&saddr, dlm_config.ci_tcp_port, &addr_len); 11056ed7257bSPatrick Caulfield 11066ed7257bSPatrick Caulfield log_print("connecting to %d", con->nodeid); 1107cb2d45daSDavid Teigland 1108cb2d45daSDavid Teigland /* Turn off Nagle's algorithm */ 1109cb2d45daSDavid Teigland kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one, 1110cb2d45daSDavid Teigland sizeof(one)); 1111cb2d45daSDavid Teigland 111236b71a8bSDavid Teigland result = sock->ops->connect(sock, (struct sockaddr *)&saddr, addr_len, 11136ed7257bSPatrick Caulfield O_NONBLOCK); 11146ed7257bSPatrick Caulfield if (result == -EINPROGRESS) 11156ed7257bSPatrick Caulfield result = 0; 11166ed7257bSPatrick Caulfield if (result == 0) 11176ed7257bSPatrick Caulfield goto out; 11186ed7257bSPatrick Caulfield 11196ed7257bSPatrick Caulfield out_err: 11206ed7257bSPatrick Caulfield if (con->sock) { 11216ed7257bSPatrick Caulfield sock_release(con->sock); 11226ed7257bSPatrick Caulfield con->sock = NULL; 1123a89d63a1SCasey Dahlin } else if (sock) { 1124a89d63a1SCasey Dahlin sock_release(sock); 11256ed7257bSPatrick Caulfield } 11266ed7257bSPatrick Caulfield /* 11276ed7257bSPatrick Caulfield * Some errors are fatal and this list might need adjusting. For other 11286ed7257bSPatrick Caulfield * errors we try again until the max number of retries is reached. 11296ed7257bSPatrick Caulfield */ 113036b71a8bSDavid Teigland if (result != -EHOSTUNREACH && 113136b71a8bSDavid Teigland result != -ENETUNREACH && 113236b71a8bSDavid Teigland result != -ENETDOWN && 113336b71a8bSDavid Teigland result != -EINVAL && 113436b71a8bSDavid Teigland result != -EPROTONOSUPPORT) { 113536b71a8bSDavid Teigland log_print("connect %d try %d error %d", con->nodeid, 113636b71a8bSDavid Teigland con->retries, result); 113736b71a8bSDavid Teigland mutex_unlock(&con->sock_mutex); 113836b71a8bSDavid Teigland msleep(1000); 11396ed7257bSPatrick Caulfield lowcomms_connect_sock(con); 114036b71a8bSDavid Teigland return; 11416ed7257bSPatrick Caulfield } 11426ed7257bSPatrick Caulfield out: 11436ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 11446ed7257bSPatrick Caulfield return; 11456ed7257bSPatrick Caulfield } 11466ed7257bSPatrick Caulfield 11476ed7257bSPatrick Caulfield static struct socket *tcp_create_listen_sock(struct connection *con, 11486ed7257bSPatrick Caulfield struct sockaddr_storage *saddr) 11496ed7257bSPatrick Caulfield { 11506ed7257bSPatrick Caulfield struct socket *sock = NULL; 11516ed7257bSPatrick Caulfield int result = 0; 11526ed7257bSPatrick Caulfield int one = 1; 11536ed7257bSPatrick Caulfield int addr_len; 11546ed7257bSPatrick Caulfield 11556ed7257bSPatrick Caulfield if (dlm_local_addr[0]->ss_family == AF_INET) 11566ed7257bSPatrick Caulfield addr_len = sizeof(struct sockaddr_in); 11576ed7257bSPatrick Caulfield else 11586ed7257bSPatrick Caulfield addr_len = sizeof(struct sockaddr_in6); 11596ed7257bSPatrick Caulfield 11606ed7257bSPatrick Caulfield /* Create a socket to communicate with */ 1161617e82e1SDavid Teigland result = sock_create_kern(dlm_local_addr[0]->ss_family, SOCK_STREAM, 1162617e82e1SDavid Teigland IPPROTO_TCP, &sock); 11636ed7257bSPatrick Caulfield if (result < 0) { 1164617e82e1SDavid Teigland log_print("Can't create listening comms socket"); 11656ed7257bSPatrick Caulfield goto create_out; 11666ed7257bSPatrick Caulfield } 11676ed7257bSPatrick Caulfield 1168cb2d45daSDavid Teigland /* Turn off Nagle's algorithm */ 1169cb2d45daSDavid Teigland kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (char *)&one, 1170cb2d45daSDavid Teigland sizeof(one)); 1171cb2d45daSDavid Teigland 11726ed7257bSPatrick Caulfield result = kernel_setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, 11736ed7257bSPatrick Caulfield (char *)&one, sizeof(one)); 11746ed7257bSPatrick Caulfield 11756ed7257bSPatrick Caulfield if (result < 0) { 1176617e82e1SDavid Teigland log_print("Failed to set SO_REUSEADDR on socket: %d", result); 11776ed7257bSPatrick Caulfield } 11786ed7257bSPatrick Caulfield sock->sk->sk_user_data = con; 11796ed7257bSPatrick Caulfield con->rx_action = tcp_accept_from_sock; 11806ed7257bSPatrick Caulfield con->connect_action = tcp_connect_to_sock; 11816ed7257bSPatrick Caulfield con->sock = sock; 11826ed7257bSPatrick Caulfield 11836ed7257bSPatrick Caulfield /* Bind to our port */ 11846ed7257bSPatrick Caulfield make_sockaddr(saddr, dlm_config.ci_tcp_port, &addr_len); 11856ed7257bSPatrick Caulfield result = sock->ops->bind(sock, (struct sockaddr *) saddr, addr_len); 11866ed7257bSPatrick Caulfield if (result < 0) { 1187617e82e1SDavid Teigland log_print("Can't bind to port %d", dlm_config.ci_tcp_port); 11886ed7257bSPatrick Caulfield sock_release(sock); 11896ed7257bSPatrick Caulfield sock = NULL; 11906ed7257bSPatrick Caulfield con->sock = NULL; 11916ed7257bSPatrick Caulfield goto create_out; 11926ed7257bSPatrick Caulfield } 11936ed7257bSPatrick Caulfield result = kernel_setsockopt(sock, SOL_SOCKET, SO_KEEPALIVE, 11946ed7257bSPatrick Caulfield (char *)&one, sizeof(one)); 11956ed7257bSPatrick Caulfield if (result < 0) { 1196617e82e1SDavid Teigland log_print("Set keepalive failed: %d", result); 11976ed7257bSPatrick Caulfield } 11986ed7257bSPatrick Caulfield 11996ed7257bSPatrick Caulfield result = sock->ops->listen(sock, 5); 12006ed7257bSPatrick Caulfield if (result < 0) { 1201617e82e1SDavid Teigland log_print("Can't listen on port %d", dlm_config.ci_tcp_port); 12026ed7257bSPatrick Caulfield sock_release(sock); 12036ed7257bSPatrick Caulfield sock = NULL; 12046ed7257bSPatrick Caulfield goto create_out; 12056ed7257bSPatrick Caulfield } 12066ed7257bSPatrick Caulfield 12076ed7257bSPatrick Caulfield create_out: 12086ed7257bSPatrick Caulfield return sock; 12096ed7257bSPatrick Caulfield } 12106ed7257bSPatrick Caulfield 12116ed7257bSPatrick Caulfield /* Get local addresses */ 12126ed7257bSPatrick Caulfield static void init_local(void) 12136ed7257bSPatrick Caulfield { 12146ed7257bSPatrick Caulfield struct sockaddr_storage sas, *addr; 12156ed7257bSPatrick Caulfield int i; 12166ed7257bSPatrick Caulfield 121730d3a237SPatrick Caulfield dlm_local_count = 0; 12181b189b88SDavid Teigland for (i = 0; i < DLM_MAX_ADDR_COUNT; i++) { 12196ed7257bSPatrick Caulfield if (dlm_our_addr(&sas, i)) 12206ed7257bSPatrick Caulfield break; 12216ed7257bSPatrick Caulfield 1222573c24c4SDavid Teigland addr = kmalloc(sizeof(*addr), GFP_NOFS); 12236ed7257bSPatrick Caulfield if (!addr) 12246ed7257bSPatrick Caulfield break; 12256ed7257bSPatrick Caulfield memcpy(addr, &sas, sizeof(*addr)); 12266ed7257bSPatrick Caulfield dlm_local_addr[dlm_local_count++] = addr; 12276ed7257bSPatrick Caulfield } 12286ed7257bSPatrick Caulfield } 12296ed7257bSPatrick Caulfield 1230617e82e1SDavid Teigland /* Bind to an IP address. SCTP allows multiple address so it can do 1231617e82e1SDavid Teigland multi-homing */ 1232617e82e1SDavid Teigland static int add_sctp_bind_addr(struct connection *sctp_con, 1233617e82e1SDavid Teigland struct sockaddr_storage *addr, 1234617e82e1SDavid Teigland int addr_len, int num) 12356ed7257bSPatrick Caulfield { 12366ed7257bSPatrick Caulfield int result = 0; 12376ed7257bSPatrick Caulfield 12386ed7257bSPatrick Caulfield if (num == 1) 12396ed7257bSPatrick Caulfield result = kernel_bind(sctp_con->sock, 12406ed7257bSPatrick Caulfield (struct sockaddr *) addr, 12416ed7257bSPatrick Caulfield addr_len); 12426ed7257bSPatrick Caulfield else 12436ed7257bSPatrick Caulfield result = kernel_setsockopt(sctp_con->sock, SOL_SCTP, 12446ed7257bSPatrick Caulfield SCTP_SOCKOPT_BINDX_ADD, 12456ed7257bSPatrick Caulfield (char *)addr, addr_len); 12466ed7257bSPatrick Caulfield 12476ed7257bSPatrick Caulfield if (result < 0) 12486ed7257bSPatrick Caulfield log_print("Can't bind to port %d addr number %d", 12496ed7257bSPatrick Caulfield dlm_config.ci_tcp_port, num); 12506ed7257bSPatrick Caulfield 12516ed7257bSPatrick Caulfield return result; 12526ed7257bSPatrick Caulfield } 12536ed7257bSPatrick Caulfield 12546ed7257bSPatrick Caulfield /* Initialise SCTP socket and bind to all interfaces */ 12556ed7257bSPatrick Caulfield static int sctp_listen_for_all(void) 12566ed7257bSPatrick Caulfield { 12576ed7257bSPatrick Caulfield struct socket *sock = NULL; 12586ed7257bSPatrick Caulfield struct sockaddr_storage localaddr; 12596ed7257bSPatrick Caulfield struct sctp_event_subscribe subscribe; 12606ed7257bSPatrick Caulfield int result = -EINVAL, num = 1, i, addr_len; 1261573c24c4SDavid Teigland struct connection *con = nodeid2con(0, GFP_NOFS); 12626ed7257bSPatrick Caulfield int bufsize = NEEDED_RMEM; 12636ed7257bSPatrick Caulfield 12646ed7257bSPatrick Caulfield if (!con) 12656ed7257bSPatrick Caulfield return -ENOMEM; 12666ed7257bSPatrick Caulfield 12676ed7257bSPatrick Caulfield log_print("Using SCTP for communications"); 12686ed7257bSPatrick Caulfield 12696ed7257bSPatrick Caulfield result = sock_create_kern(dlm_local_addr[0]->ss_family, SOCK_SEQPACKET, 12706ed7257bSPatrick Caulfield IPPROTO_SCTP, &sock); 12716ed7257bSPatrick Caulfield if (result < 0) { 12726ed7257bSPatrick Caulfield log_print("Can't create comms socket, check SCTP is loaded"); 12736ed7257bSPatrick Caulfield goto out; 12746ed7257bSPatrick Caulfield } 12756ed7257bSPatrick Caulfield 12766ed7257bSPatrick Caulfield /* Listen for events */ 12776ed7257bSPatrick Caulfield memset(&subscribe, 0, sizeof(subscribe)); 12786ed7257bSPatrick Caulfield subscribe.sctp_data_io_event = 1; 12796ed7257bSPatrick Caulfield subscribe.sctp_association_event = 1; 12806ed7257bSPatrick Caulfield subscribe.sctp_send_failure_event = 1; 12816ed7257bSPatrick Caulfield subscribe.sctp_shutdown_event = 1; 12826ed7257bSPatrick Caulfield subscribe.sctp_partial_delivery_event = 1; 12836ed7257bSPatrick Caulfield 1284df61c952SDavid S. Miller result = kernel_setsockopt(sock, SOL_SOCKET, SO_RCVBUFFORCE, 12856ed7257bSPatrick Caulfield (char *)&bufsize, sizeof(bufsize)); 12866ed7257bSPatrick Caulfield if (result) 1287617e82e1SDavid Teigland log_print("Error increasing buffer space on socket %d", result); 12886ed7257bSPatrick Caulfield 12896ed7257bSPatrick Caulfield result = kernel_setsockopt(sock, SOL_SCTP, SCTP_EVENTS, 12906ed7257bSPatrick Caulfield (char *)&subscribe, sizeof(subscribe)); 12916ed7257bSPatrick Caulfield if (result < 0) { 12926ed7257bSPatrick Caulfield log_print("Failed to set SCTP_EVENTS on socket: result=%d", 12936ed7257bSPatrick Caulfield result); 12946ed7257bSPatrick Caulfield goto create_delsock; 12956ed7257bSPatrick Caulfield } 12966ed7257bSPatrick Caulfield 12976ed7257bSPatrick Caulfield /* Init con struct */ 12986ed7257bSPatrick Caulfield sock->sk->sk_user_data = con; 12996ed7257bSPatrick Caulfield con->sock = sock; 13006ed7257bSPatrick Caulfield con->sock->sk->sk_data_ready = lowcomms_data_ready; 13016ed7257bSPatrick Caulfield con->rx_action = receive_from_sock; 13026ed7257bSPatrick Caulfield con->connect_action = sctp_init_assoc; 13036ed7257bSPatrick Caulfield 13046ed7257bSPatrick Caulfield /* Bind to all interfaces. */ 13056ed7257bSPatrick Caulfield for (i = 0; i < dlm_local_count; i++) { 13066ed7257bSPatrick Caulfield memcpy(&localaddr, dlm_local_addr[i], sizeof(localaddr)); 13076ed7257bSPatrick Caulfield make_sockaddr(&localaddr, dlm_config.ci_tcp_port, &addr_len); 13086ed7257bSPatrick Caulfield 13096ed7257bSPatrick Caulfield result = add_sctp_bind_addr(con, &localaddr, addr_len, num); 13106ed7257bSPatrick Caulfield if (result) 13116ed7257bSPatrick Caulfield goto create_delsock; 13126ed7257bSPatrick Caulfield ++num; 13136ed7257bSPatrick Caulfield } 13146ed7257bSPatrick Caulfield 13156ed7257bSPatrick Caulfield result = sock->ops->listen(sock, 5); 13166ed7257bSPatrick Caulfield if (result < 0) { 13176ed7257bSPatrick Caulfield log_print("Can't set socket listening"); 13186ed7257bSPatrick Caulfield goto create_delsock; 13196ed7257bSPatrick Caulfield } 13206ed7257bSPatrick Caulfield 13216ed7257bSPatrick Caulfield return 0; 13226ed7257bSPatrick Caulfield 13236ed7257bSPatrick Caulfield create_delsock: 13246ed7257bSPatrick Caulfield sock_release(sock); 13256ed7257bSPatrick Caulfield con->sock = NULL; 13266ed7257bSPatrick Caulfield out: 13276ed7257bSPatrick Caulfield return result; 13286ed7257bSPatrick Caulfield } 13296ed7257bSPatrick Caulfield 13306ed7257bSPatrick Caulfield static int tcp_listen_for_all(void) 13316ed7257bSPatrick Caulfield { 13326ed7257bSPatrick Caulfield struct socket *sock = NULL; 1333573c24c4SDavid Teigland struct connection *con = nodeid2con(0, GFP_NOFS); 13346ed7257bSPatrick Caulfield int result = -EINVAL; 13356ed7257bSPatrick Caulfield 13366ed7257bSPatrick Caulfield if (!con) 13376ed7257bSPatrick Caulfield return -ENOMEM; 13386ed7257bSPatrick Caulfield 13396ed7257bSPatrick Caulfield /* We don't support multi-homed hosts */ 13406ed7257bSPatrick Caulfield if (dlm_local_addr[1] != NULL) { 1341617e82e1SDavid Teigland log_print("TCP protocol can't handle multi-homed hosts, " 1342617e82e1SDavid Teigland "try SCTP"); 13436ed7257bSPatrick Caulfield return -EINVAL; 13446ed7257bSPatrick Caulfield } 13456ed7257bSPatrick Caulfield 13466ed7257bSPatrick Caulfield log_print("Using TCP for communications"); 13476ed7257bSPatrick Caulfield 13486ed7257bSPatrick Caulfield sock = tcp_create_listen_sock(con, dlm_local_addr[0]); 13496ed7257bSPatrick Caulfield if (sock) { 13506ed7257bSPatrick Caulfield add_sock(sock, con); 13516ed7257bSPatrick Caulfield result = 0; 13526ed7257bSPatrick Caulfield } 13536ed7257bSPatrick Caulfield else { 13546ed7257bSPatrick Caulfield result = -EADDRINUSE; 13556ed7257bSPatrick Caulfield } 13566ed7257bSPatrick Caulfield 13576ed7257bSPatrick Caulfield return result; 13586ed7257bSPatrick Caulfield } 13596ed7257bSPatrick Caulfield 13606ed7257bSPatrick Caulfield 13616ed7257bSPatrick Caulfield 13626ed7257bSPatrick Caulfield static struct writequeue_entry *new_writequeue_entry(struct connection *con, 13636ed7257bSPatrick Caulfield gfp_t allocation) 13646ed7257bSPatrick Caulfield { 13656ed7257bSPatrick Caulfield struct writequeue_entry *entry; 13666ed7257bSPatrick Caulfield 13676ed7257bSPatrick Caulfield entry = kmalloc(sizeof(struct writequeue_entry), allocation); 13686ed7257bSPatrick Caulfield if (!entry) 13696ed7257bSPatrick Caulfield return NULL; 13706ed7257bSPatrick Caulfield 13716ed7257bSPatrick Caulfield entry->page = alloc_page(allocation); 13726ed7257bSPatrick Caulfield if (!entry->page) { 13736ed7257bSPatrick Caulfield kfree(entry); 13746ed7257bSPatrick Caulfield return NULL; 13756ed7257bSPatrick Caulfield } 13766ed7257bSPatrick Caulfield 13776ed7257bSPatrick Caulfield entry->offset = 0; 13786ed7257bSPatrick Caulfield entry->len = 0; 13796ed7257bSPatrick Caulfield entry->end = 0; 13806ed7257bSPatrick Caulfield entry->users = 0; 13816ed7257bSPatrick Caulfield entry->con = con; 13826ed7257bSPatrick Caulfield 13836ed7257bSPatrick Caulfield return entry; 13846ed7257bSPatrick Caulfield } 13856ed7257bSPatrick Caulfield 1386617e82e1SDavid Teigland void *dlm_lowcomms_get_buffer(int nodeid, int len, gfp_t allocation, char **ppc) 13876ed7257bSPatrick Caulfield { 13886ed7257bSPatrick Caulfield struct connection *con; 13896ed7257bSPatrick Caulfield struct writequeue_entry *e; 13906ed7257bSPatrick Caulfield int offset = 0; 13916ed7257bSPatrick Caulfield int users = 0; 13926ed7257bSPatrick Caulfield 13936ed7257bSPatrick Caulfield con = nodeid2con(nodeid, allocation); 13946ed7257bSPatrick Caulfield if (!con) 13956ed7257bSPatrick Caulfield return NULL; 13966ed7257bSPatrick Caulfield 13976ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 13986ed7257bSPatrick Caulfield e = list_entry(con->writequeue.prev, struct writequeue_entry, list); 13996ed7257bSPatrick Caulfield if ((&e->list == &con->writequeue) || 14006ed7257bSPatrick Caulfield (PAGE_CACHE_SIZE - e->end < len)) { 14016ed7257bSPatrick Caulfield e = NULL; 14026ed7257bSPatrick Caulfield } else { 14036ed7257bSPatrick Caulfield offset = e->end; 14046ed7257bSPatrick Caulfield e->end += len; 14056ed7257bSPatrick Caulfield users = e->users++; 14066ed7257bSPatrick Caulfield } 14076ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14086ed7257bSPatrick Caulfield 14096ed7257bSPatrick Caulfield if (e) { 14106ed7257bSPatrick Caulfield got_one: 14116ed7257bSPatrick Caulfield *ppc = page_address(e->page) + offset; 14126ed7257bSPatrick Caulfield return e; 14136ed7257bSPatrick Caulfield } 14146ed7257bSPatrick Caulfield 14156ed7257bSPatrick Caulfield e = new_writequeue_entry(con, allocation); 14166ed7257bSPatrick Caulfield if (e) { 14176ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 14186ed7257bSPatrick Caulfield offset = e->end; 14196ed7257bSPatrick Caulfield e->end += len; 14206ed7257bSPatrick Caulfield users = e->users++; 14216ed7257bSPatrick Caulfield list_add_tail(&e->list, &con->writequeue); 14226ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14236ed7257bSPatrick Caulfield goto got_one; 14246ed7257bSPatrick Caulfield } 14256ed7257bSPatrick Caulfield return NULL; 14266ed7257bSPatrick Caulfield } 14276ed7257bSPatrick Caulfield 14286ed7257bSPatrick Caulfield void dlm_lowcomms_commit_buffer(void *mh) 14296ed7257bSPatrick Caulfield { 14306ed7257bSPatrick Caulfield struct writequeue_entry *e = (struct writequeue_entry *)mh; 14316ed7257bSPatrick Caulfield struct connection *con = e->con; 14326ed7257bSPatrick Caulfield int users; 14336ed7257bSPatrick Caulfield 14346ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 14356ed7257bSPatrick Caulfield users = --e->users; 14366ed7257bSPatrick Caulfield if (users) 14376ed7257bSPatrick Caulfield goto out; 14386ed7257bSPatrick Caulfield e->len = e->end - e->offset; 14396ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14406ed7257bSPatrick Caulfield 14416ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_WRITE_PENDING, &con->flags)) { 14426ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 14436ed7257bSPatrick Caulfield } 14446ed7257bSPatrick Caulfield return; 14456ed7257bSPatrick Caulfield 14466ed7257bSPatrick Caulfield out: 14476ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14486ed7257bSPatrick Caulfield return; 14496ed7257bSPatrick Caulfield } 14506ed7257bSPatrick Caulfield 14516ed7257bSPatrick Caulfield /* Send a message */ 14526ed7257bSPatrick Caulfield static void send_to_sock(struct connection *con) 14536ed7257bSPatrick Caulfield { 14546ed7257bSPatrick Caulfield int ret = 0; 14556ed7257bSPatrick Caulfield const int msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL; 14566ed7257bSPatrick Caulfield struct writequeue_entry *e; 14576ed7257bSPatrick Caulfield int len, offset; 1458f92c8dd7SBob Peterson int count = 0; 14596ed7257bSPatrick Caulfield 14606ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 14616ed7257bSPatrick Caulfield if (con->sock == NULL) 14626ed7257bSPatrick Caulfield goto out_connect; 14636ed7257bSPatrick Caulfield 14646ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 14656ed7257bSPatrick Caulfield for (;;) { 14666ed7257bSPatrick Caulfield e = list_entry(con->writequeue.next, struct writequeue_entry, 14676ed7257bSPatrick Caulfield list); 14686ed7257bSPatrick Caulfield if ((struct list_head *) e == &con->writequeue) 14696ed7257bSPatrick Caulfield break; 14706ed7257bSPatrick Caulfield 14716ed7257bSPatrick Caulfield len = e->len; 14726ed7257bSPatrick Caulfield offset = e->offset; 14736ed7257bSPatrick Caulfield BUG_ON(len == 0 && e->users == 0); 14746ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14756ed7257bSPatrick Caulfield 14766ed7257bSPatrick Caulfield ret = 0; 14776ed7257bSPatrick Caulfield if (len) { 14781329e3f2SPaolo Bonzini ret = kernel_sendpage(con->sock, e->page, offset, len, 14796ed7257bSPatrick Caulfield msg_flags); 1480d66f8277SPatrick Caulfield if (ret == -EAGAIN || ret == 0) { 1481b36930ddSDavid Miller if (ret == -EAGAIN && 1482b36930ddSDavid Miller test_bit(SOCK_ASYNC_NOSPACE, &con->sock->flags) && 1483b36930ddSDavid Miller !test_and_set_bit(CF_APP_LIMITED, &con->flags)) { 1484b36930ddSDavid Miller /* Notify TCP that we're limited by the 1485b36930ddSDavid Miller * application window size. 1486b36930ddSDavid Miller */ 1487b36930ddSDavid Miller set_bit(SOCK_NOSPACE, &con->sock->flags); 1488b36930ddSDavid Miller con->sock->sk->sk_write_pending++; 1489b36930ddSDavid Miller } 1490d66f8277SPatrick Caulfield cond_resched(); 14916ed7257bSPatrick Caulfield goto out; 1492d66f8277SPatrick Caulfield } 14936ed7257bSPatrick Caulfield if (ret <= 0) 14946ed7257bSPatrick Caulfield goto send_error; 1495d66f8277SPatrick Caulfield } 1496f92c8dd7SBob Peterson 14976ed7257bSPatrick Caulfield /* Don't starve people filling buffers */ 1498f92c8dd7SBob Peterson if (++count >= MAX_SEND_MSG_COUNT) { 14996ed7257bSPatrick Caulfield cond_resched(); 1500f92c8dd7SBob Peterson count = 0; 1501f92c8dd7SBob Peterson } 15026ed7257bSPatrick Caulfield 15036ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 15046ed7257bSPatrick Caulfield e->offset += ret; 15056ed7257bSPatrick Caulfield e->len -= ret; 15066ed7257bSPatrick Caulfield 15076ed7257bSPatrick Caulfield if (e->len == 0 && e->users == 0) { 15086ed7257bSPatrick Caulfield list_del(&e->list); 15096ed7257bSPatrick Caulfield free_entry(e); 15106ed7257bSPatrick Caulfield continue; 15116ed7257bSPatrick Caulfield } 15126ed7257bSPatrick Caulfield } 15136ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 15146ed7257bSPatrick Caulfield out: 15156ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 15166ed7257bSPatrick Caulfield return; 15176ed7257bSPatrick Caulfield 15186ed7257bSPatrick Caulfield send_error: 15196ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 15206ed7257bSPatrick Caulfield close_connection(con, false); 15216ed7257bSPatrick Caulfield lowcomms_connect_sock(con); 15226ed7257bSPatrick Caulfield return; 15236ed7257bSPatrick Caulfield 15246ed7257bSPatrick Caulfield out_connect: 15256ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 15266ed7257bSPatrick Caulfield if (!test_bit(CF_INIT_PENDING, &con->flags)) 15276ed7257bSPatrick Caulfield lowcomms_connect_sock(con); 15286ed7257bSPatrick Caulfield return; 15296ed7257bSPatrick Caulfield } 15306ed7257bSPatrick Caulfield 15316ed7257bSPatrick Caulfield static void clean_one_writequeue(struct connection *con) 15326ed7257bSPatrick Caulfield { 15335e9ccc37SChristine Caulfield struct writequeue_entry *e, *safe; 15346ed7257bSPatrick Caulfield 15356ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 15365e9ccc37SChristine Caulfield list_for_each_entry_safe(e, safe, &con->writequeue, list) { 15376ed7257bSPatrick Caulfield list_del(&e->list); 15386ed7257bSPatrick Caulfield free_entry(e); 15396ed7257bSPatrick Caulfield } 15406ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 15416ed7257bSPatrick Caulfield } 15426ed7257bSPatrick Caulfield 15436ed7257bSPatrick Caulfield /* Called from recovery when it knows that a node has 15446ed7257bSPatrick Caulfield left the cluster */ 15456ed7257bSPatrick Caulfield int dlm_lowcomms_close(int nodeid) 15466ed7257bSPatrick Caulfield { 15476ed7257bSPatrick Caulfield struct connection *con; 154836b71a8bSDavid Teigland struct dlm_node_addr *na; 15496ed7257bSPatrick Caulfield 15506ed7257bSPatrick Caulfield log_print("closing connection to node %d", nodeid); 15516ed7257bSPatrick Caulfield con = nodeid2con(nodeid, 0); 15526ed7257bSPatrick Caulfield if (con) { 1553063c4c99SLars Marowsky-Bree clear_bit(CF_CONNECT_PENDING, &con->flags); 1554063c4c99SLars Marowsky-Bree clear_bit(CF_WRITE_PENDING, &con->flags); 1555063c4c99SLars Marowsky-Bree set_bit(CF_CLOSE, &con->flags); 1556063c4c99SLars Marowsky-Bree if (cancel_work_sync(&con->swork)) 1557063c4c99SLars Marowsky-Bree log_print("canceled swork for node %d", nodeid); 1558063c4c99SLars Marowsky-Bree if (cancel_work_sync(&con->rwork)) 1559063c4c99SLars Marowsky-Bree log_print("canceled rwork for node %d", nodeid); 15606ed7257bSPatrick Caulfield clean_one_writequeue(con); 15616ed7257bSPatrick Caulfield close_connection(con, true); 15626ed7257bSPatrick Caulfield } 156336b71a8bSDavid Teigland 156436b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 156536b71a8bSDavid Teigland na = find_node_addr(nodeid); 156636b71a8bSDavid Teigland if (na) { 156736b71a8bSDavid Teigland list_del(&na->list); 156836b71a8bSDavid Teigland while (na->addr_count--) 156936b71a8bSDavid Teigland kfree(na->addr[na->addr_count]); 157036b71a8bSDavid Teigland kfree(na); 157136b71a8bSDavid Teigland } 157236b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 157336b71a8bSDavid Teigland 15746ed7257bSPatrick Caulfield return 0; 15756ed7257bSPatrick Caulfield } 15766ed7257bSPatrick Caulfield 15776ed7257bSPatrick Caulfield /* Receive workqueue function */ 15786ed7257bSPatrick Caulfield static void process_recv_sockets(struct work_struct *work) 15796ed7257bSPatrick Caulfield { 15806ed7257bSPatrick Caulfield struct connection *con = container_of(work, struct connection, rwork); 15816ed7257bSPatrick Caulfield int err; 15826ed7257bSPatrick Caulfield 15836ed7257bSPatrick Caulfield clear_bit(CF_READ_PENDING, &con->flags); 15846ed7257bSPatrick Caulfield do { 15856ed7257bSPatrick Caulfield err = con->rx_action(con); 15866ed7257bSPatrick Caulfield } while (!err); 15876ed7257bSPatrick Caulfield } 15886ed7257bSPatrick Caulfield 15896ed7257bSPatrick Caulfield /* Send workqueue function */ 15906ed7257bSPatrick Caulfield static void process_send_sockets(struct work_struct *work) 15916ed7257bSPatrick Caulfield { 15926ed7257bSPatrick Caulfield struct connection *con = container_of(work, struct connection, swork); 15936ed7257bSPatrick Caulfield 15946ed7257bSPatrick Caulfield if (test_and_clear_bit(CF_CONNECT_PENDING, &con->flags)) { 15956ed7257bSPatrick Caulfield con->connect_action(con); 1596063c4c99SLars Marowsky-Bree set_bit(CF_WRITE_PENDING, &con->flags); 15976ed7257bSPatrick Caulfield } 1598063c4c99SLars Marowsky-Bree if (test_and_clear_bit(CF_WRITE_PENDING, &con->flags)) 15996ed7257bSPatrick Caulfield send_to_sock(con); 16006ed7257bSPatrick Caulfield } 16016ed7257bSPatrick Caulfield 16026ed7257bSPatrick Caulfield 16036ed7257bSPatrick Caulfield /* Discard all entries on the write queues */ 16046ed7257bSPatrick Caulfield static void clean_writequeues(void) 16056ed7257bSPatrick Caulfield { 16065e9ccc37SChristine Caulfield foreach_conn(clean_one_writequeue); 16076ed7257bSPatrick Caulfield } 16086ed7257bSPatrick Caulfield 16096ed7257bSPatrick Caulfield static void work_stop(void) 16106ed7257bSPatrick Caulfield { 16116ed7257bSPatrick Caulfield destroy_workqueue(recv_workqueue); 16126ed7257bSPatrick Caulfield destroy_workqueue(send_workqueue); 16136ed7257bSPatrick Caulfield } 16146ed7257bSPatrick Caulfield 16156ed7257bSPatrick Caulfield static int work_start(void) 16166ed7257bSPatrick Caulfield { 1617e43f055aSDavid Teigland recv_workqueue = alloc_workqueue("dlm_recv", 1618e43f055aSDavid Teigland WQ_UNBOUND | WQ_MEM_RECLAIM, 1); 1619b9d41052SNamhyung Kim if (!recv_workqueue) { 1620b9d41052SNamhyung Kim log_print("can't start dlm_recv"); 1621b9d41052SNamhyung Kim return -ENOMEM; 16226ed7257bSPatrick Caulfield } 16236ed7257bSPatrick Caulfield 1624e43f055aSDavid Teigland send_workqueue = alloc_workqueue("dlm_send", 1625e43f055aSDavid Teigland WQ_UNBOUND | WQ_MEM_RECLAIM, 1); 1626b9d41052SNamhyung Kim if (!send_workqueue) { 1627b9d41052SNamhyung Kim log_print("can't start dlm_send"); 16286ed7257bSPatrick Caulfield destroy_workqueue(recv_workqueue); 1629b9d41052SNamhyung Kim return -ENOMEM; 16306ed7257bSPatrick Caulfield } 16316ed7257bSPatrick Caulfield 16326ed7257bSPatrick Caulfield return 0; 16336ed7257bSPatrick Caulfield } 16346ed7257bSPatrick Caulfield 16355e9ccc37SChristine Caulfield static void stop_conn(struct connection *con) 16366ed7257bSPatrick Caulfield { 16379e5f2825SPatrick Caulfield con->flags |= 0x0F; 1638391fbdc5SChristine Caulfield if (con->sock && con->sock->sk) 1639afb853fbSPatrick Caulfield con->sock->sk->sk_user_data = NULL; 1640afb853fbSPatrick Caulfield } 16415e9ccc37SChristine Caulfield 16425e9ccc37SChristine Caulfield static void free_conn(struct connection *con) 16435e9ccc37SChristine Caulfield { 16445e9ccc37SChristine Caulfield close_connection(con, true); 16455e9ccc37SChristine Caulfield if (con->othercon) 16465e9ccc37SChristine Caulfield kmem_cache_free(con_cache, con->othercon); 16475e9ccc37SChristine Caulfield hlist_del(&con->list); 16485e9ccc37SChristine Caulfield kmem_cache_free(con_cache, con); 16496ed7257bSPatrick Caulfield } 16505e9ccc37SChristine Caulfield 16515e9ccc37SChristine Caulfield void dlm_lowcomms_stop(void) 16525e9ccc37SChristine Caulfield { 16535e9ccc37SChristine Caulfield /* Set all the flags to prevent any 16545e9ccc37SChristine Caulfield socket activity. 16555e9ccc37SChristine Caulfield */ 16565e9ccc37SChristine Caulfield mutex_lock(&connections_lock); 1657513ef596SDavid Teigland dlm_allow_conn = 0; 16585e9ccc37SChristine Caulfield foreach_conn(stop_conn); 16597a936ce7SMatthias Kaehlcke mutex_unlock(&connections_lock); 16606ed7257bSPatrick Caulfield 16616ed7257bSPatrick Caulfield work_stop(); 16626ed7257bSPatrick Caulfield 16637a936ce7SMatthias Kaehlcke mutex_lock(&connections_lock); 16646ed7257bSPatrick Caulfield clean_writequeues(); 16656ed7257bSPatrick Caulfield 16665e9ccc37SChristine Caulfield foreach_conn(free_conn); 16675e9ccc37SChristine Caulfield 16687a936ce7SMatthias Kaehlcke mutex_unlock(&connections_lock); 16696ed7257bSPatrick Caulfield kmem_cache_destroy(con_cache); 16706ed7257bSPatrick Caulfield } 16716ed7257bSPatrick Caulfield 16726ed7257bSPatrick Caulfield int dlm_lowcomms_start(void) 16736ed7257bSPatrick Caulfield { 16746ed7257bSPatrick Caulfield int error = -EINVAL; 16756ed7257bSPatrick Caulfield struct connection *con; 16765e9ccc37SChristine Caulfield int i; 16775e9ccc37SChristine Caulfield 16785e9ccc37SChristine Caulfield for (i = 0; i < CONN_HASH_SIZE; i++) 16795e9ccc37SChristine Caulfield INIT_HLIST_HEAD(&connection_hash[i]); 16806ed7257bSPatrick Caulfield 16816ed7257bSPatrick Caulfield init_local(); 16826ed7257bSPatrick Caulfield if (!dlm_local_count) { 1683617e82e1SDavid Teigland error = -ENOTCONN; 16846ed7257bSPatrick Caulfield log_print("no local IP address has been set"); 1685513ef596SDavid Teigland goto fail; 16866ed7257bSPatrick Caulfield } 16876ed7257bSPatrick Caulfield 16886ed7257bSPatrick Caulfield error = -ENOMEM; 16896ed7257bSPatrick Caulfield con_cache = kmem_cache_create("dlm_conn", sizeof(struct connection), 16906ed7257bSPatrick Caulfield __alignof__(struct connection), 0, 169120c2df83SPaul Mundt NULL); 16926ed7257bSPatrick Caulfield if (!con_cache) 1693513ef596SDavid Teigland goto fail; 1694513ef596SDavid Teigland 1695513ef596SDavid Teigland error = work_start(); 1696513ef596SDavid Teigland if (error) 1697513ef596SDavid Teigland goto fail_destroy; 1698513ef596SDavid Teigland 1699513ef596SDavid Teigland dlm_allow_conn = 1; 17006ed7257bSPatrick Caulfield 17016ed7257bSPatrick Caulfield /* Start listening */ 17026ed7257bSPatrick Caulfield if (dlm_config.ci_protocol == 0) 17036ed7257bSPatrick Caulfield error = tcp_listen_for_all(); 17046ed7257bSPatrick Caulfield else 17056ed7257bSPatrick Caulfield error = sctp_listen_for_all(); 17066ed7257bSPatrick Caulfield if (error) 17076ed7257bSPatrick Caulfield goto fail_unlisten; 17086ed7257bSPatrick Caulfield 17096ed7257bSPatrick Caulfield return 0; 17106ed7257bSPatrick Caulfield 17116ed7257bSPatrick Caulfield fail_unlisten: 1712513ef596SDavid Teigland dlm_allow_conn = 0; 17136ed7257bSPatrick Caulfield con = nodeid2con(0,0); 17146ed7257bSPatrick Caulfield if (con) { 17156ed7257bSPatrick Caulfield close_connection(con, false); 17166ed7257bSPatrick Caulfield kmem_cache_free(con_cache, con); 17176ed7257bSPatrick Caulfield } 1718513ef596SDavid Teigland fail_destroy: 17196ed7257bSPatrick Caulfield kmem_cache_destroy(con_cache); 1720513ef596SDavid Teigland fail: 17216ed7257bSPatrick Caulfield return error; 17226ed7257bSPatrick Caulfield } 172336b71a8bSDavid Teigland 172436b71a8bSDavid Teigland void dlm_lowcomms_exit(void) 172536b71a8bSDavid Teigland { 172636b71a8bSDavid Teigland struct dlm_node_addr *na, *safe; 172736b71a8bSDavid Teigland 172836b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 172936b71a8bSDavid Teigland list_for_each_entry_safe(na, safe, &dlm_node_addrs, list) { 173036b71a8bSDavid Teigland list_del(&na->list); 173136b71a8bSDavid Teigland while (na->addr_count--) 173236b71a8bSDavid Teigland kfree(na->addr[na->addr_count]); 173336b71a8bSDavid Teigland kfree(na); 173436b71a8bSDavid Teigland } 173536b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 173636b71a8bSDavid Teigland } 1737