blob: 07de0d46c6dbe6f0f5dab194c2c9038e55863502 [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
Harald Weltee08da972017-11-13 01:00:26 +09006 * (C) 2014 by Harald Welte <laforge@gnumonks.org>
7 * (C) 2016-2017 by sysmocom - s.f.m.c. GmbH
Daniel Willmann1264cb42010-10-21 15:00:36 +02008 *
9 * All Rights Reserved
10 *
Harald Weltee08da972017-11-13 01:00:26 +090011 * SPDX-License-Identifier: GPL-2.0+
12 *
Daniel Willmann1264cb42010-10-21 15:00:36 +020013 * This program is free software; you can redistribute it and/or modify
14 * it under the terms of the GNU General Public License as published by
15 * the Free Software Foundation; either version 2 of the License, or
16 * (at your option) any later version.
17 *
18 * This program is distributed in the hope that it will be useful,
19 * but WITHOUT ANY WARRANTY; without even the implied warranty of
20 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 * GNU General Public License for more details.
22 *
23 * You should have received a copy of the GNU General Public License along
24 * with this program; if not, write to the Free Software Foundation, Inc.,
25 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
26 *
27 */
28
Holger Hans Peter Freyther267fd862015-03-18 21:52:38 +010029#include "config.h"
30
Daniel Willmann1264cb42010-10-21 15:00:36 +020031#include <errno.h>
Daniel Willmann0834a342011-07-19 12:12:10 +020032#include <inttypes.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020033#include <stdio.h>
34#include <stdlib.h>
35#include <string.h>
36#include <time.h>
37#include <unistd.h>
38
39#include <arpa/inet.h>
40
Holger Hans Peter Freyther5e211312014-08-21 16:31:52 +020041#include <netinet/in.h>
Harald Weltee15ac062014-12-04 14:15:36 +010042#ifdef HAVE_NETINET_TCP_H
43#include <netinet/tcp.h>
44#endif
Daniel Willmann1264cb42010-10-21 15:00:36 +020045
Harald Weltee15ac062014-12-04 14:15:36 +010046#include <fcntl.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020047#include <sys/ioctl.h>
48#include <sys/socket.h>
49#include <sys/types.h>
50
Harald Welte1238cc62014-08-20 19:50:04 +020051#include <osmocom/ctrl/control_cmd.h>
52#include <osmocom/ctrl/control_if.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020053
54#include <osmocom/core/msgb.h>
Daniel Willmanne9f58942011-04-21 11:43:55 +020055#include <osmocom/core/rate_ctr.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020056#include <osmocom/core/select.h>
Harald Welte216338c2017-10-15 19:46:19 +020057#include <osmocom/core/counter.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020058#include <osmocom/core/talloc.h>
Harald Welte1238cc62014-08-20 19:50:04 +020059#include <osmocom/core/socket.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020060
Harald Welte1238cc62014-08-20 19:50:04 +020061#include <osmocom/gsm/protocol/ipaccess.h>
Harald Welteacbb4c92014-08-20 22:51:25 +020062#include <osmocom/gsm/ipa.h>
Daniel Willmann1264cb42010-10-21 15:00:36 +020063
64#include <osmocom/vty/command.h>
65#include <osmocom/vty/vector.h>
66
Harald Welte79c137c2017-04-16 19:08:49 +020067extern int osmo_fsm_ctrl_cmds_install(void);
68
Daniel Willmann1264cb42010-10-21 15:00:36 +020069vector ctrl_node_vec;
70
Harald Welted6b1f852017-04-16 16:57:40 +020071/* global list of control interface lookup helpers */
72struct lookup_helper {
73 struct llist_head list;
74 ctrl_cmd_lookup lookup;
75};
76static LLIST_HEAD(ctrl_lookup_helpers);
77
Harald Welte528134b2014-08-21 15:13:08 +020078int ctrl_parse_get_num(vector vline, int i, long *num)
79{
80 char *token, *tmp;
81
82 if (i >= vector_active(vline))
83 return 0;
84 token = vector_slot(vline, i);
85
86 errno = 0;
87 if (token[0] == '\0')
88 return 0;
89
90 *num = strtol(token, &tmp, 10);
91 if (tmp[0] != '\0' || errno != 0)
92 return 0;
93
94 return 1;
95}
96
Daniel Willmann35a1b3b2011-05-06 16:38:11 +020097/* Send command to all */
98int ctrl_cmd_send_to_all(struct ctrl_handle *ctrl, struct ctrl_cmd *cmd)
99{
100 struct ctrl_connection *ccon;
101 int ret = 0;
102
103 llist_for_each_entry(ccon, &ctrl->ccon_list, list_entry) {
104 if (ccon == cmd->ccon)
105 continue;
106 if (ctrl_cmd_send(&ccon->write_queue, cmd))
107 ret++;
108 }
109 return ret;
110}
111
Daniel Willmann1264cb42010-10-21 15:00:36 +0200112int ctrl_cmd_send(struct osmo_wqueue *queue, struct ctrl_cmd *cmd)
113{
114 int ret;
115 struct msgb *msg;
116
117 msg = ctrl_cmd_make(cmd);
118 if (!msg) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200119 LOGP(DLCTRL, LOGL_ERROR, "Could not generate msg\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200120 return -1;
121 }
122
Harald Welteacbb4c92014-08-20 22:51:25 +0200123 ipa_prepend_header_ext(msg, IPAC_PROTO_EXT_CTRL);
124 ipa_prepend_header(msg, IPAC_PROTO_OSMO);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200125
126 ret = osmo_wqueue_enqueue(queue, msg);
127 if (ret != 0) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200128 LOGP(DLCTRL, LOGL_ERROR, "Failed to enqueue the command.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200129 msgb_free(msg);
130 }
131 return ret;
132}
133
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200134/*! Send TRAP over given Control Interface
Maxbc067eb2016-08-04 11:07:08 +0200135 * \param[in] ctrl Control Interface over which TRAP will be sent
136 * \param[in] name Name of the TRAP variable
137 * \param[in] value Value of the TRAP variable
138 * \return Negative value on error, result of ctrl_cmd_send_to_all() otherwise
139 */
140int ctrl_cmd_send_trap(struct ctrl_handle *ctrl, const char *name, char *value)
141{
142 int r;
143 struct ctrl_cmd *cmd = ctrl_cmd_create(NULL, CTRL_TYPE_TRAP);
144 if (!cmd)
145 return -ENOMEM;
146
147 cmd->id = "0"; /* It's a TRAP! */
Harald Weltef12d40f2017-02-08 15:46:53 +0000148 cmd->variable = (char *) name;
Maxbc067eb2016-08-04 11:07:08 +0200149 cmd->reply = value;
150 r = ctrl_cmd_send_to_all(ctrl, cmd);
151 talloc_free(cmd);
152 return r;
153}
154
Daniel Willmannf2e75d42011-07-22 18:08:31 +0200155struct ctrl_cmd *ctrl_cmd_trap(struct ctrl_cmd *cmd)
156{
157 struct ctrl_cmd *trap;
158
Harald Weltec9df37d2014-08-20 19:55:17 +0200159 trap = ctrl_cmd_cpy(cmd, cmd);
Daniel Willmannf2e75d42011-07-22 18:08:31 +0200160 if (!trap)
161 return NULL;
162
163 trap->ccon = cmd->ccon;
164 trap->type = CTRL_TYPE_TRAP;
165 return trap;
166}
167
Daniel Willmann1264cb42010-10-21 15:00:36 +0200168static void control_close_conn(struct ctrl_connection *ccon)
169{
Harald Welte39c9e7b2014-08-22 00:28:51 +0200170 struct ctrl_cmd_def *cd, *cd2;
Harald Welte29e27982017-12-22 16:47:18 +0100171 char *name = osmo_sock_get_name(ccon, ccon->write_queue.bfd.fd);
172
173 LOGP(DLCTRL, LOGL_INFO, "close()d CTRL connection %s\n", name);
174 talloc_free(name);
Harald Welte39c9e7b2014-08-22 00:28:51 +0200175
Harald Welte6d96dd72013-03-16 10:13:04 +0100176 osmo_wqueue_clear(&ccon->write_queue);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200177 close(ccon->write_queue.bfd.fd);
178 osmo_fd_unregister(&ccon->write_queue.bfd);
Daniel Willmann0e0cf372011-05-06 16:37:25 +0200179 llist_del(&ccon->list_entry);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200180 if (ccon->closed_cb)
181 ccon->closed_cb(ccon);
Jacob Erlbeck141de622014-03-31 13:42:11 +0200182 msgb_free(ccon->pending_msg);
Harald Welte39c9e7b2014-08-22 00:28:51 +0200183
184 /* clean up deferred commands */
185 llist_for_each_entry_safe(cd, cd2, &ccon->def_cmds, list) {
186 /* delete from list of def_cmds for this ccon */
187 llist_del(&cd->list);
188 /* not strictly needed as this is a slave to the ccon which we
189 * are about to free anyway */
190 talloc_free(cd->cmd);
191 /* set the CMD to null, this is the indication to the user that
192 * the connection for this command has gone */
193 cd->cmd = NULL;
194 }
195
Daniel Willmann1264cb42010-10-21 15:00:36 +0200196 talloc_free(ccon);
197}
198
Harald Welte528134b2014-08-21 15:13:08 +0200199int ctrl_cmd_handle(struct ctrl_handle *ctrl, struct ctrl_cmd *cmd,
200 void *data)
201{
202 char *request;
203 int i, j, ret, node;
Max33c7ba62017-05-02 16:24:12 +0200204 bool break_cycle = false;
Harald Welte528134b2014-08-21 15:13:08 +0200205 vector vline, cmdvec, cmds_vec;
206
Max2ed36592017-02-24 14:48:45 +0100207 if (cmd->type == CTRL_TYPE_SET_REPLY ||
208 cmd->type == CTRL_TYPE_GET_REPLY) {
209 if (strncmp(cmd->reply, "OK", 2) == 0) {
210 LOGP(DLCTRL, LOGL_DEBUG, "%s <%s> for %s is OK\n",
211 get_value_string(ctrl_type_vals, cmd->type),
212 cmd->id, cmd->variable);
213 return CTRL_CMD_HANDLED;
214 }
215 }
216
217 if (cmd->type == CTRL_TYPE_ERROR) {
218 LOGP(DLCTRL, LOGL_ERROR, "%s <%s> for %s is %s\n",
219 get_value_string(ctrl_type_vals, cmd->type),
220 cmd->id, cmd->variable, cmd->reply);
221 return CTRL_CMD_HANDLED;
222 }
223
Harald Welte528134b2014-08-21 15:13:08 +0200224 ret = CTRL_CMD_ERROR;
225 cmd->reply = NULL;
226 node = CTRL_NODE_ROOT;
227 cmd->node = data;
228
229 request = talloc_strdup(cmd, cmd->variable);
230 if (!request)
231 goto err;
232
233 for (i=0;i<strlen(request);i++) {
234 if (request[i] == '.')
235 request[i] = ' ';
236 }
237
238 vline = cmd_make_strvec(request);
239 talloc_free(request);
240 if (!vline) {
241 cmd->reply = "cmd_make_strvec failed.";
242 goto err;
243 }
244
245 for (i=0;i<vector_active(vline);i++) {
Harald Welted6b1f852017-04-16 16:57:40 +0200246 struct lookup_helper *lh;
Harald Welte528134b2014-08-21 15:13:08 +0200247 int rc;
248
249 if (ctrl->lookup)
250 rc = ctrl->lookup(data, vline, &node, &cmd->node, &i);
251 else
252 rc = 0;
253
Harald Welted6b1f852017-04-16 16:57:40 +0200254 if (!rc) {
255 llist_for_each_entry(lh, &ctrl_lookup_helpers, list) {
256 rc = lh->lookup(data, vline, &node, &cmd->node, &i);
257 if (rc)
258 break;
259 }
260 }
261
Max33c7ba62017-05-02 16:24:12 +0200262 switch (rc) {
263 case 1: /* do nothing */
264 break;
265 case -ENODEV:
266 cmd_free_strvec(vline);
267 cmd->type = CTRL_TYPE_ERROR;
268 cmd->reply = "Error while resolving object";
269 return ret;
270 case -ERANGE:
271 cmd_free_strvec(vline);
272 cmd->type = CTRL_TYPE_ERROR;
273 cmd->reply = "Error while parsing the index.";
274 return ret;
275 default: /* If we're here the rest must be the command */
Harald Welte528134b2014-08-21 15:13:08 +0200276 cmdvec = vector_init(vector_active(vline)-i);
277 for (j=i; j<vector_active(vline); j++) {
278 vector_set(cmdvec, vector_slot(vline, j));
279 }
280
281 /* Get the command vector of the right node */
282 cmds_vec = vector_lookup(ctrl_node_vec, node);
283
284 if (!cmds_vec) {
285 cmd->reply = "Command not found.";
286 vector_free(cmdvec);
287 break;
288 }
289
290 ret = ctrl_cmd_exec(cmdvec, cmd, cmds_vec, data);
Harald Welte528134b2014-08-21 15:13:08 +0200291 vector_free(cmdvec);
Max33c7ba62017-05-02 16:24:12 +0200292 break_cycle = true;
Harald Welte528134b2014-08-21 15:13:08 +0200293 break;
294 }
295
Max33c7ba62017-05-02 16:24:12 +0200296 if (break_cycle)
297 break;
298
Harald Welte528134b2014-08-21 15:13:08 +0200299 if (i+1 == vector_active(vline))
300 cmd->reply = "Command not present.";
301 }
302
303 cmd_free_strvec(vline);
304
305err:
306 if (!cmd->reply) {
307 if (ret == CTRL_CMD_ERROR) {
Thorsten Alteholza81055d2017-03-02 22:13:48 +0100308 cmd->reply = "An error has occurred.";
Harald Welte528134b2014-08-21 15:13:08 +0200309 LOGP(DLCTRL, LOGL_NOTICE,
310 "%s: cmd->reply has not been set (ERROR).\n",
311 cmd->variable);
312 } else if (ret == CTRL_CMD_REPLY) {
313 LOGP(DLCTRL, LOGL_NOTICE,
314 "%s: cmd->reply has not been set (type = %d).\n",
315 cmd->variable, cmd->type);
316 cmd->reply = "";
317 } else {
318 cmd->reply = "Command has been handled.";
319 }
320 }
321
322 if (ret == CTRL_CMD_ERROR)
323 cmd->type = CTRL_TYPE_ERROR;
324 return ret;
Harald Welte528134b2014-08-21 15:13:08 +0200325}
326
327
Daniel Willmann1264cb42010-10-21 15:00:36 +0200328static int handle_control_read(struct osmo_fd * bfd)
329{
Pablo Neira Ayuso10dfc262011-08-17 22:44:07 +0200330 int ret = -1;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200331 struct osmo_wqueue *queue;
332 struct ctrl_connection *ccon;
Alexander Huemer1d3634e2011-10-12 00:29:30 +0200333 struct msgb *msg = NULL;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200334 struct ctrl_handle *ctrl = bfd->data;
335
336 queue = container_of(bfd, struct osmo_wqueue, bfd);
337 ccon = container_of(queue, struct ctrl_connection, write_queue);
338
Jacob Erlbeck141de622014-03-31 13:42:11 +0200339 ret = ipa_msg_recv_buffered(bfd->fd, &msg, &ccon->pending_msg);
Pablo Neira Ayuso10dfc262011-08-17 22:44:07 +0200340 if (ret <= 0) {
Jacob Erlbeck141de622014-03-31 13:42:11 +0200341 if (ret == -EAGAIN)
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100342 /* received part of a message, it is stored in ccon->pending_msg and there's
343 * nothing left to do now. */
Jacob Erlbeck141de622014-03-31 13:42:11 +0200344 return 0;
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100345 /* msg was already discarded. */
346 if (ret == 0) {
Harald Weltef360b422017-12-22 16:43:24 +0100347 control_close_conn(ccon);
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100348 ret = -EIO;
349 }
Daniel Willmann1264cb42010-10-21 15:00:36 +0200350 else
Harald Welte7fd0c832014-08-20 19:58:13 +0200351 LOGP(DLCTRL, LOGL_ERROR, "Failed to parse ip access message: %d\n", ret);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200352
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100353 return ret;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200354 }
355
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100356 ret = ctrl_handle_msg(ctrl, ccon, msg);
357 msgb_free(msg);
358 if (ret)
359 control_close_conn(ccon);
360 return ret;
361}
362
363int ctrl_handle_msg(struct ctrl_handle *ctrl, struct ctrl_connection *ccon, struct msgb *msg)
364{
365 struct ctrl_cmd *cmd;
366 struct ipaccess_head *iph;
367 struct ipaccess_head_ext *iph_ext;
368
Daniel Willmann1264cb42010-10-21 15:00:36 +0200369 if (msg->len < sizeof(*iph) + sizeof(*iph_ext)) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200370 LOGP(DLCTRL, LOGL_ERROR, "The message is too short.\n");
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100371 return -EINVAL;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200372 }
373
374 iph = (struct ipaccess_head *) msg->data;
375 if (iph->proto != IPAC_PROTO_OSMO) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200376 LOGP(DLCTRL, LOGL_ERROR, "Protocol mismatch. We got 0x%x\n", iph->proto);
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100377 return -EINVAL;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200378 }
379
380 iph_ext = (struct ipaccess_head_ext *) iph->data;
381 if (iph_ext->proto != IPAC_PROTO_EXT_CTRL) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200382 LOGP(DLCTRL, LOGL_ERROR, "Extended protocol mismatch. We got 0x%x\n", iph_ext->proto);
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100383 return -EINVAL;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200384 }
385
386 msg->l2h = iph_ext->data;
387
Neels Hofmeyrc0b0b622017-12-16 01:12:35 +0100388 cmd = ctrl_cmd_parse2(ccon, msg);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200389
Neels Hofmeyrc0b0b622017-12-16 01:12:35 +0100390 if (!cmd) {
391 /* should never happen */
Daniel Willmann1264cb42010-10-21 15:00:36 +0200392 cmd = talloc_zero(ccon, struct ctrl_cmd);
393 if (!cmd)
Neels Hofmeyrd53d2162017-12-16 03:57:56 +0100394 return -ENOMEM;
Harald Welte7fd0c832014-08-20 19:58:13 +0200395 LOGP(DLCTRL, LOGL_ERROR, "Command parser error.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200396 cmd->type = CTRL_TYPE_ERROR;
397 cmd->id = "err";
398 cmd->reply = "Command parser error.";
Daniel Willmann1264cb42010-10-21 15:00:36 +0200399 }
400
Neels Hofmeyrc0b0b622017-12-16 01:12:35 +0100401 if (cmd->type != CTRL_TYPE_ERROR) {
402 cmd->ccon = ccon;
403 if (ctrl_cmd_handle(ctrl, cmd, ctrl->data) == CTRL_CMD_HANDLED) {
404 /* On CTRL_CMD_HANDLED, no reply needs to be sent back. */
405 talloc_free(cmd);
406 cmd = NULL;
407 }
408 }
409
410 if (cmd) {
411 /* There is a reply or error that should be reported back to the sender. */
412 ctrl_cmd_send(&ccon->write_queue, cmd);
413 talloc_free(cmd);
414 }
415
Daniel Willmann1264cb42010-10-21 15:00:36 +0200416 return 0;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200417}
418
419static int control_write_cb(struct osmo_fd *bfd, struct msgb *msg)
420{
Harald Weltef360b422017-12-22 16:43:24 +0100421 struct osmo_wqueue *queue;
422 struct ctrl_connection *ccon;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200423 int rc;
424
Harald Weltef360b422017-12-22 16:43:24 +0100425 queue = container_of(bfd, struct osmo_wqueue, bfd);
426 ccon = container_of(queue, struct ctrl_connection, write_queue);
427
Daniel Willmann1264cb42010-10-21 15:00:36 +0200428 rc = write(bfd->fd, msg->data, msg->len);
Harald Weltef360b422017-12-22 16:43:24 +0100429 if (rc == 0)
430 control_close_conn(ccon);
431 else if (rc != msg->len)
Harald Welte59952812017-12-22 18:03:44 +0100432 LOGP(DLCTRL, LOGL_ERROR, "Failed to write message to the CTRL connection.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200433
434 return rc;
435}
436
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200437/*! Allocate CTRL connection
Maxfa9e05e2017-02-20 11:53:11 +0100438 * \param[in] ctx Context from which talloc should allocate it
439 * \param[in] data caller's private data parameter which should assigned to
440 write queue's file descriptor data parameter.
441 * \return Allocated CTRL connection structure or NULL in case of errors
442 */
443struct ctrl_connection *osmo_ctrl_conn_alloc(void *ctx, void *data)
Daniel Willmann1264cb42010-10-21 15:00:36 +0200444{
445 struct ctrl_connection *ccon = talloc_zero(ctx, struct ctrl_connection);
446 if (!ccon)
447 return NULL;
448
449 osmo_wqueue_init(&ccon->write_queue, 100);
450 /* Error handling here? */
451
452 INIT_LLIST_HEAD(&ccon->cmds);
Harald Welte39c9e7b2014-08-22 00:28:51 +0200453 INIT_LLIST_HEAD(&ccon->def_cmds);
454
Maxfa9e05e2017-02-20 11:53:11 +0100455 ccon->write_queue.bfd.data = data;
456 ccon->write_queue.write_cb = control_write_cb;
Max9756c462017-02-24 13:03:32 +0100457 ccon->write_queue.read_cb = handle_control_read;
Maxfa9e05e2017-02-20 11:53:11 +0100458
Daniel Willmann1264cb42010-10-21 15:00:36 +0200459 return ccon;
460}
461
462static int listen_fd_cb(struct osmo_fd *listen_bfd, unsigned int what)
463{
464 int ret, fd, on;
Daniel Willmann0e0cf372011-05-06 16:37:25 +0200465 struct ctrl_handle *ctrl;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200466 struct ctrl_connection *ccon;
Harald Welte59952812017-12-22 18:03:44 +0100467 char *name;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200468
469
470 if (!(what & BSC_FD_READ))
471 return 0;
472
Harald Welte59952812017-12-22 18:03:44 +0100473 fd = accept(listen_bfd->fd, NULL, NULL);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200474 if (fd < 0) {
475 perror("accept");
476 return fd;
477 }
Daniel Willmann1264cb42010-10-21 15:00:36 +0200478
Harald Weltee15ac062014-12-04 14:15:36 +0100479#ifdef TCP_NODELAY
Daniel Willmann1264cb42010-10-21 15:00:36 +0200480 on = 1;
481 ret = setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &on, sizeof(on));
482 if (ret != 0) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200483 LOGP(DLCTRL, LOGL_ERROR, "Failed to set TCP_NODELAY: %s\n", strerror(errno));
Daniel Willmann1264cb42010-10-21 15:00:36 +0200484 close(fd);
485 return ret;
486 }
Harald Weltee15ac062014-12-04 14:15:36 +0100487#endif
Maxfa9e05e2017-02-20 11:53:11 +0100488 ctrl = listen_bfd->data;
489 ccon = osmo_ctrl_conn_alloc(listen_bfd->data, ctrl);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200490 if (!ccon) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200491 LOGP(DLCTRL, LOGL_ERROR, "Failed to allocate.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200492 close(fd);
493 return -1;
494 }
495
Harald Welte59952812017-12-22 18:03:44 +0100496 name = osmo_sock_get_name(ccon, fd);
497 LOGP(DLCTRL, LOGL_INFO, "accept()ed new CTRL connection from %s\n", name);
498
Daniel Willmann1264cb42010-10-21 15:00:36 +0200499 ccon->write_queue.bfd.fd = fd;
500 ccon->write_queue.bfd.when = BSC_FD_READ;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200501
502 ret = osmo_fd_register(&ccon->write_queue.bfd);
503 if (ret < 0) {
Harald Welte7fd0c832014-08-20 19:58:13 +0200504 LOGP(DLCTRL, LOGL_ERROR, "Could not register FD.\n");
Daniel Willmann1264cb42010-10-21 15:00:36 +0200505 close(ccon->write_queue.bfd.fd);
506 talloc_free(ccon);
507 }
508
Daniel Willmann0e0cf372011-05-06 16:37:25 +0200509 llist_add(&ccon->list_entry, &ctrl->ccon_list);
510
Daniel Willmann1264cb42010-10-21 15:00:36 +0200511 return ret;
512}
513
Maxb214af52017-11-24 13:51:57 +0100514static uint64_t get_rate_ctr_value(const struct rate_ctr *ctr, int intv, const char *grp)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200515{
Maxb214af52017-11-24 13:51:57 +0100516 if (intv >= RATE_CTR_INTV_NUM) {
517 LOGP(DLCTRL, LOGL_ERROR, "Unexpected interval value %d while trying to get rate counter value in %s\n",
518 intv, grp);
Daniel Willmanne9f58942011-04-21 11:43:55 +0200519 return 0;
Maxb214af52017-11-24 13:51:57 +0100520 }
Daniel Willmanne9f58942011-04-21 11:43:55 +0200521
522 /* Absolute value */
523 if (intv == -1) {
524 return ctr->current;
525 } else {
526 return ctr->intv[intv].rate;
527 }
528}
529
Daniel Willmanne9f58942011-04-21 11:43:55 +0200530static int get_rate_ctr_group_idx(const struct rate_ctr_group *ctrg, int intv, struct ctrl_cmd *cmd)
531{
Max7dc8e882017-11-24 13:53:33 +0100532 unsigned int i;
533 for (i = 0; i < ctrg->desc->num_ctr; i++) {
534 ctrl_cmd_reply_printf(cmd, "%s %"PRIu64";", ctrg->desc->ctr_desc[i].name,
535 get_rate_ctr_value(&ctrg->ctr[i], intv, ctrg->desc->group_name_prefix));
536 if (!cmd->reply) {
537 cmd->reply = "OOM";
538 return CTRL_CMD_ERROR;
539 }
540 }
Daniel Willmanne9f58942011-04-21 11:43:55 +0200541
542 return CTRL_CMD_REPLY;
Daniel Willmanne9f58942011-04-21 11:43:55 +0200543}
544
Max52a38dd2017-11-23 20:34:02 +0100545static int ctrl_rate_ctr_group_handler(struct rate_ctr_group *ctrg, void *data)
546{
547 struct ctrl_cmd *cmd = data;
548
549 cmd->reply = talloc_asprintf_append(cmd->reply, "%s.%u;", ctrg->desc->group_name_prefix, ctrg->idx);
550 if (!cmd->reply) {
551 cmd->reply = "OOM";
552 return -1;
553 }
554
555 return 0;
556}
557
Daniel Willmanne9f58942011-04-21 11:43:55 +0200558/* rate_ctr */
559CTRL_CMD_DEFINE(rate_ctr, "rate_ctr *");
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200560static int get_rate_ctr(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200561{
562 int intv;
563 unsigned int idx;
Neels Hofmeyr5b34f772016-09-19 14:15:03 +0200564 char *ctr_group, *ctr_idx, *tmp, *dup, *saveptr, *interval;
Daniel Willmanne9f58942011-04-21 11:43:55 +0200565 struct rate_ctr_group *ctrg;
566 const struct rate_ctr *ctr;
567
568 dup = talloc_strdup(cmd, cmd->variable);
569 if (!dup)
570 goto oom;
571
572 /* Skip over possible prefixes (net.) */
573 tmp = strstr(dup, "rate_ctr");
574 if (!tmp) {
575 talloc_free(dup);
576 cmd->reply = "rate_ctr not a token in rate_ctr command!";
577 goto err;
578 }
579
580 strtok_r(tmp, ".", &saveptr);
581 interval = strtok_r(NULL, ".", &saveptr);
582 if (!interval) {
583 talloc_free(dup);
584 cmd->reply = "Missing interval.";
585 goto err;
586 }
587
588 if (!strcmp(interval, "abs")) {
589 intv = -1;
590 } else if (!strcmp(interval, "per_sec")) {
591 intv = RATE_CTR_INTV_SEC;
592 } else if (!strcmp(interval, "per_min")) {
593 intv = RATE_CTR_INTV_MIN;
594 } else if (!strcmp(interval, "per_hour")) {
595 intv = RATE_CTR_INTV_HOUR;
596 } else if (!strcmp(interval, "per_day")) {
597 intv = RATE_CTR_INTV_DAY;
Max52a38dd2017-11-23 20:34:02 +0100598 } else if (!strcmp(interval, "*")) {
599 intv = rate_ctr_for_each_group(ctrl_rate_ctr_group_handler, cmd);
600 if (intv < 0)
601 return CTRL_CMD_ERROR;
602 return CTRL_CMD_REPLY;
Daniel Willmanne9f58942011-04-21 11:43:55 +0200603 } else {
604 talloc_free(dup);
Maxb4718fd2017-11-24 11:28:35 +0100605 cmd->reply = "Wrong interval. Expecting 'per_sec', 'per_min', 'per_hour', 'per_day' or 'abs' value.";
Daniel Willmanne9f58942011-04-21 11:43:55 +0200606 goto err;
607 }
608
609 ctr_group = strtok_r(NULL, ".", &saveptr);
Max22886d92016-06-14 18:17:06 +0200610 ctr_idx = strtok_r(NULL, ".", &saveptr);
611 if (!ctr_group || !ctr_idx) {
Daniel Willmanne9f58942011-04-21 11:43:55 +0200612 talloc_free(dup);
Max22886d92016-06-14 18:17:06 +0200613 cmd->reply = "Counter group must be of name.index form e. g. "
614 "e1inp.0";
Daniel Willmanne9f58942011-04-21 11:43:55 +0200615 goto err;
616 }
Daniel Willmanne9f58942011-04-21 11:43:55 +0200617
Daniel Willmanne9f58942011-04-21 11:43:55 +0200618 idx = atoi(ctr_idx);
619
620 ctrg = rate_ctr_get_group_by_name_idx(ctr_group, idx);
621 if (!ctrg) {
622 talloc_free(dup);
Max22886d92016-06-14 18:17:06 +0200623 cmd->reply = "Counter group with given name and index not found";
Daniel Willmanne9f58942011-04-21 11:43:55 +0200624 goto err;
625 }
626
Max22886d92016-06-14 18:17:06 +0200627 if (!strlen(saveptr)) {
Daniel Willmanne9f58942011-04-21 11:43:55 +0200628 talloc_free(dup);
629 return get_rate_ctr_group_idx(ctrg, intv, cmd);
630 }
631
Max22886d92016-06-14 18:17:06 +0200632 ctr = rate_ctr_get_by_name(ctrg, saveptr);
Daniel Willmanne9f58942011-04-21 11:43:55 +0200633 if (!ctr) {
634 cmd->reply = "Counter name not found.";
635 talloc_free(dup);
636 goto err;
637 }
638
639 talloc_free(dup);
640
Maxb214af52017-11-24 13:51:57 +0100641 cmd->reply = talloc_asprintf(cmd, "%"PRIu64, get_rate_ctr_value(ctr, intv, ctrg->desc->group_name_prefix));
Daniel Willmanne9f58942011-04-21 11:43:55 +0200642 if (!cmd->reply)
643 goto oom;
644
645 return CTRL_CMD_REPLY;
646oom:
647 cmd->reply = "OOM";
648err:
649 return CTRL_CMD_ERROR;
650}
651
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200652static int set_rate_ctr(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200653{
654 cmd->reply = "Can't set rate counter.";
655
656 return CTRL_CMD_ERROR;
657}
658
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200659static int verify_rate_ctr(struct ctrl_cmd *cmd, const char *value, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200660{
661 return 0;
662}
663
664/* counter */
665CTRL_CMD_DEFINE(counter, "counter *");
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200666static int get_counter(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200667{
668 char *ctr_name, *tmp, *dup, *saveptr;
669 struct osmo_counter *counter;
670
671 cmd->reply = "OOM";
672 dup = talloc_strdup(cmd, cmd->variable);
673 if (!dup)
674 goto err;
675
676
677 tmp = strstr(dup, "counter");
678 if (!tmp) {
679 talloc_free(dup);
680 goto err;
681 }
682
683 strtok_r(tmp, ".", &saveptr);
684 ctr_name = strtok_r(NULL, "\0", &saveptr);
685
686 if (!ctr_name)
687 goto err;
688
689 counter = osmo_counter_get_by_name(ctr_name);
690 if (!counter) {
691 cmd->reply = "Counter name not found.";
692 talloc_free(dup);
693 goto err;
694 }
695
696 talloc_free(dup);
697
698 cmd->reply = talloc_asprintf(cmd, "%lu", counter->value);
699 if (!cmd->reply) {
700 cmd->reply = "OOM";
701 goto err;
702 }
703
704 return CTRL_CMD_REPLY;
705err:
706 return CTRL_CMD_ERROR;
707}
708
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200709static int set_counter(struct ctrl_cmd *cmd, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200710{
711
712 cmd->reply = "Can't set counter.";
713
714 return CTRL_CMD_ERROR;
715}
716
Daniel Willmannf7c74e52011-08-05 11:48:18 +0200717static int verify_counter(struct ctrl_cmd *cmd, const char *value, void *data)
Daniel Willmanne9f58942011-04-21 11:43:55 +0200718{
719 return 0;
720}
721
Harald Welteae2fcb22014-08-21 15:20:00 +0200722struct ctrl_handle *ctrl_interface_setup(void *data, uint16_t port,
723 ctrl_cmd_lookup lookup)
Daniel Willmann1264cb42010-10-21 15:00:36 +0200724{
Neels Hofmeyr49343092016-02-24 00:09:40 +0100725 return ctrl_interface_setup_dynip(data, "127.0.0.1", port, lookup);
726}
727
Harald Welte79c137c2017-04-16 19:08:49 +0200728static int ctrl_initialized = 0;
729
730/* global ctrl initialization */
Neels Hofmeyrea668522017-10-16 16:18:01 +0200731static int ctrl_init(unsigned int node_count)
Harald Welte79c137c2017-04-16 19:08:49 +0200732{
733 int ret;
734
Neels Hofmeyrea668522017-10-16 16:18:01 +0200735 if (!node_count)
736 node_count = _LAST_CTRL_NODE;
737 OSMO_ASSERT(node_count >= _LAST_CTRL_NODE);
Harald Welte79c137c2017-04-16 19:08:49 +0200738
Neels Hofmeyrea668522017-10-16 16:18:01 +0200739 if (ctrl_initialized) {
740 OSMO_ASSERT(ctrl_initialized == node_count);
741 return 0;
742 }
743
744 ctrl_node_vec = vector_init(node_count);
Harald Welte79c137c2017-04-16 19:08:49 +0200745 if (!ctrl_node_vec)
746 goto err;
747
748 ret = ctrl_cmd_install(CTRL_NODE_ROOT, &cmd_rate_ctr);
749 if (ret)
750 goto err_vec;
751 ret = ctrl_cmd_install(CTRL_NODE_ROOT, &cmd_counter);
752 if (ret)
753 goto err_vec;
754
Harald Welte31c0fef2017-04-16 17:26:30 +0200755 ret = osmo_fsm_ctrl_cmds_install();
756 if (ret)
757 goto err_vec;
758
Neels Hofmeyrea668522017-10-16 16:18:01 +0200759 ctrl_initialized = node_count;
Harald Welte79c137c2017-04-16 19:08:49 +0200760 return 0;
761
762err_vec:
763 vector_free(ctrl_node_vec);
764 ctrl_node_vec = NULL;
765err:
766 return -1;
767}
768
Neels Hofmeyrea668522017-10-16 16:18:01 +0200769/*! Allocate a CTRL interface handle.
770 * \param[in] ctx Talloc allocation context to be used
Harald Welte79c137c2017-04-16 19:08:49 +0200771 * \param[in] data Pointer which will be made available to each
772 set_..() get_..() verify_..() control command function
773 * \param[in] lookup Lookup function pointer, can be NULL
Neels Hofmeyrea668522017-10-16 16:18:01 +0200774 * \param[in] node_count Number of CTRL nodes to allocate, 0 for default.
Harald Welte79c137c2017-04-16 19:08:49 +0200775 * \returns ctrl_handle pointer or NULL in case of errors
Neels Hofmeyrea668522017-10-16 16:18:01 +0200776 *
777 * Please see ctrl_interface_setup_dynip2() for a detailed description of \a
778 * node_count semantics.
Harald Welte79c137c2017-04-16 19:08:49 +0200779 */
Neels Hofmeyrea668522017-10-16 16:18:01 +0200780struct ctrl_handle *ctrl_handle_alloc2(void *ctx, void *data,
781 ctrl_cmd_lookup lookup,
782 unsigned int node_count)
Harald Welte79c137c2017-04-16 19:08:49 +0200783{
784 struct ctrl_handle *ctrl;
785
Neels Hofmeyrea668522017-10-16 16:18:01 +0200786 ctrl_init(node_count);
Harald Welte79c137c2017-04-16 19:08:49 +0200787
788 ctrl = talloc_zero(ctx, struct ctrl_handle);
789 if (!ctrl)
790 return NULL;
791
792 INIT_LLIST_HEAD(&ctrl->ccon_list);
793
794 ctrl->data = data;
795 ctrl->lookup = lookup;
796
797 return ctrl;
798}
799
Neels Hofmeyrea668522017-10-16 16:18:01 +0200800/*! Allocate a CTRL interface handle.
801 * \param[in] ctx Talloc allocation context to be used
802 * \param[in] data Pointer which will be made available to each
803 set_..() get_..() verify_..() control command function
804 * \param[in] lookup Lookup function pointer, can be NULL
805 * \returns ctrl_handle pointer or NULL in case of errors
806 */
807struct ctrl_handle *ctrl_handle_alloc(void *ctx, void *data, ctrl_cmd_lookup lookup)
808{
809 return ctrl_handle_alloc2(ctx, data, lookup, 0);
810}
811
812/*! Setup CTRL interface on a given address.
Max32ee5af2017-02-16 17:44:09 +0100813 * \param[in] data Pointer which will be made available to each
814 set_..() get_..() verify_..() control command function
815 * \param[in] bind_addr Address on which CTRL socket shall listen
816 * \param[in] port Port on which CTRL socket shall listen
817 * \param[in] lookup Lookup function pointer, can be NULL
Neels Hofmeyrea668522017-10-16 16:18:01 +0200818 * \param[in] node_count Number of CTRL nodes to allocate, 0 for default.
Max32ee5af2017-02-16 17:44:09 +0100819 * \returns ctrl_handle pointer or NULL in case of errors
Neels Hofmeyrea668522017-10-16 16:18:01 +0200820 *
821 * Control interface nodes are identified by a node handle; some of these are
822 * defined in enum ctrl_node_type, here in libosmocore. However, applications
823 * defining own nodes may add own control nodes without having to extend the
824 * enum in libosmocore. For example, in the calling application, define an enum
825 * like "enum more_ctrl_nodes { CTRL_NODE_FOO = _LAST_CTRL_NODE, CTRL_NODE_BAR,
826 * _LAST_CTRL_NODE_EXTENDED }". In order to provide space for the additional
827 * control nodes, pass _LAST_CTRL_NODE_EXTENDED to the \a node_count parameter.
828 * Passing 0 is identical to passing _LAST_CTRL_NODE, i.e. to not define own
829 * control nodes apart from libosmocore ones.
Max32ee5af2017-02-16 17:44:09 +0100830 */
Neels Hofmeyrea668522017-10-16 16:18:01 +0200831struct ctrl_handle *ctrl_interface_setup_dynip2(void *data,
832 const char *bind_addr,
833 uint16_t port,
834 ctrl_cmd_lookup lookup,
835 unsigned int node_count)
Neels Hofmeyr49343092016-02-24 00:09:40 +0100836{
Daniel Willmann1264cb42010-10-21 15:00:36 +0200837 int ret;
838 struct ctrl_handle *ctrl;
839
Neels Hofmeyrea668522017-10-16 16:18:01 +0200840 ctrl = ctrl_handle_alloc2(data, data, lookup, node_count);
Daniel Willmann1264cb42010-10-21 15:00:36 +0200841 if (!ctrl)
Daniel Willmann782974b2011-07-22 17:55:42 +0200842 return NULL;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200843
Daniel Willmann1264cb42010-10-21 15:00:36 +0200844 /* Listen for control connections */
Harald Welte1238cc62014-08-20 19:50:04 +0200845 ctrl->listen_fd.cb = listen_fd_cb;
846 ctrl->listen_fd.data = ctrl;
847 ret = osmo_sock_init_ofd(&ctrl->listen_fd, AF_INET, SOCK_STREAM, IPPROTO_TCP,
Neels Hofmeyr49343092016-02-24 00:09:40 +0100848 bind_addr, port, OSMO_SOCK_F_BIND);
Harald Welte79c137c2017-04-16 19:08:49 +0200849 if (ret < 0) {
850 talloc_free(ctrl);
851 return NULL;
852 }
Daniel Willmanne9f58942011-04-21 11:43:55 +0200853
Neels Hofmeyr38d232e2016-09-19 14:17:30 +0200854 LOGP(DLCTRL, LOGL_NOTICE, "CTRL at %s %u\n", bind_addr, port);
Daniel Willmann782974b2011-07-22 17:55:42 +0200855 return ctrl;
Daniel Willmann1264cb42010-10-21 15:00:36 +0200856}
Harald Welted6b1f852017-04-16 16:57:40 +0200857
Neels Hofmeyrea668522017-10-16 16:18:01 +0200858/*! Setup CTRL interface on a given address.
859 * \param[in] data Pointer which will be made available to each
860 set_..() get_..() verify_..() control command function
861 * \param[in] bind_addr Address on which CTRL socket shall listen
862 * \param[in] port Port on which CTRL socket shall listen
863 * \param[in] lookup Lookup function pointer, can be NULL
864 * \returns ctrl_handle pointer or NULL in case of errors
865 */
866struct ctrl_handle *ctrl_interface_setup_dynip(void *data,
867 const char *bind_addr,
868 uint16_t port,
869 ctrl_cmd_lookup lookup)
870{
871 return ctrl_interface_setup_dynip2(data, bind_addr, port, lookup, 0);
872}
873
874
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200875/*! Install a lookup helper function for control nodes
Harald Welted6b1f852017-04-16 16:57:40 +0200876 * This function is used by e.g. library code to install lookup helpers
877 * for additional nodes in the control interface.
878 * \param[in] lookup The lookup helper function
879 * \retuns - on success; negative on error.
880 */
881int ctrl_lookup_register(ctrl_cmd_lookup lookup)
882{
883 struct lookup_helper *lh;
884
885 /* avoid double registration */
886 llist_for_each_entry(lh, &ctrl_lookup_helpers, list) {
887 if (lh->lookup == lookup)
888 return -EEXIST;
889 }
890
891 lh = talloc_zero(NULL, struct lookup_helper);
892 if (!lh)
893 return -ENOMEM;
894
895 lh->lookup = lookup;
896 llist_add_tail(&lh->list, &ctrl_lookup_helpers);
897 return 0;
898}
Harald Weltef85861d2017-04-16 19:17:10 +0200899
Neels Hofmeyr87e45502017-06-20 00:17:59 +0200900/*! Helper for "local execution" of a CTRL command from a string
Harald Weltef85861d2017-04-16 19:17:10 +0200901 * The function will parse + execute the given control command string
902 * and return a corresponding ctrl_cmd. Caller is responsible to
903 * talloc_free() the return value.
904 * \param[in] Control Interface Command String
905 * \returns parsed command, including reply; NULL on error */
906struct ctrl_cmd *ctrl_cmd_exec_from_string(struct ctrl_handle *ch, const char *cmdstr)
907{
908 struct msgb *msg = msgb_alloc(1024, "ctrl-cmd");
909 struct ctrl_cmd *cmd;
910
911 if (!msg)
912 return NULL;
913 msg->l2h = msg->data;
914 osmo_strlcpy((char *)msg->data, cmdstr, msgb_tailroom(msg));
915 msgb_put(msg, strlen(cmdstr));
916
Neels Hofmeyrc0b0b622017-12-16 01:12:35 +0100917 cmd = ctrl_cmd_parse2(ch, msg);
Harald Weltef85861d2017-04-16 19:17:10 +0200918 msgb_free(msg);
919 if (!cmd)
920 return NULL;
Neels Hofmeyrc0b0b622017-12-16 01:12:35 +0100921 if (cmd->type == CTRL_TYPE_ERROR)
922 return cmd;
923 if (ctrl_cmd_handle(ch, cmd, NULL) == CTRL_CMD_HANDLED) {
924 /* No reply should be sent back. */
Harald Weltef85861d2017-04-16 19:17:10 +0200925 talloc_free(cmd);
Neels Hofmeyrc0b0b622017-12-16 01:12:35 +0100926 cmd = NULL;
Harald Weltef85861d2017-04-16 19:17:10 +0200927 }
928 return cmd;
929}