blob: 9b8a194ae64c8fc293aae4848698737e6907f66f [file] [log] [blame]
Neels Hofmeyr17518fe2017-06-20 04:35:06 +02001/*! \file control_if.c
2 * SNMP-like status interface. */
3/*
Daniel Willmann1264cb42010-10-21 15:00:36 +02004 * (C) 2010-2011 by Daniel Willmann <daniel@totalueberwachung.de>
5 * (C) 2010-2011 by On-Waves
6 *
7 * All Rights Reserved
8 *
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
18 *
19 * You should have received a copy of the GNU General Public License along
20 * with this program; if not, write to the Free Software Foundation, Inc.,
21 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
22 *
23 */
24
Holger Hans Peter Freyther267fd862015-03-18 21:52:38 +010025#include "config.h"
26
Daniel Willmann1264cb42010-10-21 15:00:36 +020027#include <errno.h>
Daniel Willmann0834a342011-07-19 12:12:10 +020028#include <inttypes.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020029#include <stdio.h>
30#include <stdlib.h>
31#include <string.h>
32#include <time.h>
33#include <unistd.h>
34
35#include <arpa/inet.h>
36
Holger Hans Peter Freyther5e211312014-08-21 16:31:52 +020037#include <netinet/in.h>
Harald Weltee15ac062014-12-04 14:15:36 +010038#ifdef HAVE_NETINET_TCP_H
39#include <netinet/tcp.h>
40#endif
Daniel Willmann1264cb42010-10-21 15:00:36 +020041
Harald Weltee15ac062014-12-04 14:15:36 +010042#include <fcntl.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020043#include <sys/ioctl.h>
44#include <sys/socket.h>
45#include <sys/types.h>
46
Harald Welte1238cc62014-08-20 19:50:04 +020047#include <osmocom/ctrl/control_cmd.h>
48#include <osmocom/ctrl/control_if.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020049
50#include <osmocom/core/msgb.h>
Daniel Willmanne9f58942011-04-21 11:43:55 +020051#include <osmocom/core/rate_ctr.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020052#include <osmocom/core/select.h>
Harald Welte216338c2017-10-15 19:46:19 +020053#include <osmocom/core/counter.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020054#include <osmocom/core/talloc.h>
Harald Welte1238cc62014-08-20 19:50:04 +020055#include <osmocom/core/socket.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020056
Harald Welte1238cc62014-08-20 19:50:04 +020057#include <osmocom/gsm/protocol/ipaccess.h>
Harald Welteacbb4c92014-08-20 22:51:25 +020058#include <osmocom/gsm/ipa.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020059
60#include <osmocom/vty/command.h>
61#include <osmocom/vty/vector.h>
62
Harald Welte79c137c2017-04-16 19:08:49 +020063extern int osmo_fsm_ctrl_cmds_install(void);
64
Daniel Willmann1264cb42010-10-21 15:00:36 +020065vector ctrl_node_vec;
66
Harald Welted6b1f852017-04-16 16:57:40 +020067/* global list of control interface lookup helpers */
68struct lookup_helper {
69 struct llist_head list;
70 ctrl_cmd_lookup lookup;
71};
72static LLIST_HEAD(ctrl_lookup_helpers);
73
Harald Welte528134b2014-08-21 15:13:08 +020074int ctrl_parse_get_num(vector vline, int i, long *num)
75{
76 char *token, *tmp;
77
78 if (i >= vector_active(vline))
79 return 0;
80 token = vector_slot(vline, i);
81
82 errno = 0;
83 if (token[0] == '\0')
84 return 0;
85
86 *num = strtol(token, &tmp, 10);
87 if (tmp[0] != '\0' || errno != 0)
88 return 0;
89
90 return 1;
91}
92
Daniel Willmann35a1b3b2011-05-06 16:38:11 +020093/* Send command to all */
94int ctrl_cmd_send_to_all(struct ctrl_handle *ctrl, struct ctrl_cmd *cmd)
95{
96 struct ctrl_connection *ccon;
97 int ret = 0;
98
99 llist_for_each_entry(ccon, &ctrl->ccon_list, list_entry) {
100 if (ccon == cmd->ccon)
101 continue;
102 if (ctrl_cmd_send(&ccon->write_queue, cmd))
103 ret++;
104 }
105 return ret;
106}
107
Daniel Willmann1264cb42010-10-21 15:00:36 +0200108int ctrl_cmd_send(struct osmo_wqueue *queue, struct ctrl_cmd *cmd)
109{
110 int ret;
111 struct msgb *msg;
112
113 msg = ctrl_cmd_make(cmd);
114 if (!msg) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200115 LOGP(DLCTRL, LOGL_ERROR, "Could not generate msg\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200116 return -1;
117 }
118
Harald Welteacbb4c92014-08-20 22:51:25 +0200119 ipa_prepend_header_ext(msg, IPAC_PROTO_EXT_CTRL);
120 ipa_prepend_header(msg, IPAC_PROTO_OSMO);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200121
122 ret = osmo_wqueue_enqueue(queue, msg);
123 if (ret != 0) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200124 LOGP(DLCTRL, LOGL_ERROR, "Failed to enqueue the command.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200125 msgb_free(msg);
126 }
127 return ret;
128}
129
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200130/*! Send TRAP over given Control Interface
Maxbc067eb2016-08-04 11:07:08 +0200131 * \param[in] ctrl Control Interface over which TRAP will be sent
132 * \param[in] name Name of the TRAP variable
133 * \param[in] value Value of the TRAP variable
134 * \return Negative value on error, result of ctrl_cmd_send_to_all() otherwise
135 */
136int ctrl_cmd_send_trap(struct ctrl_handle *ctrl, const char *name, char *value)
137{
138 int r;
139 struct ctrl_cmd *cmd = ctrl_cmd_create(NULL, CTRL_TYPE_TRAP);
140 if (!cmd)
141 return -ENOMEM;
142
143 cmd->id = "0"; /* It's a TRAP! */
Harald Weltef12d40f2017-02-08 15:46:53 +0000144 cmd->variable = (char *) name;
Maxbc067eb2016-08-04 11:07:08 +0200145 cmd->reply = value;
146 r = ctrl_cmd_send_to_all(ctrl, cmd);
147 talloc_free(cmd);
148 return r;
149}
150
Daniel Willmannf2e75d42011-07-22 18:08:31 +0200151struct ctrl_cmd *ctrl_cmd_trap(struct ctrl_cmd *cmd)
152{
153 struct ctrl_cmd *trap;
154
Harald Weltec9df37d2014-08-20 19:55:17 +0200155 trap = ctrl_cmd_cpy(cmd, cmd);
Daniel Willmannf2e75d42011-07-22 18:08:31 +0200156 if (!trap)
157 return NULL;
158
159 trap->ccon = cmd->ccon;
160 trap->type = CTRL_TYPE_TRAP;
161 return trap;
162}
163
Daniel Willmann1264cb42010-10-21 15:00:36 +0200164static void control_close_conn(struct ctrl_connection *ccon)
165{
Harald Welte39c9e7b2014-08-22 00:28:51 +0200166 struct ctrl_cmd_def *cd, *cd2;
167
Harald Welte6d96dd72013-03-16 10:13:04 +0100168 osmo_wqueue_clear(&ccon->write_queue);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200169 close(ccon->write_queue.bfd.fd);
170 osmo_fd_unregister(&ccon->write_queue.bfd);
Daniel Willmann0e0cf372011-05-06 16:37:25 +0200171 llist_del(&ccon->list_entry);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200172 if (ccon->closed_cb)
173 ccon->closed_cb(ccon);
Jacob Erlbeck141de622014-03-31 13:42:11 +0200174 msgb_free(ccon->pending_msg);
Harald Welte39c9e7b2014-08-22 00:28:51 +0200175
176 /* clean up deferred commands */
177 llist_for_each_entry_safe(cd, cd2, &ccon->def_cmds, list) {
178 /* delete from list of def_cmds for this ccon */
179 llist_del(&cd->list);
180 /* not strictly needed as this is a slave to the ccon which we
181 * are about to free anyway */
182 talloc_free(cd->cmd);
183 /* set the CMD to null, this is the indication to the user that
184 * the connection for this command has gone */
185 cd->cmd = NULL;
186 }
187
Daniel Willmann1264cb42010-10-21 15:00:36 +0200188 talloc_free(ccon);
189}
190
Harald Welte528134b2014-08-21 15:13:08 +0200191int ctrl_cmd_handle(struct ctrl_handle *ctrl, struct ctrl_cmd *cmd,
192 void *data)
193{
194 char *request;
195 int i, j, ret, node;
Max33c7ba62017-05-02 16:24:12 +0200196 bool break_cycle = false;
Harald Welte528134b2014-08-21 15:13:08 +0200197 vector vline, cmdvec, cmds_vec;
198
Max2ed36592017-02-24 14:48:45 +0100199 if (cmd->type == CTRL_TYPE_SET_REPLY ||
200 cmd->type == CTRL_TYPE_GET_REPLY) {
201 if (strncmp(cmd->reply, "OK", 2) == 0) {
202 LOGP(DLCTRL, LOGL_DEBUG, "%s <%s> for %s is OK\n",
203 get_value_string(ctrl_type_vals, cmd->type),
204 cmd->id, cmd->variable);
205 return CTRL_CMD_HANDLED;
206 }
207 }
208
209 if (cmd->type == CTRL_TYPE_ERROR) {
210 LOGP(DLCTRL, LOGL_ERROR, "%s <%s> for %s is %s\n",
211 get_value_string(ctrl_type_vals, cmd->type),
212 cmd->id, cmd->variable, cmd->reply);
213 return CTRL_CMD_HANDLED;
214 }
215
Harald Welte528134b2014-08-21 15:13:08 +0200216 ret = CTRL_CMD_ERROR;
217 cmd->reply = NULL;
218 node = CTRL_NODE_ROOT;
219 cmd->node = data;
220
221 request = talloc_strdup(cmd, cmd->variable);
222 if (!request)
223 goto err;
224
225 for (i=0;i<strlen(request);i++) {
226 if (request[i] == '.')
227 request[i] = ' ';
228 }
229
230 vline = cmd_make_strvec(request);
231 talloc_free(request);
232 if (!vline) {
233 cmd->reply = "cmd_make_strvec failed.";
234 goto err;
235 }
236
237 for (i=0;i<vector_active(vline);i++) {
Harald Welted6b1f852017-04-16 16:57:40 +0200238 struct lookup_helper *lh;
Harald Welte528134b2014-08-21 15:13:08 +0200239 int rc;
240
241 if (ctrl->lookup)
242 rc = ctrl->lookup(data, vline, &node, &cmd->node, &i);
243 else
244 rc = 0;
245
Harald Welted6b1f852017-04-16 16:57:40 +0200246 if (!rc) {
247 llist_for_each_entry(lh, &ctrl_lookup_helpers, list) {
248 rc = lh->lookup(data, vline, &node, &cmd->node, &i);
249 if (rc)
250 break;
251 }
252 }
253
Max33c7ba62017-05-02 16:24:12 +0200254 switch (rc) {
255 case 1: /* do nothing */
256 break;
257 case -ENODEV:
258 cmd_free_strvec(vline);
259 cmd->type = CTRL_TYPE_ERROR;
260 cmd->reply = "Error while resolving object";
261 return ret;
262 case -ERANGE:
263 cmd_free_strvec(vline);
264 cmd->type = CTRL_TYPE_ERROR;
265 cmd->reply = "Error while parsing the index.";
266 return ret;
267 default: /* If we're here the rest must be the command */
Harald Welte528134b2014-08-21 15:13:08 +0200268 cmdvec = vector_init(vector_active(vline)-i);
269 for (j=i; j<vector_active(vline); j++) {
270 vector_set(cmdvec, vector_slot(vline, j));
271 }
272
273 /* Get the command vector of the right node */
274 cmds_vec = vector_lookup(ctrl_node_vec, node);
275
276 if (!cmds_vec) {
277 cmd->reply = "Command not found.";
278 vector_free(cmdvec);
279 break;
280 }
281
282 ret = ctrl_cmd_exec(cmdvec, cmd, cmds_vec, data);
Harald Welte528134b2014-08-21 15:13:08 +0200283 vector_free(cmdvec);
Max33c7ba62017-05-02 16:24:12 +0200284 break_cycle = true;
Harald Welte528134b2014-08-21 15:13:08 +0200285 break;
286 }
287
Max33c7ba62017-05-02 16:24:12 +0200288 if (break_cycle)
289 break;
290
Harald Welte528134b2014-08-21 15:13:08 +0200291 if (i+1 == vector_active(vline))
292 cmd->reply = "Command not present.";
293 }
294
295 cmd_free_strvec(vline);
296
297err:
298 if (!cmd->reply) {
299 if (ret == CTRL_CMD_ERROR) {
Thorsten Alteholza81055d2017-03-02 22:13:48 +0100300 cmd->reply = "An error has occurred.";
Harald Welte528134b2014-08-21 15:13:08 +0200301 LOGP(DLCTRL, LOGL_NOTICE,
302 "%s: cmd->reply has not been set (ERROR).\n",
303 cmd->variable);
304 } else if (ret == CTRL_CMD_REPLY) {
305 LOGP(DLCTRL, LOGL_NOTICE,
306 "%s: cmd->reply has not been set (type = %d).\n",
307 cmd->variable, cmd->type);
308 cmd->reply = "";
309 } else {
310 cmd->reply = "Command has been handled.";
311 }
312 }
313
314 if (ret == CTRL_CMD_ERROR)
315 cmd->type = CTRL_TYPE_ERROR;
316 return ret;
Harald Welte528134b2014-08-21 15:13:08 +0200317}
318
319
Daniel Willmann1264cb42010-10-21 15:00:36 +0200320static int handle_control_read(struct osmo_fd * bfd)
321{
Pablo Neira Ayuso10dfc262011-08-17 22:44:07 +0200322 int ret = -1;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200323 struct osmo_wqueue *queue;
324 struct ctrl_connection *ccon;
325 struct ipaccess_head *iph;
326 struct ipaccess_head_ext *iph_ext;
Alexander Huemer1d3634e2011-10-12 00:29:30 +0200327 struct msgb *msg = NULL;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200328 struct ctrl_cmd *cmd;
329 struct ctrl_handle *ctrl = bfd->data;
330
331 queue = container_of(bfd, struct osmo_wqueue, bfd);
332 ccon = container_of(queue, struct ctrl_connection, write_queue);
333
Jacob Erlbeck141de622014-03-31 13:42:11 +0200334 ret = ipa_msg_recv_buffered(bfd->fd, &msg, &ccon->pending_msg);
Pablo Neira Ayuso10dfc262011-08-17 22:44:07 +0200335 if (ret <= 0) {
Jacob Erlbeck141de622014-03-31 13:42:11 +0200336 if (ret == -EAGAIN)
337 return 0;
Pablo Neira Ayuso10dfc262011-08-17 22:44:07 +0200338 if (ret == 0)
Harald Welte7fd0c832014-08-20 19:58:13 +0200339 LOGP(DLCTRL, LOGL_INFO, "The control connection was closed\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200340 else
Harald Welte7fd0c832014-08-20 19:58:13 +0200341 LOGP(DLCTRL, LOGL_ERROR, "Failed to parse ip access message: %d\n", ret);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200342
343 goto err;
344 }
345
346 if (msg->len < sizeof(*iph) + sizeof(*iph_ext)) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200347 LOGP(DLCTRL, LOGL_ERROR, "The message is too short.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200348 goto err;
349 }
350
351 iph = (struct ipaccess_head *) msg->data;
352 if (iph->proto != IPAC_PROTO_OSMO) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200353 LOGP(DLCTRL, LOGL_ERROR, "Protocol mismatch. We got 0x%x\n", iph->proto);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200354 goto err;
355 }
356
357 iph_ext = (struct ipaccess_head_ext *) iph->data;
358 if (iph_ext->proto != IPAC_PROTO_EXT_CTRL) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200359 LOGP(DLCTRL, LOGL_ERROR, "Extended protocol mismatch. We got 0x%x\n", iph_ext->proto);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200360 goto err;
361 }
362
363 msg->l2h = iph_ext->data;
364
365 cmd = ctrl_cmd_parse(ccon, msg);
366
367 if (cmd) {
368 cmd->ccon = ccon;
Harald Welte528134b2014-08-21 15:13:08 +0200369 if (ctrl_cmd_handle(ctrl, cmd, ctrl->data) != CTRL_CMD_HANDLED) {
Daniel Willmann1264cb42010-10-21 15:00:36 +0200370 ctrl_cmd_send(queue, cmd);
371 talloc_free(cmd);
372 }
373 } else {
374 cmd = talloc_zero(ccon, struct ctrl_cmd);
375 if (!cmd)
376 goto err;
Harald Welte7fd0c832014-08-20 19:58:13 +0200377 LOGP(DLCTRL, LOGL_ERROR, "Command parser error.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200378 cmd->type = CTRL_TYPE_ERROR;
379 cmd->id = "err";
380 cmd->reply = "Command parser error.";
381 ctrl_cmd_send(queue, cmd);
382 talloc_free(cmd);
383 }
384
385 msgb_free(msg);
386 return 0;
387
388err:
389 control_close_conn(ccon);
390 msgb_free(msg);
391 return ret;
392}
393
394static int control_write_cb(struct osmo_fd *bfd, struct msgb *msg)
395{
396 int rc;
397
398 rc = write(bfd->fd, msg->data, msg->len);
399 if (rc != msg->len)
Harald Welte7fd0c832014-08-20 19:58:13 +0200400 LOGP(DLCTRL, LOGL_ERROR, "Failed to write message to the control connection.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200401
402 return rc;
403}
404
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200405/*! Allocate CTRL connection
Maxfa9e05e2017-02-20 11:53:11 +0100406 * \param[in] ctx Context from which talloc should allocate it
407 * \param[in] data caller's private data parameter which should assigned to
408 write queue's file descriptor data parameter.
409 * \return Allocated CTRL connection structure or NULL in case of errors
410 */
411struct ctrl_connection *osmo_ctrl_conn_alloc(void *ctx, void *data)
Daniel Willmann1264cb42010-10-21 15:00:36 +0200412{
413 struct ctrl_connection *ccon = talloc_zero(ctx, struct ctrl_connection);
414 if (!ccon)
415 return NULL;
416
417 osmo_wqueue_init(&ccon->write_queue, 100);
418 /* Error handling here? */
419
420 INIT_LLIST_HEAD(&ccon->cmds);
Harald Welte39c9e7b2014-08-22 00:28:51 +0200421 INIT_LLIST_HEAD(&ccon->def_cmds);
422
Maxfa9e05e2017-02-20 11:53:11 +0100423 ccon->write_queue.bfd.data = data;
424 ccon->write_queue.write_cb = control_write_cb;
Max9756c462017-02-24 13:03:32 +0100425 ccon->write_queue.read_cb = handle_control_read;
Maxfa9e05e2017-02-20 11:53:11 +0100426
Daniel Willmann1264cb42010-10-21 15:00:36 +0200427 return ccon;
428}
429
430static int listen_fd_cb(struct osmo_fd *listen_bfd, unsigned int what)
431{
432 int ret, fd, on;
Daniel Willmann0e0cf372011-05-06 16:37:25 +0200433 struct ctrl_handle *ctrl;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200434 struct ctrl_connection *ccon;
435 struct sockaddr_in sa;
436 socklen_t sa_len = sizeof(sa);
437
438
439 if (!(what & BSC_FD_READ))
440 return 0;
441
442 fd = accept(listen_bfd->fd, (struct sockaddr *) &sa, &sa_len);
443 if (fd < 0) {
444 perror("accept");
445 return fd;
446 }
Harald Welte7fd0c832014-08-20 19:58:13 +0200447 LOGP(DLCTRL, LOGL_INFO, "accept()ed new control connection from %s\n",
Daniel Willmann1264cb42010-10-21 15:00:36 +0200448 inet_ntoa(sa.sin_addr));
449
Harald Weltee15ac062014-12-04 14:15:36 +0100450#ifdef TCP_NODELAY
Daniel Willmann1264cb42010-10-21 15:00:36 +0200451 on = 1;
452 ret = setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &on, sizeof(on));
453 if (ret != 0) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200454 LOGP(DLCTRL, LOGL_ERROR, "Failed to set TCP_NODELAY: %s\n", strerror(errno));
Daniel Willmann1264cb42010-10-21 15:00:36 +0200455 close(fd);
456 return ret;
457 }
Harald Weltee15ac062014-12-04 14:15:36 +0100458#endif
Maxfa9e05e2017-02-20 11:53:11 +0100459 ctrl = listen_bfd->data;
460 ccon = osmo_ctrl_conn_alloc(listen_bfd->data, ctrl);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200461 if (!ccon) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200462 LOGP(DLCTRL, LOGL_ERROR, "Failed to allocate.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200463 close(fd);
464 return -1;
465 }
466
Daniel Willmann1264cb42010-10-21 15:00:36 +0200467 ccon->write_queue.bfd.fd = fd;
468 ccon->write_queue.bfd.when = BSC_FD_READ;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200469
470 ret = osmo_fd_register(&ccon->write_queue.bfd);
471 if (ret < 0) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200472 LOGP(DLCTRL, LOGL_ERROR, "Could not register FD.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200473 close(ccon->write_queue.bfd.fd);
474 talloc_free(ccon);
475 }
476
Daniel Willmann0e0cf372011-05-06 16:37:25 +0200477 llist_add(&ccon->list_entry, &ctrl->ccon_list);
478
Daniel Willmann1264cb42010-10-21 15:00:36 +0200479 return ret;
480}
481
Daniel Willmanne9f58942011-04-21 11:43:55 +0200482static uint64_t get_rate_ctr_value(const struct rate_ctr *ctr, int intv)
483{
484 if (intv >= RATE_CTR_INTV_NUM)
485 return 0;
486
487 /* Absolute value */
488 if (intv == -1) {
489 return ctr->current;
490 } else {
491 return ctr->intv[intv].rate;
492 }
493}
494
Harald Weltec9df37d2014-08-20 19:55:17 +0200495static char *get_all_rate_ctr_in_group(void *ctx, const struct rate_ctr_group *ctrg, int intv)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200496{
497 int i;
Harald Weltec9df37d2014-08-20 19:55:17 +0200498 char *counters = talloc_strdup(ctx, "");
Daniel Willmanne9f58942011-04-21 11:43:55 +0200499 if (!counters)
500 return NULL;
501
502 for (i=0;i<ctrg->desc->num_ctr;i++) {
Daniel Willmann0834a342011-07-19 12:12:10 +0200503 counters = talloc_asprintf_append(counters, "\n%s.%u.%s %"PRIu64,
Daniel Willmanne9f58942011-04-21 11:43:55 +0200504 ctrg->desc->group_name_prefix, ctrg->idx,
505 ctrg->desc->ctr_desc[i].name,
506 get_rate_ctr_value(&ctrg->ctr[i], intv));
507 if (!counters)
508 return NULL;
509 }
510 return counters;
511}
512
Daniel Willmanne9f58942011-04-21 11:43:55 +0200513static int get_rate_ctr_group_idx(const struct rate_ctr_group *ctrg, int intv, struct ctrl_cmd *cmd)
514{
515 char *counters;
516
Harald Weltec9df37d2014-08-20 19:55:17 +0200517 counters = get_all_rate_ctr_in_group(cmd, ctrg, intv);
Daniel Willmanne9f58942011-04-21 11:43:55 +0200518 if (!counters)
519 goto oom;
520
521 cmd->reply = talloc_asprintf(cmd, "All counters in %s.%u%s",
522 ctrg->desc->group_name_prefix, ctrg->idx, counters);
523 talloc_free(counters);
524 if (!cmd->reply)
525 goto oom;
526
527 return CTRL_CMD_REPLY;
528oom:
529 cmd->reply = "OOM.";
530 return CTRL_CMD_ERROR;
531}
532
533/* rate_ctr */
534CTRL_CMD_DEFINE(rate_ctr, "rate_ctr *");
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200535static int get_rate_ctr(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200536{
537 int intv;
538 unsigned int idx;
Neels Hofmeyr5b34f772016-09-19 14:15:03 +0200539 char *ctr_group, *ctr_idx, *tmp, *dup, *saveptr, *interval;
Daniel Willmanne9f58942011-04-21 11:43:55 +0200540 struct rate_ctr_group *ctrg;
541 const struct rate_ctr *ctr;
542
543 dup = talloc_strdup(cmd, cmd->variable);
544 if (!dup)
545 goto oom;
546
547 /* Skip over possible prefixes (net.) */
548 tmp = strstr(dup, "rate_ctr");
549 if (!tmp) {
550 talloc_free(dup);
551 cmd->reply = "rate_ctr not a token in rate_ctr command!";
552 goto err;
553 }
554
555 strtok_r(tmp, ".", &saveptr);
556 interval = strtok_r(NULL, ".", &saveptr);
557 if (!interval) {
558 talloc_free(dup);
559 cmd->reply = "Missing interval.";
560 goto err;
561 }
562
563 if (!strcmp(interval, "abs")) {
564 intv = -1;
565 } else if (!strcmp(interval, "per_sec")) {
566 intv = RATE_CTR_INTV_SEC;
567 } else if (!strcmp(interval, "per_min")) {
568 intv = RATE_CTR_INTV_MIN;
569 } else if (!strcmp(interval, "per_hour")) {
570 intv = RATE_CTR_INTV_HOUR;
571 } else if (!strcmp(interval, "per_day")) {
572 intv = RATE_CTR_INTV_DAY;
573 } else {
574 talloc_free(dup);
575 cmd->reply = "Wrong interval.";
576 goto err;
577 }
578
579 ctr_group = strtok_r(NULL, ".", &saveptr);
Max22886d92016-06-14 18:17:06 +0200580 ctr_idx = strtok_r(NULL, ".", &saveptr);
581 if (!ctr_group || !ctr_idx) {
Daniel Willmanne9f58942011-04-21 11:43:55 +0200582 talloc_free(dup);
Max22886d92016-06-14 18:17:06 +0200583 cmd->reply = "Counter group must be of name.index form e. g. "
584 "e1inp.0";
Daniel Willmanne9f58942011-04-21 11:43:55 +0200585 goto err;
586 }
Daniel Willmanne9f58942011-04-21 11:43:55 +0200587
Daniel Willmanne9f58942011-04-21 11:43:55 +0200588 idx = atoi(ctr_idx);
589
590 ctrg = rate_ctr_get_group_by_name_idx(ctr_group, idx);
591 if (!ctrg) {
592 talloc_free(dup);
Max22886d92016-06-14 18:17:06 +0200593 cmd->reply = "Counter group with given name and index not found";
Daniel Willmanne9f58942011-04-21 11:43:55 +0200594 goto err;
595 }
596
Max22886d92016-06-14 18:17:06 +0200597 if (!strlen(saveptr)) {
Daniel Willmanne9f58942011-04-21 11:43:55 +0200598 talloc_free(dup);
599 return get_rate_ctr_group_idx(ctrg, intv, cmd);
600 }
601
Max22886d92016-06-14 18:17:06 +0200602 ctr = rate_ctr_get_by_name(ctrg, saveptr);
Daniel Willmanne9f58942011-04-21 11:43:55 +0200603 if (!ctr) {
604 cmd->reply = "Counter name not found.";
605 talloc_free(dup);
606 goto err;
607 }
608
609 talloc_free(dup);
610
Holger Hans Peter Freytherad6451f2012-03-16 12:18:39 +0100611 cmd->reply = talloc_asprintf(cmd, "%"PRIu64, get_rate_ctr_value(ctr, intv));
Daniel Willmanne9f58942011-04-21 11:43:55 +0200612 if (!cmd->reply)
613 goto oom;
614
615 return CTRL_CMD_REPLY;
616oom:
617 cmd->reply = "OOM";
618err:
619 return CTRL_CMD_ERROR;
620}
621
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200622static int set_rate_ctr(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200623{
624 cmd->reply = "Can't set rate counter.";
625
626 return CTRL_CMD_ERROR;
627}
628
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200629static int verify_rate_ctr(struct ctrl_cmd *cmd, const char *value, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200630{
631 return 0;
632}
633
634/* counter */
635CTRL_CMD_DEFINE(counter, "counter *");
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200636static int get_counter(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200637{
638 char *ctr_name, *tmp, *dup, *saveptr;
639 struct osmo_counter *counter;
640
641 cmd->reply = "OOM";
642 dup = talloc_strdup(cmd, cmd->variable);
643 if (!dup)
644 goto err;
645
646
647 tmp = strstr(dup, "counter");
648 if (!tmp) {
649 talloc_free(dup);
650 goto err;
651 }
652
653 strtok_r(tmp, ".", &saveptr);
654 ctr_name = strtok_r(NULL, "\0", &saveptr);
655
656 if (!ctr_name)
657 goto err;
658
659 counter = osmo_counter_get_by_name(ctr_name);
660 if (!counter) {
661 cmd->reply = "Counter name not found.";
662 talloc_free(dup);
663 goto err;
664 }
665
666 talloc_free(dup);
667
668 cmd->reply = talloc_asprintf(cmd, "%lu", counter->value);
669 if (!cmd->reply) {
670 cmd->reply = "OOM";
671 goto err;
672 }
673
674 return CTRL_CMD_REPLY;
675err:
676 return CTRL_CMD_ERROR;
677}
678
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200679static int set_counter(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200680{
681
682 cmd->reply = "Can't set counter.";
683
684 return CTRL_CMD_ERROR;
685}
686
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200687static int verify_counter(struct ctrl_cmd *cmd, const char *value, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200688{
689 return 0;
690}
691
Harald Welteae2fcb22014-08-21 15:20:00 +0200692struct ctrl_handle *ctrl_interface_setup(void *data, uint16_t port,
693 ctrl_cmd_lookup lookup)
Daniel Willmann1264cb42010-10-21 15:00:36 +0200694{
Neels Hofmeyr49343092016-02-24 00:09:40 +0100695 return ctrl_interface_setup_dynip(data, "127.0.0.1", port, lookup);
696}
697
Harald Welte79c137c2017-04-16 19:08:49 +0200698static int ctrl_initialized = 0;
699
700/* global ctrl initialization */
Neels Hofmeyrea668522017-10-16 16:18:01 +0200701static int ctrl_init(unsigned int node_count)
Harald Welte79c137c2017-04-16 19:08:49 +0200702{
703 int ret;
704
Neels Hofmeyrea668522017-10-16 16:18:01 +0200705 if (!node_count)
706 node_count = _LAST_CTRL_NODE;
707 OSMO_ASSERT(node_count >= _LAST_CTRL_NODE);
Harald Welte79c137c2017-04-16 19:08:49 +0200708
Neels Hofmeyrea668522017-10-16 16:18:01 +0200709 if (ctrl_initialized) {
710 OSMO_ASSERT(ctrl_initialized == node_count);
711 return 0;
712 }
713
714 ctrl_node_vec = vector_init(node_count);
Harald Welte79c137c2017-04-16 19:08:49 +0200715 if (!ctrl_node_vec)
716 goto err;
717
718 ret = ctrl_cmd_install(CTRL_NODE_ROOT, &cmd_rate_ctr);
719 if (ret)
720 goto err_vec;
721 ret = ctrl_cmd_install(CTRL_NODE_ROOT, &cmd_counter);
722 if (ret)
723 goto err_vec;
724
Harald Welte31c0fef2017-04-16 17:26:30 +0200725 ret = osmo_fsm_ctrl_cmds_install();
726 if (ret)
727 goto err_vec;
728
Neels Hofmeyrea668522017-10-16 16:18:01 +0200729 ctrl_initialized = node_count;
Harald Welte79c137c2017-04-16 19:08:49 +0200730 return 0;
731
732err_vec:
733 vector_free(ctrl_node_vec);
734 ctrl_node_vec = NULL;
735err:
736 return -1;
737}
738
Neels Hofmeyrea668522017-10-16 16:18:01 +0200739/*! Allocate a CTRL interface handle.
740 * \param[in] ctx Talloc allocation context to be used
Harald Welte79c137c2017-04-16 19:08:49 +0200741 * \param[in] data Pointer which will be made available to each
742 set_..() get_..() verify_..() control command function
743 * \param[in] lookup Lookup function pointer, can be NULL
Neels Hofmeyrea668522017-10-16 16:18:01 +0200744 * \param[in] node_count Number of CTRL nodes to allocate, 0 for default.
Harald Welte79c137c2017-04-16 19:08:49 +0200745 * \returns ctrl_handle pointer or NULL in case of errors
Neels Hofmeyrea668522017-10-16 16:18:01 +0200746 *
747 * Please see ctrl_interface_setup_dynip2() for a detailed description of \a
748 * node_count semantics.
Harald Welte79c137c2017-04-16 19:08:49 +0200749 */
Neels Hofmeyrea668522017-10-16 16:18:01 +0200750struct ctrl_handle *ctrl_handle_alloc2(void *ctx, void *data,
751 ctrl_cmd_lookup lookup,
752 unsigned int node_count)
Harald Welte79c137c2017-04-16 19:08:49 +0200753{
754 struct ctrl_handle *ctrl;
755
Neels Hofmeyrea668522017-10-16 16:18:01 +0200756 ctrl_init(node_count);
Harald Welte79c137c2017-04-16 19:08:49 +0200757
758 ctrl = talloc_zero(ctx, struct ctrl_handle);
759 if (!ctrl)
760 return NULL;
761
762 INIT_LLIST_HEAD(&ctrl->ccon_list);
763
764 ctrl->data = data;
765 ctrl->lookup = lookup;
766
767 return ctrl;
768}
769
Neels Hofmeyrea668522017-10-16 16:18:01 +0200770/*! Allocate a CTRL interface handle.
771 * \param[in] ctx Talloc allocation context to be used
772 * \param[in] data Pointer which will be made available to each
773 set_..() get_..() verify_..() control command function
774 * \param[in] lookup Lookup function pointer, can be NULL
775 * \returns ctrl_handle pointer or NULL in case of errors
776 */
777struct ctrl_handle *ctrl_handle_alloc(void *ctx, void *data, ctrl_cmd_lookup lookup)
778{
779 return ctrl_handle_alloc2(ctx, data, lookup, 0);
780}
781
782/*! Setup CTRL interface on a given address.
Max32ee5af2017-02-16 17:44:09 +0100783 * \param[in] data Pointer which will be made available to each
784 set_..() get_..() verify_..() control command function
785 * \param[in] bind_addr Address on which CTRL socket shall listen
786 * \param[in] port Port on which CTRL socket shall listen
787 * \param[in] lookup Lookup function pointer, can be NULL
Neels Hofmeyrea668522017-10-16 16:18:01 +0200788 * \param[in] node_count Number of CTRL nodes to allocate, 0 for default.
Max32ee5af2017-02-16 17:44:09 +0100789 * \returns ctrl_handle pointer or NULL in case of errors
Neels Hofmeyrea668522017-10-16 16:18:01 +0200790 *
791 * Control interface nodes are identified by a node handle; some of these are
792 * defined in enum ctrl_node_type, here in libosmocore. However, applications
793 * defining own nodes may add own control nodes without having to extend the
794 * enum in libosmocore. For example, in the calling application, define an enum
795 * like "enum more_ctrl_nodes { CTRL_NODE_FOO = _LAST_CTRL_NODE, CTRL_NODE_BAR,
796 * _LAST_CTRL_NODE_EXTENDED }". In order to provide space for the additional
797 * control nodes, pass _LAST_CTRL_NODE_EXTENDED to the \a node_count parameter.
798 * Passing 0 is identical to passing _LAST_CTRL_NODE, i.e. to not define own
799 * control nodes apart from libosmocore ones.
Max32ee5af2017-02-16 17:44:09 +0100800 */
Neels Hofmeyrea668522017-10-16 16:18:01 +0200801struct ctrl_handle *ctrl_interface_setup_dynip2(void *data,
802 const char *bind_addr,
803 uint16_t port,
804 ctrl_cmd_lookup lookup,
805 unsigned int node_count)
Neels Hofmeyr49343092016-02-24 00:09:40 +0100806{
Daniel Willmann1264cb42010-10-21 15:00:36 +0200807 int ret;
808 struct ctrl_handle *ctrl;
809
Neels Hofmeyrea668522017-10-16 16:18:01 +0200810 ctrl = ctrl_handle_alloc2(data, data, lookup, node_count);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200811 if (!ctrl)
Daniel Willmann782974b2011-07-22 17:55:42 +0200812 return NULL;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200813
Daniel Willmann1264cb42010-10-21 15:00:36 +0200814 /* Listen for control connections */
Harald Welte1238cc62014-08-20 19:50:04 +0200815 ctrl->listen_fd.cb = listen_fd_cb;
816 ctrl->listen_fd.data = ctrl;
817 ret = osmo_sock_init_ofd(&ctrl->listen_fd, AF_INET, SOCK_STREAM, IPPROTO_TCP,
Neels Hofmeyr49343092016-02-24 00:09:40 +0100818 bind_addr, port, OSMO_SOCK_F_BIND);
Harald Welte79c137c2017-04-16 19:08:49 +0200819 if (ret < 0) {
820 talloc_free(ctrl);
821 return NULL;
822 }
Daniel Willmanne9f58942011-04-21 11:43:55 +0200823
Neels Hofmeyr38d232e2016-09-19 14:17:30 +0200824 LOGP(DLCTRL, LOGL_NOTICE, "CTRL at %s %u\n", bind_addr, port);
Daniel Willmann782974b2011-07-22 17:55:42 +0200825 return ctrl;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200826}
Harald Welted6b1f852017-04-16 16:57:40 +0200827
Neels Hofmeyrea668522017-10-16 16:18:01 +0200828/*! Setup CTRL interface on a given address.
829 * \param[in] data Pointer which will be made available to each
830 set_..() get_..() verify_..() control command function
831 * \param[in] bind_addr Address on which CTRL socket shall listen
832 * \param[in] port Port on which CTRL socket shall listen
833 * \param[in] lookup Lookup function pointer, can be NULL
834 * \returns ctrl_handle pointer or NULL in case of errors
835 */
836struct ctrl_handle *ctrl_interface_setup_dynip(void *data,
837 const char *bind_addr,
838 uint16_t port,
839 ctrl_cmd_lookup lookup)
840{
841 return ctrl_interface_setup_dynip2(data, bind_addr, port, lookup, 0);
842}
843
844
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200845/*! Install a lookup helper function for control nodes
Harald Welted6b1f852017-04-16 16:57:40 +0200846 * This function is used by e.g. library code to install lookup helpers
847 * for additional nodes in the control interface.
848 * \param[in] lookup The lookup helper function
849 * \retuns - on success; negative on error.
850 */
851int ctrl_lookup_register(ctrl_cmd_lookup lookup)
852{
853 struct lookup_helper *lh;
854
855 /* avoid double registration */
856 llist_for_each_entry(lh, &ctrl_lookup_helpers, list) {
857 if (lh->lookup == lookup)
858 return -EEXIST;
859 }
860
861 lh = talloc_zero(NULL, struct lookup_helper);
862 if (!lh)
863 return -ENOMEM;
864
865 lh->lookup = lookup;
866 llist_add_tail(&lh->list, &ctrl_lookup_helpers);
867 return 0;
868}
Harald Weltef85861d2017-04-16 19:17:10 +0200869
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200870/*! Helper for "local execution" of a CTRL command from a string
Harald Weltef85861d2017-04-16 19:17:10 +0200871 * The function will parse + execute the given control command string
872 * and return a corresponding ctrl_cmd. Caller is responsible to
873 * talloc_free() the return value.
874 * \param[in] Control Interface Command String
875 * \returns parsed command, including reply; NULL on error */
876struct ctrl_cmd *ctrl_cmd_exec_from_string(struct ctrl_handle *ch, const char *cmdstr)
877{
878 struct msgb *msg = msgb_alloc(1024, "ctrl-cmd");
879 struct ctrl_cmd *cmd;
880
881 if (!msg)
882 return NULL;
883 msg->l2h = msg->data;
884 osmo_strlcpy((char *)msg->data, cmdstr, msgb_tailroom(msg));
885 msgb_put(msg, strlen(cmdstr));
886
887 cmd = ctrl_cmd_parse(ch, msg);
888 msgb_free(msg);
889 if (!cmd)
890 return NULL;
891 if (ctrl_cmd_handle(ch, cmd, NULL) < 0) {
892 talloc_free(cmd);
893 return NULL;
894 }
895 return cmd;
896}