打造自己的HTTP代理服务器

韩冰

/*========================================================================= FILE:HTTPPROXY.cpp HTTP Proxy v1.0 8 }* d# S. \% k0 y3 ppowered by shadow 2004/11/14 3 i$ Y8 V; k7 ^* Y( X7 pmy web:http://www.codehome.6600.org * E5 {, N/ [% r) d1 u7 W& [0 ^QQ:176017352 / L6 Q) v) n) j$ R& m( T A1 A$ v请先阅读有关http代理协议的相关资料,转载请著明出处 ==========================================================================*/ * F. |8 q& X2 W6 D8 p1 a$ ~

#include "stdafx.h" 8 l# X. _/ A* Y J& m/ |3 m& Z4 i#include "HTTPPROXY.h" 7 R; Y* @ Q( M( `#define FD_NOEVENT 0 & q( v, p8 U; T" N9 U// 6 m. }# K& ~: [#define BACKUP 100 : R+ V6 J% |- ]$ ?! _#define MAXDATALEN 65535 + a3 V2 V4 P5 F. aint HttpListenPort; 9 o6 }. [3 ` M, A# zSOCKET HttpListenSock; 6 y8 M! K/ U# w+ W% G// typedef struct _CLIENTINFO{ SOCKET clientsock; SOCKET udpsock; sockaddr_in clientsock_addr; }CLIENTINFO,*LPCLIENTINFO; 6 K; ]8 }0 z6 c$ sCLIENTINFO HttpClientInfo; % ]5 s# u+ m9 k// 7 s5 b9 S6 ?/ R6 H' V+ `typedef struct _SOCKINFO{ # Q' ?, q* R0 S" ^7 z& T& SSOCKET sourcesock; 3 L+ L( }1 d& F* CSOCKET destsock; ; o+ x" b) ?. d& H3 j7 D}SOCKINFO,*LPSOCKINFO; 5 w* d6 `+ V" S+ F6 `, d' I* ESOCKINFO httpsockinfo; // extern long GetSocketEventId(SOCKET remotesock); extern unsigned long GetLocalIp(); 2 b$ c/ ]4 g5 R {9 ~0 aextern unsigned long GetDomainIp(char domainname[250]); ' O1 B) y& C3 u2 N! R/* long GetSocketEventId(SOCKET remotesock){ long EventId; HANDLE hevent; hevent=CreateEvent(NULL,0,0,0); 5 l- n; W% E% ]- n. PWSANETWORKEVENTS socket_events; EventId=FD_NOEVENT; ; n) b0 ~6 e' Dif(WSAEventSelect(remotesock,hevent,FD_ACCEPT|FD_CONNECT|FD_READ|FD_WRITE|FD_CLOSE)==SOCKET_ERROR) return EventId; WSAEnumNetworkEvents(remotesock,hevent,&socket_events); " F! k6 X7 r5 r V# J8 Sif(socket_events.lNetworkEvents!=0){ switch(socket_events.lNetworkEvents){ case FD_ACCEPT:EventId=FD_ACCEPT;break; + w s2 y6 D7 O9 X [+ [case FD_CONNECT:EventId=FD_CONNECT;break; case FD_READ:EventId=FD_READ;break; case FD_WRITE:EventId=FD_WRITE;break; 2 T: u; x6 m) f+ ~& [case FD_CLOSE:EventId=FD_CLOSE;break; ! l p" H. O' K/ S4 }) c9 Z9 }case FD_OOB:EventId=FD_OOB;break; 7 [* @) v" V( t L' ^0 K9 ]& X' adefault:EventId=FD_NOEVENT;break; } } else EventId=FD_NOEVENT; return EventId; } // , h# z4 Q) [2 sunsigned long GetLocalIp() + C* J$ B: Z6 ^6 O C{ 1 j [" S+ I( s$ @. s) L' Bchar IP[MAX_PATH],*ip; ) X& j) _, X1 G z" `char pc_name[80]; struct in_addr in; ; b) ~1 g6 l# A0 M+ H1 {struct hostent *host; WORD wVersionRequested; WSADATA wsaData; 0 R. c& \; M b* J( ?. owVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; : `7 X& ]# W0 w" J: f: S5 h9 Y9 yif(gethostname(pc_name,80)==SOCKET_ERROR){ WSACleanup(); return 0; . T' w: ]5 \9 t( i% l} 9 [9 r. F' n0 }; T9 N; j) \- ]if(!(host=gethostbyname(pc_name))){ , \( s' v) w4 {4 u( y- AWSACleanup(); return 0; 1 J- o! d2 ^7 [/ N} # e: S" J( h9 L+ l+ ]' G( Zin.s_addr=*((unsigned long *)host->h_addr_list[0]); ( T2 X; X3 o: e8 E! R9 m/ Cstrcpy(ip,inet_ntoa(in)); & k8 `9 R2 f" V7 f% gWSACleanup(); , ?: c! _% ]3 u# z% }return in.s_addr; 4 z5 W/ n$ L* c% K8 O} ) ?* _, U r9 M; N: L. u& O// unsigned long GetDomainIp(char domainname[250]) u# J# U" p, L+ ? w" q- s{ char IP[MAX_PATH],*ip; 9 m8 ?) W; Z* b z+ ystruct in_addr in; # V- ? W5 w' l: s4 s" qstruct hostent *host; 4 f3 W5 {4 p+ b& Q3 k8 B, \WORD wVersionRequested; # }% h% {: u, r. S7 sWSADATA wsaData; 3 h& F& N( Z! A8 X' V R6 }wVersionRequested=MAKEWORD(2,0); ip=IP; / r+ g5 @( {& S* w" Q/ Jstrcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; 2 i1 ~. U* L! R# t: S- |5 kif(!(host=gethostbyname(domainname))){ 3 O6 w1 h# a. ^) b' P# nWSACleanup(); 5 @7 ?8 r" {4 o. n+ c$ Xreturn 0; 2 u& I5 E; o& M* |, a$ c} in.s_addr=*((unsigned long *)host->h_addr_list[0]); 9 b2 T6 C' P# X2 h- e7 a1 l2 L: [7 ]+ ]strcpy(ip,inet_ntoa(in)); WSACleanup(); . E* E0 p H# _( o( V- kreturn in.s_addr; ( W$ Z; A- q5 B4 L3 y) t} */ 9 {/ ]2 N9 I+ O% x$ V- n, u+ [& S// . p2 A" @+ I" c& Q8 r// UINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 LPSOCKINFO psockinfo; SOCKET sourcesock,destsock; 3 Y) e" C) m% z0 {: u3 E/ |- vchar data[MAXDATALEN]; ) }" y, N5 w: G- S; Z: \long eventid; int datalen; psockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; : s% H8 P2 a5 H1 ?# \TRACE("deail recive thread ok!\r\n"); while(true){ ; j6 @- Z5 l- Y X9 `eventid=GetSocketEventId(sourcesock); switch(eventid){ case FD_CLOSE: TRACE("s fdclosed\r\n"); closesocket(destsock); return 1; 4 [8 n* c& b! r6 l* ?. ?0 ~break; default:break; } eventid=GetSocketEventId(destsock); switch(eventid){ case FD_CLOSE: closesocket(sourcesock); TRACE("d fdclosed\r\n"); return 1; break; default:break; } datalen=recv(sourcesock,data,sizeof(data),0); if(datalen==0){ 3 O/ Y, N/ L! o: |* J, R7 tclosesocket(sourcesock); 3 | ^, d5 H& I' @' P% U, ?closesocket(destsock); 5 u: V3 Y, C6 P3 @$ r9 bTRACE("s fdclosed\r\n"); break; 4 y4 g% f4 F6 [! R! A3 l} 8 O# O: \6 c4 Y8 D# ?6 N5 yif(datalen>0){ while(!send(destsock,data,datalen,0)); % M: I* q) p% q4 e G, W; z} ) S+ f" |, g: G) ^2 c1 XSleep(1); H" q0 s' P2 a" s0 \5 R9 x( i} - X5 W. j6 Y- s1 dreturn 1; } // UINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 7 J: a, d* P9 U1 ~/ o! z' ?4 gLPSOCKINFO psockinfo; SOCKET sourcesock,destsock; ' o P, D0 Q6 g8 z# K* z/ `+ z, D8 kchar data[MAXDATALEN]; " [, C. T1 v9 G6 Q/ j, Y' A- G6 ?" hlong eventid; int datalen; psockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; $ _( p! Q8 o9 ?6 i u4 @) DTRACE("deail send thread ok!\r\n"); 2 F* v0 i$ [% C7 X# Dwhile(true){ + x6 J, M) j+ @, }9 geventid=GetSocketEventId(sourcesock); + c" S$ D( Z$ N" B, v- T4 G" y0 U* Cswitch(eventid){ case FD_CLOSE: " B) K) O m. {* n9 i( {! p5 LTRACE("s fdclosed\r\n"); closesocket(destsock); return 1; break; default:break; } 5 J7 Y% S2 z2 u; [( ^, p& M! Aeventid=GetSocketEventId(destsock); switch(eventid){ 8 Q& Z- [+ g2 n3 L& Zcase FD_CLOSE: 8 @: L" C; J0 k: dclosesocket(sourcesock); " t" d8 X$ L) h2 J& a- Z6 pTRACE("d fdclosed\r\n"); : I2 b3 D" O, ]return 1; * z- h, X% q6 g3 M. N% Vbreak; : c1 J2 D, B: w: D! idefault:break; } datalen=recv(destsock,data,sizeof(data),0); 4 O a( g8 q% A0 C# {if(datalen==0){ ) M+ \* U2 w2 g# e/ L# Bclosesocket(sourcesock); ' f P" D& l4 X8 ?closesocket(destsock); % |5 ~$ g7 n/ F" KTRACE("d fdclosed\r\n"); break; " \; ?6 i; Y# w3 B; l! a( D} if(datalen>0){ 5 B- M& v& D7 t5 Vwhile(!send(sourcesock,data,datalen,0)); } & f k5 l l) g5 kSleep(1); : h X# b$ G' G5 G: g} return 1; ) y9 X9 } t' f' l8 `/ ]- x} // 8 U0 v# \+ `% S; d. l// UINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 ( ^. A! W' d1 W' ILPCLIENTINFO pclientinfo; SOCKET 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; 1 V2 L) }5 l/ _% L9 [# J0 ~clientsock=pclientinfo->clientsock; ZeroMemory((void *)data,sizeof(data)); datalen=recv(clientsock,data,sizeof(data),0); / G3 Y* I1 L6 n9 c% Yif(datalen<=0){ closesocket(clientsock); $ v- l' m V2 l' N' I8 Ereturn 0; & L5 G( W* F1 {% j7 Q

