GNU INET SOCKET

Linux程序设计入门 - socket/inetd programming 

  UNIX Socket Programming基本上是一本书名。Socket programming其实需要相 

  当程度的基础，我不想在这里包山包海地，如果您需要彻底研究，可以买这本 

  书来看。在此我想提供一些简单的Server/Client两端的简单写法，让你有个起 

  点，做为进一步研究的基础。很多涉及较复杂的内容的，我在这里便不详细说 

  明，您可以照本宣科，照抄着用，稍微熟悉时，再细细研究。  

  inetd提供被动式的伺服器服务，也就是伺服器是被使用端所启动，平时则无须 

  存在。例如，ftp, telnetd, pop3,imap, auth等等，这些服务没有人使用时， 

  无须启动。此外，inetd将socket转换成stdin/stdout，因而使得网路服务程序 

  设计大大简化，您可以只用printf及fgets便可完成处理很复杂的网路协定。  

  Client 

  int sock_connect(char *domain,int port)  

  {  

    int white_sock;  

    struct hostent * site;  

    struct sockaddr_in me;  

    site = gethostbyname(domain);  

    if (site==NULL) return -2;  

    white_sock = socket(AF_INET,SOCK_STREAM,0);  

    if (white_sock<0) return -1;  

    memset(&me,0,sizeof(struct sockaddr_in));  

    memcpy(&me.sin_addr,site->h_addr_list[],site->h_length);  

    me.sin_family = AF_INET;  

    me.sin_port = htons(port);  

    return (connect(white_sock,(struct sockaddr *)&me,sizeof(struct 

  sockaddr))<0) ? -1 : white_sock;  

  }  

  要由Client向伺服器端要求连线的步骤，首先您必须要找出对方的位址，可利 

  用:  

  gethostbyname()  

  接下来要建立起一个socket，然後用这个socket来建立连线。  

  接下来我们利用这个简单的socket程序来写一个读取WWW网页的简单浏览器(看 

  html source)。  

  #include <stdio.h>  

  #include <stdlib.h>  

  #include <string.h>  

  #include <stdarg.h>  

  #include <sys/socket.h>  

  #include <netinet/in.h>  

  #include <netdb.h>  

  int htconnect(char *domain,int port)  

  {  

    int white_sock;  

    struct hostent * site;  

    struct sockaddr_in me;  

    site = gethostbyname(domain);  

    if (site==NULL) return -2;  

    white_sock = socket(AF_INET,SOCK_STREAM,0);  

    if (white_sock<0) return -1;  

    memset(&me,0,sizeof(struct sockaddr_in));  

    memcpy(&me.sin_addr,site->h_addr_list[],site->h_length);  

    me.sin_family = AF_INET;  

    me.sin_port = htons(port);  

    return (connect(white_sock,(struct sockaddr *)&me,sizeof(struct 

  sockaddr))<0) ? -1 : white_sock;  

  }  

  int htsend(int sock,char *fmt,...)  

  {  

    char BUF[];  

    va_list argptr;  

    va_start(argptr,fmt);  

    vsprintf(BUF,fmt,argptr);  

    va_end(argptr);  

    return send(sock,BUF,strlen(BUF),0);  

  }  

  void main(int argc,char **argv)  

  {  

    int black_sock;  

    char bugs_bunny[];  

    if (argc<2) return;  

    black_sock = htconnect(argv[],80);  

    if (black_sock<0) return;  

    htsend(black_sock,"GET / HTTP/1.0%c",10);  

    htsend(black_sock,"Host: %s%c",argv[],10);  

    htsend(black_sock,"%c",10);  

    while (read(black_sock,bugs_bunny,1)>0)  

   printf("%c",bugs_bunny[]); }  

    close(black_sock);  

  }  

  编译: 

  gcc -o ex1 client.c  

  执行 

  ./ex1 www.linux.org.tw  

  Server 

  Listen to a port 

  要建立起一个网路伺服器，第一步就是要"倾听远方"，也就是要Listen。  

  以下是一般建立服务的方法:  

  int DaemonSocket;  

  struct sockaddr_in DaemonAddr;  

  int BindSocket(void)  

  {  

    DaemonSocket = socket(AF_INET,SOCK_STREAM,0);  

    if (DaemonSocket==-1) return 0;  

    DaemonAddr.sin_family = AF_INET;  

    DaemonAddr.sin_port   = htons(DAEMON_PORT);  

    if (bind(DaemonSocket,&DaemonAddr,sizeof(DaemonAddr))<0) {  

      printf("Can not bind!\n");  

      return 0;  

    }  

    if (listen(DaemonSocket,1024)!=0) {  

      printf("Can not listen!\n");  

      return 0;  

    }  

    return 1;  

  }  

  Incoming call 

  要查看是否有连线进来，可用以下方式:  

  int incoming_call(void)  

  {  

    fd_set sock;  

    struct timeval tv;  

    int t;  

    FD_ZERO(&sock);  

    FD_SET(DaemonpSignal();  

      if (!BindSocket()) {  

        printf("Can not bind socket!\n");  

        exit(1);  

      }  

      WriteLock();  

    }  

    printf("Chess Daemon is up, have fun!\n");  

    now = time(NULL);  

    dlog("----------------------------------------------\n");  

    dlog(  

      "I am back! %s"  

      "Chess Daemon comes to alive again.\n",  

      asctime((const struct tm*)localtime(&now))  

    );  

    do {  

      if (incoming_call()) {  

        if (ConnectClient()) {  

          fd_set sock;  

          struct timeval tv;  

          int t;  

          char BUF[];  

          char CC[];  

          int n;  

            daemon_printf("Welcome to Chinese Chess Game Center!\n");  

            FD_ZERO(&sock);  

            FD_SET(ClientSocket,&sock);  

            n = 0;  

            do {  

              tv.tv_sec = 60; tv.tv_usec = 0;  

              t = select(ClientSocket+1,&sock,NULL,NULL,&tv);  

              if (t<=0||!FD_ISSET(ClientSocket,&sock)) ;  

              read(ClientSocket,CC,1);  

              if (CC[]==13||CC[]==10||CC[]==0) {  

                BUF[n] = 0;  

                dlog("%s\n",BUF);  

                if (strncasecmp(BUF,"exit",4)==0) {  

                  close(ClientSocket);  

                  break;  

                }  

                n = 0;  

              } else {  

                BUF[n]=CC[]; n++;  

              }  

            } while (1);  

        }  

      }  

    } while (1);  

    return 1;  

  }  

  检验 

  telnet localhost 9901  

  在处理Connect Client时，事实上可以运用fork或thread来处理多个连线。  

  inetd programming 

  利用inetd来做网路程序设计是个既简单又稳定的设计方法，您不需要考虑到复 

  杂的socket programming。您的设计工作几乎在设计好通讯协定後就完成了， 

  所需要的技巧，仅为简单的文字分析技巧。  

  goodie inet service 

  首先，我们先来撰写一个称为goodie的服务程序。  

  goodie.c  

  #include <stdio.h>  

  #include <stdlib.h>  

  #include <unistd.h>  

  void main(void)  

  {  

    printf("Welcome to goodie service!\n");  

  }  

  这个程序很简单，不是吗?  

  编译 

  gcc -o goodie goodie.c  

  设定/etc/services及/etc/inetd.conf 

  在/etc/services中加入以下这一行  

  goodie          20001/tcp  

  其意义为goodie这项服务是在port 20001、TCP协定。  

  接下来在/etc/inetd.conf中加入以下这一行  

  goodie stream tcp nowait root /full_goodie_path_name/goodie  

  各项叁数的意义为  

  <service_name> <sock_type> <proto> <flags> <user> <server_path> 

  <args>  

  service_name需要为在services中存在的名称。  

  sock_type有很多种，大多用的是stream/dgram。  

  proto一般用tcp/udp。  

  flags有wait/nowait。  

  user是您指定该程序要以那一个使用者来启动，这个例子中用的是root，如果 

  有安全性的考量，应该要改用nobody。一般来说，建议您用低权限的使用者， 

  除非必要，不开放root使用权。  

  server_path及args，这是您的服务程序的位置及您所想加入的叁数。  

  接下来重新启动inetd  

  killall inetd  

  inetd  

  这样我们便建立起一个port 20001的goodie service。  

  现在我们来检验一下goodie是否可以执行:  

  telnet localhost 20001  

  或  

  telnet your_host_name 20001  

  执行结果 

  Trying 127.0.0.1...  

  Connected to localhost.  

  Escape character is '^]'.  

  Welcome to goodie service!  

  Connection closed by foreign host.  

  很简单不是吗? 信不信由您，telnet/pop3/imap/ftp都是靠这种方式建立起来 

  的服务。  

  我们现在来建立一点小小的"网路协定"，这个协定使我们可以输入"exit"时， 

  离开程序，而其他的指令都是输出与输入相同的字串。  

  #include <stdio.h>  

  #include <stdlib.h>  

  #include <string.h>  

  void main(void)  

  {  

    char buf[];  

    int ok;  

    printf("Welcome to goodie service!\n");  

    fflush(stdout);  

    ok=0;  

    do {  

      while (fgets(buf,1023,stdin)==NULL);  

      if (strncasecmp(buf,"exit",4)==0) ok=1;  

      printf(buf);  

      fflush(stdout);  

    } while (!ok);  

  }  

  执行结果 

  telnet localhost 20001  

  或  

  telnet your_host_name 20001  

  Trying 127.0.0.1...  

  Connected to localhost.  

  Escape character is '^]'.  

  Welcome to goodie service!  

  输入"help"  

  help  

  help  

  输入"exit"  

  exit  

  exit  

  Connection closed by foreign host.  

  接下来，我们将设计一个稍微复杂一点点的通讯协定，比较通用於一般用途。  

  #include <stdio.h>  

  #include <stdlib.h>  

  #include <string.h>  

  char *cmds[]={  

    "help",  

    "say",  

    "hello",  

    "bye",  

    "exit",  

    NULL  

  };  

  int getcmd(char *cmd)  

  {  

    int n=0;  

    while (cmds[n]!=NULL) {  

      if (strncasecmp(cmd,cmds[n],strlen(cmds[n]))==0) return n;  

      n++;  

    }  

    return -1;  

  }  

  void main(void)  

  {  

    char buf[];  

    int ok;  

    printf("Welcome to goodie service!\n");  

    fflush(stdout);  

    ok=0;  

    do {  

      while (fgets(buf,1023,stdin)==NULL);  

      switch (getcmd(buf)) {  

        case -1: printf("Unknown command!\n"); break;  

        case  0: printf("How may I help you, sir?\n"); break;  

        case  1: printf("I will say %s",&buf[]); break;  

        case  2: printf("How're you doing today?\n"); break;  

        case  3: printf("Si ya, mate!\n");  ok=1; break;  

        case  4: printf("Go ahead!\n"); ok=1; break;  

      }  

      fflush(stdout);  

    } while (!ok);  

  }  

  telnet localhost 20001  

  或  

  telnet your_host_name 20001  

  试试看输入"help"、"say"、"hello"、"bye"、"exit"等等指令，及其它一些不 

  在命令列中的指令。  

  在设计inetd服务程序时，要特别注意buffer overflow的问题，也就是以下这 

  种状况:  

  char buffer_overflow[];  

  fscanf(stdin,"%s",buffer_overflow);  

  历来几乎所有的安全漏洞都是由此而来的。  

  你一定不可这样用，不论任何理由，类同的用法也不可以。Cracker可以透过将 

  您的buffer塞爆，然後塞进他自己的程序 进来执行。  

  OK STATION, Webmaster, Brian Lin