Chinaunix首页 | 论坛 | 博客

shoela950

首页　| 　博文目录　| 　关于我

shoela950

  • 博客访问： 640446
  • 博文数量： 133
  • 博客积分： 1566
  • 博客等级： 上尉
  • 技术积分： 1230
  • 用 户 组： 普通用户
  • 注册时间： 2010-12-01 09:31
文章分类

全部博文（133）

文章存档

2019年（1）

2018年（1）

2017年（8）

2016年（9）

2015年（17）

2014年（4）

2013年（31）

2012年（25）

2011年（36）

2010年（1）

我的朋友
最近访客
推荐博文
相关博文
MimeFilter 实现, 纯C++

分类： C/C++

2013-04-18 11:38:44


点击(此处)折叠或打开

  1. ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
  2. #include "../NSimple.h"
  3. #include "../class/stdEx.h"

  5. /*
  6. Yofoo-HtmlFilter
  7. Code By    Yofoo
  8. Last Edit: 2010-12-10
  9. */

  11. #define MIDL_DEFINE_GUID(type,name,l,w1,w2,b1,b2,b3,b4,b5,b6,b7,b8) /
  12.     const type name = {l,w1,w2,{b1,b2,b3,b4,b5,b6,b7,b8}}

  14. MIDL_DEFINE_GUID(CLSID, CLSID_HtmlFilter, 0x6F666F59, 0x2D6F, 0x7448, 0x6D, 0x6C, 0x46, 0x69, 0x6C, 0x74, 0x65, 0x72);


  17. #if 0
  18. #define MEM_TRACE        ATLTRACE
  19. #else
  20. #define MEM_TRACE        __noop
  21. #endif

  23. #if 0
  24. #define FLT_TRACE        ATLTRACE
  25. #else
  26. #define FLT_TRACE        __noop
  27. #endif

  29. #define FunInlineDefine        {return S_OK;}

  31. class CoInterface : public IClassFactory
  32. {
  33. public:
  34.     STDMETHOD(GetTypeInfoCount)(unsigned int*)FunInlineDefine
  35.     STDMETHOD(GetTypeInfo)(unsigned int, LCID, ITypeInfo**)FunInlineDefine
  36.     STDMETHOD(GetIDsOfNames)(REFIID, LPOLESTR*, unsigned int, LCID, DISPID*)FunInlineDefine
  37.     STDMETHOD(Invoke)(DISPID, REFIID, LCID, unsigned short, DISPPARAMS*,
  38.         VARIANT*, EXCEPINFO*, unsigned int*)FunInlineDefine

  40.     STDMETHOD(Start)(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink, IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved)FunInlineDefine
  41.     STDMETHOD(Continue)(PROTOCOLDATA *pProtocolData)FunInlineDefine
  42.     STDMETHOD(Abort)(HRESULT hrReason, DWORD dwOptions)FunInlineDefine
  43.     STDMETHOD(Terminate)(DWORD dwOptions)FunInlineDefine
  44.     STDMETHOD(Suspend)()FunInlineDefine
  45.     STDMETHOD(Resume)()FunInlineDefine

  47.     STDMETHOD(Read)(void *pv, ULONG cb, ULONG *pcbRead)FunInlineDefine
  48.     STDMETHOD(Seek)(LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)FunInlineDefine
  49.     STDMETHOD(UnlockRequest)()FunInlineDefine
  50.     STDMETHOD(LockRequest)(DWORD dwOptions)FunInlineDefine

  52.     STDMETHOD (ReportData)( DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)FunInlineDefine
  53.     STDMETHOD (ReportProgress)( ULONG ulStatusCode, LPCWSTR szStatusText)FunInlineDefine
  54.     STDMETHOD (ReportResult)( HRESULT hrResult, DWORD dwError, LPCWSTR szResult)FunInlineDefine
  55.     STDMETHOD (Switch)( PROTOCOLDATA *pProtocolData)FunInlineDefine

  57.     STDMETHOD (SetPriority)(LONG nPriority)FunInlineDefine
  58.     STDMETHOD (GetPriority)(LONG * pnPriority)FunInlineDefine

  60.     virtual ~CoInterface(){}
  61. };

  63. class CHtmlFilter;
  64. //#define FunInlineDefine        ;

  66. #define DefAddRefFun    /

  68. #define Co_AgnetFun0(_CoVar, _FunRet, _FunName)    /
  69.     _FunRet STDMETHODCALLTYPE _FunName()/
  70.     {/
  71.         return _CoVar->_FunName();/
  72.     }
  73. #define Co_AgnetFun1(_CoVar, _FunRet, _FunName, _P1Type, _P1)    /
  74.     _FunRet STDMETHODCALLTYPE _FunName(_P1Type _P1)/
  75.     {/
  76.         return _CoVar->_FunName(_P1);/
  77.     }
  78. #define Co_AgnetFun2(_CoVar, _FunRet, _FunName, _P1Type, _P1, _P2Type, _P2)    /
  79.     _FunRet STDMETHODCALLTYPE _FunName(_P1Type _P1, _P2Type _P2)/
  80.     {/
  81.         return _CoVar->_FunName(_P1, _P2);/
  82.     }
  83. #define Co_AgnetFun3(_CoVar, _FunRet, _FunName, _P1Type, _P1, _P2Type, _P2, _P3Type, _P3)    /
  84.     _FunRet STDMETHODCALLTYPE _FunName(_P1Type _P1, _P2Type _P2, _P3Type _P3)/
  85.     {/
  86.         return _CoVar->_FunName(_P1, _P2, _P3);/
  87.     }
  88. #define Co_AgnetFun4(_CoVar, _FunRet, _FunName, _P1Type, _P1, _P2Type, _P2, _P3Type, _P3, _P4Type, _P4)    /
  89.     _FunRet STDMETHODCALLTYPE _FunName(_P1Type _P1, _P2Type _P2, _P3Type _P3, _P4Type _P4)/
  90.     {/
  91.         return _CoVar->_FunName(_P1, _P2, _P3, _P4);/
  92.     }
  93. #define Co_AgnetFun5(_CoVar, _FunRet, _FunName, _P1Type, _P1, _P2Type, _P2, _P3Type, _P3, _P4Type, _P4, _P5Type, _P5)    /
  94.     _FunRet STDMETHODCALLTYPE _FunName(_P1Type _P1, _P2Type _P2, _P3Type _P3, _P4Type _P4, _P5Type _P5)/
  95.     {/
  96.         return _CoVar->_FunName(_P1, _P2, _P3, _P4, _P5);/
  97.     }
  98. #define Co_AgnetFun8(_CoVar, _FunRet, _FunName, _P1Type, _P1, _P2Type, _P2, _P3Type, _P3, _P4Type, _P4, _P5Type, _P5, _P6Type, _P6, _P7Type, _P7, _P8Type, _P8)    /
  99.     _FunRet STDMETHODCALLTYPE _FunName(_P1Type _P1, _P2Type _P2, _P3Type _P3, _P4Type _P4, _P5Type _P5, _P6Type _P6, _P7Type _P7, _P8Type _P8)/
  100.     {/
  101.         return _CoVar->_FunName(_P1, _P2, _P3, _P4, _P5, _P6, _P7, _P8);/
  102.     }

  104. #define CoAgentJmp0(_CoVar, _FunRet, _FunName)    /
  105.     _FunRet STDMETHODCALLTYPE _FunName()/
  106.     {/
  107.         __asm jmp _CoVar->_FunName;/
  108.     }

  110. class CHtmlFilter_IDispatch: public IDispatch
  111. {
  112. public:
  113.     CHtmlFilter_IDispatch(CoInterface *pFacClass)
  114.     {
  115.         m_pFacClass = pFacClass;
  116.     }
  117.     CoInterface        *m_pFacClass;

  119.     Co_AgnetFun2(m_pFacClass, HRESULT, QueryInterface, REFIID, riid, LPVOID*, ppv);
  120.     Co_AgnetFun0(m_pFacClass, ULONG, AddRef);
  121.     Co_AgnetFun0(m_pFacClass, ULONG, Release);

  123.     Co_AgnetFun1(m_pFacClass, HRESULT, GetTypeInfoCount, unsigned int*, pCount);
  124.     Co_AgnetFun3(m_pFacClass, HRESULT, GetTypeInfo, unsigned int, iTInfo, LCID, lcid, ITypeInfo**, ppTInfo);
  125.     Co_AgnetFun5(m_pFacClass, HRESULT, GetIDsOfNames, REFIID, riid, LPOLESTR *, rgszNames, UINT, cNames, LCID, lcid, DISPID *, rgDispId);
  126.     Co_AgnetFun8(m_pFacClass, HRESULT, Invoke, DISPID, dispIdMember, REFIID, riid, LCID, lcid, WORD, wFlags, DISPPARAMS *, pDispParams, VARIANT *, pVarResult, EXCEPINFO *, pExcepInfo, UINT *, puArgErr);
  127. };

  129. class CHtmlFilter_IInternetPriority : public IInternetPriority
  130. {
  131. public:
  132.     CHtmlFilter_IInternetPriority(CoInterface *pFacClass)
  133.     {
  134.         m_pFacClass = pFacClass;
  135.     }
  136.     CoInterface        *m_pFacClass;

  138.     Co_AgnetFun2(m_pFacClass, HRESULT, QueryInterface, REFIID, riid, LPVOID*, ppv);
  139.     Co_AgnetFun0(m_pFacClass, ULONG, AddRef);
  140.     Co_AgnetFun0(m_pFacClass, ULONG, Release);

  142.     Co_AgnetFun1(m_pFacClass, HRESULT, SetPriority, LONG, nPriority);
  143.     Co_AgnetFun1(m_pFacClass, HRESULT, GetPriority, LONG *, pnPriority);
  144. };

  146. class CHtmlFilter_IInternetProtocol: public IInternetProtocol
  147. {
  148. public:
  149.     CHtmlFilter_IInternetProtocol(CoInterface * pFacClass)
  150.     {
  151.         m_pFacClass = pFacClass;
  152.     }
  153.     CoInterface        *m_pFacClass;

  155.     Co_AgnetFun2(m_pFacClass, HRESULT, QueryInterface, REFIID, riid, LPVOID*, ppv);
  156.     Co_AgnetFun0(m_pFacClass, ULONG, AddRef);
  157.     Co_AgnetFun0(m_pFacClass, ULONG, Release);

  159.     //    IInternetProtocolRoot
  160.     Co_AgnetFun5(m_pFacClass, HRESULT, Start, LPCWSTR, szUrl,IInternetProtocolSink *, pOIProtSink,
  161.         IInternetBindInfo *, pOIBindInfo, DWORD, grfPI, HANDLE_PTR, dwReserved);
  162.     Co_AgnetFun1(m_pFacClass, HRESULT, Continue, PROTOCOLDATA *, pProtocolData);
  163.     Co_AgnetFun2(m_pFacClass, HRESULT, Abort, HRESULT, hrReason, DWORD, dwOptions);
  164.     Co_AgnetFun1(m_pFacClass, HRESULT, Terminate, DWORD, dwOptions);
  165.     Co_AgnetFun0(m_pFacClass, HRESULT, Suspend);
  166.     Co_AgnetFun0(m_pFacClass, HRESULT, Resume);

  168.     //    IInternetProtocol
  169.     Co_AgnetFun3(m_pFacClass, HRESULT, Read, void *, pv, ULONG, cb, ULONG *, pcbRead);
  170.     Co_AgnetFun3(m_pFacClass, HRESULT, Seek, LARGE_INTEGER, dlibMove, DWORD, dwOrigin, ULARGE_INTEGER *, plibNewPosition);
  171.     Co_AgnetFun0(m_pFacClass, HRESULT, UnlockRequest);
  172.     Co_AgnetFun1(m_pFacClass, HRESULT, LockRequest, DWORD, dwOptions);
  173. };

  175. class CHtmlFilter_IInternetProtocolSink: public IInternetProtocolSink
  176. {
  177. public:
  178.     CHtmlFilter_IInternetProtocolSink(CoInterface *pFacClass)
  179.     {
  180.         m_pFacClass = pFacClass;
  181.     }
  182.     CoInterface        *m_pFacClass;

  184.     Co_AgnetFun2(m_pFacClass, HRESULT, QueryInterface, REFIID, riid, LPVOID*, ppv);
  185.     Co_AgnetFun0(m_pFacClass, ULONG, AddRef);
  186.     Co_AgnetFun0(m_pFacClass, ULONG, Release);

  188.     Co_AgnetFun3(m_pFacClass, HRESULT, ReportData, DWORD, grfBSCF, ULONG, ulProgress, ULONG, ulProgressMax);
  189.     Co_AgnetFun2(m_pFacClass, HRESULT, ReportProgress, ULONG, ulStatusCode, LPCWSTR, szStatusText);
  190.     Co_AgnetFun3(m_pFacClass, HRESULT, ReportResult, HRESULT, hrResult, DWORD, dwError, LPCWSTR, szResult);
  191.     Co_AgnetFun1(m_pFacClass, HRESULT, Switch, PROTOCOLDATA *, pProtocolData);
  192. };

  194. typedef enum _FilterType
  195. {
  196.     Filter_namespcae    = 0x1000,
  197.     Filter_Mime            = 0x2000
  198. }FilterType;

  200. class CHtmlFilter: public CoInterface
  201. {
  202. public:
  203.     CHtmlFilter(int nFilterType = Filter_Mime):
  204.      m_xIDispatch(this),
  205.      m_xIInternetProtocol(this),
  206.      m_xIInternetProtocolSink(this),
  207.      m_xIInternetPriority(this)
  208.     {
  209.         m_nRefCount = 0;
  210.         m_nFilterType = nFilterType;
  211.     }

  213.     virtual ~CHtmlFilter()
  214.     {

  216.     }

  218.     CHtmlFilter_IDispatch                m_xIDispatch;
  219.     CHtmlFilter_IInternetProtocol        m_xIInternetProtocol;
  220.     CHtmlFilter_IInternetProtocolSink    m_xIInternetProtocolSink;
  221.     CHtmlFilter_IInternetPriority        m_xIInternetPriority;


  224.     CComPtr<IInternetProtocol>        m_UrlMonProtocol;
  225.     CComPtr<IInternetProtocolSink>    m_UrlMonProtocolSink;
  226.     std::wstring                    m_sCacheFilePath;
  227.     std::wstring                    m_sCurUrl;
  228.     std::wstring                    m_sCurMime;

  230.     LONG            m_nRefCount;
  231.     int            m_nFilterType;

  233.     STDMETHOD(QueryInterface)(REFIID riid, LPVOID* ppv)
  234.     {
  235.         if(IsEqualIID(riid, IID_IUnknown))
  236.         {
  237.             *ppv = this;
  238.             AddRef();
  239.         }
  240.         else if(IsEqualIID(riid, IID_IDispatch))
  241.         {
  242.             *ppv = &m_xIDispatch;
  243.         }
  244.         else if(IsEqualIID(riid, IID_IInternetProtocol))
  245.         {
  246.             *ppv = &m_xIInternetProtocol;
  247.         }
  248.         else if(IsEqualIID(riid, IID_IInternetProtocolSink))
  249.         {
  250.             *ppv = &m_xIInternetProtocolSink;
  251.         }
  252.         else if(IsEqualIID(riid, IID_IInternetPriority))
  253.         {
  254.             *ppv = &m_xIInternetPriority;
  255.         }
  256.         else
  257.         {
  258.             assert(0);
  259.             return ResultFromScode(E_NOINTERFACE);
  260.         }
  261.         AddRef();
  262.         return S_OK;
  263.     }

  265.     STDMETHOD_(ULONG, AddRef)()
  266.     {
  267.         return InterlockedIncrement(&m_nRefCount);
  268.     }
  269.     STDMETHOD_(ULONG, Release)()
  270.     {
  271.         int        nRef;

  273.         nRef = InterlockedDecrement(&m_nRefCount);
  274.         if(nRef <= 0)
  275.         {
  276.             MEM_TRACE("delete: %08X/n", this);
  277.             delete this;
  278.         }

  280.         return nRef;
  281.     }
  282.     virtual /* [local] */ HRESULT STDMETHODCALLTYPE CreateInstance(
  283.         /* [unique][in] */ IUnknown *pUnkOuter,
  284.         /* [in] */ REFIID riid,
  285.         /* [iid_is][out] */ void **ppvObject)
  286.     {
  287.         CHtmlFilter        *pFacClass;
  288.         HRESULT            hr;

  290.         pFacClass = new CHtmlFilter(m_nFilterType);
  291.         MEM_TRACE("New: %08X/n", pFacClass);
  292.         hr = pFacClass->QueryInterface(riid, ppvObject);
  293.         if(FAILED(hr))
  294.             delete pFacClass;

  296.         return hr;
  297.     }

  299.     virtual /* [local] */ HRESULT STDMETHODCALLTYPE LockServer(
  300.     /* [in] */ BOOL fLock)
  301.     {
  302.         return S_OK;
  303.     }


  306. public:
  307.     //    IInternetProtocolRoot
  308.     STDMETHOD(Start)(LPCWSTR szUrl,    IInternetProtocolSink *pOIProtSink,    IInternetBindInfo *pOIBindInfo,
  309.         DWORD grfPI, HANDLE_PTR dwReserved)
  310.     {
  311. #define        OT_MAX        64

  313.         HRESULT        hr;
  314.         BINDINFO    BdInfo;
  315.         ULONG        uFetched;
  316.         LPOLESTR    pzUrl[OT_MAX+1];
  317.         DWORD        dwBdf;
  318.         PROTOCOLFILTERDATA    *pFiltDat;

  320.         m_UrlMonProtocolSink = pOIProtSink;

  322.         memset(pzUrl, 0, sizeof(pzUrl));
  323.         hr = pOIBindInfo->GetBindString(BINDSTRING_ACCEPT_MIMES, pzUrl, OT_MAX, &uFetched);
  324.         if(uFetched)
  325.         {
  326.             if(wcscmp(pzUrl[0], L"*/*") != 0)
  327.                 FLT_TRACE("BINDSTRING_ACCEPT_MIMES: %S/n", pzUrl[0]);
  328.         }

  330.         memset(pzUrl, 0, sizeof(pzUrl));
  331.         hr = pOIBindInfo->GetBindString(BINDSTRING_USER_AGENT, pzUrl, OT_MAX, &uFetched);
  332.         if(uFetched && pzUrl[0])
  333.             FLT_TRACE("BINDSTRING_USER_AGENT: %S/n", pzUrl[0]);
  334.         memset(pzUrl, 0, sizeof(pzUrl));
  335.         hr = pOIBindInfo->GetBindString(BINDSTRING_POST_COOKIE, pzUrl, OT_MAX, &uFetched);
  336.         if(uFetched && pzUrl[0])
  337.             FLT_TRACE("BINDSTRING_POST_COOKIE: %S/n", pzUrl[0]);
  338.         memset(pzUrl, 0, sizeof(pzUrl));
  339.         hr = pOIBindInfo->GetBindString(BINDSTRING_POST_DATA_MIME, pzUrl, OT_MAX, &uFetched);
  340.         if(uFetched && pzUrl[0])
  341.             FLT_TRACE("BINDSTRING_POST_DATA_MIME: %S/n", pzUrl[0]);

  343.         memset(&BdInfo, 0, sizeof(BdInfo));
  344.         BdInfo.cbSize = sizeof(BdInfo);
  345.         hr = pOIBindInfo->GetBindInfo(&dwBdf, &BdInfo);

  347.         if(m_nFilterType & Filter_namespcae)
  348.         {
  349.             CHAR            szCurUrl[1024];

  351.             FLT_TRACE("IInternetProtocolRoot::Start: %S/n", szUrl);
  352.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  354.             NStr::WChar2Char((WCHAR *)szUrl, szCurUrl, sizeof(szCurUrl));
  355.             if(NStr::StrTailCmp(szCurUrl, ".js") == 0 ||
  356.                 NStr::StrTailCmp(szCurUrl, ".jpg") == 0 ||
  357.                 NStr::StrTailCmp(szCurUrl, ".swf") == 0 ||
  358.                 0)
  359.             {
  360.                 m_UrlMonProtocolSink->ReportResult(S_OK, 0, NULL);
  361.                 m_UrlMonProtocol.Release();
  362.                 return S_OK;
  363.             }

  365.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;
  366.         }

  368.         pFiltDat = (PROTOCOLFILTERDATA *)dwReserved;
  369.         hr = pOIProtSink->QueryInterface(IID_IInternetProtocol, (void**)&m_UrlMonProtocol);
  370.         ClearBuffer();

  372.         memset(pzUrl, 0, sizeof(pzUrl));
  373.         m_sCurMime = szUrl;

  375.         try
  376.         {
  377.             hr = pOIBindInfo->GetBindString(BINDSTRING_URL, pzUrl, OT_MAX, &uFetched);
  378.             m_sCurUrl = pzUrl[0];
  379.             FLT_TRACE("IInternetProtocolRoot::Start: %S, %S/n", szUrl, m_sCurUrl.c_str());
  380.         }
  381.         catch (CMemoryException* e)
  382.         {
  383.         }
  384.         catch (...)
  385.         {
  386.         }

  388.         return S_OK;
  389.     }
  390.     STDMETHOD(Continue)(PROTOCOLDATA * pProtocolData)
  391.     {
  392.         if(m_nFilterType & Filter_namespcae)
  393.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  395.         HRESULT hr;
  396.         hr = m_UrlMonProtocol->Continue(pProtocolData);
  397.         return hr;
  398.     }
  399.     STDMETHOD(Abort)(HRESULT hrReason, DWORD dwOptions)
  400.     {
  401.         if(m_nFilterType & Filter_namespcae)
  402.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  404.         HRESULT hr;
  405.         hr = m_UrlMonProtocol->Abort(hrReason, dwOptions);
  406.         return hr;
  407.     }
  408.     STDMETHOD(Terminate)(DWORD dwOptions)
  409.     {
  410.         if(m_nFilterType & Filter_namespcae)
  411.             return S_OK;

  413.         HRESULT hr;
  414.         ClearBuffer();
  415.         hr = m_UrlMonProtocol->Terminate(dwOptions);
  416.         return hr;
  417.     }
  418.     STDMETHOD(Suspend)()
  419.     {
  420.         if(m_nFilterType & Filter_namespcae)
  421.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  423.         HRESULT hr;
  424.         hr = m_UrlMonProtocol->Suspend();
  425.         return hr;
  426.     }
  427.     STDMETHOD(Resume)()
  428.     {
  429.         if(m_nFilterType & Filter_namespcae)
  430.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  432.         HRESULT hr;
  433.         hr = m_UrlMonProtocol->Resume();
  434.         return hr;
  435.     }

  437.     //    IInternetProtocol
  438.     ByteMem            m_ReadBuffer;
  439.     ByteMem            m_OneBuffer;
  440.     BOOL            m_bReadFull;

  442.     BOOL    ClearBuffer()
  443.     {
  444.         m_ReadBuffer.clear();
  445.         m_OneBuffer.resize(16*1024);
  446.         m_bReadFull = FALSE;
  447.         return TRUE;
  448.     }


  451.     //    <mce:script .... </script><!--
  452.     替换成空格
  453.     static BOOL WINAPI LoopFindPro(CHAR *pFind, VOID *pParam)
  454.     {
  455.         CHAR    *pEnd;
  456.         int        nSize;
  457.         CHAR    *pTailStr;

  459.         pTailStr = "
  460. // -->";
  461.         pEnd = StrStrIA(pFind, pTailStr);
  462.         if(pEnd == NULL)
  463.             return FALSE;

  465.         nSize = (int)(pEnd - pFind + strlen(pTailStr));
  466.         memset(pFind, ' ', nSize);
  467.         return TRUE;
  468.     }

  470.     BOOL    ProcessBuffer()
  471.     {
  472.         WCHAR        *pUrl;
  473.         CHAR        *pData;
  474.         DWORD        dwSize;

  476.         dwSize = (DWORD)m_ReadBuffer.size();
  477.         m_ReadBuffer.append((BYTE *)"", 1);    //    添加字符串结束标记
  478.         pData = (CHAR*)m_ReadBuffer.getptr();
  479.         pUrl = (WCHAR *)m_sCurUrl.c_str();
  480.         if(dwSize <= 0)
  481.             return TRUE;

  483.         if(wcsicmp(m_sCurMime.c_str(), L"text/html") != 0 &&
  484.             wcsicmp(m_sCurMime.c_str(), L"application/x-javascript") != 0 &&
  485.             wcsicmp(m_sCurMime.c_str(), L"text/css") != 0 &&
  486.             1)
  487.             return TRUE;

  489.         if(NStr::StrHeadCmp((WCHAR *)m_sCurUrl.c_str(), L"res://") == 0)
  490.         {
  491.             NStr::StrReplace(pData, "宋体", "雅黑");
  492.             return TRUE;
  493.         }

  495. //        return TRUE;
  496.         //    自己在这里处理
  497. //        NStr::LoopFindNoCaseString(LoopFindPro, pData, ", pData);    //简单去掉脚本元素
  498.         CHAR        szOld[64];
  499.         CHAR        szNew[64];
  500.         int            nCount;

  502.         nCount = 0;
  503.         nCount += NStr::StrReplace(pData, "天涯", "太阳");        //    普通编码

  505.         NStr::CodePageConvert(936, CP_UTF8, "天涯", szOld, sizeof(szOld));
  506.         NStr::CodePageConvert(936, CP_UTF8, "太阳", szNew, sizeof(szNew));
  507.         nCount += NStr::StrReplace(pData, szOld, szNew);        //    UTF8编码

  509.         NStr::String2HtmTranMean("天涯", szOld);
  510.         NStr::String2HtmTranMean("太阳", szNew);
  511.         nCount += NStr::StrReplace(pData, szOld, szNew);        //    Htm转义字符10进制

  513.         NStr::String2HtmTranMean("天涯", szOld, TRUE);
  514.         NStr::String2HtmTranMean("太阳", szNew, TRUE);
  515.         nCount += NStr::StrIReplace(pData, szOld, szNew);        //    Htm转义字符16进制


  518.         nCount += NStr::StrReplace(pData, "宋体", "雅黑");        //    普通编码

  520.         CHAR *pFind = strstr(pData, "宋体");
  521.         if(pFind)
  522.             TRACE("Find: 宋体/n");

  524.         FLT_TRACE("Replace: %d, %S/n", nCount, m_sCurUrl.c_str());

  526.         //    处理完毕
  527.         m_ReadBuffer.pop_back();            //    删除添加的字符串结束标记

  529.         return TRUE;
  530.     }

  532.     BOOL    RealFulltoBuffer()
  533.     {
  534.         HRESULT            hr;
  535.         DWORD            dwRead;

  537.         if(m_bReadFull)
  538.             return TRUE;

  540.         for(;;)
  541.         {
  542.             dwRead = 0;
  543.             hr = m_UrlMonProtocol->Read(m_OneBuffer.getptr(), (ULONG)m_OneBuffer.size(), &dwRead);
  544.             if(dwRead != 0)
  545.                 m_ReadBuffer.append(m_OneBuffer.getptr(), dwRead);

  547.             if(hr == S_OK)
  548.                 continue;
  549.             else if(hr == S_FALSE)
  550.                 break;
  551.             else
  552.                 return FALSE;
  553.         }

  555.         m_bReadFull = TRUE;
  556.         ProcessBuffer();
  557.         return TRUE;
  558.     }

  560.     BOOL    ReadFromBuffer(void *pv, ULONG cb, ULONG *pcbRead)
  561.     {
  562.         if(m_ReadBuffer.size() <= 0)
  563.             return FALSE;

  565.         m_ReadBuffer.pop_head(pv, cb, (int *)pcbRead);
  566.         return TRUE;
  567.     }

  569.     STDMETHOD(Read)(void *pv, ULONG cb, ULONG *pcbRead)
  570.     {
  571.         if(m_nFilterType & Filter_namespcae)
  572.             return S_OK;

  574. //        return m_UrlMonProtocol->Read(pv, cb, pcbRead);

  576.         BOOL        bRetVal;

  578.         bRetVal = RealFulltoBuffer();
  579.         if(bRetVal == FALSE)
  580.         {
  581.             *pcbRead = 0;
  582.             return E_PENDING;
  583.         }

  585.         bRetVal = ReadFromBuffer(pv, cb, pcbRead);
  586.         if(bRetVal == FALSE)
  587.             return S_FALSE;

  589.         return S_OK;
  590.     }

  592.     STDMETHOD(Seek)(LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition)
  593.     {
  594.         if(m_nFilterType & Filter_namespcae)
  595.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  597.         HRESULT            hr;
  598.         hr = m_UrlMonProtocol->Seek(dlibMove, dwOrigin, plibNewPosition);
  599.         return hr;
  600.     }
  601.     STDMETHOD(UnlockRequest)()
  602.     {
  603.         if(m_nFilterType & Filter_namespcae)
  604.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  606.         HRESULT            hr;
  607.         hr = m_UrlMonProtocol->UnlockRequest();
  608.         return hr;
  609.     }
  610.     STDMETHOD(LockRequest)(DWORD dwOptions)
  611.     {
  612.         if(m_nFilterType & Filter_namespcae)
  613.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  615.         HRESULT            hr;
  616.         hr = m_UrlMonProtocol->LockRequest(dwOptions);
  617.         return hr;
  618.     }

  620. //    IInternetProtocolSink
  621.     STDMETHOD (ReportData)( DWORD grfBSCF, ULONG ulProgress, ULONG ulProgressMax)
  622.     {
  623.         if(m_nFilterType & Filter_namespcae)
  624.             return INET_E_USE_DEFAULT_PROTOCOLHANDLER;

  626.         HRESULT            hr;
  627.         hr = m_UrlMonProtocolSink->ReportData(grfBSCF, ulProgress, ulProgressMax);
  628.         return hr;
  629.     }
  630.     STDMETHOD (ReportProgress)( ULONG ulStatusCode, LPCWSTR szStatusText)
  631.     {
  632.         HRESULT            hr;

  634.         switch(ulStatusCode)
  635.         {
  636.         case BINDSTATUS_DECODING:
  637.             FLT_TRACE("ReportProgress-BINDSTATUS_DECODING: %S/n", szStatusText);
  638.             break;
  639.         case BINDSTATUS_MIMETYPEAVAILABLE:
  640.             FLT_TRACE("ReportProgress-BINDSTATUS_MIMETYPEAVAILABLE: %S/n", szStatusText);
  641.             break;
  642.         case BINDSTATUS_CACHEFILENAMEAVAILABLE:
  643.             FLT_TRACE("ReportProgress-BINDSTATUS_CACHEFILENAMEAVAILABLE: %S/n", szStatusText);
  644.             m_sCacheFilePath = szStatusText;
  645.             break;
  646.         default:
  647.             FLT_TRACE("ReportProgress-Unknow: %S/n", szStatusText);
  648.         }
  649.         hr = m_UrlMonProtocolSink->ReportProgress(ulStatusCode, szStatusText);
  650.         return hr;
  651.     }
  652.     STDMETHOD (ReportResult)( HRESULT hrResult, DWORD dwError, LPCWSTR szResult)
  653.     {
  654.         HRESULT            hr;
  655.         hr = m_UrlMonProtocolSink->ReportResult(hrResult, dwError, szResult);
  656.         return hr;
  657.     }
  658.     STDMETHOD (Switch)( PROTOCOLDATA *pProtocolData)
  659.     {
  660.         HRESULT            hr;
  661.         hr = m_UrlMonProtocolSink->Switch(pProtocolData);
  662.         return hr;
  663.     }

  665.     STDMETHOD (SetPriority)(LONG nPriority)
  666.     {
  667.         return S_OK;
  668.     }

  670.     STDMETHOD (GetPriority)(LONG * pnPriority)
  671.     {
  672.         return S_OK;
  673.     }

  675. };


  678. ////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

阅读(1902) | 评论(0) | 转发(0) |
0

上一篇：MFC控制浏览器, UI, 动作...

下一篇：关于CHTMLView有内存泄漏的问题及解决办法

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6