123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726 |
- /*
- * Asterisk -- An open source telephony toolkit.
- *
- * Copyright (C) 1999 - 2014, Digium, Inc.
- *
- * Alexandr Dubovikov <alexandr.dubovikov@sipcapture.org>
- * Matt Jordan <mjordan@digium.com>
- *
- * See http://www.asterisk.org for more information about
- * the Asterisk project. Please do not directly contact
- * any of the maintainers of this project for assistance;
- * the project provides a web site, mailing lists and IRC
- * channels for your use.
- *
- * This program is free software, distributed under the terms of
- * the GNU General Public License Version 2. See the LICENSE file
- * at the top of the source tree.
- */
- /*!
- * \file
- * \brief Routines for integration with Homer using HEPv3
- *
- * \author Alexandr Dubovikov <alexandr.dubovikov@sipcapture.org>
- * \author Matt Jordan <mjordan@digium.com>
- *
- */
- /*!
- * \li \ref res_hep.c uses the configuration file \ref hep.conf
- * \addtogroup configuration_file Configuration Files
- */
- /*!
- * \page hep.conf hep.conf
- * \verbinclude hep.conf.sample
- */
- /*** MODULEINFO
- <support_level>extended</support_level>
- ***/
- /*** DOCUMENTATION
- <configInfo name="res_hep" language="en_US">
- <synopsis>Resource for integration with Homer using HEPv3</synopsis>
- <configFile name="hep.conf">
- <configObject name="general">
- <synopsis>General settings.</synopsis>
- <description><para>
- The <emphasis>general</emphasis> settings section contains information
- to configure Asterisk as a Homer capture agent.
- </para>
- </description>
- <configOption name="enabled" default="yes">
- <synopsis>Enable or disable packet capturing.</synopsis>
- <description>
- <enumlist>
- <enum name="no" />
- <enum name="yes" />
- </enumlist>
- </description>
- </configOption>
- <configOption name="uuid_type" default="call-id">
- <synopsis>The preferred type of UUID to pass to Homer.</synopsis>
- <description>
- <enumlist>
- <enum name="call-id"><para>Use the PJSIP Call-Id</para></enum>
- <enum name="channel"><para>Use the Asterisk channel name</para></enum>
- </enumlist>
- </description>
- </configOption>
- <configOption name="capture_address">
- <synopsis>The address and port of the Homer server to send packets to.</synopsis>
- </configOption>
- <configOption name="capture_password">
- <synopsis>If set, the authentication password to send to Homer.</synopsis>
- </configOption>
- <configOption name="capture_id" default="0">
- <synopsis>The ID for this capture agent.</synopsis>
- </configOption>
- </configObject>
- </configFile>
- </configInfo>
- ***/
- #include "asterisk.h"
- ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
- #include "asterisk/module.h"
- #include "asterisk/astobj2.h"
- #include "asterisk/config_options.h"
- #include "asterisk/taskprocessor.h"
- #include "asterisk/res_hep.h"
- #include <netinet/ip.h>
- #include <netinet/tcp.h>
- #include <netinet/udp.h>
- #include <netinet/ip6.h>
- /*! Generic vendor ID. Used for HEPv3 standard packets */
- #define GENERIC_VENDOR_ID 0x0000
- /*! Asterisk vendor ID. Used for custom data to send to a capture node */
- #define ASTERISK_VENDOR_ID 0x0004
- /*! Chunk types from the HEPv3 Spec */
- enum hepv3_chunk_types {
- /*! THE IP PROTOCOL FAMILY */
- CHUNK_TYPE_IP_PROTOCOL_FAMILY = 0X0001,
- /*! THE IP PROTOCOL ID (UDP, TCP, ETC.) */
- CHUNK_TYPE_IP_PROTOCOL_ID = 0X0002,
- /*! IF IPV4, THE SOURCE ADDRESS */
- CHUNK_TYPE_IPV4_SRC_ADDR = 0X0003,
- /*! IF IPV4, THE DESTINATION ADDRESS */
- CHUNK_TYPE_IPV4_DST_ADDR = 0X0004,
- /*! IF IPV6, THE SOURCE ADDRESS */
- CHUNK_TYPE_IPV6_SRC_ADDR = 0X0005,
- /*! IF IPV6, THE DESTINATION ADDRESS */
- CHUNK_TYPE_IPV6_DST_ADDR = 0X0006,
- /*! THE SOURCE PORT */
- CHUNK_TYPE_SRC_PORT = 0X0007,
- /*! THE DESTINATION PORT */
- CHUNK_TYPE_DST_PORT = 0X0008,
- /*! THE CAPTURE TIME (SECONDS) */
- CHUNK_TYPE_TIMESTAMP_SEC = 0X0009,
- /*! THE CAPTURE TIME (MICROSECONDS) */
- CHUNK_TYPE_TIMESTAMP_USEC = 0X000A,
- /*! THE PROTOCOL PACKET TYPE. SEE /REF HEPV3_CAPTURE_TYPE */
- CHUNK_TYPE_PROTOCOL_TYPE = 0X000B,
- /*! OUR CAPTURE AGENT ID */
- CHUNK_TYPE_CAPTURE_AGENT_ID = 0X000C,
- /*! A KEEP ALIVE TIMER */
- CHUNK_TYPE_KEEP_ALIVE_TIMER = 0X000D,
- /*! THE \REF CAPTURE_PASSWORD IF DEFINED */
- CHUNK_TYPE_AUTH_KEY = 0X000E,
- /*! THE ONE AND ONLY PAYLOAD */
- CHUNK_TYPE_PAYLOAD = 0X000F,
- /*! THE ONE AND ONLY (ZIPPED) PAYLOAD */
- CHUNK_TYPE_PAYLOAD_ZIP = 0X0010,
- /*! THE UUID FOR THIS PACKET */
- CHUNK_TYPE_UUID = 0X0011,
- };
- #define INITIALIZE_GENERIC_HEP_IDS(hep_chunk, type) do { \
- (hep_chunk)->vendor_id = htons(GENERIC_VENDOR_ID); \
- (hep_chunk)->type_id = htons((type)); \
- } while (0)
- #define INITIALIZE_GENERIC_HEP_IDS_VAR(hep_chunk, type, len) do { \
- INITIALIZE_GENERIC_HEP_IDS((hep_chunk), (type)); \
- (hep_chunk)->length = htons(sizeof(*(hep_chunk)) + len); \
- } while (0)
- #define INITIALIZE_GENERIC_HEP_CHUNK(hep_item, type) do { \
- INITIALIZE_GENERIC_HEP_IDS(&(hep_item)->chunk, (type)); \
- (hep_item)->chunk.length = htons(sizeof(*(hep_item))); \
- } while (0)
- #define INITIALIZE_GENERIC_HEP_CHUNK_DATA(hep_item, type, value) do { \
- INITIALIZE_GENERIC_HEP_CHUNK((hep_item), (type)); \
- (hep_item)->data = (value); \
- } while (0)
- /*
- * HEPv3 Types.
- * Note that the content in these is stored in network byte order.
- */
- struct hep_chunk {
- u_int16_t vendor_id;
- u_int16_t type_id;
- u_int16_t length;
- } __attribute__((packed));
- struct hep_chunk_uint8 {
- struct hep_chunk chunk;
- u_int8_t data;
- } __attribute__((packed));
- struct hep_chunk_uint16 {
- struct hep_chunk chunk;
- u_int16_t data;
- } __attribute__((packed));
- struct hep_chunk_uint32 {
- struct hep_chunk chunk;
- u_int32_t data;
- } __attribute__((packed));
- struct hep_chunk_ip4 {
- struct hep_chunk chunk;
- struct in_addr data;
- } __attribute__((packed));
- struct hep_chunk_ip6 {
- struct hep_chunk chunk;
- struct in6_addr data;
- } __attribute__((packed));
- struct hep_ctrl {
- char id[4];
- u_int16_t length;
- } __attribute__((packed));
- /* HEP structures */
- struct hep_generic {
- struct hep_ctrl header;
- struct hep_chunk_uint8 ip_family;
- struct hep_chunk_uint8 ip_proto;
- struct hep_chunk_uint16 src_port;
- struct hep_chunk_uint16 dst_port;
- struct hep_chunk_uint32 time_sec;
- struct hep_chunk_uint32 time_usec;
- struct hep_chunk_uint8 proto_t;
- struct hep_chunk_uint32 capt_id;
- } __attribute__((packed));
- /*! \brief Global configuration for the module */
- struct hepv3_global_config {
- unsigned int enabled; /*!< Whether or not sending is enabled */
- unsigned int capture_id; /*!< Capture ID for this agent */
- enum hep_uuid_type uuid_type; /*!< The preferred type of the UUID */
- AST_DECLARE_STRING_FIELDS(
- AST_STRING_FIELD(capture_address); /*!< Address to send to */
- AST_STRING_FIELD(capture_password); /*!< Password for Homer server */
- );
- };
- /*! \brief The actual module config */
- struct module_config {
- struct hepv3_global_config *general; /*!< The general config settings */
- };
- /*! \brief Run-time data derived from \ref hepv3_global_config */
- struct hepv3_runtime_data {
- struct ast_sockaddr remote_addr; /*!< The address to send to */
- int sockfd; /*!< The socket file descriptor */
- };
- static struct aco_type global_option = {
- .type = ACO_GLOBAL,
- .name = "general",
- .item_offset = offsetof(struct module_config, general),
- .category_match = ACO_WHITELIST_EXACT,
- .category = "general",
- };
- struct aco_type *global_options[] = ACO_TYPES(&global_option);
- struct aco_file hepv3_conf = {
- .filename = "hep.conf",
- .types = ACO_TYPES(&global_option),
- };
- /*! \brief The module configuration container */
- static AO2_GLOBAL_OBJ_STATIC(global_config);
- /*! \brief Current module data */
- static AO2_GLOBAL_OBJ_STATIC(global_data);
- static struct ast_taskprocessor *hep_queue_tp;
- static void *module_config_alloc(void);
- static int hepv3_config_pre_apply(void);
- static void hepv3_config_post_apply(void);
- /*! \brief Register information about the configs being processed by this module */
- CONFIG_INFO_STANDARD(cfg_info, global_config, module_config_alloc,
- .files = ACO_FILES(&hepv3_conf),
- .pre_apply_config = hepv3_config_pre_apply,
- .post_apply_config = hepv3_config_post_apply,
- );
- static void hepv3_config_dtor(void *obj)
- {
- struct hepv3_global_config *config = obj;
- ast_string_field_free_memory(config);
- }
- /*! \brief HEPv3 configuration object allocation */
- static void *hepv3_config_alloc(void)
- {
- struct hepv3_global_config *config;
- config = ao2_alloc(sizeof(*config), hepv3_config_dtor);
- if (!config || ast_string_field_init(config, 32)) {
- return NULL;
- }
- return config;
- }
- /*! \brief Configuration object destructor */
- static void module_config_dtor(void *obj)
- {
- struct module_config *config = obj;
- if (config->general) {
- ao2_ref(config->general, -1);
- }
- }
- /*! \brief Module config constructor */
- static void *module_config_alloc(void)
- {
- struct module_config *config;
- config = ao2_alloc(sizeof(*config), module_config_dtor);
- if (!config) {
- return NULL;
- }
- config->general = hepv3_config_alloc();
- if (!config->general) {
- ao2_ref(config, -1);
- config = NULL;
- }
- return config;
- }
- /*! \brief Handler for the uuid_type attribute */
- static int uuid_type_handler(const struct aco_option *opt, struct ast_variable *var, void *obj)
- {
- struct hepv3_global_config *global_config = obj;
- if (strcasecmp(var->name, "uuid_type")) {
- return -1;
- }
- if (!strcasecmp(var->value, "channel")) {
- global_config->uuid_type = HEP_UUID_TYPE_CHANNEL;
- } else if (!strcasecmp(var->value, "call-id")) {
- global_config->uuid_type = HEP_UUID_TYPE_CALL_ID;
- } else {
- return -1;
- }
- return 0;
- }
- /*! \brief HEPv3 run-time data destructor */
- static void hepv3_data_dtor(void *obj)
- {
- struct hepv3_runtime_data *data = obj;
- if (data->sockfd > -1) {
- close(data->sockfd);
- data->sockfd = -1;
- }
- }
- /*! \brief Pulls first resolved address and returns it */
- static int ast_sockaddr_resolve_first_af(struct ast_sockaddr *addr,
- const char* name, int flag, int family)
- {
- struct ast_sockaddr *addrs;
- int addrs_cnt;
- addrs_cnt = ast_sockaddr_resolve(&addrs, name, flag, family);
- if (addrs_cnt <= 0) {
- return 1;
- }
- if (addrs_cnt > 1) {
- ast_debug(1, "Multiple addresses resolving %s, using the first one only\n", name);
- }
- ast_sockaddr_copy(addr, &addrs[0]);
- ast_free(addrs);
- return 0;
- }
- /*! \brief Allocate the HEPv3 run-time data */
- static struct hepv3_runtime_data *hepv3_data_alloc(struct hepv3_global_config *config)
- {
- struct hepv3_runtime_data *data;
- data = ao2_alloc(sizeof(*data), hepv3_data_dtor);
- if (!data) {
- return NULL;
- }
- data->sockfd = -1;
- if (ast_sockaddr_resolve_first_af(&data->remote_addr, config->capture_address, PARSE_PORT_REQUIRE, AST_AF_UNSPEC)) {
- ast_log(AST_LOG_WARNING, "Failed to create address from %s\n", config->capture_address);
- ao2_ref(data, -1);
- return NULL;
- }
- data->sockfd = socket(ast_sockaddr_is_ipv6(&data->remote_addr) ? AF_INET6 : AF_INET, SOCK_DGRAM, 0);
- if (data->sockfd < 0) {
- ast_log(AST_LOG_WARNING, "Failed to create socket for address %s: %s\n",
- config->capture_address, strerror(errno));
- ao2_ref(data, -1);
- return NULL;
- }
- return data;
- }
- /*! \brief Destructor for a \ref hepv3_capture_info object */
- static void capture_info_dtor(void *obj)
- {
- struct hepv3_capture_info *info = obj;
- ast_free(info->uuid);
- ast_free(info->payload);
- }
- enum hep_uuid_type hepv3_get_uuid_type(void)
- {
- RAII_VAR(struct module_config *, config, ao2_global_obj_ref(global_config), ao2_cleanup);
- if (!config) {
- /* Well, that's unfortunate. Return something. */
- return HEP_UUID_TYPE_CALL_ID;
- }
- return config->general->uuid_type;
- }
- int hepv3_is_loaded(void)
- {
- RAII_VAR(struct module_config *, config, ao2_global_obj_ref(global_config), ao2_cleanup);
- return config && config->general->enabled;
- }
- struct hepv3_capture_info *hepv3_create_capture_info(const void *payload, size_t len)
- {
- struct hepv3_capture_info *info;
- info = ao2_alloc(sizeof(*info), capture_info_dtor);
- if (!info) {
- return NULL;
- }
- info->payload = ast_malloc(len);
- if (!info->payload) {
- ao2_ref(info, -1);
- return NULL;
- }
- memcpy(info->payload, payload, len);
- info->len = len;
- /* Set a reasonable default */
- info->protocol_id = IPPROTO_UDP;
- return info;
- }
- /*! \brief Callback function for the \ref hep_queue_tp taskprocessor */
- static int hep_queue_cb(void *data)
- {
- RAII_VAR(struct module_config *, config, ao2_global_obj_ref(global_config), ao2_cleanup);
- RAII_VAR(struct hepv3_runtime_data *, hepv3_data, ao2_global_obj_ref(global_data), ao2_cleanup);
- RAII_VAR(struct hepv3_capture_info *, capture_info, data, ao2_cleanup);
- struct hep_generic hg_pkt;
- unsigned int packet_len = 0, sock_buffer_len;
- struct hep_chunk_ip4 ipv4_src, ipv4_dst;
- struct hep_chunk_ip6 ipv6_src, ipv6_dst;
- struct hep_chunk auth_key, payload, uuid;
- void *sock_buffer;
- int res;
- if (!capture_info || !config || !hepv3_data) {
- return 0;
- }
- if (ast_sockaddr_is_ipv4(&capture_info->src_addr) != ast_sockaddr_is_ipv4(&capture_info->dst_addr)) {
- ast_log(AST_LOG_NOTICE, "Unable to send packet: Address Family mismatch between source/destination\n");
- return -1;
- }
- packet_len = sizeof(hg_pkt);
- /* Build HEPv3 header, capture info, and calculate the total packet size */
- memcpy(hg_pkt.header.id, "\x48\x45\x50\x33", 4);
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.ip_proto, CHUNK_TYPE_IP_PROTOCOL_ID, capture_info->protocol_id);
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.src_port, CHUNK_TYPE_SRC_PORT, htons(ast_sockaddr_port(&capture_info->src_addr)));
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.dst_port, CHUNK_TYPE_DST_PORT, htons(ast_sockaddr_port(&capture_info->dst_addr)));
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.time_sec, CHUNK_TYPE_TIMESTAMP_SEC, htonl(capture_info->capture_time.tv_sec));
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.time_usec, CHUNK_TYPE_TIMESTAMP_USEC, htonl(capture_info->capture_time.tv_usec));
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.proto_t, CHUNK_TYPE_PROTOCOL_TYPE, capture_info->capture_type);
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.capt_id, CHUNK_TYPE_CAPTURE_AGENT_ID, htonl(config->general->capture_id));
- if (ast_sockaddr_is_ipv4(&capture_info->src_addr)) {
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.ip_family,
- CHUNK_TYPE_IP_PROTOCOL_FAMILY, AF_INET);
- INITIALIZE_GENERIC_HEP_CHUNK(&ipv4_src, CHUNK_TYPE_IPV4_SRC_ADDR);
- inet_pton(AF_INET, ast_sockaddr_stringify_addr(&capture_info->src_addr), &ipv4_src.data);
- INITIALIZE_GENERIC_HEP_CHUNK(&ipv4_dst, CHUNK_TYPE_IPV4_DST_ADDR);
- inet_pton(AF_INET, ast_sockaddr_stringify_addr(&capture_info->dst_addr), &ipv4_dst.data);
- packet_len += (sizeof(ipv4_src) + sizeof(ipv4_dst));
- } else {
- INITIALIZE_GENERIC_HEP_CHUNK_DATA(&hg_pkt.ip_family,
- CHUNK_TYPE_IP_PROTOCOL_FAMILY, AF_INET6);
- INITIALIZE_GENERIC_HEP_CHUNK(&ipv6_src, CHUNK_TYPE_IPV6_SRC_ADDR);
- inet_pton(AF_INET6, ast_sockaddr_stringify_addr(&capture_info->src_addr), &ipv6_src.data);
- INITIALIZE_GENERIC_HEP_CHUNK(&ipv6_dst, CHUNK_TYPE_IPV6_DST_ADDR);
- inet_pton(AF_INET6, ast_sockaddr_stringify_addr(&capture_info->dst_addr), &ipv6_dst.data);
- packet_len += (sizeof(ipv6_src) + sizeof(ipv6_dst));
- }
- if (!ast_strlen_zero(config->general->capture_password)) {
- INITIALIZE_GENERIC_HEP_IDS_VAR(&auth_key, CHUNK_TYPE_AUTH_KEY, strlen(config->general->capture_password));
- packet_len += (sizeof(auth_key) + strlen(config->general->capture_password));
- }
- INITIALIZE_GENERIC_HEP_IDS_VAR(&uuid, CHUNK_TYPE_UUID, strlen(capture_info->uuid));
- packet_len += (sizeof(uuid) + strlen(capture_info->uuid));
- INITIALIZE_GENERIC_HEP_IDS_VAR(&payload,
- capture_info->zipped ? CHUNK_TYPE_PAYLOAD_ZIP : CHUNK_TYPE_PAYLOAD, capture_info->len);
- packet_len += (sizeof(payload) + capture_info->len);
- hg_pkt.header.length = htons(packet_len);
- /* Build the buffer to send */
- sock_buffer = ast_malloc(packet_len);
- if (!sock_buffer) {
- return -1;
- }
- /* Copy in the header */
- memcpy(sock_buffer, &hg_pkt, sizeof(hg_pkt));
- sock_buffer_len = sizeof(hg_pkt);
- /* Addresses */
- if (ast_sockaddr_is_ipv4(&capture_info->src_addr)) {
- memcpy(sock_buffer + sock_buffer_len, &ipv4_src, sizeof(ipv4_src));
- sock_buffer_len += sizeof(ipv4_src);
- memcpy(sock_buffer + sock_buffer_len, &ipv4_dst, sizeof(ipv4_dst));
- sock_buffer_len += sizeof(ipv4_dst);
- } else {
- memcpy(sock_buffer + sock_buffer_len, &ipv6_src, sizeof(ipv6_src));
- sock_buffer_len += sizeof(ipv6_src);
- memcpy(sock_buffer + sock_buffer_len, &ipv6_dst, sizeof(ipv6_dst));
- sock_buffer_len += sizeof(ipv6_dst);
- }
- /* Auth Key */
- if (!ast_strlen_zero(config->general->capture_password)) {
- memcpy(sock_buffer + sock_buffer_len, &auth_key, sizeof(auth_key));
- sock_buffer_len += sizeof(auth_key);
- memcpy(sock_buffer + sock_buffer_len, config->general->capture_password, strlen(config->general->capture_password));
- sock_buffer_len += strlen(config->general->capture_password);
- }
- /* UUID */
- memcpy(sock_buffer + sock_buffer_len, &uuid, sizeof(uuid));
- sock_buffer_len += sizeof(uuid);
- memcpy(sock_buffer + sock_buffer_len, capture_info->uuid, strlen(capture_info->uuid));
- sock_buffer_len += strlen(capture_info->uuid);
- /* Packet! */
- memcpy(sock_buffer + sock_buffer_len, &payload, sizeof(payload));
- sock_buffer_len += sizeof(payload);
- memcpy(sock_buffer + sock_buffer_len, capture_info->payload, capture_info->len);
- sock_buffer_len += capture_info->len;
- ast_assert(sock_buffer_len == packet_len);
- res = ast_sendto(hepv3_data->sockfd, sock_buffer, sock_buffer_len, 0, &hepv3_data->remote_addr);
- if (res < 0) {
- ast_log(AST_LOG_ERROR, "Error [%d] while sending packet to HEPv3 server: %s\n",
- errno, strerror(errno));
- } else if (res != sock_buffer_len) {
- ast_log(AST_LOG_WARNING, "Failed to send complete packet to HEPv3 server: %d of %u sent\n",
- res, sock_buffer_len);
- res = -1;
- }
- ast_free(sock_buffer);
- return res;
- }
- int hepv3_send_packet(struct hepv3_capture_info *capture_info)
- {
- RAII_VAR(struct module_config *, config, ao2_global_obj_ref(global_config), ao2_cleanup);
- int res;
- if (!config || !config->general->enabled) {
- ao2_ref(capture_info, -1);
- return 0;
- }
- res = ast_taskprocessor_push(hep_queue_tp, hep_queue_cb, capture_info);
- if (res == -1) {
- ao2_ref(capture_info, -1);
- }
- return res;
- }
- /*!
- * \brief Pre-apply callback for the config framework.
- *
- * This validates that required fields exist and are populated.
- */
- static int hepv3_config_pre_apply(void)
- {
- struct module_config *config = aco_pending_config(&cfg_info);
- if (!config->general->enabled) {
- /* If we're not enabled, we don't care about anything else */
- return 0;
- }
- if (ast_strlen_zero(config->general->capture_address)) {
- ast_log(AST_LOG_ERROR, "Missing required configuration option 'capture_address'\n");
- return -1;
- }
- return 0;
- }
- /*!
- * \brief Post-apply callback for the config framework.
- *
- * This will create the run-time information from the supplied
- * configuration.
- */
- static void hepv3_config_post_apply(void)
- {
- RAII_VAR(struct module_config *, mod_cfg, ao2_global_obj_ref(global_config), ao2_cleanup);
- struct hepv3_runtime_data *data;
- data = hepv3_data_alloc(mod_cfg->general);
- if (!data) {
- return;
- }
- ao2_global_obj_replace_unref(global_data, data);
- ao2_ref(data, -1);
- }
- /*!
- * \brief Reload the module
- */
- static int reload_module(void)
- {
- if (aco_process_config(&cfg_info, 1) == ACO_PROCESS_ERROR) {
- return -1;
- }
- return 0;
- }
- /*!
- * \brief Unload the module
- */
- static int unload_module(void)
- {
- hep_queue_tp = ast_taskprocessor_unreference(hep_queue_tp);
- ao2_global_obj_release(global_config);
- ao2_global_obj_release(global_data);
- aco_info_destroy(&cfg_info);
- return 0;
- }
- /*!
- * \brief Load the module
- */
- static int load_module(void)
- {
- if (aco_info_init(&cfg_info)) {
- goto error;
- }
- hep_queue_tp = ast_taskprocessor_get("hep_queue_tp", TPS_REF_DEFAULT);
- if (!hep_queue_tp) {
- goto error;
- }
- aco_option_register(&cfg_info, "enabled", ACO_EXACT, global_options, "yes", OPT_BOOL_T, 1, FLDSET(struct hepv3_global_config, enabled));
- aco_option_register(&cfg_info, "capture_address", ACO_EXACT, global_options, "", OPT_STRINGFIELD_T, 1, STRFLDSET(struct hepv3_global_config, capture_address));
- aco_option_register(&cfg_info, "capture_password", ACO_EXACT, global_options, "", OPT_STRINGFIELD_T, 0, STRFLDSET(struct hepv3_global_config, capture_password));
- aco_option_register(&cfg_info, "capture_id", ACO_EXACT, global_options, "0", OPT_UINT_T, 0, STRFLDSET(struct hepv3_global_config, capture_id));
- aco_option_register_custom(&cfg_info, "uuid_type", ACO_EXACT, global_options, "call-id", uuid_type_handler, 0);
- if (aco_process_config(&cfg_info, 0) == ACO_PROCESS_ERROR) {
- goto error;
- }
- return AST_MODULE_LOAD_SUCCESS;
- error:
- aco_info_destroy(&cfg_info);
- return AST_MODULE_LOAD_DECLINE;
- }
- AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_GLOBAL_SYMBOLS | AST_MODFLAG_LOAD_ORDER, "HEPv3 API",
- .support_level = AST_MODULE_SUPPORT_EXTENDED,
- .load = load_module,
- .unload = unload_module,
- .reload = reload_module,
- .load_pri = AST_MODPRI_APP_DEPEND,
- );
|