打造自己的HTTP代理服务器

韩冰

/*========================================================================= " ~8 z. k: L+ E+ w( C8 {FILE:HTTPPROXY.cpp HTTP Proxy v1.0 / Y# u. b7 }6 P& v3 upowered by shadow 2004/11/14 my web:http://www.codehome.6600.org ! g. D( h$ T& Y# R7 \QQ:176017352 / k' J; x( j; W$ L) F. {请先阅读有关http代理协议的相关资料,转载请著明出处 ==========================================================================*/

#include "stdafx.h" 3 m8 D! K( s$ x1 ?5 R2 v+ @#include "HTTPPROXY.h" #define FD_NOEVENT 0 // #define BACKUP 100 " J1 w" o( a- h5 {9 O2 x( _3 V#define MAXDATALEN 65535 int HttpListenPort; SOCKET HttpListenSock; // 1 A* T0 X/ _0 c/ |typedef struct _CLIENTINFO{ ) g b# A; z' R- E8 [0 K* h( ESOCKET clientsock; + n* ?. G# z0 N; b/ gSOCKET udpsock; sockaddr_in clientsock_addr; }CLIENTINFO,*LPCLIENTINFO; * k* f) w9 ~# s9 DCLIENTINFO HttpClientInfo; // : V% ^: _& i: |$ |9 ftypedef struct _SOCKINFO{ . I" }6 u1 |9 a) d" qSOCKET sourcesock; SOCKET destsock; }SOCKINFO,*LPSOCKINFO; 7 j1 }7 C' V5 E4 }SOCKINFO httpsockinfo; , \: W2 }! ?# z# \// extern long GetSocketEventId(SOCKET remotesock); : }5 S- S5 c0 p5 [2 a# c4 A9 mextern unsigned long GetLocalIp(); extern unsigned long GetDomainIp(char domainname[250]); 9 ^3 s7 f% X0 A" T/* 2 a/ N5 i; T, E, Q' R. xlong GetSocketEventId(SOCKET remotesock){ long EventId; , N" n) G" t( J' \: `HANDLE hevent; hevent=CreateEvent(NULL,0,0,0); WSANETWORKEVENTS socket_events; 8 c$ a8 k. X4 |- vEventId=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) y4 M8 m' E: \& Q) b' e2 gif(socket_events.lNetworkEvents!=0){ 9 |2 H( M9 v, Q2 ^switch(socket_events.lNetworkEvents){ ! z. L) I% ]. |5 L4 ^% a1 Ccase FD_ACCEPT:EventId=FD_ACCEPT;break; . \+ a/ m) H+ a3 a r K9 w9 O6 ]8 _case FD_CONNECT:EventId=FD_CONNECT;break; ' Q# Y% T8 y) l. `! L N8 X9 _case FD_READ:EventId=FD_READ;break; & F: ?8 W8 q! n* `# a" acase FD_WRITE:EventId=FD_WRITE;break; case FD_CLOSE:EventId=FD_CLOSE;break; case FD_OOB:EventId=FD_OOB;break; 9 M; C8 |9 s/ L0 H" ~0 e! gdefault:EventId=FD_NOEVENT;break; - J2 @) f/ |8 }& ]} } else EventId=FD_NOEVENT; return EventId; } // 7 K1 C3 W1 Z1 Punsigned long GetLocalIp() ! J# p5 C5 ~0 S" l{ 2 N* y/ {( B4 F+ U, x6 S: `char IP[MAX_PATH],*ip; char pc_name[80]; @' c: A1 C3 e N% ]* {struct in_addr in; 1 X7 `/ _: x9 K# J, T: [( u- hstruct hostent *host; . `" Y6 Z# G. kWORD wVersionRequested; $ l1 m; D8 w) T/ ] aWSADATA wsaData; wVersionRequested=MAKEWORD(2,0); ip=IP; & P" R7 V& Q/ W) A% l$ F7 Jstrcpy(ip,"Ip not get!"); ! Y, s% I" f) Qif(WSAStartup(wVersionRequested,&wsaData)) return 0; 2 ^, W& v7 [/ L' c$ n' \! K# {if(gethostname(pc_name,80)==SOCKET_ERROR){ 8 N# u+ O [$ a }7 s* M! v6 \- [WSACleanup(); - e% B' A! H0 treturn 0; 7 c; L+ R5 t/ Z8 @5 R} if(!(host=gethostbyname(pc_name))){ WSACleanup(); $ G2 H8 N. h2 I0 Wreturn 0; } 0 V& T9 m; N4 g7 c0 X8 y% Cin.s_addr=*((unsigned long *)host->h_addr_list[0]); . @; R! M5 O: Lstrcpy(ip,inet_ntoa(in)); " p& E9 j5 K5 B$ A% }1 d$ ~7 |- rWSACleanup(); return in.s_addr; : o" f2 }) ^2 m! V} // & J! }, j0 o6 w U# W( p1 Junsigned long GetDomainIp(char domainname[250]) { 2 z0 _0 ^. X( E1 Lchar IP[MAX_PATH],*ip; / u. R. r/ J& x6 H q' q T/ Bstruct in_addr in; struct hostent *host; # r/ `2 l3 {6 @; DWORD wVersionRequested; WSADATA wsaData; ! H8 L8 B1 u0 R1 hwVersionRequested=MAKEWORD(2,0); ip=IP; 9 _9 P; ~5 Q8 N) J% }8 J: Ostrcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; if(!(host=gethostbyname(domainname))){ WSACleanup(); ' C$ D# U* u* g% nreturn 0; 6 `0 ^7 f. q0 b Y1 ^} * w6 S: G* ^$ r* M# O/ qin.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); WSACleanup(); & G) x2 s0 l8 u: K% H) x3 dreturn in.s_addr; ( T9 Q$ D* G- \, i! I2 t} */ // ! K% i# M4 ^% Q$ R [/ z// + s M6 Q0 u" N" H, m" bUINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 4 k B* P# J% m+ w) I3 RLPSOCKINFO psockinfo; SOCKET sourcesock,destsock; ' @+ ]: ^ \9 I' B& I4 M3 vchar data[MAXDATALEN]; ! y3 F% V, ]+ M" V6 `long eventid; - K; z& s$ ?8 c4 v: W# oint datalen; 2 N4 x4 C2 A- S; H: ?8 @: N7 lpsockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; TRACE("deail recive thread ok!\r\n"); while(true){ eventid=GetSocketEventId(sourcesock); switch(eventid){ # X) n& Q* z$ V {1 jcase FD_CLOSE: TRACE("s fdclosed\r\n"); closesocket(destsock); % \, l# N2 a" D$ d* Vreturn 1; 0 v6 z h3 \6 a# S% I% Y Pbreak; default:break; } 7 V& t6 u' ?4 y+ qeventid=GetSocketEventId(destsock); $ j9 }# b- z% l" j6 X& O0 W; o- Uswitch(eventid){ * m' o! W% r+ o ^case FD_CLOSE: 9 s$ _" t: X ?, ~: y! eclosesocket(sourcesock); TRACE("d fdclosed\r\n"); return 1; break; ' |* i# G& {2 c6 Xdefault:break; } . w; r$ x' f% i6 _datalen=recv(sourcesock,data,sizeof(data),0); 6 ]; J' I. f* c9 \( bif(datalen==0){ closesocket(sourcesock); ' Z7 m8 M* w A9 T! o& \3 tclosesocket(destsock); TRACE("s fdclosed\r\n"); . N8 P8 s2 h- Wbreak; ' y! R! X: C2 C7 r} ) K* [7 n) D/ ]if(datalen>0){ while(!send(destsock,data,datalen,0)); } . b( x4 W. d5 p$ Q7 r& `Sleep(1); 2 x, l; T, u# I} return 1; } // 8 k( c, O- [$ I1 |* {% dUINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 LPSOCKINFO psockinfo; SOCKET sourcesock,destsock; char data[MAXDATALEN]; long eventid; + _! x) g/ H8 J4 Z' B5 jint datalen; psockinfo=(LPSOCKINFO)info; $ q4 N& l8 R, w m) Fsourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; TRACE("deail send thread ok!\r\n"); while(true){ eventid=GetSocketEventId(sourcesock); switch(eventid){ 9 o1 I* F J% kcase FD_CLOSE: . R" D; z; A) S* {/ XTRACE("s fdclosed\r\n"); closesocket(destsock); return 1; 9 }/ S$ R& _+ l1 K/ f% abreak; default:break; 8 M6 z. i5 W4 S9 _6 n} eventid=GetSocketEventId(destsock); 9 B, v" w$ m" ~; _9 Wswitch(eventid){ " y/ i; t, y {2 kcase FD_CLOSE: ; e6 ?7 A& k- r9 uclosesocket(sourcesock); TRACE("d fdclosed\r\n"); ! I8 x$ R. q6 ureturn 1; break; default:break; } datalen=recv(destsock,data,sizeof(data),0); if(datalen==0){ & h" [7 F- H" K3 ^1 M5 H0 zclosesocket(sourcesock); closesocket(destsock); 5 o% w3 {9 }, R& D* ]1 X2 UTRACE("d fdclosed\r\n"); ( a- x/ S; p8 F! r, \! ]5 Wbreak; } 8 {1 r2 q8 k' c: M: K+ T2 Hif(datalen>0){ while(!send(sourcesock,data,datalen,0)); 7 j" @- C" U1 ?- r, e} # D2 K1 s2 x8 A7 q4 t8 fSleep(1); } : k! U# c2 Q& F* Xreturn 1; 0 ~: \9 H6 }1 X# F- I; P0 f; j' s} // , z1 t/ J% N* y9 o, a// 1 N, q8 _3 S# x n6 fUINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 LPCLIENTINFO pclientinfo; SOCKET connectsock,clientsock; sockaddr_in remotesock_addr; , M: d- ~( B! c" Cchar data[MAXDATALEN],url[250],temp[250],httpurl[250],portnum[10]; int datalen,i,index_start,index_end,port; , [% h1 w7 J+ r. D; m) rCString HttpString,UrlString,PortString; 5 @! F d4 I* q5 m/ w- opclientinfo=(LPCLIENTINFO)info; : e& x! N, W( O3 H/ a. U2 Iclientsock=pclientinfo->clientsock; ZeroMemory((void *)data,sizeof(data)); datalen=recv(clientsock,data,sizeof(data),0); 4 h( O9 e, y5 X$ R) r' {+ pif(datalen<=0){ - H9 m# Q I* F1 h! B/ dclosesocket(clientsock); return 0; 9 O2 o# V5 q; D) n" D' X

