/* stream client example. */ #include #include #include #include #include #include #include #include #include #include #include #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); }