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
121
122
123
124
125
|
/*
* Copyright (C) 2003 Tommi Maekitalo
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* As a special exception, you may use this file as part of a free
* software library without restriction. Specifically, if other files
* instantiate templates or use macros or inline functions from this
* file, or you compile this file and link it with other files to
* produce an executable, this file does not by itself cause the
* resulting executable to be covered by the GNU General Public
* License. This exception does not however invalidate any other
* reasons why the executable file might be covered by the GNU Library
* General Public License.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, 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&, const tnt::Urlmapper&, tnt::Comploader&);
};
tnt::Component* FstaticFactory::doCreate(const tnt::Compident&, const tnt::Urlmapper&, tnt::Comploader&)
{ return new Fstatic(); }
TNT_COMPONENTFACTORY(fstatic, FstaticFactory)
//////////////////////////////////////////////////////////////////////
// component definition
//
unsigned Fstatic::operator() (tnt::HttpRequest& request, tnt::HttpReply& reply, tnt::QueryParams&)
{
if (!tnt::HttpRequest::checkUrl(request.getPathInfo()) ||
request.getPathInfo().find('\0') != std::string::npos)
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
log_info("send static file \"" << file << "\" size " << st.st_size << " bytes");
reply.setDirectMode();
reply.out() << in.rdbuf();
return HTTP_OK;
}
void Fstatic::drop()
{ factory.drop(this); }
}
|