韩冰

} % G( Q. b8 @8 j/ k) THttpString.Format("%s",data); UrlString=HttpString; TRACE("get http string:\r\n"); / U7 I5 R, R( ~' g9 eTRACE(HttpString); index_start=HttpString.Find("Host: ",0); //寻找url标记 if(index_start<=0){ 1 J! O. D6 d8 l; gclosesocket(clientsock); return 0; ) M; H$ i' D: v9 `% c} index_end=HttpString.Find("\r\n",index_start); if(index_end<=0){ closesocket(clientsock); return 0; 6 q/ B" ~/ E' `! J) }} ! s/ F x, Q3 v. fUrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 TRACE("\r\n get url:"); - v. @1 c z' |TRACE(UrlString); wsprintf(url,"%s",UrlString); 0 J+ V, u0 j+ l& C! fstrcpy(temp,url); |) v' f$ G# u# } }strcat(temp,":"); 5 l$ f& J' E, ~$ J! N7 o1 Sdatalen=strlen(temp); ' P- P5 x3 u6 ]! K. hif(HttpString.Find("GET",0)==0){ //判断get命令，并处理 - p2 T: C D" ~# l- j: b2 ?! oindex_start=HttpString.Find(temp,0); strcpy(httpurl,"http://"); 3 M7 v( r0 i, y% L3 f, g" D6 g, }if(index_start>0){ index_end=HttpString.Find("/",index_start); % l, }( {. v4 a9 Zif(index_end<=0){ + ~4 ^9 P* t* l' _( a& Hclosesocket(clientsock); 4 S2 h" t! s/ p$ O: rreturn 0; } 9 H% \5 K9 E- c* D( b8 B3 t' VPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); port=atoi(PortString); & L7 v7 h) r+ V) `7 v2 o+ wstrcat(httpurl,temp); 9 B& v/ Z( d4 j) Fitoa(port,portnum,sizeof(portnum)); 6 q% Z, g% }# e6 Y6 v: R% |strcat(httpurl,portnum); 7 Q! ]8 i! b: R0 c6 astrcat(httpurl,"/"); } " e0 x. L5 [# j2 h6 [0 t% R# Selse{ 4 P6 @) v' E$ @( }& H" bport=80; 7 \0 d J& T* e ustrcat(httpurl,url); strcat(httpurl,"/"); ( `3 [9 t+ B: g6 E} 0 n" D% a) h0 R) h$ @TRACE("get http url:%s\r\n",httpurl); HttpString.Replace(httpurl,"/"); $ \) N7 L/ J/ K' n8 ~9 z7 R% h7 M( B; EHttpString.Replace("Proxy-",""); HttpString.Replace("HTTP/1.0","HTTP/1.1"); # ^4 f4 X. L* ^1 t* q) g} & a5 A- T$ X( ?else if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 3 \; E& q7 `8 Mindex_start=HttpString.Find(temp,0); & R! z3 i9 ^8 c+ {1 V9 q' Cif(index_start>0){ index_end=HttpString.Find(" ",index_start); ' l* b v4 [# i+ gif(index_end<=0){ closesocket(clientsock); 8 C8 }' S, L Dreturn 0; 7 X" [7 y( b/ o; N} 4 [( V5 b$ t% G) b, _) h5 gPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); port=atoi(PortString); } else{ 3 T/ H3 H6 a1 Aclosesocket(clientsock); / F, {" q! H8 Yreturn 0; } } . M1 |$ N4 o9 F' h1 m9 jTRACE("get new http string:\r\n"); TRACE(HttpString); remotesock_addr.sin_family=AF_INET; 4 G" Z5 w% t' \$ {" m- Y9 wremotesock_addr.sin_port=htons(port); remotesock_addr.sin_addr.S_un.S_addr=GetDomainIp(url); ) j, e* l9 e% R. M6 F7 z7 Lconnectsock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); ) P1 B7 {; f6 pif(connect(connectsock,(const sockaddr *)&remotesock_addr,sizeof(remotesock_addr))==INVALID_SOCKET){ //连接远程主机 % P9 n) ~6 c3 z8 {closesocket(clientsock); return 0; ' p# p2 ` ~: c) ~( B% L# k& ~} / M# S, W6 g. Q8 CTRACE("\r\nconnect to remote ip ok\r\n"); & X, ~! a9 c- A6 F; UZeroMemory((void *)data,sizeof(data)); " Y( g p% x! n* H* vwsprintf(data,"%s",HttpString); datalen=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"); J0 C# I" j& x2 X8 [/ c6 j# Ndatalen=strlen(data); while(!send(clientsock,data,datalen,0)); P& f- H) U5 y! j0 O9 l9 x" q} httpsockinfo.sourcesock=clientsock; ( A; @0 z) |3 y7 r5 c" ]httpsockinfo.destsock=connectsock; $ d( W \$ }9 F* c) UAfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 2 P2 G. {* t3 }% s) xAfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // Sleep(100); 7 H" J* _1 w2 P3 N- R9 S( Q0 ireturn 1; 1 Q* `7 [0 q& l1 F} 6 u# V) l4 m1 v) f. i3 J1 O2 ~// UINT StartHttpProxy(LPVOID info){ //端口监听线程 SOCKET NewSock; 2 B# W6 K1 p# { S- S3 O% H6 Eint socklen; sockaddr_in serversock,remotesock_addr; 6 D+ h& T+ e3 f z: `% hserversock.sin_family=AF_INET; * B+ y6 o- _2 v0 q3 ?8 L. Dserversock.sin_addr.S_un.S_addr=INADDR_ANY; serversock.sin_port=htons(HttpListenPort); HttpListenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); ) X. t: y( q+ h7 `" n u4 R: R v4 Pif(HttpListenSock==INVALID_SOCKET) return 0; & |0 m/ M; z; p, o4 Wif(bind(HttpListenSock,(const sockaddr *)&serversock,sizeof(serversock))==SOCKET_ERROR) return 0; listen(HttpListenSock,BACKUP); 7 J2 {- I" |/ q7 ~ _ bsocklen=sizeof(remotesock_addr); . [. i5 q) e0 M) c$ c& H# @TRACE("start http proxy thread while\r\n"); X( [2 v3 `* h3 cwhile(true){ NewSock=accept(HttpListenSock,(sockaddr *)&remotesock_addr,&socklen); 5 F% f5 X' j$ y5 u" eTRACE("waitting ok...\r\n"); if(NewSock==INVALID_SOCKET){ Sleep(1); continue; } ZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); HttpClientInfo.clientsock=NewSock; 2 m/ }% s5 t- n- c+ F% I7 ~HttpClientInfo.clientsock_addr=remotesock_addr; 6 ^' d$ q% A/ a% `& |TRACE("start proxy thread\r\n"); AfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); * w$ v3 {* O0 o% r6 L) M9 ?# R" YSleep(100); } 9 Z) {/ ]$ F2 i8 `% ]+ Sreturn 1; } + O3 X8 n7 h/ P# D// - E5 c# v- [2 D; [( D/ {0 GCHTTPPROXY::CHTTPPROXY() U8 v) g i. w1 q( Z+ f{ WSADATA WsaData; WORD wsaVer; 3 E: `, _- Y: k5 n* I! I* X% [1 wwsaVer=MAKEWORD(2,0); 0 a2 R2 R& T- W' @! J1 s: n% b4 eWsaStartupOk=false; if(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; }

