gnrc_netif: add fetch-address and fetch-netif-flag functionalities.

RIOT-OS · Dec 1, 2016 · 670f70c · 670f70c
1 parent e6ad438
commit 670f70c
Show file tree

Hide file tree

Showing 2 changed files with 90 additions and 0 deletions.
diff --git a/sys/include/net/gnrc/netif/hdr.h b/sys/include/net/gnrc/netif/hdr.h
@@ -199,6 +199,37 @@ gnrc_pktsnip_t *gnrc_netif_hdr_build(uint8_t *src, uint8_t src_len, uint8_t *dst
  */
 void gnrc_netif_hdr_print(gnrc_netif_hdr_t *hdr);
 
+/* @brief   Fetch the netif header flags of a gnrc packet
+ *
+ * @param[in]   pkt     gnrc packet from whom to fetch
+ *
+ * @return              netif header flags of @p pkt
+ * @return              0, if no header is present
+ */
+uint8_t gnrc_netif_hdr_get_flag(gnrc_pktsnip_t* pkt);
+
+/* @brief   Extract the destination address out of a gnrc packet
+ *
+ * @param[in]   pkt                 gnrc packet from whom to extract
+ * @param[out]  pointer_to_addr     pointer to address will be stored here
+ *
+ * @return                          length of destination address
+ * @return                          -ENOENT, if no netif header is presented in @p pkt or if no
+ *                                           destination address field presented in netif header.
+ */
+int gnrc_netif_hdr_get_dstaddr(gnrc_pktsnip_t* pkt, uint8_t** pointer_to_addr);
+
+/* @brief   Extract the source address out of a gnrc packet
+ *
+ * @param[in]   pkt                 gnrc packet from whom to extract
+ * @param[out]  pointer_to_addr     pointer to address will be stored here
+ *
+ * @return                          length of source address
+ * @return                          -ENOENT, if no netif header is presented in @p pkt or if no
+ *                                           source address field presented in netif header.
+ */
+int gnrc_netif_hdr_get_srcaddr(gnrc_pktsnip_t* pkt, uint8_t** pointer_to_addr);
+
 #ifdef __cplusplus
 }
 #endif

diff --git a/sys/net/gnrc/netif/hdr/gnrc_netif_hdr.c b/sys/net/gnrc/netif/hdr/gnrc_netif_hdr.c
@@ -38,4 +38,63 @@ gnrc_pktsnip_t *gnrc_netif_hdr_build(uint8_t *src, uint8_t src_len, uint8_t *dst
     return pkt;
 }
 
+uint8_t gnrc_netif_hdr_get_flag(gnrc_pktsnip_t* pkt)
+{
+    int res;
+    gnrc_netif_hdr_t* netif_hdr;
+
+    assert(pkt != NULL);
+    pkt = gnrc_pktsnip_search_type(pkt, GNRC_NETTYPE_NETIF);
+    if (pkt) {
+        netif_hdr = pkt->data;
+        if (netif_hdr) {
+            return netif_hdr->flags;
+        }
+    }
+
+    return 0U;
+}
+
+int gnrc_netif_hdr_get_dstaddr(gnrc_pktsnip_t* pkt, uint8_t** pointer_to_addr)
+{
+    int res;
+    gnrc_netif_hdr_t* netif_hdr;
+
+    assert(pkt != NULL);
+    pkt = gnrc_pktsnip_search_type(pkt, GNRC_NETTYPE_NETIF);
+    if (pkt) {
+        netif_hdr = pkt->data;
+        if (netif_hdr) {
+            if ((res = netif_hdr->dst_l2addr_len) <= 0) {
+                return -ENOENT;
+            }
+            *pointer_to_addr = gnrc_netif_hdr_get_dst_addr(netif_hdr);
+            return res;
+        }
+    }
+
+    return -ENOENT;
+}
+
+int gnrc_netif_hdr_get_srcaddr(gnrc_pktsnip_t* pkt, uint8_t** pointer_to_addr)
+{
+    int res;
+    gnrc_netif_hdr_t* netif_hdr;
+
+    assert(pkt != NULL);
+    pkt = gnrc_pktsnip_search_type(pkt, GNRC_NETTYPE_NETIF);
+    if (pkt) {
+        netif_hdr = pkt->data;
+        if (netif_hdr) {
+            if ((res = netif_hdr.src_l2addr_len) <= 0) {
+                return -ENOENT;
+            }
+            *pointer_to_addr = gnrc_netif_hdr_get_src_addr(netif_hdr);
+            return res;
+        }
+    }
+
+    return -ENOENT;
+}
+
 /** @} */