2 * rpcd - UBUS RPC server
4 * Copyright (C) 2013-2014 Jo-Philipp Wich <jow@openwrt.org>
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19 #include <rpcd/plugin.h>
21 static struct blob_buf buf
;
23 struct rpc_plugin_lookup_context
{
30 rpc_plugin_lookup_plugin_cb(struct ubus_context
*ctx
,
31 struct ubus_object_data
*obj
, void *priv
)
33 struct rpc_plugin_lookup_context
*c
= priv
;
38 sprintf(c
->name
, "%s", obj
->path
);
43 rpc_plugin_lookup_plugin(struct ubus_context
*ctx
, struct ubus_object
*obj
,
46 struct rpc_plugin_lookup_context c
= { .id
= obj
->id
, .name
= strptr
};
48 if (ubus_lookup(ctx
, NULL
, rpc_plugin_lookup_plugin_cb
, &c
))
66 rpc_plugin_call_stdin_cb(struct ustream
*s
, void *priv
)
68 struct call_context
*c
= priv
;
72 ustream_write(s
, c
->input
, strlen(c
->input
), false);
80 rpc_plugin_call_stdout_cb(struct blob_buf
*blob
, char *buf
, int len
, void *priv
)
82 struct call_context
*c
= priv
;
86 c
->obj
= json_tokener_parse_ex(c
->tok
, buf
, len
);
88 if (json_tokener_get_error(c
->tok
) != json_tokener_continue
)
89 c
->output_done
= true;
96 rpc_plugin_call_stderr_cb(struct blob_buf
*blob
, char *buf
, int len
, void *priv
)
102 rpc_plugin_call_finish_cb(struct blob_buf
*blob
, int stat
, void *priv
)
104 struct call_context
*c
= priv
;
105 int rv
= UBUS_STATUS_INVALID_ARGUMENT
;
107 if (json_tokener_get_error(c
->tok
) == json_tokener_success
)
111 if (json_object_get_type(c
->obj
) == json_type_object
&&
112 blobmsg_add_object(blob
, c
->obj
))
115 json_object_put(c
->obj
);
119 rv
= UBUS_STATUS_NO_DATA
;
123 json_tokener_free(c
->tok
);
131 rpc_plugin_call(struct ubus_context
*ctx
, struct ubus_object
*obj
,
132 struct ubus_request_data
*req
, const char *method
,
133 struct blob_attr
*msg
)
135 int rv
= UBUS_STATUS_UNKNOWN_ERROR
;
136 struct call_context
*c
;
139 c
= calloc_a(sizeof(*c
), &mptr
, strlen(method
) + 1);
144 c
->method
= strcpy(mptr
, method
);
145 c
->input
= blobmsg_format_json(msg
, true);
146 c
->tok
= json_tokener_new();
148 if (!c
->input
|| !c
->tok
)
151 plugin
= c
->path
+ sprintf(c
->path
, "%s/", RPC_PLUGIN_DIRECTORY
);
153 if (!rpc_plugin_lookup_plugin(ctx
, obj
, plugin
))
155 rv
= UBUS_STATUS_NOT_FOUND
;
159 c
->argv
[0] = c
->path
;
161 c
->argv
[2] = c
->method
;
163 rv
= rpc_exec(c
->argv
, rpc_plugin_call_stdin_cb
,
164 rpc_plugin_call_stdout_cb
, rpc_plugin_call_stderr_cb
,
165 rpc_plugin_call_finish_cb
, c
, ctx
, req
);
167 if (rv
== UBUS_STATUS_OK
)
177 json_tokener_free(c
->tok
);
186 rpc_plugin_parse_signature(struct blob_attr
*sig
, struct ubus_method
*method
)
189 enum blobmsg_type type
;
190 struct blob_attr
*attr
;
191 struct blobmsg_policy
*policy
= NULL
;
193 if (!sig
|| blobmsg_type(sig
) != BLOBMSG_TYPE_TABLE
)
198 blobmsg_for_each_attr(attr
, sig
, rem
)
203 policy
= calloc(n_attr
, sizeof(*policy
));
210 blobmsg_for_each_attr(attr
, sig
, rem
)
212 type
= blobmsg_type(attr
);
214 if (type
== BLOBMSG_TYPE_INT32
)
216 switch (blobmsg_get_u32(attr
))
219 type
= BLOBMSG_TYPE_INT8
;
223 type
= BLOBMSG_TYPE_INT16
;
227 type
= BLOBMSG_TYPE_INT64
;
231 type
= BLOBMSG_TYPE_INT32
;
236 policy
[n_attr
].name
= strdup(blobmsg_name(attr
));
237 policy
[n_attr
].type
= type
;
243 method
->name
= strdup(blobmsg_name(sig
));
244 method
->handler
= rpc_plugin_call
;
245 method
->policy
= policy
;
246 method
->n_policy
= n_attr
;
251 static struct ubus_object
*
252 rpc_plugin_parse_exec(const char *name
, int fd
)
254 int len
, rem
, n_method
;
255 struct blob_attr
*cur
;
256 struct ubus_method
*methods
;
257 struct ubus_object_type
*obj_type
;
258 struct ubus_object
*obj
;
264 blob_buf_init(&buf
, 0);
266 tok
= json_tokener_new();
271 while ((len
= read(fd
, outbuf
, sizeof(outbuf
))) > 0)
273 jsobj
= json_tokener_parse_ex(tok
, outbuf
, len
);
275 if (json_tokener_get_error(tok
) == json_tokener_continue
)
278 if (json_tokener_get_error(tok
) != json_tokener_success
)
283 if (json_object_get_type(jsobj
) == json_type_object
)
284 blobmsg_add_object(&buf
, jsobj
);
286 json_object_put(jsobj
);
291 json_tokener_free(tok
);
295 blob_for_each_attr(cur
, buf
.head
, rem
)
301 methods
= calloc(n_method
, sizeof(*methods
));
308 blob_for_each_attr(cur
, buf
.head
, rem
)
310 if (!rpc_plugin_parse_signature(cur
, &methods
[n_method
]))
316 obj
= calloc(1, sizeof(*obj
));
321 obj_type
= calloc(1, sizeof(*obj_type
));
328 if (asprintf((char **)&obj_type
->name
, "luci-rpc-plugin-%s", name
) < 0) {
334 obj_type
->methods
= methods
;
335 obj_type
->n_methods
= n_method
;
337 obj
->name
= strdup(name
);
338 obj
->type
= obj_type
;
339 obj
->methods
= methods
;
340 obj
->n_methods
= n_method
;
346 rpc_plugin_register_exec(struct ubus_context
*ctx
, const char *path
)
349 int rv
= UBUS_STATUS_NO_DATA
, fd
, fds
[2];
351 struct ubus_object
*plugin
;
353 name
= strrchr(path
, '/');
356 return UBUS_STATUS_INVALID_ARGUMENT
;
359 return UBUS_STATUS_UNKNOWN_ERROR
;
361 switch ((pid
= fork()))
364 return UBUS_STATUS_UNKNOWN_ERROR
;
367 fd
= open("/dev/null", O_RDWR
);
383 if (execl(path
, path
, "list", NULL
))
384 return UBUS_STATUS_UNKNOWN_ERROR
;
387 plugin
= rpc_plugin_parse_exec(name
+ 1, fds
[0]);
392 rv
= ubus_add_object(ctx
, plugin
);
397 waitpid(pid
, NULL
, 0);
404 static LIST_HEAD(plugins
);
406 static const struct rpc_daemon_ops ops
= {
407 .session_access
= rpc_session_access
,
408 .session_create_cb
= rpc_session_create_cb
,
409 .session_destroy_cb
= rpc_session_destroy_cb
,
411 .exec_timeout
= &rpc_exec_timeout
,
415 rpc_plugin_register_library(struct ubus_context
*ctx
, const char *path
)
417 struct rpc_plugin
*p
;
420 dlh
= dlopen(path
, RTLD_LAZY
| RTLD_LOCAL
);
423 return UBUS_STATUS_UNKNOWN_ERROR
;
425 p
= dlsym(dlh
, "rpc_plugin");
428 return UBUS_STATUS_NOT_FOUND
;
430 list_add(&p
->list
, &plugins
);
432 return p
->init(&ops
, ctx
);
435 int rpc_plugin_api_init(struct ubus_context
*ctx
)
443 if ((d
= opendir(RPC_PLUGIN_DIRECTORY
)) != NULL
)
445 while ((e
= readdir(d
)) != NULL
)
447 snprintf(path
, sizeof(path
) - 1,
448 RPC_PLUGIN_DIRECTORY
"/%s", e
->d_name
);
450 if (stat(path
, &s
) || !S_ISREG(s
.st_mode
) || !(s
.st_mode
& S_IXUSR
))
453 rv
|= rpc_plugin_register_exec(ctx
, path
);
459 if ((d
= opendir(RPC_LIBRARY_DIRECTORY
)) != NULL
)
461 while ((e
= readdir(d
)) != NULL
)
463 snprintf(path
, sizeof(path
) - 1,
464 RPC_LIBRARY_DIRECTORY
"/%s", e
->d_name
);
466 if (stat(path
, &s
) || !S_ISREG(s
.st_mode
))
469 rv
|= rpc_plugin_register_library(ctx
, path
);