find.c File Reference

#include <stdio.h>
#include <stdlib.h>
#include <windef.h>
#include <winbase.h>
#include <winnls.h>
#include <winuser.h>
#include <conutils.h>
#include <strsafe.h>
#include "resource.h"

Include dependency graph for find.c:

Go to the source code of this file.

Macros
#define	FIND_LINE_BUFFER_SIZE   4096

Functions
StrStrCase
@implemented Locates a substring inside a NULL-terminated wide string. Parameters [in] pszStr The NULL-terminated string to be scanned. [in] pszSearch The NULL-terminated string to search for. [in] bIgnoreCase TRUE if case has to be ignored, FALSE otherwise. Returns Returns a pointer to the first occurrence of pszSearch in pszStr, or NULL if pszSearch does not appear in pszStr. If pszSearch points to a string of zero length, the function returns pszStr.
static PWSTR	StrStrCase (IN PCWSTR pszStr, IN PCWSTR pszSearch, IN BOOL bIgnoreCase)
FindString
@implemented Prints all lines of the stream that contain a string. Parameters [in] pStream The stream to read from. [in] pszFilePath The file name to print out. Can be NULL. [in] pszSearchString The NULL-terminated string to search for. Returns 0 if the string was found at least once, 1 otherwise.
static int	FindString (IN FILE *pStream, IN PCWSTR pszFilePath OPTIONAL, IN PCWSTR pszSearchString)
int	wmain (int argc, WCHAR *argv[])

Variables
static BOOL	bInvertSearch = FALSE
static BOOL	bCountLines = FALSE
static BOOL	bDisplayLineNumbers = FALSE
static BOOL	bIgnoreCase = FALSE
static BOOL	bDoNotSkipOfflineFiles = FALSE

Macro Definition Documentation

FIND_LINE_BUFFER_SIZE

#define FIND_LINE_BUFFER_SIZE   4096

Definition at line 23 of file find.c.

Function Documentation

FindString()

static int FindString ( IN FILE *  pStream, IN PCWSTR pszFilePath  OPTIONAL, IN PCWSTR  pszSearchString  )

static

Definition at line 106 of file find.c.

{
    LONG lLineCount = 0;
    LONG lLineNumber = 0;
    BOOL bSubstringFound;
    int iReturnValue = 1;
    WCHAR szLineBuffer[FIND_LINE_BUFFER_SIZE];
 
    if (pszFilePath != NULL)
    {
        /* Print the file's header */
        ConPrintf(StdOut, L"\n---------- %s%s",
                  pszFilePath, bCountLines ? L": " : L"\n");
    }
 
    /* Loop through every line in the file */
    // FIXME: What if the string we search for crosses the boundary of our szLineBuffer ?
    while (fgetws(szLineBuffer, _countof(szLineBuffer), pStream) != NULL)
    {
        ++lLineNumber;
 
        bSubstringFound = (StrStrCase(szLineBuffer, pszSearchString, bIgnoreCase) != NULL);
 
        /* Check if this line can be counted */
        if (bSubstringFound != bInvertSearch)
        {
            iReturnValue = 0;
 
            if (bCountLines)
            {
                ++lLineCount;
            }
            else
            {
                /* Display the line number if needed */
                if (bDisplayLineNumbers)
                {
                    ConPrintf(StdOut, L"[%ld]", lLineNumber);
                }
                ConPrintf(StdOut, L"%s", szLineBuffer);
            }
        }
    }
 
    if (bCountLines)
    {
        /* Print the matching line count */
        ConPrintf(StdOut, L"%ld\n", lLineCount);
    }
#if 0
    else if (pszFilePath != NULL && iReturnValue == 0)
    {
        /* Print a newline for formatting */
        ConPrintf(StdOut, L"\n");
    }
#endif
 
    return iReturnValue;
}

Referenced by wmain().

StrStrCase()

static PWSTR StrStrCase ( IN PCWSTR  pszStr, IN PCWSTR  pszSearch, IN BOOL  bIgnoreCase  )

static

Definition at line 52 of file find.c.

{
    if (bIgnoreCase)
    {
        LCID LocaleId;
        INT i, cch1, cch2;
 
        LocaleId = GetThreadLocale();
 
        cch1 = wcslen(pszStr);
        cch2 = wcslen(pszSearch);
 
        if (cch2 == 0)
            return (PWSTR)pszStr;
 
        for (i = 0; i <= cch1 - cch2; ++i)
        {
            if (CompareStringW(LocaleId /* LOCALE_SYSTEM_DEFAULT */,
                               NORM_IGNORECASE /* | NORM_LINGUISTIC_CASING */,
                               pszStr + i, cch2, pszSearch, cch2) == CSTR_EQUAL)
            {
                return (PWSTR)(pszStr + i);
            }
        }
        return NULL;
    }
    else
    {
        return wcsstr(pszStr, pszSearch);
    }
}

Referenced by FindString().

wmain()

int wmain	(	int	argc,
		WCHAR *	argv[]
	)

Definition at line 169 of file find.c.

{
    int i;
    int iReturnValue = 2;
    int iSearchedStringIndex = -1;
    BOOL bFoundFileParameter = FALSE;
    HANDLE hFindFile;
    WIN32_FIND_DATAW FindData;
    FILE* pOpenedFile;
    PWCHAR ptr;
    WCHAR szFullFilePath[MAX_PATH];
 
    /* Initialize the Console Standard Streams */
    ConInitStdStreams();
 
    if (argc == 1)
    {
        /* If no argument were provided by the user, display program usage and exit */
        ConResPuts(StdOut, IDS_USAGE);
        return 0;
    }
 
    /* Parse the command line arguments */
    for (i = 1; i < argc; ++i)
    {
        /* Check if this argument contains a switch */
        if (wcslen(argv[i]) == 2 && argv[i][0] == L'/')
        {
            switch (towupper(argv[i][1]))
            {
                case L'?':
                    ConResPuts(StdOut, IDS_USAGE);
                    return 0;
                case L'V':
                    bInvertSearch = TRUE;
                    break;
                case L'C':
                    bCountLines = TRUE;
                    break;
                case L'N':
                    bDisplayLineNumbers = TRUE;
                    break;
                case L'I':
                    bIgnoreCase = TRUE;
                    break;
                default:
                    /* Report invalid switch error */
                    ConResPuts(StdErr, IDS_INVALID_SWITCH);
                    return 2;
            }
        }
        else if (wcslen(argv[i]) > 2 && argv[i][0] == L'/')
        {
            /* Check if this parameter is /OFF or /OFFLINE */
            if (_wcsicmp(argv[i], L"/off") == 0 || _wcsicmp(argv[i], L"/offline") == 0)
            {
                bDoNotSkipOfflineFiles = TRUE;
            }
            else
            {
                /* Report invalid switch error */
                ConResPuts(StdErr, IDS_INVALID_SWITCH);
                return 2;
            }
        }
        else
        {
            if (iSearchedStringIndex == -1)
            {
                iSearchedStringIndex = i;
            }
            else
            {
                /* There's a file specified in the parameters, no need to read from stdin */
                bFoundFileParameter = TRUE;
            }
        }
    }
 
    if (iSearchedStringIndex == -1)
    {
        /* User didn't provide the string to search for, display program usage and exit */
        ConResPuts(StdErr, IDS_USAGE);
        return 2;
    }
 
    if (bFoundFileParameter)
    {
        /* After the command line arguments were parsed, iterate through them again to get the filenames */
        for (i = 1; i < argc; ++i)
        {
            /* If the value is a switch or the searched string, continue */
            if ((wcslen(argv[i]) > 0 && argv[i][0] == L'/') || i == iSearchedStringIndex)
            {
                continue;
            }
 
            hFindFile = FindFirstFileW(argv[i], &FindData);
            if (hFindFile == INVALID_HANDLE_VALUE)
            {
                ConResPrintf(StdErr, IDS_NO_SUCH_FILE, argv[i]);
                continue;
            }
 
            do
            {
                /* Check if the file contains offline attribute and should be skipped */
                if ((FindData.dwFileAttributes & FILE_ATTRIBUTE_OFFLINE) && !bDoNotSkipOfflineFiles)
                {
                    continue;
                }
 
                /* Skip directory */
                // FIXME: Implement recursivity?
                if (FindData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
                {
                    continue;
                }
 
                /*
                 * Build the full file path from the file specification pattern.
                 *
                 * Note that we could use GetFullPathName() instead, however
                 * we want to keep compatibility with Windows' find.exe utility
                 * that does not use this function as it keeps the file name
                 * directly based on the pattern.
                 */
                ptr = wcsrchr(argv[i], L'\\');    // Check for last directory.
                if (!ptr)
                    ptr = wcsrchr(argv[i], L':'); // Check for drive.
                if (ptr)
                {
                    /* The pattern contains a drive or directory part: keep it and concatenate the full file name */
                    StringCchCopyNW(szFullFilePath, _countof(szFullFilePath),
                                    argv[i], ptr + 1 - argv[i]);
                    StringCchCatW(szFullFilePath, _countof(szFullFilePath),
                                  FindData.cFileName);
                }
                else
                {
                    /* The pattern does not contain any drive or directory part: just copy the full file name */
                    StringCchCopyW(szFullFilePath, _countof(szFullFilePath),
                                   FindData.cFileName);
                }
 
                // FIXME: Windows' find.exe supports searching inside binary files.
                pOpenedFile = _wfopen(szFullFilePath, L"r");
                if (pOpenedFile == NULL)
                {
                    ConResPrintf(StdErr, IDS_CANNOT_OPEN, szFullFilePath);
                    continue;
                }
 
                /* NOTE: Convert the file path to uppercase for formatting */
                if (FindString(pOpenedFile, _wcsupr(szFullFilePath), argv[iSearchedStringIndex]) == 0)
                {
                    iReturnValue = 0;
                }
                else if (iReturnValue != 0)
                {
                    iReturnValue = 1;
                }
 
                fclose(pOpenedFile);
            } while (FindNextFileW(hFindFile, &FindData));
 
            FindClose(hFindFile);
        }
    }
    else
    {
        iReturnValue = FindString(stdin, NULL, argv[iSearchedStringIndex]);
    }
 
    return iReturnValue;
}

Variable Documentation

bCountLines

BOOL bCountLines = FALSE

static

Definition at line 26 of file find.c.

Referenced by FindString(), and wmain().

bDisplayLineNumbers

BOOL bDisplayLineNumbers = FALSE

static

Definition at line 27 of file find.c.

Referenced by FindString(), and wmain().

bDoNotSkipOfflineFiles

BOOL bDoNotSkipOfflineFiles = FALSE

static

Definition at line 29 of file find.c.

Referenced by wmain().

bIgnoreCase

BOOL bIgnoreCase = FALSE

static

Definition at line 28 of file find.c.

Referenced by FindString(), GetHash(), StrStrCase(), and wmain().

bInvertSearch

BOOL bInvertSearch = FALSE

static

Definition at line 25 of file find.c.

Referenced by FindString(), and wmain().