diff options
Diffstat (limited to 'test/decoders/rtp/rtp_test.cpp')
| -rw-r--r-- | test/decoders/rtp/rtp_test.cpp | 325 |
1 files changed, 325 insertions, 0 deletions
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(); +} + |