12522fe45SThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only 26ed7257bSPatrick Caulfield /****************************************************************************** 36ed7257bSPatrick Caulfield ******************************************************************************* 46ed7257bSPatrick Caulfield ** 56ed7257bSPatrick Caulfield ** Copyright (C) Sistina Software, Inc. 1997-2003 All rights reserved. 65e9ccc37SChristine Caulfield ** Copyright (C) 2004-2009 Red Hat, Inc. All rights reserved. 76ed7257bSPatrick Caulfield ** 86ed7257bSPatrick Caulfield ** 96ed7257bSPatrick Caulfield ******************************************************************************* 106ed7257bSPatrick Caulfield ******************************************************************************/ 116ed7257bSPatrick Caulfield 126ed7257bSPatrick Caulfield /* 136ed7257bSPatrick Caulfield * lowcomms.c 146ed7257bSPatrick Caulfield * 156ed7257bSPatrick Caulfield * This is the "low-level" comms layer. 166ed7257bSPatrick Caulfield * 176ed7257bSPatrick Caulfield * It is responsible for sending/receiving messages 186ed7257bSPatrick Caulfield * from other nodes in the cluster. 196ed7257bSPatrick Caulfield * 206ed7257bSPatrick Caulfield * Cluster nodes are referred to by their nodeids. nodeids are 216ed7257bSPatrick Caulfield * simply 32 bit numbers to the locking module - if they need to 222cf12c0bSJoe Perches * be expanded for the cluster infrastructure then that is its 236ed7257bSPatrick Caulfield * responsibility. It is this layer's 246ed7257bSPatrick Caulfield * responsibility to resolve these into IP address or 256ed7257bSPatrick Caulfield * whatever it needs for inter-node communication. 266ed7257bSPatrick Caulfield * 276ed7257bSPatrick Caulfield * The comms level is two kernel threads that deal mainly with 286ed7257bSPatrick Caulfield * the receiving of messages from other nodes and passing them 296ed7257bSPatrick Caulfield * up to the mid-level comms layer (which understands the 306ed7257bSPatrick Caulfield * message format) for execution by the locking core, and 316ed7257bSPatrick Caulfield * a send thread which does all the setting up of connections 326ed7257bSPatrick Caulfield * to remote nodes and the sending of data. Threads are not allowed 336ed7257bSPatrick Caulfield * to send their own data because it may cause them to wait in times 346ed7257bSPatrick Caulfield * of high load. Also, this way, the sending thread can collect together 356ed7257bSPatrick Caulfield * messages bound for one node and send them in one block. 366ed7257bSPatrick Caulfield * 372cf12c0bSJoe Perches * lowcomms will choose to use either TCP or SCTP as its transport layer 386ed7257bSPatrick Caulfield * depending on the configuration variable 'protocol'. This should be set 396ed7257bSPatrick Caulfield * to 0 (default) for TCP or 1 for SCTP. It should be configured using a 406ed7257bSPatrick Caulfield * cluster-wide mechanism as it must be the same on all nodes of the cluster 416ed7257bSPatrick Caulfield * for the DLM to function. 426ed7257bSPatrick Caulfield * 436ed7257bSPatrick Caulfield */ 446ed7257bSPatrick Caulfield 456ed7257bSPatrick Caulfield #include <asm/ioctls.h> 466ed7257bSPatrick Caulfield #include <net/sock.h> 476ed7257bSPatrick Caulfield #include <net/tcp.h> 486ed7257bSPatrick Caulfield #include <linux/pagemap.h> 496ed7257bSPatrick Caulfield #include <linux/file.h> 507a936ce7SMatthias Kaehlcke #include <linux/mutex.h> 516ed7257bSPatrick Caulfield #include <linux/sctp.h> 525a0e3ad6STejun Heo #include <linux/slab.h> 532f2d76ccSBenjamin Poirier #include <net/sctp/sctp.h> 5444ad532bSJoe Perches #include <net/ipv6.h> 556ed7257bSPatrick Caulfield 566ed7257bSPatrick Caulfield #include "dlm_internal.h" 576ed7257bSPatrick Caulfield #include "lowcomms.h" 586ed7257bSPatrick Caulfield #include "midcomms.h" 596ed7257bSPatrick Caulfield #include "config.h" 606ed7257bSPatrick Caulfield 616ed7257bSPatrick Caulfield #define NEEDED_RMEM (4*1024*1024) 625e9ccc37SChristine Caulfield #define CONN_HASH_SIZE 32 636ed7257bSPatrick Caulfield 64f92c8dd7SBob Peterson /* Number of messages to send before rescheduling */ 65f92c8dd7SBob Peterson #define MAX_SEND_MSG_COUNT 25 66055923bfSAlexander Aring #define DLM_SHUTDOWN_WAIT_TIMEOUT msecs_to_jiffies(10000) 67f92c8dd7SBob Peterson 686ed7257bSPatrick Caulfield struct connection { 696ed7257bSPatrick Caulfield struct socket *sock; /* NULL if not connected */ 706ed7257bSPatrick Caulfield uint32_t nodeid; /* So we know who we are in the list */ 716ed7257bSPatrick Caulfield struct mutex sock_mutex; 726ed7257bSPatrick Caulfield unsigned long flags; 736ed7257bSPatrick Caulfield #define CF_READ_PENDING 1 748a4abb08Stsutomu.owa@toshiba.co.jp #define CF_WRITE_PENDING 2 756ed7257bSPatrick Caulfield #define CF_INIT_PENDING 4 766ed7257bSPatrick Caulfield #define CF_IS_OTHERCON 5 77063c4c99SLars Marowsky-Bree #define CF_CLOSE 6 78b36930ddSDavid Miller #define CF_APP_LIMITED 7 79b2a66629Stsutomu.owa@toshiba.co.jp #define CF_CLOSING 8 80055923bfSAlexander Aring #define CF_SHUTDOWN 9 8119633c7eSAlexander Aring #define CF_CONNECTED 10 826ed7257bSPatrick Caulfield struct list_head writequeue; /* List of outgoing writequeue_entries */ 836ed7257bSPatrick Caulfield spinlock_t writequeue_lock; 846ed7257bSPatrick Caulfield void (*connect_action) (struct connection *); /* What to do to connect */ 85055923bfSAlexander Aring void (*shutdown_action)(struct connection *con); /* What to do to shutdown */ 866ed7257bSPatrick Caulfield int retries; 876ed7257bSPatrick Caulfield #define MAX_CONNECT_RETRIES 3 885e9ccc37SChristine Caulfield struct hlist_node list; 896ed7257bSPatrick Caulfield struct connection *othercon; 906ed7257bSPatrick Caulfield struct work_struct rwork; /* Receive workqueue */ 916ed7257bSPatrick Caulfield struct work_struct swork; /* Send workqueue */ 92055923bfSAlexander Aring wait_queue_head_t shutdown_wait; /* wait for graceful shutdown */ 934798cbbfSAlexander Aring unsigned char *rx_buf; 944798cbbfSAlexander Aring int rx_buflen; 954798cbbfSAlexander Aring int rx_leftover; 96a47666ebSAlexander Aring struct rcu_head rcu; 976ed7257bSPatrick Caulfield }; 986ed7257bSPatrick Caulfield #define sock2con(x) ((struct connection *)(x)->sk_user_data) 996ed7257bSPatrick Caulfield 100d11ccd45SAlexander Aring struct listen_connection { 101d11ccd45SAlexander Aring struct socket *sock; 102d11ccd45SAlexander Aring struct work_struct rwork; 103d11ccd45SAlexander Aring }; 104d11ccd45SAlexander Aring 105f0747ebfSAlexander Aring #define DLM_WQ_REMAIN_BYTES(e) (PAGE_SIZE - e->end) 106f0747ebfSAlexander Aring #define DLM_WQ_LENGTH_BYTES(e) (e->end - e->offset) 107f0747ebfSAlexander Aring 1086ed7257bSPatrick Caulfield /* An entry waiting to be sent */ 1096ed7257bSPatrick Caulfield struct writequeue_entry { 1106ed7257bSPatrick Caulfield struct list_head list; 1116ed7257bSPatrick Caulfield struct page *page; 1126ed7257bSPatrick Caulfield int offset; 1136ed7257bSPatrick Caulfield int len; 1146ed7257bSPatrick Caulfield int end; 1156ed7257bSPatrick Caulfield int users; 116b38bc9c2SAlexander Aring int idx; /* get()/commit() idx exchange */ 1176ed7257bSPatrick Caulfield struct connection *con; 1186ed7257bSPatrick Caulfield }; 1196ed7257bSPatrick Caulfield 12036b71a8bSDavid Teigland struct dlm_node_addr { 12136b71a8bSDavid Teigland struct list_head list; 12236b71a8bSDavid Teigland int nodeid; 123e125fbebSAlexander Aring int mark; 12436b71a8bSDavid Teigland int addr_count; 12598e1b60eSMike Christie int curr_addr_index; 12636b71a8bSDavid Teigland struct sockaddr_storage *addr[DLM_MAX_ADDR_COUNT]; 12736b71a8bSDavid Teigland }; 12836b71a8bSDavid Teigland 129cc661fc9SBob Peterson static struct listen_sock_callbacks { 130cc661fc9SBob Peterson void (*sk_error_report)(struct sock *); 131cc661fc9SBob Peterson void (*sk_data_ready)(struct sock *); 132cc661fc9SBob Peterson void (*sk_state_change)(struct sock *); 133cc661fc9SBob Peterson void (*sk_write_space)(struct sock *); 134cc661fc9SBob Peterson } listen_sock; 135cc661fc9SBob Peterson 13636b71a8bSDavid Teigland static LIST_HEAD(dlm_node_addrs); 13736b71a8bSDavid Teigland static DEFINE_SPINLOCK(dlm_node_addrs_spin); 13836b71a8bSDavid Teigland 139d11ccd45SAlexander Aring static struct listen_connection listen_con; 1406ed7257bSPatrick Caulfield static struct sockaddr_storage *dlm_local_addr[DLM_MAX_ADDR_COUNT]; 1416ed7257bSPatrick Caulfield static int dlm_local_count; 14251746163SAlexander Aring int dlm_allow_conn; 1436ed7257bSPatrick Caulfield 1446ed7257bSPatrick Caulfield /* Work queues */ 1456ed7257bSPatrick Caulfield static struct workqueue_struct *recv_workqueue; 1466ed7257bSPatrick Caulfield static struct workqueue_struct *send_workqueue; 1476ed7257bSPatrick Caulfield 1485e9ccc37SChristine Caulfield static struct hlist_head connection_hash[CONN_HASH_SIZE]; 149a47666ebSAlexander Aring static DEFINE_SPINLOCK(connections_lock); 150a47666ebSAlexander Aring DEFINE_STATIC_SRCU(connections_srcu); 1516ed7257bSPatrick Caulfield 1526ed7257bSPatrick Caulfield static void process_recv_sockets(struct work_struct *work); 1536ed7257bSPatrick Caulfield static void process_send_sockets(struct work_struct *work); 1546ed7257bSPatrick Caulfield 1550672c3c2SAlexander Aring static void sctp_connect_to_sock(struct connection *con); 1560672c3c2SAlexander Aring static void tcp_connect_to_sock(struct connection *con); 15742873c90SAlexander Aring static void dlm_tcp_shutdown(struct connection *con); 1585e9ccc37SChristine Caulfield 1595e9ccc37SChristine Caulfield /* This is deliberately very simple because most clusters have simple 1605e9ccc37SChristine Caulfield sequential nodeids, so we should be able to go straight to a connection 1615e9ccc37SChristine Caulfield struct in the array */ 1625e9ccc37SChristine Caulfield static inline int nodeid_hash(int nodeid) 1635e9ccc37SChristine Caulfield { 1645e9ccc37SChristine Caulfield return nodeid & (CONN_HASH_SIZE-1); 1655e9ccc37SChristine Caulfield } 1665e9ccc37SChristine Caulfield 167b38bc9c2SAlexander Aring static struct connection *__find_con(int nodeid, int r) 1685e9ccc37SChristine Caulfield { 1695e9ccc37SChristine Caulfield struct connection *con; 1705e9ccc37SChristine Caulfield 171a47666ebSAlexander Aring hlist_for_each_entry_rcu(con, &connection_hash[r], list) { 172b38bc9c2SAlexander Aring if (con->nodeid == nodeid) 1735e9ccc37SChristine Caulfield return con; 1745e9ccc37SChristine Caulfield } 175a47666ebSAlexander Aring 1765e9ccc37SChristine Caulfield return NULL; 1775e9ccc37SChristine Caulfield } 1785e9ccc37SChristine Caulfield 1796cde210aSAlexander Aring static int dlm_con_init(struct connection *con, int nodeid) 1806ed7257bSPatrick Caulfield { 1814798cbbfSAlexander Aring con->rx_buflen = dlm_config.ci_buffer_size; 1824798cbbfSAlexander Aring con->rx_buf = kmalloc(con->rx_buflen, GFP_NOFS); 1836cde210aSAlexander Aring if (!con->rx_buf) 1846cde210aSAlexander Aring return -ENOMEM; 1854798cbbfSAlexander Aring 1866ed7257bSPatrick Caulfield con->nodeid = nodeid; 1876ed7257bSPatrick Caulfield mutex_init(&con->sock_mutex); 1886ed7257bSPatrick Caulfield INIT_LIST_HEAD(&con->writequeue); 1896ed7257bSPatrick Caulfield spin_lock_init(&con->writequeue_lock); 1906ed7257bSPatrick Caulfield INIT_WORK(&con->swork, process_send_sockets); 1916ed7257bSPatrick Caulfield INIT_WORK(&con->rwork, process_recv_sockets); 192055923bfSAlexander Aring init_waitqueue_head(&con->shutdown_wait); 1936ed7257bSPatrick Caulfield 19442873c90SAlexander Aring if (dlm_config.ci_protocol == 0) { 1950672c3c2SAlexander Aring con->connect_action = tcp_connect_to_sock; 19642873c90SAlexander Aring con->shutdown_action = dlm_tcp_shutdown; 19742873c90SAlexander Aring } else { 1980672c3c2SAlexander Aring con->connect_action = sctp_connect_to_sock; 19942873c90SAlexander Aring } 2006ed7257bSPatrick Caulfield 2016cde210aSAlexander Aring return 0; 2026cde210aSAlexander Aring } 2036cde210aSAlexander Aring 2046cde210aSAlexander Aring /* 2056cde210aSAlexander Aring * If 'allocation' is zero then we don't attempt to create a new 2066cde210aSAlexander Aring * connection structure for this node. 2076cde210aSAlexander Aring */ 2086cde210aSAlexander Aring static struct connection *nodeid2con(int nodeid, gfp_t alloc) 2096cde210aSAlexander Aring { 2106cde210aSAlexander Aring struct connection *con, *tmp; 2116cde210aSAlexander Aring int r, ret; 2126cde210aSAlexander Aring 213b38bc9c2SAlexander Aring r = nodeid_hash(nodeid); 214b38bc9c2SAlexander Aring con = __find_con(nodeid, r); 2156cde210aSAlexander Aring if (con || !alloc) 2166cde210aSAlexander Aring return con; 2176cde210aSAlexander Aring 2186cde210aSAlexander Aring con = kzalloc(sizeof(*con), alloc); 2196cde210aSAlexander Aring if (!con) 2206cde210aSAlexander Aring return NULL; 2216cde210aSAlexander Aring 2226cde210aSAlexander Aring ret = dlm_con_init(con, nodeid); 2236cde210aSAlexander Aring if (ret) { 2246cde210aSAlexander Aring kfree(con); 2256cde210aSAlexander Aring return NULL; 2266cde210aSAlexander Aring } 2276cde210aSAlexander Aring 228a47666ebSAlexander Aring spin_lock(&connections_lock); 2294f2b30fdSAlexander Aring /* Because multiple workqueues/threads calls this function it can 2304f2b30fdSAlexander Aring * race on multiple cpu's. Instead of locking hot path __find_con() 2314f2b30fdSAlexander Aring * we just check in rare cases of recently added nodes again 2324f2b30fdSAlexander Aring * under protection of connections_lock. If this is the case we 2334f2b30fdSAlexander Aring * abort our connection creation and return the existing connection. 2344f2b30fdSAlexander Aring */ 235b38bc9c2SAlexander Aring tmp = __find_con(nodeid, r); 2364f2b30fdSAlexander Aring if (tmp) { 2374f2b30fdSAlexander Aring spin_unlock(&connections_lock); 2384f2b30fdSAlexander Aring kfree(con->rx_buf); 2394f2b30fdSAlexander Aring kfree(con); 2404f2b30fdSAlexander Aring return tmp; 2414f2b30fdSAlexander Aring } 2424f2b30fdSAlexander Aring 243a47666ebSAlexander Aring hlist_add_head_rcu(&con->list, &connection_hash[r]); 244a47666ebSAlexander Aring spin_unlock(&connections_lock); 245a47666ebSAlexander Aring 2466ed7257bSPatrick Caulfield return con; 2476ed7257bSPatrick Caulfield } 2486ed7257bSPatrick Caulfield 2495e9ccc37SChristine Caulfield /* Loop round all connections */ 2505e9ccc37SChristine Caulfield static void foreach_conn(void (*conn_func)(struct connection *c)) 2515e9ccc37SChristine Caulfield { 252b38bc9c2SAlexander Aring int i; 2535e9ccc37SChristine Caulfield struct connection *con; 2545e9ccc37SChristine Caulfield 2555e9ccc37SChristine Caulfield for (i = 0; i < CONN_HASH_SIZE; i++) { 256a47666ebSAlexander Aring hlist_for_each_entry_rcu(con, &connection_hash[i], list) 2575e9ccc37SChristine Caulfield conn_func(con); 2585e9ccc37SChristine Caulfield } 2596ed7257bSPatrick Caulfield } 2606ed7257bSPatrick Caulfield 26136b71a8bSDavid Teigland static struct dlm_node_addr *find_node_addr(int nodeid) 2626ed7257bSPatrick Caulfield { 26336b71a8bSDavid Teigland struct dlm_node_addr *na; 26436b71a8bSDavid Teigland 26536b71a8bSDavid Teigland list_for_each_entry(na, &dlm_node_addrs, list) { 26636b71a8bSDavid Teigland if (na->nodeid == nodeid) 26736b71a8bSDavid Teigland return na; 26836b71a8bSDavid Teigland } 26936b71a8bSDavid Teigland return NULL; 27036b71a8bSDavid Teigland } 27136b71a8bSDavid Teigland 27240c6b83eSAlexander Aring static int addr_compare(const struct sockaddr_storage *x, 27340c6b83eSAlexander Aring const struct sockaddr_storage *y) 27436b71a8bSDavid Teigland { 27536b71a8bSDavid Teigland switch (x->ss_family) { 27636b71a8bSDavid Teigland case AF_INET: { 27736b71a8bSDavid Teigland struct sockaddr_in *sinx = (struct sockaddr_in *)x; 27836b71a8bSDavid Teigland struct sockaddr_in *siny = (struct sockaddr_in *)y; 27936b71a8bSDavid Teigland if (sinx->sin_addr.s_addr != siny->sin_addr.s_addr) 28036b71a8bSDavid Teigland return 0; 28136b71a8bSDavid Teigland if (sinx->sin_port != siny->sin_port) 28236b71a8bSDavid Teigland return 0; 28336b71a8bSDavid Teigland break; 28436b71a8bSDavid Teigland } 28536b71a8bSDavid Teigland case AF_INET6: { 28636b71a8bSDavid Teigland struct sockaddr_in6 *sinx = (struct sockaddr_in6 *)x; 28736b71a8bSDavid Teigland struct sockaddr_in6 *siny = (struct sockaddr_in6 *)y; 28836b71a8bSDavid Teigland if (!ipv6_addr_equal(&sinx->sin6_addr, &siny->sin6_addr)) 28936b71a8bSDavid Teigland return 0; 29036b71a8bSDavid Teigland if (sinx->sin6_port != siny->sin6_port) 29136b71a8bSDavid Teigland return 0; 29236b71a8bSDavid Teigland break; 29336b71a8bSDavid Teigland } 29436b71a8bSDavid Teigland default: 29536b71a8bSDavid Teigland return 0; 29636b71a8bSDavid Teigland } 29736b71a8bSDavid Teigland return 1; 29836b71a8bSDavid Teigland } 29936b71a8bSDavid Teigland 30036b71a8bSDavid Teigland static int nodeid_to_addr(int nodeid, struct sockaddr_storage *sas_out, 301e125fbebSAlexander Aring struct sockaddr *sa_out, bool try_new_addr, 302e125fbebSAlexander Aring unsigned int *mark) 30336b71a8bSDavid Teigland { 30436b71a8bSDavid Teigland struct sockaddr_storage sas; 30536b71a8bSDavid Teigland struct dlm_node_addr *na; 3066ed7257bSPatrick Caulfield 3076ed7257bSPatrick Caulfield if (!dlm_local_count) 3086ed7257bSPatrick Caulfield return -1; 3096ed7257bSPatrick Caulfield 31036b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 31136b71a8bSDavid Teigland na = find_node_addr(nodeid); 31298e1b60eSMike Christie if (na && na->addr_count) { 313ee44b4bcSMarcelo Ricardo Leitner memcpy(&sas, na->addr[na->curr_addr_index], 314ee44b4bcSMarcelo Ricardo Leitner sizeof(struct sockaddr_storage)); 315ee44b4bcSMarcelo Ricardo Leitner 31698e1b60eSMike Christie if (try_new_addr) { 31798e1b60eSMike Christie na->curr_addr_index++; 31898e1b60eSMike Christie if (na->curr_addr_index == na->addr_count) 31998e1b60eSMike Christie na->curr_addr_index = 0; 32098e1b60eSMike Christie } 32198e1b60eSMike Christie } 32236b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 32336b71a8bSDavid Teigland 32436b71a8bSDavid Teigland if (!na) 32536b71a8bSDavid Teigland return -EEXIST; 32636b71a8bSDavid Teigland 32736b71a8bSDavid Teigland if (!na->addr_count) 32836b71a8bSDavid Teigland return -ENOENT; 32936b71a8bSDavid Teigland 330e125fbebSAlexander Aring *mark = na->mark; 331e125fbebSAlexander Aring 33236b71a8bSDavid Teigland if (sas_out) 33336b71a8bSDavid Teigland memcpy(sas_out, &sas, sizeof(struct sockaddr_storage)); 33436b71a8bSDavid Teigland 33536b71a8bSDavid Teigland if (!sa_out) 33636b71a8bSDavid Teigland return 0; 3376ed7257bSPatrick Caulfield 3386ed7257bSPatrick Caulfield if (dlm_local_addr[0]->ss_family == AF_INET) { 33936b71a8bSDavid Teigland struct sockaddr_in *in4 = (struct sockaddr_in *) &sas; 34036b71a8bSDavid Teigland struct sockaddr_in *ret4 = (struct sockaddr_in *) sa_out; 3416ed7257bSPatrick Caulfield ret4->sin_addr.s_addr = in4->sin_addr.s_addr; 3426ed7257bSPatrick Caulfield } else { 34336b71a8bSDavid Teigland struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) &sas; 34436b71a8bSDavid Teigland struct sockaddr_in6 *ret6 = (struct sockaddr_in6 *) sa_out; 3454e3fd7a0SAlexey Dobriyan ret6->sin6_addr = in6->sin6_addr; 3466ed7257bSPatrick Caulfield } 3476ed7257bSPatrick Caulfield 3486ed7257bSPatrick Caulfield return 0; 3496ed7257bSPatrick Caulfield } 3506ed7257bSPatrick Caulfield 351e125fbebSAlexander Aring static int addr_to_nodeid(struct sockaddr_storage *addr, int *nodeid, 352e125fbebSAlexander Aring unsigned int *mark) 35336b71a8bSDavid Teigland { 35436b71a8bSDavid Teigland struct dlm_node_addr *na; 35536b71a8bSDavid Teigland int rv = -EEXIST; 35698e1b60eSMike Christie int addr_i; 35736b71a8bSDavid Teigland 35836b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 35936b71a8bSDavid Teigland list_for_each_entry(na, &dlm_node_addrs, list) { 36036b71a8bSDavid Teigland if (!na->addr_count) 36136b71a8bSDavid Teigland continue; 36236b71a8bSDavid Teigland 36398e1b60eSMike Christie for (addr_i = 0; addr_i < na->addr_count; addr_i++) { 36498e1b60eSMike Christie if (addr_compare(na->addr[addr_i], addr)) { 36536b71a8bSDavid Teigland *nodeid = na->nodeid; 366e125fbebSAlexander Aring *mark = na->mark; 36736b71a8bSDavid Teigland rv = 0; 36898e1b60eSMike Christie goto unlock; 36936b71a8bSDavid Teigland } 37098e1b60eSMike Christie } 37198e1b60eSMike Christie } 37298e1b60eSMike Christie unlock: 37336b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 37436b71a8bSDavid Teigland return rv; 37536b71a8bSDavid Teigland } 37636b71a8bSDavid Teigland 3774f19d071SAlexander Aring /* caller need to held dlm_node_addrs_spin lock */ 3784f19d071SAlexander Aring static bool dlm_lowcomms_na_has_addr(const struct dlm_node_addr *na, 3794f19d071SAlexander Aring const struct sockaddr_storage *addr) 3804f19d071SAlexander Aring { 3814f19d071SAlexander Aring int i; 3824f19d071SAlexander Aring 3834f19d071SAlexander Aring for (i = 0; i < na->addr_count; i++) { 3844f19d071SAlexander Aring if (addr_compare(na->addr[i], addr)) 3854f19d071SAlexander Aring return true; 3864f19d071SAlexander Aring } 3874f19d071SAlexander Aring 3884f19d071SAlexander Aring return false; 3894f19d071SAlexander Aring } 3904f19d071SAlexander Aring 39136b71a8bSDavid Teigland int dlm_lowcomms_addr(int nodeid, struct sockaddr_storage *addr, int len) 39236b71a8bSDavid Teigland { 39336b71a8bSDavid Teigland struct sockaddr_storage *new_addr; 39436b71a8bSDavid Teigland struct dlm_node_addr *new_node, *na; 3954f19d071SAlexander Aring bool ret; 39636b71a8bSDavid Teigland 39736b71a8bSDavid Teigland new_node = kzalloc(sizeof(struct dlm_node_addr), GFP_NOFS); 39836b71a8bSDavid Teigland if (!new_node) 39936b71a8bSDavid Teigland return -ENOMEM; 40036b71a8bSDavid Teigland 40136b71a8bSDavid Teigland new_addr = kzalloc(sizeof(struct sockaddr_storage), GFP_NOFS); 40236b71a8bSDavid Teigland if (!new_addr) { 40336b71a8bSDavid Teigland kfree(new_node); 40436b71a8bSDavid Teigland return -ENOMEM; 40536b71a8bSDavid Teigland } 40636b71a8bSDavid Teigland 40736b71a8bSDavid Teigland memcpy(new_addr, addr, len); 40836b71a8bSDavid Teigland 40936b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 41036b71a8bSDavid Teigland na = find_node_addr(nodeid); 41136b71a8bSDavid Teigland if (!na) { 41236b71a8bSDavid Teigland new_node->nodeid = nodeid; 41336b71a8bSDavid Teigland new_node->addr[0] = new_addr; 41436b71a8bSDavid Teigland new_node->addr_count = 1; 415e125fbebSAlexander Aring new_node->mark = dlm_config.ci_mark; 41636b71a8bSDavid Teigland list_add(&new_node->list, &dlm_node_addrs); 41736b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 41836b71a8bSDavid Teigland return 0; 41936b71a8bSDavid Teigland } 42036b71a8bSDavid Teigland 4214f19d071SAlexander Aring ret = dlm_lowcomms_na_has_addr(na, addr); 4224f19d071SAlexander Aring if (ret) { 4234f19d071SAlexander Aring spin_unlock(&dlm_node_addrs_spin); 4244f19d071SAlexander Aring kfree(new_addr); 4254f19d071SAlexander Aring kfree(new_node); 4264f19d071SAlexander Aring return -EEXIST; 4274f19d071SAlexander Aring } 4284f19d071SAlexander Aring 42936b71a8bSDavid Teigland if (na->addr_count >= DLM_MAX_ADDR_COUNT) { 43036b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 43136b71a8bSDavid Teigland kfree(new_addr); 43236b71a8bSDavid Teigland kfree(new_node); 43336b71a8bSDavid Teigland return -ENOSPC; 43436b71a8bSDavid Teigland } 43536b71a8bSDavid Teigland 43636b71a8bSDavid Teigland na->addr[na->addr_count++] = new_addr; 43736b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 43836b71a8bSDavid Teigland kfree(new_node); 43936b71a8bSDavid Teigland return 0; 44036b71a8bSDavid Teigland } 44136b71a8bSDavid Teigland 4426ed7257bSPatrick Caulfield /* Data available on socket or listen socket received a connect */ 443676d2369SDavid S. Miller static void lowcomms_data_ready(struct sock *sk) 4446ed7257bSPatrick Caulfield { 44593eaadebStsutomu.owa@toshiba.co.jp struct connection *con; 44693eaadebStsutomu.owa@toshiba.co.jp 44793eaadebStsutomu.owa@toshiba.co.jp read_lock_bh(&sk->sk_callback_lock); 44893eaadebStsutomu.owa@toshiba.co.jp con = sock2con(sk); 449afb853fbSPatrick Caulfield if (con && !test_and_set_bit(CF_READ_PENDING, &con->flags)) 4506ed7257bSPatrick Caulfield queue_work(recv_workqueue, &con->rwork); 45193eaadebStsutomu.owa@toshiba.co.jp read_unlock_bh(&sk->sk_callback_lock); 4526ed7257bSPatrick Caulfield } 4536ed7257bSPatrick Caulfield 454d11ccd45SAlexander Aring static void lowcomms_listen_data_ready(struct sock *sk) 455d11ccd45SAlexander Aring { 456d11ccd45SAlexander Aring queue_work(recv_workqueue, &listen_con.rwork); 457d11ccd45SAlexander Aring } 458d11ccd45SAlexander Aring 4596ed7257bSPatrick Caulfield static void lowcomms_write_space(struct sock *sk) 4606ed7257bSPatrick Caulfield { 46193eaadebStsutomu.owa@toshiba.co.jp struct connection *con; 4626ed7257bSPatrick Caulfield 46393eaadebStsutomu.owa@toshiba.co.jp read_lock_bh(&sk->sk_callback_lock); 46493eaadebStsutomu.owa@toshiba.co.jp con = sock2con(sk); 465b36930ddSDavid Miller if (!con) 46693eaadebStsutomu.owa@toshiba.co.jp goto out; 467b36930ddSDavid Miller 46819633c7eSAlexander Aring if (!test_and_set_bit(CF_CONNECTED, &con->flags)) { 46919633c7eSAlexander Aring log_print("successful connected to node %d", con->nodeid); 47019633c7eSAlexander Aring queue_work(send_workqueue, &con->swork); 47119633c7eSAlexander Aring goto out; 47219633c7eSAlexander Aring } 47319633c7eSAlexander Aring 474b36930ddSDavid Miller clear_bit(SOCK_NOSPACE, &con->sock->flags); 475b36930ddSDavid Miller 476b36930ddSDavid Miller if (test_and_clear_bit(CF_APP_LIMITED, &con->flags)) { 477b36930ddSDavid Miller con->sock->sk->sk_write_pending--; 4789cd3e072SEric Dumazet clear_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags); 479b36930ddSDavid Miller } 480b36930ddSDavid Miller 4816ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 48293eaadebStsutomu.owa@toshiba.co.jp out: 48393eaadebStsutomu.owa@toshiba.co.jp read_unlock_bh(&sk->sk_callback_lock); 4846ed7257bSPatrick Caulfield } 4856ed7257bSPatrick Caulfield 4866ed7257bSPatrick Caulfield static inline void lowcomms_connect_sock(struct connection *con) 4876ed7257bSPatrick Caulfield { 488063c4c99SLars Marowsky-Bree if (test_bit(CF_CLOSE, &con->flags)) 489063c4c99SLars Marowsky-Bree return; 4906ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 49161d9102bSBob Peterson cond_resched(); 4926ed7257bSPatrick Caulfield } 4936ed7257bSPatrick Caulfield 4946ed7257bSPatrick Caulfield static void lowcomms_state_change(struct sock *sk) 4956ed7257bSPatrick Caulfield { 496ee44b4bcSMarcelo Ricardo Leitner /* SCTP layer is not calling sk_data_ready when the connection 497ee44b4bcSMarcelo Ricardo Leitner * is done, so we catch the signal through here. Also, it 498ee44b4bcSMarcelo Ricardo Leitner * doesn't switch socket state when entering shutdown, so we 499ee44b4bcSMarcelo Ricardo Leitner * skip the write in that case. 500ee44b4bcSMarcelo Ricardo Leitner */ 501ee44b4bcSMarcelo Ricardo Leitner if (sk->sk_shutdown) { 502ee44b4bcSMarcelo Ricardo Leitner if (sk->sk_shutdown == RCV_SHUTDOWN) 503ee44b4bcSMarcelo Ricardo Leitner lowcomms_data_ready(sk); 504ee44b4bcSMarcelo Ricardo Leitner } else if (sk->sk_state == TCP_ESTABLISHED) { 5056ed7257bSPatrick Caulfield lowcomms_write_space(sk); 5066ed7257bSPatrick Caulfield } 507ee44b4bcSMarcelo Ricardo Leitner } 5086ed7257bSPatrick Caulfield 509391fbdc5SChristine Caulfield int dlm_lowcomms_connect_node(int nodeid) 510391fbdc5SChristine Caulfield { 511391fbdc5SChristine Caulfield struct connection *con; 512b38bc9c2SAlexander Aring int idx; 513391fbdc5SChristine Caulfield 514391fbdc5SChristine Caulfield if (nodeid == dlm_our_nodeid()) 515391fbdc5SChristine Caulfield return 0; 516391fbdc5SChristine Caulfield 517b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 518391fbdc5SChristine Caulfield con = nodeid2con(nodeid, GFP_NOFS); 519b38bc9c2SAlexander Aring if (!con) { 520b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 521391fbdc5SChristine Caulfield return -ENOMEM; 522b38bc9c2SAlexander Aring } 523b38bc9c2SAlexander Aring 524391fbdc5SChristine Caulfield lowcomms_connect_sock(con); 525b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 526b38bc9c2SAlexander Aring 527391fbdc5SChristine Caulfield return 0; 528391fbdc5SChristine Caulfield } 529391fbdc5SChristine Caulfield 530e125fbebSAlexander Aring int dlm_lowcomms_nodes_set_mark(int nodeid, unsigned int mark) 531e125fbebSAlexander Aring { 532e125fbebSAlexander Aring struct dlm_node_addr *na; 533e125fbebSAlexander Aring 534e125fbebSAlexander Aring spin_lock(&dlm_node_addrs_spin); 535e125fbebSAlexander Aring na = find_node_addr(nodeid); 536e125fbebSAlexander Aring if (!na) { 537e125fbebSAlexander Aring spin_unlock(&dlm_node_addrs_spin); 538e125fbebSAlexander Aring return -ENOENT; 539e125fbebSAlexander Aring } 540e125fbebSAlexander Aring 541e125fbebSAlexander Aring na->mark = mark; 542e125fbebSAlexander Aring spin_unlock(&dlm_node_addrs_spin); 543e125fbebSAlexander Aring 544e125fbebSAlexander Aring return 0; 545e125fbebSAlexander Aring } 546e125fbebSAlexander Aring 547b3a5bbfdSBob Peterson static void lowcomms_error_report(struct sock *sk) 548b3a5bbfdSBob Peterson { 549b81171cbSBob Peterson struct connection *con; 550b3a5bbfdSBob Peterson struct sockaddr_storage saddr; 551b81171cbSBob Peterson void (*orig_report)(struct sock *) = NULL; 552b3a5bbfdSBob Peterson 553b81171cbSBob Peterson read_lock_bh(&sk->sk_callback_lock); 554b81171cbSBob Peterson con = sock2con(sk); 555b81171cbSBob Peterson if (con == NULL) 556b81171cbSBob Peterson goto out; 557b81171cbSBob Peterson 558cc661fc9SBob Peterson orig_report = listen_sock.sk_error_report; 5591a31833dSBob Peterson if (con->sock == NULL || 5609b2c45d4SDenys Vlasenko kernel_getpeername(con->sock, (struct sockaddr *)&saddr) < 0) { 561b3a5bbfdSBob Peterson printk_ratelimited(KERN_ERR "dlm: node %d: socket error " 562b3a5bbfdSBob Peterson "sending to node %d, port %d, " 563b3a5bbfdSBob Peterson "sk_err=%d/%d\n", dlm_our_nodeid(), 564b3a5bbfdSBob Peterson con->nodeid, dlm_config.ci_tcp_port, 565b3a5bbfdSBob Peterson sk->sk_err, sk->sk_err_soft); 566b3a5bbfdSBob Peterson } else if (saddr.ss_family == AF_INET) { 567b3a5bbfdSBob Peterson struct sockaddr_in *sin4 = (struct sockaddr_in *)&saddr; 568b3a5bbfdSBob Peterson 569b3a5bbfdSBob Peterson printk_ratelimited(KERN_ERR "dlm: node %d: socket error " 570b3a5bbfdSBob Peterson "sending to node %d at %pI4, port %d, " 571b3a5bbfdSBob Peterson "sk_err=%d/%d\n", dlm_our_nodeid(), 572b3a5bbfdSBob Peterson con->nodeid, &sin4->sin_addr.s_addr, 573b3a5bbfdSBob Peterson dlm_config.ci_tcp_port, sk->sk_err, 574b3a5bbfdSBob Peterson sk->sk_err_soft); 575b3a5bbfdSBob Peterson } else { 576b3a5bbfdSBob Peterson struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)&saddr; 577b3a5bbfdSBob Peterson 578b3a5bbfdSBob Peterson printk_ratelimited(KERN_ERR "dlm: node %d: socket error " 579b3a5bbfdSBob Peterson "sending to node %d at %u.%u.%u.%u, " 580b3a5bbfdSBob Peterson "port %d, sk_err=%d/%d\n", dlm_our_nodeid(), 581b3a5bbfdSBob Peterson con->nodeid, sin6->sin6_addr.s6_addr32[0], 582b3a5bbfdSBob Peterson sin6->sin6_addr.s6_addr32[1], 583b3a5bbfdSBob Peterson sin6->sin6_addr.s6_addr32[2], 584b3a5bbfdSBob Peterson sin6->sin6_addr.s6_addr32[3], 585b3a5bbfdSBob Peterson dlm_config.ci_tcp_port, sk->sk_err, 586b3a5bbfdSBob Peterson sk->sk_err_soft); 587b3a5bbfdSBob Peterson } 588b81171cbSBob Peterson out: 589b81171cbSBob Peterson read_unlock_bh(&sk->sk_callback_lock); 590b81171cbSBob Peterson if (orig_report) 591b81171cbSBob Peterson orig_report(sk); 592b81171cbSBob Peterson } 593b81171cbSBob Peterson 594b81171cbSBob Peterson /* Note: sk_callback_lock must be locked before calling this function. */ 595cc661fc9SBob Peterson static void save_listen_callbacks(struct socket *sock) 596b81171cbSBob Peterson { 597cc661fc9SBob Peterson struct sock *sk = sock->sk; 598cc661fc9SBob Peterson 599cc661fc9SBob Peterson listen_sock.sk_data_ready = sk->sk_data_ready; 600cc661fc9SBob Peterson listen_sock.sk_state_change = sk->sk_state_change; 601cc661fc9SBob Peterson listen_sock.sk_write_space = sk->sk_write_space; 602cc661fc9SBob Peterson listen_sock.sk_error_report = sk->sk_error_report; 603b81171cbSBob Peterson } 604b81171cbSBob Peterson 605cc661fc9SBob Peterson static void restore_callbacks(struct socket *sock) 606b81171cbSBob Peterson { 607cc661fc9SBob Peterson struct sock *sk = sock->sk; 608cc661fc9SBob Peterson 609b81171cbSBob Peterson write_lock_bh(&sk->sk_callback_lock); 610b81171cbSBob Peterson sk->sk_user_data = NULL; 611cc661fc9SBob Peterson sk->sk_data_ready = listen_sock.sk_data_ready; 612cc661fc9SBob Peterson sk->sk_state_change = listen_sock.sk_state_change; 613cc661fc9SBob Peterson sk->sk_write_space = listen_sock.sk_write_space; 614cc661fc9SBob Peterson sk->sk_error_report = listen_sock.sk_error_report; 615b81171cbSBob Peterson write_unlock_bh(&sk->sk_callback_lock); 616b3a5bbfdSBob Peterson } 617b3a5bbfdSBob Peterson 618d11ccd45SAlexander Aring static void add_listen_sock(struct socket *sock, struct listen_connection *con) 619d11ccd45SAlexander Aring { 620d11ccd45SAlexander Aring struct sock *sk = sock->sk; 621d11ccd45SAlexander Aring 622d11ccd45SAlexander Aring write_lock_bh(&sk->sk_callback_lock); 623d11ccd45SAlexander Aring save_listen_callbacks(sock); 624d11ccd45SAlexander Aring con->sock = sock; 625d11ccd45SAlexander Aring 626d11ccd45SAlexander Aring sk->sk_user_data = con; 627d11ccd45SAlexander Aring sk->sk_allocation = GFP_NOFS; 628d11ccd45SAlexander Aring /* Install a data_ready callback */ 629d11ccd45SAlexander Aring sk->sk_data_ready = lowcomms_listen_data_ready; 630d11ccd45SAlexander Aring write_unlock_bh(&sk->sk_callback_lock); 631d11ccd45SAlexander Aring } 632d11ccd45SAlexander Aring 6336ed7257bSPatrick Caulfield /* Make a socket active */ 634988419a9Stsutomu.owa@toshiba.co.jp static void add_sock(struct socket *sock, struct connection *con) 6356ed7257bSPatrick Caulfield { 636b81171cbSBob Peterson struct sock *sk = sock->sk; 637b81171cbSBob Peterson 638b81171cbSBob Peterson write_lock_bh(&sk->sk_callback_lock); 6396ed7257bSPatrick Caulfield con->sock = sock; 6406ed7257bSPatrick Caulfield 641b81171cbSBob Peterson sk->sk_user_data = con; 6426ed7257bSPatrick Caulfield /* Install a data_ready callback */ 643b81171cbSBob Peterson sk->sk_data_ready = lowcomms_data_ready; 644b81171cbSBob Peterson sk->sk_write_space = lowcomms_write_space; 645b81171cbSBob Peterson sk->sk_state_change = lowcomms_state_change; 646b81171cbSBob Peterson sk->sk_allocation = GFP_NOFS; 647b81171cbSBob Peterson sk->sk_error_report = lowcomms_error_report; 648b81171cbSBob Peterson write_unlock_bh(&sk->sk_callback_lock); 6496ed7257bSPatrick Caulfield } 6506ed7257bSPatrick Caulfield 6516ed7257bSPatrick Caulfield /* Add the port number to an IPv6 or 4 sockaddr and return the address 6526ed7257bSPatrick Caulfield length */ 6536ed7257bSPatrick Caulfield static void make_sockaddr(struct sockaddr_storage *saddr, uint16_t port, 6546ed7257bSPatrick Caulfield int *addr_len) 6556ed7257bSPatrick Caulfield { 6566ed7257bSPatrick Caulfield saddr->ss_family = dlm_local_addr[0]->ss_family; 6576ed7257bSPatrick Caulfield if (saddr->ss_family == AF_INET) { 6586ed7257bSPatrick Caulfield struct sockaddr_in *in4_addr = (struct sockaddr_in *)saddr; 6596ed7257bSPatrick Caulfield in4_addr->sin_port = cpu_to_be16(port); 6606ed7257bSPatrick Caulfield *addr_len = sizeof(struct sockaddr_in); 6616ed7257bSPatrick Caulfield memset(&in4_addr->sin_zero, 0, sizeof(in4_addr->sin_zero)); 6626ed7257bSPatrick Caulfield } else { 6636ed7257bSPatrick Caulfield struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)saddr; 6646ed7257bSPatrick Caulfield in6_addr->sin6_port = cpu_to_be16(port); 6656ed7257bSPatrick Caulfield *addr_len = sizeof(struct sockaddr_in6); 6666ed7257bSPatrick Caulfield } 66701c8cab2SPatrick Caulfield memset((char *)saddr + *addr_len, 0, sizeof(struct sockaddr_storage) - *addr_len); 6686ed7257bSPatrick Caulfield } 6696ed7257bSPatrick Caulfield 670d11ccd45SAlexander Aring static void dlm_close_sock(struct socket **sock) 671d11ccd45SAlexander Aring { 672d11ccd45SAlexander Aring if (*sock) { 673d11ccd45SAlexander Aring restore_callbacks(*sock); 674d11ccd45SAlexander Aring sock_release(*sock); 675d11ccd45SAlexander Aring *sock = NULL; 676d11ccd45SAlexander Aring } 677d11ccd45SAlexander Aring } 678d11ccd45SAlexander Aring 6796ed7257bSPatrick Caulfield /* Close a remote connection and tidy up */ 6800d737a8cSMarcelo Ricardo Leitner static void close_connection(struct connection *con, bool and_other, 6810d737a8cSMarcelo Ricardo Leitner bool tx, bool rx) 6826ed7257bSPatrick Caulfield { 683b2a66629Stsutomu.owa@toshiba.co.jp bool closing = test_and_set_bit(CF_CLOSING, &con->flags); 684b2a66629Stsutomu.owa@toshiba.co.jp 6850aa18464Stsutomu.owa@toshiba.co.jp if (tx && !closing && cancel_work_sync(&con->swork)) { 6860d737a8cSMarcelo Ricardo Leitner log_print("canceled swork for node %d", con->nodeid); 6870aa18464Stsutomu.owa@toshiba.co.jp clear_bit(CF_WRITE_PENDING, &con->flags); 6880aa18464Stsutomu.owa@toshiba.co.jp } 6890aa18464Stsutomu.owa@toshiba.co.jp if (rx && !closing && cancel_work_sync(&con->rwork)) { 6900d737a8cSMarcelo Ricardo Leitner log_print("canceled rwork for node %d", con->nodeid); 6910aa18464Stsutomu.owa@toshiba.co.jp clear_bit(CF_READ_PENDING, &con->flags); 6920aa18464Stsutomu.owa@toshiba.co.jp } 6936ed7257bSPatrick Caulfield 6940d737a8cSMarcelo Ricardo Leitner mutex_lock(&con->sock_mutex); 695d11ccd45SAlexander Aring dlm_close_sock(&con->sock); 696d11ccd45SAlexander Aring 6976ed7257bSPatrick Caulfield if (con->othercon && and_other) { 6986ed7257bSPatrick Caulfield /* Will only re-enter once. */ 6990d737a8cSMarcelo Ricardo Leitner close_connection(con->othercon, false, true, true); 7006ed7257bSPatrick Caulfield } 7019e5f2825SPatrick Caulfield 7024798cbbfSAlexander Aring con->rx_leftover = 0; 7036ed7257bSPatrick Caulfield con->retries = 0; 70419633c7eSAlexander Aring clear_bit(CF_CONNECTED, &con->flags); 7056ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 706b2a66629Stsutomu.owa@toshiba.co.jp clear_bit(CF_CLOSING, &con->flags); 7076ed7257bSPatrick Caulfield } 7086ed7257bSPatrick Caulfield 709055923bfSAlexander Aring static void shutdown_connection(struct connection *con) 710055923bfSAlexander Aring { 711055923bfSAlexander Aring int ret; 712055923bfSAlexander Aring 713eec054b5SAlexander Aring flush_work(&con->swork); 714055923bfSAlexander Aring 715055923bfSAlexander Aring mutex_lock(&con->sock_mutex); 716055923bfSAlexander Aring /* nothing to shutdown */ 717055923bfSAlexander Aring if (!con->sock) { 718055923bfSAlexander Aring mutex_unlock(&con->sock_mutex); 719055923bfSAlexander Aring return; 720055923bfSAlexander Aring } 721055923bfSAlexander Aring 722055923bfSAlexander Aring set_bit(CF_SHUTDOWN, &con->flags); 723055923bfSAlexander Aring ret = kernel_sock_shutdown(con->sock, SHUT_WR); 724055923bfSAlexander Aring mutex_unlock(&con->sock_mutex); 725055923bfSAlexander Aring if (ret) { 726055923bfSAlexander Aring log_print("Connection %p failed to shutdown: %d will force close", 727055923bfSAlexander Aring con, ret); 728055923bfSAlexander Aring goto force_close; 729055923bfSAlexander Aring } else { 730055923bfSAlexander Aring ret = wait_event_timeout(con->shutdown_wait, 731055923bfSAlexander Aring !test_bit(CF_SHUTDOWN, &con->flags), 732055923bfSAlexander Aring DLM_SHUTDOWN_WAIT_TIMEOUT); 733055923bfSAlexander Aring if (ret == 0) { 734055923bfSAlexander Aring log_print("Connection %p shutdown timed out, will force close", 735055923bfSAlexander Aring con); 736055923bfSAlexander Aring goto force_close; 737055923bfSAlexander Aring } 738055923bfSAlexander Aring } 739055923bfSAlexander Aring 740055923bfSAlexander Aring return; 741055923bfSAlexander Aring 742055923bfSAlexander Aring force_close: 743055923bfSAlexander Aring clear_bit(CF_SHUTDOWN, &con->flags); 744055923bfSAlexander Aring close_connection(con, false, true, true); 745055923bfSAlexander Aring } 746055923bfSAlexander Aring 747055923bfSAlexander Aring static void dlm_tcp_shutdown(struct connection *con) 748055923bfSAlexander Aring { 749055923bfSAlexander Aring if (con->othercon) 750055923bfSAlexander Aring shutdown_connection(con->othercon); 751055923bfSAlexander Aring shutdown_connection(con); 752055923bfSAlexander Aring } 753055923bfSAlexander Aring 7544798cbbfSAlexander Aring static int con_realloc_receive_buf(struct connection *con, int newlen) 7554798cbbfSAlexander Aring { 7564798cbbfSAlexander Aring unsigned char *newbuf; 7574798cbbfSAlexander Aring 7584798cbbfSAlexander Aring newbuf = kmalloc(newlen, GFP_NOFS); 7594798cbbfSAlexander Aring if (!newbuf) 7604798cbbfSAlexander Aring return -ENOMEM; 7614798cbbfSAlexander Aring 7624798cbbfSAlexander Aring /* copy any leftover from last receive */ 7634798cbbfSAlexander Aring if (con->rx_leftover) 7644798cbbfSAlexander Aring memmove(newbuf, con->rx_buf, con->rx_leftover); 7654798cbbfSAlexander Aring 7664798cbbfSAlexander Aring /* swap to new buffer space */ 7674798cbbfSAlexander Aring kfree(con->rx_buf); 7684798cbbfSAlexander Aring con->rx_buflen = newlen; 7694798cbbfSAlexander Aring con->rx_buf = newbuf; 7704798cbbfSAlexander Aring 7714798cbbfSAlexander Aring return 0; 7724798cbbfSAlexander Aring } 7734798cbbfSAlexander Aring 7746ed7257bSPatrick Caulfield /* Data received from remote end */ 7756ed7257bSPatrick Caulfield static int receive_from_sock(struct connection *con) 7766ed7257bSPatrick Caulfield { 7776ed7257bSPatrick Caulfield int call_again_soon = 0; 7784798cbbfSAlexander Aring struct msghdr msg; 7794798cbbfSAlexander Aring struct kvec iov; 7804798cbbfSAlexander Aring int ret, buflen; 7816ed7257bSPatrick Caulfield 7826ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 7836ed7257bSPatrick Caulfield 7846ed7257bSPatrick Caulfield if (con->sock == NULL) { 7856ed7257bSPatrick Caulfield ret = -EAGAIN; 7866ed7257bSPatrick Caulfield goto out_close; 7876ed7257bSPatrick Caulfield } 7884798cbbfSAlexander Aring 7894798cbbfSAlexander Aring /* realloc if we get new buffer size to read out */ 7904798cbbfSAlexander Aring buflen = dlm_config.ci_buffer_size; 7914798cbbfSAlexander Aring if (con->rx_buflen != buflen && con->rx_leftover <= buflen) { 7924798cbbfSAlexander Aring ret = con_realloc_receive_buf(con, buflen); 7934798cbbfSAlexander Aring if (ret < 0) 7946ed7257bSPatrick Caulfield goto out_resched; 7956ed7257bSPatrick Caulfield } 7966ed7257bSPatrick Caulfield 7974798cbbfSAlexander Aring /* calculate new buffer parameter regarding last receive and 7984798cbbfSAlexander Aring * possible leftover bytes 7996ed7257bSPatrick Caulfield */ 8004798cbbfSAlexander Aring iov.iov_base = con->rx_buf + con->rx_leftover; 8014798cbbfSAlexander Aring iov.iov_len = con->rx_buflen - con->rx_leftover; 8026ed7257bSPatrick Caulfield 8034798cbbfSAlexander Aring memset(&msg, 0, sizeof(msg)); 8044798cbbfSAlexander Aring msg.msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL; 8054798cbbfSAlexander Aring ret = kernel_recvmsg(con->sock, &msg, &iov, 1, iov.iov_len, 8064798cbbfSAlexander Aring msg.msg_flags); 8076ed7257bSPatrick Caulfield if (ret <= 0) 8086ed7257bSPatrick Caulfield goto out_close; 8094798cbbfSAlexander Aring else if (ret == iov.iov_len) 810ee44b4bcSMarcelo Ricardo Leitner call_again_soon = 1; 8116ed7257bSPatrick Caulfield 8124798cbbfSAlexander Aring /* new buflen according readed bytes and leftover from last receive */ 8134798cbbfSAlexander Aring buflen = ret + con->rx_leftover; 8144798cbbfSAlexander Aring ret = dlm_process_incoming_buffer(con->nodeid, con->rx_buf, buflen); 8154798cbbfSAlexander Aring if (ret < 0) 8164798cbbfSAlexander Aring goto out_close; 8176ed7257bSPatrick Caulfield 8184798cbbfSAlexander Aring /* calculate leftover bytes from process and put it into begin of 8194798cbbfSAlexander Aring * the receive buffer, so next receive we have the full message 8204798cbbfSAlexander Aring * at the start address of the receive buffer. 8214798cbbfSAlexander Aring */ 8224798cbbfSAlexander Aring con->rx_leftover = buflen - ret; 8234798cbbfSAlexander Aring if (con->rx_leftover) { 8244798cbbfSAlexander Aring memmove(con->rx_buf, con->rx_buf + ret, 8254798cbbfSAlexander Aring con->rx_leftover); 8264798cbbfSAlexander Aring call_again_soon = true; 8276ed7257bSPatrick Caulfield } 8286ed7257bSPatrick Caulfield 8296ed7257bSPatrick Caulfield if (call_again_soon) 8306ed7257bSPatrick Caulfield goto out_resched; 8314798cbbfSAlexander Aring 8326ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 8336ed7257bSPatrick Caulfield return 0; 8346ed7257bSPatrick Caulfield 8356ed7257bSPatrick Caulfield out_resched: 8366ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_READ_PENDING, &con->flags)) 8376ed7257bSPatrick Caulfield queue_work(recv_workqueue, &con->rwork); 8386ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 8396ed7257bSPatrick Caulfield return -EAGAIN; 8406ed7257bSPatrick Caulfield 8416ed7257bSPatrick Caulfield out_close: 8426ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 8439e5f2825SPatrick Caulfield if (ret != -EAGAIN) { 8446ed7257bSPatrick Caulfield /* Reconnect when there is something to send */ 845055923bfSAlexander Aring close_connection(con, false, true, false); 846055923bfSAlexander Aring if (ret == 0) { 847055923bfSAlexander Aring log_print("connection %p got EOF from %d", 848055923bfSAlexander Aring con, con->nodeid); 849055923bfSAlexander Aring /* handling for tcp shutdown */ 850055923bfSAlexander Aring clear_bit(CF_SHUTDOWN, &con->flags); 851055923bfSAlexander Aring wake_up(&con->shutdown_wait); 852055923bfSAlexander Aring /* signal to breaking receive worker */ 853055923bfSAlexander Aring ret = -1; 8546ed7257bSPatrick Caulfield } 855055923bfSAlexander Aring } 8566ed7257bSPatrick Caulfield return ret; 8576ed7257bSPatrick Caulfield } 8586ed7257bSPatrick Caulfield 8596ed7257bSPatrick Caulfield /* Listening socket is busy, accept a connection */ 860d11ccd45SAlexander Aring static int accept_from_sock(struct listen_connection *con) 8616ed7257bSPatrick Caulfield { 8626ed7257bSPatrick Caulfield int result; 8636ed7257bSPatrick Caulfield struct sockaddr_storage peeraddr; 8646ed7257bSPatrick Caulfield struct socket *newsock; 865b38bc9c2SAlexander Aring int len, idx; 8666ed7257bSPatrick Caulfield int nodeid; 8676ed7257bSPatrick Caulfield struct connection *newcon; 8686ed7257bSPatrick Caulfield struct connection *addcon; 8693f78cd7dSAlexander Aring unsigned int mark; 8706ed7257bSPatrick Caulfield 871513ef596SDavid Teigland if (!dlm_allow_conn) { 872513ef596SDavid Teigland return -1; 873513ef596SDavid Teigland } 874513ef596SDavid Teigland 875d11ccd45SAlexander Aring if (!con->sock) 8763421fb15Stsutomu.owa@toshiba.co.jp return -ENOTCONN; 8776ed7257bSPatrick Caulfield 8783421fb15Stsutomu.owa@toshiba.co.jp result = kernel_accept(con->sock, &newsock, O_NONBLOCK); 8796ed7257bSPatrick Caulfield if (result < 0) 8806ed7257bSPatrick Caulfield goto accept_err; 8816ed7257bSPatrick Caulfield 8826ed7257bSPatrick Caulfield /* Get the connected socket's peer */ 8836ed7257bSPatrick Caulfield memset(&peeraddr, 0, sizeof(peeraddr)); 8849b2c45d4SDenys Vlasenko len = newsock->ops->getname(newsock, (struct sockaddr *)&peeraddr, 2); 8859b2c45d4SDenys Vlasenko if (len < 0) { 8866ed7257bSPatrick Caulfield result = -ECONNABORTED; 8876ed7257bSPatrick Caulfield goto accept_err; 8886ed7257bSPatrick Caulfield } 8896ed7257bSPatrick Caulfield 8906ed7257bSPatrick Caulfield /* Get the new node's NODEID */ 8916ed7257bSPatrick Caulfield make_sockaddr(&peeraddr, 0, &len); 892e125fbebSAlexander Aring if (addr_to_nodeid(&peeraddr, &nodeid, &mark)) { 893bcaadf5cSMasatake YAMATO unsigned char *b=(unsigned char *)&peeraddr; 894617e82e1SDavid Teigland log_print("connect from non cluster node"); 895bcaadf5cSMasatake YAMATO print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE, 896bcaadf5cSMasatake YAMATO b, sizeof(struct sockaddr_storage)); 8976ed7257bSPatrick Caulfield sock_release(newsock); 8986ed7257bSPatrick Caulfield return -1; 8996ed7257bSPatrick Caulfield } 9006ed7257bSPatrick Caulfield 9016ed7257bSPatrick Caulfield log_print("got connection from %d", nodeid); 9026ed7257bSPatrick Caulfield 9036ed7257bSPatrick Caulfield /* Check to see if we already have a connection to this node. This 9046ed7257bSPatrick Caulfield * could happen if the two nodes initiate a connection at roughly 9056ed7257bSPatrick Caulfield * the same time and the connections cross on the wire. 9066ed7257bSPatrick Caulfield * In this case we store the incoming one in "othercon" 9076ed7257bSPatrick Caulfield */ 908b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 909748285ccSDavid Teigland newcon = nodeid2con(nodeid, GFP_NOFS); 9106ed7257bSPatrick Caulfield if (!newcon) { 911b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 9126ed7257bSPatrick Caulfield result = -ENOMEM; 9136ed7257bSPatrick Caulfield goto accept_err; 9146ed7257bSPatrick Caulfield } 915d11ccd45SAlexander Aring 916e125fbebSAlexander Aring sock_set_mark(newsock->sk, mark); 917e125fbebSAlexander Aring 918d11ccd45SAlexander Aring mutex_lock(&newcon->sock_mutex); 9196ed7257bSPatrick Caulfield if (newcon->sock) { 9206ed7257bSPatrick Caulfield struct connection *othercon = newcon->othercon; 9216ed7257bSPatrick Caulfield 9226ed7257bSPatrick Caulfield if (!othercon) { 923a47666ebSAlexander Aring othercon = kzalloc(sizeof(*othercon), GFP_NOFS); 9246ed7257bSPatrick Caulfield if (!othercon) { 925617e82e1SDavid Teigland log_print("failed to allocate incoming socket"); 9266ed7257bSPatrick Caulfield mutex_unlock(&newcon->sock_mutex); 927b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 9286ed7257bSPatrick Caulfield result = -ENOMEM; 9296ed7257bSPatrick Caulfield goto accept_err; 9306ed7257bSPatrick Caulfield } 9314798cbbfSAlexander Aring 9326cde210aSAlexander Aring result = dlm_con_init(othercon, nodeid); 9336cde210aSAlexander Aring if (result < 0) { 9344798cbbfSAlexander Aring kfree(othercon); 9352fd8db2dSYang Yingliang mutex_unlock(&newcon->sock_mutex); 936b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 9374798cbbfSAlexander Aring goto accept_err; 9384798cbbfSAlexander Aring } 9394798cbbfSAlexander Aring 940e9a470acSAlexander Aring lockdep_set_subclass(&othercon->sock_mutex, 1); 941*7443bc96SAlexander Aring set_bit(CF_IS_OTHERCON, &othercon->flags); 9426cde210aSAlexander Aring newcon->othercon = othercon; 943ba3ab3caSAlexander Aring } else { 944ba3ab3caSAlexander Aring /* close other sock con if we have something new */ 945ba3ab3caSAlexander Aring close_connection(othercon, false, true, false); 94661d96be0SPatrick Caulfield } 947ba3ab3caSAlexander Aring 948e9a470acSAlexander Aring mutex_lock(&othercon->sock_mutex); 949988419a9Stsutomu.owa@toshiba.co.jp add_sock(newsock, othercon); 9506ed7257bSPatrick Caulfield addcon = othercon; 951c7355827Stsutomu.owa@toshiba.co.jp mutex_unlock(&othercon->sock_mutex); 9526ed7257bSPatrick Caulfield } 9536ed7257bSPatrick Caulfield else { 9543735b4b9SBob Peterson /* accept copies the sk after we've saved the callbacks, so we 9553735b4b9SBob Peterson don't want to save them a second time or comm errors will 9563735b4b9SBob Peterson result in calling sk_error_report recursively. */ 957988419a9Stsutomu.owa@toshiba.co.jp add_sock(newsock, newcon); 9586ed7257bSPatrick Caulfield addcon = newcon; 9596ed7257bSPatrick Caulfield } 9606ed7257bSPatrick Caulfield 961b30a624fSAlexander Aring set_bit(CF_CONNECTED, &addcon->flags); 9626ed7257bSPatrick Caulfield mutex_unlock(&newcon->sock_mutex); 9636ed7257bSPatrick Caulfield 9646ed7257bSPatrick Caulfield /* 9656ed7257bSPatrick Caulfield * Add it to the active queue in case we got data 96625985edcSLucas De Marchi * between processing the accept adding the socket 9676ed7257bSPatrick Caulfield * to the read_sockets list 9686ed7257bSPatrick Caulfield */ 9696ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags)) 9706ed7257bSPatrick Caulfield queue_work(recv_workqueue, &addcon->rwork); 9716ed7257bSPatrick Caulfield 972b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 973b38bc9c2SAlexander Aring 9746ed7257bSPatrick Caulfield return 0; 9756ed7257bSPatrick Caulfield 9766ed7257bSPatrick Caulfield accept_err: 9773421fb15Stsutomu.owa@toshiba.co.jp if (newsock) 9786ed7257bSPatrick Caulfield sock_release(newsock); 9796ed7257bSPatrick Caulfield 9806ed7257bSPatrick Caulfield if (result != -EAGAIN) 981617e82e1SDavid Teigland log_print("error accepting connection from node: %d", result); 9826ed7257bSPatrick Caulfield return result; 9836ed7257bSPatrick Caulfield } 9846ed7257bSPatrick Caulfield 9856ed7257bSPatrick Caulfield static void free_entry(struct writequeue_entry *e) 9866ed7257bSPatrick Caulfield { 9876ed7257bSPatrick Caulfield __free_page(e->page); 9886ed7257bSPatrick Caulfield kfree(e); 9896ed7257bSPatrick Caulfield } 9906ed7257bSPatrick Caulfield 9915d689871SMike Christie /* 9925d689871SMike Christie * writequeue_entry_complete - try to delete and free write queue entry 9935d689871SMike Christie * @e: write queue entry to try to delete 9945d689871SMike Christie * @completed: bytes completed 9955d689871SMike Christie * 9965d689871SMike Christie * writequeue_lock must be held. 9975d689871SMike Christie */ 9985d689871SMike Christie static void writequeue_entry_complete(struct writequeue_entry *e, int completed) 9995d689871SMike Christie { 10005d689871SMike Christie e->offset += completed; 10015d689871SMike Christie e->len -= completed; 10025d689871SMike Christie 10035d689871SMike Christie if (e->len == 0 && e->users == 0) { 10045d689871SMike Christie list_del(&e->list); 10055d689871SMike Christie free_entry(e); 10065d689871SMike Christie } 10075d689871SMike Christie } 10085d689871SMike Christie 1009ee44b4bcSMarcelo Ricardo Leitner /* 1010ee44b4bcSMarcelo Ricardo Leitner * sctp_bind_addrs - bind a SCTP socket to all our addresses 1011ee44b4bcSMarcelo Ricardo Leitner */ 101213004e8aSAlexander Aring static int sctp_bind_addrs(struct socket *sock, uint16_t port) 1013ee44b4bcSMarcelo Ricardo Leitner { 1014ee44b4bcSMarcelo Ricardo Leitner struct sockaddr_storage localaddr; 1015c0425a42SChristoph Hellwig struct sockaddr *addr = (struct sockaddr *)&localaddr; 1016ee44b4bcSMarcelo Ricardo Leitner int i, addr_len, result = 0; 1017ee44b4bcSMarcelo Ricardo Leitner 1018ee44b4bcSMarcelo Ricardo Leitner for (i = 0; i < dlm_local_count; i++) { 1019ee44b4bcSMarcelo Ricardo Leitner memcpy(&localaddr, dlm_local_addr[i], sizeof(localaddr)); 1020ee44b4bcSMarcelo Ricardo Leitner make_sockaddr(&localaddr, port, &addr_len); 1021ee44b4bcSMarcelo Ricardo Leitner 1022ee44b4bcSMarcelo Ricardo Leitner if (!i) 102313004e8aSAlexander Aring result = kernel_bind(sock, addr, addr_len); 1024ee44b4bcSMarcelo Ricardo Leitner else 102513004e8aSAlexander Aring result = sock_bind_add(sock->sk, addr, addr_len); 1026ee44b4bcSMarcelo Ricardo Leitner 1027ee44b4bcSMarcelo Ricardo Leitner if (result < 0) { 1028ee44b4bcSMarcelo Ricardo Leitner log_print("Can't bind to %d addr number %d, %d.\n", 1029ee44b4bcSMarcelo Ricardo Leitner port, i + 1, result); 1030ee44b4bcSMarcelo Ricardo Leitner break; 1031ee44b4bcSMarcelo Ricardo Leitner } 1032ee44b4bcSMarcelo Ricardo Leitner } 1033ee44b4bcSMarcelo Ricardo Leitner return result; 1034ee44b4bcSMarcelo Ricardo Leitner } 1035ee44b4bcSMarcelo Ricardo Leitner 10366ed7257bSPatrick Caulfield /* Initiate an SCTP association. 10376ed7257bSPatrick Caulfield This is a special case of send_to_sock() in that we don't yet have a 10386ed7257bSPatrick Caulfield peeled-off socket for this association, so we use the listening socket 10396ed7257bSPatrick Caulfield and add the primary IP address of the remote node. 10406ed7257bSPatrick Caulfield */ 1041ee44b4bcSMarcelo Ricardo Leitner static void sctp_connect_to_sock(struct connection *con) 10426ed7257bSPatrick Caulfield { 1043ee44b4bcSMarcelo Ricardo Leitner struct sockaddr_storage daddr; 1044ee44b4bcSMarcelo Ricardo Leitner int result; 1045ee44b4bcSMarcelo Ricardo Leitner int addr_len; 1046ee44b4bcSMarcelo Ricardo Leitner struct socket *sock; 10479c9f168fSAlexander Aring unsigned int mark; 1048ee44b4bcSMarcelo Ricardo Leitner 10495d689871SMike Christie mutex_lock(&con->sock_mutex); 10506ed7257bSPatrick Caulfield 1051ee44b4bcSMarcelo Ricardo Leitner /* Some odd races can cause double-connects, ignore them */ 1052ee44b4bcSMarcelo Ricardo Leitner if (con->retries++ > MAX_CONNECT_RETRIES) 1053ee44b4bcSMarcelo Ricardo Leitner goto out; 1054ee44b4bcSMarcelo Ricardo Leitner 1055ee44b4bcSMarcelo Ricardo Leitner if (con->sock) { 1056ee44b4bcSMarcelo Ricardo Leitner log_print("node %d already connected.", con->nodeid); 1057ee44b4bcSMarcelo Ricardo Leitner goto out; 1058ee44b4bcSMarcelo Ricardo Leitner } 1059ee44b4bcSMarcelo Ricardo Leitner 1060ee44b4bcSMarcelo Ricardo Leitner memset(&daddr, 0, sizeof(daddr)); 1061e125fbebSAlexander Aring result = nodeid_to_addr(con->nodeid, &daddr, NULL, true, &mark); 1062ee44b4bcSMarcelo Ricardo Leitner if (result < 0) { 10636ed7257bSPatrick Caulfield log_print("no address for nodeid %d", con->nodeid); 1064ee44b4bcSMarcelo Ricardo Leitner goto out; 106504bedd79SDavid Teigland } 10666ed7257bSPatrick Caulfield 1067ee44b4bcSMarcelo Ricardo Leitner /* Create a socket to communicate with */ 1068ee44b4bcSMarcelo Ricardo Leitner result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family, 1069ee44b4bcSMarcelo Ricardo Leitner SOCK_STREAM, IPPROTO_SCTP, &sock); 1070ee44b4bcSMarcelo Ricardo Leitner if (result < 0) 1071ee44b4bcSMarcelo Ricardo Leitner goto socket_err; 10726ed7257bSPatrick Caulfield 10739c9f168fSAlexander Aring sock_set_mark(sock->sk, mark); 10749c9f168fSAlexander Aring 1075988419a9Stsutomu.owa@toshiba.co.jp add_sock(sock, con); 10766ed7257bSPatrick Caulfield 1077ee44b4bcSMarcelo Ricardo Leitner /* Bind to all addresses. */ 107813004e8aSAlexander Aring if (sctp_bind_addrs(con->sock, 0)) 1079ee44b4bcSMarcelo Ricardo Leitner goto bind_err; 108098e1b60eSMike Christie 1081ee44b4bcSMarcelo Ricardo Leitner make_sockaddr(&daddr, dlm_config.ci_tcp_port, &addr_len); 10826ed7257bSPatrick Caulfield 10832df6b762SAlexander Aring log_print_ratelimited("connecting to %d", con->nodeid); 10846ed7257bSPatrick Caulfield 1085ee44b4bcSMarcelo Ricardo Leitner /* Turn off Nagle's algorithm */ 108640ef92c6SChristoph Hellwig sctp_sock_set_nodelay(sock->sk); 1087ee44b4bcSMarcelo Ricardo Leitner 1088f706d830SGang He /* 1089f706d830SGang He * Make sock->ops->connect() function return in specified time, 1090f706d830SGang He * since O_NONBLOCK argument in connect() function does not work here, 1091f706d830SGang He * then, we should restore the default value of this attribute. 1092f706d830SGang He */ 109376ee0785SChristoph Hellwig sock_set_sndtimeo(sock->sk, 5); 1094ee44b4bcSMarcelo Ricardo Leitner result = sock->ops->connect(sock, (struct sockaddr *)&daddr, addr_len, 1095da3627c3SGang He 0); 109676ee0785SChristoph Hellwig sock_set_sndtimeo(sock->sk, 0); 1097f706d830SGang He 1098ee44b4bcSMarcelo Ricardo Leitner if (result == -EINPROGRESS) 1099ee44b4bcSMarcelo Ricardo Leitner result = 0; 110019633c7eSAlexander Aring if (result == 0) { 110119633c7eSAlexander Aring if (!test_and_set_bit(CF_CONNECTED, &con->flags)) 110219633c7eSAlexander Aring log_print("successful connected to node %d", con->nodeid); 1103ee44b4bcSMarcelo Ricardo Leitner goto out; 110419633c7eSAlexander Aring } 1105ee44b4bcSMarcelo Ricardo Leitner 1106ee44b4bcSMarcelo Ricardo Leitner bind_err: 1107ee44b4bcSMarcelo Ricardo Leitner con->sock = NULL; 1108ee44b4bcSMarcelo Ricardo Leitner sock_release(sock); 1109ee44b4bcSMarcelo Ricardo Leitner 1110ee44b4bcSMarcelo Ricardo Leitner socket_err: 1111ee44b4bcSMarcelo Ricardo Leitner /* 1112ee44b4bcSMarcelo Ricardo Leitner * Some errors are fatal and this list might need adjusting. For other 1113ee44b4bcSMarcelo Ricardo Leitner * errors we try again until the max number of retries is reached. 1114ee44b4bcSMarcelo Ricardo Leitner */ 1115ee44b4bcSMarcelo Ricardo Leitner if (result != -EHOSTUNREACH && 1116ee44b4bcSMarcelo Ricardo Leitner result != -ENETUNREACH && 1117ee44b4bcSMarcelo Ricardo Leitner result != -ENETDOWN && 1118ee44b4bcSMarcelo Ricardo Leitner result != -EINVAL && 1119ee44b4bcSMarcelo Ricardo Leitner result != -EPROTONOSUPPORT) { 1120ee44b4bcSMarcelo Ricardo Leitner log_print("connect %d try %d error %d", con->nodeid, 1121ee44b4bcSMarcelo Ricardo Leitner con->retries, result); 1122ee44b4bcSMarcelo Ricardo Leitner mutex_unlock(&con->sock_mutex); 1123ee44b4bcSMarcelo Ricardo Leitner msleep(1000); 1124ee44b4bcSMarcelo Ricardo Leitner lowcomms_connect_sock(con); 1125ee44b4bcSMarcelo Ricardo Leitner return; 11266ed7257bSPatrick Caulfield } 11275d689871SMike Christie 1128ee44b4bcSMarcelo Ricardo Leitner out: 11295d689871SMike Christie mutex_unlock(&con->sock_mutex); 11306ed7257bSPatrick Caulfield } 11316ed7257bSPatrick Caulfield 11326ed7257bSPatrick Caulfield /* Connect a new socket to its peer */ 11336ed7257bSPatrick Caulfield static void tcp_connect_to_sock(struct connection *con) 11346ed7257bSPatrick Caulfield { 11356bd8fedaSLon Hohberger struct sockaddr_storage saddr, src_addr; 1136e125fbebSAlexander Aring unsigned int mark; 11376ed7257bSPatrick Caulfield int addr_len; 1138a89d63a1SCasey Dahlin struct socket *sock = NULL; 113936b71a8bSDavid Teigland int result; 11406ed7257bSPatrick Caulfield 11416ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 11426ed7257bSPatrick Caulfield if (con->retries++ > MAX_CONNECT_RETRIES) 11436ed7257bSPatrick Caulfield goto out; 11446ed7257bSPatrick Caulfield 11456ed7257bSPatrick Caulfield /* Some odd races can cause double-connects, ignore them */ 114636b71a8bSDavid Teigland if (con->sock) 11476ed7257bSPatrick Caulfield goto out; 11486ed7257bSPatrick Caulfield 11496ed7257bSPatrick Caulfield /* Create a socket to communicate with */ 1150eeb1bd5cSEric W. Biederman result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family, 1151eeb1bd5cSEric W. Biederman SOCK_STREAM, IPPROTO_TCP, &sock); 11526ed7257bSPatrick Caulfield if (result < 0) 11536ed7257bSPatrick Caulfield goto out_err; 11546ed7257bSPatrick Caulfield 11556ed7257bSPatrick Caulfield memset(&saddr, 0, sizeof(saddr)); 1156e125fbebSAlexander Aring result = nodeid_to_addr(con->nodeid, &saddr, NULL, false, &mark); 115736b71a8bSDavid Teigland if (result < 0) { 115836b71a8bSDavid Teigland log_print("no address for nodeid %d", con->nodeid); 11596ed7257bSPatrick Caulfield goto out_err; 116036b71a8bSDavid Teigland } 11616ed7257bSPatrick Caulfield 1162e125fbebSAlexander Aring sock_set_mark(sock->sk, mark); 1163e125fbebSAlexander Aring 1164988419a9Stsutomu.owa@toshiba.co.jp add_sock(sock, con); 11656ed7257bSPatrick Caulfield 11666bd8fedaSLon Hohberger /* Bind to our cluster-known address connecting to avoid 11676bd8fedaSLon Hohberger routing problems */ 11686bd8fedaSLon Hohberger memcpy(&src_addr, dlm_local_addr[0], sizeof(src_addr)); 11696bd8fedaSLon Hohberger make_sockaddr(&src_addr, 0, &addr_len); 11706bd8fedaSLon Hohberger result = sock->ops->bind(sock, (struct sockaddr *) &src_addr, 11716bd8fedaSLon Hohberger addr_len); 11726bd8fedaSLon Hohberger if (result < 0) { 11736bd8fedaSLon Hohberger log_print("could not bind for connect: %d", result); 11746bd8fedaSLon Hohberger /* This *may* not indicate a critical error */ 11756bd8fedaSLon Hohberger } 11766bd8fedaSLon Hohberger 11776ed7257bSPatrick Caulfield make_sockaddr(&saddr, dlm_config.ci_tcp_port, &addr_len); 11786ed7257bSPatrick Caulfield 11792df6b762SAlexander Aring log_print_ratelimited("connecting to %d", con->nodeid); 1180cb2d45daSDavid Teigland 1181cb2d45daSDavid Teigland /* Turn off Nagle's algorithm */ 118212abc5eeSChristoph Hellwig tcp_sock_set_nodelay(sock->sk); 1183cb2d45daSDavid Teigland 118436b71a8bSDavid Teigland result = sock->ops->connect(sock, (struct sockaddr *)&saddr, addr_len, 11856ed7257bSPatrick Caulfield O_NONBLOCK); 11866ed7257bSPatrick Caulfield if (result == -EINPROGRESS) 11876ed7257bSPatrick Caulfield result = 0; 11886ed7257bSPatrick Caulfield if (result == 0) 11896ed7257bSPatrick Caulfield goto out; 11906ed7257bSPatrick Caulfield 11916ed7257bSPatrick Caulfield out_err: 11926ed7257bSPatrick Caulfield if (con->sock) { 11936ed7257bSPatrick Caulfield sock_release(con->sock); 11946ed7257bSPatrick Caulfield con->sock = NULL; 1195a89d63a1SCasey Dahlin } else if (sock) { 1196a89d63a1SCasey Dahlin sock_release(sock); 11976ed7257bSPatrick Caulfield } 11986ed7257bSPatrick Caulfield /* 11996ed7257bSPatrick Caulfield * Some errors are fatal and this list might need adjusting. For other 12006ed7257bSPatrick Caulfield * errors we try again until the max number of retries is reached. 12016ed7257bSPatrick Caulfield */ 120236b71a8bSDavid Teigland if (result != -EHOSTUNREACH && 120336b71a8bSDavid Teigland result != -ENETUNREACH && 120436b71a8bSDavid Teigland result != -ENETDOWN && 120536b71a8bSDavid Teigland result != -EINVAL && 120636b71a8bSDavid Teigland result != -EPROTONOSUPPORT) { 120736b71a8bSDavid Teigland log_print("connect %d try %d error %d", con->nodeid, 120836b71a8bSDavid Teigland con->retries, result); 120936b71a8bSDavid Teigland mutex_unlock(&con->sock_mutex); 121036b71a8bSDavid Teigland msleep(1000); 12116ed7257bSPatrick Caulfield lowcomms_connect_sock(con); 121236b71a8bSDavid Teigland return; 12136ed7257bSPatrick Caulfield } 12146ed7257bSPatrick Caulfield out: 12156ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 12166ed7257bSPatrick Caulfield return; 12176ed7257bSPatrick Caulfield } 12186ed7257bSPatrick Caulfield 1219d11ccd45SAlexander Aring /* On error caller must run dlm_close_sock() for the 1220d11ccd45SAlexander Aring * listen connection socket. 1221d11ccd45SAlexander Aring */ 1222d11ccd45SAlexander Aring static int tcp_create_listen_sock(struct listen_connection *con, 12236ed7257bSPatrick Caulfield struct sockaddr_storage *saddr) 12246ed7257bSPatrick Caulfield { 12256ed7257bSPatrick Caulfield struct socket *sock = NULL; 12266ed7257bSPatrick Caulfield int result = 0; 12276ed7257bSPatrick Caulfield int addr_len; 12286ed7257bSPatrick Caulfield 12296ed7257bSPatrick Caulfield if (dlm_local_addr[0]->ss_family == AF_INET) 12306ed7257bSPatrick Caulfield addr_len = sizeof(struct sockaddr_in); 12316ed7257bSPatrick Caulfield else 12326ed7257bSPatrick Caulfield addr_len = sizeof(struct sockaddr_in6); 12336ed7257bSPatrick Caulfield 12346ed7257bSPatrick Caulfield /* Create a socket to communicate with */ 1235eeb1bd5cSEric W. Biederman result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family, 1236eeb1bd5cSEric W. Biederman SOCK_STREAM, IPPROTO_TCP, &sock); 12376ed7257bSPatrick Caulfield if (result < 0) { 1238617e82e1SDavid Teigland log_print("Can't create listening comms socket"); 12396ed7257bSPatrick Caulfield goto create_out; 12406ed7257bSPatrick Caulfield } 12416ed7257bSPatrick Caulfield 1242a5b7ab63SAlexander Aring sock_set_mark(sock->sk, dlm_config.ci_mark); 1243a5b7ab63SAlexander Aring 1244cb2d45daSDavid Teigland /* Turn off Nagle's algorithm */ 124512abc5eeSChristoph Hellwig tcp_sock_set_nodelay(sock->sk); 1246cb2d45daSDavid Teigland 1247b58f0e8fSChristoph Hellwig sock_set_reuseaddr(sock->sk); 12486ed7257bSPatrick Caulfield 1249d11ccd45SAlexander Aring add_listen_sock(sock, con); 12506ed7257bSPatrick Caulfield 12516ed7257bSPatrick Caulfield /* Bind to our port */ 12526ed7257bSPatrick Caulfield make_sockaddr(saddr, dlm_config.ci_tcp_port, &addr_len); 12536ed7257bSPatrick Caulfield result = sock->ops->bind(sock, (struct sockaddr *) saddr, addr_len); 12546ed7257bSPatrick Caulfield if (result < 0) { 1255617e82e1SDavid Teigland log_print("Can't bind to port %d", dlm_config.ci_tcp_port); 12566ed7257bSPatrick Caulfield goto create_out; 12576ed7257bSPatrick Caulfield } 1258ce3d9544SChristoph Hellwig sock_set_keepalive(sock->sk); 12596ed7257bSPatrick Caulfield 12606ed7257bSPatrick Caulfield result = sock->ops->listen(sock, 5); 12616ed7257bSPatrick Caulfield if (result < 0) { 1262617e82e1SDavid Teigland log_print("Can't listen on port %d", dlm_config.ci_tcp_port); 12636ed7257bSPatrick Caulfield goto create_out; 12646ed7257bSPatrick Caulfield } 12656ed7257bSPatrick Caulfield 1266d11ccd45SAlexander Aring return 0; 1267d11ccd45SAlexander Aring 12686ed7257bSPatrick Caulfield create_out: 1269d11ccd45SAlexander Aring return result; 12706ed7257bSPatrick Caulfield } 12716ed7257bSPatrick Caulfield 12726ed7257bSPatrick Caulfield /* Get local addresses */ 12736ed7257bSPatrick Caulfield static void init_local(void) 12746ed7257bSPatrick Caulfield { 12756ed7257bSPatrick Caulfield struct sockaddr_storage sas, *addr; 12766ed7257bSPatrick Caulfield int i; 12776ed7257bSPatrick Caulfield 127830d3a237SPatrick Caulfield dlm_local_count = 0; 12791b189b88SDavid Teigland for (i = 0; i < DLM_MAX_ADDR_COUNT; i++) { 12806ed7257bSPatrick Caulfield if (dlm_our_addr(&sas, i)) 12816ed7257bSPatrick Caulfield break; 12826ed7257bSPatrick Caulfield 12835c93f56fSAmitoj Kaur Chawla addr = kmemdup(&sas, sizeof(*addr), GFP_NOFS); 12846ed7257bSPatrick Caulfield if (!addr) 12856ed7257bSPatrick Caulfield break; 12866ed7257bSPatrick Caulfield dlm_local_addr[dlm_local_count++] = addr; 12876ed7257bSPatrick Caulfield } 12886ed7257bSPatrick Caulfield } 12896ed7257bSPatrick Caulfield 1290043697f0SAlexander Aring static void deinit_local(void) 1291043697f0SAlexander Aring { 1292043697f0SAlexander Aring int i; 1293043697f0SAlexander Aring 1294043697f0SAlexander Aring for (i = 0; i < dlm_local_count; i++) 1295043697f0SAlexander Aring kfree(dlm_local_addr[i]); 1296043697f0SAlexander Aring } 1297043697f0SAlexander Aring 1298d11ccd45SAlexander Aring /* Initialise SCTP socket and bind to all interfaces 1299d11ccd45SAlexander Aring * On error caller must run dlm_close_sock() for the 1300d11ccd45SAlexander Aring * listen connection socket. 1301d11ccd45SAlexander Aring */ 1302d11ccd45SAlexander Aring static int sctp_listen_for_all(struct listen_connection *con) 13036ed7257bSPatrick Caulfield { 13046ed7257bSPatrick Caulfield struct socket *sock = NULL; 1305ee44b4bcSMarcelo Ricardo Leitner int result = -EINVAL; 13066ed7257bSPatrick Caulfield 13076ed7257bSPatrick Caulfield log_print("Using SCTP for communications"); 13086ed7257bSPatrick Caulfield 1309eeb1bd5cSEric W. Biederman result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family, 1310ee44b4bcSMarcelo Ricardo Leitner SOCK_STREAM, IPPROTO_SCTP, &sock); 13116ed7257bSPatrick Caulfield if (result < 0) { 13126ed7257bSPatrick Caulfield log_print("Can't create comms socket, check SCTP is loaded"); 13136ed7257bSPatrick Caulfield goto out; 13146ed7257bSPatrick Caulfield } 13156ed7257bSPatrick Caulfield 131626cfabf9SChristoph Hellwig sock_set_rcvbuf(sock->sk, NEEDED_RMEM); 1317a5b7ab63SAlexander Aring sock_set_mark(sock->sk, dlm_config.ci_mark); 131840ef92c6SChristoph Hellwig sctp_sock_set_nodelay(sock->sk); 131986e92ad2SMike Christie 1320d11ccd45SAlexander Aring add_listen_sock(sock, con); 1321b81171cbSBob Peterson 1322ee44b4bcSMarcelo Ricardo Leitner /* Bind to all addresses. */ 1323d11ccd45SAlexander Aring result = sctp_bind_addrs(con->sock, dlm_config.ci_tcp_port); 1324d11ccd45SAlexander Aring if (result < 0) 1325d11ccd45SAlexander Aring goto out; 13266ed7257bSPatrick Caulfield 13276ed7257bSPatrick Caulfield result = sock->ops->listen(sock, 5); 13286ed7257bSPatrick Caulfield if (result < 0) { 13296ed7257bSPatrick Caulfield log_print("Can't set socket listening"); 1330d11ccd45SAlexander Aring goto out; 13316ed7257bSPatrick Caulfield } 13326ed7257bSPatrick Caulfield 13336ed7257bSPatrick Caulfield return 0; 13346ed7257bSPatrick Caulfield 13356ed7257bSPatrick Caulfield out: 13366ed7257bSPatrick Caulfield return result; 13376ed7257bSPatrick Caulfield } 13386ed7257bSPatrick Caulfield 13396ed7257bSPatrick Caulfield static int tcp_listen_for_all(void) 13406ed7257bSPatrick Caulfield { 13416ed7257bSPatrick Caulfield /* We don't support multi-homed hosts */ 13421a26bfafSAlexander Aring if (dlm_local_count > 1) { 1343617e82e1SDavid Teigland log_print("TCP protocol can't handle multi-homed hosts, " 1344617e82e1SDavid Teigland "try SCTP"); 13456ed7257bSPatrick Caulfield return -EINVAL; 13466ed7257bSPatrick Caulfield } 13476ed7257bSPatrick Caulfield 13486ed7257bSPatrick Caulfield log_print("Using TCP for communications"); 13496ed7257bSPatrick Caulfield 1350d11ccd45SAlexander Aring return tcp_create_listen_sock(&listen_con, dlm_local_addr[0]); 13516ed7257bSPatrick Caulfield } 13526ed7257bSPatrick Caulfield 13536ed7257bSPatrick Caulfield 13546ed7257bSPatrick Caulfield 13556ed7257bSPatrick Caulfield static struct writequeue_entry *new_writequeue_entry(struct connection *con, 13566ed7257bSPatrick Caulfield gfp_t allocation) 13576ed7257bSPatrick Caulfield { 13586ed7257bSPatrick Caulfield struct writequeue_entry *entry; 13596ed7257bSPatrick Caulfield 1360f0747ebfSAlexander Aring entry = kzalloc(sizeof(*entry), allocation); 13616ed7257bSPatrick Caulfield if (!entry) 13626ed7257bSPatrick Caulfield return NULL; 13636ed7257bSPatrick Caulfield 1364e1a7cbceSAlexander Aring entry->page = alloc_page(allocation | __GFP_ZERO); 13656ed7257bSPatrick Caulfield if (!entry->page) { 13666ed7257bSPatrick Caulfield kfree(entry); 13676ed7257bSPatrick Caulfield return NULL; 13686ed7257bSPatrick Caulfield } 13696ed7257bSPatrick Caulfield 13706ed7257bSPatrick Caulfield entry->con = con; 1371f0747ebfSAlexander Aring entry->users = 1; 13726ed7257bSPatrick Caulfield 13736ed7257bSPatrick Caulfield return entry; 13746ed7257bSPatrick Caulfield } 13756ed7257bSPatrick Caulfield 1376f0747ebfSAlexander Aring static struct writequeue_entry *new_wq_entry(struct connection *con, int len, 1377f0747ebfSAlexander Aring gfp_t allocation, char **ppc) 1378f0747ebfSAlexander Aring { 1379f0747ebfSAlexander Aring struct writequeue_entry *e; 1380f0747ebfSAlexander Aring 1381f0747ebfSAlexander Aring spin_lock(&con->writequeue_lock); 1382f0747ebfSAlexander Aring if (!list_empty(&con->writequeue)) { 1383f0747ebfSAlexander Aring e = list_last_entry(&con->writequeue, struct writequeue_entry, list); 1384f0747ebfSAlexander Aring if (DLM_WQ_REMAIN_BYTES(e) >= len) { 1385f0747ebfSAlexander Aring *ppc = page_address(e->page) + e->end; 1386f0747ebfSAlexander Aring e->end += len; 1387f0747ebfSAlexander Aring e->users++; 1388f0747ebfSAlexander Aring spin_unlock(&con->writequeue_lock); 1389f0747ebfSAlexander Aring 1390f0747ebfSAlexander Aring return e; 1391f0747ebfSAlexander Aring } 1392f0747ebfSAlexander Aring } 1393f0747ebfSAlexander Aring spin_unlock(&con->writequeue_lock); 1394f0747ebfSAlexander Aring 1395f0747ebfSAlexander Aring e = new_writequeue_entry(con, allocation); 1396f0747ebfSAlexander Aring if (!e) 1397f0747ebfSAlexander Aring return NULL; 1398f0747ebfSAlexander Aring 1399f0747ebfSAlexander Aring *ppc = page_address(e->page); 1400f0747ebfSAlexander Aring e->end += len; 1401f0747ebfSAlexander Aring 1402f0747ebfSAlexander Aring spin_lock(&con->writequeue_lock); 1403f0747ebfSAlexander Aring list_add_tail(&e->list, &con->writequeue); 1404f0747ebfSAlexander Aring spin_unlock(&con->writequeue_lock); 1405f0747ebfSAlexander Aring 1406f0747ebfSAlexander Aring return e; 1407f0747ebfSAlexander Aring }; 1408f0747ebfSAlexander Aring 1409617e82e1SDavid Teigland void *dlm_lowcomms_get_buffer(int nodeid, int len, gfp_t allocation, char **ppc) 14106ed7257bSPatrick Caulfield { 1411b38bc9c2SAlexander Aring struct writequeue_entry *e; 14126ed7257bSPatrick Caulfield struct connection *con; 1413b38bc9c2SAlexander Aring int idx; 14146ed7257bSPatrick Caulfield 1415c45674fbSAlexander Aring if (len > DEFAULT_BUFFER_SIZE || 1416c45674fbSAlexander Aring len < sizeof(struct dlm_header)) { 1417c45674fbSAlexander Aring BUILD_BUG_ON(PAGE_SIZE < DEFAULT_BUFFER_SIZE); 1418692f51c8SAlexander Aring log_print("failed to allocate a buffer of size %d", len); 1419c45674fbSAlexander Aring WARN_ON(1); 1420692f51c8SAlexander Aring return NULL; 1421692f51c8SAlexander Aring } 1422692f51c8SAlexander Aring 1423b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 14246ed7257bSPatrick Caulfield con = nodeid2con(nodeid, allocation); 1425b38bc9c2SAlexander Aring if (!con) { 1426b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 14276ed7257bSPatrick Caulfield return NULL; 1428b38bc9c2SAlexander Aring } 14296ed7257bSPatrick Caulfield 1430b38bc9c2SAlexander Aring e = new_wq_entry(con, len, allocation, ppc); 1431b38bc9c2SAlexander Aring if (!e) { 1432b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 1433b38bc9c2SAlexander Aring return NULL; 1434b38bc9c2SAlexander Aring } 1435b38bc9c2SAlexander Aring 1436b38bc9c2SAlexander Aring /* we assume if successful commit must called */ 1437b38bc9c2SAlexander Aring e->idx = idx; 1438b38bc9c2SAlexander Aring 1439b38bc9c2SAlexander Aring return e; 14406ed7257bSPatrick Caulfield } 14416ed7257bSPatrick Caulfield 14426ed7257bSPatrick Caulfield void dlm_lowcomms_commit_buffer(void *mh) 14436ed7257bSPatrick Caulfield { 14446ed7257bSPatrick Caulfield struct writequeue_entry *e = (struct writequeue_entry *)mh; 14456ed7257bSPatrick Caulfield struct connection *con = e->con; 14466ed7257bSPatrick Caulfield int users; 14476ed7257bSPatrick Caulfield 14486ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 14496ed7257bSPatrick Caulfield users = --e->users; 14506ed7257bSPatrick Caulfield if (users) 14516ed7257bSPatrick Caulfield goto out; 1452f0747ebfSAlexander Aring 1453f0747ebfSAlexander Aring e->len = DLM_WQ_LENGTH_BYTES(e); 14546ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14556ed7257bSPatrick Caulfield 14566ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 1457b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, e->idx); 14586ed7257bSPatrick Caulfield return; 14596ed7257bSPatrick Caulfield 14606ed7257bSPatrick Caulfield out: 14616ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 1462b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, e->idx); 14636ed7257bSPatrick Caulfield return; 14646ed7257bSPatrick Caulfield } 14656ed7257bSPatrick Caulfield 14666ed7257bSPatrick Caulfield /* Send a message */ 14676ed7257bSPatrick Caulfield static void send_to_sock(struct connection *con) 14686ed7257bSPatrick Caulfield { 14696ed7257bSPatrick Caulfield int ret = 0; 14706ed7257bSPatrick Caulfield const int msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL; 14716ed7257bSPatrick Caulfield struct writequeue_entry *e; 14726ed7257bSPatrick Caulfield int len, offset; 1473f92c8dd7SBob Peterson int count = 0; 14746ed7257bSPatrick Caulfield 14756ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 14766ed7257bSPatrick Caulfield if (con->sock == NULL) 14776ed7257bSPatrick Caulfield goto out_connect; 14786ed7257bSPatrick Caulfield 14796ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 14806ed7257bSPatrick Caulfield for (;;) { 1481f0747ebfSAlexander Aring if (list_empty(&con->writequeue)) 14826ed7257bSPatrick Caulfield break; 14836ed7257bSPatrick Caulfield 1484f0747ebfSAlexander Aring e = list_first_entry(&con->writequeue, struct writequeue_entry, list); 14856ed7257bSPatrick Caulfield len = e->len; 14866ed7257bSPatrick Caulfield offset = e->offset; 14876ed7257bSPatrick Caulfield BUG_ON(len == 0 && e->users == 0); 14886ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14896ed7257bSPatrick Caulfield 14906ed7257bSPatrick Caulfield ret = 0; 14916ed7257bSPatrick Caulfield if (len) { 14921329e3f2SPaolo Bonzini ret = kernel_sendpage(con->sock, e->page, offset, len, 14936ed7257bSPatrick Caulfield msg_flags); 1494d66f8277SPatrick Caulfield if (ret == -EAGAIN || ret == 0) { 1495b36930ddSDavid Miller if (ret == -EAGAIN && 14969cd3e072SEric Dumazet test_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags) && 1497b36930ddSDavid Miller !test_and_set_bit(CF_APP_LIMITED, &con->flags)) { 1498b36930ddSDavid Miller /* Notify TCP that we're limited by the 1499b36930ddSDavid Miller * application window size. 1500b36930ddSDavid Miller */ 1501b36930ddSDavid Miller set_bit(SOCK_NOSPACE, &con->sock->flags); 1502b36930ddSDavid Miller con->sock->sk->sk_write_pending++; 1503b36930ddSDavid Miller } 1504d66f8277SPatrick Caulfield cond_resched(); 15056ed7257bSPatrick Caulfield goto out; 15069c5bef58SYing Xue } else if (ret < 0) 15076ed7257bSPatrick Caulfield goto send_error; 1508d66f8277SPatrick Caulfield } 1509f92c8dd7SBob Peterson 15106ed7257bSPatrick Caulfield /* Don't starve people filling buffers */ 1511f92c8dd7SBob Peterson if (++count >= MAX_SEND_MSG_COUNT) { 15126ed7257bSPatrick Caulfield cond_resched(); 1513f92c8dd7SBob Peterson count = 0; 1514f92c8dd7SBob Peterson } 15156ed7257bSPatrick Caulfield 15166ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 15175d689871SMike Christie writequeue_entry_complete(e, ret); 15186ed7257bSPatrick Caulfield } 15196ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 15206ed7257bSPatrick Caulfield out: 15216ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 15226ed7257bSPatrick Caulfield return; 15236ed7257bSPatrick Caulfield 15246ed7257bSPatrick Caulfield send_error: 15256ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 1526ba3ab3caSAlexander Aring close_connection(con, false, false, true); 152701da24d3SBob Peterson /* Requeue the send work. When the work daemon runs again, it will try 152801da24d3SBob Peterson a new connection, then call this function again. */ 152901da24d3SBob Peterson queue_work(send_workqueue, &con->swork); 15306ed7257bSPatrick Caulfield return; 15316ed7257bSPatrick Caulfield 15326ed7257bSPatrick Caulfield out_connect: 15336ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 153401da24d3SBob Peterson queue_work(send_workqueue, &con->swork); 153501da24d3SBob Peterson cond_resched(); 15366ed7257bSPatrick Caulfield } 15376ed7257bSPatrick Caulfield 15386ed7257bSPatrick Caulfield static void clean_one_writequeue(struct connection *con) 15396ed7257bSPatrick Caulfield { 15405e9ccc37SChristine Caulfield struct writequeue_entry *e, *safe; 15416ed7257bSPatrick Caulfield 15426ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 15435e9ccc37SChristine Caulfield list_for_each_entry_safe(e, safe, &con->writequeue, list) { 15446ed7257bSPatrick Caulfield list_del(&e->list); 15456ed7257bSPatrick Caulfield free_entry(e); 15466ed7257bSPatrick Caulfield } 15476ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 15486ed7257bSPatrick Caulfield } 15496ed7257bSPatrick Caulfield 15506ed7257bSPatrick Caulfield /* Called from recovery when it knows that a node has 15516ed7257bSPatrick Caulfield left the cluster */ 15526ed7257bSPatrick Caulfield int dlm_lowcomms_close(int nodeid) 15536ed7257bSPatrick Caulfield { 15546ed7257bSPatrick Caulfield struct connection *con; 155536b71a8bSDavid Teigland struct dlm_node_addr *na; 1556b38bc9c2SAlexander Aring int idx; 15576ed7257bSPatrick Caulfield 15586ed7257bSPatrick Caulfield log_print("closing connection to node %d", nodeid); 1559b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 15606ed7257bSPatrick Caulfield con = nodeid2con(nodeid, 0); 15616ed7257bSPatrick Caulfield if (con) { 1562063c4c99SLars Marowsky-Bree set_bit(CF_CLOSE, &con->flags); 15630d737a8cSMarcelo Ricardo Leitner close_connection(con, true, true, true); 15646ed7257bSPatrick Caulfield clean_one_writequeue(con); 156553a5edaaSAlexander Aring if (con->othercon) 156653a5edaaSAlexander Aring clean_one_writequeue(con->othercon); 15676ed7257bSPatrick Caulfield } 1568b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 156936b71a8bSDavid Teigland 157036b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 157136b71a8bSDavid Teigland na = find_node_addr(nodeid); 157236b71a8bSDavid Teigland if (na) { 157336b71a8bSDavid Teigland list_del(&na->list); 157436b71a8bSDavid Teigland while (na->addr_count--) 157536b71a8bSDavid Teigland kfree(na->addr[na->addr_count]); 157636b71a8bSDavid Teigland kfree(na); 157736b71a8bSDavid Teigland } 157836b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 157936b71a8bSDavid Teigland 15806ed7257bSPatrick Caulfield return 0; 15816ed7257bSPatrick Caulfield } 15826ed7257bSPatrick Caulfield 15836ed7257bSPatrick Caulfield /* Receive workqueue function */ 15846ed7257bSPatrick Caulfield static void process_recv_sockets(struct work_struct *work) 15856ed7257bSPatrick Caulfield { 15866ed7257bSPatrick Caulfield struct connection *con = container_of(work, struct connection, rwork); 15876ed7257bSPatrick Caulfield int err; 15886ed7257bSPatrick Caulfield 15896ed7257bSPatrick Caulfield clear_bit(CF_READ_PENDING, &con->flags); 15906ed7257bSPatrick Caulfield do { 1591d11ccd45SAlexander Aring err = receive_from_sock(con); 15926ed7257bSPatrick Caulfield } while (!err); 15936ed7257bSPatrick Caulfield } 15946ed7257bSPatrick Caulfield 1595d11ccd45SAlexander Aring static void process_listen_recv_socket(struct work_struct *work) 1596d11ccd45SAlexander Aring { 1597d11ccd45SAlexander Aring accept_from_sock(&listen_con); 1598d11ccd45SAlexander Aring } 1599d11ccd45SAlexander Aring 16006ed7257bSPatrick Caulfield /* Send workqueue function */ 16016ed7257bSPatrick Caulfield static void process_send_sockets(struct work_struct *work) 16026ed7257bSPatrick Caulfield { 16036ed7257bSPatrick Caulfield struct connection *con = container_of(work, struct connection, swork); 16046ed7257bSPatrick Caulfield 1605*7443bc96SAlexander Aring WARN_ON(test_bit(CF_IS_OTHERCON, &con->flags)); 1606*7443bc96SAlexander Aring 16078a4abb08Stsutomu.owa@toshiba.co.jp clear_bit(CF_WRITE_PENDING, &con->flags); 160861d9102bSBob Peterson if (con->sock == NULL) /* not mutex protected so check it inside too */ 16096ed7257bSPatrick Caulfield con->connect_action(con); 161001da24d3SBob Peterson if (!list_empty(&con->writequeue)) 16116ed7257bSPatrick Caulfield send_to_sock(con); 16126ed7257bSPatrick Caulfield } 16136ed7257bSPatrick Caulfield 16146ed7257bSPatrick Caulfield static void work_stop(void) 16156ed7257bSPatrick Caulfield { 1616b355516fSDavid Windsor if (recv_workqueue) 16176ed7257bSPatrick Caulfield destroy_workqueue(recv_workqueue); 1618b355516fSDavid Windsor if (send_workqueue) 16196ed7257bSPatrick Caulfield destroy_workqueue(send_workqueue); 16206ed7257bSPatrick Caulfield } 16216ed7257bSPatrick Caulfield 16226ed7257bSPatrick Caulfield static int work_start(void) 16236ed7257bSPatrick Caulfield { 1624e43f055aSDavid Teigland recv_workqueue = alloc_workqueue("dlm_recv", 1625e43f055aSDavid Teigland WQ_UNBOUND | WQ_MEM_RECLAIM, 1); 1626b9d41052SNamhyung Kim if (!recv_workqueue) { 1627b9d41052SNamhyung Kim log_print("can't start dlm_recv"); 1628b9d41052SNamhyung Kim return -ENOMEM; 16296ed7257bSPatrick Caulfield } 16306ed7257bSPatrick Caulfield 1631e43f055aSDavid Teigland send_workqueue = alloc_workqueue("dlm_send", 1632e43f055aSDavid Teigland WQ_UNBOUND | WQ_MEM_RECLAIM, 1); 1633b9d41052SNamhyung Kim if (!send_workqueue) { 1634b9d41052SNamhyung Kim log_print("can't start dlm_send"); 16356ed7257bSPatrick Caulfield destroy_workqueue(recv_workqueue); 1636b9d41052SNamhyung Kim return -ENOMEM; 16376ed7257bSPatrick Caulfield } 16386ed7257bSPatrick Caulfield 16396ed7257bSPatrick Caulfield return 0; 16406ed7257bSPatrick Caulfield } 16416ed7257bSPatrick Caulfield 16429d232469SAlexander Aring static void shutdown_conn(struct connection *con) 16439d232469SAlexander Aring { 16449d232469SAlexander Aring if (con->shutdown_action) 16459d232469SAlexander Aring con->shutdown_action(con); 16469d232469SAlexander Aring } 16479d232469SAlexander Aring 16489d232469SAlexander Aring void dlm_lowcomms_shutdown(void) 16499d232469SAlexander Aring { 1650b38bc9c2SAlexander Aring int idx; 1651b38bc9c2SAlexander Aring 16529d232469SAlexander Aring /* Set all the flags to prevent any 16539d232469SAlexander Aring * socket activity. 16549d232469SAlexander Aring */ 16559d232469SAlexander Aring dlm_allow_conn = 0; 16569d232469SAlexander Aring 16579d232469SAlexander Aring if (recv_workqueue) 16589d232469SAlexander Aring flush_workqueue(recv_workqueue); 16599d232469SAlexander Aring if (send_workqueue) 16609d232469SAlexander Aring flush_workqueue(send_workqueue); 16619d232469SAlexander Aring 16629d232469SAlexander Aring dlm_close_sock(&listen_con.sock); 16639d232469SAlexander Aring 1664b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 16659d232469SAlexander Aring foreach_conn(shutdown_conn); 1666b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 16679d232469SAlexander Aring } 16689d232469SAlexander Aring 1669f0fb83cbStsutomu.owa@toshiba.co.jp static void _stop_conn(struct connection *con, bool and_other) 16706ed7257bSPatrick Caulfield { 1671f0fb83cbStsutomu.owa@toshiba.co.jp mutex_lock(&con->sock_mutex); 1672173a31feStsutomu.owa@toshiba.co.jp set_bit(CF_CLOSE, &con->flags); 1673f0fb83cbStsutomu.owa@toshiba.co.jp set_bit(CF_READ_PENDING, &con->flags); 16748a4abb08Stsutomu.owa@toshiba.co.jp set_bit(CF_WRITE_PENDING, &con->flags); 167593eaadebStsutomu.owa@toshiba.co.jp if (con->sock && con->sock->sk) { 167693eaadebStsutomu.owa@toshiba.co.jp write_lock_bh(&con->sock->sk->sk_callback_lock); 1677afb853fbSPatrick Caulfield con->sock->sk->sk_user_data = NULL; 167893eaadebStsutomu.owa@toshiba.co.jp write_unlock_bh(&con->sock->sk->sk_callback_lock); 167993eaadebStsutomu.owa@toshiba.co.jp } 1680f0fb83cbStsutomu.owa@toshiba.co.jp if (con->othercon && and_other) 1681f0fb83cbStsutomu.owa@toshiba.co.jp _stop_conn(con->othercon, false); 1682f0fb83cbStsutomu.owa@toshiba.co.jp mutex_unlock(&con->sock_mutex); 1683f0fb83cbStsutomu.owa@toshiba.co.jp } 1684f0fb83cbStsutomu.owa@toshiba.co.jp 1685f0fb83cbStsutomu.owa@toshiba.co.jp static void stop_conn(struct connection *con) 1686f0fb83cbStsutomu.owa@toshiba.co.jp { 1687f0fb83cbStsutomu.owa@toshiba.co.jp _stop_conn(con, true); 1688afb853fbSPatrick Caulfield } 16895e9ccc37SChristine Caulfield 16904798cbbfSAlexander Aring static void connection_release(struct rcu_head *rcu) 16914798cbbfSAlexander Aring { 16924798cbbfSAlexander Aring struct connection *con = container_of(rcu, struct connection, rcu); 16934798cbbfSAlexander Aring 16944798cbbfSAlexander Aring kfree(con->rx_buf); 16954798cbbfSAlexander Aring kfree(con); 16964798cbbfSAlexander Aring } 16974798cbbfSAlexander Aring 16985e9ccc37SChristine Caulfield static void free_conn(struct connection *con) 16995e9ccc37SChristine Caulfield { 17000d737a8cSMarcelo Ricardo Leitner close_connection(con, true, true, true); 1701a47666ebSAlexander Aring spin_lock(&connections_lock); 1702a47666ebSAlexander Aring hlist_del_rcu(&con->list); 1703a47666ebSAlexander Aring spin_unlock(&connections_lock); 1704948c47e9SAlexander Aring if (con->othercon) { 1705948c47e9SAlexander Aring clean_one_writequeue(con->othercon); 17065cbec208SAlexander Aring call_srcu(&connections_srcu, &con->othercon->rcu, 17075cbec208SAlexander Aring connection_release); 1708948c47e9SAlexander Aring } 17090de98432SAlexander Aring clean_one_writequeue(con); 17105cbec208SAlexander Aring call_srcu(&connections_srcu, &con->rcu, connection_release); 17116ed7257bSPatrick Caulfield } 17125e9ccc37SChristine Caulfield 1713f0fb83cbStsutomu.owa@toshiba.co.jp static void work_flush(void) 1714f0fb83cbStsutomu.owa@toshiba.co.jp { 1715b38bc9c2SAlexander Aring int ok; 1716f0fb83cbStsutomu.owa@toshiba.co.jp int i; 1717f0fb83cbStsutomu.owa@toshiba.co.jp struct connection *con; 1718f0fb83cbStsutomu.owa@toshiba.co.jp 1719f0fb83cbStsutomu.owa@toshiba.co.jp do { 1720f0fb83cbStsutomu.owa@toshiba.co.jp ok = 1; 1721f0fb83cbStsutomu.owa@toshiba.co.jp foreach_conn(stop_conn); 1722b355516fSDavid Windsor if (recv_workqueue) 1723f0fb83cbStsutomu.owa@toshiba.co.jp flush_workqueue(recv_workqueue); 1724b355516fSDavid Windsor if (send_workqueue) 1725f0fb83cbStsutomu.owa@toshiba.co.jp flush_workqueue(send_workqueue); 1726f0fb83cbStsutomu.owa@toshiba.co.jp for (i = 0; i < CONN_HASH_SIZE && ok; i++) { 1727a47666ebSAlexander Aring hlist_for_each_entry_rcu(con, &connection_hash[i], 1728a47666ebSAlexander Aring list) { 1729f0fb83cbStsutomu.owa@toshiba.co.jp ok &= test_bit(CF_READ_PENDING, &con->flags); 17308a4abb08Stsutomu.owa@toshiba.co.jp ok &= test_bit(CF_WRITE_PENDING, &con->flags); 17318a4abb08Stsutomu.owa@toshiba.co.jp if (con->othercon) { 1732f0fb83cbStsutomu.owa@toshiba.co.jp ok &= test_bit(CF_READ_PENDING, 1733f0fb83cbStsutomu.owa@toshiba.co.jp &con->othercon->flags); 17348a4abb08Stsutomu.owa@toshiba.co.jp ok &= test_bit(CF_WRITE_PENDING, 17358a4abb08Stsutomu.owa@toshiba.co.jp &con->othercon->flags); 17368a4abb08Stsutomu.owa@toshiba.co.jp } 1737f0fb83cbStsutomu.owa@toshiba.co.jp } 1738f0fb83cbStsutomu.owa@toshiba.co.jp } 1739f0fb83cbStsutomu.owa@toshiba.co.jp } while (!ok); 1740f0fb83cbStsutomu.owa@toshiba.co.jp } 1741f0fb83cbStsutomu.owa@toshiba.co.jp 17425e9ccc37SChristine Caulfield void dlm_lowcomms_stop(void) 17435e9ccc37SChristine Caulfield { 1744b38bc9c2SAlexander Aring int idx; 1745b38bc9c2SAlexander Aring 1746b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 1747f0fb83cbStsutomu.owa@toshiba.co.jp work_flush(); 17483a8db798SMarcelo Ricardo Leitner foreach_conn(free_conn); 1749b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 17506ed7257bSPatrick Caulfield work_stop(); 1751043697f0SAlexander Aring deinit_local(); 17526ed7257bSPatrick Caulfield } 17536ed7257bSPatrick Caulfield 17546ed7257bSPatrick Caulfield int dlm_lowcomms_start(void) 17556ed7257bSPatrick Caulfield { 17566ed7257bSPatrick Caulfield int error = -EINVAL; 17575e9ccc37SChristine Caulfield int i; 17585e9ccc37SChristine Caulfield 17595e9ccc37SChristine Caulfield for (i = 0; i < CONN_HASH_SIZE; i++) 17605e9ccc37SChristine Caulfield INIT_HLIST_HEAD(&connection_hash[i]); 17616ed7257bSPatrick Caulfield 17626ed7257bSPatrick Caulfield init_local(); 17636ed7257bSPatrick Caulfield if (!dlm_local_count) { 1764617e82e1SDavid Teigland error = -ENOTCONN; 17656ed7257bSPatrick Caulfield log_print("no local IP address has been set"); 1766513ef596SDavid Teigland goto fail; 17676ed7257bSPatrick Caulfield } 17686ed7257bSPatrick Caulfield 1769d11ccd45SAlexander Aring INIT_WORK(&listen_con.rwork, process_listen_recv_socket); 1770d11ccd45SAlexander Aring 1771513ef596SDavid Teigland error = work_start(); 1772513ef596SDavid Teigland if (error) 1773a47666ebSAlexander Aring goto fail; 1774513ef596SDavid Teigland 1775513ef596SDavid Teigland dlm_allow_conn = 1; 17766ed7257bSPatrick Caulfield 17776ed7257bSPatrick Caulfield /* Start listening */ 17786ed7257bSPatrick Caulfield if (dlm_config.ci_protocol == 0) 17796ed7257bSPatrick Caulfield error = tcp_listen_for_all(); 17806ed7257bSPatrick Caulfield else 1781d11ccd45SAlexander Aring error = sctp_listen_for_all(&listen_con); 17826ed7257bSPatrick Caulfield if (error) 17836ed7257bSPatrick Caulfield goto fail_unlisten; 17846ed7257bSPatrick Caulfield 17856ed7257bSPatrick Caulfield return 0; 17866ed7257bSPatrick Caulfield 17876ed7257bSPatrick Caulfield fail_unlisten: 1788513ef596SDavid Teigland dlm_allow_conn = 0; 1789d11ccd45SAlexander Aring dlm_close_sock(&listen_con.sock); 1790513ef596SDavid Teigland fail: 17916ed7257bSPatrick Caulfield return error; 17926ed7257bSPatrick Caulfield } 179336b71a8bSDavid Teigland 179436b71a8bSDavid Teigland void dlm_lowcomms_exit(void) 179536b71a8bSDavid Teigland { 179636b71a8bSDavid Teigland struct dlm_node_addr *na, *safe; 179736b71a8bSDavid Teigland 179836b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 179936b71a8bSDavid Teigland list_for_each_entry_safe(na, safe, &dlm_node_addrs, list) { 180036b71a8bSDavid Teigland list_del(&na->list); 180136b71a8bSDavid Teigland while (na->addr_count--) 180236b71a8bSDavid Teigland kfree(na->addr[na->addr_count]); 180336b71a8bSDavid Teigland kfree(na); 180436b71a8bSDavid Teigland } 180536b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 180636b71a8bSDavid Teigland } 1807