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/debug.h"
  24 #include "resources.h"
  25
  26 WINE_DEFAULT_DEBUG_CHANNEL(find);
  27
  28 static BOOL read_char_from_handle(HANDLE handle, char *char_out)
  29 {
  30     static char buffer[4096];
  31     static DWORD buffer_max = 0;
  32     static DWORD buffer_pos = 0;
  33
  34     /* Read next content into buffer */
  35     if (buffer_pos >= buffer_max)
  36     {
  37         BOOL success = ReadFile(handle, buffer, 4096, &buffer_max, NULL);
  38         if (!success || !buffer_max)
  39             return FALSE;
  40         buffer_pos = 0;
  41     }
  42
  43     *char_out = buffer[buffer_pos++];
  44     return TRUE;
  45 }
  46
  47 /* Read a line from a handle, returns NULL if the end is reached */
  48 static WCHAR* read_line_from_handle(HANDLE handle)
  49 {
  50     int line_max = 4096;
  51     int length = 0;
  52     WCHAR *line_converted;
  53     int line_converted_length;
  54     BOOL success;
  55     char *line = malloc(line_max);
  56
  57     for (;;)
  58     {
  59         char c;
  60         success = read_char_from_handle(handle, &c);
  61
  62         /* Check for EOF */
  63         if (!success)
  64         {
  65             if (length == 0)
  66                 return NULL;
  67             else
  68                 break;
  69         }
  70
  71         if (c == '\n')
  72             break;
  73
  74         /* Make sure buffer is large enough */
  75         if (length + 1 >= line_max)
  76         {
  77             line_max *= 2;
  78             line = realloc(line, line_max);
  79         }
  80
  81         line[length++] = c;
  82     }
  83
  84     line[length] = 0;
  85     if (length - 1 >= 0 && line[length - 1] == '\r') /* Strip \r of windows line endings */
  86         line[length - 1] = 0;
  87
  88     line_converted_length = MultiByteToWideChar(CP_ACP, 0, line, -1, 0, 0);
  89     line_converted = malloc(line_converted_length * sizeof(WCHAR));
  90     MultiByteToWideChar(CP_ACP, 0, line, -1, line_converted, line_converted_length);
  91
  92     free(line);
  93
  94     return line_converted;
  95 }
  96
  97 static void write_to_stdout(const WCHAR *str)
  98 {
  99     char *str_converted;
 100     UINT str_converted_length;
 101     DWORD bytes_written;
 102     UINT str_length = lstrlenW(str);
 103     int codepage = CP_ACP;
 104
 105     str_converted_length = WideCharToMultiByte(codepage, 0, str, str_length, NULL, 0, NULL, NULL);
 106     str_converted = malloc(str_converted_length);
 107     WideCharToMultiByte(codepage, 0, str, str_length, str_converted, str_converted_length, NULL, NULL);
 108
 109     WriteFile(GetStdHandle(STD_OUTPUT_HANDLE), str_converted, str_converted_length, &bytes_written, NULL);
 110     if (bytes_written < str_converted_length)
 111         ERR("Failed to write output\n");
 112
 113     free(str_converted);
 114 }
 115
 116 static BOOL run_find_for_line(const WCHAR *line, const WCHAR *tofind)
 117 {
 118     void *found;
 119
 120     if (lstrlenW(line) == 0 || lstrlenW(tofind) == 0)
 121         return FALSE;
 122
 123     found = wcsstr(line, tofind);
 124
 125     if (found)
 126     {
 127         write_to_stdout(line);
 128         write_to_stdout(L"\r\n");
 129         return TRUE;
 130     }
 131
 132     return FALSE;
 133 }
 134
 135 static void output_resource_message(int id)
 136 {
 137     WCHAR buffer[64];
 138     LoadStringW(GetModuleHandleW(NULL), id, buffer, ARRAY_SIZE(buffer));
 139     write_to_stdout(buffer);
 140 }
 141
 142 int __cdecl wmain(int argc, WCHAR *argv[])
 143 {
 144     WCHAR *line;
 145     WCHAR *tofind = NULL;
 146     int i;
 147     int exitcode;
 148     int file_paths_len = 0;
 149     int file_paths_max = 0;
 150     WCHAR** file_paths = NULL;
 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     for (i = 1; i < argc; i++)
 160     {
 161         if (argv[i][0] == '/')
 162         {
 163             output_resource_message(IDS_INVALID_SWITCH);
 164             return 2;
 165         }
 166         else if (tofind == NULL)
 167         {
 168             tofind = argv[i];
 169         }
 170         else
 171         {
 172             if (file_paths_len >= file_paths_max)
 173             {
 174                 file_paths_max = file_paths_max ? file_paths_max * 2 : 2;
 175                 file_paths = realloc(file_paths, sizeof(WCHAR*) * file_paths_max);
 176             }
 177             file_paths[file_paths_len++] = argv[i];
 178         }
 179     }
 180
 181     if (tofind == NULL)
 182     {
 183         output_resource_message(IDS_INVALID_PARAMETER);
 184         return 2;
 185     }
 186
 187     exitcode = 1;
 188
 189     if (file_paths_len > 0)
 190     {
 191         for (i = 0; i < file_paths_len; i++)
 192         {
 193             HANDLE input;
 194             WCHAR file_path_upper[MAX_PATH];
 195
 196             wcscpy(file_path_upper, file_paths[i]);
 197             wcsupr(file_path_upper);
 198
 199             input = CreateFileW(file_paths[i], GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
 200
 201             if (input == INVALID_HANDLE_VALUE)
 202             {
 203                 WCHAR buffer_message[64];
 204                 WCHAR message[300];
 205
 206                 LoadStringW(GetModuleHandleW(NULL), IDS_FILE_NOT_FOUND, buffer_message, ARRAY_SIZE(buffer_message));
 207
 208                 wsprintfW(message, buffer_message, file_path_upper);
 209                 write_to_stdout(message);
 210                 continue;
 211             }
 212
 213             write_to_stdout(L"\r\n---------- ");
 214             write_to_stdout(file_path_upper);
 215             write_to_stdout(L"\r\n");
 216             while ((line = read_line_from_handle(input)) != NULL)
 217             {
 218                 if (run_find_for_line(line, tofind))
 219                     exitcode = 0;
 220
 221                 free(line);
 222             }
 223             CloseHandle(input);
 224         }
 225     }
 226     else
 227     {
 228         HANDLE input = GetStdHandle(STD_INPUT_HANDLE);
 229         while ((line = read_line_from_handle(input)) != NULL)
 230         {
 231             if (run_find_for_line(line, tofind))
 232                 exitcode = 0;
 233
 234             free(line);
 235         }
 236     }
 237
 238     free(file_paths);
 239     return exitcode;
 240 }