- Main Page
- Namespaces
- Classes
- Files
- Directories

- File List
- File Members

# IPinHook.cpp

    00001 /*
    00002  *      Copyright (C) 2003-2005 Gabest
    00003  *      http://www.gabest.org
    00004  *
    00005  *  This Program is free software; you can redistribute it and/or modify
    00006  *  it under the terms of the GNU General Public License as published by
    00007  *  the Free Software Foundation; either version 2, or (at your option)
    00008  *  any later version.
    00009  *
    00010  *  This Program is distributed in the hope that it will be useful,
    00011  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
    00012  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
    00013  *  GNU General Public License for more details.
    00014  *
    00015  *  You should have received a copy of the GNU General Public License
    00016  *  along with GNU Make; see the file COPYING.  If not, write to
    00017  *  the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
    00018  *  http://www.gnu.org/copyleft/gpl.html
    00019  *
    00020  */
    00021
    00022 #include "stdafx.h"
    00023 #include "IPinHook.h"
    00024
    00025 REFERENCE_TIME g_tSegmentStart = 0;
    00026
    00027 static HRESULT (STDMETHODCALLTYPE * NewSegmentOrg)(IPinC * This, /* [in] */ REFERENCE_TIME tStart, /* [in] */ REFERENCE_TIME tStop, /* [in] */ double dRate) = NULL;
    00028
    00029 static HRESULT STDMETHODCALLTYPE NewSegmentMine(IPinC * This, /* [in] */ REFERENCE_TIME tStart, /* [in] */ REFERENCE_TIME tStop, /* [in] */ double dRate)
    00030 {
    00031         g_tSegmentStart = tStart;
    00032         return NewSegmentOrg(This, tStart, tStop, dRate);
    00033 }
    00034
    00035 void HookNewSegment(IPinC* pPinC)
    00036 {
    00037         if(NewSegmentOrg == NULL)
    00038         {
    00039                 NewSegmentOrg = pPinC->lpVtbl->NewSegment;
    00040         }
    00041
    00042         g_tSegmentStart = 0;
    00043
    00044         BOOL res;
    00045         DWORD flOldProtect = 0;
    00046         res = VirtualProtect(pPinC->lpVtbl, sizeof(IPinC), PAGE_WRITECOPY, &flOldProtect);
    00047
    00048         pPinC->lpVtbl->NewSegment = NewSegmentMine;
    00049
    00050         res = VirtualProtect(pPinC->lpVtbl, sizeof(IPinC), PAGE_EXECUTE, &flOldProtect);
    00051 }