打造自己的HTTP代理服务器

韩冰

/*========================================================================= * Z& X! K! _- p* A- \FILE:HTTPPROXY.cpp HTTP Proxy v1.0 * U% m8 N* c5 u' V [7 Mpowered by shadow 2004/11/14 my web:http://www.codehome.6600.org QQ:176017352 请先阅读有关http代理协议的相关资料,转载请著明出处 8 u2 {3 |8 R2 L. M) R- F==========================================================================*/ 2 {# D4 n+ u& `

#include "stdafx.h" % d ]3 [8 _! Y% m#include "HTTPPROXY.h" 4 V, c: z) y" w6 ~6 b o#define FD_NOEVENT 0 // #define BACKUP 100 #define MAXDATALEN 65535 int HttpListenPort; - s% y0 R, o/ vSOCKET HttpListenSock; // typedef struct _CLIENTINFO{ # l0 l" `7 F4 U9 S+ k4 o4 |* ?. wSOCKET clientsock; 6 @- M9 D! n) N5 ?% E' X0 jSOCKET udpsock; sockaddr_in clientsock_addr; }CLIENTINFO,*LPCLIENTINFO; CLIENTINFO HttpClientInfo; , ?3 z" v# }! b// typedef struct _SOCKINFO{ C0 r5 J% m O- [7 g2 V1 k9 ISOCKET sourcesock; ) F3 n! Y. W# A3 R* ^+ y5 hSOCKET destsock; * l6 `6 p' o/ V* i0 ~0 Z}SOCKINFO,*LPSOCKINFO; SOCKINFO httpsockinfo; // extern long GetSocketEventId(SOCKET remotesock); extern unsigned long GetLocalIp(); extern unsigned long GetDomainIp(char domainname[250]); /* % |$ V+ i1 g6 w q: R2 a4 @4 Along GetSocketEventId(SOCKET remotesock){ long EventId; HANDLE hevent; hevent=CreateEvent(NULL,0,0,0); 8 i- n& e) _8 Z) v l- UWSANETWORKEVENTS socket_events; 4 W5 P: L9 X+ C- g5 M/ l3 ^0 CEventId=FD_NOEVENT; if(WSAEventSelect(remotesock,hevent,FD_ACCEPT|FD_CONNECT|FD_READ|FD_WRITE|FD_CLOSE)==SOCKET_ERROR) return EventId; , z& {/ A+ m8 x/ |$ Y8 oWSAEnumNetworkEvents(remotesock,hevent,&socket_events); if(socket_events.lNetworkEvents!=0){ 8 X# n) Y! Y8 u# t+ x: F' vswitch(socket_events.lNetworkEvents){ : @/ j9 P5 N# p/ k, E: Z! G: ^3 h; Tcase FD_ACCEPT:EventId=FD_ACCEPT;break; case FD_CONNECT:EventId=FD_CONNECT;break; 2 H- F7 Y% j7 k/ V; l" S, k+ f8 |case FD_READ:EventId=FD_READ;break; case FD_WRITE:EventId=FD_WRITE;break; 4 ~$ s& s5 T7 U$ Q0 ~% Xcase FD_CLOSE:EventId=FD_CLOSE;break; 6 [3 [5 L6 T6 Q) z/ @. Icase FD_OOB:EventId=FD_OOB;break; default:EventId=FD_NOEVENT;break; } 3 w6 O) ]8 W6 y} : q6 T( g6 }6 A o+ }( welse EventId=FD_NOEVENT; 4 x0 K4 l" ?* l4 @# x) Ereturn EventId; } 4 s8 X2 R0 T# Q- Y B1 K- f- q// / x7 s9 A3 z* Q/ Munsigned long GetLocalIp() { char IP[MAX_PATH],*ip; . v6 P: S Z8 R% J& O7 }2 `8 Wchar pc_name[80]; struct in_addr in; struct hostent *host; " \8 `0 ^6 ~, ^; rWORD wVersionRequested; WSADATA wsaData; : w5 |" x! ]" _1 _% X- e4 E( PwVersionRequested=MAKEWORD(2,0); ip=IP; strcpy(ip,"Ip not get!"); 9 t+ D" {) S6 }$ Gif(WSAStartup(wVersionRequested,&wsaData)) return 0; * {0 f0 s- h3 E* j" |' ]if(gethostname(pc_name,80)==SOCKET_ERROR){ , R! k2 V% m1 V% ?, ~" p: D' AWSACleanup(); ) c9 E- e$ Z: F2 W; Lreturn 0; 0 X; `8 I1 {) w4 s7 C4 B1 H4 P} if(!(host=gethostbyname(pc_name))){ x% l: ]9 ]5 {8 M) B* X0 f- y+ }2 nWSACleanup(); 5 }2 @! S: x1 k9 K# a& y! ^return 0; 0 |, k* ?$ w, p' k+ G8 ^% m} in.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); / G4 l( z6 V' J4 D" hWSACleanup(); return in.s_addr; } // 2 x, z _6 v% z3 d+ S# Punsigned long GetDomainIp(char domainname[250]) & u) `5 \* X. F& P( _' i# G{ char IP[MAX_PATH],*ip; struct in_addr in; struct hostent *host; 6 l$ Q: I! N5 W% D# ]$ o& bWORD wVersionRequested; WSADATA wsaData; wVersionRequested=MAKEWORD(2,0); $ R& Z6 f& L5 p) w9 d2 hip=IP; strcpy(ip,"Ip not get!"); if(WSAStartup(wVersionRequested,&wsaData)) return 0; if(!(host=gethostbyname(domainname))){ % |9 I/ F6 F% t5 WWSACleanup(); return 0; 3 ]+ f9 `3 Q% {% l4 n0 ~} / v8 ?& n; @8 Q6 ~in.s_addr=*((unsigned long *)host->h_addr_list[0]); strcpy(ip,inet_ntoa(in)); $ U; G2 n7 U5 L( R, @3 h8 qWSACleanup(); return in.s_addr; } */ 5 ]1 V2 n; {% d1 [ e// // ; i1 Y0 R. U6 b3 @UINT HttpReciveThread(LPVOID info){ //针对客户端的接收处理线程 LPSOCKINFO psockinfo; / M7 r. O% A2 }3 @SOCKET sourcesock,destsock; char data[MAXDATALEN]; ( a; a7 K# B: U( a/ k4 Mlong eventid; int datalen; psockinfo=(LPSOCKINFO)info; 0 K7 v3 _5 I& h$ A Q5 Y% ssourcesock=psockinfo->sourcesock; $ N: l8 w4 v! `8 @) Bdestsock=psockinfo->destsock; TRACE("deail recive thread ok!\r\n"); while(true){ 9 \' }! b+ r2 Y) U$ E% neventid=GetSocketEventId(sourcesock); $ g3 ]. R1 D' Wswitch(eventid){ ' g5 O: @/ ^, l1 h0 P4 o: Ncase FD_CLOSE: ' r$ c9 C+ }3 _7 T9 l/ h- jTRACE("s fdclosed\r\n"); closesocket(destsock); $ ~* k0 s- d2 Ereturn 1; ! n* K6 q) G% z& N4 S+ B3 w* wbreak; * N; a4 T, ~; J9 Z5 r% L1 f! Odefault:break; 4 M4 F; B! V7 s8 h4 Q, u} eventid=GetSocketEventId(destsock); switch(eventid){ . S& ~ L5 X5 S1 }3 u1 Hcase FD_CLOSE: ( ?; ?9 j/ H2 M$ E, Q' ~6 `closesocket(sourcesock); TRACE("d fdclosed\r\n"); return 1; 1 P9 [; D- h0 f. O: } s: Y! U# \break; 0 a4 D1 V e2 G$ p4 jdefault:break; } 3 i6 ^& N+ j3 N0 `3 F2 Cdatalen=recv(sourcesock,data,sizeof(data),0); 4 Y2 X9 f. N3 t+ c" iif(datalen==0){ closesocket(sourcesock); / `1 U9 y. i: {; ]closesocket(destsock); TRACE("s fdclosed\r\n"); / V$ B& i; L& ~break; : }( g0 |# y# {} ' r+ N; p; j- C* cif(datalen>0){ P2 @5 _: g2 x" ]" o( H* f/ @. Awhile(!send(destsock,data,datalen,0)); $ v; |! a( y) z+ C} ! `. j4 @4 h* u8 ^: d- ?6 BSleep(1); ; A2 K! h- \9 s: w( w} ' I7 Z' q" |& o7 a: Ereturn 1; } - o: j4 y5 y3 F, N+ V// UINT HttpSendThread(LPVOID info){ //针对远程端的接收处理线程 LPSOCKINFO psockinfo; SOCKET sourcesock,destsock; 8 }, u; g9 Q9 T' Echar data[MAXDATALEN]; 6 c6 C3 ?8 b, P8 o4 {$ d* f8 Dlong eventid; % f7 e" A$ |& d, U1 Cint datalen; # q7 A- l+ e) d( ~) Bpsockinfo=(LPSOCKINFO)info; . w U/ y8 [$ E, V& Zsourcesock=psockinfo->sourcesock; 6 s3 H. v" O' r/ m! l1 B( Bdestsock=psockinfo->destsock; TRACE("deail send thread ok!\r\n"); while(true){ & w- d+ }" v0 {. `) M, K9 Z: oeventid=GetSocketEventId(sourcesock); 8 [) L5 v) R" j/ b' P" Y: Nswitch(eventid){ case FD_CLOSE: ) ^. g) m* [8 Z' h- }4 k6 yTRACE("s fdclosed\r\n"); * j! v9 E2 p8 W8 d$ |: @8 yclosesocket(destsock); ) v' x+ f" u+ U# Sreturn 1; break; default:break; } eventid=GetSocketEventId(destsock); - q# A3 `6 M- rswitch(eventid){ case FD_CLOSE: 6 n0 F& G* S9 `/ Q) `; s! Q2 Oclosesocket(sourcesock); " _# @$ g% z; D0 ~/ N; }6 QTRACE("d fdclosed\r\n"); return 1; break; default:break; } : l! b2 Z) Q: F7 y4 w( ]" ydatalen=recv(destsock,data,sizeof(data),0); if(datalen==0){ - E }( @2 `" O7 I. y, `3 ~- @" P$ yclosesocket(sourcesock); 8 v6 H7 {' c$ A+ Xclosesocket(destsock); TRACE("d fdclosed\r\n"); break; } / \% q2 J# d rif(datalen>0){ \; k/ b. J/ fwhile(!send(sourcesock,data,datalen,0)); } $ @5 L" J4 a' K. d4 kSleep(1); 1 I& A! w2 |4 |: j- ~} : m$ y0 w& X; y- R* A9 O4 Treturn 1; } ) z, p, y" f! d7 e& v. u8 M// // UINT HttpProxyServerThread(LPVOID info){ //针对一次服务的线程 LPCLIENTINFO pclientinfo; SOCKET connectsock,clientsock; sockaddr_in remotesock_addr; # U& o3 E9 s$ T0 C5 ?char data[MAXDATALEN],url[250],temp[250],httpurl[250],portnum[10]; int datalen,i,index_start,index_end,port; 6 @; R0 `8 v* ` BCString HttpString,UrlString,PortString; % z0 v3 O0 o, A' Jpclientinfo=(LPCLIENTINFO)info; clientsock=pclientinfo->clientsock; ZeroMemory((void *)data,sizeof(data)); datalen=recv(clientsock,data,sizeof(data),0); if(datalen<=0){ 3 \+ ^; E0 V2 Dclosesocket(clientsock); 1 X: D* C4 s# X+ dreturn 0;

韩冰

} 0 I4 v9 Q) M0 Y* ?* gHttpString.Format("%s",data); & z3 v$ w! R: s/ v; h( [5 A3 o6 y- e( x" kUrlString=HttpString; ; X" C" _8 I, f+ B5 ITRACE("get http string:\r\n"); 3 f* E0 ^* W1 F; ~TRACE(HttpString); index_start=HttpString.Find("Host: ",0); //寻找url标记 if(index_start<=0){ E* m7 G# i' D* p7 U; x8 eclosesocket(clientsock); ; H5 r4 Z- M+ L9 Nreturn 0; } index_end=HttpString.Find("\r\n",index_start); 3 A( `3 k' q6 _" P+ v0 @if(index_end<=0){ closesocket(clientsock); ( X' N/ m' V9 Y- B/ F) H+ Vreturn 0; } UrlString=HttpString.Mid(index_start+6,index_end-index_start-6); //读取 url字符串 TRACE("\r\n get url:"); TRACE(UrlString); wsprintf(url,"%s",UrlString); strcpy(temp,url); ?* |' Z, U2 k0 X+ d. xstrcat(temp,":"); / h! B- e& K* q J# M9 u( Bdatalen=strlen(temp); if(HttpString.Find("GET",0)==0){ //判断get命令，并处理 7 H! X( s" ~( `8 m$ E& |4 W8 jindex_start=HttpString.Find(temp,0); strcpy(httpurl,"http://"); / F6 L' z+ L/ n# }/ T. O' u7 d( v) p2 sif(index_start>0){ 5 y% M' K% d2 W$ T" X$ Eindex_end=HttpString.Find("/",index_start); " q8 @. J f. v1 L. {if(index_end<=0){ 4 T( N' P& p J+ l0 U- K, A' c6 Mclosesocket(clientsock); return 0; , S6 M9 i, ~: D" G! w1 E} . ?# J# V' o- `1 O3 t6 a. MPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); 1 r+ }' g/ O8 t1 B6 nport=atoi(PortString); strcat(httpurl,temp); + u F" c% ~: t; A8 Eitoa(port,portnum,sizeof(portnum)); strcat(httpurl,portnum); strcat(httpurl,"/"); } 7 O1 w) b% o( R; N/ Felse{ port=80; & ^0 h; k H# K5 R( R( }+ G4 @strcat(httpurl,url); ! w: C8 P" L5 x6 u$ S- g: F! qstrcat(httpurl,"/"); ! X- m% X; p5 Y& A2 T7 @6 ^. M} TRACE("get http url:%s\r\n",httpurl); HttpString.Replace(httpurl,"/"); HttpString.Replace("Proxy-",""); 3 u) f- `' r, {HttpString.Replace("HTTP/1.0","HTTP/1.1"); 3 q/ G2 I- t* D$ _} else if(HttpString.Find("CONNECT",0)==0){ //判断connect命令并处理 8 v+ a% z" `, F* S' W9 P rindex_start=HttpString.Find(temp,0); if(index_start>0){ ) p9 I" G" g7 F3 Y Uindex_end=HttpString.Find(" ",index_start); 7 @+ F5 N) H" w4 E1 vif(index_end<=0){ - U* a7 D: l$ u a/ x) Jclosesocket(clientsock); return 0; } ) y* s- e5 t& M; E/ A9 g9 gPortString=HttpString.Mid(index_start+datalen,index_end-index_start-datalen); port=atoi(PortString); } else{ % p! M! l/ T; k+ w% o3 d8 F* Bclosesocket(clientsock); return 0; } } TRACE("get new http string:\r\n"); TRACE(HttpString); remotesock_addr.sin_family=AF_INET; remotesock_addr.sin_port=htons(port); remotesock_addr.sin_addr.S_un.S_addr=GetDomainIp(url); . b4 F6 q) v( {, L+ Wconnectsock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); if(connect(connectsock,(const sockaddr *)&remotesock_addr,sizeof(remotesock_addr))==INVALID_SOCKET){ //连接远程主机 closesocket(clientsock); , `5 j: q6 `& @/ greturn 0; 3 N. a* z6 j1 _# X3 L5 Q; t} ) ~ O! {. K) X2 Q4 e2 [" }TRACE("\r\nconnect to remote ip ok\r\n"); * s0 p- D Y! A( d2 ? Q3 VZeroMemory((void *)data,sizeof(data)); wsprintf(data,"%s",HttpString); datalen=strlen(data); 5 A! S C- ]9 I4 B" y% X" q7 qif(HttpString.Find("CONNECT",0)<0) while(!send(connectsock,data,datalen,0)); else{ 3 ` }; a1 _- w, Y# F2 _strcpy(data,"HTTP/1.0 200 Connection established\r\nProxy-agent: CHTTPPROXY V1.0 powered by shadow\r\n\r\n"); : s3 {& f! v8 b* Z2 o4 qdatalen=strlen(data); # R- y3 Q1 G1 s0 ^while(!send(clientsock,data,datalen,0)); : z0 u* p; y1 G2 P" q} httpsockinfo.sourcesock=clientsock; httpsockinfo.destsock=connectsock; 4 T% G* h, I2 |3 C9 O% [" [AfxBeginThread(HttpReciveThread,(LPVOID)&httpsockinfo); //抛出处理线程 - _1 z$ {% J! x5 wAfxBeginThread(HttpSendThread,(LPVOID)&httpsockinfo); // Sleep(100); return 1; . B# o5 z* ^$ R/ ]1 m} // UINT StartHttpProxy(LPVOID info){ //端口监听线程 : M, f" m( R- K* b" ]+ ^SOCKET NewSock; . M# K$ N! j+ |. bint socklen; / n: e1 V7 P+ m6 b4 F; `sockaddr_in serversock,remotesock_addr; serversock.sin_family=AF_INET; serversock.sin_addr.S_un.S_addr=INADDR_ANY; 8 n( i/ O, F J3 r2 W8 B; A. w" xserversock.sin_port=htons(HttpListenPort); HttpListenSock=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP); . l4 _+ K& r' W1 W' K! u8 M0 ?; Vif(HttpListenSock==INVALID_SOCKET) return 0; & D7 Y0 X) E; t! i5 R* U, cif(bind(HttpListenSock,(const sockaddr *)&serversock,sizeof(serversock))==SOCKET_ERROR) return 0; listen(HttpListenSock,BACKUP); * Q$ i; [ s7 a9 Isocklen=sizeof(remotesock_addr); TRACE("start http proxy thread while\r\n"); - b8 C9 H( H/ p+ r5 B( C5 y" Nwhile(true){ NewSock=accept(HttpListenSock,(sockaddr *)&remotesock_addr,&socklen); TRACE("waitting ok...\r\n"); 8 ], Z3 H$ A* l# f' P1 f( ]if(NewSock==INVALID_SOCKET){ Sleep(1); continue; } ZeroMemory((void *)&HttpClientInfo,sizeof(CLIENTINFO)); HttpClientInfo.clientsock=NewSock; HttpClientInfo.clientsock_addr=remotesock_addr; TRACE("start proxy thread\r\n"); AfxBeginThread(HttpProxyServerThread,(LPVOID)&HttpClientInfo); Sleep(100); 9 g4 F" ?) K9 {- K} 2 n; F% p) Z) t* a$ N0 Greturn 1; / }1 s* a5 z3 j3 b3 u! {& j! Q7 }} // CHTTPPROXY::CHTTPPROXY() { WSADATA WsaData; WORD wsaVer; / g, j s; V# y: kwsaVer=MAKEWORD(2,0); 9 m* i. r! c: R1 ]6 P, L, o, N& U/ hWsaStartupOk=false; if(WSAStartup(wsaVer,&WsaData)!=SOCKET_ERROR) WsaStartupOk=true; }

CHTTPPROXY::~CHTTPPROXY() { : ]7 y7 o% s8 L6 N$ t: Z: sif(WsaStartupOk){ 7 v0 Z# x& I) Q$ x0 dWSACleanup(); } }

int CHTTPPROXY::StartProxy(int listenport) ( p6 F6 L9 j9 V) g) E{ HttpListenPort=listenport; AfxBeginThread(StartHttpProxy,(LPVOID)NULL); 0 Z+ d6 o7 @% H9 k( w5 nreturn 1; }

/*========================================================================= FILE:HTTPPROXY.h ; _: o+ M- L. p3 C- E+ Z. x==========================================================================*/ class CHTTPPROXY { public: 4 z. c% r$ Y. k9 qint StartProxy(int listenport); bool WsaStartupOk; CHTTPPROXY(); virtual ~CHTTPPROXY();

};

/*========================================================================= FILE:stdafx.h ==========================================================================*/ * ^% D2 Y- a8 d3 T$ _#include #include #include 6 _9 |; V6 {- T- W9 j#include

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

本代码在win2k和vc6.0下编译成功~~ 9 y4 d' E4 M" E/ S3 X! h. Z用法,把这几个文件添加到你的项目中，在WinMain()中添加如下代码: CHTTPPROXY httpproxy; httpproxy.StartProxy(7890); 有问题mailto me！