一个检测二进制文件内容的工具

// incorporate.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "assert.h"
#include "stdio.h"
#include<windows.h>
#include <fstream>
#include <iostream>
#include <string>
using namespace std;
#include "jtzCompress.h"
#include "jtcCompress.h"

static bool checkContext(const char * fileName, const char * szCheck);
static bool checkIfDataHasStr(const char * fileName, const char * szCheck, Byte * pData, UInt32 nFileLen);

//单字节、双字节互转
static char * wstrTostr(const wchar_t * lpcwszStr){
    DWORD dwNum = WideCharToMultiByte(CP_OEMCP,NULL,lpcwszStr,-1,NULL,0,NULL,FALSE);
    char *psText = NULL;
    psText = new char[dwNum*sizeof(char)];
    if(!psText)
    {
        delete(psText);
        psText = NULL;
        return psText;
    }
    WideCharToMultiByte(CP_OEMCP,NULL,lpcwszStr,-1,psText,dwNum,NULL,FALSE);
    return psText;
}

static wchar_t * strToWstr(const char * lpcwszStr){
    DWORD dwNum = MultiByteToWideChar (CP_ACP, 0, lpcwszStr, -1, NULL, 0);
    wchar_t *psText = NULL;
    psText = new wchar_t[dwNum];;
    if(!psText)
    {
        delete(psText);
        psText = NULL;
        return psText;
    }
    MultiByteToWideChar (CP_ACP, 0, lpcwszStr, -1, psText, dwNum);  
    return psText;
}

ostream & green(ostream & s)
{
    HANDLE hstdhadle = GetStdHandle(STD_OUTPUT_HANDLE);
    SetConsoleTextAttribute(hstdhadle, FOREGROUND_GREEN|FOREGROUND_INTENSITY);
    return s;
}

ostream & white(ostream & s)
{
    HANDLE hstdhadle = GetStdHandle(STD_OUTPUT_HANDLE);
    SetConsoleTextAttribute(hstdhadle, FOREGROUND_RED|FOREGROUND_GREEN|FOREGROUND_BLUE|FOREGROUND_INTENSITY);
    return s;
}


class CFile{
private:
    fstream m_fout;

public:
    CFile(){
        m_fout.open("out.txt", ios::out | ios::binary);
    }

    //检查要测试文件的路径
    void checkFolder(const char* lpPath, const char * szCheck, bool bNeedCheckAllFile){
        assert(lpPath);

        char szFind[1024];
        char szFile[1024];
        WIN32_FIND_DATA FindFileData;
        strcpy_s(szFind,lpPath);
        strcat_s(szFind,"*.*");
        wchar_t * wszFind = strToWstr(szFind);
        HANDLE hFind=FindFirstFile(wszFind,&FindFileData);
        delete wszFind;
        if (INVALID_HANDLE_VALUE == hFind)
            return;

        bool bHasFileFinded = false;//如果有文件已经找到匹配了，不用全部文件遍历
        while(true)
        {
            if (FindFileData.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY)
            {
                if (FindFileData.cFileName[0]!='.')
                {
                    strcpy_s(szFile,lpPath);
                    char * pFileName = wstrTostr(FindFileData.cFileName);
                    strcat_s(szFile,pFileName);
                    strcat_s(szFile,"\\");
                    delete pFileName;
                    checkFolder(szFile, szCheck, bNeedCheckAllFile);
                }
            }
            else if(bHasFileFinded == false || bNeedCheckAllFile == true)
            {
                char * pFileName = wstrTostr(FindFileData.cFileName);
                char pFilePath[1024] = {0};
                strcpy_s(pFilePath,lpPath);
                strcat_s(pFilePath,pFileName);
                bHasFileFinded = checkContext(pFilePath, szCheck);
                delete pFileName;
            }
            if (!FindNextFile(hFind,&FindFileData))
                break;
        }
        FindClose(hFind);

    }

