agent.c

Go to the documentation of this file.

/*
 * Copyright (c) 2020 Bitdefender
 * SPDX-License-Identifier: Apache-2.0
 */
#include "agent.h"
#include "winagent.h"
#include "lixagent.h"
#include "guests.h"


INTSTATUS
IntAgentHandleInt3(
    _In_ QWORD Rip,
    _In_ DWORD CpuNumber
    )
{
    if (gGuest.OSType == introGuestWindows)
    {
        return IntWinAgentHandleInt3(Rip, CpuNumber);
    }
    else if (gGuest.OSType == introGuestLinux)
    {
        return IntLixAgentHandleInt3(Rip);
    }
    else
    {
        return INT_STATUS_NOT_SUPPORTED;
    }
}


INTSTATUS
IntAgentHandleVmcall(
    _In_ QWORD Rip
    )
{
    if (gGuest.OSType == introGuestLinux)
    {
        return IntLixAgentHandleVmcall(Rip);
    }
    else if (gGuest.OSType == introGuestWindows)
    {
        return IntWinAgentHandleVmcall(Rip);
    }
    else
    {
        return INT_STATUS_NOT_SUPPORTED;
    }
}


INTSTATUS
IntAgentActivatePendingAgent(
    void
    )
{
    if (gGuest.OSType == introGuestUnknown)
    {
        return INT_STATUS_NOT_INITIALIZED_HINT;
    }

    if (gGuest.OSType == introGuestWindows)
    {
        return IntWinAgentActivatePendingAgent();
    }
    else if (gGuest.OSType == introGuestLinux)
    {
        return IntLixAgentActivatePendingAgent();
    }
    else
    {
        return INT_STATUS_NOT_SUPPORTED;
    }
}


void
IntAgentDisablePendingAgents(
    void
    )
{
    if (gGuest.OSType == introGuestWindows)
    {
        IntWinAgentDisablePendingAgents();
    }
    else if (gGuest.OSType == introGuestLinux)
    {
        IntLixAgentDisablePendingAgents();
    }
}


AG_WAITSTATE
IntAgentGetState(
    _Out_opt_ DWORD *Tag
    )
{
    if (gGuest.OSType == introGuestWindows)
    {
        return IntWinAgentGetState(Tag);
    }
    else if (gGuest.OSType == introGuestLinux)
    {
        return IntLixAgentGetState(Tag);
    }
    else
    {
        if (Tag != NULL)
        {
            *Tag = 0;
        }
        return agNone;
    }
}


INTSTATUS
IntAgentEnableInjection(
    void
    )
{
    if (gGuest.OSType == introGuestWindows)
    {
        return IntWinAgentEnableInjection();
    }
    else if (gGuest.OSType == introGuestLinux)
    {
        IntLixAgentEnableInjection();

        return INT_STATUS_SUCCESS;
    }
    else
    {
        return INT_STATUS_NOT_SUPPORTED;
    }
}


BOOLEAN
IntAgentIsPtrInTrampoline(
    _In_ QWORD Ptr,
    _In_ THS_PTR_TYPE Type
    )
{
    if (gGuest.OSType == introGuestWindows)
    {
        return IntWinAgentIsPtrInTrampoline(Ptr, Type);
    }

    return FALSE;
}