]> git.proxmox.com Git - mirror_ubuntu-artful-kernel.git/blobdiff - net/ceph/messenger.c
libceph: use local variables for message positions
[mirror_ubuntu-artful-kernel.git] / net / ceph / messenger.c
index fb5f6e7d57a3b0e247abca0657e26a5aa76cd0db..7788170524e3c410741163cc08159c74f92d8dc9 100644 (file)
@@ -725,22 +725,23 @@ static void iter_bio_next(struct bio **bio_iter, unsigned int *seg)
 static void prepare_write_message_data(struct ceph_connection *con)
 {
        struct ceph_msg *msg = con->out_msg;
+       struct ceph_msg_pos *msg_pos = &con->out_msg_pos;
 
        BUG_ON(!msg);
        BUG_ON(!msg->hdr.data_len);
 
        /* initialize page iterator */
-       con->out_msg_pos.page = 0;
+       msg_pos->page = 0;
        if (msg->pages)
-               con->out_msg_pos.page_pos = msg->page_alignment;
+               msg_pos->page_pos = msg->page_alignment;
        else
-               con->out_msg_pos.page_pos = 0;
+               msg_pos->page_pos = 0;
 #ifdef CONFIG_BLOCK
        if (msg->bio)
                init_bio_iter(msg->bio, &msg->bio_iter, &msg->bio_seg);
 #endif
-       con->out_msg_pos.data_pos = 0;
-       con->out_msg_pos.did_page_crc = false;
+       msg_pos->data_pos = 0;
+       msg_pos->did_page_crc = false;
        con->out_more = 1;  /* data + footer will follow */
 }
 
@@ -804,16 +805,11 @@ static void prepare_write_message(struct ceph_connection *con)
                m->hdr.seq = cpu_to_le64(++con->out_seq);
                m->needs_out_seq = false;
        }
-#ifdef CONFIG_BLOCK
-       else
-               m->bio_iter = NULL;
-#endif
 
-       dout("prepare_write_message %p seq %lld type %d len %d+%d+%d %d pgs\n",
+       dout("prepare_write_message %p seq %lld type %d len %d+%d+%d (%zd)\n",
             m, con->out_seq, le16_to_cpu(m->hdr.type),
             le32_to_cpu(m->hdr.front_len), le32_to_cpu(m->hdr.middle_len),
-            le32_to_cpu(m->hdr.data_len),
-            m->page_count);
+            le32_to_cpu(m->hdr.data_len), m->length);
        BUG_ON(le32_to_cpu(m->hdr.front_len) != m->front.iov_len);
 
        /* tag + hdr + front + middle */
