102 lines
3.0 KiB
C
102 lines
3.0 KiB
C
#ifndef _TEST_STATE_MACHINE_H
|
|
#define _TEST_STATE_MACHINE_H
|
|
|
|
#ifdef __cpluscplus
|
|
extern "C"
|
|
{
|
|
#endif
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "session_private.h"
|
|
#include "timestamp.h"
|
|
#include "session_manager.h"
|
|
#include "ipv4_helpers.h"
|
|
#include "tcp_helpers.h"
|
|
#include "test_packets.h"
|
|
|
|
struct session_manager_config config = {
|
|
// max session number
|
|
.max_tcp_session_num = 3,
|
|
.max_udp_session_num = 3,
|
|
|
|
// tcp timeout
|
|
.tcp_timeout_init = 2,
|
|
.tcp_timeout_handshake = 2,
|
|
.tcp_timeout_data = 2,
|
|
.tcp_timeout_half_closed = 2,
|
|
.tcp_timeout_time_wait = 2,
|
|
.tcp_timeout_discard = 2,
|
|
|
|
// udp timeout
|
|
.udp_timeout_data = 1,
|
|
|
|
// tcp duplicate packet filter
|
|
.tcp_dupkt_filter_enable = 1,
|
|
.tcp_dupkt_filter_capacity = 1000,
|
|
.tcp_dupkt_filter_timeout = 10,
|
|
.tcp_dupkt_filter_error_rate = 0.0001,
|
|
|
|
// udp eviction filter
|
|
.udp_eviction_filter_enable = 1,
|
|
.udp_eviction_filter_capacity = 1000,
|
|
.udp_eviction_filter_timeout = 10,
|
|
.udp_eviction_filter_error_rate = 0.0001,
|
|
};
|
|
|
|
static void __session_dispatch(struct session *sess)
|
|
{
|
|
if (sess == NULL)
|
|
{
|
|
return;
|
|
}
|
|
|
|
printf("\n");
|
|
printf("=> session dispatch: %p\n", sess);
|
|
session_dump(sess);
|
|
printf("<= session dispatch\n");
|
|
printf("\n");
|
|
|
|
session_set0_cur_pkt(sess, NULL);
|
|
session_set_cur_dir(sess, SESSION_DIR_NONE);
|
|
}
|
|
|
|
static void __session_manager_dispatch(struct session_manager *mgr, enum closing_reasion reason)
|
|
{
|
|
struct session *sess;
|
|
for (int i = 0; i < 3; i++)
|
|
{
|
|
timestamp_update();
|
|
sess = session_manager_get_expired_session(mgr);
|
|
__session_dispatch(sess);
|
|
if (sess && session_get_state(sess) == SESSION_STATE_CLOSING)
|
|
{
|
|
EXPECT_TRUE(session_get_closing_reasion(sess) == reason);
|
|
}
|
|
sess = session_manager_get_evicted_session(mgr);
|
|
__session_dispatch(sess);
|
|
if (sess && session_get_state(sess) == SESSION_STATE_CLOSING)
|
|
{
|
|
EXPECT_TRUE(session_get_closing_reasion(sess) == reason);
|
|
}
|
|
sleep(1);
|
|
}
|
|
}
|
|
|
|
static void __session_manager_count(struct session_manager *mgr, uint64_t tcp_opening_num, uint64_t tcp_active_num, uint64_t tcp_closing_num, uint64_t udp_opening_num, uint64_t udp_active_num, uint64_t udp_closing_num)
|
|
{
|
|
EXPECT_TRUE(session_manager_get_session_number(mgr, SESSION_TYPE_TCP, SESSION_STATE_OPENING) == tcp_opening_num);
|
|
EXPECT_TRUE(session_manager_get_session_number(mgr, SESSION_TYPE_TCP, SESSION_STATE_ACTIVE) == tcp_active_num);
|
|
EXPECT_TRUE(session_manager_get_session_number(mgr, SESSION_TYPE_TCP, SESSION_STATE_CLOSING) == tcp_closing_num);
|
|
|
|
EXPECT_TRUE(session_manager_get_session_number(mgr, SESSION_TYPE_UDP, SESSION_STATE_OPENING) == udp_opening_num);
|
|
EXPECT_TRUE(session_manager_get_session_number(mgr, SESSION_TYPE_UDP, SESSION_STATE_ACTIVE) == udp_active_num);
|
|
EXPECT_TRUE(session_manager_get_session_number(mgr, SESSION_TYPE_UDP, SESSION_STATE_CLOSING) == udp_closing_num);
|
|
}
|
|
|
|
#ifdef __cpluscplus
|
|
}
|
|
#endif
|
|
|
|
#endif
|