1 #include <libubox/avl-cmp.h>
6 struct avl_tree services
;
7 static struct blob_buf b
;
10 service_instance_add(struct service
*s
, struct blob_attr
*attr
)
12 struct service_instance
*in
;
13 const char *name
= blobmsg_name(attr
);
15 if (blobmsg_type(attr
) != BLOBMSG_TYPE_TABLE
)
18 in
= calloc(1, sizeof(*in
));
22 instance_init(in
, s
, attr
);
23 vlist_add(&s
->instances
, &in
->node
, (void *) name
);
27 service_instance_update(struct vlist_tree
*tree
, struct vlist_node
*node_new
,
28 struct vlist_node
*node_old
)
30 struct service_instance
*in_o
= NULL
, *in_n
= NULL
;
33 in_o
= container_of(node_old
, struct service_instance
, node
);
36 in_n
= container_of(node_new
, struct service_instance
, node
);
39 DPRINTF("Update instance %s::%s\n", in_o
->srv
->name
, in_o
->name
);
40 instance_update(in_o
, in_n
);
43 DPRINTF("Free instance %s::%s\n", in_o
->srv
->name
, in_o
->name
);
44 instance_stop(in_o
, false);
47 DPRINTF("Create instance %s::%s\n", in_n
->srv
->name
, in_n
->name
);
52 static struct service
*
53 service_alloc(const char *name
)
58 s
= calloc(1, sizeof(*s
) + strlen(name
) + 1);
60 new_name
= (char *) (s
+ 1);
61 strcpy(new_name
, name
);
63 vlist_init(&s
->instances
, avl_strcmp
, service_instance_update
);
64 s
->instances
.keep_old
= true;
74 SERVICE_SET_INSTANCES
,
78 static const struct blobmsg_policy service_set_attrs
[__SERVICE_SET_MAX
] = {
79 [SERVICE_SET_NAME
] = { "name", BLOBMSG_TYPE_STRING
},
80 [SERVICE_SET_SCRIPT
] = { "script", BLOBMSG_TYPE_STRING
},
81 [SERVICE_SET_INSTANCES
] = { "instances", BLOBMSG_TYPE_TABLE
},
86 service_update(struct service
*s
, struct blob_attr
*config
, struct blob_attr
**tb
, bool add
)
88 struct blob_attr
*cur
;
91 if (tb
[SERVICE_SET_INSTANCES
]) {
93 vlist_update(&s
->instances
);
94 blobmsg_for_each_attr(cur
, tb
[SERVICE_SET_INSTANCES
], rem
) {
95 service_instance_add(s
, cur
);
98 vlist_flush(&s
->instances
);
105 service_delete(struct service
*s
)
107 vlist_flush_all(&s
->instances
);
108 avl_delete(&services
, &s
->avl
);
118 static const struct blobmsg_policy service_attrs
[__SERVICE_ATTR_MAX
] = {
119 [SERVICE_ATTR_NAME
] = { "name", BLOBMSG_TYPE_STRING
},
124 service_handle_set(struct ubus_context
*ctx
, struct ubus_object
*obj
,
125 struct ubus_request_data
*req
, const char *method
,
126 struct blob_attr
*msg
)
128 struct blob_attr
*tb
[__SERVICE_SET_MAX
], *cur
;
129 struct service
*s
= NULL
;
131 int ret
= UBUS_STATUS_INVALID_ARGUMENT
;
132 bool add
= !strcmp(method
, "add");
134 blobmsg_parse(service_set_attrs
, __SERVICE_SET_MAX
, tb
, blob_data(msg
), blob_len(msg
));
135 cur
= tb
[SERVICE_ATTR_NAME
];
139 name
= blobmsg_data(cur
);
141 s
= avl_find_element(&services
, name
, s
, avl
);
143 DPRINTF("Update service %s\n", name
);
144 return service_update(s
, msg
, tb
, add
);
147 DPRINTF("Create service %s\n", name
);
148 s
= service_alloc(name
);
150 return UBUS_STATUS_UNKNOWN_ERROR
;
152 ret
= service_update(s
, msg
, tb
, add
);
156 avl_insert(&services
, &s
->avl
);
166 service_dump(struct service
*s
)
168 struct service_instance
*in
;
171 c
= blobmsg_open_table(&b
, s
->name
);
172 i
= blobmsg_open_table(&b
, "instances");
173 vlist_for_each_element(&s
->instances
, in
, node
)
174 instance_dump(&b
, in
);
175 blobmsg_close_table(&b
, i
);
176 blobmsg_close_table(&b
, c
);
180 service_handle_list(struct ubus_context
*ctx
, struct ubus_object
*obj
,
181 struct ubus_request_data
*req
, const char *method
,
182 struct blob_attr
*msg
)
186 blob_buf_init(&b
, 0);
187 avl_for_each_element(&services
, s
, avl
)
190 ubus_send_reply(ctx
, req
, b
.head
);
196 service_handle_delete(struct ubus_context
*ctx
, struct ubus_object
*obj
,
197 struct ubus_request_data
*req
, const char *method
,
198 struct blob_attr
*msg
)
200 struct blob_attr
*tb
[__SERVICE_ATTR_MAX
], *cur
;
201 struct service
*s
, *tmp
;
203 blobmsg_parse(service_attrs
, __SERVICE_ATTR_MAX
, tb
, blob_data(msg
), blob_len(msg
));
205 cur
= tb
[SERVICE_ATTR_NAME
];
207 avl_for_each_element_safe(&services
, s
, avl
, tmp
)
212 s
= avl_find_element(&services
, blobmsg_data(cur
), s
, avl
);
214 return UBUS_STATUS_NOT_FOUND
;
221 service_handle_update(struct ubus_context
*ctx
, struct ubus_object
*obj
,
222 struct ubus_request_data
*req
, const char *method
,
223 struct blob_attr
*msg
)
225 struct blob_attr
*tb
[__SERVICE_ATTR_MAX
], *cur
;
228 blobmsg_parse(service_attrs
, __SERVICE_ATTR_MAX
, tb
, blob_data(msg
), blob_len(msg
));
230 cur
= tb
[SERVICE_ATTR_NAME
];
232 return UBUS_STATUS_INVALID_ARGUMENT
;
234 s
= avl_find_element(&services
, blobmsg_data(cur
), s
, avl
);
236 return UBUS_STATUS_NOT_FOUND
;
238 if (!strcmp(method
, "update_start"))
239 vlist_update(&s
->instances
);
241 vlist_flush(&s
->instances
);
246 static struct ubus_method main_object_methods
[] = {
247 UBUS_METHOD("set", service_handle_set
, service_set_attrs
),
248 UBUS_METHOD("add", service_handle_set
, service_set_attrs
),
249 UBUS_METHOD("list", service_handle_list
, service_attrs
),
250 UBUS_METHOD("delete", service_handle_delete
, service_attrs
),
251 UBUS_METHOD("update_start", service_handle_update
, service_attrs
),
252 UBUS_METHOD("update_complete", service_handle_update
, service_attrs
),
255 static struct ubus_object_type main_object_type
=
256 UBUS_OBJECT_TYPE("service", main_object_methods
);
258 static struct ubus_object main_object
= {
260 .type
= &main_object_type
,
261 .methods
= main_object_methods
,
262 .n_methods
= ARRAY_SIZE(main_object_methods
),
265 void procd_init_service(struct ubus_context
*ctx
)
267 avl_init(&services
, avl_strcmp
, false, NULL
);
268 ubus_add_object(ctx
, &main_object
);