Linux网络编程：一个简单的正向代理服务器的实现

Linux是一个可靠性非常高的操作系统，但是所有用过Linux的朋友都会感觉到， Linux和Windows这样的"傻瓜"操作系统（这里丝毫没有贬低Windows的意思，相反这应该是Windows的优点）相比，后者无疑在易操作 性上更胜一筹。但是为什么又有那么多的爱好者钟情于Linux呢，当然自由是最吸引人的一点，另外Linux强大的功能也是一个非常重要的原因，尤其是 Linux强大的网络功能更是引人注目。放眼今天的WAP业务、银行网络业务和曾经红透半边天的电子商务，都越来越倚重基于Linux的解决方案。因此 Linux网络编程是非常重要的，而且当我们一接触到Linux网络编程，我们就会发现这是一件非常有意思的事情，因为以前一些关于网络通信概念似是而非的地方，在这一段段代码面前马上就豁然开朗了。在刚开始学习编程的时候总是让人感觉有点理不清头绪，不过只要多读几段代码，很快我们就能体会到其中的乐趣 了。下面我就从一段Proxy源代码开始，谈谈如何进行Linux网络编程。

首先声明，这段源代码不是我编写的，让我们感谢这位名叫Carl Harris的大虾，是他编写了这段代码并将其散播到网上供大家学习讨论。这段代码虽然只是描述了最简单的proxy操作，但它的确是经典，它不仅清晰地 描述了客户机/服务器系统的概念，而且几乎包括了Linux网络编程的方方面面，非常适合Linux网络编程的初学者学习。
这段Proxy程序的用法是这样的，我们可以使用这个proxy登录其它主机的服务端口。假如编译后生成了名为Proxy的可执行文件，那么命令及其参数的描述为：
./Proxy <proxy_port> <remote_host> <service_port>
其中参数proxy_port是指由我们指定的代理服务器端口。参数remote_host是指我们希望连接的远程主机的主机名，IP地址也同样有 效。这个主机名在网络上应该是唯一的，如果您不确定的话，可以在远程主机上使用uname -n命令查看一下。参数service_port是远程主机可提供的服务名，也可直接键入服务对应的端口号。这个命令的相应操作是将代理服务器的 proxy_port端口绑定到remote_host的service_port端口。然后我们就可以通过代理服务器的proxy_port端口访问 remote_host了。例如一台计算机，网络主机名是legends，IP地址为10.10.8.221，如果在我的计算机上执行：
[root@lee /root]#./proxy 8000 legends telnet
那么我们就可以通过下面这条命令访问legends的telnet端口。

以下代码有多次修改。运行环境：Ubuntu 9.04 Server Gcc 4.3

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <errno.h>
#include <signal.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/file.h>
#include <sys/ioctl.h>
#include <sys/wait.h>
#include <netdb.h>
#define TCP_PROTO "tcp"
int proxy_port;
struct sockaddr_in hostaddr;
extern int errno;
//extern char *sys_errlist[];
void parse_args(int argc, char **argv);
void daemonize(int servfd);
void do_proxy(int usersocket);
void reap_status(void);
void errorout(char *msg);

typedef void Sigfunc(int);

int main(int argc, char **argv)
{
int clilen;
pid_t childpid;

int sockfd, newsockfd;
struct sockaddr_in servaddr, cliaddr;
parse_args(argc, argv);
bzero((char *) &servaddr, sizeof(servaddr));

servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = proxy_port;

if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
fputs("failed to create server socket /r/n", stderr);
exit(1);
}

if (bind(sockfd, (struct sockaddr *) &servaddr, sizeof(servaddr)) < 0) {
fputs("failed to bind server socket to specified port /r/n", stderr);
exit(1);
}

listen(sockfd, 5);
daemonize(sockfd);

for(;;) {
clilen = sizeof(cliaddr);
newsockfd = accept(sockfd, (struct sockaddr *) &cliaddr, &clilen);
if (newsockfd <0 && errno == EINTR)
continue;
else if (newsockfd < 0)
errorout("failed to accept connection");

if ((childpid = fork()) == 0) {
close(sockfd);
do_proxy(newsockfd);
exit(0);
}
close(newsockfd);
}
}

/***********************************************************************
function:parse_args
description: Parse the command line args.
arguments:argc, argv you know what these are.
return value: none.
calls:none.
globals:writes proxy_port, writes hostaddr.
**********************************************************************/

