1 | // ******************************************************************************************************* |
---|
2 | // Servicio: ogAdmServer |
---|
3 | // Autor: José Manuel Alonso (E.T.S.I.I.) Universidad de Sevilla |
---|
4 | // Fecha Creación: Marzo-2010 |
---|
5 | // Fecha Última modificación: Marzo-2010 |
---|
6 | // Nombre del fichero: ogAdmServer.cpp |
---|
7 | // Descripción :Este fichero implementa el servicio de administración general del sistema |
---|
8 | // ******************************************************************************************************* |
---|
9 | #include "ogAdmServer.h" |
---|
10 | #include "dbi.h" |
---|
11 | #include "utils.h" |
---|
12 | #include "list.h" |
---|
13 | #include "rest.h" |
---|
14 | #include "client.h" |
---|
15 | #include "json.h" |
---|
16 | #include "schedule.h" |
---|
17 | #include <syslog.h> |
---|
18 | #include <sys/ioctl.h> |
---|
19 | #include <ifaddrs.h> |
---|
20 | #include <sys/types.h> |
---|
21 | #include <sys/stat.h> |
---|
22 | #include <fcntl.h> |
---|
23 | #include <jansson.h> |
---|
24 | #include <time.h> |
---|
25 | |
---|
26 | static char usuario[LONPRM]; // Usuario de acceso a la base de datos |
---|
27 | static char pasguor[LONPRM]; // Password del usuario |
---|
28 | static char datasource[LONPRM]; // Dirección IP del gestor de base de datos |
---|
29 | static char catalog[LONPRM]; // Nombre de la base de datos |
---|
30 | static char interface[LONPRM]; // Interface name |
---|
31 | char auth_token[LONPRM]; // API token |
---|
32 | |
---|
33 | struct og_dbi_config dbi_config = { |
---|
34 | .user = usuario, |
---|
35 | .passwd = pasguor, |
---|
36 | .host = datasource, |
---|
37 | .database = catalog, |
---|
38 | }; |
---|
39 | |
---|
40 | //________________________________________________________________________________________________________ |
---|
41 | // Función: tomaConfiguracion |
---|
42 | // |
---|
43 | // Descripción: |
---|
44 | // Lee el fichero de configuración del servicio |
---|
45 | // Parámetros: |
---|
46 | // filecfg : Ruta completa al fichero de configuración |
---|
47 | // Devuelve: |
---|
48 | // true: Si el proceso es correcto |
---|
49 | // false: En caso de ocurrir algún error |
---|
50 | //________________________________________________________________________________________________________ |
---|
51 | static bool tomaConfiguracion(const char *filecfg) |
---|
52 | { |
---|
53 | char buf[1024], *line; |
---|
54 | char *key, *value; |
---|
55 | FILE *fcfg; |
---|
56 | |
---|
57 | if (filecfg == NULL || strlen(filecfg) == 0) { |
---|
58 | syslog(LOG_ERR, "No configuration file has been specified\n"); |
---|
59 | return false; |
---|
60 | } |
---|
61 | |
---|
62 | fcfg = fopen(filecfg, "rt"); |
---|
63 | if (fcfg == NULL) { |
---|
64 | syslog(LOG_ERR, "Cannot open configuration file `%s'\n", |
---|
65 | filecfg); |
---|
66 | return false; |
---|
67 | } |
---|
68 | |
---|
69 | servidoradm[0] = '\0'; //inicializar variables globales |
---|
70 | |
---|
71 | line = fgets(buf, sizeof(buf), fcfg); |
---|
72 | while (line != NULL) { |
---|
73 | const char *delim = "="; |
---|
74 | |
---|
75 | line[strlen(line) - 1] = '\0'; |
---|
76 | |
---|
77 | key = strtok(line, delim); |
---|
78 | value = strtok(NULL, delim); |
---|
79 | |
---|
80 | if (!strcmp(str_toupper(key), "SERVIDORADM")) |
---|
81 | snprintf(servidoradm, sizeof(servidoradm), "%s", value); |
---|
82 | else if (!strcmp(str_toupper(key), "PUERTO")) |
---|
83 | snprintf(puerto, sizeof(puerto), "%s", value); |
---|
84 | else if (!strcmp(str_toupper(key), "USUARIO")) |
---|
85 | snprintf(usuario, sizeof(usuario), "%s", value); |
---|
86 | else if (!strcmp(str_toupper(key), "PASSWORD")) |
---|
87 | snprintf(pasguor, sizeof(pasguor), "%s", value); |
---|
88 | else if (!strcmp(str_toupper(key), "DATASOURCE")) |
---|
89 | snprintf(datasource, sizeof(datasource), "%s", value); |
---|
90 | else if (!strcmp(str_toupper(key), "CATALOG")) |
---|
91 | snprintf(catalog, sizeof(catalog), "%s", value); |
---|
92 | else if (!strcmp(str_toupper(key), "INTERFACE")) |
---|
93 | snprintf(interface, sizeof(interface), "%s", value); |
---|
94 | else if (!strcmp(str_toupper(key), "APITOKEN")) |
---|
95 | snprintf(auth_token, sizeof(auth_token), "%s", value); |
---|
96 | |
---|
97 | line = fgets(buf, sizeof(buf), fcfg); |
---|
98 | } |
---|
99 | |
---|
100 | fclose(fcfg); |
---|
101 | |
---|
102 | if (!servidoradm[0]) { |
---|
103 | syslog(LOG_ERR, "Missing SERVIDORADM in configuration file\n"); |
---|
104 | return false; |
---|
105 | } |
---|
106 | if (!puerto[0]) { |
---|
107 | syslog(LOG_ERR, "Missing PUERTO in configuration file\n"); |
---|
108 | return false; |
---|
109 | } |
---|
110 | if (!usuario[0]) { |
---|
111 | syslog(LOG_ERR, "Missing USUARIO in configuration file\n"); |
---|
112 | return false; |
---|
113 | } |
---|
114 | if (!pasguor[0]) { |
---|
115 | syslog(LOG_ERR, "Missing PASSWORD in configuration file\n"); |
---|
116 | return false; |
---|
117 | } |
---|
118 | if (!datasource[0]) { |
---|
119 | syslog(LOG_ERR, "Missing DATASOURCE in configuration file\n"); |
---|
120 | return false; |
---|
121 | } |
---|
122 | if (!catalog[0]) { |
---|
123 | syslog(LOG_ERR, "Missing CATALOG in configuration file\n"); |
---|
124 | return false; |
---|
125 | } |
---|
126 | if (!interface[0]) |
---|
127 | syslog(LOG_ERR, "Missing INTERFACE in configuration file\n"); |
---|
128 | |
---|
129 | return true; |
---|
130 | } |
---|
131 | |
---|
132 | #define OG_CMD_MAXLEN 64 |
---|
133 | |
---|
134 | /* Shut down connection if there is no complete message after 10 seconds. */ |
---|
135 | #define OG_CLIENT_TIMEOUT 10 |
---|
136 | |
---|
137 | /* Agent client operation might take longer, shut down after 30 seconds. */ |
---|
138 | #define OG_AGENT_CLIENT_TIMEOUT 30 |
---|
139 | |
---|
140 | // ________________________________________________________________________________________________________ |
---|
141 | // Función: clienteDisponible |
---|
142 | // |
---|
143 | // Descripción: |
---|
144 | // Comprueba la disponibilidad del cliente para recibir comandos interactivos |
---|
145 | // Parametros: |
---|
146 | // - ip : La ip del cliente a buscar |
---|
147 | // - idx: (Salida) Indice que ocupa el cliente, de estar ya registrado |
---|
148 | // Devuelve: |
---|
149 | // true: Si el cliente está disponible |
---|
150 | // false: En caso contrario |
---|
151 | // ________________________________________________________________________________________________________ |
---|
152 | bool clienteDisponible(char *ip, int* idx) |
---|
153 | { |
---|
154 | int estado; |
---|
155 | |
---|
156 | if (clienteExistente(ip, idx)) { |
---|
157 | estado = strcmp(tbsockets[*idx].estado, CLIENTE_OCUPADO); // Cliente ocupado |
---|
158 | if (estado == 0) |
---|
159 | return false; |
---|
160 | |
---|
161 | estado = strcmp(tbsockets[*idx].estado, CLIENTE_APAGADO); // Cliente apagado |
---|
162 | if (estado == 0) |
---|
163 | return false; |
---|
164 | |
---|
165 | estado = strcmp(tbsockets[*idx].estado, CLIENTE_INICIANDO); // Cliente en proceso de inclusión |
---|
166 | if (estado == 0) |
---|
167 | return false; |
---|
168 | |
---|
169 | return true; // En caso contrario el cliente está disponible |
---|
170 | } |
---|
171 | return false; // Cliente no está registrado en el sistema |
---|
172 | } |
---|
173 | // ________________________________________________________________________________________________________ |
---|
174 | // Función: clienteExistente |
---|
175 | // |
---|
176 | // Descripción: |
---|
177 | // Comprueba si el cliente está registrado en la tabla de socket del sistema |
---|
178 | // Parametros: |
---|
179 | // - ip : La ip del cliente a buscar |
---|
180 | // - idx:(Salida) Indice que ocupa el cliente, de estar ya registrado |
---|
181 | // Devuelve: |
---|
182 | // true: Si el cliente está registrado |
---|
183 | // false: En caso contrario |
---|
184 | // ________________________________________________________________________________________________________ |
---|
185 | bool clienteExistente(char *ip, int* idx) |
---|
186 | { |
---|
187 | int i; |
---|
188 | for (i = 0; i < MAXIMOS_CLIENTES; i++) { |
---|
189 | if (contieneIP(ip, tbsockets[i].ip)) { // Si existe la IP en la cadena |
---|
190 | *idx = i; |
---|
191 | return true; |
---|
192 | } |
---|
193 | } |
---|
194 | return false; |
---|
195 | } |
---|
196 | // ________________________________________________________________________________________________________ |
---|
197 | // Función: actualizaConfiguracion |
---|
198 | // |
---|
199 | // Descripción: |
---|
200 | // Esta función actualiza la base de datos con la configuracion de particiones de un cliente |
---|
201 | // Parámetros: |
---|
202 | // - db: Objeto base de datos (ya operativo) |
---|
203 | // - tbl: Objeto tabla |
---|
204 | // - cfg: cadena con una Configuración |
---|
205 | // - ido: Identificador del ordenador cliente |
---|
206 | // Devuelve: |
---|
207 | // true: Si el proceso es correcto |
---|
208 | // false: En caso de ocurrir algún error |
---|
209 | // Especificaciones: |
---|
210 | // Los parametros de la configuración son: |
---|
211 | // par= Número de partición |
---|
212 | // cpt= Codigo o tipo de partición |
---|
213 | // sfi= Sistema de ficheros que está implementado en la partición |
---|
214 | // soi= Nombre del sistema de ficheros instalado en la partición |
---|
215 | // tam= Tamaño de la partición |
---|
216 | // ________________________________________________________________________________________________________ |
---|
217 | bool actualizaConfiguracion(struct og_dbi *dbi, char *cfg, int ido) |
---|
218 | { |
---|
219 | int lon, p, c,i, dato, swu, idsoi, idsfi,k; |
---|
220 | char *ptrPar[MAXPAR], *ptrCfg[7], *ptrDual[2], tbPar[LONSTD]; |
---|
221 | char *ser, *disk, *par, *cpt, *sfi, *soi, *tam, *uso; // Parametros de configuración. |
---|
222 | dbi_result result, result_update; |
---|
223 | const char *msglog; |
---|
224 | |
---|
225 | lon = 0; |
---|
226 | p = splitCadena(ptrPar, cfg, '\n'); |
---|
227 | for (i = 0; i < p; i++) { |
---|
228 | c = splitCadena(ptrCfg, ptrPar[i], '\t'); |
---|
229 | |
---|
230 | // Si la 1ª línea solo incluye el número de serie del equipo; actualizar BD. |
---|
231 | if (i == 0 && c == 1) { |
---|
232 | splitCadena(ptrDual, ptrCfg[0], '='); |
---|
233 | ser = ptrDual[1]; |
---|
234 | if (strlen(ser) > 0) { |
---|
235 | // Solo actualizar si número de serie no existía. |
---|
236 | result = dbi_conn_queryf(dbi->conn, |
---|
237 | "UPDATE ordenadores SET numserie='%s'" |
---|
238 | " WHERE idordenador=%d AND numserie IS NULL", |
---|
239 | ser, ido); |
---|
240 | if (!result) { |
---|
241 | dbi_conn_error(dbi->conn, &msglog); |
---|
242 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
243 | __func__, __LINE__, msglog); |
---|
244 | return false; |
---|
245 | } |
---|
246 | dbi_result_free(result); |
---|
247 | } |
---|
248 | continue; |
---|
249 | } |
---|
250 | |
---|
251 | // Distribución de particionado. |
---|
252 | disk = par = cpt = sfi = soi = tam = uso = NULL; |
---|
253 | |
---|
254 | splitCadena(ptrDual, ptrCfg[0], '='); |
---|
255 | disk = ptrDual[1]; // Número de disco |
---|
256 | |
---|
257 | splitCadena(ptrDual, ptrCfg[1], '='); |
---|
258 | par = ptrDual[1]; // Número de partición |
---|
259 | |
---|
260 | k=splitCadena(ptrDual, ptrCfg[2], '='); |
---|
261 | if(k==2){ |
---|
262 | cpt = ptrDual[1]; // Código de partición |
---|
263 | }else{ |
---|
264 | cpt = (char*)"0"; |
---|
265 | } |
---|
266 | |
---|
267 | k=splitCadena(ptrDual, ptrCfg[3], '='); |
---|
268 | if(k==2){ |
---|
269 | sfi = ptrDual[1]; // Sistema de ficheros |
---|
270 | /* Comprueba existencia del s0xistema de ficheros instalado */ |
---|
271 | idsfi = checkDato(dbi, sfi, "sistemasficheros", "descripcion","idsistemafichero"); |
---|
272 | } |
---|
273 | else |
---|
274 | idsfi=0; |
---|
275 | |
---|
276 | k=splitCadena(ptrDual, ptrCfg[4], '='); |
---|
277 | if(k==2){ // Sistema operativo detecdtado |
---|
278 | soi = ptrDual[1]; // Nombre del S.O. instalado |
---|
279 | /* Comprueba existencia del sistema operativo instalado */ |
---|
280 | idsoi = checkDato(dbi, soi, "nombresos", "nombreso", "idnombreso"); |
---|
281 | } |
---|
282 | else |
---|
283 | idsoi=0; |
---|
284 | |
---|
285 | splitCadena(ptrDual, ptrCfg[5], '='); |
---|
286 | tam = ptrDual[1]; // Tamaño de la partición |
---|
287 | |
---|
288 | splitCadena(ptrDual, ptrCfg[6], '='); |
---|
289 | uso = ptrDual[1]; // Porcentaje de uso del S.F. |
---|
290 | |
---|
291 | lon += sprintf(tbPar + lon, "(%s, %s),", disk, par); |
---|
292 | |
---|
293 | result = dbi_conn_queryf(dbi->conn, |
---|
294 | "SELECT numdisk, numpar, tamano, uso, idsistemafichero, idnombreso" |
---|
295 | " FROM ordenadores_particiones" |
---|
296 | " WHERE idordenador=%d AND numdisk=%s AND numpar=%s", |
---|
297 | ido, disk, par); |
---|
298 | if (!result) { |
---|
299 | dbi_conn_error(dbi->conn, &msglog); |
---|
300 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
301 | __func__, __LINE__, msglog); |
---|
302 | return false; |
---|
303 | } |
---|
304 | if (!dbi_result_next_row(result)) { |
---|
305 | result_update = dbi_conn_queryf(dbi->conn, |
---|
306 | "INSERT INTO ordenadores_particiones(idordenador,numdisk,numpar,codpar,tamano,uso,idsistemafichero,idnombreso,idimagen)" |
---|
307 | " VALUES(%d,%s,%s,0x%s,%s,%s,%d,%d,0)", |
---|
308 | ido, disk, par, cpt, tam, uso, idsfi, idsoi); |
---|
309 | if (!result_update) { |
---|
310 | dbi_conn_error(dbi->conn, &msglog); |
---|
311 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
312 | __func__, __LINE__, msglog); |
---|
313 | return false; |
---|
314 | } |
---|
315 | dbi_result_free(result_update); |
---|
316 | |
---|
317 | } else { // Existe el registro |
---|
318 | swu = true; // Se supone que algún dato ha cambiado |
---|
319 | |
---|
320 | dato = dbi_result_get_uint(result, "tamano"); |
---|
321 | if (atoi(tam) == dato) {// Parámetro tamaño igual al almacenado |
---|
322 | dato = dbi_result_get_uint(result, "idsistemafichero"); |
---|
323 | if (idsfi == dato) {// Parámetro sistema de fichero igual al almacenado |
---|
324 | dato = dbi_result_get_uint(result, "idnombreso"); |
---|
325 | if (idsoi == dato) {// Parámetro sistema de fichero distinto al almacenado |
---|
326 | swu = false; // Todos los parámetros de la partición son iguales, no se actualiza |
---|
327 | } |
---|
328 | } |
---|
329 | } |
---|
330 | if (swu) { // Hay que actualizar los parámetros de la partición |
---|
331 | result_update = dbi_conn_queryf(dbi->conn, |
---|
332 | "UPDATE ordenadores_particiones SET " |
---|
333 | " codpar=0x%s," |
---|
334 | " tamano=%s," |
---|
335 | " uso=%s," |
---|
336 | " idsistemafichero=%d," |
---|
337 | " idnombreso=%d," |
---|
338 | " idimagen=0," |
---|
339 | " idperfilsoft=0," |
---|
340 | " fechadespliegue=NULL" |
---|
341 | " WHERE idordenador=%d AND numdisk=%s AND numpar=%s", |
---|
342 | cpt, tam, uso, idsfi, idsoi, ido, disk, par); |
---|
343 | } else { // Actualizar porcentaje de uso. |
---|
344 | result_update = dbi_conn_queryf(dbi->conn, |
---|
345 | "UPDATE ordenadores_particiones SET " |
---|
346 | " codpar=0x%s," |
---|
347 | " uso=%s" |
---|
348 | " WHERE idordenador=%d AND numdisk=%s AND numpar=%s", |
---|
349 | cpt, uso, ido, disk, par); |
---|
350 | } |
---|
351 | if (!result_update) { |
---|
352 | dbi_conn_error(dbi->conn, &msglog); |
---|
353 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
354 | __func__, __LINE__, msglog); |
---|
355 | return false; |
---|
356 | } |
---|
357 | |
---|
358 | dbi_result_free(result_update); |
---|
359 | } |
---|
360 | dbi_result_free(result); |
---|
361 | } |
---|
362 | lon += sprintf(tbPar + lon, "(0,0)"); |
---|
363 | // Eliminar particiones almacenadas que ya no existen |
---|
364 | result_update = dbi_conn_queryf(dbi->conn, |
---|
365 | "DELETE FROM ordenadores_particiones WHERE idordenador=%d AND (numdisk, numpar) NOT IN (%s)", |
---|
366 | ido, tbPar); |
---|
367 | if (!result_update) { |
---|
368 | dbi_conn_error(dbi->conn, &msglog); |
---|
369 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
370 | __func__, __LINE__, msglog); |
---|
371 | return false; |
---|
372 | } |
---|
373 | dbi_result_free(result_update); |
---|
374 | |
---|
375 | return true; |
---|
376 | } |
---|
377 | // ________________________________________________________________________________________________________ |
---|
378 | // Función: checkDato |
---|
379 | // |
---|
380 | // Descripción: |
---|
381 | // Esta función comprueba si existe un dato en una tabla y si no es así lo incluye. devuelve en |
---|
382 | // cualquier caso el identificador del registro existenet o del insertado |
---|
383 | // Parámetros: |
---|
384 | // - db: Objeto base de datos (ya operativo) |
---|
385 | // - tbl: Objeto tabla |
---|
386 | // - dato: Dato |
---|
387 | // - tabla: Nombre de la tabla |
---|
388 | // - nomdato: Nombre del dato en la tabla |
---|
389 | // - nomidentificador: Nombre del identificador en la tabla |
---|
390 | // Devuelve: |
---|
391 | // El identificador del registro existente o el del insertado |
---|
392 | // |
---|
393 | // Especificaciones: |
---|
394 | // En caso de producirse algún error se devuelve el valor 0 |
---|
395 | // ________________________________________________________________________________________________________ |
---|
396 | |
---|
397 | int checkDato(struct og_dbi *dbi, char *dato, const char *tabla, |
---|
398 | const char *nomdato, const char *nomidentificador) |
---|
399 | { |
---|
400 | const char *msglog; |
---|
401 | int identificador; |
---|
402 | dbi_result result; |
---|
403 | |
---|
404 | if (strlen(dato) == 0) |
---|
405 | return (0); // EL dato no tiene valor |
---|
406 | result = dbi_conn_queryf(dbi->conn, |
---|
407 | "SELECT %s FROM %s WHERE %s ='%s'", nomidentificador, |
---|
408 | tabla, nomdato, dato); |
---|
409 | |
---|
410 | // Ejecuta consulta |
---|
411 | if (!result) { |
---|
412 | dbi_conn_error(dbi->conn, &msglog); |
---|
413 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
414 | __func__, __LINE__, msglog); |
---|
415 | return (0); |
---|
416 | } |
---|
417 | if (!dbi_result_next_row(result)) { // Software NO existente |
---|
418 | dbi_result_free(result); |
---|
419 | |
---|
420 | result = dbi_conn_queryf(dbi->conn, |
---|
421 | "INSERT INTO %s (%s) VALUES('%s')", tabla, nomdato, dato); |
---|
422 | if (!result) { |
---|
423 | dbi_conn_error(dbi->conn, &msglog); |
---|
424 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
425 | __func__, __LINE__, msglog); |
---|
426 | return (0); |
---|
427 | } |
---|
428 | // Recupera el identificador del software |
---|
429 | identificador = dbi_conn_sequence_last(dbi->conn, NULL); |
---|
430 | } else { |
---|
431 | identificador = dbi_result_get_uint(result, nomidentificador); |
---|
432 | } |
---|
433 | dbi_result_free(result); |
---|
434 | |
---|
435 | return (identificador); |
---|
436 | } |
---|
437 | |
---|
438 | // ________________________________________________________________________________________________________ |
---|
439 | // Función: Levanta |
---|
440 | // |
---|
441 | // Descripción: |
---|
442 | // Enciende ordenadores a través de la red cuyas macs se pasan como parámetro |
---|
443 | // Parámetros: |
---|
444 | // - iph: Cadena de direcciones ip separadas por ";" |
---|
445 | // - mac: Cadena de direcciones mac separadas por ";" |
---|
446 | // - mar: Método de arranque (1=Broadcast, 2=Unicast) |
---|
447 | // Devuelve: |
---|
448 | // true: Si el proceso es correcto |
---|
449 | // false: En caso de ocurrir algún error |
---|
450 | // ________________________________________________________________________________________________________ |
---|
451 | |
---|
452 | bool Levanta(char *ptrIP[], char *ptrMacs[], int lon, char *mar) |
---|
453 | { |
---|
454 | unsigned int on = 1; |
---|
455 | struct sockaddr_in local; |
---|
456 | int i, res; |
---|
457 | int s; |
---|
458 | |
---|
459 | /* Creación de socket para envío de magig packet */ |
---|
460 | s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP); |
---|
461 | if (s < 0) { |
---|
462 | syslog(LOG_ERR, "cannot create socket for magic packet\n"); |
---|
463 | return false; |
---|
464 | } |
---|
465 | res = setsockopt(s, SOL_SOCKET, SO_BROADCAST, (unsigned int *) &on, |
---|
466 | sizeof(on)); |
---|
467 | if (res < 0) { |
---|
468 | syslog(LOG_ERR, "cannot set broadcast socket\n"); |
---|
469 | return false; |
---|
470 | } |
---|
471 | memset(&local, 0, sizeof(local)); |
---|
472 | local.sin_family = AF_INET; |
---|
473 | local.sin_port = htons(PUERTO_WAKEUP); |
---|
474 | local.sin_addr.s_addr = htonl(INADDR_ANY); |
---|
475 | |
---|
476 | for (i = 0; i < lon; i++) { |
---|
477 | if (!WakeUp(s, ptrIP[i], ptrMacs[i], mar)) { |
---|
478 | syslog(LOG_ERR, "problem sending magic packet\n"); |
---|
479 | close(s); |
---|
480 | return false; |
---|
481 | } |
---|
482 | } |
---|
483 | close(s); |
---|
484 | return true; |
---|
485 | } |
---|
486 | |
---|
487 | #define OG_WOL_SEQUENCE 6 |
---|
488 | #define OG_WOL_MACADDR_LEN 6 |
---|
489 | #define OG_WOL_REPEAT 16 |
---|
490 | |
---|
491 | struct wol_msg { |
---|
492 | char secuencia_FF[OG_WOL_SEQUENCE]; |
---|
493 | char macbin[OG_WOL_REPEAT][OG_WOL_MACADDR_LEN]; |
---|
494 | }; |
---|
495 | |
---|
496 | static bool wake_up_broadcast(int sd, struct sockaddr_in *client, |
---|
497 | const struct wol_msg *msg) |
---|
498 | { |
---|
499 | struct sockaddr_in *broadcast_addr; |
---|
500 | struct ifaddrs *ifaddr, *ifa; |
---|
501 | int ret; |
---|
502 | |
---|
503 | if (getifaddrs(&ifaddr) < 0) { |
---|
504 | syslog(LOG_ERR, "cannot get list of addresses\n"); |
---|
505 | return false; |
---|
506 | } |
---|
507 | |
---|
508 | client->sin_addr.s_addr = htonl(INADDR_BROADCAST); |
---|
509 | |
---|
510 | for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) { |
---|
511 | if (ifa->ifa_addr == NULL || |
---|
512 | ifa->ifa_addr->sa_family != AF_INET || |
---|
513 | strcmp(ifa->ifa_name, interface) != 0) |
---|
514 | continue; |
---|
515 | |
---|
516 | broadcast_addr = |
---|
517 | (struct sockaddr_in *)ifa->ifa_ifu.ifu_broadaddr; |
---|
518 | client->sin_addr.s_addr = broadcast_addr->sin_addr.s_addr; |
---|
519 | break; |
---|
520 | } |
---|
521 | freeifaddrs(ifaddr); |
---|
522 | |
---|
523 | ret = sendto(sd, msg, sizeof(*msg), 0, |
---|
524 | (struct sockaddr *)client, sizeof(*client)); |
---|
525 | if (ret < 0) { |
---|
526 | syslog(LOG_ERR, "failed to send broadcast wol\n"); |
---|
527 | return false; |
---|
528 | } |
---|
529 | |
---|
530 | return true; |
---|
531 | } |
---|
532 | |
---|
533 | static bool wake_up_unicast(int sd, struct sockaddr_in *client, |
---|
534 | const struct wol_msg *msg, |
---|
535 | const struct in_addr *addr) |
---|
536 | { |
---|
537 | int ret; |
---|
538 | |
---|
539 | client->sin_addr.s_addr = addr->s_addr; |
---|
540 | |
---|
541 | ret = sendto(sd, msg, sizeof(*msg), 0, |
---|
542 | (struct sockaddr *)client, sizeof(*client)); |
---|
543 | if (ret < 0) { |
---|
544 | syslog(LOG_ERR, "failed to send unicast wol\n"); |
---|
545 | return false; |
---|
546 | } |
---|
547 | |
---|
548 | return true; |
---|
549 | } |
---|
550 | |
---|
551 | enum wol_delivery_type { |
---|
552 | OG_WOL_BROADCAST = 1, |
---|
553 | OG_WOL_UNICAST = 2 |
---|
554 | }; |
---|
555 | |
---|
556 | //_____________________________________________________________________________________________________________ |
---|
557 | // Función: WakeUp |
---|
558 | // |
---|
559 | // Descripción: |
---|
560 | // Enciende el ordenador cuya MAC se pasa como parámetro |
---|
561 | // Parámetros: |
---|
562 | // - s : Socket para enviar trama magic packet |
---|
563 | // - iph : Cadena con la dirección ip |
---|
564 | // - mac : Cadena con la dirección mac en formato XXXXXXXXXXXX |
---|
565 | // - mar: Método de arranque (1=Broadcast, 2=Unicast) |
---|
566 | // Devuelve: |
---|
567 | // true: Si el proceso es correcto |
---|
568 | // false: En caso de ocurrir algún error |
---|
569 | //_____________________________________________________________________________________________________________ |
---|
570 | // |
---|
571 | bool WakeUp(int s, char* iph, char *mac, char *mar) |
---|
572 | { |
---|
573 | unsigned int macaddr[OG_WOL_MACADDR_LEN]; |
---|
574 | char HDaddress_bin[OG_WOL_MACADDR_LEN]; |
---|
575 | struct sockaddr_in WakeUpCliente; |
---|
576 | struct wol_msg Trama_WakeUp; |
---|
577 | struct in_addr addr; |
---|
578 | bool ret; |
---|
579 | int i; |
---|
580 | |
---|
581 | for (i = 0; i < 6; i++) // Primera secuencia de la trama Wake Up (0xFFFFFFFFFFFF) |
---|
582 | Trama_WakeUp.secuencia_FF[i] = 0xFF; |
---|
583 | |
---|
584 | sscanf(mac, "%02x%02x%02x%02x%02x%02x", |
---|
585 | &macaddr[0], &macaddr[1], &macaddr[2], |
---|
586 | &macaddr[3], &macaddr[4], &macaddr[5]); |
---|
587 | |
---|
588 | for (i = 0; i < 6; i++) |
---|
589 | HDaddress_bin[i] = (uint8_t)macaddr[i]; |
---|
590 | |
---|
591 | for (i = 0; i < 16; i++) // Segunda secuencia de la trama Wake Up , repetir 16 veces su la MAC |
---|
592 | memcpy(&Trama_WakeUp.macbin[i][0], &HDaddress_bin, 6); |
---|
593 | |
---|
594 | /* Creación de socket del cliente que recibe la trama magic packet */ |
---|
595 | WakeUpCliente.sin_family = AF_INET; |
---|
596 | WakeUpCliente.sin_port = htons((short) PUERTO_WAKEUP); |
---|
597 | |
---|
598 | switch (atoi(mar)) { |
---|
599 | case OG_WOL_BROADCAST: |
---|
600 | ret = wake_up_broadcast(s, &WakeUpCliente, &Trama_WakeUp); |
---|
601 | break; |
---|
602 | case OG_WOL_UNICAST: |
---|
603 | if (inet_aton(iph, &addr) < 0) { |
---|
604 | syslog(LOG_ERR, "bad IP address for unicast wol\n"); |
---|
605 | ret = false; |
---|
606 | break; |
---|
607 | } |
---|
608 | ret = wake_up_unicast(s, &WakeUpCliente, &Trama_WakeUp, &addr); |
---|
609 | break; |
---|
610 | default: |
---|
611 | syslog(LOG_ERR, "unknown wol type\n"); |
---|
612 | ret = false; |
---|
613 | break; |
---|
614 | } |
---|
615 | return ret; |
---|
616 | } |
---|
617 | |
---|
618 | // ________________________________________________________________________________________________________ |
---|
619 | // Función: actualizaCreacionImagen |
---|
620 | // |
---|
621 | // Descripción: |
---|
622 | // Esta función actualiza la base de datos con el resultado de la creación de una imagen |
---|
623 | // Parámetros: |
---|
624 | // - db: Objeto base de datos (ya operativo) |
---|
625 | // - tbl: Objeto tabla |
---|
626 | // - idi: Identificador de la imagen |
---|
627 | // - dsk: Disco de donde se creó |
---|
628 | // - par: Partición de donde se creó |
---|
629 | // - cpt: Código de partición |
---|
630 | // - ipr: Ip del repositorio |
---|
631 | // - ido: Identificador del ordenador modelo |
---|
632 | // Devuelve: |
---|
633 | // true: Si el proceso es correcto |
---|
634 | // false: En caso de ocurrir algún error |
---|
635 | // ________________________________________________________________________________________________________ |
---|
636 | bool actualizaCreacionImagen(struct og_dbi *dbi, char *idi, char *dsk, |
---|
637 | char *par, char *cpt, char *ipr, char *ido) |
---|
638 | { |
---|
639 | const char *msglog; |
---|
640 | dbi_result result; |
---|
641 | int idr,ifs; |
---|
642 | |
---|
643 | /* Toma identificador del repositorio correspondiente al ordenador modelo */ |
---|
644 | result = dbi_conn_queryf(dbi->conn, |
---|
645 | "SELECT repositorios.idrepositorio" |
---|
646 | " FROM repositorios" |
---|
647 | " LEFT JOIN ordenadores USING (idrepositorio)" |
---|
648 | " WHERE repositorios.ip='%s' AND ordenadores.idordenador=%s", ipr, ido); |
---|
649 | |
---|
650 | if (!result) { |
---|
651 | dbi_conn_error(dbi->conn, &msglog); |
---|
652 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
653 | __func__, __LINE__, msglog); |
---|
654 | return false; |
---|
655 | } |
---|
656 | if (!dbi_result_next_row(result)) { |
---|
657 | syslog(LOG_ERR, |
---|
658 | "repository does not exist in database (%s:%d)\n", |
---|
659 | __func__, __LINE__); |
---|
660 | dbi_result_free(result); |
---|
661 | return false; |
---|
662 | } |
---|
663 | idr = dbi_result_get_uint(result, "idrepositorio"); |
---|
664 | dbi_result_free(result); |
---|
665 | |
---|
666 | /* Toma identificador del perfilsoftware */ |
---|
667 | result = dbi_conn_queryf(dbi->conn, |
---|
668 | "SELECT idperfilsoft" |
---|
669 | " FROM ordenadores_particiones" |
---|
670 | " WHERE idordenador=%s AND numdisk=%s AND numpar=%s", ido, dsk, par); |
---|
671 | |
---|
672 | if (!result) { |
---|
673 | dbi_conn_error(dbi->conn, &msglog); |
---|
674 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
675 | __func__, __LINE__, msglog); |
---|
676 | return false; |
---|
677 | } |
---|
678 | if (!dbi_result_next_row(result)) { |
---|
679 | syslog(LOG_ERR, |
---|
680 | "software profile does not exist in database (%s:%d)\n", |
---|
681 | __func__, __LINE__); |
---|
682 | dbi_result_free(result); |
---|
683 | return false; |
---|
684 | } |
---|
685 | ifs = dbi_result_get_uint(result, "idperfilsoft"); |
---|
686 | dbi_result_free(result); |
---|
687 | |
---|
688 | /* Actualizar los datos de la imagen */ |
---|
689 | result = dbi_conn_queryf(dbi->conn, |
---|
690 | "UPDATE imagenes" |
---|
691 | " SET idordenador=%s, numdisk=%s, numpar=%s, codpar=%s," |
---|
692 | " idperfilsoft=%d, idrepositorio=%d," |
---|
693 | " fechacreacion=NOW(), revision=revision+1" |
---|
694 | " WHERE idimagen=%s", ido, dsk, par, cpt, ifs, idr, idi); |
---|
695 | |
---|
696 | if (!result) { |
---|
697 | dbi_conn_error(dbi->conn, &msglog); |
---|
698 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
699 | __func__, __LINE__, msglog); |
---|
700 | return false; |
---|
701 | } |
---|
702 | dbi_result_free(result); |
---|
703 | |
---|
704 | /* Actualizar los datos en el cliente */ |
---|
705 | result = dbi_conn_queryf(dbi->conn, |
---|
706 | "UPDATE ordenadores_particiones" |
---|
707 | " SET idimagen=%s, revision=(SELECT revision FROM imagenes WHERE idimagen=%s)," |
---|
708 | " fechadespliegue=NOW()" |
---|
709 | " WHERE idordenador=%s AND numdisk=%s AND numpar=%s", |
---|
710 | idi, idi, ido, dsk, par); |
---|
711 | if (!result) { |
---|
712 | dbi_conn_error(dbi->conn, &msglog); |
---|
713 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
714 | __func__, __LINE__, msglog); |
---|
715 | return false; |
---|
716 | } |
---|
717 | dbi_result_free(result); |
---|
718 | |
---|
719 | return true; |
---|
720 | } |
---|
721 | |
---|
722 | // ________________________________________________________________________________________________________ |
---|
723 | // Función: actualizaRestauracionImagen |
---|
724 | // |
---|
725 | // Descripción: |
---|
726 | // Esta función actualiza la base de datos con el resultado de la restauración de una imagen |
---|
727 | // Parámetros: |
---|
728 | // - db: Objeto base de datos (ya operativo) |
---|
729 | // - tbl: Objeto tabla |
---|
730 | // - idi: Identificador de la imagen |
---|
731 | // - dsk: Disco de donde se restauró |
---|
732 | // - par: Partición de donde se restauró |
---|
733 | // - ido: Identificador del cliente donde se restauró |
---|
734 | // - ifs: Identificador del perfil software contenido en la imagen |
---|
735 | // Devuelve: |
---|
736 | // true: Si el proceso es correcto |
---|
737 | // false: En caso de ocurrir algún error |
---|
738 | // ________________________________________________________________________________________________________ |
---|
739 | bool actualizaRestauracionImagen(struct og_dbi *dbi, char *idi, |
---|
740 | char *dsk, char *par, char *ido, char *ifs) |
---|
741 | { |
---|
742 | const char *msglog; |
---|
743 | dbi_result result; |
---|
744 | |
---|
745 | /* Actualizar los datos de la imagen */ |
---|
746 | result = dbi_conn_queryf(dbi->conn, |
---|
747 | "UPDATE ordenadores_particiones" |
---|
748 | " SET idimagen=%s, idperfilsoft=%s, fechadespliegue=NOW()," |
---|
749 | " revision=(SELECT revision FROM imagenes WHERE idimagen=%s)," |
---|
750 | " idnombreso=IFNULL((SELECT idnombreso FROM perfilessoft WHERE idperfilsoft=%s),0)" |
---|
751 | " WHERE idordenador=%s AND numdisk=%s AND numpar=%s", idi, ifs, idi, ifs, ido, dsk, par); |
---|
752 | |
---|
753 | if (!result) { |
---|
754 | dbi_conn_error(dbi->conn, &msglog); |
---|
755 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
756 | __func__, __LINE__, msglog); |
---|
757 | return false; |
---|
758 | } |
---|
759 | dbi_result_free(result); |
---|
760 | |
---|
761 | return true; |
---|
762 | } |
---|
763 | // ________________________________________________________________________________________________________ |
---|
764 | // Función: actualizaHardware |
---|
765 | // |
---|
766 | // Descripción: |
---|
767 | // Actualiza la base de datos con la configuracion hardware del cliente |
---|
768 | // Parámetros: |
---|
769 | // - db: Objeto base de datos (ya operativo) |
---|
770 | // - tbl: Objeto tabla |
---|
771 | // - hrd: cadena con el inventario hardware |
---|
772 | // - ido: Identificador del ordenador |
---|
773 | // - npc: Nombre del ordenador |
---|
774 | // - idc: Identificador del centro o Unidad organizativa |
---|
775 | // ________________________________________________________________________________________________________ |
---|
776 | // |
---|
777 | bool actualizaHardware(struct og_dbi *dbi, char *hrd, char *ido, char *npc, |
---|
778 | char *idc) |
---|
779 | { |
---|
780 | const char *msglog; |
---|
781 | int idtipohardware, idperfilhard; |
---|
782 | int lon, i, j, aux; |
---|
783 | bool retval; |
---|
784 | char *whard; |
---|
785 | int tbidhardware[MAXHARDWARE]; |
---|
786 | char *tbHardware[MAXHARDWARE],*dualHardware[2], strInt[LONINT], *idhardwares; |
---|
787 | dbi_result result; |
---|
788 | |
---|
789 | /* Toma Centro (Unidad Organizativa) */ |
---|
790 | result = dbi_conn_queryf(dbi->conn, |
---|
791 | "SELECT idperfilhard FROM ordenadores WHERE idordenador=%s", |
---|
792 | ido); |
---|
793 | if (!result) { |
---|
794 | dbi_conn_error(dbi->conn, &msglog); |
---|
795 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
796 | __func__, __LINE__, msglog); |
---|
797 | return false; |
---|
798 | } |
---|
799 | if (!dbi_result_next_row(result)) { |
---|
800 | syslog(LOG_ERR, "client does not exist in database (%s:%d)\n", |
---|
801 | __func__, __LINE__); |
---|
802 | dbi_result_free(result); |
---|
803 | return false; |
---|
804 | } |
---|
805 | idperfilhard = dbi_result_get_uint(result, "idperfilhard"); |
---|
806 | dbi_result_free(result); |
---|
807 | |
---|
808 | whard=escaparCadena(hrd); // Codificar comillas simples |
---|
809 | if(!whard) |
---|
810 | return false; |
---|
811 | /* Recorre componentes hardware*/ |
---|
812 | lon = splitCadena(tbHardware, whard, '\n'); |
---|
813 | if (lon > MAXHARDWARE) |
---|
814 | lon = MAXHARDWARE; // Limita el número de componentes hardware |
---|
815 | /* |
---|
816 | for (i=0;i<lon;i++){ |
---|
817 | sprintf(msglog,"Linea de inventario: %s",tbHardware[i]); |
---|
818 | RegistraLog(msglog,false); |
---|
819 | } |
---|
820 | */ |
---|
821 | for (i = 0; i < lon; i++) { |
---|
822 | splitCadena(dualHardware, rTrim(tbHardware[i]), '='); |
---|
823 | //sprintf(msglog,"nemonico: %s",dualHardware[0]); |
---|
824 | //RegistraLog(msglog,false); |
---|
825 | //sprintf(msglog,"valor: %s",dualHardware[1]); |
---|
826 | //RegistraLog(msglog,false); |
---|
827 | result = dbi_conn_queryf(dbi->conn, |
---|
828 | "SELECT idtipohardware,descripcion FROM tipohardwares WHERE nemonico='%s'", |
---|
829 | dualHardware[0]); |
---|
830 | if (!result) { |
---|
831 | dbi_conn_error(dbi->conn, &msglog); |
---|
832 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
833 | __func__, __LINE__, msglog); |
---|
834 | return false; |
---|
835 | } |
---|
836 | if (!dbi_result_next_row(result)) { // Tipo de Hardware NO existente |
---|
837 | dbi_result_free(result); |
---|
838 | return false; |
---|
839 | } else { // Tipo de Hardware Existe |
---|
840 | idtipohardware = dbi_result_get_uint(result, "idtipohardware"); |
---|
841 | dbi_result_free(result); |
---|
842 | |
---|
843 | result = dbi_conn_queryf(dbi->conn, |
---|
844 | "SELECT idhardware FROM hardwares WHERE idtipohardware=%d AND descripcion='%s'", |
---|
845 | idtipohardware, dualHardware[1]); |
---|
846 | |
---|
847 | if (!result) { |
---|
848 | dbi_conn_error(dbi->conn, &msglog); |
---|
849 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
850 | __func__, __LINE__, msglog); |
---|
851 | return false; |
---|
852 | } |
---|
853 | |
---|
854 | if (!dbi_result_next_row(result)) { // Hardware NO existente |
---|
855 | dbi_result_free(result); |
---|
856 | result = dbi_conn_queryf(dbi->conn, |
---|
857 | "INSERT hardwares (idtipohardware,descripcion,idcentro,grupoid) " |
---|
858 | " VALUES(%d,'%s',%s,0)", idtipohardware, |
---|
859 | dualHardware[1], idc); |
---|
860 | if (!result) { |
---|
861 | dbi_conn_error(dbi->conn, &msglog); |
---|
862 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
863 | __func__, __LINE__, msglog); |
---|
864 | return false; |
---|
865 | } |
---|
866 | |
---|
867 | // Recupera el identificador del hardware |
---|
868 | tbidhardware[i] = dbi_conn_sequence_last(dbi->conn, NULL); |
---|
869 | } else { |
---|
870 | tbidhardware[i] = dbi_result_get_uint(result, "idhardware"); |
---|
871 | } |
---|
872 | dbi_result_free(result); |
---|
873 | } |
---|
874 | } |
---|
875 | // Ordena tabla de identificadores para cosultar si existe un pefil con esas especificaciones |
---|
876 | |
---|
877 | for (i = 0; i < lon - 1; i++) { |
---|
878 | for (j = i + 1; j < lon; j++) { |
---|
879 | if (tbidhardware[i] > tbidhardware[j]) { |
---|
880 | aux = tbidhardware[i]; |
---|
881 | tbidhardware[i] = tbidhardware[j]; |
---|
882 | tbidhardware[j] = aux; |
---|
883 | } |
---|
884 | } |
---|
885 | } |
---|
886 | /* Crea cadena de identificadores de componentes hardware separados por coma */ |
---|
887 | sprintf(strInt, "%d", tbidhardware[lon - 1]); // Pasa a cadena el último identificador que es de mayor longitud |
---|
888 | aux = strlen(strInt); // Calcula longitud de cadena para reservar espacio a todos los perfiles |
---|
889 | idhardwares = calloc(1, sizeof(aux) * lon + lon); |
---|
890 | if (idhardwares == NULL) { |
---|
891 | syslog(LOG_ERR, "%s:%d OOM\n", __FILE__, __LINE__); |
---|
892 | return false; |
---|
893 | } |
---|
894 | aux = sprintf(idhardwares, "%d", tbidhardware[0]); |
---|
895 | for (i = 1; i < lon; i++) |
---|
896 | aux += sprintf(idhardwares + aux, ",%d", tbidhardware[i]); |
---|
897 | |
---|
898 | if (!cuestionPerfilHardware(dbi, idc, ido, idperfilhard, idhardwares, |
---|
899 | npc, tbidhardware, lon)) { |
---|
900 | syslog(LOG_ERR, "Problem updating client hardware\n"); |
---|
901 | retval=false; |
---|
902 | } else { |
---|
903 | retval=true; |
---|
904 | } |
---|
905 | free(whard); |
---|
906 | free(idhardwares); |
---|
907 | |
---|
908 | return (retval); |
---|
909 | } |
---|
910 | // ________________________________________________________________________________________________________ |
---|
911 | // Función: cuestionPerfilHardware |
---|
912 | // |
---|
913 | // Descripción: |
---|
914 | // Comprueba existencia de perfil hardware y actualización de éste para el ordenador |
---|
915 | // Parámetros: |
---|
916 | // - db: Objeto base de datos (ya operativo) |
---|
917 | // - tbl: Objeto tabla |
---|
918 | // - idc: Identificador de la Unidad organizativa donde se encuentra el cliente |
---|
919 | // - ido: Identificador del ordenador |
---|
920 | // - tbidhardware: Identificador del tipo de hardware |
---|
921 | // - con: Número de componentes detectados para configurar un el perfil hardware |
---|
922 | // - npc: Nombre del cliente |
---|
923 | // ________________________________________________________________________________________________________ |
---|
924 | bool cuestionPerfilHardware(struct og_dbi *dbi, char *idc, char *ido, |
---|
925 | int idperfilhardware, char *idhardwares, char *npc, int *tbidhardware, |
---|
926 | int lon) |
---|
927 | { |
---|
928 | const char *msglog; |
---|
929 | dbi_result result; |
---|
930 | int i; |
---|
931 | int nwidperfilhard; |
---|
932 | |
---|
933 | // Busca perfil hard del ordenador que contenga todos los componentes hardware encontrados |
---|
934 | result = dbi_conn_queryf(dbi->conn, |
---|
935 | "SELECT idperfilhard FROM" |
---|
936 | " (SELECT perfileshard_hardwares.idperfilhard as idperfilhard," |
---|
937 | " group_concat(cast(perfileshard_hardwares.idhardware AS char( 11) )" |
---|
938 | " ORDER BY perfileshard_hardwares.idhardware SEPARATOR ',' ) AS idhardwares" |
---|
939 | " FROM perfileshard_hardwares" |
---|
940 | " GROUP BY perfileshard_hardwares.idperfilhard) AS temp" |
---|
941 | " WHERE idhardwares LIKE '%s'", idhardwares); |
---|
942 | |
---|
943 | if (!result) { |
---|
944 | dbi_conn_error(dbi->conn, &msglog); |
---|
945 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
946 | __func__, __LINE__, msglog); |
---|
947 | return false; |
---|
948 | } |
---|
949 | if (!dbi_result_next_row(result)) { |
---|
950 | // No existe un perfil hardware con esos componentes de componentes hardware, lo crea |
---|
951 | dbi_result_free(result); |
---|
952 | result = dbi_conn_queryf(dbi->conn, |
---|
953 | "INSERT perfileshard (descripcion,idcentro,grupoid)" |
---|
954 | " VALUES('Perfil hardware (%s) ',%s,0)", npc, idc); |
---|
955 | if (!result) { |
---|
956 | dbi_conn_error(dbi->conn, &msglog); |
---|
957 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
958 | __func__, __LINE__, msglog); |
---|
959 | return false; |
---|
960 | } |
---|
961 | dbi_result_free(result); |
---|
962 | |
---|
963 | // Recupera el identificador del nuevo perfil hardware |
---|
964 | nwidperfilhard = dbi_conn_sequence_last(dbi->conn, NULL); |
---|
965 | |
---|
966 | // Crea la relación entre perfiles y componenetes hardware |
---|
967 | for (i = 0; i < lon; i++) { |
---|
968 | result = dbi_conn_queryf(dbi->conn, |
---|
969 | "INSERT perfileshard_hardwares (idperfilhard,idhardware)" |
---|
970 | " VALUES(%d,%d)", nwidperfilhard, tbidhardware[i]); |
---|
971 | if (!result) { |
---|
972 | dbi_conn_error(dbi->conn, &msglog); |
---|
973 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
974 | __func__, __LINE__, msglog); |
---|
975 | return false; |
---|
976 | } |
---|
977 | dbi_result_free(result); |
---|
978 | } |
---|
979 | } else { // Existe un perfil con todos esos componentes |
---|
980 | nwidperfilhard = dbi_result_get_uint(result, "idperfilhard"); |
---|
981 | dbi_result_free(result); |
---|
982 | } |
---|
983 | if (idperfilhardware != nwidperfilhard) { // No coinciden los perfiles |
---|
984 | // Actualiza el identificador del perfil hardware del ordenador |
---|
985 | result = dbi_conn_queryf(dbi->conn, |
---|
986 | "UPDATE ordenadores SET idperfilhard=%d" |
---|
987 | " WHERE idordenador=%s", nwidperfilhard, ido); |
---|
988 | if (!result) { |
---|
989 | dbi_conn_error(dbi->conn, &msglog); |
---|
990 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
991 | __func__, __LINE__, msglog); |
---|
992 | return false; |
---|
993 | } |
---|
994 | dbi_result_free(result); |
---|
995 | } |
---|
996 | /* Eliminar Relación de hardwares con Perfiles hardware que quedan húerfanos */ |
---|
997 | result = dbi_conn_queryf(dbi->conn, |
---|
998 | "DELETE FROM perfileshard_hardwares WHERE idperfilhard IN " |
---|
999 | " (SELECT idperfilhard FROM perfileshard WHERE idperfilhard NOT IN" |
---|
1000 | " (SELECT DISTINCT idperfilhard from ordenadores))"); |
---|
1001 | if (!result) { |
---|
1002 | dbi_conn_error(dbi->conn, &msglog); |
---|
1003 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1004 | __func__, __LINE__, msglog); |
---|
1005 | return false; |
---|
1006 | } |
---|
1007 | dbi_result_free(result); |
---|
1008 | |
---|
1009 | /* Eliminar Perfiles hardware que quedan húerfanos */ |
---|
1010 | result = dbi_conn_queryf(dbi->conn, |
---|
1011 | "DELETE FROM perfileshard WHERE idperfilhard NOT IN" |
---|
1012 | " (SELECT DISTINCT idperfilhard FROM ordenadores)"); |
---|
1013 | if (!result) { |
---|
1014 | dbi_conn_error(dbi->conn, &msglog); |
---|
1015 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1016 | __func__, __LINE__, msglog); |
---|
1017 | return false; |
---|
1018 | } |
---|
1019 | dbi_result_free(result); |
---|
1020 | |
---|
1021 | /* Eliminar Relación de hardwares con Perfiles hardware que quedan húerfanos */ |
---|
1022 | result = dbi_conn_queryf(dbi->conn, |
---|
1023 | "DELETE FROM perfileshard_hardwares WHERE idperfilhard NOT IN" |
---|
1024 | " (SELECT idperfilhard FROM perfileshard)"); |
---|
1025 | if (!result) { |
---|
1026 | dbi_conn_error(dbi->conn, &msglog); |
---|
1027 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1028 | __func__, __LINE__, msglog); |
---|
1029 | return false; |
---|
1030 | } |
---|
1031 | dbi_result_free(result); |
---|
1032 | |
---|
1033 | return true; |
---|
1034 | } |
---|
1035 | // ________________________________________________________________________________________________________ |
---|
1036 | // Función: actualizaSoftware |
---|
1037 | // |
---|
1038 | // Descripción: |
---|
1039 | // Actualiza la base de datos con la configuración software del cliente |
---|
1040 | // Parámetros: |
---|
1041 | // - db: Objeto base de datos (ya operativo) |
---|
1042 | // - tbl: Objeto tabla |
---|
1043 | // - sft: cadena con el inventario software |
---|
1044 | // - par: Número de la partición |
---|
1045 | // - ido: Identificador del ordenador del cliente en la tabla |
---|
1046 | // - npc: Nombre del ordenador |
---|
1047 | // - idc: Identificador del centro o Unidad organizativa |
---|
1048 | // Devuelve: |
---|
1049 | // true: Si el proceso es correcto |
---|
1050 | // false: En caso de ocurrir algún error |
---|
1051 | // |
---|
1052 | // Versión 1.1.0: Se incluye el sistema operativo. Autora: Irina Gómez - ETSII Universidad Sevilla |
---|
1053 | // ________________________________________________________________________________________________________ |
---|
1054 | bool actualizaSoftware(struct og_dbi *dbi, char *sft, char *par,char *ido, |
---|
1055 | char *npc, char *idc) |
---|
1056 | { |
---|
1057 | int i, j, lon, aux, idperfilsoft, idnombreso; |
---|
1058 | bool retval; |
---|
1059 | char *wsft; |
---|
1060 | int tbidsoftware[MAXSOFTWARE]; |
---|
1061 | char *tbSoftware[MAXSOFTWARE], strInt[LONINT], *idsoftwares; |
---|
1062 | const char *msglog; |
---|
1063 | dbi_result result; |
---|
1064 | |
---|
1065 | /* Toma Centro (Unidad Organizativa) y perfil software */ |
---|
1066 | result = dbi_conn_queryf(dbi->conn, |
---|
1067 | "SELECT idperfilsoft,numpar" |
---|
1068 | " FROM ordenadores_particiones" |
---|
1069 | " WHERE idordenador=%s", ido); |
---|
1070 | if (!result) { |
---|
1071 | dbi_conn_error(dbi->conn, &msglog); |
---|
1072 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1073 | __func__, __LINE__, msglog); |
---|
1074 | return false; |
---|
1075 | } |
---|
1076 | idperfilsoft = 0; // Por defecto se supone que el ordenador no tiene aún detectado el perfil software |
---|
1077 | while (dbi_result_next_row(result)) { |
---|
1078 | aux = dbi_result_get_uint(result, "numpar"); |
---|
1079 | if (aux == atoi(par)) { // Se encuentra la partición |
---|
1080 | idperfilsoft = dbi_result_get_uint(result, "idperfilsoft"); |
---|
1081 | break; |
---|
1082 | } |
---|
1083 | } |
---|
1084 | dbi_result_free(result); |
---|
1085 | wsft=escaparCadena(sft); // Codificar comillas simples |
---|
1086 | if(!wsft) |
---|
1087 | return false; |
---|
1088 | |
---|
1089 | /* Recorre componentes software*/ |
---|
1090 | lon = splitCadena(tbSoftware, wsft, '\n'); |
---|
1091 | |
---|
1092 | if (lon == 0) |
---|
1093 | return true; // No hay lineas que procesar |
---|
1094 | if (lon > MAXSOFTWARE) |
---|
1095 | lon = MAXSOFTWARE; // Limita el número de componentes software |
---|
1096 | |
---|
1097 | idnombreso = 0; |
---|
1098 | for (i = 0; i < lon; i++) { |
---|
1099 | // Primera línea es el sistema operativo: se obtiene identificador |
---|
1100 | if (i == 0) { |
---|
1101 | idnombreso = checkDato(dbi, rTrim(tbSoftware[i]), "nombresos", "nombreso", "idnombreso"); |
---|
1102 | continue; |
---|
1103 | } |
---|
1104 | |
---|
1105 | result = dbi_conn_queryf(dbi->conn, |
---|
1106 | "SELECT idsoftware FROM softwares WHERE descripcion ='%s'", |
---|
1107 | rTrim(tbSoftware[i])); |
---|
1108 | if (!result) { |
---|
1109 | dbi_conn_error(dbi->conn, &msglog); |
---|
1110 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1111 | __func__, __LINE__, msglog); |
---|
1112 | return false; |
---|
1113 | } |
---|
1114 | |
---|
1115 | if (!dbi_result_next_row(result)) { |
---|
1116 | dbi_result_free(result); |
---|
1117 | result = dbi_conn_queryf(dbi->conn, |
---|
1118 | "INSERT INTO softwares (idtiposoftware,descripcion,idcentro,grupoid)" |
---|
1119 | " VALUES(2,'%s',%s,0)", tbSoftware[i], idc); |
---|
1120 | if (!result) { // Error al insertar |
---|
1121 | dbi_conn_error(dbi->conn, &msglog); |
---|
1122 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1123 | __func__, __LINE__, msglog); |
---|
1124 | return false; |
---|
1125 | } |
---|
1126 | |
---|
1127 | // Recupera el identificador del software |
---|
1128 | tbidsoftware[i] = dbi_conn_sequence_last(dbi->conn, NULL); |
---|
1129 | } else { |
---|
1130 | tbidsoftware[i] = dbi_result_get_uint(result, "idsoftware"); |
---|
1131 | } |
---|
1132 | dbi_result_free(result); |
---|
1133 | } |
---|
1134 | |
---|
1135 | // Ordena tabla de identificadores para cosultar si existe un pefil con esas especificaciones |
---|
1136 | |
---|
1137 | for (i = 0; i < lon - 1; i++) { |
---|
1138 | for (j = i + 1; j < lon; j++) { |
---|
1139 | if (tbidsoftware[i] > tbidsoftware[j]) { |
---|
1140 | aux = tbidsoftware[i]; |
---|
1141 | tbidsoftware[i] = tbidsoftware[j]; |
---|
1142 | tbidsoftware[j] = aux; |
---|
1143 | } |
---|
1144 | } |
---|
1145 | } |
---|
1146 | /* Crea cadena de identificadores de componentes software separados por coma */ |
---|
1147 | sprintf(strInt, "%d", tbidsoftware[lon - 1]); // Pasa a cadena el último identificador que es de mayor longitud |
---|
1148 | aux = strlen(strInt); // Calcula longitud de cadena para reservar espacio a todos los perfiles |
---|
1149 | idsoftwares = calloc(1, (sizeof(aux)+1) * lon + lon); |
---|
1150 | if (idsoftwares == NULL) { |
---|
1151 | syslog(LOG_ERR, "%s:%d OOM\n", __FILE__, __LINE__); |
---|
1152 | return false; |
---|
1153 | } |
---|
1154 | aux = sprintf(idsoftwares, "%d", tbidsoftware[0]); |
---|
1155 | for (i = 1; i < lon; i++) |
---|
1156 | aux += sprintf(idsoftwares + aux, ",%d", tbidsoftware[i]); |
---|
1157 | |
---|
1158 | // Comprueba existencia de perfil software y actualización de éste para el ordenador |
---|
1159 | if (!cuestionPerfilSoftware(dbi, idc, ido, idperfilsoft, idnombreso, idsoftwares, |
---|
1160 | npc, par, tbidsoftware, lon)) { |
---|
1161 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1162 | __func__, __LINE__, msglog); |
---|
1163 | retval=false; |
---|
1164 | } else { |
---|
1165 | retval=true; |
---|
1166 | } |
---|
1167 | free(wsft); |
---|
1168 | free(idsoftwares); |
---|
1169 | |
---|
1170 | return retval; |
---|
1171 | } |
---|
1172 | // ________________________________________________________________________________________________________ |
---|
1173 | // Función: CuestionPerfilSoftware |
---|
1174 | // |
---|
1175 | // Parámetros: |
---|
1176 | // - db: Objeto base de datos (ya operativo) |
---|
1177 | // - tbl: Objeto tabla |
---|
1178 | // - idcentro: Identificador del centro en la tabla |
---|
1179 | // - ido: Identificador del ordenador del cliente en la tabla |
---|
1180 | // - idnombreso: Identificador del sistema operativo |
---|
1181 | // - idsoftwares: Cadena con los identificadores de componentes software separados por comas |
---|
1182 | // - npc: Nombre del ordenador del cliente |
---|
1183 | // - particion: Número de la partición |
---|
1184 | // - tbidsoftware: Array con los identificadores de componentes software |
---|
1185 | // - lon: Número de componentes |
---|
1186 | // Devuelve: |
---|
1187 | // true: Si el proceso es correcto |
---|
1188 | // false: En caso de ocurrir algún error |
---|
1189 | // |
---|
1190 | // Versión 1.1.0: Se incluye el sistema operativo. Autora: Irina Gómez - ETSII Universidad Sevilla |
---|
1191 | //_________________________________________________________________________________________________________ |
---|
1192 | bool cuestionPerfilSoftware(struct og_dbi *dbi, char *idc, char *ido, |
---|
1193 | int idperfilsoftware, int idnombreso, |
---|
1194 | char *idsoftwares, char *npc, char *par, |
---|
1195 | int *tbidsoftware, int lon) |
---|
1196 | { |
---|
1197 | int i, nwidperfilsoft; |
---|
1198 | const char *msglog; |
---|
1199 | dbi_result result; |
---|
1200 | |
---|
1201 | // Busca perfil soft del ordenador que contenga todos los componentes software encontrados |
---|
1202 | result = dbi_conn_queryf(dbi->conn, |
---|
1203 | "SELECT idperfilsoft FROM" |
---|
1204 | " (SELECT perfilessoft_softwares.idperfilsoft as idperfilsoft," |
---|
1205 | " group_concat(cast(perfilessoft_softwares.idsoftware AS char( 11) )" |
---|
1206 | " ORDER BY perfilessoft_softwares.idsoftware SEPARATOR ',' ) AS idsoftwares" |
---|
1207 | " FROM perfilessoft_softwares" |
---|
1208 | " GROUP BY perfilessoft_softwares.idperfilsoft) AS temp" |
---|
1209 | " WHERE idsoftwares LIKE '%s'", idsoftwares); |
---|
1210 | |
---|
1211 | if (!result) { |
---|
1212 | dbi_conn_error(dbi->conn, &msglog); |
---|
1213 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1214 | __func__, __LINE__, msglog); |
---|
1215 | return false; |
---|
1216 | } |
---|
1217 | if (!dbi_result_next_row(result)) { // No existe un perfil software con esos componentes de componentes software, lo crea |
---|
1218 | dbi_result_free(result); |
---|
1219 | result = dbi_conn_queryf(dbi->conn, |
---|
1220 | "INSERT perfilessoft (descripcion, idcentro, grupoid, idnombreso)" |
---|
1221 | " VALUES('Perfil Software (%s, Part:%s) ',%s,0,%i)", npc, par, idc,idnombreso); |
---|
1222 | if (!result) { |
---|
1223 | dbi_conn_error(dbi->conn, &msglog); |
---|
1224 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1225 | __func__, __LINE__, msglog); |
---|
1226 | return false; |
---|
1227 | } |
---|
1228 | |
---|
1229 | dbi_result_free(result); |
---|
1230 | // Recupera el identificador del nuevo perfil software |
---|
1231 | nwidperfilsoft = dbi_conn_sequence_last(dbi->conn, NULL); |
---|
1232 | |
---|
1233 | // Crea la relación entre perfiles y componenetes software |
---|
1234 | for (i = 0; i < lon; i++) { |
---|
1235 | result = dbi_conn_queryf(dbi->conn, |
---|
1236 | "INSERT perfilessoft_softwares (idperfilsoft,idsoftware)" |
---|
1237 | " VALUES(%d,%d)", nwidperfilsoft, tbidsoftware[i]); |
---|
1238 | if (!result) { |
---|
1239 | dbi_conn_error(dbi->conn, &msglog); |
---|
1240 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1241 | __func__, __LINE__, msglog); |
---|
1242 | return false; |
---|
1243 | } |
---|
1244 | dbi_result_free(result); |
---|
1245 | } |
---|
1246 | } else { // Existe un perfil con todos esos componentes |
---|
1247 | nwidperfilsoft = dbi_result_get_uint(result, "idperfilsoft"); |
---|
1248 | dbi_result_free(result); |
---|
1249 | } |
---|
1250 | |
---|
1251 | if (idperfilsoftware != nwidperfilsoft) { // No coinciden los perfiles |
---|
1252 | // Actualiza el identificador del perfil software del ordenador |
---|
1253 | result = dbi_conn_queryf(dbi->conn, |
---|
1254 | "UPDATE ordenadores_particiones SET idperfilsoft=%d,idimagen=0" |
---|
1255 | " WHERE idordenador=%s AND numpar=%s", nwidperfilsoft, ido, par); |
---|
1256 | if (!result) { // Error al insertar |
---|
1257 | dbi_conn_error(dbi->conn, &msglog); |
---|
1258 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1259 | __func__, __LINE__, msglog); |
---|
1260 | return false; |
---|
1261 | } |
---|
1262 | dbi_result_free(result); |
---|
1263 | } |
---|
1264 | |
---|
1265 | /* DEPURACIÓN DE PERFILES SOFTWARE */ |
---|
1266 | |
---|
1267 | /* Eliminar Relación de softwares con Perfiles software que quedan húerfanos */ |
---|
1268 | result = dbi_conn_queryf(dbi->conn, |
---|
1269 | "DELETE FROM perfilessoft_softwares WHERE idperfilsoft IN "\ |
---|
1270 | " (SELECT idperfilsoft FROM perfilessoft WHERE idperfilsoft NOT IN"\ |
---|
1271 | " (SELECT DISTINCT idperfilsoft from ordenadores_particiones) AND idperfilsoft NOT IN"\ |
---|
1272 | " (SELECT DISTINCT idperfilsoft from imagenes))"); |
---|
1273 | if (!result) { |
---|
1274 | dbi_conn_error(dbi->conn, &msglog); |
---|
1275 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1276 | __func__, __LINE__, msglog); |
---|
1277 | return false; |
---|
1278 | } |
---|
1279 | dbi_result_free(result), |
---|
1280 | /* Eliminar Perfiles software que quedan húerfanos */ |
---|
1281 | result = dbi_conn_queryf(dbi->conn, |
---|
1282 | "DELETE FROM perfilessoft WHERE idperfilsoft NOT IN" |
---|
1283 | " (SELECT DISTINCT idperfilsoft from ordenadores_particiones)"\ |
---|
1284 | " AND idperfilsoft NOT IN"\ |
---|
1285 | " (SELECT DISTINCT idperfilsoft from imagenes)"); |
---|
1286 | if (!result) { |
---|
1287 | dbi_conn_error(dbi->conn, &msglog); |
---|
1288 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1289 | __func__, __LINE__, msglog); |
---|
1290 | return false; |
---|
1291 | } |
---|
1292 | dbi_result_free(result), |
---|
1293 | |
---|
1294 | /* Eliminar Relación de softwares con Perfiles software que quedan húerfanos */ |
---|
1295 | result = dbi_conn_queryf(dbi->conn, |
---|
1296 | "DELETE FROM perfilessoft_softwares WHERE idperfilsoft NOT IN" |
---|
1297 | " (SELECT idperfilsoft from perfilessoft)"); |
---|
1298 | if (!result) { |
---|
1299 | dbi_conn_error(dbi->conn, &msglog); |
---|
1300 | syslog(LOG_ERR, "failed to query database (%s:%d) %s\n", |
---|
1301 | __func__, __LINE__, msglog); |
---|
1302 | return false; |
---|
1303 | } |
---|
1304 | dbi_result_free(result); |
---|
1305 | |
---|
1306 | return true; |
---|
1307 | } |
---|
1308 | |
---|
1309 | static void og_client_release(struct ev_loop *loop, struct og_client *cli) |
---|
1310 | { |
---|
1311 | if (cli->keepalive_idx >= 0) { |
---|
1312 | syslog(LOG_DEBUG, "closing keepalive connection for %s:%hu in slot %d\n", |
---|
1313 | inet_ntoa(cli->addr.sin_addr), |
---|
1314 | ntohs(cli->addr.sin_port), cli->keepalive_idx); |
---|
1315 | tbsockets[cli->keepalive_idx].cli = NULL; |
---|
1316 | } |
---|
1317 | |
---|
1318 | list_del(&cli->list); |
---|
1319 | ev_io_stop(loop, &cli->io); |
---|
1320 | close(cli->io.fd); |
---|
1321 | free(cli); |
---|
1322 | } |
---|
1323 | |
---|
1324 | static void og_client_keepalive(struct ev_loop *loop, struct og_client *cli) |
---|
1325 | { |
---|
1326 | struct og_client *old_cli; |
---|
1327 | |
---|
1328 | old_cli = tbsockets[cli->keepalive_idx].cli; |
---|
1329 | if (old_cli && old_cli != cli) { |
---|
1330 | syslog(LOG_DEBUG, "closing old keepalive connection for %s:%hu\n", |
---|
1331 | inet_ntoa(old_cli->addr.sin_addr), |
---|
1332 | ntohs(old_cli->addr.sin_port)); |
---|
1333 | |
---|
1334 | og_client_release(loop, old_cli); |
---|
1335 | } |
---|
1336 | tbsockets[cli->keepalive_idx].cli = cli; |
---|
1337 | } |
---|
1338 | |
---|
1339 | static void og_client_reset_state(struct og_client *cli) |
---|
1340 | { |
---|
1341 | cli->state = OG_CLIENT_RECEIVING_HEADER; |
---|
1342 | cli->buf_len = 0; |
---|
1343 | } |
---|
1344 | |
---|
1345 | static int og_client_payload_too_large(struct og_client *cli) |
---|
1346 | { |
---|
1347 | char buf[] = "HTTP/1.1 413 Payload Too Large\r\n" |
---|
1348 | "Content-Length: 0\r\n\r\n"; |
---|
1349 | |
---|
1350 | send(og_client_socket(cli), buf, strlen(buf), 0); |
---|
1351 | |
---|
1352 | return -1; |
---|
1353 | } |
---|
1354 | |
---|
1355 | static int og_client_state_recv_hdr_rest(struct og_client *cli) |
---|
1356 | { |
---|
1357 | char *ptr; |
---|
1358 | |
---|
1359 | ptr = strstr(cli->buf, "\r\n\r\n"); |
---|
1360 | if (!ptr) |
---|
1361 | return 0; |
---|
1362 | |
---|
1363 | cli->msg_len = ptr - cli->buf + 4; |
---|
1364 | |
---|
1365 | ptr = strstr(cli->buf, "Content-Length: "); |
---|
1366 | if (ptr) { |
---|
1367 | sscanf(ptr, "Content-Length: %i[^\r\n]", &cli->content_length); |
---|
1368 | if (cli->content_length < 0) |
---|
1369 | return -1; |
---|
1370 | cli->msg_len += cli->content_length; |
---|
1371 | } |
---|
1372 | |
---|
1373 | ptr = strstr(cli->buf, "Authorization: "); |
---|
1374 | if (ptr) |
---|
1375 | sscanf(ptr, "Authorization: %63[^\r\n]", cli->auth_token); |
---|
1376 | |
---|
1377 | return 1; |
---|
1378 | } |
---|
1379 | |
---|
1380 | static int og_client_recv(struct og_client *cli, int events) |
---|
1381 | { |
---|
1382 | struct ev_io *io = &cli->io; |
---|
1383 | int ret; |
---|
1384 | |
---|
1385 | if (events & EV_ERROR) { |
---|
1386 | syslog(LOG_ERR, "unexpected error event from client %s:%hu\n", |
---|
1387 | inet_ntoa(cli->addr.sin_addr), |
---|
1388 | ntohs(cli->addr.sin_port)); |
---|
1389 | return 0; |
---|
1390 | } |
---|
1391 | |
---|
1392 | ret = recv(io->fd, cli->buf + cli->buf_len, |
---|
1393 | sizeof(cli->buf) - cli->buf_len, 0); |
---|
1394 | if (ret <= 0) { |
---|
1395 | if (ret < 0) { |
---|
1396 | syslog(LOG_ERR, "error reading from client %s:%hu (%s)\n", |
---|
1397 | inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port), |
---|
1398 | strerror(errno)); |
---|
1399 | } else { |
---|
1400 | syslog(LOG_DEBUG, "closed connection by %s:%hu\n", |
---|
1401 | inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port)); |
---|
1402 | } |
---|
1403 | return ret; |
---|
1404 | } |
---|
1405 | |
---|
1406 | return ret; |
---|
1407 | } |
---|
1408 | |
---|
1409 | static void og_client_read_cb(struct ev_loop *loop, struct ev_io *io, int events) |
---|
1410 | { |
---|
1411 | struct og_client *cli; |
---|
1412 | int ret; |
---|
1413 | |
---|
1414 | cli = container_of(io, struct og_client, io); |
---|
1415 | |
---|
1416 | ret = og_client_recv(cli, events); |
---|
1417 | if (ret <= 0) |
---|
1418 | goto close; |
---|
1419 | |
---|
1420 | if (cli->keepalive_idx >= 0) |
---|
1421 | return; |
---|
1422 | |
---|
1423 | ev_timer_again(loop, &cli->timer); |
---|
1424 | |
---|
1425 | cli->buf_len += ret; |
---|
1426 | if (cli->buf_len >= sizeof(cli->buf)) { |
---|
1427 | syslog(LOG_ERR, "client request from %s:%hu is too long\n", |
---|
1428 | inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port)); |
---|
1429 | og_client_payload_too_large(cli); |
---|
1430 | goto close; |
---|
1431 | } |
---|
1432 | |
---|
1433 | switch (cli->state) { |
---|
1434 | case OG_CLIENT_RECEIVING_HEADER: |
---|
1435 | ret = og_client_state_recv_hdr_rest(cli); |
---|
1436 | if (ret < 0) |
---|
1437 | goto close; |
---|
1438 | if (!ret) |
---|
1439 | return; |
---|
1440 | |
---|
1441 | cli->state = OG_CLIENT_RECEIVING_PAYLOAD; |
---|
1442 | /* Fall through. */ |
---|
1443 | case OG_CLIENT_RECEIVING_PAYLOAD: |
---|
1444 | /* Still not enough data to process request. */ |
---|
1445 | if (cli->buf_len < cli->msg_len) |
---|
1446 | return; |
---|
1447 | |
---|
1448 | cli->state = OG_CLIENT_PROCESSING_REQUEST; |
---|
1449 | /* fall through. */ |
---|
1450 | case OG_CLIENT_PROCESSING_REQUEST: |
---|
1451 | ret = og_client_state_process_payload_rest(cli); |
---|
1452 | if (ret < 0) { |
---|
1453 | syslog(LOG_ERR, "Failed to process HTTP request from %s:%hu\n", |
---|
1454 | inet_ntoa(cli->addr.sin_addr), |
---|
1455 | ntohs(cli->addr.sin_port)); |
---|
1456 | } |
---|
1457 | if (ret < 0) |
---|
1458 | goto close; |
---|
1459 | |
---|
1460 | if (cli->keepalive_idx < 0) { |
---|
1461 | syslog(LOG_DEBUG, "server closing connection to %s:%hu\n", |
---|
1462 | inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port)); |
---|
1463 | goto close; |
---|
1464 | } else { |
---|
1465 | syslog(LOG_DEBUG, "leaving client %s:%hu in keepalive mode\n", |
---|
1466 | inet_ntoa(cli->addr.sin_addr), |
---|
1467 | ntohs(cli->addr.sin_port)); |
---|
1468 | og_client_keepalive(loop, cli); |
---|
1469 | og_client_reset_state(cli); |
---|
1470 | } |
---|
1471 | break; |
---|
1472 | default: |
---|
1473 | syslog(LOG_ERR, "unknown state, critical internal error\n"); |
---|
1474 | goto close; |
---|
1475 | } |
---|
1476 | return; |
---|
1477 | close: |
---|
1478 | ev_timer_stop(loop, &cli->timer); |
---|
1479 | og_client_release(loop, cli); |
---|
1480 | } |
---|
1481 | |
---|
1482 | enum og_agent_state { |
---|
1483 | OG_AGENT_RECEIVING_HEADER = 0, |
---|
1484 | OG_AGENT_RECEIVING_PAYLOAD, |
---|
1485 | OG_AGENT_PROCESSING_RESPONSE, |
---|
1486 | }; |
---|
1487 | |
---|
1488 | static int og_agent_state_recv_hdr_rest(struct og_client *cli) |
---|
1489 | { |
---|
1490 | char *ptr; |
---|
1491 | |
---|
1492 | ptr = strstr(cli->buf, "\r\n\r\n"); |
---|
1493 | if (!ptr) |
---|
1494 | return 0; |
---|
1495 | |
---|
1496 | cli->msg_len = ptr - cli->buf + 4; |
---|
1497 | |
---|
1498 | ptr = strstr(cli->buf, "Content-Length: "); |
---|
1499 | if (ptr) { |
---|
1500 | sscanf(ptr, "Content-Length: %i[^\r\n]", &cli->content_length); |
---|
1501 | if (cli->content_length < 0) |
---|
1502 | return -1; |
---|
1503 | cli->msg_len += cli->content_length; |
---|
1504 | } |
---|
1505 | |
---|
1506 | return 1; |
---|
1507 | } |
---|
1508 | |
---|
1509 | static void og_agent_reset_state(struct og_client *cli) |
---|
1510 | { |
---|
1511 | cli->state = OG_AGENT_RECEIVING_HEADER; |
---|
1512 | cli->buf_len = 0; |
---|
1513 | cli->content_length = 0; |
---|
1514 | memset(cli->buf, 0, sizeof(cli->buf)); |
---|
1515 | } |
---|
1516 | |
---|
1517 | static void og_agent_deliver_pending_cmd(struct og_client *cli) |
---|
1518 | { |
---|
1519 | const struct og_cmd *cmd; |
---|
1520 | |
---|
1521 | cmd = og_cmd_find(inet_ntoa(cli->addr.sin_addr)); |
---|
1522 | if (!cmd) |
---|
1523 | return; |
---|
1524 | |
---|
1525 | og_send_request(cmd->method, cmd->type, &cmd->params, cmd->json); |
---|
1526 | cli->last_cmd_id = cmd->id; |
---|
1527 | |
---|
1528 | og_cmd_free(cmd); |
---|
1529 | } |
---|
1530 | |
---|
1531 | static void og_agent_read_cb(struct ev_loop *loop, struct ev_io *io, int events) |
---|
1532 | { |
---|
1533 | struct og_client *cli; |
---|
1534 | int ret; |
---|
1535 | |
---|
1536 | cli = container_of(io, struct og_client, io); |
---|
1537 | |
---|
1538 | ret = og_client_recv(cli, events); |
---|
1539 | if (ret <= 0) |
---|
1540 | goto close; |
---|
1541 | |
---|
1542 | ev_timer_again(loop, &cli->timer); |
---|
1543 | |
---|
1544 | cli->buf_len += ret; |
---|
1545 | if (cli->buf_len >= sizeof(cli->buf)) { |
---|
1546 | syslog(LOG_ERR, "client request from %s:%hu is too long\n", |
---|
1547 | inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port)); |
---|
1548 | goto close; |
---|
1549 | } |
---|
1550 | |
---|
1551 | switch (cli->state) { |
---|
1552 | case OG_AGENT_RECEIVING_HEADER: |
---|
1553 | ret = og_agent_state_recv_hdr_rest(cli); |
---|
1554 | if (ret < 0) |
---|
1555 | goto close; |
---|
1556 | if (!ret) |
---|
1557 | return; |
---|
1558 | |
---|
1559 | cli->state = OG_AGENT_RECEIVING_PAYLOAD; |
---|
1560 | /* Fall through. */ |
---|
1561 | case OG_AGENT_RECEIVING_PAYLOAD: |
---|
1562 | /* Still not enough data to process request. */ |
---|
1563 | if (cli->buf_len < cli->msg_len) |
---|
1564 | return; |
---|
1565 | |
---|
1566 | cli->state = OG_AGENT_PROCESSING_RESPONSE; |
---|
1567 | /* fall through. */ |
---|
1568 | case OG_AGENT_PROCESSING_RESPONSE: |
---|
1569 | ret = og_agent_state_process_response(cli); |
---|
1570 | if (ret < 0) { |
---|
1571 | syslog(LOG_ERR, "Failed to process HTTP request from %s:%hu\n", |
---|
1572 | inet_ntoa(cli->addr.sin_addr), |
---|
1573 | ntohs(cli->addr.sin_port)); |
---|
1574 | goto close; |
---|
1575 | } else if (ret == 0) { |
---|
1576 | og_agent_deliver_pending_cmd(cli); |
---|
1577 | } |
---|
1578 | |
---|
1579 | syslog(LOG_DEBUG, "leaving client %s:%hu in keepalive mode\n", |
---|
1580 | inet_ntoa(cli->addr.sin_addr), |
---|
1581 | ntohs(cli->addr.sin_port)); |
---|
1582 | og_agent_reset_state(cli); |
---|
1583 | break; |
---|
1584 | default: |
---|
1585 | syslog(LOG_ERR, "unknown state, critical internal error\n"); |
---|
1586 | goto close; |
---|
1587 | } |
---|
1588 | return; |
---|
1589 | close: |
---|
1590 | ev_timer_stop(loop, &cli->timer); |
---|
1591 | og_client_release(loop, cli); |
---|
1592 | } |
---|
1593 | |
---|
1594 | static void og_client_timer_cb(struct ev_loop *loop, ev_timer *timer, int events) |
---|
1595 | { |
---|
1596 | struct og_client *cli; |
---|
1597 | |
---|
1598 | cli = container_of(timer, struct og_client, timer); |
---|
1599 | if (cli->keepalive_idx >= 0) { |
---|
1600 | ev_timer_again(loop, &cli->timer); |
---|
1601 | return; |
---|
1602 | } |
---|
1603 | syslog(LOG_ERR, "timeout request for client %s:%hu\n", |
---|
1604 | inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port)); |
---|
1605 | |
---|
1606 | og_client_release(loop, cli); |
---|
1607 | } |
---|
1608 | |
---|
1609 | static void og_agent_send_refresh(struct og_client *cli) |
---|
1610 | { |
---|
1611 | struct og_msg_params params; |
---|
1612 | int err; |
---|
1613 | |
---|
1614 | params.ips_array[0] = inet_ntoa(cli->addr.sin_addr); |
---|
1615 | params.ips_array_len = 1; |
---|
1616 | |
---|
1617 | err = og_send_request(OG_METHOD_GET, OG_CMD_REFRESH, ¶ms, NULL); |
---|
1618 | if (err < 0) { |
---|
1619 | syslog(LOG_ERR, "Can't send refresh to: %s\n", |
---|
1620 | params.ips_array[0]); |
---|
1621 | } else { |
---|
1622 | syslog(LOG_INFO, "Sent refresh to: %s\n", |
---|
1623 | params.ips_array[0]); |
---|
1624 | } |
---|
1625 | } |
---|
1626 | |
---|
1627 | static int socket_rest, socket_agent_rest; |
---|
1628 | |
---|
1629 | static void og_server_accept_cb(struct ev_loop *loop, struct ev_io *io, |
---|
1630 | int events) |
---|
1631 | { |
---|
1632 | struct sockaddr_in client_addr; |
---|
1633 | socklen_t addrlen = sizeof(client_addr); |
---|
1634 | struct og_client *cli; |
---|
1635 | int client_sd; |
---|
1636 | |
---|
1637 | if (events & EV_ERROR) |
---|
1638 | return; |
---|
1639 | |
---|
1640 | client_sd = accept(io->fd, (struct sockaddr *)&client_addr, &addrlen); |
---|
1641 | if (client_sd < 0) { |
---|
1642 | syslog(LOG_ERR, "cannot accept client connection\n"); |
---|
1643 | return; |
---|
1644 | } |
---|
1645 | |
---|
1646 | cli = (struct og_client *)calloc(1, sizeof(struct og_client)); |
---|
1647 | if (!cli) { |
---|
1648 | close(client_sd); |
---|
1649 | return; |
---|
1650 | } |
---|
1651 | memcpy(&cli->addr, &client_addr, sizeof(client_addr)); |
---|
1652 | if (io->fd == socket_agent_rest) |
---|
1653 | cli->keepalive_idx = 0; |
---|
1654 | else |
---|
1655 | cli->keepalive_idx = -1; |
---|
1656 | |
---|
1657 | if (io->fd == socket_rest) |
---|
1658 | cli->rest = true; |
---|
1659 | else if (io->fd == socket_agent_rest) |
---|
1660 | cli->agent = true; |
---|
1661 | |
---|
1662 | syslog(LOG_DEBUG, "connection from client %s:%hu\n", |
---|
1663 | inet_ntoa(cli->addr.sin_addr), ntohs(cli->addr.sin_port)); |
---|
1664 | |
---|
1665 | if (io->fd == socket_agent_rest) |
---|
1666 | ev_io_init(&cli->io, og_agent_read_cb, client_sd, EV_READ); |
---|
1667 | else |
---|
1668 | ev_io_init(&cli->io, og_client_read_cb, client_sd, EV_READ); |
---|
1669 | |
---|
1670 | ev_io_start(loop, &cli->io); |
---|
1671 | if (io->fd == socket_agent_rest) { |
---|
1672 | ev_timer_init(&cli->timer, og_client_timer_cb, |
---|
1673 | OG_AGENT_CLIENT_TIMEOUT, 0.); |
---|
1674 | } else { |
---|
1675 | ev_timer_init(&cli->timer, og_client_timer_cb, |
---|
1676 | OG_CLIENT_TIMEOUT, 0.); |
---|
1677 | } |
---|
1678 | ev_timer_start(loop, &cli->timer); |
---|
1679 | og_client_add(cli); |
---|
1680 | |
---|
1681 | if (io->fd == socket_agent_rest) { |
---|
1682 | og_agent_send_refresh(cli); |
---|
1683 | } |
---|
1684 | } |
---|
1685 | |
---|
1686 | static int og_socket_server_init(const char *port) |
---|
1687 | { |
---|
1688 | struct sockaddr_in local; |
---|
1689 | int sd, on = 1; |
---|
1690 | |
---|
1691 | sd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); |
---|
1692 | if (sd < 0) { |
---|
1693 | syslog(LOG_ERR, "cannot create main socket\n"); |
---|
1694 | return -1; |
---|
1695 | } |
---|
1696 | setsockopt(sd, SOL_SOCKET, SO_REUSEPORT, &on, sizeof(int)); |
---|
1697 | |
---|
1698 | local.sin_addr.s_addr = htonl(INADDR_ANY); |
---|
1699 | local.sin_family = AF_INET; |
---|
1700 | local.sin_port = htons(atoi(port)); |
---|
1701 | |
---|
1702 | if (bind(sd, (struct sockaddr *) &local, sizeof(local)) < 0) { |
---|
1703 | close(sd); |
---|
1704 | syslog(LOG_ERR, "cannot bind socket\n"); |
---|
1705 | return -1; |
---|
1706 | } |
---|
1707 | |
---|
1708 | listen(sd, 250); |
---|
1709 | |
---|
1710 | return sd; |
---|
1711 | } |
---|
1712 | |
---|
1713 | struct ev_loop *og_loop; |
---|
1714 | |
---|
1715 | int main(int argc, char *argv[]) |
---|
1716 | { |
---|
1717 | struct ev_io ev_io_server_rest, ev_io_agent_rest; |
---|
1718 | int i; |
---|
1719 | |
---|
1720 | og_loop = ev_default_loop(0); |
---|
1721 | |
---|
1722 | if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) |
---|
1723 | exit(EXIT_FAILURE); |
---|
1724 | |
---|
1725 | openlog("ogAdmServer", LOG_PID, LOG_DAEMON); |
---|
1726 | |
---|
1727 | /*-------------------------------------------------------------------------------------------------------- |
---|
1728 | Validación de parámetros de ejecución y lectura del fichero de configuración del servicio |
---|
1729 | ---------------------------------------------------------------------------------------------------------*/ |
---|
1730 | if (!validacionParametros(argc, argv, 1)) // Valida parámetros de ejecución |
---|
1731 | exit(EXIT_FAILURE); |
---|
1732 | |
---|
1733 | if (!tomaConfiguracion(szPathFileCfg)) { // Toma parametros de configuracion |
---|
1734 | exit(EXIT_FAILURE); |
---|
1735 | } |
---|
1736 | |
---|
1737 | /*-------------------------------------------------------------------------------------------------------- |
---|
1738 | // Inicializa array de información de los clientes |
---|
1739 | ---------------------------------------------------------------------------------------------------------*/ |
---|
1740 | for (i = 0; i < MAXIMOS_CLIENTES; i++) { |
---|
1741 | tbsockets[i].ip[0] = '\0'; |
---|
1742 | tbsockets[i].cli = NULL; |
---|
1743 | } |
---|
1744 | /*-------------------------------------------------------------------------------------------------------- |
---|
1745 | Creación y configuración del socket del servicio |
---|
1746 | ---------------------------------------------------------------------------------------------------------*/ |
---|
1747 | |
---|
1748 | socket_rest = og_socket_server_init("8888"); |
---|
1749 | if (socket_rest < 0) |
---|
1750 | exit(EXIT_FAILURE); |
---|
1751 | |
---|
1752 | ev_io_init(&ev_io_server_rest, og_server_accept_cb, socket_rest, EV_READ); |
---|
1753 | ev_io_start(og_loop, &ev_io_server_rest); |
---|
1754 | |
---|
1755 | socket_agent_rest = og_socket_server_init("8889"); |
---|
1756 | if (socket_agent_rest < 0) |
---|
1757 | exit(EXIT_FAILURE); |
---|
1758 | |
---|
1759 | ev_io_init(&ev_io_agent_rest, og_server_accept_cb, socket_agent_rest, EV_READ); |
---|
1760 | ev_io_start(og_loop, &ev_io_agent_rest); |
---|
1761 | |
---|
1762 | if (og_dbi_schedule_get() < 0) |
---|
1763 | exit(EXIT_FAILURE); |
---|
1764 | |
---|
1765 | og_schedule_next(og_loop); |
---|
1766 | |
---|
1767 | syslog(LOG_INFO, "Waiting for connections\n"); |
---|
1768 | |
---|
1769 | while (1) |
---|
1770 | ev_loop(og_loop, 0); |
---|
1771 | |
---|
1772 | exit(EXIT_SUCCESS); |
---|
1773 | } |
---|