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
108
109
110
111
112
113
114
115
116
117
118
119
120
|
/* fstatic.cpp
* Copyright (C) 2003 Tommi Maekitalo
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License as
* published by the Free Software Foundation; either version 2 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful, but
* is provided AS IS, WITHOUT ANY WARRANTY; without even the implied
* warranty of MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, and
* NON-INFRINGEMENT. See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
*/
#include "fstatic.h"
#include <tnt/componentfactory.h>
#include <tnt/httprequest.h>
#include <tnt/httpreply.h>
#include <tnt/httperror.h>
#include <tnt/http.h>
#include <tnt/httpheader.h>
#include <tnt/httperror.h>
#include <tnt/comploader.h>
#include <fstream>
#include <cxxtools/log.h>
#include <sys/types.h>
#include <sys/stat.h>
log_define("tntnet.fstatic")
namespace tnt
{
////////////////////////////////////////////////////////////////////////
// factory
//
class FstaticFactory : public tnt::SingletonComponentFactory
{
public:
FstaticFactory(const std::string& componentName)
: tnt::SingletonComponentFactory(componentName)
{ }
virtual tnt::Component* doCreate(const tnt::Compident& ci,
const tnt::Urlmapper& um, tnt::Comploader& cl);
};
tnt::Component* FstaticFactory::doCreate(const tnt::Compident&,
const tnt::Urlmapper&, tnt::Comploader&)
{
return new Fstatic();
}
TNT_COMPONENTFACTORY(fstatic, FstaticFactory)
//////////////////////////////////////////////////////////////////////
// componentdefinition
//
unsigned Fstatic::operator() (tnt::HttpRequest& request,
tnt::HttpReply& reply, cxxtools::QueryParams& qparams)
{
if (!tnt::HttpRequest::checkUrl(request.getPathInfo()))
throw tnt::HttpError(HTTP_BAD_REQUEST, "illegal url");
std::string file;
if (!getDocumentRoot().empty())
file = getDocumentRoot() + '/';
file += request.getPathInfo();
log_debug("file: " << file);
struct stat st;
if (stat(file.c_str(), &st) != 0)
{
log_warn("error in stat for file \"" << file << "\"");
reply.throwNotFound(request.getPathInfo());
}
if (!S_ISREG(st.st_mode))
{
log_warn("no regular file \"" << file << "\"");
reply.throwNotFound(request.getPathInfo());
}
std::ifstream in(file.c_str());
if (!in)
{
log_warn("file \"" << file << "\" not found");
reply.throwNotFound(request.getPathInfo());
}
// set Content-Type
if (request.getArgs().size() > 0)
reply.setContentType(request.getArg(0));
// set Content-Length
reply.setContentLengthHeader(st.st_size);
// set Keep-Alive
if (request.keepAlive())
reply.setHeader(tnt::httpheader::connection,
tnt::httpheader::connectionKeepAlive);
// send data
reply.setDirectMode();
reply.out() << in.rdbuf();
return HTTP_OK;
}
void Fstatic::drop()
{
factory.drop(this);
}
}
|
