打造自己的HTTP代理服务器

韩冰

/*========================================================================= FILE:HTTPPROXY.cpp HTTP Proxy v1.0 powered by shadow 2004/11/14 2 ^) `* K( I9 X# m6 Omy web:http://www.codehome.6600.org 3 n+ T/ E% f2 w$ PQQ:176017352 0 y. K9 \7 f+ W0 h请先阅读有关http代理协议的相关资料,转载请著明出处 ==========================================================================*/ : w: v# A% y# {& m+ e

#include "stdafx.h" #include "HTTPPROXY.h" #define FD_NOEVENT 0 // & z v, S0 b5 c u1 |' g#define BACKUP 100 #define MAXDATALEN 65535 # I9 K* f2 O }$ t- z3 ^$ }9 q3 F1 Fint HttpListenPort; : A/ q/ p/ F" t4 wSOCKET HttpListenSock; // $ M- z h; E: S9 E% y% Ntypedef struct _CLIENTINFO{ SOCKET clientsock; 1 {- P$ O& H' @ o: ~2 G# }4 fSOCKET udpsock; sockaddr_in clientsock_addr; }CLIENTINFO,*LPCLIENTINFO; 3 [! `. S$ R O6 v5 _" n" e% @CLIENTINFO HttpClientInfo; o/ _+ u H5 S// ( X3 L7 L" P+ m( O3 l Ftypedef struct _SOCKINFO{ * A0 M# Q* Z7 t4 n; C% PSOCKET sourcesock; SOCKET destsock; }SOCKINFO,*LPSOCKINFO; SOCKINFO httpsockinfo; // extern long GetSocketEventId(SOCKET remotesock); extern unsigned long GetLocalIp(); extern unsigned long GetDomainIp(char domainname[250]); /* ( M5 ]: u; {- u( {7 E- v4 Nlong GetSocketEventId(SOCKET remotesock){ long EventId; + T4 N' @7 M1 OHANDLE hevent; ' x6 o- B# F; r" e% U8 Y Thevent=CreateEvent(NULL,0,0,0); G; k! \; i" P) F+ B5 Z7 PWSANETWORKEVENTS socket_events; " H6 |" v0 v* J* d# L0 i4 xEventId=FD_NOEVENT; if(WSAEventSelect(remotesock,hevent,FD_ACCEPT|FD_CONNECT|FD_READ|FD_WRITE|FD_CLOSE)==SOCKET_ERROR) return EventId; WSAEnumNetworkEvents(remotesock,hevent,&socket_events); & R' _- P( i% K% A* \5 d: t( i& sif(socket_events.lNetworkEvents!=0){ switch(socket_events.lNetworkEvents){ case FD_ACCEPT:EventId=FD_ACCEPT;break; 6 E7 ^! g0 |/ `+ L4 ?2 L$ bcase FD_CONNECT:EventId=FD_CONNECT;break; case FD_READ:EventId=FD_READ;break; case FD_WRITE:EventId=FD_WRITE;break; & U7 t+ f' g2 a% \1 C! bcase FD_CLOSE:EventId=FD_CLOSE;break; 3 K4 a, d) ^, ^+ R! P( Scase FD_OOB:EventId=FD_OOB;break; 7 R- T y$ L4 n, Xdefault:EventId=FD_NOEVENT;break; " w* i+ X. K, P* b# W} } ! d5 D$ m+ {9 n+ S4 Z8 Selse EventId=FD_NOEVENT; - s% W) Z7 l* a2 @' Breturn EventId; } % k5 F$ { F' B9 q// / D2 q9 e3 q( munsigned long GetLocalIp() { ( k8 {: x. C3 S3 x# ~char IP[MAX_PATH],*ip; char pc_name[80]; struct in_addr in; struct hostent *host; WORD wVersionRequested; % ^; b" ~' O c l' Y$ w+ UWSADATA wsaData; wVersionRequested=MAKEWORD(2,0); ( D6 J! o2 _4 Z: Gip=IP; strcpy(ip,"Ip not get!"); * C w# Q: j4 B3 zif(WSAStartup(wVersionRequested,&wsaData)) return 0; if(gethostname(pc_name,80)==SOCKET_ERROR){ WSACleanup(); ; i' {( q5 s# ?# Z1 g4 H( C+ k$ J* `' ireturn 0; , R2 R9 n- V$ t/ e6 [2 v} 4 F+ o$ z! N4 J' b3 U% Bif(!(host=gethostbyname(pc_name))){ WSACleanup(); return 0; } 9 x0 ]6 E+ c. K0 p( qin.s_addr=*((unsigned long *)host->h_addr_list[0]); , g6 E( N+ z2 Z& P* }& v4 Istrcpy(ip,inet_ntoa(in)); 8 s& y ]( f4 U/ X3 bWSACleanup(); $ L/ Y! q! d. }! f4 vreturn in.s_addr; } // unsigned long GetDomainIp(char domainname[250]) { : a* K2 m4 O* ~7 k/ Cchar IP[MAX_PATH],*ip; 5 ~! o, O9 E5 V* P$ T% ]. g/ f& p N1 N/ `struct in_addr in; struct hostent *host; * r9 h2 q: B6 N0 d( IWORD wVersionRequested; 6 y3 ~ o% r: N6 oWSADATA wsaData; wVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; $ T6 e! O. {5 q! Dif(!(host=gethostbyname(domainname))){ WSACleanup(); / @; K6 d2 J& n* n4 Y& Hreturn 0; } - k( u( P9 Y0 \" I7 Q4 Uin.s_addr=*((unsigned long *)host->h_addr_list[0]); ) a1 [: D% V& r. istrcpy(ip,inet_ntoa(in)); : z8 `; y; n$ jWSACleanup(); return in.s_addr; } 0 A" C B) }. R& C+ s" c*/ // ; |' m9 c) _+ `8 W// " Y5 K7 f0 r3 yUINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 LPSOCKINFO psockinfo; $ A$ C2 V3 O& P/ @$ {SOCKET sourcesock,destsock; " [; h% J+ u/ p( Zchar data[MAXDATALEN]; long eventid; int datalen; 5 W+ N7 s8 v8 s* z$ t/ [' i$ j; Vpsockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; 7 b0 {/ j: C# i1 C& W4 g1 S$ Sdestsock=psockinfo->destsock; 6 E" x/ l# v# y; @TRACE("deail recive thread ok!\r\n"); while(true){ : ~. w* |# C( W; C/ Deventid=GetSocketEventId(sourcesock); switch(eventid){ 2 M4 B& P2 S$ ?- @case FD_CLOSE: 2 i$ G; P p* I/ ]+ OTRACE("s fdclosed\r\n"); closesocket(destsock); , o' m: p: L0 S7 W/ W+ i& ~% J; g2 \return 1; break; % ]0 V# X; l* Adefault:break; ' b& B+ m3 V4 p+ ?( s5 B} eventid=GetSocketEventId(destsock); switch(eventid){ % s! J; t$ s9 u% ~case FD_CLOSE: ) x% L+ ^# Z6 p3 i! A3 Iclosesocket(sourcesock); 8 H" e, d5 f% Q! [2 R# qTRACE("d fdclosed\r\n"); 9 h' e+ T2 G& i+ Y/ ]return 1; break; ( s1 t( ]% Y, T- q8 fdefault:break; } datalen=recv(sourcesock,data,sizeof(data),0); " E3 F, U, ~2 U+ \if(datalen==0){ closesocket(sourcesock); closesocket(destsock); ! W3 K) J; h1 N& T! NTRACE("s fdclosed\r\n"); 1 ^4 f& L. X$ J+ @9 i) G* C- Dbreak; } 6 F3 U# L4 t. q& w, Gif(datalen>0){ 6 y7 V9 Q! z7 G6 Z D& Hwhile(!send(destsock,data,datalen,0)); " C+ @: e, [$ o' C# ~& Y} Sleep(1); 9 i0 L. M3 v/ p8 X$ V} return 1; } // 3 e' _1 i" P: k8 GUINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 LPSOCKINFO psockinfo; # h. Q7 d" k- p# f2 ]" aSOCKET sourcesock,destsock; char data[MAXDATALEN]; long eventid; 6 o7 @. E5 M2 v1 wint datalen; % w; @) P. t2 y1 s+ _2 y- jpsockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; 7 k$ e/ P1 z3 Y1 G5 t# ^$ qdestsock=psockinfo->destsock; 3 Z# J4 G4 \2 c; o: sTRACE("deail send thread ok!\r\n"); while(true){ eventid=GetSocketEventId(sourcesock); ; B# ~8 j* g7 j! d5 k5 B- Uswitch(eventid){ case FD_CLOSE: TRACE("s fdclosed\r\n"); closesocket(destsock); ; ~& S6 Y% C' }0 P5 c; Areturn 1; / D5 r' Z/ O% ?0 h3 T! ]/ C/ v! vbreak; default:break; } 1 f8 j2 K6 N" O1 f$ V Ceventid=GetSocketEventId(destsock); switch(eventid){ case FD_CLOSE: 4 e2 R1 N- S5 Wclosesocket(sourcesock); I! R, L. L# O& tTRACE("d fdclosed\r\n"); return 1; & H: C( |5 K6 }! Dbreak; default:break; } datalen=recv(destsock,data,sizeof(data),0); if(datalen==0){ closesocket(sourcesock); 8 L& H- C+ t( v: P9 _closesocket(destsock); 6 v; v9 {8 N0 v* y& p0 S1 |TRACE("d fdclosed\r\n"); 7 _+ m* Y- q; Nbreak; % _0 f! M& ~. s* d2 [7 F; x} if(datalen>0){ while(!send(sourcesock,data,datalen,0)); } + l9 U1 V0 l8 A6 _8 N+ k% ASleep(1); 1 k3 w* e# g$ H' t& R} return 1; , s) `4 S. }* [* R1 E} // // 0 p; a+ {. ]' R; FUINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 LPCLIENTINFO pclientinfo; SOCKET connectsock,clientsock; & q( I6 ~% W3 m' u; F* M/ h' Isockaddr_in remotesock_addr; char data[MAXDATALEN],url[250],temp[250],httpurl[250],portnum[10]; . e& n3 }* }& f; Z% Wint datalen,i,index_start,index_end,port; CString HttpString,UrlString,PortString; pclientinfo=(LPCLIENTINFO)info; $ C' f; X' g6 Wclientsock=pclientinfo->clientsock; ZeroMemory((void *)data,sizeof(data)); : R2 }4 t& A/ ?) O% G0 S6 l6 Qdatalen=recv(clientsock,data,sizeof(data),0); if(datalen<=0){ closesocket(clientsock); , ? Y% u7 G/ J- ]5 sreturn 0;

