打造自己的HTTP代理服务器

韩冰

/*========================================================================= 0 [ C4 O% C# E9 i; B# z" N h3 U1 O WFILE:HTTPPROXY.cpp HTTP Proxy v1.0 powered by shadow 2004/11/14 my web:http://www.codehome.6600.org QQ:176017352 请先阅读有关http代理协议的相关资料,转载请著明出处 5 S, Y6 y S; t Z==========================================================================*/ / m L* C c- \ n+ ^! j

#include "stdafx.h" #include "HTTPPROXY.h" #define FD_NOEVENT 0 % _9 A& x3 q6 H- ^, p// ! J3 K3 K7 H" J+ l#define BACKUP 100 " M9 B; m' o* D( p' R) v#define MAXDATALEN 65535 int HttpListenPort; SOCKET HttpListenSock; 7 d" |5 o# @/ _8 S2 P// typedef struct _CLIENTINFO{ ) D. K; q* r( m' P* E9 x4 m g' nSOCKET clientsock; 3 o- T3 f+ i& [, A% ASOCKET udpsock; 6 m6 D1 `" w+ t+ G& v1 @! gsockaddr_in clientsock_addr; $ t. ]" y6 v0 i/ \}CLIENTINFO,*LPCLIENTINFO; CLIENTINFO HttpClientInfo; // typedef struct _SOCKINFO{ " j, M- ~" `1 TSOCKET sourcesock; SOCKET destsock; " l6 P A! L( }& F- F/ Z}SOCKINFO,*LPSOCKINFO; SOCKINFO httpsockinfo; & }4 ^" y* j6 g. M" D4 Y! e// extern long GetSocketEventId(SOCKET remotesock); extern unsigned long GetLocalIp(); # `, e9 J( \) |( Bextern unsigned long GetDomainIp(char domainname[250]); /* + \, k2 K+ I( V1 S. {4 ~long GetSocketEventId(SOCKET remotesock){ & b Z+ x, u! g* W" J! [long EventId; # j3 |8 m* R5 F; F) ~6 h- A& n1 _HANDLE hevent; hevent=CreateEvent(NULL,0,0,0); % ~4 e) E# t; D; h) CWSANETWORKEVENTS socket_events; 4 N" {" f( A& q. 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); if(socket_events.lNetworkEvents!=0){ switch(socket_events.lNetworkEvents){ $ O1 @! Q6 I# K1 k7 e. _case FD_ACCEPT:EventId=FD_ACCEPT;break; 1 M5 `; Z; c' V) b4 {9 {4 Ecase 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; " @5 ~) M- P8 E) Mcase FD_OOB:EventId=FD_OOB;break; default:EventId=FD_NOEVENT;break; } ' Y: e$ F/ O# r* x2 f( L} ' H N, X* S5 u" Eelse EventId=FD_NOEVENT; return EventId; } 8 i, }$ v! b- Z// 8 t' U% R3 b7 Munsigned long GetLocalIp() % v$ D- @4 t/ g& `" t8 e1 R{ char IP[MAX_PATH],*ip; char pc_name[80]; ( @& Q: K2 O" I( p ~% h) i. M; _struct in_addr in; 9 Q( n3 U) @+ N. b: qstruct hostent *host; 1 g# s+ y+ ~1 n$ O; {# g6 Q3 bWORD wVersionRequested; & D) r& ]& u# |% y1 CWSADATA wsaData; 6 x; @/ Z, ]2 E( i' S/ ]' ZwVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; if(gethostname(pc_name,80)==SOCKET_ERROR){ + E g8 u0 ~3 J3 FWSACleanup(); 0 [/ B, I ?0 S. I0 L9 q3 ?) dreturn 0; } 0 V, m, D& F1 L, @3 {if(!(host=gethostbyname(pc_name))){ , _% m* A, u! c9 d9 R, YWSACleanup(); return 0; + o9 r* I0 x/ O1 S% t! F} in.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); , M' C5 c; b/ I( MWSACleanup(); " s4 [6 a6 }, }( t Yreturn in.s_addr; + k4 u( J9 H6 T p# {} // 1 w+ b2 u. n0 j R9 R: p; \- m2 \; Dunsigned long GetDomainIp(char domainname[250]) " v# {- B- N/ m& X! G+ Z{ $ y: A m2 T' S) [' Xchar IP[MAX_PATH],*ip; . e, ?* D, z0 q1 }/ wstruct in_addr in; + E: ]( }# D0 Z& g Z4 xstruct hostent *host; % k2 t( G2 ~& f+ UWORD wVersionRequested; 6 N! x" j& p/ Y& e, a9 @" `WSADATA wsaData; 1 v% m8 x. r4 X7 |0 X1 p1 vwVersionRequested=MAKEWORD(2,0); ip=IP; 4 v* }+ G3 M6 w/ i+ @' Rstrcpy(ip,"Ip not get!"); $ M0 K# o: D( _# }* Oif(WSAStartup(wVersionRequested,&wsaData)) return 0; , V/ I7 e1 N6 _1 p/ W7 c4 r& dif(!(host=gethostbyname(domainname))){ WSACleanup(); 7 T! x' A: ?# n8 g/ [return 0; - C& g+ @" ~5 m, K8 ~} ' {# Q- d/ Q7 B3 Min.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); 9 f1 J* @! w( zWSACleanup(); return in.s_addr; & u" u/ J3 H. v/ }3 \! E/ _} 5 Y/ p8 G& O: R$ L& }% S8 _ T3 ?* c*/ // // : b" [4 J" `. DUINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 LPSOCKINFO psockinfo; / i# X' t, V: C( B9 V# J+ HSOCKET sourcesock,destsock; char data[MAXDATALEN]; long eventid; int datalen; ( D" f! j! o! x; }2 v1 [# ~psockinfo=(LPSOCKINFO)info; 2 _8 G: A; }) u- Vsourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; / d& k: A5 G0 c: TTRACE("deail recive thread ok!\r\n"); while(true){ b, b* s6 M/ s! N9 Eeventid=GetSocketEventId(sourcesock); ) P! u' {3 s1 r+ v5 ]switch(eventid){ 6 d9 ? x: Y1 m/ d. N* {# Tcase FD_CLOSE: TRACE("s fdclosed\r\n"); 7 X( _: U1 y& A5 b$ T, {closesocket(destsock); return 1; break; default:break; } 5 [; Y4 s( G$ n; Geventid=GetSocketEventId(destsock); : S6 O0 Z) }: B. Oswitch(eventid){ case FD_CLOSE: : v* J- F- T$ L' }( z3 yclosesocket(sourcesock); TRACE("d fdclosed\r\n"); 0 x2 U0 y: J/ G. m6 Sreturn 1; + T4 @5 K9 r( @) ? Y! W% fbreak; 1 T1 I; |, V5 e1 H, [default:break; } ( Q9 X# \* M& X% W7 Rdatalen=recv(sourcesock,data,sizeof(data),0); " r6 |- s* _% {2 k- Bif(datalen==0){ closesocket(sourcesock); 1 ?! K+ \' \5 Z- Hclosesocket(destsock); TRACE("s fdclosed\r\n"); break; , \. H; t4 m! V( S+ M} & e+ P) }: A) f2 ^/ k4 W% cif(datalen>0){ while(!send(destsock,data,datalen,0)); } Sleep(1); , j6 x$ _8 Y0 X# l} 2 R5 l3 H% d! P/ O Preturn 1; } 8 f8 |1 C5 l) _// / d# a( X1 h) ~# F; b4 lUINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 , R. Z5 v9 X2 x( n; y1 Y) Z$ QLPSOCKINFO psockinfo; r5 I. v/ e/ ISOCKET sourcesock,destsock; char data[MAXDATALEN]; 3 g; f A$ X1 R0 H1 R5 M ?long eventid; int datalen; - p+ F7 A' C/ q5 H2 l) e& Vpsockinfo=(LPSOCKINFO)info; sourcesock=psockinfo->sourcesock; destsock=psockinfo->destsock; TRACE("deail send thread ok!\r\n"); - t) K' p$ N9 \. L. k' i7 Swhile(true){ eventid=GetSocketEventId(sourcesock); " g3 t8 M) S3 T* a8 s3 n" M5 gswitch(eventid){ 2 [' f, \5 y$ ~. e' Q1 x" Q: h; Zcase FD_CLOSE: TRACE("s fdclosed\r\n"); : P+ T# u- U4 \2 X! O& Rclosesocket(destsock); return 1; $ O- x I3 Y+ i5 Ebreak; default:break; 9 A& m1 R; N1 L} 0 u/ @7 { Y9 ]: _* L1 xeventid=GetSocketEventId(destsock); 7 o. I2 ?3 r; F! c, kswitch(eventid){ case FD_CLOSE: 7 `2 m6 q7 k9 H( `" [6 l! Oclosesocket(sourcesock); TRACE("d fdclosed\r\n"); return 1; break; default:break; } # { \1 x J- t7 r- ^0 q6 c. M. Odatalen=recv(destsock,data,sizeof(data),0); {, ^& c6 \# sif(datalen==0){ 0 r/ U0 s9 i" o/ o' F1 Zclosesocket(sourcesock); closesocket(destsock); 4 Z2 b4 N) o1 TTRACE("d fdclosed\r\n"); break; 5 k! y2 h- z0 p4 [} if(datalen>0){ while(!send(sourcesock,data,datalen,0)); } Sleep(1); } return 1; } 0 L$ d+ Q" Y& i// // UINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 LPCLIENTINFO pclientinfo; SOCKET connectsock,clientsock; 3 ?8 K2 i7 M m1 |sockaddr_in remotesock_addr; char data[MAXDATALEN],url[250],temp[250],httpurl[250],portnum[10]; 1 `3 Z' ?5 L1 p, n+ ?int datalen,i,index_start,index_end,port; . R J+ m; r8 o* O5 BCString HttpString,UrlString,PortString; pclientinfo=(LPCLIENTINFO)info; ; V" ?1 t+ {2 z- `/ `( w8 `clientsock=pclientinfo->clientsock; 3 R7 T' ]. R: f; [ZeroMemory((void *)data,sizeof(data)); + R$ W% l5 ~% udatalen=recv(clientsock,data,sizeof(data),0); if(datalen<=0){ closesocket(clientsock); 6 B4 @3 t9 W8 X2 ?/ u }return 0; % v+ n9 t+ B \8 `- y

