Category

Development Platform
Home > SourceCode/Document > Windows Develop > Windows Kernel > View Code
virusfct.cpp
Package: shell.rar　[view]
Upload User: xhy777
Upload Date: 2007-02-14
Package Size: 24088k
Code Size: 1k
Category:

Windows Kernel

Development Platform:

Visual C++

virusfct.cpp：Code Content
  1. #include "viruspch.h"
  2. #include "virusmn.h"
  3. #include "virusfct.h"
  4. #include "vrsscan.h"
  5. #include "viruschk.h"
  7. CVirusFactory *pcf;
  9. STDMETHODIMP CVirusFactory::QueryInterface(REFIID riid, void **ppv)
  10. {
  11.    if((riid == IID_IClassFactory) || (riid == IID_IUnknown))
  12.    {
  13.       cRef++;
  14.       *ppv = (void *)this;
  15.       return NOERROR;
  16.    }
  18.    *ppv = NULL;
  19.    return E_NOINTERFACE;
  20. }
  22. STDMETHODIMP_(ULONG) CVirusFactory::AddRef()
  23. {
  24.    return(++cRef);
  25. }
  27. STDMETHODIMP_(ULONG) CVirusFactory::Release()
  28. {
  29.    return(--cRef);
  30. }
  32. STDMETHODIMP CVirusFactory::CreateInstance(IUnknown *pUnkOuter, REFIID riid, void **ppv)
  33. {
  34.    CVirusCheck *vc = NULL;
  35.    IUnknown *punk;
  36.    HRESULT hr;
  38.    vc = new CVirusCheck(pUnkOuter, &punk);
  39.    if(!vc)
  40.       return (E_OUTOFMEMORY);
  42.    if(punk == NULL)
  43.       return CLASS_E_NOAGGREGATION;
  45.    hr = punk->QueryInterface(riid, ppv);
  46.    if(FAILED(hr))
  47.       delete vc;
  48.    else
  49.       DllAddRef();
  50.    
  51.    punk->Release();
  52.    return hr;
  53. }
  55. STDMETHODIMP CVirusFactory::LockServer(BOOL fLock)
  56. {
  57.    if(fLock)
  58.       DllAddRef();
  59.    else
  60.       DllRelease();
  61.    return NOERROR;
  62. }