VBoxClient.h

Go to the documentation of this file.

#pragma once
#include <bzsnet/BufferedSocket.h>
#include <bzscore/buffer.h>
#include <bzscore/string.h>
#include <map>
#include <vector>
#include <string>
#include <hash_map>

#define USE_REG_WORKAROUND_DLL

namespace VBoxGDB
{

    class VBoxClient
    {
    private:
        BazisLib::Network::TCPSocketEx m_Socket;
        BazisLib::DynamicStringA m_Buffer;
        size_t m_BufferAllocationSize;
        bool m_b64Bit;
        int m_CPUCount;
        int m_BreakpointIDOfLastStopEvent;  //-1 if not a breakpoint

        typedef std::hash_map<int, ULONGLONG> BreakpointAddressMap;
        BreakpointAddressMap m_BreakpointMap;

        BazisLib::BasicBuffer m_LastBlockReadBuffer;
        ULONGLONG m_LastBlockReadAddr;  //-1LL if not valid

        BazisLib::Network::InternetAddress m_TargetAddr;

    public:

        VBoxClient(const BazisLib::Network::InternetAddress &addr);
        ~VBoxClient();

        bool Initialize();

    private:
        bool RecvTillCommandPrompt();
        bool ReadMemoryBlockAligned(ULONGLONG address, BazisLib::BasicBuffer &buffer);  //The read size is determined by VirtualBox; address should be aligned to 4 bytes
        void AdjustPCAfterBreakpoint(int breakpointNumber);
        ULONGLONG GetCurrentPC();

    public:
        std::vector<BazisLib::DynamicStringA> ExecuteCommand(const char *pCommand);

    public:
        std::map<BazisLib::DynamicStringA, ULONGLONG> ReadRegisters();
        bool WriteRegister(const char *pName, ULONGLONG value);
        bool ReadRegister(const char *pName, ULONGLONG *pValue);

        bool WaitForStop();

        void RequestStopAsync();


        void Disconnect();


        bool ReadMemory(ULONGLONG address, void *pData, size_t *pLength, bool allowCaching = true);

        bool WriteMemory(ULONGLONG address, const void *pData, size_t length);

        bool StepAndWait();

        bool ContinueAndWait();


        bool IsSuccessfulReply(const std::vector<BazisLib::DynamicStringA> &reply, bool report = true);


        int CreateBreakpoint(ULONGLONG addr);
        
        bool RemoveBreakpoint(int idx);


    protected:

        virtual void ReportError(const char *pErrorDescription);
    };
}