Chinaunix首页 | 论坛 | 博客

_小林_

首页　| 　博文目录　| 　关于我

SCUT_Lin

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

全部博文（21）

文章存档

2015年（21）

我的朋友
最近访客
推荐博文
相关博文
一个简单的wed服务器SHTTPD（4）————SHTTPD支持CGI的实现

分类： LINUX

2015-08-07 23:54:50


点击(此处)折叠或打开

  1. //start from the very beginning,and to create greatness
  2. //@author: Chuangwei Lin
  3. //@E-mail：979951191@qq.com
  4. //@brief： SHTTPD支持CGI的实现
  5. #include "lcw_shttpd.h"
  6. /******************************************************
  7. 函数名： GenerateDirFile(struct worker_ctl *wctl)
  8. 参数：
  9. 功能：生成目录下的文件列表
  10. *******************************************************/
  11. int GenerateDirFile(struct worker_ctl *wctl)
  12. {
  13.     struct conn_request *req = &wctl->conn.con_req;
  14.     struct conn_response *res = &wctl->conn.con_res;
  15.     char *command = strstr(req->uri, CGISTR) + strlen(CGISTR);
  16.     char *arg[ARGNUM];
  17.     int num = 0;
  18.     char *rpath = wctl->conn.con_req.rpath;
  19.     stat *fs = &wctl->conn.con_res.fsate;
  20.     //打开目录
  21.     DIR *dir = opendir(rpath);
  22.     if(dir == NULL)
  23.     {
  24.         //错误
  25.         res->status = 500;
  26.         retval = -1;
  27.         goto EXITgenerateIndex;
  28.     }
  29.     //建立临时文件保存目录列表
  30.     File *tmpfile;
  31.     char tmpbuff[2048];
  32.     int filesize = 0;
  33.     char *uri = wctl->conn.con_req.uri;
  34.     //以wb+形式创建一个临时二进制文件
  35.     tmpfile = tmpfile();
  36.     //标题部分
  37.     sprintf(tmpbuff,
  38.         "%s%s%s",
  39.         "<!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 3.2 Final//EN\">\n<HTML><HEAD><TITLE>",
  40.         uri,
  41.         "</TITLE></HEAD>\n"    );
  42.     fprintf(tmpfile, "%s", tmpbuff);
  43.     filesize += strlen(tmpbuff);
  44.     //标识部分
  45.     sprintf(tmpbuff,
  46.         "%s %s %s",
  47.         "<BODY><H1>Index of:",
  48.         uri,
  49.         " </H1> <HR><P><I>Date: </I> <I>Size: </I></P><HR>");
  50.     fprintf(tmpfile, "%s", tmpbuff);
  51.     filesize += strlen(tmpbuff);    
  52.     //读取目录中的文件列表
  53.     struct dirent *de;
  54.     #define PATHLENGTH 2048
  55.     char path[PATHLENGTH];
  56.     char tmpath[PATHLENGTH];
  57.     char linkname[PATHLENGTH];
  58.     struct stat fs;//stat 结构定义于：/usr/include/sys/stat.h 文件中
  59.     strcpy(path, rpath);
  60.     if(rpath[strlen(rpath)]!='/')
  61.     {
  62.         rpath[strlen(rpath)]='/';
  63.     }
  64.     while ((de = readdir(dir)) != NULL)//读取一个文件
  65.     {
  66.         menset(tmpath, 0, sizeof(tmpath));
  67.         menset(linkname, 0, sizeof(linkname));
  68.         if(strcmp(de->d_name, "."))//不是当前目录
  69.         {
  70.             if(strcmp(de->d_name, ".."))//不是父目录            
  71.             {
  72.                 strcpy(linkname,de->d_name);//将目录名称作为链接名称
  73.             }
  74.             else//是父目录
  75.             {
  76.                 strcpy(linkname, "Parent Directory");//将父目录作为链接名称
  77.             }

  79.             sprintf(tmpath, "%s%s",path, de->d_name);//构建当前文件的全路径
  80.             stat(tmpath, &fs);//获得文件信息
  81.             if(S_ISDIR(fs.st_mode))//是一个目录
  82.             {
  83.                 //打印目录的连接为目录名称
  84.                 sprintf(tmpbuff, "<A HREF=\"%s/\">%s/</A><BR>\n", de->d_name,tmpath);
  85.             }
  86.             else//正常文件
  87.             {
  88.                 char size_str[32];
  89.                 off_t size_int;
  90.                 size_int = fs.st_size;//文件大小
  91.                 if (size_int < 1024)//不到1K
  92.                     sprintf(size_str, "%d bytes", (int) size_int);
  93.                 else if (size_int < 1024*1024)//不到1M
  94.                     sprintf(size_str, "%1.2f Kbytes", (float) size_int / 1024);
  95.                 else//其他
  96.                     sprintf(size_str, "%1.2f Mbytes", (float) size_int / (1024*1024));
  97.                 //输出文件大小
  98.                 sprintf(tmpbuff, "<A HREF=\"%s\">%s</A> (%s)<BR>\n", de->d_name, linkname, size_int);
  99.             }
  100.             //将形成的字符串写入临时文件
  101.             fprintf(tmpfile, "%s", tmpbuff);
  102.             filesize += strlen(tmpbuff);                
  103.         }
  104.     }
  105.     //生成临时的文件信息，主要是文件大小
  106.     fs.st_ctime = time(NULL);
  107.     fs.st_mtime = time(NULL);
  108.     fs.st_size = filesize;
  109.     fseek(tmpfile, (long) 0, SEEK_SET);//移动文件指针到头部
  110. EXITgenerateIndex:
  111.     return 0;
  112. }

  114. #define CGISTR "/cgi-bin/"//CGI目录的字符串
  115. #define ARGNUM 16 //CGI程序变量的最大个数
  116. #define READIN 0 //读出管道
  117. #define WRITEOUT 1 //写入管道
  118. /******************************************************
  119. 函数名：cgiHandler(struct worker_ctl *wctl)
  120. 参数：
  121. 功能：
  122. *******************************************************/
  123. int cgiHandler(struct worker_ctl *wctl)
  124. {
  125.     struct conn_request *req = &wctl->conn.con_req;
  126.     struct conn_response *res = &wctl->conn.con_res;
  127.     //strstr(str1,str2);str1:被查找目标 str2:要查找对象　
  128.     char *command = strstr(req->uri, CGISTR) + strlen(CGISTR);//获得匹配字符串/cgi-bin/
  129.     char *arg[ARGNUM];
  130.     int num = 0;
  131.     char *rpath = wctl->conn.con_req.rpath;
  132.     stat *fs = &wctl->conn.con_res.fsate;
  133.     int retval = -1;
  134.     char *pos = command;//查找CGI的命令
  135.     for(;*pos != '?' && *pos !='\0';pos++);//找到命令尾
  136.      {
  137.          *pos = '\0';
  138.      }
  139.     sprintf(rpath, "%s%s",conf_para.CGIRoot,command);//构建全路径
  140.     //查找CGI的参数
  141.     pos++;
  142.     for(;*pos != '\0' && num < ARGNUM;)
  143.     {    //CGI的参数为紧跟CGI命令后的？的字符串，多个变量之间用+连接起来，所以可以根据加号的个数确定参数的个数
  144.         arg[num] = pos;//参数头
  145.         for(;*pos != '+' && *pos!='\0';pos++);
  146.         if(*pos == '+')
  147.         {
  148.             *pos = '\0';//参数尾
  149.             pos++;
  150.             num++;
  151.         }
  152.     }
  153.     arg[num] = NULL;
  154.     //命令的属性
  155.     if(stat(rpath,fs)<0)
  156.     {
  157.         //错误
  158.         res->status = 403;
  159.         retval = -1;
  160.         goto EXITcgiHandler;
  161.     }
  162.     else if((fs->st_mode & S_IFDIR) == S_IFDIR)
  163.     {
  164.         //是一个目录,列出目录下的文件
  165.         GenerateDirFile(wctl);
  166.         retval = 0;
  167.         goto EXITcgiHandler;
  168.     }
  169.     else if((fs->st_mode & S_IXUSR) != S_IXUSR)
  170.     {
  171.         //所指文件不能执行
  172.         res->status = 403;
  173.         retval = -1;
  174.         goto EXITcgiHandler;
  175.     }
  176.     //创建进程间通信的管道    
  177.     int pipe_in[2];
  178.     int pipe_out[2];

  180.     if(pipe(pipe_in) < 0)//创建管道
  181.     {
  182.         res->status = 500;
  183.         retval = -1;
  184.         goto EXITcgiHandler;
  185.     }
  186.     if(pipe(pipe_out) < 0)
  187.     {
  188.         res->status = 500;
  189.         retval = -1;
  190.         goto EXITcgiHandler;
  191.     }
  192.     //进程分叉
  193.     int pid = 0;
  194.     pid = fork();
  195.     if(pid < 0)//错误
  196.     {
  197.         res->status = 500;
  198.         retval = -1;
  199.         goto EXITcgiHandler;
  200.     }
  201.     else if(pid > 0)//父进程
  202.     {
  203.         close(pipe_out[WRITEOUT]);//关闭写端
  204.         close(pipe_in[READIN]);//关闭读端
  205.         //主进程从CGI的标准输出读取数据    ，并将数据发送到网络资源请求的客户端
  206.         int size = 0;//这里初始化为0，怎么进入while循环？改为下面的情况
  207.         int end = 0;
  208.         //读取CGI进程数据
  209.         size = read(pipe_out[READIN], res->res.ptr, sizeof(wctl->conn.dres));
  210.         while(size > 0 && !end)
  211.         {
  212.             if(size > 0)
  213.             {//将数据发送给客户端
  214.                 send(wctl->conn.cs, res->res.ptr, strlen(res->res.ptr));
  215.             }
  216.             else
  217.             {
  218.                 end = 1;
  219.             }
  220.             size = read(pipe_out[READIN], res->res.ptr, sizeof(wctl->conn.dres));
  221.         }
  222.         wait(&end);//等待其子进程全部结束
  223.         close(pipe_out[READIN]);//关闭管道
  224.         close(pipe_in[WRITEOUT]);
  225.         retval = 0;
  226.     }
  227.     else//子进程
  228.     {
  229.         char cmdarg[2048];
  230.         char onearg[2048];
  231.         char *pos = NULL;
  232.         int i = 0;
  233.         //形成执行命令
  234.         memset(onearg, 0, 2048];
  235.         for(i = 0;i<num;i++)
  236.             sprintf(cmdarg,"%s %s", onearg, arg[i]);
  237.         //将写入的管道绑定到标注输出
  238.         close(pipe_out[READIN]); //关闭无用的读管道
  239.         dup2(pipe_out[WRITEOUT], 1); //将写管道绑定到标准输出
  240.         close(pipe_out[WRITEOUT]); //关闭写管道

  242.         close(pipe_in[WRITEOUT]); // 关闭无用的写管道
  243.         dup2(pipe_in[READIN], 0); // 将读管道绑定到标准输入
  244.         close(pipe_in[READIN]); // 关闭写管道
  245.         //execlp()会从PATH 环境变量所指的目录中查找符合参数file的文件名，
  246.         //找到后便执行该文件，然后将第二个以后的参数当做该文件的argv[0]、
  247.         //argv[1]……，最后一个参数必须用空指针(NULL)作结束
  248.         execlp(rpath, arg);//执行命令,命令的输出需要为标准输出     
  249.     }
  250. EXITcgiHandler:
  251.     return retval;
  252. }

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

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

下一篇： 一个简单的wed服务器SHTTPD（5）————服务器SHTTPD请求方法解析

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

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

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

16024965号-6