1 /**
2 * collectd - src/exec.c
3 * Copyright (C) 2007-2010 Florian octo Forster
4 * Copyright (C) 2007-2009 Sebastian Harl
5 * Copyright (C) 2008 Peter Holik
6 *
7 * This program is free software; you can redistribute it and/or modify it
8 * under the terms of the GNU General Public License as published by the
9 * Free Software Foundation; only version 2 of the License is applicable.
10 *
11 * This program is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along
17 * with this program; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
19 *
20 * Authors:
21 * Florian octo Forster <octo at verplant.org>
22 * Sebastian Harl <sh at tokkee.org>
23 * Peter Holik <peter at holik.at>
24 **/
26 #define _BSD_SOURCE /* For setgroups */
28 #include "collectd.h"
29 #include "common.h"
30 #include "plugin.h"
32 #include "utils_cmd_putval.h"
33 #include "utils_cmd_putnotif.h"
35 #include <sys/types.h>
36 #include <pwd.h>
37 #include <grp.h>
38 #include <signal.h>
40 #include <pthread.h>
42 #define PL_NORMAL 0x01
43 #define PL_NOTIF_ACTION 0x02
45 #define PL_RUNNING 0x10
47 /*
48 * Private data types
49 */
50 /*
51 * Access to this structure is serialized using the `pl_lock' lock and the
52 * `PL_RUNNING' flag. The execution of notifications is *not* serialized, so
53 * all functions used to handle notifications MUST NOT write to this structure.
54 * The `pid' and `status' fields are thus unused if the `PL_NOTIF_ACTION' flag
55 * is set.
56 * The `PL_RUNNING' flag is set in `exec_read' and unset in `exec_read_one'.
57 */
58 struct program_list_s;
59 typedef struct program_list_s program_list_t;
60 struct program_list_s
61 {
62 char *user;
63 char *group;
64 char *exec;
65 char **argv;
66 int pid;
67 int status;
68 int flags;
69 program_list_t *next;
70 };
72 typedef struct program_list_and_notification_s
73 {
74 program_list_t *pl;
75 notification_t n;
76 } program_list_and_notification_t;
78 /*
79 * Private variables
80 */
81 static program_list_t *pl_head = NULL;
82 static pthread_mutex_t pl_lock = PTHREAD_MUTEX_INITIALIZER;
84 /*
85 * Functions
86 */
87 static void sigchld_handler (int __attribute__((unused)) signal) /* {{{ */
88 {
89 pid_t pid;
90 int status;
91 while ((pid = waitpid (-1, &status, WNOHANG)) > 0)
92 {
93 program_list_t *pl;
94 for (pl = pl_head; pl != NULL; pl = pl->next)
95 if (pl->pid == pid)
96 break;
97 if (pl != NULL)
98 pl->status = status;
99 } /* while (waitpid) */
100 } /* void sigchld_handler }}} */
102 static int exec_config_exec (oconfig_item_t *ci) /* {{{ */
103 {
104 program_list_t *pl;
105 char buffer[128];
106 int i;
108 if (ci->children_num != 0)
109 {
110 WARNING ("exec plugin: The config option `%s' may not be a block.",
111 ci->key);
112 return (-1);
113 }
114 if (ci->values_num < 2)
115 {
116 WARNING ("exec plugin: The config option `%s' needs at least two "
117 "arguments.", ci->key);
118 return (-1);
119 }
120 if ((ci->values[0].type != OCONFIG_TYPE_STRING)
121 || (ci->values[1].type != OCONFIG_TYPE_STRING))
122 {
123 WARNING ("exec plugin: The first two arguments to the `%s' option must "
124 "be string arguments.", ci->key);
125 return (-1);
126 }
128 pl = (program_list_t *) malloc (sizeof (program_list_t));
129 if (pl == NULL)
130 {
131 ERROR ("exec plugin: malloc failed.");
132 return (-1);
133 }
134 memset (pl, '\0', sizeof (program_list_t));
136 if (strcasecmp ("NotificationExec", ci->key) == 0)
137 pl->flags |= PL_NOTIF_ACTION;
138 else
139 pl->flags |= PL_NORMAL;
141 pl->user = strdup (ci->values[0].value.string);
142 if (pl->user == NULL)
143 {
144 ERROR ("exec plugin: strdup failed.");
145 sfree (pl);
146 return (-1);
147 }
149 pl->group = strchr (pl->user, ':');
150 if (pl->group != NULL)
151 {
152 *pl->group = '\0';
153 pl->group++;
154 }
156 pl->exec = strdup (ci->values[1].value.string);
157 if (pl->exec == NULL)
158 {
159 ERROR ("exec plugin: strdup failed.");
160 sfree (pl->user);
161 sfree (pl);
162 return (-1);
163 }
165 pl->argv = (char **) malloc (ci->values_num * sizeof (char *));
166 if (pl->argv == NULL)
167 {
168 ERROR ("exec plugin: malloc failed.");
169 sfree (pl->exec);
170 sfree (pl->user);
171 sfree (pl);
172 return (-1);
173 }
174 memset (pl->argv, '\0', ci->values_num * sizeof (char *));
176 {
177 char *tmp = strrchr (ci->values[1].value.string, '/');
178 if (tmp == NULL)
179 sstrncpy (buffer, ci->values[1].value.string, sizeof (buffer));
180 else
181 sstrncpy (buffer, tmp + 1, sizeof (buffer));
182 }
183 pl->argv[0] = strdup (buffer);
184 if (pl->argv[0] == NULL)
185 {
186 ERROR ("exec plugin: malloc failed.");
187 sfree (pl->argv);
188 sfree (pl->exec);
189 sfree (pl->user);
190 sfree (pl);
191 return (-1);
192 }
194 for (i = 1; i < (ci->values_num - 1); i++)
195 {
196 if (ci->values[i + 1].type == OCONFIG_TYPE_STRING)
197 {
198 pl->argv[i] = strdup (ci->values[i + 1].value.string);
199 }
200 else
201 {
202 if (ci->values[i + 1].type == OCONFIG_TYPE_NUMBER)
203 {
204 ssnprintf (buffer, sizeof (buffer), "%lf",
205 ci->values[i + 1].value.number);
206 }
207 else
208 {
209 if (ci->values[i + 1].value.boolean)
210 sstrncpy (buffer, "true", sizeof (buffer));
211 else
212 sstrncpy (buffer, "false", sizeof (buffer));
213 }
215 pl->argv[i] = strdup (buffer);
216 }
218 if (pl->argv[i] == NULL)
219 {
220 ERROR ("exec plugin: strdup failed.");
221 break;
222 }
223 } /* for (i) */
225 if (i < (ci->values_num - 1))
226 {
227 while ((--i) >= 0)
228 {
229 sfree (pl->argv[i]);
230 }
231 sfree (pl->argv);
232 sfree (pl->exec);
233 sfree (pl->user);
234 sfree (pl);
235 return (-1);
236 }
238 for (i = 0; pl->argv[i] != NULL; i++)
239 {
240 DEBUG ("exec plugin: argv[%i] = %s", i, pl->argv[i]);
241 }
243 pl->next = pl_head;
244 pl_head = pl;
246 return (0);
247 } /* int exec_config_exec }}} */
249 static int exec_config (oconfig_item_t *ci) /* {{{ */
250 {
251 int i;
253 for (i = 0; i < ci->children_num; i++)
254 {
255 oconfig_item_t *child = ci->children + i;
256 if ((strcasecmp ("Exec", child->key) == 0)
257 || (strcasecmp ("NotificationExec", child->key) == 0))
258 exec_config_exec (child);
259 else
260 {
261 WARNING ("exec plugin: Unknown config option `%s'.", child->key);
262 }
263 } /* for (i) */
265 return (0);
266 } /* int exec_config }}} */
268 static void set_environment (void) /* {{{ */
269 {
270 char buffer[1024];
272 #ifdef HAVE_SETENV
273 ssnprintf (buffer, sizeof (buffer), "%.3f",
274 CDTIME_T_TO_DOUBLE (plugin_get_interval ()));
275 setenv ("COLLECTD_INTERVAL", buffer, /* overwrite = */ 1);
277 ssnprintf (buffer, sizeof (buffer), "%s", hostname_g);
278 setenv ("COLLECTD_HOSTNAME", buffer, /* overwrite = */ 1);
279 #else
280 ssnprintf (buffer, sizeof (buffer), "COLLECTD_INTERVAL=%.3f",
281 CDTIME_T_TO_DOUBLE (plugin_get_interval ()));
282 putenv (buffer);
284 ssnprintf (buffer, sizeof (buffer), "COLLECTD_HOSTNAME=%s", hostname_g);
285 putenv (buffer);
286 #endif
287 } /* }}} void set_environment */
289 __attribute__((noreturn))
290 static void exec_child (program_list_t *pl, int uid, int gid, int egid) /* {{{ */
291 {
292 int status;
293 char errbuf[1024];
295 #if HAVE_SETGROUPS
296 if (getuid () == 0)
297 {
298 gid_t glist[2];
299 size_t glist_len;
301 glist[0] = gid;
302 glist_len = 1;
304 if ((gid != egid) && (egid != -1))
305 {
306 glist[1] = egid;
307 glist_len = 2;
308 }
310 setgroups (glist_len, glist);
311 }
312 #endif /* HAVE_SETGROUPS */
314 status = setgid (gid);
315 if (status != 0)
316 {
317 ERROR ("exec plugin: setgid (%i) failed: %s",
318 gid, sstrerror (errno, errbuf, sizeof (errbuf)));
319 exit (-1);
320 }
322 if (egid != -1)
323 {
324 status = setegid (egid);
325 if (status != 0)
326 {
327 ERROR ("exec plugin: setegid (%i) failed: %s",
328 egid, sstrerror (errno, errbuf, sizeof (errbuf)));
329 exit (-1);
330 }
331 }
333 status = setuid (uid);
334 if (status != 0)
335 {
336 ERROR ("exec plugin: setuid (%i) failed: %s",
337 uid, sstrerror (errno, errbuf, sizeof (errbuf)));
338 exit (-1);
339 }
341 execvp (pl->exec, pl->argv);
343 ERROR ("exec plugin: Failed to execute ``%s'': %s",
344 pl->exec, sstrerror (errno, errbuf, sizeof (errbuf)));
345 exit (-1);
346 } /* void exec_child }}} */
348 static void reset_signal_mask (void) /* {{{ */
349 {
350 sigset_t ss;
352 memset (&ss, 0, sizeof (ss));
353 sigemptyset (&ss);
354 sigprocmask (SIG_SETMASK, &ss, /* old mask = */ NULL);
355 } /* }}} void reset_signal_mask */
357 static int create_pipe (int fd_pipe[2]) /* {{{ */
358 {
359 char errbuf[1024];
360 int status;
362 status = pipe (fd_pipe);
363 if (status != 0)
364 {
365 ERROR ("exec plugin: pipe failed: %s",
366 sstrerror (errno, errbuf, sizeof (errbuf)));
367 return (-1);
368 }
370 return 0;
371 } /* }}} int create_pipe */
373 static void close_pipe (int fd_pipe[2]) /* {{{ */
374 {
375 if (fd_pipe[0] != -1)
376 close (fd_pipe[0]);
378 if (fd_pipe[1] != -1)
379 close (fd_pipe[1]);
380 } /* }}} void close_pipe */
382 /*
383 * Creates three pipes (one for reading, one for writing and one for errors),
384 * forks a child, sets up the pipes so that fd_in is connected to STDIN of
385 * the child and fd_out is connected to STDOUT and fd_err is connected to STDERR
386 * of the child. Then is calls `exec_child'.
387 */
388 static int fork_child (program_list_t *pl, int *fd_in, int *fd_out, int *fd_err) /* {{{ */
389 {
390 int fd_pipe_in[2] = {-1, -1};
391 int fd_pipe_out[2] = {-1, -1};
392 int fd_pipe_err[2] = {-1, -1};
393 char errbuf[1024];
394 int status;
395 int pid;
397 int uid;
398 int gid;
399 int egid;
401 struct passwd *sp_ptr;
402 struct passwd sp;
403 char nambuf[2048];
405 if (pl->pid != 0)
406 return (-1);
408 if ((create_pipe(fd_pipe_in) == -1)
409 || (create_pipe(fd_pipe_out) == -1)
410 || (create_pipe(fd_pipe_err) == -1))
411 goto failed;
413 sp_ptr = NULL;
414 status = getpwnam_r (pl->user, &sp, nambuf, sizeof (nambuf), &sp_ptr);
415 if (status != 0)
416 {
417 ERROR ("exec plugin: Failed to get user information for user ``%s'': %s",
418 pl->user, sstrerror (errno, errbuf, sizeof (errbuf)));
419 goto failed;
420 }
422 if (sp_ptr == NULL)
423 {
424 ERROR ("exec plugin: No such user: `%s'", pl->user);
425 goto failed;
426 }
428 uid = sp.pw_uid;
429 gid = sp.pw_gid;
430 if (uid == 0)
431 {
432 ERROR ("exec plugin: Cowardly refusing to exec program as root.");
433 goto failed;
434 }
436 /* The group configured in the configfile is set as effective group, because
437 * this way the forked process can (re-)gain the user's primary group. */
438 egid = -1;
439 if (NULL != pl->group)
440 {
441 if ('\0' != *pl->group) {
442 struct group *gr_ptr = NULL;
443 struct group gr;
445 status = getgrnam_r (pl->group, &gr, nambuf, sizeof (nambuf), &gr_ptr);
446 if (0 != status)
447 {
448 ERROR ("exec plugin: Failed to get group information "
449 "for group ``%s'': %s", pl->group,
450 sstrerror (errno, errbuf, sizeof (errbuf)));
451 goto failed;
452 }
453 if (NULL == gr_ptr)
454 {
455 ERROR ("exec plugin: No such group: `%s'", pl->group);
456 goto failed;
457 }
459 egid = gr.gr_gid;
460 }
461 else
462 {
463 egid = gid;
464 }
465 } /* if (pl->group == NULL) */
467 pid = fork ();
468 if (pid < 0)
469 {
470 ERROR ("exec plugin: fork failed: %s",
471 sstrerror (errno, errbuf, sizeof (errbuf)));
472 goto failed;
473 }
474 else if (pid == 0)
475 {
476 int fd_num;
477 int fd;
479 /* Close all file descriptors but the pipe end we need. */
480 fd_num = getdtablesize ();
481 for (fd = 0; fd < fd_num; fd++)
482 {
483 if ((fd == fd_pipe_in[0])
484 || (fd == fd_pipe_out[1])
485 || (fd == fd_pipe_err[1]))
486 continue;
487 close (fd);
488 }
490 /* Connect the `in' pipe to STDIN */
491 if (fd_pipe_in[0] != STDIN_FILENO)
492 {
493 dup2 (fd_pipe_in[0], STDIN_FILENO);
494 close (fd_pipe_in[0]);
495 }
497 /* Now connect the `out' pipe to STDOUT */
498 if (fd_pipe_out[1] != STDOUT_FILENO)
499 {
500 dup2 (fd_pipe_out[1], STDOUT_FILENO);
501 close (fd_pipe_out[1]);
502 }
504 /* Now connect the `err' pipe to STDERR */
505 if (fd_pipe_err[1] != STDERR_FILENO)
506 {
507 dup2 (fd_pipe_err[1], STDERR_FILENO);
508 close (fd_pipe_err[1]);
509 }
511 set_environment ();
513 /* Unblock all signals */
514 reset_signal_mask ();
516 exec_child (pl, uid, gid, egid);
517 /* does not return */
518 }
520 close (fd_pipe_in[0]);
521 close (fd_pipe_out[1]);
522 close (fd_pipe_err[1]);
524 if (fd_in != NULL)
525 *fd_in = fd_pipe_in[1];
526 else
527 close (fd_pipe_in[1]);
529 if (fd_out != NULL)
530 *fd_out = fd_pipe_out[0];
531 else
532 close (fd_pipe_out[0]);
534 if (fd_err != NULL)
535 *fd_err = fd_pipe_err[0];
536 else
537 close (fd_pipe_err[0]);
539 return (pid);
541 failed:
542 close_pipe(fd_pipe_in);
543 close_pipe(fd_pipe_out);
544 close_pipe(fd_pipe_err);
546 return (-1);
547 } /* int fork_child }}} */
549 static int parse_line (char *buffer) /* {{{ */
550 {
551 if (strncasecmp ("PUTVAL", buffer, strlen ("PUTVAL")) == 0)
552 return (handle_putval (stdout, buffer));
553 else if (strncasecmp ("PUTNOTIF", buffer, strlen ("PUTNOTIF")) == 0)
554 return (handle_putnotif (stdout, buffer));
555 else
556 {
557 ERROR ("exec plugin: Unable to parse command, ignoring line: \"%s\"",
558 buffer);
559 return (-1);
560 }
561 } /* int parse_line }}} */
563 static void *exec_read_one (void *arg) /* {{{ */
564 {
565 program_list_t *pl = (program_list_t *) arg;
566 int fd, fd_err, highest_fd;
567 fd_set fdset, copy;
568 int status;
569 char buffer[1200]; /* if not completely read */
570 char buffer_err[1024];
571 char *pbuffer = buffer;
572 char *pbuffer_err = buffer_err;
574 status = fork_child (pl, NULL, &fd, &fd_err);
575 if (status < 0)
576 {
577 /* Reset the "running" flag */
578 pthread_mutex_lock (&pl_lock);
579 pl->flags &= ~PL_RUNNING;
580 pthread_mutex_unlock (&pl_lock);
581 pthread_exit ((void *) 1);
582 }
583 pl->pid = status;
585 assert (pl->pid != 0);
587 FD_ZERO( &fdset );
588 FD_SET(fd, &fdset);
589 FD_SET(fd_err, &fdset);
591 /* Determine the highest file descriptor */
592 highest_fd = (fd > fd_err) ? fd : fd_err;
594 /* We use a copy of fdset, as select modifies it */
595 copy = fdset;
597 while (1)
598 {
599 int len;
601 status = select (highest_fd + 1, ©, NULL, NULL, NULL);
602 if (status < 0)
603 {
604 if (errno == EINTR)
605 continue;
606 break;
607 }
609 if (FD_ISSET(fd, ©))
610 {
611 char *pnl;
613 len = read(fd, pbuffer, sizeof(buffer) - 1 - (pbuffer - buffer));
615 if (len < 0)
616 {
617 if (errno == EAGAIN || errno == EINTR) continue;
618 break;
619 }
620 else if (len == 0) break; /* We've reached EOF */
622 pbuffer[len] = '\0';
624 len += pbuffer - buffer;
625 pbuffer = buffer;
627 while ((pnl = strchr(pbuffer, '\n')))
628 {
629 *pnl = '\0';
630 if (*(pnl-1) == '\r' ) *(pnl-1) = '\0';
632 parse_line (pbuffer);
634 pbuffer = ++pnl;
635 }
636 /* not completely read ? */
637 if (pbuffer - buffer < len)
638 {
639 len -= pbuffer - buffer;
640 memmove(buffer, pbuffer, len);
641 pbuffer = buffer + len;
642 }
643 else
644 pbuffer = buffer;
645 }
646 else if (FD_ISSET(fd_err, ©))
647 {
648 char *pnl;
650 len = read(fd_err, pbuffer_err, sizeof(buffer_err) - 1 - (pbuffer_err - buffer_err));
652 if (len < 0)
653 {
654 if (errno == EAGAIN || errno == EINTR)
655 continue;
656 break;
657 }
658 else if (len == 0)
659 {
660 /* We've reached EOF */
661 NOTICE ("exec plugin: Program `%s' has closed STDERR.", pl->exec);
663 /* Remove file descriptor form select() set. */
664 FD_CLR (fd_err, &fdset);
665 copy = fdset;
666 highest_fd = fd;
668 /* Clean up file descriptor */
669 close (fd_err);
670 fd_err = -1;
671 continue;
672 }
674 pbuffer_err[len] = '\0';
676 len += pbuffer_err - buffer_err;
677 pbuffer_err = buffer_err;
679 while ((pnl = strchr(pbuffer_err, '\n')))
680 {
681 *pnl = '\0';
682 if (*(pnl-1) == '\r' ) *(pnl-1) = '\0';
684 ERROR ("exec plugin: exec_read_one: error = %s", pbuffer_err);
686 pbuffer_err = ++pnl;
687 }
688 /* not completely read ? */
689 if (pbuffer_err - buffer_err < len)
690 {
691 len -= pbuffer_err - buffer_err;
692 memmove(buffer_err, pbuffer_err, len);
693 pbuffer_err = buffer_err + len;
694 }
695 else
696 pbuffer_err = buffer_err;
697 }
698 /* reset copy */
699 copy = fdset;
700 }
702 DEBUG ("exec plugin: exec_read_one: Waiting for `%s' to exit.", pl->exec);
703 if (waitpid (pl->pid, &status, 0) > 0)
704 pl->status = status;
706 DEBUG ("exec plugin: Child %i exited with status %i.",
707 (int) pl->pid, pl->status);
709 pl->pid = 0;
711 pthread_mutex_lock (&pl_lock);
712 pl->flags &= ~PL_RUNNING;
713 pthread_mutex_unlock (&pl_lock);
715 close (fd);
716 if (fd_err >= 0)
717 close (fd_err);
719 pthread_exit ((void *) 0);
720 return (NULL);
721 } /* void *exec_read_one }}} */
723 static void *exec_notification_one (void *arg) /* {{{ */
724 {
725 program_list_t *pl = ((program_list_and_notification_t *) arg)->pl;
726 notification_t *n = &((program_list_and_notification_t *) arg)->n;
727 notification_meta_t *meta;
728 int fd;
729 FILE *fh;
730 int pid;
731 int status;
732 const char *severity;
734 pid = fork_child (pl, &fd, NULL, NULL);
735 if (pid < 0) {
736 sfree (arg);
737 pthread_exit ((void *) 1);
738 }
740 fh = fdopen (fd, "w");
741 if (fh == NULL)
742 {
743 char errbuf[1024];
744 ERROR ("exec plugin: fdopen (%i) failed: %s", fd,
745 sstrerror (errno, errbuf, sizeof (errbuf)));
746 kill (pl->pid, SIGTERM);
747 pl->pid = 0;
748 close (fd);
749 sfree (arg);
750 pthread_exit ((void *) 1);
751 }
753 severity = "FAILURE";
754 if (n->severity == NOTIF_WARNING)
755 severity = "WARNING";
756 else if (n->severity == NOTIF_OKAY)
757 severity = "OKAY";
759 fprintf (fh,
760 "Severity: %s\n"
761 "Time: %.3f\n",
762 severity, CDTIME_T_TO_DOUBLE (n->time));
764 /* Print the optional fields */
765 if (strlen (n->host) > 0)
766 fprintf (fh, "Host: %s\n", n->host);
767 if (strlen (n->plugin) > 0)
768 fprintf (fh, "Plugin: %s\n", n->plugin);
769 if (strlen (n->plugin_instance) > 0)
770 fprintf (fh, "PluginInstance: %s\n", n->plugin_instance);
771 if (strlen (n->type) > 0)
772 fprintf (fh, "Type: %s\n", n->type);
773 if (strlen (n->type_instance) > 0)
774 fprintf (fh, "TypeInstance: %s\n", n->type_instance);
776 for (meta = n->meta; meta != NULL; meta = meta->next)
777 {
778 if (meta->type == NM_TYPE_STRING)
779 fprintf (fh, "%s: %s\n", meta->name, meta->nm_value.nm_string);
780 else if (meta->type == NM_TYPE_SIGNED_INT)
781 fprintf (fh, "%s: %"PRIi64"\n", meta->name, meta->nm_value.nm_signed_int);
782 else if (meta->type == NM_TYPE_UNSIGNED_INT)
783 fprintf (fh, "%s: %"PRIu64"\n", meta->name, meta->nm_value.nm_unsigned_int);
784 else if (meta->type == NM_TYPE_DOUBLE)
785 fprintf (fh, "%s: %e\n", meta->name, meta->nm_value.nm_double);
786 else if (meta->type == NM_TYPE_BOOLEAN)
787 fprintf (fh, "%s: %s\n", meta->name,
788 meta->nm_value.nm_boolean ? "true" : "false");
789 }
791 fprintf (fh, "\n%s\n", n->message);
793 fflush (fh);
794 fclose (fh);
796 waitpid (pid, &status, 0);
798 DEBUG ("exec plugin: Child %i exited with status %i.",
799 pid, status);
801 if (n->meta != NULL)
802 plugin_notification_meta_free (n->meta);
803 n->meta = NULL;
804 sfree (arg);
805 pthread_exit ((void *) 0);
806 return (NULL);
807 } /* void *exec_notification_one }}} */
809 static int exec_init (void) /* {{{ */
810 {
811 struct sigaction sa;
813 memset (&sa, '\0', sizeof (sa));
814 sa.sa_handler = sigchld_handler;
815 sigaction (SIGCHLD, &sa, NULL);
817 return (0);
818 } /* int exec_init }}} */
820 static int exec_read (void) /* {{{ */
821 {
822 program_list_t *pl;
824 for (pl = pl_head; pl != NULL; pl = pl->next)
825 {
826 pthread_t t;
827 pthread_attr_t attr;
829 /* Only execute `normal' style executables here. */
830 if ((pl->flags & PL_NORMAL) == 0)
831 continue;
833 pthread_mutex_lock (&pl_lock);
834 /* Skip if a child is already running. */
835 if ((pl->flags & PL_RUNNING) != 0)
836 {
837 pthread_mutex_unlock (&pl_lock);
838 continue;
839 }
840 pl->flags |= PL_RUNNING;
841 pthread_mutex_unlock (&pl_lock);
843 pthread_attr_init (&attr);
844 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
845 plugin_thread_create (&t, &attr, exec_read_one, (void *) pl);
846 pthread_attr_destroy (&attr);
847 } /* for (pl) */
849 return (0);
850 } /* int exec_read }}} */
852 static int exec_notification (const notification_t *n, /* {{{ */
853 user_data_t __attribute__((unused)) *user_data)
854 {
855 program_list_t *pl;
856 program_list_and_notification_t *pln;
858 for (pl = pl_head; pl != NULL; pl = pl->next)
859 {
860 pthread_t t;
861 pthread_attr_t attr;
863 /* Only execute `notification' style executables here. */
864 if ((pl->flags & PL_NOTIF_ACTION) == 0)
865 continue;
867 /* Skip if a child is already running. */
868 if (pl->pid != 0)
869 continue;
871 pln = (program_list_and_notification_t *) malloc (sizeof
872 (program_list_and_notification_t));
873 if (pln == NULL)
874 {
875 ERROR ("exec plugin: malloc failed.");
876 continue;
877 }
879 pln->pl = pl;
880 memcpy (&pln->n, n, sizeof (notification_t));
882 /* Set the `meta' member to NULL, otherwise `plugin_notification_meta_copy'
883 * will run into an endless loop. */
884 pln->n.meta = NULL;
885 plugin_notification_meta_copy (&pln->n, n);
887 pthread_attr_init (&attr);
888 pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
889 plugin_thread_create (&t, &attr, exec_notification_one, (void *) pln);
890 pthread_attr_destroy (&attr);
891 } /* for (pl) */
893 return (0);
894 } /* }}} int exec_notification */
896 static int exec_shutdown (void) /* {{{ */
897 {
898 program_list_t *pl;
899 program_list_t *next;
901 pl = pl_head;
902 while (pl != NULL)
903 {
904 next = pl->next;
906 if (pl->pid > 0)
907 {
908 kill (pl->pid, SIGTERM);
909 INFO ("exec plugin: Sent SIGTERM to %hu", (unsigned short int) pl->pid);
910 }
912 sfree (pl->user);
913 sfree (pl);
915 pl = next;
916 } /* while (pl) */
917 pl_head = NULL;
919 return (0);
920 } /* int exec_shutdown }}} */
922 void module_register (void)
923 {
924 plugin_register_complex_config ("exec", exec_config);
925 plugin_register_init ("exec", exec_init);
926 plugin_register_read ("exec", exec_read);
927 plugin_register_notification ("exec", exec_notification,
928 /* user_data = */ NULL);
929 plugin_register_shutdown ("exec", exec_shutdown);
930 } /* void module_register */
932 /*
933 * vim:shiftwidth=2:softtabstop=2:tabstop=8:fdm=marker
934 */