打造自己的HTTP代理服务器

韩冰

/*========================================================================= FILE:HTTPPROXY.cpp HTTP Proxy v1.0 4 Y2 z- N) k( V9 L+ `powered by shadow 2004/11/14 2 Z% E& Y% u% H: H9 u6 pmy web:http://www.codehome.6600.org . r q. d. J) E' z" u U% SQQ:176017352 请先阅读有关http代理协议的相关资料,转载请著明出处 ==========================================================================*/

#include "stdafx.h" #include "HTTPPROXY.h" #define FD_NOEVENT 0 S; N3 z; X/ ~) B8 p2 K// ! U7 w: \, j" M2 m5 p" y#define BACKUP 100 , K3 Y. Q6 C4 d: I. z#define MAXDATALEN 65535 1 l) N6 t2 W! o0 gint HttpListenPort; 0 Y2 G5 I/ w3 W; g. I/ ISOCKET HttpListenSock; // typedef struct _CLIENTINFO{ * I% F2 H. e) DSOCKET clientsock; SOCKET udpsock; sockaddr_in clientsock_addr; $ T+ C! q, H% q H}CLIENTINFO,*LPCLIENTINFO; CLIENTINFO HttpClientInfo; // ' `2 h5 g) ^, X& Z, Otypedef struct _SOCKINFO{ & A7 F" q5 Y- E$ i4 J; ^SOCKET sourcesock; 7 n& d% ]7 l1 l* B2 }1 USOCKET destsock; % j) ^9 O0 v( n, Y3 P/ D}SOCKINFO,*LPSOCKINFO; 4 h& H% i6 h. {4 M. TSOCKINFO httpsockinfo; // extern long GetSocketEventId(SOCKET remotesock); 4 `5 u3 a% G3 x5 L: h" j, C$ `3 o1 k( o2 |extern unsigned long GetLocalIp(); * v E4 c4 \0 i( d% m$ e% Wextern unsigned long GetDomainIp(char domainname[250]); /* long GetSocketEventId(SOCKET remotesock){ ; F- A/ }4 R( J0 ^long EventId; HANDLE hevent; / _- G) w: z3 u' W8 shevent=CreateEvent(NULL,0,0,0); 8 R& j% d& Z& [- F) P; oWSANETWORKEVENTS socket_events; EventId=FD_NOEVENT; 0 t6 H' [0 L& A8 ]if(WSAEventSelect(remotesock,hevent,FD_ACCEPT|FD_CONNECT|FD_READ|FD_WRITE|FD_CLOSE)==SOCKET_ERROR) return EventId; 3 N) @& l% g2 t- G( J) QWSAEnumNetworkEvents(remotesock,hevent,&socket_events); * y/ i5 L4 L. Y; Q+ fif(socket_events.lNetworkEvents!=0){ switch(socket_events.lNetworkEvents){ case FD_ACCEPT:EventId=FD_ACCEPT;break; case FD_CONNECT:EventId=FD_CONNECT;break; case FD_READ:EventId=FD_READ;break; case FD_WRITE:EventId=FD_WRITE;break; case FD_CLOSE:EventId=FD_CLOSE;break; case FD_OOB:EventId=FD_OOB;break; 5 C2 u: s' J5 u' P4 y- |. V, B/ adefault:EventId=FD_NOEVENT;break; } - @6 |% M6 _4 `5 K} - i Z8 Z- R+ P& x) Q Qelse EventId=FD_NOEVENT; . A6 X: X4 ?. T9 E5 Greturn EventId; } - I: j0 |+ V5 q+ N+ }2 p! y. Q" v// 2 S) U$ N0 \7 s+ W( `unsigned long GetLocalIp() . ^7 n- _# ~ ^; w{ char IP[MAX_PATH],*ip; char pc_name[80]; 2 }/ I9 ^) |2 {8 dstruct in_addr in; ' z% h7 G+ r0 p2 C* Jstruct hostent *host; WORD wVersionRequested; WSADATA wsaData; wVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; if(gethostname(pc_name,80)==SOCKET_ERROR){ WSACleanup(); / l' Q7 o* a0 [. Rreturn 0; } ; l" ]# E! V) K' O* v" M' R3 Xif(!(host=gethostbyname(pc_name))){ WSACleanup(); return 0; } 1 R) e' i3 W- ?in.s_addr=*((unsigned long *)host->h_addr_list[0]); 3 w, d8 _- l: O) W9 [5 v/ `strcpy(ip,inet_ntoa(in)); 4 v- }4 m" O) S3 |% BWSACleanup(); return in.s_addr; 9 {: r& K" C2 \+ e; H} // / ]( A4 o+ I6 X5 j5 h2 @5 tunsigned long GetDomainIp(char domainname[250]) { char IP[MAX_PATH],*ip; ' g* @1 e( R# r* A0 m k" ?% Bstruct in_addr in; ) b5 X) F( o$ F/ k/ istruct hostent *host; 8 M& r4 \% E) [1 p0 B5 oWORD wVersionRequested; ( ^$ O9 ~0 n. j# rWSADATA wsaData; wVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); 2 d. }$ T( n8 Zif(WSAStartup(wVersionRequested,&wsaData)) return 0; if(!(host=gethostbyname(domainname))){ % }# a6 \- e# DWSACleanup(); / L# j+ J0 v; T9 ?return 0; . `& y: g( d+ }- h; r+ l3 S} in.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); WSACleanup(); C" p; B, Y8 L; _! kreturn in.s_addr; ~4 D( I: T0 ^! ]- a8 a) Z} */ j/ m1 P) x1 v0 u: N0 a, t// // UINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 1 s4 g" M! x; RLPSOCKINFO psockinfo; + x! N9 u9 N7 M4 j$ @9 N9 cSOCKET sourcesock,destsock; char data[MAXDATALEN]; ; W6 E8 w" s- G; d3 z9 `( _, Z& a1 Ilong eventid; int datalen; psockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; 4 G# Y4 Z1 ]$ b5 N. f1 P& x/ Ddestsock=psockinfo->destsock; # \, V) E% s0 ~# z ^9 x5 Q( ZTRACE("deail recive thread ok!\r\n"); 3 \# }& C( v: c4 |2 c' Z. @while(true){ ( S/ \& Q; W8 j7 {/ Y6 Neventid=GetSocketEventId(sourcesock); switch(eventid){ case FD_CLOSE: . {( Q2 S% r6 Y2 u. M+ ITRACE("s fdclosed\r\n"); closesocket(destsock); ' J1 q% t6 R% z3 |+ C wreturn 1; break; default:break; ( f1 l/ P5 t( I% e1 T4 l, Y} eventid=GetSocketEventId(destsock); switch(eventid){ case FD_CLOSE: closesocket(sourcesock); 7 Q- v+ J/ R! h: B2 ~+ {TRACE("d fdclosed\r\n"); return 1; 2 A5 _, J& S9 ^break; $ w% ]* n4 n4 u% {0 D$ f6 Zdefault:break; $ `# k3 u" c _; }6 \8 Y} datalen=recv(sourcesock,data,sizeof(data),0); if(datalen==0){ closesocket(sourcesock); ! U0 q ?" |7 g h8 |closesocket(destsock); ( h7 Y* w1 y% d1 `+ V+ i5 H q0 `TRACE("s fdclosed\r\n"); break; ' _ [! D0 ^* F# X* p5 d} if(datalen>0){ b$ k) ~& _! r/ ^+ A) a* x! Cwhile(!send(destsock,data,datalen,0)); } 2 ^, L F6 Z' P7 ESleep(1); 7 ], K \$ @7 L: }} return 1; 1 Q) L6 e9 P8 ~$ _6 U; f; u, S} + f4 v" Y$ W2 _# o// 3 f2 |, `. T9 \1 Q [5 u, Y# _UINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 $ e$ I& z$ [: R4 B, C/ D4 |5 Z5 ELPSOCKINFO psockinfo; " z& X! N! Y' QSOCKET sourcesock,destsock; # x! s8 @7 O& L1 `char data[MAXDATALEN]; long eventid; int datalen; 8 _6 o2 E0 E1 f) cpsockinfo=(LPSOCKINFO)info; : c# A) d; L2 I7 }9 P4 _" jsourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; TRACE("deail send thread ok!\r\n"); + K! T |- w( x: ~# N. B I# Mwhile(true){ eventid=GetSocketEventId(sourcesock); 9 R0 B. N3 o& Q7 ^6 Q: G! Rswitch(eventid){ . \' {- s3 C4 ]: ^2 `5 N0 R( vcase FD_CLOSE: TRACE("s fdclosed\r\n"); 6 {$ B$ j) z7 O% W5 T7 Eclosesocket(destsock); return 1; ! U E% O3 z( Q* Y& P, B) r3 Xbreak; ! P. R- v* ]6 C# Y7 T- Xdefault:break; } eventid=GetSocketEventId(destsock); $ Q5 y* j7 n. Y0 B7 I' C& \switch(eventid){ case FD_CLOSE: closesocket(sourcesock); ) B/ G9 {6 m3 C F7 Z( N: nTRACE("d fdclosed\r\n"); return 1; 9 j: b9 e$ k/ O3 T; Ebreak; : C( c) W' M/ M, Y f2 ^default:break; 2 a- U0 C; O& ]% D% ~% Y$ ^} datalen=recv(destsock,data,sizeof(data),0); _+ V' _1 Z: R" p; g+ _if(datalen==0){ , Q6 S& J8 @( \+ [! ]+ V+ q* B4 M9 Zclosesocket(sourcesock); ! ?. G8 L! }9 O+ ~, z) kclosesocket(destsock); TRACE("d fdclosed\r\n"); 8 [: ]7 c% ~$ B8 ]) U2 v" d; vbreak; 4 B7 `. e: h' ]5 N9 i0 @6 n} if(datalen>0){ while(!send(sourcesock,data,datalen,0)); + X- D" g$ j$ X v} 3 q* ]* T/ r8 hSleep(1); 5 [7 C7 F6 P i T1 D4 E$ h% |} 9 R8 w& T; s& Z/ Nreturn 1; - V' u; \2 U/ p9 G5 J6 a" i! l} g0 L9 Q8 {# u- l1 i6 V8 H: T// // UINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 LPCLIENTINFO pclientinfo; SOCKET connectsock,clientsock; & t9 ]& n j7 s- [sockaddr_in remotesock_addr; ) E1 X8 n7 r! S: }, e1 Schar data[MAXDATALEN],url[250],temp[250],httpurl[250],portnum[10]; . L; ?& V: @: Z" ~( U, D7 b5 @. f `) xint datalen,i,index_start,index_end,port; CString HttpString,UrlString,PortString; pclientinfo=(LPCLIENTINFO)info; clientsock=pclientinfo->clientsock; ZeroMemory((void *)data,sizeof(data)); ! t; x- l1 l6 A6 c9 y5 ], Mdatalen=recv(clientsock,data,sizeof(data),0); + [% Q& @$ g) z- K7 y" O! gif(datalen<=0){ % l% w& j; L, R2 v, W- L0 b7 Sclosesocket(clientsock); / O% L! H0 @& y. {) [return 0; ! T7 z$ S- Y {5 M

