The incoming udp header in lowpan_compress_udp_header function is
already in network byte order.

Everytime we read this values for source and destination port we need
to convert this value to host byte order.

In the outcoming header we need to set this value in network byteorder
which the upcoming process assumes.

Signed-off-by: Alexander Aring <alex.ar...@gmail.com>
---
 net/ieee802154/6lowpan.c | 33 +++++++++++++++++++--------------
 1 file changed, 19 insertions(+), 14 deletions(-)

diff --git a/net/ieee802154/6lowpan.c b/net/ieee802154/6lowpan.c
index 769b2b7..4477659 100644
--- a/net/ieee802154/6lowpan.c
+++ b/net/ieee802154/6lowpan.c
@@ -341,28 +341,30 @@ lowpan_compress_udp_header(u8 **hc06_ptr, struct sk_buff 
*skb)
 {
        struct udphdr *uh = udp_hdr(skb);
 
-       if (((uh->source & LOWPAN_NHC_UDP_4BIT_MASK) ==
+       if (((ntohs(uh->source) & LOWPAN_NHC_UDP_4BIT_MASK) ==
                                LOWPAN_NHC_UDP_4BIT_PORT) &&
-           ((uh->dest & LOWPAN_NHC_UDP_4BIT_MASK) ==
+           ((ntohs(uh->dest) & LOWPAN_NHC_UDP_4BIT_MASK) ==
                                LOWPAN_NHC_UDP_4BIT_PORT)) {
                pr_debug("UDP header: both ports compression to 4 bits\n");
                **hc06_ptr = LOWPAN_NHC_UDP_CS_P_11;
                *(*hc06_ptr + 1) = /* subtraction is faster */
-                  (u8)((uh->dest - LOWPAN_NHC_UDP_4BIT_PORT) +
-                      ((uh->source & LOWPAN_NHC_UDP_4BIT_PORT) << 4));
+                  (u8)((ntohs(uh->dest) - LOWPAN_NHC_UDP_4BIT_PORT) +
+                      ((ntohs(uh->source) & LOWPAN_NHC_UDP_4BIT_PORT) << 4));
                *hc06_ptr += 2;
-       } else if ((uh->dest & LOWPAN_NHC_UDP_8BIT_MASK) ==
+       } else if ((ntohs(uh->dest) & LOWPAN_NHC_UDP_8BIT_MASK) ==
                        LOWPAN_NHC_UDP_8BIT_PORT) {
                pr_debug("UDP header: remove 8 bits of dest\n");
                **hc06_ptr = LOWPAN_NHC_UDP_CS_P_01;
                memcpy(*hc06_ptr + 1, &uh->source, 2);
-               *(*hc06_ptr + 3) = (u8)(uh->dest - LOWPAN_NHC_UDP_8BIT_PORT);
+               *(*hc06_ptr + 3) = (u8)(ntohs(uh->dest) -
+                               LOWPAN_NHC_UDP_8BIT_PORT);
                *hc06_ptr += 4;
-       } else if ((uh->source & LOWPAN_NHC_UDP_8BIT_MASK) ==
+       } else if ((ntohs(uh->source) & LOWPAN_NHC_UDP_8BIT_MASK) ==
                        LOWPAN_NHC_UDP_8BIT_PORT) {
                pr_debug("UDP header: remove 8 bits of source\n");
                **hc06_ptr = LOWPAN_NHC_UDP_CS_P_10;
-               *(*hc06_ptr + 1) = (u8)(uh->source - LOWPAN_NHC_UDP_8BIT_PORT);
+               *(*hc06_ptr + 1) = (u8)(ntohs(uh->source) -
+                               LOWPAN_NHC_UDP_8BIT_PORT);
                memcpy(*hc06_ptr + 2, &uh->dest, 2);
                *hc06_ptr += 4;
        } else {
@@ -427,19 +429,22 @@ lowpan_uncompress_udp_header(struct sk_buff *skb,
                case LOWPAN_NHC_UDP_CS_P_01:
                        memcpy(&uh->source, &skb->data[0], 2);
                        uh->dest =
-                          skb->data[2] + LOWPAN_NHC_UDP_8BIT_PORT;
+                          htons(skb->data[2] + LOWPAN_NHC_UDP_8BIT_PORT);
                        skb_pull(skb, 3);
                        break;
                case LOWPAN_NHC_UDP_CS_P_10:
-                       uh->source = skb->data[0] + LOWPAN_NHC_UDP_8BIT_PORT;
+                       uh->source = htons(skb->data[0] +
+                                       LOWPAN_NHC_UDP_8BIT_PORT);
                        memcpy(&uh->dest, &skb->data[1], 2);
                        skb_pull(skb, 3);
                        break;
                case LOWPAN_NHC_UDP_CS_P_11:
                        uh->source =
-                          LOWPAN_NHC_UDP_4BIT_PORT + (skb->data[0] >> 4);
+                          htons(LOWPAN_NHC_UDP_4BIT_PORT +
+                                          (skb->data[0] >> 4));
                        uh->dest =
-                          LOWPAN_NHC_UDP_4BIT_PORT + (skb->data[0] & 0x0f);
+                          htons(LOWPAN_NHC_UDP_4BIT_PORT +
+                                          (skb->data[0] & 0x0f));
                        skb_pull(skb, 1);
                        break;
                default:
@@ -449,7 +454,7 @@ lowpan_uncompress_udp_header(struct sk_buff *skb,
                }
 
                pr_debug("uncompressed UDP ports: src = %d, dst = %d\n",
-                        uh->source, uh->dest);
+                               ntohs(uh->source), ntohs(uh->dest));
 
                /* copy checksum */
                memcpy(&uh->check, &skb->data[0], 2);
@@ -465,7 +470,7 @@ lowpan_uncompress_udp_header(struct sk_buff *skb,
                else
                        uh->len = htons(skb->len + sizeof(struct udphdr));
 
-               pr_debug("uncompressed UDP length: src = %d", uh->len);
+               pr_debug("uncompressed UDP length: src = %d", ntohs(uh->len));
        } else {
                pr_debug("ERROR: unsupported NH format\n");
                goto err;
-- 
1.8.4


------------------------------------------------------------------------------
October Webinars: Code for Performance
Free Intel webinars can help you accelerate application performance.
Explore tips for MPI, OpenMP, advanced profiling, and more. Get the most from 
the latest Intel processors and coprocessors. See abstracts and register >
http://pubads.g.doubleclick.net/gampad/clk?id=60134071&iu=/4140/ostg.clktrk
_______________________________________________
Linux-zigbee-devel mailing list
Linux-zigbee-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/linux-zigbee-devel

Reply via email to