2 * Copyright (C) 2011 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.
18 static struct ubus_context
*ctx
;
19 static struct ubus_watch_object test_event
;
20 static struct blob_buf b
;
28 static const struct blobmsg_policy hello_policy
[] = {
29 [HELLO_ID
] = { .name
= "id", .type
= BLOBMSG_TYPE_INT32
},
30 [HELLO_MSG
] = { .name
= "msg", .type
= BLOBMSG_TYPE_STRING
},
33 struct hello_request
{
34 struct ubus_request_data req
;
35 struct uloop_timeout timeout
;
39 static void test_hello_reply(struct uloop_timeout
*t
)
41 struct hello_request
*req
= container_of(t
, struct hello_request
, timeout
);
44 blobmsg_add_string(&b
, "message", req
->data
);
45 ubus_send_reply(ctx
, &req
->req
, b
.head
);
46 ubus_complete_deferred_request(ctx
, &req
->req
, 0);
50 static int test_hello(struct ubus_context
*ctx
, struct ubus_object
*obj
,
51 struct ubus_request_data
*req
, const char *method
,
52 struct blob_attr
*msg
)
54 struct hello_request
*hreq
;
55 struct blob_attr
*tb
[__HELLO_MAX
];
56 const char *format
= "%s received a message: %s";
57 const char *msgstr
= "(unknown)";
59 blobmsg_parse(hello_policy
, ARRAY_SIZE(hello_policy
), tb
, blob_data(msg
), blob_len(msg
));
62 msgstr
= blobmsg_data(tb
[HELLO_MSG
]);
64 hreq
= calloc(1, sizeof(*hreq
) + strlen(format
) + strlen(obj
->name
) + strlen(msgstr
) + 1);
65 sprintf(hreq
->data
, format
, obj
->name
, msgstr
);
66 ubus_defer_request(ctx
, req
, &hreq
->req
);
67 hreq
->timeout
.cb
= test_hello_reply
;
68 uloop_timeout_set(&hreq
->timeout
, 1000);
78 static const struct blobmsg_policy watch_policy
[__WATCH_MAX
] = {
79 [WATCH_ID
] = { .name
= "id", .type
= BLOBMSG_TYPE_INT32
},
82 static void test_handle_event(struct ubus_context
*ctx
, struct ubus_watch_object
*w
,
85 fprintf(stderr
, "Object %08x went away\n", id
);
88 static int test_watch(struct ubus_context
*ctx
, struct ubus_object
*obj
,
89 struct ubus_request_data
*req
, const char *method
,
90 struct blob_attr
*msg
)
92 struct blob_attr
*tb
[__WATCH_MAX
];
95 blobmsg_parse(watch_policy
, __WATCH_MAX
, tb
, blob_data(msg
), blob_len(msg
));
97 return UBUS_STATUS_INVALID_ARGUMENT
;
99 test_event
.cb
= test_handle_event
;
100 ret
= ubus_watch_object_add(ctx
, &test_event
, blobmsg_get_u32(tb
[WATCH_ID
]));
101 fprintf(stderr
, "Watching object %08x: %s\n", blobmsg_get_u32(tb
[WATCH_ID
]), ubus_strerror(ret
));
105 static const struct ubus_method test_methods
[] = {
106 UBUS_METHOD("hello", test_hello
, hello_policy
),
107 UBUS_METHOD("watch", test_watch
, watch_policy
),
110 static struct ubus_object_type test_object_type
=
111 UBUS_OBJECT_TYPE("test", test_methods
);
113 static struct ubus_object test_object
= {
115 .type
= &test_object_type
,
116 .methods
= test_methods
,
117 .n_methods
= ARRAY_SIZE(test_methods
),
120 static struct ubus_object test_client_object
= {
121 .type
= &test_object_type
,
122 .methods
= test_methods
,
123 .n_methods
= ARRAY_SIZE(test_methods
),
126 static void server_main(void)
130 ret
= ubus_add_object(ctx
, &test_object
);
132 fprintf(stderr
, "Failed to add object: %s\n", ubus_strerror(ret
));
134 ret
= ubus_register_watch_object(ctx
, &test_event
);
136 fprintf(stderr
, "Failed to add watch handler: %s\n", ubus_strerror(ret
));
141 static void client_main(void)
146 ret
= ubus_add_object(ctx
, &test_client_object
);
148 fprintf(stderr
, "Failed to add_object object: %s\n", ubus_strerror(ret
));
152 if (ubus_lookup_id(ctx
, test_object
.name
, &id
)) {
153 fprintf(stderr
, "Failed to look up test object\n");
157 blob_buf_init(&b
, 0);
158 blobmsg_add_u32(&b
, "id", test_client_object
.id
);
159 ubus_invoke(ctx
, id
, "watch", b
.head
, NULL
, 0, 3000);
163 int main(int argc
, char **argv
)
165 const char *ubus_socket
= NULL
;
169 while ((ch
= getopt(argc
, argv
, "cs:")) != -1) {
172 ubus_socket
= optarg
;
187 ctx
= ubus_connect(ubus_socket
);
189 fprintf(stderr
, "Failed to connect to ubus\n");