Context Navigation

rest.c @ b116081

Last change on this file since b116081 was 7f5ab4a, checked in by OpenGnSys Support Team <soporte-og@…>, 4 years ago

#942 Add GET /session REST API function

This function returns the installed and available OSs in a client to be
booted from.

Request:
GET /session
{

"client": 192.168.56.11?

}

Response
200 OK
{

"sessions": [

{

"disk": 1,
"name": "Ubuntu 18.04.4 LTS",
"partition": 1

}

]

}

Property mode set to 100644

File size: 102.6 KB

Line
1	/*
2	* Copyright (C) 2020 Soleta Networks <info@soleta.eu>
3	*
4	* This program is free software: you can redistribute it and/or modify it under
5	* the terms of the GNU Affero General Public License as published by the
6	* Free Software Foundation, version 3.
7	*/
8
9	#include "ogAdmServer.h"
10	#include "dbi.h"
11	#include "utils.h"
12	#include "list.h"
13	#include "rest.h"
14	#include "cfg.h"
15	#include "schedule.h"
16	#include <ev.h>
17	#include <syslog.h>
18	#include <sys/ioctl.h>
19	#include <ifaddrs.h>
20	#include <sys/types.h>
21	#include <sys/stat.h>
22	#include <fcntl.h>
23	#include <jansson.h>
24	#include <dirent.h>
25	#include <time.h>
26	#include <stdlib.h>
27	#include <unistd.h>
28	#include <sys/wait.h>
29	#include <sys/statvfs.h>
30
31	struct ev_loop *og_loop;
32
33	#define OG_REST_PARAM_ADDR (1UL << 0)
34	#define OG_REST_PARAM_MAC (1UL << 1)
35	#define OG_REST_PARAM_WOL_TYPE (1UL << 2)
36	#define OG_REST_PARAM_RUN_CMD (1UL << 3)
37	#define OG_REST_PARAM_DISK (1UL << 4)
38	#define OG_REST_PARAM_PARTITION (1UL << 5)
39	#define OG_REST_PARAM_REPO (1UL << 6)
40	#define OG_REST_PARAM_NAME (1UL << 7)
41	#define OG_REST_PARAM_ID (1UL << 8)
42	#define OG_REST_PARAM_CODE (1UL << 9)
43	#define OG_REST_PARAM_TYPE (1UL << 10)
44	#define OG_REST_PARAM_PROFILE (1UL << 11)
45	#define OG_REST_PARAM_CACHE (1UL << 12)
46	#define OG_REST_PARAM_CACHE_SIZE (1UL << 13)
47	#define OG_REST_PARAM_PART_0 (1UL << 14)
48	#define OG_REST_PARAM_PART_1 (1UL << 15)
49	#define OG_REST_PARAM_PART_2 (1UL << 16)
50	#define OG_REST_PARAM_PART_3 (1UL << 17)
51	#define OG_REST_PARAM_SYNC_SYNC (1UL << 18)
52	#define OG_REST_PARAM_SYNC_DIFF (1UL << 19)
53	#define OG_REST_PARAM_SYNC_REMOVE (1UL << 20)
54	#define OG_REST_PARAM_SYNC_COMPRESS (1UL << 21)
55	#define OG_REST_PARAM_SYNC_CLEANUP (1UL << 22)
56	#define OG_REST_PARAM_SYNC_CACHE (1UL << 23)
57	#define OG_REST_PARAM_SYNC_CLEANUP_CACHE (1UL << 24)
58	#define OG_REST_PARAM_SYNC_REMOVE_DST (1UL << 25)
59	#define OG_REST_PARAM_SYNC_DIFF_ID (1UL << 26)
60	#define OG_REST_PARAM_SYNC_DIFF_NAME (1UL << 27)
61	#define OG_REST_PARAM_SYNC_PATH (1UL << 28)
62	#define OG_REST_PARAM_SYNC_METHOD (1UL << 29)
63	#define OG_REST_PARAM_ECHO (1UL << 30)
64	#define OG_REST_PARAM_TASK (1UL << 31)
65	#define OG_REST_PARAM_TIME_YEARS (1UL << 32)
66	#define OG_REST_PARAM_TIME_MONTHS (1UL << 33)
67	#define OG_REST_PARAM_TIME_WEEKS (1UL << 34)
68	#define OG_REST_PARAM_TIME_WEEK_DAYS (1UL << 35)
69	#define OG_REST_PARAM_TIME_DAYS (1UL << 36)
70	#define OG_REST_PARAM_TIME_HOURS (1UL << 37)
71	#define OG_REST_PARAM_TIME_AM_PM (1UL << 38)
72	#define OG_REST_PARAM_TIME_MINUTES (1UL << 39)
73	#define OG_REST_PARAM_NETMASK (1UL << 40)
74	#define OG_REST_PARAM_SCOPE (1UL << 41)
75	#define OG_REST_PARAM_MODE (1UL << 42)
76
77	static LIST_HEAD(client_list);
78
79	void og_client_add(struct og_client *cli)
80	{
81	list_add(&cli->list, &client_list);
82	}
83
84	static struct og_client og_client_find(const char ip)
85	{
86	struct og_client *client;
87	struct in_addr addr;
88	int res;
89
90	res = inet_aton(ip, &addr);
91	if (!res) {
92	syslog(LOG_ERR, "Invalid IP string: %s\n", ip);
93	return NULL;
94	}
95
96	list_for_each_entry(client, &client_list, list) {
97	if (client->addr.sin_addr.s_addr == addr.s_addr && client->agent) {
98	return client;
99	}
100	}
101
102	return NULL;
103	}
104
105	static const char og_client_status(const struct og_client cli)
106	{
107	switch (cli->last_cmd) {
108	case OG_CMD_UNSPEC:
109	case OG_CMD_PROBE:
110	break;
111	default:
112	return "BSY";
113	}
114
115	switch (cli->status) {
116	case OG_CLIENT_STATUS_BUSY:
117	return "BSY";
118	case OG_CLIENT_STATUS_OGLIVE:
119	return "OPG";
120	case OG_CLIENT_STATUS_VIRTUAL:
121	return "VDI";
122	default:
123	return "OFF";
124	}
125	}
126
127	static bool og_msg_params_validate(const struct og_msg_params *params,
128	const uint64_t flags)
129	{
130	return (params->flags & flags) == flags;
131	}
132
133	static bool og_flags_validate(const uint64_t flags,
134	const uint64_t required_flags)
135	{
136	return (flags & required_flags) == required_flags;
137	}
138
139	static int og_json_parse_clients(json_t element, struct og_msg_params params)
140	{
141	unsigned int i;
142	json_t *k;
143
144	if (json_typeof(element) != JSON_ARRAY)
145	return -1;
146
147	for (i = 0; i < json_array_size(element); i++) {
148	k = json_array_get(element, i);
149	if (json_typeof(k) != JSON_STRING)
150	return -1;
151
152	params->ips_array[params->ips_array_len++] =
153	json_string_value(k);
154
155	params->flags \|= OG_REST_PARAM_ADDR;
156	}
157
158	return 0;
159	}
160
161	static int og_json_parse_partition_setup(json_t *element,
162	struct og_msg_params *params)
163	{
164	unsigned int i;
165	json_t *k;
166
167	if (json_typeof(element) != JSON_ARRAY)
168	return -1;
169
170	for (i = 0; i < json_array_size(element) && i < OG_PARTITION_MAX; ++i) {
171	k = json_array_get(element, i);
172
173	if (json_typeof(k) != JSON_OBJECT)
174	return -1;
175
176	if (og_json_parse_partition(k, &params->partition_setup[i],
177	OG_PARAM_PART_NUMBER \|
178	OG_PARAM_PART_CODE \|
179	OG_PARAM_PART_FILESYSTEM \|
180	OG_PARAM_PART_SIZE \|
181	OG_PARAM_PART_FORMAT) < 0)
182	return -1;
183
184	params->flags \|= (OG_REST_PARAM_PART_0 << i);
185	}
186	return 0;
187	}
188
189	static int og_json_parse_time_params(json_t *element,
190	struct og_msg_params *params)
191	{
192	const char *key;
193	json_t *value;
194	int err = 0;
195
196	json_object_foreach(element, key, value) {
197	if (!strcmp(key, "years")) {
198	err = og_json_parse_uint(value, &params->time.years);
199	params->flags \|= OG_REST_PARAM_TIME_YEARS;
200	} else if (!strcmp(key, "months")) {
201	err = og_json_parse_uint(value, &params->time.months);
202	params->flags \|= OG_REST_PARAM_TIME_MONTHS;
203	} else if (!strcmp(key, "weeks")) {
204	err = og_json_parse_uint(value, &params->time.weeks);
205	params->flags \|= OG_REST_PARAM_TIME_WEEKS;
206	} else if (!strcmp(key, "week_days")) {
207	err = og_json_parse_uint(value, &params->time.week_days);
208	params->flags \|= OG_REST_PARAM_TIME_WEEK_DAYS;
209	} else if (!strcmp(key, "days")) {
210	err = og_json_parse_uint(value, &params->time.days);
211	params->flags \|= OG_REST_PARAM_TIME_DAYS;
212	} else if (!strcmp(key, "hours")) {
213	err = og_json_parse_uint(value, &params->time.hours);
214	params->flags \|= OG_REST_PARAM_TIME_HOURS;
215	} else if (!strcmp(key, "am_pm")) {
216	err = og_json_parse_uint(value, &params->time.am_pm);
217	params->flags \|= OG_REST_PARAM_TIME_AM_PM;
218	} else if (!strcmp(key, "minutes")) {
219	err = og_json_parse_uint(value, &params->time.minutes);
220	params->flags \|= OG_REST_PARAM_TIME_MINUTES;
221	}
222	if (err != 0)
223	return err;
224	}
225
226	return err;
227	}
228
229	static const char *og_cmd_to_uri[OG_CMD_MAX] = {
230	[OG_CMD_WOL] = "wol",
231	[OG_CMD_PROBE] = "probe",
232	[OG_CMD_SHELL_RUN] = "shell/run",
233	[OG_CMD_SESSION] = "session",
234	[OG_CMD_POWEROFF] = "poweroff",
235	[OG_CMD_REFRESH] = "refresh",
236	[OG_CMD_REBOOT] = "reboot",
237	[OG_CMD_STOP] = "stop",
238	[OG_CMD_HARDWARE] = "hardware",
239	[OG_CMD_SOFTWARE] = "software",
240	[OG_CMD_IMAGE_CREATE] = "image/create",
241	[OG_CMD_IMAGE_RESTORE] = "image/restore",
242	[OG_CMD_SETUP] = "setup",
243	[OG_CMD_RUN_SCHEDULE] = "run/schedule",
244	[OG_CMD_IMAGES] = "images",
245	};
246
247	static bool og_client_is_busy(const struct og_client *cli,
248	enum og_cmd_type type)
249	{
250	switch (type) {
251	case OG_CMD_REBOOT:
252	case OG_CMD_POWEROFF:
253	case OG_CMD_STOP:
254	break;
255	default:
256	if (cli->last_cmd != OG_CMD_UNSPEC)
257	return true;
258	break;
259	}
260
261	return false;
262	}
263
264	int og_send_request(enum og_rest_method method, enum og_cmd_type type,
265	const struct og_msg_params *params,
266	const json_t *data)
267	{
268	const char *content_type = "Content-Type: application/json";
269	char content [OG_MSG_REQUEST_MAXLEN - 700] = {};
270	char buf[OG_MSG_REQUEST_MAXLEN] = {};
271	unsigned int content_length;
272	char method_str[5] = {};
273	struct og_client *cli;
274	const char *uri;
275	unsigned int i;
276	int client_sd;
277
278	if (method == OG_METHOD_GET)
279	snprintf(method_str, 5, "GET");
280	else if (method == OG_METHOD_POST)
281	snprintf(method_str, 5, "POST");
282	else
283	return -1;
284
285	if (!data)
286	content_length = 0;
287	else
288	content_length = json_dumpb(data, content,
289	OG_MSG_REQUEST_MAXLEN - 700,
290	JSON_COMPACT);
291
292	uri = og_cmd_to_uri[type];
293	snprintf(buf, OG_MSG_REQUEST_MAXLEN,
294	"%s /%s HTTP/1.1\r\nContent-Length: %d\r\n%s\r\n\r\n%s",
295	method_str, uri, content_length, content_type, content);
296
297	for (i = 0; i < params->ips_array_len; i++) {
298	cli = og_client_find(params->ips_array[i]);
299	if (!cli)
300	continue;
301
302	if (og_client_is_busy(cli, type))
303	continue;
304
305	client_sd = cli->io.fd;
306	if (client_sd < 0) {
307	syslog(LOG_INFO, "Client %s not conected\n",
308	params->ips_array[i]);
309	continue;
310	}
311
312	if (send(client_sd, buf, strlen(buf), 0) < 0)
313	continue;
314
315	cli->last_cmd = type;
316	}
317
318	return 0;
319	}
320
321	static int og_cmd_post_clients(json_t element, struct og_msg_params params)
322	{
323	const char *key;
324	json_t *value;
325	int err = 0;
326
327	if (json_typeof(element) != JSON_OBJECT)
328	return -1;
329
330	json_object_foreach(element, key, value) {
331	if (!strcmp(key, "clients"))
332	err = og_json_parse_clients(value, params);
333
334	if (err < 0)
335	break;
336	}
337
338	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
339	return -1;
340
341	return og_send_request(OG_METHOD_POST, OG_CMD_PROBE, params, NULL);
342	}
343
344	struct og_buffer {
345	char *data;
346	int len;
347	};
348
349	static int og_json_dump_clients(const char buffer, size_t size, void data)
350	{
351	struct og_buffer og_buffer = (struct og_buffer )data;
352
353	memcpy(og_buffer->data + og_buffer->len, buffer, size);
354	og_buffer->len += size;
355
356	return 0;
357	}
358
359	static int og_cmd_get_clients(json_t element, struct og_msg_params params,
360	char *buffer_reply)
361	{
362	json_t root, array, addr, state, *object;
363	struct og_client *client;
364	struct og_buffer og_buffer = {
365	.data = buffer_reply,
366	};
367
368	array = json_array();
369	if (!array)
370	return -1;
371
372	list_for_each_entry(client, &client_list, list) {
373	if (!client->agent)
374	continue;
375
376	object = json_object();
377	if (!object) {
378	json_decref(array);
379	return -1;
380	}
381	addr = json_string(inet_ntoa(client->addr.sin_addr));
382	if (!addr) {
383	json_decref(object);
384	json_decref(array);
385	return -1;
386	}
387	json_object_set_new(object, "addr", addr);
388	state = json_string(og_client_status(client));
389	if (!state) {
390	json_decref(object);
391	json_decref(array);
392	return -1;
393	}
394	json_object_set_new(object, "state", state);
395	json_array_append_new(array, object);
396	}
397	root = json_pack("{s:o}", "clients", array);
398	if (!root) {
399	json_decref(array);
400	return -1;
401	}
402
403	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
404	json_decref(root);
405
406	return 0;
407	}
408
409	static int og_json_parse_target(json_t element, struct og_msg_params params)
410	{
411	const char *key;
412	json_t *value;
413
414	if (json_typeof(element) != JSON_OBJECT) {
415	return -1;
416	}
417
418	json_object_foreach(element, key, value) {
419	if (!strcmp(key, "addr")) {
420	if (json_typeof(value) != JSON_STRING)
421	return -1;
422
423	params->ips_array[params->ips_array_len] =
424	json_string_value(value);
425
426	params->flags \|= OG_REST_PARAM_ADDR;
427	} else if (!strcmp(key, "mac")) {
428	if (json_typeof(value) != JSON_STRING)
429	return -1;
430
431	params->mac_array[params->ips_array_len] =
432	json_string_value(value);
433
434	params->flags \|= OG_REST_PARAM_MAC;
435	} else if (!strcmp(key, "netmask")) {
436	if (json_typeof(value) != JSON_STRING)
437	return -1;
438
439	params->netmask_array[params->ips_array_len] =
440	json_string_value(value);
441
442	params->flags \|= OG_REST_PARAM_NETMASK;
443	}
444	}
445
446	return 0;
447	}
448
449	static int og_json_parse_targets(json_t element, struct og_msg_params params)
450	{
451	unsigned int i;
452	json_t *k;
453	int err;
454
455	if (json_typeof(element) != JSON_ARRAY)
456	return -1;
457
458	for (i = 0; i < json_array_size(element); i++) {
459	k = json_array_get(element, i);
460
461	if (json_typeof(k) != JSON_OBJECT)
462	return -1;
463
464	err = og_json_parse_target(k, params);
465	if (err < 0)
466	return err;
467
468	params->ips_array_len++;
469	}
470	return 0;
471	}
472
473	static int og_json_parse_type(json_t element, struct og_msg_params params)
474	{
475	const char *type;
476
477	if (json_typeof(element) != JSON_STRING)
478	return -1;
479
480	params->wol_type = json_string_value(element);
481
482	type = json_string_value(element);
483	if (!strcmp(type, "unicast"))
484	params->wol_type = "2";
485	else if (!strcmp(type, "broadcast"))
486	params->wol_type = "1";
487
488	params->flags \|= OG_REST_PARAM_WOL_TYPE;
489
490	return 0;
491	}
492
493	static int og_cmd_wol(json_t element, struct og_msg_params params)
494	{
495	const char *key;
496	json_t *value;
497	int err = 0;
498
499	if (json_typeof(element) != JSON_OBJECT)
500	return -1;
501
502	json_object_foreach(element, key, value) {
503	if (!strcmp(key, "clients")) {
504	err = og_json_parse_targets(value, params);
505	} else if (!strcmp(key, "type")) {
506	err = og_json_parse_type(value, params);
507	}
508
509	if (err < 0)
510	break;
511	}
512
513	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR \|
514	OG_REST_PARAM_MAC \|
515	OG_REST_PARAM_NETMASK \|
516	OG_REST_PARAM_WOL_TYPE))
517	return -1;
518
519	if (!Levanta((char )params->ips_array, (char )params->mac_array,
520	(char **)params->netmask_array, params->ips_array_len,
521	(char *)params->wol_type))
522	return -1;
523
524	return 0;
525	}
526
527	static int og_json_parse_run(json_t element, struct og_msg_params params)
528	{
529	if (json_typeof(element) != JSON_STRING)
530	return -1;
531
532	snprintf(params->run_cmd, sizeof(params->run_cmd), "%s",
533	json_string_value(element));
534
535	params->flags \|= OG_REST_PARAM_RUN_CMD;
536
537	return 0;
538	}
539
540	static int og_cmd_run_post(json_t element, struct og_msg_params params)
541	{
542	json_t value, clients;
543	const char *key;
544	unsigned int i;
545	int err = 0;
546
547	if (json_typeof(element) != JSON_OBJECT)
548	return -1;
549
550	json_object_foreach(element, key, value) {
551	if (!strcmp(key, "clients"))
552	err = og_json_parse_clients(value, params);
553	else if (!strcmp(key, "run"))
554	err = og_json_parse_run(value, params);
555	else if (!strcmp(key, "echo")) {
556	err = og_json_parse_bool(value, &params->echo);
557	params->flags \|= OG_REST_PARAM_ECHO;
558	}
559
560	if (err < 0)
561	break;
562	}
563
564	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR \|
565	OG_REST_PARAM_RUN_CMD \|
566	OG_REST_PARAM_ECHO))
567	return -1;
568
569	clients = json_copy(element);
570	json_object_del(clients, "clients");
571
572	err = og_send_request(OG_METHOD_POST, OG_CMD_SHELL_RUN, params, clients);
573	if (err < 0)
574	return err;
575
576	for (i = 0; i < params->ips_array_len; i++) {
577	char filename[4096];
578	FILE *f;
579
580	sprintf(filename, "/tmp/_Seconsola_%s", params->ips_array[i]);
581	f = fopen(filename, "wt");
582	fclose(f);
583	}
584
585	return 0;
586	}
587
588	static int og_cmd_run_get(json_t element, struct og_msg_params params,
589	char *buffer_reply)
590	{
591	struct og_buffer og_buffer = {
592	.data = buffer_reply,
593	};
594	json_t root, value, *array;
595	const char *key;
596	unsigned int i;
597	int err = 0;
598
599	if (json_typeof(element) != JSON_OBJECT)
600	return -1;
601
602	json_object_foreach(element, key, value) {
603	if (!strcmp(key, "clients"))
604	err = og_json_parse_clients(value, params);
605
606	if (err < 0)
607	return err;
608	}
609
610	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
611	return -1;
612
613	array = json_array();
614	if (!array)
615	return -1;
616
617	for (i = 0; i < params->ips_array_len; i++) {
618	json_t object, output, *addr;
619	char data[4096] = {};
620	char filename[4096];
621	int fd, numbytes;
622
623	sprintf(filename, "/tmp/_Seconsola_%s", params->ips_array[i]);
624
625	fd = open(filename, O_RDONLY);
626	if (!fd)
627	return -1;
628
629	numbytes = read(fd, data, sizeof(data));
630	if (numbytes < 0) {
631	close(fd);
632	return -1;
633	}
634	data[sizeof(data) - 1] = '\0';
635	close(fd);
636
637	object = json_object();
638	if (!object) {
639	json_decref(array);
640	return -1;
641	}
642	addr = json_string(params->ips_array[i]);
643	if (!addr) {
644	json_decref(object);
645	json_decref(array);
646	return -1;
647	}
648	json_object_set_new(object, "addr", addr);
649
650	output = json_string(data);
651	if (!output) {
652	json_decref(object);
653	json_decref(array);
654	return -1;
655	}
656	json_object_set_new(object, "output", output);
657
658	json_array_append_new(array, object);
659	}
660
661	root = json_pack("{s:o}", "clients", array);
662	if (!root)
663	return -1;
664
665	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
666	json_decref(root);
667
668	return 0;
669	}
670
671	static int og_cmd_session(json_t element, struct og_msg_params params)
672	{
673	json_t clients, value;
674	const char *key;
675	int err = 0;
676
677	if (json_typeof(element) != JSON_OBJECT)
678	return -1;
679
680	json_object_foreach(element, key, value) {
681	if (!strcmp(key, "clients")) {
682	err = og_json_parse_clients(value, params);
683	} else if (!strcmp(key, "disk")) {
684	err = og_json_parse_string(value, &params->disk);
685	params->flags \|= OG_REST_PARAM_DISK;
686	} else if (!strcmp(key, "partition")) {
687	err = og_json_parse_string(value, &params->partition);
688	params->flags \|= OG_REST_PARAM_PARTITION;
689	}
690
691	if (err < 0)
692	return err;
693	}
694
695	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR \|
696	OG_REST_PARAM_DISK \|
697	OG_REST_PARAM_PARTITION))
698	return -1;
699
700	clients = json_copy(element);
701	json_object_del(clients, "clients");
702
703	return og_send_request(OG_METHOD_POST, OG_CMD_SESSION, params, clients);
704	}
705
706	static int og_cmd_get_session(json_t element, struct og_msg_params params,
707	char *buffer_reply)
708	{
709	json_t value, root, array, item;
710	const char key, msglog, *os_name;
711	unsigned int disk, partition;
712	struct og_dbi *dbi;
713	dbi_result result;
714	int err = 0;
715
716	struct og_buffer og_buffer = {
717	.data = buffer_reply
718	};
719
720	json_object_foreach(element, key, value) {
721	if (!strcmp(key, "client"))
722	err = og_json_parse_clients(value, params);
723	else
724	err = -1;
725
726	if (err < 0)
727	return err;
728	}
729
730	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
731	return -1;
732
733	dbi = og_dbi_open(&ogconfig.db);
734	if (!dbi) {
735	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
736	__func__, __LINE__);
737	return -1;
738	}
739
740	result = dbi_conn_queryf(dbi->conn,
741	"SELECT op.numdisk, op.numpar, nom.nombreso "
742	"FROM ordenadores o "
743	"INNER JOIN ordenadores_particiones op "
744	" ON o.idordenador = op.idordenador "
745	"INNER JOIN nombresos nom "
746	" ON op.idnombreso = nom.idnombreso "
747	"WHERE o.ip = '%s'",
748	params->ips_array[0]);
749	if (!result) {
750	dbi_conn_error(dbi->conn, &msglog);
751	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
752	__func__, __LINE__, msglog);
753	og_dbi_close(dbi);
754	return -1;
755	}
756
757	array = json_array();
758	if (!array) {
759	dbi_result_free(result);
760	og_dbi_close(dbi);
761	return -1;
762	}
763
764	while (dbi_result_next_row(result)) {
765	item = json_object();
766	if (!item) {
767	dbi_result_free(result);
768	og_dbi_close(dbi);
769	json_decref(array);
770	return -1;
771	}
772
773	disk = dbi_result_get_uint(result, "numdisk");
774	partition = dbi_result_get_uint(result, "numpar");
775	os_name = dbi_result_get_string(result, "nombreso");
776
777	json_object_set_new(item, "disk", json_integer(disk));
778	json_object_set_new(item, "partition", json_integer(partition));
779	json_object_set_new(item, "name", json_string(os_name));
780	json_array_append_new(array, item);
781	}
782
783	dbi_result_free(result);
784	og_dbi_close(dbi);
785
786	root = json_object();
787	if (!root){
788	json_decref(array);
789	return -1;
790	}
791
792	json_object_set_new(root, "sessions", array);
793
794	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
795	json_decref(root);
796	return 0;
797	}
798
799	static int og_cmd_poweroff(json_t element, struct og_msg_params params)
800	{
801	const char *key;
802	json_t *value;
803	int err = 0;
804
805	if (json_typeof(element) != JSON_OBJECT)
806	return -1;
807
808	json_object_foreach(element, key, value) {
809	if (!strcmp(key, "clients"))
810	err = og_json_parse_clients(value, params);
811
812	if (err < 0)
813	break;
814	}
815
816	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
817	return -1;
818
819	return og_send_request(OG_METHOD_POST, OG_CMD_POWEROFF, params, NULL);
820	}
821
822	static int og_cmd_refresh(json_t element, struct og_msg_params params)
823	{
824	const char *key;
825	json_t *value;
826	int err = 0;
827
828	if (json_typeof(element) != JSON_OBJECT)
829	return -1;
830
831	json_object_foreach(element, key, value) {
832	if (!strcmp(key, "clients"))
833	err = og_json_parse_clients(value, params);
834
835	if (err < 0)
836	break;
837	}
838
839	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
840	return -1;
841
842	return og_send_request(OG_METHOD_GET, OG_CMD_REFRESH, params, NULL);
843	}
844
845	static int og_cmd_reboot(json_t element, struct og_msg_params params)
846	{
847	const char *key;
848	json_t *value;
849	int err = 0;
850
851	if (json_typeof(element) != JSON_OBJECT)
852	return -1;
853
854	json_object_foreach(element, key, value) {
855	if (!strcmp(key, "clients"))
856	err = og_json_parse_clients(value, params);
857
858	if (err < 0)
859	break;
860	}
861
862	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
863	return -1;
864
865	return og_send_request(OG_METHOD_POST, OG_CMD_REBOOT, params, NULL);
866	}
867
868	#define OG_TFTP_TMPL_PATH "/opt/opengnsys/tftpboot/menu.lst/templates"
869
870	static int og_cmd_get_modes(json_t element, struct og_msg_params params,
871	char *buffer_reply)
872	{
873	struct og_buffer og_buffer = {
874	.data = buffer_reply
875	};
876	json_t root, modes;
877	struct dirent *dent;
878	DIR *d = NULL;
879
880	root = json_object();
881	if (!root)
882	return -1;
883
884	modes = json_array();
885	if (!modes) {
886	json_decref(root);
887	return -1;
888	}
889
890	d = opendir(OG_TFTP_TMPL_PATH);
891	if (!d) {
892	json_decref(modes);
893	json_decref(root);
894	syslog(LOG_ERR, "Cannot open directory %s\n",
895	OG_TFTP_TMPL_PATH);
896	return -1;
897	}
898
899	dent = readdir(d);
900	while (dent) {
901	if (dent->d_type != DT_REG) {
902	dent = readdir(d);
903	continue;
904	}
905	json_array_append_new(modes, json_string(dent->d_name));
906	dent = readdir(d);
907	}
908
909	json_object_set_new(root, "modes", modes);
910	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
911	json_decref(root);
912	closedir(d);
913
914	return 0;
915	}
916
917	static int og_change_db_mode(struct og_dbi dbi, const char mac,
918	const char * mode)
919	{
920	const char *msglog;
921	dbi_result result;
922
923	result = dbi_conn_queryf(dbi->conn,
924	"UPDATE ordenadores SET arranque='%s' "
925	"WHERE mac='%s'",
926	mode, mac);
927
928	if (!result) {
929	dbi_conn_error(dbi->conn, &msglog);
930	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
931	__func__, __LINE__, msglog);
932	return -1;
933	}
934
935	dbi_result_free(result);
936	return 0;
937	}
938
939	static int og_set_client_mode(struct og_dbi dbi, const char mac,
940	const char mode, const char template_name)
941	{
942	char filename[PATH_MAX + 1] = "/tmp/mode_params_XXXXXX";
943	char cmd_params[16384] = {};
944	char params[4096] = "\0";
945	const char *msglog;
946	dbi_result result;
947	unsigned int i;
948	int numbytes;
949	int status;
950	int fd;
951
952	result = dbi_conn_queryf(dbi->conn,
953	"SELECT ' LANG=%s', ' ip=', CONCAT_WS(':', ordenadores.ip, (SELECT (@serverip:=ipserveradm) FROM entornos LIMIT 1), aulas.router, aulas.netmask, ordenadores.nombreordenador, ordenadores.netiface, 'none'), ' group=', REPLACE(TRIM(aulas.nombreaula), ' ', '_'), ' ogrepo=', (@repoip:=IFNULL(repositorios.ip, '')), ' oglive=', @serverip, ' oglog=', @serverip, ' ogshare=', @serverip, ' oglivedir=', ordenadores.oglivedir, ' ogprof=', IF(ordenadores.idordenador=aulas.idordprofesor, 'true', 'false'), IF(perfileshard.descripcion<>'', CONCAT(' hardprofile=', REPLACE(TRIM(perfileshard.descripcion), ' ', '_')), ''), IF(aulas.ntp<>'', CONCAT(' ogntp=', aulas.ntp), ''), IF(aulas.dns<>'', CONCAT(' ogdns=', aulas.dns), ''), IF(aulas.proxy<>'', CONCAT(' ogproxy=', aulas.proxy), ''), IF(entidades.ogunit=1 AND NOT centros.directorio='', CONCAT(' ogunit=', centros.directorio), ''), CASE WHEN menus.resolucion IS NULL THEN '' WHEN menus.resolucion <= '999' THEN CONCAT(' vga=', menus.resolucion) WHEN menus.resolucion LIKE '%:%' THEN CONCAT(' video=', menus.resolucion) ELSE menus.resolucion END FROM ordenadores JOIN aulas USING(idaula) JOIN centros USING(idcentro) JOIN entidades USING(identidad) LEFT JOIN repositorios USING(idrepositorio) LEFT JOIN perfileshard USING(idperfilhard) LEFT JOIN menus USING(idmenu) WHERE ordenadores.mac='%s'", getenv("LANG"), mac);
954
955	if (dbi_result_get_numrows(result) != 1) {
956	dbi_conn_error(dbi->conn, &msglog);
957	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
958	__FILE__, __LINE__, msglog);
959	dbi_result_free(result);
960	return -1;
961	}
962	dbi_result_next_row(result);
963
964	for (i = 1; i <= dbi_result_get_numfields(result); ++i)
965	strcat(params, dbi_result_get_string_idx(result, i));
966
967	dbi_result_free(result);
968
969	snprintf(cmd_params, sizeof(cmd_params),
970	"MODE_FILE='%s'\nMAC='%s'\nDATA='%s'\n"
971	"MODE='PERM'\nTEMPLATE_NAME='%s'",
972	mode, mac, params, template_name);
973
974	fd = mkstemp(filename);
975	if (fd < 0) {
976	syslog(LOG_ERR, "cannot generate temp file (%s:%d)\n",
977	__func__, __LINE__);
978	return -1;
979	}
980
981	numbytes = write(fd, cmd_params, strlen(cmd_params) + 1);
982	close(fd);
983
984	if (numbytes < 0) {
985	syslog(LOG_ERR, "cannot write file\n");
986	unlink(filename);
987	return -1;
988	}
989
990	if (fork() == 0) {
991	execlp("/bin/bash", "/bin/bash",
992	"/opt/opengnsys/bin/setclientmode", filename, NULL);
993	syslog(LOG_ERR, "failed script execution (%s:%d)\n",
994	__func__, __LINE__);
995	exit(EXIT_FAILURE);
996	} else {
997	wait(&status);
998	}
999	unlink(filename);
1000
1001	if (!WIFEXITED(status) \|\| WEXITSTATUS(status) != 0) {
1002	syslog(LOG_ERR, "failed script execution (%s:%d)\n",
1003	__func__, __LINE__);
1004	return -1;
1005	}
1006
1007	if (og_change_db_mode(dbi, mac, mode) < 0) {
1008	syslog(LOG_ERR, "failed to change db mode (%s:%d)\n",
1009	__func__, __LINE__);
1010	return -1;
1011	}
1012
1013	return 0;
1014	}
1015
1016	static int og_cmd_post_modes(json_t element, struct og_msg_params params)
1017	{
1018	char ips_str[(OG_DB_IP_MAXLEN + 1) * OG_CLIENTS_MAX + 1] = {};
1019	char template_file[PATH_MAX + 1] = {};
1020	char template_name[PATH_MAX + 1] = {};
1021	char first_line[PATH_MAX + 1] = {};
1022	const char mode_str, mac;
1023	int ips_str_len = 0;
1024	struct og_dbi *dbi;
1025	uint64_t flags = 0;
1026	dbi_result result;
1027	const char *key;
1028	json_t *value;
1029	int err = 0;
1030	FILE *f;
1031	int i;
1032
1033	json_object_foreach(element, key, value) {
1034	if (!strcmp(key, "clients")) {
1035	err = og_json_parse_clients(value, params);
1036	} else if (!strcmp(key, "mode")) {
1037	err = og_json_parse_string(value, &mode_str);
1038	flags \|= OG_REST_PARAM_MODE;
1039	} else {
1040	err = -1;
1041	}
1042
1043	if (err < 0)
1044	break;
1045	}
1046
1047	if (!og_flags_validate(flags, OG_REST_PARAM_MODE) \|\|
1048	!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1049	return -1;
1050
1051	snprintf(template_file, sizeof(template_file), "%s/%s",
1052	OG_TFTP_TMPL_PATH, mode_str);
1053	f = fopen(template_file, "r");
1054	if (!f) {
1055	syslog(LOG_ERR, "cannot open file (%s:%d)\n",
1056	__func__, __LINE__);
1057	return -1;
1058	}
1059
1060	if (!fgets(first_line, sizeof(first_line), f)) {
1061	fclose(f);
1062	syslog(LOG_ERR, "cannot read file (%s:%d)\n",
1063	__func__, __LINE__);
1064	return -1;
1065	}
1066
1067	fclose(f);
1068
1069	if (sscanf(first_line, "##NO-TOCAR-ESTA-LINEA %s", template_name) != 1) {
1070	syslog(LOG_ERR, "malformed template: %s", first_line);
1071	return -1;
1072	}
1073
1074	for (i = 0; i < params->ips_array_len; ++i) {
1075	ips_str_len += snprintf(ips_str + ips_str_len,
1076	sizeof(ips_str) - ips_str_len,
1077	"'%s',", params->ips_array[i]);
1078	}
1079	ips_str[ips_str_len - 1] = '\0';
1080
1081	dbi = og_dbi_open(&ogconfig.db);
1082	if (!dbi) {
1083	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
1084	__func__, __LINE__);
1085	return -1;
1086	}
1087
1088	result = dbi_conn_queryf(dbi->conn,
1089	"SELECT mac FROM ordenadores "
1090	"WHERE ip IN (%s)", ips_str);
1091
1092	while (dbi_result_next_row(result)) {
1093	mac = dbi_result_get_string(result, "mac");
1094	err = og_set_client_mode(dbi, mac, mode_str, template_name);
1095	if (err != 0) {
1096	dbi_result_free(result);
1097	og_dbi_close(dbi);
1098	return -1;
1099	}
1100	}
1101
1102	dbi_result_free(result);
1103	og_dbi_close(dbi);
1104
1105	return 0;
1106	}
1107
1108	static int og_cmd_get_client_setup(json_t *element,
1109	struct og_msg_params *params,
1110	char *buffer_reply)
1111	{
1112	json_t value, root, partitions_array, partition_json;
1113	const char key, msglog;
1114	unsigned int len_part;
1115	struct og_dbi *dbi;
1116	dbi_result result;
1117	int err = 0;
1118
1119	struct og_buffer og_buffer = {
1120	.data = buffer_reply
1121	};
1122
1123	struct {
1124	int disk;
1125	int number;
1126	int code;
1127	int size;
1128	int filesystem;
1129	int format;
1130	int os;
1131	int used_size;
1132	int image;
1133	int software;
1134	} partition;
1135
1136	json_object_foreach(element, key, value) {
1137	if (!strcmp(key, "client")) {
1138	err = og_json_parse_clients(value, params);
1139	}
1140
1141	if (err < 0)
1142	break;
1143	}
1144
1145	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1146	return -1;
1147
1148	if (params->ips_array_len != 1)
1149	return -1;
1150
1151	root = json_object();
1152	if (!root)
1153	return -1;
1154
1155	partitions_array = json_array();
1156	if (!partitions_array) {
1157	json_decref(root);
1158	return -1;
1159	}
1160	json_object_set_new(root, "partitions", partitions_array);
1161
1162	dbi = og_dbi_open(&ogconfig.db);
1163	if (!dbi) {
1164	json_decref(root);
1165	syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1166	__func__, __LINE__);
1167	return -1;
1168	}
1169
1170	result = dbi_conn_queryf(dbi->conn,
1171	"SELECT numdisk, numpar, codpar, tamano, "
1172	" uso, idsistemafichero, idnombreso, "
1173	" idimagen, idperfilsoft "
1174	"FROM ordenadores_particiones "
1175	"INNER JOIN ordenadores "
1176	"ON ordenadores.idordenador = ordenadores_particiones.idordenador "
1177	"WHERE ordenadores.ip='%s'",
1178	params->ips_array[0]);
1179	if (!result) {
1180	json_decref(root);
1181	dbi_conn_error(dbi->conn, &msglog);
1182	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1183	__func__, __LINE__, msglog);
1184	og_dbi_close(dbi);
1185	return -1;
1186	}
1187
1188	len_part = 0;
1189	/* partition 0 represents the full disk, hence OG_PARTITION_MAX + 1. */
1190	while (dbi_result_next_row(result) && len_part < OG_PARTITION_MAX + 1) {
1191	partition.disk = dbi_result_get_int(result, "numdisk");
1192	partition.number = dbi_result_get_int(result, "numpar");
1193	partition.code = dbi_result_get_int(result, "codpar");
1194	partition.size = dbi_result_get_int(result, "tamano");
1195	partition.used_size = dbi_result_get_int(result, "uso");
1196	partition.filesystem = dbi_result_get_int(result, "idsistemafichero");
1197	partition.os = dbi_result_get_int(result, "idnombreso");
1198	partition.image = dbi_result_get_int(result, "idimagen");
1199	partition.software = dbi_result_get_int(result, "idperfilsoft");
1200
1201	partition_json = json_object();
1202	if (!partition_json) {
1203	json_decref(root);
1204	dbi_result_free(result);
1205	og_dbi_close(dbi);
1206	return -1;
1207	}
1208
1209	json_object_set_new(partition_json, "disk",
1210	json_integer(partition.disk));
1211	json_object_set_new(partition_json, "partition",
1212	json_integer(partition.number));
1213	json_object_set_new(partition_json, "code",
1214	json_integer(partition.code));
1215	json_object_set_new(partition_json, "size",
1216	json_integer(partition.size));
1217	json_object_set_new(partition_json, "used_size",
1218	json_integer(partition.used_size));
1219	json_object_set_new(partition_json, "filesystem",
1220	json_integer(partition.filesystem));
1221	json_object_set_new(partition_json, "os",
1222	json_integer(partition.os));
1223	json_object_set_new(partition_json, "image",
1224	json_integer(partition.image));
1225	json_object_set_new(partition_json, "software",
1226	json_integer(partition.software));
1227	json_array_append_new(partitions_array, partition_json);
1228
1229	++len_part;
1230	}
1231
1232	dbi_result_free(result);
1233	og_dbi_close(dbi);
1234
1235	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1236	json_decref(root);
1237	return 0;
1238	}
1239
1240	static int og_cmd_get_client_info(json_t *element,
1241	struct og_msg_params *params,
1242	char *buffer_reply)
1243	{
1244	struct og_computer computer = {};
1245	json_t value, root;
1246	struct in_addr addr;
1247	struct og_dbi *dbi;
1248	const char *key;
1249	int err = 0;
1250
1251	struct og_buffer og_buffer = {
1252	.data = buffer_reply
1253	};
1254
1255	json_object_foreach(element, key, value) {
1256	if (!strcmp(key, "client")) {
1257	err = og_json_parse_clients(value, params);
1258	}
1259
1260	if (err < 0)
1261	break;
1262	}
1263
1264	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1265	return -1;
1266
1267	if (params->ips_array_len != 1)
1268	return -1;
1269
1270	if (inet_aton(params->ips_array[0], &addr) == 0)
1271	return -1;
1272
1273	dbi = og_dbi_open(&ogconfig.db);
1274	if (!dbi) {
1275	syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1276	__func__, __LINE__);
1277	return -1;
1278	}
1279
1280	if (og_dbi_get_computer_info(dbi, &computer, addr)) {
1281	og_dbi_close(dbi);
1282	return -1;
1283	}
1284
1285	og_dbi_close(dbi);
1286
1287	root = json_object();
1288	if (!root)
1289	return -1;
1290
1291	json_object_set_new(root, "serial_number",
1292	json_string(computer.serial_number));
1293	json_object_set_new(root, "hardware_id",
1294	json_integer(computer.hardware_id));
1295	json_object_set_new(root, "netdriver", json_string(computer.netdriver));
1296	json_object_set_new(root, "maintenance", json_boolean(computer.name));
1297	json_object_set_new(root, "netiface", json_string(computer.netiface));
1298	json_object_set_new(root, "repo_id", json_integer(computer.repo_id));
1299	json_object_set_new(root, "livedir", json_string(computer.livedir));
1300	json_object_set_new(root, "netmask", json_string(computer.netmask));
1301	json_object_set_new(root, "center", json_integer(computer.center));
1302	json_object_set_new(root, "remote", json_boolean(computer.remote));
1303	json_object_set_new(root, "room", json_integer(computer.room));
1304	json_object_set_new(root, "name", json_string(computer.name));
1305	json_object_set_new(root, "boot", json_string(computer.boot));
1306	json_object_set_new(root, "mac", json_string(computer.mac));
1307	json_object_set_new(root, "id", json_integer(computer.id));
1308	json_object_set_new(root, "ip", json_string(computer.ip));
1309
1310	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1311	json_decref(root);
1312	return 0;
1313	}
1314
1315	static int og_cmd_post_client_add(json_t *element,
1316	struct og_msg_params *params,
1317	char *buffer_reply)
1318	{
1319	struct og_computer computer = {};
1320	const char key, msglog;
1321	struct og_dbi *dbi;
1322	dbi_result result;
1323	json_t *value;
1324	int err = 0;
1325
1326	json_object_foreach(element, key, value) {
1327	if (!strcmp(key, "serial_number")) {
1328	err = og_json_parse_string_copy(value,
1329	computer.serial_number,
1330	sizeof(computer.serial_number));
1331	} else if (!strcmp(key, "hardware_id")) {
1332	err = og_json_parse_uint(value, &computer.hardware_id);
1333	} else if (!strcmp(key, "netdriver")) {
1334	err = og_json_parse_string_copy(value,
1335	computer.netdriver,
1336	sizeof(computer.netdriver));
1337	} else if (!strcmp(key, "maintenance")) {
1338	err = og_json_parse_bool(value, &computer.maintenance);
1339	} else if (!strcmp(key, "netiface")) {
1340	err = og_json_parse_string_copy(value,
1341	computer.netiface,
1342	sizeof(computer.netiface));
1343	} else if (!strcmp(key, "repo_id")) {
1344	err = og_json_parse_uint(value, &computer.repo_id);
1345	} else if (!strcmp(key, "livedir")) {
1346	err = og_json_parse_string_copy(value,
1347	computer.livedir,
1348	sizeof(computer.livedir));
1349	} else if (!strcmp(key, "netmask")) {
1350	err = og_json_parse_string_copy(value,
1351	computer.netmask,
1352	sizeof(computer.netmask));
1353	} else if (!strcmp(key, "remote")) {
1354	err = og_json_parse_bool(value, &computer.remote);
1355	} else if (!strcmp(key, "room")) {
1356	err = og_json_parse_uint(value, &computer.room);
1357	} else if (!strcmp(key, "name")) {
1358	err = og_json_parse_string_copy(value,
1359	computer.name,
1360	sizeof(computer.name));
1361	} else if (!strcmp(key, "boot")) {
1362	err = og_json_parse_string_copy(value,
1363	computer.boot,
1364	sizeof(computer.boot));
1365	} else if (!strcmp(key, "mac")) {
1366	err = og_json_parse_string_copy(value,
1367	computer.mac,
1368	sizeof(computer.mac));
1369	} else if (!strcmp(key, "ip")) {
1370	err = og_json_parse_string_copy(value,
1371	computer.ip,
1372	sizeof(computer.ip));
1373	}
1374
1375	if (err < 0)
1376	break;
1377	}
1378
1379	dbi = og_dbi_open(&ogconfig.db);
1380	if (!dbi) {
1381	syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1382	__func__, __LINE__);
1383	return -1;
1384	}
1385
1386	result = dbi_conn_queryf(dbi->conn,
1387	"SELECT ip FROM ordenadores WHERE ip='%s'",
1388	computer.ip);
1389
1390	if (!result) {
1391	dbi_conn_error(dbi->conn, &msglog);
1392	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1393	__func__, __LINE__, msglog);
1394	og_dbi_close(dbi);
1395	return -1;
1396	}
1397
1398	if (dbi_result_get_numrows(result) > 0) {
1399	syslog(LOG_ERR, "client with the same IP already exists: %s\n",
1400	computer.ip);
1401	dbi_result_free(result);
1402	og_dbi_close(dbi);
1403	return -1;
1404	}
1405	dbi_result_free(result);
1406
1407	result = dbi_conn_queryf(dbi->conn,
1408	"INSERT INTO ordenadores("
1409	" nombreordenador,"
1410	" numserie,"
1411	" ip,"
1412	" mac,"
1413	" idaula,"
1414	" idperfilhard,"
1415	" idrepositorio,"
1416	" mascara,"
1417	" arranque,"
1418	" netiface,"
1419	" netdriver,"
1420	" oglivedir,"
1421	" inremotepc,"
1422	" maintenance"
1423	") VALUES ('%s', '%s', '%s', '%s', %u, %u,"
1424	" %u, '%s', '%s', '%s', '%s',"
1425	" '%s', %u, %u)",
1426	computer.name, computer.serial_number,
1427	computer.ip, computer.mac, computer.room,
1428	computer.hardware_id, computer.repo_id,
1429	computer.netmask, computer.boot,
1430	computer.netiface, computer.netdriver,
1431	computer.livedir, computer.remote,
1432	computer.maintenance);
1433
1434	if (!result) {
1435	dbi_conn_error(dbi->conn, &msglog);
1436	syslog(LOG_ERR, "failed to add client to database (%s:%d) %s\n",
1437	__func__, __LINE__, msglog);
1438	og_dbi_close(dbi);
1439	return -1;
1440	}
1441
1442	dbi_result_free(result);
1443	og_dbi_close(dbi);
1444	return 0;
1445	}
1446
1447	static int og_cmd_post_client_delete(json_t *element,
1448	struct og_msg_params *params)
1449	{
1450	const char key, msglog;
1451	struct og_dbi *dbi;
1452	dbi_result result;
1453	unsigned int i;
1454	json_t *value;
1455	int err = 0;
1456
1457	json_object_foreach(element, key, value) {
1458	if (!strcmp(key, "clients"))
1459	err = og_json_parse_clients(value, params);
1460	else
1461	err = -1;
1462
1463	if (err < 0)
1464	break;
1465	}
1466
1467	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1468	return -1;
1469
1470	dbi = og_dbi_open(&ogconfig.db);
1471	if (!dbi) {
1472	syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1473	__func__, __LINE__);
1474	return -1;
1475	}
1476
1477	for (i = 0; i < params->ips_array_len; i++) {
1478	result = dbi_conn_queryf(dbi->conn,
1479	"DELETE FROM ordenadores WHERE ip='%s'",
1480	params->ips_array[i]);
1481
1482	if (!result) {
1483	dbi_conn_error(dbi->conn, &msglog);
1484	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1485	__func__, __LINE__, msglog);
1486	og_dbi_close(dbi);
1487	return -1;
1488	}
1489
1490	dbi_result_free(result);
1491	}
1492
1493	og_dbi_close(dbi);
1494	return 0;
1495	}
1496
1497	static int og_cmd_stop(json_t element, struct og_msg_params params)
1498	{
1499	const char *key;
1500	json_t *value;
1501	int err = 0;
1502
1503	if (json_typeof(element) != JSON_OBJECT)
1504	return -1;
1505
1506	json_object_foreach(element, key, value) {
1507	if (!strcmp(key, "clients"))
1508	err = og_json_parse_clients(value, params);
1509
1510	if (err < 0)
1511	break;
1512	}
1513
1514	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1515	return -1;
1516
1517	return og_send_request(OG_METHOD_POST, OG_CMD_STOP, params, NULL);
1518	}
1519
1520	static int og_cmd_hardware(json_t element, struct og_msg_params params)
1521	{
1522	const char *key;
1523	json_t *value;
1524	int err = 0;
1525
1526	if (json_typeof(element) != JSON_OBJECT)
1527	return -1;
1528
1529	json_object_foreach(element, key, value) {
1530	if (!strcmp(key, "clients"))
1531	err = og_json_parse_clients(value, params);
1532
1533	if (err < 0)
1534	break;
1535	}
1536
1537	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1538	return -1;
1539
1540	return og_send_request(OG_METHOD_GET, OG_CMD_HARDWARE, params, NULL);
1541	}
1542
1543	static int og_cmd_get_hardware(json_t element, struct og_msg_params params,
1544	char *buffer_reply)
1545	{
1546	const char key, msglog, hw_item, hw_type;
1547	json_t value, root, array, item;
1548	struct og_dbi *dbi;
1549	dbi_result result;
1550	int err = 0;
1551
1552	struct og_buffer og_buffer = {
1553	.data = buffer_reply
1554	};
1555
1556	json_object_foreach(element, key, value) {
1557	if (!strcmp(key, "client"))
1558	err = og_json_parse_clients(value, params);
1559	else
1560	err = -1;
1561
1562	if (err < 0)
1563	return err;
1564	}
1565
1566	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1567	return -1;
1568
1569	dbi = og_dbi_open(&ogconfig.db);
1570	if (!dbi) {
1571	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
1572	__func__, __LINE__);
1573	return -1;
1574	}
1575
1576	result = dbi_conn_queryf(dbi->conn,
1577	"SELECT hardwares.descripcion AS item, "
1578	" tipohardwares.descripcion AS type "
1579	"FROM hardwares "
1580	"INNER JOIN perfileshard_hardwares "
1581	" ON hardwares.idhardware = perfileshard_hardwares.idhardware "
1582	"INNER JOIN ordenadores "
1583	" ON perfileshard_hardwares.idperfilhard = ordenadores.idperfilhard "
1584	"INNER JOIN tipohardwares "
1585	" ON hardwares.idtipohardware = tipohardwares.idtipohardware "
1586	"WHERE ordenadores.ip = '%s'",
1587	params->ips_array[0]);
1588	if (!result) {
1589	dbi_conn_error(dbi->conn, &msglog);
1590	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1591	__func__, __LINE__, msglog);
1592	og_dbi_close(dbi);
1593	return -1;
1594	}
1595
1596	array = json_array();
1597	if (!array) {
1598	dbi_result_free(result);
1599	og_dbi_close(dbi);
1600	return -1;
1601	}
1602
1603	while (dbi_result_next_row(result)) {
1604	item = json_object();
1605	if (!item) {
1606	dbi_result_free(result);
1607	og_dbi_close(dbi);
1608	json_decref(array);
1609	return -1;
1610	}
1611
1612	hw_item = dbi_result_get_string(result, "item");
1613	hw_type = dbi_result_get_string(result, "type");
1614
1615	json_object_set_new(item, "type", json_string(hw_type));
1616	json_object_set_new(item, "description", json_string(hw_item));
1617	json_array_append_new(array, item);
1618	}
1619
1620	dbi_result_free(result);
1621	og_dbi_close(dbi);
1622
1623	root = json_object();
1624	if (!root){
1625	json_decref(array);
1626	return -1;
1627	}
1628
1629	json_object_set_new(root, "hardware", array);
1630
1631	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1632	json_decref(root);
1633	return 0;
1634	}
1635
1636	static int og_cmd_software(json_t element, struct og_msg_params params)
1637	{
1638	json_t clients, value;
1639	const char *key;
1640	int err = 0;
1641
1642	if (json_typeof(element) != JSON_OBJECT)
1643	return -1;
1644
1645	json_object_foreach(element, key, value) {
1646	if (!strcmp(key, "clients"))
1647	err = og_json_parse_clients(value, params);
1648	else if (!strcmp(key, "disk")) {
1649	err = og_json_parse_string(value, &params->disk);
1650	params->flags \|= OG_REST_PARAM_DISK;
1651	}
1652	else if (!strcmp(key, "partition")) {
1653	err = og_json_parse_string(value, &params->partition);
1654	params->flags \|= OG_REST_PARAM_PARTITION;
1655	}
1656
1657	if (err < 0)
1658	break;
1659	}
1660
1661	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR \|
1662	OG_REST_PARAM_DISK \|
1663	OG_REST_PARAM_PARTITION))
1664	return -1;
1665
1666	clients = json_copy(element);
1667	json_object_del(clients, "clients");
1668
1669	return og_send_request(OG_METHOD_POST, OG_CMD_SOFTWARE, params, clients);
1670	}
1671
1672	static int og_cmd_get_software(json_t element, struct og_msg_params params,
1673	char *buffer_reply)
1674	{
1675	json_t value, software, *root;
1676	const char key, msglog, *name;
1677	uint64_t disk, partition;
1678	uint64_t flags = 0;
1679	struct og_dbi *dbi;
1680	dbi_result result;
1681	int err = 0;
1682
1683	struct og_buffer og_buffer = {
1684	.data = buffer_reply
1685	};
1686
1687	if (json_typeof(element) != JSON_OBJECT)
1688	return -1;
1689
1690	json_object_foreach(element, key, value) {
1691	if (!strcmp(key, "client")) {
1692	err = og_json_parse_clients(value, params);
1693	} else if (!strcmp(key, "disk")) {
1694	err = og_json_parse_uint64(value, &disk);
1695	flags \|= OG_REST_PARAM_DISK;
1696	} else if (!strcmp(key, "partition")) {
1697	err = og_json_parse_uint64(value, &partition);
1698	flags \|= OG_REST_PARAM_PARTITION;
1699	}
1700
1701	if (err < 0)
1702	break;
1703	}
1704
1705	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR) \|\|
1706	!og_flags_validate(flags, OG_REST_PARAM_DISK \|
1707	OG_REST_PARAM_PARTITION))
1708	return -1;
1709
1710	dbi = og_dbi_open(&ogconfig.db);
1711	if (!dbi) {
1712	syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1713	__func__, __LINE__);
1714	return -1;
1715	}
1716
1717	result = dbi_conn_queryf(dbi->conn,
1718	"SELECT s.descripcion "
1719	"FROM softwares s "
1720	"INNER JOIN perfilessoft_softwares pss "
1721	"ON s.idsoftware = pss.idsoftware "
1722	"INNER JOIN ordenadores_particiones op "
1723	"ON pss.idperfilsoft = op.idperfilsoft "
1724	"INNER JOIN ordenadores o "
1725	"ON o.idordenador = op.idordenador "
1726	"WHERE o.ip='%s' AND "
1727	" op.numdisk=%lu AND "
1728	" op.numpar=%lu",
1729	params->ips_array[0], disk, partition);
1730	if (!result) {
1731	dbi_conn_error(dbi->conn, &msglog);
1732	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1733	__func__, __LINE__, msglog);
1734	return -1;
1735	}
1736	software = json_array();
1737	if (!software) {
1738	dbi_result_free(result);
1739	og_dbi_close(dbi);
1740	return -1;
1741	}
1742
1743	while (dbi_result_next_row(result)) {
1744	name = dbi_result_get_string(result, "descripcion");
1745	json_array_append_new(software, json_string(name));
1746	}
1747
1748	dbi_result_free(result);
1749	og_dbi_close(dbi);
1750
1751	root = json_object();
1752	if (!root) {
1753	json_decref(software);
1754	return -1;
1755	}
1756	json_object_set_new(root, "software", software);
1757
1758	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1759	json_decref(root);
1760	return 0;
1761	}
1762
1763	#define OG_IMAGE_TYPE_MAXLEN 4
1764
1765	static int og_get_image_stats(const char *name,
1766	struct stat *image_stats)
1767	{
1768	const char *dir = ogconfig.repo.dir;
1769	char filename[PATH_MAX + 1];
1770
1771	snprintf(filename, sizeof(filename), "%s/%s.img", dir, name);
1772	if (stat(filename, image_stats) < 0) {
1773	syslog(LOG_ERR, "%s image does not exists", name);
1774	return -1;
1775	}
1776	return 0;
1777	}
1778
1779	static json_t *og_json_disk_alloc()
1780	{
1781	const char *dir = ogconfig.repo.dir;
1782	struct statvfs buffer;
1783	json_t *disk_json;
1784	int ret;
1785
1786	ret = statvfs(dir, &buffer);
1787	if (ret)
1788	return NULL;
1789
1790	disk_json = json_object();
1791	if (!disk_json)
1792	return NULL;
1793
1794	json_object_set_new(disk_json, "total",
1795	json_integer(buffer.f_blocks * buffer.f_frsize));
1796	json_object_set_new(disk_json, "free",
1797	json_integer(buffer.f_bfree * buffer.f_frsize));
1798
1799	return disk_json;
1800	}
1801
1802	#define OG_PERMS_IRWX (S_IRWXU \| S_IRWXG \| S_IRWXO)
1803	#define OG_PERMS_MAXLEN 4
1804
1805	static json_t og_json_image_alloc(struct og_image image)
1806	{
1807	char perms_string[OG_PERMS_MAXLEN];
1808	json_t *image_json;
1809	char *modified;
1810	uint16_t perms;
1811
1812	image_json = json_object();
1813	if (!image_json)
1814	return NULL;
1815
1816	perms = image->image_stats.st_mode & OG_PERMS_IRWX;
1817	snprintf(perms_string, sizeof(perms_string), "%o", perms);
1818
1819	modified = ctime(&image->image_stats.st_mtime);
1820	modified[strlen(modified) - 1] = '\0';
1821
1822	json_object_set_new(image_json, "name",
1823	json_string(image->name));
1824	json_object_set_new(image_json, "datasize",
1825	json_integer(image->datasize));
1826	json_object_set_new(image_json, "size",
1827	json_integer(image->image_stats.st_size));
1828	json_object_set_new(image_json, "modified",
1829	json_string(modified));
1830	json_object_set_new(image_json, "permissions",
1831	json_string(perms_string));
1832	json_object_set_new(image_json, "software_id",
1833	json_integer(image->software_id));
1834	json_object_set_new(image_json, "type",
1835	json_integer(image->type));
1836	json_object_set_new(image_json, "id",
1837	json_integer(image->id));
1838
1839	return image_json;
1840	}
1841
1842	static int og_cmd_images(char *buffer_reply)
1843	{
1844	json_t root, images, image_json, disk_json;
1845	struct og_buffer og_buffer = {
1846	.data = buffer_reply
1847	};
1848	struct og_image image;
1849	struct og_dbi *dbi;
1850	dbi_result result;
1851
1852	root = json_object();
1853	if (!root)
1854	return -1;
1855
1856	images = json_array();
1857	if (!images) {
1858	json_decref(root);
1859	return -1;
1860	}
1861
1862	json_object_set_new(root, "images", images);
1863
1864	dbi = og_dbi_open(&ogconfig.db);
1865	if (!dbi) {
1866	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
1867	__func__, __LINE__);
1868	json_decref(root);
1869	return -1;
1870	}
1871
1872	result = dbi_conn_queryf(dbi->conn,
1873	"SELECT i.nombreca, o.nombreordenador, "
1874	" i.clonator, i.compressor, "
1875	" i.filesystem, i.datasize, "
1876	" i.idperfilsoft, i.tipo, "
1877	" i.idimagen "
1878	"FROM imagenes i "
1879	"LEFT JOIN ordenadores o "
1880	"ON i.idordenador = o.idordenador");
1881
1882	while (dbi_result_next_row(result)) {
1883	image = (struct og_image){0};
1884	image.datasize = dbi_result_get_ulonglong(result, "datasize");
1885	image.software_id = dbi_result_get_ulonglong(result, "idperfilsoft");
1886	image.type = dbi_result_get_ulonglong(result, "tipo");
1887	image.id = dbi_result_get_ulonglong(result, "idimagen");
1888	snprintf(image.name, sizeof(image.name), "%s",
1889	dbi_result_get_string(result, "nombreca"));
1890
1891	if (og_get_image_stats(image.name, &image.image_stats)) {
1892	continue;
1893	}
1894
1895	image_json = og_json_image_alloc(&image);
1896	if (!image_json) {
1897	dbi_result_free(result);
1898	og_dbi_close(dbi);
1899	json_decref(root);
1900	return -1;
1901	}
1902
1903	json_array_append(images, image_json);
1904	}
1905
1906	dbi_result_free(result);
1907	og_dbi_close(dbi);
1908
1909	disk_json = og_json_disk_alloc();
1910	if (!disk_json) {
1911	syslog(LOG_ERR, "cannot allocate disk json");
1912	json_decref(root);
1913	return -1;
1914	}
1915
1916	json_object_set_new(root, "disk", disk_json);
1917
1918	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1919	json_decref(root);
1920
1921	return 0;
1922	}
1923
1924	static int og_cmd_create_image(json_t element, struct og_msg_params params)
1925	{
1926	char new_image_id[OG_DB_INT_MAXLEN + 1];
1927	struct og_image image = {};
1928	json_t value, clients;
1929	struct og_dbi *dbi;
1930	const char *key;
1931	int err = 0;
1932
1933	if (json_typeof(element) != JSON_OBJECT)
1934	return -1;
1935
1936	json_object_foreach(element, key, value) {
1937	if (!strcmp(key, "disk")) {
1938	err = og_json_parse_string(value, &params->disk);
1939	params->flags \|= OG_REST_PARAM_DISK;
1940	} else if (!strcmp(key, "partition")) {
1941	err = og_json_parse_string(value, &params->partition);
1942	params->flags \|= OG_REST_PARAM_PARTITION;
1943	} else if (!strcmp(key, "name")) {
1944	err = og_json_parse_string(value, &params->name);
1945	params->flags \|= OG_REST_PARAM_NAME;
1946	} else if (!strcmp(key, "repository")) {
1947	err = og_json_parse_string(value, &params->repository);
1948	params->flags \|= OG_REST_PARAM_REPO;
1949	} else if (!strcmp(key, "clients")) {
1950	err = og_json_parse_clients(value, params);
1951	} else if (!strcmp(key, "id")) {
1952	err = og_json_parse_string(value, &params->id);
1953	params->flags \|= OG_REST_PARAM_ID;
1954	} else if (!strcmp(key, "code")) {
1955	err = og_json_parse_string(value, &params->code);
1956	params->flags \|= OG_REST_PARAM_CODE;
1957	} else if (!strcmp(key, "description")) {
1958	err = og_json_parse_string_copy(value,
1959	image.description,
1960	sizeof(image.description));
1961	} else if (!strcmp(key, "group_id")) {
1962	err = og_json_parse_uint64(value, &image.group_id);
1963	} else if (!strcmp(key, "center_id")) {
1964	err = og_json_parse_uint64(value, &image.center_id);
1965	}
1966
1967
1968	if (err < 0)
1969	break;
1970	}
1971
1972	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR \|
1973	OG_REST_PARAM_DISK \|
1974	OG_REST_PARAM_PARTITION \|
1975	OG_REST_PARAM_CODE \|
1976	OG_REST_PARAM_ID \|
1977	OG_REST_PARAM_NAME \|
1978	OG_REST_PARAM_REPO))
1979	return -1;
1980
1981	/* If there is a description, this means the image is not in the DB. */
1982	if (image.description[0]) {
1983	snprintf(image.name, sizeof(image.name), "%s", params->name);
1984
1985	dbi = og_dbi_open(&ogconfig.db);
1986	if (!dbi) {
1987	syslog(LOG_ERR,
1988	"cannot open connection database (%s:%d)\n",
1989	__func__, __LINE__);
1990	return -1;
1991	}
1992
1993	err = og_dbi_add_image(dbi, &image);
1994
1995	og_dbi_close(dbi);
1996	if (err < 0)
1997	return err;
1998
1999	snprintf(new_image_id, sizeof(new_image_id), "%u", err);
2000	params->id = new_image_id;
2001	}
2002
2003	clients = json_copy(element);
2004	json_object_del(clients, "clients");
2005
2006	return og_send_request(OG_METHOD_POST, OG_CMD_IMAGE_CREATE, params,
2007	clients);
2008	}
2009
2010	static int og_cmd_restore_image(json_t element, struct og_msg_params params)
2011	{
2012	json_t clients, value;
2013	const char *key;
2014	int err = 0;
2015
2016	if (json_typeof(element) != JSON_OBJECT)
2017	return -1;
2018
2019	json_object_foreach(element, key, value) {
2020	if (!strcmp(key, "disk")) {
2021	err = og_json_parse_string(value, &params->disk);
2022	params->flags \|= OG_REST_PARAM_DISK;
2023	} else if (!strcmp(key, "partition")) {
2024	err = og_json_parse_string(value, &params->partition);
2025	params->flags \|= OG_REST_PARAM_PARTITION;
2026	} else if (!strcmp(key, "name")) {
2027	err = og_json_parse_string(value, &params->name);
2028	params->flags \|= OG_REST_PARAM_NAME;
2029	} else if (!strcmp(key, "repository")) {
2030	err = og_json_parse_string(value, &params->repository);
2031	params->flags \|= OG_REST_PARAM_REPO;
2032	} else if (!strcmp(key, "clients")) {
2033	err = og_json_parse_clients(value, params);
2034	} else if (!strcmp(key, "type")) {
2035	err = og_json_parse_string(value, &params->type);
2036	params->flags \|= OG_REST_PARAM_TYPE;
2037	} else if (!strcmp(key, "profile")) {
2038	err = og_json_parse_string(value, &params->profile);
2039	params->flags \|= OG_REST_PARAM_PROFILE;
2040	} else if (!strcmp(key, "id")) {
2041	err = og_json_parse_string(value, &params->id);
2042	params->flags \|= OG_REST_PARAM_ID;
2043	}
2044
2045	if (err < 0)
2046	break;
2047	}
2048
2049	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR \|
2050	OG_REST_PARAM_DISK \|
2051	OG_REST_PARAM_PARTITION \|
2052	OG_REST_PARAM_NAME \|
2053	OG_REST_PARAM_REPO \|
2054	OG_REST_PARAM_TYPE \|
2055	OG_REST_PARAM_PROFILE \|
2056	OG_REST_PARAM_ID))
2057	return -1;
2058
2059	clients = json_copy(element);
2060	json_object_del(clients, "clients");
2061
2062	return og_send_request(OG_METHOD_POST, OG_CMD_IMAGE_RESTORE, params,
2063	clients);
2064	}
2065
2066	static int og_cmd_setup(json_t element, struct og_msg_params params)
2067	{
2068	json_t value, clients;
2069	const char *key;
2070	int err = 0;
2071
2072	if (json_typeof(element) != JSON_OBJECT)
2073	return -1;
2074
2075	json_object_foreach(element, key, value) {
2076	if (!strcmp(key, "clients")) {
2077	err = og_json_parse_clients(value, params);
2078	} else if (!strcmp(key, "disk")) {
2079	err = og_json_parse_string(value, &params->disk);
2080	params->flags \|= OG_REST_PARAM_DISK;
2081	} else if (!strcmp(key, "cache")) {
2082	err = og_json_parse_string(value, &params->cache);
2083	params->flags \|= OG_REST_PARAM_CACHE;
2084	} else if (!strcmp(key, "cache_size")) {
2085	err = og_json_parse_string(value, &params->cache_size);
2086	params->flags \|= OG_REST_PARAM_CACHE_SIZE;
2087	} else if (!strcmp(key, "partition_setup")) {
2088	err = og_json_parse_partition_setup(value, params);
2089	}
2090
2091	if (err < 0)
2092	break;
2093	}
2094
2095	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR \|
2096	OG_REST_PARAM_DISK \|
2097	OG_REST_PARAM_CACHE \|
2098	OG_REST_PARAM_CACHE_SIZE \|
2099	OG_REST_PARAM_PART_0 \|
2100	OG_REST_PARAM_PART_1 \|
2101	OG_REST_PARAM_PART_2 \|
2102	OG_REST_PARAM_PART_3))
2103	return -1;
2104
2105	clients = json_copy(element);
2106	json_object_del(clients, "clients");
2107
2108	return og_send_request(OG_METHOD_POST, OG_CMD_SETUP, params, clients);
2109	}
2110
2111	static int og_cmd_run_schedule(json_t element, struct og_msg_params params)
2112	{
2113	const char *key;
2114	json_t *value;
2115	int err = 0;
2116
2117	json_object_foreach(element, key, value) {
2118	if (!strcmp(key, "clients"))
2119	err = og_json_parse_clients(value, params);
2120
2121	if (err < 0)
2122	break;
2123	}
2124
2125	if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
2126	return -1;
2127
2128	return og_send_request(OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, params,
2129	NULL);
2130	}
2131
2132	static LIST_HEAD(cmd_list);
2133
2134	const struct og_cmd og_cmd_find(const char client_ip)
2135	{
2136	struct og_cmd cmd, next;
2137
2138	list_for_each_entry_safe(cmd, next, &cmd_list, list) {
2139	if (strcmp(cmd->ip, client_ip))
2140	continue;
2141
2142	list_del(&cmd->list);
2143	return cmd;
2144	}
2145
2146	return NULL;
2147	}
2148
2149	void og_cmd_free(const struct og_cmd *cmd)
2150	{
2151	struct og_msg_params params = (struct og_msg_params )&cmd->params;
2152	int i;
2153
2154	for (i = 0; i < params->ips_array_len; i++) {
2155	free((void *)params->ips_array[i]);
2156	free((void *)params->mac_array[i]);
2157	}
2158	free((void *)params->wol_type);
2159
2160	if (cmd->json)
2161	json_decref(cmd->json);
2162
2163	free((void *)cmd->ip);
2164	free((void *)cmd->mac);
2165	free((void *)cmd);
2166	}
2167
2168	static void og_cmd_init(struct og_cmd *cmd, enum og_rest_method method,
2169	enum og_cmd_type type, json_t *root)
2170	{
2171	cmd->type = type;
2172	cmd->method = method;
2173	cmd->params.ips_array[0] = strdup(cmd->ip);
2174	cmd->params.ips_array_len = 1;
2175	cmd->json = root;
2176	}
2177
2178	static int og_cmd_legacy_wol(const char input, struct og_cmd cmd)
2179	{
2180	char wol_type[2] = {};
2181
2182	if (sscanf(input, "mar=%s", wol_type) != 1) {
2183	syslog(LOG_ERR, "malformed database legacy input\n");
2184	return -1;
2185	}
2186
2187	og_cmd_init(cmd, OG_METHOD_NO_HTTP, OG_CMD_WOL, NULL);
2188	cmd->params.mac_array[0] = strdup(cmd->mac);
2189	cmd->params.wol_type = strdup(wol_type);
2190
2191	return 0;
2192	}
2193
2194	static int og_cmd_legacy_shell_run(const char input, struct og_cmd cmd)
2195	{
2196	json_t root, script, *echo;
2197
2198	script = json_string(input + 4);
2199	echo = json_boolean(false);
2200
2201	root = json_object();
2202	if (!root)
2203	return -1;
2204	json_object_set_new(root, "run", script);
2205	json_object_set_new(root, "echo", echo);
2206
2207	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_SHELL_RUN, root);
2208
2209	return 0;
2210	}
2211
2212	static int og_cmd_legacy_session(const char input, struct og_cmd cmd)
2213	{
2214	char part_str[OG_DB_SMALLINT_MAXLEN + 1];
2215	char disk_str[OG_DB_SMALLINT_MAXLEN + 1];
2216	json_t root, disk, *partition;
2217
2218	if (sscanf(input, "dsk=%s\rpar=%s\r", disk_str, part_str) != 2)
2219	return -1;
2220	partition = json_string(part_str);
2221	disk = json_string(disk_str);
2222
2223	root = json_object();
2224	if (!root)
2225	return -1;
2226	json_object_set_new(root, "partition", partition);
2227	json_object_set_new(root, "disk", disk);
2228
2229	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_SESSION, root);
2230
2231	return 0;
2232	}
2233
2234	static int og_cmd_legacy_poweroff(const char input, struct og_cmd cmd)
2235	{
2236	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_POWEROFF, NULL);
2237
2238	return 0;
2239	}
2240
2241	static int og_cmd_legacy_refresh(const char input, struct og_cmd cmd)
2242	{
2243	og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_REFRESH, NULL);
2244
2245	return 0;
2246	}
2247
2248	static int og_cmd_legacy_reboot(const char input, struct og_cmd cmd)
2249	{
2250	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_REBOOT, NULL);
2251
2252	return 0;
2253	}
2254
2255	static int og_cmd_legacy_stop(const char input, struct og_cmd cmd)
2256	{
2257	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_STOP, NULL);
2258
2259	return 0;
2260	}
2261
2262	static int og_cmd_legacy_hardware(const char input, struct og_cmd cmd)
2263	{
2264	og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_HARDWARE, NULL);
2265
2266	return 0;
2267	}
2268
2269	static int og_cmd_legacy_software(const char input, struct og_cmd cmd)
2270	{
2271	og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_SOFTWARE, NULL);
2272
2273	return 0;
2274	}
2275
2276	static int og_cmd_legacy_image_create(const char input, struct og_cmd cmd)
2277	{
2278	json_t root, disk, partition, code, image_id, name, *repo;
2279	struct og_image_legacy img = {};
2280
2281	if (sscanf(input, "dsk=%s\rpar=%s\rcpt=%s\ridi=%s\rnci=%s\ripr=%s\r",
2282	img.disk, img.part, img.code, img.image_id, img.name,
2283	img.repo) != 6)
2284	return -1;
2285	image_id = json_string(img.image_id);
2286	partition = json_string(img.part);
2287	code = json_string(img.code);
2288	name = json_string(img.name);
2289	repo = json_string(img.repo);
2290	disk = json_string(img.disk);
2291
2292	root = json_object();
2293	if (!root)
2294	return -1;
2295	json_object_set_new(root, "partition", partition);
2296	json_object_set_new(root, "repository", repo);
2297	json_object_set_new(root, "id", image_id);
2298	json_object_set_new(root, "code", code);
2299	json_object_set_new(root, "name", name);
2300	json_object_set_new(root, "disk", disk);
2301
2302	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_IMAGE_CREATE, root);
2303
2304	return 0;
2305	}
2306
2307	#define OG_DB_RESTORE_TYPE_MAXLEN 64
2308
2309	static int og_cmd_legacy_image_restore(const char input, struct og_cmd cmd)
2310	{
2311	json_t root, disk, partition, image_id, name, repo;
2312	char restore_type_str[OG_DB_RESTORE_TYPE_MAXLEN + 1] = {};
2313	char software_id_str[OG_DB_INT_MAXLEN + 1] = {};
2314	json_t software_id, restore_type;
2315	struct og_image_legacy img = {};
2316
2317	if (sscanf(input,
2318	"dsk=%s\rpar=%s\ridi=%s\rnci=%s\ripr=%s\rifs=%s\rptc=%s\r",
2319	img.disk, img.part, img.image_id, img.name, img.repo,
2320	software_id_str, restore_type_str) != 7)
2321	return -1;
2322
2323	restore_type = json_string(restore_type_str);
2324	software_id = json_string(software_id_str);
2325	image_id = json_string(img.image_id);
2326	partition = json_string(img.part);
2327	name = json_string(img.name);
2328	repo = json_string(img.repo);
2329	disk = json_string(img.disk);
2330
2331	root = json_object();
2332	if (!root)
2333	return -1;
2334	json_object_set_new(root, "profile", software_id);
2335	json_object_set_new(root, "partition", partition);
2336	json_object_set_new(root, "type", restore_type);
2337	json_object_set_new(root, "repository", repo);
2338	json_object_set_new(root, "id", image_id);
2339	json_object_set_new(root, "name", name);
2340	json_object_set_new(root, "disk", disk);
2341
2342	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_IMAGE_RESTORE, root);
2343
2344	return 0;
2345	}
2346
2347	static int og_cmd_legacy_setup(const char input, struct og_cmd cmd)
2348	{
2349	json_t root, disk, cache, cache_size, partition_setup, object;
2350	struct og_legacy_partition part_cfg[OG_PARTITION_MAX] = {};
2351	char cache_size_str [OG_DB_INT_MAXLEN + 1];
2352	char disk_str [OG_DB_SMALLINT_MAXLEN + 1];
2353	json_t part, code, fs, size, *format;
2354	unsigned int partition_len = 0;
2355	const char *in_ptr;
2356	char cache_str[2];
2357
2358	if (sscanf(input, "dsk=%s\rcfg=dis=%[^]che=%[^]*tch=%[^!]!",
2359	disk_str, cache_str, cache_size_str) != 3)
2360	return -1;
2361
2362	in_ptr = strstr(input, "!") + 1;
2363	while (strlen(in_ptr) > 0) {
2364	if(sscanf(in_ptr,
2365	"par=%[^]cpt=%[^]sfi=%[^]tam=%[^]ope=%[^%%]%%",
2366	part_cfg[partition_len].partition,
2367	part_cfg[partition_len].code,
2368	part_cfg[partition_len].filesystem,
2369	part_cfg[partition_len].size,
2370	part_cfg[partition_len].format) != 5)
2371	return -1;
2372	in_ptr = strstr(in_ptr, "%") + 1;
2373	partition_len++;
2374	}
2375
2376	root = json_object();
2377	if (!root)
2378	return -1;
2379
2380	cache_size = json_string(cache_size_str);
2381	cache = json_string(cache_str);
2382	partition_setup = json_array();
2383	disk = json_string(disk_str);
2384
2385	for (unsigned int i = 0; i < partition_len; ++i) {
2386	object = json_object();
2387	if (!object) {
2388	json_decref(root);
2389	return -1;
2390	}
2391
2392	part = json_string(part_cfg[i].partition);
2393	fs = json_string(part_cfg[i].filesystem);
2394	format = json_string(part_cfg[i].format);
2395	code = json_string(part_cfg[i].code);
2396	size = json_string(part_cfg[i].size);
2397
2398	json_object_set_new(object, "partition", part);
2399	json_object_set_new(object, "filesystem", fs);
2400	json_object_set_new(object, "format", format);
2401	json_object_set_new(object, "code", code);
2402	json_object_set_new(object, "size", size);
2403
2404	json_array_append_new(partition_setup, object);
2405	}
2406
2407	json_object_set_new(root, "partition_setup", partition_setup);
2408	json_object_set_new(root, "cache_size", cache_size);
2409	json_object_set_new(root, "cache", cache);
2410	json_object_set_new(root, "disk", disk);
2411
2412	og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_SETUP, root);
2413
2414	return 0;
2415	}
2416
2417	static int og_cmd_legacy_run_schedule(const char input, struct og_cmd cmd)
2418	{
2419	og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, NULL);
2420
2421	return 0;
2422	}
2423
2424	static int og_cmd_legacy(const char input, struct og_cmd cmd)
2425	{
2426	char legacy_cmd[32] = {};
2427	int err = -1;
2428
2429	if (sscanf(input, "nfn=%31s\r", legacy_cmd) != 1) {
2430	syslog(LOG_ERR, "malformed database legacy input\n");
2431	return -1;
2432	}
2433	input = strchr(input, '\r') + 1;
2434
2435	if (!strcmp(legacy_cmd, "Arrancar")) {
2436	err = og_cmd_legacy_wol(input, cmd);
2437	} else if (!strcmp(legacy_cmd, "EjecutarScript")) {
2438	err = og_cmd_legacy_shell_run(input, cmd);
2439	} else if (!strcmp(legacy_cmd, "IniciarSesion")) {
2440	err = og_cmd_legacy_session(input, cmd);
2441	} else if (!strcmp(legacy_cmd, "Apagar")) {
2442	err = og_cmd_legacy_poweroff(input, cmd);
2443	} else if (!strcmp(legacy_cmd, "Actualizar")) {
2444	err = og_cmd_legacy_refresh(input, cmd);
2445	} else if (!strcmp(legacy_cmd, "Reiniciar")) {
2446	err = og_cmd_legacy_reboot(input, cmd);
2447	} else if (!strcmp(legacy_cmd, "Purgar")) {
2448	err = og_cmd_legacy_stop(input, cmd);
2449	} else if (!strcmp(legacy_cmd, "InventarioHardware")) {
2450	err = og_cmd_legacy_hardware(input, cmd);
2451	} else if (!strcmp(legacy_cmd, "InventarioSoftware")) {
2452	err = og_cmd_legacy_software(input, cmd);
2453	} else if (!strcmp(legacy_cmd, "CrearImagen")) {
2454	err = og_cmd_legacy_image_create(input, cmd);
2455	} else if (!strcmp(legacy_cmd, "RestaurarImagen")) {
2456	err = og_cmd_legacy_image_restore(input, cmd);
2457	} else if (!strcmp(legacy_cmd, "Configurar")) {
2458	err = og_cmd_legacy_setup(input, cmd);
2459	} else if (!strcmp(legacy_cmd, "EjecutaComandosPendientes") \|\|
2460	!strcmp(legacy_cmd, "Actualizar")) {
2461	err = og_cmd_legacy_run_schedule(input, cmd);
2462	}
2463
2464	return err;
2465	}
2466
2467	static int og_dbi_add_action(const struct og_dbi dbi, const struct og_task task,
2468	struct og_cmd *cmd)
2469	{
2470	char start_date_string[24];
2471	struct tm *start_date;
2472	const char *msglog;
2473	dbi_result result;
2474	time_t now;
2475
2476	time(&now);
2477	start_date = localtime(&now);
2478
2479	sprintf(start_date_string, "%hu/%hhu/%hhu %hhu:%hhu:%hhu",
2480	start_date->tm_year + 1900, start_date->tm_mon + 1,
2481	start_date->tm_mday, start_date->tm_hour, start_date->tm_min,
2482	start_date->tm_sec);
2483	result = dbi_conn_queryf(dbi->conn,
2484	"INSERT INTO acciones (idordenador, "
2485	"tipoaccion, idtipoaccion, descriaccion, ip, "
2486	"sesion, idcomando, parametros, fechahorareg, "
2487	"estado, resultado, ambito, idambito, "
2488	"restrambito, idprocedimiento, idcentro, "
2489	"idprogramacion) "
2490	"VALUES (%d, %d, %d, '%s', '%s', %d, %d, '%s', "
2491	"'%s', %d, %d, %d, %d, '%s', %d, %d, %d)",
2492	cmd->client_id, EJECUCION_TAREA, task->task_id,
2493	"", cmd->ip, 0, task->command_id,
2494	task->params, start_date_string,
2495	ACCION_INICIADA, ACCION_SINRESULTADO,
2496	task->type_scope, task->scope, "",
2497	task->procedure_id, task->center_id,
2498	task->schedule_id);
2499	if (!result) {
2500	dbi_conn_error(dbi->conn, &msglog);
2501	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2502	__func__, __LINE__, msglog);
2503	return -1;
2504	}
2505	cmd->id = dbi_conn_sequence_last(dbi->conn, NULL);
2506	dbi_result_free(result);
2507
2508	return 0;
2509	}
2510
2511	static int og_queue_task_command(struct og_dbi dbi, const struct og_task task,
2512	char *query)
2513	{
2514	struct og_cmd *cmd;
2515	const char *msglog;
2516	dbi_result result;
2517
2518	result = dbi_conn_queryf(dbi->conn, query);
2519	if (!result) {
2520	dbi_conn_error(dbi->conn, &msglog);
2521	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2522	__func__, __LINE__, msglog);
2523	return -1;
2524	}
2525
2526	while (dbi_result_next_row(result)) {
2527	cmd = (struct og_cmd *)calloc(1, sizeof(struct og_cmd));
2528	if (!cmd) {
2529	dbi_result_free(result);
2530	return -1;
2531	}
2532
2533	cmd->client_id = dbi_result_get_uint(result, "idordenador");
2534	cmd->ip = strdup(dbi_result_get_string(result, "ip"));
2535	cmd->mac = strdup(dbi_result_get_string(result, "mac"));
2536
2537	og_cmd_legacy(task->params, cmd);
2538
2539	if (task->procedure_id) {
2540	if (og_dbi_add_action(dbi, task, cmd)) {
2541	dbi_result_free(result);
2542	return -1;
2543	}
2544	} else {
2545	cmd->id = task->task_id;
2546	}
2547
2548	list_add_tail(&cmd->list, &cmd_list);
2549	}
2550
2551	dbi_result_free(result);
2552
2553	return 0;
2554	}
2555
2556	static int og_queue_task_group_clients(struct og_dbi dbi, struct og_task task,
2557	char *query)
2558	{
2559
2560	const char *msglog;
2561	dbi_result result;
2562
2563	result = dbi_conn_queryf(dbi->conn, query);
2564	if (!result) {
2565	dbi_conn_error(dbi->conn, &msglog);
2566	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2567	__func__, __LINE__, msglog);
2568	return -1;
2569	}
2570
2571	while (dbi_result_next_row(result)) {
2572	uint32_t group_id = dbi_result_get_uint(result, "idgrupo");
2573
2574	sprintf(query, "SELECT idgrupo FROM gruposordenadores "
2575	"WHERE grupoid=%d", group_id);
2576	if (og_queue_task_group_clients(dbi, task, query)) {
2577	dbi_result_free(result);
2578	return -1;
2579	}
2580
2581	sprintf(query,"SELECT ip, mac, idordenador FROM ordenadores "
2582	"WHERE grupoid=%d", group_id);
2583	if (og_queue_task_command(dbi, task, query)) {
2584	dbi_result_free(result);
2585	return -1;
2586	}
2587
2588	}
2589
2590	dbi_result_free(result);
2591
2592	return 0;
2593	}
2594
2595	static int og_queue_task_group_classrooms(struct og_dbi *dbi,
2596	struct og_task task, char query)
2597	{
2598
2599	const char *msglog;
2600	dbi_result result;
2601
2602	result = dbi_conn_queryf(dbi->conn, query);
2603	if (!result) {
2604	dbi_conn_error(dbi->conn, &msglog);
2605	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2606	__func__, __LINE__, msglog);
2607	return -1;
2608	}
2609
2610	while (dbi_result_next_row(result)) {
2611	uint32_t group_id = dbi_result_get_uint(result, "idgrupo");
2612
2613	sprintf(query, "SELECT idgrupo FROM grupos "
2614	"WHERE grupoid=%d AND tipo=%d", group_id, AMBITO_GRUPOSAULAS);
2615	if (og_queue_task_group_classrooms(dbi, task, query)) {
2616	dbi_result_free(result);
2617	return -1;
2618	}
2619
2620	sprintf(query,
2621	"SELECT ip,mac,idordenador "
2622	"FROM ordenadores INNER JOIN aulas "
2623	"WHERE ordenadores.idaula=aulas.idaula "
2624	"AND aulas.grupoid=%d",
2625	group_id);
2626	if (og_queue_task_command(dbi, task, query)) {
2627	dbi_result_free(result);
2628	return -1;
2629	}
2630
2631	}
2632
2633	dbi_result_free(result);
2634
2635	return 0;
2636	}
2637
2638	static int og_queue_task_clients(struct og_dbi dbi, struct og_task task)
2639	{
2640	char query[4096];
2641
2642	switch (task->type_scope) {
2643	case AMBITO_CENTROS:
2644	sprintf(query,
2645	"SELECT ip,mac,idordenador "
2646	"FROM ordenadores INNER JOIN aulas "
2647	"WHERE ordenadores.idaula=aulas.idaula "
2648	"AND idcentro=%d",
2649	task->scope);
2650	return og_queue_task_command(dbi, task, query);
2651	case AMBITO_GRUPOSAULAS:
2652	sprintf(query,
2653	"SELECT idgrupo FROM grupos "
2654	"WHERE idgrupo=%i AND tipo=%d",
2655	task->scope, AMBITO_GRUPOSAULAS);
2656	return og_queue_task_group_classrooms(dbi, task, query);
2657	case AMBITO_AULAS:
2658	sprintf(query,
2659	"SELECT ip,mac,idordenador FROM ordenadores "
2660	"WHERE idaula=%d",
2661	task->scope);
2662	return og_queue_task_command(dbi, task, query);
2663	case AMBITO_GRUPOSORDENADORES:
2664	sprintf(query,
2665	"SELECT idgrupo FROM gruposordenadores "
2666	"WHERE idgrupo = %d",
2667	task->scope);
2668	return og_queue_task_group_clients(dbi, task, query);
2669	case AMBITO_ORDENADORES:
2670	sprintf(query,
2671	"SELECT ip, mac, idordenador FROM ordenadores "
2672	"WHERE idordenador = %d",
2673	task->scope);
2674	return og_queue_task_command(dbi, task, query);
2675	}
2676	return 0;
2677	}
2678
2679	int og_dbi_queue_procedure(struct og_dbi dbi, struct og_task task)
2680	{
2681	uint32_t procedure_id;
2682	const char *msglog;
2683	dbi_result result;
2684
2685	result = dbi_conn_queryf(dbi->conn,
2686	"SELECT parametros, procedimientoid, idcomando "
2687	"FROM procedimientos_acciones "
2688	"WHERE idprocedimiento=%d ORDER BY orden", task->procedure_id);
2689	if (!result) {
2690	dbi_conn_error(dbi->conn, &msglog);
2691	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2692	__func__, __LINE__, msglog);
2693	return -1;
2694	}
2695
2696	while (dbi_result_next_row(result)) {
2697	procedure_id = dbi_result_get_uint(result, "procedimientoid");
2698	if (procedure_id > 0) {
2699	task->procedure_id = procedure_id;
2700	if (og_dbi_queue_procedure(dbi, task))
2701	return -1;
2702	continue;
2703	}
2704
2705	task->params = strdup(dbi_result_get_string(result, "parametros"));
2706	task->command_id = dbi_result_get_uint(result, "idcomando");
2707	if (og_queue_task_clients(dbi, task))
2708	return -1;
2709	}
2710
2711	dbi_result_free(result);
2712
2713	return 0;
2714	}
2715
2716	static int og_dbi_queue_task(struct og_dbi *dbi, uint32_t task_id,
2717	uint32_t schedule_id)
2718	{
2719	struct og_task task = {};
2720	uint32_t task_id_next;
2721	const char *msglog;
2722	dbi_result result;
2723
2724	task.schedule_id = schedule_id;
2725
2726	result = dbi_conn_queryf(dbi->conn,
2727	"SELECT tareas_acciones.orden, "
2728	"tareas_acciones.idprocedimiento, "
2729	"tareas_acciones.tareaid, "
2730	"tareas.idtarea, "
2731	"tareas.idcentro, "
2732	"tareas.ambito, "
2733	"tareas.idambito, "
2734	"tareas.restrambito "
2735	" FROM tareas"
2736	" INNER JOIN tareas_acciones ON tareas_acciones.idtarea=tareas.idtarea"
2737	" WHERE tareas_acciones.idtarea=%u ORDER BY tareas_acciones.orden ASC", task_id);
2738	if (!result) {
2739	dbi_conn_error(dbi->conn, &msglog);
2740	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2741	__func__, __LINE__, msglog);
2742	return -1;
2743	}
2744
2745	while (dbi_result_next_row(result)) {
2746	task_id_next = dbi_result_get_uint(result, "tareaid");
2747
2748	if (task_id_next > 0) {
2749	if (og_dbi_queue_task(dbi, task_id_next, schedule_id))
2750	return -1;
2751
2752	continue;
2753	}
2754	task.task_id = dbi_result_get_uint(result, "idtarea");
2755	task.center_id = dbi_result_get_uint(result, "idcentro");
2756	task.procedure_id = dbi_result_get_uint(result, "idprocedimiento");
2757	task.type_scope = dbi_result_get_uint(result, "ambito");
2758	task.scope = dbi_result_get_uint(result, "idambito");
2759	task.filtered_scope = dbi_result_get_string(result, "restrambito");
2760
2761	og_dbi_queue_procedure(dbi, &task);
2762	}
2763
2764	dbi_result_free(result);
2765
2766	return 0;
2767	}
2768
2769	static int og_dbi_queue_command(struct og_dbi *dbi, uint32_t task_id,
2770	uint32_t schedule_id)
2771	{
2772	struct og_task task = {};
2773	const char *msglog;
2774	dbi_result result;
2775	char query[4096];
2776
2777	result = dbi_conn_queryf(dbi->conn,
2778	"SELECT idaccion, idcentro, idordenador, parametros "
2779	"FROM acciones "
2780	"WHERE sesion = %u", task_id);
2781	if (!result) {
2782	dbi_conn_error(dbi->conn, &msglog);
2783	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2784	__func__, __LINE__, msglog);
2785	return -1;
2786	}
2787
2788	while (dbi_result_next_row(result)) {
2789	task.task_id = dbi_result_get_uint(result, "idaccion");
2790	task.center_id = dbi_result_get_uint(result, "idcentro");
2791	task.scope = dbi_result_get_uint(result, "idordenador");
2792	task.params = strdup(dbi_result_get_string(result, "parametros"));
2793
2794	sprintf(query,
2795	"SELECT ip, mac, idordenador FROM ordenadores "
2796	"WHERE idordenador = %d",
2797	task.scope);
2798	if (og_queue_task_command(dbi, &task, query)) {
2799	dbi_result_free(result);
2800	return -1;
2801	}
2802	}
2803
2804	dbi_result_free(result);
2805
2806	return 0;
2807	}
2808
2809	int og_dbi_update_action(uint32_t id, bool success)
2810	{
2811	char end_date_string[24];
2812	struct tm *end_date;
2813	const char *msglog;
2814	struct og_dbi *dbi;
2815	uint8_t status = 2;
2816	dbi_result result;
2817	time_t now;
2818
2819	if (!id)
2820	return 0;
2821
2822	dbi = og_dbi_open(&ogconfig.db);
2823	if (!dbi) {
2824	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2825	__func__, __LINE__);
2826	return -1;
2827	}
2828
2829	time(&now);
2830	end_date = localtime(&now);
2831
2832	sprintf(end_date_string, "%hu/%hhu/%hhu %hhu:%hhu:%hhu",
2833	end_date->tm_year + 1900, end_date->tm_mon + 1,
2834	end_date->tm_mday, end_date->tm_hour, end_date->tm_min,
2835	end_date->tm_sec);
2836	result = dbi_conn_queryf(dbi->conn,
2837	"UPDATE acciones SET fechahorafin='%s', "
2838	"estado=%d, resultado=%d WHERE idaccion=%d",
2839	end_date_string, ACCION_FINALIZADA,
2840	status - success, id);
2841
2842	if (!result) {
2843	dbi_conn_error(dbi->conn, &msglog);
2844	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2845	__func__, __LINE__, msglog);
2846	og_dbi_close(dbi);
2847	return -1;
2848	}
2849	dbi_result_free(result);
2850	og_dbi_close(dbi);
2851
2852	return 0;
2853	}
2854
2855	void og_schedule_run(unsigned int task_id, unsigned int schedule_id,
2856	enum og_schedule_type type)
2857	{
2858	struct og_msg_params params = {};
2859	bool duplicated = false;
2860	struct og_cmd cmd, next;
2861	struct og_dbi *dbi;
2862	unsigned int i;
2863
2864	dbi = og_dbi_open(&ogconfig.db);
2865	if (!dbi) {
2866	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2867	__func__, __LINE__);
2868	return;
2869	}
2870
2871	switch (type) {
2872	case OG_SCHEDULE_TASK:
2873	og_dbi_queue_task(dbi, task_id, schedule_id);
2874	break;
2875	case OG_SCHEDULE_PROCEDURE:
2876	case OG_SCHEDULE_COMMAND:
2877	og_dbi_queue_command(dbi, task_id, schedule_id);
2878	break;
2879	}
2880	og_dbi_close(dbi);
2881
2882	list_for_each_entry(cmd, &cmd_list, list) {
2883	for (i = 0; i < params.ips_array_len; i++) {
2884	if (!strncmp(cmd->ip, params.ips_array[i],
2885	OG_DB_IP_MAXLEN)) {
2886	duplicated = true;
2887	break;
2888	}
2889	}
2890
2891	if (!duplicated)
2892	params.ips_array[params.ips_array_len++] = cmd->ip;
2893	else
2894	duplicated = false;
2895	}
2896
2897	list_for_each_entry_safe(cmd, next, &cmd_list, list) {
2898	if (cmd->type != OG_CMD_WOL)
2899	continue;
2900
2901	if (Levanta((char **)cmd->params.ips_array,
2902	(char **)cmd->params.mac_array,
2903	(char **)cmd->params.netmask_array,
2904	cmd->params.ips_array_len,
2905	(char *)cmd->params.wol_type))
2906	og_dbi_update_action(cmd->id, true);
2907
2908	list_del(&cmd->list);
2909	og_cmd_free(cmd);
2910	}
2911
2912	og_send_request(OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, &params, NULL);
2913	}
2914
2915	static int og_cmd_task_post(json_t element, struct og_msg_params params)
2916	{
2917	struct og_cmd *cmd;
2918	struct og_dbi *dbi;
2919	const char *key;
2920	json_t *value;
2921	int err;
2922
2923	if (json_typeof(element) != JSON_OBJECT)
2924	return -1;
2925
2926	json_object_foreach(element, key, value) {
2927	if (!strcmp(key, "task")) {
2928	err = og_json_parse_string(value, &params->task_id);
2929	params->flags \|= OG_REST_PARAM_TASK;
2930	}
2931
2932	if (err < 0)
2933	break;
2934	}
2935
2936	if (!og_msg_params_validate(params, OG_REST_PARAM_TASK))
2937	return -1;
2938
2939	dbi = og_dbi_open(&ogconfig.db);
2940	if (!dbi) {
2941	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2942	__func__, __LINE__);
2943	return -1;
2944	}
2945
2946	og_schedule_run(atoi(params->task_id), 0, OG_SCHEDULE_TASK);
2947	og_dbi_close(dbi);
2948
2949	list_for_each_entry(cmd, &cmd_list, list)
2950	params->ips_array[params->ips_array_len++] = cmd->ip;
2951
2952	return og_send_request(OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, params,
2953	NULL);
2954	}
2955
2956	static int og_dbi_scope_get_computer(struct og_dbi dbi, json_t array,
2957	uint32_t room_id)
2958	{
2959	const char computer_name, computer_ip;
2960	uint32_t computer_id;
2961	const char *msglog;
2962	dbi_result result;
2963	json_t *computer;
2964
2965	result = dbi_conn_queryf(dbi->conn,
2966	"SELECT idordenador, nombreordenador, ip "
2967	"FROM ordenadores WHERE idaula=%d",
2968	room_id);
2969	if (!result) {
2970	dbi_conn_error(dbi->conn, &msglog);
2971	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2972	__func__, __LINE__, msglog);
2973	return -1;
2974	}
2975
2976	while (dbi_result_next_row(result)) {
2977	computer_id = dbi_result_get_uint(result, "idordenador");
2978	computer_name = dbi_result_get_string(result, "nombreordenador");
2979	computer_ip = dbi_result_get_string(result, "ip");
2980
2981	computer = json_object();
2982	if (!computer) {
2983	dbi_result_free(result);
2984	return -1;
2985	}
2986
2987	json_object_set_new(computer, "name", json_string(computer_name));
2988	json_object_set_new(computer, "type", json_string("computer"));
2989	json_object_set_new(computer, "id", json_integer(computer_id));
2990	json_object_set_new(computer, "scope", json_array());
2991	json_object_set_new(computer, "ip", json_string(computer_ip));
2992	json_array_append(array, computer);
2993	json_decref(computer);
2994	}
2995	dbi_result_free(result);
2996
2997	return 0;
2998	}
2999
3000	static int og_dbi_scope_get_room(struct og_dbi dbi, json_t array,
3001	uint32_t center_id)
3002	{
3003	char room_name[OG_DB_ROOM_NAME_MAXLEN + 1] = {};
3004	json_t room, room_array;
3005	const char *msglog;
3006	dbi_result result;
3007	uint32_t room_id;
3008
3009	result = dbi_conn_queryf(dbi->conn,
3010	"SELECT idaula, nombreaula FROM aulas WHERE "
3011	"idcentro=%d",
3012	center_id);
3013	if (!result) {
3014	dbi_conn_error(dbi->conn, &msglog);
3015	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3016	__func__, __LINE__, msglog);
3017	return -1;
3018	}
3019
3020	while (dbi_result_next_row(result)) {
3021	room_id = dbi_result_get_uint(result, "idaula");
3022	strncpy(room_name,
3023	dbi_result_get_string(result, "nombreaula"),
3024	OG_DB_CENTER_NAME_MAXLEN);
3025
3026	room = json_object();
3027	if (!room) {
3028	dbi_result_free(result);
3029	return -1;
3030	}
3031
3032	json_object_set_new(room, "name", json_string(room_name));
3033	json_object_set_new(room, "type", json_string("room"));
3034	json_object_set_new(room, "id", json_integer(room_id));
3035	json_object_set_new(room, "scope", json_array());
3036	json_array_append(array, room);
3037	json_decref(room);
3038
3039	room_array = json_object_get(room, "scope");
3040	if (!room_array) {
3041	dbi_result_free(result);
3042	return -1;
3043	}
3044
3045	if (og_dbi_scope_get_computer(dbi, room_array, room_id)) {
3046	dbi_result_free(result);
3047	return -1;
3048	}
3049	}
3050	dbi_result_free(result);
3051
3052	return 0;
3053	}
3054
3055	static int og_dbi_scope_get(struct og_dbi dbi, json_t array)
3056	{
3057	char center_name[OG_DB_CENTER_NAME_MAXLEN + 1] = {};
3058	json_t center, array_room;
3059	const char *msglog;
3060	uint32_t center_id;
3061	dbi_result result;
3062
3063	result = dbi_conn_queryf(dbi->conn,
3064	"SELECT nombrecentro, idcentro FROM centros");
3065	if (!result) {
3066	dbi_conn_error(dbi->conn, &msglog);
3067	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3068	__func__, __LINE__, msglog);
3069	return -1;
3070	}
3071
3072	while (dbi_result_next_row(result)) {
3073	center_id = dbi_result_get_uint(result, "idcentro");
3074	strncpy(center_name,
3075	dbi_result_get_string(result, "nombrecentro"),
3076	OG_DB_CENTER_NAME_MAXLEN);
3077
3078	center = json_object();
3079	if (!center) {
3080	dbi_result_free(result);
3081	return -1;
3082	}
3083
3084	array_room = json_array();
3085	if (!array_room) {
3086	dbi_result_free(result);
3087	json_decref(center);
3088	return -1;
3089	}
3090
3091	json_object_set_new(center, "name", json_string(center_name));
3092	json_object_set_new(center, "type", json_string("center"));
3093	json_object_set_new(center, "id", json_integer(center_id));
3094	json_object_set_new(center, "scope", array_room);
3095	json_array_append(array, center);
3096	json_decref(center);
3097
3098	if (og_dbi_scope_get_room(dbi, array_room, center_id)) {
3099	dbi_result_free(result);
3100	return -1;
3101	}
3102	}
3103
3104	dbi_result_free(result);
3105
3106	return 0;
3107	}
3108
3109	static int og_cmd_scope_get(json_t element, struct og_msg_params params,
3110	char *buffer_reply)
3111	{
3112	struct og_buffer og_buffer = {
3113	.data = buffer_reply
3114	};
3115	json_t root, array;
3116	struct og_dbi *dbi;
3117
3118	root = json_object();
3119	if (!root)
3120	return -1;
3121
3122	array = json_array();
3123	if (!array) {
3124	json_decref(root);
3125	return -1;
3126	}
3127	json_object_set_new(root, "scope", array);
3128
3129	dbi = og_dbi_open(&ogconfig.db);
3130	if (!dbi) {
3131	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3132	__func__, __LINE__);
3133	json_decref(root);
3134	return -1;
3135	}
3136
3137	if (og_dbi_scope_get(dbi, array)) {
3138	og_dbi_close(dbi);
3139	json_decref(root);
3140	return -1;
3141	}
3142
3143	og_dbi_close(dbi);
3144
3145	json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
3146	json_decref(root);
3147
3148	return 0;
3149	}
3150
3151	int og_dbi_schedule_get(void)
3152	{
3153	uint32_t schedule_id, task_id;
3154	struct og_schedule_time time;
3155	struct og_dbi *dbi;
3156	const char *msglog;
3157	dbi_result result;
3158
3159	dbi = og_dbi_open(&ogconfig.db);
3160	if (!dbi) {
3161	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3162	__func__, __LINE__);
3163	return -1;
3164	}
3165
3166	result = dbi_conn_queryf(dbi->conn,
3167	"SELECT idprogramacion, tipoaccion, identificador, "
3168	"sesion, annos, meses, diario, dias, semanas, horas, "
3169	"ampm, minutos FROM programaciones "
3170	"WHERE suspendida = 0");
3171	if (!result) {
3172	dbi_conn_error(dbi->conn, &msglog);
3173	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3174	__func__, __LINE__, msglog);
3175	og_dbi_close(dbi);
3176	return -1;
3177	}
3178
3179	while (dbi_result_next_row(result)) {
3180	memset(&time, 0, sizeof(time));
3181	schedule_id = dbi_result_get_uint(result, "idprogramacion");
3182	task_id = dbi_result_get_uint(result, "identificador");
3183	time.years = dbi_result_get_uint(result, "annos");
3184	time.months = dbi_result_get_uint(result, "meses");
3185	time.weeks = dbi_result_get_uint(result, "semanas");
3186	time.week_days = dbi_result_get_uint(result, "dias");
3187	time.days = dbi_result_get_uint(result, "diario");
3188	time.hours = dbi_result_get_uint(result, "horas");
3189	time.am_pm = dbi_result_get_uint(result, "ampm");
3190	time.minutes = dbi_result_get_uint(result, "minutos");
3191	time.on_start = true;
3192
3193	og_schedule_create(schedule_id, task_id, OG_SCHEDULE_TASK,
3194	&time);
3195	}
3196
3197	dbi_result_free(result);
3198	og_dbi_close(dbi);
3199
3200	return 0;
3201	}
3202
3203	static int og_dbi_schedule_create(struct og_dbi *dbi,
3204	struct og_msg_params *params,
3205	uint32_t *schedule_id,
3206	enum og_schedule_type schedule_type)
3207	{
3208	uint8_t suspended = 0;
3209	uint32_t session = 0;
3210	const char *msglog;
3211	dbi_result result;
3212	uint8_t type;
3213
3214	switch (schedule_type) {
3215	case OG_SCHEDULE_TASK:
3216	type = 3;
3217	break;
3218	case OG_SCHEDULE_PROCEDURE:
3219	type = 2;
3220	break;
3221	case OG_SCHEDULE_COMMAND:
3222	session = atoi(params->task_id);
3223	type = 1;
3224	break;
3225	}
3226
3227	result = dbi_conn_queryf(dbi->conn,
3228	"INSERT INTO programaciones (tipoaccion,"
3229	" identificador, nombrebloque, annos, meses,"
3230	" semanas, dias, diario, horas, ampm, minutos,"
3231	" suspendida, sesion) VALUES (%d, %s, '%s',"
3232	" %d, %d, %d, %d, %d, %d, %d, %d, %d, %d)",
3233	type, params->task_id, params->name,
3234	params->time.years, params->time.months,
3235	params->time.weeks, params->time.week_days,
3236	params->time.days, params->time.hours,
3237	params->time.am_pm, params->time.minutes,
3238	suspended, session);
3239	if (!result) {
3240	dbi_conn_error(dbi->conn, &msglog);
3241	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3242	__func__, __LINE__, msglog);
3243	return -1;
3244	}
3245	dbi_result_free(result);
3246
3247	*schedule_id = dbi_conn_sequence_last(dbi->conn, NULL);
3248
3249	return 0;
3250	}
3251
3252	static int og_dbi_schedule_update(struct og_dbi *dbi,
3253	struct og_msg_params *params)
3254	{
3255	const char *msglog;
3256	dbi_result result;
3257	uint8_t type = 3;
3258
3259	result = dbi_conn_queryf(dbi->conn,
3260	"UPDATE programaciones SET tipoaccion=%d, "
3261	"identificador='%s', nombrebloque='%s', "
3262	"annos=%d, meses=%d, "
3263	"diario=%d, horas=%d, ampm=%d, minutos=%d "
3264	"WHERE idprogramacion='%s'",
3265	type, params->task_id, params->name,
3266	params->time.years, params->time.months,
3267	params->time.days, params->time.hours,
3268	params->time.am_pm, params->time.minutes,
3269	params->id);
3270
3271	if (!result) {
3272	dbi_conn_error(dbi->conn, &msglog);
3273	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3274	__func__, __LINE__, msglog);
3275	return -1;
3276	}
3277	dbi_result_free(result);
3278
3279	return 0;
3280	}
3281
3282	static int og_dbi_schedule_delete(struct og_dbi *dbi, uint32_t id)
3283	{
3284	const char *msglog;
3285	dbi_result result;
3286
3287	result = dbi_conn_queryf(dbi->conn,
3288	"DELETE FROM programaciones WHERE idprogramacion=%d",
3289	id);
3290	if (!result) {
3291	dbi_conn_error(dbi->conn, &msglog);
3292	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3293	__func__, __LINE__, msglog);
3294	return -1;
3295	}
3296	dbi_result_free(result);
3297
3298	return 0;
3299	}
3300
3301	struct og_db_schedule {
3302	uint32_t id;
3303	uint32_t task_id;
3304	const char *name;
3305	struct og_schedule_time time;
3306	uint32_t week_days;
3307	uint32_t weeks;
3308	uint32_t suspended;
3309	uint32_t session;
3310	};
3311
3312	static int og_dbi_schedule_get_json(struct og_dbi dbi, json_t root,
3313	const char task_id, const char schedule_id)
3314	{
3315	struct og_db_schedule schedule;
3316	json_t obj, array;
3317	const char *msglog;
3318	dbi_result result;
3319	int err = 0;
3320
3321	if (task_id) {
3322	result = dbi_conn_queryf(dbi->conn,
3323	"SELECT idprogramacion,"
3324	" identificador, nombrebloque,"
3325	" annos, meses, diario, dias,"
3326	" semanas, horas, ampm,"
3327	" minutos,suspendida, sesion "
3328	"FROM programaciones "
3329	"WHERE identificador=%d",
3330	atoi(task_id));
3331	} else if (schedule_id) {
3332	result = dbi_conn_queryf(dbi->conn,
3333	"SELECT idprogramacion,"
3334	" identificador, nombrebloque,"
3335	" annos, meses, diario, dias,"
3336	" semanas, horas, ampm,"
3337	" minutos,suspendida, sesion "
3338	"FROM programaciones "
3339	"WHERE idprogramacion=%d",
3340	atoi(schedule_id));
3341	} else {
3342	result = dbi_conn_queryf(dbi->conn,
3343	"SELECT idprogramacion,"
3344	" identificador, nombrebloque,"
3345	" annos, meses, diario, dias,"
3346	" semanas, horas, ampm,"
3347	" minutos,suspendida, sesion "
3348	"FROM programaciones");
3349	}
3350
3351	if (!result) {
3352	dbi_conn_error(dbi->conn, &msglog);
3353	syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3354	__func__, __LINE__, msglog);
3355	return -1;
3356	}
3357
3358	array = json_array();
3359	if (!array)
3360	return -1;
3361
3362	while (dbi_result_next_row(result)) {
3363	schedule.id = dbi_result_get_uint(result, "idprogramacion");
3364	schedule.task_id = dbi_result_get_uint(result, "identificador");
3365	schedule.name = dbi_result_get_string(result, "nombrebloque");
3366	schedule.time.years = dbi_result_get_uint(result, "annos");
3367	schedule.time.months = dbi_result_get_uint(result, "meses");
3368	schedule.time.days = dbi_result_get_uint(result, "diario");
3369	schedule.time.hours = dbi_result_get_uint(result, "horas");
3370	schedule.time.am_pm = dbi_result_get_uint(result, "ampm");
3371	schedule.time.minutes = dbi_result_get_uint(result, "minutos");
3372	schedule.week_days = dbi_result_get_uint(result, "dias");
3373	schedule.weeks = dbi_result_get_uint(result, "semanas");
3374	schedule.suspended = dbi_result_get_uint(result, "suspendida");
3375	schedule.session = dbi_result_get_uint(result, "sesion");
3376
3377	obj = json_object();
3378	if (!obj) {
3379	err = -1;
3380	break;
3381	}
3382	json_object_set_new(obj, "id", json_integer(schedule.id));
3383	json_object_set_new(obj, "task", json_integer(schedule.task_id));
3384	json_object_set_new(obj, "name", json_string(schedule.name));
3385	json_object_set_new(obj, "years", json_integer(schedule.time.years));
3386	json_object_set_new(obj, "months", json_integer(schedule.time.months));
3387	json_object_set_new(obj, "days", json_integer(schedule.time.days));
3388	json_object_set_new(obj, "hours", json_integer(schedule.time.hours));
3389	json_object_set_new(obj, "am_pm", json_integer(schedule.time.am_pm));
3390	json_object_set_new(obj, "minutes", json_integer(schedule.time.minutes));
3391	json_object_set_new(obj, "week_days", json_integer(schedule.week_days));
3392	json_object_set_new(obj, "weeks", json_integer(schedule.weeks));
3393	json_object_set_new(obj, "suspended", json_integer(schedule.suspended));
3394	json_object_set_new(obj, "session", json_integer(schedule.session));
3395
3396	json_array_append_new(array, obj);
3397	}
3398
3399	json_object_set_new(root, "schedule", array);
3400
3401	dbi_result_free(result);
3402
3403	return err;
3404	}
3405
3406	static int og_task_schedule_create(struct og_msg_params *params)
3407	{
3408	enum og_schedule_type type;
3409	uint32_t schedule_id;
3410	struct og_dbi *dbi;
3411	int err;
3412
3413	if (!strcmp(params->type, "task"))
3414	type = OG_SCHEDULE_TASK;
3415	else if (!strcmp(params->type, "procedure"))
3416	type = OG_SCHEDULE_PROCEDURE;
3417	else if (!strcmp(params->type, "command"))
3418	type = OG_SCHEDULE_COMMAND;
3419	else
3420	return -1;
3421
3422	dbi = og_dbi_open(&ogconfig.db);
3423	if (!dbi) {
3424	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3425	__func__, __LINE__);
3426	return -1;
3427	}
3428
3429	err = og_dbi_schedule_create(dbi, params, &schedule_id, type);
3430	if (err < 0) {
3431	og_dbi_close(dbi);
3432	return -1;
3433	}
3434	og_schedule_create(schedule_id, atoi(params->task_id), type,
3435	&params->time);
3436	og_schedule_refresh(og_loop);
3437	og_dbi_close(dbi);
3438
3439	return 0;
3440	}
3441
3442	static int og_cmd_schedule_create(json_t element, struct og_msg_params params)
3443	{
3444	const char *key;
3445	json_t *value;
3446	int err;
3447
3448	if (json_typeof(element) != JSON_OBJECT)
3449	return -1;
3450
3451	json_object_foreach(element, key, value) {
3452	if (!strcmp(key, "task")) {
3453	err = og_json_parse_string(value, &params->task_id);
3454	params->flags \|= OG_REST_PARAM_TASK;
3455	} else if (!strcmp(key, "name")) {
3456	err = og_json_parse_string(value, &params->name);
3457	params->flags \|= OG_REST_PARAM_NAME;
3458	} else if (!strcmp(key, "when")) {
3459	err = og_json_parse_time_params(value, params);
3460	} else if (!strcmp(key, "type")) {
3461	err = og_json_parse_string(value, &params->type);
3462	params->flags \|= OG_REST_PARAM_TYPE;
3463	}
3464
3465	if (err < 0)
3466	break;
3467	}
3468
3469	if (!og_msg_params_validate(params, OG_REST_PARAM_TASK \|
3470	OG_REST_PARAM_NAME \|
3471	OG_REST_PARAM_TIME_YEARS \|
3472	OG_REST_PARAM_TIME_MONTHS \|
3473	OG_REST_PARAM_TIME_WEEKS \|
3474	OG_REST_PARAM_TIME_WEEK_DAYS \|
3475	OG_REST_PARAM_TIME_DAYS \|
3476	OG_REST_PARAM_TIME_HOURS \|
3477	OG_REST_PARAM_TIME_MINUTES \|
3478	OG_REST_PARAM_TIME_AM_PM \|
3479	OG_REST_PARAM_TYPE))
3480	return -1;
3481
3482	return og_task_schedule_create(params);
3483	}
3484
3485	static int og_cmd_schedule_update(json_t element, struct og_msg_params params)
3486	{
3487	struct og_dbi *dbi;
3488	const char *key;
3489	json_t *value;
3490	int err;
3491
3492	if (json_typeof(element) != JSON_OBJECT)
3493	return -1;
3494
3495	json_object_foreach(element, key, value) {
3496	if (!strcmp(key, "id")) {
3497	err = og_json_parse_string(value, &params->id);
3498	params->flags \|= OG_REST_PARAM_ID;
3499	} else if (!strcmp(key, "task")) {
3500	err = og_json_parse_string(value, &params->task_id);
3501	params->flags \|= OG_REST_PARAM_TASK;
3502	} else if (!strcmp(key, "name")) {
3503	err = og_json_parse_string(value, &params->name);
3504	params->flags \|= OG_REST_PARAM_NAME;
3505	} else if (!strcmp(key, "when"))
3506	err = og_json_parse_time_params(value, params);
3507
3508	if (err < 0)
3509	break;
3510	}
3511
3512	if (!og_msg_params_validate(params, OG_REST_PARAM_ID \|
3513	OG_REST_PARAM_TASK \|
3514	OG_REST_PARAM_NAME \|
3515	OG_REST_PARAM_TIME_YEARS \|
3516	OG_REST_PARAM_TIME_MONTHS \|
3517	OG_REST_PARAM_TIME_DAYS \|
3518	OG_REST_PARAM_TIME_HOURS \|
3519	OG_REST_PARAM_TIME_MINUTES \|
3520	OG_REST_PARAM_TIME_AM_PM))
3521	return -1;
3522
3523	dbi = og_dbi_open(&ogconfig.db);
3524	if (!dbi) {
3525	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3526	__func__, __LINE__);
3527	return -1;
3528	}
3529
3530	err = og_dbi_schedule_update(dbi, params);
3531	og_dbi_close(dbi);
3532
3533	if (err < 0)
3534	return err;
3535
3536	og_schedule_update(og_loop, atoi(params->id), atoi(params->task_id),
3537	&params->time);
3538	og_schedule_refresh(og_loop);
3539
3540	return err;
3541	}
3542
3543	static int og_cmd_schedule_delete(json_t element, struct og_msg_params params)
3544	{
3545	struct og_dbi *dbi;
3546	const char *key;
3547	json_t *value;
3548	int err;
3549
3550	if (json_typeof(element) != JSON_OBJECT)
3551	return -1;
3552
3553	json_object_foreach(element, key, value) {
3554	if (!strcmp(key, "id")) {
3555	err = og_json_parse_string(value, &params->id);
3556	params->flags \|= OG_REST_PARAM_ID;
3557	} else {
3558	return -1;
3559	}
3560
3561	if (err < 0)
3562	break;
3563	}
3564
3565	if (!og_msg_params_validate(params, OG_REST_PARAM_ID))
3566	return -1;
3567
3568	dbi = og_dbi_open(&ogconfig.db);
3569	if (!dbi) {
3570	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3571	__func__, __LINE__);
3572	return -1;
3573	}
3574
3575	err = og_dbi_schedule_delete(dbi, atoi(params->id));
3576	og_dbi_close(dbi);
3577
3578	og_schedule_delete(og_loop, atoi(params->id));
3579
3580	return err;
3581	}
3582
3583	static int og_cmd_schedule_get(json_t element, struct og_msg_params params,
3584	char *buffer_reply)
3585	{
3586	struct og_buffer og_buffer = {
3587	.data = buffer_reply,
3588	};
3589	json_t *schedule_root;
3590	struct og_dbi *dbi;
3591	const char *key;
3592	json_t *value;
3593	int err;
3594
3595	if (element) {
3596	if (json_typeof(element) != JSON_OBJECT)
3597	return -1;
3598
3599	json_object_foreach(element, key, value) {
3600	if (!strcmp(key, "task")) {
3601	err = og_json_parse_string(value,
3602	&params->task_id);
3603	} else if (!strcmp(key, "id")) {
3604	err = og_json_parse_string(value, &params->id);
3605	} else {
3606	return -1;
3607	}
3608
3609	if (err < 0)
3610	break;
3611	}
3612	}
3613
3614	dbi = og_dbi_open(&ogconfig.db);
3615	if (!dbi) {
3616	syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3617	__func__, __LINE__);
3618	return -1;
3619	}
3620
3621	schedule_root = json_object();
3622	if (!schedule_root) {
3623	og_dbi_close(dbi);
3624	return -1;
3625	}
3626
3627	err = og_dbi_schedule_get_json(dbi, schedule_root,
3628	params->task_id, params->id);
3629	og_dbi_close(dbi);
3630
3631	if (err >= 0)
3632	json_dump_callback(schedule_root, og_json_dump_clients, &og_buffer, 0);
3633
3634	json_decref(schedule_root);
3635
3636	return err;
3637	}
3638
3639	static int og_client_method_not_found(struct og_client *cli)
3640	{
3641	/* To meet RFC 7231, this function MUST generate an Allow header field
3642	* containing the correct methods. For example: "Allow: POST\r\n"
3643	*/
3644	char buf[] = "HTTP/1.1 405 Method Not Allowed\r\n"
3645	"Content-Length: 0\r\n\r\n";
3646
3647	send(og_client_socket(cli), buf, strlen(buf), 0);
3648
3649	return -1;
3650	}
3651
3652	static int og_client_bad_request(struct og_client *cli)
3653	{
3654	char buf[] = "HTTP/1.1 400 Bad Request\r\nContent-Length: 0\r\n\r\n";
3655
3656	send(og_client_socket(cli), buf, strlen(buf), 0);
3657
3658	return -1;
3659	}
3660
3661	static int og_client_not_found(struct og_client *cli)
3662	{
3663	char buf[] = "HTTP/1.1 404 Not Found\r\nContent-Length: 0\r\n\r\n";
3664
3665	send(og_client_socket(cli), buf, strlen(buf), 0);
3666
3667	return -1;
3668	}
3669
3670	static int og_client_not_authorized(struct og_client *cli)
3671	{
3672	char buf[] = "HTTP/1.1 401 Unauthorized\r\n"
3673	"WWW-Authenticate: Basic\r\n"
3674	"Content-Length: 0\r\n\r\n";
3675
3676	send(og_client_socket(cli), buf, strlen(buf), 0);
3677
3678	return -1;
3679	}
3680
3681	static int og_server_internal_error(struct og_client *cli)
3682	{
3683	char buf[] = "HTTP/1.1 500 Internal Server Error\r\n"
3684	"Content-Length: 0\r\n\r\n";
3685
3686	send(og_client_socket(cli), buf, strlen(buf), 0);
3687
3688	return -1;
3689	}
3690
3691	#define OG_MSG_RESPONSE_MAXLEN 65536
3692
3693	static int og_client_ok(struct og_client cli, char buf_reply)
3694	{
3695	char buf[OG_MSG_RESPONSE_MAXLEN] = {};
3696	int err = 0, len;
3697
3698	len = snprintf(buf, sizeof(buf),
3699	"HTTP/1.1 200 OK\r\nContent-Length: %ld\r\n\r\n%s",
3700	strlen(buf_reply), buf_reply);
3701	if (len >= (int)sizeof(buf))
3702	err = og_server_internal_error(cli);
3703
3704	send(og_client_socket(cli), buf, strlen(buf), 0);
3705
3706	return err;
3707	}
3708
3709	int og_client_state_process_payload_rest(struct og_client *cli)
3710	{
3711	char buf_reply[OG_MSG_RESPONSE_MAXLEN] = {};
3712	struct og_msg_params params = {};
3713	enum og_rest_method method;
3714	const char cmd, body;
3715	json_error_t json_err;
3716	json_t *root = NULL;
3717	int err = 0;
3718
3719	syslog(LOG_DEBUG, "%s:%hu %.32s ...\n",
3720	inet_ntoa(cli->addr.sin_addr),
3721	ntohs(cli->addr.sin_port), cli->buf);
3722
3723	if (!strncmp(cli->buf, "GET", strlen("GET"))) {
3724	method = OG_METHOD_GET;
3725	cmd = cli->buf + strlen("GET") + 2;
3726	} else if (!strncmp(cli->buf, "POST", strlen("POST"))) {
3727	method = OG_METHOD_POST;
3728	cmd = cli->buf + strlen("POST") + 2;
3729	} else
3730	return og_client_method_not_found(cli);
3731
3732	body = strstr(cli->buf, "\r\n\r\n") + 4;
3733
3734	if (strcmp(cli->auth_token, ogconfig.rest.api_token)) {
3735	syslog(LOG_ERR, "wrong Authentication key\n");
3736	return og_client_not_authorized(cli);
3737	}
3738
3739	if (cli->content_length) {
3740	root = json_loads(body, 0, &json_err);
3741	if (!root) {
3742	syslog(LOG_ERR, "malformed json line %d: %s\n",
3743	json_err.line, json_err.text);
3744	return og_client_not_found(cli);
3745	}
3746	}
3747
3748	if (!strncmp(cmd, "clients", strlen("clients"))) {
3749	if (method != OG_METHOD_POST &&
3750	method != OG_METHOD_GET)
3751	return og_client_method_not_found(cli);
3752
3753	if (method == OG_METHOD_POST && !root) {
3754	syslog(LOG_ERR, "command clients with no payload\n");
3755	return og_client_bad_request(cli);
3756	}
3757	switch (method) {
3758	case OG_METHOD_POST:
3759	err = og_cmd_post_clients(root, &params);
3760	break;
3761	case OG_METHOD_GET:
3762	err = og_cmd_get_clients(root, &params, buf_reply);
3763	break;
3764	default:
3765	return og_client_bad_request(cli);
3766	}
3767	} else if (!strncmp(cmd, "client/setup",
3768	strlen("client/setup"))) {
3769	if (method != OG_METHOD_GET)
3770	return og_client_method_not_found(cli);
3771
3772	if (!root) {
3773	syslog(LOG_ERR,
3774	"command client partitions with no payload\n");
3775	return og_client_bad_request(cli);
3776	}
3777
3778	err = og_cmd_get_client_setup(root, &params, buf_reply);
3779	} else if (!strncmp(cmd, "client/info",
3780	strlen("client/info"))) {
3781	if (method != OG_METHOD_GET)
3782	return og_client_method_not_found(cli);
3783
3784	if (!root) {
3785	syslog(LOG_ERR,
3786	"command client info with no payload\n");
3787	return og_client_bad_request(cli);
3788	}
3789
3790	err = og_cmd_get_client_info(root, &params, buf_reply);
3791	} else if (!strncmp(cmd, "client/add", strlen("client/add"))) {
3792	if (method != OG_METHOD_POST)
3793	return og_client_method_not_found(cli);
3794
3795	if (!root) {
3796	syslog(LOG_ERR,
3797	"command client info with no payload\n");
3798	return og_client_bad_request(cli);
3799	}
3800
3801	err = og_cmd_post_client_add(root, &params, buf_reply);
3802	} else if (!strncmp(cmd, "client/delete", strlen("client/delete"))) {
3803	if (method != OG_METHOD_POST)
3804	return og_client_method_not_found(cli);
3805
3806	if (!root) {
3807	syslog(LOG_ERR,
3808	"command client delete with no payload\n");
3809	return og_client_bad_request(cli);
3810	}
3811
3812	err = og_cmd_post_client_delete(root, &params);
3813	} else if (!strncmp(cmd, "wol", strlen("wol"))) {
3814	if (method != OG_METHOD_POST)
3815	return og_client_method_not_found(cli);
3816
3817	if (!root) {
3818	syslog(LOG_ERR, "command wol with no payload\n");
3819	return og_client_bad_request(cli);
3820	}
3821	err = og_cmd_wol(root, &params);
3822	} else if (!strncmp(cmd, "shell/run", strlen("shell/run"))) {
3823	if (method != OG_METHOD_POST)
3824	return og_client_method_not_found(cli);
3825
3826	if (!root) {
3827	syslog(LOG_ERR, "command run with no payload\n");
3828	return og_client_bad_request(cli);
3829	}
3830	err = og_cmd_run_post(root, &params);
3831	} else if (!strncmp(cmd, "shell/output", strlen("shell/output"))) {
3832	if (method != OG_METHOD_POST)
3833	return og_client_method_not_found(cli);
3834
3835	if (!root) {
3836	syslog(LOG_ERR, "command output with no payload\n");
3837	return og_client_bad_request(cli);
3838	}
3839
3840	err = og_cmd_run_get(root, &params, buf_reply);
3841	} else if (!strncmp(cmd, "session", strlen("session"))) {
3842	if (method != OG_METHOD_POST && method != OG_METHOD_GET)
3843	return og_client_method_not_found(cli);
3844
3845	if (!root) {
3846	syslog(LOG_ERR, "command session with no payload\n");
3847	return og_client_bad_request(cli);
3848	}
3849
3850	if (method == OG_METHOD_POST)
3851	err = og_cmd_session(root, &params);
3852	else
3853	err = og_cmd_get_session(root, &params, buf_reply);
3854	} else if (!strncmp(cmd, "scopes", strlen("scopes"))) {
3855	if (method != OG_METHOD_GET)
3856	return og_client_method_not_found(cli);
3857
3858	err = og_cmd_scope_get(root, &params, buf_reply);
3859	} else if (!strncmp(cmd, "poweroff", strlen("poweroff"))) {
3860	if (method != OG_METHOD_POST)
3861	return og_client_method_not_found(cli);
3862
3863	if (!root) {
3864	syslog(LOG_ERR, "command poweroff with no payload\n");
3865	return og_client_bad_request(cli);
3866	}
3867	err = og_cmd_poweroff(root, &params);
3868	} else if (!strncmp(cmd, "reboot", strlen("reboot"))) {
3869	if (method != OG_METHOD_POST)
3870	return og_client_method_not_found(cli);
3871
3872	if (!root) {
3873	syslog(LOG_ERR, "command reboot with no payload\n");
3874	return og_client_bad_request(cli);
3875	}
3876	err = og_cmd_reboot(root, &params);
3877	} else if (!strncmp(cmd, "mode", strlen("mode"))) {
3878	if (method != OG_METHOD_GET && method != OG_METHOD_POST)
3879	return og_client_method_not_found(cli);
3880
3881	if (method == OG_METHOD_POST && !root) {
3882	syslog(LOG_ERR, "command mode with no payload\n");
3883	return og_client_bad_request(cli);
3884	}
3885
3886	if (method == OG_METHOD_GET)
3887	err = og_cmd_get_modes(root, &params, buf_reply);
3888	else if (method == OG_METHOD_POST)
3889	err = og_cmd_post_modes(root, &params);
3890	} else if (!strncmp(cmd, "stop", strlen("stop"))) {
3891	if (method != OG_METHOD_POST)
3892	return og_client_method_not_found(cli);
3893
3894	if (!root) {
3895	syslog(LOG_ERR, "command stop with no payload\n");
3896	return og_client_bad_request(cli);
3897	}
3898	err = og_cmd_stop(root, &params);
3899	} else if (!strncmp(cmd, "refresh", strlen("refresh"))) {
3900	if (method != OG_METHOD_POST)
3901	return og_client_method_not_found(cli);
3902
3903	if (!root) {
3904	syslog(LOG_ERR, "command refresh with no payload\n");
3905	return og_client_bad_request(cli);
3906	}
3907	err = og_cmd_refresh(root, &params);
3908	} else if (!strncmp(cmd, "hardware", strlen("hardware"))) {
3909	if (method != OG_METHOD_GET && method != OG_METHOD_POST)
3910	return og_client_method_not_found(cli);
3911
3912	if (!root) {
3913	syslog(LOG_ERR, "command hardware with no payload\n");
3914	return og_client_bad_request(cli);
3915	}
3916
3917	if (method == OG_METHOD_GET)
3918	err = og_cmd_get_hardware(root, &params, buf_reply);
3919	else if (method == OG_METHOD_POST)
3920	err = og_cmd_hardware(root, &params);
3921	} else if (!strncmp(cmd, "software", strlen("software"))) {
3922	if (method != OG_METHOD_POST && method != OG_METHOD_GET)
3923	return og_client_method_not_found(cli);
3924
3925	if (!root) {
3926	syslog(LOG_ERR, "command software with no payload\n");
3927	return og_client_bad_request(cli);
3928	}
3929
3930	if (method == OG_METHOD_POST)
3931	err = og_cmd_software(root, &params);
3932	else
3933	err = og_cmd_get_software(root, &params, buf_reply);
3934	} else if (!strncmp(cmd, "images", strlen("images"))) {
3935	if (method != OG_METHOD_GET)
3936	return og_client_method_not_found(cli);
3937
3938	if (!root)
3939	err = og_cmd_images(buf_reply);
3940	else
3941	return og_client_method_not_found(cli);
3942	} else if (!strncmp(cmd, "image/create", strlen("image/create"))) {
3943	if (method != OG_METHOD_POST)
3944	return og_client_method_not_found(cli);
3945
3946	if (!root) {
3947	syslog(LOG_ERR, "command create with no payload\n");
3948	return og_client_bad_request(cli);
3949	}
3950	err = og_cmd_create_image(root, &params);
3951	} else if (!strncmp(cmd, "image/restore", strlen("image/restore"))) {
3952	if (method != OG_METHOD_POST)
3953	return og_client_method_not_found(cli);
3954
3955	if (!root) {
3956	syslog(LOG_ERR, "command create with no payload\n");
3957	return og_client_bad_request(cli);
3958	}
3959	err = og_cmd_restore_image(root, &params);
3960	} else if (!strncmp(cmd, "setup", strlen("setup"))) {
3961	if (method != OG_METHOD_POST)
3962	return og_client_method_not_found(cli);
3963
3964	if (!root) {
3965	syslog(LOG_ERR, "command create with no payload\n");
3966	return og_client_bad_request(cli);
3967	}
3968	err = og_cmd_setup(root, &params);
3969	} else if (!strncmp(cmd, "run/schedule", strlen("run/schedule"))) {
3970	if (method != OG_METHOD_POST)
3971	return og_client_method_not_found(cli);
3972
3973	if (!root) {
3974	syslog(LOG_ERR, "command create with no payload\n");
3975	return og_client_bad_request(cli);
3976	}
3977
3978	err = og_cmd_run_schedule(root, &params);
3979	} else if (!strncmp(cmd, "task/run", strlen("task/run"))) {
3980	if (method != OG_METHOD_POST)
3981	return og_client_method_not_found(cli);
3982
3983	if (!root) {
3984	syslog(LOG_ERR, "command task with no payload\n");
3985	return og_client_bad_request(cli);
3986	}
3987	err = og_cmd_task_post(root, &params);
3988	} else if (!strncmp(cmd, "schedule/create",
3989	strlen("schedule/create"))) {
3990	if (method != OG_METHOD_POST)
3991	return og_client_method_not_found(cli);
3992
3993	if (!root) {
3994	syslog(LOG_ERR, "command task with no payload\n");
3995	return og_client_bad_request(cli);
3996	}
3997	err = og_cmd_schedule_create(root, &params);
3998	} else if (!strncmp(cmd, "schedule/delete",
3999	strlen("schedule/delete"))) {
4000	if (method != OG_METHOD_POST)
4001	return og_client_method_not_found(cli);
4002
4003	if (!root) {
4004	syslog(LOG_ERR, "command task with no payload\n");
4005	return og_client_bad_request(cli);
4006	}
4007	err = og_cmd_schedule_delete(root, &params);
4008	} else if (!strncmp(cmd, "schedule/update",
4009	strlen("schedule/update"))) {
4010	if (method != OG_METHOD_POST)
4011	return og_client_method_not_found(cli);
4012
4013	if (!root) {
4014	syslog(LOG_ERR, "command task with no payload\n");
4015	return og_client_bad_request(cli);
4016	}
4017	err = og_cmd_schedule_update(root, &params);
4018	} else if (!strncmp(cmd, "schedule/get",
4019	strlen("schedule/get"))) {
4020	if (method != OG_METHOD_POST)
4021	return og_client_method_not_found(cli);
4022
4023	err = og_cmd_schedule_get(root, &params, buf_reply);
4024	} else {
4025	syslog(LOG_ERR, "unknown command: %.32s ...\n", cmd);
4026	err = og_client_not_found(cli);
4027	}
4028
4029	if (root)
4030	json_decref(root);
4031
4032	if (err < 0)
4033	return og_client_bad_request(cli);
4034
4035	err = og_client_ok(cli, buf_reply);
4036	if (err < 0) {
4037	syslog(LOG_ERR, "HTTP response to %s:%hu is too large\n",
4038	inet_ntoa(cli->addr.sin_addr),
4039	ntohs(cli->addr.sin_port));
4040	}
4041
4042	return err;
4043	}

Note: See TracBrowser for help on using the repository browser.

Context Navigation

source: ogServer-Git/src/rest.c @ b116081

Download in other formats: