programs/findstr/main.c

   1 /*
   2  * Copyright 2012 Qian Hong
   3  * Copyright 2023 Zhiyi Zhang for CodeWeavers
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  18  */
  19
  20 #include <windows.h>
  21 #include <stdio.h>
  22 #include "findstr.h"
  23 #include "wine/debug.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(findstr);
  26
  27 static int WINAPIV findstr_error_wprintf(int msg, ...)
  28 {
  29     WCHAR msg_buffer[MAXSTRING];
  30     va_list va_args;
  31     int ret;
  32
  33     LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
  34     va_start(va_args, msg);
  35     ret = vfwprintf(stderr, msg_buffer, va_args);
  36     va_end(va_args);
  37     return ret;
  38 }
  39
  40 static int findstr_message(int msg)
  41 {
  42     WCHAR msg_buffer[MAXSTRING];
  43
  44     LoadStringW(GetModuleHandleW(NULL), msg, msg_buffer, ARRAY_SIZE(msg_buffer));
  45     return wprintf(msg_buffer);
  46 }
  47
  48 static BOOL add_file(struct findstr_file **head, const WCHAR *path)
  49 {
  50     struct findstr_file **ptr, *new_file;
  51
  52     ptr = head;
  53     while (*ptr)
  54         ptr = &((*ptr)->next);
  55
  56     new_file = calloc(1, sizeof(*new_file));
  57     if (!new_file)
  58     {
  59         WINE_ERR("Out of memory.\n");
  60         return FALSE;
  61     }
  62
  63     if (!path)
  64     {
  65         new_file->file = stdin;
  66     }
  67     else
  68     {
  69         new_file->file = _wfopen(path, L"rt,ccs=unicode");
  70         if (!new_file->file)
  71         {
  72             findstr_error_wprintf(STRING_CANNOT_OPEN, path);
  73             return FALSE;
  74         }
  75     }
  76
  77     *ptr = new_file;
  78     return TRUE;
  79 }
  80
  81 static void add_string(struct findstr_string **head, const WCHAR *string)
  82 {
  83     struct findstr_string **ptr, *new_string;
  84
  85     ptr = head;
  86     while (*ptr)
  87         ptr = &((*ptr)->next);
  88
  89     new_string = calloc(1, sizeof(*new_string));
  90     if (!new_string)
  91     {
  92         WINE_ERR("Out of memory.\n");
  93         return;
  94     }
  95
  96     new_string->string = string;
  97     *ptr = new_string;
  98 }
  99
 100 int __cdecl wmain(int argc, WCHAR *argv[])
 101 {
 102     struct findstr_string *string_head = NULL, *current_string, *next_string;
 103     struct findstr_file *file_head = NULL, *current_file, *next_file;
 104     WCHAR *string, *ptr, *buffer, line[MAXSTRING];
 105     BOOL has_string = FALSE, has_file = FALSE;
 106     int ret = 1, i, j;
 107
 108     for (i = 0; i < argc; i++)
 109         WINE_TRACE("%s ", wine_dbgstr_w(argv[i]));
 110     WINE_TRACE("\n");
 111
 112     if (argc == 1)
 113     {
 114         findstr_error_wprintf(STRING_BAD_COMMAND_LINE);
 115         return 2;
 116     }
 117
 118     for (i = 1; i < argc; i++)
 119     {
 120         if (argv[i][0] == '/')
 121         {
 122             if (argv[i][1] == '\0')
 123             {
 124                 findstr_error_wprintf(STRING_BAD_COMMAND_LINE);
 125                 return 2;
 126             }
 127
 128             j = 1;
 129             while (argv[i][j] != '\0')
 130             {
 131                 switch(argv[i][j])
 132                 {
 133                 case '?':
 134                     findstr_message(STRING_USAGE);
 135                     ret = 0;
 136                     goto done;
 137                 case 'C':
 138                 case 'c':
 139                     if (argv[i][j + 1] == ':')
 140                     {
 141                         ptr = argv[i] + j + 2;
 142                         if (*ptr == '"')
 143                             ptr++;
 144
 145                         string = ptr;
 146                         while (*ptr != '"' && *ptr != '\0' )
 147                             ptr++;
 148                         *ptr = '\0';
 149                         j = ptr - argv[i] - 1;
 150                         add_string(&string_head, string);
 151                         has_string = TRUE;
 152                     }
 153                     break;
 154                 default:
 155                     findstr_error_wprintf(STRING_IGNORED, argv[i][j]);
 156                     break;
 157                 }
 158
 159                 j++;
 160             }
 161         }
 162         else if (!has_string)
 163         {
 164             string = wcstok(argv[i], L" ", &buffer);
 165             if (string)
 166             {
 167                 add_string(&string_head, string);
 168                 has_string = TRUE;
 169             }
 170             while ((string = wcstok(NULL, L" ", &buffer)))
 171                 add_string(&string_head, string);
 172         }
 173         else
 174         {
 175             if (!add_file(&file_head, argv[i]))
 176                 goto done;
 177             has_file = TRUE;
 178         }
 179     }
 180
 181     if (!has_string)
 182     {
 183         findstr_error_wprintf(STRING_BAD_COMMAND_LINE);
 184         ret = 2;
 185         goto done;
 186     }
 187
 188     if (!has_file)
 189         add_file(&file_head, NULL);
 190
 191     current_file = file_head;
 192     while (current_file)
 193     {
 194         while (fgetws(line, ARRAY_SIZE(line), current_file->file))
 195         {
 196             current_string = string_head;
 197             while (current_string)
 198             {
 199                 if (wcsstr(line, current_string->string))
 200                 {
 201                     wprintf(line);
 202                     if (current_file->file == stdin)
 203                         wprintf(L"\n");
 204                     ret = 0;
 205                 }
 206
 207                 current_string = current_string->next;
 208             }
 209         }
 210
 211         current_file = current_file->next;
 212     }
 213
 214 done:
 215     current_file = file_head;
 216     while (current_file)
 217     {
 218         next_file = current_file->next;
 219         if (current_file->file != stdin)
 220             fclose(current_file->file);
 221         free(current_file);
 222         current_file = next_file;
 223     }
 224
 225     current_string = string_head;
 226     while (current_string)
 227     {
 228         next_string = current_string->next;
 229         free(current_string);
 230         current_string = next_string;
 231     }
 232     return ret;
 233 }