Reduce packet_get_xxx() calls
This commit is contained in:
72
src/packet/tcp_helpers.h
Normal file
72
src/packet/tcp_helpers.h
Normal file
@@ -0,0 +1,72 @@
|
||||
#ifndef _TCP_HELPERS_H
|
||||
#define _TCP_HELPERS_H
|
||||
|
||||
#ifdef __cpluscplus
|
||||
extern "C"
|
||||
{
|
||||
#endif
|
||||
|
||||
#include <arpa/inet.h>
|
||||
#define __FAVOR_BSD 1
|
||||
#include <netinet/tcp.h>
|
||||
|
||||
static inline uint16_t tcp_hdr_get_sport(const struct tcphdr *hdr)
|
||||
{
|
||||
return ntohs(hdr->source);
|
||||
}
|
||||
|
||||
static inline uint16_t tcp_hdr_get_dport(const struct tcphdr *hdr)
|
||||
{
|
||||
return ntohs(hdr->dest);
|
||||
}
|
||||
|
||||
static inline uint32_t tcp_hdr_get_seq(const struct tcphdr *hdr)
|
||||
{
|
||||
return ntohl(hdr->seq);
|
||||
}
|
||||
|
||||
static inline uint32_t tcp_hdr_get_ack(const struct tcphdr *hdr)
|
||||
{
|
||||
return ntohl(hdr->ack_seq);
|
||||
}
|
||||
|
||||
static inline uint8_t tcp_hdr_get_flags(const struct tcphdr *hdr)
|
||||
{
|
||||
return hdr->th_flags;
|
||||
}
|
||||
|
||||
static inline bool tcp_hdr_has_flag_urg(const struct tcphdr *hdr)
|
||||
{
|
||||
return hdr->th_flags & TH_URG;
|
||||
}
|
||||
|
||||
static inline bool tcp_hdr_has_flag_ack(const struct tcphdr *hdr)
|
||||
{
|
||||
return hdr->th_flags & TH_ACK;
|
||||
}
|
||||
|
||||
static inline bool tcp_hdr_has_flag_psh(const struct tcphdr *hdr)
|
||||
{
|
||||
return hdr->th_flags & TH_PUSH;
|
||||
}
|
||||
|
||||
static inline bool tcp_hdr_has_flag_rst(const struct tcphdr *hdr)
|
||||
{
|
||||
return hdr->th_flags & TH_RST;
|
||||
}
|
||||
|
||||
static inline bool tcp_hdr_has_flag_syn(const struct tcphdr *hdr)
|
||||
{
|
||||
return hdr->th_flags & TH_SYN;
|
||||
}
|
||||
|
||||
static inline bool tcp_hdr_has_flag_fin(const struct tcphdr *hdr)
|
||||
{
|
||||
return hdr->th_flags & TH_FIN;
|
||||
}
|
||||
|
||||
#ifdef __cpluscplus
|
||||
}
|
||||
#endif
|
||||
|
||||
#endif
|
||||
Reference in New Issue
Block a user