void parse_args(int argc, char ** argv)
{
int i;
struct hostent *hostp;
struct servent *servp;
unsigned long inaddr;
struct {
char proxy_port[16];
char isolated_host[64];
char service_name[32];
} pargs;

if (argc < 4) {
printf("usage: %s <proxy-port> <host> <service-name|port-number> /r/n", argv[0]);
exit(1);
}

strcpy(pargs.proxy_port, argv[1]);
strcpy(pargs.isolated_host, argv[2]);
strcpy(pargs.service_name, argv[3]);

for (i = 0; i < strlen(pargs.proxy_port); i++)
if (!isdigit(*(pargs.proxy_port + i))) break;

if (i == strlen(pargs.proxy_port))
proxy_port = htons(atoi(pargs.proxy_port));
else {
printf("%s: invalid proxy port /r/n", pargs.proxy_port);
exit(0);
}

bzero(&hostaddr, sizeof(hostaddr));
hostaddr.sin_family = AF_INET;
if ((inaddr = inet_addr(pargs.isolated_host)) != INADDR_NONE)
bcopy(&inaddr, &hostaddr.sin_addr, sizeof(inaddr));
else if ((hostp = gethostbyname(pargs.isolated_host)) != NULL)
bcopy(hostp->h_addr, &hostaddr.sin_addr, hostp->h_length);
else {
printf("%s: unknown host /r/n", pargs.isolated_host);
exit(1);
}

if ((servp = getservbyname(pargs.service_name, TCP_PROTO)) != NULL)
hostaddr.sin_port = servp->s_port;
else if (atoi(pargs.service_name) > 0)
hostaddr.sin_port = htons(atoi(pargs.service_name));
else {
printf("%s: invalid/unknown service name or port number /r/n", pargs.service_name);
exit(1);
}
}

/*****************************************************************
function: daemonize
description: detach the server process from the current context,
creating a pristine, predictable environment in which it will execute.
arguments: servfd file descriptor in use by server.
return value: none
calls: none
globals: none
*****************************************************************/
void daemonize(int servfd)
{
pid_t childpid;
int fd, fdtablesize;
/* ignore terminal I/O, stop signals */
signal(SIGTTOU, SIG_IGN);
signal(SIGTTIN, SIG_IGN);
signal(SIGTSTP, SIG_IGN);

/* fork to put us in the background (whether or not the user specified
'&' on the command line */
if ((childpid = fork()) < 0) {
fputs("failed to fork first child /r/n", stderr);
exit(1);
}
else if (childpid > 0)
exit(0);

if (setpgrp(0, getpid()) < 0 ) {
fputs("failed to become process group leader/r/n", stderr);
exit(1);
}

if ((fd = open("/dev/tty", O_RDWR)) >= 0) {
ioctl(fd, TIOCNOTTY, NULL);
close(fd);
}

for(fd= 0, fdtablesize = getdtablesize(); fd < fdtablesize; fd++)
if (fd != servfd) close(fd);

chdir("/");

umask(0);
signal(SIGCLD, (Sigfunc *)reap_status);
}

/*****************************************************************
function:reap_status
desc: Handle a SIGCLD signal by reaping the exit status of the
perished child, and discarding it.
arguments:none
return value : none.
calls: none
globals : none
*******************************************************************/
void reap_status(void)
{
pid_t pid;
union wait status;
while ((pid = wait3(&status, WNOHANG, NULL)) > 0);
}

/******************************************************************
function : do_proxy
desc : does the actual work of virtually connecting a client to the
telnet service on the isolated host.
arguments: usersockfd socket to which the client is connected.
return value : none
calls : none
globals : reads hostaddr.
*******************************************************************/

void do_proxy(int usersockfd)
{
int isosockfd;
fd_set rdfdset;
int connstat;
int iolen;
char buf[2048];
/*open a socket to connect to the isolated host */
if ((isosockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
errorout("failed to create socket to host");
connstat = connect(isosockfd, (struct sockaddr *)&hostaddr, sizeof(hostaddr));

switch (connstat){
case 0: break;
case ETIMEDOUT:
case ECONNREFUSED:
case ENETUNREACH:
strcpy(buf, strerror(errno));
strcat(buf, "/r/n");
write(usersockfd, buf, strlen(buf));
close(usersockfd);
exit(1);
break;
default:
errorout("failed to connect to host");
}

for(;;) {
/* Select for readability on either of our two sockedts */
FD_ZERO(&rdfdset);
FD_SET(usersockfd, &rdfdset);
FD_SET(isosockfd, &rdfdset);
if (select(FD_SETSIZE, &rdfdset, NULL, NULL, NULL) < 0)
errorout("Select failed");

/* is the client sending data? */
if (FD_ISSET(usersockfd, &rdfdset)) {
/* zero length means the client disconnected */
if ((iolen = read(usersockfd, buf, sizeof(buf))) <= 0) break;
write(isosockfd, buf, iolen);
}

/*is the host sending data? */
if (FD_ISSET(isosockfd, &rdfdset)) {
if ((iolen = read(isosockfd, buf, sizeof(buf))) <= 0) break;
write(usersockfd, buf, iolen);
}
}
close(isosockfd);
close(usersockfd);
}

/**************************************************************
function : errorout
desc : displays an error message on the console and kill the
current process.
arguments : msg-- message to be displayed.
return value : none;
calls : none
globals: none
**************************************************************/
void errorout(char *msg)
{
FILE *console;
console = fopen("/dev/console", "a");
fprintf(console, "proxyd: %s /r/n", msg);
fclose(console);
exit(1);
}