打造自己的HTTP代理服务器

韩冰

/*========================================================================= : R% R, U5 \% r+ R. @! `. }FILE:HTTPPROXY.cpp HTTP Proxy v1.0 * ]6 [8 }. V( R! ?1 lpowered by shadow 2004/11/14 my web:http://www.codehome.6600.org QQ:176017352 请先阅读有关http代理协议的相关资料,转载请著明出处 4 ]4 j# d' [ T# @- n/ ?+ N/ H" P" O==========================================================================*/

#include "stdafx.h" 6 q$ b' y8 M* t$ ^7 A: `#include "HTTPPROXY.h" ! l; n; z1 F( e8 Q% H) v" |4 F#define FD_NOEVENT 0 8 Z7 O$ U4 |' @; F// 9 Z, P2 l) n0 Q: \8 X#define BACKUP 100 #define MAXDATALEN 65535 int HttpListenPort; 9 \& L( s; K D% ?SOCKET HttpListenSock; // typedef struct _CLIENTINFO{ SOCKET clientsock; SOCKET udpsock; 3 T* @$ v" \* msockaddr_in clientsock_addr; ! N' |2 N3 P, k3 q' c( N$ m}CLIENTINFO,*LPCLIENTINFO; CLIENTINFO HttpClientInfo; : ?6 `7 N4 d% D2 B; a1 i5 V+ a// 3 p& J& v; t6 \$ H# o2 v. Wtypedef struct _SOCKINFO{ SOCKET sourcesock; # i l# k- Y2 t6 h2 G* ySOCKET destsock; }SOCKINFO,*LPSOCKINFO; SOCKINFO httpsockinfo; // : ^% E, A( x8 K2 |9 m+ l: Nextern long GetSocketEventId(SOCKET remotesock); % D! G% t* {% ^& X3 V2 ?- ^extern unsigned long GetLocalIp(); 3 n* M* K: i+ }) yextern unsigned long GetDomainIp(char domainname[250]); # t; R# |& d- ~# Y: f# w" ?/* 1 O! H/ R) }* w' `9 N9 P- {long GetSocketEventId(SOCKET remotesock){ long EventId; . }+ I# Y7 w' j1 k( t/ g1 VHANDLE hevent; hevent=CreateEvent(NULL,0,0,0); " f/ |$ v6 j9 V- [3 b" r$ L7 }WSANETWORKEVENTS socket_events; 2 t: p/ i( t3 P; d) ` P: z) @" [$ x6 YEventId=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); if(socket_events.lNetworkEvents!=0){ switch(socket_events.lNetworkEvents){ case FD_ACCEPT:EventId=FD_ACCEPT;break; case FD_CONNECT:EventId=FD_CONNECT;break; . u0 ]- K: e2 l% \. a# [) O; I: U) dcase FD_READ:EventId=FD_READ;break; 0 a* R3 n9 F. c( o1 V0 m4 Ccase FD_WRITE:EventId=FD_WRITE;break; case FD_CLOSE:EventId=FD_CLOSE;break; 0 U1 T/ C, l! d+ X1 Scase FD_OOB:EventId=FD_OOB;break; default:EventId=FD_NOEVENT;break; } } ( }$ D7 |! x, J* m# Xelse EventId=FD_NOEVENT; return EventId; } 8 p' e" s# h: @1 s// unsigned long GetLocalIp() { char IP[MAX_PATH],*ip; char pc_name[80]; 3 d+ ?- U, j. cstruct in_addr in; struct hostent *host; WORD wVersionRequested; WSADATA wsaData; wVersionRequested=MAKEWORD(2,0); ) s' I) O& P2 Nip=IP; + m0 a' c- U' ustrcpy(ip,"Ip not get!"); % G! g6 F) S1 u. w+ t0 oif(WSAStartup(wVersionRequested,&wsaData)) return 0; ' I& }# ~4 g8 {3 C s1 Eif(gethostname(pc_name,80)==SOCKET_ERROR){ WSACleanup(); return 0; " M |0 N# y4 j# h: a5 U} if(!(host=gethostbyname(pc_name))){ WSACleanup(); return 0; } in.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); . E/ n% Z6 h' t; o1 {WSACleanup(); ' I! J% i f4 V2 Z/ n1 vreturn in.s_addr; 8 L' R# U/ A: R- f} : F% S) M- n' b3 b/ B5 n0 V* [// unsigned long GetDomainIp(char domainname[250]) { char IP[MAX_PATH],*ip; struct in_addr in; 2 s4 \$ m* Z9 x9 R8 }% jstruct hostent *host; & `2 O) V1 }& p# l- CWORD wVersionRequested; 4 o9 [6 j: @1 M% }9 O0 x( o" CWSADATA wsaData; ( E0 `0 [' k3 }wVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; + l) Z0 n% Z6 G7 n0 ]4 {if(!(host=gethostbyname(domainname))){ WSACleanup(); 0 s1 c( l' Y: Z/ Ireturn 0; 5 a+ C; s+ q1 q! g- c0 o* }; c% l} in.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); WSACleanup(); return in.s_addr; / ~$ L% x, _3 A: y& M6 g} */ . `2 ` z5 n# C) v// - F# b, g1 Q7 d% [+ o t// / S6 H b9 j a m' ]UINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 & M& X7 f/ o8 B6 p- l8 hLPSOCKINFO psockinfo; ' D1 F f: Z3 e$ K; w9 ISOCKET sourcesock,destsock; char data[MAXDATALEN]; long eventid; $ O1 P8 \% m0 d5 oint datalen; ! A1 i6 J/ S; M. T* Y1 Y) [psockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; " Y, g5 q3 ?7 J% PTRACE("deail recive thread ok!\r\n"); " Y3 f& q+ i. E4 S+ Q6 k; e/ o xwhile(true){ eventid=GetSocketEventId(sourcesock); switch(eventid){ case FD_CLOSE: TRACE("s fdclosed\r\n"); closesocket(destsock); return 1; $ @6 B# O! P; }* X sbreak; default:break; & D+ S/ ?) e% E! m: K* |5 X} ( i" U* [- ~* f* `, N6 teventid=GetSocketEventId(destsock); , h5 T% b7 H4 D/ d9 qswitch(eventid){ case FD_CLOSE: closesocket(sourcesock); & y$ X1 f$ F% J. g3 l6 p7 @TRACE("d fdclosed\r\n"); 5 S: i$ T5 ?8 Dreturn 1; 9 L9 \+ P# j1 t Abreak; default:break; ) m! u. s& e9 K" k# X) D' H, `: q) ?- v} , _1 K& x1 |+ L0 g" c$ Mdatalen=recv(sourcesock,data,sizeof(data),0); if(datalen==0){ 1 U, O, P u2 D4 T- M% p5 Gclosesocket(sourcesock); % U; k( o/ s. `* M/ \9 }; d: Uclosesocket(destsock); TRACE("s fdclosed\r\n"); break; , U2 g4 |9 h( j; O7 `5 g+ [} 6 ?6 S. G; C P2 S2 ] E5 vif(datalen>0){ ( X2 M6 n8 b, D( B8 Swhile(!send(destsock,data,datalen,0)); } & ~( C7 J& C5 p& R" fSleep(1); & i, V( }9 g, {} " n+ l0 S4 O. I; I" s: ^return 1; ( `: B* r4 o" J; V* e} // 7 ?- W0 I8 G/ N/ [1 lUINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 LPSOCKINFO psockinfo; SOCKET sourcesock,destsock; char data[MAXDATALEN]; " Q2 ~8 A) m$ W) ~0 |long eventid; int datalen; $ w, R+ S+ q `& q6 t5 k, X- fpsockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; 3 c5 I! o+ H8 _& J: C" Pdestsock=psockinfo->destsock; " c% c" P! j3 G) F$ {TRACE("deail send thread ok!\r\n"); ) [3 y9 d4 o: | W$ x8 v" Nwhile(true){ ) z( m9 _# T' B0 a6 eeventid=GetSocketEventId(sourcesock); ( z6 v. C- Z2 i4 Xswitch(eventid){ case FD_CLOSE: ( S5 k: _# X/ L" _TRACE("s fdclosed\r\n"); closesocket(destsock); & Z* p. f S) A$ m: ]return 1; break; default:break; } % y" \6 j5 {. e7 n9 C9 Q! p$ Oeventid=GetSocketEventId(destsock); & e# I# [4 Y9 ~$ J& E6 x5 {switch(eventid){ 1 \/ A/ l) v3 x5 Q# ncase FD_CLOSE: closesocket(sourcesock); . `& H' D0 @ A- s, ~TRACE("d fdclosed\r\n"); return 1; 1 r7 e3 b! ^" t* C5 o0 fbreak; 4 `' h n4 Q4 m6 L) B) Fdefault:break; } datalen=recv(destsock,data,sizeof(data),0); if(datalen==0){ ( u* z+ p& h; H) }closesocket(sourcesock); closesocket(destsock); TRACE("d fdclosed\r\n"); 6 O* E# L: ~" x3 L* `" ]break; + K, h& `% ^# \# Z} if(datalen>0){ 2 t. }' a4 K) T1 fwhile(!send(sourcesock,data,datalen,0)); , p0 D# n6 T) g( `} ; `1 U; Y" S0 B& M6 v- ZSleep(1); 4 o( p3 v- S* ^4 w5 X6 X} 9 U6 v( y; J7 m+ [return 1; 0 t, x% d% F8 j+ e( Y} // 9 A, t7 J7 Z: v- x8 d3 Q0 |// " u. X. Y0 B% h: v; FUINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 % v$ o: E" Z9 u* S M7 ]+ ZLPCLIENTINFO pclientinfo; SOCKET connectsock,clientsock; 1 I. I6 o+ m' v/ [/ Vsockaddr_in remotesock_addr; char data[MAXDATALEN],url[250],temp[250],httpurl[250],portnum[10]; int datalen,i,index_start,index_end,port; CString HttpString,UrlString,PortString; 2 d0 ~+ b* k u6 j6 G9 _7 r4 @1 Spclientinfo=(LPCLIENTINFO)info; clientsock=pclientinfo->clientsock; ZeroMemory((void *)data,sizeof(data)); datalen=recv(clientsock,data,sizeof(data),0); if(datalen<=0){ % w" H& `; x8 G4 r1 Y2 n: f# i9 _closesocket(clientsock); return 0; + o. m: E2 F m8 B

