DNF_DLL/test/sock.cpp

#include "pch.h"
#include "sock.h"

extern  std::map<std::string, std::string>Json_STL;

void sock::Pack_Control(int idx, int code, void* p3, void* p4)
{
    DWORD Size;
    Net_Get_Dword(Size);
    char* Buffer = new char[Size + 1];
    Net_Get_Buffer(Buffer, Size);
    Buffer[Size] = '\0';

    //std::cout << Buffer << std::endl;
    /*
    std::fstream OutputFile;
    OutputFile.open("Yosin_Log.txt", std::ios::app);
    std::string line;
    OutputFile << Buffer << "\n";
    OutputFile.close();
    */

    rapidjson::Document Dom;
    Dom.Parse(DNFTOOL::UtfToGbk(Buffer).c_str());//<2F><><EFBFBD><EFBFBD> <20>ַ<EFBFBD><D6B7><EFBFBD>
    int Op = Dom["op"].GetInt();//<2F>ж<EFBFBD><D0B6><EFBFBD><EFBFBD><EFBFBD>


#if !defined GOLD_LIGHTT
    wchar_t* ss = DNFTOOL::charTowchar_t(Buffer);
    uint32_t v = GetSqVm();
    SQPushRootTable(v);
    SQPushString(v, L"Sq_Pack_Control", -1);
    SQ_Get(v, -2);
    SQPushRootTable(v);
    SQPushString(v, ss, -1);
    SQ_Call(v, 2, 0, 1);
    SQPop(v, 2);
    delete[]ss;
    if (Op > 0 && Op <= 10)Json_STL["DragonBox"] = Buffer;
#else
    //<2F><><EFBFBD>а<EFBFBD>
#if defined RANK_SWITCH
    if (Op >= 30 && Op <= 40)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //<2F><><EFBFBD><EFBFBD>
#if defined DRAGONBOX_SWITCH
    if (Op > 0 && Op <= 10)
    {
        Json_STL["DragonBox"] = Buffer;
    }
#endif
    //<2F><><EFBFBD><EFBFBD>
#if defined TUGUAN_SWITCH
    if (Op > 20008000 && Op <= 20008999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //ǩ<><C7A9>
#if defined SIGNIN
    if (Op > 20051000 && Op <= 20051999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //<2F><><EFBFBD><EFBFBD>Boss
#if defined WORLDBOSS_SWITCH
    if (Op > 10 && Op <= 20)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //<2F>һ<EFBFBD>CDK
#if defined CDKCHANGE_SWITCH
    if (Op > 20005000 && Op <= 20005999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //<2F><><EFBFBD><EFBFBD>DPS
#if defined TEAM_DPS_SWITCH
    if (Op > 610 && Op <= 620)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //<2F><>ͼ<EFBFBD><CDBC>
#if defined ANTON_SWITCH
    if (Op > 1000 && Op <= 1100)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //¬<><C2AC>
#if defined LUKE_SWITCH
    if (Op > 25600000 && Op <= 25600999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif
    //ѫ<><D1AB>
#if defined MEDAL_SWITCH
    if (Op > 20009000 && Op <= 20009999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif

    //<2F><>ʯ
#if defined HUSHI_SWITCH
    if (Op > 29991000 && Op <= 29991999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif

    //buff<66><66>װ
#ifdef BUFFSWITCHING_SWITCH
    if (Op > 20013000 && Op <= 20013999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //ǿ<><C7BF>ϵͳ
#ifdef NEWUPGRADE_SWITCH
    if (Op > 20018000 && Op <= 20018999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //ǩ<><C7A9><EFBFBD><EFBFBD>
#ifdef SIGN_IN_SWITCH
    if (Op >= 20051000 && Op <= 20051999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //ȫ<><C8AB><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
#ifdef SERVER_AUCTION_SWITCH
    if (Op >= 25701000 && Op <= 25701999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //<2F><><EFBFBD><EFBFBD>ǩ<EFBFBD><C7A9><EFBFBD>
#ifdef ONLINEEVENT_SYSTEM
    if (Op >= 20052000 && Op <= 20052999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //<2F><><EFBFBD><EFBFBD>ϵͳ
#ifdef MARRY_SYSTEM
    if (Op >= 20019000 && Op <= 20019999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //<2F><><EFBFBD>˵ľ<CBB5><C4BE><EFBFBD>
#ifdef FATE_SELECT_SYSTEM
    if (Op >= 20023000 && Op <= 20023999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //<2F><><EFBFBD><EFBFBD><EFBFBD>ֿ<EFBFBD>
#ifdef FONT_BANK_SYSTEM
    if (Op >= 20024000 && Op <= 20024999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 


    //<2F>ű<EFBFBD><C5B1><EFBFBD><EFBFBD><EFBFBD>
#ifdef RAID_AUCTION_SWITCH
    if (Op >= 25700000 && Op <= 25700999)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 

    //<2F><><EFBFBD>Ļ<EFBFBD><C4BB><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
#ifdef YOSIN_BASE
    if (Op >= 30000 && Op <= 40000)
    {
        uint32_t v = GetSqVm();
        SQPushRootTable(v);
        SQPushString(v, L"Sq_Pack_Control", -1);
        SQ_Get(v, -2);
        SQPushRootTable(v);
        SQPushString(v, DNFTOOL::charTowchar_t(Buffer), -1);
        SQ_Call(v, 2, 0, 1);
        SQPop(v, 2);
    }
#endif 
#endif
}

void sock::Pack_Control2(int idx, int code, void* p3, void* p4)
{
    DWORD Size;
    Net_Get_Dword(Size);
    char* Buffer = new char[Size + 1];
    Net_Get_Buffer(Buffer, Size);
    Buffer[Size] = '\0';

    std::cout << Size << std::endl;
    std::cout << Buffer << std::endl;

}

void sock::R_Register_Pack()
{
    auto Registerfunc = reinterpret_cast<register_pack_handler_t>(0x7186D0);

    Registerfunc(130, Pack_Control, 0);
    //Registerfunc(625, Pack_Control2, 0);
}

bool sock::Net_Get_Byte(BYTE& v)
{
    typedef bool(__cdecl* func_t)(BYTE*);
    return reinterpret_cast<func_t>(0x011AEA00)(&v);
}

bool sock::Net_Get_Short(short int& v)
{
    typedef bool(__cdecl* func_t)(short int*);
    return reinterpret_cast<func_t>(0x011AEA30)(&v);
}

bool sock::Net_Get_Dword(DWORD& v)
{
    typedef bool(__cdecl* func_t)(DWORD*);
    return reinterpret_cast<func_t>(0x011AEA60)(&v);
}

bool sock::Net_Get_Buffer(char* v, int sz)
{
    typedef bool(__cdecl* func_t)(char*, int);
    return reinterpret_cast<func_t>(0x011AEA90)(v, sz);
}
/*
bool sock::HookRegistPackHandler(int idx, pack_handler_t handler)
{
    return handleTable.emplace(idx, handler).second;
}
*/