    bool checkContext(const char * fileName, const char * szCheck)
    {
        fstream fin;
        bool bRet = false;
        if(ifFileTypeEqual(fileName, "jtz"))
        {
            Byte* fileContext = NULL;
            UInt32 nFileLen = 0;

            wchar_t * pFileName = strToWstr(fileName);
            fin.open(pFileName, ios::in | ios::binary);//防止有中文名

            fin.seekg(0, ios::end);
            nFileLen = fin.tellg();
            fileContext = new Byte[nFileLen];
            fin.seekg(0, ios::beg);
            fin.read((char*)fileContext, nFileLen);
            fin.close();

            Byte* unzipMem = NULL;
            UInt32 unzipSize = 0;
            Unzip(&unzipMem, &unzipSize, fileContext, nFileLen);
            delete fileContext;

            //检测
            bRet = checkIfDataHasStr(fileName, szCheck, unzipMem, unzipSize);
            delete unzipMem;
            delete pFileName;
            return bRet;
        }
        if(ifFileTypeEqual(fileName, "jtc"))
        {
            Byte* unzipMem = NULL;
            UInt32 unzipSize = 0;
            CJtcCompress * jtcInstance = new CJtcCompress(fileName);
            unzipMem = jtcInstance->Decode(unzipSize);
            delete jtcInstance;

            wchar_t * pFileName = strToWstr(fileName);

            //检测
            bRet = checkIfDataHasStr(fileName, szCheck, unzipMem, unzipSize);
            delete unzipMem;
            delete pFileName;
            return bRet;
        }
        //其他文件格式,直接用strstr即可
        if(ifFileTypeEqual(fileName, "ini")
            || ifFileTypeEqual(fileName, "cfg")
            || ifFileTypeEqual(fileName, "txt")
            || ifFileTypeEqual(fileName, "mod"))
        {
            Byte* unzipMem = NULL;
            UInt32 unzipSize = 0;

            wchar_t * pFileName = strToWstr(fileName);
            fin.open(pFileName, ios::in | ios::binary);//防止有中文名

            assert(fin);
            fin.seekg(0, ios::end);
            unzipSize = fin.tellg();
            unzipMem = new Byte[unzipSize];
            fin.seekg(0, ios::beg);
            fin.read((char*)unzipMem, unzipSize);
            fin.close();

            //检测
            bRet = checkIfDataHasStr(fileName, szCheck, unzipMem, unzipSize);
            delete unzipMem;
            delete pFileName;
            return bRet;
        }
        if(ifFileTypeEqual(fileName, "bar"))
        {
            char copyLocalCmd[512];
            char delLocalCmd[512];
            char brewrcCmd[512];
            const wchar_t * wszBarName = L"demo.bar";
            const wchar_t * wszBrxName = L"demo.brx";
            const char * szBarName = "demo.bar";

            wchar_t * pFileName = strToWstr(fileName);
            copyFileToOtherPath(pFileName, wszBarName);

            sprintf_s(brewrcCmd, 512, "brewrc.exe -d %s > 1.txt", szBarName);
            system(brewrcCmd);//执行Brew反解析工具

            DeleteFile(wszBarName);

            Byte* unzipMem = NULL;
            UInt32 unzipSize = 0;

            fin.open(wszBrxName, ios::in | ios::binary);//防止有中文名
            fin.seekg(0, ios::end);
            unzipSize = fin.tellg();
            unzipMem = new Byte[unzipSize];
            fin.seekg(0, ios::beg);
            fin.read((char*)unzipMem, unzipSize);
            fin.close();

            //检测
            bRet = checkIfDataHasStr(fileName, szCheck, unzipMem, unzipSize);
            delete unzipMem;
            delete pFileName;
            return bRet;
        }
        return bRet;
    }


    //判定文件后缀是否匹配
    bool ifFileTypeEqual(const char * fullName, const char * szCompare){
        assert(fullName && szCompare);

        bool bRet = false;
        string * str = new string(fullName);
        int nLen = str->find_last_of('.');
        if (nLen <= 0)
            bRet = false;
        else
        {
            const char * sz = fullName + nLen + 1;
            if(strcmp(sz, szCompare) == 0)
                bRet = true;
        }

        delete str;
        return bRet;
    }

    //判定文件后缀是否匹配
    const char * getBrxName(const char * fullName){
        assert(fullName);

        char * szRet = NULL;
        string * str = new string(fullName);
        int nLen = str->find_last_of('.');
        if (nLen > 0)
        {
            szRet = strdup(fullName);
            nLen++;
            szRet[nLen] = 'b';
            nLen++;
            szRet[nLen] = 'r';
            nLen++;
            szRet[nLen] = 'x';
        }

        delete str;
        return szRet;
    }


    void copyFileToOtherPath(const wchar_t * wszSrc, const wchar_t * wszDes)
    {
        assert(wszDes && wszSrc);
        fstream fin, fout;
        fin.open(wszSrc, ios::in | ios::binary);
        if(fin.is_open())
        {
            Byte * buf = NULL;
            UInt32 nFileLen = 0;

            fin.seekg(0, ios::end);
            nFileLen = fin.tellg();

            buf = new Byte[nFileLen];
            fin.seekg(0, ios::beg);
            fin.read((char *)buf, nFileLen);
            fin.close();

            fout.open(wszDes, ios::out | ios::binary);
            if (fout.is_open())
            {
                fout.write((char *)buf, nFileLen);
                fout.close();
            }
        }
    }


    bool checkIfDataHasStr(const char * fileName, const char * szCheck, Byte * pData, UInt32 nFileLen)
    {
        assert(pData && fileName && szCheck);

        bool bRet = false;

        UInt32 nCheckLen = strlen(szCheck);
        UInt32 nCheckOffest = 0;
        UInt32 nOffest = 0;

        while (nOffest < nFileLen){
            while(nOffest < nFileLen && pData[nOffest++] == szCheck[nCheckOffest++])
            {
                if(nCheckOffest == nCheckLen)
                {
                    const char * c_file = "文件：";
                    const char * c_have = " 有 ";
                    const char * c_changeline = "\r\n";

                    m_fout.write(c_file, strlen(c_file));
                    m_fout.write(fileName, strlen(fileName));
                    m_fout.write(c_have, strlen(c_have));
                    m_fout.write(szCheck, strlen(szCheck));
                    m_fout.write(c_changeline, strlen(c_changeline));
                    return true;
                }
            }
            nCheckOffest = 0;
        }

        cout << white << "检测文件：" << green << fileName << white << endl;

        return bRet;
    }


    ~CFile(){
        if(m_fout)
            m_fout.close();
    }
};

int main(int argc, char* argv[]/* f */)
{
    //说明
    if(argc >= 2 && strcmp(argv[1], "/?") == 0)
    {
        cout << "参数1：要检查的目录，以‘/’结束，如“Res/”" << endl;
        cout << "参数2：要检查的字符串" << endl;
        cout << "参数3（Y/N）：是否检测到一个文件存在要查字符串，就不再查这个目录下的其他文件" << endl;
        cout << "目前会查的文件有：bar、jtz、jtc、mod、cfg、ini、txt" << endl;
        return 0;
    }

    CFile * pFile = new CFile();
    if(argc > 3)
    {
        if (toupper(*argv[3]) == 'Y')
            pFile->checkFolder(argv[1], argv[2], true);
        else
            pFile->checkFolder(argv[1], argv[2], false);
    }
    else
        pFile->checkFolder("res\\", "base_url", false);

    delete pFile;
    getchar();
    return 0;
}