/**
 * @file switch.cpp
 * @author Sina Karvandi (sina@hyperdbg.org)
 * @brief .switch command
 * @details
 * @version 0.1
 * @date 2022-01-31
 *
 * @copyright This project is released under the GNU Public License v3.
 *
 */
#include "pch.h"

/**
 * @brief help of the .switch command
 *
 * @return VOID
 */
VOID
CommandSwitchHelp()
{
    ShowMessages(".switch : shows the list of active debugging threads and switches "
                 "between processes and threads in VMI Mode.\n\n");

    ShowMessages("syntax : \t.switch \n");
    ShowMessages("syntax : \t.switch [pid ProcessId (hex)]\n");
    ShowMessages("syntax : \t.switch [tid ThreadId (hex)]\n");

    ShowMessages("\n");
    ShowMessages("\t\te.g : .switch\n");
    ShowMessages("\t\te.g : .switch list\n");
    ShowMessages("\t\te.g : .switch pid b60 \n");
    ShowMessages("\t\te.g : .switch tid b60 \n");
}

/**
 * @brief .switch command handler
 *
 * @param CommandTokens
 * @param Command
 *
 * @return VOID
 */
VOID
CommandSwitch(vector<CommandToken> CommandTokens, string Command)
{
    UINT32 PidOrTid = NULL;

    if (CommandTokens.size() > 3)
    {
        ShowMessages("incorrect use of the '%s'\n\n",
                     GetCaseSensitiveStringFromCommandToken(CommandTokens.at(0)).c_str());
        CommandSwitchHelp();
        return;
    }

    //
    // Perform switching or listing the threads
    //
    if (CommandTokens.size() == 1 || (CommandTokens.size() == 2 && CompareLowerCaseStrings(CommandTokens.at(1), "list")))
    {
        UdShowListActiveDebuggingProcessesAndThreads();
    }
    else if (CompareLowerCaseStrings(CommandTokens.at(1), "pid"))
    {
        if (!ConvertTokenToUInt32(CommandTokens.at(2), &PidOrTid))
        {
            ShowMessages("please specify a correct hex value for process id\n\n");
            return;
        }

        //
        // Switch by pid
        //
        if (UdSetActiveDebuggingThreadByPidOrTid(PidOrTid, FALSE))
        {
            ShowMessages("switched to process id: %x\n", PidOrTid);
        }
    }
    else if (CompareLowerCaseStrings(CommandTokens.at(1), "tid"))
    {
        if (!ConvertTokenToUInt32(CommandTokens.at(2), &PidOrTid))
        {
            ShowMessages("please specify a correct hex value for thread id\n\n");
            return;
        }

        //
        // Switch by tid
        //
        if (UdSetActiveDebuggingThreadByPidOrTid(PidOrTid, TRUE))
        {
            ShowMessages("switched to thread id: %x\n", PidOrTid);
        }
    }
    else
    {
        ShowMessages("err, couldn't resolve error at '%s'\n",
                     GetCaseSensitiveStringFromCommandToken(CommandTokens.at(1)).c_str());
        return;
    }
}