summaryrefslogtreecommitdiff
path: root/src/VBox/Additions/WINNT/VBoxTray/VBoxMMR.cpp
blob: 4708c0bcd880b97b235d8616560accb2e6711bfb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
/* $Id: VBoxMMR.cpp $ */
/** @file
 * VBoxMMR - Multimedia Redirection
 */

/*
 * Copyright (C) 2012 Oracle Corporation
 *
 * This file is part of VirtualBox Open Source Edition (OSE), as
 * available from http://www.virtualbox.org. This file is free software;
 * you can redistribute it and/or modify it under the terms of the GNU
 * General Public License (GPL) as published by the Free Software
 * Foundation, in version 2 as it comes in the "COPYING" file of the
 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
 */

#include "VBoxTray.h"
#include "VBoxMMR.h"
#include <iprt/ldr.h>

struct VBOXMMRCONTEXT
{
    RTLDRMOD  hModHook;
    HHOOK     hHook;
};

static VBOXMMRCONTEXT gCtx = {0};

static const char *g_pszMMRDLL  = "VBoxMMRHook";
static const char *g_pszMMRPROC = "CBTProc";

void VBoxMMRCleanup(VBOXMMRCONTEXT *pCtx)
{
    if (pCtx->hHook)
    {
        UnhookWindowsHookEx(pCtx->hHook);
        pCtx->hHook = NULL;
    }

    if (pCtx->hModHook != NIL_RTLDRMOD)
    {
        RTLdrClose(pCtx->hModHook);
        pCtx->hModHook = NIL_RTLDRMOD;
    }
}

int VBoxMMRInit(const VBOXSERVICEENV *pEnv, void **ppInstance, bool *pfStartThread)
{
    LogRel2(("VBoxMMR: Initializing\n"));

    int rc = RTLdrLoadAppPriv(g_pszMMRDLL, &gCtx.hModHook);
    if (RT_SUCCESS(rc))
    {
        HOOKPROC pHook = (HOOKPROC)RTLdrGetFunction(gCtx.hModHook, g_pszMMRPROC);
        if (pHook)
        {
            HMODULE hMod = (HMODULE)RTLdrGetNativeHandle(gCtx.hModHook);
            Assert(hMod != (HMODULE)~(uintptr_t)0);
            gCtx.hHook = SetWindowsHookEx(WH_CBT, pHook, hMod, 0);
            if (gCtx.hHook)
            {
                *ppInstance = &gCtx;
                return VINF_SUCCESS;
            }

            rc = RTErrConvertFromWin32(GetLastError());
            LogRel2(("VBoxMMR: Error installing hooking proc: %Rrc\n", rc));
        }
        else
        {
            LogRel2(("VBoxMMR: Hooking proc not found\n"));
            rc = VERR_NOT_FOUND;
        }

        RTLdrClose(gCtx.hModHook);
        gCtx.hModHook = NIL_RTLDRMOD;
    }
    else
        LogRel2(("VBoxMMR: Hooking library not found (%Rrc)\n", rc));

    return rc;
}

void VBoxMMRDestroy(const VBOXSERVICEENV *pEnv, void *pInstance)
{
    VBOXMMRCONTEXT *pCtx = (VBOXMMRCONTEXT *) pInstance;

    VBoxMMRCleanup(pCtx);
}

unsigned __stdcall VBoxMMRThread(void *pInstance)
{
    return 0;
}