Merge pull request #2575 from authmillenon/ng_net/feat/hdr-build

ng_net: header building facilities

sys/include/net/ng_netif/hdr.h

@@ -15,6 +15,7 @@
  * @brief       Generic network interface header
  *
  * @author      Hauke Petersen <[email protected]>
+ * @author      Martine Lenders <[email protected]>
  */
 
 #ifndef NETIF_HDR_H_
@@ -24,6 +25,8 @@
 #include <stdint.h>
 
 #include "kernel.h"
+#include "net/ng_pkt.h"
+#include "net/ng_pktbuf.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -134,6 +137,44 @@ static inline void ng_netif_hdr_set_dst_addr(ng_netif_hdr_t *hdr, uint8_t *addr,
     memcpy(((uint8_t *)(hdr + 1)) + hdr->src_l2addr_len, addr, addr_len);
 }
 
+/**
+ * @brief   Builds a generic network interface header for sending and
+ *          adds it to the packet buffer.
+ *
+ * @param[in] src       Source address for the header. Can be NULL if not
+ *                      known or required.
+ * @param[in] src_len   Length of @p src. Can be 0 if not known or required.
+ * @param[in] dst       Destination address for the header. Can be NULL if not
+ *                      known or required.
+ * @param[in] dst_len   Length of @p dst. Can be 0 if not known or required.
+ *
+ * @return  The generic network layer header on success.
+ * @return  NULL on error.
+ */
+static inline ng_pktsnip_t *ng_netif_hdr_build(uint8_t *src, uint8_t src_len,
+                                               uint8_t *dst, uint8_t dst_len)
+{
+    ng_pktsnip_t *pkt = ng_pktbuf_add(NULL, NULL,
+                                      sizeof(ng_netif_hdr_t) + src_len + dst_len,
+                                      NG_NETTYPE_NETIF);
+
+    if (pkt == NULL) {
+        return NULL;
+    }
+
+    ng_netif_hdr_init(pkt->data, src_len, dst_len);
+
+    if (src != NULL && src_len > 0) {
+        ng_netif_hdr_set_src_addr(pkt->data, src, src_len);
+    }
+
+    if (dst != NULL && dst_len > 0) {
+        ng_netif_hdr_set_dst_addr(pkt->data, dst, dst_len);
+    }
+
+    return pkt;
+}
+
 #ifdef __cplusplus
 }
 #endif

sys/include/net/ng_netreg.h

@@ -24,6 +24,7 @@
 
 #include "kernel_types.h"
 #include "net/ng_nettype.h"
+#include "net/ng_pkt.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -126,6 +127,25 @@ int ng_netreg_num(ng_nettype_t type, uint32_t demux_ctx);
  */
 ng_netreg_entry_t *ng_netreg_getnext(ng_netreg_entry_t *entry);
 
+/**
+ * @brief   Builds a header for sending and adds it to the packet buffer.
+ *
+ * @param[in] type      Type of the header.
+ * @param[in] payload   Payload for the packet.
+ * @param[in] src       Source address for the header. Can be NULL if not
+ *                      known or required.
+ * @param[in] src_len   Length of @p src. Can be 0 if not known or required.
+ * @param[in] dst       Destination address for the header. Can be NULL if not
+ *                      known or required.
+ * @param[in] dst_len   Length of @p dst. Can be 0 if not known or required.
+ *
+ * @return  The header for the protocol on success.
+ * @return  NULL on error.
+ */
+ng_pktsnip_t *ng_netreg_hdr_build(ng_nettype_t type, ng_pktsnip_t *payload,
+                                  uint8_t *src, uint8_t src_len,
+                                  uint8_t *dst, uint8_t dst_len);
+
 #ifdef __cplusplus
 }
 #endif

sys/net/crosslayer/ng_netreg/ng_netreg.c

@@ -19,6 +19,7 @@
 #include "utlist.h"
 #include "net/ng_netreg.h"
 #include "net/ng_nettype.h"
+#include "net/ng_ipv6.h"
 
 #define _INVALID_TYPE(type) (((type) < NG_NETTYPE_UNDEF) || ((type) >= NG_NETTYPE_NUMOF))
 
@@ -101,4 +102,30 @@ ng_netreg_entry_t *ng_netreg_getnext(ng_netreg_entry_t *entry)
     return entry;
 }
 
+ng_pktsnip_t *ng_netreg_hdr_build(ng_nettype_t type, ng_pktsnip_t *payload,
+                                  uint8_t *src, uint8_t src_len,
+                                  uint8_t *dst, uint8_t dst_len)
+{
+    switch (type) {
+#ifdef MODULE_NG_IPV6
+
+        case NG_NETTYPE_IPV6:
+            return ng_ipv6_hdr_build(payload, src, src_len, dst, dst_len);
+#endif
+#ifdef MODULE_NG_TCP
+
+        case NG_NETTYPE_TCP:
+            return ng_tcp_hdr_build(payload, src, src_len, dst, dst_len);
+#endif
+#ifdef MODULE_NG_UDP
+
+        case NG_NETTYPE_UDP:
+            return ng_udp_hdr_build(payload, src, src_len, dst, dst_len);
+#endif
+
+        default:
+            return NULL;
+    }
+}
+
 /** @} */