/**
 * @file hyperdbg-test.cpp
 * @author Sina Karvandi (sina@hyperdbg.org)
 * @brief perform tests on a remote process (this is the remote process)
 * @details
 * @version 0.1
 * @date 2020-09-16
 *
 * @copyright This project is released under the GNU Public License v3.
 *
 */
#include "pch.h"

using namespace std;

/**
 * @brief Main function of test process
 *
 * @param argc
 * @param argv
 * @return int
 */
int
main(int argc, char * argv[])
{
    HANDLE  PipeHandle;
    BOOLEAN SentMessageResult;
    UINT32  ReadBytes;
    char *  Buffer;

    if (argc != 2)
    {
        printf("you should not test functionalities directly, instead use 'test' "
               "command from HyperDbg...\n");
        return 1;
    }

    Buffer = (char *)malloc(TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE);
    RtlZeroMemory(Buffer, TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE);
    strcpy_s(Buffer, TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE, "Hey there, Are you HyperDbg?");

    if (!strcmp(argv[1], "im-hyperdbg"))
    {
        //
        // Perform our shaking with HyperDbg
        //

        //
        // It's not called directly, it's probably from HyperDbg
        //
        PipeHandle = NamedPipeClientCreatePipe("\\\\.\\Pipe\\HyperDbgTests");

        if (!PipeHandle)
        {
            //
            // Unable to create handle
            //
            free(Buffer);
            return 1;
        }

        SentMessageResult =
            NamedPipeClientSendMessage(PipeHandle, Buffer, strlen(Buffer) + 1);

        if (!SentMessageResult)
        {
            //
            // Sending error
            //
            free(Buffer);
            return 1;
        }

        ReadBytes = NamedPipeClientReadMessage(PipeHandle, Buffer, TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE);

        if (!ReadBytes)
        {
            //
            // Nothing to read
            //
            free(Buffer);
            return 1;
        }

        if (strcmp(Buffer,
                   "Hello, Dear Test Process... Yes, I'm HyperDbg Debugger :)") ==
            0)
        {
            //
            // *** Connected to the HyperDbg debugger ***
            //

            //
            // Now we should request the test case number from the HyperDbg Debugger
            //
            RtlZeroMemory(Buffer, TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE);

            strcpy_s(
                Buffer,
                TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE,
                "Wow! I miss you... Would you plz send me the kernel information?");

            SentMessageResult =
                NamedPipeClientSendMessage(PipeHandle, Buffer, strlen(Buffer) + 1);

            if (!SentMessageResult)
            {
                //
                // Sending error
                //
                free(Buffer);
                return 1;
            }

            //
            // Read the test case number
            //
            RtlZeroMemory(Buffer, TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE);
            ReadBytes = NamedPipeClientReadMessage(PipeHandle, Buffer, TEST_CASE_MAXIMUM_BUFFERS_TO_COMMUNICATE);

            if (!ReadBytes)
            {
                //
                // Nothing to read
                //
                free(Buffer);
                return 1;
            }

            //
            // Dispatch the test case number
            //
            TestCreateLookupTable(PipeHandle, (PVOID)Buffer, ReadBytes);

            //
            // Close the pipe connection
            //
            NamedPipeClientClosePipe(PipeHandle);

            //
            // Make sure to exit the test program
            //
            exit(0);
        }
    }
    else
    {
        printf("you should not test functionalities directly, instead use 'test' "
               "command from HyperDbg...\n");

        free(Buffer);
        return 1;
    }

    free(Buffer);
    return 0;
}