A generic message passing library is used to communicate with all other components in the Ceph file system. The messenger library provides ordered, reliable delivery of messages between two nodes in the system, or notifies the higher layer when it is unable to do so. This implementation is based on TCP. Signed-off-by: Sage Weil <sage@xxxxxxxxxxxx> --- fs/ceph/decode.h | 151 ++++ fs/ceph/messenger.c | 2389 +++++++++++++++++++++++++++++++++++++++++++++++++++ fs/ceph/messenger.h | 267 ++++++ 3 files changed, 2807 insertions(+), 0 deletions(-) create mode 100644 fs/ceph/decode.h create mode 100644 fs/ceph/messenger.c create mode 100644 fs/ceph/messenger.h diff --git a/fs/ceph/decode.h b/fs/ceph/decode.h new file mode 100644 index 0000000..c34de0a --- /dev/null +++ b/fs/ceph/decode.h @@ -0,0 +1,151 @@ +#ifndef __CEPH_DECODE_H +#define __CEPH_DECODE_H + +/* + * in all cases, + * void **p pointer to position pointer + * void *end pointer to end of buffer (last byte + 1) + */ + +/* + * bounds check input. + */ +#define ceph_decode_need(p, end, n, bad) \ + do { \ + if (unlikely(*(p) + (n) > (end))) \ + goto bad; \ + } while (0) + +#define ceph_decode_64(p, v) \ + do { \ + v = le64_to_cpu(*(__le64 *)*(p)); \ + *(p) += sizeof(u64); \ + } while (0) +#define ceph_decode_32(p, v) \ + do { \ + v = le32_to_cpu(*(__le32 *)*(p)); \ + *(p) += sizeof(u32); \ + } while (0) +#define ceph_decode_16(p, v) \ + do { \ + v = le16_to_cpu(*(__le16 *)*(p)); \ + *(p) += sizeof(u16); \ + } while (0) +#define ceph_decode_8(p, v) \ + do { \ + v = *(u8 *)*(p); \ + (*p)++; \ + } while (0) + +/* decode into an __le## */ +#define ceph_decode_64_le(p, v) \ + do { \ + v = *(__le64 *)*(p); \ + *(p) += sizeof(u64); \ + } while (0) +#define ceph_decode_32_le(p, v) \ + do { \ + v = *(__le32 *)*(p); \ + *(p) += sizeof(u32); \ + } while (0) +#define ceph_decode_16_le(p, v) \ + do { \ + v = *(__le16 *)*(p); \ + *(p) += sizeof(u16); \ + } while (0) + +#define ceph_decode_copy(p, pv, n) \ + do { \ + memcpy(pv, *(p), n); \ + *(p) += n; \ + } while (0) + +/* bounds check too */ +#define ceph_decode_64_safe(p, end, v, bad) \ + do { \ + ceph_decode_need(p, end, sizeof(u64), bad); \ + ceph_decode_64(p, v); \ + } while (0) +#define ceph_decode_32_safe(p, end, v, bad) \ + do { \ + ceph_decode_need(p, end, sizeof(u32), bad); \ + ceph_decode_32(p, v); \ + } while (0) +#define ceph_decode_16_safe(p, end, v, bad) \ + do { \ + ceph_decode_need(p, end, sizeof(u16), bad); \ + ceph_decode_16(p, v); \ + } while (0) + +#define ceph_decode_copy_safe(p, end, pv, n, bad) \ + do { \ + ceph_decode_need(p, end, n, bad); \ + ceph_decode_copy(p, pv, n); \ + } while (0) + +/* + * struct ceph_timespec <-> struct timespec + */ +#define ceph_decode_timespec(ts, tv) \ + do { \ + (ts)->tv_sec = le32_to_cpu((tv)->tv_sec); \ + (ts)->tv_nsec = le32_to_cpu((tv)->tv_nsec); \ + } while (0) +#define ceph_encode_timespec(tv, ts) \ + do { \ + (tv)->tv_sec = cpu_to_le32((ts)->tv_sec); \ + (tv)->tv_nsec = cpu_to_le32((ts)->tv_nsec); \ + } while (0) + + +/* + * encoders + */ +#define ceph_encode_64(p, v) \ + do { \ + *(__le64 *)*(p) = cpu_to_le64((v)); \ + *(p) += sizeof(u64); \ + } while (0) +#define ceph_encode_32(p, v) \ + do { \ + *(__le32 *)*(p) = cpu_to_le32((v)); \ + *(p) += sizeof(u32); \ + } while (0) +#define ceph_encode_16(p, v) \ + do { \ + *(__le16 *)*(p) = cpu_to_le16((v)); \ + *(p) += sizeof(u16); \ + } while (0) +#define ceph_encode_8(p, v) \ + do { \ + *(u8 *)*(p) = v; \ + (*(p))++; \ + } while (0) + +/* + * filepath, string encoders + */ +static inline void ceph_encode_filepath(void **p, void *end, + u64 ino, const char *path) +{ + u32 len = path ? strlen(path) : 0; + BUG_ON(*p + sizeof(ino) + sizeof(len) + len > end); + ceph_encode_64(p, ino); + ceph_encode_32(p, len); + if (len) + memcpy(*p, path, len); + *p += len; +} + +static inline void ceph_encode_string(void **p, void *end, + const char *s, u32 len) +{ + BUG_ON(*p + sizeof(len) + len > end); + ceph_encode_32(p, len); + if (len) + memcpy(*p, s, len); + *p += len; +} + + +#endif diff --git a/fs/ceph/messenger.c b/fs/ceph/messenger.c new file mode 100644 index 0000000..89d7289 --- /dev/null +++ b/fs/ceph/messenger.c @@ -0,0 +1,2389 @@ +#include <linux/crc32c.h> +#include <linux/kthread.h> +#include <linux/socket.h> +#include <linux/net.h> +#include <linux/string.h> +#include <linux/highmem.h> +#include <linux/ctype.h> +#include <net/tcp.h> + +#include "ceph_debug.h" +int ceph_debug_msgr __read_mostly; +#define DOUT_MASK DOUT_MASK_MSGR +#define DOUT_VAR ceph_debug_msgr + +#include "super.h" +#include "messenger.h" + + + +/* static tag bytes (protocol control messages) */ +static char tag_msg = CEPH_MSGR_TAG_MSG; +static char tag_ack = CEPH_MSGR_TAG_ACK; + + +static void ceph_queue_con(struct ceph_connection *con); +static void con_work(struct work_struct *); +static void ceph_fault(struct ceph_connection *con); + + +/* + * work queue for all reading and writing to/from the socket. + */ +struct workqueue_struct *ceph_msgr_wq; + +int ceph_msgr_init(void) +{ + ceph_msgr_wq = create_workqueue("ceph-msgr"); + if (IS_ERR(ceph_msgr_wq)) { + int ret = PTR_ERR(ceph_msgr_wq); + derr(0, "failed to create workqueue: %d\n", ret); + ceph_msgr_wq = NULL; + return ret; + } + return 0; +} + +void ceph_msgr_exit(void) +{ + destroy_workqueue(ceph_msgr_wq); +} + +/* from slub.c */ +static void print_section(char *text, u8 *addr, unsigned int length) +{ + int i, offset; + int newline = 1; + char ascii[17]; + + ascii[16] = 0; + + for (i = 0; i < length; i++) { + if (newline) { + printk(KERN_ERR "%8s 0x%p: ", text, addr + i); + newline = 0; + } + printk(KERN_CONT " %02x", addr[i]); + offset = i % 16; + ascii[offset] = isgraph(addr[i]) ? addr[i] : '.'; + if (offset == 15) { + printk(KERN_CONT " %s\n", ascii); + newline = 1; + } + } + if (!newline) { + i %= 16; + while (i < 16) { + printk(KERN_CONT " "); + ascii[i] = ' '; + i++; + } + printk(KERN_CONT " %s\n", ascii); + } +} + +/* + * socket callback functions + */ + +/* listen socket received a connection */ +static void ceph_accept_ready(struct sock *sk, int count_unused) +{ + struct ceph_messenger *msgr = (struct ceph_messenger *)sk->sk_user_data; + + dout(30, "ceph_accept_ready messenger %p sk_state = %u\n", + msgr, sk->sk_state); + if (sk->sk_state == TCP_LISTEN) + queue_work(ceph_msgr_wq, &msgr->awork); +} + +/* data available on socket, or listen socket received a connect */ +static void ceph_data_ready(struct sock *sk, int count_unused) +{ + struct ceph_connection *con = + (struct ceph_connection *)sk->sk_user_data; + if (sk->sk_state != TCP_CLOSE_WAIT) { + dout(30, "ceph_data_ready on %p state = %lu, queueing work\n", + con, con->state); + ceph_queue_con(con); + } +} + +/* socket has buffer space for writing */ +static void ceph_write_space(struct sock *sk) +{ + struct ceph_connection *con = + (struct ceph_connection *)sk->sk_user_data; + + /* only queue to workqueue if there is data we want to write. */ + if (test_bit(WRITE_PENDING, &con->state)) { + dout(30, "ceph_write_space %p queueing write work\n", con); + ceph_queue_con(con); + } else { + dout(30, "ceph_write_space %p nothing to write\n", con); + } + + /* since we have our own write_space, clear the SOCK_NOSPACE flag */ + clear_bit(SOCK_NOSPACE, &sk->sk_socket->flags); +} + +/* socket's state has changed */ +static void ceph_state_change(struct sock *sk) +{ + struct ceph_connection *con = + (struct ceph_connection *)sk->sk_user_data; + + dout(30, "ceph_state_change %p state = %lu sk_state = %u\n", + con, con->state, sk->sk_state); + + if (test_bit(CLOSED, &con->state)) + return; + + switch (sk->sk_state) { + case TCP_CLOSE: + dout(30, "ceph_state_change TCP_CLOSE\n"); + case TCP_CLOSE_WAIT: + dout(30, "ceph_state_change TCP_CLOSE_WAIT\n"); + set_bit(SOCK_CLOSED, &con->state); + if (test_bit(CONNECTING, &con->state)) + con->error_msg = "connection failed"; + else + con->error_msg = "socket closed"; + ceph_queue_con(con); + break; + case TCP_ESTABLISHED: + dout(30, "ceph_state_change TCP_ESTABLISHED\n"); + ceph_queue_con(con); + break; + } +} + +/* + * set up socket callbacks + */ +static void listen_sock_callbacks(struct socket *sock, + struct ceph_messenger *msgr) +{ + struct sock *sk = sock->sk; + sk->sk_user_data = (void *)msgr; + sk->sk_data_ready = ceph_accept_ready; +} + +static void set_sock_callbacks(struct socket *sock, + struct ceph_connection *con) +{ + struct sock *sk = sock->sk; + sk->sk_user_data = (void *)con; + sk->sk_data_ready = ceph_data_ready; + sk->sk_write_space = ceph_write_space; + sk->sk_state_change = ceph_state_change; +} + + +/* + * socket helpers + */ + +/* + * initiate connection to a remote socket. + */ +static struct socket *ceph_tcp_connect(struct ceph_connection *con) +{ + struct sockaddr *paddr = (struct sockaddr *)&con->peer_addr.ipaddr; + struct socket *sock; + int ret; + + BUG_ON(con->sock); + ret = sock_create_kern(AF_INET, SOCK_STREAM, IPPROTO_TCP, &sock); + if (ret) + return ERR_PTR(ret); + con->sock = sock; + sock->sk->sk_allocation = GFP_NOFS; + + set_sock_callbacks(sock, con); + + dout(20, "connect %u.%u.%u.%u:%u\n", + IPQUADPORT(*(struct sockaddr_in *)paddr)); + + ret = sock->ops->connect(sock, paddr, + sizeof(struct sockaddr_in), O_NONBLOCK); + if (ret == -EINPROGRESS) { + dout(20, "connect %u.%u.%u.%u:%u EINPROGRESS sk_state = %u\n", + IPQUADPORT(*(struct sockaddr_in *)paddr), + sock->sk->sk_state); + ret = 0; + } + if (ret < 0) { + derr(1, "connect %u.%u.%u.%u:%u error %d\n", + IPQUADPORT(*(struct sockaddr_in *)paddr), ret); + sock_release(sock); + con->sock = NULL; + con->error_msg = "connect error"; + } + + if (ret < 0) + return ERR_PTR(ret); + return sock; +} + +/* + * set up listening socket + */ +static int ceph_tcp_listen(struct ceph_messenger *msgr) +{ + int ret; + int optval = 1; + struct sockaddr_in *myaddr = &msgr->inst.addr.ipaddr; + int nlen; + struct socket *sock; + + ret = sock_create_kern(AF_INET, SOCK_STREAM, IPPROTO_TCP, &sock); + if (ret) + return ret; + sock->sk->sk_allocation = GFP_NOFS; + ret = kernel_setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, + (char *)&optval, sizeof(optval)); + if (ret < 0) { + derr(0, "failed to set SO_REUSEADDR: %d\n", ret); + goto err; + } + + ret = sock->ops->bind(sock, (struct sockaddr *)myaddr, + sizeof(*myaddr)); + if (ret < 0) { + derr(0, "Failed to bind: %d\n", ret); + goto err; + } + + /* what port did we bind to? */ + nlen = sizeof(*myaddr); + ret = sock->ops->getname(sock, (struct sockaddr *)myaddr, &nlen, + 0); + if (ret < 0) { + derr(0, "failed to getsockname: %d\n", ret); + goto err; + } + dout(0, "listening on %u.%u.%u.%u:%u\n", IPQUADPORT(*myaddr)); + + /* we don't care too much if this works or not */ + sock->ops->listen(sock, CEPH_MSGR_BACKUP); + + /* ok! */ + msgr->listen_sock = sock; + listen_sock_callbacks(sock, msgr); + return 0; + +err: + sock_release(sock); + return ret; +} + +/* + * accept a connection + */ +static int ceph_tcp_accept(struct socket *lsock, struct ceph_connection *con) +{ + struct socket *sock; + int ret; + + ret = sock_create_kern(AF_INET, SOCK_STREAM, IPPROTO_TCP, &sock); + if (ret) + return ret; + con->sock = sock; + sock->sk->sk_allocation = GFP_NOFS; + + ret = lsock->ops->accept(lsock, sock, O_NONBLOCK); + if (ret < 0) { + derr(0, "accept error: %d\n", ret); + goto err; + } + + sock->ops = lsock->ops; + sock->type = lsock->type; + set_sock_callbacks(sock, con); + return ret; + +err: + sock->ops->shutdown(sock, SHUT_RDWR); + sock_release(sock); + return ret; +} + +static int ceph_tcp_recvmsg(struct socket *sock, void *buf, size_t len) +{ + struct kvec iov = {buf, len}; + struct msghdr msg = { .msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL }; + + return kernel_recvmsg(sock, &msg, &iov, 1, len, msg.msg_flags); +} + +/* + * write something. @more is true if caller will be sending more data + * shortly. + */ +static int ceph_tcp_sendmsg(struct socket *sock, struct kvec *iov, + size_t kvlen, size_t len, int more) +{ + struct msghdr msg = { .msg_flags = MSG_DONTWAIT | MSG_NOSIGNAL }; + + if (more) + msg.msg_flags |= MSG_MORE; + else + msg.msg_flags |= MSG_EOR; /* superfluous, but what the hell */ + + return kernel_sendmsg(sock, &msg, iov, kvlen, len); +} + + +/* + * create a new connection. + */ +static struct ceph_connection *new_connection(struct ceph_messenger *msgr) +{ + struct ceph_connection *con; + + con = kzalloc(sizeof(struct ceph_connection), GFP_NOFS); + if (con == NULL) + return NULL; + con->msgr = msgr; + atomic_set(&con->nref, 1); + INIT_LIST_HEAD(&con->list_all); + INIT_LIST_HEAD(&con->list_bucket); + spin_lock_init(&con->out_queue_lock); + INIT_LIST_HEAD(&con->out_queue); + INIT_LIST_HEAD(&con->out_sent); + INIT_DELAYED_WORK(&con->work, con_work); + + dout(20, "new connection: %p\n", con); + return con; +} + +/* + * The con_tree radix_tree has an unsigned long key and void * value. + * Since ceph_entity_addr is bigger than that, we use a trivial hash + * key, and point to a list_head in ceph_connection, as you would with + * a hash table. If the trivial hash collides, we just traverse the + * (hopefully short) list until we find what we want. + */ +static unsigned long hash_addr(struct ceph_entity_addr *addr) +{ + unsigned long key; + + key = *(u32 *)&addr->ipaddr.sin_addr.s_addr; + key ^= *(u16 *)&addr->ipaddr.sin_port; + return key; +} + +/* + * Get an existing connection, if any, for given addr. Note that we + * may need to traverse the list_bucket list, which has to "head." + * + * called under con_lock. + */ +static struct ceph_connection *__get_connection(struct ceph_messenger *msgr, + struct ceph_entity_addr *addr) +{ + struct ceph_connection *con = NULL; + struct list_head *head, *p; + unsigned long key = hash_addr(addr); + + head = radix_tree_lookup(&msgr->con_tree, key); + if (head == NULL) + return NULL; + con = list_entry(head, struct ceph_connection, list_bucket); + if (memcmp(&con->peer_addr, addr, sizeof(addr)) == 0) + goto yes; + list_for_each(p, head) { + con = list_entry(p, struct ceph_connection, list_bucket); + if (memcmp(&con->peer_addr, addr, sizeof(addr)) == 0) + goto yes; + } + return NULL; + +yes: + atomic_inc(&con->nref); + dout(20, "get_connection %p nref = %d -> %d\n", con, + atomic_read(&con->nref) - 1, atomic_read(&con->nref)); + return con; +} + + +/* + * Shutdown/close the socket for the given connection. + */ +static int con_close_socket(struct ceph_connection *con) +{ + int rc; + + dout(10, "con_close_socket on %p sock %p\n", con, con->sock); + if (!con->sock) + return 0; + rc = con->sock->ops->shutdown(con->sock, SHUT_RDWR); + sock_release(con->sock); + con->sock = NULL; + return rc; +} + +/* + * drop a reference + */ +static void put_connection(struct ceph_connection *con) +{ + dout(20, "put_connection %p nref = %d -> %d\n", con, + atomic_read(&con->nref), atomic_read(&con->nref) - 1); + BUG_ON(atomic_read(&con->nref) == 0); + if (atomic_dec_and_test(&con->nref)) { + dout(20, "put_connection %p destroying\n", con); + ceph_msg_put_list(&con->out_queue); + ceph_msg_put_list(&con->out_sent); + set_bit(CLOSED, &con->state); + con_close_socket(con); /* silently ignore possible errors */ + kfree(con); + } +} + +/* + * add a connection to the con_tree. + * + * called under con_lock. + */ +static int __register_connection(struct ceph_messenger *msgr, + struct ceph_connection *con) +{ + struct list_head *head; + unsigned long key = hash_addr(&con->peer_addr); + int rc = 0; + + dout(20, "register_connection %p %d -> %d\n", con, + atomic_read(&con->nref), atomic_read(&con->nref) + 1); + atomic_inc(&con->nref); + + /* if were just ACCEPTING this connection, it is already on the + * con_all and con_accepting lists. */ + if (test_and_clear_bit(ACCEPTING, &con->state)) { + list_del_init(&con->list_bucket); + put_connection(con); + } else { + list_add(&con->list_all, &msgr->con_all); + } + + head = radix_tree_lookup(&msgr->con_tree, key); + if (head) { + dout(20, "register_connection %p in old bucket %lu head %p\n", + con, key, head); + list_add(&con->list_bucket, head); + } else { + dout(20, "register_connection %p in new bucket %lu head %p\n", + con, key, &con->list_bucket); + INIT_LIST_HEAD(&con->list_bucket); /* empty */ + rc = radix_tree_insert(&msgr->con_tree, key, &con->list_bucket); + if (rc < 0) { + list_del(&con->list_all); + put_connection(con); + return rc; + } + } + set_bit(REGISTERED, &con->state); + return 0; +} + +/* + * called under con_lock. + */ +static void add_connection_accepting(struct ceph_messenger *msgr, + struct ceph_connection *con) +{ + dout(20, "add_connection_accepting %p nref = %d -> %d\n", con, + atomic_read(&con->nref), atomic_read(&con->nref) + 1); + atomic_inc(&con->nref); + spin_lock(&msgr->con_lock); + list_add(&con->list_all, &msgr->con_all); + spin_unlock(&msgr->con_lock); +} + +/* + * Remove connection from all list. Also, from con_tree, if it should + * have been there. + * + * called under con_lock. + */ +static void __remove_connection(struct ceph_messenger *msgr, + struct ceph_connection *con) +{ + unsigned long key; + void **slot, *val; + + dout(0, "__remove_connection: %p\n", con); + dout(20, "__remove_connection %p\n", con); + if (list_empty(&con->list_all)) { + dout(20, "__remove_connection %p not registered\n", con); + return; + } + list_del_init(&con->list_all); + if (test_bit(REGISTERED, &con->state)) { + key = hash_addr(&con->peer_addr); + if (list_empty(&con->list_bucket)) { + /* last one in this bucket */ + dout(20, "__remove_connection %p and bucket %lu\n", + con, key); + radix_tree_delete(&msgr->con_tree, key); + } else { + /* if we share this bucket, and the radix tree points + * to us, adjust it to point to the next guy. */ + slot = radix_tree_lookup_slot(&msgr->con_tree, key); + val = radix_tree_deref_slot(slot); + dout(20, "__remove_connection %p from bucket %lu " + "head %p\n", con, key, val); + if (val == &con->list_bucket) { + dout(20, "__remove_connection adjusting bucket" + " for %lu to next item, %p\n", key, + con->list_bucket.next); + radix_tree_replace_slot(slot, + con->list_bucket.next); + } + list_del_init(&con->list_bucket); + } + } + if (test_and_clear_bit(ACCEPTING, &con->state)) + list_del_init(&con->list_bucket); + put_connection(con); +} + +static void remove_connection(struct ceph_messenger *msgr, + struct ceph_connection *con) +{ + spin_lock(&msgr->con_lock); + __remove_connection(msgr, con); + spin_unlock(&msgr->con_lock); +} + +/* + * replace another connection + * (old and new should be for the _same_ peer, + * and thus in the same bucket in the radix tree) + */ +static void __replace_connection(struct ceph_messenger *msgr, + struct ceph_connection *old, + struct ceph_connection *new) +{ + unsigned long key = hash_addr(&new->peer_addr); + void **slot; + + dout(0, "replace_connection %p with %p\n", old, new); + dout(10, "replace_connection %p with %p\n", old, new); + + /* replace in con_tree */ + slot = radix_tree_lookup_slot(&msgr->con_tree, key); + if (*slot == &old->list_bucket) + radix_tree_replace_slot(slot, &new->list_bucket); + else + BUG_ON(list_empty(&old->list_bucket)); + if (!list_empty(&old->list_bucket)) { + /* replace old with new in bucket list */ + list_add(&new->list_bucket, &old->list_bucket); + list_del_init(&old->list_bucket); + } + + /* take old connections message queue */ + spin_lock(&old->out_queue_lock); + if (!list_empty(&old->out_queue)) + list_splice_init(&new->out_queue, &old->out_queue); + spin_unlock(&old->out_queue_lock); + + new->connect_seq = le32_to_cpu(new->in_connect.connect_seq); + new->out_seq = old->out_seq; + new->peer_name = old->peer_name; + + set_bit(CLOSED, &old->state); + put_connection(old); /* dec reference count */ + + clear_bit(ACCEPTING, &new->state); +} + + + + +/* + * We maintain a global counter to order connection attempts. Get + * a unique seq greater than @gt. + */ +static u32 get_global_seq(struct ceph_messenger *msgr, u32 gt) +{ + u32 ret; + + spin_lock(&msgr->global_seq_lock); + if (msgr->global_seq < gt) + msgr->global_seq = gt; + ret = ++msgr->global_seq; + spin_unlock(&msgr->global_seq_lock); + return ret; +} + + + + +/* + * Prepare footer for currently outgoing message, and finish things + * off. Assumes out_kvec* are already valid.. we just add on to the end. + */ +static void prepare_write_message_footer(struct ceph_connection *con, int v) +{ + struct ceph_msg *m = con->out_msg; + + dout(10, "prepare_write_message_footer %p\n", con); + con->out_kvec[v].iov_base = &m->footer; + con->out_kvec[v].iov_len = sizeof(m->footer); + con->out_kvec_bytes += sizeof(m->footer); + con->out_kvec_left++; + con->out_msg = NULL; /* we're done with this one */ + con->out_more = 0; /* end of message */ +} + +/* + * Prepare headers for the next outgoing message. + */ +static void prepare_write_message(struct ceph_connection *con) +{ + struct ceph_msg *m; + int v = 0; + + con->out_kvec_bytes = 0; + + /* Sneak an ack in there first? If we can get it into the same + * TCP packet that's a good thing. */ + if (con->in_seq > con->in_seq_acked) { + con->in_seq_acked = con->in_seq; + con->out_kvec[v].iov_base = &tag_ack; + con->out_kvec[v++].iov_len = 1; + con->out_temp_ack = cpu_to_le32(con->in_seq_acked); + con->out_kvec[v].iov_base = &con->out_temp_ack; + con->out_kvec[v++].iov_len = 4; + con->out_kvec_bytes = 1 + 4; + } + + /* move message to sending/sent list */ + m = list_entry(con->out_queue.next, + struct ceph_msg, list_head); + list_del_init(&m->list_head); + list_add_tail(&m->list_head, &con->out_sent); + con->out_msg = m; /* we don't bother taking a reference here. */ + + dout(20, "prepare_write_message %p seq %lld type %d len %d+%d %d pgs\n", + m, le64_to_cpu(m->hdr.seq), le16_to_cpu(m->hdr.type), + le32_to_cpu(m->hdr.front_len), le32_to_cpu(m->hdr.data_len), + m->nr_pages); + BUG_ON(le32_to_cpu(m->hdr.front_len) != m->front.iov_len); + + /* tag + hdr + front */ + con->out_kvec[v].iov_base = &tag_msg; + con->out_kvec[v++].iov_len = 1; + con->out_kvec[v].iov_base = &m->hdr; + con->out_kvec[v++].iov_len = sizeof(m->hdr); + con->out_kvec[v++] = m->front; + con->out_kvec_left = v; + con->out_kvec_bytes += 1 + sizeof(m->hdr) + m->front.iov_len; + con->out_kvec_cur = con->out_kvec; + + /* fill in crc (except data pages), footer */ + con->out_msg->hdr.crc = + cpu_to_le32(crc32c_le(0, (void *)&m->hdr, + sizeof(m->hdr) - sizeof(m->hdr.crc))); + con->out_msg->footer.flags = 0; + con->out_msg->footer.front_crc = + cpu_to_le32(crc32c_le(0, m->front.iov_base, m->front.iov_len)); + con->out_msg->footer.data_crc = 0; + + /* is there a data payload? */ + if (le32_to_cpu(m->hdr.data_len) > 0) { + /* initialize page iterator */ + con->out_msg_pos.page = 0; + con->out_msg_pos.page_pos = + le16_to_cpu(m->hdr.data_off) & ~PAGE_MASK; + con->out_msg_pos.data_pos = 0; + con->out_msg_pos.did_page_crc = 0; + con->out_more = 1; /* data + footer will follow */ + } else { + /* no, queue up footer too and be done */ + prepare_write_message_footer(con, v); + } + + set_bit(WRITE_PENDING, &con->state); +} + +/* + * Prepare an ack. + */ +static void prepare_write_ack(struct ceph_connection *con) +{ + dout(20, "prepare_write_ack %p %u -> %u\n", con, + con->in_seq_acked, con->in_seq); + con->in_seq_acked = con->in_seq; + + con->out_kvec[0].iov_base = &tag_ack; + con->out_kvec[0].iov_len = 1; + con->out_temp_ack = cpu_to_le32(con->in_seq_acked); + con->out_kvec[1].iov_base = &con->out_temp_ack; + con->out_kvec[1].iov_len = 4; + con->out_kvec_left = 2; + con->out_kvec_bytes = 1 + 4; + con->out_kvec_cur = con->out_kvec; + con->out_more = 1; /* more will follow.. eventually.. */ + set_bit(WRITE_PENDING, &con->state); +} + +/* + * Connection negotiation. + */ + +/* + * We connected to a peer and are saying hello. + */ +static void prepare_write_connect(struct ceph_messenger *msgr, + struct ceph_connection *con) +{ + int len = strlen(CEPH_BANNER); + + dout(10, "prepare_write_connect %p\n", con); + con->out_connect.host_type = cpu_to_le32(CEPH_ENTITY_TYPE_CLIENT); + con->out_connect.connect_seq = cpu_to_le32(con->connect_seq); + con->out_connect.global_seq = + cpu_to_le32(get_global_seq(con->msgr, 0)); + con->out_connect.flags = 0; + if (test_bit(LOSSYTX, &con->state)) + con->out_connect.flags = CEPH_MSG_CONNECT_LOSSY; + + con->out_kvec[0].iov_base = CEPH_BANNER; + con->out_kvec[0].iov_len = len; + con->out_kvec[1].iov_base = &msgr->inst.addr; + con->out_kvec[1].iov_len = sizeof(msgr->inst.addr); + con->out_kvec[2].iov_base = &con->out_connect; + con->out_kvec[2].iov_len = sizeof(con->out_connect); + con->out_kvec_left = 3; + con->out_kvec_bytes = len + sizeof(msgr->inst.addr) + + sizeof(con->out_connect); + con->out_kvec_cur = con->out_kvec; + con->out_more = 0; + set_bit(WRITE_PENDING, &con->state); +} + +static void prepare_write_connect_retry(struct ceph_messenger *msgr, + struct ceph_connection *con) +{ + dout(10, "prepare_write_connect_retry %p\n", con); + con->out_connect.connect_seq = cpu_to_le32(con->connect_seq); + con->out_connect.global_seq = + cpu_to_le32(get_global_seq(con->msgr, 0)); + + con->out_kvec[0].iov_base = &con->out_connect; + con->out_kvec[0].iov_len = sizeof(con->out_connect); + con->out_kvec_left = 1; + con->out_kvec_bytes = sizeof(con->out_connect); + con->out_kvec_cur = con->out_kvec; + con->out_more = 0; + set_bit(WRITE_PENDING, &con->state); +} + +/* + * We accepted a connection and are saying hello. + */ +static void prepare_write_accept_hello(struct ceph_messenger *msgr, + struct ceph_connection *con) +{ + int len = strlen(CEPH_BANNER); + + dout(10, "prepare_write_accept_hello %p\n", con); + con->out_kvec[0].iov_base = CEPH_BANNER; + con->out_kvec[0].iov_len = len; + con->out_kvec[1].iov_base = &msgr->inst.addr; + con->out_kvec[1].iov_len = sizeof(msgr->inst.addr); + con->out_kvec_left = 2; + con->out_kvec_bytes = len + sizeof(msgr->inst.addr); + con->out_kvec_cur = con->out_kvec; + con->out_more = 0; + set_bit(WRITE_PENDING, &con->state); +} + +/* + * Reply to a connect attempt, indicating whether the negotiation has + * succeeded or must continue. + */ +static void prepare_write_accept_reply(struct ceph_connection *con, bool retry) +{ + dout(10, "prepare_write_accept_reply %p\n", con); + con->out_reply.flags = 0; + if (test_bit(LOSSYTX, &con->state)) + con->out_reply.flags = CEPH_MSG_CONNECT_LOSSY; + + con->out_kvec[0].iov_base = &con->out_reply; + con->out_kvec[0].iov_len = sizeof(con->out_reply); + con->out_kvec_left = 1; + con->out_kvec_bytes = sizeof(con->out_reply); + con->out_kvec_cur = con->out_kvec; + con->out_more = 0; + set_bit(WRITE_PENDING, &con->state); + + if (retry) + /* we'll re-read the connect request, sans the hello + addr */ + con->in_base_pos = strlen(CEPH_BANNER) + + sizeof(con->msgr->inst.addr); +} + + + +/* + * write as much of pending kvecs to the socket as we can. + * 1 -> done + * 0 -> socket full, but more to do + * <0 -> error + */ +static int write_partial_kvec(struct ceph_connection *con) +{ + int ret; + + dout(10, "write_partial_kvec %p %d left\n", con, con->out_kvec_bytes); + while (con->out_kvec_bytes > 0) { + ret = ceph_tcp_sendmsg(con->sock, con->out_kvec_cur, + con->out_kvec_left, con->out_kvec_bytes, + con->out_more); + if (ret <= 0) + goto out; + con->out_kvec_bytes -= ret; + if (con->out_kvec_bytes == 0) + break; /* done */ + while (ret > 0) { + if (ret >= con->out_kvec_cur->iov_len) { + ret -= con->out_kvec_cur->iov_len; + con->out_kvec_cur++; + con->out_kvec_left--; + } else { + con->out_kvec_cur->iov_len -= ret; + con->out_kvec_cur->iov_base += ret; + ret = 0; + break; + } + } + } + con->out_kvec_left = 0; + ret = 1; +out: + dout(30, "write_partial_kvec %p %d left in %d kvecs ret = %d\n", con, + con->out_kvec_bytes, con->out_kvec_left, ret); + return ret; /* done! */ +} + +/* + * Write as much message data payload as we can. If we finish, queue + * up the footer. + * 1 -> done, footer is now queued in out_kvec[]. + * 0 -> socket full, but more to do + * <0 -> error + */ +static int write_partial_msg_pages(struct ceph_connection *con) +{ + struct ceph_client *client = con->msgr->parent; + struct ceph_msg *msg = con->out_msg; + unsigned data_len = le32_to_cpu(msg->hdr.data_len); + size_t len; + int crc = !(client->mount_args.flags & CEPH_MOUNT_NOCRC); + int ret; + + dout(30, "write_partial_msg_pages %p msg %p page %d/%d offset %d\n", + con, con->out_msg, con->out_msg_pos.page, con->out_msg->nr_pages, + con->out_msg_pos.page_pos); + + while (con->out_msg_pos.page < con->out_msg->nr_pages) { + struct page *page = NULL; + void *kaddr = NULL; + + /* + * if we are calculating the data crc (the default), we need + * to map the page. if our pages[] has been revoked, use the + * zero page. + */ + mutex_lock(&msg->page_mutex); + if (msg->pages) { + page = msg->pages[con->out_msg_pos.page]; + if (crc) + kaddr = kmap(page); + } else { + page = con->msgr->zero_page; + if (crc) + kaddr = page_address(con->msgr->zero_page); + } + len = min((int)(PAGE_SIZE - con->out_msg_pos.page_pos), + (int)(data_len - con->out_msg_pos.data_pos)); + if (crc && !con->out_msg_pos.did_page_crc) { + void *base = kaddr + con->out_msg_pos.page_pos; + u32 tmpcrc = le32_to_cpu(con->out_msg->footer.data_crc); + + con->out_msg->footer.data_crc = + cpu_to_le32(crc32c_le(tmpcrc, base, len)); + con->out_msg_pos.did_page_crc = 1; + } + + ret = kernel_sendpage(con->sock, page, + con->out_msg_pos.page_pos, len, + MSG_DONTWAIT | MSG_NOSIGNAL | + MSG_MORE); + + if (crc && msg->pages) + kunmap(page); + + mutex_unlock(&msg->page_mutex); + if (ret <= 0) + goto out; + + con->out_msg_pos.data_pos += ret; + con->out_msg_pos.page_pos += ret; + if (ret == len) { + con->out_msg_pos.page_pos = 0; + con->out_msg_pos.page++; + con->out_msg_pos.did_page_crc = 0; + } + } + + dout(30, "write_partial_msg_pages %p msg %p done\n", con, msg); + + /* prepare and queue up footer, too */ + if (!crc) + con->out_msg->footer.flags |= + cpu_to_le32(CEPH_MSG_FOOTER_NOCRC); + con->out_kvec_bytes = 0; + con->out_kvec_left = 0; + con->out_kvec_cur = con->out_kvec; + prepare_write_message_footer(con, 0); + ret = 1; +out: + return ret; +} + + + +/* + * Prepare to read connection handshake, or an ack. + */ +static void prepare_read_connect(struct ceph_connection *con) +{ + dout(10, "prepare_read_connect %p\n", con); + con->in_base_pos = 0; +} + +static void prepare_read_ack(struct ceph_connection *con) +{ + dout(10, "prepare_read_ack %p\n", con); + con->in_base_pos = 0; +} + +static void prepare_read_tag(struct ceph_connection *con) +{ + dout(10, "prepare_read_tag %p\n", con); + con->in_base_pos = 0; + con->in_tag = CEPH_MSGR_TAG_READY; +} + +/* + * Prepare to read a message. + */ +static int prepare_read_message(struct ceph_connection *con) +{ + int err; + + dout(10, "prepare_read_message %p\n", con); + con->in_base_pos = 0; + BUG_ON(con->in_msg != NULL); + con->in_msg = ceph_msg_new(0, 0, 0, 0, NULL); + if (IS_ERR(con->in_msg)) { + err = PTR_ERR(con->in_msg); + con->in_msg = NULL; + con->error_msg = "out of memory for incoming message"; + return err; + } + con->in_front_crc = con->in_data_crc = 0; + return 0; +} + + +static int read_partial(struct ceph_connection *con, + int *to, int size, void *object) +{ + *to += size; + while (con->in_base_pos < *to) { + int left = *to - con->in_base_pos; + int have = size - left; + int ret = ceph_tcp_recvmsg(con->sock, object + have, left); + if (ret <= 0) + return ret; + con->in_base_pos += ret; + } + return 1; +} + + +/* + * Read all or part of the connect-side handshake on a new connection + */ +static int read_partial_connect(struct ceph_connection *con) +{ + int ret, to = 0; + + dout(20, "read_partial_connect %p at %d\n", con, con->in_base_pos); + + /* peer's banner */ + ret = read_partial(con, &to, strlen(CEPH_BANNER), con->in_banner); + if (ret <= 0) + goto out; + ret = read_partial(con, &to, sizeof(con->actual_peer_addr), + &con->actual_peer_addr); + if (ret <= 0) + goto out; + ret = read_partial(con, &to, sizeof(con->in_reply), &con->in_reply); + if (ret <= 0) + goto out; + + dout(20, "read_partial_connect %p connect_seq = %u, global_seq = %u\n", + con, le32_to_cpu(con->in_reply.connect_seq), + le32_to_cpu(con->in_reply.global_seq)); +out: + return ret; +} + +/* + * Verify the hello banner looks okay. + */ +static int verify_hello(struct ceph_connection *con) +{ + if (memcmp(con->in_banner, CEPH_BANNER, strlen(CEPH_BANNER))) { + derr(10, "connection to/from %u.%u.%u.%u:%u has bad banner\n", + IPQUADPORT(con->peer_addr.ipaddr)); + con->error_msg = "protocol error, bad banner"; + return -1; + } + return 0; +} + +/* + * Reset a connection. Discard all incoming and outgoing messages + * and clear *_seq state. + */ +static void reset_connection(struct ceph_connection *con) +{ + derr(1, "%s%d %u.%u.%u.%u:%u connection reset\n", + ENTITY_NAME(con->peer_name), + IPQUADPORT(con->peer_addr.ipaddr)); + + /* reset connection, out_queue, msg_ and connect_seq */ + /* discard existing out_queue and msg_seq */ + spin_lock(&con->out_queue_lock); + ceph_msg_put_list(&con->out_queue); + ceph_msg_put_list(&con->out_sent); + + con->connect_seq = 0; + con->out_seq = 0; + con->out_msg = NULL; + con->in_seq = 0; + con->in_msg = NULL; + spin_unlock(&con->out_queue_lock); +} + + +static int process_connect(struct ceph_connection *con) +{ + dout(20, "process_connect on %p tag %d\n", con, (int)con->in_tag); + + if (verify_hello(con) < 0) + return -1; + + /* + * Make sure the other end is who we wanted. note that the other + * end may not yet know their ip address, so if it's 0.0.0.0, give + * them the benefit of the doubt. + */ + if (!ceph_entity_addr_is_local(&con->peer_addr, + &con->actual_peer_addr) && + con->actual_peer_addr.ipaddr.sin_addr.s_addr != 0) { + derr(1, "process_connect wrong peer, want %u.%u.%u.%u:%u/%d, " + "got %u.%u.%u.%u:%u/%d, wtf\n", + IPQUADPORT(con->peer_addr.ipaddr), + con->peer_addr.nonce, + IPQUADPORT(con->actual_peer_addr.ipaddr), + con->actual_peer_addr.nonce); + con->error_msg = "protocol error, wrong peer"; + return -1; + } + + switch (con->in_reply.tag) { + case CEPH_MSGR_TAG_RESETSESSION: + /* + * If we connected with a large connect_seq but the peer + * has no record of a session with us (no connection, or + * connect_seq == 0), they will send RESETSESION to indicate + * that they must have reset their session, and may have + * dropped messages. + */ + dout(10, "process_connect got RESET peer seq %u\n", + le32_to_cpu(con->in_connect.connect_seq)); + reset_connection(con); + prepare_write_connect_retry(con->msgr, con); + prepare_read_connect(con); + + /* Tell ceph about it. */ + con->msgr->peer_reset(con->msgr->parent, &con->peer_addr, + &con->peer_name); + break; + + case CEPH_MSGR_TAG_RETRY_SESSION: + /* + * If we sent a smaller connect_seq than the peer has, try + * again with a larger value. + */ + dout(10, + "process_connect got RETRY my seq = %u, peer_seq = %u\n", + le32_to_cpu(con->out_connect.connect_seq), + le32_to_cpu(con->in_connect.connect_seq)); + con->connect_seq = le32_to_cpu(con->in_connect.connect_seq); + prepare_write_connect_retry(con->msgr, con); + prepare_read_connect(con); + break; + + case CEPH_MSGR_TAG_RETRY_GLOBAL: + /* + * If we sent a smaller global_seq than the peer has, try + * again with a larger value. + */ + dout(10, + "process_connect got RETRY_GLOBAL my %u, peer_gseq = %u\n", + con->peer_global_seq, + le32_to_cpu(con->in_connect.global_seq)); + get_global_seq(con->msgr, + le32_to_cpu(con->in_connect.global_seq)); + prepare_write_connect_retry(con->msgr, con); + prepare_read_connect(con); + break; + + case CEPH_MSGR_TAG_WAIT: + /* + * If there is a connection race (we are opening connections to + * each other), one of us may just have to WAIT. We will keep + * our queued messages, in expectation of being replaced by an + * incoming connection. + */ + dout(10, "process_connect peer connecting WAIT\n"); + set_bit(WAIT, &con->state); + con_close_socket(con); + break; + + case CEPH_MSGR_TAG_READY: + clear_bit(CONNECTING, &con->state); + if (con->in_reply.flags & CEPH_MSG_CONNECT_LOSSY) + set_bit(LOSSYRX, &con->state); + con->peer_global_seq = le32_to_cpu(con->in_reply.global_seq); + con->connect_seq++; + dout(10, "process_connect got READY gseq %d cseq %d (%d)\n", + con->peer_global_seq, + le32_to_cpu(con->in_reply.connect_seq), + con->connect_seq); + WARN_ON(con->connect_seq != + le32_to_cpu(con->in_reply.connect_seq)); + + con->delay = 0; /* reset backoff memory */ + prepare_read_tag(con); + break; + + default: + derr(1, "process_connect protocol error, will retry\n"); + con->error_msg = "protocol error, garbage tag during connect"; + return -1; + } + return 0; +} + + +/* + * Read all or part of the accept-side handshake on a newly accepted + * connection. + */ +static int read_partial_accept(struct ceph_connection *con) +{ + int ret; + int to = 0; + + /* banner */ + ret = read_partial(con, &to, strlen(CEPH_BANNER), con->in_banner); + if (ret <= 0) + return ret; + ret = read_partial(con, &to, sizeof(con->peer_addr), &con->peer_addr); + if (ret <= 0) + return ret; + ret = read_partial(con, &to, sizeof(con->in_connect), &con->in_connect); + if (ret <= 0) + return ret; + return 1; +} + +/* + * Call after a new connection's handshake has been read. + */ +static int process_accept(struct ceph_connection *con) +{ + struct ceph_connection *existing; + struct ceph_messenger *msgr = con->msgr; + u32 peer_gseq = le32_to_cpu(con->in_connect.global_seq); + u32 peer_cseq = le32_to_cpu(con->in_connect.connect_seq); + bool retry = true; + bool replace = false; + + dout(10, "process_accept %p got gseq %d cseq %d\n", con, + peer_gseq, peer_cseq); + + if (verify_hello(con) < 0) + return -1; + + /* note flags */ + if (con->in_connect.flags & CEPH_MSG_CONNECT_LOSSY) + set_bit(LOSSYRX, &con->state); + + /* do we have an existing connection for this peer? */ + if (radix_tree_preload(GFP_NOFS) < 0) { + derr(10, "ENOMEM in process_accept\n"); + con->error_msg = "out of memory"; + return -1; + } + + memset(&con->out_reply, 0, sizeof(con->out_reply)); + + spin_lock(&msgr->con_lock); + existing = __get_connection(msgr, &con->peer_addr); + if (existing) { + if (peer_gseq < existing->peer_global_seq) { + /* out of order connection attempt */ + con->out_reply.tag = CEPH_MSGR_TAG_RETRY_GLOBAL; + con->out_reply.global_seq = + cpu_to_le32(con->peer_global_seq); + goto reply; + } + if (test_bit(LOSSYTX, &existing->state)) { + dout(20, "process_accept %p replacing LOSSYTX %p\n", + con, existing); + replace = true; + goto accept; + } + if (peer_cseq < existing->connect_seq) { + if (peer_cseq == 0) { + /* peer reset, then connected to us */ + reset_connection(existing); + con->msgr->peer_reset(con->msgr->parent, + &con->peer_addr, + &con->peer_name); + replace = true; + goto accept; + } + + /* old attempt or peer didn't get the READY */ + con->out_reply.tag = CEPH_MSGR_TAG_RETRY_SESSION; + con->out_reply.connect_seq = + cpu_to_le32(existing->connect_seq); + goto reply; + } + + if (peer_cseq == existing->connect_seq) { + /* connection race */ + dout(20, "process_accept connection race state = %lu\n", + con->state); + if (ceph_entity_addr_equal(&msgr->inst.addr, + &con->peer_addr)) { + /* incoming connection wins.. */ + replace = true; + goto accept; + } + + /* our existing outgoing connection wins, tell peer + to wait for our outging connection to go through */ + con->out_reply.tag = CEPH_MSGR_TAG_WAIT; + goto reply; + } + + if (existing->connect_seq == 0 && + peer_cseq > existing->connect_seq) { + /* we reset and already reconnecting */ + con->out_reply.tag = CEPH_MSGR_TAG_RESETSESSION; + goto reply; + } + + WARN_ON(le32_to_cpu(con->in_connect.connect_seq) <= + existing->connect_seq); + WARN_ON(le32_to_cpu(con->in_connect.global_seq) < + existing->peer_global_seq); + if (existing->connect_seq == 0) { + /* we reset, sending RESETSESSION */ + con->out_reply.tag = CEPH_MSGR_TAG_RESETSESSION; + goto reply; + } + + /* reconnect, replace connection */ + replace = true; + goto accept; + } + + if (peer_cseq == 0) { + dout(20, "process_accept no existing connection, opening\n"); + goto accept; + } else { + dout(20, "process_accept no existing connection, we reset\n"); + con->out_reply.tag = CEPH_MSGR_TAG_RESETSESSION; + goto reply; + } + + +accept: + /* accept this connection */ + con->connect_seq = peer_cseq + 1; + con->peer_global_seq = peer_gseq; + dout(10, "process_accept %p cseq %d peer_gseq %d %s\n", con, + con->connect_seq, peer_gseq, replace ? "replace" : "new"); + + con->out_reply.tag = CEPH_MSGR_TAG_READY; + con->out_reply.global_seq = cpu_to_le32(get_global_seq(con->msgr, 0)); + con->out_reply.connect_seq = cpu_to_le32(peer_cseq + 1); + + retry = false; + prepare_read_tag(con); + + /* do this _after_ con is ready to go */ + if (replace) + __replace_connection(msgr, existing, con); + else + __register_connection(msgr, con); + put_connection(con); + +reply: + if (existing) + put_connection(existing); + prepare_write_accept_reply(con, retry); + + spin_unlock(&msgr->con_lock); + radix_tree_preload_end(); + + ceph_queue_con(con); + return 0; +} + +/* + * read (part of) an ack + */ +static int read_partial_ack(struct ceph_connection *con) +{ + int to = 0; + + return read_partial(con, &to, sizeof(con->in_temp_ack), + &con->in_temp_ack); +} + + +/* + * We can finally discard anything that's been acked. + */ +static void process_ack(struct ceph_connection *con) +{ + struct ceph_msg *m; + u32 ack = le32_to_cpu(con->in_temp_ack); + u64 seq; + + spin_lock(&con->out_queue_lock); + while (!list_empty(&con->out_sent)) { + m = list_entry(con->out_sent.next, struct ceph_msg, list_head); + seq = le64_to_cpu(m->hdr.seq); + if (seq > ack) + break; + dout(5, "got ack for seq %llu type %d at %p\n", seq, + le16_to_cpu(m->hdr.type), m); + list_del_init(&m->list_head); + ceph_msg_put(m); + } + spin_unlock(&con->out_queue_lock); + prepare_read_tag(con); +} + + + + + + +/* + * read (part of) a message. + */ +static int read_partial_message(struct ceph_connection *con) +{ + struct ceph_msg *m = con->in_msg; + void *p; + int ret; + int to, want, left; + unsigned front_len, data_len, data_off; + struct ceph_client *client = con->msgr->parent; + int datacrc = !(client->mount_args.flags & CEPH_MOUNT_NOCRC); + + dout(20, "read_partial_message con %p msg %p\n", con, m); + + /* header */ + while (con->in_base_pos < sizeof(m->hdr)) { + left = sizeof(m->hdr) - con->in_base_pos; + ret = ceph_tcp_recvmsg(con->sock, + (char *)&m->hdr + con->in_base_pos, + left); + if (ret <= 0) + return ret; + con->in_base_pos += ret; + if (con->in_base_pos == sizeof(m->hdr)) { + u32 crc = crc32c_le(0, (void *)&m->hdr, + sizeof(m->hdr) - sizeof(m->hdr.crc)); + if (crc != le32_to_cpu(m->hdr.crc)) { + print_section("hdr", (u8 *)&m->hdr, sizeof(m->hdr)); + derr(0, "read_partial_message %p bad hdr crc" + " %u != expected %u\n", + m, crc, m->hdr.crc); + return -EBADMSG; + } + } + } + + /* front */ + front_len = le32_to_cpu(m->hdr.front_len); + if (front_len > CEPH_MSG_MAX_FRONT_LEN) + return -EIO; + + while (m->front.iov_len < front_len) { + if (m->front.iov_base == NULL) { + m->front.iov_base = kmalloc(front_len, GFP_NOFS); + if (m->front.iov_base == NULL) + return -ENOMEM; + } + left = front_len - m->front.iov_len; + ret = ceph_tcp_recvmsg(con->sock, (char *)m->front.iov_base + + m->front.iov_len, left); + if (ret <= 0) + return ret; + m->front.iov_len += ret; + if (m->front.iov_len == front_len) + con->in_front_crc = crc32c_le(0, m->front.iov_base, + m->front.iov_len); + } + + /* (page) data */ + data_len = le32_to_cpu(m->hdr.data_len); + if (data_len > CEPH_MSG_MAX_DATA_LEN) + return -EIO; + + data_off = le16_to_cpu(m->hdr.data_off); + if (data_len == 0) + goto no_data; + + if (m->nr_pages == 0) { + con->in_msg_pos.page = 0; + con->in_msg_pos.page_pos = data_off & ~PAGE_MASK; + con->in_msg_pos.data_pos = 0; + /* find pages for data payload */ + want = calc_pages_for(data_off & ~PAGE_MASK, data_len); + ret = 0; + BUG_ON(!con->msgr->prepare_pages); + ret = con->msgr->prepare_pages(con->msgr->parent, m, want); + if (ret < 0) { + dout(10, "prepare_pages failed, skipping payload\n"); + con->in_base_pos = -data_len - sizeof(m->footer); + ceph_msg_put(con->in_msg); + con->in_msg = NULL; + con->in_tag = CEPH_MSGR_TAG_READY; + return 0; + } + BUG_ON(m->nr_pages < want); + } + while (con->in_msg_pos.data_pos < data_len) { + left = min((int)(data_len - con->in_msg_pos.data_pos), + (int)(PAGE_SIZE - con->in_msg_pos.page_pos)); + mutex_lock(&m->page_mutex); + if (!m->pages) { + dout(10, "pages revoked during msg read\n"); + mutex_unlock(&m->page_mutex); + con->in_base_pos = con->in_msg_pos.data_pos - data_len - + sizeof(m->footer); + ceph_msg_put(m); + con->in_msg = NULL; + con->in_tag = CEPH_MSGR_TAG_READY; + return 0; + } + p = kmap(m->pages[con->in_msg_pos.page]); + ret = ceph_tcp_recvmsg(con->sock, p + con->in_msg_pos.page_pos, + left); + if (ret > 0 && datacrc) + con->in_data_crc = + crc32c_le(con->in_data_crc, + p + con->in_msg_pos.page_pos, ret); + kunmap(m->pages[con->in_msg_pos.page]); + mutex_unlock(&m->page_mutex); + if (ret <= 0) + return ret; + con->in_msg_pos.data_pos += ret; + con->in_msg_pos.page_pos += ret; + if (con->in_msg_pos.page_pos == PAGE_SIZE) { + con->in_msg_pos.page_pos = 0; + con->in_msg_pos.page++; + } + } + +no_data: + /* footer */ + to = sizeof(m->hdr) + sizeof(m->footer); + while (con->in_base_pos < to) { + left = to - con->in_base_pos; + ret = ceph_tcp_recvmsg(con->sock, (char *)&m->footer + + (con->in_base_pos - sizeof(m->hdr)), + left); + if (ret <= 0) + return ret; + con->in_base_pos += ret; + } + dout(20, "read_partial_message got msg %p\n", m); + + /* crc ok? */ + if (con->in_front_crc != le32_to_cpu(m->footer.front_crc)) { + derr(0, "read_partial_message %p front crc %u != expected %u\n", + con->in_msg, + con->in_front_crc, m->footer.front_crc); + print_section("front", (u8 *)&m->front.iov_base, sizeof(m->front.iov_len)); + return -EBADMSG; + } + if (datacrc && + (le32_to_cpu(m->footer.flags) & CEPH_MSG_FOOTER_NOCRC) == 0 && + con->in_data_crc != le32_to_cpu(m->footer.data_crc)) { + int cur_page, data_pos; + derr(0, "read_partial_message %p data crc %u != expected %u\n", + con->in_msg, + con->in_data_crc, m->footer.data_crc); + for (data_pos = 0, cur_page = 0; data_pos < data_len; data_pos += PAGE_SIZE, cur_page++) { + left = min((int)(data_len - data_pos), + (int)(PAGE_SIZE)); + mutex_lock(&m->page_mutex); + + if (!m->pages) { + derr(0, "m->pages == NULL\n"); + mutex_unlock(&m->page_mutex); + break; + } + + p = kmap(m->pages[cur_page]); + printk("data page %d len %d\n", cur_page, left); + print_section("data", p, left); + + kunmap(m->pages[0]); + mutex_unlock(&m->page_mutex); + } + return -EBADMSG; + } + + /* did i learn my ip? */ + if (con->msgr->inst.addr.ipaddr.sin_addr.s_addr == htonl(INADDR_ANY)) { + /* + * in practice, we learn our ip from the first incoming mon + * message, before anyone else knows we exist, so this is + * safe. + */ + con->msgr->inst.addr.ipaddr = con->in_msg->hdr.dst.addr.ipaddr; + dout(10, "read_partial_message learned my addr is " + "%u.%u.%u.%u:%u\n", + IPQUADPORT(con->msgr->inst.addr.ipaddr)); + } + + return 1; /* done! */ +} + +/* + * Process message. This happens in the worker thread. The callback should + * be careful not to do anything that waits on other incoming messages or it + * may deadlock. + */ +static void process_message(struct ceph_connection *con) +{ + /* if first message, set peer_name */ + if (con->peer_name.type == 0) + con->peer_name = con->in_msg->hdr.src.name; + + spin_lock(&con->out_queue_lock); + con->in_seq++; + spin_unlock(&con->out_queue_lock); + + dout(1, "===== %p %llu from %s%d %d=%s len %d+%d (%u %u) =====\n", + con->in_msg, le64_to_cpu(con->in_msg->hdr.seq), + ENTITY_NAME(con->in_msg->hdr.src.name), + le16_to_cpu(con->in_msg->hdr.type), + ceph_msg_type_name(le16_to_cpu(con->in_msg->hdr.type)), + le32_to_cpu(con->in_msg->hdr.front_len), + le32_to_cpu(con->in_msg->hdr.data_len), + con->in_front_crc, con->in_data_crc); + con->msgr->dispatch(con->msgr->parent, con->in_msg); + con->in_msg = NULL; + prepare_read_tag(con); +} + + + + + + + + +/* + * Write something to the socket. Called in a worker thread when the + * socket appears to be writeable and we have something ready to send. + */ +static int try_write(struct ceph_connection *con) +{ + struct ceph_messenger *msgr = con->msgr; + int ret = 1; + + dout(30, "try_write start %p state %lu nref %d\n", con, con->state, + atomic_read(&con->nref)); + +more: + dout(30, "try_write out_kvec_bytes %d\n", con->out_kvec_bytes); + + /* open the socket first? */ + if (con->sock == NULL) { + /* + * if we were STANDBY and are reconnecting _this_ + * connection, bump connect_seq now. Always bump + * global_seq. + */ + if (test_and_clear_bit(STANDBY, &con->state)) + con->connect_seq++; + + prepare_write_connect(msgr, con); + prepare_read_connect(con); + set_bit(CONNECTING, &con->state); + + con->in_tag = CEPH_MSGR_TAG_READY; + dout(5, "try_write initiating connect on %p new state %lu\n", + con, con->state); + con->sock = ceph_tcp_connect(con); + if (IS_ERR(con->sock)) { + con->sock = NULL; + con->error_msg = "connect error"; + ret = -1; + goto out; + } + } + +more_kvec: + /* kvec data queued? */ + if (con->out_kvec_left) { + ret = write_partial_kvec(con); + if (ret <= 0) + goto done; + if (ret < 0) { + dout(30, "try_write write_partial_kvec err %d\n", ret); + goto done; + } + } + + /* msg pages? */ + if (con->out_msg) { + ret = write_partial_msg_pages(con); + if (ret == 1) + goto more_kvec; /* we need to send the footer, too! */ + if (ret == 0) + goto done; + if (ret < 0) { + dout(30, "try_write write_partial_msg_pages err %d\n", + ret); + goto done; + } + } + + if (!test_bit(CONNECTING, &con->state)) { + /* is anything else pending? */ + spin_lock(&con->out_queue_lock); + if (!list_empty(&con->out_queue)) { + prepare_write_message(con); + spin_unlock(&con->out_queue_lock); + goto more; + } + if (con->in_seq > con->in_seq_acked) { + prepare_write_ack(con); + spin_unlock(&con->out_queue_lock); + goto more; + } + spin_unlock(&con->out_queue_lock); + } + + /* Nothing to do! */ + clear_bit(WRITE_PENDING, &con->state); + dout(30, "try_write nothing else to write.\n"); +done: + ret = 0; +out: + dout(30, "try_write done on %p\n", con); + return ret; +} + + + +/* + * Read what we can from the socket. + */ +static int try_read(struct ceph_connection *con) +{ + struct ceph_messenger *msgr; + int ret = -1; + + if (!con->sock) + return 0; + + if (test_bit(STANDBY, &con->state)) + return 0; + + dout(20, "try_read start on %p\n", con); + msgr = con->msgr; + +more: + dout(20, "try_read tag %d in_base_pos %d\n", (int)con->in_tag, + con->in_base_pos); + if (test_bit(ACCEPTING, &con->state)) { + dout(20, "try_read accepting\n"); + ret = read_partial_accept(con); + if (ret <= 0) + goto done; + if (process_accept(con) < 0) { + ret = -1; + goto out; + } + goto more; + } + if (test_bit(CONNECTING, &con->state)) { + dout(20, "try_read connecting\n"); + ret = read_partial_connect(con); + if (ret <= 0) + goto done; + if (process_connect(con) < 0) { + ret = -1; + goto out; + } + goto more; + } + + if (con->in_base_pos < 0) { + /* + * skipping + discarding content. + * + * FIXME: there must be a better way to do this! + */ + static char buf[1024]; + int skip = min(1024, -con->in_base_pos); + dout(20, "skipping %d / %d bytes\n", skip, -con->in_base_pos); + ret = ceph_tcp_recvmsg(con->sock, buf, skip); + if (ret <= 0) + goto done; + con->in_base_pos += ret; + if (con->in_base_pos) + goto more; + } + if (con->in_tag == CEPH_MSGR_TAG_READY) { + /* + * what's next? + */ + ret = ceph_tcp_recvmsg(con->sock, &con->in_tag, 1); + if (ret <= 0) + goto done; + dout(30, "try_read got tag %d\n", (int)con->in_tag); + switch (con->in_tag) { + case CEPH_MSGR_TAG_MSG: + prepare_read_message(con); + break; + case CEPH_MSGR_TAG_ACK: + prepare_read_ack(con); + break; + case CEPH_MSGR_TAG_CLOSE: + set_bit(CLOSED, &con->state); /* fixme */ + goto done; + default: + goto bad_tag; + } + } + if (con->in_tag == CEPH_MSGR_TAG_MSG) { + ret = read_partial_message(con); + if (ret <= 0) { + switch (ret) { + case -EBADMSG: + con->error_msg = "bad crc"; + ret = -EIO; + goto out; + case -EIO: + con->error_msg = "io error"; + goto out; + default: + goto done; + } + } + if (con->in_tag == CEPH_MSGR_TAG_READY) + goto more; + process_message(con); + goto more; + } + if (con->in_tag == CEPH_MSGR_TAG_ACK) { + ret = read_partial_ack(con); + if (ret <= 0) + goto done; + process_ack(con); + goto more; + } + +done: + ret = 0; +out: + dout(20, "try_read done on %p\n", con); + return ret; + +bad_tag: + derr(2, "try_read bad con->in_tag = %d\n", (int)con->in_tag); + con->error_msg = "protocol error, garbage tag"; + ret = -1; + goto out; +} + + +/* + * Atomically queue work on a connection. Bump @con reference to + * avoid races with connection teardown. + * + * There is some trickery going on with QUEUED and BUSY because we + * only want a _single_ thread operating on each connection at any + * point in time, but we want to use all available CPUs. + * + * The worker thread only proceeds if it can atomically set BUSY. It + * clears QUEUED and does it's thing. When it thinks it's done, it + * clears BUSY, then rechecks QUEUED.. if it's set again, it loops + * (tries again to set BUSY). + * + * To queue work, we first set QUEUED, _then_ if BUSY isn't set, we + * try to queue work. If that fails (work is already queued, or BUSY) + * we give up (work also already being done or is queued) but leave QUEUED + * set so that the worker thread will loop if necessary. + */ +static void ceph_queue_con(struct ceph_connection *con) +{ + if (test_bit(WAIT, &con->state) || + test_bit(CLOSED, &con->state)) { + dout(40, "ceph_queue_con %p ignoring: WAIT|CLOSED\n", + con); + return; + } + + atomic_inc(&con->nref); + dout(40, "ceph_queue_con %p %d -> %d\n", con, + atomic_read(&con->nref) - 1, atomic_read(&con->nref)); + + set_bit(QUEUED, &con->state); + if (test_bit(BUSY, &con->state) || + !queue_work(ceph_msgr_wq, &con->work.work)) { + dout(40, "ceph_queue_con %p - already BUSY or queued\n", con); + put_connection(con); + } +} + +/* + * Do some work on a connection. Drop a connection ref when we're done. + */ +static void con_work(struct work_struct *work) +{ + struct ceph_connection *con = container_of(work, struct ceph_connection, + work.work); + int backoff = 0; + +more: + if (test_and_set_bit(BUSY, &con->state) != 0) { + dout(10, "con_work %p BUSY already set\n", con); + goto out; + } + dout(10, "con_work %p start, clearing QUEUED\n", con); + clear_bit(QUEUED, &con->state); + + if (test_bit(CLOSED, &con->state)) { /* e.g. if we are replaced */ + dout(5, "con_work CLOSED\n"); + goto done; + } + if (test_bit(WAIT, &con->state)) { /* we are a zombie */ + dout(5, "con_work WAIT\n"); + goto done; + } + + if (test_and_clear_bit(SOCK_CLOSED, &con->state) || + try_read(con) < 0 || + try_write(con) < 0) { + backoff = 1; + ceph_fault(con); /* error/fault path */ + } + +done: + clear_bit(BUSY, &con->state); + dout(10, "con->state=%lu\n", con->state); + if (test_bit(QUEUED, &con->state)) { + if (!backoff) { + dout(10, "con_work %p QUEUED reset, looping\n", con); + goto more; + } + dout(10, "con_work %p QUEUED reset, but just faulted\n", con); + clear_bit(QUEUED, &con->state); + } + dout(10, "con_work %p done\n", con); + +out: + put_connection(con); +} + + +/* + * Generic error/fault handler. A retry mechanism is used with + * exponential backoff + */ +static void ceph_fault(struct ceph_connection *con) +{ + derr(1, "%s%d %u.%u.%u.%u:%u %s\n", ENTITY_NAME(con->peer_name), + IPQUADPORT(con->peer_addr.ipaddr), con->error_msg); + dout(10, "fault %p state %lu to peer %u.%u.%u.%u:%u\n", + con, con->state, IPQUADPORT(con->peer_addr.ipaddr)); + + if (test_bit(LOSSYTX, &con->state)) { + dout(30, "fault on LOSSYTX channel\n"); + remove_connection(con->msgr, con); + return; + } + + clear_bit(BUSY, &con->state); /* to avoid an improbable race */ + + con_close_socket(con); + con->in_msg = NULL; + + /* If there are no messages in the queue, place the connection + * in a STANDBY state (i.e., don't try to reconnect just yet). */ + spin_lock(&con->out_queue_lock); + if (list_empty(&con->out_queue)) { + dout(10, "fault setting STANDBY\n"); + set_bit(STANDBY, &con->state); + spin_unlock(&con->out_queue_lock); + return; + } + + /* Requeue anything that hasn't been acked, and retry after a + * delay. */ + list_splice_init(&con->out_sent, &con->out_queue); + spin_unlock(&con->out_queue_lock); + + if (con->delay == 0) + con->delay = BASE_DELAY_INTERVAL; + else if (con->delay < MAX_DELAY_INTERVAL) + con->delay *= 2; + + /* explicitly schedule work to try to reconnect again later. */ + dout(40, "fault queueing %p %d -> %d delay %lu\n", con, + atomic_read(&con->nref), atomic_read(&con->nref) + 1, + con->delay); + atomic_inc(&con->nref); + queue_delayed_work(ceph_msgr_wq, &con->work, + round_jiffies_relative(con->delay)); +} + + +/* + * Handle an incoming connection. + */ +static void accept_work(struct work_struct *work) +{ + struct ceph_connection *newcon = NULL; + struct ceph_messenger *msgr = container_of(work, struct ceph_messenger, + awork); + + /* initialize the msgr connection */ + newcon = new_connection(msgr); + if (newcon == NULL) { + derr(1, "kmalloc failure accepting new connection\n"); + return; + } + + set_bit(ACCEPTING, &newcon->state); + newcon->connect_seq = 1; + newcon->in_tag = CEPH_MSGR_TAG_READY; /* eventually, hopefully */ + + if (ceph_tcp_accept(msgr->listen_sock, newcon) < 0) { + derr(1, "error accepting connection\n"); + put_connection(newcon); + return; + } + dout(5, "accepted connection \n"); + + prepare_write_accept_hello(msgr, newcon); + add_connection_accepting(msgr, newcon); + + /* queue work explicitly; we may have missed the socket state + * change before setting the socket callbacks. */ + ceph_queue_con(newcon); +} + + + +/* + * create a new messenger instance, creates listening socket + */ +struct ceph_messenger *ceph_messenger_create(struct ceph_entity_addr *myaddr) +{ + struct ceph_messenger *msgr; + int ret = 0; + + msgr = kzalloc(sizeof(*msgr), GFP_KERNEL); + if (msgr == NULL) + return ERR_PTR(-ENOMEM); + + INIT_WORK(&msgr->awork, accept_work); + spin_lock_init(&msgr->con_lock); + INIT_LIST_HEAD(&msgr->con_all); + INIT_LIST_HEAD(&msgr->con_accepting); + INIT_RADIX_TREE(&msgr->con_tree, GFP_ATOMIC); + spin_lock_init(&msgr->global_seq_lock); + + /* the zero page is needed if a request is "canceled" while the message + * is being written over the socket */ + msgr->zero_page = alloc_page(GFP_KERNEL | __GFP_ZERO); + if (!msgr->zero_page) { + kfree(msgr); + return ERR_PTR(-ENOMEM); + } + kmap(msgr->zero_page); + + /* pick listening address */ + if (myaddr) { + msgr->inst.addr = *myaddr; + } else { + dout(10, "create my ip not specified, binding to INADDR_ANY\n"); + msgr->inst.addr.ipaddr.sin_addr.s_addr = htonl(INADDR_ANY); + msgr->inst.addr.ipaddr.sin_port = htons(0); /* any port */ + } + msgr->inst.addr.ipaddr.sin_family = AF_INET; + + /* create listening socket */ + ret = ceph_tcp_listen(msgr); + if (ret < 0) { + kfree(msgr); + return ERR_PTR(ret); + } + if (myaddr) + msgr->inst.addr.ipaddr.sin_addr = myaddr->ipaddr.sin_addr; + + dout(1, "messenger %p listening on %u.%u.%u.%u:%u\n", msgr, + IPQUADPORT(msgr->inst.addr.ipaddr)); + return msgr; +} + +void ceph_messenger_destroy(struct ceph_messenger *msgr) +{ + struct ceph_connection *con; + + dout(2, "destroy %p\n", msgr); + + /* stop listener */ + msgr->listen_sock->ops->shutdown(msgr->listen_sock, SHUT_RDWR); + sock_release(msgr->listen_sock); + cancel_work_sync(&msgr->awork); + + /* kill off connections */ + spin_lock(&msgr->con_lock); + while (!list_empty(&msgr->con_all)) { + con = list_entry(msgr->con_all.next, struct ceph_connection, + list_all); + dout(10, "destroy removing connection %p\n", con); + set_bit(CLOSED, &con->state); + atomic_inc(&con->nref); + dout(40, " get %p %d -> %d\n", con, + atomic_read(&con->nref) - 1, atomic_read(&con->nref)); + __remove_connection(msgr, con); + + /* in case there's queued work. drop a reference if + * we successfully cancel work. */ + spin_unlock(&msgr->con_lock); + if (cancel_delayed_work_sync(&con->work)) + put_connection(con); + put_connection(con); + dout(10, "destroy removed connection %p\n", con); + + spin_lock(&msgr->con_lock); + } + spin_unlock(&msgr->con_lock); + + kunmap(msgr->zero_page); + __free_page(msgr->zero_page); + + kfree(msgr); + dout(10, "destroyed messenger %p\n", msgr); +} + +/* + * mark a peer down. drop any open connections. + */ +void ceph_messenger_mark_down(struct ceph_messenger *msgr, + struct ceph_entity_addr *addr) +{ + struct ceph_connection *con; + + dout(2, "mark_down peer %u.%u.%u.%u:%u\n", + IPQUADPORT(addr->ipaddr)); + + spin_lock(&msgr->con_lock); + con = __get_connection(msgr, addr); + if (con) { + dout(1, "mark_down %s%d %u.%u.%u.%u:%u (%p)\n", + ENTITY_NAME(con->peer_name), + IPQUADPORT(con->peer_addr.ipaddr), con); + set_bit(CLOSED, &con->state); /* in case there's queued work */ + __remove_connection(msgr, con); + } + spin_unlock(&msgr->con_lock); + if (con) + put_connection(con); +} + + +/* + * A single ceph_msg can't be queued for send twice, unless it's + * already been delivered (i.e. we have the only remaining reference), + * because of the list_head indicating which queue it is on. + * + * So, we dup the message if there is more than once reference. If it has + * pages (a data payload), steal the pages away from the old message. + */ +struct ceph_msg *ceph_msg_maybe_dup(struct ceph_msg *old) +{ + struct ceph_msg *dup; + + if (atomic_read(&old->nref) == 1) + return old; /* we have only ref, all is well */ + + dup = ceph_msg_new(le16_to_cpu(old->hdr.type), + le32_to_cpu(old->hdr.front_len), + le32_to_cpu(old->hdr.data_len), + le16_to_cpu(old->hdr.data_off), + old->pages); + if (!dup) + return ERR_PTR(-ENOMEM); + memcpy(dup->front.iov_base, old->front.iov_base, + le32_to_cpu(old->hdr.front_len)); + + /* revoke old message's pages */ + mutex_lock(&old->page_mutex); + old->pages = NULL; + old->footer.flags |= cpu_to_le32(CEPH_MSG_FOOTER_ABORTED); + mutex_unlock(&old->page_mutex); + + ceph_msg_put(old); + return dup; +} + + +/* + * Queue up an outgoing message. + * + * This consumes a msg reference. That is, if the caller wants to + * keep @msg around, it had better call ceph_msg_get first. + */ +int ceph_msg_send(struct ceph_messenger *msgr, struct ceph_msg *msg, + unsigned long timeout) +{ + struct ceph_connection *con, *newcon; + int ret = 0; + + /* set source */ + msg->hdr.src = msgr->inst; + msg->hdr.orig_src = msgr->inst; + + /* do we have the connection? */ + spin_lock(&msgr->con_lock); + con = __get_connection(msgr, &msg->hdr.dst.addr); + if (!con) { + /* drop lock while we allocate a new connection */ + spin_unlock(&msgr->con_lock); + + newcon = new_connection(msgr); + if (IS_ERR(newcon)) + return PTR_ERR(con); + + newcon->out_connect.flags = 0; + if (!timeout) { + dout(10, "ceph_msg_send setting LOSSYTX\n"); + newcon->out_connect.flags |= CEPH_MSG_CONNECT_LOSSY; + set_bit(LOSSYTX, &newcon->state); + } + + ret = radix_tree_preload(GFP_NOFS); + if (ret < 0) { + derr(10, "ENOMEM in ceph_msg_send\n"); + return ret; + } + + spin_lock(&msgr->con_lock); + con = __get_connection(msgr, &msg->hdr.dst.addr); + if (con) { + put_connection(newcon); + dout(10, "ceph_msg_send (lost race and) had connection " + "%p to peer %u.%u.%u.%u:%u\n", con, + IPQUADPORT(msg->hdr.dst.addr.ipaddr)); + } else { + con = newcon; + con->peer_addr = msg->hdr.dst.addr; + con->peer_name = msg->hdr.dst.name; + __register_connection(msgr, con); + dout(5, "ceph_msg_send new connection %p to peer " + "%u.%u.%u.%u:%u\n", con, + IPQUADPORT(msg->hdr.dst.addr.ipaddr)); + } + spin_unlock(&msgr->con_lock); + radix_tree_preload_end(); + } else { + dout(10, "ceph_msg_send had connection %p to peer " + "%u.%u.%u.%u:%u con->sock=%p\n", con, + IPQUADPORT(msg->hdr.dst.addr.ipaddr), con->sock); + spin_unlock(&msgr->con_lock); + } + + /* queue */ + spin_lock(&con->out_queue_lock); + + /* avoid queuing multiple PING messages in a row. */ + if (unlikely(le16_to_cpu(msg->hdr.type) == CEPH_MSG_PING && + !list_empty(&con->out_queue) && + le16_to_cpu(list_entry(con->out_queue.prev, + struct ceph_msg, + list_head)->hdr.type) == CEPH_MSG_PING)) { + dout(2, "ceph_msg_send dropping dup ping\n"); + ceph_msg_put(msg); + } else { + msg->hdr.seq = cpu_to_le64(++con->out_seq); + dout(1, "----- %p %u to %s%d %d=%s len %d+%d -----\n", msg, + (unsigned)con->out_seq, + ENTITY_NAME(msg->hdr.dst.name), le16_to_cpu(msg->hdr.type), + ceph_msg_type_name(le16_to_cpu(msg->hdr.type)), + le32_to_cpu(msg->hdr.front_len), + le32_to_cpu(msg->hdr.data_len)); + dout(2, "ceph_msg_send %p seq %llu for %s%d on %p pgs %d\n", + msg, le64_to_cpu(msg->hdr.seq), + ENTITY_NAME(msg->hdr.dst.name), con, msg->nr_pages); + list_add_tail(&msg->list_head, &con->out_queue); + } + spin_unlock(&con->out_queue_lock); + + /* if there wasn't anything waiting to send before, queue + * new work */ + if (test_and_set_bit(WRITE_PENDING, &con->state) == 0) + ceph_queue_con(con); + + put_connection(con); + dout(30, "ceph_msg_send done\n"); + return ret; +} + + +/* + * construct a new message with given type, size + * the new msg has a ref count of 1. + */ +struct ceph_msg *ceph_msg_new(int type, int front_len, + int page_len, int page_off, struct page **pages) +{ + struct ceph_msg *m; + + m = kmalloc(sizeof(*m), GFP_NOFS); + if (m == NULL) + goto out; + atomic_set(&m->nref, 1); + mutex_init(&m->page_mutex); + INIT_LIST_HEAD(&m->list_head); + + m->hdr.type = cpu_to_le16(type); + m->hdr.front_len = cpu_to_le32(front_len); + m->hdr.data_len = cpu_to_le32(page_len); + m->hdr.data_off = cpu_to_le16(page_off); + m->hdr.priority = 0; + m->hdr.mon_protocol = CEPH_MON_PROTOCOL; + m->hdr.monc_protocol = CEPH_MONC_PROTOCOL; + m->hdr.osd_protocol = CEPH_OSD_PROTOCOL; + m->hdr.osdc_protocol = CEPH_OSDC_PROTOCOL; + m->hdr.mds_protocol = CEPH_MDS_PROTOCOL; + m->hdr.mdsc_protocol = CEPH_MDSC_PROTOCOL; + m->footer.front_crc = 0; + m->footer.data_crc = 0; + m->front_is_vmalloc = false; + + /* front */ + if (front_len) { + if (front_len > PAGE_CACHE_SIZE) { + m->front.iov_base = vmalloc(front_len); + m->front_is_vmalloc = true; + } else { + m->front.iov_base = kmalloc(front_len, GFP_NOFS); + } + if (m->front.iov_base == NULL) { + derr(0, "ceph_msg_new can't allocate %d bytes\n", + front_len); + goto out2; + } + } else { + m->front.iov_base = NULL; + } + m->front.iov_len = front_len; + + /* pages */ + m->nr_pages = calc_pages_for(page_off, page_len); + m->pages = pages; + + dout(20, "ceph_msg_new %p page %d~%d -> %d\n", m, page_off, page_len, + m->nr_pages); + return m; + +out2: + ceph_msg_put(m); +out: + derr(0, "msg_new can't create msg type %d len %d\n", type, front_len); + return ERR_PTR(-ENOMEM); +} + +void ceph_msg_put(struct ceph_msg *m) +{ + dout(20, "ceph_msg_put %p %d -> %d\n", m, atomic_read(&m->nref), + atomic_read(&m->nref)-1); + if (atomic_read(&m->nref) <= 0) { + derr(0, "bad ceph_msg_put on %p %llu %s%d->%s%d %d=%s %d+%d\n", + m, le64_to_cpu(m->hdr.seq), + ENTITY_NAME(m->hdr.src.name), + ENTITY_NAME(m->hdr.dst.name), + le16_to_cpu(m->hdr.type), + ceph_msg_type_name(le16_to_cpu(m->hdr.type)), + le32_to_cpu(m->hdr.front_len), + le32_to_cpu(m->hdr.data_len)); + WARN_ON(1); + } + if (atomic_dec_and_test(&m->nref)) { + dout(20, "ceph_msg_put last one on %p\n", m); + WARN_ON(!list_empty(&m->list_head)); + if (m->front_is_vmalloc) + vfree(m->front.iov_base); + else + kfree(m->front.iov_base); + kfree(m); + } +} + +void ceph_ping(struct ceph_messenger *msgr, struct ceph_entity_name name, + struct ceph_entity_addr *addr) +{ + struct ceph_msg *m; + + m = ceph_msg_new(CEPH_MSG_PING, 0, 0, 0, NULL); + if (!m) + return; + memset(m->front.iov_base, 0, m->front.iov_len); + m->hdr.dst.name = name; + m->hdr.dst.addr = *addr; + ceph_msg_send(msgr, m, BASE_DELAY_INTERVAL); +} diff --git a/fs/ceph/messenger.h b/fs/ceph/messenger.h new file mode 100644 index 0000000..a13a6d1 --- /dev/null +++ b/fs/ceph/messenger.h @@ -0,0 +1,267 @@ +#ifndef __FS_CEPH_MESSENGER_H +#define __FS_CEPH_MESSENGER_H + +#include <linux/kobject.h> +#include <linux/mutex.h> +#include <linux/net.h> +#include <linux/radix-tree.h> +#include <linux/uio.h> +#include <linux/version.h> +#include <linux/workqueue.h> + +#include "types.h" + +/* + * Ceph uses the messenger to exchange ceph_msg messages with + * other hosts in the system. The messenger provides ordered and + * reliable delivery. It tolerates TCP disconnects by reconnecting + * (with exponential backoff) in the case of a fault (disconnection, + * bad crc, protocol error). Acks allow sent messages to be discarded + * by the sender. + * + * The network topology is flat: there is no "client" or "server," and + * any node can initiate a connection (i.e., send messages) to any other + * node. There is a fair bit of complexity to handle the "connection + * race" case where two nodes are simultaneously connecting to each other + * so that the end result is a single session. + * + * The messenger can also send messages in "lossy" mode, where there is + * no error recovery or connect retry... the message is just dropped if + * something goes wrong. + */ + +struct ceph_msg; + +#define IPQUADPORT(n) \ + (unsigned int)((be32_to_cpu((n).sin_addr.s_addr) >> 24)) & 0xFF, \ + (unsigned int)((be32_to_cpu((n).sin_addr.s_addr)) >> 16) & 0xFF, \ + (unsigned int)((be32_to_cpu((n).sin_addr.s_addr))>>8) & 0xFF, \ + (unsigned int)((be32_to_cpu((n).sin_addr.s_addr))) & 0xFF, \ + (unsigned int)(ntohs((n).sin_port)) + + +extern struct workqueue_struct *ceph_msgr_wq; /* receive work queue */ + +/* + * Ceph defines these callbacks for handling events: + */ +/* handle an incoming message. */ +typedef void (*ceph_msgr_dispatch_t) (void *p, struct ceph_msg *m); +/* an incoming message has a data payload; tell me what pages I + * should read the data into. */ +typedef int (*ceph_msgr_prepare_pages_t) (void *p, struct ceph_msg *m, + int want); +/* a remote host as terminated a message exchange session, and messages + * we sent (or they tried to send us) may be lost. */ +typedef void (*ceph_msgr_peer_reset_t) (void *p, struct ceph_entity_addr *addr, + struct ceph_entity_name *pn); + +static inline const char *ceph_name_type_str(int t) +{ + switch (t) { + case CEPH_ENTITY_TYPE_MON: return "mon"; + case CEPH_ENTITY_TYPE_MDS: return "mds"; + case CEPH_ENTITY_TYPE_OSD: return "osd"; + case CEPH_ENTITY_TYPE_CLIENT: return "client"; + case CEPH_ENTITY_TYPE_ADMIN: return "admin"; + default: return "???"; + } +} + +#define CEPH_MSGR_BACKUP 10 /* backlogged incoming connections */ + +/* use format string %s%d */ +#define ENTITY_NAME(n) \ + ceph_name_type_str(le32_to_cpu((n).type)), \ + le32_to_cpu((n).num) + +struct ceph_messenger { + void *parent; /* normally struct ceph_client * */ + ceph_msgr_dispatch_t dispatch; + ceph_msgr_peer_reset_t peer_reset; + ceph_msgr_prepare_pages_t prepare_pages; + + struct ceph_entity_inst inst; /* my name+address */ + + struct socket *listen_sock; /* listening socket */ + struct work_struct awork; /* accept work */ + + spinlock_t con_lock; + struct list_head con_all; /* all open connections */ + struct list_head con_accepting; /* accepting */ + struct radix_tree_root con_tree; /* established */ + + struct page *zero_page; /* used in certain error cases */ + + /* + * the global_seq counts connections i (attempt to) initiate + * in order to disambiguate certain connect race conditions. + */ + u32 global_seq; + spinlock_t global_seq_lock; +}; + +/* + * a single message. it contains a header (src, dest, message type, etc.), + * footer (crc values, mainly), a "front" message body, and possibly a + * data payload (stored in some number of pages). The page_mutex protects + * access to the page vector. + */ +struct ceph_msg { + struct ceph_msg_header hdr; /* header */ + struct ceph_msg_footer footer; /* footer */ + struct kvec front; /* first bit of message */ + struct mutex page_mutex; + struct page **pages; /* data payload. NOT OWNER. */ + unsigned nr_pages; /* size of page array */ + struct list_head list_head; + atomic_t nref; + bool front_is_vmalloc; +}; + +struct ceph_msg_pos { + int page, page_pos; /* which page; offset in page */ + int data_pos; /* offset in data payload */ + int did_page_crc; /* true if we've calculated crc for current page */ +}; + +/* ceph connection fault delay defaults */ +#define BASE_DELAY_INTERVAL (HZ/2) +#define MAX_DELAY_INTERVAL (5 * 60 * HZ) + +/* + * ceph_connection state bit flags + * + * QUEUED and BUSY are used together to ensure that only a single + * thread is currently opening, reading or writing data to the socket. + */ +#define LOSSYTX 0 /* we can close channel or drop messages on errors */ +#define LOSSYRX 1 /* peer may reset/drop messages */ +#define CONNECTING 2 +#define ACCEPTING 3 +#define WRITE_PENDING 4 /* we have data ready to send */ +#define QUEUED 5 /* there is work queued on this connection */ +#define BUSY 6 /* work is being done */ +#define STANDBY 8 /* no outgoing messages, socket closed. we keep + * the ceph_connection around to maintain shared + * state with the peer. */ +#define WAIT 9 /* waiting for peer to connect to us (during a + * connection race) */ +#define CLOSED 10 /* we've closed the connection */ +#define SOCK_CLOSED 11 /* socket state changed to closed */ +#define REGISTERED 12 /* connection appears in con_tree */ + +/* + * A single connection with another host. + * + * We maintain a queue of outgoing messages, and some session state to + * ensure that we can preserve the lossless, ordered delivery of + * messages in the case of a TCP disconnect. + */ +struct ceph_connection { + struct ceph_messenger *msgr; + struct socket *sock; + unsigned long state; /* connection state (see flags above) */ + const char *error_msg; /* error message, if any */ + + atomic_t nref; + + struct list_head list_all; /* msgr->con_all */ + struct list_head list_bucket; /* msgr->con_tree or con_accepting */ + + struct ceph_entity_addr peer_addr; /* peer address */ + struct ceph_entity_name peer_name; /* peer name */ + u32 connect_seq; /* identify the most recent connection + attempt for this connection, client */ + u32 peer_global_seq; /* peer's global seq for this connection */ + + /* out queue */ + spinlock_t out_queue_lock; /* protects out_queue, out_sent, out_seq */ + struct list_head out_queue; + struct list_head out_sent; /* sending/sent but unacked */ + u32 out_seq; /* last message queued for send */ + + u32 in_seq, in_seq_acked; /* last message received, acked */ + + /* connection negotiation temps */ + char in_banner[CEPH_BANNER_MAX_LEN]; + union { + struct { /* outgoing connection */ + struct ceph_msg_connect out_connect; + struct ceph_msg_connect_reply in_reply; + }; + struct { /* incoming */ + struct ceph_msg_connect in_connect; + struct ceph_msg_connect_reply out_reply; + }; + }; + struct ceph_entity_addr actual_peer_addr; + + /* message out temps */ + struct ceph_msg *out_msg; /* sending message (== tail of + out_sent) */ + struct ceph_msg_pos out_msg_pos; + + struct kvec out_kvec[6], /* sending header/footer data */ + *out_kvec_cur; + int out_kvec_left; /* kvec's left in out_kvec */ + int out_kvec_bytes; /* total bytes left */ + int out_more; /* there is more data after the kvecs */ + __le32 out_temp_ack; /* for writing an ack */ + + /* message in temps */ + struct ceph_msg *in_msg; + struct ceph_msg_pos in_msg_pos; + u32 in_front_crc, in_data_crc; /* calculated crc, for comparison + message footer */ + + char in_tag; /* protocol control byte */ + int in_base_pos; /* bytes read */ + __le32 in_temp_ack; /* for reading an ack */ + + struct delayed_work work; /* send|recv work */ + unsigned long delay; /* current delay interval */ +}; + +extern int ceph_msgr_init(void); +extern void ceph_msgr_exit(void); + +extern struct ceph_messenger * +ceph_messenger_create(struct ceph_entity_addr *myaddr); +extern void ceph_messenger_destroy(struct ceph_messenger *); +extern void ceph_messenger_mark_down(struct ceph_messenger *msgr, + struct ceph_entity_addr *addr); + +extern struct ceph_msg *ceph_msg_new(int type, int front_len, + int page_len, int page_off, + struct page **pages); + +static inline struct ceph_msg *ceph_msg_get(struct ceph_msg *msg) +{ + /*printk("ceph_msg_get %p %d -> %d\n", msg, atomic_read(&msg->nref), + atomic_read(&msg->nref)+1);*/ + atomic_inc(&msg->nref); + return msg; +} + +extern void ceph_msg_put(struct ceph_msg *msg); + +static inline void ceph_msg_put_list(struct list_head *head) +{ + while (!list_empty(head)) { + struct ceph_msg *msg = list_first_entry(head, struct ceph_msg, + list_head); + list_del_init(&msg->list_head); + ceph_msg_put(msg); + } +} + +extern struct ceph_msg *ceph_msg_maybe_dup(struct ceph_msg *msg); + +extern int ceph_msg_send(struct ceph_messenger *msgr, struct ceph_msg *msg, + unsigned long timeout); + +extern void ceph_ping(struct ceph_messenger *msgr, struct ceph_entity_name name, + struct ceph_entity_addr *addr); + +#endif -- 1.5.6.5 -- To unsubscribe from this list: send the line "unsubscribe linux-fsdevel" in the body of a message to majordomo@xxxxxxxxxxxxxxx More majordomo info at http://vger.kernel.org/majordomo-info.html