| /* |
| * (C) 2015 by Sysmocom s.f.m.c. GmbH |
| * |
| * Author: Jacob Erlbeck <jerlbeck@sysmocom.de> |
| * |
| * All Rights Reserved |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License as published by |
| * the Free Software Foundation; either version 2 of the License, or |
| * (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License along |
| * with this program; if not, write to the Free Software Foundation, Inc., |
| * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| */ |
| |
| #include "config.h" |
| #if !defined(EMBEDDED) |
| |
| #include <osmocom/core/byteswap.h> |
| #include <osmocom/core/stats.h> |
| |
| #include <unistd.h> |
| #include <string.h> |
| #include <stdint.h> |
| #include <errno.h> |
| #include <stdio.h> |
| #include <sys/types.h> |
| |
| #ifdef HAVE_SYS_SOCKET_H |
| #include <sys/socket.h> |
| #include <netinet/in.h> |
| #include <arpa/inet.h> |
| #endif |
| |
| #include <osmocom/core/utils.h> |
| #include <osmocom/core/logging.h> |
| #include <osmocom/core/rate_ctr.h> |
| #include <osmocom/core/stat_item.h> |
| #include <osmocom/core/timer.h> |
| #include <osmocom/core/statistics.h> |
| #include <osmocom/core/msgb.h> |
| |
| #define STATS_DEFAULT_INTERVAL 5 /* secs */ |
| #define STATS_DEFAULT_BUFLEN 256 |
| |
| static LLIST_HEAD(osmo_stats_reporter_list); |
| static void *osmo_stats_ctx = NULL; |
| static int is_initialised = 0; |
| static int32_t current_stat_item_index = 0; |
| |
| static struct osmo_stats_config s_stats_config = { |
| .interval = STATS_DEFAULT_INTERVAL, |
| }; |
| struct osmo_stats_config *osmo_stats_config = &s_stats_config; |
| |
| static struct osmo_timer_list osmo_stats_timer; |
| |
| static int osmo_stats_reporter_log_send_counter(struct osmo_stats_reporter *srep, |
| const struct rate_ctr_group *ctrg, |
| const struct rate_ctr_desc *desc, |
| int64_t value, int64_t delta); |
| static int osmo_stats_reporter_log_send_item(struct osmo_stats_reporter *srep, |
| const struct osmo_stat_item_group *statg, |
| const struct osmo_stat_item_desc *desc, int64_t value); |
| |
| static int update_srep_config(struct osmo_stats_reporter *srep) |
| { |
| int rc = 0; |
| |
| if (srep->running) { |
| if (srep->close) |
| rc = srep->close(srep); |
| srep->running = 0; |
| } |
| |
| if (!srep->enabled) |
| return rc; |
| |
| if (srep->open) |
| rc = srep->open(srep); |
| else |
| rc = 0; |
| |
| if (rc < 0) |
| srep->enabled = 0; |
| else |
| srep->running = 1; |
| |
| srep->force_single_flush = 1; |
| |
| return rc; |
| } |
| |
| static void osmo_stats_timer_cb(void *data) |
| { |
| int interval = osmo_stats_config->interval; |
| |
| if (!llist_empty(&osmo_stats_reporter_list)) |
| osmo_stats_report(); |
| |
| osmo_timer_schedule(&osmo_stats_timer, interval, 0); |
| } |
| |
| static int start_timer() |
| { |
| if (!is_initialised) |
| return -ESRCH; |
| |
| osmo_timer_setup(&osmo_stats_timer, osmo_stats_timer_cb, NULL); |
| osmo_timer_schedule(&osmo_stats_timer, 0, 1); |
| |
| return 0; |
| } |
| |
| struct osmo_stats_reporter *osmo_stats_reporter_alloc(enum osmo_stats_reporter_type type, |
| const char *name) |
| { |
| struct osmo_stats_reporter *srep; |
| srep = talloc_zero(osmo_stats_ctx, struct osmo_stats_reporter); |
| OSMO_ASSERT(srep); |
| srep->type = type; |
| if (name) |
| srep->name = talloc_strdup(srep, name); |
| srep->fd = -1; |
| |
| llist_add(&srep->list, &osmo_stats_reporter_list); |
| |
| return srep; |
| } |
| |
| void osmo_stats_reporter_free(struct osmo_stats_reporter *srep) |
| { |
| osmo_stats_reporter_disable(srep); |
| llist_del(&srep->list); |
| talloc_free(srep); |
| } |
| |
| void osmo_stats_init(void *ctx) |
| { |
| osmo_stats_ctx = ctx; |
| osmo_stat_item_discard_all(¤t_stat_item_index); |
| |
| is_initialised = 1; |
| start_timer(); |
| } |
| |
| struct osmo_stats_reporter *osmo_stats_reporter_find(enum osmo_stats_reporter_type type, |
| const char *name) |
| { |
| struct osmo_stats_reporter *srep; |
| llist_for_each_entry(srep, &osmo_stats_reporter_list, list) { |
| if (srep->type != type) |
| continue; |
| if (srep->name != name) { |
| if (name == NULL || srep->name == NULL || |
| strcmp(name, srep->name) != 0) |
| continue; |
| } |
| return srep; |
| } |
| return NULL; |
| } |
| |
| #ifdef HAVE_SYS_SOCKET_H |
| |
| int osmo_stats_reporter_set_remote_addr(struct osmo_stats_reporter *srep, const char *addr) |
| { |
| int rc; |
| struct sockaddr_in *sock_addr = (struct sockaddr_in *)&srep->dest_addr; |
| struct in_addr inaddr; |
| |
| if (!srep->have_net_config) |
| return -ENOTSUP; |
| |
| OSMO_ASSERT(addr != NULL); |
| |
| rc = inet_pton(AF_INET, addr, &inaddr); |
| if (rc <= 0) |
| return -EINVAL; |
| |
| sock_addr->sin_addr = inaddr; |
| sock_addr->sin_family = AF_INET; |
| srep->dest_addr_len = sizeof(*sock_addr); |
| |
| talloc_free(srep->dest_addr_str); |
| srep->dest_addr_str = talloc_strdup(srep, addr); |
| |
| return update_srep_config(srep); |
| } |
| |
| int osmo_stats_reporter_set_remote_port(struct osmo_stats_reporter *srep, int port) |
| { |
| struct sockaddr_in *sock_addr = (struct sockaddr_in *)&srep->dest_addr; |
| |
| if (!srep->have_net_config) |
| return -ENOTSUP; |
| |
| srep->dest_port = port; |
| sock_addr->sin_port = osmo_htons(port); |
| |
| return update_srep_config(srep); |
| } |
| |
| int osmo_stats_reporter_set_local_addr(struct osmo_stats_reporter *srep, const char *addr) |
| { |
| int rc; |
| struct sockaddr_in *sock_addr = (struct sockaddr_in *)&srep->bind_addr; |
| struct in_addr inaddr; |
| |
| if (!srep->have_net_config) |
| return -ENOTSUP; |
| |
| if (addr) { |
| rc = inet_pton(AF_INET, addr, &inaddr); |
| if (rc <= 0) |
| return -EINVAL; |
| } else { |
| inaddr.s_addr = INADDR_ANY; |
| } |
| |
| sock_addr->sin_addr = inaddr; |
| sock_addr->sin_family = AF_INET; |
| srep->bind_addr_len = addr ? sizeof(*sock_addr) : 0; |
| |
| talloc_free(srep->bind_addr_str); |
| srep->bind_addr_str = addr ? talloc_strdup(srep, addr) : NULL; |
| |
| return update_srep_config(srep); |
| } |
| |
| int osmo_stats_reporter_set_mtu(struct osmo_stats_reporter *srep, int mtu) |
| { |
| if (!srep->have_net_config) |
| return -ENOTSUP; |
| |
| if (mtu < 0) |
| return -EINVAL; |
| |
| srep->mtu = mtu; |
| |
| return update_srep_config(srep); |
| } |
| #endif /* HAVE_SYS_SOCKETS_H */ |
| |
| int osmo_stats_reporter_set_max_class(struct osmo_stats_reporter *srep, |
| enum osmo_stats_class class_id) |
| { |
| if (class_id == OSMO_STATS_CLASS_UNKNOWN) |
| return -EINVAL; |
| |
| srep->max_class = class_id; |
| |
| return 0; |
| } |
| |
| int osmo_stats_set_interval(int interval) |
| { |
| if (interval <= 0) |
| return -EINVAL; |
| |
| osmo_stats_config->interval = interval; |
| if (is_initialised) |
| start_timer(); |
| |
| return 0; |
| } |
| |
| int osmo_stats_reporter_set_name_prefix(struct osmo_stats_reporter *srep, const char *prefix) |
| { |
| talloc_free(srep->name_prefix); |
| srep->name_prefix = prefix && strlen(prefix) > 0 ? |
| talloc_strdup(srep, prefix) : NULL; |
| |
| return update_srep_config(srep); |
| } |
| |
| int osmo_stats_reporter_enable(struct osmo_stats_reporter *srep) |
| { |
| srep->enabled = 1; |
| |
| return update_srep_config(srep); |
| } |
| |
| int osmo_stats_reporter_disable(struct osmo_stats_reporter *srep) |
| { |
| srep->enabled = 0; |
| |
| return update_srep_config(srep); |
| } |
| |
| /*** i/o helper functions ***/ |
| |
| #ifdef HAVE_SYS_SOCKET_H |
| |
| int osmo_stats_reporter_udp_open(struct osmo_stats_reporter *srep) |
| { |
| int sock; |
| int rc; |
| int buffer_size = STATS_DEFAULT_BUFLEN; |
| |
| if (srep->fd != -1 && srep->close) |
| srep->close(srep); |
| |
| sock = socket(AF_INET, SOCK_DGRAM, 0); |
| if (sock == -1) |
| return -errno; |
| |
| #if defined(__APPLE__) && !defined(MSG_NOSIGNAL) |
| { |
| static int val = 1; |
| |
| rc = setsockopt(sock, SOL_SOCKET, SO_NOSIGPIPE, (void*)&val, sizeof(val)); |
| goto failed; |
| } |
| #endif |
| if (srep->bind_addr_len > 0) { |
| rc = bind(sock, &srep->bind_addr, srep->bind_addr_len); |
| if (rc == -1) |
| goto failed; |
| } |
| |
| srep->fd = sock; |
| |
| if (srep->mtu > 0) { |
| buffer_size = srep->mtu - 20 /* IP */ - 8 /* UDP */; |
| srep->agg_enabled = 1; |
| } |
| |
| srep->buffer = msgb_alloc(buffer_size, "stats buffer"); |
| |
| return 0; |
| |
| failed: |
| rc = -errno; |
| close(sock); |
| |
| return rc; |
| } |
| |
| int osmo_stats_reporter_udp_close(struct osmo_stats_reporter *srep) |
| { |
| int rc; |
| if (srep->fd == -1) |
| return -EBADF; |
| |
| osmo_stats_reporter_send_buffer(srep); |
| |
| rc = close(srep->fd); |
| srep->fd = -1; |
| msgb_free(srep->buffer); |
| srep->buffer = NULL; |
| return rc == -1 ? -errno : 0; |
| } |
| |
| int osmo_stats_reporter_send(struct osmo_stats_reporter *srep, const char *data, |
| int data_len) |
| { |
| int rc; |
| |
| rc = sendto(srep->fd, data, data_len, |
| #ifdef MSG_NOSIGNAL |
| MSG_NOSIGNAL | |
| #endif |
| MSG_DONTWAIT, |
| &srep->dest_addr, srep->dest_addr_len); |
| |
| if (rc == -1) |
| rc = -errno; |
| |
| return rc; |
| } |
| |
| int osmo_stats_reporter_send_buffer(struct osmo_stats_reporter *srep) |
| { |
| int rc; |
| |
| if (!srep->buffer || msgb_length(srep->buffer) == 0) |
| return 0; |
| |
| rc = osmo_stats_reporter_send(srep, |
| (const char *)msgb_data(srep->buffer), msgb_length(srep->buffer)); |
| |
| msgb_trim(srep->buffer, 0); |
| |
| return rc; |
| } |
| #endif /* HAVE_SYS_SOCKET_H */ |
| |
| /*** log reporter ***/ |
| |
| struct osmo_stats_reporter *osmo_stats_reporter_create_log(const char *name) |
| { |
| struct osmo_stats_reporter *srep; |
| srep = osmo_stats_reporter_alloc(OSMO_STATS_REPORTER_LOG, name); |
| |
| srep->have_net_config = 0; |
| |
| srep->send_counter = osmo_stats_reporter_log_send_counter; |
| srep->send_item = osmo_stats_reporter_log_send_item; |
| |
| return srep; |
| } |
| |
| static int osmo_stats_reporter_log_send(struct osmo_stats_reporter *srep, |
| const char *type, |
| const char *name1, unsigned int index1, const char *name2, int value, |
| const char *unit) |
| { |
| LOGP(DLSTATS, LOGL_INFO, |
| "stats t=%s p=%s g=%s i=%u n=%s v=%d u=%s\n", |
| type, srep->name_prefix ? srep->name_prefix : "", |
| name1 ? name1 : "", index1, |
| name2, value, unit ? unit : ""); |
| |
| return 0; |
| } |
| |
| |
| static int osmo_stats_reporter_log_send_counter(struct osmo_stats_reporter *srep, |
| const struct rate_ctr_group *ctrg, |
| const struct rate_ctr_desc *desc, |
| int64_t value, int64_t delta) |
| { |
| if (ctrg) |
| return osmo_stats_reporter_log_send(srep, "c", |
| ctrg->desc->group_name_prefix, |
| ctrg->idx, |
| desc->name, value, NULL); |
| else |
| return osmo_stats_reporter_log_send(srep, "c", |
| NULL, 0, |
| desc->name, value, NULL); |
| } |
| |
| static int osmo_stats_reporter_log_send_item(struct osmo_stats_reporter *srep, |
| const struct osmo_stat_item_group *statg, |
| const struct osmo_stat_item_desc *desc, int64_t value) |
| { |
| return osmo_stats_reporter_log_send(srep, "i", |
| statg->desc->group_name_prefix, statg->idx, |
| desc->name, value, desc->unit); |
| } |
| |
| /*** helper for reporting ***/ |
| |
| static int osmo_stats_reporter_check_config(struct osmo_stats_reporter *srep, |
| unsigned int index, int class_id) |
| { |
| if (class_id == OSMO_STATS_CLASS_UNKNOWN) |
| class_id = index != 0 ? |
| OSMO_STATS_CLASS_SUBSCRIBER : OSMO_STATS_CLASS_GLOBAL; |
| |
| return class_id <= srep->max_class; |
| } |
| |
| /*** generic rate counter support ***/ |
| |
| static int osmo_stats_reporter_send_counter(struct osmo_stats_reporter *srep, |
| const struct rate_ctr_group *ctrg, |
| const struct rate_ctr_desc *desc, |
| int64_t value, int64_t delta) |
| { |
| if (!srep->send_counter) |
| return 0; |
| |
| return srep->send_counter(srep, ctrg, desc, value, delta); |
| } |
| |
| static int rate_ctr_handler( |
| struct rate_ctr_group *ctrg, struct rate_ctr *ctr, |
| const struct rate_ctr_desc *desc, void *sctx_) |
| { |
| struct osmo_stats_reporter *srep; |
| int64_t delta = rate_ctr_difference(ctr); |
| |
| llist_for_each_entry(srep, &osmo_stats_reporter_list, list) { |
| if (!srep->running) |
| continue; |
| |
| if (delta == 0 && !srep->force_single_flush) |
| continue; |
| |
| if (!osmo_stats_reporter_check_config(srep, |
| ctrg->idx, ctrg->desc->class_id)) |
| continue; |
| |
| osmo_stats_reporter_send_counter(srep, ctrg, desc, |
| ctr->current, delta); |
| |
| /* TODO: handle result (log?, inc counter(!)?) or remove it */ |
| } |
| |
| return 0; |
| } |
| |
| static int rate_ctr_group_handler(struct rate_ctr_group *ctrg, void *sctx_) |
| { |
| rate_ctr_for_each_counter(ctrg, rate_ctr_handler, sctx_); |
| |
| return 0; |
| } |
| |
| /*** stat item support ***/ |
| |
| static int osmo_stats_reporter_send_item(struct osmo_stats_reporter *srep, |
| const struct osmo_stat_item_group *statg, |
| const struct osmo_stat_item_desc *desc, |
| int32_t value) |
| { |
| if (!srep->send_item) |
| return 0; |
| |
| return srep->send_item(srep, statg, desc, value); |
| } |
| |
| static int osmo_stat_item_handler( |
| struct osmo_stat_item_group *statg, struct osmo_stat_item *item, void *sctx_) |
| { |
| struct osmo_stats_reporter *srep; |
| int32_t idx = current_stat_item_index; |
| int32_t value; |
| int have_value; |
| |
| have_value = osmo_stat_item_get_next(item, &idx, &value) > 0; |
| if (!have_value) |
| /* Send the last value in case a flush is requested */ |
| value = osmo_stat_item_get_last(item); |
| |
| do { |
| llist_for_each_entry(srep, &osmo_stats_reporter_list, list) { |
| if (!srep->running) |
| continue; |
| |
| if (!have_value && !srep->force_single_flush) |
| continue; |
| |
| if (!osmo_stats_reporter_check_config(srep, |
| statg->idx, statg->desc->class_id)) |
| continue; |
| |
| osmo_stats_reporter_send_item(srep, statg, |
| item->desc, value); |
| } |
| |
| if (!have_value) |
| break; |
| |
| have_value = osmo_stat_item_get_next(item, &idx, &value) > 0; |
| } while (have_value); |
| |
| return 0; |
| } |
| |
| static int osmo_stat_item_group_handler(struct osmo_stat_item_group *statg, void *sctx_) |
| { |
| osmo_stat_item_for_each_item(statg, osmo_stat_item_handler, sctx_); |
| |
| return 0; |
| } |
| |
| /*** osmo counter support ***/ |
| |
| static int handle_counter(struct osmo_counter *counter, void *sctx_) |
| { |
| struct osmo_stats_reporter *srep; |
| struct rate_ctr_desc desc = {0}; |
| /* Fake a rate counter description */ |
| desc.name = counter->name; |
| desc.description = counter->description; |
| |
| int delta = osmo_counter_difference(counter); |
| |
| llist_for_each_entry(srep, &osmo_stats_reporter_list, list) { |
| if (!srep->running) |
| continue; |
| |
| if (delta == 0 && !srep->force_single_flush) |
| continue; |
| |
| osmo_stats_reporter_send_counter(srep, NULL, &desc, |
| counter->value, delta); |
| |
| /* TODO: handle result (log?, inc counter(!)?) */ |
| } |
| |
| return 0; |
| } |
| |
| |
| /*** main reporting function ***/ |
| |
| static void flush_all_reporters() |
| { |
| struct osmo_stats_reporter *srep; |
| |
| llist_for_each_entry(srep, &osmo_stats_reporter_list, list) { |
| if (!srep->running) |
| continue; |
| |
| osmo_stats_reporter_send_buffer(srep); |
| srep->force_single_flush = 0; |
| } |
| } |
| |
| int osmo_stats_report() |
| { |
| /* per group actions */ |
| osmo_counters_for_each(handle_counter, NULL); |
| rate_ctr_for_each_group(rate_ctr_group_handler, NULL); |
| osmo_stat_item_for_each_group(osmo_stat_item_group_handler, NULL); |
| |
| /* global actions */ |
| osmo_stat_item_discard_all(¤t_stat_item_index); |
| flush_all_reporters(); |
| |
| return 0; |
| } |
| |
| #endif /* !EMBEDDED */ |