打造自己的HTTP代理服务器

韩冰

/*========================================================================= 3 H1 D: [, V. Z2 Q8 f' `FILE:HTTPPROXY.cpp ! ]0 i& y0 a) V- ? ? }HTTP Proxy v1.0 powered by shadow 2004/11/14 my web:http://www.codehome.6600.org QQ:176017352 / @: g2 P5 p; R: G" d1 x# b4 T请先阅读有关http代理协议的相关资料,转载请著明出处 ==========================================================================*/ $ `, F5 S8 ] l" |( T5 w7 \

#include "stdafx.h" / o0 h& s b& O/ x3 _! T0 A#include "HTTPPROXY.h" #define FD_NOEVENT 0 // ( J2 o- w& u* H: M. z: K( L; g d#define BACKUP 100 #define MAXDATALEN 65535 int HttpListenPort; + o2 t, }9 G/ y" ?7 Y* p/ u- |8 _SOCKET HttpListenSock; // typedef struct _CLIENTINFO{ SOCKET clientsock; z% P P& z j/ g1 SSOCKET udpsock; sockaddr_in clientsock_addr; }CLIENTINFO,*LPCLIENTINFO; 0 ^! P* v+ @- u$ d$ PCLIENTINFO HttpClientInfo; 5 l3 v+ [2 Z3 }9 O8 i// + _3 M" b e! d' C$ `& K& X' Dtypedef struct _SOCKINFO{ : O( \. J0 p kSOCKET sourcesock; SOCKET destsock; , G; g7 |% d1 w2 z r1 W}SOCKINFO,*LPSOCKINFO; + Q; I* k! T- u" b7 O/ T; J% iSOCKINFO httpsockinfo; // extern long GetSocketEventId(SOCKET remotesock); 3 _4 d7 r& i" Gextern unsigned long GetLocalIp(); ) x& |4 N/ U. ~. a; ~extern unsigned long GetDomainIp(char domainname[250]); # R2 T3 x; C a' z) \" Y. U/* long GetSocketEventId(SOCKET remotesock){ . b4 k% [8 @" l( l Y: u. Qlong EventId; ( y0 i+ n* a; r0 B$ mHANDLE hevent; hevent=CreateEvent(NULL,0,0,0); , ]& ]% B! m9 M) S' o4 rWSANETWORKEVENTS socket_events; EventId=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); 7 q& N7 w! y5 @' d/ h9 wif(socket_events.lNetworkEvents!=0){ switch(socket_events.lNetworkEvents){ 2 l, U4 j1 f" @case FD_ACCEPT:EventId=FD_ACCEPT;break; ' q* r& z5 O$ E. hcase FD_CONNECT:EventId=FD_CONNECT;break; case FD_READ:EventId=FD_READ;break; case FD_WRITE:EventId=FD_WRITE;break; * O, b0 i" F; c! l- Qcase FD_CLOSE:EventId=FD_CLOSE;break; $ r3 z$ Y9 B" _5 M& ~1 }' r* p% Icase FD_OOB:EventId=FD_OOB;break; default:EventId=FD_NOEVENT;break; } } 7 E# G, M0 i% d# u( I1 Ielse EventId=FD_NOEVENT; , P9 i) M1 o# Jreturn EventId; , P7 L7 `8 K7 r" V2 b* F9 a6 Q} 6 |. a: E' R! M |) t/ k// unsigned long GetLocalIp() 5 m e( X5 u! _2 e r# A; z" h{ char IP[MAX_PATH],*ip; char pc_name[80]; 0 ^8 \3 D! X- q I5 estruct in_addr in; struct hostent *host; WORD wVersionRequested; WSADATA wsaData; wVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; 6 B* M% w8 n' K/ ]1 {if(gethostname(pc_name,80)==SOCKET_ERROR){ & q; E6 c; N9 b% nWSACleanup(); ; A, ~3 Q! F9 L greturn 0; } 9 e' l( E$ _9 l+ I: P9 C7 @if(!(host=gethostbyname(pc_name))){ * R+ p# ~5 @ V6 G- d9 p4 uWSACleanup(); 4 [ [& \9 ^0 q4 f6 Z" p$ Oreturn 0; } ' s9 R5 S' R4 {- Y, @6 Y( Tin.s_addr=*((unsigned long *)host->h_addr_list[0]); 0 x+ `& E1 y. C0 e; d! {. g& Kstrcpy(ip,inet_ntoa(in)); WSACleanup(); return in.s_addr; } // " r |2 e3 @, u/ c2 C Gunsigned long GetDomainIp(char domainname[250]) ( @3 v; e; j7 |{ char IP[MAX_PATH],*ip; " g' z+ O5 Q- [+ gstruct in_addr in; 4 Q" Y" `) q5 Jstruct hostent *host; WORD wVersionRequested; WSADATA wsaData; wVersionRequested=MAKEWORD(2,0); / g+ o) s$ C2 v8 F0 h9 a1 G' Kip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; if(!(host=gethostbyname(domainname))){ 1 X+ h7 R! A/ K( B5 W9 @% \3 U2 lWSACleanup(); ; b. g9 }6 V8 g; |1 qreturn 0; } + l0 I4 L6 M2 C& R0 S% uin.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); * \% ]8 [- {/ V* u$ q6 o7 t) n: \WSACleanup(); 5 y; K- z2 Y5 q+ Z( V3 N6 F6 Rreturn in.s_addr; } */ , X2 i. v% ]+ ?* t1 l/ @// // - w" O8 l q! D; Q( iUINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 LPSOCKINFO psockinfo; q! ]/ h0 \' n/ f- GSOCKET sourcesock,destsock; char data[MAXDATALEN]; long eventid; int datalen; psockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; ( G+ v6 P0 G& U( [* gdestsock=psockinfo->destsock; 3 @0 {- U( b6 x! L% WTRACE("deail recive thread ok!\r\n"); 1 C/ s# v8 A$ D6 D% Zwhile(true){ eventid=GetSocketEventId(sourcesock); switch(eventid){ , J: L8 i" P4 g& c( s7 T" Q$ |case FD_CLOSE: TRACE("s fdclosed\r\n"); closesocket(destsock); return 1; break; - ]0 |+ s* `1 L" i! ~( Edefault:break; } eventid=GetSocketEventId(destsock); switch(eventid){ 5 M' W: ?; F, Rcase FD_CLOSE: closesocket(sourcesock); TRACE("d fdclosed\r\n"); return 1; / m; L) |2 A; m, w7 tbreak; default:break; " h2 g+ F) c, [- P9 p} datalen=recv(sourcesock,data,sizeof(data),0); $ |. F( y8 K0 |if(datalen==0){ 8 a" ?6 J7 `& `* L: Vclosesocket(sourcesock); closesocket(destsock); TRACE("s fdclosed\r\n"); break; ' `/ S: K# d" x% ]# U, B} ; J1 A' V3 c y+ y% ~if(datalen>0){ 8 l6 X8 i$ a6 T4 Swhile(!send(destsock,data,datalen,0)); ! Q; p5 B: e8 G8 m3 n, ^} ; I8 F2 `3 W- u9 K2 y2 ?( nSleep(1); , v. B# Z" E) I* v, c* F0 Y} 8 P3 e( S- F0 {) Ureturn 1; : d. R" S3 }& @6 ]7 |, C& z9 X} // 7 W& q% _7 x& D1 xUINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 ' f7 k5 h3 i% `0 V8 t3 Q" _/ ]LPSOCKINFO psockinfo; & L' I" I: q2 ^( YSOCKET sourcesock,destsock; char data[MAXDATALEN]; long eventid; - r! {+ L( u j7 uint datalen; : o. g4 @6 ~; P/ Hpsockinfo=(LPSOCKINFO)info; 1 B! _& T: a" [$ e. { N7 esourcesock=psockinfo->sourcesock; 4 j$ L t" H' Y( h5 [9 Hdestsock=psockinfo->destsock; TRACE("deail send thread ok!\r\n"); _3 ?3 v8 ~; swhile(true){ eventid=GetSocketEventId(sourcesock); switch(eventid){ case FD_CLOSE: ! p5 V' _; R) ^' Z7 d4 c. d& f& YTRACE("s fdclosed\r\n"); " b, r& J4 C# \2 j Y, yclosesocket(destsock); : I6 D8 w0 [4 O. R% O8 C+ |8 X+ Ureturn 1; break; $ R z% H8 G( Udefault:break; } % V! S8 h4 C4 c* A2 D3 k- ?/ zeventid=GetSocketEventId(destsock); switch(eventid){ 3 `7 k5 }/ u Ocase FD_CLOSE: closesocket(sourcesock); 0 o" {/ y2 }; {+ e/ ~TRACE("d fdclosed\r\n"); return 1; ( f" L- f' l- j1 ]* Nbreak; ! f1 x. }. T8 tdefault:break; } datalen=recv(destsock,data,sizeof(data),0); # T* I8 c) d: k# b# v W+ Y4 kif(datalen==0){ . c/ w0 u. f& bclosesocket(sourcesock); - O8 E- t5 i: b4 cclosesocket(destsock); TRACE("d fdclosed\r\n"); & s+ E; }7 O3 Jbreak; ) O+ u! p4 z J" N* l} if(datalen>0){ while(!send(sourcesock,data,datalen,0)); } / Y! h; y; C& ?. X7 k7 _" iSleep(1); " [: t$ [2 B$ v( b} return 1; . t+ F8 N: k- z) j# i, e' k} ; b- r5 Q6 ~' z& ? ?1 i// ; t- \" r' G: C. {2 G0 v! p// UINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 0 W5 l# U9 j5 t+ n: o% y! dLPCLIENTINFO pclientinfo; + Q/ C4 k- R' _/ `4 z! DSOCKET connectsock,clientsock; sockaddr_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; pclientinfo=(LPCLIENTINFO)info; $ M* j* k8 _0 z' T) o7 vclientsock=pclientinfo->clientsock; / U' ]/ Q9 ]( h2 f( |$ T# u% ^1 f# _ZeroMemory((void *)data,sizeof(data)); ! F2 P- ]; |% n9 x% Ndatalen=recv(clientsock,data,sizeof(data),0); / G$ ]4 @% X% M9 k" uif(datalen<=0){ 5 |# d( \( R, n$ Q$ Sclosesocket(clientsock); . w# {/ r* s' N4 f9 Ereturn 0; ) g4 O, N. d1 x, B! E$ X; ]

韩冰

} HttpString.Format("%s",data); 5 t5 h- Y% T. n# IUrlString=HttpString; # t4 _/ G: s8 p3 FTRACE("get http string:\r\n"); TRACE(HttpString); 4 h" j! L# j; @0 qindex_start=HttpString.Find("Host: ",0); //寻找url标记 * ?% y; \8 p2 @) J: h1 R' S8 ^if(index_start<=0){ closesocket(clientsock); + l6 S) f( {" treturn 0; } & W H( L9 O* B; [index_end=HttpString.Find("\r\n",index_start); j$ e- |' z# _6 k+ O5 Hif(index_end<=0){ closesocket(clientsock); 5 \+ E$ d. m1 ~- h) d2 H. @- Yreturn 0; ) s+ l* H% {; m) p} UrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 TRACE("\r\n get url:"); ! h" w2 {% ~+ I7 TTRACE(UrlString); wsprintf(url,"%s",UrlString); }# a. u3 l8 |5 k2 N- | ystrcpy(temp,url); strcat(temp,":"); datalen=strlen(temp); * o/ ~3 U# ?# ]2 }, p$ i* j zif(HttpString.Find("GET",0)==0){ //判断get命令，并处理 index_start=HttpString.Find(temp,0); * F1 \! c0 V2 t9 Gstrcpy(httpurl,"http://"); 4 j* b* O+ `9 |( | {if(index_start>0){ index_end=HttpString.Find("/",index_start); if(index_end<=0){ & \2 I7 t9 d& p. Sclosesocket(clientsock); ]/ l p" t! Lreturn 0; 5 O" k. v1 {: R2 G4 Q$ w) H6 `( I} PortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); port=atoi(PortString); strcat(httpurl,temp); itoa(port,portnum,sizeof(portnum)); strcat(httpurl,portnum); strcat(httpurl,"/"); ! h8 y6 U" t/ t1 F/ V8 z} 7 h% y2 R; @; x' j0 Q- delse{ 0 N) s% ]$ M9 D0 cport=80; strcat(httpurl,url); strcat(httpurl,"/"); } TRACE("get http url:%s\r\n",httpurl); HttpString.Replace(httpurl,"/"); + X& f4 G" C1 P$ t2 \HttpString.Replace("Proxy-",""); HttpString.Replace("HTTP/1.0","HTTP/1.1"); } . L x7 g0 M. Melse if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 index_start=HttpString.Find(temp,0); if(index_start>0){ index_end=HttpString.Find(" ",index_start); if(index_end<=0){ closesocket(clientsock); " Y0 Z; ~% B C8 n" ?return 0; & X( Q' e& W: G} % u, ^6 d: q2 K. X/ I8 f9 s6 XPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); 9 p& M1 ^* a. R. @4 xport=atoi(PortString); } 4 C3 Y ^) u5 B- Eelse{ closesocket(clientsock); , o1 V* }/ \! }' ?! lreturn 0; } 9 G, ^2 q- H* A3 O& c* \} TRACE("get new http string:\r\n"); TRACE(HttpString); % k, e' P; d* C5 [4 {remotesock_addr.sin_family=AF_INET; 3 }1 Y) V1 [& G7 {2 hremotesock_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){ //连接远程主机 closesocket(clientsock); ; Y) a0 t* c B+ s4 mreturn 0; } - I, }! j" e" t4 y/ ~ cTRACE("\r\nconnect to remote ip ok\r\n"); ZeroMemory((void *)data,sizeof(data)); 1 X d9 d# a# twsprintf(data,"%s",HttpString); datalen=strlen(data); 5 O1 s Y2 A4 e; Tif(HttpString.Find("CONNECT",0)<0) while(!send(connectsock,data,datalen,0)); else{ $ p, \3 ^; b) l. q, ^$ Ystrcpy(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)); - f$ B1 ?* z: |) M3 V9 N' v} httpsockinfo.sourcesock=clientsock; 9 G1 k. |4 s0 |, U- I7 Y ehttpsockinfo.destsock=connectsock; AfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 # `( ]9 z7 j/ _3 `8 s6 D& z' ?4 q9 zAfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // Sleep(100); , Z. u7 C- B% G) Y! \6 Rreturn 1; } // 1 k8 X5 |7 a: ~! l4 D' J) _UINT StartHttpProxy(LPVOID info){ //端口监听线程 SOCKET NewSock; int socklen; ) h/ a5 r9 u+ e) G, A: ~; zsockaddr_in serversock,remotesock_addr; ( Q$ D" |2 e. x3 i! n: V Zserversock.sin_family=AF_INET; 4 O3 [7 O r6 N! U K2 c- `serversock.sin_addr.S_un.S_addr=INADDR_ANY; & t& g& U: o8 \ x$ u* Iserversock.sin_port=htons(HttpListenPort); ' N4 c! h! g0 B% N/ [ mHttpListenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); . c0 B4 O' t0 {8 I a( Kif(HttpListenSock==INVALID_SOCKET) return 0; if(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); ; V' e& X! i$ n. l' l7 A% z+ xTRACE("waitting ok...\r\n"); 3 _) i. v* b3 x0 H% Pif(NewSock==INVALID_SOCKET){ Sleep(1); continue; } ' Y; i0 W% u1 F* i: [ZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); 6 j7 n& c4 u7 e6 E' K3 x$ jHttpClientInfo.clientsock=NewSock; HttpClientInfo.clientsock_addr=remotesock_addr; & H8 q% |: v+ B# f6 a. l3 y0 p6 l. { YTRACE("start proxy thread\r\n"); ( n# r5 F0 L9 H& A' m9 TAfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); 7 }8 U8 S N5 ]Sleep(100); ' m1 U7 S# N Z) @6 `} 0 g4 B3 E" r- X/ Q+ lreturn 1; } : ]* G3 R l$ Q6 M2 l// F/ v; J! ~( V* p- Y3 f; J YCHTTPPROXY::CHTTPPROXY() { ) C: c/ g- G; \# |& ]) w6 uWSADATA WsaData; % [3 H) v* A1 I1 FWORD wsaVer; wsaVer=MAKEWORD(2,0); " C: O P) c, F* C7 p: \WsaStartupOk=false; 7 W' m2 e, J$ m" Mif(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; }

CHTTPPROXY::~CHTTPPROXY() ' A6 u3 q8 h( |8 ~& D. y1 W{ if(WsaStartupOk){ ' S' S- X5 |0 q8 z3 I: xWSACleanup(); } % C) Q6 |# s! G0 X" K+ `" u+ W}

int CHTTPPROXY::StartProxy(int listenport) { * M7 N: ~' Q9 @2 X+ s( r$ H8 p9 {HttpListenPort=listenport; AfxBeginThread(StartHttpProxy,(LPVOID)NULL); 5 e/ s' a, v% j8 f; E3 f) ]; X0 Z% R2 lreturn 1; 9 k! I0 N$ l4 D d}

/*========================================================================= FILE:HTTPPROXY.h ==========================================================================*/ class CHTTPPROXY / @$ E5 s0 T) C" L( g! q( O0 H{ public: 1 x( \+ f0 g$ R5 Q$ H" v. v4 sint StartProxy(int listenport); bool WsaStartupOk; 1 g- L4 \3 C1 ]) l' `/ c0 lCHTTPPROXY(); 6 C& ~: j: p1 Z' |1 s+ g5 ?; Qvirtual ~CHTTPPROXY();

};

/*========================================================================= - G5 J. ^ T% Y( {0 }FILE:stdafx.h ==========================================================================*/ #include 3 Q( y# G8 W7 R# {& @#include #include 0 d) ?1 B3 G, q2 C6 N, P$ V#include

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

本代码在win2k和vc6.0下编译成功~~ 用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: 6 u9 v9 J# j. j2 n: S& ECHTTPPROXY httpproxy; 9 ?/ x& _4 k3 X4 x: d1 }' bhttpproxy.StartProxy(7890); 有问题mailto me！