socket通信之http-server

 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include
 #include

 //#define DEFAULTIP "127.0.0.1"
 #define DEFAULTIP "192.168.1.33"
 #define DEFAULTPORT "80"
 #define DEFAULTBACK "10"
 #define DEFAULTDIR "/home"
 #define DEFAULTLOG "/tmp/http-server.log"

 void prterrmsg(char *msg);
 #define prterrmsg(msg)        { perror(msg); abort(); }
 void wrterrmsg(char *msg);
 #define wrterrmsg(msg)        { fputs(msg, logfp); fputs(strerror(errno), logfp);fflush(logfp); abort(); }

 void prtinfomsg(char *msg);
 #define prtinfomsg(msg)        { fputs(msg, stdout);  }
 void wrtinfomsg(char *msg);
 #define wrtinfomsg(msg)        {  fputs(msg, logfp); fflush(logfp);}

 #define MAXBUF        1024

 char buffer[MAXBUF + 1];
 char *host = 0;
 char *port = 0;
 char *back = 0;
 char *dirroot = 0;
 char *logdir = 0;
 unsigned char daemon_y_n = 0;
 FILE *logfp;

 #define MAXPATH        150

 char *dir_up(char *dirpath)
 {
     static char Path[MAXPATH];
     int len;

     strcpy(Path, dirpath);
     len = strlen(Path);
     if (len > 1 && Path[len - 1] == '/')
         len--;
     while (Path[len - 1] != '/' && len > 1)
         len--;
     Path[len] = 0;
     return Path;
 }

 void AllocateMemory(char **s, int l, char *d)
 {
     *s = malloc(l + 1);
     bzero(*s, l + 1);
     memcpy(*s, d, l);
 }



 void GiveResponse(FILE * client_sock, char *Path)
 {
     struct dirent *dirent;
     struct stat info;
     char Filename[MAXPATH];
     DIR *dir;
     int fd, len, ret;
     char *p, *realPath, *realFilename, *nport;

     len = strlen(dirroot) + strlen(Path) + 1;
     realPath = malloc(len + 1);
     bzero(realPath, len + 1);
     sprintf(realPath, "%s/%s", dirroot, Path);

     len = strlen(port) + 1;
     nport = malloc(len + 1);
     bzero(nport, len + 1);
    sprintf(nport, ":%s", port);


     if (stat(realPath, &info)) {
         fprintf(client_sock,
                 "HTTP/1.1 200 OK\r\nServer: DAS by Chengang\r\nConnection: close\r\n\r\n%d - %s"
                 "Linux 下目录访问服务器"
                 "", errno,
                 strerror(errno));
         fprintf(client_sock,
                 "请向管理员咨询为何出现如下错误提示：\n%s %s",
                 Path, strerror(errno));
         goto out;
     }

     if (S_ISREG(info.st_mode)) {
         fd = open(realPath, O_RDONLY);
         len = lseek(fd, 0, SEEK_END);
         p = (char *) malloc(len + 1);
         bzero(p, len + 1);
         lseek(fd, 0, SEEK_SET);
         ret = read(fd, p, len);
         close(fd);
         fprintf(client_sock,
                 "HTTP/1.1 200 OK\r\nServer: DAS by Chengang\r\nConnection: keep-alive\r\nContent-type: application
         dir = opendir(realPath);
         fprintf(client_sock,
                 "HTTP/1.1 200 OK\r\nServer: DAS by Chengang\r\nConnection: close\r\n\r\n%s"
                 "Linux 下目录访问服务器"
                 "", Path);
         fprintf(client_sock,
                 "目录 %s\n",
                 Path);
         fprintf(client_sock,
                 "名称大小修改时间\n");
         if (dir == 0) {
             fprintf(client_sock,
                     "%s",
                     strerror(errno));
             return;
         }

         while ((dirent = readdir(dir)) != 0) {
             if (strcmp(Path, "/") == 0)
                 sprintf(Filename, "/%s", dirent->d_name);
             else
                 sprintf(Filename, "%s/%s", Path, dirent->d_name);
             fprintf(client_sock, "");
             len = strlen(dirroot) + strlen(Filename) + 1;
             realFilename = malloc(len + 1);
             bzero(realFilename, len + 1);
             sprintf(realFilename, "%s/%s", dirroot, Filename);
             if (stat(realFilename, &info) == 0) {
                 if (strcmp(dirent->d_name, "..") == 0)
                     fprintf(client_sock,
                             "(parent)",
                             host, atoi(port) == 80 ? "" : nport,
                             dir_up(Path));
                 else
                     fprintf(client_sock,
                             "%s",
                             host, atoi(port) == 80 ? "" : nport, Filename,
                             dirent->d_name);
                if (S_ISDIR(info.st_mode))
                     fprintf(client_sock, "目录");
                 else if (S_ISREG(info.st_mode))
                     fprintf(client_sock, "%d", info.st_size);
                 else if (S_ISLNK(info.st_mode))
                     fprintf(client_sock, "链接");
                 else if (S_ISCHR(info.st_mode))
                     fprintf(client_sock, "字符设备");
                 else if (S_ISBLK(info.st_mode))
                     fprintf(client_sock, "块设备");
                 else if (S_ISFIFO(info.st_mode))
                     fprintf(client_sock, "FIFO");
                 else if (S_ISSOCK(info.st_mode))
                     fprintf(client_sock, "Socket");
                 else
                     fprintf(client_sock, "(未知)");
                 fprintf(client_sock, "%s", ctime(&info.st_ctime));
             }
             fprintf(client_sock, "\n");
             free(realFilename);
         }
         fprintf(client_sock, "");
     } else {

         fprintf(client_sock,
                 "HTTP/1.1 200 OK\r\nServer: DAS by ZhouLifa\r\nConnection: close\r\n\r\npermission denied"
                 "Linux 下目录访问服务器"
                 "");
         fprintf(client_sock,
                 "你访问的资源'%s'被禁止访问，请联系管理员解决！",
                 Path);
     }
   out:
     free(realPath);
     free(nport);
 }

 void getoption(int argc, char **argv)
 {
     int c, len;
     char *p = 0;

     opterr = 0;
     while (1) {
         int option_index = 0;

   static struct option long_options[] = {
             {"host", 1, 0, 'H'},
             {"port", 1, 0, 'P'},
             {"back", 1, 0, 'B'},
             {"dir", 1, 0, 'D'},
             {"log", 1, 0, 'L'},
             {"daemon", 0, 0, 0},
             {0, 0, 0, 0}
         };


     c = getopt_long(argc, argv, "H:P:B:D:L",
                         long_options, &option_index);
         if (c == -1 || c == '?')
             break;

         if(optarg)        len = strlen(optarg);
         else        len = 0;

         if ((!c && !(strcasecmp(long_options[option_index].name, "host")))
             || c == 'H')
             p = host = malloc(len + 1);
         else if ((!c
                   &&
                   !(strcasecmp(long_options[option_index].name, "port")))
                  || c == 'P')
             p = port = malloc(len + 1);
         else if ((!c
                   &&
                   !(strcasecmp(long_options[option_index].name, "back")))
                  || c == 'B')
             p = back = malloc(len + 1);
         else if ((!c
                   && !(strcasecmp(long_options[option_index].name, "dir")))
                  || c == 'D')
             p = dirroot = malloc(len + 1);
         else if ((!c
                   && !(strcasecmp(long_options[option_index].name, "log")))
                  || c == 'L')
             p = logdir = malloc(len + 1);
         else if ((!c
                   &&
                   !(strcasecmp
                     (long_options[option_index].name, "daemon")))) {
             daemon_y_n = 1;
             continue;
         }
         else
             break;
         bzero(p, len + 1);
         memcpy(p, optarg, len);
     }
 }

 int main(int argc, char **argv)
 {
     struct sockaddr_in addr;
     int sock_fd, addrlen;

     getoption(argc, argv);

     if (!host) {
         addrlen = strlen(DEFAULTIP);
         AllocateMemory(&host, addrlen, DEFAULTIP);
     }
     if (!port) {
         addrlen = strlen(DEFAULTPORT);
         AllocateMemory(&port, addrlen, DEFAULTPORT);
     }
     if (!back) {
         addrlen = strlen(DEFAULTBACK);
         AllocateMemory(&back, addrlen, DEFAULTBACK);
     }
     if (!dirroot) {
         addrlen = strlen(DEFAULTDIR);
         AllocateMemory(&dirroot, addrlen, DEFAULTDIR);
     }
     if (!logdir) {
         addrlen = strlen(DEFAULTLOG);
         AllocateMemory(&logdir, addrlen, DEFAULTLOG);
     }

     printf
         ("host=%s port=%s back=%s dirroot=%s logdir=%s %s  backgroud mode(process ID:%d)\n",
          host, port, back, dirroot, logdir, daemon_y_n?"":"not", getpid());


     if (daemon_y_n) {
         if (fork())
             exit(0);
         if (fork())
             exit(0);
         close(0), close(1), close(2);
         logfp = fopen(logdir, "a+");
         if (!logfp)
             exit(0);
     }

     signal(SIGCHLD, SIG_IGN);

     if ((sock_fd = socket(PF_INET, SOCK_STREAM, 0)) < 0) {
         if (!daemon_y_n) {
             prterrmsg("socket()");
         } else {
             wrterrmsg("socket()");
         }
     }

     addrlen = 1;
     setsockopt(sock_fd, SOL_SOCKET, SO_REUSEADDR, &addrlen,
                sizeof(addrlen));

     addr.sin_family = AF_INET;
     addr.sin_port = htons(atoi(port));
     addr.sin_addr.s_addr = inet_addr(host);
     addrlen = sizeof(struct sockaddr_in);

     if (bind(sock_fd, (struct sockaddr *) &addr, addrlen) < 0) {
         if (!daemon_y_n) {
             prterrmsg("bind()");
         } else {
             wrterrmsg("bind()");
         }
     }

     if (listen(sock_fd, atoi(back)) < 0) {
         if (!daemon_y_n) {
             prterrmsg("listen()");
         } else {
             wrterrmsg("listen()");
         }
     }
     while (1) {
         int len;
         int new_fd;
         addrlen = sizeof(struct sockaddr_in);

         new_fd = accept(sock_fd, (struct sockaddr *) &addr, &addrlen);
         if (new_fd < 0) {
             if (!daemon_y_n) {
                 prterrmsg("accept()");
             } else {
                 wrterrmsg("accept()");
             }
             break;
         }
         bzero(buffer, MAXBUF + 1);
         sprintf(buffer, "Connection comes from:\n %s:%d\n",
                 inet_ntoa(addr.sin_addr), ntohs(addr.sin_port));
         if (!daemon_y_n) {
             prtinfomsg(buffer);
         } else {
             wrtinfomsg(buffer);
         }

         if (!fork()) {
             bzero(buffer, MAXBUF + 1);
             if ((len = recv(new_fd, buffer, MAXBUF, 0)) > 0) {
                 FILE *ClientFP = fdopen(new_fd, "w");
                 if (ClientFP == NULL) {
                     if (!daemon_y_n) {
                         prterrmsg("fdopen()");
                     } else {
                         prterrmsg("fdopen()");
                     }
                 } else {
                     char Req[MAXPATH + 1] = "";
                     sscanf(buffer, "GET %s HTTP", Req);  //取buffer内容到Req中（以字符串的形式）
                     bzero(buffer, MAXBUF + 1);
                     sprintf(buffer, "Request acquire files: \"%s\"\n", Req);//以字符串的形式将Req的内容传到buffer中
                     if (!daemon_y_n) {
                         prtinfomsg(buffer);
                     } else {
                         wrtinfomsg(buffer);
                     }

                     GiveResponse(ClientFP, Req);
                     fclose(ClientFP);
                 }
             }
             exit(0);
         }
         close(new_fd);
     }
     close(sock_fd);
     return 0;
 }