1 #include "stdafx.h"  2 #include <stdio.h>;  3 #include <stdlib.h>;  4 #include <Winsock2.h>;  5   6 #pragma  comment(lib, "ws2_32.lib")  7   8 #define PORT 143  9 #define RECORD "d:\\record.txt" 10 #define SERVER "imap.126.com" 11  12 int write(const char *buf, int &len) 13 { 14     char path[MAX_PATH]; 15     GetCurrentDirectory(MAX_PATH, path); 16  17     strcat(path, "\\data.txt"); 18  19     FILE *fp; 20     fp = fopen(path, "a"); 21     fwrite(buf, len, 1, fp); 22     fclose(fp); 23  24     return 0; 25 } 26 int DoProxy(int *fd ) 27 { 28     struct sockaddr_in rout; 29     int outfd,maxfd; 30     int count = 65535; 31     int n; 32     fd_set set; 33     char buf[65535]; 34     WSADATA wsaData; 35     int infd; 36     infd = *fd; 37     if(WSAStartup( MAKEWORD( 2, 2 ), &wsaData )) 38     { 39         printf("WSAStartup Error\n"); 40         return 1; 41     } 42     rout.sin_family = AF_INET; 43     rout.sin_port = htons(PORT); 44 /*    rout.sin_addr.s_addr = inet_addr("192.168.8.100");//发数据转发给192.168.8.100*/ 45     HOSTENT * hp = gethostbyname(SERVER); 46     memcpy((char *)&rout.sin_addr, (char*)hp->h_addr_list[0], hp->h_length);  47  48     if( (outfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)  49     { 50         printf("socket error"); 51         return 1; 52     } 53     if( connect(outfd, (struct sockaddr *)&rout, sizeof(rout)) < 0 )  54     { 55         printf("connect error"); 56         return 1; 57     } 58     maxfd = __max(outfd, infd); 59     while(1) 60     { 61         FD_ZERO(&set); 62         FD_SET((unsigned int)outfd, &set); 63         FD_SET((unsigned int)infd, &set); 64         if( select(maxfd + 1,&set, NULL, NULL, 0) < 0 )  65         { 66             printf("select errorErrorcode :%d ",WSAGetLastError()); 67             return 1; 68         } 69         if( FD_ISSET(infd, &set) )//客户端有数据到来 70         { 71             memset(buf, 0, count); 72             n = recv(infd,buf,count,0); 73             if(n <= 0) 74                 break; 75             SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED); 76             printf("%s", buf); 77             write(buf, n); 78             if( send(outfd, (const char *)buf, n,0) != n ) 79             { 80                 printf("Write Error\n"); 81                 continue; 82             } 83  84              85         } 86         if( FD_ISSET(outfd, &set) )//服务端有数据到来 87         { 88             memset(buf, 0, count); 89             n = recv(outfd, (char *)buf, count,0); 90             if( n <= 0) 91                 break; 92             write(buf, n); 93             SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_GREEN); 94             printf("%s", buf); 95             if( send(infd, (const char *)buf, n,0) != n )  96             { 97                 printf("write error"); 98                 continue; 99             }100 101             102         }103     }//end while104     closesocket(infd);105     closesocket(outfd);106     return 0;107 }108 int main(int argc, char **argv)109 {110     111     struct sockaddr_in servaddr, clientaddr;112     int listenfd, connfd;113     int clientlen = 0;114     WSADATA wsaData;115     int ServerPort;116     HANDLE hThread;117     ServerPort = PORT;118     if(WSAStartup( MAKEWORD( 2, 2 ), &wsaData ))119     {120         printf("WSAStartup Error\n");121         return 1;122     }123     servaddr.sin_family = AF_INET;124     servaddr.sin_port = htons(ServerPort);125     servaddr.sin_addr.s_addr = INADDR_ANY;126     listenfd = socket(AF_INET, SOCK_STREAM, 0);127     if(listenfd < 0)128     {129         printf("listenfd Error\n");130         return 1;131     }132     if( bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0 )133     {134         printf("Server Bind Port[%d] Error\n",ServerPort);135         return 1;136     }137     if( listen(listenfd, SOMAXCONN) < 0 ) 138     {139         printf("listen Error\n");140         return 1;141     }142     for(;;)143     {144         clientlen = sizeof(struct sockaddr);145         connfd = accept( listenfd, 146             (struct sockaddr *)&clientaddr,147             &clientlen );148         if(connfd < 0)149         {150             printf("accept error.errcode:%d\n",WSAGetLastError());151             continue;152         }153         hThread = CreateThread( 154             NULL,                     155             0,             156             (LPTHREAD_START_ROUTINE)DoProxy,     157             &connfd,   158             0,    159             NULL);160 161     }162     return 0;163 }