韩冰

} $ F6 m8 f6 l: i# M' \/ RHttpString.Format("%s",data); " Y5 v- v5 F& c# x/ tUrlString=HttpString; : Z6 b0 U r/ n) G2 ]TRACE("get http string:\r\n"); TRACE(HttpString); $ w4 N- Z" I+ K" @ Pindex_start=HttpString.Find("Host: ",0); //寻找url标记 if(index_start<=0){ closesocket(clientsock); return 0; ( B u: D6 Y1 z2 N3 P} index_end=HttpString.Find("\r\n",index_start); if(index_end<=0){ closesocket(clientsock); 3 ]4 I+ r, H) G. n* V# N/ f. ?return 0; * y/ ^% b% }5 O' ]8 p( @9 c} 6 r8 b7 y9 L* Z/ w0 b7 ]UrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 " ]; d& l% R/ NTRACE("\r\n get url:"); TRACE(UrlString); wsprintf(url,"%s",UrlString); 7 Q1 e: S8 o/ H, k, ~( ]strcpy(temp,url); 8 y: G4 {1 U: c) b& q8 Y, I( C2 G6 kstrcat(temp,":"); datalen=strlen(temp); if(HttpString.Find("GET",0)==0){ //判断get命令，并处理 5 N7 l- Y: ?9 P) @# X' @! `index_start=HttpString.Find(temp,0); strcpy(httpurl,"http://"); 4 n/ ]- J7 e5 S+ v* r) G5 ]- p. iif(index_start>0){ a. F5 [! i; B- j* @index_end=HttpString.Find("/",index_start); if(index_end<=0){ closesocket(clientsock); $ Z& R1 A5 T/ `2 breturn 0; , |" r6 n8 Q% |1 h+ _3 t} - r# g* K. ^4 M, l) O) I( rPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); 4 V. M7 c6 y! |7 e6 d/ o$ w+ ]port=atoi(PortString); : h$ _( |. N9 l; wstrcat(httpurl,temp); % ?# G4 x. [6 g: A- s4 A1 P5 s6 xitoa(port,portnum,sizeof(portnum)); strcat(httpurl,portnum); . ?& A N7 c8 G! D: n$ V0 ustrcat(httpurl,"/"); ; s( `& Q7 _/ l/ ~ @+ Z! x9 s+ \} - l& f. [8 ]0 r+ c5 b7 H6 V# @else{ port=80; strcat(httpurl,url); ) e2 s2 `) l; Q2 P8 z; i5 `- J& vstrcat(httpurl,"/"); } TRACE("get http url:%s\r\n",httpurl); % T7 J6 e. B. C$ KHttpString.Replace(httpurl,"/"); HttpString.Replace("Proxy-",""); 2 o8 n/ w5 U8 E$ `0 u+ G& _HttpString.Replace("HTTP/1.0","HTTP/1.1"); 5 t3 i5 i( ~! b5 G( Y} else if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 ! W# \+ ~7 F vindex_start=HttpString.Find(temp,0); if(index_start>0){ % m$ x. |" m/ q! q$ @% E" l7 q; r9 e* Bindex_end=HttpString.Find(" ",index_start); if(index_end<=0){ closesocket(clientsock); 5 n; e3 g% s( @' xreturn 0; } 6 u0 J" _, [0 j6 k! X( z2 }PortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); 1 U% M, i8 m( o! n. [2 ^7 J5 |port=atoi(PortString); } else{ closesocket(clientsock); 7 F1 Z( B, c+ ^4 Z; e2 |2 @return 0; } } TRACE("get new http string:\r\n"); TRACE(HttpString); + @& h( l; `( B, Uremotesock_addr.sin_family=AF_INET; remotesock_addr.sin_port=htons(port); * d0 J& {2 G) P- q' Jremotesock_addr.sin_addr.S_un.S_addr=GetDomainIp(url); connectsock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); * T. v: `1 P. |) q, u8 |if(connect(connectsock,(const sockaddr *)&remotesock_addr,sizeof(remotesock_addr))==INVALID_SOCKET){ //连接远程主机 : s; ~. w ^) kclosesocket(clientsock); return 0; } / O( p ^# {0 s I4 v% LTRACE("\r\nconnect to remote ip ok\r\n"); ZeroMemory((void *)data,sizeof(data)); * ^/ Q2 d9 x# vwsprintf(data,"%s",HttpString); datalen=strlen(data); 1 h7 J% L" y& j' eif(HttpString.Find("CONNECT",0)<0) while(!send(connectsock,data,datalen,0)); % v8 j0 L! F D' ]else{ strcpy(data,"HTTP/1.0 200 Connection established\r\nProxy-agent: CHTTPPROXY V1.0 powered by shadow\r\n\r\n"); datalen=strlen(data); while(!send(clientsock,data,datalen,0)); } httpsockinfo.sourcesock=clientsock; httpsockinfo.destsock=connectsock; % r% e# k# V Q8 j& u0 O2 a, k2 }" bAfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 AfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // Sleep(100); return 1; } ; G" i/ r( L. O6 L0 o// ' m9 j# p: Z% IUINT StartHttpProxy(LPVOID info){ //端口监听线程 SOCKET NewSock; int socklen; sockaddr_in serversock,remotesock_addr; serversock.sin_family=AF_INET; serversock.sin_addr.S_un.S_addr=INADDR_ANY; ; C8 J8 ?" |) A3 w4 N; V# e( T/ b' Sserversock.sin_port=htons(HttpListenPort); " [8 g. v5 b) ~, c& z; LHttpListenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); ' t9 _+ \. Y' ]" ]7 ^if(HttpListenSock==INVALID_SOCKET) return 0; 2 V0 o8 ~" X; k5 E5 wif(bind(HttpListenSock,(const sockaddr *)&serversock,sizeof(serversock))==SOCKET_ERROR) return 0; listen(HttpListenSock,BACKUP); socklen=sizeof(remotesock_addr); TRACE("start http proxy thread while\r\n"); while(true){ NewSock=accept(HttpListenSock,(sockaddr *)&remotesock_addr,&socklen); % Z0 T3 Z5 c. X' e2 S% ?% Q! @* m2 vTRACE("waitting ok...\r\n"); if(NewSock==INVALID_SOCKET){ . u4 U$ j# G7 |$ |Sleep(1); 7 ]/ f5 k3 H Y( l1 z/ Z0 \continue; } ZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); # R& ?1 q, ?# Q( I' l9 g0 HHttpClientInfo.clientsock=NewSock; HttpClientInfo.clientsock_addr=remotesock_addr; 4 u# L% O' a- K: @. `8 C: WTRACE("start proxy thread\r\n"); ! W" E& A$ _& m9 M( B5 D4 IAfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); , x* p6 P# m5 ]. eSleep(100); . @- }' M0 y3 q* P- t' W! W} ; N+ ^8 M ?6 K$ Greturn 1; } // + u$ @ g3 H. f* g3 fCHTTPPROXY::CHTTPPROXY() ) |6 V0 K2 ~+ K1 A+ X4 B{ # i$ U& j+ [: `$ S6 E# Y( QWSADATA WsaData; WORD wsaVer; , P! g' j' g9 A* f& g, b3 jwsaVer=MAKEWORD(2,0); WsaStartupOk=false; if(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; }

