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
 121     if (lstrlenW(line) == 0 || lstrlenW(tofind) == 0)
 122         return FALSE;
 123
 124     found = wcsstr(line, tofind);
 125
 126     if (found)
 127     {
 128         write_to_stdout(line);
 129         write_to_stdout(L"\r\n");
 130         return TRUE;
 131     }
 132
 133     return FALSE;
 134 }
 135
 136 static void output_resource_message(int id)
 137 {
 138     WCHAR buffer[64];
 139     LoadStringW(GetModuleHandleW(NULL), id, buffer, ARRAY_SIZE(buffer));
 140     write_to_stdout(buffer);
 141 }
 142
 143 int __cdecl wmain(int argc, WCHAR *argv[])
 144 {
 145     WCHAR *line;
 146     WCHAR *tofind = NULL;
 147     int i;
 148     int exitcode;
 149     int file_paths_len = 0;
 150     int file_paths_max = 0;
 151     WCHAR** file_paths = NULL;
 152
 153     TRACE("running find:");
 154     for (i = 0; i < argc; i++)
 155     {
 156         TRACE(" %s", wine_dbgstr_w(argv[i]));
 157     }
 158     TRACE("\n");
 159
 160     for (i = 1; i < argc; i++)
 161     {
 162         if (argv[i][0] == '/')
 163         {
 164             output_resource_message(IDS_INVALID_SWITCH);
 165             return 2;
 166         }
 167         else if (tofind == NULL)
 168         {
 169             tofind = argv[i];
 170         }
 171         else
 172         {
 173             if (file_paths_len >= file_paths_max)
 174             {
 175                 file_paths_max = file_paths_max ? file_paths_max * 2 : 2;
 176                 file_paths = heap_realloc(file_paths, sizeof(WCHAR*) * file_paths_max);
 177             }
 178             file_paths[file_paths_len++] = argv[i];
 179         }
 180     }
 181
 182     if (tofind == NULL)
 183     {
 184         output_resource_message(IDS_INVALID_PARAMETER);
 185         return 2;
 186     }
 187
 188     exitcode = 1;
 189
 190     if (file_paths_len > 0)
 191     {
 192         for (i = 0; i < file_paths_len; i++)
 193         {
 194             HANDLE input;
 195             WCHAR file_path_upper[MAX_PATH];
 196
 197             wcscpy(file_path_upper, file_paths[i]);
 198             wcsupr(file_path_upper);
 199
 200             input = CreateFileW(file_paths[i], GENERIC_READ, 0, NULL, OPEN_EXISTING, 0, NULL);
 201
 202             if (input == INVALID_HANDLE_VALUE)
 203             {
 204                 WCHAR buffer_message[64];
 205                 WCHAR message[300];
 206
 207                 LoadStringW(GetModuleHandleW(NULL), IDS_FILE_NOT_FOUND, buffer_message, ARRAY_SIZE(buffer_message));
 208
 209                 wsprintfW(message, buffer_message, file_path_upper);
 210                 write_to_stdout(message);
 211                 continue;
 212             }
 213
 214             write_to_stdout(L"\r\n---------- ");
 215             write_to_stdout(file_path_upper);
 216             write_to_stdout(L"\r\n");
 217             while ((line = read_line_from_handle(input)) != NULL)
 218             {
 219                 if (run_find_for_line(line, tofind))
 220                     exitcode = 0;
 221
 222                 heap_free(line);
 223             }
 224             CloseHandle(input);
 225         }
 226     }
 227     else
 228     {
 229         HANDLE input = GetStdHandle(STD_INPUT_HANDLE);
 230         while ((line = read_line_from_handle(input)) != NULL)
 231         {
 232             if (run_find_for_line(line, tofind))
 233                 exitcode = 0;
 234
 235             heap_free(line);
 236         }
 237     }
 238
 239     heap_free(file_paths);
 240     return exitcode;
 241 }