Reduce packet_get_xxx() calls

2023-12-21 20:03:43 +08:00
parent cd47b46e7a
commit fa4cc898d8
4 changed files with 125 additions and 18 deletions
--- a/src/packet/udp_helpers.h
+++ b/src/packet/udp_helpers.h
@@ -0,0 +1,27 @@
+#ifndef _UDP_HELPERS_H
+#define _UDP_HELPERS_H
+
+#ifdef __cpluscplus
+extern "C"
+{
+#endif
+
+#include <arpa/inet.h>
+#define __FAVOR_BSD 1
+#include <netinet/udp.h>
+
+static inline uint16_t udp_hdr_get_sport(const struct udphdr *hdr)
+{
+    return ntohs(hdr->uh_sport);
+}
+
+static inline uint16_t udp_hdr_get_dport(const struct udphdr *hdr)
+{
+    return ntohs(hdr->uh_dport);
+}
+
+#ifdef __cpluscplus
+}
+#endif
+
+#endif