韩冰

} ' _4 @& z: \) ]/ zHttpString.Format("%s",data); UrlString=HttpString; ! I# }, |2 Q% i7 a6 [, PTRACE("get http string:\r\n"); TRACE(HttpString); index_start=HttpString.Find("Host: ",0); //寻找url标记 if(index_start<=0){ , {+ R3 ]1 M) r$ D9 o( c( Qclosesocket(clientsock); return 0; # A# C, u( e: [) K5 P} index_end=HttpString.Find("\r\n",index_start); if(index_end<=0){ + X3 |; c6 q; @: A5 F$ k6 l4 fclosesocket(clientsock); " B b* L4 Z2 _% ]return 0; - n1 U: g/ i$ }} 5 \, k+ C$ ?0 XUrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 6 a+ c; P6 B0 a& e& q8 y- uTRACE("\r\n get url:"); TRACE(UrlString); wsprintf(url,"%s",UrlString); 3 a: x, C& t N/ e( |& _strcpy(temp,url); / X, \/ z$ n5 B9 U# lstrcat(temp,":"); 9 v# ^: A# v# ?0 K/ ]datalen=strlen(temp); / f% [$ R- U# Wif(HttpString.Find("GET",0)==0){ //判断get命令，并处理 index_start=HttpString.Find(temp,0); strcpy(httpurl,"http://"); if(index_start>0){ index_end=HttpString.Find("/",index_start); : }; p0 P$ s2 \, Bif(index_end<=0){ - Z$ e: r8 Z) |3 @8 `closesocket(clientsock); return 0; ! [2 S1 N# v( b* j} 9 s! ~( n: S! R8 H( {PortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); a; a$ ?/ G( eport=atoi(PortString); strcat(httpurl,temp); , K/ u2 Z+ y, } N- ditoa(port,portnum,sizeof(portnum)); strcat(httpurl,portnum); strcat(httpurl,"/"); } % @8 Q, Q' d: d4 relse{ h$ c- Z, m2 Y6 lport=80; strcat(httpurl,url); 8 x. y; o9 A+ _* A/ Tstrcat(httpurl,"/"); ; p+ S/ z8 _7 V& k1 B} % u9 S9 n2 F# H" f# B/ RTRACE("get http url:%s\r\n",httpurl); HttpString.Replace(httpurl,"/"); HttpString.Replace("Proxy-",""); 2 M4 a* p) w1 x! j+ @ L$ LHttpString.Replace("HTTP/1.0","HTTP/1.1"); } else if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 8 B4 ~3 D% W1 A. k) dindex_start=HttpString.Find(temp,0); if(index_start>0){ / V! V2 Y; |1 n" z7 jindex_end=HttpString.Find(" ",index_start); / c) a( p' d R' p+ R. D' \3 B6 jif(index_end<=0){ closesocket(clientsock); return 0; } 3 S& X6 q3 J, f# K& T) T$ w9 hPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); p2 ~1 Y3 |2 @1 t# J7 Fport=atoi(PortString); } else{ + y v9 I, ^4 `% I; Gclosesocket(clientsock); " w5 c( y, C# [. Creturn 0; } } M! L3 Y; H1 U- _( ^' f! u7 bTRACE("get new http string:\r\n"); TRACE(HttpString); remotesock_addr.sin_family=AF_INET; 0 Y, E( d3 J& t& T( tremotesock_addr.sin_port=htons(port); remotesock_addr.sin_addr.S_un.S_addr=GetDomainIp(url); connectsock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); ; ]+ _) R7 h, e- Aif(connect(connectsock,(const sockaddr *)&remotesock_addr,sizeof(remotesock_addr))==INVALID_SOCKET){ //连接远程主机 f! Q* F' f. J# l8 q- g, rclosesocket(clientsock); 5 L9 a @2 W. ]0 ?$ h2 e) rreturn 0; 0 |% X8 t/ P: H1 M$ W5 G! w} 1 h- E l, D! i, }TRACE("\r\nconnect to remote ip ok\r\n"); - y/ S U( l# g/ p, P8 ? CZeroMemory((void *)data,sizeof(data)); wsprintf(data,"%s",HttpString); & w8 P4 F! `. U0 y h, H& Hdatalen=strlen(data); if(HttpString.Find("CONNECT",0)<0) while(!send(connectsock,data,datalen,0)); else{ strcpy(data,"HTTP/1.0 200 Connection established\r\nProxy-agent: CHTTPPROXY V1.0 powered by shadow\r\n\r\n"); 4 A* O" m! f5 k. e- sdatalen=strlen(data); while(!send(clientsock,data,datalen,0)); . Q' Y, t% ~! w} httpsockinfo.sourcesock=clientsock; httpsockinfo.destsock=connectsock; f# }4 G% @: f) ?- e" qAfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 : b8 A' s) T# pAfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // Sleep(100); & u* v3 r( d5 h( d+ [ O1 creturn 1; ! K- f! ?+ T9 R$ W* {} " w5 e, o7 u: H6 V// UINT StartHttpProxy(LPVOID info){ //端口监听线程 SOCKET NewSock; int socklen; $ g% }( N2 H. S: R! s; k. {sockaddr_in serversock,remotesock_addr; serversock.sin_family=AF_INET; serversock.sin_addr.S_un.S_addr=INADDR_ANY; serversock.sin_port=htons(HttpListenPort); 4 u. J& C8 t* sHttpListenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); if(HttpListenSock==INVALID_SOCKET) return 0; if(bind(HttpListenSock,(const sockaddr *)&serversock,sizeof(serversock))==SOCKET_ERROR) return 0; listen(HttpListenSock,BACKUP); , [4 a, A) S& s- H# A5 k1 zsocklen=sizeof(remotesock_addr); . ?0 K/ C# u8 e% hTRACE("start http proxy thread while\r\n"); 3 ]) ]9 k. |0 U0 Q4 Uwhile(true){ ' Q v2 M1 k2 ]4 x- YNewSock=accept(HttpListenSock,(sockaddr *)&remotesock_addr,&socklen); TRACE("waitting ok...\r\n"); if(NewSock==INVALID_SOCKET){ Sleep(1); : l( l' j) z( ^% B# N ]6 d2 {continue; ( x3 \ g r% g9 B* D} ZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); HttpClientInfo.clientsock=NewSock; - x6 {. c% J3 `; J# w2 A' O/ ~HttpClientInfo.clientsock_addr=remotesock_addr; % f# L5 j' |7 W, e) q2 k9 XTRACE("start proxy thread\r\n"); AfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); Sleep(100); } return 1; 7 s1 S: E2 R1 Z/ Z+ e4 |} // ( R- i# z, j3 H! L3 cCHTTPPROXY::CHTTPPROXY() { . K! f: b- n6 L! @6 EWSADATA WsaData; % D* n$ ~* B0 P( e( X6 m7 e6 @# GWORD wsaVer; 8 g9 e0 @4 j" ^7 a0 _. ywsaVer=MAKEWORD(2,0); WsaStartupOk=false; * ?5 n# x6 ~ q# l$ e+ qif(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; & r+ Y+ n/ l) m W5 c& k: h% R+ s}