@@ -1027,25 +1023,24 @@ static void out_msg_pos_next(struct ceph_connection *con, struct page *page,
                        size_t len, size_t sent, bool in_trail)
 {
        struct ceph_msg *msg = con->out_msg;
+       struct ceph_msg_pos *msg_pos = &con->out_msg_pos;
 
        BUG_ON(!msg);
        BUG_ON(!sent);
 
-       con->out_msg_pos.data_pos += sent;
-       con->out_msg_pos.page_pos += sent;
+       msg_pos->data_pos += sent;
+       msg_pos->page_pos += sent;
        if (sent < len)
                return;
 
        BUG_ON(sent != len);
-       con->out_msg_pos.page_pos = 0;
-       con->out_msg_pos.page++;
-       con->out_msg_pos.did_page_crc = false;
+       msg_pos->page_pos = 0;
+       msg_pos->page++;
+       msg_pos->did_page_crc = false;
        if (in_trail)
-               list_move_tail(&page->lru,
-                              &msg->trail->head);
+               list_rotate_left(&msg->trail->head);
        else if (msg->pagelist)
-               list_move_tail(&page->lru,
-                              &msg->pagelist->head);
+               list_rotate_left(&msg->pagelist->head);
 #ifdef CONFIG_BLOCK
        else if (msg->bio)
                iter_bio_next(&msg->bio_iter, &msg->bio_seg);
@@ -1056,18 +1051,19 @@ static void in_msg_pos_next(struct ceph_connection *con, size_t len,
                                size_t received)
 {
        struct ceph_msg *msg = con->in_msg;
+       struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
 
        BUG_ON(!msg);
        BUG_ON(!received);
 
-       con->in_msg_pos.data_pos += received;
-       con->in_msg_pos.page_pos += received;
+       msg_pos->data_pos += received;
+       msg_pos->page_pos += received;
        if (received < len)
                return;
 
        BUG_ON(received != len);
-       con->in_msg_pos.page_pos = 0;
-       con->in_msg_pos.page++;
+       msg_pos->page_pos = 0;
+       msg_pos->page++;
 #ifdef CONFIG_BLOCK
        if (msg->bio)
                iter_bio_next(&msg->bio_iter, &msg->bio_seg);
@@ -1084,6 +1080,7 @@ static void in_msg_pos_next(struct ceph_connection *con, size_t len,
 static int write_partial_msg_pages(struct ceph_connection *con)
 {
        struct ceph_msg *msg = con->out_msg;
+       struct ceph_msg_pos *msg_pos = &con->out_msg_pos;
        unsigned int data_len = le32_to_cpu(msg->hdr.data_len);
        size_t len;
        bool do_datacrc = !con->msgr->nocrc;
@@ -1093,9 +1090,8 @@ static int write_partial_msg_pages(struct ceph_connection *con)
        const size_t trail_len = (msg->trail ? msg->trail->length : 0);
        const size_t trail_off = data_len - trail_len;
 
-       dout("write_partial_msg_pages %p msg %p page %d/%d offset %d\n",
-            con, msg, con->out_msg_pos.page, msg->page_count,
-            con->out_msg_pos.page_pos);
+       dout("write_partial_msg_pages %p msg %p page %d offset %d\n",
+            con, msg, msg_pos->page, msg_pos->page_pos);
 
        /*
         * Iterate through each page that contains data to be
@@ -1105,22 +1101,22 @@ static int write_partial_msg_pages(struct ceph_connection *con)
         * need to map the page.  If we have no pages, they have
         * been revoked, so use the zero page.
         */
-       while (data_len > con->out_msg_pos.data_pos) {
+       while (data_len > msg_pos->data_pos) {
                struct page *page = NULL;
                int max_write = PAGE_SIZE;
                int bio_offset = 0;
 
-               in_trail = in_trail || con->out_msg_pos.data_pos >= trail_off;
+               in_trail = in_trail || msg_pos->data_pos >= trail_off;
                if (!in_trail)
-                       total_max_write = trail_off - con->out_msg_pos.data_pos;
+                       total_max_write = trail_off - msg_pos->data_pos;
 
                if (in_trail) {
-                       total_max_write = data_len - con->out_msg_pos.data_pos;
+                       total_max_write = data_len - msg_pos->data_pos;
 
                        page = list_first_entry(&msg->trail->head,
                                                struct page, lru);
                } else if (msg->pages) {
-                       page = msg->pages[con->out_msg_pos.page];
+                       page = msg->pages[msg_pos->page];
                } else if (msg->pagelist) {
                        page = list_first_entry(&msg->pagelist->head,
                                                struct page, lru);
@@ -1136,24 +1132,24 @@ static int write_partial_msg_pages(struct ceph_connection *con)
                } else {
                        page = zero_page;
                }
-               len = min_t(int, max_write - con->out_msg_pos.page_pos,
+               len = min_t(int, max_write - msg_pos->page_pos,
                            total_max_write);
 
-               if (do_datacrc && !con->out_msg_pos.did_page_crc) {
+               if (do_datacrc && !msg_pos->did_page_crc) {
                        void *base;
                        u32 crc = le32_to_cpu(msg->footer.data_crc);
                        char *kaddr;
 
                        kaddr = kmap(page);
                        BUG_ON(kaddr == NULL);
-                       base = kaddr + con->out_msg_pos.page_pos + bio_offset;
+                       base = kaddr + msg_pos->page_pos + bio_offset;
                        crc = crc32c(crc, base, len);
                        kunmap(page);
                        msg->footer.data_crc = cpu_to_le32(crc);
-                       con->out_msg_pos.did_page_crc = true;
+                       msg_pos->did_page_crc = true;
                }
                ret = ceph_tcp_sendpage(con->sock, page,
-                                     con->out_msg_pos.page_pos + bio_offset,
+                                     msg_pos->page_pos + bio_offset,
                                      len, true);
                if (ret <= 0)
                        goto out;
@@ -1811,22 +1807,23 @@ static int read_partial_message_pages(struct ceph_connection *con,
                                      struct page **pages,
                                      unsigned int data_len, bool do_datacrc)
 {
+       struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
        struct page *page;
        void *p;
        int ret;
        int left;
 
-       left = min((int)(data_len - con->in_msg_pos.data_pos),
-                  (int)(PAGE_SIZE - con->in_msg_pos.page_pos));
+       left = min((int)(data_len - msg_pos->data_pos),
+                  (int)(PAGE_SIZE - msg_pos->page_pos));
        /* (page) data */
        BUG_ON(pages == NULL);
-       page = pages[con->in_msg_pos.page];
+       page = pages[msg_pos->page];
        p = kmap(page);
-       ret = ceph_tcp_recvmsg(con->sock, p + con->in_msg_pos.page_pos, left);
+       ret = ceph_tcp_recvmsg(con->sock, p + msg_pos->page_pos, left);
        if (ret > 0 && do_datacrc)
                con->in_data_crc =
                        crc32c(con->in_data_crc,
-                                 p + con->in_msg_pos.page_pos, ret);
+                                 p + msg_pos->page_pos, ret);
        kunmap(page);
        if (ret <= 0)
                return ret;
@@ -1841,6 +1838,7 @@ static int read_partial_message_bio(struct ceph_connection *con,
                                    unsigned int data_len, bool do_datacrc)
 {
        struct ceph_msg *msg = con->in_msg;
+       struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
        struct bio_vec *bv;
        struct page *page;
        void *p;
@@ -1850,17 +1848,17 @@ static int read_partial_message_bio(struct ceph_connection *con,
        BUG_ON(!msg->bio_iter);
        bv = bio_iovec_idx(msg->bio_iter, msg->bio_seg);
 
-       left = min((int)(data_len - con->in_msg_pos.data_pos),
-                  (int)(bv->bv_len - con->in_msg_pos.page_pos));
+       left = min((int)(data_len - msg_pos->data_pos),
+                  (int)(bv->bv_len - msg_pos->page_pos));
 
        page = bv->bv_page;
        p = kmap(page) + bv->bv_offset;
 
-       ret = ceph_tcp_recvmsg(con->sock, p + con->in_msg_pos.page_pos, left);
+       ret = ceph_tcp_recvmsg(con->sock, p + msg_pos->page_pos, left);
        if (ret > 0 && do_datacrc)
                con->in_data_crc =
                        crc32c(con->in_data_crc,
-                                 p + con->in_msg_pos.page_pos, ret);
+                                 p + msg_pos->page_pos, ret);
        kunmap(page);
        if (ret <= 0)
                return ret;
@@ -1877,6 +1875,7 @@ static int read_partial_message_bio(struct ceph_connection *con,
 static int read_partial_message(struct ceph_connection *con)
 {
        struct ceph_msg *m = con->in_msg;
+       struct ceph_msg_pos *msg_pos = &con->in_msg_pos;
        int size;
        int end;
        int ret;
@@ -1957,12 +1956,12 @@ static int read_partial_message(struct ceph_connection *con)
                if (m->middle)
                        m->middle->vec.iov_len = 0;
 
-               con->in_msg_pos.page = 0;
+               msg_pos->page = 0;
                if (m->pages)
-                       con->in_msg_pos.page_pos = m->page_alignment;
+                       msg_pos->page_pos = m->page_alignment;
                else
-                       con->in_msg_pos.page_pos = 0;
-               con->in_msg_pos.data_pos = 0;
+                       msg_pos->page_pos = 0;
+               msg_pos->data_pos = 0;
 
 #ifdef CONFIG_BLOCK
                if (m->bio)
@@ -1986,7 +1985,7 @@ static int read_partial_message(struct ceph_connection *con)
        }
 
        /* (page) data */
-       while (con->in_msg_pos.data_pos < data_len) {
+       while (msg_pos->data_pos < data_len) {
                if (m->pages) {
                        ret = read_partial_message_pages(con, m->pages,
                                                 data_len, do_datacrc);
@@ -2691,6 +2690,49 @@ void ceph_con_keepalive(struct ceph_connection *con)
 }
 EXPORT_SYMBOL(ceph_con_keepalive);
 
+void ceph_msg_data_set_pages(struct ceph_msg *msg, struct page **pages,
+               size_t length, size_t alignment)
+{
+       BUG_ON(!pages);
+       BUG_ON(!length);
+       BUG_ON(msg->pages);
+       BUG_ON(msg->length);
+
+       msg->pages = pages;
+       msg->length = length;
+       msg->page_alignment = alignment & ~PAGE_MASK;
+}
+EXPORT_SYMBOL(ceph_msg_data_set_pages);
+
+void ceph_msg_data_set_pagelist(struct ceph_msg *msg,
+                               struct ceph_pagelist *pagelist)
+{
+       BUG_ON(!pagelist);
+       BUG_ON(!pagelist->length);
+       BUG_ON(msg->pagelist);
+
+       msg->pagelist = pagelist;
+}
+EXPORT_SYMBOL(ceph_msg_data_set_pagelist);
+
+void ceph_msg_data_set_bio(struct ceph_msg *msg, struct bio *bio)
+{
+       BUG_ON(!bio);
+       BUG_ON(msg->bio);
+
+       msg->bio = bio;
+}
+EXPORT_SYMBOL(ceph_msg_data_set_bio);
+
+void ceph_msg_data_set_trail(struct ceph_msg *msg, struct ceph_pagelist *trail)
+{
+       BUG_ON(!trail);
+       BUG_ON(!trail->length);
+       BUG_ON(msg->trail);
+
+       msg->trail = trail;
+}
+EXPORT_SYMBOL(ceph_msg_data_set_trail);
 
 /*
  * construct a new message with given type, size
@@ -2701,49 +2743,19 @@ struct ceph_msg *ceph_msg_new(int type, int front_len, gfp_t flags,
 {
        struct ceph_msg *m;
 
-       m = kmalloc(sizeof(*m), flags);
+       m = kzalloc(sizeof(*m), flags);
        if (m == NULL)
                goto out;
-       kref_init(&m->kref);
-
-       m->con = NULL;
-       INIT_LIST_HEAD(&m->list_head);
 
-       m->hdr.tid = 0;
        m->hdr.type = cpu_to_le16(type);
        m->hdr.priority = cpu_to_le16(CEPH_MSG_PRIO_DEFAULT);
-       m->hdr.version = 0;
        m->hdr.front_len = cpu_to_le32(front_len);
-       m->hdr.middle_len = 0;
-       m->hdr.data_len = 0;
-       m->hdr.data_off = 0;
-       m->hdr.reserved = 0;
-       m->footer.front_crc = 0;
-       m->footer.middle_crc = 0;
-       m->footer.data_crc = 0;
-       m->footer.flags = 0;
-       m->front_max = front_len;
-       m->front_is_vmalloc = false;
-       m->more_to_follow = false;
-       m->ack_stamp = 0;
-       m->pool = NULL;
-
-       /* middle */
-       m->middle = NULL;
 
-       /* data */
-       m->page_count = 0;
-       m->page_alignment = 0;
-       m->pages = NULL;
-       m->pagelist = NULL;
-#ifdef CONFIG_BLOCK
-       m->bio = NULL;
-       m->bio_iter = NULL;
-       m->bio_seg = 0;
-#endif /* CONFIG_BLOCK */
-       m->trail = NULL;
+       INIT_LIST_HEAD(&m->list_head);
+       kref_init(&m->kref);
 
        /* front */
+       m->front_max = front_len;
        if (front_len) {
                if (front_len > PAGE_CACHE_SIZE) {
                        m->front.iov_base = __vmalloc(front_len, flags,
@@ -2895,7 +2907,7 @@ void ceph_msg_last_put(struct kref *kref)
                ceph_buffer_put(m->middle);
                m->middle = NULL;
        }
-       m->page_count = 0;
+       m->length = 0;
        m->pages = NULL;
 
        if (m->pagelist) {
@@ -2915,8 +2927,8 @@ EXPORT_SYMBOL(ceph_msg_last_put);
 
 void ceph_msg_dump(struct ceph_msg *msg)
 {
-       pr_debug("msg_dump %p (front_max %d page_count %d)\n", msg,
-                msg->front_max, msg->page_count);
+       pr_debug("msg_dump %p (front_max %d length %zd)\n", msg,
+                msg->front_max, msg->length);
        print_hex_dump(KERN_DEBUG, "header: ",
                       DUMP_PREFIX_OFFSET, 16, 1,
                       &msg->hdr, sizeof(msg->hdr), true);