Chinaunix首页 | 论坛 | 博客

_小林_

首页　| 　博文目录　| 　关于我

SCUT_Lin

  • 博客访问： 64220
  • 博文数量： 21
  • 博客积分： 0
  • 博客等级： 民兵
  • 技术积分： 236
  • 用 户 组： 普通用户
  • 注册时间： 2014-09-05 21:34
文章分类

全部博文（21）

文章存档

2015年（21）

我的朋友
最近访客
推荐博文
相关博文
一个简单的wed服务器SHTTPD（2）———— 客户端请求分析

分类： LINUX

2015-08-07 23:52:29


点击(此处)折叠或打开

  1. //start from the very beginning,and to create greatness
  2. //@author: Chuangwei Lin
  3. //@E-mail：979951191@qq.com
  4. //@brief：客户端请求分析
  5. #include "lcw_shttpd.h"
  6. extern void Error_400(struct worker_ctl* wctl);
  7. extern void Error_403(struct worker_ctl* wctl);
  8. extern void Error_404(struct worker_ctl* wctl);
  9. extern void Error_505(struct worker_ctl* wctl);

  11. static struct http_header http_headers[] =
  12. {
  13.     {16,    HDR_INT,    OFFSET(cl),            "Content-Length: "        },
  14.     {14,    HDR_STRING,    OFFSET(ct),            "Content-Type: "        },
  15.     {12,    HDR_STRING,    OFFSET(useragent),    "User-Agent: "            },
  16.     {19,    HDR_DATE,    OFFSET(ims),        "If-Modified-Since: "    },
  17.     {15,    HDR_STRING,    OFFSET(auth),        "Authorization: "        },
  18.     {9,        HDR_STRING,    OFFSET(referer),    "Referer: "                },
  19.     {8,        HDR_STRING,    OFFSET(cookie),        "Cookie: "                },
  20.     {10,    HDR_STRING,    OFFSET(location),    "Location: "            },
  21.     {8,        HDR_INT,    OFFSET(status),        "Status: "                },
  22.     {7,        HDR_STRING,    OFFSET(range),        "Range: "                },
  23.     {12,    HDR_STRING,    OFFSET(connection),    "Connection: "            },
  24.     {19,    HDR_STRING,    OFFSET(transenc),    "Transfer-Encoding: "    },
  25.     {0,        HDR_INT,    0,                    NULL                    }
  26. };
  27. extern struct vec _shttpd_methods[];
  28. struct conf_opts conf_para;
  29. /******************************************************
  30. 函数名：montoi(char *s)
  31. 参数：
  32. 功能：将月份转化为数字
  33. *******************************************************/
  34. static int montoi(char *s)
  35. {
  36.     DBGPRINT("LCW==>montoi\n");
  37.     int retval = -1;
  38.     static char *ar[] =
  39.      { //月份数组
  40.         "Jan", "Feb", "Mar", "Apr", "May", "Jun",
  41.         "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
  42.     };
  43.     size_t i;
  44.     for (i = 0; i < sizeof(ar) / sizeof(ar[0]); i++)
  45.     {
  46.         if (!strcmp(s, ar[i]))
  47.         {//遍历查找
  48.             retval = i;
  49.             goto EXITmontoi;
  50.         }
  51.     }    
  52.     DBGPRINT("LCW<==montoi\n");
  53. EXITmontoi:
  54.     return retval;
  55. }
  56. /******************************************************
  57. 函数名：date_to_epoch(char *s)
  58. 参数：
  59. 功能：Parse date-time string, and return the corresponding time_t value
  60. *******************************************************/
  61. static time_t date_to_epoch(char *s)
  62. {
  63.     DBGPRINT("LCW==>date_to_epoch\n");
  64.     struct tm    tm;
  65.     char mon[32];
  66.     int    sec, min, hour, mday, month, year;

  68.     (void) memset(&tm, 0, sizeof(tm));
  69.     sec = min = hour = mday = month = year = 0;

  71.     if (((sscanf(s, "%d/%3s/%d %d:%d:%d",&mday, mon, &year, &hour, &min, &sec) == 6) ||
  72.      (sscanf(s, "%d %3s %d %d:%d:%d",&mday, mon, &year, &hour, &min, &sec) == 6) ||
  73.      (sscanf(s, "%*3s, %d %3s %d %d:%d:%d", &mday, mon, &year, &hour, &min, &sec) == 6) ||
  74.      (sscanf(s, "%d-%3s-%d %d:%d:%d",&mday, mon, &year, &hour, &min, &sec) == 6)) && (month = montoi(mon)) != -1)
  75.     {
  76.         tm.tm_mday    = mday;
  77.         tm.tm_mon    = month;
  78.         tm.tm_year    = year;
  79.         tm.tm_hour    = hour;
  80.         tm.tm_min    = min;
  81.         tm.tm_sec    = sec;
  82.     }

  84.     if (tm.tm_year > 1900)
  85.         tm.tm_year -= 1900;
  86.     else if (tm.tm_year < 70)
  87.         tm.tm_year += 100;

  89.     DBGPRINT("LCW<==date_to_epoch\n");
  90.     return (mktime(&tm));
  91. }
  92. /******************************************************
  93. 函数名：Request_HeaderParse(char *s, int len, struct headers *parsed)
  94. 参数：
  95. 功能：
  96. *******************************************************/
  97. void Request_HeaderParse(char *s, int len, struct headers *parsed)
  98. {
  99.     DBGPRINT("LCW==>Request_HeaderParse\n");
  100.     struct http_header *h;//结构struct http_header指针
  101.     union variant *v;//通用参数
  102.     char *p,*e = s + len;//p当前位置,e尾部
  103.     // 查找请求字符串中的头部关键字
  104.     while (s < e)
  105.     {
  106.         for (p = s; p < e && *p != '\n'; ) //查找一行末尾
  107.         {
  108.             p++;
  109.         }
  110.         for (h = http_headers; h->len != 0; h++)//已知方法
  111.         { //最上面定义的那个结构体数组
  112.             if (e - s > h->len && !strncasecmp(s, h->name, h->len))//字符串匹配
  113.             {
  114.                 break;
  115.             }
  116.         }
  117.         if (h->len != 0) //将此方法放入
  118.         {
  119.             s += h->len;//请求字符串中值的位置
  120.             v = (union variant *) ((char *) parsed + h->offset);//将值存放到参数parsed中
  121.             // 根据头部选项不同,计算不同的值*/
  122.             if (h->type == HDR_STRING) //字符串类型
  123.             {
  124.                 v->v_vec.ptr = s;//字符串开始
  125.                 v->v_vec.len = p - s;//字符串长度
  126.                 if (p[-1] == '\r' && v->v_vec.len > 0)
  127.                 {
  128.                     v->v_vec.len--;
  129.                 }
  130.             }
  131.             else if (h->type == HDR_INT) //INT类型
  132.             {
  133.                 v->v_big_int = strtoul(s, NULL, 10);
  134.             }
  135.             else if (h->type == HDR_DATE) //时间格式
  136.             {
  137.                 v->v_time = date_to_epoch(s);
  138.             }
  139.         }

  141.         s = p + 1;//转到下一个头部
  142.     }
  143.     DBGPRINT("LCW<==Request_HeaderParse\n");
  144. }
  145. #define JUMPOVER_CHAR(p,over) do{for(;*p== over;p++);}while(0);
  146. #define JUMPTO_CHAR(p,to) do{for(;* to;p++);}while(0);
  147. /******************************************************
  148. 函数名：Request_Parse(struct worker_ctl *wctl)
  149. 参数：
  150. 功能：分析获得的请求
  151. *******************************************************/
  152. int Request_Parse(struct worker_ctl *wctl)
  153. {
  154.     DBGPRINT("LCW==>Request_Parse\n");
  155.     struct worker_conn *c = &wctl->conn;
  156.     struct conn_request *req = &c->con_req;
  157.     struct conn_response *res = &c->con_res;
  158.     int retval = 200;
  159.     char *p = req->req.ptr;
  160.     int len = req->req.len;
  161.     char *pos = NULL;
  162.     //处理第一行
  163.     //[GET /root/default.html HTTP/1.1\r\n]*/
  164.     //查找一行末尾
  165.     pos = memchr(p, '\n', len);
  166.     if(*(pos-1) == '\r')
  167.     {
  168.         *(pos-1) = '\0';
  169.     }
  170.     *pos = '\0';
  171.     pos = p;
  172.     //方法    
  173.     int found = 0;
  174.     JUMPOVER_CHAR(pos,' ');//跳过空格
  175.     struct vec *m= NULL;
  176.     //查找比较方法字符串
  177.     for(m = &_shttpd_methods[0];m->ptr!=NULL;m++)
  178.     {
  179.         if(!strncmp(m->ptr, pos, m->len))//比较字符串
  180.         {
  181.             req->method = m->type;//更新头部方法
  182.             found = 1;
  183.             break;
  184.         }
  185.     }
  186.     if(!found)
  187.     {
  188.         retval = 400;
  189.         goto EXITRequest_Parse;
  190.     }

  192.     //URI分析*/
  193.     pos += m->len;//跳过方法
  194.     JUMPOVER_CHAR(pos,' ');//跳过空格
  195.     len -= pos -p;
  196.     p = pos;
  197.     JUMPTO_CHAR(pos, ' ');//跳到空格
  198.     *pos = '\0';
  199.     req->uri = (char*)p;
  200.     //文件
  201.     snprintf(req->rpath, URI_MAX, "%s/%s",conf_para.DocumentRoot, req->uri);
  202.     res->fd = open(req->rpath, O_RDONLY , 0644);
  203.     if(res->fd != -1)
  204.     {
  205.         fstat(res->fd, &res->fsate);    
  206.         if(S_ISDIR(res->fsate.st_mode))
  207.         {
  208.             retval = 403;
  209.             goto EXITRequest_Parse;
  210.         }
  211.     }
  212.     else
  213.     {
  214.         
  215.         retval = 404;
  216.         goto EXITRequest_Parse;
  217.     }
  218.         
  219.     //HTTP版本:
  220.     //    HTTP/[1|0].[1|0|9]
  221.     pos += 1;
  222.     JUMPOVER_CHAR(pos,' ');//跳过空格
  223.     len -= pos -p;
  224.     p = pos;
  225.     sscanf(p,"HTTP/%lu.%lu",&req->major, &req->minor);
  226.     if(!((req->major == 0 && req->minor == 9)||(req->major == 1 && req->minor == 0)||(req->major == 1 && req->minor == 1)))
  227.     {
  228.         retval = 505;
  229.         goto EXITRequest_Parse;
  230.     }
  231.     //其他头部信息
  232.     JUMPTO_CHAR(pos, '\0');
  233.     JUMPOVER_CHAR(pos,'\0');//跳过空字符
  234.     len -= pos - p;
  235.     p = pos;
  236.     Request_HeaderParse(p, len, & req->ch);    
  237.     DBGPRINT("URI:'%s',patch:'%s'\n",req->uri,req->rpath);
  238. EXITRequest_Parse:
  239.     DBGPRINT("LCW<==Request_Parse\n");
  240.     return retval;
  241. }
  242. /******************************************************
  243. 函数名：Request_Handle(struct worker_ctl* wctl)
  244. 参数：
  245. 功能：处理并响应客户端请求
  246. *******************************************************/
  247. int Request_Handle(struct worker_ctl* wctl)
  248. {
  249.     DBGPRINT("LCW==>Request_Handle\n");
  250.     int err = wctl->conn.con_req.err;        
  251.     int cs = wctl->conn.cs;
  252.     int cl = -1;
  253.     char *ptr = wctl->conn.con_res.res.ptr;
  254.     int len = -1;
  255.     int n = -1;
  256.     switch(err)
  257.     {
  258.         case 200:
  259.             Method_Do(wctl);
  260.             int fd = wctl->conn.con_res.fd;
  261.             cl = wctl->conn.con_res.cl;
  262.             len = strlen(wctl->conn.con_res.res.ptr);
  263.             n = write(cs, ptr, len);
  264.             printf("echo header:%s, write to client %d bytes, status:%d\n",ptr,n,wctl->conn.con_res.status);
  265.             if(fd != -1)
  266.             {        
  267.                 lseek(fd, 0, SEEK_SET);
  268.                 len = sizeof(wctl->conn.dres);
  269.                 printf("response len:%d, content length:%d\n",len,wctl->conn.con_res.cl);
  270.                 for(n = 0; cl>0; cl -= n)
  271.                 {
  272.                     n = read(fd,ptr,len>cl?cl:sizeof(wctl->conn.dres));
  273.                     printf("read %d bytes,",n);
  274.                     if(n > 0)
  275.                     {
  276.                         n =write(cs, ptr, n);
  277.                         printf("write %d bytes\n",n);
  278.                     }
  279.                 }
  280.                 close(fd);
  281.                 wctl->conn.con_res.fd = -1;
  282.             }

  284.             break;
  285.         default:
  286.         case 400:
  287.             Error_400(wctl);
  288.             cl = wctl->conn.con_res.cl;
  289.             len = strlen(wctl->conn.con_res.res.ptr);        
  290.             n = write(cs, ptr, len);            
  291.             break;    
  292.         case 403:
  293.             Error_403(wctl);
  294.             cl = wctl->conn.con_res.cl;
  295.             len = strlen(wctl->conn.con_res.res.ptr);        
  296.             n = write(cs, ptr, len);            
  297.             break;    
  298.         case 404:
  299.             Error_404(wctl);
  300.             cl = wctl->conn.con_res.cl;
  301.             len = strlen(wctl->conn.con_res.res.ptr);    
  302.             n = write(cs, ptr, len);            
  303.             break;    
  304.         case 505:
  305.             Error_505(wctl);
  306.             cl = wctl->conn.con_res.cl;
  307.             len = strlen(wctl->conn.con_res.res.ptr);        
  308.             n = write(cs, ptr, len);            
  309.             break;            
  310.     }
  311.     DBGPRINT("LCW<==Request_Handle\n");
  312.     return 0;
  313. }

阅读(1233) | 评论(0) | 转发(0) |
0

上一篇：进程间管道通信

下一篇： 一个简单的wed服务器SHTTPD（3）————SHTTPD多客户端支持的实现

给主人留下些什么吧！~~
关于我们 | 关于IT168 | 联系方式 | 广告合作 | 法律声明 | 免费注册

Copyright 2001-2010 ChinaUnix.net All Rights Reserved 北京皓辰网域网络信息技术有限公司. 版权所有

感谢所有关心和支持过ChinaUnix的朋友们

16024965号-6