CHTTPPROXY::~CHTTPPROXY() { 5 ~* U# c6 d1 j% l. ~3 Lif(WsaStartupOk){ 0 a: E x7 u5 O3 `WSACleanup(); ( e& w. W7 z* p9 o' Q1 @2 R+ H} 0 L7 s9 q9 z' t5 r# I}

int CHTTPPROXY::StartProxy(int listenport) { ) ~7 v4 ^, }% W; WHttpListenPort=listenport; AfxBeginThread(StartHttpProxy,(LPVOID)NULL); D r3 f- b& C3 @, Ureturn 1; }

, _% V7 s6 M g8 [; R1 U h/*========================================================================= FILE:HTTPPROXY.h ==========================================================================*/ 0 r7 S+ _% ~7 ]5 v" j" ~) q/ Oclass CHTTPPROXY { public: ' |( ]6 ]- G. a6 `6 B$ qint StartProxy(int listenport); - c& B, @# S8 M6 r2 F: k; Z8 j& abool WsaStartupOk; CHTTPPROXY(); ' @/ Z: f6 Q) Y- x& _9 kvirtual ~CHTTPPROXY();

};

/*========================================================================= , u7 j" Z6 X" W' pFILE:stdafx.h " r4 j7 Z' ?3 ~1 R==========================================================================*/ #include #include #include #include

注:不要忘了在link选项中添加wsock32.lib和ws2_32.lib，或在文件前部加上如下语句： #paragma comment(lib,"wsock32.lib") # q- r9 W' l+ n, k#paragma comment(lib,"ws2_32.lib")

本代码在win2k和vc6.0下编译成功~~ # R `$ t( N3 p0 Q用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: * p) j1 B% h$ y iCHTTPPROXY httpproxy; httpproxy.StartProxy(7890); 有问题mailto me！