source: ogServer-Git/src/rest.c @ 24c8b94

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

Add REST POST /client/delete method

This method deletes a client from the DB.

Request:
POST /client/delete
{

"clients": 192.168.56.30?

}

Response:
200 OK

Co-authored-by: Javier Sánchez Parra <jsanchez@…>

  • Property mode set to 100644
File size: 96.9 KB
RevLine 
[04ca20e]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"
[866b6c5]14#include "cfg.h"
[04ca20e]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>
[608709f]24#include <dirent.h>
[04ca20e]25#include <time.h>
[532b081]26#include <stdlib.h>
[139633a]27#include <unistd.h>
28#include <sys/wait.h>
[403e7c3]29#include <sys/statvfs.h>
[04ca20e]30
[48de515]31struct ev_loop *og_loop;
32
[04ca20e]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)
[06af0c2]73#define OG_REST_PARAM_NETMASK                   (1UL << 40)
[33b0c6f]74#define OG_REST_PARAM_SCOPE                     (1UL << 41)
75#define OG_REST_PARAM_MODE                      (1UL << 42)
[04ca20e]76
77static LIST_HEAD(client_list);
78
79void og_client_add(struct og_client *cli)
80{
81        list_add(&cli->list, &client_list);
82}
83
84static 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
105static const char *og_client_status(const struct og_client *cli)
106{
[a1aaad4]107        switch (cli->last_cmd) {
108        case OG_CMD_UNSPEC:
109        case OG_CMD_PROBE:
110                break;
111        default:
[04ca20e]112                return "BSY";
[a1aaad4]113        }
[04ca20e]114
115        switch (cli->status) {
116        case OG_CLIENT_STATUS_BUSY:
117                return "BSY";
118        case OG_CLIENT_STATUS_OGLIVE:
119                return "OPG";
[c0f5d2c]120        case OG_CLIENT_STATUS_VIRTUAL:
121                return "VDI";
[04ca20e]122        default:
123                return "OFF";
124        }
125}
126
127static 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
[33b0c6f]133static bool og_flags_validate(const uint64_t flags,
134                              const uint64_t required_flags)
135{
136        return (flags & required_flags) == required_flags;
137}
138
[04ca20e]139static 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
161static 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
189static 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
229static 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",
[403e7c3]244        [OG_CMD_IMAGES]         = "images",
[04ca20e]245};
246
247static 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
264int 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
321static 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
344struct og_buffer {
345        char    *data;
346        int     len;
347};
348
349static 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
359static 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
409static 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;
[06af0c2]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;
[04ca20e]443                }
444        }
445
446        return 0;
447}
448
449static 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
473static 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
493static 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 |
[06af0c2]515                                            OG_REST_PARAM_NETMASK |
[04ca20e]516                                            OG_REST_PARAM_WOL_TYPE))
517                return -1;
518
519        if (!Levanta((char **)params->ips_array, (char **)params->mac_array,
[06af0c2]520                     (char **)params->netmask_array, params->ips_array_len,
521                     (char *)params->wol_type))
[04ca20e]522                return -1;
523
524        return 0;
525}
526
527static 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
540static 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
588static 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
671static 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
706static int og_cmd_poweroff(json_t *element, struct og_msg_params *params)
707{
708        const char *key;
709        json_t *value;
710        int err = 0;
711
712        if (json_typeof(element) != JSON_OBJECT)
713                return -1;
714
715        json_object_foreach(element, key, value) {
716                if (!strcmp(key, "clients"))
717                        err = og_json_parse_clients(value, params);
718
719                if (err < 0)
720                        break;
721        }
722
723        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
724                return -1;
725
726        return og_send_request(OG_METHOD_POST, OG_CMD_POWEROFF, params, NULL);
727}
728
729static int og_cmd_refresh(json_t *element, struct og_msg_params *params)
730{
731        const char *key;
732        json_t *value;
733        int err = 0;
734
735        if (json_typeof(element) != JSON_OBJECT)
736                return -1;
737
738        json_object_foreach(element, key, value) {
739                if (!strcmp(key, "clients"))
740                        err = og_json_parse_clients(value, params);
741
742                if (err < 0)
743                        break;
744        }
745
746        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
747                return -1;
748
749        return og_send_request(OG_METHOD_GET, OG_CMD_REFRESH, params, NULL);
750}
751
752static int og_cmd_reboot(json_t *element, struct og_msg_params *params)
753{
754        const char *key;
755        json_t *value;
756        int err = 0;
757
758        if (json_typeof(element) != JSON_OBJECT)
759                return -1;
760
761        json_object_foreach(element, key, value) {
762                if (!strcmp(key, "clients"))
763                        err = og_json_parse_clients(value, params);
764
765                if (err < 0)
766                        break;
767        }
768
769        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
770                return -1;
771
772        return og_send_request(OG_METHOD_POST, OG_CMD_REBOOT, params, NULL);
773}
774
[608709f]775#define OG_TFTP_TMPL_PATH "/opt/opengnsys/tftpboot/menu.lst/templates"
776
777static int og_cmd_get_modes(json_t *element, struct og_msg_params *params,
778                            char *buffer_reply)
779{
780        struct og_buffer og_buffer = {
781                .data = buffer_reply
782        };
783        json_t *root, *modes;
784        struct dirent *dent;
785        DIR *d = NULL;
786
787        root = json_object();
788        if (!root)
789                return -1;
790
791        modes = json_array();
792        if (!modes) {
793                json_decref(root);
794                return -1;
795        }
796
797        d = opendir(OG_TFTP_TMPL_PATH);
798        if (!d) {
799                json_decref(modes);
800                json_decref(root);
801                syslog(LOG_ERR, "Cannot open directory %s\n",
802                       OG_TFTP_TMPL_PATH);
803                return -1;
804        }
805
806        dent = readdir(d);
807        while (dent) {
808                if (dent->d_type != DT_REG) {
809                        dent = readdir(d);
810                        continue;
811                }
812                json_array_append_new(modes, json_string(dent->d_name));
813                dent = readdir(d);
814        }
815
816        json_object_set_new(root, "modes", modes);
817        json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
818        json_decref(root);
[e54c5ca]819        closedir(d);
[608709f]820
821        return 0;
822}
823
[532b081]824static int og_change_db_mode(struct og_dbi *dbi, const char *mac,
825                             const char * mode)
826{
827        const char *msglog;
828        dbi_result result;
829
830        result = dbi_conn_queryf(dbi->conn,
831                                 "UPDATE ordenadores SET arranque='%s' "
832                                 "WHERE mac='%s'",
833                                 mode, mac);
834
835        if (!result) {
836                dbi_conn_error(dbi->conn, &msglog);
837                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
838                       __func__, __LINE__, msglog);
839                return -1;
840        }
841
842        dbi_result_free(result);
843        return 0;
844}
845
846static int og_set_client_mode(struct og_dbi *dbi, const char *mac,
[4ae9903]847                              const char *mode, const char *template_name)
[532b081]848{
849        char filename[PATH_MAX + 1] = "/tmp/mode_params_XXXXXX";
[1855b68]850        char cmd_params[16384] = {};
851        char params[4096] = "\0";
[532b081]852        const char *msglog;
853        dbi_result result;
854        unsigned int i;
855        int numbytes;
[139633a]856        int status;
[532b081]857        int fd;
858
859        result = dbi_conn_queryf(dbi->conn,
860                "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);
861
862        if (dbi_result_get_numrows(result) != 1) {
863                dbi_conn_error(dbi->conn, &msglog);
864                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
865                       __FILE__, __LINE__, msglog);
866                dbi_result_free(result);
867                return -1;
868        }
869        dbi_result_next_row(result);
870
871        for (i = 1; i <= dbi_result_get_numfields(result); ++i)
872                strcat(params, dbi_result_get_string_idx(result, i));
873
874        dbi_result_free(result);
875
876        snprintf(cmd_params, sizeof(cmd_params),
877                 "MODE_FILE='%s'\nMAC='%s'\nDATA='%s'\n"
878                 "MODE='PERM'\nTEMPLATE_NAME='%s'",
879                 mode, mac, params, template_name);
880
881        fd = mkstemp(filename);
882        if (fd < 0) {
883                syslog(LOG_ERR, "cannot generate temp file (%s:%d)\n",
884                       __func__, __LINE__);
885                return -1;
886        }
887
888        numbytes = write(fd, cmd_params, strlen(cmd_params) + 1);
889        close(fd);
890
891        if (numbytes < 0) {
892                syslog(LOG_ERR, "cannot write file\n");
893                unlink(filename);
894                return -1;
895        }
896
[139633a]897        if (fork() == 0) {
898                execlp("/bin/bash", "/bin/bash",
899                       "/opt/opengnsys/bin/setclientmode", filename, NULL);
[5286c93]900                syslog(LOG_ERR, "failed script execution (%s:%d)\n",
901                       __func__, __LINE__);
902                exit(EXIT_FAILURE);
[139633a]903        } else {
904                wait(&status);
905        }
[532b081]906        unlink(filename);
[139633a]907
908        if (!WIFEXITED(status) || WEXITSTATUS(status) != 0) {
[532b081]909                syslog(LOG_ERR, "failed script execution (%s:%d)\n",
910                       __func__, __LINE__);
911                return -1;
912        }
913
914        if (og_change_db_mode(dbi, mac, mode) < 0) {
915                syslog(LOG_ERR, "failed to change db mode (%s:%d)\n",
916                       __func__, __LINE__);
917                return -1;
918        }
919
920        return 0;
921}
922
[33b0c6f]923static int og_cmd_post_modes(json_t *element, struct og_msg_params *params)
924{
[4ae9903]925        char ips_str[(OG_DB_IP_MAXLEN + 1) * OG_CLIENTS_MAX + 1] = {};
[532b081]926        char template_file[PATH_MAX + 1] = {};
927        char template_name[PATH_MAX + 1] = {};
928        char first_line[PATH_MAX + 1] = {};
[4ae9903]929        const char *mode_str, *mac;
930        int ips_str_len = 0;
[33b0c6f]931        struct og_dbi *dbi;
932        uint64_t flags = 0;
933        dbi_result result;
934        const char *key;
935        json_t *value;
936        int err = 0;
[532b081]937        FILE *f;
[4ae9903]938        int i;
[33b0c6f]939
940        json_object_foreach(element, key, value) {
[4ae9903]941                if (!strcmp(key, "clients")) {
942                        err = og_json_parse_clients(value, params);
[33b0c6f]943                } else if (!strcmp(key, "mode")) {
944                        err = og_json_parse_string(value, &mode_str);
945                        flags |= OG_REST_PARAM_MODE;
946                } else {
947                        err = -1;
948                }
949
950                if (err < 0)
951                        break;
952        }
953
[4ae9903]954        if (!og_flags_validate(flags, OG_REST_PARAM_MODE) ||
955            !og_msg_params_validate(params, OG_REST_PARAM_ADDR))
[33b0c6f]956                return -1;
957
[532b081]958        snprintf(template_file, sizeof(template_file), "%s/%s",
959                 OG_TFTP_TMPL_PATH, mode_str);
960        f = fopen(template_file, "r");
961        if (!f) {
962                syslog(LOG_ERR, "cannot open file (%s:%d)\n",
[33b0c6f]963                       __func__, __LINE__);
964                return -1;
965        }
966
[532b081]967        if (!fgets(first_line, sizeof(first_line), f)) {
968                fclose(f);
969                syslog(LOG_ERR, "cannot read file (%s:%d)\n",
[33b0c6f]970                       __func__, __LINE__);
971                return -1;
972        }
973
[532b081]974        fclose(f);
975
976        if (sscanf(first_line, "##NO-TOCAR-ESTA-LINEA %s", template_name) != 1) {
977                syslog(LOG_ERR, "malformed template: %s", first_line);
978                return -1;
979        }
980
[4ae9903]981        for (i = 0; i < params->ips_array_len; ++i) {
982                ips_str_len += snprintf(ips_str + ips_str_len,
983                                        sizeof(ips_str) - ips_str_len,
984                                        "'%s',", params->ips_array[i]);
985        }
986        ips_str[ips_str_len - 1] = '\0';
987
[fe1ce97]988        dbi = og_dbi_open(&ogconfig.db);
[532b081]989        if (!dbi) {
990                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
991                       __func__, __LINE__);
[33b0c6f]992                return -1;
993        }
994
[532b081]995        result = dbi_conn_queryf(dbi->conn,
996                                 "SELECT mac FROM ordenadores "
[4ae9903]997                                 "WHERE ip IN (%s)", ips_str);
[532b081]998
999        while (dbi_result_next_row(result)) {
1000                mac = dbi_result_get_string(result, "mac");
[4ae9903]1001                err = og_set_client_mode(dbi, mac, mode_str, template_name);
[532b081]1002                if (err != 0) {
1003                        dbi_result_free(result);
1004                        og_dbi_close(dbi);
1005                        return -1;
1006                }
1007        }
1008
[33b0c6f]1009        dbi_result_free(result);
1010        og_dbi_close(dbi);
1011
1012        return 0;
1013}
1014
[b59ff7c]1015static int og_cmd_get_client_setup(json_t *element,
1016                                   struct og_msg_params *params,
1017                                   char *buffer_reply)
1018{
1019        json_t *value, *root, *partitions_array, *partition_json;
1020        const char *key, *msglog;
1021        unsigned int len_part;
1022        struct og_dbi *dbi;
1023        dbi_result result;
1024        int err = 0;
1025
1026        struct og_buffer og_buffer = {
1027                .data = buffer_reply
1028        };
1029
1030        struct {
1031                int disk;
1032                int number;
1033                int code;
1034                int size;
1035                int filesystem;
1036                int format;
1037                int os;
1038                int used_size;
1039                int image;
1040                int software;
1041        } partition;
1042
1043        json_object_foreach(element, key, value) {
1044                if (!strcmp(key, "client")) {
1045                        err = og_json_parse_clients(value, params);
1046                }
1047
1048                if (err < 0)
1049                        break;
1050        }
1051
1052        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1053                return -1;
1054
1055        if (params->ips_array_len != 1)
1056                return -1;
1057
1058        root = json_object();
1059        if (!root)
1060                return -1;
1061
1062        partitions_array = json_array();
1063        if (!partitions_array) {
1064                json_decref(root);
1065                return -1;
1066        }
1067        json_object_set_new(root, "partitions", partitions_array);
1068
[fe1ce97]1069        dbi = og_dbi_open(&ogconfig.db);
[b59ff7c]1070        if (!dbi) {
1071                json_decref(root);
1072                syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1073                       __func__, __LINE__);
1074                return -1;
1075        }
1076
1077        result = dbi_conn_queryf(dbi->conn,
1078                                 "SELECT numdisk, numpar, codpar, tamano, "
1079                                 "       uso, idsistemafichero, idnombreso, "
1080                                 "       idimagen, idperfilsoft "
1081                                 "FROM ordenadores_particiones "
1082                                 "INNER JOIN ordenadores "
1083                                 "ON ordenadores.idordenador = ordenadores_particiones.idordenador "
1084                                 "WHERE ordenadores.ip='%s'",
1085                                 params->ips_array[0]);
1086        if (!result) {
1087                json_decref(root);
1088                dbi_conn_error(dbi->conn, &msglog);
1089                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1090                       __func__, __LINE__, msglog);
1091                og_dbi_close(dbi);
1092                return -1;
1093        }
1094
1095        len_part = 0;
[ff71a2a]1096        /* partition 0 represents the full disk, hence OG_PARTITION_MAX + 1. */
1097        while (dbi_result_next_row(result) && len_part < OG_PARTITION_MAX + 1) {
[b59ff7c]1098                partition.disk = dbi_result_get_int(result, "numdisk");
1099                partition.number = dbi_result_get_int(result, "numpar");
1100                partition.code = dbi_result_get_int(result, "codpar");
1101                partition.size = dbi_result_get_int(result, "tamano");
1102                partition.used_size = dbi_result_get_int(result, "uso");
1103                partition.filesystem = dbi_result_get_int(result, "idsistemafichero");
1104                partition.os = dbi_result_get_int(result, "idnombreso");
1105                partition.image = dbi_result_get_int(result, "idimagen");
1106                partition.software = dbi_result_get_int(result, "idperfilsoft");
1107
1108                partition_json = json_object();
1109                if (!partition_json) {
1110                        json_decref(root);
1111                        dbi_result_free(result);
1112                        og_dbi_close(dbi);
1113                        return -1;
1114                }
1115
1116                json_object_set_new(partition_json, "disk",
1117                                    json_integer(partition.disk));
1118                json_object_set_new(partition_json, "partition",
1119                                    json_integer(partition.number));
1120                json_object_set_new(partition_json, "code",
1121                                    json_integer(partition.code));
1122                json_object_set_new(partition_json, "size",
1123                                    json_integer(partition.size));
1124                json_object_set_new(partition_json, "used_size",
1125                                    json_integer(partition.used_size));
1126                json_object_set_new(partition_json, "filesystem",
1127                                    json_integer(partition.filesystem));
1128                json_object_set_new(partition_json, "os",
1129                                    json_integer(partition.os));
1130                json_object_set_new(partition_json, "image",
1131                                    json_integer(partition.image));
1132                json_object_set_new(partition_json, "software",
1133                                    json_integer(partition.software));
1134                json_array_append_new(partitions_array, partition_json);
1135
1136                ++len_part;
1137        }
1138
1139        dbi_result_free(result);
1140        og_dbi_close(dbi);
1141
1142        json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1143        json_decref(root);
1144        return 0;
1145}
1146
[af47a08]1147static int og_cmd_get_client_info(json_t *element,
1148                                  struct og_msg_params *params,
1149                                  char *buffer_reply)
1150{
1151        struct og_computer computer = {};
1152        json_t *value, *root;
1153        struct in_addr addr;
1154        struct og_dbi *dbi;
1155        const char *key;
1156        int err = 0;
1157
1158        struct og_buffer og_buffer = {
1159                .data = buffer_reply
1160        };
1161
1162        json_object_foreach(element, key, value) {
1163                if (!strcmp(key, "client")) {
1164                        err = og_json_parse_clients(value, params);
1165                }
1166
1167                if (err < 0)
1168                        break;
1169        }
1170
1171        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1172                return -1;
1173
1174        if (params->ips_array_len != 1)
1175                return -1;
1176
1177        if (inet_aton(params->ips_array[0], &addr) == 0)
1178                return -1;
1179
[fe1ce97]1180        dbi = og_dbi_open(&ogconfig.db);
[af47a08]1181        if (!dbi) {
1182                syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1183                       __func__, __LINE__);
1184                return -1;
1185        }
1186
1187        if (og_dbi_get_computer_info(dbi, &computer, addr)) {
1188                og_dbi_close(dbi);
1189                return -1;
1190        }
1191
1192        og_dbi_close(dbi);
1193
1194        root = json_object();
1195        if (!root)
1196                return -1;
1197
1198        json_object_set_new(root, "serial_number",
1199                            json_string(computer.serial_number));
1200        json_object_set_new(root, "hardware_id",
1201                            json_integer(computer.hardware_id));
1202        json_object_set_new(root, "netdriver", json_string(computer.netdriver));
1203        json_object_set_new(root, "maintenance", json_boolean(computer.name));
1204        json_object_set_new(root, "netiface", json_string(computer.netiface));
1205        json_object_set_new(root, "repo_id", json_integer(computer.repo_id));
1206        json_object_set_new(root, "livedir", json_string(computer.livedir));
1207        json_object_set_new(root, "netmask", json_string(computer.netmask));
1208        json_object_set_new(root, "center", json_integer(computer.center));
1209        json_object_set_new(root, "remote", json_boolean(computer.remote));
1210        json_object_set_new(root, "room", json_integer(computer.room));
1211        json_object_set_new(root, "name", json_string(computer.name));
1212        json_object_set_new(root, "boot", json_string(computer.boot));
1213        json_object_set_new(root, "mac", json_string(computer.mac));
1214        json_object_set_new(root, "id", json_integer(computer.id));
1215        json_object_set_new(root, "ip", json_string(computer.ip));
1216
1217        json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1218        json_decref(root);
1219        return 0;
1220}
1221
[65a14e5]1222static int og_cmd_post_client_add(json_t *element,
1223                                  struct og_msg_params *params,
1224                                  char *buffer_reply)
1225{
1226        struct og_computer computer = {};
[f537daf]1227        const char *key, *msglog;
[65a14e5]1228        struct og_dbi *dbi;
1229        dbi_result result;
1230        json_t *value;
1231        int err = 0;
1232
1233        json_object_foreach(element, key, value) {
1234                if (!strcmp(key, "serial_number")) {
[f537daf]1235                        err = og_json_parse_string_copy(value,
1236                                                        computer.serial_number,
1237                                                        sizeof(computer.serial_number));
[65a14e5]1238                } else if (!strcmp(key, "hardware_id")) {
1239                        err = og_json_parse_uint(value, &computer.hardware_id);
1240                } else if (!strcmp(key, "netdriver")) {
[f537daf]1241                        err = og_json_parse_string_copy(value,
1242                                                        computer.netdriver,
1243                                                        sizeof(computer.netdriver));
[65a14e5]1244                } else if (!strcmp(key, "maintenance")) {
1245                        err = og_json_parse_bool(value, &computer.maintenance);
1246                } else if (!strcmp(key, "netiface")) {
[f537daf]1247                        err = og_json_parse_string_copy(value,
1248                                                        computer.netiface,
1249                                                        sizeof(computer.netiface));
[65a14e5]1250                } else if (!strcmp(key, "repo_id")) {
1251                        err = og_json_parse_uint(value, &computer.repo_id);
1252                } else if (!strcmp(key, "livedir")) {
[f537daf]1253                        err = og_json_parse_string_copy(value,
1254                                                        computer.livedir,
1255                                                        sizeof(computer.livedir));
[65a14e5]1256                } else if (!strcmp(key, "netmask")) {
[f537daf]1257                        err = og_json_parse_string_copy(value,
1258                                                        computer.netmask,
1259                                                        sizeof(computer.netmask));
[65a14e5]1260                } else if (!strcmp(key, "remote")) {
1261                        err = og_json_parse_bool(value, &computer.remote);
1262                } else if (!strcmp(key, "room")) {
1263                        err = og_json_parse_uint(value, &computer.room);
1264                } else if (!strcmp(key, "name")) {
[f537daf]1265                        err = og_json_parse_string_copy(value,
1266                                                        computer.name,
1267                                                        sizeof(computer.name));
[65a14e5]1268                } else if (!strcmp(key, "boot")) {
[f537daf]1269                        err = og_json_parse_string_copy(value,
1270                                                        computer.boot,
1271                                                        sizeof(computer.boot));
[65a14e5]1272                } else if (!strcmp(key, "mac")) {
[f537daf]1273                        err = og_json_parse_string_copy(value,
1274                                                        computer.mac,
1275                                                        sizeof(computer.mac));
[65a14e5]1276                } else if (!strcmp(key, "ip")) {
[f537daf]1277                        err = og_json_parse_string_copy(value,
1278                                                        computer.ip,
1279                                                        sizeof(computer.ip));
[65a14e5]1280                }
1281
1282                if (err < 0)
1283                        break;
1284        }
1285
[fe1ce97]1286        dbi = og_dbi_open(&ogconfig.db);
[65a14e5]1287        if (!dbi) {
1288                syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1289                       __func__, __LINE__);
1290                return -1;
1291        }
1292
1293        result = dbi_conn_queryf(dbi->conn,
1294                                 "SELECT ip FROM ordenadores WHERE ip='%s'",
1295                                 computer.ip);
1296
1297        if (!result) {
1298                dbi_conn_error(dbi->conn, &msglog);
1299                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1300                       __func__, __LINE__, msglog);
1301                og_dbi_close(dbi);
1302                return -1;
1303        }
1304
1305        if (dbi_result_get_numrows(result) > 0) {
1306                syslog(LOG_ERR, "client with the same IP already exists: %s\n",
1307                       computer.ip);
1308                dbi_result_free(result);
1309                og_dbi_close(dbi);
1310                return -1;
1311        }
1312        dbi_result_free(result);
1313
1314        result = dbi_conn_queryf(dbi->conn,
1315                                 "INSERT INTO ordenadores("
1316                                 "  nombreordenador,"
1317                                 "  numserie,"
1318                                 "  ip,"
1319                                 "  mac,"
1320                                 "  idaula,"
1321                                 "  idperfilhard,"
1322                                 "  idrepositorio,"
1323                                 "  mascara,"
1324                                 "  arranque,"
1325                                 "  netiface,"
1326                                 "  netdriver,"
1327                                 "  oglivedir,"
1328                                 "  inremotepc,"
1329                                 "  maintenance"
1330                                 ") VALUES ('%s', '%s', '%s', '%s', %u, %u,"
1331                                 "           %u, '%s', '%s', '%s', '%s',"
1332                                 "          '%s', %u, %u)",
1333                                 computer.name, computer.serial_number,
1334                                 computer.ip, computer.mac, computer.room,
1335                                 computer.hardware_id, computer.repo_id,
1336                                 computer.netmask, computer.boot,
1337                                 computer.netiface, computer.netdriver,
1338                                 computer.livedir, computer.remote,
1339                                 computer.maintenance);
1340
1341        if (!result) {
1342                dbi_conn_error(dbi->conn, &msglog);
1343                syslog(LOG_ERR, "failed to add client to database (%s:%d) %s\n",
1344                       __func__, __LINE__, msglog);
1345                og_dbi_close(dbi);
1346                return -1;
1347        }
1348
1349        dbi_result_free(result);
1350        og_dbi_close(dbi);
1351        return 0;
1352}
1353
[24c8b94]1354static int og_cmd_post_client_delete(json_t *element,
1355                                     struct og_msg_params *params)
1356{
1357        const char *key, *msglog;
1358        struct og_dbi *dbi;
1359        dbi_result result;
1360        unsigned int i;
1361        json_t *value;
1362        int err = 0;
1363
1364        json_object_foreach(element, key, value) {
1365                if (!strcmp(key, "clients"))
1366                        err = og_json_parse_clients(value, params);
1367                else
1368                        err = -1;
1369
1370                if (err < 0)
1371                        break;
1372        }
1373
1374        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1375                return -1;
1376
1377        dbi = og_dbi_open(&ogconfig.db);
1378        if (!dbi) {
1379                syslog(LOG_ERR, "cannot open conection database (%s:%d)\n",
1380                       __func__, __LINE__);
1381                return -1;
1382        }
1383
1384        for (i = 0; i < params->ips_array_len; i++) {
1385                result = dbi_conn_queryf(dbi->conn,
1386                                         "DELETE FROM ordenadores WHERE ip='%s'",
1387                                         params->ips_array[i]);
1388
1389                if (!result) {
1390                        dbi_conn_error(dbi->conn, &msglog);
1391                        syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1392                               __func__, __LINE__, msglog);
1393                        og_dbi_close(dbi);
1394                        return -1;
1395                }
1396
1397                dbi_result_free(result);
1398        }
1399
1400        og_dbi_close(dbi);
1401        return 0;
1402}
1403
[04ca20e]1404static int og_cmd_stop(json_t *element, struct og_msg_params *params)
1405{
1406        const char *key;
1407        json_t *value;
1408        int err = 0;
1409
1410        if (json_typeof(element) != JSON_OBJECT)
1411                return -1;
1412
1413        json_object_foreach(element, key, value) {
1414                if (!strcmp(key, "clients"))
1415                        err = og_json_parse_clients(value, params);
1416
1417                if (err < 0)
1418                        break;
1419        }
1420
1421        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1422                return -1;
1423
1424        return og_send_request(OG_METHOD_POST, OG_CMD_STOP, params, NULL);
1425}
1426
1427static int og_cmd_hardware(json_t *element, struct og_msg_params *params)
1428{
1429        const char *key;
1430        json_t *value;
1431        int err = 0;
1432
1433        if (json_typeof(element) != JSON_OBJECT)
1434                return -1;
1435
1436        json_object_foreach(element, key, value) {
1437                if (!strcmp(key, "clients"))
1438                        err = og_json_parse_clients(value, params);
1439
1440                if (err < 0)
1441                        break;
1442        }
1443
1444        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1445                return -1;
1446
1447        return og_send_request(OG_METHOD_GET, OG_CMD_HARDWARE, params, NULL);
1448}
1449
[a8eccba]1450static int og_cmd_get_hardware(json_t *element, struct og_msg_params *params,
1451                               char *buffer_reply)
1452{
1453        const char *key, *msglog, *hw_item, *hw_type;
1454        json_t *value, *root, *array, *item;
1455        struct og_scope scope = {};
1456        uint64_t flags = 0;
1457        struct og_dbi *dbi;
1458        dbi_result result;
1459        int err = 0;
1460
1461        struct og_buffer og_buffer = {
1462                .data = buffer_reply
1463        };
1464
1465        json_object_foreach(element, key, value) {
1466                if (!strcmp(key, "scope")) {
1467                        err = og_json_parse_scope(value, &scope,
1468                                                  OG_PARAM_SCOPE_ID |
1469                                                  OG_PARAM_SCOPE_TYPE);
1470                        flags |= OG_REST_PARAM_SCOPE;
1471                } else {
1472                        err = -1;
1473                }
1474
1475                if (err < 0)
1476                        return err;
1477        }
1478
1479        if (!og_flags_validate(flags, OG_REST_PARAM_SCOPE))
1480                return -1;
1481
1482        if (strcmp(scope.type, "computer")) {
1483                syslog(LOG_ERR, "incorrect scope type\n");
1484                return -1;
1485        }
1486
[fe1ce97]1487        dbi = og_dbi_open(&ogconfig.db);
[a8eccba]1488        if (!dbi) {
1489                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
1490                       __func__, __LINE__);
1491                return -1;
1492        }
1493
1494        result = dbi_conn_queryf(dbi->conn,
1495                                 "SELECT hardwares.descripcion AS item, "
1496                                 "       tipohardwares.descripcion AS type "
1497                                 "FROM hardwares "
1498                                 "INNER JOIN perfileshard_hardwares "
1499                                 "    ON hardwares.idhardware = perfileshard_hardwares.idhardware "
1500                                 "INNER JOIN ordenadores "
1501                                 "    ON perfileshard_hardwares.idperfilhard = ordenadores.idperfilhard "
1502                                 "INNER JOIN tipohardwares "
1503                                 "    ON hardwares.idtipohardware = tipohardwares.idtipohardware "
1504                                 "WHERE ordenadores.idordenador = %u",
1505                                 scope.id);
1506        if (!result) {
1507                dbi_conn_error(dbi->conn, &msglog);
1508                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
1509                       __func__, __LINE__, msglog);
1510                og_dbi_close(dbi);
1511                return -1;
1512        }
1513
1514        array = json_array();
1515        if (!array) {
1516                dbi_result_free(result);
1517                og_dbi_close(dbi);
1518                return -1;
1519        }
1520
1521        while (dbi_result_next_row(result)) {
1522                item = json_object();
1523                if (!item) {
1524                        dbi_result_free(result);
1525                        og_dbi_close(dbi);
1526                        json_decref(array);
1527                        return -1;
1528                }
1529
1530                hw_item = dbi_result_get_string(result, "item");
1531                hw_type = dbi_result_get_string(result, "type");
1532
1533                json_object_set_new(item, "type", json_string(hw_type));
1534                json_object_set_new(item, "description", json_string(hw_item));
1535                json_array_append_new(array, item);
1536        }
1537
1538        dbi_result_free(result);
1539        og_dbi_close(dbi);
1540
1541        root = json_object();
1542        if (!root){
1543                json_decref(array);
1544                return -1;
1545        }
1546
1547        json_object_set_new(root, "hardware", array);
1548
1549        json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1550        json_decref(root);
1551        return 0;
1552}
1553
[04ca20e]1554static int og_cmd_software(json_t *element, struct og_msg_params *params)
1555{
1556        json_t *clients, *value;
1557        const char *key;
1558        int err = 0;
1559
1560        if (json_typeof(element) != JSON_OBJECT)
1561                return -1;
1562
1563        json_object_foreach(element, key, value) {
1564                if (!strcmp(key, "clients"))
1565                        err = og_json_parse_clients(value, params);
1566                else if (!strcmp(key, "disk")) {
1567                        err = og_json_parse_string(value, &params->disk);
1568                        params->flags |= OG_REST_PARAM_DISK;
1569                }
1570                else if (!strcmp(key, "partition")) {
1571                        err = og_json_parse_string(value, &params->partition);
1572                        params->flags |= OG_REST_PARAM_PARTITION;
1573                }
1574
1575                if (err < 0)
1576                        break;
1577        }
1578
1579        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR |
1580                                            OG_REST_PARAM_DISK |
1581                                            OG_REST_PARAM_PARTITION))
1582                return -1;
1583
1584        clients = json_copy(element);
1585        json_object_del(clients, "clients");
1586
1587        return og_send_request(OG_METHOD_POST, OG_CMD_SOFTWARE, params, clients);
1588}
1589
[403e7c3]1590#define OG_IMAGE_TYPE_MAXLEN    4
1591
1592struct og_image {
1593        const char              *filename;
1594        uint64_t                datasize;
1595        struct stat             image_stats;
1596};
1597
1598static int og_get_image_stats(const char *name,
1599                              struct stat *image_stats)
1600{
[3cea4bb]1601        const char *dir = ogconfig.repo.dir;
[403e7c3]1602        char filename[PATH_MAX + 1];
1603
1604        snprintf(filename, sizeof(filename), "%s/%s", dir, name);
1605        if (stat(filename, image_stats) < 0) {
1606                syslog(LOG_ERR, "%s image does not exists", name);
1607                return -1;
1608        }
1609        return 0;
1610}
1611
1612static json_t *og_json_disk_alloc()
1613{
[3cea4bb]1614        const char *dir = ogconfig.repo.dir;
[403e7c3]1615        struct statvfs buffer;
1616        json_t *disk_json;
1617        int ret;
1618
1619        ret = statvfs(dir, &buffer);
1620        if (ret)
1621                return NULL;
1622
1623        disk_json = json_object();
1624        if (!disk_json)
1625                return NULL;
1626
1627        json_object_set_new(disk_json, "total",
1628                            json_integer(buffer.f_blocks * buffer.f_frsize));
1629        json_object_set_new(disk_json, "free",
1630                            json_integer(buffer.f_bfree * buffer.f_frsize));
1631
1632        return disk_json;
1633}
1634
1635#define OG_PERMS_IRWX (S_IRWXU | S_IRWXG | S_IRWXO)
1636#define OG_PERMS_MAXLEN 4
1637
1638static json_t *og_json_image_alloc(struct og_image *image)
1639{
1640        char perms_string[OG_PERMS_MAXLEN];
1641        json_t *image_json;
1642        char *modified;
1643        uint16_t perms;
1644
1645        image_json = json_object();
1646        if (!image_json)
1647                return NULL;
1648
1649        perms = image->image_stats.st_mode & OG_PERMS_IRWX;
1650        snprintf(perms_string, sizeof(perms_string), "%o", perms);
1651
1652        modified = ctime(&image->image_stats.st_mtime);
1653        modified[strlen(modified) - 1] = '\0';
1654
1655        json_object_set_new(image_json, "filename",
1656                            json_string(image->filename));
1657        json_object_set_new(image_json, "datasize",
1658                            json_integer(image->datasize));
1659        json_object_set_new(image_json, "size",
1660                            json_integer(image->image_stats.st_size));
1661        json_object_set_new(image_json, "modified",
1662                            json_string(modified));
1663        json_object_set_new(image_json, "permissions",
1664                            json_string(perms_string));
1665
1666        return image_json;
1667}
1668
1669static int og_cmd_images(char *buffer_reply)
1670{
1671        json_t *root, *images, *image_json, *disk_json;
1672        struct og_buffer og_buffer = {
1673                .data = buffer_reply
1674        };
1675        struct og_image image;
1676        struct og_dbi *dbi;
1677        dbi_result result;
1678
1679        root = json_object();
1680        if (!root)
1681                return -1;
1682
1683        images = json_array();
1684        if (!images) {
1685                json_decref(root);
1686                return -1;
1687        }
1688
1689        json_object_set_new(root, "images", images);
1690
[fe1ce97]1691        dbi = og_dbi_open(&ogconfig.db);
[403e7c3]1692        if (!dbi) {
1693                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
1694                       __func__, __LINE__);
1695                json_decref(root);
1696                return -1;
1697        }
1698
1699        result = dbi_conn_queryf(dbi->conn,
1700                                 "SELECT i.nombreca, o.nombreordenador, "
1701                                 "       i.clonator, i.compressor, "
1702                                 "       i.filesystem, i.datasize "
1703                                 "FROM imagenes i "
1704                                 "LEFT JOIN ordenadores o "
1705                                 "ON i.idordenador = o.idordenador");
1706
1707        while (dbi_result_next_row(result)) {
1708                image = (struct og_image){0};
1709                image.filename = dbi_result_get_string(result, "nombreca");
1710                image.datasize = dbi_result_get_ulonglong(result, "datasize");
1711
1712                if (og_get_image_stats(image.filename, &image.image_stats)) {
1713                        continue;
1714                }
1715
1716                image_json = og_json_image_alloc(&image);
1717                if (!image_json) {
1718                        dbi_result_free(result);
1719                        og_dbi_close(dbi);
1720                        json_decref(root);
1721                        return -1;
1722                }
1723
1724                json_array_append(images, image_json);
1725        }
1726
1727        dbi_result_free(result);
1728        og_dbi_close(dbi);
1729
1730        disk_json = og_json_disk_alloc();
1731        if (!disk_json) {
1732                syslog(LOG_ERR, "cannot allocate disk json");
1733                json_decref(root);
1734                return -1;
1735        }
1736
1737        json_object_set_new(root, "disk", disk_json);
1738
1739        json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
1740        json_decref(root);
1741
1742        return 0;
1743}
1744
[04ca20e]1745static int og_cmd_create_image(json_t *element, struct og_msg_params *params)
1746{
1747        json_t *value, *clients;
1748        const char *key;
1749        int err = 0;
1750
1751        if (json_typeof(element) != JSON_OBJECT)
1752                return -1;
1753
1754        json_object_foreach(element, key, value) {
1755                if (!strcmp(key, "disk")) {
1756                        err = og_json_parse_string(value, &params->disk);
1757                        params->flags |= OG_REST_PARAM_DISK;
1758                } else if (!strcmp(key, "partition")) {
1759                        err = og_json_parse_string(value, &params->partition);
1760                        params->flags |= OG_REST_PARAM_PARTITION;
1761                } else if (!strcmp(key, "name")) {
1762                        err = og_json_parse_string(value, &params->name);
1763                        params->flags |= OG_REST_PARAM_NAME;
1764                } else if (!strcmp(key, "repository")) {
1765                        err = og_json_parse_string(value, &params->repository);
1766                        params->flags |= OG_REST_PARAM_REPO;
1767                } else if (!strcmp(key, "clients")) {
1768                        err = og_json_parse_clients(value, params);
1769                } else if (!strcmp(key, "id")) {
1770                        err = og_json_parse_string(value, &params->id);
1771                        params->flags |= OG_REST_PARAM_ID;
1772                } else if (!strcmp(key, "code")) {
1773                        err = og_json_parse_string(value, &params->code);
1774                        params->flags |= OG_REST_PARAM_CODE;
1775                }
1776
1777                if (err < 0)
1778                        break;
1779        }
1780
1781        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR |
1782                                            OG_REST_PARAM_DISK |
1783                                            OG_REST_PARAM_PARTITION |
1784                                            OG_REST_PARAM_CODE |
1785                                            OG_REST_PARAM_ID |
1786                                            OG_REST_PARAM_NAME |
1787                                            OG_REST_PARAM_REPO))
1788                return -1;
1789
1790        clients = json_copy(element);
1791        json_object_del(clients, "clients");
1792
1793        return og_send_request(OG_METHOD_POST, OG_CMD_IMAGE_CREATE, params,
1794                               clients);
1795}
1796
1797static int og_cmd_restore_image(json_t *element, struct og_msg_params *params)
1798{
1799        json_t *clients, *value;
1800        const char *key;
1801        int err = 0;
1802
1803        if (json_typeof(element) != JSON_OBJECT)
1804                return -1;
1805
1806        json_object_foreach(element, key, value) {
1807                if (!strcmp(key, "disk")) {
1808                        err = og_json_parse_string(value, &params->disk);
1809                        params->flags |= OG_REST_PARAM_DISK;
1810                } else if (!strcmp(key, "partition")) {
1811                        err = og_json_parse_string(value, &params->partition);
1812                        params->flags |= OG_REST_PARAM_PARTITION;
1813                } else if (!strcmp(key, "name")) {
1814                        err = og_json_parse_string(value, &params->name);
1815                        params->flags |= OG_REST_PARAM_NAME;
1816                } else if (!strcmp(key, "repository")) {
1817                        err = og_json_parse_string(value, &params->repository);
1818                        params->flags |= OG_REST_PARAM_REPO;
1819                } else if (!strcmp(key, "clients")) {
1820                        err = og_json_parse_clients(value, params);
1821                } else if (!strcmp(key, "type")) {
1822                        err = og_json_parse_string(value, &params->type);
1823                        params->flags |= OG_REST_PARAM_TYPE;
1824                } else if (!strcmp(key, "profile")) {
1825                        err = og_json_parse_string(value, &params->profile);
1826                        params->flags |= OG_REST_PARAM_PROFILE;
1827                } else if (!strcmp(key, "id")) {
1828                        err = og_json_parse_string(value, &params->id);
1829                        params->flags |= OG_REST_PARAM_ID;
1830                }
1831
1832                if (err < 0)
1833                        break;
1834        }
1835
1836        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR |
1837                                            OG_REST_PARAM_DISK |
1838                                            OG_REST_PARAM_PARTITION |
1839                                            OG_REST_PARAM_NAME |
1840                                            OG_REST_PARAM_REPO |
1841                                            OG_REST_PARAM_TYPE |
1842                                            OG_REST_PARAM_PROFILE |
1843                                            OG_REST_PARAM_ID))
1844                return -1;
1845
1846        clients = json_copy(element);
1847        json_object_del(clients, "clients");
1848
1849        return og_send_request(OG_METHOD_POST, OG_CMD_IMAGE_RESTORE, params,
1850                               clients);
1851}
1852
1853static int og_cmd_setup(json_t *element, struct og_msg_params *params)
1854{
1855        json_t *value, *clients;
1856        const char *key;
1857        int err = 0;
1858
1859        if (json_typeof(element) != JSON_OBJECT)
1860                return -1;
1861
1862        json_object_foreach(element, key, value) {
1863                if (!strcmp(key, "clients")) {
1864                        err = og_json_parse_clients(value, params);
1865                } else if (!strcmp(key, "disk")) {
1866                        err = og_json_parse_string(value, &params->disk);
1867                        params->flags |= OG_REST_PARAM_DISK;
1868                } else if (!strcmp(key, "cache")) {
1869                        err = og_json_parse_string(value, &params->cache);
1870                        params->flags |= OG_REST_PARAM_CACHE;
1871                } else if (!strcmp(key, "cache_size")) {
1872                        err = og_json_parse_string(value, &params->cache_size);
1873                        params->flags |= OG_REST_PARAM_CACHE_SIZE;
1874                } else if (!strcmp(key, "partition_setup")) {
1875                        err = og_json_parse_partition_setup(value, params);
1876                }
1877
1878                if (err < 0)
1879                        break;
1880        }
1881
1882        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR |
1883                                            OG_REST_PARAM_DISK |
1884                                            OG_REST_PARAM_CACHE |
1885                                            OG_REST_PARAM_CACHE_SIZE |
1886                                            OG_REST_PARAM_PART_0 |
1887                                            OG_REST_PARAM_PART_1 |
1888                                            OG_REST_PARAM_PART_2 |
1889                                            OG_REST_PARAM_PART_3))
1890                return -1;
1891
1892        clients = json_copy(element);
1893        json_object_del(clients, "clients");
1894
1895        return og_send_request(OG_METHOD_POST, OG_CMD_SETUP, params, clients);
1896}
1897
1898static int og_cmd_run_schedule(json_t *element, struct og_msg_params *params)
1899{
1900        const char *key;
1901        json_t *value;
1902        int err = 0;
1903
1904        json_object_foreach(element, key, value) {
1905                if (!strcmp(key, "clients"))
1906                        err = og_json_parse_clients(value, params);
1907
1908                if (err < 0)
1909                        break;
1910        }
1911
1912        if (!og_msg_params_validate(params, OG_REST_PARAM_ADDR))
1913                return -1;
1914
1915        return og_send_request(OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, params,
1916                               NULL);
1917}
1918
1919static LIST_HEAD(cmd_list);
1920
1921const struct og_cmd *og_cmd_find(const char *client_ip)
1922{
1923        struct og_cmd *cmd, *next;
1924
1925        list_for_each_entry_safe(cmd, next, &cmd_list, list) {
1926                if (strcmp(cmd->ip, client_ip))
1927                        continue;
1928
1929                list_del(&cmd->list);
1930                return cmd;
1931        }
1932
1933        return NULL;
1934}
1935
1936void og_cmd_free(const struct og_cmd *cmd)
1937{
1938        struct og_msg_params *params = (struct og_msg_params *)&cmd->params;
1939        int i;
1940
1941        for (i = 0; i < params->ips_array_len; i++) {
1942                free((void *)params->ips_array[i]);
1943                free((void *)params->mac_array[i]);
1944        }
1945        free((void *)params->wol_type);
1946
1947        if (cmd->json)
1948                json_decref(cmd->json);
1949
1950        free((void *)cmd->ip);
1951        free((void *)cmd->mac);
1952        free((void *)cmd);
1953}
1954
1955static void og_cmd_init(struct og_cmd *cmd, enum og_rest_method method,
1956                        enum og_cmd_type type, json_t *root)
1957{
1958        cmd->type = type;
1959        cmd->method = method;
1960        cmd->params.ips_array[0] = strdup(cmd->ip);
1961        cmd->params.ips_array_len = 1;
1962        cmd->json = root;
1963}
1964
1965static int og_cmd_legacy_wol(const char *input, struct og_cmd *cmd)
1966{
1967        char wol_type[2] = {};
1968
1969        if (sscanf(input, "mar=%s", wol_type) != 1) {
1970                syslog(LOG_ERR, "malformed database legacy input\n");
1971                return -1;
1972        }
1973
1974        og_cmd_init(cmd, OG_METHOD_NO_HTTP, OG_CMD_WOL, NULL);
1975        cmd->params.mac_array[0] = strdup(cmd->mac);
1976        cmd->params.wol_type = strdup(wol_type);
1977
1978        return 0;
1979}
1980
1981static int og_cmd_legacy_shell_run(const char *input, struct og_cmd *cmd)
1982{
1983        json_t *root, *script, *echo;
1984
1985        script = json_string(input + 4);
1986        echo = json_boolean(false);
1987
1988        root = json_object();
1989        if (!root)
1990                return -1;
1991        json_object_set_new(root, "run", script);
1992        json_object_set_new(root, "echo", echo);
1993
1994        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_SHELL_RUN, root);
1995
1996        return 0;
1997}
1998
1999static int og_cmd_legacy_session(const char *input, struct og_cmd *cmd)
2000{
2001        char part_str[OG_DB_SMALLINT_MAXLEN + 1];
2002        char disk_str[OG_DB_SMALLINT_MAXLEN + 1];
2003        json_t *root, *disk, *partition;
2004
2005        if (sscanf(input, "dsk=%s\rpar=%s\r", disk_str, part_str) != 2)
2006                return -1;
2007        partition = json_string(part_str);
2008        disk = json_string(disk_str);
2009
2010        root = json_object();
2011        if (!root)
2012                return -1;
2013        json_object_set_new(root, "partition", partition);
2014        json_object_set_new(root, "disk", disk);
2015
2016        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_SESSION, root);
2017
2018        return 0;
2019}
2020
2021static int og_cmd_legacy_poweroff(const char *input, struct og_cmd *cmd)
2022{
2023        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_POWEROFF, NULL);
2024
2025        return 0;
2026}
2027
2028static int og_cmd_legacy_refresh(const char *input, struct og_cmd *cmd)
2029{
2030        og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_REFRESH, NULL);
2031
2032        return 0;
2033}
2034
2035static int og_cmd_legacy_reboot(const char *input, struct og_cmd *cmd)
2036{
2037        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_REBOOT, NULL);
2038
2039        return 0;
2040}
2041
2042static int og_cmd_legacy_stop(const char *input, struct og_cmd *cmd)
2043{
2044        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_STOP, NULL);
2045
2046        return 0;
2047}
2048
2049static int og_cmd_legacy_hardware(const char *input, struct og_cmd *cmd)
2050{
2051        og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_HARDWARE, NULL);
2052
2053        return 0;
2054}
2055
2056static int og_cmd_legacy_software(const char *input, struct og_cmd *cmd)
2057{
2058        og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_SOFTWARE, NULL);
2059
2060        return 0;
2061}
2062
2063static int og_cmd_legacy_image_create(const char *input, struct og_cmd *cmd)
2064{
2065        json_t *root, *disk, *partition, *code, *image_id, *name, *repo;
2066        struct og_image_legacy img = {};
2067
2068        if (sscanf(input, "dsk=%s\rpar=%s\rcpt=%s\ridi=%s\rnci=%s\ripr=%s\r",
2069                   img.disk, img.part, img.code, img.image_id, img.name,
2070                   img.repo) != 6)
2071                return -1;
2072        image_id = json_string(img.image_id);
2073        partition = json_string(img.part);
2074        code = json_string(img.code);
2075        name = json_string(img.name);
2076        repo = json_string(img.repo);
2077        disk = json_string(img.disk);
2078
2079        root = json_object();
2080        if (!root)
2081                return -1;
2082        json_object_set_new(root, "partition", partition);
2083        json_object_set_new(root, "repository", repo);
2084        json_object_set_new(root, "id", image_id);
2085        json_object_set_new(root, "code", code);
2086        json_object_set_new(root, "name", name);
2087        json_object_set_new(root, "disk", disk);
2088
2089        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_IMAGE_CREATE, root);
2090
2091        return 0;
2092}
2093
2094#define OG_DB_RESTORE_TYPE_MAXLEN       64
2095
2096static int og_cmd_legacy_image_restore(const char *input, struct og_cmd *cmd)
2097{
2098        json_t *root, *disk, *partition, *image_id, *name, *repo;
2099        char restore_type_str[OG_DB_RESTORE_TYPE_MAXLEN + 1] = {};
2100        char software_id_str[OG_DB_INT_MAXLEN + 1] = {};
2101        json_t *software_id, *restore_type;
2102        struct og_image_legacy img = {};
2103
2104        if (sscanf(input,
2105                   "dsk=%s\rpar=%s\ridi=%s\rnci=%s\ripr=%s\rifs=%s\rptc=%s\r",
2106                   img.disk, img.part, img.image_id, img.name, img.repo,
2107                   software_id_str, restore_type_str) != 7)
2108                return -1;
2109
2110        restore_type = json_string(restore_type_str);
2111        software_id = json_string(software_id_str);
2112        image_id = json_string(img.image_id);
2113        partition = json_string(img.part);
2114        name = json_string(img.name);
2115        repo = json_string(img.repo);
2116        disk = json_string(img.disk);
2117
2118        root = json_object();
2119        if (!root)
2120                return -1;
2121        json_object_set_new(root, "profile", software_id);
2122        json_object_set_new(root, "partition", partition);
2123        json_object_set_new(root, "type", restore_type);
2124        json_object_set_new(root, "repository", repo);
2125        json_object_set_new(root, "id", image_id);
2126        json_object_set_new(root, "name", name);
2127        json_object_set_new(root, "disk", disk);
2128
2129        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_IMAGE_RESTORE, root);
2130
2131        return 0;
2132}
2133
2134static int og_cmd_legacy_setup(const char *input, struct og_cmd *cmd)
2135{
2136        json_t *root, *disk, *cache, *cache_size, *partition_setup, *object;
2137        struct og_legacy_partition part_cfg[OG_PARTITION_MAX] = {};
2138        char cache_size_str [OG_DB_INT_MAXLEN + 1];
2139        char disk_str [OG_DB_SMALLINT_MAXLEN + 1];
2140        json_t *part, *code, *fs, *size, *format;
2141        unsigned int partition_len = 0;
2142        const char *in_ptr;
2143        char cache_str[2];
2144
2145        if (sscanf(input, "dsk=%s\rcfg=dis=%*[^*]*che=%[^*]*tch=%[^!]!",
2146                   disk_str, cache_str, cache_size_str) != 3)
2147                return -1;
2148
2149        in_ptr = strstr(input, "!") + 1;
2150        while (strlen(in_ptr) > 0) {
2151                if(sscanf(in_ptr,
2152                          "par=%[^*]*cpt=%[^*]*sfi=%[^*]*tam=%[^*]*ope=%[^%%]%%",
2153                          part_cfg[partition_len].partition,
2154                          part_cfg[partition_len].code,
2155                          part_cfg[partition_len].filesystem,
2156                          part_cfg[partition_len].size,
2157                          part_cfg[partition_len].format) != 5)
2158                        return -1;
2159                in_ptr = strstr(in_ptr, "%") + 1;
2160                partition_len++;
2161        }
2162
2163        root = json_object();
2164        if (!root)
2165                return -1;
2166
2167        cache_size = json_string(cache_size_str);
2168        cache = json_string(cache_str);
2169        partition_setup = json_array();
2170        disk = json_string(disk_str);
2171
2172        for (unsigned int i = 0; i < partition_len; ++i) {
2173                object = json_object();
2174                if (!object) {
2175                        json_decref(root);
2176                        return -1;
2177                }
2178
2179                part = json_string(part_cfg[i].partition);
2180                fs = json_string(part_cfg[i].filesystem);
2181                format = json_string(part_cfg[i].format);
2182                code = json_string(part_cfg[i].code);
2183                size = json_string(part_cfg[i].size);
2184
2185                json_object_set_new(object, "partition", part);
2186                json_object_set_new(object, "filesystem", fs);
2187                json_object_set_new(object, "format", format);
2188                json_object_set_new(object, "code", code);
2189                json_object_set_new(object, "size", size);
2190
2191                json_array_append_new(partition_setup, object);
2192        }
2193
2194        json_object_set_new(root, "partition_setup", partition_setup);
2195        json_object_set_new(root, "cache_size", cache_size);
2196        json_object_set_new(root, "cache", cache);
2197        json_object_set_new(root, "disk", disk);
2198
2199        og_cmd_init(cmd, OG_METHOD_POST, OG_CMD_SETUP, root);
2200
2201        return 0;
2202}
2203
2204static int og_cmd_legacy_run_schedule(const char *input, struct og_cmd *cmd)
2205{
2206        og_cmd_init(cmd, OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, NULL);
2207
2208        return 0;
2209}
2210
2211static int og_cmd_legacy(const char *input, struct og_cmd *cmd)
2212{
2213        char legacy_cmd[32] = {};
2214        int err = -1;
2215
2216        if (sscanf(input, "nfn=%31s\r", legacy_cmd) != 1) {
2217                syslog(LOG_ERR, "malformed database legacy input\n");
2218                return -1;
2219        }
2220        input = strchr(input, '\r') + 1;
2221
2222        if (!strcmp(legacy_cmd, "Arrancar")) {
2223                err = og_cmd_legacy_wol(input, cmd);
2224        } else if (!strcmp(legacy_cmd, "EjecutarScript")) {
2225                err = og_cmd_legacy_shell_run(input, cmd);
2226        } else if (!strcmp(legacy_cmd, "IniciarSesion")) {
2227                err = og_cmd_legacy_session(input, cmd);
2228        } else if (!strcmp(legacy_cmd, "Apagar")) {
2229                err = og_cmd_legacy_poweroff(input, cmd);
2230        } else if (!strcmp(legacy_cmd, "Actualizar")) {
2231                err = og_cmd_legacy_refresh(input, cmd);
2232        } else if (!strcmp(legacy_cmd, "Reiniciar")) {
2233                err = og_cmd_legacy_reboot(input, cmd);
2234        } else if (!strcmp(legacy_cmd, "Purgar")) {
2235                err = og_cmd_legacy_stop(input, cmd);
2236        } else if (!strcmp(legacy_cmd, "InventarioHardware")) {
2237                err = og_cmd_legacy_hardware(input, cmd);
2238        } else if (!strcmp(legacy_cmd, "InventarioSoftware")) {
2239                err = og_cmd_legacy_software(input, cmd);
2240        } else if (!strcmp(legacy_cmd, "CrearImagen")) {
2241                err = og_cmd_legacy_image_create(input, cmd);
2242        } else if (!strcmp(legacy_cmd, "RestaurarImagen")) {
2243                err = og_cmd_legacy_image_restore(input, cmd);
2244        } else if (!strcmp(legacy_cmd, "Configurar")) {
2245                err = og_cmd_legacy_setup(input, cmd);
2246        } else if (!strcmp(legacy_cmd, "EjecutaComandosPendientes") ||
2247                   !strcmp(legacy_cmd, "Actualizar")) {
2248                err = og_cmd_legacy_run_schedule(input, cmd);
2249        }
2250
2251        return err;
2252}
2253
2254static int og_dbi_add_action(const struct og_dbi *dbi, const struct og_task *task,
2255                             struct og_cmd *cmd)
2256{
2257        char start_date_string[24];
2258        struct tm *start_date;
2259        const char *msglog;
2260        dbi_result result;
2261        time_t now;
2262
2263        time(&now);
2264        start_date = localtime(&now);
2265
2266        sprintf(start_date_string, "%hu/%hhu/%hhu %hhu:%hhu:%hhu",
2267                start_date->tm_year + 1900, start_date->tm_mon + 1,
2268                start_date->tm_mday, start_date->tm_hour, start_date->tm_min,
2269                start_date->tm_sec);
2270        result = dbi_conn_queryf(dbi->conn,
2271                                "INSERT INTO acciones (idordenador, "
2272                                "tipoaccion, idtipoaccion, descriaccion, ip, "
2273                                "sesion, idcomando, parametros, fechahorareg, "
2274                                "estado, resultado, ambito, idambito, "
2275                                "restrambito, idprocedimiento, idcentro, "
2276                                "idprogramacion) "
2277                                "VALUES (%d, %d, %d, '%s', '%s', %d, %d, '%s', "
2278                                "'%s', %d, %d, %d, %d, '%s', %d, %d, %d)",
2279                                cmd->client_id, EJECUCION_TAREA, task->task_id,
2280                                "", cmd->ip, 0, task->command_id,
2281                                task->params, start_date_string,
2282                                ACCION_INICIADA, ACCION_SINRESULTADO,
2283                                task->type_scope, task->scope, "",
2284                                task->procedure_id, task->center_id,
2285                                task->schedule_id);
2286        if (!result) {
2287                dbi_conn_error(dbi->conn, &msglog);
2288                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2289                       __func__, __LINE__, msglog);
2290                return -1;
2291        }
2292        cmd->id = dbi_conn_sequence_last(dbi->conn, NULL);
2293        dbi_result_free(result);
2294
2295        return 0;
2296}
2297
2298static int og_queue_task_command(struct og_dbi *dbi, const struct og_task *task,
2299                                 char *query)
2300{
2301        struct og_cmd *cmd;
2302        const char *msglog;
2303        dbi_result result;
2304
2305        result = dbi_conn_queryf(dbi->conn, query);
2306        if (!result) {
2307                dbi_conn_error(dbi->conn, &msglog);
2308                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2309                       __func__, __LINE__, msglog);
2310                return -1;
2311        }
2312
2313        while (dbi_result_next_row(result)) {
2314                cmd = (struct og_cmd *)calloc(1, sizeof(struct og_cmd));
2315                if (!cmd) {
2316                        dbi_result_free(result);
2317                        return -1;
2318                }
2319
2320                cmd->client_id  = dbi_result_get_uint(result, "idordenador");
2321                cmd->ip         = strdup(dbi_result_get_string(result, "ip"));
2322                cmd->mac        = strdup(dbi_result_get_string(result, "mac"));
2323
2324                og_cmd_legacy(task->params, cmd);
2325
2326                if (task->procedure_id) {
2327                        if (og_dbi_add_action(dbi, task, cmd)) {
2328                                dbi_result_free(result);
2329                                return -1;
2330                        }
2331                } else {
2332                        cmd->id = task->task_id;
2333                }
2334
2335                list_add_tail(&cmd->list, &cmd_list);
2336        }
2337
2338        dbi_result_free(result);
2339
2340        return 0;
2341}
2342
2343static int og_queue_task_group_clients(struct og_dbi *dbi, struct og_task *task,
2344                                       char *query)
2345{
2346
2347        const char *msglog;
2348        dbi_result result;
2349
2350        result = dbi_conn_queryf(dbi->conn, query);
2351        if (!result) {
2352                dbi_conn_error(dbi->conn, &msglog);
2353                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2354                       __func__, __LINE__, msglog);
2355                return -1;
2356        }
2357
2358        while (dbi_result_next_row(result)) {
2359                uint32_t group_id = dbi_result_get_uint(result, "idgrupo");
2360
2361                sprintf(query, "SELECT idgrupo FROM gruposordenadores "
2362                                "WHERE grupoid=%d", group_id);
2363                if (og_queue_task_group_clients(dbi, task, query)) {
2364                        dbi_result_free(result);
2365                        return -1;
2366                }
2367
2368                sprintf(query,"SELECT ip, mac, idordenador FROM ordenadores "
2369                              "WHERE grupoid=%d", group_id);
2370                if (og_queue_task_command(dbi, task, query)) {
2371                        dbi_result_free(result);
2372                        return -1;
2373                }
2374
2375        }
2376
2377        dbi_result_free(result);
2378
2379        return 0;
2380}
2381
2382static int og_queue_task_group_classrooms(struct og_dbi *dbi,
2383                                          struct og_task *task, char *query)
2384{
2385
2386        const char *msglog;
2387        dbi_result result;
2388
2389        result = dbi_conn_queryf(dbi->conn, query);
2390        if (!result) {
2391                dbi_conn_error(dbi->conn, &msglog);
2392                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2393                       __func__, __LINE__, msglog);
2394                return -1;
2395        }
2396
2397        while (dbi_result_next_row(result)) {
2398                uint32_t group_id = dbi_result_get_uint(result, "idgrupo");
2399
2400                sprintf(query, "SELECT idgrupo FROM grupos "
2401                                "WHERE grupoid=%d AND tipo=%d", group_id, AMBITO_GRUPOSAULAS);
2402                if (og_queue_task_group_classrooms(dbi, task, query)) {
2403                        dbi_result_free(result);
2404                        return -1;
2405                }
2406
2407                sprintf(query,
2408                        "SELECT ip,mac,idordenador "
2409                        "FROM ordenadores INNER JOIN aulas "
2410                        "WHERE ordenadores.idaula=aulas.idaula "
2411                        "AND aulas.grupoid=%d",
2412                        group_id);
2413                if (og_queue_task_command(dbi, task, query)) {
2414                        dbi_result_free(result);
2415                        return -1;
2416                }
2417
2418        }
2419
2420        dbi_result_free(result);
2421
2422        return 0;
2423}
2424
2425static int og_queue_task_clients(struct og_dbi *dbi, struct og_task *task)
2426{
2427        char query[4096];
2428
2429        switch (task->type_scope) {
2430                case AMBITO_CENTROS:
2431                        sprintf(query,
2432                                "SELECT ip,mac,idordenador "
2433                                "FROM ordenadores INNER JOIN aulas "
2434                                "WHERE ordenadores.idaula=aulas.idaula "
2435                                "AND idcentro=%d",
2436                                task->scope);
2437                        return og_queue_task_command(dbi, task, query);
2438                case AMBITO_GRUPOSAULAS:
2439                        sprintf(query,
2440                                "SELECT idgrupo FROM grupos "
2441                                "WHERE idgrupo=%i AND tipo=%d",
2442                                task->scope, AMBITO_GRUPOSAULAS);
2443                        return og_queue_task_group_classrooms(dbi, task, query);
2444                case AMBITO_AULAS:
2445                        sprintf(query,
2446                                "SELECT ip,mac,idordenador FROM ordenadores "
2447                                "WHERE idaula=%d",
2448                                task->scope);
2449                        return og_queue_task_command(dbi, task, query);
2450                case AMBITO_GRUPOSORDENADORES:
2451                        sprintf(query,
2452                                "SELECT idgrupo FROM gruposordenadores "
2453                                "WHERE idgrupo = %d",
2454                                task->scope);
2455                        return og_queue_task_group_clients(dbi, task, query);
2456                case AMBITO_ORDENADORES:
2457                        sprintf(query,
2458                                "SELECT ip, mac, idordenador FROM ordenadores "
2459                                "WHERE idordenador = %d",
2460                                task->scope);
2461                        return og_queue_task_command(dbi, task, query);
2462        }
2463        return 0;
2464}
2465
2466int og_dbi_queue_procedure(struct og_dbi *dbi, struct og_task *task)
2467{
2468        uint32_t procedure_id;
2469        const char *msglog;
2470        dbi_result result;
2471
2472        result = dbi_conn_queryf(dbi->conn,
2473                        "SELECT parametros, procedimientoid, idcomando "
2474                        "FROM procedimientos_acciones "
2475                        "WHERE idprocedimiento=%d ORDER BY orden", task->procedure_id);
2476        if (!result) {
2477                dbi_conn_error(dbi->conn, &msglog);
2478                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2479                       __func__, __LINE__, msglog);
2480                return -1;
2481        }
2482
2483        while (dbi_result_next_row(result)) {
2484                procedure_id = dbi_result_get_uint(result, "procedimientoid");
2485                if (procedure_id > 0) {
2486                        task->procedure_id = procedure_id;
2487                        if (og_dbi_queue_procedure(dbi, task))
2488                                return -1;
2489                        continue;
2490                }
2491
2492                task->params    = strdup(dbi_result_get_string(result, "parametros"));
2493                task->command_id = dbi_result_get_uint(result, "idcomando");
2494                if (og_queue_task_clients(dbi, task))
2495                        return -1;
2496        }
2497
2498        dbi_result_free(result);
2499
2500        return 0;
2501}
2502
2503static int og_dbi_queue_task(struct og_dbi *dbi, uint32_t task_id,
2504                             uint32_t schedule_id)
2505{
2506        struct og_task task = {};
2507        uint32_t task_id_next;
2508        const char *msglog;
2509        dbi_result result;
2510
2511        task.schedule_id = schedule_id;
2512
2513        result = dbi_conn_queryf(dbi->conn,
2514                        "SELECT tareas_acciones.orden, "
2515                                "tareas_acciones.idprocedimiento, "
2516                                "tareas_acciones.tareaid, "
2517                                "tareas.idtarea, "
2518                                "tareas.idcentro, "
2519                                "tareas.ambito, "
2520                                "tareas.idambito, "
2521                                "tareas.restrambito "
2522                        " FROM tareas"
2523                                " INNER JOIN tareas_acciones ON tareas_acciones.idtarea=tareas.idtarea"
2524                        " WHERE tareas_acciones.idtarea=%u ORDER BY tareas_acciones.orden ASC", task_id);
2525        if (!result) {
2526                dbi_conn_error(dbi->conn, &msglog);
2527                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2528                       __func__, __LINE__, msglog);
2529                return -1;
2530        }
2531
2532        while (dbi_result_next_row(result)) {
2533                task_id_next = dbi_result_get_uint(result, "tareaid");
2534
2535                if (task_id_next > 0) {
2536                        if (og_dbi_queue_task(dbi, task_id_next, schedule_id))
2537                                return -1;
2538
2539                        continue;
2540                }
2541                task.task_id = dbi_result_get_uint(result, "idtarea");
2542                task.center_id = dbi_result_get_uint(result, "idcentro");
2543                task.procedure_id = dbi_result_get_uint(result, "idprocedimiento");
2544                task.type_scope = dbi_result_get_uint(result, "ambito");
2545                task.scope = dbi_result_get_uint(result, "idambito");
2546                task.filtered_scope = dbi_result_get_string(result, "restrambito");
2547
2548                og_dbi_queue_procedure(dbi, &task);
2549        }
2550
2551        dbi_result_free(result);
2552
2553        return 0;
2554}
2555
2556static int og_dbi_queue_command(struct og_dbi *dbi, uint32_t task_id,
2557                                uint32_t schedule_id)
2558{
2559        struct og_task task = {};
2560        const char *msglog;
2561        dbi_result result;
2562        char query[4096];
2563
2564        result = dbi_conn_queryf(dbi->conn,
2565                        "SELECT idaccion, idcentro, idordenador, parametros "
2566                        "FROM acciones "
2567                        "WHERE sesion = %u", task_id);
2568        if (!result) {
2569                dbi_conn_error(dbi->conn, &msglog);
2570                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2571                       __func__, __LINE__, msglog);
2572                return -1;
2573        }
2574
2575        while (dbi_result_next_row(result)) {
2576                task.task_id = dbi_result_get_uint(result, "idaccion");
2577                task.center_id = dbi_result_get_uint(result, "idcentro");
2578                task.scope = dbi_result_get_uint(result, "idordenador");
2579                task.params = strdup(dbi_result_get_string(result, "parametros"));
2580
2581                sprintf(query,
2582                        "SELECT ip, mac, idordenador FROM ordenadores "
2583                        "WHERE idordenador = %d",
2584                        task.scope);
2585                if (og_queue_task_command(dbi, &task, query)) {
2586                        dbi_result_free(result);
2587                        return -1;
2588                }
2589        }
2590
2591        dbi_result_free(result);
2592
2593        return 0;
2594}
2595
2596int og_dbi_update_action(uint32_t id, bool success)
2597{
2598        char end_date_string[24];
2599        struct tm *end_date;
2600        const char *msglog;
2601        struct og_dbi *dbi;
2602        uint8_t status = 2;
2603        dbi_result result;
2604        time_t now;
2605
2606        if (!id)
2607                return 0;
2608
[fe1ce97]2609        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]2610        if (!dbi) {
2611                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2612                       __func__, __LINE__);
2613                return -1;
2614        }
2615
2616        time(&now);
2617        end_date = localtime(&now);
2618
2619        sprintf(end_date_string, "%hu/%hhu/%hhu %hhu:%hhu:%hhu",
2620                end_date->tm_year + 1900, end_date->tm_mon + 1,
2621                end_date->tm_mday, end_date->tm_hour, end_date->tm_min,
2622                end_date->tm_sec);
2623        result = dbi_conn_queryf(dbi->conn,
2624                                 "UPDATE acciones SET fechahorafin='%s', "
2625                                 "estado=%d, resultado=%d WHERE idaccion=%d",
2626                                 end_date_string, ACCION_FINALIZADA,
2627                                 status - success, id);
2628
2629        if (!result) {
2630                dbi_conn_error(dbi->conn, &msglog);
2631                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2632                       __func__, __LINE__, msglog);
2633                og_dbi_close(dbi);
2634                return -1;
2635        }
2636        dbi_result_free(result);
2637        og_dbi_close(dbi);
2638
2639        return 0;
2640}
2641
2642void og_schedule_run(unsigned int task_id, unsigned int schedule_id,
2643                     enum og_schedule_type type)
2644{
2645        struct og_msg_params params = {};
2646        bool duplicated = false;
2647        struct og_cmd *cmd, *next;
2648        struct og_dbi *dbi;
2649        unsigned int i;
2650
[fe1ce97]2651        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]2652        if (!dbi) {
2653                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2654                       __func__, __LINE__);
2655                return;
2656        }
2657
2658        switch (type) {
2659        case OG_SCHEDULE_TASK:
2660                og_dbi_queue_task(dbi, task_id, schedule_id);
2661                break;
2662        case OG_SCHEDULE_PROCEDURE:
2663        case OG_SCHEDULE_COMMAND:
2664                og_dbi_queue_command(dbi, task_id, schedule_id);
2665                break;
2666        }
2667        og_dbi_close(dbi);
2668
2669        list_for_each_entry(cmd, &cmd_list, list) {
2670                for (i = 0; i < params.ips_array_len; i++) {
2671                        if (!strncmp(cmd->ip, params.ips_array[i],
2672                                     OG_DB_IP_MAXLEN)) {
2673                                duplicated = true;
2674                                break;
2675                        }
2676                }
2677
2678                if (!duplicated)
2679                        params.ips_array[params.ips_array_len++] = cmd->ip;
2680                else
2681                        duplicated = false;
2682        }
2683
2684        list_for_each_entry_safe(cmd, next, &cmd_list, list) {
2685                if (cmd->type != OG_CMD_WOL)
2686                        continue;
2687
2688                if (Levanta((char **)cmd->params.ips_array,
2689                            (char **)cmd->params.mac_array,
[06af0c2]2690                            (char **)cmd->params.netmask_array,
[04ca20e]2691                            cmd->params.ips_array_len,
2692                            (char *)cmd->params.wol_type))
2693                        og_dbi_update_action(cmd->id, true);
2694
2695                list_del(&cmd->list);
2696                og_cmd_free(cmd);
2697        }
2698
2699        og_send_request(OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, &params, NULL);
2700}
2701
2702static int og_cmd_task_post(json_t *element, struct og_msg_params *params)
2703{
2704        struct og_cmd *cmd;
2705        struct og_dbi *dbi;
2706        const char *key;
2707        json_t *value;
2708        int err;
2709
2710        if (json_typeof(element) != JSON_OBJECT)
2711                return -1;
2712
2713        json_object_foreach(element, key, value) {
2714                if (!strcmp(key, "task")) {
2715                        err = og_json_parse_string(value, &params->task_id);
2716                        params->flags |= OG_REST_PARAM_TASK;
2717                }
2718
2719                if (err < 0)
2720                        break;
2721        }
2722
2723        if (!og_msg_params_validate(params, OG_REST_PARAM_TASK))
2724                return -1;
2725
[fe1ce97]2726        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]2727        if (!dbi) {
2728                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2729                           __func__, __LINE__);
2730                return -1;
2731        }
2732
2733        og_schedule_run(atoi(params->task_id), 0, OG_SCHEDULE_TASK);
2734        og_dbi_close(dbi);
2735
2736        list_for_each_entry(cmd, &cmd_list, list)
2737                params->ips_array[params->ips_array_len++] = cmd->ip;
2738
2739        return og_send_request(OG_METHOD_GET, OG_CMD_RUN_SCHEDULE, params,
2740                               NULL);
2741}
2742
[f520a57]2743static int og_dbi_scope_get_computer(struct og_dbi *dbi, json_t *array,
2744                                     uint32_t room_id)
[c46fa3c]2745{
[91c3a28]2746        const char *computer_name, *computer_ip;
[f520a57]2747        uint32_t computer_id;
[3d253e6]2748        const char *msglog;
2749        dbi_result result;
[f520a57]2750        json_t *computer;
[3d253e6]2751
2752        result = dbi_conn_queryf(dbi->conn,
[f520a57]2753                                 "SELECT idordenador, nombreordenador, ip "
2754                                 "FROM ordenadores WHERE idaula=%d",
2755                                 room_id);
[3d253e6]2756        if (!result) {
2757                dbi_conn_error(dbi->conn, &msglog);
2758                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2759                       __func__, __LINE__, msglog);
2760                return -1;
2761        }
2762
2763        while (dbi_result_next_row(result)) {
[f520a57]2764                computer_id = dbi_result_get_uint(result, "idordenador");
[91c3a28]2765                computer_name = dbi_result_get_string(result, "nombreordenador");
2766                computer_ip = dbi_result_get_string(result, "ip");
[3d253e6]2767
[f520a57]2768                computer = json_object();
2769                if (!computer) {
[3d253e6]2770                        dbi_result_free(result);
2771                        return -1;
2772                }
2773
[f520a57]2774                json_object_set_new(computer, "name", json_string(computer_name));
2775                json_object_set_new(computer, "type", json_string("computer"));
2776                json_object_set_new(computer, "id", json_integer(computer_id));
2777                json_object_set_new(computer, "scope", json_array());
[91c3a28]2778                json_object_set_new(computer, "ip", json_string(computer_ip));
[f520a57]2779                json_array_append(array, computer);
2780                json_decref(computer);
[3d253e6]2781        }
2782        dbi_result_free(result);
2783
2784        return 0;
2785}
2786
[c46fa3c]2787static int og_dbi_scope_get_room(struct og_dbi *dbi, json_t *array,
2788                                 uint32_t center_id)
2789{
[3d253e6]2790        char room_name[OG_DB_ROOM_NAME_MAXLEN + 1] = {};
[f520a57]2791        json_t *room, *room_array;
[3d253e6]2792        const char *msglog;
2793        dbi_result result;
2794        uint32_t room_id;
2795
2796        result = dbi_conn_queryf(dbi->conn,
2797                                 "SELECT idaula, nombreaula FROM aulas WHERE "
2798                                 "idcentro=%d",
2799                                 center_id);
2800        if (!result) {
2801                dbi_conn_error(dbi->conn, &msglog);
2802                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2803                       __func__, __LINE__, msglog);
2804                return -1;
2805        }
2806
2807        while (dbi_result_next_row(result)) {
2808                room_id = dbi_result_get_uint(result, "idaula");
2809                strncpy(room_name,
2810                        dbi_result_get_string(result, "nombreaula"),
2811                        OG_DB_CENTER_NAME_MAXLEN);
2812
2813                room = json_object();
2814                if (!room) {
2815                        dbi_result_free(result);
2816                        return -1;
2817                }
2818
2819                json_object_set_new(room, "name", json_string(room_name));
2820                json_object_set_new(room, "type", json_string("room"));
2821                json_object_set_new(room, "id", json_integer(room_id));
2822                json_object_set_new(room, "scope", json_array());
2823                json_array_append(array, room);
2824                json_decref(room);
[f520a57]2825
2826                room_array = json_object_get(room, "scope");
2827                if (!room_array) {
2828                        dbi_result_free(result);
2829                        return -1;
2830                }
2831
2832                if (og_dbi_scope_get_computer(dbi, room_array, room_id)) {
2833                        dbi_result_free(result);
2834                        return -1;
2835                }
[3d253e6]2836        }
2837        dbi_result_free(result);
2838
2839        return 0;
2840}
2841
[f520a57]2842static int og_dbi_scope_get(struct og_dbi *dbi, json_t *array)
[c46fa3c]2843{
[f520a57]2844        char center_name[OG_DB_CENTER_NAME_MAXLEN + 1] = {};
2845        json_t *center, *array_room;
[3d253e6]2846        const char *msglog;
[f520a57]2847        uint32_t center_id;
[3d253e6]2848        dbi_result result;
2849
2850        result = dbi_conn_queryf(dbi->conn,
[f520a57]2851                                 "SELECT nombrecentro, idcentro FROM centros");
[3d253e6]2852        if (!result) {
2853                dbi_conn_error(dbi->conn, &msglog);
2854                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2855                       __func__, __LINE__, msglog);
2856                return -1;
2857        }
2858
2859        while (dbi_result_next_row(result)) {
[f520a57]2860                center_id = dbi_result_get_uint(result, "idcentro");
2861                strncpy(center_name,
2862                        dbi_result_get_string(result, "nombrecentro"),
[3d253e6]2863                        OG_DB_CENTER_NAME_MAXLEN);
2864
[f520a57]2865                center = json_object();
2866                if (!center) {
[3d253e6]2867                        dbi_result_free(result);
2868                        return -1;
2869                }
2870
[f520a57]2871                array_room = json_array();
2872                if (!array_room) {
2873                        dbi_result_free(result);
2874                        json_decref(center);
2875                        return -1;
2876                }
2877
2878                json_object_set_new(center, "name", json_string(center_name));
2879                json_object_set_new(center, "type", json_string("center"));
2880                json_object_set_new(center, "id", json_integer(center_id));
2881                json_object_set_new(center, "scope", array_room);
2882                json_array_append(array, center);
2883                json_decref(center);
2884
2885                if (og_dbi_scope_get_room(dbi, array_room, center_id)) {
2886                        dbi_result_free(result);
2887                        return -1;
2888                }
[3d253e6]2889        }
[f520a57]2890
[3d253e6]2891        dbi_result_free(result);
2892
2893        return 0;
2894}
2895
2896static int og_cmd_scope_get(json_t *element, struct og_msg_params *params,
2897                            char *buffer_reply)
2898{
2899        struct og_buffer og_buffer = {
2900                .data = buffer_reply
2901        };
[f520a57]2902        json_t *root, *array;
2903        struct og_dbi *dbi;
[3d253e6]2904
2905        root = json_object();
[76c0a05]2906        if (!root)
2907                return -1;
2908
[f520a57]2909        array = json_array();
2910        if (!array) {
[76c0a05]2911                json_decref(root);
[3d253e6]2912                return -1;
[76c0a05]2913        }
[f520a57]2914        json_object_set_new(root, "scope", array);
[3d253e6]2915
[fe1ce97]2916        dbi = og_dbi_open(&ogconfig.db);
[3d253e6]2917        if (!dbi) {
2918                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2919                       __func__, __LINE__);
[f520a57]2920                json_decref(root);
[3d253e6]2921                return -1;
2922        }
2923
[f520a57]2924        if (og_dbi_scope_get(dbi, array)) {
[3d253e6]2925                og_dbi_close(dbi);
[f520a57]2926                json_decref(root);
[3d253e6]2927                return -1;
2928        }
2929
2930        og_dbi_close(dbi);
2931
2932        json_dump_callback(root, og_json_dump_clients, &og_buffer, 0);
2933        json_decref(root);
2934
2935        return 0;
2936}
2937
[04ca20e]2938int og_dbi_schedule_get(void)
2939{
2940        uint32_t schedule_id, task_id;
2941        struct og_schedule_time time;
2942        struct og_dbi *dbi;
2943        const char *msglog;
2944        dbi_result result;
2945
[fe1ce97]2946        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]2947        if (!dbi) {
2948                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
2949                       __func__, __LINE__);
2950                return -1;
2951        }
2952
2953        result = dbi_conn_queryf(dbi->conn,
2954                                 "SELECT idprogramacion, tipoaccion, identificador, "
2955                                 "sesion, annos, meses, diario, dias, semanas, horas, "
2956                                 "ampm, minutos FROM programaciones "
2957                                 "WHERE suspendida = 0");
2958        if (!result) {
2959                dbi_conn_error(dbi->conn, &msglog);
2960                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
2961                       __func__, __LINE__, msglog);
2962                og_dbi_close(dbi);
2963                return -1;
2964        }
2965
2966        while (dbi_result_next_row(result)) {
2967                memset(&time, 0, sizeof(time));
2968                schedule_id = dbi_result_get_uint(result, "idprogramacion");
2969                task_id = dbi_result_get_uint(result, "identificador");
2970                time.years = dbi_result_get_uint(result, "annos");
2971                time.months = dbi_result_get_uint(result, "meses");
2972                time.weeks = dbi_result_get_uint(result, "semanas");
2973                time.week_days = dbi_result_get_uint(result, "dias");
2974                time.days = dbi_result_get_uint(result, "diario");
2975                time.hours = dbi_result_get_uint(result, "horas");
2976                time.am_pm = dbi_result_get_uint(result, "ampm");
2977                time.minutes = dbi_result_get_uint(result, "minutos");
2978                time.on_start = true;
2979
2980                og_schedule_create(schedule_id, task_id, OG_SCHEDULE_TASK,
2981                                   &time);
2982        }
2983
2984        dbi_result_free(result);
2985        og_dbi_close(dbi);
2986
2987        return 0;
2988}
2989
2990static int og_dbi_schedule_create(struct og_dbi *dbi,
2991                                  struct og_msg_params *params,
2992                                  uint32_t *schedule_id,
2993                                  enum og_schedule_type schedule_type)
2994{
2995        uint8_t suspended = 0;
2996        uint32_t session = 0;
2997        const char *msglog;
2998        dbi_result result;
2999        uint8_t type;
3000
3001        switch (schedule_type) {
3002        case OG_SCHEDULE_TASK:
3003                type = 3;
3004                break;
3005        case OG_SCHEDULE_PROCEDURE:
3006                type = 2;
3007                break;
3008        case OG_SCHEDULE_COMMAND:
3009                session = atoi(params->task_id);
3010                type = 1;
3011                break;
3012        }
3013
3014        result = dbi_conn_queryf(dbi->conn,
3015                                 "INSERT INTO programaciones (tipoaccion,"
3016                                 " identificador, nombrebloque, annos, meses,"
3017                                 " semanas, dias, diario, horas, ampm, minutos,"
3018                                 " suspendida, sesion) VALUES (%d, %s, '%s',"
3019                                 " %d, %d, %d, %d, %d, %d, %d, %d, %d, %d)",
3020                                 type, params->task_id, params->name,
3021                                 params->time.years, params->time.months,
3022                                 params->time.weeks, params->time.week_days,
3023                                 params->time.days, params->time.hours,
3024                                 params->time.am_pm, params->time.minutes,
3025                                 suspended, session);
3026        if (!result) {
3027                dbi_conn_error(dbi->conn, &msglog);
3028                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3029                       __func__, __LINE__, msglog);
3030                return -1;
3031        }
3032        dbi_result_free(result);
3033
3034        *schedule_id = dbi_conn_sequence_last(dbi->conn, NULL);
3035
3036        return 0;
3037}
3038
3039static int og_dbi_schedule_update(struct og_dbi *dbi,
3040                                  struct og_msg_params *params)
3041{
3042        const char *msglog;
3043        dbi_result result;
3044        uint8_t type = 3;
3045
3046        result = dbi_conn_queryf(dbi->conn,
3047                                 "UPDATE programaciones SET tipoaccion=%d, "
3048                                 "identificador='%s', nombrebloque='%s', "
3049                                 "annos=%d, meses=%d, "
3050                                 "diario=%d, horas=%d, ampm=%d, minutos=%d "
3051                                 "WHERE idprogramacion='%s'",
3052                                 type, params->task_id, params->name,
3053                                 params->time.years, params->time.months,
3054                                 params->time.days, params->time.hours,
3055                                 params->time.am_pm, params->time.minutes,
3056                                 params->id);
3057
3058        if (!result) {
3059                dbi_conn_error(dbi->conn, &msglog);
3060                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3061                       __func__, __LINE__, msglog);
3062                return -1;
3063        }
3064        dbi_result_free(result);
3065
3066        return 0;
3067}
3068
3069static int og_dbi_schedule_delete(struct og_dbi *dbi, uint32_t id)
3070{
3071        const char *msglog;
3072        dbi_result result;
3073
3074        result = dbi_conn_queryf(dbi->conn,
3075                                 "DELETE FROM programaciones WHERE idprogramacion=%d",
3076                                 id);
3077        if (!result) {
3078                dbi_conn_error(dbi->conn, &msglog);
3079                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3080                       __func__, __LINE__, msglog);
3081                return -1;
3082        }
3083        dbi_result_free(result);
3084
3085        return 0;
3086}
3087
3088struct og_db_schedule {
3089        uint32_t                id;
3090        uint32_t                task_id;
3091        const char              *name;
3092        struct og_schedule_time time;
3093        uint32_t                week_days;
3094        uint32_t                weeks;
3095        uint32_t                suspended;
3096        uint32_t                session;
3097};
3098
3099static int og_dbi_schedule_get_json(struct og_dbi *dbi, json_t *root,
3100                                    const char *task_id, const char *schedule_id)
3101{
3102        struct og_db_schedule schedule;
3103        json_t *obj, *array;
3104        const char *msglog;
3105        dbi_result result;
3106        int err = 0;
3107
3108        if (task_id) {
3109                result = dbi_conn_queryf(dbi->conn,
3110                                         "SELECT idprogramacion,"
3111                                         "       identificador, nombrebloque,"
3112                                         "       annos, meses, diario, dias,"
3113                                         "       semanas, horas, ampm,"
3114                                         "       minutos,suspendida, sesion "
3115                                         "FROM programaciones "
3116                                         "WHERE identificador=%d",
3117                                         atoi(task_id));
3118        } else if (schedule_id) {
3119                result = dbi_conn_queryf(dbi->conn,
3120                                         "SELECT idprogramacion,"
3121                                         "       identificador, nombrebloque,"
3122                                         "       annos, meses, diario, dias,"
3123                                         "       semanas, horas, ampm,"
3124                                         "       minutos,suspendida, sesion "
3125                                         "FROM programaciones "
3126                                         "WHERE idprogramacion=%d",
3127                                         atoi(schedule_id));
3128        } else {
3129                result = dbi_conn_queryf(dbi->conn,
3130                                         "SELECT idprogramacion,"
3131                                         "       identificador, nombrebloque,"
3132                                         "       annos, meses, diario, dias,"
3133                                         "       semanas, horas, ampm,"
3134                                         "       minutos,suspendida, sesion "
3135                                         "FROM programaciones");
3136        }
3137
3138        if (!result) {
3139                dbi_conn_error(dbi->conn, &msglog);
3140                syslog(LOG_ERR, "failed to query database (%s:%d) %s\n",
3141                       __func__, __LINE__, msglog);
3142                return -1;
3143        }
3144
3145        array = json_array();
3146        if (!array)
3147                return -1;
3148
3149        while (dbi_result_next_row(result)) {
3150                schedule.id = dbi_result_get_uint(result, "idprogramacion");
3151                schedule.task_id = dbi_result_get_uint(result, "identificador");
3152                schedule.name = dbi_result_get_string(result, "nombrebloque");
3153                schedule.time.years = dbi_result_get_uint(result, "annos");
3154                schedule.time.months = dbi_result_get_uint(result, "meses");
3155                schedule.time.days = dbi_result_get_uint(result, "diario");
3156                schedule.time.hours = dbi_result_get_uint(result, "horas");
3157                schedule.time.am_pm = dbi_result_get_uint(result, "ampm");
3158                schedule.time.minutes = dbi_result_get_uint(result, "minutos");
3159                schedule.week_days = dbi_result_get_uint(result, "dias");
3160                schedule.weeks = dbi_result_get_uint(result, "semanas");
3161                schedule.suspended = dbi_result_get_uint(result, "suspendida");
3162                schedule.session = dbi_result_get_uint(result, "sesion");
3163
3164                obj = json_object();
3165                if (!obj) {
3166                        err = -1;
3167                        break;
3168                }
3169                json_object_set_new(obj, "id", json_integer(schedule.id));
3170                json_object_set_new(obj, "task", json_integer(schedule.task_id));
3171                json_object_set_new(obj, "name", json_string(schedule.name));
3172                json_object_set_new(obj, "years", json_integer(schedule.time.years));
3173                json_object_set_new(obj, "months", json_integer(schedule.time.months));
3174                json_object_set_new(obj, "days", json_integer(schedule.time.days));
3175                json_object_set_new(obj, "hours", json_integer(schedule.time.hours));
3176                json_object_set_new(obj, "am_pm", json_integer(schedule.time.am_pm));
3177                json_object_set_new(obj, "minutes", json_integer(schedule.time.minutes));
3178                json_object_set_new(obj, "week_days", json_integer(schedule.week_days));
3179                json_object_set_new(obj, "weeks", json_integer(schedule.weeks));
3180                json_object_set_new(obj, "suspended", json_integer(schedule.suspended));
3181                json_object_set_new(obj, "session", json_integer(schedule.session));
3182
3183                json_array_append_new(array, obj);
3184        }
3185
3186        json_object_set_new(root, "schedule", array);
3187
3188        dbi_result_free(result);
3189
3190        return err;
3191}
3192
3193static int og_task_schedule_create(struct og_msg_params *params)
3194{
3195        enum og_schedule_type type;
3196        uint32_t schedule_id;
3197        struct og_dbi *dbi;
3198        int err;
3199
3200        if (!strcmp(params->type, "task"))
3201                type = OG_SCHEDULE_TASK;
3202        else if (!strcmp(params->type, "procedure"))
3203                type = OG_SCHEDULE_PROCEDURE;
3204        else if (!strcmp(params->type, "command"))
3205                type = OG_SCHEDULE_COMMAND;
3206        else
3207                return -1;
3208
[fe1ce97]3209        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]3210        if (!dbi) {
3211                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3212                       __func__, __LINE__);
3213                return -1;
3214        }
3215
3216        err = og_dbi_schedule_create(dbi, params, &schedule_id, type);
3217        if (err < 0) {
3218                og_dbi_close(dbi);
3219                return -1;
3220        }
3221        og_schedule_create(schedule_id, atoi(params->task_id), type,
3222                           &params->time);
3223        og_schedule_refresh(og_loop);
3224        og_dbi_close(dbi);
3225
3226        return 0;
3227}
3228
3229static int og_cmd_schedule_create(json_t *element, struct og_msg_params *params)
3230{
3231        const char *key;
3232        json_t *value;
3233        int err;
3234
3235        if (json_typeof(element) != JSON_OBJECT)
3236                return -1;
3237
3238        json_object_foreach(element, key, value) {
3239                if (!strcmp(key, "task")) {
3240                        err = og_json_parse_string(value, &params->task_id);
3241                        params->flags |= OG_REST_PARAM_TASK;
3242                } else if (!strcmp(key, "name")) {
3243                        err = og_json_parse_string(value, &params->name);
3244                        params->flags |= OG_REST_PARAM_NAME;
3245                } else if (!strcmp(key, "when")) {
3246                        err = og_json_parse_time_params(value, params);
3247                } else if (!strcmp(key, "type")) {
3248                        err = og_json_parse_string(value, &params->type);
3249                        params->flags |= OG_REST_PARAM_TYPE;
3250                }
3251
3252                if (err < 0)
3253                        break;
3254        }
3255
3256        if (!og_msg_params_validate(params, OG_REST_PARAM_TASK |
3257                                            OG_REST_PARAM_NAME |
3258                                            OG_REST_PARAM_TIME_YEARS |
3259                                            OG_REST_PARAM_TIME_MONTHS |
3260                                            OG_REST_PARAM_TIME_WEEKS |
3261                                            OG_REST_PARAM_TIME_WEEK_DAYS |
3262                                            OG_REST_PARAM_TIME_DAYS |
3263                                            OG_REST_PARAM_TIME_HOURS |
3264                                            OG_REST_PARAM_TIME_MINUTES |
3265                                            OG_REST_PARAM_TIME_AM_PM |
3266                                            OG_REST_PARAM_TYPE))
3267                return -1;
3268
3269        return og_task_schedule_create(params);
3270}
3271
3272static int og_cmd_schedule_update(json_t *element, struct og_msg_params *params)
3273{
3274        struct og_dbi *dbi;
3275        const char *key;
3276        json_t *value;
3277        int err;
3278
3279        if (json_typeof(element) != JSON_OBJECT)
3280                return -1;
3281
3282        json_object_foreach(element, key, value) {
3283                if (!strcmp(key, "id")) {
3284                        err = og_json_parse_string(value, &params->id);
3285                        params->flags |= OG_REST_PARAM_ID;
3286                } else if (!strcmp(key, "task")) {
3287                        err = og_json_parse_string(value, &params->task_id);
3288                        params->flags |= OG_REST_PARAM_TASK;
3289                } else if (!strcmp(key, "name")) {
3290                        err = og_json_parse_string(value, &params->name);
3291                        params->flags |= OG_REST_PARAM_NAME;
3292                } else if (!strcmp(key, "when"))
3293                        err = og_json_parse_time_params(value, params);
3294
3295                if (err < 0)
3296                        break;
3297        }
3298
3299        if (!og_msg_params_validate(params, OG_REST_PARAM_ID |
3300                                            OG_REST_PARAM_TASK |
3301                                            OG_REST_PARAM_NAME |
3302                                            OG_REST_PARAM_TIME_YEARS |
3303                                            OG_REST_PARAM_TIME_MONTHS |
3304                                            OG_REST_PARAM_TIME_DAYS |
3305                                            OG_REST_PARAM_TIME_HOURS |
3306                                            OG_REST_PARAM_TIME_MINUTES |
3307                                            OG_REST_PARAM_TIME_AM_PM))
3308                return -1;
3309
[fe1ce97]3310        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]3311        if (!dbi) {
3312                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3313                           __func__, __LINE__);
3314                return -1;
3315        }
3316
3317        err = og_dbi_schedule_update(dbi, params);
3318        og_dbi_close(dbi);
3319
3320        if (err < 0)
3321                return err;
3322
3323        og_schedule_update(og_loop, atoi(params->id), atoi(params->task_id),
3324                           &params->time);
3325        og_schedule_refresh(og_loop);
3326
3327        return err;
3328}
3329
3330static int og_cmd_schedule_delete(json_t *element, struct og_msg_params *params)
3331{
3332        struct og_dbi *dbi;
3333        const char *key;
3334        json_t *value;
3335        int err;
3336
3337        if (json_typeof(element) != JSON_OBJECT)
3338                return -1;
3339
3340        json_object_foreach(element, key, value) {
3341                if (!strcmp(key, "id")) {
3342                        err = og_json_parse_string(value, &params->id);
3343                        params->flags |= OG_REST_PARAM_ID;
3344                } else {
3345                        return -1;
3346                }
3347
3348                if (err < 0)
3349                        break;
3350        }
3351
3352        if (!og_msg_params_validate(params, OG_REST_PARAM_ID))
3353                return -1;
3354
[fe1ce97]3355        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]3356        if (!dbi) {
3357                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3358                           __func__, __LINE__);
3359                return -1;
3360        }
3361
3362        err = og_dbi_schedule_delete(dbi, atoi(params->id));
3363        og_dbi_close(dbi);
3364
3365        og_schedule_delete(og_loop, atoi(params->id));
3366
3367        return err;
3368}
3369
3370static int og_cmd_schedule_get(json_t *element, struct og_msg_params *params,
3371                               char *buffer_reply)
3372{
3373        struct og_buffer og_buffer = {
3374                .data   = buffer_reply,
3375        };
3376        json_t *schedule_root;
3377        struct og_dbi *dbi;
3378        const char *key;
3379        json_t *value;
3380        int err;
3381
3382        if (element) {
3383                if (json_typeof(element) != JSON_OBJECT)
3384                        return -1;
3385
3386                json_object_foreach(element, key, value) {
3387                        if (!strcmp(key, "task")) {
3388                                err = og_json_parse_string(value,
3389                                                           &params->task_id);
3390                        } else if (!strcmp(key, "id")) {
3391                                err = og_json_parse_string(value, &params->id);
3392                        } else {
3393                                return -1;
3394                        }
3395
3396                        if (err < 0)
3397                                break;
3398                }
3399        }
3400
[fe1ce97]3401        dbi = og_dbi_open(&ogconfig.db);
[04ca20e]3402        if (!dbi) {
3403                syslog(LOG_ERR, "cannot open connection database (%s:%d)\n",
3404                           __func__, __LINE__);
3405                return -1;
3406        }
3407
3408        schedule_root = json_object();
3409        if (!schedule_root) {
3410                og_dbi_close(dbi);
3411                return -1;
3412        }
3413
3414        err = og_dbi_schedule_get_json(dbi, schedule_root,
3415                                       params->task_id, params->id);
3416        og_dbi_close(dbi);
3417
3418        if (err >= 0)
3419                json_dump_callback(schedule_root, og_json_dump_clients, &og_buffer, 0);
3420
3421        json_decref(schedule_root);
3422
3423        return err;
3424}
3425
3426static int og_client_method_not_found(struct og_client *cli)
3427{
3428        /* To meet RFC 7231, this function MUST generate an Allow header field
3429         * containing the correct methods. For example: "Allow: POST\r\n"
3430         */
3431        char buf[] = "HTTP/1.1 405 Method Not Allowed\r\n"
3432                     "Content-Length: 0\r\n\r\n";
3433
3434        send(og_client_socket(cli), buf, strlen(buf), 0);
3435
3436        return -1;
3437}
3438
3439static int og_client_bad_request(struct og_client *cli)
3440{
3441        char buf[] = "HTTP/1.1 400 Bad Request\r\nContent-Length: 0\r\n\r\n";
3442
3443        send(og_client_socket(cli), buf, strlen(buf), 0);
3444
3445        return -1;
3446}
3447
3448static int og_client_not_found(struct og_client *cli)
3449{
3450        char buf[] = "HTTP/1.1 404 Not Found\r\nContent-Length: 0\r\n\r\n";
3451
3452        send(og_client_socket(cli), buf, strlen(buf), 0);
3453
3454        return -1;
3455}
3456
3457static int og_client_not_authorized(struct og_client *cli)
3458{
3459        char buf[] = "HTTP/1.1 401 Unauthorized\r\n"
3460                     "WWW-Authenticate: Basic\r\n"
3461                     "Content-Length: 0\r\n\r\n";
3462
3463        send(og_client_socket(cli), buf, strlen(buf), 0);
3464
3465        return -1;
3466}
3467
3468static int og_server_internal_error(struct og_client *cli)
3469{
3470        char buf[] = "HTTP/1.1 500 Internal Server Error\r\n"
3471                     "Content-Length: 0\r\n\r\n";
3472
3473        send(og_client_socket(cli), buf, strlen(buf), 0);
3474
3475        return -1;
3476}
3477
3478#define OG_MSG_RESPONSE_MAXLEN  65536
3479
3480static int og_client_ok(struct og_client *cli, char *buf_reply)
3481{
3482        char buf[OG_MSG_RESPONSE_MAXLEN] = {};
3483        int err = 0, len;
3484
3485        len = snprintf(buf, sizeof(buf),
3486                       "HTTP/1.1 200 OK\r\nContent-Length: %ld\r\n\r\n%s",
3487                       strlen(buf_reply), buf_reply);
3488        if (len >= (int)sizeof(buf))
3489                err = og_server_internal_error(cli);
3490
3491        send(og_client_socket(cli), buf, strlen(buf), 0);
3492
3493        return err;
3494}
3495
3496int og_client_state_process_payload_rest(struct og_client *cli)
3497{
3498        char buf_reply[OG_MSG_RESPONSE_MAXLEN] = {};
3499        struct og_msg_params params = {};
3500        enum og_rest_method method;
3501        const char *cmd, *body;
3502        json_error_t json_err;
3503        json_t *root = NULL;
3504        int err = 0;
3505
3506        syslog(LOG_DEBUG, "%s:%hu %.32s ...\n",
3507               inet_ntoa(cli->addr.sin_addr),
3508               ntohs(cli->addr.sin_port), cli->buf);
3509
3510        if (!strncmp(cli->buf, "GET", strlen("GET"))) {
3511                method = OG_METHOD_GET;
3512                cmd = cli->buf + strlen("GET") + 2;
3513        } else if (!strncmp(cli->buf, "POST", strlen("POST"))) {
3514                method = OG_METHOD_POST;
3515                cmd = cli->buf + strlen("POST") + 2;
3516        } else
3517                return og_client_method_not_found(cli);
3518
3519        body = strstr(cli->buf, "\r\n\r\n") + 4;
3520
[fe1ce97]3521        if (strcmp(cli->auth_token, ogconfig.rest.api_token)) {
[04ca20e]3522                syslog(LOG_ERR, "wrong Authentication key\n");
3523                return og_client_not_authorized(cli);
3524        }
3525
3526        if (cli->content_length) {
3527                root = json_loads(body, 0, &json_err);
3528                if (!root) {
3529                        syslog(LOG_ERR, "malformed json line %d: %s\n",
3530                               json_err.line, json_err.text);
3531                        return og_client_not_found(cli);
3532                }
3533        }
3534
3535        if (!strncmp(cmd, "clients", strlen("clients"))) {
3536                if (method != OG_METHOD_POST &&
3537                    method != OG_METHOD_GET)
3538                        return og_client_method_not_found(cli);
3539
3540                if (method == OG_METHOD_POST && !root) {
3541                        syslog(LOG_ERR, "command clients with no payload\n");
3542                        return og_client_bad_request(cli);
3543                }
3544                switch (method) {
3545                case OG_METHOD_POST:
3546                        err = og_cmd_post_clients(root, &params);
3547                        break;
3548                case OG_METHOD_GET:
3549                        err = og_cmd_get_clients(root, &params, buf_reply);
3550                        break;
3551                default:
3552                        return og_client_bad_request(cli);
3553                }
[b59ff7c]3554        } else if (!strncmp(cmd, "client/setup",
3555                            strlen("client/setup"))) {
3556                if (method != OG_METHOD_GET)
3557                        return og_client_method_not_found(cli);
3558
3559                if (!root) {
3560                        syslog(LOG_ERR,
3561                               "command client partitions with no payload\n");
3562                        return og_client_bad_request(cli);
3563                }
3564
3565                err = og_cmd_get_client_setup(root, &params, buf_reply);
[af47a08]3566        } else if (!strncmp(cmd, "client/info",
3567                            strlen("client/info"))) {
3568                if (method != OG_METHOD_GET)
3569                        return og_client_method_not_found(cli);
3570
3571                if (!root) {
3572                        syslog(LOG_ERR,
3573                               "command client info with no payload\n");
3574                        return og_client_bad_request(cli);
3575                }
3576
3577                err = og_cmd_get_client_info(root, &params, buf_reply);
[65a14e5]3578        } else if (!strncmp(cmd, "client/add", strlen("client/add"))) {
3579                if (method != OG_METHOD_POST)
3580                        return og_client_method_not_found(cli);
3581
3582                if (!root) {
3583                        syslog(LOG_ERR,
3584                               "command client info with no payload\n");
3585                        return og_client_bad_request(cli);
3586                }
3587
3588                err = og_cmd_post_client_add(root, &params, buf_reply);
[24c8b94]3589        } else if (!strncmp(cmd, "client/delete", strlen("client/delete"))) {
3590                if (method != OG_METHOD_POST)
3591                        return og_client_method_not_found(cli);
3592
3593                if (!root) {
3594                        syslog(LOG_ERR,
3595                               "command client delete with no payload\n");
3596                        return og_client_bad_request(cli);
3597                }
3598
3599                err = og_cmd_post_client_delete(root, &params);
[04ca20e]3600        } else if (!strncmp(cmd, "wol", strlen("wol"))) {
3601                if (method != OG_METHOD_POST)
3602                        return og_client_method_not_found(cli);
3603
3604                if (!root) {
3605                        syslog(LOG_ERR, "command wol with no payload\n");
3606                        return og_client_bad_request(cli);
3607                }
3608                err = og_cmd_wol(root, &params);
3609        } else if (!strncmp(cmd, "shell/run", strlen("shell/run"))) {
3610                if (method != OG_METHOD_POST)
3611                        return og_client_method_not_found(cli);
3612
3613                if (!root) {
3614                        syslog(LOG_ERR, "command run with no payload\n");
3615                        return og_client_bad_request(cli);
3616                }
3617                err = og_cmd_run_post(root, &params);
3618        } else if (!strncmp(cmd, "shell/output", strlen("shell/output"))) {
3619                if (method != OG_METHOD_POST)
3620                        return og_client_method_not_found(cli);
3621
3622                if (!root) {
3623                        syslog(LOG_ERR, "command output with no payload\n");
3624                        return og_client_bad_request(cli);
3625                }
3626
3627                err = og_cmd_run_get(root, &params, buf_reply);
3628        } else if (!strncmp(cmd, "session", strlen("session"))) {
3629                if (method != OG_METHOD_POST)
3630                        return og_client_method_not_found(cli);
3631
3632                if (!root) {
3633                        syslog(LOG_ERR, "command session with no payload\n");
3634                        return og_client_bad_request(cli);
3635                }
3636                err = og_cmd_session(root, &params);
[3d253e6]3637        } else if (!strncmp(cmd, "scopes", strlen("scopes"))) {
3638                if (method != OG_METHOD_GET)
3639                        return og_client_method_not_found(cli);
3640
3641                err = og_cmd_scope_get(root, &params, buf_reply);
[04ca20e]3642        } else if (!strncmp(cmd, "poweroff", strlen("poweroff"))) {
3643                if (method != OG_METHOD_POST)
3644                        return og_client_method_not_found(cli);
3645
3646                if (!root) {
3647                        syslog(LOG_ERR, "command poweroff with no payload\n");
3648                        return og_client_bad_request(cli);
3649                }
3650                err = og_cmd_poweroff(root, &params);
3651        } else if (!strncmp(cmd, "reboot", strlen("reboot"))) {
3652                if (method != OG_METHOD_POST)
3653                        return og_client_method_not_found(cli);
3654
3655                if (!root) {
3656                        syslog(LOG_ERR, "command reboot with no payload\n");
3657                        return og_client_bad_request(cli);
3658                }
3659                err = og_cmd_reboot(root, &params);
[b8a509b]3660        } else if (!strncmp(cmd, "mode", strlen("mode"))) {
[33b0c6f]3661                if (method != OG_METHOD_GET && method != OG_METHOD_POST)
[608709f]3662                        return og_client_method_not_found(cli);
3663
[33b0c6f]3664                if (method == OG_METHOD_POST && !root) {
[b8a509b]3665                        syslog(LOG_ERR, "command mode with no payload\n");
[33b0c6f]3666                        return og_client_bad_request(cli);
3667                }
3668
3669                if (method == OG_METHOD_GET)
3670                        err = og_cmd_get_modes(root, &params, buf_reply);
3671                else if (method == OG_METHOD_POST)
3672                        err = og_cmd_post_modes(root, &params);
[04ca20e]3673        } else if (!strncmp(cmd, "stop", strlen("stop"))) {
3674                if (method != OG_METHOD_POST)
3675                        return og_client_method_not_found(cli);
3676
3677                if (!root) {
3678                        syslog(LOG_ERR, "command stop with no payload\n");
3679                        return og_client_bad_request(cli);
3680                }
3681                err = og_cmd_stop(root, &params);
3682        } else if (!strncmp(cmd, "refresh", strlen("refresh"))) {
3683                if (method != OG_METHOD_POST)
3684                        return og_client_method_not_found(cli);
3685
3686                if (!root) {
3687                        syslog(LOG_ERR, "command refresh with no payload\n");
3688                        return og_client_bad_request(cli);
3689                }
3690                err = og_cmd_refresh(root, &params);
3691        } else if (!strncmp(cmd, "hardware", strlen("hardware"))) {
[a8eccba]3692                if (method != OG_METHOD_GET && method != OG_METHOD_POST)
[04ca20e]3693                        return og_client_method_not_found(cli);
3694
3695                if (!root) {
3696                        syslog(LOG_ERR, "command hardware with no payload\n");
3697                        return og_client_bad_request(cli);
3698                }
[a8eccba]3699
3700                if (method == OG_METHOD_GET)
3701                        err = og_cmd_get_hardware(root, &params, buf_reply);
3702                else if (method == OG_METHOD_POST)
3703                        err = og_cmd_hardware(root, &params);
[04ca20e]3704        } else if (!strncmp(cmd, "software", strlen("software"))) {
3705                if (method != OG_METHOD_POST)
3706                        return og_client_method_not_found(cli);
3707
3708                if (!root) {
3709                        syslog(LOG_ERR, "command software with no payload\n");
3710                        return og_client_bad_request(cli);
3711                }
3712                err = og_cmd_software(root, &params);
[403e7c3]3713        } else if (!strncmp(cmd, "images", strlen("images"))) {
3714                if (method != OG_METHOD_GET)
3715                        return og_client_method_not_found(cli);
3716
3717                if (!root)
3718                        err = og_cmd_images(buf_reply);
3719                else
3720                        return og_client_method_not_found(cli);
[04ca20e]3721        } else if (!strncmp(cmd, "image/create", strlen("image/create"))) {
3722                if (method != OG_METHOD_POST)
3723                        return og_client_method_not_found(cli);
3724
3725                if (!root) {
3726                        syslog(LOG_ERR, "command create with no payload\n");
3727                        return og_client_bad_request(cli);
3728                }
3729                err = og_cmd_create_image(root, &params);
3730        } else if (!strncmp(cmd, "image/restore", strlen("image/restore"))) {
3731                if (method != OG_METHOD_POST)
3732                        return og_client_method_not_found(cli);
3733
3734                if (!root) {
3735                        syslog(LOG_ERR, "command create with no payload\n");
3736                        return og_client_bad_request(cli);
3737                }
3738                err = og_cmd_restore_image(root, &params);
3739        } else if (!strncmp(cmd, "setup", strlen("setup"))) {
3740                if (method != OG_METHOD_POST)
3741                        return og_client_method_not_found(cli);
3742
3743                if (!root) {
3744                        syslog(LOG_ERR, "command create with no payload\n");
3745                        return og_client_bad_request(cli);
3746                }
3747                err = og_cmd_setup(root, &params);
3748        } else if (!strncmp(cmd, "run/schedule", strlen("run/schedule"))) {
3749                if (method != OG_METHOD_POST)
3750                        return og_client_method_not_found(cli);
3751
3752                if (!root) {
3753                        syslog(LOG_ERR, "command create with no payload\n");
3754                        return og_client_bad_request(cli);
3755                }
3756
3757                err = og_cmd_run_schedule(root, &params);
3758        } else if (!strncmp(cmd, "task/run", strlen("task/run"))) {
3759                if (method != OG_METHOD_POST)
3760                        return og_client_method_not_found(cli);
3761
3762                if (!root) {
3763                        syslog(LOG_ERR, "command task with no payload\n");
3764                        return og_client_bad_request(cli);
3765                }
3766                err = og_cmd_task_post(root, &params);
3767        } else if (!strncmp(cmd, "schedule/create",
3768                            strlen("schedule/create"))) {
3769                if (method != OG_METHOD_POST)
3770                        return og_client_method_not_found(cli);
3771
3772                if (!root) {
3773                        syslog(LOG_ERR, "command task with no payload\n");
3774                        return og_client_bad_request(cli);
3775                }
3776                err = og_cmd_schedule_create(root, &params);
3777        } else if (!strncmp(cmd, "schedule/delete",
3778                            strlen("schedule/delete"))) {
3779                if (method != OG_METHOD_POST)
3780                        return og_client_method_not_found(cli);
3781
3782                if (!root) {
3783                        syslog(LOG_ERR, "command task with no payload\n");
3784                        return og_client_bad_request(cli);
3785                }
3786                err = og_cmd_schedule_delete(root, &params);
3787        } else if (!strncmp(cmd, "schedule/update",
3788                            strlen("schedule/update"))) {
3789                if (method != OG_METHOD_POST)
3790                        return og_client_method_not_found(cli);
3791
3792                if (!root) {
3793                        syslog(LOG_ERR, "command task with no payload\n");
3794                        return og_client_bad_request(cli);
3795                }
3796                err = og_cmd_schedule_update(root, &params);
3797        } else if (!strncmp(cmd, "schedule/get",
3798                            strlen("schedule/get"))) {
3799                if (method != OG_METHOD_POST)
3800                        return og_client_method_not_found(cli);
3801
3802                err = og_cmd_schedule_get(root, &params, buf_reply);
3803        } else {
3804                syslog(LOG_ERR, "unknown command: %.32s ...\n", cmd);
3805                err = og_client_not_found(cli);
3806        }
3807
3808        if (root)
3809                json_decref(root);
3810
3811        if (err < 0)
3812                return og_client_bad_request(cli);
3813
3814        err = og_client_ok(cli, buf_reply);
3815        if (err < 0) {
3816                syslog(LOG_ERR, "HTTP response to %s:%hu is too large\n",
3817                       inet_ntoa(cli->addr.sin_addr),
3818                       ntohs(cli->addr.sin_port));
3819        }
3820
3821        return err;
3822}
Note: See TracBrowser for help on using the repository browser.