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 5692732376SAlexander Aring #include <trace/events/dlm.h> 5792732376SAlexander Aring 586ed7257bSPatrick Caulfield #include "dlm_internal.h" 596ed7257bSPatrick Caulfield #include "lowcomms.h" 606ed7257bSPatrick Caulfield #include "midcomms.h" 616ed7257bSPatrick Caulfield #include "config.h" 626ed7257bSPatrick Caulfield 636ed7257bSPatrick Caulfield #define NEEDED_RMEM (4*1024*1024) 646ed7257bSPatrick Caulfield 65f92c8dd7SBob Peterson /* Number of messages to send before rescheduling */ 66f92c8dd7SBob Peterson #define MAX_SEND_MSG_COUNT 25 67055923bfSAlexander Aring #define DLM_SHUTDOWN_WAIT_TIMEOUT msecs_to_jiffies(10000) 68f92c8dd7SBob Peterson 696ed7257bSPatrick Caulfield struct connection { 706ed7257bSPatrick Caulfield struct socket *sock; /* NULL if not connected */ 716ed7257bSPatrick Caulfield uint32_t nodeid; /* So we know who we are in the list */ 726ed7257bSPatrick Caulfield struct mutex sock_mutex; 736ed7257bSPatrick Caulfield unsigned long flags; 746ed7257bSPatrick Caulfield #define CF_READ_PENDING 1 758a4abb08Stsutomu.owa@toshiba.co.jp #define CF_WRITE_PENDING 2 766ed7257bSPatrick Caulfield #define CF_INIT_PENDING 4 776ed7257bSPatrick Caulfield #define CF_IS_OTHERCON 5 78063c4c99SLars Marowsky-Bree #define CF_CLOSE 6 79b36930ddSDavid Miller #define CF_APP_LIMITED 7 80b2a66629Stsutomu.owa@toshiba.co.jp #define CF_CLOSING 8 81055923bfSAlexander Aring #define CF_SHUTDOWN 9 8219633c7eSAlexander Aring #define CF_CONNECTED 10 83ba868d9dSAlexander Aring #define CF_RECONNECT 11 84ba868d9dSAlexander Aring #define CF_DELAY_CONNECT 12 858aa31cbfSAlexander Aring #define CF_EOF 13 866ed7257bSPatrick Caulfield struct list_head writequeue; /* List of outgoing writequeue_entries */ 876ed7257bSPatrick Caulfield spinlock_t writequeue_lock; 888aa31cbfSAlexander Aring atomic_t writequeue_cnt; 89c51b0221SAlexander Aring struct mutex wq_alloc; 906ed7257bSPatrick Caulfield int retries; 916ed7257bSPatrick Caulfield #define MAX_CONNECT_RETRIES 3 925e9ccc37SChristine Caulfield struct hlist_node list; 936ed7257bSPatrick Caulfield struct connection *othercon; 94ba868d9dSAlexander Aring struct connection *sendcon; 956ed7257bSPatrick Caulfield struct work_struct rwork; /* Receive workqueue */ 966ed7257bSPatrick Caulfield struct work_struct swork; /* Send workqueue */ 97055923bfSAlexander Aring wait_queue_head_t shutdown_wait; /* wait for graceful shutdown */ 984798cbbfSAlexander Aring unsigned char *rx_buf; 994798cbbfSAlexander Aring int rx_buflen; 1004798cbbfSAlexander Aring int rx_leftover; 101a47666ebSAlexander Aring struct rcu_head rcu; 1026ed7257bSPatrick Caulfield }; 1036ed7257bSPatrick Caulfield #define sock2con(x) ((struct connection *)(x)->sk_user_data) 1046ed7257bSPatrick Caulfield 105d11ccd45SAlexander Aring struct listen_connection { 106d11ccd45SAlexander Aring struct socket *sock; 107d11ccd45SAlexander Aring struct work_struct rwork; 108d11ccd45SAlexander Aring }; 109d11ccd45SAlexander Aring 110f0747ebfSAlexander Aring #define DLM_WQ_REMAIN_BYTES(e) (PAGE_SIZE - e->end) 111f0747ebfSAlexander Aring #define DLM_WQ_LENGTH_BYTES(e) (e->end - e->offset) 112f0747ebfSAlexander Aring 1136ed7257bSPatrick Caulfield /* An entry waiting to be sent */ 1146ed7257bSPatrick Caulfield struct writequeue_entry { 1156ed7257bSPatrick Caulfield struct list_head list; 1166ed7257bSPatrick Caulfield struct page *page; 1176ed7257bSPatrick Caulfield int offset; 1186ed7257bSPatrick Caulfield int len; 1196ed7257bSPatrick Caulfield int end; 1206ed7257bSPatrick Caulfield int users; 121706474fbSAlexander Aring bool dirty; 1226ed7257bSPatrick Caulfield struct connection *con; 1238f2dc78dSAlexander Aring struct list_head msgs; 1248f2dc78dSAlexander Aring struct kref ref; 1258f2dc78dSAlexander Aring }; 1268f2dc78dSAlexander Aring 1278f2dc78dSAlexander Aring struct dlm_msg { 1288f2dc78dSAlexander Aring struct writequeue_entry *entry; 1292874d1a6SAlexander Aring struct dlm_msg *orig_msg; 1302874d1a6SAlexander Aring bool retransmit; 1318f2dc78dSAlexander Aring void *ppc; 1328f2dc78dSAlexander Aring int len; 1338f2dc78dSAlexander Aring int idx; /* new()/commit() idx exchange */ 1348f2dc78dSAlexander Aring 1358f2dc78dSAlexander Aring struct list_head list; 1368f2dc78dSAlexander Aring struct kref ref; 1376ed7257bSPatrick Caulfield }; 1386ed7257bSPatrick Caulfield 13936b71a8bSDavid Teigland struct dlm_node_addr { 14036b71a8bSDavid Teigland struct list_head list; 14136b71a8bSDavid Teigland int nodeid; 142e125fbebSAlexander Aring int mark; 14336b71a8bSDavid Teigland int addr_count; 14498e1b60eSMike Christie int curr_addr_index; 14536b71a8bSDavid Teigland struct sockaddr_storage *addr[DLM_MAX_ADDR_COUNT]; 14636b71a8bSDavid Teigland }; 14736b71a8bSDavid Teigland 148a66c008cSAlexander Aring struct dlm_proto_ops { 1498728a455SAlexander Aring bool try_new_addr; 1502dc6b115SAlexander Aring const char *name; 1512dc6b115SAlexander Aring int proto; 1522dc6b115SAlexander Aring 1538728a455SAlexander Aring int (*connect)(struct connection *con, struct socket *sock, 1548728a455SAlexander Aring struct sockaddr *addr, int addr_len); 1558728a455SAlexander Aring void (*sockopts)(struct socket *sock); 1568728a455SAlexander Aring int (*bind)(struct socket *sock); 1572dc6b115SAlexander Aring int (*listen_validate)(void); 1582dc6b115SAlexander Aring void (*listen_sockopts)(struct socket *sock); 1592dc6b115SAlexander Aring int (*listen_bind)(struct socket *sock); 160a66c008cSAlexander Aring /* What to do to shutdown */ 161a66c008cSAlexander Aring void (*shutdown_action)(struct connection *con); 162a66c008cSAlexander Aring /* What to do to eof check */ 163a66c008cSAlexander Aring bool (*eof_condition)(struct connection *con); 164a66c008cSAlexander Aring }; 165a66c008cSAlexander Aring 166cc661fc9SBob Peterson static struct listen_sock_callbacks { 167cc661fc9SBob Peterson void (*sk_error_report)(struct sock *); 168cc661fc9SBob Peterson void (*sk_data_ready)(struct sock *); 169cc661fc9SBob Peterson void (*sk_state_change)(struct sock *); 170cc661fc9SBob Peterson void (*sk_write_space)(struct sock *); 171cc661fc9SBob Peterson } listen_sock; 172cc661fc9SBob Peterson 17336b71a8bSDavid Teigland static LIST_HEAD(dlm_node_addrs); 17436b71a8bSDavid Teigland static DEFINE_SPINLOCK(dlm_node_addrs_spin); 17536b71a8bSDavid Teigland 176d11ccd45SAlexander Aring static struct listen_connection listen_con; 1776ed7257bSPatrick Caulfield static struct sockaddr_storage *dlm_local_addr[DLM_MAX_ADDR_COUNT]; 1786ed7257bSPatrick Caulfield static int dlm_local_count; 17951746163SAlexander Aring int dlm_allow_conn; 1806ed7257bSPatrick Caulfield 1816ed7257bSPatrick Caulfield /* Work queues */ 1826ed7257bSPatrick Caulfield static struct workqueue_struct *recv_workqueue; 1836ed7257bSPatrick Caulfield static struct workqueue_struct *send_workqueue; 1846ed7257bSPatrick Caulfield 1855e9ccc37SChristine Caulfield static struct hlist_head connection_hash[CONN_HASH_SIZE]; 186a47666ebSAlexander Aring static DEFINE_SPINLOCK(connections_lock); 187a47666ebSAlexander Aring DEFINE_STATIC_SRCU(connections_srcu); 1886ed7257bSPatrick Caulfield 189a66c008cSAlexander Aring static const struct dlm_proto_ops *dlm_proto_ops; 190a66c008cSAlexander Aring 1916ed7257bSPatrick Caulfield static void process_recv_sockets(struct work_struct *work); 1926ed7257bSPatrick Caulfield static void process_send_sockets(struct work_struct *work); 1936ed7257bSPatrick Caulfield 19466d5955aSAlexander Aring /* need to held writequeue_lock */ 19566d5955aSAlexander Aring static struct writequeue_entry *con_next_wq(struct connection *con) 19666d5955aSAlexander Aring { 19766d5955aSAlexander Aring struct writequeue_entry *e; 19866d5955aSAlexander Aring 19966d5955aSAlexander Aring if (list_empty(&con->writequeue)) 20066d5955aSAlexander Aring return NULL; 20166d5955aSAlexander Aring 20266d5955aSAlexander Aring e = list_first_entry(&con->writequeue, struct writequeue_entry, 20366d5955aSAlexander Aring list); 20466d5955aSAlexander Aring if (e->len == 0) 20566d5955aSAlexander Aring return NULL; 20666d5955aSAlexander Aring 20766d5955aSAlexander Aring return e; 20866d5955aSAlexander Aring } 20966d5955aSAlexander Aring 210b38bc9c2SAlexander Aring static struct connection *__find_con(int nodeid, int r) 2115e9ccc37SChristine Caulfield { 2125e9ccc37SChristine Caulfield struct connection *con; 2135e9ccc37SChristine Caulfield 214a47666ebSAlexander Aring hlist_for_each_entry_rcu(con, &connection_hash[r], list) { 215b38bc9c2SAlexander Aring if (con->nodeid == nodeid) 2165e9ccc37SChristine Caulfield return con; 2175e9ccc37SChristine Caulfield } 218a47666ebSAlexander Aring 2195e9ccc37SChristine Caulfield return NULL; 2205e9ccc37SChristine Caulfield } 2215e9ccc37SChristine Caulfield 2228aa31cbfSAlexander Aring static bool tcp_eof_condition(struct connection *con) 2238aa31cbfSAlexander Aring { 2248aa31cbfSAlexander Aring return atomic_read(&con->writequeue_cnt); 2258aa31cbfSAlexander Aring } 2268aa31cbfSAlexander Aring 2276cde210aSAlexander Aring static int dlm_con_init(struct connection *con, int nodeid) 2286ed7257bSPatrick Caulfield { 2294798cbbfSAlexander Aring con->rx_buflen = dlm_config.ci_buffer_size; 2304798cbbfSAlexander Aring con->rx_buf = kmalloc(con->rx_buflen, GFP_NOFS); 2316cde210aSAlexander Aring if (!con->rx_buf) 2326cde210aSAlexander Aring return -ENOMEM; 2334798cbbfSAlexander Aring 2346ed7257bSPatrick Caulfield con->nodeid = nodeid; 2356ed7257bSPatrick Caulfield mutex_init(&con->sock_mutex); 2366ed7257bSPatrick Caulfield INIT_LIST_HEAD(&con->writequeue); 2376ed7257bSPatrick Caulfield spin_lock_init(&con->writequeue_lock); 2388aa31cbfSAlexander Aring atomic_set(&con->writequeue_cnt, 0); 2396ed7257bSPatrick Caulfield INIT_WORK(&con->swork, process_send_sockets); 2406ed7257bSPatrick Caulfield INIT_WORK(&con->rwork, process_recv_sockets); 241055923bfSAlexander Aring init_waitqueue_head(&con->shutdown_wait); 2426ed7257bSPatrick Caulfield 2436cde210aSAlexander Aring return 0; 2446cde210aSAlexander Aring } 2456cde210aSAlexander Aring 2466cde210aSAlexander Aring /* 2476cde210aSAlexander Aring * If 'allocation' is zero then we don't attempt to create a new 2486cde210aSAlexander Aring * connection structure for this node. 2496cde210aSAlexander Aring */ 2506cde210aSAlexander Aring static struct connection *nodeid2con(int nodeid, gfp_t alloc) 2516cde210aSAlexander Aring { 2526cde210aSAlexander Aring struct connection *con, *tmp; 2536cde210aSAlexander Aring int r, ret; 2546cde210aSAlexander Aring 255b38bc9c2SAlexander Aring r = nodeid_hash(nodeid); 256b38bc9c2SAlexander Aring con = __find_con(nodeid, r); 2576cde210aSAlexander Aring if (con || !alloc) 2586cde210aSAlexander Aring return con; 2596cde210aSAlexander Aring 2606cde210aSAlexander Aring con = kzalloc(sizeof(*con), alloc); 2616cde210aSAlexander Aring if (!con) 2626cde210aSAlexander Aring return NULL; 2636cde210aSAlexander Aring 2646cde210aSAlexander Aring ret = dlm_con_init(con, nodeid); 2656cde210aSAlexander Aring if (ret) { 2666cde210aSAlexander Aring kfree(con); 2676cde210aSAlexander Aring return NULL; 2686cde210aSAlexander Aring } 2696cde210aSAlexander Aring 270c51b0221SAlexander Aring mutex_init(&con->wq_alloc); 271c51b0221SAlexander Aring 272a47666ebSAlexander Aring spin_lock(&connections_lock); 2734f2b30fdSAlexander Aring /* Because multiple workqueues/threads calls this function it can 2744f2b30fdSAlexander Aring * race on multiple cpu's. Instead of locking hot path __find_con() 2754f2b30fdSAlexander Aring * we just check in rare cases of recently added nodes again 2764f2b30fdSAlexander Aring * under protection of connections_lock. If this is the case we 2774f2b30fdSAlexander Aring * abort our connection creation and return the existing connection. 2784f2b30fdSAlexander Aring */ 279b38bc9c2SAlexander Aring tmp = __find_con(nodeid, r); 2804f2b30fdSAlexander Aring if (tmp) { 2814f2b30fdSAlexander Aring spin_unlock(&connections_lock); 2824f2b30fdSAlexander Aring kfree(con->rx_buf); 2834f2b30fdSAlexander Aring kfree(con); 2844f2b30fdSAlexander Aring return tmp; 2854f2b30fdSAlexander Aring } 2864f2b30fdSAlexander Aring 287a47666ebSAlexander Aring hlist_add_head_rcu(&con->list, &connection_hash[r]); 288a47666ebSAlexander Aring spin_unlock(&connections_lock); 289a47666ebSAlexander Aring 2906ed7257bSPatrick Caulfield return con; 2916ed7257bSPatrick Caulfield } 2926ed7257bSPatrick Caulfield 2935e9ccc37SChristine Caulfield /* Loop round all connections */ 2945e9ccc37SChristine Caulfield static void foreach_conn(void (*conn_func)(struct connection *c)) 2955e9ccc37SChristine Caulfield { 296b38bc9c2SAlexander Aring int i; 2975e9ccc37SChristine Caulfield struct connection *con; 2985e9ccc37SChristine Caulfield 2995e9ccc37SChristine Caulfield for (i = 0; i < CONN_HASH_SIZE; i++) { 300a47666ebSAlexander Aring hlist_for_each_entry_rcu(con, &connection_hash[i], list) 3015e9ccc37SChristine Caulfield conn_func(con); 3025e9ccc37SChristine Caulfield } 3036ed7257bSPatrick Caulfield } 3046ed7257bSPatrick Caulfield 30536b71a8bSDavid Teigland static struct dlm_node_addr *find_node_addr(int nodeid) 3066ed7257bSPatrick Caulfield { 30736b71a8bSDavid Teigland struct dlm_node_addr *na; 30836b71a8bSDavid Teigland 30936b71a8bSDavid Teigland list_for_each_entry(na, &dlm_node_addrs, list) { 31036b71a8bSDavid Teigland if (na->nodeid == nodeid) 31136b71a8bSDavid Teigland return na; 31236b71a8bSDavid Teigland } 31336b71a8bSDavid Teigland return NULL; 31436b71a8bSDavid Teigland } 31536b71a8bSDavid Teigland 31640c6b83eSAlexander Aring static int addr_compare(const struct sockaddr_storage *x, 31740c6b83eSAlexander Aring const struct sockaddr_storage *y) 31836b71a8bSDavid Teigland { 31936b71a8bSDavid Teigland switch (x->ss_family) { 32036b71a8bSDavid Teigland case AF_INET: { 32136b71a8bSDavid Teigland struct sockaddr_in *sinx = (struct sockaddr_in *)x; 32236b71a8bSDavid Teigland struct sockaddr_in *siny = (struct sockaddr_in *)y; 32336b71a8bSDavid Teigland if (sinx->sin_addr.s_addr != siny->sin_addr.s_addr) 32436b71a8bSDavid Teigland return 0; 32536b71a8bSDavid Teigland if (sinx->sin_port != siny->sin_port) 32636b71a8bSDavid Teigland return 0; 32736b71a8bSDavid Teigland break; 32836b71a8bSDavid Teigland } 32936b71a8bSDavid Teigland case AF_INET6: { 33036b71a8bSDavid Teigland struct sockaddr_in6 *sinx = (struct sockaddr_in6 *)x; 33136b71a8bSDavid Teigland struct sockaddr_in6 *siny = (struct sockaddr_in6 *)y; 33236b71a8bSDavid Teigland if (!ipv6_addr_equal(&sinx->sin6_addr, &siny->sin6_addr)) 33336b71a8bSDavid Teigland return 0; 33436b71a8bSDavid Teigland if (sinx->sin6_port != siny->sin6_port) 33536b71a8bSDavid Teigland return 0; 33636b71a8bSDavid Teigland break; 33736b71a8bSDavid Teigland } 33836b71a8bSDavid Teigland default: 33936b71a8bSDavid Teigland return 0; 34036b71a8bSDavid Teigland } 34136b71a8bSDavid Teigland return 1; 34236b71a8bSDavid Teigland } 34336b71a8bSDavid Teigland 34436b71a8bSDavid Teigland static int nodeid_to_addr(int nodeid, struct sockaddr_storage *sas_out, 345e125fbebSAlexander Aring struct sockaddr *sa_out, bool try_new_addr, 346e125fbebSAlexander Aring unsigned int *mark) 34736b71a8bSDavid Teigland { 34836b71a8bSDavid Teigland struct sockaddr_storage sas; 34936b71a8bSDavid Teigland struct dlm_node_addr *na; 3506ed7257bSPatrick Caulfield 3516ed7257bSPatrick Caulfield if (!dlm_local_count) 3526ed7257bSPatrick Caulfield return -1; 3536ed7257bSPatrick Caulfield 35436b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 35536b71a8bSDavid Teigland na = find_node_addr(nodeid); 35698e1b60eSMike Christie if (na && na->addr_count) { 357ee44b4bcSMarcelo Ricardo Leitner memcpy(&sas, na->addr[na->curr_addr_index], 358ee44b4bcSMarcelo Ricardo Leitner sizeof(struct sockaddr_storage)); 359ee44b4bcSMarcelo Ricardo Leitner 36098e1b60eSMike Christie if (try_new_addr) { 36198e1b60eSMike Christie na->curr_addr_index++; 36298e1b60eSMike Christie if (na->curr_addr_index == na->addr_count) 36398e1b60eSMike Christie na->curr_addr_index = 0; 36498e1b60eSMike Christie } 36598e1b60eSMike Christie } 36636b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 36736b71a8bSDavid Teigland 36836b71a8bSDavid Teigland if (!na) 36936b71a8bSDavid Teigland return -EEXIST; 37036b71a8bSDavid Teigland 37136b71a8bSDavid Teigland if (!na->addr_count) 37236b71a8bSDavid Teigland return -ENOENT; 37336b71a8bSDavid Teigland 374e125fbebSAlexander Aring *mark = na->mark; 375e125fbebSAlexander Aring 37636b71a8bSDavid Teigland if (sas_out) 37736b71a8bSDavid Teigland memcpy(sas_out, &sas, sizeof(struct sockaddr_storage)); 37836b71a8bSDavid Teigland 37936b71a8bSDavid Teigland if (!sa_out) 38036b71a8bSDavid Teigland return 0; 3816ed7257bSPatrick Caulfield 3826ed7257bSPatrick Caulfield if (dlm_local_addr[0]->ss_family == AF_INET) { 38336b71a8bSDavid Teigland struct sockaddr_in *in4 = (struct sockaddr_in *) &sas; 38436b71a8bSDavid Teigland struct sockaddr_in *ret4 = (struct sockaddr_in *) sa_out; 3856ed7257bSPatrick Caulfield ret4->sin_addr.s_addr = in4->sin_addr.s_addr; 3866ed7257bSPatrick Caulfield } else { 38736b71a8bSDavid Teigland struct sockaddr_in6 *in6 = (struct sockaddr_in6 *) &sas; 38836b71a8bSDavid Teigland struct sockaddr_in6 *ret6 = (struct sockaddr_in6 *) sa_out; 3894e3fd7a0SAlexey Dobriyan ret6->sin6_addr = in6->sin6_addr; 3906ed7257bSPatrick Caulfield } 3916ed7257bSPatrick Caulfield 3926ed7257bSPatrick Caulfield return 0; 3936ed7257bSPatrick Caulfield } 3946ed7257bSPatrick Caulfield 395e125fbebSAlexander Aring static int addr_to_nodeid(struct sockaddr_storage *addr, int *nodeid, 396e125fbebSAlexander Aring unsigned int *mark) 39736b71a8bSDavid Teigland { 39836b71a8bSDavid Teigland struct dlm_node_addr *na; 39936b71a8bSDavid Teigland int rv = -EEXIST; 40098e1b60eSMike Christie int addr_i; 40136b71a8bSDavid Teigland 40236b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 40336b71a8bSDavid Teigland list_for_each_entry(na, &dlm_node_addrs, list) { 40436b71a8bSDavid Teigland if (!na->addr_count) 40536b71a8bSDavid Teigland continue; 40636b71a8bSDavid Teigland 40798e1b60eSMike Christie for (addr_i = 0; addr_i < na->addr_count; addr_i++) { 40898e1b60eSMike Christie if (addr_compare(na->addr[addr_i], addr)) { 40936b71a8bSDavid Teigland *nodeid = na->nodeid; 410e125fbebSAlexander Aring *mark = na->mark; 41136b71a8bSDavid Teigland rv = 0; 41298e1b60eSMike Christie goto unlock; 41336b71a8bSDavid Teigland } 41498e1b60eSMike Christie } 41598e1b60eSMike Christie } 41698e1b60eSMike Christie unlock: 41736b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 41836b71a8bSDavid Teigland return rv; 41936b71a8bSDavid Teigland } 42036b71a8bSDavid Teigland 4214f19d071SAlexander Aring /* caller need to held dlm_node_addrs_spin lock */ 4224f19d071SAlexander Aring static bool dlm_lowcomms_na_has_addr(const struct dlm_node_addr *na, 4234f19d071SAlexander Aring const struct sockaddr_storage *addr) 4244f19d071SAlexander Aring { 4254f19d071SAlexander Aring int i; 4264f19d071SAlexander Aring 4274f19d071SAlexander Aring for (i = 0; i < na->addr_count; i++) { 4284f19d071SAlexander Aring if (addr_compare(na->addr[i], addr)) 4294f19d071SAlexander Aring return true; 4304f19d071SAlexander Aring } 4314f19d071SAlexander Aring 4324f19d071SAlexander Aring return false; 4334f19d071SAlexander Aring } 4344f19d071SAlexander Aring 43536b71a8bSDavid Teigland int dlm_lowcomms_addr(int nodeid, struct sockaddr_storage *addr, int len) 43636b71a8bSDavid Teigland { 43736b71a8bSDavid Teigland struct sockaddr_storage *new_addr; 43836b71a8bSDavid Teigland struct dlm_node_addr *new_node, *na; 4394f19d071SAlexander Aring bool ret; 44036b71a8bSDavid Teigland 44136b71a8bSDavid Teigland new_node = kzalloc(sizeof(struct dlm_node_addr), GFP_NOFS); 44236b71a8bSDavid Teigland if (!new_node) 44336b71a8bSDavid Teigland return -ENOMEM; 44436b71a8bSDavid Teigland 44536b71a8bSDavid Teigland new_addr = kzalloc(sizeof(struct sockaddr_storage), GFP_NOFS); 44636b71a8bSDavid Teigland if (!new_addr) { 44736b71a8bSDavid Teigland kfree(new_node); 44836b71a8bSDavid Teigland return -ENOMEM; 44936b71a8bSDavid Teigland } 45036b71a8bSDavid Teigland 45136b71a8bSDavid Teigland memcpy(new_addr, addr, len); 45236b71a8bSDavid Teigland 45336b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 45436b71a8bSDavid Teigland na = find_node_addr(nodeid); 45536b71a8bSDavid Teigland if (!na) { 45636b71a8bSDavid Teigland new_node->nodeid = nodeid; 45736b71a8bSDavid Teigland new_node->addr[0] = new_addr; 45836b71a8bSDavid Teigland new_node->addr_count = 1; 459e125fbebSAlexander Aring new_node->mark = dlm_config.ci_mark; 46036b71a8bSDavid Teigland list_add(&new_node->list, &dlm_node_addrs); 46136b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 46236b71a8bSDavid Teigland return 0; 46336b71a8bSDavid Teigland } 46436b71a8bSDavid Teigland 4654f19d071SAlexander Aring ret = dlm_lowcomms_na_has_addr(na, addr); 4664f19d071SAlexander Aring if (ret) { 4674f19d071SAlexander Aring spin_unlock(&dlm_node_addrs_spin); 4684f19d071SAlexander Aring kfree(new_addr); 4694f19d071SAlexander Aring kfree(new_node); 4704f19d071SAlexander Aring return -EEXIST; 4714f19d071SAlexander Aring } 4724f19d071SAlexander Aring 47336b71a8bSDavid Teigland if (na->addr_count >= DLM_MAX_ADDR_COUNT) { 47436b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 47536b71a8bSDavid Teigland kfree(new_addr); 47636b71a8bSDavid Teigland kfree(new_node); 47736b71a8bSDavid Teigland return -ENOSPC; 47836b71a8bSDavid Teigland } 47936b71a8bSDavid Teigland 48036b71a8bSDavid Teigland na->addr[na->addr_count++] = new_addr; 48136b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 48236b71a8bSDavid Teigland kfree(new_node); 48336b71a8bSDavid Teigland return 0; 48436b71a8bSDavid Teigland } 48536b71a8bSDavid Teigland 4866ed7257bSPatrick Caulfield /* Data available on socket or listen socket received a connect */ 487676d2369SDavid S. Miller static void lowcomms_data_ready(struct sock *sk) 4886ed7257bSPatrick Caulfield { 48993eaadebStsutomu.owa@toshiba.co.jp struct connection *con; 49093eaadebStsutomu.owa@toshiba.co.jp 49193eaadebStsutomu.owa@toshiba.co.jp con = sock2con(sk); 492afb853fbSPatrick Caulfield if (con && !test_and_set_bit(CF_READ_PENDING, &con->flags)) 4936ed7257bSPatrick Caulfield queue_work(recv_workqueue, &con->rwork); 4946ed7257bSPatrick Caulfield } 4956ed7257bSPatrick Caulfield 496d11ccd45SAlexander Aring static void lowcomms_listen_data_ready(struct sock *sk) 497d11ccd45SAlexander Aring { 4989a4139a7SAlexander Aring if (!dlm_allow_conn) 4999a4139a7SAlexander Aring return; 5009a4139a7SAlexander Aring 501d11ccd45SAlexander Aring queue_work(recv_workqueue, &listen_con.rwork); 502d11ccd45SAlexander Aring } 503d11ccd45SAlexander Aring 5046ed7257bSPatrick Caulfield static void lowcomms_write_space(struct sock *sk) 5056ed7257bSPatrick Caulfield { 50693eaadebStsutomu.owa@toshiba.co.jp struct connection *con; 5076ed7257bSPatrick Caulfield 50893eaadebStsutomu.owa@toshiba.co.jp con = sock2con(sk); 509b36930ddSDavid Miller if (!con) 51092c44605SAlexander Aring return; 511b36930ddSDavid Miller 51219633c7eSAlexander Aring if (!test_and_set_bit(CF_CONNECTED, &con->flags)) { 51319633c7eSAlexander Aring log_print("successful connected to node %d", con->nodeid); 51419633c7eSAlexander Aring queue_work(send_workqueue, &con->swork); 51592c44605SAlexander Aring return; 51619633c7eSAlexander Aring } 51719633c7eSAlexander Aring 518b36930ddSDavid Miller clear_bit(SOCK_NOSPACE, &con->sock->flags); 519b36930ddSDavid Miller 520b36930ddSDavid Miller if (test_and_clear_bit(CF_APP_LIMITED, &con->flags)) { 521b36930ddSDavid Miller con->sock->sk->sk_write_pending--; 5229cd3e072SEric Dumazet clear_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags); 523b36930ddSDavid Miller } 524b36930ddSDavid Miller 5256ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 5266ed7257bSPatrick Caulfield } 5276ed7257bSPatrick Caulfield 5286ed7257bSPatrick Caulfield static inline void lowcomms_connect_sock(struct connection *con) 5296ed7257bSPatrick Caulfield { 530063c4c99SLars Marowsky-Bree if (test_bit(CF_CLOSE, &con->flags)) 531063c4c99SLars Marowsky-Bree return; 5326ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 53361d9102bSBob Peterson cond_resched(); 5346ed7257bSPatrick Caulfield } 5356ed7257bSPatrick Caulfield 5366ed7257bSPatrick Caulfield static void lowcomms_state_change(struct sock *sk) 5376ed7257bSPatrick Caulfield { 538ee44b4bcSMarcelo Ricardo Leitner /* SCTP layer is not calling sk_data_ready when the connection 539ee44b4bcSMarcelo Ricardo Leitner * is done, so we catch the signal through here. Also, it 540ee44b4bcSMarcelo Ricardo Leitner * doesn't switch socket state when entering shutdown, so we 541ee44b4bcSMarcelo Ricardo Leitner * skip the write in that case. 542ee44b4bcSMarcelo Ricardo Leitner */ 543ee44b4bcSMarcelo Ricardo Leitner if (sk->sk_shutdown) { 544ee44b4bcSMarcelo Ricardo Leitner if (sk->sk_shutdown == RCV_SHUTDOWN) 545ee44b4bcSMarcelo Ricardo Leitner lowcomms_data_ready(sk); 546ee44b4bcSMarcelo Ricardo Leitner } else if (sk->sk_state == TCP_ESTABLISHED) { 5476ed7257bSPatrick Caulfield lowcomms_write_space(sk); 5486ed7257bSPatrick Caulfield } 549ee44b4bcSMarcelo Ricardo Leitner } 5506ed7257bSPatrick Caulfield 551391fbdc5SChristine Caulfield int dlm_lowcomms_connect_node(int nodeid) 552391fbdc5SChristine Caulfield { 553391fbdc5SChristine Caulfield struct connection *con; 554b38bc9c2SAlexander Aring int idx; 555391fbdc5SChristine Caulfield 556391fbdc5SChristine Caulfield if (nodeid == dlm_our_nodeid()) 557391fbdc5SChristine Caulfield return 0; 558391fbdc5SChristine Caulfield 559b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 560391fbdc5SChristine Caulfield con = nodeid2con(nodeid, GFP_NOFS); 561b38bc9c2SAlexander Aring if (!con) { 562b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 563391fbdc5SChristine Caulfield return -ENOMEM; 564b38bc9c2SAlexander Aring } 565b38bc9c2SAlexander Aring 566391fbdc5SChristine Caulfield lowcomms_connect_sock(con); 567b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 568b38bc9c2SAlexander Aring 569391fbdc5SChristine Caulfield return 0; 570391fbdc5SChristine Caulfield } 571391fbdc5SChristine Caulfield 572e125fbebSAlexander Aring int dlm_lowcomms_nodes_set_mark(int nodeid, unsigned int mark) 573e125fbebSAlexander Aring { 574e125fbebSAlexander Aring struct dlm_node_addr *na; 575e125fbebSAlexander Aring 576e125fbebSAlexander Aring spin_lock(&dlm_node_addrs_spin); 577e125fbebSAlexander Aring na = find_node_addr(nodeid); 578e125fbebSAlexander Aring if (!na) { 579e125fbebSAlexander Aring spin_unlock(&dlm_node_addrs_spin); 580e125fbebSAlexander Aring return -ENOENT; 581e125fbebSAlexander Aring } 582e125fbebSAlexander Aring 583e125fbebSAlexander Aring na->mark = mark; 584e125fbebSAlexander Aring spin_unlock(&dlm_node_addrs_spin); 585e125fbebSAlexander Aring 586e125fbebSAlexander Aring return 0; 587e125fbebSAlexander Aring } 588e125fbebSAlexander Aring 589b3a5bbfdSBob Peterson static void lowcomms_error_report(struct sock *sk) 590b3a5bbfdSBob Peterson { 591b81171cbSBob Peterson struct connection *con; 592b81171cbSBob Peterson void (*orig_report)(struct sock *) = NULL; 5934c3d9057SAlexander Aring struct inet_sock *inet; 594b3a5bbfdSBob Peterson 595b81171cbSBob Peterson con = sock2con(sk); 596b81171cbSBob Peterson if (con == NULL) 597b81171cbSBob Peterson goto out; 598b81171cbSBob Peterson 599cc661fc9SBob Peterson orig_report = listen_sock.sk_error_report; 600b3a5bbfdSBob Peterson 6014c3d9057SAlexander Aring inet = inet_sk(sk); 6024c3d9057SAlexander Aring switch (sk->sk_family) { 6034c3d9057SAlexander Aring case AF_INET: 604b3a5bbfdSBob Peterson printk_ratelimited(KERN_ERR "dlm: node %d: socket error " 6054c3d9057SAlexander Aring "sending to node %d at %pI4, dport %d, " 606b3a5bbfdSBob Peterson "sk_err=%d/%d\n", dlm_our_nodeid(), 6074c3d9057SAlexander Aring con->nodeid, &inet->inet_daddr, 6084c3d9057SAlexander Aring ntohs(inet->inet_dport), sk->sk_err, 609b3a5bbfdSBob Peterson sk->sk_err_soft); 6104c3d9057SAlexander Aring break; 611*1b9beda8SAlexander Aring #if IS_ENABLED(CONFIG_IPV6) 6124c3d9057SAlexander Aring case AF_INET6: 613b3a5bbfdSBob Peterson printk_ratelimited(KERN_ERR "dlm: node %d: socket error " 6144c3d9057SAlexander Aring "sending to node %d at %pI6c, " 6154c3d9057SAlexander Aring "dport %d, sk_err=%d/%d\n", dlm_our_nodeid(), 6164c3d9057SAlexander Aring con->nodeid, &sk->sk_v6_daddr, 6174c3d9057SAlexander Aring ntohs(inet->inet_dport), sk->sk_err, 618b3a5bbfdSBob Peterson sk->sk_err_soft); 6194c3d9057SAlexander Aring break; 620*1b9beda8SAlexander Aring #endif 6214c3d9057SAlexander Aring default: 6224c3d9057SAlexander Aring printk_ratelimited(KERN_ERR "dlm: node %d: socket error " 6234c3d9057SAlexander Aring "invalid socket family %d set, " 6244c3d9057SAlexander Aring "sk_err=%d/%d\n", dlm_our_nodeid(), 6254c3d9057SAlexander Aring sk->sk_family, sk->sk_err, sk->sk_err_soft); 6264c3d9057SAlexander Aring goto out; 627b3a5bbfdSBob Peterson } 628ba868d9dSAlexander Aring 629ba868d9dSAlexander Aring /* below sendcon only handling */ 630ba868d9dSAlexander Aring if (test_bit(CF_IS_OTHERCON, &con->flags)) 631ba868d9dSAlexander Aring con = con->sendcon; 632ba868d9dSAlexander Aring 633ba868d9dSAlexander Aring switch (sk->sk_err) { 634ba868d9dSAlexander Aring case ECONNREFUSED: 635ba868d9dSAlexander Aring set_bit(CF_DELAY_CONNECT, &con->flags); 636ba868d9dSAlexander Aring break; 637ba868d9dSAlexander Aring default: 638ba868d9dSAlexander Aring break; 639ba868d9dSAlexander Aring } 640ba868d9dSAlexander Aring 641ba868d9dSAlexander Aring if (!test_and_set_bit(CF_RECONNECT, &con->flags)) 642ba868d9dSAlexander Aring queue_work(send_workqueue, &con->swork); 643ba868d9dSAlexander Aring 644b81171cbSBob Peterson out: 645b81171cbSBob Peterson if (orig_report) 646b81171cbSBob Peterson orig_report(sk); 647b81171cbSBob Peterson } 648b81171cbSBob Peterson 649b81171cbSBob Peterson /* Note: sk_callback_lock must be locked before calling this function. */ 650cc661fc9SBob Peterson static void save_listen_callbacks(struct socket *sock) 651b81171cbSBob Peterson { 652cc661fc9SBob Peterson struct sock *sk = sock->sk; 653cc661fc9SBob Peterson 654cc661fc9SBob Peterson listen_sock.sk_data_ready = sk->sk_data_ready; 655cc661fc9SBob Peterson listen_sock.sk_state_change = sk->sk_state_change; 656cc661fc9SBob Peterson listen_sock.sk_write_space = sk->sk_write_space; 657cc661fc9SBob Peterson listen_sock.sk_error_report = sk->sk_error_report; 658b81171cbSBob Peterson } 659b81171cbSBob Peterson 660cc661fc9SBob Peterson static void restore_callbacks(struct socket *sock) 661b81171cbSBob Peterson { 662cc661fc9SBob Peterson struct sock *sk = sock->sk; 663cc661fc9SBob Peterson 66492c44605SAlexander Aring lock_sock(sk); 665b81171cbSBob Peterson sk->sk_user_data = NULL; 666cc661fc9SBob Peterson sk->sk_data_ready = listen_sock.sk_data_ready; 667cc661fc9SBob Peterson sk->sk_state_change = listen_sock.sk_state_change; 668cc661fc9SBob Peterson sk->sk_write_space = listen_sock.sk_write_space; 669cc661fc9SBob Peterson sk->sk_error_report = listen_sock.sk_error_report; 67092c44605SAlexander Aring release_sock(sk); 671b3a5bbfdSBob Peterson } 672b3a5bbfdSBob Peterson 673d11ccd45SAlexander Aring static void add_listen_sock(struct socket *sock, struct listen_connection *con) 674d11ccd45SAlexander Aring { 675d11ccd45SAlexander Aring struct sock *sk = sock->sk; 676d11ccd45SAlexander Aring 67792c44605SAlexander Aring lock_sock(sk); 678d11ccd45SAlexander Aring save_listen_callbacks(sock); 679d11ccd45SAlexander Aring con->sock = sock; 680d11ccd45SAlexander Aring 681d11ccd45SAlexander Aring sk->sk_user_data = con; 682d11ccd45SAlexander Aring sk->sk_allocation = GFP_NOFS; 683d11ccd45SAlexander Aring /* Install a data_ready callback */ 684d11ccd45SAlexander Aring sk->sk_data_ready = lowcomms_listen_data_ready; 68592c44605SAlexander Aring release_sock(sk); 686d11ccd45SAlexander Aring } 687d11ccd45SAlexander Aring 6886ed7257bSPatrick Caulfield /* Make a socket active */ 689988419a9Stsutomu.owa@toshiba.co.jp static void add_sock(struct socket *sock, struct connection *con) 6906ed7257bSPatrick Caulfield { 691b81171cbSBob Peterson struct sock *sk = sock->sk; 692b81171cbSBob Peterson 69392c44605SAlexander Aring lock_sock(sk); 6946ed7257bSPatrick Caulfield con->sock = sock; 6956ed7257bSPatrick Caulfield 696b81171cbSBob Peterson sk->sk_user_data = con; 6976ed7257bSPatrick Caulfield /* Install a data_ready callback */ 698b81171cbSBob Peterson sk->sk_data_ready = lowcomms_data_ready; 699b81171cbSBob Peterson sk->sk_write_space = lowcomms_write_space; 700b81171cbSBob Peterson sk->sk_state_change = lowcomms_state_change; 701b81171cbSBob Peterson sk->sk_allocation = GFP_NOFS; 702b81171cbSBob Peterson sk->sk_error_report = lowcomms_error_report; 70392c44605SAlexander Aring release_sock(sk); 7046ed7257bSPatrick Caulfield } 7056ed7257bSPatrick Caulfield 7066ed7257bSPatrick Caulfield /* Add the port number to an IPv6 or 4 sockaddr and return the address 7076ed7257bSPatrick Caulfield length */ 7086ed7257bSPatrick Caulfield static void make_sockaddr(struct sockaddr_storage *saddr, uint16_t port, 7096ed7257bSPatrick Caulfield int *addr_len) 7106ed7257bSPatrick Caulfield { 7116ed7257bSPatrick Caulfield saddr->ss_family = dlm_local_addr[0]->ss_family; 7126ed7257bSPatrick Caulfield if (saddr->ss_family == AF_INET) { 7136ed7257bSPatrick Caulfield struct sockaddr_in *in4_addr = (struct sockaddr_in *)saddr; 7146ed7257bSPatrick Caulfield in4_addr->sin_port = cpu_to_be16(port); 7156ed7257bSPatrick Caulfield *addr_len = sizeof(struct sockaddr_in); 7166ed7257bSPatrick Caulfield memset(&in4_addr->sin_zero, 0, sizeof(in4_addr->sin_zero)); 7176ed7257bSPatrick Caulfield } else { 7186ed7257bSPatrick Caulfield struct sockaddr_in6 *in6_addr = (struct sockaddr_in6 *)saddr; 7196ed7257bSPatrick Caulfield in6_addr->sin6_port = cpu_to_be16(port); 7206ed7257bSPatrick Caulfield *addr_len = sizeof(struct sockaddr_in6); 7216ed7257bSPatrick Caulfield } 72201c8cab2SPatrick Caulfield memset((char *)saddr + *addr_len, 0, sizeof(struct sockaddr_storage) - *addr_len); 7236ed7257bSPatrick Caulfield } 7246ed7257bSPatrick Caulfield 725706474fbSAlexander Aring static void dlm_page_release(struct kref *kref) 726706474fbSAlexander Aring { 727706474fbSAlexander Aring struct writequeue_entry *e = container_of(kref, struct writequeue_entry, 728706474fbSAlexander Aring ref); 729706474fbSAlexander Aring 730706474fbSAlexander Aring __free_page(e->page); 731706474fbSAlexander Aring kfree(e); 732706474fbSAlexander Aring } 733706474fbSAlexander Aring 734706474fbSAlexander Aring static void dlm_msg_release(struct kref *kref) 735706474fbSAlexander Aring { 736706474fbSAlexander Aring struct dlm_msg *msg = container_of(kref, struct dlm_msg, ref); 737706474fbSAlexander Aring 738706474fbSAlexander Aring kref_put(&msg->entry->ref, dlm_page_release); 739706474fbSAlexander Aring kfree(msg); 740706474fbSAlexander Aring } 741706474fbSAlexander Aring 742706474fbSAlexander Aring static void free_entry(struct writequeue_entry *e) 743706474fbSAlexander Aring { 744706474fbSAlexander Aring struct dlm_msg *msg, *tmp; 745706474fbSAlexander Aring 746706474fbSAlexander Aring list_for_each_entry_safe(msg, tmp, &e->msgs, list) { 747706474fbSAlexander Aring if (msg->orig_msg) { 748706474fbSAlexander Aring msg->orig_msg->retransmit = false; 749706474fbSAlexander Aring kref_put(&msg->orig_msg->ref, dlm_msg_release); 750706474fbSAlexander Aring } 751706474fbSAlexander Aring 752706474fbSAlexander Aring list_del(&msg->list); 753706474fbSAlexander Aring kref_put(&msg->ref, dlm_msg_release); 754706474fbSAlexander Aring } 755706474fbSAlexander Aring 756706474fbSAlexander Aring list_del(&e->list); 757706474fbSAlexander Aring atomic_dec(&e->con->writequeue_cnt); 758706474fbSAlexander Aring kref_put(&e->ref, dlm_page_release); 759706474fbSAlexander Aring } 760706474fbSAlexander Aring 761d11ccd45SAlexander Aring static void dlm_close_sock(struct socket **sock) 762d11ccd45SAlexander Aring { 763d11ccd45SAlexander Aring if (*sock) { 764d11ccd45SAlexander Aring restore_callbacks(*sock); 765d11ccd45SAlexander Aring sock_release(*sock); 766d11ccd45SAlexander Aring *sock = NULL; 767d11ccd45SAlexander Aring } 768d11ccd45SAlexander Aring } 769d11ccd45SAlexander Aring 7706ed7257bSPatrick Caulfield /* Close a remote connection and tidy up */ 7710d737a8cSMarcelo Ricardo Leitner static void close_connection(struct connection *con, bool and_other, 7720d737a8cSMarcelo Ricardo Leitner bool tx, bool rx) 7736ed7257bSPatrick Caulfield { 774b2a66629Stsutomu.owa@toshiba.co.jp bool closing = test_and_set_bit(CF_CLOSING, &con->flags); 775706474fbSAlexander Aring struct writequeue_entry *e; 776b2a66629Stsutomu.owa@toshiba.co.jp 7770aa18464Stsutomu.owa@toshiba.co.jp if (tx && !closing && cancel_work_sync(&con->swork)) { 7780d737a8cSMarcelo Ricardo Leitner log_print("canceled swork for node %d", con->nodeid); 7790aa18464Stsutomu.owa@toshiba.co.jp clear_bit(CF_WRITE_PENDING, &con->flags); 7800aa18464Stsutomu.owa@toshiba.co.jp } 7810aa18464Stsutomu.owa@toshiba.co.jp if (rx && !closing && cancel_work_sync(&con->rwork)) { 7820d737a8cSMarcelo Ricardo Leitner log_print("canceled rwork for node %d", con->nodeid); 7830aa18464Stsutomu.owa@toshiba.co.jp clear_bit(CF_READ_PENDING, &con->flags); 7840aa18464Stsutomu.owa@toshiba.co.jp } 7856ed7257bSPatrick Caulfield 7860d737a8cSMarcelo Ricardo Leitner mutex_lock(&con->sock_mutex); 787d11ccd45SAlexander Aring dlm_close_sock(&con->sock); 788d11ccd45SAlexander Aring 7896ed7257bSPatrick Caulfield if (con->othercon && and_other) { 7906ed7257bSPatrick Caulfield /* Will only re-enter once. */ 791c6aa00e3SAlexander Aring close_connection(con->othercon, false, tx, rx); 7926ed7257bSPatrick Caulfield } 7939e5f2825SPatrick Caulfield 794706474fbSAlexander Aring /* if we send a writequeue entry only a half way, we drop the 795706474fbSAlexander Aring * whole entry because reconnection and that we not start of the 796706474fbSAlexander Aring * middle of a msg which will confuse the other end. 797706474fbSAlexander Aring * 798706474fbSAlexander Aring * we can always drop messages because retransmits, but what we 799706474fbSAlexander Aring * cannot allow is to transmit half messages which may be processed 800706474fbSAlexander Aring * at the other side. 801706474fbSAlexander Aring * 802706474fbSAlexander Aring * our policy is to start on a clean state when disconnects, we don't 803706474fbSAlexander Aring * know what's send/received on transport layer in this case. 804706474fbSAlexander Aring */ 805706474fbSAlexander Aring spin_lock(&con->writequeue_lock); 806706474fbSAlexander Aring if (!list_empty(&con->writequeue)) { 807706474fbSAlexander Aring e = list_first_entry(&con->writequeue, struct writequeue_entry, 808706474fbSAlexander Aring list); 809706474fbSAlexander Aring if (e->dirty) 810706474fbSAlexander Aring free_entry(e); 811706474fbSAlexander Aring } 812706474fbSAlexander Aring spin_unlock(&con->writequeue_lock); 813706474fbSAlexander Aring 8144798cbbfSAlexander Aring con->rx_leftover = 0; 8156ed7257bSPatrick Caulfield con->retries = 0; 816052849beSAlexander Aring clear_bit(CF_APP_LIMITED, &con->flags); 81719633c7eSAlexander Aring clear_bit(CF_CONNECTED, &con->flags); 818ba868d9dSAlexander Aring clear_bit(CF_DELAY_CONNECT, &con->flags); 819ba868d9dSAlexander Aring clear_bit(CF_RECONNECT, &con->flags); 8208aa31cbfSAlexander Aring clear_bit(CF_EOF, &con->flags); 8216ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 822b2a66629Stsutomu.owa@toshiba.co.jp clear_bit(CF_CLOSING, &con->flags); 8236ed7257bSPatrick Caulfield } 8246ed7257bSPatrick Caulfield 825055923bfSAlexander Aring static void shutdown_connection(struct connection *con) 826055923bfSAlexander Aring { 827055923bfSAlexander Aring int ret; 828055923bfSAlexander Aring 829eec054b5SAlexander Aring flush_work(&con->swork); 830055923bfSAlexander Aring 831055923bfSAlexander Aring mutex_lock(&con->sock_mutex); 832055923bfSAlexander Aring /* nothing to shutdown */ 833055923bfSAlexander Aring if (!con->sock) { 834055923bfSAlexander Aring mutex_unlock(&con->sock_mutex); 835055923bfSAlexander Aring return; 836055923bfSAlexander Aring } 837055923bfSAlexander Aring 838055923bfSAlexander Aring set_bit(CF_SHUTDOWN, &con->flags); 839055923bfSAlexander Aring ret = kernel_sock_shutdown(con->sock, SHUT_WR); 840055923bfSAlexander Aring mutex_unlock(&con->sock_mutex); 841055923bfSAlexander Aring if (ret) { 842055923bfSAlexander Aring log_print("Connection %p failed to shutdown: %d will force close", 843055923bfSAlexander Aring con, ret); 844055923bfSAlexander Aring goto force_close; 845055923bfSAlexander Aring } else { 846055923bfSAlexander Aring ret = wait_event_timeout(con->shutdown_wait, 847055923bfSAlexander Aring !test_bit(CF_SHUTDOWN, &con->flags), 848055923bfSAlexander Aring DLM_SHUTDOWN_WAIT_TIMEOUT); 849055923bfSAlexander Aring if (ret == 0) { 850055923bfSAlexander Aring log_print("Connection %p shutdown timed out, will force close", 851055923bfSAlexander Aring con); 852055923bfSAlexander Aring goto force_close; 853055923bfSAlexander Aring } 854055923bfSAlexander Aring } 855055923bfSAlexander Aring 856055923bfSAlexander Aring return; 857055923bfSAlexander Aring 858055923bfSAlexander Aring force_close: 859055923bfSAlexander Aring clear_bit(CF_SHUTDOWN, &con->flags); 860055923bfSAlexander Aring close_connection(con, false, true, true); 861055923bfSAlexander Aring } 862055923bfSAlexander Aring 863055923bfSAlexander Aring static void dlm_tcp_shutdown(struct connection *con) 864055923bfSAlexander Aring { 865055923bfSAlexander Aring if (con->othercon) 866055923bfSAlexander Aring shutdown_connection(con->othercon); 867055923bfSAlexander Aring shutdown_connection(con); 868055923bfSAlexander Aring } 869055923bfSAlexander Aring 8704798cbbfSAlexander Aring static int con_realloc_receive_buf(struct connection *con, int newlen) 8714798cbbfSAlexander Aring { 8724798cbbfSAlexander Aring unsigned char *newbuf; 8734798cbbfSAlexander Aring 8744798cbbfSAlexander Aring newbuf = kmalloc(newlen, GFP_NOFS); 8754798cbbfSAlexander Aring if (!newbuf) 8764798cbbfSAlexander Aring return -ENOMEM; 8774798cbbfSAlexander Aring 8784798cbbfSAlexander Aring /* copy any leftover from last receive */ 8794798cbbfSAlexander Aring if (con->rx_leftover) 8804798cbbfSAlexander Aring memmove(newbuf, con->rx_buf, con->rx_leftover); 8814798cbbfSAlexander Aring 8824798cbbfSAlexander Aring /* swap to new buffer space */ 8834798cbbfSAlexander Aring kfree(con->rx_buf); 8844798cbbfSAlexander Aring con->rx_buflen = newlen; 8854798cbbfSAlexander Aring con->rx_buf = newbuf; 8864798cbbfSAlexander Aring 8874798cbbfSAlexander Aring return 0; 8884798cbbfSAlexander Aring } 8894798cbbfSAlexander Aring 8906ed7257bSPatrick Caulfield /* Data received from remote end */ 8916ed7257bSPatrick Caulfield static int receive_from_sock(struct connection *con) 8926ed7257bSPatrick Caulfield { 8934798cbbfSAlexander Aring struct msghdr msg; 8944798cbbfSAlexander Aring struct kvec iov; 8954798cbbfSAlexander Aring int ret, buflen; 8966ed7257bSPatrick Caulfield 8976ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 8986ed7257bSPatrick Caulfield 8996ed7257bSPatrick Caulfield if (con->sock == NULL) { 9006ed7257bSPatrick Caulfield ret = -EAGAIN; 9016ed7257bSPatrick Caulfield goto out_close; 9026ed7257bSPatrick Caulfield } 9034798cbbfSAlexander Aring 9044798cbbfSAlexander Aring /* realloc if we get new buffer size to read out */ 9054798cbbfSAlexander Aring buflen = dlm_config.ci_buffer_size; 9064798cbbfSAlexander Aring if (con->rx_buflen != buflen && con->rx_leftover <= buflen) { 9074798cbbfSAlexander Aring ret = con_realloc_receive_buf(con, buflen); 9084798cbbfSAlexander Aring if (ret < 0) 9096ed7257bSPatrick Caulfield goto out_resched; 9106ed7257bSPatrick Caulfield } 9116ed7257bSPatrick Caulfield 91262699b3fSAlexander Aring for (;;) { 9134798cbbfSAlexander Aring /* calculate new buffer parameter regarding last receive and 9144798cbbfSAlexander Aring * possible leftover bytes 9156ed7257bSPatrick Caulfield */ 9164798cbbfSAlexander Aring iov.iov_base = con->rx_buf + con->rx_leftover; 9174798cbbfSAlexander Aring iov.iov_len = con->rx_buflen - con->rx_leftover; 9186ed7257bSPatrick Caulfield 9194798cbbfSAlexander Aring memset(&msg, 0, sizeof(msg)); 9204798cbbfSAlexander Aring msg.msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL; 9214798cbbfSAlexander Aring ret = kernel_recvmsg(con->sock, &msg, &iov, 1, iov.iov_len, 9224798cbbfSAlexander Aring msg.msg_flags); 92392732376SAlexander Aring trace_dlm_recv(con->nodeid, ret); 92462699b3fSAlexander Aring if (ret == -EAGAIN) 92562699b3fSAlexander Aring break; 92662699b3fSAlexander Aring else if (ret <= 0) 9276ed7257bSPatrick Caulfield goto out_close; 9286ed7257bSPatrick Caulfield 9294798cbbfSAlexander Aring /* new buflen according readed bytes and leftover from last receive */ 9304798cbbfSAlexander Aring buflen = ret + con->rx_leftover; 9314798cbbfSAlexander Aring ret = dlm_process_incoming_buffer(con->nodeid, con->rx_buf, buflen); 9324798cbbfSAlexander Aring if (ret < 0) 9334798cbbfSAlexander Aring goto out_close; 9346ed7257bSPatrick Caulfield 9354798cbbfSAlexander Aring /* calculate leftover bytes from process and put it into begin of 9364798cbbfSAlexander Aring * the receive buffer, so next receive we have the full message 9374798cbbfSAlexander Aring * at the start address of the receive buffer. 9384798cbbfSAlexander Aring */ 9394798cbbfSAlexander Aring con->rx_leftover = buflen - ret; 9404798cbbfSAlexander Aring if (con->rx_leftover) { 9414798cbbfSAlexander Aring memmove(con->rx_buf, con->rx_buf + ret, 9424798cbbfSAlexander Aring con->rx_leftover); 9436ed7257bSPatrick Caulfield } 94462699b3fSAlexander Aring } 9454798cbbfSAlexander Aring 946b97f8525SAlexander Aring dlm_midcomms_receive_done(con->nodeid); 9476ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 9486ed7257bSPatrick Caulfield return 0; 9496ed7257bSPatrick Caulfield 9506ed7257bSPatrick Caulfield out_resched: 9516ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_READ_PENDING, &con->flags)) 9526ed7257bSPatrick Caulfield queue_work(recv_workqueue, &con->rwork); 9536ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 9546ed7257bSPatrick Caulfield return -EAGAIN; 9556ed7257bSPatrick Caulfield 9566ed7257bSPatrick Caulfield out_close: 957055923bfSAlexander Aring if (ret == 0) { 958055923bfSAlexander Aring log_print("connection %p got EOF from %d", 959055923bfSAlexander Aring con, con->nodeid); 9608aa31cbfSAlexander Aring 961a66c008cSAlexander Aring if (dlm_proto_ops->eof_condition && 962a66c008cSAlexander Aring dlm_proto_ops->eof_condition(con)) { 9638aa31cbfSAlexander Aring set_bit(CF_EOF, &con->flags); 9648aa31cbfSAlexander Aring mutex_unlock(&con->sock_mutex); 9658aa31cbfSAlexander Aring } else { 9668aa31cbfSAlexander Aring mutex_unlock(&con->sock_mutex); 9678aa31cbfSAlexander Aring close_connection(con, false, true, false); 9688aa31cbfSAlexander Aring 969055923bfSAlexander Aring /* handling for tcp shutdown */ 970055923bfSAlexander Aring clear_bit(CF_SHUTDOWN, &con->flags); 971055923bfSAlexander Aring wake_up(&con->shutdown_wait); 9728aa31cbfSAlexander Aring } 9738aa31cbfSAlexander Aring 974055923bfSAlexander Aring /* signal to breaking receive worker */ 975055923bfSAlexander Aring ret = -1; 9768aa31cbfSAlexander Aring } else { 9778aa31cbfSAlexander Aring mutex_unlock(&con->sock_mutex); 9786ed7257bSPatrick Caulfield } 9796ed7257bSPatrick Caulfield return ret; 9806ed7257bSPatrick Caulfield } 9816ed7257bSPatrick Caulfield 9826ed7257bSPatrick Caulfield /* Listening socket is busy, accept a connection */ 983d11ccd45SAlexander Aring static int accept_from_sock(struct listen_connection *con) 9846ed7257bSPatrick Caulfield { 9856ed7257bSPatrick Caulfield int result; 9866ed7257bSPatrick Caulfield struct sockaddr_storage peeraddr; 9876ed7257bSPatrick Caulfield struct socket *newsock; 988b38bc9c2SAlexander Aring int len, idx; 9896ed7257bSPatrick Caulfield int nodeid; 9906ed7257bSPatrick Caulfield struct connection *newcon; 9916ed7257bSPatrick Caulfield struct connection *addcon; 9923f78cd7dSAlexander Aring unsigned int mark; 9936ed7257bSPatrick Caulfield 994d11ccd45SAlexander Aring if (!con->sock) 9953421fb15Stsutomu.owa@toshiba.co.jp return -ENOTCONN; 9966ed7257bSPatrick Caulfield 9973421fb15Stsutomu.owa@toshiba.co.jp result = kernel_accept(con->sock, &newsock, O_NONBLOCK); 9986ed7257bSPatrick Caulfield if (result < 0) 9996ed7257bSPatrick Caulfield goto accept_err; 10006ed7257bSPatrick Caulfield 10016ed7257bSPatrick Caulfield /* Get the connected socket's peer */ 10026ed7257bSPatrick Caulfield memset(&peeraddr, 0, sizeof(peeraddr)); 10039b2c45d4SDenys Vlasenko len = newsock->ops->getname(newsock, (struct sockaddr *)&peeraddr, 2); 10049b2c45d4SDenys Vlasenko if (len < 0) { 10056ed7257bSPatrick Caulfield result = -ECONNABORTED; 10066ed7257bSPatrick Caulfield goto accept_err; 10076ed7257bSPatrick Caulfield } 10086ed7257bSPatrick Caulfield 10096ed7257bSPatrick Caulfield /* Get the new node's NODEID */ 10106ed7257bSPatrick Caulfield make_sockaddr(&peeraddr, 0, &len); 1011e125fbebSAlexander Aring if (addr_to_nodeid(&peeraddr, &nodeid, &mark)) { 1012bcaadf5cSMasatake YAMATO unsigned char *b=(unsigned char *)&peeraddr; 1013617e82e1SDavid Teigland log_print("connect from non cluster node"); 1014bcaadf5cSMasatake YAMATO print_hex_dump_bytes("ss: ", DUMP_PREFIX_NONE, 1015bcaadf5cSMasatake YAMATO b, sizeof(struct sockaddr_storage)); 10166ed7257bSPatrick Caulfield sock_release(newsock); 10176ed7257bSPatrick Caulfield return -1; 10186ed7257bSPatrick Caulfield } 10196ed7257bSPatrick Caulfield 10206ed7257bSPatrick Caulfield log_print("got connection from %d", nodeid); 10216ed7257bSPatrick Caulfield 10226ed7257bSPatrick Caulfield /* Check to see if we already have a connection to this node. This 10236ed7257bSPatrick Caulfield * could happen if the two nodes initiate a connection at roughly 10246ed7257bSPatrick Caulfield * the same time and the connections cross on the wire. 10256ed7257bSPatrick Caulfield * In this case we store the incoming one in "othercon" 10266ed7257bSPatrick Caulfield */ 1027b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 1028748285ccSDavid Teigland newcon = nodeid2con(nodeid, GFP_NOFS); 10296ed7257bSPatrick Caulfield if (!newcon) { 1030b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 10316ed7257bSPatrick Caulfield result = -ENOMEM; 10326ed7257bSPatrick Caulfield goto accept_err; 10336ed7257bSPatrick Caulfield } 1034d11ccd45SAlexander Aring 1035e125fbebSAlexander Aring sock_set_mark(newsock->sk, mark); 1036e125fbebSAlexander Aring 1037d11ccd45SAlexander Aring mutex_lock(&newcon->sock_mutex); 10386ed7257bSPatrick Caulfield if (newcon->sock) { 10396ed7257bSPatrick Caulfield struct connection *othercon = newcon->othercon; 10406ed7257bSPatrick Caulfield 10416ed7257bSPatrick Caulfield if (!othercon) { 1042a47666ebSAlexander Aring othercon = kzalloc(sizeof(*othercon), GFP_NOFS); 10436ed7257bSPatrick Caulfield if (!othercon) { 1044617e82e1SDavid Teigland log_print("failed to allocate incoming socket"); 10456ed7257bSPatrick Caulfield mutex_unlock(&newcon->sock_mutex); 1046b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 10476ed7257bSPatrick Caulfield result = -ENOMEM; 10486ed7257bSPatrick Caulfield goto accept_err; 10496ed7257bSPatrick Caulfield } 10504798cbbfSAlexander Aring 10516cde210aSAlexander Aring result = dlm_con_init(othercon, nodeid); 10526cde210aSAlexander Aring if (result < 0) { 10534798cbbfSAlexander Aring kfree(othercon); 10542fd8db2dSYang Yingliang mutex_unlock(&newcon->sock_mutex); 1055b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 10564798cbbfSAlexander Aring goto accept_err; 10574798cbbfSAlexander Aring } 10584798cbbfSAlexander Aring 1059e9a470acSAlexander Aring lockdep_set_subclass(&othercon->sock_mutex, 1); 10607443bc96SAlexander Aring set_bit(CF_IS_OTHERCON, &othercon->flags); 10616cde210aSAlexander Aring newcon->othercon = othercon; 1062ba868d9dSAlexander Aring othercon->sendcon = newcon; 1063ba3ab3caSAlexander Aring } else { 1064ba3ab3caSAlexander Aring /* close other sock con if we have something new */ 1065ba3ab3caSAlexander Aring close_connection(othercon, false, true, false); 106661d96be0SPatrick Caulfield } 1067ba3ab3caSAlexander Aring 1068e9a470acSAlexander Aring mutex_lock(&othercon->sock_mutex); 1069988419a9Stsutomu.owa@toshiba.co.jp add_sock(newsock, othercon); 10706ed7257bSPatrick Caulfield addcon = othercon; 1071c7355827Stsutomu.owa@toshiba.co.jp mutex_unlock(&othercon->sock_mutex); 10726ed7257bSPatrick Caulfield } 10736ed7257bSPatrick Caulfield else { 10743735b4b9SBob Peterson /* accept copies the sk after we've saved the callbacks, so we 10753735b4b9SBob Peterson don't want to save them a second time or comm errors will 10763735b4b9SBob Peterson result in calling sk_error_report recursively. */ 1077988419a9Stsutomu.owa@toshiba.co.jp add_sock(newsock, newcon); 10786ed7257bSPatrick Caulfield addcon = newcon; 10796ed7257bSPatrick Caulfield } 10806ed7257bSPatrick Caulfield 1081b30a624fSAlexander Aring set_bit(CF_CONNECTED, &addcon->flags); 10826ed7257bSPatrick Caulfield mutex_unlock(&newcon->sock_mutex); 10836ed7257bSPatrick Caulfield 10846ed7257bSPatrick Caulfield /* 10856ed7257bSPatrick Caulfield * Add it to the active queue in case we got data 108625985edcSLucas De Marchi * between processing the accept adding the socket 10876ed7257bSPatrick Caulfield * to the read_sockets list 10886ed7257bSPatrick Caulfield */ 10896ed7257bSPatrick Caulfield if (!test_and_set_bit(CF_READ_PENDING, &addcon->flags)) 10906ed7257bSPatrick Caulfield queue_work(recv_workqueue, &addcon->rwork); 10916ed7257bSPatrick Caulfield 1092b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 1093b38bc9c2SAlexander Aring 10946ed7257bSPatrick Caulfield return 0; 10956ed7257bSPatrick Caulfield 10966ed7257bSPatrick Caulfield accept_err: 10973421fb15Stsutomu.owa@toshiba.co.jp if (newsock) 10986ed7257bSPatrick Caulfield sock_release(newsock); 10996ed7257bSPatrick Caulfield 11006ed7257bSPatrick Caulfield if (result != -EAGAIN) 1101617e82e1SDavid Teigland log_print("error accepting connection from node: %d", result); 11026ed7257bSPatrick Caulfield return result; 11036ed7257bSPatrick Caulfield } 11046ed7257bSPatrick Caulfield 11055d689871SMike Christie /* 11065d689871SMike Christie * writequeue_entry_complete - try to delete and free write queue entry 11075d689871SMike Christie * @e: write queue entry to try to delete 11085d689871SMike Christie * @completed: bytes completed 11095d689871SMike Christie * 11105d689871SMike Christie * writequeue_lock must be held. 11115d689871SMike Christie */ 11125d689871SMike Christie static void writequeue_entry_complete(struct writequeue_entry *e, int completed) 11135d689871SMike Christie { 11145d689871SMike Christie e->offset += completed; 11155d689871SMike Christie e->len -= completed; 1116706474fbSAlexander Aring /* signal that page was half way transmitted */ 1117706474fbSAlexander Aring e->dirty = true; 11185d689871SMike Christie 11198f2dc78dSAlexander Aring if (e->len == 0 && e->users == 0) 11205d689871SMike Christie free_entry(e); 11215d689871SMike Christie } 11225d689871SMike Christie 1123ee44b4bcSMarcelo Ricardo Leitner /* 1124ee44b4bcSMarcelo Ricardo Leitner * sctp_bind_addrs - bind a SCTP socket to all our addresses 1125ee44b4bcSMarcelo Ricardo Leitner */ 112613004e8aSAlexander Aring static int sctp_bind_addrs(struct socket *sock, uint16_t port) 1127ee44b4bcSMarcelo Ricardo Leitner { 1128ee44b4bcSMarcelo Ricardo Leitner struct sockaddr_storage localaddr; 1129c0425a42SChristoph Hellwig struct sockaddr *addr = (struct sockaddr *)&localaddr; 1130ee44b4bcSMarcelo Ricardo Leitner int i, addr_len, result = 0; 1131ee44b4bcSMarcelo Ricardo Leitner 1132ee44b4bcSMarcelo Ricardo Leitner for (i = 0; i < dlm_local_count; i++) { 1133ee44b4bcSMarcelo Ricardo Leitner memcpy(&localaddr, dlm_local_addr[i], sizeof(localaddr)); 1134ee44b4bcSMarcelo Ricardo Leitner make_sockaddr(&localaddr, port, &addr_len); 1135ee44b4bcSMarcelo Ricardo Leitner 1136ee44b4bcSMarcelo Ricardo Leitner if (!i) 113713004e8aSAlexander Aring result = kernel_bind(sock, addr, addr_len); 1138ee44b4bcSMarcelo Ricardo Leitner else 113913004e8aSAlexander Aring result = sock_bind_add(sock->sk, addr, addr_len); 1140ee44b4bcSMarcelo Ricardo Leitner 1141ee44b4bcSMarcelo Ricardo Leitner if (result < 0) { 1142ee44b4bcSMarcelo Ricardo Leitner log_print("Can't bind to %d addr number %d, %d.\n", 1143ee44b4bcSMarcelo Ricardo Leitner port, i + 1, result); 1144ee44b4bcSMarcelo Ricardo Leitner break; 1145ee44b4bcSMarcelo Ricardo Leitner } 1146ee44b4bcSMarcelo Ricardo Leitner } 1147ee44b4bcSMarcelo Ricardo Leitner return result; 1148ee44b4bcSMarcelo Ricardo Leitner } 1149ee44b4bcSMarcelo Ricardo Leitner 11506ed7257bSPatrick Caulfield /* Get local addresses */ 11516ed7257bSPatrick Caulfield static void init_local(void) 11526ed7257bSPatrick Caulfield { 11536ed7257bSPatrick Caulfield struct sockaddr_storage sas, *addr; 11546ed7257bSPatrick Caulfield int i; 11556ed7257bSPatrick Caulfield 115630d3a237SPatrick Caulfield dlm_local_count = 0; 11571b189b88SDavid Teigland for (i = 0; i < DLM_MAX_ADDR_COUNT; i++) { 11586ed7257bSPatrick Caulfield if (dlm_our_addr(&sas, i)) 11596ed7257bSPatrick Caulfield break; 11606ed7257bSPatrick Caulfield 11615c93f56fSAmitoj Kaur Chawla addr = kmemdup(&sas, sizeof(*addr), GFP_NOFS); 11626ed7257bSPatrick Caulfield if (!addr) 11636ed7257bSPatrick Caulfield break; 11646ed7257bSPatrick Caulfield dlm_local_addr[dlm_local_count++] = addr; 11656ed7257bSPatrick Caulfield } 11666ed7257bSPatrick Caulfield } 11676ed7257bSPatrick Caulfield 1168043697f0SAlexander Aring static void deinit_local(void) 1169043697f0SAlexander Aring { 1170043697f0SAlexander Aring int i; 1171043697f0SAlexander Aring 1172043697f0SAlexander Aring for (i = 0; i < dlm_local_count; i++) 1173043697f0SAlexander Aring kfree(dlm_local_addr[i]); 1174043697f0SAlexander Aring } 1175043697f0SAlexander Aring 11766ed7257bSPatrick Caulfield static struct writequeue_entry *new_writequeue_entry(struct connection *con, 11776ed7257bSPatrick Caulfield gfp_t allocation) 11786ed7257bSPatrick Caulfield { 11796ed7257bSPatrick Caulfield struct writequeue_entry *entry; 11806ed7257bSPatrick Caulfield 1181f0747ebfSAlexander Aring entry = kzalloc(sizeof(*entry), allocation); 11826ed7257bSPatrick Caulfield if (!entry) 11836ed7257bSPatrick Caulfield return NULL; 11846ed7257bSPatrick Caulfield 1185e1a7cbceSAlexander Aring entry->page = alloc_page(allocation | __GFP_ZERO); 11866ed7257bSPatrick Caulfield if (!entry->page) { 11876ed7257bSPatrick Caulfield kfree(entry); 11886ed7257bSPatrick Caulfield return NULL; 11896ed7257bSPatrick Caulfield } 11906ed7257bSPatrick Caulfield 11916ed7257bSPatrick Caulfield entry->con = con; 1192f0747ebfSAlexander Aring entry->users = 1; 11938f2dc78dSAlexander Aring kref_init(&entry->ref); 11948f2dc78dSAlexander Aring INIT_LIST_HEAD(&entry->msgs); 11956ed7257bSPatrick Caulfield 11966ed7257bSPatrick Caulfield return entry; 11976ed7257bSPatrick Caulfield } 11986ed7257bSPatrick Caulfield 1199f0747ebfSAlexander Aring static struct writequeue_entry *new_wq_entry(struct connection *con, int len, 12008f2dc78dSAlexander Aring gfp_t allocation, char **ppc, 12015c16febbSAlexander Aring void (*cb)(void *data), void *data) 1202f0747ebfSAlexander Aring { 1203f0747ebfSAlexander Aring struct writequeue_entry *e; 1204f0747ebfSAlexander Aring 1205f0747ebfSAlexander Aring spin_lock(&con->writequeue_lock); 1206f0747ebfSAlexander Aring if (!list_empty(&con->writequeue)) { 1207f0747ebfSAlexander Aring e = list_last_entry(&con->writequeue, struct writequeue_entry, list); 1208f0747ebfSAlexander Aring if (DLM_WQ_REMAIN_BYTES(e) >= len) { 12098f2dc78dSAlexander Aring kref_get(&e->ref); 12108f2dc78dSAlexander Aring 1211f0747ebfSAlexander Aring *ppc = page_address(e->page) + e->end; 12128f2dc78dSAlexander Aring if (cb) 12135c16febbSAlexander Aring cb(data); 12148f2dc78dSAlexander Aring 1215f0747ebfSAlexander Aring e->end += len; 1216f0747ebfSAlexander Aring e->users++; 1217f0747ebfSAlexander Aring spin_unlock(&con->writequeue_lock); 1218f0747ebfSAlexander Aring 1219f0747ebfSAlexander Aring return e; 1220f0747ebfSAlexander Aring } 1221f0747ebfSAlexander Aring } 1222f0747ebfSAlexander Aring spin_unlock(&con->writequeue_lock); 1223f0747ebfSAlexander Aring 1224f0747ebfSAlexander Aring e = new_writequeue_entry(con, allocation); 1225f0747ebfSAlexander Aring if (!e) 1226f0747ebfSAlexander Aring return NULL; 1227f0747ebfSAlexander Aring 12288f2dc78dSAlexander Aring kref_get(&e->ref); 1229f0747ebfSAlexander Aring *ppc = page_address(e->page); 1230f0747ebfSAlexander Aring e->end += len; 12318aa31cbfSAlexander Aring atomic_inc(&con->writequeue_cnt); 1232f0747ebfSAlexander Aring 1233f0747ebfSAlexander Aring spin_lock(&con->writequeue_lock); 12348f2dc78dSAlexander Aring if (cb) 12355c16febbSAlexander Aring cb(data); 12368f2dc78dSAlexander Aring 1237f0747ebfSAlexander Aring list_add_tail(&e->list, &con->writequeue); 1238f0747ebfSAlexander Aring spin_unlock(&con->writequeue_lock); 1239f0747ebfSAlexander Aring 1240f0747ebfSAlexander Aring return e; 1241f0747ebfSAlexander Aring }; 1242f0747ebfSAlexander Aring 12432874d1a6SAlexander Aring static struct dlm_msg *dlm_lowcomms_new_msg_con(struct connection *con, int len, 12442874d1a6SAlexander Aring gfp_t allocation, char **ppc, 12455c16febbSAlexander Aring void (*cb)(void *data), 12465c16febbSAlexander Aring void *data) 12472874d1a6SAlexander Aring { 12482874d1a6SAlexander Aring struct writequeue_entry *e; 12492874d1a6SAlexander Aring struct dlm_msg *msg; 1250c51b0221SAlexander Aring bool sleepable; 12512874d1a6SAlexander Aring 12522874d1a6SAlexander Aring msg = kzalloc(sizeof(*msg), allocation); 12532874d1a6SAlexander Aring if (!msg) 12542874d1a6SAlexander Aring return NULL; 12552874d1a6SAlexander Aring 1256c51b0221SAlexander Aring /* this mutex is being used as a wait to avoid multiple "fast" 1257c51b0221SAlexander Aring * new writequeue page list entry allocs in new_wq_entry in 1258c51b0221SAlexander Aring * normal operation which is sleepable context. Without it 1259c51b0221SAlexander Aring * we could end in multiple writequeue entries with one 1260c51b0221SAlexander Aring * dlm message because multiple callers were waiting at 1261c51b0221SAlexander Aring * the writequeue_lock in new_wq_entry(). 1262c51b0221SAlexander Aring */ 1263c51b0221SAlexander Aring sleepable = gfpflags_normal_context(allocation); 1264c51b0221SAlexander Aring if (sleepable) 1265c51b0221SAlexander Aring mutex_lock(&con->wq_alloc); 1266c51b0221SAlexander Aring 12672874d1a6SAlexander Aring kref_init(&msg->ref); 12682874d1a6SAlexander Aring 12695c16febbSAlexander Aring e = new_wq_entry(con, len, allocation, ppc, cb, data); 12702874d1a6SAlexander Aring if (!e) { 1271c51b0221SAlexander Aring if (sleepable) 1272c51b0221SAlexander Aring mutex_unlock(&con->wq_alloc); 1273c51b0221SAlexander Aring 12742874d1a6SAlexander Aring kfree(msg); 12752874d1a6SAlexander Aring return NULL; 12762874d1a6SAlexander Aring } 12772874d1a6SAlexander Aring 1278c51b0221SAlexander Aring if (sleepable) 1279c51b0221SAlexander Aring mutex_unlock(&con->wq_alloc); 1280c51b0221SAlexander Aring 12812874d1a6SAlexander Aring msg->ppc = *ppc; 12822874d1a6SAlexander Aring msg->len = len; 12832874d1a6SAlexander Aring msg->entry = e; 12842874d1a6SAlexander Aring 12852874d1a6SAlexander Aring return msg; 12862874d1a6SAlexander Aring } 12872874d1a6SAlexander Aring 12888f2dc78dSAlexander Aring struct dlm_msg *dlm_lowcomms_new_msg(int nodeid, int len, gfp_t allocation, 12895c16febbSAlexander Aring char **ppc, void (*cb)(void *data), 12905c16febbSAlexander Aring void *data) 12916ed7257bSPatrick Caulfield { 12926ed7257bSPatrick Caulfield struct connection *con; 12938f2dc78dSAlexander Aring struct dlm_msg *msg; 1294b38bc9c2SAlexander Aring int idx; 12956ed7257bSPatrick Caulfield 1296d10a0b88SAlexander Aring if (len > DLM_MAX_SOCKET_BUFSIZE || 1297c45674fbSAlexander Aring len < sizeof(struct dlm_header)) { 1298d10a0b88SAlexander Aring BUILD_BUG_ON(PAGE_SIZE < DLM_MAX_SOCKET_BUFSIZE); 1299692f51c8SAlexander Aring log_print("failed to allocate a buffer of size %d", len); 1300c45674fbSAlexander Aring WARN_ON(1); 1301692f51c8SAlexander Aring return NULL; 1302692f51c8SAlexander Aring } 1303692f51c8SAlexander Aring 1304b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 13056ed7257bSPatrick Caulfield con = nodeid2con(nodeid, allocation); 1306b38bc9c2SAlexander Aring if (!con) { 1307b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 13086ed7257bSPatrick Caulfield return NULL; 1309b38bc9c2SAlexander Aring } 13106ed7257bSPatrick Caulfield 13115c16febbSAlexander Aring msg = dlm_lowcomms_new_msg_con(con, len, allocation, ppc, cb, data); 13128f2dc78dSAlexander Aring if (!msg) { 1313b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 1314b38bc9c2SAlexander Aring return NULL; 1315b38bc9c2SAlexander Aring } 1316b38bc9c2SAlexander Aring 13178f2dc78dSAlexander Aring /* we assume if successful commit must called */ 13188f2dc78dSAlexander Aring msg->idx = idx; 13198f2dc78dSAlexander Aring return msg; 13208f2dc78dSAlexander Aring } 13218f2dc78dSAlexander Aring 13222874d1a6SAlexander Aring static void _dlm_lowcomms_commit_msg(struct dlm_msg *msg) 13236ed7257bSPatrick Caulfield { 13248f2dc78dSAlexander Aring struct writequeue_entry *e = msg->entry; 13256ed7257bSPatrick Caulfield struct connection *con = e->con; 13266ed7257bSPatrick Caulfield int users; 13276ed7257bSPatrick Caulfield 13286ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 13298f2dc78dSAlexander Aring kref_get(&msg->ref); 13308f2dc78dSAlexander Aring list_add(&msg->list, &e->msgs); 13318f2dc78dSAlexander Aring 13326ed7257bSPatrick Caulfield users = --e->users; 13336ed7257bSPatrick Caulfield if (users) 13346ed7257bSPatrick Caulfield goto out; 1335f0747ebfSAlexander Aring 1336f0747ebfSAlexander Aring e->len = DLM_WQ_LENGTH_BYTES(e); 13376ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 13386ed7257bSPatrick Caulfield 13396ed7257bSPatrick Caulfield queue_work(send_workqueue, &con->swork); 13406ed7257bSPatrick Caulfield return; 13416ed7257bSPatrick Caulfield 13426ed7257bSPatrick Caulfield out: 13436ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 13446ed7257bSPatrick Caulfield return; 13456ed7257bSPatrick Caulfield } 13466ed7257bSPatrick Caulfield 13472874d1a6SAlexander Aring void dlm_lowcomms_commit_msg(struct dlm_msg *msg) 13482874d1a6SAlexander Aring { 13492874d1a6SAlexander Aring _dlm_lowcomms_commit_msg(msg); 13502874d1a6SAlexander Aring srcu_read_unlock(&connections_srcu, msg->idx); 13512874d1a6SAlexander Aring } 13522874d1a6SAlexander Aring 13538f2dc78dSAlexander Aring void dlm_lowcomms_put_msg(struct dlm_msg *msg) 13548f2dc78dSAlexander Aring { 13558f2dc78dSAlexander Aring kref_put(&msg->ref, dlm_msg_release); 13568f2dc78dSAlexander Aring } 13578f2dc78dSAlexander Aring 13582874d1a6SAlexander Aring /* does not held connections_srcu, usage workqueue only */ 13592874d1a6SAlexander Aring int dlm_lowcomms_resend_msg(struct dlm_msg *msg) 13602874d1a6SAlexander Aring { 13612874d1a6SAlexander Aring struct dlm_msg *msg_resend; 13622874d1a6SAlexander Aring char *ppc; 13632874d1a6SAlexander Aring 13642874d1a6SAlexander Aring if (msg->retransmit) 13652874d1a6SAlexander Aring return 1; 13662874d1a6SAlexander Aring 13672874d1a6SAlexander Aring msg_resend = dlm_lowcomms_new_msg_con(msg->entry->con, msg->len, 13682874d1a6SAlexander Aring GFP_ATOMIC, &ppc, NULL, NULL); 13692874d1a6SAlexander Aring if (!msg_resend) 13702874d1a6SAlexander Aring return -ENOMEM; 13712874d1a6SAlexander Aring 13722874d1a6SAlexander Aring msg->retransmit = true; 13732874d1a6SAlexander Aring kref_get(&msg->ref); 13742874d1a6SAlexander Aring msg_resend->orig_msg = msg; 13752874d1a6SAlexander Aring 13762874d1a6SAlexander Aring memcpy(ppc, msg->ppc, msg->len); 13772874d1a6SAlexander Aring _dlm_lowcomms_commit_msg(msg_resend); 13782874d1a6SAlexander Aring dlm_lowcomms_put_msg(msg_resend); 13792874d1a6SAlexander Aring 13802874d1a6SAlexander Aring return 0; 13812874d1a6SAlexander Aring } 13822874d1a6SAlexander Aring 13836ed7257bSPatrick Caulfield /* Send a message */ 13846ed7257bSPatrick Caulfield static void send_to_sock(struct connection *con) 13856ed7257bSPatrick Caulfield { 13866ed7257bSPatrick Caulfield const int msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL; 13876ed7257bSPatrick Caulfield struct writequeue_entry *e; 138866d5955aSAlexander Aring int len, offset, ret; 1389f92c8dd7SBob Peterson int count = 0; 13906ed7257bSPatrick Caulfield 13916ed7257bSPatrick Caulfield mutex_lock(&con->sock_mutex); 13926ed7257bSPatrick Caulfield if (con->sock == NULL) 13936ed7257bSPatrick Caulfield goto out_connect; 13946ed7257bSPatrick Caulfield 13956ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 13966ed7257bSPatrick Caulfield for (;;) { 139766d5955aSAlexander Aring e = con_next_wq(con); 139866d5955aSAlexander Aring if (!e) 13996ed7257bSPatrick Caulfield break; 14006ed7257bSPatrick Caulfield 14016ed7257bSPatrick Caulfield len = e->len; 14026ed7257bSPatrick Caulfield offset = e->offset; 14036ed7257bSPatrick Caulfield BUG_ON(len == 0 && e->users == 0); 14046ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14056ed7257bSPatrick Caulfield 14061329e3f2SPaolo Bonzini ret = kernel_sendpage(con->sock, e->page, offset, len, 14076ed7257bSPatrick Caulfield msg_flags); 140892732376SAlexander Aring trace_dlm_send(con->nodeid, ret); 1409d66f8277SPatrick Caulfield if (ret == -EAGAIN || ret == 0) { 1410b36930ddSDavid Miller if (ret == -EAGAIN && 14119cd3e072SEric Dumazet test_bit(SOCKWQ_ASYNC_NOSPACE, &con->sock->flags) && 1412b36930ddSDavid Miller !test_and_set_bit(CF_APP_LIMITED, &con->flags)) { 1413b36930ddSDavid Miller /* Notify TCP that we're limited by the 1414b36930ddSDavid Miller * application window size. 1415b36930ddSDavid Miller */ 1416b36930ddSDavid Miller set_bit(SOCK_NOSPACE, &con->sock->flags); 1417b36930ddSDavid Miller con->sock->sk->sk_write_pending++; 1418b36930ddSDavid Miller } 1419d66f8277SPatrick Caulfield cond_resched(); 14206ed7257bSPatrick Caulfield goto out; 14219c5bef58SYing Xue } else if (ret < 0) 1422ba868d9dSAlexander Aring goto out; 1423f92c8dd7SBob Peterson 14246ed7257bSPatrick Caulfield /* Don't starve people filling buffers */ 1425f92c8dd7SBob Peterson if (++count >= MAX_SEND_MSG_COUNT) { 14266ed7257bSPatrick Caulfield cond_resched(); 1427f92c8dd7SBob Peterson count = 0; 1428f92c8dd7SBob Peterson } 14296ed7257bSPatrick Caulfield 14306ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 14315d689871SMike Christie writequeue_entry_complete(e, ret); 14326ed7257bSPatrick Caulfield } 14336ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14348aa31cbfSAlexander Aring 14358aa31cbfSAlexander Aring /* close if we got EOF */ 14368aa31cbfSAlexander Aring if (test_and_clear_bit(CF_EOF, &con->flags)) { 14378aa31cbfSAlexander Aring mutex_unlock(&con->sock_mutex); 14388aa31cbfSAlexander Aring close_connection(con, false, false, true); 14398aa31cbfSAlexander Aring 14408aa31cbfSAlexander Aring /* handling for tcp shutdown */ 14418aa31cbfSAlexander Aring clear_bit(CF_SHUTDOWN, &con->flags); 14428aa31cbfSAlexander Aring wake_up(&con->shutdown_wait); 14438aa31cbfSAlexander Aring } else { 14448aa31cbfSAlexander Aring mutex_unlock(&con->sock_mutex); 14458aa31cbfSAlexander Aring } 14468aa31cbfSAlexander Aring 14478aa31cbfSAlexander Aring return; 14488aa31cbfSAlexander Aring 14496ed7257bSPatrick Caulfield out: 14506ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 14516ed7257bSPatrick Caulfield return; 14526ed7257bSPatrick Caulfield 14536ed7257bSPatrick Caulfield out_connect: 14546ed7257bSPatrick Caulfield mutex_unlock(&con->sock_mutex); 145501da24d3SBob Peterson queue_work(send_workqueue, &con->swork); 145601da24d3SBob Peterson cond_resched(); 14576ed7257bSPatrick Caulfield } 14586ed7257bSPatrick Caulfield 14596ed7257bSPatrick Caulfield static void clean_one_writequeue(struct connection *con) 14606ed7257bSPatrick Caulfield { 14615e9ccc37SChristine Caulfield struct writequeue_entry *e, *safe; 14626ed7257bSPatrick Caulfield 14636ed7257bSPatrick Caulfield spin_lock(&con->writequeue_lock); 14645e9ccc37SChristine Caulfield list_for_each_entry_safe(e, safe, &con->writequeue, list) { 14656ed7257bSPatrick Caulfield free_entry(e); 14666ed7257bSPatrick Caulfield } 14676ed7257bSPatrick Caulfield spin_unlock(&con->writequeue_lock); 14686ed7257bSPatrick Caulfield } 14696ed7257bSPatrick Caulfield 14706ed7257bSPatrick Caulfield /* Called from recovery when it knows that a node has 14716ed7257bSPatrick Caulfield left the cluster */ 14726ed7257bSPatrick Caulfield int dlm_lowcomms_close(int nodeid) 14736ed7257bSPatrick Caulfield { 14746ed7257bSPatrick Caulfield struct connection *con; 147536b71a8bSDavid Teigland struct dlm_node_addr *na; 1476b38bc9c2SAlexander Aring int idx; 14776ed7257bSPatrick Caulfield 14786ed7257bSPatrick Caulfield log_print("closing connection to node %d", nodeid); 1479b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 14806ed7257bSPatrick Caulfield con = nodeid2con(nodeid, 0); 14816ed7257bSPatrick Caulfield if (con) { 1482063c4c99SLars Marowsky-Bree set_bit(CF_CLOSE, &con->flags); 14830d737a8cSMarcelo Ricardo Leitner close_connection(con, true, true, true); 14846ed7257bSPatrick Caulfield clean_one_writequeue(con); 148553a5edaaSAlexander Aring if (con->othercon) 148653a5edaaSAlexander Aring clean_one_writequeue(con->othercon); 14876ed7257bSPatrick Caulfield } 1488b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 148936b71a8bSDavid Teigland 149036b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 149136b71a8bSDavid Teigland na = find_node_addr(nodeid); 149236b71a8bSDavid Teigland if (na) { 149336b71a8bSDavid Teigland list_del(&na->list); 149436b71a8bSDavid Teigland while (na->addr_count--) 149536b71a8bSDavid Teigland kfree(na->addr[na->addr_count]); 149636b71a8bSDavid Teigland kfree(na); 149736b71a8bSDavid Teigland } 149836b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 149936b71a8bSDavid Teigland 15006ed7257bSPatrick Caulfield return 0; 15016ed7257bSPatrick Caulfield } 15026ed7257bSPatrick Caulfield 15036ed7257bSPatrick Caulfield /* Receive workqueue function */ 15046ed7257bSPatrick Caulfield static void process_recv_sockets(struct work_struct *work) 15056ed7257bSPatrick Caulfield { 15066ed7257bSPatrick Caulfield struct connection *con = container_of(work, struct connection, rwork); 15076ed7257bSPatrick Caulfield 15086ed7257bSPatrick Caulfield clear_bit(CF_READ_PENDING, &con->flags); 150962699b3fSAlexander Aring receive_from_sock(con); 15106ed7257bSPatrick Caulfield } 15116ed7257bSPatrick Caulfield 1512d11ccd45SAlexander Aring static void process_listen_recv_socket(struct work_struct *work) 1513d11ccd45SAlexander Aring { 1514d11ccd45SAlexander Aring accept_from_sock(&listen_con); 1515d11ccd45SAlexander Aring } 1516d11ccd45SAlexander Aring 15178728a455SAlexander Aring static void dlm_connect(struct connection *con) 15188728a455SAlexander Aring { 15198728a455SAlexander Aring struct sockaddr_storage addr; 15208728a455SAlexander Aring int result, addr_len; 15218728a455SAlexander Aring struct socket *sock; 15228728a455SAlexander Aring unsigned int mark; 15238728a455SAlexander Aring 15248728a455SAlexander Aring /* Some odd races can cause double-connects, ignore them */ 15258728a455SAlexander Aring if (con->retries++ > MAX_CONNECT_RETRIES) 15268728a455SAlexander Aring return; 15278728a455SAlexander Aring 15288728a455SAlexander Aring if (con->sock) { 15298728a455SAlexander Aring log_print("node %d already connected.", con->nodeid); 15308728a455SAlexander Aring return; 15318728a455SAlexander Aring } 15328728a455SAlexander Aring 15338728a455SAlexander Aring memset(&addr, 0, sizeof(addr)); 15348728a455SAlexander Aring result = nodeid_to_addr(con->nodeid, &addr, NULL, 15358728a455SAlexander Aring dlm_proto_ops->try_new_addr, &mark); 15368728a455SAlexander Aring if (result < 0) { 15378728a455SAlexander Aring log_print("no address for nodeid %d", con->nodeid); 15388728a455SAlexander Aring return; 15398728a455SAlexander Aring } 15408728a455SAlexander Aring 15418728a455SAlexander Aring /* Create a socket to communicate with */ 15428728a455SAlexander Aring result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family, 15438728a455SAlexander Aring SOCK_STREAM, dlm_proto_ops->proto, &sock); 15448728a455SAlexander Aring if (result < 0) 15458728a455SAlexander Aring goto socket_err; 15468728a455SAlexander Aring 15478728a455SAlexander Aring sock_set_mark(sock->sk, mark); 15488728a455SAlexander Aring dlm_proto_ops->sockopts(sock); 15498728a455SAlexander Aring 15508728a455SAlexander Aring add_sock(sock, con); 15518728a455SAlexander Aring 15528728a455SAlexander Aring result = dlm_proto_ops->bind(sock); 15538728a455SAlexander Aring if (result < 0) 15548728a455SAlexander Aring goto add_sock_err; 15558728a455SAlexander Aring 15568728a455SAlexander Aring log_print_ratelimited("connecting to %d", con->nodeid); 15578728a455SAlexander Aring make_sockaddr(&addr, dlm_config.ci_tcp_port, &addr_len); 15588728a455SAlexander Aring result = dlm_proto_ops->connect(con, sock, (struct sockaddr *)&addr, 15598728a455SAlexander Aring addr_len); 15608728a455SAlexander Aring if (result < 0) 15618728a455SAlexander Aring goto add_sock_err; 15628728a455SAlexander Aring 15638728a455SAlexander Aring return; 15648728a455SAlexander Aring 15658728a455SAlexander Aring add_sock_err: 15668728a455SAlexander Aring dlm_close_sock(&con->sock); 15678728a455SAlexander Aring 15688728a455SAlexander Aring socket_err: 15698728a455SAlexander Aring /* 15708728a455SAlexander Aring * Some errors are fatal and this list might need adjusting. For other 15718728a455SAlexander Aring * errors we try again until the max number of retries is reached. 15728728a455SAlexander Aring */ 15738728a455SAlexander Aring if (result != -EHOSTUNREACH && 15748728a455SAlexander Aring result != -ENETUNREACH && 15758728a455SAlexander Aring result != -ENETDOWN && 15768728a455SAlexander Aring result != -EINVAL && 15778728a455SAlexander Aring result != -EPROTONOSUPPORT) { 15788728a455SAlexander Aring log_print("connect %d try %d error %d", con->nodeid, 15798728a455SAlexander Aring con->retries, result); 15808728a455SAlexander Aring msleep(1000); 15818728a455SAlexander Aring lowcomms_connect_sock(con); 15828728a455SAlexander Aring } 15838728a455SAlexander Aring } 15848728a455SAlexander Aring 15856ed7257bSPatrick Caulfield /* Send workqueue function */ 15866ed7257bSPatrick Caulfield static void process_send_sockets(struct work_struct *work) 15876ed7257bSPatrick Caulfield { 15886ed7257bSPatrick Caulfield struct connection *con = container_of(work, struct connection, swork); 15896ed7257bSPatrick Caulfield 15907443bc96SAlexander Aring WARN_ON(test_bit(CF_IS_OTHERCON, &con->flags)); 15917443bc96SAlexander Aring 15928a4abb08Stsutomu.owa@toshiba.co.jp clear_bit(CF_WRITE_PENDING, &con->flags); 1593ba868d9dSAlexander Aring 1594489d8e55SAlexander Aring if (test_and_clear_bit(CF_RECONNECT, &con->flags)) { 1595ba868d9dSAlexander Aring close_connection(con, false, false, true); 1596489d8e55SAlexander Aring dlm_midcomms_unack_msg_resend(con->nodeid); 1597489d8e55SAlexander Aring } 1598ba868d9dSAlexander Aring 15998728a455SAlexander Aring if (con->sock == NULL) { 1600ba868d9dSAlexander Aring if (test_and_clear_bit(CF_DELAY_CONNECT, &con->flags)) 1601ba868d9dSAlexander Aring msleep(1000); 16028728a455SAlexander Aring 16038728a455SAlexander Aring mutex_lock(&con->sock_mutex); 16048728a455SAlexander Aring dlm_connect(con); 16058728a455SAlexander Aring mutex_unlock(&con->sock_mutex); 1606ba868d9dSAlexander Aring } 16078728a455SAlexander Aring 160801da24d3SBob Peterson if (!list_empty(&con->writequeue)) 16096ed7257bSPatrick Caulfield send_to_sock(con); 16106ed7257bSPatrick Caulfield } 16116ed7257bSPatrick Caulfield 16126ed7257bSPatrick Caulfield static void work_stop(void) 16136ed7257bSPatrick Caulfield { 1614fcef0e6cSAlexander Aring if (recv_workqueue) { 16156ed7257bSPatrick Caulfield destroy_workqueue(recv_workqueue); 1616fcef0e6cSAlexander Aring recv_workqueue = NULL; 1617fcef0e6cSAlexander Aring } 1618fcef0e6cSAlexander Aring 1619fcef0e6cSAlexander Aring if (send_workqueue) { 16206ed7257bSPatrick Caulfield destroy_workqueue(send_workqueue); 1621fcef0e6cSAlexander Aring send_workqueue = NULL; 1622fcef0e6cSAlexander Aring } 16236ed7257bSPatrick Caulfield } 16246ed7257bSPatrick Caulfield 16256ed7257bSPatrick Caulfield static int work_start(void) 16266ed7257bSPatrick Caulfield { 16276c6a1cc6SAlexander Aring recv_workqueue = alloc_ordered_workqueue("dlm_recv", WQ_MEM_RECLAIM); 1628b9d41052SNamhyung Kim if (!recv_workqueue) { 1629b9d41052SNamhyung Kim log_print("can't start dlm_recv"); 1630b9d41052SNamhyung Kim return -ENOMEM; 16316ed7257bSPatrick Caulfield } 16326ed7257bSPatrick Caulfield 16336c6a1cc6SAlexander Aring send_workqueue = alloc_ordered_workqueue("dlm_send", WQ_MEM_RECLAIM); 1634b9d41052SNamhyung Kim if (!send_workqueue) { 1635b9d41052SNamhyung Kim log_print("can't start dlm_send"); 16366ed7257bSPatrick Caulfield destroy_workqueue(recv_workqueue); 1637fcef0e6cSAlexander Aring recv_workqueue = NULL; 1638b9d41052SNamhyung Kim return -ENOMEM; 16396ed7257bSPatrick Caulfield } 16406ed7257bSPatrick Caulfield 16416ed7257bSPatrick Caulfield return 0; 16426ed7257bSPatrick Caulfield } 16436ed7257bSPatrick Caulfield 16449d232469SAlexander Aring static void shutdown_conn(struct connection *con) 16459d232469SAlexander Aring { 1646a66c008cSAlexander Aring if (dlm_proto_ops->shutdown_action) 1647a66c008cSAlexander Aring dlm_proto_ops->shutdown_action(con); 16489d232469SAlexander Aring } 16499d232469SAlexander Aring 16509d232469SAlexander Aring void dlm_lowcomms_shutdown(void) 16519d232469SAlexander Aring { 1652b38bc9c2SAlexander Aring int idx; 1653b38bc9c2SAlexander Aring 16549d232469SAlexander Aring /* Set all the flags to prevent any 16559d232469SAlexander Aring * socket activity. 16569d232469SAlexander Aring */ 16579d232469SAlexander Aring dlm_allow_conn = 0; 16589d232469SAlexander Aring 16599d232469SAlexander Aring if (recv_workqueue) 16609d232469SAlexander Aring flush_workqueue(recv_workqueue); 16619d232469SAlexander Aring if (send_workqueue) 16629d232469SAlexander Aring flush_workqueue(send_workqueue); 16639d232469SAlexander Aring 16649d232469SAlexander Aring dlm_close_sock(&listen_con.sock); 16659d232469SAlexander Aring 1666b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 16679d232469SAlexander Aring foreach_conn(shutdown_conn); 1668b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 16699d232469SAlexander Aring } 16709d232469SAlexander Aring 1671f0fb83cbStsutomu.owa@toshiba.co.jp static void _stop_conn(struct connection *con, bool and_other) 16726ed7257bSPatrick Caulfield { 1673f0fb83cbStsutomu.owa@toshiba.co.jp mutex_lock(&con->sock_mutex); 1674173a31feStsutomu.owa@toshiba.co.jp set_bit(CF_CLOSE, &con->flags); 1675f0fb83cbStsutomu.owa@toshiba.co.jp set_bit(CF_READ_PENDING, &con->flags); 16768a4abb08Stsutomu.owa@toshiba.co.jp set_bit(CF_WRITE_PENDING, &con->flags); 167793eaadebStsutomu.owa@toshiba.co.jp if (con->sock && con->sock->sk) { 167892c44605SAlexander Aring lock_sock(con->sock->sk); 1679afb853fbSPatrick Caulfield con->sock->sk->sk_user_data = NULL; 168092c44605SAlexander Aring release_sock(con->sock->sk); 168193eaadebStsutomu.owa@toshiba.co.jp } 1682f0fb83cbStsutomu.owa@toshiba.co.jp if (con->othercon && and_other) 1683f0fb83cbStsutomu.owa@toshiba.co.jp _stop_conn(con->othercon, false); 1684f0fb83cbStsutomu.owa@toshiba.co.jp mutex_unlock(&con->sock_mutex); 1685f0fb83cbStsutomu.owa@toshiba.co.jp } 1686f0fb83cbStsutomu.owa@toshiba.co.jp 1687f0fb83cbStsutomu.owa@toshiba.co.jp static void stop_conn(struct connection *con) 1688f0fb83cbStsutomu.owa@toshiba.co.jp { 1689f0fb83cbStsutomu.owa@toshiba.co.jp _stop_conn(con, true); 1690afb853fbSPatrick Caulfield } 16915e9ccc37SChristine Caulfield 16924798cbbfSAlexander Aring static void connection_release(struct rcu_head *rcu) 16934798cbbfSAlexander Aring { 16944798cbbfSAlexander Aring struct connection *con = container_of(rcu, struct connection, rcu); 16954798cbbfSAlexander Aring 16964798cbbfSAlexander Aring kfree(con->rx_buf); 16974798cbbfSAlexander Aring kfree(con); 16984798cbbfSAlexander Aring } 16994798cbbfSAlexander Aring 17005e9ccc37SChristine Caulfield static void free_conn(struct connection *con) 17015e9ccc37SChristine Caulfield { 17020d737a8cSMarcelo Ricardo Leitner close_connection(con, true, true, true); 1703a47666ebSAlexander Aring spin_lock(&connections_lock); 1704a47666ebSAlexander Aring hlist_del_rcu(&con->list); 1705a47666ebSAlexander Aring spin_unlock(&connections_lock); 1706948c47e9SAlexander Aring if (con->othercon) { 1707948c47e9SAlexander Aring clean_one_writequeue(con->othercon); 17085cbec208SAlexander Aring call_srcu(&connections_srcu, &con->othercon->rcu, 17095cbec208SAlexander Aring connection_release); 1710948c47e9SAlexander Aring } 17110de98432SAlexander Aring clean_one_writequeue(con); 17125cbec208SAlexander Aring call_srcu(&connections_srcu, &con->rcu, connection_release); 17136ed7257bSPatrick Caulfield } 17145e9ccc37SChristine Caulfield 1715f0fb83cbStsutomu.owa@toshiba.co.jp static void work_flush(void) 1716f0fb83cbStsutomu.owa@toshiba.co.jp { 1717b38bc9c2SAlexander Aring int ok; 1718f0fb83cbStsutomu.owa@toshiba.co.jp int i; 1719f0fb83cbStsutomu.owa@toshiba.co.jp struct connection *con; 1720f0fb83cbStsutomu.owa@toshiba.co.jp 1721f0fb83cbStsutomu.owa@toshiba.co.jp do { 1722f0fb83cbStsutomu.owa@toshiba.co.jp ok = 1; 1723f0fb83cbStsutomu.owa@toshiba.co.jp foreach_conn(stop_conn); 1724b355516fSDavid Windsor if (recv_workqueue) 1725f0fb83cbStsutomu.owa@toshiba.co.jp flush_workqueue(recv_workqueue); 1726b355516fSDavid Windsor if (send_workqueue) 1727f0fb83cbStsutomu.owa@toshiba.co.jp flush_workqueue(send_workqueue); 1728f0fb83cbStsutomu.owa@toshiba.co.jp for (i = 0; i < CONN_HASH_SIZE && ok; i++) { 1729a47666ebSAlexander Aring hlist_for_each_entry_rcu(con, &connection_hash[i], 1730a47666ebSAlexander Aring list) { 1731f0fb83cbStsutomu.owa@toshiba.co.jp ok &= test_bit(CF_READ_PENDING, &con->flags); 17328a4abb08Stsutomu.owa@toshiba.co.jp ok &= test_bit(CF_WRITE_PENDING, &con->flags); 17338a4abb08Stsutomu.owa@toshiba.co.jp if (con->othercon) { 1734f0fb83cbStsutomu.owa@toshiba.co.jp ok &= test_bit(CF_READ_PENDING, 1735f0fb83cbStsutomu.owa@toshiba.co.jp &con->othercon->flags); 17368a4abb08Stsutomu.owa@toshiba.co.jp ok &= test_bit(CF_WRITE_PENDING, 17378a4abb08Stsutomu.owa@toshiba.co.jp &con->othercon->flags); 17388a4abb08Stsutomu.owa@toshiba.co.jp } 1739f0fb83cbStsutomu.owa@toshiba.co.jp } 1740f0fb83cbStsutomu.owa@toshiba.co.jp } 1741f0fb83cbStsutomu.owa@toshiba.co.jp } while (!ok); 1742f0fb83cbStsutomu.owa@toshiba.co.jp } 1743f0fb83cbStsutomu.owa@toshiba.co.jp 17445e9ccc37SChristine Caulfield void dlm_lowcomms_stop(void) 17455e9ccc37SChristine Caulfield { 1746b38bc9c2SAlexander Aring int idx; 1747b38bc9c2SAlexander Aring 1748b38bc9c2SAlexander Aring idx = srcu_read_lock(&connections_srcu); 1749f0fb83cbStsutomu.owa@toshiba.co.jp work_flush(); 17503a8db798SMarcelo Ricardo Leitner foreach_conn(free_conn); 1751b38bc9c2SAlexander Aring srcu_read_unlock(&connections_srcu, idx); 17526ed7257bSPatrick Caulfield work_stop(); 1753043697f0SAlexander Aring deinit_local(); 1754a66c008cSAlexander Aring 1755a66c008cSAlexander Aring dlm_proto_ops = NULL; 17566ed7257bSPatrick Caulfield } 17576ed7257bSPatrick Caulfield 17582dc6b115SAlexander Aring static int dlm_listen_for_all(void) 17592dc6b115SAlexander Aring { 17602dc6b115SAlexander Aring struct socket *sock; 17612dc6b115SAlexander Aring int result; 17622dc6b115SAlexander Aring 17632dc6b115SAlexander Aring log_print("Using %s for communications", 17642dc6b115SAlexander Aring dlm_proto_ops->name); 17652dc6b115SAlexander Aring 17662dc6b115SAlexander Aring result = dlm_proto_ops->listen_validate(); 17672dc6b115SAlexander Aring if (result < 0) 17682dc6b115SAlexander Aring return result; 17692dc6b115SAlexander Aring 17702dc6b115SAlexander Aring result = sock_create_kern(&init_net, dlm_local_addr[0]->ss_family, 17712dc6b115SAlexander Aring SOCK_STREAM, dlm_proto_ops->proto, &sock); 17722dc6b115SAlexander Aring if (result < 0) { 1773fe933675SAlexander Aring log_print("Can't create comms socket: %d", result); 17742dc6b115SAlexander Aring goto out; 17752dc6b115SAlexander Aring } 17762dc6b115SAlexander Aring 17772dc6b115SAlexander Aring sock_set_mark(sock->sk, dlm_config.ci_mark); 17782dc6b115SAlexander Aring dlm_proto_ops->listen_sockopts(sock); 17792dc6b115SAlexander Aring 17802dc6b115SAlexander Aring result = dlm_proto_ops->listen_bind(sock); 17812dc6b115SAlexander Aring if (result < 0) 17822dc6b115SAlexander Aring goto out; 17832dc6b115SAlexander Aring 17842dc6b115SAlexander Aring save_listen_callbacks(sock); 17852dc6b115SAlexander Aring add_listen_sock(sock, &listen_con); 17862dc6b115SAlexander Aring 17872dc6b115SAlexander Aring INIT_WORK(&listen_con.rwork, process_listen_recv_socket); 17882dc6b115SAlexander Aring result = sock->ops->listen(sock, 5); 17892dc6b115SAlexander Aring if (result < 0) { 17902dc6b115SAlexander Aring dlm_close_sock(&listen_con.sock); 17912dc6b115SAlexander Aring goto out; 17922dc6b115SAlexander Aring } 17932dc6b115SAlexander Aring 17942dc6b115SAlexander Aring return 0; 17952dc6b115SAlexander Aring 17962dc6b115SAlexander Aring out: 17972dc6b115SAlexander Aring sock_release(sock); 17982dc6b115SAlexander Aring return result; 17992dc6b115SAlexander Aring } 18002dc6b115SAlexander Aring 18018728a455SAlexander Aring static int dlm_tcp_bind(struct socket *sock) 18028728a455SAlexander Aring { 18038728a455SAlexander Aring struct sockaddr_storage src_addr; 18048728a455SAlexander Aring int result, addr_len; 18058728a455SAlexander Aring 18068728a455SAlexander Aring /* Bind to our cluster-known address connecting to avoid 18078728a455SAlexander Aring * routing problems. 18088728a455SAlexander Aring */ 18098728a455SAlexander Aring memcpy(&src_addr, dlm_local_addr[0], sizeof(src_addr)); 18108728a455SAlexander Aring make_sockaddr(&src_addr, 0, &addr_len); 18118728a455SAlexander Aring 18128728a455SAlexander Aring result = sock->ops->bind(sock, (struct sockaddr *)&src_addr, 18138728a455SAlexander Aring addr_len); 18148728a455SAlexander Aring if (result < 0) { 18158728a455SAlexander Aring /* This *may* not indicate a critical error */ 18168728a455SAlexander Aring log_print("could not bind for connect: %d", result); 18178728a455SAlexander Aring } 18188728a455SAlexander Aring 18198728a455SAlexander Aring return 0; 18208728a455SAlexander Aring } 18218728a455SAlexander Aring 18228728a455SAlexander Aring static int dlm_tcp_connect(struct connection *con, struct socket *sock, 18238728a455SAlexander Aring struct sockaddr *addr, int addr_len) 18248728a455SAlexander Aring { 18258728a455SAlexander Aring int ret; 18268728a455SAlexander Aring 18278728a455SAlexander Aring ret = sock->ops->connect(sock, addr, addr_len, O_NONBLOCK); 18288728a455SAlexander Aring switch (ret) { 18298728a455SAlexander Aring case -EINPROGRESS: 18308728a455SAlexander Aring fallthrough; 18318728a455SAlexander Aring case 0: 18328728a455SAlexander Aring return 0; 18338728a455SAlexander Aring } 18348728a455SAlexander Aring 18358728a455SAlexander Aring return ret; 18368728a455SAlexander Aring } 18378728a455SAlexander Aring 18382dc6b115SAlexander Aring static int dlm_tcp_listen_validate(void) 18392dc6b115SAlexander Aring { 18402dc6b115SAlexander Aring /* We don't support multi-homed hosts */ 18412dc6b115SAlexander Aring if (dlm_local_count > 1) { 18422dc6b115SAlexander Aring log_print("TCP protocol can't handle multi-homed hosts, try SCTP"); 18432dc6b115SAlexander Aring return -EINVAL; 18442dc6b115SAlexander Aring } 18452dc6b115SAlexander Aring 18462dc6b115SAlexander Aring return 0; 18472dc6b115SAlexander Aring } 18482dc6b115SAlexander Aring 18492dc6b115SAlexander Aring static void dlm_tcp_sockopts(struct socket *sock) 18502dc6b115SAlexander Aring { 18512dc6b115SAlexander Aring /* Turn off Nagle's algorithm */ 18522dc6b115SAlexander Aring tcp_sock_set_nodelay(sock->sk); 18532dc6b115SAlexander Aring } 18542dc6b115SAlexander Aring 18552dc6b115SAlexander Aring static void dlm_tcp_listen_sockopts(struct socket *sock) 18562dc6b115SAlexander Aring { 18572dc6b115SAlexander Aring dlm_tcp_sockopts(sock); 18582dc6b115SAlexander Aring sock_set_reuseaddr(sock->sk); 18592dc6b115SAlexander Aring } 18602dc6b115SAlexander Aring 18612dc6b115SAlexander Aring static int dlm_tcp_listen_bind(struct socket *sock) 18622dc6b115SAlexander Aring { 18632dc6b115SAlexander Aring int addr_len; 18642dc6b115SAlexander Aring 18652dc6b115SAlexander Aring /* Bind to our port */ 18662dc6b115SAlexander Aring make_sockaddr(dlm_local_addr[0], dlm_config.ci_tcp_port, &addr_len); 18672dc6b115SAlexander Aring return sock->ops->bind(sock, (struct sockaddr *)dlm_local_addr[0], 18682dc6b115SAlexander Aring addr_len); 18692dc6b115SAlexander Aring } 18702dc6b115SAlexander Aring 1871a66c008cSAlexander Aring static const struct dlm_proto_ops dlm_tcp_ops = { 18722dc6b115SAlexander Aring .name = "TCP", 18732dc6b115SAlexander Aring .proto = IPPROTO_TCP, 18748728a455SAlexander Aring .connect = dlm_tcp_connect, 18758728a455SAlexander Aring .sockopts = dlm_tcp_sockopts, 18768728a455SAlexander Aring .bind = dlm_tcp_bind, 18772dc6b115SAlexander Aring .listen_validate = dlm_tcp_listen_validate, 18782dc6b115SAlexander Aring .listen_sockopts = dlm_tcp_listen_sockopts, 18792dc6b115SAlexander Aring .listen_bind = dlm_tcp_listen_bind, 1880a66c008cSAlexander Aring .shutdown_action = dlm_tcp_shutdown, 1881a66c008cSAlexander Aring .eof_condition = tcp_eof_condition, 1882a66c008cSAlexander Aring }; 1883a66c008cSAlexander Aring 18848728a455SAlexander Aring static int dlm_sctp_bind(struct socket *sock) 18858728a455SAlexander Aring { 18868728a455SAlexander Aring return sctp_bind_addrs(sock, 0); 18878728a455SAlexander Aring } 18888728a455SAlexander Aring 18898728a455SAlexander Aring static int dlm_sctp_connect(struct connection *con, struct socket *sock, 18908728a455SAlexander Aring struct sockaddr *addr, int addr_len) 18918728a455SAlexander Aring { 18928728a455SAlexander Aring int ret; 18938728a455SAlexander Aring 18948728a455SAlexander Aring /* 18958728a455SAlexander Aring * Make sock->ops->connect() function return in specified time, 18968728a455SAlexander Aring * since O_NONBLOCK argument in connect() function does not work here, 18978728a455SAlexander Aring * then, we should restore the default value of this attribute. 18988728a455SAlexander Aring */ 18998728a455SAlexander Aring sock_set_sndtimeo(sock->sk, 5); 19008728a455SAlexander Aring ret = sock->ops->connect(sock, addr, addr_len, 0); 19018728a455SAlexander Aring sock_set_sndtimeo(sock->sk, 0); 19028728a455SAlexander Aring if (ret < 0) 19038728a455SAlexander Aring return ret; 19048728a455SAlexander Aring 19058728a455SAlexander Aring if (!test_and_set_bit(CF_CONNECTED, &con->flags)) 19068728a455SAlexander Aring log_print("successful connected to node %d", con->nodeid); 19078728a455SAlexander Aring 19088728a455SAlexander Aring return 0; 19098728a455SAlexander Aring } 19108728a455SAlexander Aring 191190d21fc0SAlexander Aring static int dlm_sctp_listen_validate(void) 191290d21fc0SAlexander Aring { 191390d21fc0SAlexander Aring if (!IS_ENABLED(CONFIG_IP_SCTP)) { 191490d21fc0SAlexander Aring log_print("SCTP is not enabled by this kernel"); 191590d21fc0SAlexander Aring return -EOPNOTSUPP; 191690d21fc0SAlexander Aring } 191790d21fc0SAlexander Aring 191890d21fc0SAlexander Aring request_module("sctp"); 191990d21fc0SAlexander Aring return 0; 192090d21fc0SAlexander Aring } 192190d21fc0SAlexander Aring 19222dc6b115SAlexander Aring static int dlm_sctp_bind_listen(struct socket *sock) 19232dc6b115SAlexander Aring { 19242dc6b115SAlexander Aring return sctp_bind_addrs(sock, dlm_config.ci_tcp_port); 19252dc6b115SAlexander Aring } 19262dc6b115SAlexander Aring 19272dc6b115SAlexander Aring static void dlm_sctp_sockopts(struct socket *sock) 19282dc6b115SAlexander Aring { 19292dc6b115SAlexander Aring /* Turn off Nagle's algorithm */ 19302dc6b115SAlexander Aring sctp_sock_set_nodelay(sock->sk); 19312dc6b115SAlexander Aring sock_set_rcvbuf(sock->sk, NEEDED_RMEM); 19322dc6b115SAlexander Aring } 19332dc6b115SAlexander Aring 1934a66c008cSAlexander Aring static const struct dlm_proto_ops dlm_sctp_ops = { 19352dc6b115SAlexander Aring .name = "SCTP", 19362dc6b115SAlexander Aring .proto = IPPROTO_SCTP, 19378728a455SAlexander Aring .try_new_addr = true, 19388728a455SAlexander Aring .connect = dlm_sctp_connect, 19398728a455SAlexander Aring .sockopts = dlm_sctp_sockopts, 19408728a455SAlexander Aring .bind = dlm_sctp_bind, 194190d21fc0SAlexander Aring .listen_validate = dlm_sctp_listen_validate, 19422dc6b115SAlexander Aring .listen_sockopts = dlm_sctp_sockopts, 19432dc6b115SAlexander Aring .listen_bind = dlm_sctp_bind_listen, 1944a66c008cSAlexander Aring }; 1945a66c008cSAlexander Aring 19466ed7257bSPatrick Caulfield int dlm_lowcomms_start(void) 19476ed7257bSPatrick Caulfield { 19486ed7257bSPatrick Caulfield int error = -EINVAL; 19495e9ccc37SChristine Caulfield int i; 19505e9ccc37SChristine Caulfield 19515e9ccc37SChristine Caulfield for (i = 0; i < CONN_HASH_SIZE; i++) 19525e9ccc37SChristine Caulfield INIT_HLIST_HEAD(&connection_hash[i]); 19536ed7257bSPatrick Caulfield 19546ed7257bSPatrick Caulfield init_local(); 19556ed7257bSPatrick Caulfield if (!dlm_local_count) { 1956617e82e1SDavid Teigland error = -ENOTCONN; 19576ed7257bSPatrick Caulfield log_print("no local IP address has been set"); 1958513ef596SDavid Teigland goto fail; 19596ed7257bSPatrick Caulfield } 19606ed7257bSPatrick Caulfield 1961d11ccd45SAlexander Aring INIT_WORK(&listen_con.rwork, process_listen_recv_socket); 1962d11ccd45SAlexander Aring 1963513ef596SDavid Teigland error = work_start(); 1964513ef596SDavid Teigland if (error) 1965fcef0e6cSAlexander Aring goto fail_local; 1966513ef596SDavid Teigland 1967513ef596SDavid Teigland dlm_allow_conn = 1; 19686ed7257bSPatrick Caulfield 19696ed7257bSPatrick Caulfield /* Start listening */ 1970ac7d5d03SAlexander Aring switch (dlm_config.ci_protocol) { 1971ac7d5d03SAlexander Aring case DLM_PROTO_TCP: 1972a66c008cSAlexander Aring dlm_proto_ops = &dlm_tcp_ops; 1973ac7d5d03SAlexander Aring break; 1974ac7d5d03SAlexander Aring case DLM_PROTO_SCTP: 1975a66c008cSAlexander Aring dlm_proto_ops = &dlm_sctp_ops; 1976ac7d5d03SAlexander Aring break; 1977ac7d5d03SAlexander Aring default: 1978ac7d5d03SAlexander Aring log_print("Invalid protocol identifier %d set", 1979ac7d5d03SAlexander Aring dlm_config.ci_protocol); 1980ac7d5d03SAlexander Aring error = -EINVAL; 19812dc6b115SAlexander Aring goto fail_proto_ops; 1982ac7d5d03SAlexander Aring } 19832dc6b115SAlexander Aring 19842dc6b115SAlexander Aring error = dlm_listen_for_all(); 19856ed7257bSPatrick Caulfield if (error) 19862dc6b115SAlexander Aring goto fail_listen; 19876ed7257bSPatrick Caulfield 19886ed7257bSPatrick Caulfield return 0; 19896ed7257bSPatrick Caulfield 19902dc6b115SAlexander Aring fail_listen: 19912dc6b115SAlexander Aring dlm_proto_ops = NULL; 19922dc6b115SAlexander Aring fail_proto_ops: 1993513ef596SDavid Teigland dlm_allow_conn = 0; 1994d11ccd45SAlexander Aring dlm_close_sock(&listen_con.sock); 1995fcef0e6cSAlexander Aring work_stop(); 1996fcef0e6cSAlexander Aring fail_local: 1997fcef0e6cSAlexander Aring deinit_local(); 1998513ef596SDavid Teigland fail: 19996ed7257bSPatrick Caulfield return error; 20006ed7257bSPatrick Caulfield } 200136b71a8bSDavid Teigland 200236b71a8bSDavid Teigland void dlm_lowcomms_exit(void) 200336b71a8bSDavid Teigland { 200436b71a8bSDavid Teigland struct dlm_node_addr *na, *safe; 200536b71a8bSDavid Teigland 200636b71a8bSDavid Teigland spin_lock(&dlm_node_addrs_spin); 200736b71a8bSDavid Teigland list_for_each_entry_safe(na, safe, &dlm_node_addrs, list) { 200836b71a8bSDavid Teigland list_del(&na->list); 200936b71a8bSDavid Teigland while (na->addr_count--) 201036b71a8bSDavid Teigland kfree(na->addr[na->addr_count]); 201136b71a8bSDavid Teigland kfree(na); 201236b71a8bSDavid Teigland } 201336b71a8bSDavid Teigland spin_unlock(&dlm_node_addrs_spin); 201436b71a8bSDavid Teigland } 2015