CHTTPPROXY::~CHTTPPROXY() { if(WsaStartupOk){ WSACleanup(); 4 V7 E$ l" a! M8 @} }

int CHTTPPROXY::StartProxy(int listenport) { HttpListenPort=listenport; AfxBeginThread(StartHttpProxy,(LPVOID)NULL); return 1; , q6 ]/ u( j: E/ ^7 S, l( Y}

) _4 b/ o1 }+ W$ ^- S: I/*========================================================================= 6 s0 W& e% l. e" P4 O7 ]$ CFILE:HTTPPROXY.h , v4 L; _8 b; l. d==========================================================================*/ class CHTTPPROXY ' o5 {2 y7 x- c{ public: , D2 c, G5 X1 m5 ?6 V6 i" Yint StartProxy(int listenport); bool WsaStartupOk; 8 w: o1 i. Y/ @& U# [6 I; ^( s FCHTTPPROXY(); virtual ~CHTTPPROXY();

};

/*========================================================================= FILE:stdafx.h . S: n) R( U4 T' [4 t, H==========================================================================*/ # c) S& w. {" j R#include / L& a: c8 i m% y1 l6 R d#include #include : {3 X4 s4 G4 L) y$ U# }- Q# _#include

注:不要忘了在link选项中添加wsock32.lib和ws2_32.lib，或在文件前部加上如下语句： 9 j3 t1 T5 T8 P7 y- t#paragma comment(lib,"wsock32.lib") : N4 d% R8 T. ~$ c, c, A#paragma comment(lib,"ws2_32.lib")

本代码在win2k和vc6.0下编译成功~~ / F) {3 i; \6 I用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: 3 v2 w4 o( |6 i* X8 z+ @CHTTPPROXY httpproxy; 4 L# E# b+ q. _" B2 W2 A2 W% Xhttpproxy.StartProxy(7890); 7 ^- R7 w- W9 A% L2 K( l" o# k有问题mailto me！ ( K8 z8 U5 x2 o. B Y