一个简单的 HTTP 服务器 - 编程论坛

程序代码：

#include <stdio.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include "hacking.h"
#include "hacking-network.h"

#define PORT 50000   // 用户连接端口
#define WEBROOT "./webroot" //浏览器的默认目录

void handle_connection(int, struct sockaddr_in *); //处理請求的函数
int get_file_size(int); // 获得打开文件的大小

int main(void) {
   int sockfd, new_sockfd, yes=1;
   struct sockaddr_in host_addr, client_addr;   //
   socklen_t sin_size;

   printf("接受来自端口 %d 的請求\n", PORT);

   if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1)
      fatal("无法生成套接字");

   if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1)
      fatal("設定套接字为SO_REUSEADDR ");

   host_addr.sin_family = AF_INET;
   host_addr.sin_port = htons(PORT);
   host_addr.sin_addr.s_addr = INADDR_ANY; // 自动設定自身ＩＰ
   memset(&(host_addr.sin_zero), '\0', 8); // 结构体的最后８字节清０
   if (bind(sockfd, (struct sockaddr *)&host_addr, sizeof(struct sockaddr)) == -1)
      fatal("Bind失败");

   if (listen(sockfd, 20) == -1)
      fatal("Listen失败");

   while(1) {
      sin_size = sizeof(struct sockaddr_in);
      new_sockfd = accept(sockfd, (struct sockaddr *)&client_addr, &sin_size);
      if(new_sockfd == -1)
         fatal("新连接失败");

      handle_connection(new_sockfd, &client_addr);
   }
   return 0;
}

void handle_connection(int sockfd, struct sockaddr_in *client_addr_ptr) {
   unsigned char *ptr, request[500], resource[500];
   int fd, length;

   length = recv_line(sockfd, request);

   printf("来自%s：%d　的請求  \"%s\"\n", inet_ntoa(client_addr_ptr->sin_addr), ntohs(client_addr_ptr->sin_port), request);

   ptr = strstr(request, " HTTP/"); // 注意第一个字符是空格
   if(ptr == NULL) { // 找不到HTTP就说明不是一个合法的請求
      printf(" 非HTTP！\n");
   } else {
      *ptr = 0; //找到了HTTP，" HTTP/“　就结束使命了，在第一个空格处加'\0'
      ptr = NULL; // 接下来找"GET "或是" HEAD "，若没有找到，ptr设为NULL就有用了
      if(strncmp(request, "GET ", 4) == 0)  // GET請求
         ptr = request+4; // 請求会是类似于 GET /image.jpg HTTP/1.1"这种形式，就可以理解为什么加４了
      if(strncmp(request, "HEAD ", 5) == 0) // 同理
         ptr = request+5; // 同理

      if(ptr == NULL) { // 找不到"GET " 或是"HEAD "的任何一个
         printf("\t未知請求！\n");
      } else { // 找到"GET "或"HEAD "中任何一个，就开始处理后续的比如"GET /image.jpg" 的"/image.jpg"部分
         if (ptr[strlen(ptr) - 1] == '/')  // 若只有一个"/"，则默认返回index.html
            strcat(ptr, "index.html");
         strcpy(resource, WEBROOT);     // 开始构建一个完整的路径
         strcat(resource, ptr);
         fd = open(resource, O_RDONLY, 0); //  试着打开这个文件
         printf("\t打开\'%s\' \t", resource);
         if(fd == -1) { // 没有找到文件
            printf(" 404 Not Found\n");
            send_string(sockfd, "HTTP/1.0 404 NOT FOUND\r\n");
            send_string(sockfd, "Server: Tiny webserver\r\n\r\n");
            send_string(sockfd, "<html><head><title>404 Not Found</title></head>");
            send_string(sockfd, "<body><h1>URL not found</h1></body></html>\r\n");
         } else {      // 找到文件则发送，
            printf(" 200 OK\n");
            send_string(sockfd, "HTTP/1.0 200 OK\r\n");
            send_string(sockfd, "Server: Tiny webserver\r\n\r\n");
            if(ptr == request + 4) { // 上面ptr 已经被設定好，现在看ptr如果是GET 命令的话
               if( (length = get_file_size(fd)) == -1)
                  fatal("取得文件大小失败。");
               if( (ptr = (unsigned char *) malloc(length)) == NULL)
                  fatal("内存的分配失败。");
               read(fd, ptr, length); // 将文件读入内存
               send(sockfd, ptr, length, 0);  // 发送
               free(ptr); // 释放存储文件的内存
            }
            close(fd); // 关闭文件
         }
      }
   }
   shutdown(sockfd, SHUT_RDWR);
}

int get_file_size(int fd) {
   struct stat stat_struct;

   if(fstat(fd, &stat_struct) == -1)
      return -1;
   return (int) stat_struct.st_size;
}

[ 本帖最后由 madfrogme 于 2012-10-12 23:00 编辑 ]