Merge branch 'master' of git+ssh://gnunet.org/gnunet
[oweals/gnunet.git] / src / util / w32cat.c
1 /*
2      W32 version of 'cat' program
3      Copyright (C) 2012 LRN
4
5      cat 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      cat 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 cat; 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      SPDX-License-Identifier: AGPL3.0-or-later
21 */
22
23 #include <stdio.h>
24 #include <windows.h>
25 #include <stdint.h>
26 #include <signal.h>
27
28 DWORD WINAPI
29 parent_control_thread (LPVOID lpParameter)
30 {
31   HANDLE h = (HANDLE) lpParameter;
32   while (TRUE)
33   {
34     DWORD dw;
35     BOOL b;
36     unsigned char c;
37     b = ReadFile (h, &c, 1, &dw, NULL);
38     if (!b)
39     {
40       ExitProcess (0);
41     }
42     raise ((int) c);
43   }
44 }
45
46 void
47 install_parent_control_handler ()
48 {
49   const char *env_buf;
50   char *env_buf_end;
51   uint64_t pipe_fd;
52   HANDLE pipe_handle;
53
54   env_buf = getenv ("GNUNET_OS_CONTROL_PIPE");
55   if ( (NULL == env_buf) || (strlen (env_buf) <= 0) )
56     return;
57   errno = 0;
58   pipe_fd = strtoull (env_buf, &env_buf_end, 16);
59   if ((0 != errno) || (env_buf == env_buf_end))
60     return;
61   /* Gcc will issue a warning here. What to do with it? */
62   pipe_handle = (HANDLE) (uintptr_t) pipe_fd;
63   CreateThread (NULL, 0, parent_control_thread, (LPVOID) pipe_handle, 0, NULL);
64 }
65
66 int
67 main (int argc, char **argv)
68 {
69   HANDLE stdi, stdo;
70   BOOL b;
71   wchar_t *commandlinew, **argvw;
72   int argcw;
73   int i;
74
75   stdo = GetStdHandle (STD_OUTPUT_HANDLE);
76   if (stdo == INVALID_HANDLE_VALUE || stdo == NULL)
77     return 1;
78
79   commandlinew = GetCommandLineW ();
80   argvw = CommandLineToArgvW (commandlinew, &argcw);
81   if (argvw == NULL)
82     return 1;
83
84   install_parent_control_handler ();
85
86   for (i = 1; i < argcw || argcw == 1; i++)
87   {
88     DWORD r, w;
89     int is_dash = wcscmp (NULL == argvw[i] ? L"-" : argvw[i], L"-") == 0;
90     if (argcw == 1 || is_dash)
91     {
92       stdi = GetStdHandle (STD_INPUT_HANDLE);
93       if (stdi == INVALID_HANDLE_VALUE)
94       {
95         fprintf (stderr, "cat: Failed to obtain stdin handle.\n");
96         return 4;
97       }
98       if (stdi == NULL)
99       {
100         fprintf (stderr, "cat: Have no stdin.\n");
101         return 5;
102       }
103     }
104     else
105     {
106       stdi = CreateFileW (argvw[i], GENERIC_READ, FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
107       if (stdi == INVALID_HANDLE_VALUE)
108       {
109         wchar_t *msgbuf;
110         DWORD le = GetLastError ();
111         if (0 < FormatMessageW (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, 0, le, 0, (wchar_t *) &msgbuf, 0, NULL))
112         {
113           fprintf (stderr, "cat: Failed to open file `%S'. Error %lu.\n", argvw[i], le);
114           return 3;
115         }
116         fprintf (stderr, "cat: Failed to open file `%S'. Error %lu: %S\n", argvw[i], le, msgbuf);
117         if (msgbuf != NULL)
118           LocalFree (msgbuf);
119         return 2;
120       }
121     }
122     do
123     {
124       unsigned char c;
125       b = ReadFile (stdi, &c, 1, &r, NULL);
126       if (b && r > 0)
127       {
128         b = WriteFile (stdo, &c, 1, &w, NULL);
129         if (b == 0)
130         {
131           wchar_t *msgbuf;
132           DWORD le = GetLastError ();
133           if (0 < FormatMessageW (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, 0, le, 0, (wchar_t *) &msgbuf, 0, NULL))
134           {
135             fprintf (stderr, "cat: Failed to write into stdout. Error %lu.\n", le);
136             return 3;
137           }
138           fprintf (stderr, "cat: Failed to write into stdout. Error %lu: %S\n", le, msgbuf);
139           if (msgbuf != NULL)
140             LocalFree (msgbuf);
141           return 6;
142         }
143       }
144     } while (b && r > 0);
145     if (argcw == 1)
146       break;
147     if (!is_dash)
148       CloseHandle (stdi);
149   }
150   LocalFree (argvw);
151   return 0;
152 }