2 * Copyright (C) 2016, 2017 "IoT.bzh"
3 * Author José Bollo <jose.bollo@iot.bzh>
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
24 #include <json-c/json.h>
26 #include <afb/afb-binding-v1.h>
27 #include <afb/afb-binding-v2.h>
31 #include "afb-common.h"
37 /**********************************************
39 **********************************************/
40 static void vverbose_cb(void *closure, int level, const char *file, int line, const char *function, const char *fmt, va_list args)
43 struct afb_ditf *ditf = closure;
45 if (vasprintf(&p, fmt, args) < 0)
46 vverbose(level, file, line, function, fmt, args);
48 verbose(level, file, line, function, "[API %s] %s", ditf->api, p);
53 static void old_vverbose_cb(void *closure, int level, const char *file, int line, const char *fmt, va_list args)
55 vverbose_cb(closure, level, file, line, "?", fmt, args);
58 static struct afb_event event_make_cb(void *closure, const char *name)
62 struct afb_ditf *ditf = closure;
64 /* makes the event name */
65 plen = strlen(ditf->api);
67 event = alloca(nlen + plen + 2);
68 memcpy(event, ditf->api, plen);
70 memcpy(event + plen + 1, name, nlen + 1);
72 /* create the event */
73 return afb_evt_create_event(event);
76 static int event_broadcast_cb(void *closure, const char *name, struct json_object *object)
80 struct afb_ditf *ditf = closure;
82 /* makes the event name */
83 plen = strlen(ditf->api);
85 event = alloca(nlen + plen + 2);
86 memcpy(event, ditf->api, plen);
88 memcpy(event + plen + 1, name, nlen + 1);
90 /* broadcast the event */
91 return afb_evt_broadcast(event, object);
94 static int rootdir_open_locale_cb(void *closure, const char *filename, int flags, const char *locale)
96 return afb_common_rootdir_open_locale(filename, flags, locale);
99 static int queue_job_cb(void *closure, void (*callback)(int signum, void *arg), void *argument, void *group, int timeout)
101 return jobs_queue(group, timeout, callback, argument);
104 static struct afb_req unstore_req_cb(void *closure, struct afb_stored_req *sreq)
106 return afb_xreq_unstore(sreq);
109 /**********************************************
111 **********************************************/
112 static void hooked_vverbose_cb(void *closure, int level, const char *file, int line, const char *function, const char *fmt, va_list args)
114 struct afb_ditf *ditf = closure;
117 vverbose_cb(closure, level, file, line, function, fmt, args);
118 afb_hook_ditf_vverbose(ditf, level, file, line, function, fmt, ap);
122 static void hooked_old_vverbose_cb(void *closure, int level, const char *file, int line, const char *fmt, va_list args)
124 hooked_vverbose_cb(closure, level, file, line, "?", fmt, args);
127 static struct afb_event hooked_event_make_cb(void *closure, const char *name)
129 struct afb_ditf *ditf = closure;
130 struct afb_event r = event_make_cb(closure, name);
131 return afb_hook_ditf_event_make(ditf, name, r);
134 static int hooked_event_broadcast_cb(void *closure, const char *name, struct json_object *object)
137 struct afb_ditf *ditf = closure;
138 json_object_get(object);
139 afb_hook_ditf_event_broadcast_before(ditf, name, json_object_get(object));
140 r = event_broadcast_cb(closure, name, object);
141 afb_hook_ditf_event_broadcast_after(ditf, name, object, r);
142 json_object_put(object);
146 static struct sd_event *hooked_get_event_loop(void *closure)
148 struct afb_ditf *ditf = closure;
149 struct sd_event *r = afb_common_get_event_loop();
150 return afb_hook_ditf_get_event_loop(ditf, r);
153 static struct sd_bus *hooked_get_user_bus(void *closure)
155 struct afb_ditf *ditf = closure;
156 struct sd_bus *r = afb_common_get_user_bus();
157 return afb_hook_ditf_get_user_bus(ditf, r);
160 static struct sd_bus *hooked_get_system_bus(void *closure)
162 struct afb_ditf *ditf = closure;
163 struct sd_bus *r = afb_common_get_system_bus();
164 return afb_hook_ditf_get_system_bus(ditf, r);
167 static int hooked_rootdir_get_fd(void *closure)
169 struct afb_ditf *ditf = closure;
170 int r = afb_common_rootdir_get_fd();
171 return afb_hook_ditf_rootdir_get_fd(ditf, r);
174 static int hooked_rootdir_open_locale_cb(void *closure, const char *filename, int flags, const char *locale)
176 struct afb_ditf *ditf = closure;
177 int r = rootdir_open_locale_cb(closure, filename, flags, locale);
178 return afb_hook_ditf_rootdir_open_locale(ditf, filename, flags, locale, r);
181 static int hooked_queue_job_cb(void *closure, void (*callback)(int signum, void *arg), void *argument, void *group, int timeout)
183 struct afb_ditf *ditf = closure;
184 int r = queue_job_cb(closure, callback, argument, group, timeout);
185 return afb_hook_ditf_queue_job(ditf, callback, argument, group, timeout, r);
188 static struct afb_req hooked_unstore_req_cb(void *closure, struct afb_stored_req *sreq)
190 struct afb_ditf *ditf = closure;
191 afb_hook_ditf_unstore_req(ditf, sreq);
192 return unstore_req_cb(closure, sreq);
195 /**********************************************
197 **********************************************/
198 static const struct afb_daemon_itf daemon_itf = {
199 .vverbose_v1 = old_vverbose_cb,
200 .vverbose_v2 = vverbose_cb,
201 .event_make = event_make_cb,
202 .event_broadcast = event_broadcast_cb,
203 .get_event_loop = afb_common_get_event_loop,
204 .get_user_bus = afb_common_get_user_bus,
205 .get_system_bus = afb_common_get_system_bus,
206 .rootdir_get_fd = afb_common_rootdir_get_fd,
207 .rootdir_open_locale = rootdir_open_locale_cb,
208 .queue_job = queue_job_cb,
209 .unstore_req = unstore_req_cb
212 static const struct afb_daemon_itf hooked_daemon_itf = {
213 .vverbose_v1 = hooked_old_vverbose_cb,
214 .vverbose_v2 = hooked_vverbose_cb,
215 .event_make = hooked_event_make_cb,
216 .event_broadcast = hooked_event_broadcast_cb,
217 .get_event_loop = hooked_get_event_loop,
218 .get_user_bus = hooked_get_user_bus,
219 .get_system_bus = hooked_get_system_bus,
220 .rootdir_get_fd = hooked_rootdir_get_fd,
221 .rootdir_open_locale = hooked_rootdir_open_locale_cb,
222 .queue_job = hooked_queue_job_cb,
223 .unstore_req = hooked_unstore_req_cb
226 void afb_ditf_init_v2(struct afb_ditf *ditf, const char *api, struct afb_binding_data_v2 *data)
230 data->daemon.closure = ditf;
231 afb_ditf_rename(ditf, api);
234 void afb_ditf_init_v1(struct afb_ditf *ditf, const char *api, struct afb_binding_interface_v1 *itf)
238 itf->verbosity = verbosity;
239 itf->mode = AFB_MODE_LOCAL;
240 itf->daemon.closure = ditf;
241 afb_ditf_rename(ditf, api);
244 void afb_ditf_rename(struct afb_ditf *ditf, const char *api)
247 afb_ditf_update_hook(ditf);
250 void afb_ditf_update_hook(struct afb_ditf *ditf)
252 int hooked = !!afb_hook_flags_ditf(ditf->api);
253 switch (ditf->version) {
255 ditf->v1->daemon.itf = hooked ? &hooked_daemon_itf : &daemon_itf;
259 ditf->v2->daemon.itf = hooked ? &hooked_daemon_itf : &daemon_itf;