Staging: batman-adv: reassemble fragmented skb if mtu allows it

[firefly-linux-kernel-4.4.55.git] / drivers / staging / batman-adv / unicast.c

diff --git a/drivers/staging/batman-adv/unicast.c b/drivers/staging/batman-adv/unicast.c
index 12afae618ce232cf7be8b8ea1375260ca5722e53..e58e634c28060f20e1f20b3a2584f54121b5b827 100644 (file)
--- a/drivers/staging/batman-adv/unicast.c
+++ b/drivers/staging/batman-adv/unicast.c
@@ -36,6 +36,9 @@ static struct sk_buff *frag_merge_packet(struct list_head *head,
        struct unicast_frag_packet *up =
                (struct unicast_frag_packet *)skb->data;
        struct sk_buff *tmp_skb;
+       struct unicast_packet *unicast_packet;
+       int hdr_len = sizeof(struct unicast_packet),
+           uni_diff = sizeof(struct unicast_frag_packet) - hdr_len;
 
        /* set skb to the first part and tmp_skb to the second part */
        if (up->flags & UNI_FRAG_HEAD) {
@@ -59,6 +62,11 @@ static struct sk_buff *frag_merge_packet(struct list_head *head,
 
        memcpy(skb_put(skb, tmp_skb->len), tmp_skb->data, tmp_skb->len);
        kfree_skb(tmp_skb);
+
+       memmove(skb->data + uni_diff, skb->data, hdr_len);
+       unicast_packet = (struct unicast_packet *) skb_pull(skb, uni_diff);
+       unicast_packet->packet_type = BAT_UNICAST;
+
        return skb;
 }