CHTTPPROXY::~CHTTPPROXY() { if(WsaStartupOk){ WSACleanup(); } }

int CHTTPPROXY::StartProxy(int listenport) { HttpListenPort=listenport; AfxBeginThread(StartHttpProxy,(LPVOID)NULL); return 1; : s0 n v( I# j7 y# r}

% i5 y: Z( o" ]! t/*========================================================================= 8 e0 V" F7 Y% }1 c& _5 d0 oFILE:HTTPPROXY.h # g1 y6 n4 V4 j3 T+ F9 r8 C7 z0 x==========================================================================*/ class CHTTPPROXY : R$ k" A( f3 c) j' _/ W{ . ^. z6 _. w: O! Lpublic: int StartProxy(int listenport); bool WsaStartupOk; * B1 Y% [6 b* G3 V O7 WCHTTPPROXY(); $ [! P% m: R3 ?. k! j5 @virtual ~CHTTPPROXY();

};

/*========================================================================= FILE:stdafx.h ==========================================================================*/ #include 1 J$ h# ^' x( b$ U, H#include #include + A3 C4 ^( u6 x! w: {0 @( q3 f: U#include

, u( f# Y# v. U' V9 ^5 S! _注:不要忘了在link选项中添加wsock32.lib和ws2_32.lib，或在文件前部加上如下语句： #paragma comment(lib,"wsock32.lib") #paragma comment(lib,"ws2_32.lib")

本代码在win2k和vc6.0下编译成功~~ # y- f$ w3 l! A: q- e) F" t用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: CHTTPPROXY httpproxy; httpproxy.StartProxy(7890); 有问题mailto me！