韩冰

} 4 L; r6 Y: U* |/ j" c4 ~HttpString.Format("%s",data); UrlString=HttpString; $ u/ F0 h5 G* z* D/ dTRACE("get http string:\r\n"); TRACE(HttpString); 0 w1 J4 L* m- Q3 jindex_start=HttpString.Find("Host: ",0); //寻找url标记 9 N( Z4 `4 n. Dif(index_start<=0){ closesocket(clientsock); 8 X4 w9 Q4 j: w) T' i7 h, zreturn 0; % i3 q4 b$ I1 S6 t0 H! [} $ `: N+ e v7 J# X$ y- \index_end=HttpString.Find("\r\n",index_start); if(index_end<=0){ & `6 a0 l6 U& y' y: J7 I/ |closesocket(clientsock); . ^# m( u9 z6 L7 ?/ d( \return 0; - t; H- I. |- k' q6 N0 O, h- `9 v} UrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 # U5 |! O7 ~; B" x0 [0 A4 J4 HTRACE("\r\n get url:"); 2 _/ K: h# F4 Z$ T; v0 s$ t2 d1 ETRACE(UrlString); i' I* S8 x L j1 V) }! V7 E' R6 @wsprintf(url,"%s",UrlString); strcpy(temp,url); strcat(temp,":"); ( p- v; r0 Z* ldatalen=strlen(temp); ; s1 t+ X- X* O+ Yif(HttpString.Find("GET",0)==0){ //判断get命令，并处理 index_start=HttpString.Find(temp,0); strcpy(httpurl,"http://"); 5 o$ p- ], D z3 }# R7 [( ~if(index_start>0){ T7 ?" I: c! J3 _, D" W- b, mindex_end=HttpString.Find("/",index_start); if(index_end<=0){ ' k: z6 f+ H: K. Eclosesocket(clientsock); return 0; + B i5 `" V" @1 R+ c} $ }' I0 [" c5 z8 A6 s1 Z. `1 VPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); port=atoi(PortString); * c% n6 \% ` ^$ a. c+ \strcat(httpurl,temp); ) V0 k- v- @+ s4 N* Yitoa(port,portnum,sizeof(portnum)); strcat(httpurl,portnum); strcat(httpurl,"/"); 5 n9 p7 h R- _: ]; W! _} else{ 7 |$ I3 g0 y5 Q: Gport=80; * l$ q7 C# z& Pstrcat(httpurl,url); : _- v# K0 N' Q% Astrcat(httpurl,"/"); } * d4 W" M* ]5 O4 Q, S1 V5 d$ l4 S+ NTRACE("get http url:%s\r\n",httpurl); 4 ?0 `& ]7 p" D* RHttpString.Replace(httpurl,"/"); HttpString.Replace("Proxy-",""); 3 @. t; ~: T' Z" vHttpString.Replace("HTTP/1.0","HTTP/1.1"); : \! _7 v$ g1 O) a z4 q} 0 Y' ^1 x8 a! D( V, Z; e9 F# Xelse if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 8 a5 ?4 ]1 D0 c3 `1 mindex_start=HttpString.Find(temp,0); if(index_start>0){ index_end=HttpString.Find(" ",index_start); if(index_end<=0){ closesocket(clientsock); & Z3 D/ q' W" nreturn 0; 2 S* w( a; `5 x3 i} . r& G$ R, P+ I) `% @PortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); port=atoi(PortString); } else{ closesocket(clientsock); return 0; 4 ]% R! B+ u/ z} } ( s2 i/ \2 q" p, @TRACE("get new http string:\r\n"); . _$ ]9 U: T+ g8 o" p$ O6 Q7 MTRACE(HttpString); ; Z6 Q0 x( R# r0 m" e8 x; \6 sremotesock_addr.sin_family=AF_INET; ) f; R7 u- C7 Q9 k/ d. i0 wremotesock_addr.sin_port=htons(port); remotesock_addr.sin_addr.S_un.S_addr=GetDomainIp(url); + K, y8 y1 f& _: G! Mconnectsock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); ) p$ b9 ?$ Q" A. [3 R2 ]$ h, cif(connect(connectsock,(const sockaddr *)&remotesock_addr,sizeof(remotesock_addr))==INVALID_SOCKET){ //连接远程主机 - ^) L R+ E2 g- g; l; Oclosesocket(clientsock); return 0; 1 S+ X% {: |& B+ S4 Y8 V} TRACE("\r\nconnect to remote ip ok\r\n"); ZeroMemory((void *)data,sizeof(data)); wsprintf(data,"%s",HttpString); ! I B3 w' Z, e2 C5 Z! ddatalen=strlen(data); ' ^7 ^+ K; W* l! d: s" ]+ N6 G( Uif(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"); datalen=strlen(data); 8 a0 P9 R; j0 h4 z4 `6 ?while(!send(clientsock,data,datalen,0)); } 2 |! G8 t9 L; K5 U3 }' f lhttpsockinfo.sourcesock=clientsock; # Y- o9 S8 r/ ~httpsockinfo.destsock=connectsock; 2 i% f3 U* l8 N5 w$ fAfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 AfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // $ c- i$ u" T1 c5 \" S( CSleep(100); % J# S* O1 G; b5 treturn 1; + ]& R* m" A+ l6 p$ }} + c1 K5 Z3 U( r! s, M// UINT StartHttpProxy(LPVOID info){ //端口监听线程 / u( e7 j% ]& y8 NSOCKET NewSock; int socklen; 3 U6 T: \# B5 x; N6 F' d- osockaddr_in serversock,remotesock_addr; 9 m) f* l C, P& o: b6 p# Z" `7 e j, Bserversock.sin_family=AF_INET; serversock.sin_addr.S_un.S_addr=INADDR_ANY; serversock.sin_port=htons(HttpListenPort); HttpListenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); / G0 A) {/ y* }4 T7 k! q, s& Vif(HttpListenSock==INVALID_SOCKET) return 0; " B. x9 }5 {8 ]. Xif(bind(HttpListenSock,(const sockaddr *)&serversock,sizeof(serversock))==SOCKET_ERROR) return 0; 9 I6 y: T* O2 L& }5 nlisten(HttpListenSock,BACKUP); . h& u, j2 N* W' J' a8 ^socklen=sizeof(remotesock_addr); TRACE("start http proxy thread while\r\n"); " Y& y$ _9 w* j: L2 n( Bwhile(true){ NewSock=accept(HttpListenSock,(sockaddr *)&remotesock_addr,&socklen); " B2 Z; | \4 x" `$ h+ YTRACE("waitting ok...\r\n"); if(NewSock==INVALID_SOCKET){ 8 ~& w$ _" x9 s- dSleep(1); continue; } 9 W: n0 x1 [2 ~' _0 `ZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); HttpClientInfo.clientsock=NewSock; HttpClientInfo.clientsock_addr=remotesock_addr; TRACE("start proxy thread\r\n"); 8 {+ S8 s8 U. ]( jAfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); # E5 N+ V. P3 O) ~- f6 [; ASleep(100); 1 R+ G E& i+ m' f/ v v} " Q5 g8 o2 i n& O& q3 Rreturn 1; 1 a; f q( R p' r$ t} * D* g2 S2 F$ W1 {3 \# Y// CHTTPPROXY::CHTTPPROXY() { WSADATA WsaData; WORD wsaVer; wsaVer=MAKEWORD(2,0); . B5 h0 x% @4 m. i3 x8 A, B! L4 a3 QWsaStartupOk=false; if(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; }

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

int CHTTPPROXY::StartProxy(int listenport) { HttpListenPort=listenport; AfxBeginThread(StartHttpProxy,(LPVOID)NULL); & L$ ?" S2 @4 Z A7 l) Dreturn 1; }

: Q$ W" s8 Z9 H6 I' a/*========================================================================= FILE:HTTPPROXY.h ==========================================================================*/ class CHTTPPROXY ) S$ @9 v9 y( w4 G+ f/ E{ 0 O. J/ ?% ~. b# Ppublic: % U+ E3 o( b. ~4 v/ w! ]int StartProxy(int listenport); bool WsaStartupOk; CHTTPPROXY(); virtual ~CHTTPPROXY();

};

/*========================================================================= FILE:stdafx.h 0 X) w$ [. h1 U==========================================================================*/ #include #include % H$ o' E' v# G7 G# s: N1 p( A#include #include

1 ~, S8 d# o9 t8 F! Q/ Q7 y注:不要忘了在link选项中添加wsock32.lib和ws2_32.lib，或在文件前部加上如下语句： ! \9 q8 M" ]0 T# B7 ]#paragma comment(lib,"wsock32.lib") + V3 Y' ?& K. `* X0 @7 r; a#paragma comment(lib,"ws2_32.lib")

本代码在win2k和vc6.0下编译成功~~ , `$ h! F* V& y5 \7 \用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: CHTTPPROXY httpproxy; httpproxy.StartProxy(7890); u7 Z9 k0 E( F7 s2 x有问题mailto me！ C" K R9 | a& c' g; O