187 lines
4.1 KiB
C
187 lines
4.1 KiB
C
/* stream client example. */
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <netinet/in.h>
|
|
|
|
#include <osmocom/core/select.h>
|
|
#include <osmocom/core/talloc.h>
|
|
#include <osmocom/core/msgb.h>
|
|
#include <osmocom/core/logging.h>
|
|
#include <osmocom/core/application.h>
|
|
|
|
#include <osmocom/netif/stream.h>
|
|
|
|
#define DSTREAMTEST 0
|
|
|
|
struct log_info_cat osmo_stream_cli_test_cat[] = {
|
|
[DSTREAMTEST] = {
|
|
.name = "DSTREAMTEST",
|
|
.description = "STREAMCLIENT-mode test",
|
|
.color = "\033[1;35m",
|
|
.enabled = 1, .loglevel = LOGL_DEBUG,
|
|
},
|
|
};
|
|
|
|
const struct log_info osmo_stream_cli_test_log_info = {
|
|
.filter_fn = NULL,
|
|
.cat = osmo_stream_cli_test_cat,
|
|
.num_cat = ARRAY_SIZE(osmo_stream_cli_test_cat),
|
|
};
|
|
|
|
static struct osmo_stream_cli *conn;
|
|
|
|
static bool quit = false;
|
|
|
|
void sighandler(int foo)
|
|
{
|
|
LOGP(DSTREAMTEST, LOGL_NOTICE, "closing stream.\n");
|
|
quit = true;
|
|
signal(SIGINT, SIG_DFL);
|
|
}
|
|
|
|
static int connect_cb(struct osmo_stream_cli *conn)
|
|
{
|
|
LOGP(DSTREAMTEST, LOGL_NOTICE, "connected: %s\n", osmo_stream_cli_get_sockname(conn));
|
|
return 0;
|
|
}
|
|
|
|
static int disconnect_cb(struct osmo_stream_cli *conn)
|
|
{
|
|
LOGP(DSTREAMTEST, LOGL_NOTICE, "disconnected: %s\n", osmo_stream_cli_get_sockname(conn));
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int read_cb(struct osmo_stream_cli *conn, int res, struct msgb *msg)
|
|
{
|
|
LOGP(DSTREAMTEST, LOGL_NOTICE, "receiving message from stream... ");
|
|
|
|
if (res < 0) {
|
|
LOGPC(DSTREAMTEST, LOGL_ERROR, "cannot receive message (res = %d)\n", res);
|
|
msgb_free(msg);
|
|
return 0;
|
|
}
|
|
|
|
|
|
LOGPC(DSTREAMTEST, LOGL_NOTICE, "got %d bytes: %s\n", msg->len, msgb_hexdump(msg));
|
|
|
|
msgb_free(msg);
|
|
return 0;
|
|
}
|
|
|
|
static void *tall_test;
|
|
|
|
static int kbd_cb(struct osmo_fd *fd, unsigned int what)
|
|
{
|
|
char buf[1024];
|
|
struct msgb *msg;
|
|
uint8_t *ptr;
|
|
int ret;
|
|
|
|
ret = read(STDIN_FILENO, buf, sizeof(buf));
|
|
LOGP(DSTREAMTEST, LOGL_NOTICE, "read %d byte from keyboard\n", ret);
|
|
if (ret < 0)
|
|
return ret;
|
|
|
|
msg = msgb_alloc(1024, "STREAMCLIENT/test");
|
|
if (msg == NULL) {
|
|
LOGP(DSTREAMTEST, LOGL_ERROR, "cannot allocate message\n");
|
|
return 0;
|
|
}
|
|
ptr = msgb_put(msg, ret);
|
|
memcpy(ptr, buf, ret);
|
|
|
|
LOGP(DSTREAMTEST, LOGL_NOTICE, "sending %d bytes message: %s\n", msg->len, msgb_hexdump(msg));
|
|
|
|
osmo_stream_cli_send(conn, msg);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void signal_handler(int signum)
|
|
{
|
|
switch (signum) {
|
|
case SIGUSR1:
|
|
talloc_report_full(tall_test, stdout);
|
|
break;
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
struct osmo_fd *kbd_ofd;
|
|
bool use_sctp = false;
|
|
const char *use_remote_addr = "127.0.0.1";
|
|
int opt, rc;
|
|
|
|
while ((opt = getopt(argc, argv, "sr:")) != -1) {
|
|
switch (opt) {
|
|
case 's':
|
|
use_sctp = true;
|
|
break;
|
|
case 'r':
|
|
use_remote_addr = optarg;
|
|
break;
|
|
default:
|
|
exit(0);
|
|
}
|
|
}
|
|
|
|
signal(SIGUSR1, &signal_handler);
|
|
|
|
tall_test = talloc_named_const(NULL, 1, "osmo_stream_cli_test");
|
|
msgb_talloc_ctx_init(tall_test, 0);
|
|
osmo_init_logging2(tall_test, &osmo_stream_cli_test_log_info);
|
|
log_set_log_level(osmo_stderr_target, LOGL_DEBUG);
|
|
|
|
/*
|
|
* initialize stream cli.
|
|
*/
|
|
|
|
conn = osmo_stream_cli_create(tall_test);
|
|
if (conn == NULL) {
|
|
fprintf(stderr, "cannot create cli\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
osmo_stream_cli_set_name(conn, "stream_client");
|
|
osmo_stream_cli_set_addr(conn, use_remote_addr);
|
|
osmo_stream_cli_set_port(conn, 10000);
|
|
if (use_sctp)
|
|
osmo_stream_cli_set_proto(conn, IPPROTO_SCTP);
|
|
osmo_stream_cli_set_connect_cb(conn, connect_cb);
|
|
osmo_stream_cli_set_disconnect_cb(conn, disconnect_cb);
|
|
osmo_stream_cli_set_read_cb2(conn, read_cb);
|
|
|
|
rc = osmo_stream_cli_open(conn);
|
|
if (rc < 0) {
|
|
fprintf(stderr, "cannot open cli: %d\n", rc);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
kbd_ofd = talloc_zero(tall_test, struct osmo_fd);
|
|
if (!kbd_ofd) {
|
|
LOGP(DSTREAMTEST, LOGL_ERROR, "OOM\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
osmo_fd_setup(kbd_ofd, STDIN_FILENO, OSMO_FD_READ, kbd_cb, conn, 0);
|
|
rc = osmo_fd_register(kbd_ofd);
|
|
if (rc < 0) {
|
|
LOGP(DSTREAMTEST, LOGL_ERROR, "FD Register\n");
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
signal(SIGINT, sighandler);
|
|
|
|
LOGP(DSTREAMTEST, LOGL_NOTICE, "Entering main loop\n");
|
|
|
|
while (!quit) {
|
|
osmo_select_main(0);
|
|
}
|
|
|
|
osmo_fd_unregister(kbd_ofd);
|
|
|
|
osmo_stream_cli_destroy(conn);
|
|
}
|