天天看点

opendpi 源码分析(二)

先介绍opendpi如何设置各个协议的处理,然后介绍如何处理每一个包,最后是软件的流程图。

第一:首先看下要用到的重要的宏

View Code

#define IPOQUE_SAVE_AS_BITMASK(bitmask,value) (bitmask)=(((IPOQUE_PROTOCOL_BITMASK)1)<<(value))

#define IPOQUE_BITMASK_COMPARE(a,b) ((a) & (b))

#define IPOQUE_BITMASK_MATCH(x,y) ((x) == (y))

#define IPOQUE_BITMASK_COMPARE(a,b) ((a) & (b))

#define IPOQUE_COMPARE_PROTOCOL_TO_BITMASK(bmask,value) IPOQUE_BITMASK_COMPARE(bmask,IPOQUE_CONVERT_PROTOCOL_TO_BITMASK(value))

#define IPOQUE_CONVERT_PROTOCOL_TO_BITMASK(p) ( ((IPOQUE_PROTOCOL_BITMASK)1) << (p) )

#define IPOQUE_ADD_PROTOCOL_TO_BITMASK(bmask,value) IPOQUE_BITMASK_ADD(bmask,IPOQUE_CONVERT_PROTOCOL_TO_BITMASK(value))

#define IPOQUE_BITMASK_ADD(a,b) (a)|=(b)

#define IPOQUE_BITMASK_SET(a,b) (a)=(b)

#define IPOQUE_DEL_PROTOCOL_FROM_BITMASK(bmask,value) IPOQUE_BITMASK_DEL(bmask,IPOQUE_CONVERT_PROTOCOL_TO_BITMASK(value))

#define IPOQUE_BITMASK_DEL(a,b) (a)=((a) & (~(b)))

#define IPOQUE_BITMASK_SET_ALL(a) (a) = ((IPOQUE_PROTOCOL_BITMASK)0xFFFFFFFFFFFFFFFFULL)

#define IPOQUE_BITMASK_RESET(a) (a) = 0

然后是标识协议掩码的数据结构:

#if IPOQUE_MAX_SUPPORTED_PROTOCOLS >= 128

typedef struct ipoque_protocol_bitmask_struct {

u64 bitmask[ 3 ];

} ipoque_protocol_bitmask_struct_t;

#define IPOQUE_PROTOCOL_BITMASK struct ipoque_protocol_bitmask_struct

#elif IPOQUE_MAX_SUPPORTED_PROTOCOLS >= 64

typedef struct ipoque_protocol_bitmask_struct {

u64 bitmask[ 2 ];

} ipoque_protocol_bitmask_struct_t;

#define IPOQUE_PROTOCOL_BITMASK struct ipoque_protocol_bitmask_struct

#if IPOQUE_MAX_SUPPORTED_PROTOCOLS >= 128

typedef struct ipoque_protocol_bitmask_struct {

u64 bitmask[ 3 ];

} ipoque_protocol_bitmask_struct_t;

#define IPOQUE_PROTOCOL_BITMASK struct ipoque_protocol_bitmask_struct

#elif IPOQUE_MAX_SUPPORTED_PROTOCOLS >= 64

typedef struct ipoque_protocol_bitmask_struct {

u64 bitmask[ 2 ];

} ipoque_protocol_bitmask_struct_t;

#define IPOQUE_PROTOCOL_BITMASK struct ipoque_protocol_bitmask_struct

#else

#define IPOQUE_PROTOCOL_BITMASK u64

#endif

#ifndef u64

#define u64 unsigned long long

#endif

#ifndef u32

#define u32 unsigned int

#endif

#ifndef u16

#define u16 unsigned short

#endif

#ifndef u8

#define u8 unsigned char

#endif

我开始的时候一直以为 IPOQUE_PROTOCOL_BITMASK是整型,没想到是个结构体。

在函数ipoque_set_protocol_detection_bitmask2中开启了对某个固定协议的处理,比如这样就开启了所有协议来检测包:

// enable all protocols

IPOQUE_BITMASK_SET_ALL(all);

ipoque_set_protocol_detection_bitmask2(ipoque_struct, & all);

在具体的一种协议的实现中是这样处理的,比如YAHOO:

View Code

#ifdef IPOQUE_PROTOCOL_YAHOO

if (IPOQUE_COMPARE_PROTOCOL_TO_BITMASK( * detection_bitmask, IPOQUE_PROTOCOL_YAHOO) != 0 ) {

ipoque_struct -> callback_buffer[a].func = ipoque_search_yahoo;

ipoque_struct -> callback_buffer[a].ipq_selection_bitmask = IPQ_SELECTION_BITMASK_PROTOCOL_TCP_OR_UDP;

IPOQUE_SAVE_AS_BITMASK(ipoque_struct -> callback_buffer[a].detection_bitmask, IPOQUE_PROTOCOL_UNKNOWN);

IPOQUE_ADD_PROTOCOL_TO_BITMASK(ipoque_struct -> callback_buffer[a].detection_bitmask, IPOQUE_PROTOCOL_YAHOO);

IPOQUE_SAVE_AS_BITMASK(ipoque_struct -> callback_buffer[a].excluded_protocol_bitmask, IPOQUE_PROTOCOL_YAHOO);

a ++ ;

}

而函数ipoque_search_yahoo就是存在于目录protocol中,对于YAHOO的具体检查,实现如下:

