summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/CMakeLists.txt3
-rw-r--r--test/decoders/rtp/CMakeLists.txt46
-rw-r--r--test/decoders/rtp/conf/spec.toml11
-rw-r--r--test/decoders/rtp/conf/stellar.toml79
-rw-r--r--test/decoders/rtp/pcap/01-rtp-10-packet-PCMA.pcapbin0 -> 2324 bytes
-rw-r--r--test/decoders/rtp/result/rtp_result.json119
-rw-r--r--test/decoders/rtp/rtp_test.cpp325
7 files changed, 582 insertions, 1 deletions
diff --git a/test/CMakeLists.txt b/test/CMakeLists.txt
index a402bf4..6eb0986 100644
--- a/test/CMakeLists.txt
+++ b/test/CMakeLists.txt
@@ -5,4 +5,5 @@ add_subdirectory(lpi_plus)
#add_subdirectory(decoders/http)
#add_subdirectory(decoders/socks)
#add_subdirectory(decoders/stratum)
-#add_subdirectory(decoders/session_flags) \ No newline at end of file
+#add_subdirectory(decoders/session_flags)
+add_subdirectory(decoders/rtp)
diff --git a/test/decoders/rtp/CMakeLists.txt b/test/decoders/rtp/CMakeLists.txt
new file mode 100644
index 0000000..c9389e8
--- /dev/null
+++ b/test/decoders/rtp/CMakeLists.txt
@@ -0,0 +1,46 @@
+set(TEST_NAME rtp_test)
+set(TEST_MAIN ${CMAKE_CURRENT_BINARY_DIR}/${TEST_NAME})
+file(GLOB TEST_SRC "${TEST_NAME}*.cpp")
+
+add_executable(
+ ${TEST_NAME}
+ ${TEST_SRC}
+)
+
+target_include_directories(
+ ${TEST_NAME} PRIVATE
+ ${CMAKE_SOURCE_DIR}/deps/
+ ${CMAKE_SOURCE_DIR}/decoders/
+)
+
+target_link_libraries(
+ ${TEST_NAME}
+ rtp
+ stellar_lib
+ cjson-static
+ dl "-rdynamic"
+ gtest
+ gmock
+)
+
+add_test(
+ NAME ${TEST_NAME}.SETUP
+ COMMAND sh -c "
+ cat ${CMAKE_CURRENT_SOURCE_DIR}/conf/stellar.toml > ${CMAKE_CURRENT_BINARY_DIR}/stellar.toml &&
+ cat ${CMAKE_CURRENT_SOURCE_DIR}/conf/spec.toml >> ${CMAKE_CURRENT_BINARY_DIR}/stellar.toml &&
+ cat ${CMAKE_CURRENT_SOURCE_DIR}/result/rtp_result.json >> ${CMAKE_CURRENT_BINARY_DIR}/rtp_result.json
+ "
+)
+
+add_test(
+ NAME ${TEST_NAME}
+ COMMAND sh -c "
+ find ${CMAKE_CURRENT_SOURCE_DIR}/pcap/ -type f | sort -V > ${CMAKE_CURRENT_BINARY_DIR}/pcaplist.txt &&
+ ${TEST_MAIN}
+ "
+)
+
+set_tests_properties(
+ ${TEST_NAME} PROPERTIES
+ FIXTURES_REQUIRED ${TEST_NAME}.SETUP
+)
diff --git a/test/decoders/rtp/conf/spec.toml b/test/decoders/rtp/conf/spec.toml
new file mode 100644
index 0000000..1360252
--- /dev/null
+++ b/test/decoders/rtp/conf/spec.toml
@@ -0,0 +1,11 @@
+# stellar_plugin.toml
+#
+[[module]]
+path = ""
+init = "rtp_init"
+exit = "rtp_exit"
+
+[[module]]
+path = ""
+init = "rtp_test_init"
+exit = "rtp_test_exit"
diff --git a/test/decoders/rtp/conf/stellar.toml b/test/decoders/rtp/conf/stellar.toml
new file mode 100644
index 0000000..c7ce725
--- /dev/null
+++ b/test/decoders/rtp/conf/stellar.toml
@@ -0,0 +1,79 @@
+[instance]
+ id = 1 # range: [0, 4095] (12 bit)
+
+[packet_io]
+ mode = "pcaplist" # pcapfile, pcaplist, marsio
+ app_symbol = "stellar"
+ dev_symbol = "nf_0_fw"
+ pcap_path = "pcaplist.txt"
+ thread_num = 1 # range: [1, 256]
+ cpu_mask = [5, 6, 7, 8, 9, 10, 11, 12]
+ idle_yield_ms = 900 # range: [0, 60000] (ms)
+
+ [packet_io.packet_pool]
+ capacity = 1024 # range: [1, 4294967295]
+
+ [packet_io.ip_reassembly]
+ fail_action = 1 # 0: bypass, 1: drop
+ timeout_ms = 1000 # range: [1, 60000] (ms)
+ frag_queue_num = 1024 # range: [1, 4294967295]
+ frag_queue_size = 64 # range: [2, 65535]
+
+[session_manager]
+ tcp_session_max = 50000
+ udp_session_max = 50000
+
+ evict_old_on_tcp_table_limit = 1 # range: [0, 1]
+ evict_old_on_udp_table_limit = 1 # range: [0, 1]
+
+ expire_period_ms = 0 # range: [0, 60000] (ms)
+ expire_batch_max = 1024 # range: [1, 1024]
+
+ [session_manager.tcp_timeout_ms]
+ init = 5000 # range: [1, 60000] (ms)
+ handshake = 5000 # range: [1, 60000] (ms)
+ data = 5000 # range: [1, 15999999000] (ms)
+ half_closed = 5000 # range: [1, 604800000] (ms)
+ time_wait = 5000 # range: [1, 600000] (ms)
+ discard_default = 10000 # range: [1, 15999999000] (ms)
+ unverified_rst = 5000 # range: [1, 600000] (ms)
+
+ [session_manager.udp_timeout_ms]
+ data = 5000 # range: [1, 15999999000] (ms)
+ discard_default = 5000 # range: [1, 15999999000] (ms)
+
+ [session_manager.duplicated_packet_bloom_filter]
+ enable = 1
+ capacity = 1000000 # range: [1, 4294967295]
+ time_window_ms = 10000 # range: [1, 60000] (ms)
+ error_rate = 0.00001 # range: [0.0, 1.0]
+
+ [session_manager.evicted_session_bloom_filter]
+ enable = 1 # range: [0, 1]
+ capacity = 1000000 # range: [1, 4294967295]
+ time_window_ms = 10000 # range: [1, 60000] (ms)
+ error_rate = 0.00001 # range: [0.0, 1.0]
+
+ [session_manager.tcp_reassembly]
+ enable = 1 # range: [0, 1]
+ timeout_ms = 10000 # range: [1, 60000] (ms)
+ buffered_segments_max = 256 # range: [2, 4096] per flow
+
+[log]
+ output = "both" # stderr, file, both
+ file = "stellar.log"
+ level = "INFO" # TRACE, DEBUG, INFO, WARN, ERROR, FATAL
+
+[[module]]
+ path = ""
+ init = "packet_manager_on_init"
+ exit = "packet_manager_on_exit"
+ thread_init = "packet_manager_on_thread_init"
+ thread_exit = "packet_manager_on_thread_exit"
+
+[[module]]
+ path = ""
+ init = "session_manager_on_init"
+ exit = "session_manager_on_exit"
+ thread_init = "session_manager_on_thread_init"
+ thread_exit = "session_manager_on_thread_exit"
diff --git a/test/decoders/rtp/pcap/01-rtp-10-packet-PCMA.pcap b/test/decoders/rtp/pcap/01-rtp-10-packet-PCMA.pcap
new file mode 100644
index 0000000..a8d063f
--- /dev/null
+++ b/test/decoders/rtp/pcap/01-rtp-10-packet-PCMA.pcap
Binary files differ
diff --git a/test/decoders/rtp/result/rtp_result.json b/test/decoders/rtp/result/rtp_result.json
new file mode 100644
index 0000000..fa626b7
--- /dev/null
+++ b/test/decoders/rtp/result/rtp_result.json
@@ -0,0 +1,119 @@
+[
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 2,
+ "TIMESTAMP": 320,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 0
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 3,
+ "TIMESTAMP": 480,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 1
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 4,
+ "TIMESTAMP": 640,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 2
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 5,
+ "TIMESTAMP": 800,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 3
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 6,
+ "TIMESTAMP": 960,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 4
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 7,
+ "TIMESTAMP": 9440,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 5
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 8,
+ "TIMESTAMP": 9600,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 6
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 9,
+ "TIMESTAMP": 9760,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 7
+ },
+ {
+ "CSRC_LEN": 0,
+ "EXTENSION": 0,
+ "PADDING": 0,
+ "VERSION": 2,
+ "PAYLOAD_TYPE": "PCMA",
+ "MARKER": 0,
+ "SEQ": 10,
+ "TIMESTAMP": 9920,
+ "SSRC": 3535621694,
+ "PAYLOAD_LEN": 160,
+ "PAYLOAD_SEQ": 8
+ }
+]
diff --git a/test/decoders/rtp/rtp_test.cpp b/test/decoders/rtp/rtp_test.cpp
new file mode 100644
index 0000000..ddd4d80
--- /dev/null
+++ b/test/decoders/rtp/rtp_test.cpp
@@ -0,0 +1,325 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "stellar/stellar.h"
+#include "stellar/module_manager.h"
+#include "stellar/session.h"
+#include "stellar/utils.h"
+#include "stellar/rtp.h"
+
+#include <gtest/gtest.h>
+#include "cjson/cJSON.h"
+
+#define RTP_TEST_MODULE_NAME "RTP_TEST_MODULE"
+#define RTP_TEST_RESULT_EXPECT_ENV "RTP_TEST_RESULT_EXPECT"
+#define RTP_TEST_STELLAR_CONFIG_FILE "RTP_TEST_STELLAR_CONFIG"
+
+struct rtp_test_result {
+ cJSON *test_json;
+ cJSON *expect_json;
+ int count;
+};
+
+struct rtp_test_module_ctx {
+ struct rtp_test_result *result;
+
+ // depends a single session pcap
+ int rtp_count;
+};
+
+static void rtp_test_result_add(struct rtp_test_result *result, cJSON *json)
+{
+ cJSON_AddItemToArray(result->test_json, json);
+}
+
+static void rtp_test_result_compare(struct rtp_test_result *result)
+{
+ EXPECT_TRUE(result->expect_json != NULL && result->test_json != NULL);
+
+ int i;
+ int test_result_count, expect_result_count;
+ char *test_str, *expect_str;
+ cJSON *tmp_test, *tmp_expect;
+
+ /*
+ expect_str = cJSON_Print(result->expect_json);
+ test_str = cJSON_Print(result->test_json);
+ printf("LOAD Raw:\n%s\n", expect_str);
+ printf("TEST Raw:\n%s\n", test_str);
+ */
+
+ test_result_count = cJSON_GetArraySize(result->test_json);
+ expect_result_count = cJSON_GetArraySize(result->expect_json);
+
+ EXPECT_EQ(test_result_count, expect_result_count);
+
+ for (i = 0; i < MIN(test_result_count, expect_result_count); i++) {
+ tmp_test = cJSON_GetArrayItem(result->test_json, i);
+ tmp_expect = cJSON_GetArrayItem(result->expect_json, i);
+ expect_str = cJSON_Print(tmp_expect);
+ test_str = cJSON_Print(tmp_test);
+
+ EXPECT_STREQ(expect_str, test_str);
+ free(expect_str);
+ free(test_str);
+ break;
+ }
+}
+
+static void rtp_test_result_exit(struct rtp_test_result *result)
+{
+ if (result->expect_json) {
+ cJSON_Delete(result->expect_json);
+ }
+ if (result->test_json) {
+ cJSON_Delete(result->test_json);
+ }
+}
+
+static struct rtp_test_result * rtp_test_result_init(void)
+{
+ long filesize;
+ char *buffer;
+ FILE *file;
+ struct rtp_test_result *result;
+
+ result = (struct rtp_test_result *)calloc(1, sizeof(struct rtp_test_result));
+
+ file = fopen(getenv(RTP_TEST_RESULT_EXPECT_ENV), "rb");
+ if (file) {
+ fseek(file, 0, SEEK_END);
+ filesize = ftell(file);
+ rewind(file);
+ buffer = (char *)calloc(filesize + 1, 1);
+ fread(buffer, 1, filesize, file);
+
+ result->expect_json = cJSON_Parse(buffer);
+
+ free(buffer);
+ fclose(file);
+ }
+
+ result->test_json = cJSON_CreateArray();
+
+ return result;
+}
+
+static const char* rtp_test_rtp_payload_to_string(unsigned char payload_type)
+{
+ switch (payload_type) {
+ case PT_PCMU:
+ return "PCMU";
+ case PT_1016:
+ return "1016";
+ case PT_G721:
+ return "G721";
+ case PT_GSM:
+ return "GSM";
+ case PT_G723:
+ return "G723";
+ case PT_DVI4_8000:
+ return "DVI4_8000";
+ case PT_DVI4_16000:
+ return "DVI4_16000";
+ case PT_LPC:
+ return "LPC";
+ case PT_PCMA:
+ return "PCMA";
+ case PT_G722:
+ return "G722";
+ case PT_L16_STEREO:
+ return "L16_STEREO";
+ case PT_L16_MONO:
+ return "L16_MONO";
+ case PT_QCELP:
+ return "QCELP";
+ case PT_CN:
+ return "CN";
+ case PT_MPA:
+ return "MPA";
+ case PT_G728:
+ return "G728";
+ case PT_DVI4_11025:
+ return "DVI4_11025";
+ case PT_DVI4_22050:
+ return "DVI4_22050";
+ case PT_G729:
+ return "G729";
+ case PT_CN_OLD:
+ return "CN_OLD";
+ case PT_CELB:
+ return "CELB";
+ case PT_JPEG:
+ return "JPEG";
+ case PT_NV:
+ return "NV";
+ case PT_H261:
+ return "H261";
+ case PT_MPV:
+ return "MPV";
+ case PT_MP2T:
+ return "MP2T";
+ case PT_H263:
+ return "H263";
+ case PT_UNDF_96:
+ return "UNDF_96";
+ case PT_UNDF_97:
+ return "UNDF_97";
+ case PT_UNDF_98:
+ return "UNDF_98";
+ case PT_UNDF_99:
+ return "UNDF_99";
+ case PT_UNDF_100:
+ return "UNDF_100";
+ case PT_UNDF_101:
+ return "UNDF_101";
+ case PT_UNDF_102:
+ return "UNDF_102";
+ case PT_UNDF_103:
+ return "UNDF_103";
+ case PT_UNDF_104:
+ return "UNDF_104";
+ case PT_UNDF_105:
+ return "UNDF_105";
+ case PT_UNDF_106:
+ return "UNDF_106";
+ case PT_UNDF_107:
+ return "UNDF_107";
+ case PT_UNDF_108:
+ return "UNDF_108";
+ case PT_UNDF_109:
+ return "UNDF_109";
+ case PT_UNDF_110:
+ return "UNDF_110";
+ case PT_UNDF_111:
+ return "UNDF_111";
+ case PT_UNDF_112:
+ return "UNDF_112";
+ case PT_UNDF_113:
+ return "UNDF_113";
+ case PT_UNDF_114:
+ return "UNDF_114";
+ case PT_UNDF_115:
+ return "UNDF_115";
+ case PT_UNDF_116:
+ return "UNDF_116";
+ case PT_UNDF_117:
+ return "UNDF_117";
+ case PT_UNDF_118:
+ return "UNDF_118";
+ case PT_UNDF_119:
+ return "UNDF_119";
+ case PT_UNDF_120:
+ return "UNDF_120";
+ case PT_UNDF_121:
+ return "UNDF_121";
+ case PT_UNDF_122:
+ return "UNDF_122";
+ case PT_UNDF_123:
+ return "UNDF_123";
+ case PT_UNDF_124:
+ return "UNDF_124";
+ case PT_UNDF_125:
+ return "UNDF_125";
+ case PT_UNDF_126:
+ return "UNDF_126";
+ case PT_UNDF_127:
+ return "UNDF_127";
+ default:
+ return "UNKNOWN";
+ }
+ return NULL;
+}
+
+static void rtp_test_rtp_callback(struct session *sess, const struct rtp_header *hdr, const char *payload, size_t payload_len, void *arg)
+{
+ (void)(sess);
+ (void)(payload);
+ struct rtp_test_module_ctx *mod_ctx = (struct rtp_test_module_ctx *)arg;
+
+ cJSON *json = cJSON_CreateObject();
+
+ cJSON_AddNumberToObject(json, "CSRC_LEN", hdr->csrc_len);
+ cJSON_AddNumberToObject(json, "EXTENSION", hdr->extension);
+ cJSON_AddNumberToObject(json, "PADDING", hdr->padding);
+ cJSON_AddNumberToObject(json, "VERSION", hdr->version);
+ cJSON_AddStringToObject(json, "PAYLOAD_TYPE", rtp_test_rtp_payload_to_string(hdr->payload_type));
+ cJSON_AddNumberToObject(json, "MARKER", hdr->marker);
+ cJSON_AddNumberToObject(json, "SEQ", ntohs(hdr->seq));
+ cJSON_AddNumberToObject(json, "TIMESTAMP", ntohl(hdr->timestamp));
+ cJSON_AddNumberToObject(json, "SSRC", ntohl(hdr->ssrc));
+
+ cJSON_AddNumberToObject(json, "PAYLOAD_LEN", payload_len);
+ cJSON_AddNumberToObject(json, "PAYLOAD_SEQ", mod_ctx->rtp_count++);
+
+ rtp_test_result_add(mod_ctx->result, json);
+ return;
+}
+
+extern "C" void rtp_test_exit(struct stellar_module_manager *mod_mgr, struct stellar_module *mod)
+{
+ struct rtp_test_module_ctx *mod_ctx;
+
+ if (mod_mgr && mod) {
+ mod_ctx = (struct rtp_test_module_ctx *)stellar_module_get_ctx(mod);
+ if (mod_ctx) {
+ rtp_test_result_compare(mod_ctx->result);
+ rtp_test_result_exit(mod_ctx->result);
+ free(mod_ctx);
+ }
+ stellar_module_free(mod);
+ }
+}
+
+extern "C" struct stellar_module *rtp_test_init(struct stellar_module_manager *mod_mgr)
+{
+ int ret;
+ struct stellar_module *mod;
+ struct rtp_test_module_ctx *mod_ctx;
+ struct session_manager *sess_mgr;
+
+ mod_ctx = (struct rtp_test_module_ctx *)calloc(1, sizeof(struct rtp_test_module_ctx));
+ mod = stellar_module_new(RTP_TEST_MODULE_NAME, mod_ctx);
+ sess_mgr = stellar_module_get_session_manager(mod_mgr);
+
+ if (mod_mgr == NULL || sess_mgr == NULL) {
+ goto exit;
+ }
+
+ mod_ctx->result = rtp_test_result_init();
+ if (mod_ctx->result == NULL) {
+ goto exit;
+ }
+
+ ret = rtp_subscribe(mod_mgr, rtp_test_rtp_callback, mod_ctx);
+ if (ret < 0) {
+ goto exit;
+ }
+
+ return mod;
+exit:
+ printf("rtp_test module init failed!\n");
+ rtp_test_exit(mod_mgr, mod);
+ return NULL;
+}
+
+TEST(rtp, rtp_module)
+{
+ struct stellar *st;
+
+ setenv(RTP_TEST_RESULT_EXPECT_ENV, "./rtp_result.json", 0);
+
+ st = stellar_new(getenv(RTP_TEST_STELLAR_CONFIG_FILE));
+
+ stellar_run(st);
+ stellar_free(st);
+}
+
+int main(int argc, char ** argv)
+{
+ setenv(RTP_TEST_STELLAR_CONFIG_FILE, "./stellar.toml", 0);
+
+ ::testing::InitGoogleTest(&argc, argv);
+ return RUN_ALL_TESTS();
+}
+
generated by cgit v1.2.3 (git 2.39.1) at 2025-12-30 18:17:24 +0000