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