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 }