Reworked the posix and w32 exechelpers.
[gnupg.git] / common / init.c
1 /* init.c - Various initializations
2  *      Copyright (C) 2007 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 3 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, see <http://www.gnu.org/licenses/>.
18  */
19
20 #include <config.h>
21
22 #ifdef WITHOUT_GNU_PTH /* Give the Makefile a chance to build without Pth.  */
23 #undef HAVE_PTH
24 #undef USE_GNU_PTH
25 #endif
26
27 #ifdef HAVE_W32_SYSTEM
28 #include <windows.h>
29 #endif
30 #ifdef HAVE_PTH      
31 #include <pth.h>
32 #endif
33 #ifdef HAVE_W32CE_SYSTEM
34 # include <assuan.h> /* For _assuan_w32ce_finish_pipe. */
35 #endif
36
37 #include "util.h"
38
39
40 #ifdef HAVE_W32CE_SYSTEM
41 static void parse_std_file_handles (int *argcp, char ***argvp);
42 static void
43 sleep_on_exit (void)
44 {
45   /* The sshd on CE swallows some of the command output.  Sleeping a
46      while usually helps.  */
47   Sleep (400);
48 }
49 #endif /*HAVE_W32CE_SYSTEM*/
50
51
52 /* If STRING is not NULL write string to es_stdout or es_stderr.  MODE
53    must be 1 or 2.  If STRING is NULL flush the respective stream.  */
54 static int
55 writestring_via_estream (int mode, const char *string)
56 {
57   if (mode == 1 || mode == 2)
58     {
59       if (string)
60         return es_fputs (string, mode == 1? es_stdout : es_stderr);
61       else
62         return es_fflush (mode == 1? es_stdout : es_stderr);
63     }
64   else
65     return -1;
66 }
67
68
69 /* This function is to be used early at program startup to make sure
70    that some subsystems are initialized.  This is in particular
71    important for W32 to initialize the sockets so that our socket
72    emulation code used directly as well as in libassuan may be used.
73    It should best be called before any I/O is done so that setup
74    required for logging is ready.  ARGCP and ARGVP are the addresses
75    of the parameters given to main.  This function may modify them.
76
77    CAUTION: This might be called while running suid(root).  */
78 void
79 init_common_subsystems (int *argcp, char ***argvp)
80 {
81   /* Try to auto set the character set.  */
82   set_native_charset (NULL); 
83
84 #ifdef HAVE_W32_SYSTEM
85   /* For W32 we need to initialize the socket layer.  This is because
86      we use recv and send in libassuan as well as at some other
87      places.  If we are building with PTH we let pth_init do it.  We
88      can't do much on error so we ignore them.  An error would anyway
89      later pop up if one of the socket functions is used. */
90 # ifdef HAVE_PTH
91   pth_init ();
92 # else
93   {
94     WSADATA wsadat;
95     
96     WSAStartup (0x202, &wsadat);
97   }
98 # endif /*!HAVE_PTH*/
99 #endif
100   
101 #ifdef HAVE_W32CE_SYSTEM
102   /* Register the sleep exit function before the estream init so that
103      the sleep will be called after the estream registered atexit
104      function which flushes the left open estream streams and in
105      particular es_stdout.  */
106   atexit (sleep_on_exit);
107 #endif
108
109   /* Initialize the Estream library. */
110   es_init ();
111
112   /* Special hack for Windows CE: We extract some options from arg
113      to setup the standard handles.  */
114 #ifdef HAVE_W32CE_SYSTEM
115   parse_std_file_handles (argcp, argvp);
116 #else
117   (void)argcp;
118   (void)argvp;
119 #endif
120
121   /* --version et al shall use estream as well.  */
122   argparse_register_outfnc (writestring_via_estream);
123
124 }
125
126
127
128 /* WindowsCE uses a very strange way of handling the standard streams.
129    There is a function SetStdioPath to associate a standard stream
130    with a file or a device but what we really want is to use pipes as
131    standard streams.  Despite that we implement pipes using a device,
132    we would have some limitations on the number of open pipes due to
133    the 3 character limit of device file name.  Thus we don't take this
134    path.  Another option would be to install a file system driver with
135    support for pipes; this would allow us to get rid of the device
136    name length limitation.  However, with GnuPG we can get away be
137    redefining the standard streams and passing the handles to be used
138    on the command line.  This has also the advantage that it makes
139    creating a process much easier and does not require the
140    SetStdioPath set and restore game.  The caller needs to pass the
141    rendezvous ids using up to three options:
142
143      -&S0=<rvid> -&S1=<rvid> -&S2=<rvid>
144
145    They are all optional but they must be the first arguments on the
146    command line.  Parsing stops as soon as an invalid option is found.
147    These rendezvous ids are then used to finish the pipe creation.*/
148 #ifdef HAVE_W32CE_SYSTEM
149 static void
150 parse_std_file_handles (int *argcp, char ***argvp)
151 {
152   int argc = *argcp;
153   char **argv = *argvp;
154   const char *s;
155   assuan_fd_t fd;
156   int i;
157   int fixup = 0;
158
159   if (!argc)
160     return;
161
162   for (argc--, argv++; argc; argc--, argv++)
163     {
164       s = *argv;
165       if (*s == '-' && s[1] == '&' && s[2] == 'S'
166           && (s[3] == '0' || s[3] == '1' || s[3] == '2')
167           && s[4] == '=' 
168           && (strchr ("-01234567890", s[5]) || !strcmp (s+5, "null")))
169         {
170           if (s[5] == 'n')
171             fd = ASSUAN_INVALID_FD;
172           else
173             fd = _assuan_w32ce_finish_pipe (atoi (s+5), s[3] != '0');
174           _es_set_std_fd (s[3] - '0', (int)fd);
175           fixup++;
176         }
177       else
178         break;
179     }
180
181   if (fixup)
182     {
183       argc = *argcp;
184       argc -= fixup;
185       *argcp = argc;
186
187       argv = *argvp;
188       for (i=1; i < argc; i++)
189         argv[i] = argv[i + fixup];
190       for (; i < argc + fixup; i++)
191         argv[i] = NULL;
192     }
193
194
195 }
196 #endif /*HAVE_W32CE_SYSTEM*/