将kni合并到tfe中

common/src/tfe_utils.cpp

@@ -6,6 +6,12 @@
 #include <tfe_utils.h>
 #include <time.h>
 #include <assert.h>
+#include <netinet/ip.h>
+#include <netinet/ether.h>
+#include <arpa/inet.h>
+#include <net/if.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
 
 //functioned as strdup, for dictator compatible.
 char* tfe_strdup(const char* s)
@@ -215,4 +221,173 @@ int tfe_decode_base64url(u_char *dst, u_char *src)
     };
 
     return tfe_decode_base64_internal(dst, src, basis64);
-}
+}
+
+/******************************************************************************
+ * sids
+ ******************************************************************************/
+
+void sids_write_once(struct sids *dst, struct sids *src)
+{
+    if (dst && src)
+    {
+        if (dst->num == 0 && src->num > 0)
+        {
+            sids_copy(dst, src);
+        }
+    }
+}
+
+void sids_copy(struct sids *dst, struct sids *src)
+{
+    if (dst && src)
+    {
+        dst->num = src->num;
+        memcpy(dst->elems, src->elems, sizeof(dst->elems[0]) * dst->num);
+    }
+}
+
+/******************************************************************************
+ * route_ctx
+ ******************************************************************************/
+
+int route_ctx_is_empty(struct route_ctx *ctx)
+{
+    if (ctx->len == 0)
+    {
+        return 1;
+    }
+    else
+    {
+        return 0;
+    }
+}
+
+void route_ctx_copy(struct route_ctx *dst, struct route_ctx *src)
+{
+    memcpy(dst->data, src->data, src->len);
+    dst->len = src->len;
+}
+
+
+/******************************************************************************
+ * protocol
+ ******************************************************************************/
+
+#define CHECKSUM_CARRY(x) (x = (x >> 16) + (x & 0xffff), (~(x + (x >> 16)) & 0xffff))
+
+static int checksum(u_int16_t *addr, int len)
+{
+    int sum = 0;
+    int nleft = len;
+    u_int16_t ans = 0;
+    u_int16_t *w = addr;
+
+    while (nleft > 1)
+    {
+        sum += *w++;
+        nleft -= 2;
+    }
+
+    if (nleft == 1)
+    {
+        *(char *)(&ans) = *(char *)w;
+        sum += ans;
+    }
+
+    return sum;
+}
+
+void build_udp_header(const char *l3_hdr, int l3_hdr_len, struct udp_hdr *udp_hdr, u_int16_t udp_sport, u_int16_t udp_dport, int payload_len)
+{
+    memset(udp_hdr, 0, sizeof(struct udp_hdr));
+
+    int udp_hlen = sizeof(struct udp_hdr) + payload_len;
+
+    udp_hdr->uh_sport = htons(udp_sport);
+    udp_hdr->uh_dport = htons(udp_dport);
+
+    udp_hdr->uh_ulen = htons(udp_hlen);
+    udp_hdr->uh_sum = 0;
+
+    int sum = checksum((u_int16_t *)l3_hdr, l3_hdr_len);
+    sum += ntohs(IPPROTO_UDP + udp_hlen);
+    sum += checksum((u_int16_t *)udp_hdr, udp_hlen);
+    udp_hdr->uh_sum = CHECKSUM_CARRY(sum);
+}
+
+void build_ip_header(struct ip *ip_hdr, u_int8_t next_protocol, const char *src_addr, const char *dst_addr, uint16_t payload_len)
+{
+    memset(ip_hdr, 0, sizeof(struct ip));
+
+    ip_hdr->ip_hl = 5;               /* 20 byte header */
+    ip_hdr->ip_v = 4;                /* version 4 */
+    ip_hdr->ip_tos = 0;              /* IP tos */
+    ip_hdr->ip_id = htons(random()); /* IP ID */
+    ip_hdr->ip_ttl = 80;             /* time to live */
+    ip_hdr->ip_p = next_protocol;    /* transport protocol */
+    ip_hdr->ip_src.s_addr = inet_addr(src_addr);
+    ip_hdr->ip_dst.s_addr = inet_addr(dst_addr);
+    ip_hdr->ip_len = htons(sizeof(struct ip) + payload_len); /* total length */
+    ip_hdr->ip_off = htons(0);                               /* fragmentation flags */
+    ip_hdr->ip_sum = 0;                                      /* do this later */
+
+    int sum = checksum((u_int16_t *)ip_hdr, 20);
+    ip_hdr->ip_sum = CHECKSUM_CARRY(sum);
+}
+
+// l3_protocol: ETH_P_IPV6/ETH_P_IP
+void build_ether_header(struct ethhdr *eth_hdr, uint16_t next_protocol, const char *src_mac, const char *dst_mac)
+{
+    memset(eth_hdr, 0, sizeof(struct ethhdr));
+
+    str_to_mac(src_mac, (char *)eth_hdr->h_source);
+    str_to_mac(dst_mac, (char *)eth_hdr->h_dest);
+    eth_hdr->h_proto = htons(next_protocol);
+}
+
+int str_to_mac(const char *str, char *mac_buff)
+{
+    if (sscanf(str, "%02hhx:%02hhx:%02hhx:%02hhx:%02hhx:%02hhx", &(mac_buff[0]), &(mac_buff[1]), &(mac_buff[2]), &(mac_buff[3]), &(mac_buff[4]), &(mac_buff[5])) == 6)
+    {
+        return 0;
+    }
+    else
+    {
+        return -1;
+    }
+}
+
+int get_mac_by_device_name(const char *dev_name, char *mac_buff)
+{
+    int fd = socket(PF_INET, SOCK_DGRAM, IPPROTO_IP);
+    if (fd == -1)
+    {
+        return -1;
+    }
+
+    struct ifreq ifr;
+    memset(&ifr, 0, sizeof(struct ifreq));
+    strcpy(ifr.ifr_name, dev_name);
+    if (ioctl(fd, SIOCGIFHWADDR, &ifr) != 0)
+    {
+        close(fd);
+        return -1;
+    }
+
+    unsigned char *mac = (unsigned char *)ifr.ifr_hwaddr.sa_data;
+    sprintf(mac_buff, "%02hhx:%02hhx:%02hhx:%02hhx:%02hhx:%02hhx", mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);
+    close(fd);
+
+    return 0;
+}
+
+/******************************************************************************
+ * throughput_metrics
+ ******************************************************************************/
+
+void throughput_metrics_inc(struct throughput_metrics *iterm, uint64_t n_pkts, uint64_t n_bytes)
+{
+    __atomic_fetch_add(&iterm->n_bytes, n_bytes, __ATOMIC_RELAXED);
+    __atomic_fetch_add(&iterm->n_pkts, n_pkts, __ATOMIC_RELAXED);
+}