programs/find/find.c

   1 /*
   2  * Copyright 2018 Fabian Maurer
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18
  19 #include <windows.h>
  20 #include <stdlib.h>
  21 #include <shlwapi.h>
  22
  23 #include "wine/heap.h"
  24 #include "wine/debug.h"
  25 #include "resources.h"
  26
  27 WINE_DEFAULT_DEBUG_CHANNEL(find);
  28
  29 static BOOL read_char_from_handle(HANDLE handle, char *char_out)
  30 {
  31     static char buffer[4096];
  32     static UINT buffer_max = 0;
  33     static UINT buffer_pos = 0;
  34
  35     /* Read next content into buffer */
  36     if (buffer_pos >= buffer_max)
  37     {
  38         BOOL success = ReadFile(handle, buffer, 4096, &buffer_max, NULL);
  39         if (!success || !buffer_max)
  40             return FALSE;
  41         buffer_pos = 0;
  42     }
  43
  44     *char_out = buffer[buffer_pos++];
  45     return TRUE;
  46 }
  47
  48 /* Read a line from a handle, returns NULL if the end is reached */
  49 static WCHAR* read_line_from_handle(HANDLE handle)
  50 {
  51     int line_max = 4096;
  52     int length = 0;
  53     WCHAR *line_converted;
  54     int line_converted_length;
  55     BOOL success;
  56     char *line = heap_alloc(line_max);
  57
  58     for (;;)
  59     {
  60         char c;
  61         success = read_char_from_handle(handle, &c);
  62
  63         /* Check for EOF */
  64         if (!success)
  65         {
  66             if (length == 0)
  67                 return NULL;
  68             else
  69                 break;
  70         }
  71
  72         if (c == '\n')
  73             break;
  74
  75         /* Make sure buffer is large enough */
  76         if (length + 1 >= line_max)
  77         {
  78             line_max *= 2;
  79             line = heap_realloc(line, line_max);
  80         }
  81
  82         line[length++] = c;
  83     }
  84
  85     line[length] = 0;
  86     if (length - 1 >= 0 && line[length - 1] == '\r') /* Strip \r of windows line endings */
  87         line[length - 1] = 0;
  88
  89     line_converted_length = MultiByteToWideChar(CP_ACP, 0, line, -1, 0, 0);
  90     line_converted = heap_alloc(line_converted_length * sizeof(WCHAR));
  91     MultiByteToWideChar(CP_ACP, 0, line, -1, line_converted, line_converted_length);
  92
  93     heap_free(line);
  94
  95     return line_converted;
  96 }
  97
  98 static void write_to_stdout(const WCHAR *str)
  99 {
 100     char *str_converted;
 101     UINT str_converted_length;
 102     DWORD bytes_written;
 103     UINT str_length = lstrlenW(str);
 104     int codepage = CP_ACP;
 105
 106     str_converted_length = WideCharToMultiByte(codepage, 0, str, str_length, NULL, 0, NULL, NULL);
 107     str_converted = heap_alloc(str_converted_length);
 108     WideCharToMultiByte(codepage, 0, str, str_length, str_converted, str_converted_length, NULL, NULL);
 109
 110     WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), str_converted, str_converted_length, &bytes_written, NULL);
 111     if (bytes_written < str_converted_length)
 112         ERR("Failed to write output\n");
 113
 114     heap_free(str_converted);
 115 }
 116
 117 static BOOL run_find_for_line(const WCHAR *line, const WCHAR *tofind)
 118 {
 119     void *found;
 120     WCHAR lineending[] = {'\r', '\n', 0};
 121
 122     if (lstrlenW(line) == 0 || lstrlenW(tofind) == 0)
 123         return FALSE;
 124
 125     found = wcsstr(line, tofind);
 126
 127     if (found)
 128     {
 129         write_to_stdout(line);
 130         write_to_stdout(lineending);
 131         return TRUE;
 132     }
 133
 134     return FALSE;
 135 }
 136
 137 static void output_resource_message(int id)
 138 {
 139     WCHAR buffer[64];
 140     LoadStringW(GetModuleHandleW(NULL), id, buffer, ARRAY_SIZE(buffer));
 141     write_to_stdout(buffer);
 142 }
 143
 144 int __cdecl wmain(int argc, WCHAR *argv[])
 145 {
 146     WCHAR *line;
 147     WCHAR *tofind = NULL;
 148     int i;
 149     int exitcode;
 150     HANDLE input;
 151
 152     TRACE("running find:");
 153     for (i = 0; i < argc; i++)
 154     {
 155         TRACE(" %s", wine_dbgstr_w(argv[i]));
 156     }
 157     TRACE("\n");
 158
 159     input = GetStdHandle(STD_INPUT_HANDLE);
 160
 161     for (i = 1; i < argc; i++)
 162     {
 163         if (argv[i][0] == '/')
 164         {
 165             output_resource_message(IDS_INVALID_SWITCH);
 166             return 2;
 167         }
 168         else if(tofind == NULL)
 169         {
 170             tofind = argv[i];
 171         }
 172         else
 173         {
 174             FIXME("Searching files not supported yet\n");
 175             return 1000;
 176         }
 177     }
 178
 179     if (tofind == NULL)
 180     {
 181         output_resource_message(IDS_INVALID_PARAMETER);
 182         return 2;
 183     }
 184
 185     exitcode = 1;
 186     while ((line = read_line_from_handle(input)) != NULL)
 187     {
 188         if (run_find_for_line(line, tofind))
 189             exitcode = 0;
 190
 191         heap_free(line);
 192     }
 193
 194     return exitcode;
 195 }