韩冰

} HttpString.Format("%s",data); UrlString=HttpString; TRACE("get http string:\r\n"); & s0 B2 f1 I' [9 E/ dTRACE(HttpString); . |6 z/ M, @2 \index_start=HttpString.Find("Host: ",0); //寻找url标记 " e' [; o: y9 q6 C: ^if(index_start<=0){ - U, {/ F8 ~' H' Xclosesocket(clientsock); return 0; 7 _1 E# q6 F" I: \1 W2 L} index_end=HttpString.Find("\r\n",index_start); ! q9 G. S4 u( ? L9 Z0 G# Tif(index_end<=0){ 2 j5 G) V$ w, O9 H M1 x/ d2 fclosesocket(clientsock); return 0; 4 D4 U+ s/ L6 f! D0 c7 `$ f7 h6 H- x& P} - ~6 \+ {. [2 q: a3 y! u$ GUrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 3 K9 a/ P& ]3 v$ Q/ ZTRACE("\r\n get url:"); - z" b5 }' D# p( W* MTRACE(UrlString); ; R3 w1 H' ^9 D$ N% Swsprintf(url,"%s",UrlString); & ?, C$ `, I: O! s3 U( P- `# Qstrcpy(temp,url); strcat(temp,":"); datalen=strlen(temp); _# c& q% p$ N7 k7 g. Q% eif(HttpString.Find("GET",0)==0){ //判断get命令，并处理 index_start=HttpString.Find(temp,0); # u& j2 t5 R3 I7 nstrcpy(httpurl,"http://"); if(index_start>0){ index_end=HttpString.Find("/",index_start); if(index_end<=0){ # E0 ?1 d9 U' E, r5 H1 Z$ p( gclosesocket(clientsock); ' o ?1 A y7 l& [" Lreturn 0; } PortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); & {* h0 @; z0 y- ~! n, N2 _port=atoi(PortString); strcat(httpurl,temp); itoa(port,portnum,sizeof(portnum)); strcat(httpurl,portnum); strcat(httpurl,"/"); } else{ ) Y4 P6 h4 u1 x" k/ k5 S8 @port=80; * [- {: ]2 {! g/ ^strcat(httpurl,url); " Y6 }! Y4 p/ K- s: T6 |+ y- Kstrcat(httpurl,"/"); } TRACE("get http url:%s\r\n",httpurl); 7 {. N: O' ?' J3 W, hHttpString.Replace(httpurl,"/"); 3 k! I: A4 D) M/ e0 d1 w9 b3 MHttpString.Replace("Proxy-",""); HttpString.Replace("HTTP/1.0","HTTP/1.1"); } $ Q/ o# s. j: d! |' o3 F. _, y9 Yelse if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 7 J' D% Z# ?3 K7 i& E Z. R8 iindex_start=HttpString.Find(temp,0); . q1 R: Q ?9 H+ p m0 tif(index_start>0){ 9 T Q7 \( f7 y6 o1 C1 m+ jindex_end=HttpString.Find(" ",index_start); 1 n8 N( D& s2 D% rif(index_end<=0){ + z6 v, u: k7 b" n( _* h% Xclosesocket(clientsock); return 0; 1 {& M7 r' K7 }1 s* A$ S& p} PortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); & h" M6 S* Q6 }$ J3 ?! Oport=atoi(PortString); 6 X1 x- _" T- m3 f9 _1 ]} ' E1 t6 Y$ _* [8 _else{ closesocket(clientsock); return 0; } } 9 b8 Q* p/ u' v* S: y% `4 G4 ~$ g/ }5 \3 ]TRACE("get new http string:\r\n"); 8 [ V9 G9 _) i- ?% ^) wTRACE(HttpString); 3 T Z) X" U: s" y/ }remotesock_addr.sin_family=AF_INET; remotesock_addr.sin_port=htons(port); remotesock_addr.sin_addr.S_un.S_addr=GetDomainIp(url); connectsock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); ) F5 {( \( p3 a( @- {if(connect(connectsock,(const sockaddr *)&remotesock_addr,sizeof(remotesock_addr))==INVALID_SOCKET){ //连接远程主机 closesocket(clientsock); + p5 ?9 q4 E2 w, A8 ?return 0; 9 _$ Y8 Z: L/ [} " N" E" y# u3 c* P, l: }TRACE("\r\nconnect to remote ip ok\r\n"); 0 p! F+ y6 `2 N. b( EZeroMemory((void *)data,sizeof(data)); wsprintf(data,"%s",HttpString); datalen=strlen(data); f- y1 B' M. l, L* }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"); 5 E# w+ p) h: sdatalen=strlen(data); 3 @ q, X' {* H$ K8 T7 bwhile(!send(clientsock,data,datalen,0)); - X+ Y) ^1 Z3 x} 9 N( I) H: E9 U6 R1 J4 |$ ?httpsockinfo.sourcesock=clientsock; httpsockinfo.destsock=connectsock; : B* F2 |3 P3 g C5 F! P, lAfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 ) n' p8 G+ |2 q4 {. YAfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // Sleep(100); S7 Z7 T5 I! X7 hreturn 1; , ]* O: m' U. c& X} // UINT StartHttpProxy(LPVOID info){ //端口监听线程 1 |( l" @. ~, u- I O9 b+ vSOCKET NewSock; int socklen; ' Y) q( `5 A$ T4 E* ssockaddr_in serversock,remotesock_addr; ; l5 a- P; W; g8 q" ^serversock.sin_family=AF_INET; ! e, K6 n; d0 C$ _5 z' O* Rserversock.sin_addr.S_un.S_addr=INADDR_ANY; 9 k7 D7 J6 l# `( ?; ]serversock.sin_port=htons(HttpListenPort); # N4 O R. i! u2 FHttpListenSock=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; 0 Z! y# v3 e% `listen(HttpListenSock,BACKUP); socklen=sizeof(remotesock_addr); TRACE("start http proxy thread while\r\n"); while(true){ NewSock=accept(HttpListenSock,(sockaddr *)&remotesock_addr,&socklen); TRACE("waitting ok...\r\n"); , x# F; r7 i- v- E9 tif(NewSock==INVALID_SOCKET){ Sleep(1); continue; } ZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); ' K; J, b. U7 `1 z/ ^& kHttpClientInfo.clientsock=NewSock; HttpClientInfo.clientsock_addr=remotesock_addr; * r7 N w2 a0 p, OTRACE("start proxy thread\r\n"); AfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); Sleep(100); } , [, q, H7 m+ K8 d9 e# W! areturn 1; } // CHTTPPROXY::CHTTPPROXY() { - P1 h* T% t: {) t0 j, o: cWSADATA WsaData; % X9 T& l: a! SWORD wsaVer; 8 a) Q6 A! a) ]1 G# r( awsaVer=MAKEWORD(2,0); WsaStartupOk=false; ; x; A+ ~* j7 f# G2 D9 c$ {1 Hif(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; }

CHTTPPROXY::~CHTTPPROXY() { if(WsaStartupOk){ ?' ^3 g* `* I$ A. g: EWSACleanup(); ; f" t( l# d$ z% y& D; ~6 P} # r" U; }! o t% g}

int CHTTPPROXY::StartProxy(int listenport) 8 @- W0 X3 V, u7 ?$ k& o& ]0 ]( |{ 0 A9 o9 H' I" T: P) @4 L$ sHttpListenPort=listenport; 4 @6 \1 Y9 H: M @3 F% pAfxBeginThread(StartHttpProxy,(LPVOID)NULL); & f4 y' V) W f- ~/ _! Ureturn 1; }

6 @0 y' P* H& c8 X" O2 ^/*========================================================================= % _1 j% a% a! E0 U$ @FILE:HTTPPROXY.h ==========================================================================*/ ! V( C% h8 D/ nclass CHTTPPROXY 7 f1 \7 O& j4 W) S9 h n3 Q6 M{ public: int StartProxy(int listenport); 4 f) v M! p6 `5 @$ O0 z. Kbool WsaStartupOk; CHTTPPROXY(); virtual ~CHTTPPROXY();

};

/*========================================================================= - [: b( ^' X( YFILE:stdafx.h 9 S* B# h' D5 h" P8 E/ C$ y5 E==========================================================================*/ #include #include #include #include

注:不要忘了在link选项中添加wsock32.lib和ws2_32.lib，或在文件前部加上如下语句： #paragma comment(lib,"wsock32.lib") ' v% Q) N% h8 {* p9 q* N4 e$ |#paragma comment(lib,"ws2_32.lib")

本代码在win2k和vc6.0下编译成功~~ 用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: CHTTPPROXY httpproxy; % F% d4 s" n fhttpproxy.StartProxy(7890); 6 ]) G0 y& W+ ~* O有问题mailto me！