X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=src%2Fafs-supervisor.c;h=71a00369a4c7cd6a65c73cffb4d83bba2f0d87f7;hb=ac7a95223a6314cca6250495ea59c3cf7e46e89e;hp=09df3639e606caa7ea1aa1240ce6c70a374b4525;hpb=b8c9d5de384efcfa53ebdb3f0053d7b3723777e1;p=src%2Fapp-framework-binder.git diff --git a/src/afs-supervisor.c b/src/afs-supervisor.c index 09df3639..71a00369 100644 --- a/src/afs-supervisor.c +++ b/src/afs-supervisor.c @@ -35,8 +35,7 @@ #include #include -#include "afs-supervision.h" -#include "afb-common.h" +#include "afb-systemd.h" #include "afb-session.h" #include "afb-cred.h" #include "afb-stub-ws.h" @@ -45,9 +44,15 @@ #include "afb-api-so-v2.h" #include "afb-api-ws.h" #include "afb-apiset.h" +#include "afb-fdev.h" #include "jobs.h" #include "verbose.h" #include "wrap-json.h" +#include "process-name.h" + +#include "afs-supervision.h" +#include "afs-supervisor.h" +#include "afs-discover.h" /* supervised items */ struct supervised @@ -64,9 +69,7 @@ struct supervised /* api and apiset name */ static const char supervision_apiname[] = AFS_SURPERVISION_APINAME; - -/* the main apiset */ -struct afb_apiset *main_apiset; +static const char supervisor_apiname[] = AFS_SURPERVISOR_APINAME; /* the empty apiset */ static struct afb_apiset *empty_apiset; @@ -82,7 +85,6 @@ static struct supervised *superviseds; /*************************************************************************************/ -static int afb_init_supervision_api(); /*************************************************************************************/ @@ -190,13 +192,20 @@ static void on_supervised_hangup(struct afb_stub_ws *stub) static int make_supervised(int fd, struct afb_cred *cred) { struct supervised *s; + struct fdev *fdev; s = malloc(sizeof *s); if (!s) return -1; + fdev = afb_fdev_create(fd); + if (!fdev) { + free(s); + return -1; + } + s->cred = cred; - s->stub = afb_stub_ws_create_client(fd, supervision_apiname, empty_apiset); + s->stub = afb_stub_ws_create_client(fdev, supervision_apiname, empty_apiset); if (!s->stub) { free(s); return -1; @@ -209,6 +218,22 @@ static int make_supervised(int fd, struct afb_cred *cred) return 0; } +/** + * Search the supervised of 'pid', return it or NULL. + */ +static struct supervised *supervised_of_pid(pid_t pid) +{ + struct supervised *s; + + pthread_mutex_lock(&mutex); + s = superviseds; + while (s && pid != s->cred->pid) + s = s->next; + pthread_mutex_unlock(&mutex); + + return s; +} + /* * handles incoming connection on 'sock' */ @@ -251,101 +276,27 @@ static int listening(sd_event_source *src, int fd, uint32_t revents, void *closu return 0; } - -/** - * initalize the supervision +/* */ -static int init(const char *spec) +static void discovered_cb(void *closure, pid_t pid) { - int rc, fd; - - /* check argument */ - if (!spec) { - ERROR("invalid socket spec"); - return -1; - } - - rc = afb_session_init(100, 600, ""); - /* TODO check that value */ - - /* create the apisets */ - main_apiset = afb_apiset_create(supervision_apiname, 0); - if (!main_apiset) { - ERROR("Can't create supervision's apiset"); - return -1; - } - empty_apiset = afb_apiset_create(supervision_apiname, 0); - if (!empty_apiset) { - ERROR("Can't create supervised apiset"); - return -1; - } - - - /* init the main apiset */ - rc = afb_init_supervision_api(); - if (rc < 0) { - ERROR("Can't create supervision's apiset: %m"); - return -1; - } - - /* create the supervision socket */ - fd = create_supervision_socket(supervision_socket_path); - if (fd < 0) - return fd; - - /* listen the socket */ - rc = listen(fd, 5); - if (rc < 0) { - ERROR("refused to listen on socket"); - return rc; - } - - /* integrate the socket to the loop */ - rc = sd_event_add_io(afb_common_get_event_loop(), - NULL, fd, EPOLLIN, - listening, NULL); - if (rc < 0) { - ERROR("handling socket event isn't possible"); - return rc; - } + struct supervised *s; - /* adds the server socket */ - rc = afb_api_ws_add_server(spec, main_apiset); - if (rc < 0) { - ERROR("can't start the server socket"); - return -1; + s = supervised_of_pid(pid); + if (!s) { + (*(int*)closure)++; + kill(pid, SIGHUP); } - return 0; } -/* start should not be null but */ -static void start(int signum, void *arg) +int afs_supervisor_discover() { - char *xpath = arg; - int rc; - - if (signum) - exit(1); - - rc = init(xpath); - if (rc) - exit(1); - - sd_notify(1, "READY=1"); + int n = 0; + afs_discover("afb-daemon", discovered_cb, &n); + return n; } -/** - * initalize the supervision - */ -int main(int ac, char **av) -{ - /* enter job processing */ - jobs_start(3, 0, 10, start, av[1]); - WARNING("hoops returned from jobs_enter! [report bug]"); - return 1; -} - -/*********************************************************************************************************/ +/*************************************************************************************/ static struct afb_binding_data_v2 datav2; @@ -374,6 +325,12 @@ static void f_list(struct afb_req req) afb_req_success(req, resu, NULL); } +static void f_discover(struct afb_req req) +{ + afs_supervisor_discover(); + afb_req_success(req, NULL, NULL); +} + static void propagate(struct afb_req req, const char *verb) { struct afb_xreq *xreq; @@ -394,9 +351,7 @@ static void propagate(struct afb_req req, const char *verb) afb_xreq_fail(xreq, "bad-pid", NULL); return; } - s = superviseds; - while (s && p != (int)s->cred->pid) - s = s->next; + s = supervised_of_pid((pid_t)p); if (!s) { afb_req_fail(req, "unknown-pid", NULL); return; @@ -413,6 +368,11 @@ static void f_do(struct afb_req req) propagate(req, NULL); } +static void f_config(struct afb_req req) +{ + propagate(req, NULL); +} + static void f_trace(struct afb_req req) { propagate(req, NULL); @@ -443,7 +403,49 @@ static void f_debug_break(struct afb_req req) propagate(req, "break"); } -static const struct afb_auth _afb_auths_v2_supervision[] = { +/*************************************************************************************/ + +/** + * initalize the supervisor + */ +static int init_supervisor() +{ + int rc, fd; + + /* create an empty set for superviseds */ + empty_apiset = afb_apiset_create(supervision_apiname, 0); + if (!empty_apiset) { + ERROR("Can't create supervision apiset"); + return -1; + } + + /* create the supervision socket */ + fd = create_supervision_socket(supervision_socket_path); + if (fd < 0) + return fd; + + /* listen the socket */ + rc = listen(fd, 5); + if (rc < 0) { + ERROR("refused to listen on socket"); + return rc; + } + + /* integrate the socket to the loop */ + rc = sd_event_add_io(afb_systemd_get_event_loop(), + NULL, fd, EPOLLIN, + listening, NULL); + if (rc < 0) { + ERROR("handling socket event isn't possible"); + return rc; + } + + return 0; +} + +/*************************************************************************************/ + +static const struct afb_auth _afb_auths_v2_supervisor[] = { /* 0 */ { .type = afb_auth_Permission, @@ -451,79 +453,93 @@ static const struct afb_auth _afb_auths_v2_supervision[] = { } }; -static const struct afb_verb_v2 _afb_verbs_v2_supervision[] = { +static const struct afb_verb_v2 _afb_verbs_v2_supervisor[] = { { .verb = "list", .callback = f_list, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], + .info = NULL, + .session = AFB_SESSION_NONE_V2 + }, + { + .verb = "config", + .callback = f_config, + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = "do", .callback = f_do, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = "trace", .callback = f_trace, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = "sessions", .callback = f_sessions, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = "session-close", .callback = f_session_close, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = "exit", .callback = f_exit, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = "debug-wait", .callback = f_debug_wait, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = "debug-break", .callback = f_debug_break, - .auth = &_afb_auths_v2_supervision[0], + .auth = &_afb_auths_v2_supervisor[0], + .info = NULL, + .session = AFB_SESSION_NONE_V2 + }, + { + .verb = "discover", + .callback = f_discover, + .auth = &_afb_auths_v2_supervisor[0], .info = NULL, .session = AFB_SESSION_NONE_V2 }, { .verb = NULL } }; -static const struct afb_binding_v2 _afb_binding_v2_supervision = { - .api = supervision_apiname, +static const struct afb_binding_v2 _afb_binding_v2_supervisor = { + .api = supervisor_apiname, .specification = NULL, .info = NULL, - .verbs = _afb_verbs_v2_supervision, + .verbs = _afb_verbs_v2_supervisor, .preinit = NULL, - .init = NULL, + .init = init_supervisor, .onevent = NULL, .noconcurrency = 0 }; -static int afb_init_supervision_api() +int afs_supervisor_add(struct afb_apiset *apiset) { - return afb_api_so_v2_add_binding(&_afb_binding_v2_supervision, NULL, main_apiset, &datav2); + return afb_api_so_v2_add_binding(&_afb_binding_v2_supervisor, NULL, apiset, &datav2); }