一个web服务器的实现（下）

最新推荐文章于 2017-03-18 16:13:28 发布
searchwang
最新推荐文章于 2017-03-18 16:13:28 发布
阅读量395
点赞数
分类专栏：服务器 C++
C++ 同时被 2 个专栏收录
66 篇文章 0 订阅
订阅专栏
服务器
3 篇文章 0 订阅
订阅专栏
 
  /* ------------------------------------------------------ *
 
 
 
   process_rq( char *rq, int fd )
 
 
 
   分析客户端的请求，分析然后做出相应的应答
 
 
 
   rq is HTTP command:  GET /foo/bar.html HTTP/1.0
 
 
 
   ------------------------------------------------------ */
 
 
 
 
 
 
 
process_rq( char *rq, int fd )
 
 
 
{
 
 
 
    char    cmd[BUFSIZ], arg[BUFSIZ];
 
 
 
 
 
 
 
    /* create a new process and return if not the child */
 
 
 
    if ( fork() != 0 )
 
 
 
        return;
 
 
 
 
 
 
 
    strcpy(arg, "./");      /* precede args with ./ */
 
 
 
    if ( sscanf(rq, "%s%s", cmd, arg+2) != 2 )
 
 
 
        return;
 
 
 
 
 
 
 
    if ( strcmp(cmd,"GET") != 0 )
 
 
 
        cannot_do(fd);
 
 
 
    else if ( not_exist( arg ) )
 
 
 
        do_404(arg, fd );
 
 
 
    else if ( isadir( arg ) )
 
 
 
        do_ls( arg, fd );
 
 
 
    else if ( ends_in_cgi( arg ) )
 
 
 
        do_exec( arg, fd );
 
 
 
    else
 
 
 
        do_cat( arg, fd );
 
 
 
}
 
 
 
 
 
 
 
/* ------------------------------------------------------ *
 
 
 
   the reply header thing: all functions need one
 
 
 
   if content_type is NULL then don't send content type
 
 
 
   ------------------------------------------------------ */
 
 
 
 
 
 
 
header( FILE *fp, char *content_type )
 
 
 
{
 
 
 
    fprintf(fp, "HTTP/1.0 200 OK\r\n");
 
 
 
    if ( content_type )
 
 
 
        fprintf(fp, "Content-type: %s\r\n", content_type );
 
 
 
}
 
 
 
 
 
 
 
/* ------------------------------------------------------ *
 
 
 
   simple functions first:
 
 
 
        cannot_do(fd)       unimplemented HTTP command
 
 
 
    and do_404(item,fd)     no such object
 
 
 
   ------------------------------------------------------ */
 
 
 
 
 
 
 
cannot_do(int fd)
 
 
 
{
 
 
 
    FILE    *fp = fdopen(fd,"w");
 
 
 
 
 
 
 
    fprintf(fp, "HTTP/1.0 501 Not Implemented\r\n");
 
 
 
    fprintf(fp, "Content-type: text/plain\r\n");
 
 
 
    fprintf(fp, "\r\n");
 
 
 
 
 
 
 
    fprintf(fp, "That command is not yet implemented\r\n");
 
 
 
    fclose(fp);
 
 
 
}
 
 
 
 
 
 
 
do_404(char *item, int fd)
 
 
 
{
 
 
 
    FILE    *fp = fdopen(fd,"w");
 
 
 
 
 
 
 
    fprintf(fp, "HTTP/1.0 404 Not Found\r\n");
 
 
 
    fprintf(fp, "Content-type: text/plain\r\n");
 
 
 
    fprintf(fp, "\r\n");
 
 
 
 
 
 
 
    fprintf(fp, "The item you requested: %s\r\nis not found\r\n",
 
 
 
            item);
 
 
 
    fclose(fp);
 
 
 
}
 
 
 
 
 
 
 
/* ------------------------------------------------------ *
 
 
 
   the directory listing section
 
 
 
   isadir() uses stat, not_exist() uses stat
 
 
 
   do_ls runs ls. It should not
 
 
 
   ------------------------------------------------------ */
 
 
 
 
 
 
 
