Merge branch 'master' of ssh://gnunet.org/gnunet
[oweals/gnunet.git] / src / util / gnunet-uri.c
1 /*
2      This file is part of GNUnet.
3      Copyright (C) 2012 GNUnet e.V.
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
18      Boston, MA 02110-1301, USA.
19 */
20
21 /**
22  * @file util/gnunet-uri.c
23  * @brief tool to dispatch URIs to the appropriate GNUnet helper process
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet_util_lib.h"
28
29 /**
30  * Handler exit code
31  */
32 static long unsigned int exit_code = 1;
33
34 /**
35  * Helper process we started.
36  */
37 static struct GNUNET_OS_Process *p;
38
39 /**
40  * Pipe used to communicate shutdown via signal.
41  */
42 static struct GNUNET_DISK_PipeHandle *sigpipe;
43
44
45 /**
46  * Task triggered whenever we receive a SIGCHLD (child
47  * process died) or when user presses CTRL-C.
48  *
49  * @param cls closure, NULL
50  */
51 static void
52 maint_child_death (void *cls)
53 {
54   enum GNUNET_OS_ProcessStatusType type;
55
56   if ( (GNUNET_OK !=
57         GNUNET_OS_process_status (p, &type, &exit_code)) ||
58        (type != GNUNET_OS_PROCESS_EXITED) )
59     GNUNET_break (0 == GNUNET_OS_process_kill (p, GNUNET_TERM_SIG));
60   GNUNET_OS_process_destroy (p);
61 }
62
63
64 /**
65  * Main function that will be run by the scheduler.
66  *
67  * @param cls closure
68  * @param args remaining command-line arguments
69  * @param cfgfile name of the configuration file used (for saving, can be NULL!)
70  * @param cfg configuration
71  */
72 static void
73 run (void *cls, char *const *args, const char *cfgfile,
74      const struct GNUNET_CONFIGURATION_Handle *cfg)
75 {
76   const char *uri;
77   const char *slash;
78   char *subsystem;
79   char *program;
80   struct GNUNET_SCHEDULER_Task * rt;
81
82   if (NULL == (uri = args[0]))
83   {
84     fprintf (stderr,
85              _("No URI specified on command line\n"));
86     return;
87   }
88   if (0 != strncasecmp ("gnunet://", uri, strlen ("gnunet://")))
89   {
90     fprintf (stderr,
91              _("Invalid URI: does not start with `%s'\n"),
92              "gnunet://");
93     return;
94   }
95   uri += strlen ("gnunet://");
96   if (NULL == (slash = strchr (uri, '/')))
97   {
98     fprintf (stderr, _("Invalid URI: fails to specify subsystem\n"));
99     return;
100   }
101   subsystem = GNUNET_strndup (uri, slash - uri);
102   if (GNUNET_OK !=
103       GNUNET_CONFIGURATION_get_value_string (cfg,
104                                              "uri",
105                                              subsystem,
106                                              &program))
107   {
108     fprintf (stderr, _("No handler known for subsystem `%s'\n"), subsystem);
109     GNUNET_free (subsystem);
110     return;
111   }
112   GNUNET_free (subsystem);
113   rt = GNUNET_SCHEDULER_add_read_file (GNUNET_TIME_UNIT_FOREVER_REL,
114                                        GNUNET_DISK_pipe_handle (sigpipe,
115                                                                 GNUNET_DISK_PIPE_END_READ),
116                                        &maint_child_death, NULL);
117   p = GNUNET_OS_start_process (GNUNET_NO, 0,
118                                NULL, NULL, NULL,
119                                program,
120                                program,
121                                args[0],
122                                NULL);
123   GNUNET_free (program);
124   if (NULL == p)
125     GNUNET_SCHEDULER_cancel (rt);
126 }
127
128
129 /**
130  * Signal handler called for SIGCHLD.  Triggers the
131  * respective handler by writing to the trigger pipe.
132  */
133 static void
134 sighandler_child_death ()
135 {
136   static char c;
137   int old_errno = errno;        /* back-up errno */
138
139   GNUNET_break (1 ==
140                 GNUNET_DISK_file_write (GNUNET_DISK_pipe_handle
141                                         (sigpipe, GNUNET_DISK_PIPE_END_WRITE),
142                                         &c, sizeof (c)));
143   errno = old_errno;            /* restore errno */
144 }
145
146
147 /**
148  * The main function to handle gnunet://-URIs.
149  *
150  * @param argc number of arguments from the command line
151  * @param argv command line arguments
152  * @return 0 ok, 1 on error
153  */
154 int
155 main (int argc, char *const *argv)
156 {
157   static const struct GNUNET_GETOPT_CommandLineOption options[] = {
158     GNUNET_GETOPT_OPTION_END
159   };
160   struct GNUNET_SIGNAL_Context *shc_chld;
161   int ret;
162
163   if (GNUNET_OK != GNUNET_STRINGS_get_utf8_args (argc, argv, &argc, &argv))
164     return 2;
165   sigpipe = GNUNET_DISK_pipe (GNUNET_NO, GNUNET_NO, GNUNET_NO, GNUNET_NO);
166   GNUNET_assert (sigpipe != NULL);
167   shc_chld =
168     GNUNET_SIGNAL_handler_install (GNUNET_SIGCHLD, &sighandler_child_death);
169   ret = GNUNET_PROGRAM_run (argc, argv, "gnunet-uri URI",
170                             gettext_noop ("Perform default-actions for GNUnet URIs"),
171                             options, &run, NULL);
172   GNUNET_SIGNAL_handler_uninstall (shc_chld);
173   shc_chld = NULL;
174   GNUNET_DISK_pipe_close (sigpipe);
175   sigpipe = NULL;
176   GNUNET_free ((void *) argv);
177   return ((GNUNET_OK == ret) && (0 == exit_code)) ? 0 : 1;
178 }
179
180 /* end of gnunet-uri.c */