#include <windows.h>
#include "debug.h"
#include "PatchFunctions.h"
#include "cenk.h"
#include "kernelmisc.h"
#include "cever_deps.h"

#ifdef WITHDEBUG
#define patchdebug debug
#else
#define patchdebug while(0)
#endif
void searchforinstruction(DWORD vbase, DWORD vsize, int nwanted, DWORD dwMask, DWORD dwValue, DwordPtrVector& v)
{
    patchdebug("searching vrange %08lx-%08lx for %08lx.%08lx\n", vbase, vbase+vsize, dwMask, dwValue);
    for (DWORD pv= vbase; pv<vbase+vsize ; pv+=0x1000)
    {
        DWORD pp= VirtToPhys(pv);
        if (pp==INVALID_PHYSICAL_ADDRESS)
            continue;
        DWORD *vpv= (DWORD*)PhysToVirt(pp);
        if (vpv==0)
            continue;
        for (DWORD *p=vpv ; p<vpv+0x1000/sizeof(DWORD) ; p++)
        {
            if (((*p)&dwMask)==(dwValue&dwMask)) {
                v.push_back(p);
                if (v.size()==nwanted)
                    return;
            }
        }
    }
    patchdebug("found %d matches\n", v.size());
}

void unpatch(const DwordPtrVector &offsets, const DwordVector &oldvalues)
{
    for (unsigned i=0 ; i<offsets.size() ; i++) {
        patchdebug("unpatch %08lx=%08lx -> %08lx\n", offsets[i], *offsets[i], oldvalues[i]);
        *offsets[i]= oldvalues[i];
    }
}