isadir(char *f)
 
 
 
{
 
 
 
    struct stat info;
 
 
 
    return ( stat(f, &info) != -1 && S_ISDIR(info.st_mode) );
 
 
 
}
 
 
 
 
 
 
 
not_exist(char *f)
 
 
 
{
 
 
 
    struct stat info;
 
 
 
    return( stat(f,&info) == -1 );
 
 
 
}
 
 
 
 
 
 
 
do_ls(char *dir, int fd)
 
 
 
{
 
 
 
    FILE    *fp ;
 
 
 
 
 
 
 
    fp = fdopen(fd,"w");
 
 
 
    header(fp, "text/plain");
 
 
 
    fprintf(fp,"\r\n");
 
 
 
    fflush(fp);
 
 
 
 
 
 
 
    dup2(fd,1);
 
 
 
    dup2(fd,2);
 
 
 
    close(fd);
 
 
 
    execlp("ls","ls","-l",dir,NULL);
 
 
 
    perror(dir);
 
 
 
    exit(1);
 
 
 
}
 
 
 
 
 
 
 
/* ------------------------------------------------------ *
 
 
 
   the cgi stuff.  function to check extension and
 
 
 
   one to run the program.
 
 
 
   ------------------------------------------------------ */
 
 
 
 
 
 
 
char * file_type(char *f)
 
 
 
/* returns 'extension' of file */
 
 
 
{
 
 
 
    char    *cp;
 
 
 
    if ( (cp = strrchr(f, '.' )) != NULL )
 
 
 
        return cp+1;
 
 
 
    return "";
 
 
 
}
 
 
 
 
 
 
 
ends_in_cgi(char *f)
 
 
 
{
 
 
 
    return ( strcmp( file_type(f), "cgi" ) == 0 );
 
 
 
}
 
 
 
 
 
 
 
do_exec( char *prog, int fd )
 
 
 
{
 
 
 
    FILE    *fp ;
 
 
 
 
 
 
 
    fp = fdopen(fd,"w");
 
 
 
    header(fp, NULL);
 
 
 
    fflush(fp);
 
 
 
    dup2(fd, 1);
 
 
 
    dup2(fd, 2);
 
 
 
    close(fd);
 
 
 
    execl(prog,prog,NULL);
 
 
 
    perror(prog);
 
 
 
}
 
 
 
/* ------------------------------------------------------ *
 
 
 
   do_cat(filename,fd)
 
 
 
   sends back contents after a header
 
 
 
   ------------------------------------------------------ */
 
 
 
 
 
 
 
do_cat(char *f, int fd)
 
 
 
{
 
 
 
    char    *extension = file_type(f);
 
 
 
    char    *content = "text/plain";
 
 
 
    FILE    *fpsock, *fpfile;
 
 
 
    int c;
 
 
 
 
 
 
 
    if ( strcmp(extension,"html") == 0 )
 
 
 
        content = "text/html";
 
 
 
    else if ( strcmp(extension, "gif") == 0 )
 
 
 
        content = "image/gif";
 
 
 
    else if ( strcmp(extension, "jpg") == 0 )
 
 
 
        content = "image/jpeg";
 
 
 
    else if ( strcmp(extension, "jpeg") == 0 )
 
 
 
        content = "image/jpeg";
 
 
 
 
 
 
 
    fpsock = fdopen(fd, "w");
 
 
 
    fpfile = fopen( f , "r");
 
 
 
    if ( fpsock != NULL && fpfile != NULL )
 
 
 
    {
 
 
 
        header( fpsock, content );
 
 
 
        fprintf(fpsock, "\r\n");
 
 
 
        while( (c = getc(fpfile) ) != EOF )
 
 
 
            putc(c, fpsock);
 
 
 
        fclose(fpfile);
 
 
 
        fclose(fpsock);
 
 
 
    }
 
 
 
    exit(0);
 
 
 
}