韩冰

} {( C/ k. e6 [- P' e( @3 _1 H- Y- BHttpString.Format("%s",data); 8 p5 b; @- g9 t% dUrlString=HttpString; TRACE("get http string:\r\n"); $ [/ ?0 D: F- j# wTRACE(HttpString); & t! o X3 L( n/ b3 Z6 U. Windex_start=HttpString.Find("Host: ",0); //寻找url标记 if(index_start<=0){ closesocket(clientsock); return 0; 6 q/ ~. K+ @% l' A} ; U# q+ k1 @/ r% o' T! a5 nindex_end=HttpString.Find("\r\n",index_start); if(index_end<=0){ closesocket(clientsock); return 0; 0 k) y2 z6 d' H5 q% G} UrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 ' v1 d! S. @" Z7 f1 b$ Y( d8 TTRACE("\r\n get url:"); TRACE(UrlString); wsprintf(url,"%s",UrlString); strcpy(temp,url); strcat(temp,":"); datalen=strlen(temp); if(HttpString.Find("GET",0)==0){ //判断get命令，并处理 4 y: v8 D0 ~6 T4 A( P, h O2 p. cindex_start=HttpString.Find(temp,0); strcpy(httpurl,"http://"); 2 e* F" Q3 d6 ~! y3 x, z/ X- p. Oif(index_start>0){ index_end=HttpString.Find("/",index_start); 3 e2 R, {( a% y* Vif(index_end<=0){ closesocket(clientsock); return 0; . ^( E8 M2 d! z( G7 C3 r( m" z} % G& Q' `8 ~* o) SPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); ' F) q' s1 `* G7 ^( kport=atoi(PortString); strcat(httpurl,temp); & U4 B2 N" S1 F' J9 c9 Hitoa(port,portnum,sizeof(portnum)); 4 w8 Z0 o' s$ q$ y. ~0 h1 x2 lstrcat(httpurl,portnum); # N# _) C( d! rstrcat(httpurl,"/"); 4 L7 h I. q3 I% D- j} else{ port=80; strcat(httpurl,url); strcat(httpurl,"/"); 7 _0 h6 j ]6 p- L8 A8 @' t! D} 8 O; }; [" v0 ^* K9 z7 gTRACE("get http url:%s\r\n",httpurl); HttpString.Replace(httpurl,"/"); HttpString.Replace("Proxy-",""); HttpString.Replace("HTTP/1.0","HTTP/1.1"); } else if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 index_start=HttpString.Find(temp,0); if(index_start>0){ index_end=HttpString.Find(" ",index_start); 3 a) o' {( ~# z, [% lif(index_end<=0){ closesocket(clientsock); return 0; } PortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); port=atoi(PortString); } 0 Q* J- i! f1 L! ?else{ closesocket(clientsock); 2 g$ a$ r% k+ i* F( Ereturn 0; } } TRACE("get new http string:\r\n"); , y5 l, H) a5 Q. v- g5 e# I4 qTRACE(HttpString); remotesock_addr.sin_family=AF_INET; remotesock_addr.sin_port=htons(port); remotesock_addr.sin_addr.S_un.S_addr=GetDomainIp(url); connectsock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); if(connect(connectsock,(const sockaddr *)&remotesock_addr,sizeof(remotesock_addr))==INVALID_SOCKET){ //连接远程主机 1 P, W/ {! _( w {. wclosesocket(clientsock); 0 `7 E$ f: L/ p; freturn 0; } TRACE("\r\nconnect to remote ip ok\r\n"); ZeroMemory((void *)data,sizeof(data)); wsprintf(data,"%s",HttpString); 0 f) r5 F: E' @% Q6 G: x+ A0 Udatalen=strlen(data); 0 x% `" x. E* B" R6 s, F) L4 Mif(HttpString.Find("CONNECT",0)<0) while(!send(connectsock,data,datalen,0)); 5 Z0 O% S( T; i# b3 Zelse{ strcpy(data,"HTTP/1.0 200 Connection established\r\nProxy-agent: CHTTPPROXY V1.0 powered by shadow\r\n\r\n"); 8 Z6 @3 T5 \# H& ~0 s2 U' u$ Qdatalen=strlen(data); while(!send(clientsock,data,datalen,0)); 6 I6 y& s% K g1 N) b9 c9 Q) H} httpsockinfo.sourcesock=clientsock; 9 e, T; ~% y$ N' e6 A( _httpsockinfo.destsock=connectsock; & C! |! c4 J6 jAfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 AfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // Sleep(100); return 1; } // UINT StartHttpProxy(LPVOID info){ //端口监听线程 2 R- J. g7 k+ h7 J" }SOCKET NewSock; int socklen; % F4 ?% g4 o2 B( O- u: wsockaddr_in serversock,remotesock_addr; serversock.sin_family=AF_INET; , y7 Z; n3 l6 L! E: Sserversock.sin_addr.S_un.S_addr=INADDR_ANY; serversock.sin_port=htons(HttpListenPort); HttpListenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); . h/ H* w5 }( }if(HttpListenSock==INVALID_SOCKET) return 0; $ l1 t# ]% m3 x2 T/ C0 _, qif(bind(HttpListenSock,(const sockaddr *)&serversock,sizeof(serversock))==SOCKET_ERROR) return 0; $ S" E& W K. ?" P% S; F( H$ N: [listen(HttpListenSock,BACKUP); : s3 z, [, O. ?2 G& f9 Zsocklen=sizeof(remotesock_addr); TRACE("start http proxy thread while\r\n"); while(true){ NewSock=accept(HttpListenSock,(sockaddr *)&remotesock_addr,&socklen); ! j! ?- q( j- m; Z0 \& G8 gTRACE("waitting ok...\r\n"); 8 s, U7 Q$ K) sif(NewSock==INVALID_SOCKET){ Sleep(1); ( O; ^8 I f; G5 Lcontinue; } 9 b5 f! Q; W, Y* \/ MZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); HttpClientInfo.clientsock=NewSock; HttpClientInfo.clientsock_addr=remotesock_addr; TRACE("start proxy thread\r\n"); , p6 R7 e! Q8 M j4 n: Y& mAfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); 4 E$ P( H9 b, l3 H/ M6 n7 _! ~4 XSleep(100); } ; ^4 `6 \7 q; @: j6 d& m p! A# }return 1; 4 o4 Q6 a4 Q% S6 l: q' R} // 0 ~/ A- U; h) N' G! x/ ?' kCHTTPPROXY::CHTTPPROXY() { WSADATA WsaData; 4 N3 f/ G+ Q& ]7 WWORD wsaVer; wsaVer=MAKEWORD(2,0); WsaStartupOk=false; if(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; 0 e$ |1 C0 j6 Q4 q9 _}

CHTTPPROXY::~CHTTPPROXY() { 1 `2 d# @7 U. b6 K9 R9 v$ ~0 Aif(WsaStartupOk){ WSACleanup(); } }

int CHTTPPROXY::StartProxy(int listenport) { HttpListenPort=listenport; * G, i4 s# K; F# @. q5 h+ T. `AfxBeginThread(StartHttpProxy,(LPVOID)NULL); return 1; }

2 Y9 ^" c$ b6 ~% J0 {- P9 B/*========================================================================= FILE:HTTPPROXY.h 2 y+ p; f( Q' ~: |6 ^* S==========================================================================*/ class CHTTPPROXY { public: int StartProxy(int listenport); / }7 d, F) a4 X& \bool WsaStartupOk; CHTTPPROXY(); virtual ~CHTTPPROXY();

};

/*========================================================================= FILE:stdafx.h ==========================================================================*/ 5 J4 o7 g% g( _5 J; i2 {% Q#include % Y# G. \! z2 q7 V#include 4 a5 f$ F( _9 J9 [#include 1 j* _9 ]. C& j4 Z Q: O7 e r% |#include

注:不要忘了在link选项中添加wsock32.lib和ws2_32.lib，或在文件前部加上如下语句： #paragma comment(lib,"wsock32.lib") ; W# X9 e" @. F- A t, o#paragma comment(lib,"ws2_32.lib")

本代码在win2k和vc6.0下编译成功~~ 用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: CHTTPPROXY httpproxy; httpproxy.StartProxy(7890); ( N% Z" X; m4 `0 |3 V; z# o有问题mailto me！ - h6 v. U( R! }1 t( G- d