MoeQ.cpp

#include "MoeQ.h"

#if defined(_WIN_PLATFORM_)
wchar_t DataPath[MAX_PATH + 1]; // "./data/"
#endif

#if defined(_LINUX_PLATFORM_)
char DataPath[PATH_MAX + 1];
#endif

bool DevMode = true;

PluginSystem Plugin;

// GUID 86A445BF44A2C287597618F6F36EB68C   MAC 4F923C3D4568   4F:92:3C:3D:45:68
Android Sdk("861891778567",
            "460013521635791",
            (const byte *)"\x86\xA4\x45\xBF\x44\xA2\xC2\x87\x59\x76\x18\xF6\xF3\x6E\xB6\x8C",
            (const byte *)"\x4F\x92\x3C\x3D\x45\x68",
            "XiaoMi",
            "MIX Alpha");

void Debug()
{



}

QQ::Token LoadToken(char (&QQ)[],
#if defined(_WIN_PLATFORM_)
                    const wchar_t *
#endif

#if defined(_LINUX_PLATFORM_)
                    const char *
#endif
                        DataFilePath)
{
    QQ::Token Token;

    std::ifstream input(DataFilePath);
    char Json[1000] = {0};
    if (!input.is_open())
        throw "First login";

    input.read(Json, 1000);
    Json[input.gcount()] = 0;

    input.close();

    rapidjson::Document d;
    d.Parse<rapidjson::kParseCommentsFlag>(Json);

    if (d.HasParseError())
        throw "First login";

    if (!(d.HasMember("QQ") && d.HasMember("A2") && d.HasMember("TGT") && d.HasMember("TGTkey") && d.HasMember("D2Key") && d.HasMember("wtSessionTicket") && d.HasMember("wtSessionTicketKey") && d.HasMember("token_16A") && d.HasMember("md5")))
        throw "First login";

    memcpy(QQ, d["QQ"].GetString(), d["QQ"].GetStringLength());

    if (XBin::Hex2BinEx(d["A2"].GetString(), Token.A2) != 64)
        throw "A2 len error";

    if (XBin::Hex2BinEx(d["TGT"].GetString(), Token.TGT) != 72)
        throw "TGT len error";

    if (XBin::Hex2BinEx(d["TGTkey"].GetString(), Token.TGTkey) != 16)
        throw "TGTkey len error";

    if (XBin::Hex2BinEx(d["D2Key"].GetString(), Token.D2Key) != 16)
        throw "D2Key len error";

    if (XBin::Hex2BinEx(d["wtSessionTicket"].GetString(), Token.wtSessionTicket) != 48)
        throw "wtSessionTicket len error";

    if (XBin::Hex2BinEx(d["wtSessionTicketKey"].GetString(), Token.wtSessionTicketKey) != 16)
        throw "wtSessionTicketKey len error";

    if (XBin::Hex2BinEx(d["token_16A"].GetString(), Token.token_16A) != 56)
        throw "token_16A len error";

    if (XBin::Hex2BinEx(d["md5"].GetString(), Token.md5) != 16)
        throw "md5 len error";

    return Token;
}

void SaveToken(const char *QQ,
               const QQ::Token *Token,
#if defined(_WIN_PLATFORM_)
               const wchar_t *
#endif

#if defined(_LINUX_PLATFORM_)
               const char *
#endif
                   DataFilePath)
{

    rapidjson::Document d;
    rapidjson::Document::AllocatorType &Allocator = d.GetAllocator();
    d.SetObject();
    d.AddMember("QQ", rapidjson::Value(rapidjson::kStringType).SetString(QQ, Allocator), Allocator);
    d.AddMember("A2", rapidjson::StringRef(XBin::Bin2HexEx(Token->A2, 64), 128), Allocator);
    d.AddMember("TGT", rapidjson::StringRef(XBin::Bin2HexEx(Token->TGT, 72), 144), Allocator);
    d.AddMember("D2Key", rapidjson::StringRef(XBin::Bin2HexEx(Token->D2Key, 16), 32), Allocator);
    d.AddMember("wtSessionTicket", rapidjson::StringRef(XBin::Bin2HexEx(Token->wtSessionTicket, 48), 96), Allocator);
    d.AddMember("wtSessionTicketKey", rapidjson::StringRef(XBin::Bin2HexEx(Token->wtSessionTicketKey, 16), 32), Allocator);
    d.AddMember("token_16A", rapidjson::StringRef(XBin::Bin2HexEx(Token->token_16A, 56), 112), Allocator);
    d.AddMember("md5", rapidjson::StringRef(XBin::Bin2HexEx(Token->md5, 16), 32), Allocator);
    d.AddMember("TGTkey", rapidjson::StringRef(XBin::Bin2HexEx(Token->TGTkey, 16), 32), Allocator);

#if defined(_WIN_PLATFORM_)
    FILE *fp = _wfopen(DataFilePath, L"wb");
#endif
#if defined(_LINUX_PLATFORM_)
    FILE *fp = fopen(DataFilePath, "wb");
#endif
    if (!fp)
    {
        Log::AddLog(Log::LogType::_ERROR, Log::MsgType::PROGRAM, u8"SaveData", u8"Save Json error");
        return;
    }
    char writeBuffer[10000];
    rapidjson::FileWriteStream os(fp, writeBuffer, sizeof(writeBuffer));
    rapidjson::PrettyWriter<rapidjson::FileWriteStream> writer(os);
    d.Accept(writer);
    fclose(fp);
}

