src/util/w32cat.c

   1 /*
   2      W32 version of 'cat' program
   3      (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., 59 Temple Place - Suite 330,
  18      Boston, MA 02111-1307, USA.
  19 */
  20
  21 #include <stdio.h>
  22 #include <windows.h>
  23
  24 int
  25 main (int argc, char **argv)
  26 {
  27   HANDLE stdi, stdo;
  28   BOOL b;
  29   wchar_t *commandlinew, **argvw;
  30   int argcw;
  31   int i;
  32
  33   stdo = GetStdHandle (STD_OUTPUT_HANDLE);
  34   if (stdo == INVALID_HANDLE_VALUE || stdo == NULL)
  35     return 1;
  36
  37   commandlinew = GetCommandLineW ();
  38   argvw = CommandLineToArgvW (commandlinew, &argcw);
  39   if (argvw == NULL)
  40     return 1;
  41
  42   for (i = 1; i < argcw || argcw == 1; i++)
  43   {
  44     DWORD r, w;
  45     int is_dash = wcscmp (argvw[i], L"-") == 0;
  46     if (argcw == 1 || is_dash)
  47     {
  48       stdi = GetStdHandle (STD_INPUT_HANDLE);
  49       if (stdi == INVALID_HANDLE_VALUE)
  50       {
  51         fprintf (stderr, "cat: Failed to obtain stdin handle.\n");
  52         return 4;
  53       }
  54       if (stdi == NULL)
  55       {
  56         fprintf (stderr, "cat: Have no stdin.\n");
  57         return 5;
  58       }
  59     }
  60     else
  61     {
  62       stdi = CreateFileW (argvw[i], GENERIC_READ, FILE_SHARE_DELETE | FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
  63       if (stdi == INVALID_HANDLE_VALUE)
  64       {
  65         wchar_t *msgbuf;
  66         DWORD le = GetLastError ();
  67         if (0 < FormatMessageW (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, 0, le, 0, (wchar_t *) &msgbuf, 0, NULL))
  68         {
  69           fprintf (stderr, "cat: Failed to open file `%S'. Error %lu.\n", argvw[i], le);
  70           return 3;
  71         }
  72         fprintf (stderr, "cat: Failed to open file `%S'. Error %lu: %S\n", argvw[i], le, msgbuf);
  73         if (msgbuf != NULL)
  74           LocalFree (msgbuf);
  75         return 2;
  76       }
  77     }
  78     do
  79     {
  80       unsigned char c;
  81       b = ReadFile (stdi, &c, 1, &r, NULL);
  82       if (b && r > 0)
  83       {
  84         b = WriteFile (stdo, &c, 1, &w, NULL);
  85         if (b == 0)
  86         {
  87           wchar_t *msgbuf;
  88           DWORD le = GetLastError ();
  89           if (0 < FormatMessageW (FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, 0, le, 0, (wchar_t *) &msgbuf, 0, NULL))
  90           {
  91             fprintf (stderr, "cat: Failed to write into stdout. Error %lu.\n", le);
  92             return 3;
  93           }
  94           fprintf (stderr, "cat: Failed to write into stdout. Error %lu: %S\n", le, msgbuf);
  95           if (msgbuf != NULL)
  96             LocalFree (msgbuf);
  97           return 6;
  98         }
  99       }
 100     } while (b && r > 0);
 101     if (argcw == 1)
 102       break;
 103     if (!is_dash)
 104       CloseHandle (stdi);
 105   }
 106   LocalFree (argvw);
 107   return 0;
 108 }