instance.c

   1 #include "procd.h"
   2 #include "service.h"
   3 #include "instance.h"
   4
   5 enum {
   6         INSTANCE_ATTR_COMMAND,
   7         INSTANCE_ATTR_ENV,
   8         INSTANCE_ATTR_DATA,
   9         __INSTANCE_ATTR_MAX
  10 };
  11
  12 static const struct blobmsg_policy instance_attr[__INSTANCE_ATTR_MAX] = {
  13         [INSTANCE_ATTR_COMMAND] = { "command", BLOBMSG_TYPE_ARRAY },
  14         [INSTANCE_ATTR_ENV] = { "env", BLOBMSG_TYPE_TABLE },
  15         [INSTANCE_ATTR_DATA] = { "data", BLOBMSG_TYPE_TABLE },
  16 };
  17
  18 void
  19 instance_start(struct service_instance *in)
  20 {
  21         in->restart = false;
  22         if (!in->valid)
  23                 return;
  24 }
  25
  26 static void
  27 instance_timeout(struct uloop_timeout *t)
  28 {
  29         struct service_instance *in;
  30
  31         in = container_of(t, struct service_instance, timeout);
  32         kill(in->proc.pid, SIGKILL);
  33         uloop_process_delete(&in->proc);
  34         in->proc.cb(&in->proc, -1);
  35 }
  36
  37 static void
  38 instance_exit(struct uloop_process *p, int ret)
  39 {
  40         struct service_instance *in;
  41
  42         in = container_of(p, struct service_instance, proc);
  43         uloop_timeout_cancel(&in->timeout);
  44         if (in->restart)
  45                 instance_start(in);
  46 }
  47
  48 void
  49 instance_stop(struct service_instance *in, bool restart)
  50 {
  51         if (!in->proc.pending)
  52                 return;
  53
  54         kill(in->proc.pid, SIGTERM);
  55 }
  56
  57 static bool
  58 instance_config_changed(struct service_instance *in, struct service_instance *in_new)
  59 {
  60         if (!in->valid)
  61                 return true;
  62
  63         if (!blob_attr_equal(in->command, in_new->command))
  64                 return true;
  65
  66         if (!blobmsg_list_equal(&in->env, &in_new->env))
  67                 return true;
  68
  69         if (!blobmsg_list_equal(&in->data, &in_new->data))
  70                 return true;
  71
  72         return false;
  73 }
  74
  75 static bool
  76 instance_config_parse(struct service_instance *in)
  77 {
  78         struct blob_attr *tb[__INSTANCE_ATTR_MAX];
  79         struct blob_attr *cur;
  80
  81         blobmsg_parse(instance_attr, __INSTANCE_ATTR_MAX, tb,
  82                 blobmsg_data(in->config), blobmsg_data_len(in->config));
  83
  84         if (!tb[INSTANCE_ATTR_COMMAND])
  85                 return false;
  86
  87         if ((cur = tb[INSTANCE_ATTR_ENV]))
  88                 blobmsg_list_fill(&in->env, blobmsg_data(cur), blobmsg_data_len(cur));
  89
  90         if ((cur = tb[INSTANCE_ATTR_DATA]))
  91                 blobmsg_list_fill(&in->data, blobmsg_data(cur), blobmsg_data_len(cur));
  92
  93         return true;
  94 }
  95
  96 static void
  97 instance_config_cleanup(struct service_instance *in)
  98 {
  99         blobmsg_list_free(&in->env);
 100         blobmsg_list_free(&in->data);
 101 }
 102
 103 static void
 104 instance_config_move(struct service_instance *in, struct service_instance *in_src)
 105 {
 106         instance_config_cleanup(in);
 107         blobmsg_list_move(&in->env, &in_src->env);
 108         blobmsg_list_move(&in->data, &in_src->data);
 109         in->command = in_src->command;
 110 }
 111
 112 bool
 113 instance_update(struct service_instance *in, struct service_instance *in_new)
 114 {
 115         bool changed = instance_config_changed(in, in_new);
 116
 117         in->config = in_new->config;
 118         if (!changed)
 119                 return false;
 120
 121         in->restart = true;
 122         instance_stop(in, true);
 123         instance_config_move(in, in_new);
 124         return true;
 125 }
 126
 127 void
 128 instance_free(struct service_instance *in)
 129 {
 130         uloop_process_delete(&in->proc);
 131         uloop_timeout_cancel(&in->timeout);
 132         instance_config_cleanup(in);
 133         free(in);
 134 }
 135
 136 void
 137 instance_init(struct service_instance *in, struct blob_attr *config)
 138 {
 139         in->config = config;
 140         in->timeout.cb = instance_timeout;
 141         in->proc.cb = instance_exit;
 142
 143         blobmsg_list_simple_init(&in->env);
 144         blobmsg_list_simple_init(&in->data);
 145         in->valid = instance_config_parse(in);
 146 }