int main()
{
    char QQ[12];
#if defined(_WIN_PLATFORM_)
    SetConsoleOutputCP(65001);
    wchar_t szFilePath[MAX_PATH + 1], DataFilePath[MAX_PATH + 1], DllFilePath[MAX_PATH + 1] = {0};
    GetModuleFileName(NULL, szFilePath, MAX_PATH);
    (wcsrchr(szFilePath, _T('\\')))[1] = 0;
    wcscpy(DataPath, szFilePath);
    wcscat(DataPath, L"data\\");
#endif
#if defined(_LINUX_PLATFORM_)
    char szFilePath[PATH_MAX + 1], DataFilePath[PATH_MAX + 1], DllFilePath[PATH_MAX + 1] = {0};
    getcwd(szFilePath, sizeof(szFilePath));
    strcpy(DataPath, szFilePath);
    strcat(DataPath, "/data/");
#endif

    if (!std::filesystem::exists(DataPath))
        std::filesystem::create_directory(DataPath);

    Log::Init();
    Database::Init();
    Plugin.Load(szFilePath);

#if defined(_WIN_PLATFORM_)
    wcscpy(DataFilePath, DataPath);
    wcscat(DataFilePath, L"data.json");
#endif

#if defined(_LINUX_PLATFORM_)
    strcpy(DataFilePath, DataPath);
    strcat(DataFilePath, "data.json");
#endif

    try
    {
        QQ::Token Token = LoadToken(QQ, DataFilePath);

        //
        char c[2];
        std::cout << "Second Login?(Y/N) ";
        std::cin >> c;
        //
        // char c[2] = "Y";

        if (c[0] == 'Y')
        {
            Sdk.QQ_Init(QQ);
            Sdk.QQ_Set_Token(&Token);
            int state = Sdk.QQ_Login_Second();
            if (state != LOGIN_SUCCESS)
            {
                Log::AddLog(Log::LogType::_ERROR, Log::MsgType::OTHER, u8"SyncCookie", u8"Login failed, error code: %d, error message: %s", true, state, Sdk.QQ_GetErrorMsg());
                return 0;
            }
            Sdk.QQ_Login_Finish();
        }
        else
            throw "First login";
    }
    catch (...)
    { // 一次登录

        std::cout << "Start first login\n";

    login:
        char Password[20];
        std::cout << "Please input your account:";
        std::cin >> QQ;
        Sdk.QQ_Init(QQ);

        std::cout << "Please input your password:";
        std::cin >> Password;
        int state;
        state = Sdk.QQ_Login(Password);

    check:
        char Ticket[200];
        switch (state)
        {
        case LOGIN_SUCCESS:
            Log::AddLog(Log::LogType::INFORMATION, Log::MsgType::OTHER, u8"Login", u8"Login Successfully!");
            break;
        case LOGIN_VERIY:
            Log::AddLog(Log::LogType::INFORMATION, Log::MsgType::OTHER, u8"Login", u8"Slider verification code");
            std::cout << Sdk.QQ_Get_Viery_Ticket();
            std::cin >> Ticket;
            state = Sdk.QQ_Viery_Ticket(Ticket);
            goto check;
        case LOGIN_VERIY_SMS:
            Log::AddLog(Log::LogType::INFORMATION, Log::MsgType::OTHER, u8"Login", u8"Driver Lock");

            char notice[200];
            strcpy(notice, "Send Sms to ");
            strcat(notice, Sdk.QQ_Get_Viery_PhoneNumber());
            strcat(notice, " ?(Y/N) ");
            std::cout << notice;
            char c[2];
            std::cin >> c;

            if (c[0] == 'Y')
            {
                Sdk.QQ_Send_Sms();
                char SmsCode[7];
                std::cin >> SmsCode;
                state = Sdk.QQ_Viery_Sms(SmsCode);
                goto check;
            }
            else
                goto login;
        default:
            Log::AddLog(Log::LogType::INFORMATION, Log::MsgType::OTHER, u8"Login", u8"Login failed, error code: %d, error message: %s.", true, state, Sdk.QQ_GetErrorMsg());
            goto login;
        }
        Sdk.QQ_Login_Finish();
    }

    SaveToken(QQ, Sdk.QQ_Get_Token(), DataFilePath);

    Sdk.QQ_Online();

#if defined(DEBUG)
    Debug();
#endif

    char a[99];
    std::cin >> a;

    SaveToken(QQ, Sdk.QQ_Get_Token(), DataFilePath);

    return 0;
}