httpserver/app.cpp at master · Monk-Liu/httpserver · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#include "app.h"
#include <boost/filesystem.hpp>

using namespace boost::filesystem;

const static string ROOT = "./static";
const static char * NOTFOUND = "./static/404.html";

Application::Application(HttpRequest &req, HttpResponse &res){
    handler(req,res);
}

int Application::handler(HttpRequest &req, HttpResponse &res){

    string p_str = ROOT+req.getpath(); //过滤 ..
    const char * p_char = p_str.c_str();
    path p(p_str.c_str());
    if(exists(p)){
        if(is_regular_file(p)){
            ifstream fd;
            //cout<<path_str<<endl;  //the path_str is not the same as line 18;
            fd.open(p_char);
            //cout<<p_char<<endl;
            if(!(fd.is_open())){
                return 0;
            }
            string ext = extension(p);
            map<string,string>::iterator it = HttpResponse::MIME_TYPE.find(ext);
            string key, value;
            key = "Content-Type";
            if(it != HttpResponse::MIME_TYPE.end()){
                value = it->second;
            }else{
                value = "text/html"; //MIME TYPE
            }
            res.setHeader(key,value);
            long len;
            len = file_size(p);
            /*fd.seekg(0, ios::end);
            len = fd.tellg();
            fd.seekg(0, ios::beg);a*/
            res.setDataLen(len);
            //redds.setHeader("Content-Length",string(len_cha));
            char buffer[100];
            string wbuf;
            int i = 0;
            int writesize = 0;
            while(!(fd.eof())){
                fd.read(buffer,100);
                i += 100;
                if(i>=len+100){
                    break;
                }
                if(i>len){
                   writesize = len+100-i;
                }else{
                    writesize = 100;
                }
                res.writeData(buffer,writesize);
            }
            fd.close();
        }else if(is_directory(p)){
            vector<path> v;
            string body="<html><body><h2>" + string(p_char) + "</h2><ul>";
            string key, value;
            copy(directory_iterator(p), directory_iterator(), back_inserter(v));

            for(vector<path>::iterator it=v.begin(); it!=v.end(); it++){
                body = body + "<li><a href=\""+ (it->c_str()+9)+ "\">" +it->filename().c_str()+ "</a></li>";
            }
            body = body+"</ul></body></html>";

            key = "Content-Type";
            value = "text/html";
            res.setHeader(key,value);
            long len = body.size();
            res.setDataLen(len);

            res.writeData(body);
        }
    }else{
        ifstream fd;
        fd.open(NOTFOUND);
        if(!fd.is_open()){
            return 0;
        }
        string key, value;
        key = "Content-Type";
        value = "text/html"; //MIME TYPE
        res.setHeader(key,value);
        res.setStatus(404);
        long len;
        fd.seekg(0, ios::end);
        len = fd.tellg();
        fd.seekg(0, ios::beg);
        res.setDataLen(len);
        //redds.setHeader("Content-Length",string(len_cha));
        char buffer[100];
        string wbuf;
        while(!(fd.eof())){
            fd.read(buffer,100);
            wbuf = string(buffer);  //mywrite -> string -> char *;
            res.writeData(wbuf);
        }
        fd.close();
    }
}