2 * Copyright (C) 2011-2014 Felix Fietkau <nbd@openwrt.org>
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU Lesser General Public License version 2.1
6 * as published by the Free Software Foundation
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
17 #include <libubox/blobmsg_json.h>
21 static struct ubus_context *ctx;
22 static struct ubus_subscriber test_event;
23 static struct blob_buf b;
31 static const struct blobmsg_policy hello_policy[] = {
32 [HELLO_ID] = { .name = "id", .type = BLOBMSG_TYPE_INT32 },
33 [HELLO_MSG] = { .name = "msg", .type = BLOBMSG_TYPE_STRING },
36 struct hello_request {
37 struct ubus_request_data req;
38 struct uloop_timeout timeout;
44 static void test_hello_fd_reply(struct uloop_timeout *t)
46 struct hello_request *req = container_of(t, struct hello_request, timeout);
49 data = alloca(strlen(req->data) + 32);
50 sprintf(data, "msg%d: %s\n", ++req->idx, req->data);
51 if (write(req->fd, data, strlen(data)) < 0) {
57 uloop_timeout_set(&req->timeout, 1000);
60 static void test_hello_reply(struct uloop_timeout *t)
62 struct hello_request *req = container_of(t, struct hello_request, timeout);
66 blobmsg_add_string(&b, "message", req->data);
67 ubus_send_reply(ctx, &req->req, b.head);
69 if (pipe(fds) == -1) {
70 fprintf(stderr, "Failed to create pipe\n");
73 ubus_request_set_fd(ctx, &req->req, fds[0]);
74 ubus_complete_deferred_request(ctx, &req->req, 0);
77 req->timeout.cb = test_hello_fd_reply;
78 test_hello_fd_reply(t);
81 static int test_hello(struct ubus_context *ctx, struct ubus_object *obj,
82 struct ubus_request_data *req, const char *method,
83 struct blob_attr *msg)
85 struct hello_request *hreq;
86 struct blob_attr *tb[__HELLO_MAX];
87 const char format[] = "%s received a message: %s";
88 const char *msgstr = "(unknown)";
90 blobmsg_parse(hello_policy, ARRAY_SIZE(hello_policy), tb, blob_data(msg), blob_len(msg));
93 msgstr = blobmsg_data(tb[HELLO_MSG]);
95 size_t len = sizeof(*hreq) + sizeof(format) + strlen(obj->name) + strlen(msgstr) + 1;
96 hreq = calloc(1, len);
98 return UBUS_STATUS_UNKNOWN_ERROR;
100 snprintf(hreq->data, len, format, obj->name, msgstr);
101 ubus_defer_request(ctx, req, &hreq->req);
102 hreq->timeout.cb = test_hello_reply;
103 uloop_timeout_set(&hreq->timeout, 1000);
114 static const struct blobmsg_policy watch_policy[__WATCH_MAX] = {
115 [WATCH_ID] = { .name = "id", .type = BLOBMSG_TYPE_INT32 },
116 [WATCH_COUNTER] = { .name = "counter", .type = BLOBMSG_TYPE_INT32 },
120 test_handle_remove(struct ubus_context *ctx, struct ubus_subscriber *s,
123 fprintf(stderr, "Object %08x went away\n", id);
127 test_notify(struct ubus_context *ctx, struct ubus_object *obj,
128 struct ubus_request_data *req, const char *method,
129 struct blob_attr *msg)
134 str = blobmsg_format_json(msg, true);
135 fprintf(stderr, "Received notification '%s': %s\n", method, str);
142 static int test_watch(struct ubus_context *ctx, struct ubus_object *obj,
143 struct ubus_request_data *req, const char *method,
144 struct blob_attr *msg)
146 struct blob_attr *tb[__WATCH_MAX];
149 blobmsg_parse(watch_policy, __WATCH_MAX, tb, blob_data(msg), blob_len(msg));
151 return UBUS_STATUS_INVALID_ARGUMENT;
153 test_event.remove_cb = test_handle_remove;
154 test_event.cb = test_notify;
155 ret = ubus_subscribe(ctx, &test_event, blobmsg_get_u32(tb[WATCH_ID]));
156 fprintf(stderr, "Watching object %08x: %s\n", blobmsg_get_u32(tb[WATCH_ID]), ubus_strerror(ret));
166 static const struct blobmsg_policy count_policy[__COUNT_MAX] = {
167 [COUNT_TO] = { .name = "to", .type = BLOBMSG_TYPE_INT32 },
168 [COUNT_STRING] = { .name = "string", .type = BLOBMSG_TYPE_STRING },
171 static int test_count(struct ubus_context *ctx, struct ubus_object *obj,
172 struct ubus_request_data *req, const char *method,
173 struct blob_attr *msg)
175 struct blob_attr *tb[__COUNT_MAX];
179 blobmsg_parse(count_policy, __COUNT_MAX, tb, blob_data(msg), blob_len(msg));
180 if (!tb[COUNT_TO] || !tb[COUNT_STRING])
181 return UBUS_STATUS_INVALID_ARGUMENT;
183 num = blobmsg_get_u32(tb[COUNT_TO]);
184 s1 = blobmsg_get_string(tb[COUNT_STRING]);
185 s2 = count_to_number(num);
188 return UBUS_STATUS_UNKNOWN_ERROR;
190 blob_buf_init(&b, 0);
191 blobmsg_add_u32(&b, "rc", strcmp(s1, s2));
192 ubus_send_reply(ctx, req, b.head);
198 static const struct ubus_method test_methods[] = {
199 UBUS_METHOD("hello", test_hello, hello_policy),
200 UBUS_METHOD("watch", test_watch, watch_policy),
201 UBUS_METHOD("count", test_count, count_policy),
204 static struct ubus_object_type test_object_type =
205 UBUS_OBJECT_TYPE("test", test_methods);
207 static struct ubus_object test_object = {
209 .type = &test_object_type,
210 .methods = test_methods,
211 .n_methods = ARRAY_SIZE(test_methods),
214 static void server_main(void)
218 ret = ubus_add_object(ctx, &test_object);
220 fprintf(stderr, "Failed to add object: %s\n", ubus_strerror(ret));
222 ret = ubus_register_subscriber(ctx, &test_event);
224 fprintf(stderr, "Failed to add watch handler: %s\n", ubus_strerror(ret));
229 int main(int argc, char **argv)
231 const char *ubus_socket = NULL;
234 while ((ch = getopt(argc, argv, "cs:")) != -1) {
237 ubus_socket = optarg;
245 signal(SIGPIPE, SIG_IGN);
247 ctx = ubus_connect(ubus_socket);
249 fprintf(stderr, "Failed to connect to ubus\n");