123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239 |
- /*
- * Copyright (C) 2011 Felix Fietkau <nbd@openwrt.org>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU Lesser General Public License version 2.1
- * as published by the Free Software Foundation
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- */
- #include <sys/time.h>
- #include <unistd.h>
- #include <libubox/ustream.h>
- #include "libubus.h"
- #include "count.h"
- static struct ubus_context *ctx;
- static struct blob_buf b;
- static void test_client_subscribe_cb(struct ubus_context *ctx, struct ubus_object *obj)
- {
- fprintf(stderr, "Subscribers active: %d\n", obj->has_subscribers);
- }
- static struct ubus_object test_client_object = {
- .subscribe_cb = test_client_subscribe_cb,
- };
- static void test_client_notify_cb(struct uloop_timeout *timeout)
- {
- static int counter = 0;
- int err;
- struct timeval tv1, tv2;
- int max = 1000;
- long delta;
- int i = 0;
- blob_buf_init(&b, 0);
- blobmsg_add_u32(&b, "counter", counter++);
- gettimeofday(&tv1, NULL);
- for (i = 0; i < max; i++)
- err = ubus_notify(ctx, &test_client_object, "ping", b.head, 1000);
- gettimeofday(&tv2, NULL);
- if (err)
- fprintf(stderr, "Notify failed: %s\n", ubus_strerror(err));
- delta = (tv2.tv_sec - tv1.tv_sec) * 1000000 + (tv2.tv_usec - tv1.tv_usec);
- fprintf(stderr, "Avg time per iteration: %ld usec\n", delta / max);
- uloop_timeout_set(timeout, 1000);
- }
- enum {
- RETURN_CODE,
- __RETURN_MAX,
- };
- static const struct blobmsg_policy return_policy[__RETURN_MAX] = {
- [RETURN_CODE] = { .name = "rc", .type = BLOBMSG_TYPE_INT32 },
- };
- static void test_count_data_cb(struct ubus_request *req,
- int type, struct blob_attr *msg)
- {
- struct blob_attr *tb[__RETURN_MAX];
- int rc;
- uint32_t count_to = *(uint32_t *)req->priv;
- blobmsg_parse(return_policy, __RETURN_MAX, tb, blob_data(msg), blob_len(msg));
- if (!tb[RETURN_CODE]) {
- fprintf(stderr, "No return code received from server\n");
- return;
- }
- rc = blobmsg_get_u32(tb[RETURN_CODE]);
- if (rc)
- fprintf(stderr, "Corruption of data with count up to '%u'\n", count_to);
- else
- fprintf(stderr, "Server validated our count up to '%u'\n", count_to);
- }
- static void test_count(struct uloop_timeout *timeout)
- {
- enum {
- COUNT_TO_MIN = 10000,
- COUNT_TO_MAX = 1000000,
- PROGRESSION = 100,
- };
- uint32_t id;
- static uint32_t count_to = 100000;
- static int count_progression = PROGRESSION;
- char *s;
- if (count_to <= COUNT_TO_MIN)
- count_progression = PROGRESSION;
- else if (count_to >= COUNT_TO_MAX)
- count_progression = -PROGRESSION;
- count_to += count_progression;
- s = count_to_number(count_to);
- if (!s) {
- fprintf(stderr, "Could not allocate memory to count up to '%u'\n", count_to);
- return;
- }
- fprintf(stderr, "Sending count up to '%u'; string has length '%u'\n",
- count_to, (uint32_t)strlen(s));
- blob_buf_init(&b, 0);
- blobmsg_add_u32(&b, "to", count_to);
- blobmsg_add_string(&b, "string", s);
- if (ubus_lookup_id(ctx, "test", &id)) {
- free(s);
- fprintf(stderr, "Failed to look up test object\n");
- return;
- }
- ubus_invoke(ctx, id, "count", b.head, test_count_data_cb, &count_to, 5000);
- free(s);
- uloop_timeout_set(timeout, 2000);
- }
- static struct uloop_timeout notify_timer = {
- .cb = test_client_notify_cb,
- };
- static struct uloop_timeout count_timer = {
- .cb = test_count,
- };
- static void test_client_fd_data_cb(struct ustream *s, int bytes)
- {
- char *data, *sep;
- int len;
- data = ustream_get_read_buf(s, &len);
- if (len < 1)
- return;
- sep = strchr(data, '\n');
- if (!sep)
- return;
- *sep = 0;
- fprintf(stderr, "Got line: %s\n", data);
- ustream_consume(s, sep + 1 - data);
- }
- static void test_client_fd_cb(struct ubus_request *req, int fd)
- {
- static struct ustream_fd test_fd;
- fprintf(stderr, "Got fd from the server, watching...\n");
- test_fd.stream.notify_read = test_client_fd_data_cb;
- ustream_fd_init(&test_fd, fd);
- }
- static void test_client_complete_cb(struct ubus_request *req, int ret)
- {
- fprintf(stderr, "completed request, ret: %d\n", ret);
- }
- static void client_main(void)
- {
- static struct ubus_request req;
- uint32_t id;
- int ret;
- ret = ubus_add_object(ctx, &test_client_object);
- if (ret) {
- fprintf(stderr, "Failed to add_object object: %s\n", ubus_strerror(ret));
- return;
- }
- if (ubus_lookup_id(ctx, "test", &id)) {
- fprintf(stderr, "Failed to look up test object\n");
- return;
- }
- blob_buf_init(&b, 0);
- blobmsg_add_u32(&b, "id", test_client_object.id);
- ubus_invoke(ctx, id, "watch", b.head, NULL, 0, 3000);
- test_client_notify_cb(¬ify_timer);
- blob_buf_init(&b, 0);
- blobmsg_add_string(&b, "msg", "blah");
- ubus_invoke_async(ctx, id, "hello", b.head, &req);
- req.fd_cb = test_client_fd_cb;
- req.complete_cb = test_client_complete_cb;
- ubus_complete_request_async(ctx, &req);
- uloop_timeout_set(&count_timer, 2000);
- uloop_run();
- }
- int main(int argc, char **argv)
- {
- const char *ubus_socket = NULL;
- int ch;
- while ((ch = getopt(argc, argv, "cs:")) != -1) {
- switch (ch) {
- case 's':
- ubus_socket = optarg;
- break;
- default:
- break;
- }
- }
- uloop_init();
- ctx = ubus_connect(ubus_socket);
- if (!ctx) {
- fprintf(stderr, "Failed to connect to ubus\n");
- return -1;
- }
- ubus_add_uloop(ctx);
- client_main();
- ubus_free(ctx);
- uloop_done();
- return 0;
- }
|