Merge remote branch 'origin/master'
[collectd.git] / src / exec.c
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  **/
25
26 #define _BSD_SOURCE /* For setgroups */
27
28 #include "collectd.h"
29 #include "common.h"
30 #include "plugin.h"
31
32 #include "utils_cmd_putval.h"
33 #include "utils_cmd_putnotif.h"
34
35 #include <sys/types.h>
36 #include <pwd.h>
37 #include <grp.h>
38 #include <signal.h>
39
40 #include <pthread.h>
41
42 #define PL_NORMAL        0x01
43 #define PL_NOTIF_ACTION  0x02
44
45 #define PL_RUNNING       0x10
46
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 };
71
72 typedef struct program_list_and_notification_s
73 {
74   program_list_t *pl;
75   notification_t n;
76 } program_list_and_notification_t;
77
78 /*
79  * Private variables
80  */
81 static program_list_t *pl_head = NULL;
82 static pthread_mutex_t pl_lock = PTHREAD_MUTEX_INITIALIZER;
83
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 }}} */
101
102 static int exec_config_exec (oconfig_item_t *ci) /* {{{ */
103 {
104   program_list_t *pl;
105   char buffer[128];
106   int i;
107
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   }
127
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));
135
136   if (strcasecmp ("NotificationExec", ci->key) == 0)
137     pl->flags |= PL_NOTIF_ACTION;
138   else
139     pl->flags |= PL_NORMAL;
140
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   }
148
149   pl->group = strchr (pl->user, ':');
150   if (pl->group != NULL)
151   {
152     *pl->group = '\0';
153     pl->group++;
154   }
155
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   }
164
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 *));
175
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   }
193
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       }
214
215       pl->argv[i] = strdup (buffer);
216     }
217
218     if (pl->argv[i] == NULL)
219     {
220       ERROR ("exec plugin: strdup failed.");
221       break;
222     }
223   } /* for (i) */
224
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   }
237
238   for (i = 0; pl->argv[i] != NULL; i++)
239   {
240     DEBUG ("exec plugin: argv[%i] = %s", i, pl->argv[i]);
241   }
242
243   pl->next = pl_head;
244   pl_head = pl;
245
246   return (0);
247 } /* int exec_config_exec }}} */
248
249 static int exec_config (oconfig_item_t *ci) /* {{{ */
250 {
251   int i;
252
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) */
264
265   return (0);
266 } /* int exec_config }}} */
267
268 static void set_environment (void) /* {{{ */
269 {
270   char buffer[1024];
271
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);
276
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);
283
284   ssnprintf (buffer, sizeof (buffer), "COLLECTD_HOSTNAME=%s", hostname_g);
285   putenv (buffer);
286 #endif
287 } /* }}} void set_environment */
288
289 __attribute__((noreturn))
290 static void exec_child (program_list_t *pl) /* {{{ */
291 {
292   int status;
293   int uid;
294   int gid;
295   int egid;
296
297   struct passwd *sp_ptr;
298   struct passwd sp;
299   char nambuf[2048];
300   char errbuf[1024];
301
302   sp_ptr = NULL;
303   status = getpwnam_r (pl->user, &sp, nambuf, sizeof (nambuf), &sp_ptr);
304   if (status != 0)
305   {
306     ERROR ("exec plugin: Failed to get user information for user ``%s'': %s",
307         pl->user, sstrerror (errno, errbuf, sizeof (errbuf)));
308     exit (-1);
309   }
310   if (sp_ptr == NULL)
311   {
312     ERROR ("exec plugin: No such user: `%s'", pl->user);
313     exit (-1);
314   }
315
316   uid = sp.pw_uid;
317   gid = sp.pw_gid;
318   if (uid == 0)
319   {
320     ERROR ("exec plugin: Cowardly refusing to exec program as root.");
321     exit (-1);
322   }
323
324   /* The group configured in the configfile is set as effective group, because
325    * this way the forked process can (re-)gain the user's primary group. */
326   egid = -1;
327   if (NULL != pl->group)
328   {
329     if ('\0' != *pl->group) {
330       struct group *gr_ptr = NULL;
331       struct group gr;
332
333       status = getgrnam_r (pl->group, &gr, nambuf, sizeof (nambuf), &gr_ptr);
334       if (0 != status)
335       {
336         ERROR ("exec plugin: Failed to get group information "
337             "for group ``%s'': %s", pl->group,
338             sstrerror (errno, errbuf, sizeof (errbuf)));
339         exit (-1);
340       }
341       if (NULL == gr_ptr)
342       {
343         ERROR ("exec plugin: No such group: `%s'", pl->group);
344         exit (-1);
345       }
346
347       egid = gr.gr_gid;
348     }
349     else
350     {
351       egid = gid;
352     }
353   } /* if (pl->group == NULL) */
354
355 #if HAVE_SETGROUPS
356   if (getuid () == 0)
357   {
358     gid_t  glist[2];
359     size_t glist_len;
360
361     glist[0] = gid;
362     glist_len = 1;
363
364     if ((gid != egid) && (egid != -1))
365     {
366       glist[1] = egid;
367       glist_len = 2;
368     }
369
370     setgroups (glist_len, glist);
371   }
372 #endif /* HAVE_SETGROUPS */
373
374   status = setgid (gid);
375   if (status != 0)
376   {
377     ERROR ("exec plugin: setgid (%i) failed: %s",
378         gid, sstrerror (errno, errbuf, sizeof (errbuf)));
379     exit (-1);
380   }
381
382   if (egid != -1)
383   {
384     status = setegid (egid);
385     if (status != 0)
386     {
387       ERROR ("exec plugin: setegid (%i) failed: %s",
388           egid, sstrerror (errno, errbuf, sizeof (errbuf)));
389       exit (-1);
390     }
391   }
392
393   status = setuid (uid);
394   if (status != 0)
395   {
396     ERROR ("exec plugin: setuid (%i) failed: %s",
397         uid, sstrerror (errno, errbuf, sizeof (errbuf)));
398     exit (-1);
399   }
400
401   status = execvp (pl->exec, pl->argv);
402
403   ERROR ("exec plugin: Failed to execute ``%s'': %s",
404       pl->exec, sstrerror (errno, errbuf, sizeof (errbuf)));
405   exit (-1);
406 } /* void exec_child }}} */
407
408 static void reset_signal_mask (void) /* {{{ */
409 {
410   sigset_t ss;
411
412   memset (&ss, 0, sizeof (ss));
413   sigemptyset (&ss);
414   sigprocmask (SIG_SETMASK, &ss, /* old mask = */ NULL);
415 } /* }}} void reset_signal_mask */
416
417 /*
418  * Creates three pipes (one for reading, one for writing and one for errors),
419  * forks a child, sets up the pipes so that fd_in is connected to STDIN of
420  * the child and fd_out is connected to STDOUT and fd_err is connected to STDERR
421  * of the child. Then is calls `exec_child'.
422  */
423 static int fork_child (program_list_t *pl, int *fd_in, int *fd_out, int *fd_err) /* {{{ */
424 {
425   int fd_pipe_in[2];
426   int fd_pipe_out[2];
427   int fd_pipe_err[2];
428   char errbuf[1024];
429   int status;
430   int pid;
431
432   if (pl->pid != 0)
433     return (-1);
434
435   status = pipe (fd_pipe_in);
436   if (status != 0)
437   {
438     ERROR ("exec plugin: pipe failed: %s",
439         sstrerror (errno, errbuf, sizeof (errbuf)));
440     return (-1);
441   }
442
443   status = pipe (fd_pipe_out);
444   if (status != 0)
445   {
446     ERROR ("exec plugin: pipe failed: %s",
447         sstrerror (errno, errbuf, sizeof (errbuf)));
448     return (-1);
449   }
450
451   status = pipe (fd_pipe_err);
452   if (status != 0)
453   {
454     ERROR ("exec plugin: pipe failed: %s",
455         sstrerror (errno, errbuf, sizeof (errbuf)));
456     return (-1);
457   }
458
459   pid = fork ();
460   if (pid < 0)
461   {
462     ERROR ("exec plugin: fork failed: %s",
463         sstrerror (errno, errbuf, sizeof (errbuf)));
464     return (-1);
465   }
466   else if (pid == 0)
467   {
468     int fd_num;
469     int fd;
470
471     /* Close all file descriptors but the pipe end we need. */
472     fd_num = getdtablesize ();
473     for (fd = 0; fd < fd_num; fd++)
474     {
475       if ((fd == fd_pipe_in[0])
476           || (fd == fd_pipe_out[1])
477           || (fd == fd_pipe_err[1]))
478         continue;
479       close (fd);
480     }
481
482     /* Connect the `in' pipe to STDIN */
483     if (fd_pipe_in[0] != STDIN_FILENO)
484     {
485       dup2 (fd_pipe_in[0], STDIN_FILENO);
486       close (fd_pipe_in[0]);
487     }
488
489     /* Now connect the `out' pipe to STDOUT */
490     if (fd_pipe_out[1] != STDOUT_FILENO)
491     {
492       dup2 (fd_pipe_out[1], STDOUT_FILENO);
493       close (fd_pipe_out[1]);
494     }
495
496     /* Now connect the `err' pipe to STDERR */
497     if (fd_pipe_err[1] != STDERR_FILENO)
498     {
499       dup2 (fd_pipe_err[1], STDERR_FILENO);
500       close (fd_pipe_err[1]);
501     }
502
503     set_environment ();
504
505     /* Unblock all signals */
506     reset_signal_mask ();
507
508     exec_child (pl);
509     /* does not return */
510   }
511
512   close (fd_pipe_in[0]);
513   close (fd_pipe_out[1]);
514   close (fd_pipe_err[1]);
515
516   if (fd_in != NULL)
517     *fd_in = fd_pipe_in[1];
518   else
519     close (fd_pipe_in[1]);
520
521   if (fd_out != NULL)
522     *fd_out = fd_pipe_out[0];
523   else
524     close (fd_pipe_out[0]);
525
526   if (fd_err != NULL)
527     *fd_err = fd_pipe_err[0];
528   else
529     close (fd_pipe_err[0]);
530
531   return (pid);
532 } /* int fork_child }}} */
533
534 static int parse_line (char *buffer) /* {{{ */
535 {
536   if (strncasecmp ("PUTVAL", buffer, strlen ("PUTVAL")) == 0)
537     return (handle_putval (stdout, buffer));
538   else if (strncasecmp ("PUTNOTIF", buffer, strlen ("PUTNOTIF")) == 0)
539     return (handle_putnotif (stdout, buffer));
540   else
541   {
542     ERROR ("exec plugin: Unable to parse command, ignoring line: \"%s\"",
543         buffer);
544     return (-1);
545   }
546 } /* int parse_line }}} */
547
548 static void *exec_read_one (void *arg) /* {{{ */
549 {
550   program_list_t *pl = (program_list_t *) arg;
551   int fd, fd_err, highest_fd;
552   fd_set fdset, copy;
553   int status;
554   char buffer[1200];  /* if not completely read */
555   char buffer_err[1024];
556   char *pbuffer = buffer;
557   char *pbuffer_err = buffer_err;
558
559   status = fork_child (pl, NULL, &fd, &fd_err);
560   if (status < 0)
561   {
562     /* Reset the "running" flag */
563     pthread_mutex_lock (&pl_lock);
564     pl->flags &= ~PL_RUNNING;
565     pthread_mutex_unlock (&pl_lock);
566     pthread_exit ((void *) 1);
567   }
568   pl->pid = status;
569
570   assert (pl->pid != 0);
571
572   FD_ZERO( &fdset );
573   FD_SET(fd, &fdset);
574   FD_SET(fd_err, &fdset);
575
576   /* Determine the highest file descriptor */
577   highest_fd = (fd > fd_err) ? fd : fd_err;
578
579   /* We use a copy of fdset, as select modifies it */
580   copy = fdset;
581
582   while (1)
583   {
584     int len;
585
586     status = select (highest_fd + 1, &copy, NULL, NULL, NULL);
587     if (status < 0)
588     {
589       if (errno == EINTR)
590         continue;
591       break;
592     }
593
594     if (FD_ISSET(fd, &copy))
595     {
596       char *pnl;
597
598       len = read(fd, pbuffer, sizeof(buffer) - 1 - (pbuffer - buffer));
599
600       if (len < 0)
601       {
602         if (errno == EAGAIN || errno == EINTR)  continue;
603         break;
604       }
605       else if (len == 0) break;  /* We've reached EOF */
606
607       pbuffer[len] = '\0';
608
609       len += pbuffer - buffer;
610       pbuffer = buffer;
611
612       while ((pnl = strchr(pbuffer, '\n')))
613       {
614         *pnl = '\0';
615         if (*(pnl-1) == '\r' ) *(pnl-1) = '\0';
616
617         parse_line (pbuffer);
618
619         pbuffer = ++pnl;
620       }
621       /* not completely read ? */
622       if (pbuffer - buffer < len)
623       {
624         len -= pbuffer - buffer;
625         memmove(buffer, pbuffer, len);
626         pbuffer = buffer + len;
627       }
628       else
629         pbuffer = buffer;
630     }
631     else if (FD_ISSET(fd_err, &copy))
632     {
633       char *pnl;
634
635       len = read(fd_err, pbuffer_err, sizeof(buffer_err) - 1 - (pbuffer_err - buffer_err));
636
637       if (len < 0)
638       {
639         if (errno == EAGAIN || errno == EINTR)
640           continue;
641         break;
642       }
643       else if (len == 0)
644       {
645         /* We've reached EOF */
646         NOTICE ("exec plugin: Program `%s' has closed STDERR.", pl->exec);
647
648         /* Remove file descriptor form select() set. */
649         FD_CLR (fd_err, &fdset);
650         copy = fdset;
651         highest_fd = fd;
652
653         /* Clean up file descriptor */
654         close (fd_err);
655         fd_err = -1;
656         continue;
657       }
658
659       pbuffer_err[len] = '\0';
660
661       len += pbuffer_err - buffer_err;
662       pbuffer_err = buffer_err;
663
664       while ((pnl = strchr(pbuffer_err, '\n')))
665       {
666         *pnl = '\0';
667         if (*(pnl-1) == '\r' ) *(pnl-1) = '\0';
668
669         ERROR ("exec plugin: exec_read_one: error = %s", pbuffer_err);
670
671         pbuffer_err = ++pnl;
672       }
673       /* not completely read ? */
674       if (pbuffer_err - buffer_err < len)
675       {
676         len -= pbuffer_err - buffer_err;
677         memmove(buffer_err, pbuffer_err, len);
678         pbuffer_err = buffer_err + len;
679       }
680       else
681         pbuffer_err = buffer_err;
682     }
683     /* reset copy */
684     copy = fdset;
685   }
686
687   DEBUG ("exec plugin: exec_read_one: Waiting for `%s' to exit.", pl->exec);
688   if (waitpid (pl->pid, &status, 0) > 0)
689     pl->status = status;
690
691   DEBUG ("exec plugin: Child %i exited with status %i.",
692       (int) pl->pid, pl->status);
693
694   pl->pid = 0;
695
696   pthread_mutex_lock (&pl_lock);
697   pl->flags &= ~PL_RUNNING;
698   pthread_mutex_unlock (&pl_lock);
699
700   close (fd);
701   if (fd_err >= 0)
702     close (fd_err);
703
704   pthread_exit ((void *) 0);
705   return (NULL);
706 } /* void *exec_read_one }}} */
707
708 static void *exec_notification_one (void *arg) /* {{{ */
709 {
710   program_list_t *pl = ((program_list_and_notification_t *) arg)->pl;
711   notification_t *n = &((program_list_and_notification_t *) arg)->n;
712   notification_meta_t *meta;
713   int fd;
714   FILE *fh;
715   int pid;
716   int status;
717   const char *severity;
718
719   pid = fork_child (pl, &fd, NULL, NULL);
720   if (pid < 0) {
721     sfree (arg);
722     pthread_exit ((void *) 1);
723   }
724
725   fh = fdopen (fd, "w");
726   if (fh == NULL)
727   {
728     char errbuf[1024];
729     ERROR ("exec plugin: fdopen (%i) failed: %s", fd,
730         sstrerror (errno, errbuf, sizeof (errbuf)));
731     kill (pl->pid, SIGTERM);
732     pl->pid = 0;
733     close (fd);
734     sfree (arg);
735     pthread_exit ((void *) 1);
736   }
737
738   severity = "FAILURE";
739   if (n->severity == NOTIF_WARNING)
740     severity = "WARNING";
741   else if (n->severity == NOTIF_OKAY)
742     severity = "OKAY";
743
744   fprintf (fh,
745       "Severity: %s\n"
746       "Time: %.3f\n",
747       severity, CDTIME_T_TO_DOUBLE (n->time));
748
749   /* Print the optional fields */
750   if (strlen (n->host) > 0)
751     fprintf (fh, "Host: %s\n", n->host);
752   if (strlen (n->plugin) > 0)
753     fprintf (fh, "Plugin: %s\n", n->plugin);
754   if (strlen (n->plugin_instance) > 0)
755     fprintf (fh, "PluginInstance: %s\n", n->plugin_instance);
756   if (strlen (n->type) > 0)
757     fprintf (fh, "Type: %s\n", n->type);
758   if (strlen (n->type_instance) > 0)
759     fprintf (fh, "TypeInstance: %s\n", n->type_instance);
760
761   for (meta = n->meta; meta != NULL; meta = meta->next)
762   {
763     if (meta->type == NM_TYPE_STRING)
764       fprintf (fh, "%s: %s\n", meta->name, meta->nm_value.nm_string);
765     else if (meta->type == NM_TYPE_SIGNED_INT)
766       fprintf (fh, "%s: %"PRIi64"\n", meta->name, meta->nm_value.nm_signed_int);
767     else if (meta->type == NM_TYPE_UNSIGNED_INT)
768       fprintf (fh, "%s: %"PRIu64"\n", meta->name, meta->nm_value.nm_unsigned_int);
769     else if (meta->type == NM_TYPE_DOUBLE)
770       fprintf (fh, "%s: %e\n", meta->name, meta->nm_value.nm_double);
771     else if (meta->type == NM_TYPE_BOOLEAN)
772       fprintf (fh, "%s: %s\n", meta->name,
773           meta->nm_value.nm_boolean ? "true" : "false");
774   }
775
776   fprintf (fh, "\n%s\n", n->message);
777
778   fflush (fh);
779   fclose (fh);
780
781   waitpid (pid, &status, 0);
782
783   DEBUG ("exec plugin: Child %i exited with status %i.",
784       pid, status);
785
786   if (n->meta != NULL)
787     plugin_notification_meta_free (n->meta);
788   n->meta = NULL;
789   sfree (arg);
790   pthread_exit ((void *) 0);
791   return (NULL);
792 } /* void *exec_notification_one }}} */
793
794 static int exec_init (void) /* {{{ */
795 {
796   struct sigaction sa;
797
798   memset (&sa, '\0', sizeof (sa));
799   sa.sa_handler = sigchld_handler;
800   sigaction (SIGCHLD, &sa, NULL);
801
802   return (0);
803 } /* int exec_init }}} */
804
805 static int exec_read (void) /* {{{ */
806 {
807   program_list_t *pl;
808
809   for (pl = pl_head; pl != NULL; pl = pl->next)
810   {
811     pthread_t t;
812     pthread_attr_t attr;
813
814     /* Only execute `normal' style executables here. */
815     if ((pl->flags & PL_NORMAL) == 0)
816       continue;
817
818     pthread_mutex_lock (&pl_lock);
819     /* Skip if a child is already running. */
820     if ((pl->flags & PL_RUNNING) != 0)
821     {
822       pthread_mutex_unlock (&pl_lock);
823       continue;
824     }
825     pl->flags |= PL_RUNNING;
826     pthread_mutex_unlock (&pl_lock);
827
828     pthread_attr_init (&attr);
829     pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
830     plugin_thread_create (&t, &attr, exec_read_one, (void *) pl);
831     pthread_attr_destroy (&attr);
832   } /* for (pl) */
833
834   return (0);
835 } /* int exec_read }}} */
836
837 static int exec_notification (const notification_t *n, /* {{{ */
838     user_data_t __attribute__((unused)) *user_data)
839 {
840   program_list_t *pl;
841   program_list_and_notification_t *pln;
842
843   for (pl = pl_head; pl != NULL; pl = pl->next)
844   {
845     pthread_t t;
846     pthread_attr_t attr;
847
848     /* Only execute `notification' style executables here. */
849     if ((pl->flags & PL_NOTIF_ACTION) == 0)
850       continue;
851
852     /* Skip if a child is already running. */
853     if (pl->pid != 0)
854       continue;
855
856     pln = (program_list_and_notification_t *) malloc (sizeof
857         (program_list_and_notification_t));
858     if (pln == NULL)
859     {
860       ERROR ("exec plugin: malloc failed.");
861       continue;
862     }
863
864     pln->pl = pl;
865     memcpy (&pln->n, n, sizeof (notification_t));
866
867     /* Set the `meta' member to NULL, otherwise `plugin_notification_meta_copy'
868      * will run into an endless loop. */
869     pln->n.meta = NULL;
870     plugin_notification_meta_copy (&pln->n, n);
871
872     pthread_attr_init (&attr);
873     pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED);
874     plugin_thread_create (&t, &attr, exec_notification_one, (void *) pln);
875     pthread_attr_destroy (&attr);
876   } /* for (pl) */
877
878   return (0);
879 } /* }}} int exec_notification */
880
881 static int exec_shutdown (void) /* {{{ */
882 {
883   program_list_t *pl;
884   program_list_t *next;
885
886   pl = pl_head;
887   while (pl != NULL)
888   {
889     next = pl->next;
890
891     if (pl->pid > 0)
892     {
893       kill (pl->pid, SIGTERM);
894       INFO ("exec plugin: Sent SIGTERM to %hu", (unsigned short int) pl->pid);
895     }
896
897     sfree (pl->user);
898     sfree (pl);
899
900     pl = next;
901   } /* while (pl) */
902   pl_head = NULL;
903
904   return (0);
905 } /* int exec_shutdown }}} */
906
907 void module_register (void)
908 {
909   plugin_register_complex_config ("exec", exec_config);
910   plugin_register_init ("exec", exec_init);
911   plugin_register_read ("exec", exec_read);
912   plugin_register_notification ("exec", exec_notification,
913       /* user_data = */ NULL);
914   plugin_register_shutdown ("exec", exec_shutdown);
915 } /* void module_register */
916
917 /*
918  * vim:shiftwidth=2:softtabstop=2:tabstop=8:fdm=marker
919  */