View Code

void ipoque_search_yahoo( struct ipoque_detection_module_struct * ipoque_struct)

{

struct ipoque_packet_struct * packet = & ipoque_struct -> packet;

struct ipoque_flow_struct * flow = ipoque_struct -> flow;

IPQ_LOG(IPOQUE_PROTOCOL_YAHOO, ipoque_struct, IPQ_LOG_DEBUG, " search yahoo\n " );

if (packet -> payload_packet_len > 0 && flow -> yahoo_detection_finished == 0 ) {

if (packet -> tcp != NULL && packet -> tcp_retransmission == 0 ) {

#ifdef IPOQUE_PROTOCOL_HTTP

if (packet -> detected_protocol == IPOQUE_PROTOCOL_UNKNOWN

|| packet -> detected_protocol == IPOQUE_PROTOCOL_HTTP)

#else

if (packet -> detected_protocol == IPOQUE_PROTOCOL_UNKNOWN)

#endif

{

ipoque_search_yahoo_tcp(ipoque_struct);

}

} else if (packet -> udp != NULL) {

ipoque_search_yahoo_udp(ipoque_struct);

}

}

if (packet -> payload_packet_len > 0 && flow -> yahoo_detection_finished == 2 ) {

if (packet -> tcp != NULL && packet -> tcp_retransmission == 0 ) {

ipoque_search_yahoo_tcp(ipoque_struct);

}

}

}

这个函数检查了YAHOO是基于UDP还是TCP,还有没有payload长度是不是等于0.

这样就建立了一个匹配的模式数据库。

第二:对于每一个包,在这个函数中从pcap文件中获得:

// executed for each packet in the pcap file

static void pcap_packet_callback(u_char * args, const struct pcap_pkthdr * header, const u_char * packet) 然后在函数

// here the actual detection is performed

protocol = ipoque_detection_process_packet(ipoque_struct, ipq_flow, (uint8_t * ) iph, ipsize, time, src, dst);

这是对每个包的具体处理,而返回值就是我们所想要知道的,这个包所属于的协议号。

在函数 ipoque_detection_process_packet中最重要的是如下的代码:

if (flow != NULL && ipoque_struct -> packet.tcp != NULL) {

if (ipoque_struct -> packet.payload_packet_len != 0 ) {

for (a = 0 ; a < ipoque_struct -> callback_buffer_size_tcp_payload; a ++ ) {

if ((ipoque_struct -> callback_buffer_tcp_payload[a].ipq_selection_bitmask & ipq_selection_packet) ==

ipoque_struct -> callback_buffer_tcp_payload[a].ipq_selection_bitmask

&& IPOQUE_BITMASK_COMPARE(ipoque_struct -> flow -> excluded_protocol_bitmask,

ipoque_struct -> callback_buffer_tcp_payload[a].excluded_protocol_bitmask) == 0

&& IPOQUE_BITMASK_COMPARE(ipoque_struct -> callback_buffer_tcp_payload[a].detection_bitmask,

detection_bitmask) != 0 ) {

ipoque_struct -> callback_buffer_tcp_payload[a].func(ipoque_struct);

}

}

首先判断了payload,而ipq_selection_packet依赖于下面的这些判断:

View Code

ipq_selection_packet = IPQ_SELECTION_BITMASK_PROTOCOL_COMPLETE_TRAFFIC;

if (ipoque_struct -> packet.iph != NULL) {

ipq_selection_packet |= IPQ_SELECTION_BITMASK_PROTOCOL_IP | IPQ_SELECTION_BITMASK_PROTOCOL_IPV4_OR_IPV6;

}

if (ipoque_struct -> packet.tcp != NULL) {

ipq_selection_packet |=

(IPQ_SELECTION_BITMASK_PROTOCOL_INT_TCP | IPQ_SELECTION_BITMASK_PROTOCOL_INT_TCP_OR_UDP);

}

if (ipoque_struct -> packet.udp != NULL) {

ipq_selection_packet |=

(IPQ_SELECTION_BITMASK_PROTOCOL_INT_UDP | IPQ_SELECTION_BITMASK_PROTOCOL_INT_TCP_OR_UDP);

}

if (ipoque_struct -> packet.payload_packet_len != 0 ) {

ipq_selection_packet |= IPQ_SELECTION_BITMASK_PROTOCOL_HAS_PAYLOAD;

}

if (ipoque_struct -> packet.tcp_retransmission == 0 ) {

ipq_selection_packet |= IPQ_SELECTION_BITMASK_PROTOCOL_NO_TCP_RETRANSMISSION;

}

excluded_protocol_bitmask是设置不需要检查的协议,我们上面已经设置为all。

detection_bitmask通过这样初始化:

View Code

#define IPOQUE_CONVERT_PROTOCOL_TO_BITMASK(p) ( ((IPOQUE_PROTOCOL_BITMASK)1) << (p) )

IPOQUE_SAVE_AS_BITMASK(detection_bitmask, ipoque_struct -> packet.detected_protocol);

通过gdb可以查看 detection_bitmask的值是:bitmask = {1, 0}。

这样就把每一个包放入已有的协议模式去匹配。

主要流程图:

opendpi 源码分析(二)

也许有错误,希望指教。

转载于:https://www.cnblogs.com/lxgeek/archive/2011/04/21/2024157.html