SPI网络数据过滤程序

最新推荐文章于 2025-02-17 14:53:35 发布

原创最新推荐文章于 2025-02-17 14:53:35 发布 · 2.8k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#网络 #socket #null #string #stream #delete

C/C++ 专栏收录该内容

1 篇文章

订阅专栏

网络信息异常丰富，过滤一些数据对家里的小朋友还是很有必要的。

下面我就把C语言写的部分贴出来给有需要的朋友做个参考，我不是玩C语言的，写过的代码包括下面的也不超过2000行，下面的代码也很乱，写的时候急于求成，想到什么就写什么，过程内的代码一堆堆的，呵呵，好了不废话了，代码都有注释。

效果：

InstallYunerInfo安装

RemoveYunerInfo移除

必须在公共文档下面建立关键词文件，每个关键词后面必须有一个英文逗号，这就表示最后是一个逗号结尾。

#include "stdafx.h"#include <WS2spi.h>#include <SpOrder.h>#include <tchar.h>#include <string>#include <shlobj.h>#define ZLIB_INTERNAL#include "../zlib/zlib.h"#include "../zlib/zconf.h"#pragma comment(lib,"Ws2_32.lib")#pragma comment( lib, "shell32.lib")#pragma comment(lib,"..//zlib//zlibstat.lib")//使用C++标准库using namespace std;//定义GUID// {D0D86306-49BF-49F6-9EF7-3CF238829281}GUID spiGuid = { 0xd0d86306, 0x49bf, 0x49f6, { 0x9e, 0xf7, 0x3c, 0xf2, 0x38, 0x82, 0x92, 0x81 } };// {06A7CE70-F0FD-4500-9D43-D0A6F8A4E677}GUID spiChainGuid = { 0x6a7ce70, 0xf0fd, 0x4500, { 0x9d, 0x43, 0xd0, 0xa6, 0xf8, 0xa4, 0xe6, 0x77 } };//SPI变量定义LPWSAPROTOCOL_INFOW protoInfo=NULL;//协议信息结构DWORD protoInfoSize=0;//协议信息结构缓冲区大小WSPPROC_TABLE nextTable;//协议链中的下一个int totalProtos=0;//协议链总数//关键词临时文件路径char* fKeyTemp;//次数临时文件char* fNumTemp;//定义结构struct RECV_SR{SOCKET s;//套接字句柄string httpHead;//HTTP头string httpBody;//HTTP数据体string ContentEncoding;//压缩编码string TransferEncoding;//chunk编码int ContentLength;//Content-Length字段长度BOOL FindOK;//是否已经检查过数据BOOL FindReslut;//检测结果BOOL GetComplete;//接受是否完成};#define MAX_RLIST 256//保存每条socket接受的数据RECV_SR recvDatList[MAX_RLIST];//结构索引int indexRecv=-1;BOOL initY=FALSE;//302跳转//char* falseStr="HTTP/1.1 302 Found/r/nLocation: http://www.google.com//r/nCache-Control: private/r/nContent-Type: text/html; charset=UTF-8/r/nContent-Length: 0/r/n/r/n";//直接写入代码char* falseStr;string* aryKey;//敏感信息关键词int intKeys=0;//记录关键词个数//char* to wchar_t*wchar_t* CtoW(char* c){wchar_t* w;int lSize;lSize=MultiByteToWideChar(CP_ACP,0,c,-1,NULL,0);if(lSize==0){return NULL;}w=new wchar_t[lSize+1];w[lSize]='/0/0';lSize=MultiByteToWideChar(CP_ACP,0,c,-1,w,lSize);if(lSize==0){delete w;return NULL;}return w;}//wchar_t* to UTF8char* WtoUTF8(wchar_t* w){char* c;int lSize;lSize=WideCharToMultiByte(CP_UTF8,0,w,-1,NULL,0,NULL,NULL);if(lSize==0){return NULL;}c=new char[lSize+1];c[lSize]='/0';lSize=WideCharToMultiByte(CP_UTF8,0,w,-1,c,lSize,NULL,NULL);if(lSize==0){delete c;return NULL;}return c;}//wchar_t* to char*char* WtoC(wchar_t* w){char* c;int lSize;lSize=WideCharToMultiByte(CP_ACP,0,w,-1,NULL,0,NULL,NULL);if(lSize==0){return NULL;}c=new char[lSize+1];c[lSize]='/0';lSize=WideCharToMultiByte(CP_ACP,0,w,-1,c,lSize,NULL,NULL);if(lSize==0){delete c;return NULL;}return c;}//初始化//如果没有载入关键词必须返回FALSEBOOL Init(){TCHAR szDocument[MAX_PATH]={0};LPITEMIDLIST pidl=NULL;FILE *fp;char* lpC;char* cKey;TCHAR* tcKey;char ch;DWORD lSize=0;string strTemp;TCHAR tcTemp[MAX_PATH]={0};int nPos=0;int num=0;int nextPos=0;char* c;if(initY){return TRUE;}for(int i=0;i<MAX_RLIST;i++){recvDatList[i].s=0;recvDatList[i].httpHead.clear();recvDatList[i].httpBody.clear();recvDatList[i].ContentEncoding.clear();recvDatList[i].TransferEncoding.clear();recvDatList[i].ContentLength=0;recvDatList[i].FindOK=FALSE;recvDatList[i].FindReslut=FALSE;recvDatList[i].GetComplete=FALSE;}falseStr=WtoUTF8(CtoW("HTTP/1.1 200 OK/r/nContent-Length: 36/r/nContent-Type: text/html;charset=utf-8/r/nCache-Control: private/r/n/r/n内容被河蟹，等你的神经承受得起再来.."));//用户文档路径SHGetSpecialFolderLocation(NULL,CSIDL_COMMON_DOCUMENTS,&pidl);if(pidl && SHGetPathFromIDList(pidl,szDocument)){wcscat_s(tcTemp,szDocument);wcscat_s(tcTemp,L"//httpInfo//tpYkeys.txt");fKeyTemp=WtoC(tcTemp);memset(tcTemp,0,MAX_PATH);wcscat_s(tcTemp,szDocument);wcscat_s(tcTemp,L"//httpInfo//tpYnums.txt");fNumTemp=WtoC(tcTemp);}//读取数据if((fp=fopen(fKeyTemp,"r"))==NULL){return FALSE;}//长度while(true) {ch=fgetc(fp);if(ch==EOF || ch=='/n')break;lSize++;}//恢复文件指针rewind(fp);cKey=new char[lSize+1];cKey[lSize]='/0';//读取lpC=cKey;while(true){ch=fgetc(fp);if(ch==EOF || ch=='/n')break;*lpC=ch;lpC++;}fclose(fp);//转换tcKey=CtoW(cKey);//转换为窄字节c=WtoC(tcKey);//填充到字符串strTemp.assign(c,strlen(c));delete[] c;delete[] tcKey;delete[] cKey;//拆分nPos=strTemp.find(",",0);while(nPos>0){num++;nPos=strTemp.find(",",nPos+1);}if(num==0){return FALSE;}//关键词数组个数intKeys=num;//分配关键词数组aryKey=new string[intKeys];//赋值num=0;nPos=strTemp.find(",",0);aryKey[num]=strTemp.substr(0,//开始位置nPos//长度);while(nPos>0){num++;nextPos=strTemp.find(",",nPos+1);if(nextPos<0){break;}aryKey[num]=strTemp.substr(nPos+1,//开始位置nextPos-(nPos+1)//长度);nPos=nextPos;}if(num==0){return FALSE;}initY=TRUE;return TRUE;}//查找表中是否存在socket//存在返回下标，否则返回-1int FindSocket(SOCKET s){for(int i=0;i<MAX_RLIST;i++){if(recvDatList[i].s==s){return i;}}return -1;}//添加数据void AddRecv(SOCKET s,string &str,int findS){//存在if(findS>-1){indexRecv=findS;//追加数据recvDatList[indexRecv].httpBody.append(str);return;}//下一个索引indexRecv++;//到了表低端，则需要重新回到表顶端if(indexRecv==MAX_RLIST){indexRecv=0;}//重新赋值recvDatList[indexRecv].s=s;recvDatList[indexRecv].httpBody=str;}//清除一条socket数据void ClearSocket(int i){if(i<0||i>=MAX_RLIST){return;}recvDatList[i].s=0;recvDatList[i].httpHead.clear();recvDatList[i].httpBody.clear();recvDatList[i].ContentEncoding.clear();recvDatList[i].TransferEncoding.clear();recvDatList[i].ContentLength=0;recvDatList[i].FindOK=FALSE;recvDatList[i].FindReslut=FALSE;recvDatList[i].GetComplete=FALSE;}//从s字符串中返回s1到s2之间的字符string GetSubString(string &s,char* s1,char* s2){string str;int find1,find2;//查找s1find1=s.find(s1);if(find1<0){return str;}find1+=strlen(s1);//查找s2find2=s.find(s2,find1);if(find2<1){return str;}//提取str=s.substr(find1,//开始的位置find2-find1//N个字符);return str;}//数据是否接受完成BOOL HttpGetCompete(int i){int f;if(i<0||i>=MAX_RLIST){return FALSE;}if(recvDatList[i].GetComplete){return TRUE;}if(recvDatList[i].ContentLength>0&& recvDatList[i].ContentLength==recvDatList[i].httpBody.length()){recvDatList[i].GetComplete=TRUE;return TRUE;}//chunked模式if((f=recvDatList[i].TransferEncoding.find("chunked"))>=0){//chunk结束标记if((f=recvDatList[i].httpBody.rfind("/r/n0",recvDatList[i].httpBody.length()))>=0){recvDatList[i].GetComplete=TRUE;return TRUE;}}return FALSE;}//typedef std::basic_string<Byte, std::char_traits<Byte>, std::allocator<Byte> > ByteVector;int gzip_uncompress (const Bytef *source, uLong sourceLen, string &dst){z_stream stream;stream.next_in = (Bytef*)source;stream.avail_in = (uInt)sourceLen; stream.next_out = NULL;stream.avail_out = 0;stream.zalloc = (alloc_func)0;stream.zfree = (free_func)0; int err = inflateInit2(&stream, MAX_WBITS+32);if (err != Z_OK) return err;size_t nBufferSize = 4*1024*4;//! 16KBByte *pBuffer = NULL;try{pBuffer = new Byte[nBufferSize];}catch(std::bad_alloc &r){r;return Z_BUF_ERROR;}while(true){stream.next_out = pBuffer;stream.avail_out = nBufferSize;err = inflate(&stream, Z_SYNC_FLUSH);if(err == Z_OK && stream.avail_out == 0){dst.append((char*)pBuffer, nBufferSize);}else if (err == Z_STREAM_END){dst.append((char*)pBuffer, nBufferSize - stream.avail_out);break;}else{delete [] pBuffer;inflateEnd(&stream);if (err == Z_NEED_DICT || (err == Z_BUF_ERROR && stream.avail_in == 0))return Z_DATA_ERROR;return err;}}delete [] pBuffer; err = inflateEnd(&stream);return err;}//查找关键词BOOL FindKeys(string &b){int f;HWND hWnd;COPYDATASTRUCT cds;//查找关键词for(int j=0;j<intKeys;j++){if(aryKey[j].length()==0) continue;if((f=b.find(aryKey[j].c_str()))==-1) continue;//通知应用程序hWnd=FindWindow(L"ThttpInfoProtection",L"网络过滤");if(hWnd){f=1;cds.dwData=0;cds.cbData=sizeof(f);cds.lpData=&f;SendMessage(hWnd,WM_COPYDATA,0,(LPARAM)&cds);}return TRUE;}//forreturn FALSE;}//解压GzipBOOL GzipUn(string &inStr,string &outStr){if(gzip_uncompress((const BYTE*)inStr.c_str(),inStr.length(),outStr)==Z_OK){return TRUE;}return FALSE;}//检查是否存在敏感信息BOOL IsFiterData(int i){int f;string b;string ub;int chunkSize=0;char* str;int nPos=0;BOOL gzip=FALSE;//是否已经查找过了if(recvDatList[i].FindOK){return recvDatList[i].FindReslut;}recvDatList[i].FindOK=TRUE;//先查找HTTP头过滤网址if(FindKeys(recvDatList[i].httpHead)){recvDatList[i].FindReslut=TRUE;return TRUE;}gzip=((f=recvDatList[i].ContentEncoding.find("gzip"))>=0);//chunk模式if((f=recvDatList[i].TransferEncoding.find("chunked"))>=0){//将所有数据块解压并查找关键词while(TRUE){//头部结束标记位置nPos=recvDatList[i].httpBody.find("/r/n",chunkSize);//chunk头部大小标记ub=recvDatList[i].httpBody.substr(chunkSize,nPos-chunkSize);//转换为10进制chunkSize=strtol(ub.c_str(),&str,16);if((chunkSize==0)||(strlen(str)>0))break;nPos+=2;//提取数据b=recvDatList[i].httpBody.substr(nPos,chunkSize);//解压if(gzip){ub.clear();GzipUn(b,ub);}else{ub=b;}//是否存在敏感关键词if(FindKeys(ub)){b.clear();ub.clear();recvDatList[i].FindReslut=TRUE;return TRUE;}//头部开始位置chunkSize=chunkSize+nPos+2;}b.clear();ub.clear();return FALSE;}//chunk模式//解压if(gzip){GzipUn(recvDatList[i].httpBody,ub);}else{ub=recvDatList[i].httpBody;}//是否存在敏感关键词if(FindKeys(ub)){ub.clear();recvDatList[i].FindReslut=TRUE;return TRUE;}b.clear();ub.clear();return FALSE;}//接受函数int WSPAPI NewWSPRecv(SOCKET s,LPWSABUF lpBuffers,DWORD dwBufferCount,LPDWORD lpNumberOfBytesRecvd,LPDWORD lpFlags,LPWSAOVERLAPPED lpOverlapped,LPWSAOVERLAPPED_COMPLETION_ROUTINE lpCompletionRoutine,LPWSATHREADID lpThreadId,LPINT lpErrno){int result=0;string strTemp;DWORD bytesLen;DWORD lenStr;int findIndex;int findSocket;char *findChar;if(lpOverlapped){//OutputDebugString(_T("lpOverlapped not NULL"));//完成事件if(lpOverlapped->hEvent){//OutputDebugString(_T("lpOverlapped->hEvent"));}return nextTable.lpWSPRecv(s,lpBuffers,dwBufferCount,lpNumberOfBytesRecvd,lpFlags,lpOverlapped,lpCompletionRoutine,lpThreadId,lpErrno);}for(;;){//查找指向这条socket的链表下标findSocket=FindSocket(s);//如果数据接受完成if(HttpGetCompete(findSocket)){//存在敏感信息if(IsFiterData(findSocket)){if(dwBufferCount>0){lenStr=strlen(falseStr);//缓冲区长度足够//把自定义的跳转HTTP头部拷贝到缓冲区if(lpBuffers[0].len>=lenStr){//返回实际字节数*lpNumberOfBytesRecvd=lenStr;//一次性复制memcpy(lpBuffers[0].buf,falseStr,lenStr);//清除ClearSocket(findSocket);}else//缓冲区长度不够{//返回错误*lpNumberOfBytesRecvd=0;result=SOCKET_ERROR;//清除ClearSocket(findSocket);}}break;}//存在敏感信息，复制假数据//不存在敏感信息，把真实数据返回if(dwBufferCount>0){//复制头部if(recvDatList[findSocket].httpHead.length()>0){//缓冲区长度足够if(lpBuffers[0].len>=recvDatList[findSocket].httpHead.length()){//返回实际字节数*lpNumberOfBytesRecvd=recvDatList[findSocket].httpHead.length();//一次性复制recvDatList[findSocket].httpHead.copy(lpBuffers[0].buf,//目标缓冲区recvDatList[findSocket].httpHead.length(),//长度0//开始位置);//清空头部recvDatList[findSocket].httpHead.clear();}else//缓冲区长度不足够{//返回实际字节数*lpNumberOfBytesRecvd=lpBuffers[0].len;//将缓冲区复制满recvDatList[findSocket].httpHead.copy(lpBuffers[0].buf,//目标缓冲区lpBuffers[0].len,//长度0//开始位置);//删除已复制的数据recvDatList[findSocket].httpHead=recvDatList[findSocket].httpHead.substr(lpBuffers[0].len,recvDatList[findSocket].httpHead.length()-lpBuffers[0].len);}break;}//复制头部数据//复制主体数据if(recvDatList[findSocket].httpBody.length()>0){//缓冲区长度足够if(lpBuffers[0].len>=recvDatList[findSocket].httpBody.length()){//返回实际字节数*lpNumberOfBytesRecvd=recvDatList[findSocket].httpBody.length();//一次性复制recvDatList[findSocket].httpBody.copy(lpBuffers[0].buf,recvDatList[findSocket].httpBody.length(),0);//清除ClearSocket(findSocket);}else//缓冲区长度不足够{//返回实际字节数*lpNumberOfBytesRecvd=lpBuffers[0].len;//把缓冲区复制满recvDatList[findSocket].httpBody.copy(lpBuffers[0].buf,lpBuffers[0].len,0);//删除已复制的数据recvDatList[findSocket].httpBody=recvDatList[findSocket].httpBody.substr(lpBuffers[0].len,recvDatList[findSocket].httpBody.length()-lpBuffers[0].len);}break;}//复制主体数据}//把全部真实数据返回break;}//如果数据接受完成//接受数据result = nextTable.lpWSPRecv(s,lpBuffers,dwBufferCount,lpNumberOfBytesRecvd,lpFlags,lpOverlapped,lpCompletionRoutine,lpThreadId,lpErrno);bytesLen=*lpNumberOfBytesRecvd;//错误if((result!=0) || (bytesLen<1) || (dwBufferCount<1)){break;}//处理收到的数据for(DWORD i =0;i<dwBufferCount;i++){//判断是否为HTTP数据if(strncmp("HTTP/1.1 200 OK",lpBuffers[i].buf,strlen("HTTP/1.1 200 OK"))==0){//清除已经存在的ClearSocket(findSocket);}else{//不是HTTP头部包，并且也没有找到socket存在//说明是无关数据if(findSocket==-1){return result;}}//得到字符串形式的数据strTemp.assign(lpBuffers[i].buf,bytesLen);//添加数据AddRecv(s,strTemp,findSocket);strTemp.clear();//提取HTTP头部信息if((recvDatList[indexRecv].ContentLength==0)&&(recvDatList[indexRecv].TransferEncoding.length()==0)){findChar="/r/n/r/n";findIndex=recvDatList[indexRecv].httpBody.find(findChar);//HTTP头部还没有接受完成if(findIndex<0){continue;}//把HTTP头部提取出来strTemp=recvDatList[indexRecv].httpBody.substr(0,findIndex + strlen(findChar));//查找 Content-Type://判断文件类型，如果不为文本类型if((findIndex=GetSubString(strTemp,"Content-Type:","/r/n").find(" text/"))!=0){ClearSocket(indexRecv);return result;}//查找 Content-Length://转换为数字recvDatList[indexRecv].ContentLength=atoi(GetSubString(strTemp,"Content-Length:","/r/n").c_str());//不正确的长度if(recvDatList[indexRecv].ContentLength<1){//chunkedrecvDatList[indexRecv].TransferEncoding=GetSubString(strTemp,"Transfer-Encoding:","/r/n");if(recvDatList[indexRecv].TransferEncoding.length()==0){ClearSocket(indexRecv);return result;}}//查找 Content-Encoding:recvDatList[indexRecv].ContentEncoding=GetSubString(strTemp,"Content-Encoding:","/r/n");//头部recvDatList[indexRecv].httpHead=strTemp;//HTTP主体recvDatList[indexRecv].httpBody=recvDatList[indexRecv].httpBody.substr(strTemp.length(),recvDatList[indexRecv].httpBody.length() - strTemp.length());}//提取HTTP头部信息}//循环接受缓冲区数据}//主循环return result;}//关闭连接int WSPAPI NewWSPCloseSocket(SOCKET s,LPINT lpErrno){//释放资源处理ClearSocket(FindSocket(s));return nextTable.lpWSPCloseSocket(s,lpErrno);}//枚举所有协议链BOOL EmProto(){int erCode=0;protoInfo=NULL;protoInfoSize=0;totalProtos=0;//获取协议链缓冲大小if(WSCEnumProtocols(NULL,protoInfo,&protoInfoSize,&erCode)==SOCKET_ERROR){if(erCode!=WSAENOBUFS){return FALSE;}}//分配内存空间if((protoInfo=(LPWSAPROTOCOL_INFOW)GlobalAlloc(GPTR,protoInfoSize))==NULL){return FALSE;}//枚举协议链if((totalProtos=WSCEnumProtocols(NULL,protoInfo,&protoInfoSize,&erCode))==SOCKET_ERROR){return FALSE;}return TRUE;}//释放内存void Free(){GlobalFree(protoInfo);}//SPI 接口函数int WSPAPI NewWSPStartup(WORD wVersionRequested,LPWSPDATA lpWSPData,LPWSAPROTOCOL_INFOW lpProtocolInfo,WSPUPCALLTABLE UpcallTable,LPWSPPROC_TABLE lpProcTable){int i=0;int erCode=0;DWORD layerId=0;DWORD nextLayerId=0;TCHAR* nextSpiPath;int nextSpiPathLen=MAX_PATH;HINSTANCE hDll=NULL;LPWSPSTARTUP fWSPStartup=NULL;//初始化数据if(!Init()){return FALSE;}if(lpProtocolInfo->ProtocolChain.ChainLen<=1){return FALSE;}if(!EmProto()){return FALSE;}//找到我们的协议for(i=0;i<totalProtos;i++){if(memcmp(&protoInfo[i].ProviderId,&spiGuid,sizeof(GUID))==0){layerId=protoInfo[i].dwCatalogEntryId;break;}}//协议链中下一个Chainfor(i=0;i<lpProtocolInfo->ProtocolChain.ChainLen;i++){if(lpProtocolInfo->ProtocolChain.ChainEntries[i]==layerId){nextLayerId=lpProtocolInfo->ProtocolChain.ChainEntries[i+1];break;}}//下一个协议的文件路径nextSpiPath=(TCHAR*)GlobalAlloc(GPTR,nextSpiPathLen);for(i=0;i<totalProtos;i++){if(nextLayerId==protoInfo[i].dwCatalogEntryId){if(WSCGetProviderPath(&protoInfo[i].ProviderId,nextSpiPath,&nextSpiPathLen,&erCode)==SOCKET_ERROR){Free();return WSAEPROVIDERFAILEDINIT;}break;}}//扩展环境变量if(!ExpandEnvironmentStrings(nextSpiPath,nextSpiPath,nextSpiPathLen)){Free();return WSAEPROVIDERFAILEDINIT;}//加载协议文件if((hDll=LoadLibrary(nextSpiPath))==NULL){Free();return WSAEPROVIDERFAILEDINIT;}//获取 WSPStartup 函数地址if((fWSPStartup=(LPWSPSTARTUP)GetProcAddress(hDll,"WSPStartup"))==NULL){Free();return WSAEPROVIDERFAILEDINIT;}//调用下一个协议链的 WSPStartup 函数if((erCode=fWSPStartup(wVersionRequested,lpWSPData,lpProtocolInfo,UpcallTable,lpProcTable))!=ERROR_SUCCESS){Free();return erCode;}nextTable=*lpProcTable;//处理函数lpProcTable->lpWSPRecv=NewWSPRecv;lpProcTable->lpWSPCloseSocket=NewWSPCloseSocket;Free();return 0;}//是否已安装BOOL GetInstall(){if(!EmProto()){return FALSE;}//查找安装的协议for(int i=0;i<totalProtos;i++){if(memcmp(&protoInfo[i].ProviderId,&spiGuid,sizeof(GUID))==0){Free();return TRUE;//已安装}}Free();return FALSE;}//移除协议BOOL RemoveYunerInfo(){int erCode;BOOL compFlags=TRUE;if(WSCDeinstallProvider(&spiChainGuid,&erCode)==SOCKET_ERROR){compFlags=FALSE;}if(WSCDeinstallProvider(&spiGuid,&erCode)==SOCKET_ERROR){compFlags=FALSE;}if(compFlags){return TRUE;}return FALSE;}//安装协议BOOL InstallYunerInfo(){int i;int provcnt;int cataindex;int errorcode;BOOL rawip=FALSE;BOOL tcpip=FALSE;DWORD iplayercataid=0,tcpOrigcataid=0;TCHAR chainname[WSAPROTOCOL_LEN+1];LPDWORD cataentries;WSAPROTOCOL_INFOW iplayerinfo,tcpChainInfo,chainarray[1];if(GetInstall()){return FALSE;}if(!EmProto()){return FALSE;}for(i=0;i<totalProtos;i++){if(!rawip&& protoInfo[i].iAddressFamily==AF_INET//Internet协议&& protoInfo[i].iProtocol==IPPROTO_IP)//IP协议{rawip=TRUE;memcpy(&iplayerinfo,&protoInfo[i],sizeof(WSAPROTOCOL_INFOW));iplayerinfo.dwServiceFlags1=protoInfo[i].dwServiceFlags1 & (~XP1_IFS_HANDLES);}if(!tcpip&& protoInfo[i].iAddressFamily==AF_INET //Internet协议&& protoInfo[i].iProtocol==IPPROTO_TCP) //TCP协议{tcpip=TRUE;tcpOrigcataid=protoInfo[i].dwCatalogEntryId;memcpy(&tcpChainInfo,&protoInfo[i],sizeof(WSAPROTOCOL_INFOW));tcpChainInfo.dwServiceFlags1=protoInfo[i].dwServiceFlags1 & (~XP1_IFS_HANDLES);}}wcscpy_s(iplayerinfo.szProtocol,L"httpInfo");iplayerinfo.ProtocolChain.ChainLen=LAYERED_PROTOCOL;//安装协议if(WSCInstallProvider(&spiGuid,myDllPath,&iplayerinfo,1,&errorcode)==SOCKET_ERROR){Free();return FALSE;}Free();if(!EmProto()){WSCDeinstallProvider(&spiGuid,&errorcode);return FALSE;}//查找安装的协议for(i=0;i<totalProtos;i++){if(memcmp(&protoInfo[i].ProviderId,&spiGuid,sizeof(GUID))==0){iplayercataid=protoInfo[i].dwCatalogEntryId;break;}}provcnt=0;if(tcpip){wcscpy_s(chainname,L"httpInfoCh");wcscpy_s(tcpChainInfo.szProtocol,chainname);if(tcpChainInfo.ProtocolChain.ChainLen==BASE_PROTOCOL){tcpChainInfo.ProtocolChain.ChainEntries[1]=tcpOrigcataid;}else{for(i=tcpChainInfo.ProtocolChain.ChainLen;i>0;i--){tcpChainInfo.ProtocolChain.ChainEntries[i+1]=tcpChainInfo.ProtocolChain.ChainEntries[i];}}tcpChainInfo.ProtocolChain.ChainLen++;tcpChainInfo.ProtocolChain.ChainEntries[0]=iplayercataid;memcpy(&chainarray[provcnt++],&tcpChainInfo,sizeof(WSAPROTOCOL_INFOW));}//安装Chainif(WSCInstallProvider(&spiChainGuid,myDllPath,chainarray,provcnt,&errorcode)==SOCKET_ERROR){WSCDeinstallProvider(&spiGuid,&errorcode);Free();return FALSE;}Free();if(!EmProto()){RemoveYunerInfo();return FALSE;}if((cataentries=(LPDWORD)GlobalAlloc(GPTR,totalProtos*sizeof(WSAPROTOCOL_INFOW)))==NULL){Free();RemoveYunerInfo();return FALSE;}cataindex=0;for(i=0;i<totalProtos;i++){if(memcmp(&protoInfo[i].ProviderId,&spiGuid,sizeof(GUID))==0|| memcmp(&protoInfo[i].ProviderId,&spiChainGuid,sizeof(GUID))==0){cataentries[cataindex++]=protoInfo[i].dwCatalogEntryId;}}for(i=0;i<totalProtos;i++){if(memcmp(&protoInfo[i].ProviderId,&spiGuid,sizeof(GUID))!=0&& memcmp(&protoInfo[i].ProviderId,&spiChainGuid,sizeof(GUID))!=0){cataentries[cataindex++]=protoInfo[i].dwCatalogEntryId;}}//排序if((errorcode==WSCWriteProviderOrder(cataentries,totalProtos))!=ERROR_SUCCESS){Free();RemoveYunerInfo();return FALSE;}Free();return TRUE;}

DLL入口

// dllmain.cpp : 定义 DLL 应用程序的入口点。#include "stdafx.h"BOOL APIENTRY DllMain( HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved ){switch (ul_reason_for_call){case DLL_PROCESS_ATTACH://自己的完整路径GetModuleFileName(hModule,myDllPath,MAX_PATH);case DLL_THREAD_ATTACH:case DLL_THREAD_DETACH:case DLL_PROCESS_DETACH:break;}return TRUE;}

导出.def文件

LIBRARYEXPORTSWSPStartup = NewWSPStartupInstallYunerInfoRemoveYunerInfoGetInstall