examples/client: receive data from the passed file descriptor
[project/ubus.git] / examples / server.c
1 /*
2  * Copyright (C) 2011-2014 Felix Fietkau <nbd@openwrt.org>
3  *
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
7  *
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.
12  */
13
14 #include <unistd.h>
15 #include <signal.h>
16
17 #include <libubox/blobmsg_json.h>
18 #include "libubus.h"
19
20 static struct ubus_context *ctx;
21 static struct ubus_subscriber test_event;
22 static struct blob_buf b;
23
24 enum {
25         HELLO_ID,
26         HELLO_MSG,
27         __HELLO_MAX
28 };
29
30 static const struct blobmsg_policy hello_policy[] = {
31         [HELLO_ID] = { .name = "id", .type = BLOBMSG_TYPE_INT32 },
32         [HELLO_MSG] = { .name = "msg", .type = BLOBMSG_TYPE_STRING },
33 };
34
35 struct hello_request {
36         struct ubus_request_data req;
37         struct uloop_timeout timeout;
38         int fd;
39         int idx;
40         char data[];
41 };
42
43 static void test_hello_fd_reply(struct uloop_timeout *t)
44 {
45         struct hello_request *req = container_of(t, struct hello_request, timeout);
46         char *data;
47
48         data = alloca(strlen(req->data) + 32);
49         sprintf(data, "msg%d: %s\n", ++req->idx, req->data);
50         if (write(req->fd, data, strlen(data)) < 0) {
51                 close(req->fd);
52                 free(req);
53                 return;
54         }
55
56         uloop_timeout_set(&req->timeout, 1000);
57 }
58
59 static void test_hello_reply(struct uloop_timeout *t)
60 {
61         struct hello_request *req = container_of(t, struct hello_request, timeout);
62         int fds[2];
63
64         blob_buf_init(&b, 0);
65         blobmsg_add_string(&b, "message", req->data);
66         ubus_send_reply(ctx, &req->req, b.head);
67
68         pipe(fds);
69         ubus_request_set_fd(ctx, &req->req, fds[0]);
70         ubus_complete_deferred_request(ctx, &req->req, 0);
71         close(fds[0]);
72         req->fd = fds[1];
73
74         req->timeout.cb = test_hello_fd_reply;
75         test_hello_fd_reply(t);
76 }
77
78 static int test_hello(struct ubus_context *ctx, struct ubus_object *obj,
79                       struct ubus_request_data *req, const char *method,
80                       struct blob_attr *msg)
81 {
82         struct hello_request *hreq;
83         struct blob_attr *tb[__HELLO_MAX];
84         const char *format = "%s received a message: %s";
85         const char *msgstr = "(unknown)";
86
87         blobmsg_parse(hello_policy, ARRAY_SIZE(hello_policy), tb, blob_data(msg), blob_len(msg));
88
89         if (tb[HELLO_MSG])
90                 msgstr = blobmsg_data(tb[HELLO_MSG]);
91
92         hreq = calloc(1, sizeof(*hreq) + strlen(format) + strlen(obj->name) + strlen(msgstr) + 1);
93         sprintf(hreq->data, format, obj->name, msgstr);
94         ubus_defer_request(ctx, req, &hreq->req);
95         hreq->timeout.cb = test_hello_reply;
96         uloop_timeout_set(&hreq->timeout, 1000);
97
98         return 0;
99 }
100
101 enum {
102         WATCH_ID,
103         WATCH_COUNTER,
104         __WATCH_MAX
105 };
106
107 static const struct blobmsg_policy watch_policy[__WATCH_MAX] = {
108         [WATCH_ID] = { .name = "id", .type = BLOBMSG_TYPE_INT32 },
109         [WATCH_COUNTER] = { .name = "counter", .type = BLOBMSG_TYPE_INT32 },
110 };
111
112 static void
113 test_handle_remove(struct ubus_context *ctx, struct ubus_subscriber *s,
114                    uint32_t id)
115 {
116         fprintf(stderr, "Object %08x went away\n", id);
117 }
118
119 static int
120 test_notify(struct ubus_context *ctx, struct ubus_object *obj,
121             struct ubus_request_data *req, const char *method,
122             struct blob_attr *msg)
123 {
124 #if 0
125         char *str;
126
127         str = blobmsg_format_json(msg, true);
128         fprintf(stderr, "Received notification '%s': %s\n", method, str);
129         free(str);
130 #endif
131
132         return 0;
133 }
134
135 static int test_watch(struct ubus_context *ctx, struct ubus_object *obj,
136                       struct ubus_request_data *req, const char *method,
137                       struct blob_attr *msg)
138 {
139         struct blob_attr *tb[__WATCH_MAX];
140         int ret;
141
142         blobmsg_parse(watch_policy, __WATCH_MAX, tb, blob_data(msg), blob_len(msg));
143         if (!tb[WATCH_ID])
144                 return UBUS_STATUS_INVALID_ARGUMENT;
145
146         test_event.remove_cb = test_handle_remove;
147         test_event.cb = test_notify;
148         ret = ubus_subscribe(ctx, &test_event, blobmsg_get_u32(tb[WATCH_ID]));
149         fprintf(stderr, "Watching object %08x: %s\n", blobmsg_get_u32(tb[WATCH_ID]), ubus_strerror(ret));
150         return ret;
151 }
152
153 static const struct ubus_method test_methods[] = {
154         UBUS_METHOD("hello", test_hello, hello_policy),
155         UBUS_METHOD("watch", test_watch, watch_policy),
156 };
157
158 static struct ubus_object_type test_object_type =
159         UBUS_OBJECT_TYPE("test", test_methods);
160
161 static struct ubus_object test_object = {
162         .name = "test",
163         .type = &test_object_type,
164         .methods = test_methods,
165         .n_methods = ARRAY_SIZE(test_methods),
166 };
167
168 static void server_main(void)
169 {
170         int ret;
171
172         ret = ubus_add_object(ctx, &test_object);
173         if (ret)
174                 fprintf(stderr, "Failed to add object: %s\n", ubus_strerror(ret));
175
176         ret = ubus_register_subscriber(ctx, &test_event);
177         if (ret)
178                 fprintf(stderr, "Failed to add watch handler: %s\n", ubus_strerror(ret));
179
180         uloop_run();
181 }
182
183 int main(int argc, char **argv)
184 {
185         const char *ubus_socket = NULL;
186         int ch;
187
188         while ((ch = getopt(argc, argv, "cs:")) != -1) {
189                 switch (ch) {
190                 case 's':
191                         ubus_socket = optarg;
192                         break;
193                 default:
194                         break;
195                 }
196         }
197
198         argc -= optind;
199         argv += optind;
200
201         uloop_init();
202         signal(SIGPIPE, SIG_IGN);
203
204         ctx = ubus_connect(ubus_socket);
205         if (!ctx) {
206                 fprintf(stderr, "Failed to connect to ubus\n");
207                 return -1;
208         }
209
210         ubus_add_uloop(ctx);
211
212         server_main();
213
214         ubus_free(ctx);
215         uloop_done();
216
217         return 0;
218 }