tipc: stricter behavior of message reassembly function

The function tipc_link_recv_fragment(struct sk_buff **buf) currently
leaves the value of the input buffer pointer undefined when it returns,
except when the return code indicates that the reassembly is complete.
This despite the fact that it always consumes the input buffer.

Here, we enforce a stricter behavior by this function, ensuring that
the returned buffer pointer is non-NULL if and only if the reassembly
is complete. This makes it possible to test for the buffer pointer as
criteria for successful reassembly.

We also rename the function to tipc_link_frag_rcv(), which is both
shorter and more in line with common naming practice in the network
subsystem.

Apart from the new name, these changes have no impact on current
users of the function, but makes it more practical for use in some
planned future commits.

Signed-off-by: Jon Maloy <jon.maloy@ericsson.com>
Reviewed-by: Ying Xue <ying.xue@windriver.com>
Signed-off-by: David S. Miller <davem@davemloft.net>

1 files changed, 9 insertions, 7 deletions

diff --git a/net/tipc/link.c b/net/tipc/link.c
index d4b5de41b682..17fbd15fcad8 100644
--- a/net/tipc/link.c
+++ b/net/tipc/link.c
@@ -1584,9 +1584,9 @@ deliver:
                         continue;
                 case MSG_FRAGMENTER:
                         l_ptr->stats.recv_fragments++;
-                        ret = tipc_link_recv_fragment(&l_ptr->reasm_head,
-                                                      &l_ptr->reasm_tail,
-                                                      &buf);
+                        ret = tipc_link_frag_rcv(&l_ptr->reasm_head,
+                                                 &l_ptr->reasm_tail,
+                                                 &buf);
                         if (ret == LINK_REASM_COMPLETE) {
                                 l_ptr->stats.recv_fragmented++;
                                 msg = buf_msg(buf);
@@ -2277,12 +2277,11 @@ static int link_send_long_buf(struct tipc_link *l_ptr, struct sk_buff *buf)
         return dsz;
 }
 
-/*
- * tipc_link_recv_fragment(): Called with node lock on. Returns
+/* tipc_link_frag_rcv(): Called with node lock on. Returns
  * the reassembled buffer if message is complete.
  */
-int tipc_link_recv_fragment(struct sk_buff **head, struct sk_buff **tail,
-                            struct sk_buff **fbuf)
+int tipc_link_frag_rcv(struct sk_buff **head, struct sk_buff **tail,
+                       struct sk_buff **fbuf)
 {
         struct sk_buff *frag = *fbuf;
         struct tipc_msg *msg = buf_msg(frag);
@@ -2296,6 +2295,7 @@ int tipc_link_recv_fragment(struct sk_buff **head, struct sk_buff **tail,
                         goto out_free;
                 *head = frag;
                 skb_frag_list_init(*head);
+                *fbuf = NULL;
                 return 0;
         } else if (*head &&
                    skb_try_coalesce(*head, frag, &headstolen, &delta)) {
@@ -2315,10 +2315,12 @@ int tipc_link_recv_fragment(struct sk_buff **head, struct sk_buff **tail,
                 *tail = *head = NULL;
                 return LINK_REASM_COMPLETE;
         }
+        *fbuf = NULL;
         return 0;
 out_free:
         pr_warn_ratelimited("Link unable to reassemble fragmented message\n");
         kfree_skb(*fbuf);
+        *fbuf = NULL;
         return LINK_REASM_ERROR;
 }