/* IPA stream client example. */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include static LLIST_HEAD(msg_sent_list); struct msg_sent { struct llist_head head; struct msgb *msg; int num; struct timeval tv; }; #define DIPATEST 0 struct log_info_cat osmo_stream_client_test_cat[] = { [DIPATEST] = { .name = "DIPATEST", .description = "STREAMCLIENT-mode test", .color = "\033[1;35m", .enabled = 1, .loglevel = LOGL_NOTICE, }, }; const struct log_info osmo_stream_client_test_log_info = { .filter_fn = NULL, .cat = osmo_stream_client_test_cat, .num_cat = ARRAY_SIZE(osmo_stream_client_test_cat), }; static struct osmo_stream_cli *conn; void sighandler(int foo) { LOGP(DIPATEST, LOGL_NOTICE, "closing stream.\n"); exit(EXIT_SUCCESS); } static int disconnect_cb(struct osmo_stream_cli *conn) { LOGP(DIPATEST, LOGL_NOTICE, "disconnected\n"); return 0; } static int connect_cb(struct osmo_stream_cli *conn) { int *__num_msgs = osmo_stream_cli_get_data(conn); int num_msgs = *__num_msgs, i; LOGP(DIPATEST, LOGL_NOTICE, "connected\n"); for (i=0; imsg = msg; gettimeofday(&msg_sent->tv, NULL); msg_sent->num = i; llist_add(&msg_sent->head, &msg_sent_list); osmo_ipa_msg_push_headers(msg, osmo_ipa_msgb_cb_proto(msg), osmo_ipa_msgb_cb_proto_ext(msg)); osmo_stream_cli_send(conn, msg); LOGP(DIPATEST, LOGL_DEBUG, "enqueueing msg %d of " "%d bytes to be sent\n", i, msg->len); } return 0; } static int read_cb(struct osmo_stream_cli *conn, int res, struct msgb *msg) { int num; struct msg_sent *cur, *tmp, *found = NULL; LOGP(DIPATEST, LOGL_DEBUG, "received message from stream (payload len=%d)\n", msgb_length(msg)); if (res <= 0) { LOGP(DIPATEST, LOGL_ERROR, "Event with no data! %d\n", res); msgb_free(msg); return 0; } if (osmo_ipa_process_msg(msg) < 0) { LOGP(DIPATEST, LOGL_ERROR, "bad IPA message\n"); msgb_free(msg); return 0; } num = osmo_load32be(msg->data); LOGP(DLINP, LOGL_DEBUG, "received msg number %d\n", num); llist_for_each_entry_safe(cur, tmp, &msg_sent_list, head) { if (cur->num == num) { llist_del(&cur->head); found = cur; break; } } if (found) { struct timeval tv, diff; gettimeofday(&tv, NULL); timersub(&tv, &found->tv, &diff); LOGP(DLINP, LOGL_NOTICE, "message %d replied " "in %lu.%.6lu\n", num, diff.tv_sec, diff.tv_usec); talloc_free(found); } else { LOGP(DLINP, LOGL_ERROR, "message %d not found!\n", num); } msgb_free(msg); return 0; } static void *tall_test; int main(int argc, char *argv[]) { int num_msgs; signal(SIGINT, sighandler); if (argc != 2) { printf("Usage: %s [num_msgs]\n", argv[0]); exit(EXIT_FAILURE); } num_msgs = atoi(argv[1]); tall_test = talloc_named_const(NULL, 1, "osmo_stream_client_test"); msgb_talloc_ctx_init(tall_test, 0); osmo_init_logging2(tall_test, &osmo_stream_client_test_log_info); log_set_log_level(osmo_stderr_target, LOGL_DEBUG); /* * initialize stream client. */ conn = osmo_stream_cli_create(tall_test); if (conn == NULL) { fprintf(stderr, "cannot create client\n"); exit(EXIT_FAILURE); } osmo_stream_cli_set_name(conn, "ipa_test_client"); osmo_stream_cli_set_addr(conn, "127.0.0.1"); osmo_stream_cli_set_port(conn, 10000); 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); osmo_stream_cli_set_data(conn, &num_msgs); osmo_stream_cli_set_nodelay(conn, true); if (osmo_stream_cli_open(conn) < 0) { fprintf(stderr, "cannot open client\n"); exit(EXIT_FAILURE); } osmo_stream_cli_set_segmentation_cb(conn, osmo_ipa_segmentation_cb); LOGP(DIPATEST, LOGL_NOTICE, "Entering main loop\n"); while(1) { osmo_select_main(0); } }