7 |
|
#include "Bender.hpp" |
8 |
|
#include "Matcher.hpp" |
9 |
|
|
10 |
< |
#include <xalanc/Include/PlatformDefinitions.hpp> |
11 |
< |
#include <xercesc/util/PlatformUtils.hpp> |
12 |
< |
#include <xalanc/XalanTransformer/XalanTransformer.hpp> |
10 |
> |
#include <menes-api/exename.hpp> |
11 |
> |
#include <menes-api/files.hpp> |
12 |
> |
#include <menes-api/process.hpp> |
13 |
> |
#include <menes-app/simple.hpp> |
14 |
|
|
15 |
< |
XALAN_USING_XERCES(XMLPlatformUtils) |
15 |
< |
XALAN_USING_XALAN(XalanTransformer) |
16 |
< |
|
17 |
< |
int main(int argc, char* argv[]) |
15 |
> |
struct Environment |
16 |
|
{ |
17 |
< |
XMLPlatformUtils::Initialize(); |
18 |
< |
XalanTransformer::initialize(); |
17 |
> |
ext::String get(const ext::String& name) |
18 |
> |
{ |
19 |
> |
try |
20 |
> |
{ |
21 |
> |
return api::TheEnvironment.Get(name); |
22 |
> |
} |
23 |
> |
catch (ext::Exception) |
24 |
> |
{ |
25 |
> |
return ext::String(); |
26 |
> |
} |
27 |
> |
} |
28 |
> |
} env; |
29 |
|
|
30 |
+ |
int Main(const app::Options& options) |
31 |
+ |
{ |
32 |
|
Bender bender; |
33 |
|
|
24 |
– |
XalanTransformer::terminate(); |
25 |
– |
XMLPlatformUtils::Terminate(); |
26 |
– |
XalanTransformer::ICUCleanUp(); |
27 |
– |
|
34 |
|
return 0; |
35 |
|
} |
36 |
|
|
37 |
|
Bender::Bender() |
38 |
|
{ |
39 |
< |
string path = sgetenv("PATH_TRANSLATED"); |
39 |
> |
ext::String path(env.get("PATH_TRANSLATED")); |
40 |
> |
Matcher matcher("^" + env.get("SCRIPT_NAME")); |
41 |
|
|
42 |
< |
|
36 |
< |
if (path != "") |
42 |
> |
if (!path.IsEmpty() && env.get("REQUEST_URI") != matcher) |
43 |
|
{ |
44 |
< |
ifstream file(path.c_str()); |
39 |
< |
|
40 |
< |
if (file.is_open()) |
44 |
> |
try |
45 |
|
{ |
46 |
< |
file.close(); |
46 |
> |
api::FileReader file(path); |
47 |
|
|
48 |
< |
bend(path, sgetenv("HTTP_USER_AGENT")); |
48 |
> |
bend(path, env.get("HTTP_USER_AGENT")); |
49 |
|
} |
50 |
< |
else |
50 |
> |
catch (ext::Exception e) |
51 |
|
{ |
52 |
< |
cout << "Status: 404\n" |
53 |
< |
<< "Content-Type: text/html; charset=ISO-8859-1\n\n" |
52 |
> |
api::Cout << "Status: 404\r\n" |
53 |
> |
<< "Content-Type: text/html; charset=ISO-8859-1\r\n\r\n" |
54 |
|
<< "<!DOCTYPE HTML PUBLIC \"-//IETF//DTD HTML 2.0//EN\">\n" |
55 |
|
<< "<html><head>\n" |
56 |
|
<< "<title>404 Not Found</title>\n" |
57 |
|
<< "</head><body>\n" |
58 |
|
<< "<h1>Not Found</h1>\n" |
59 |
< |
<< "<p>The requested URL " << sgetenv("PATH_INFO") << " was no" |
60 |
< |
<< "t found on this server.</p>\n" |
59 |
> |
<< "<p>The requested URL " << env.get("PATH_INFO") |
60 |
> |
<< " was not found on this server.</p>\n" |
61 |
|
<< "<hr />\n" |
62 |
< |
<< sgetenv("SERVER_SIGNATURE") |
59 |
< |
<< "</body></html>\n"; |
62 |
> |
<< env.get("SERVER_SIGNATURE") << "</body></html>\n"; |
63 |
|
} |
64 |
|
} |
65 |
|
else |
66 |
|
{ |
67 |
< |
cout << "Location: http://computers.douglasthrift.net/bender.xml\n\n"; |
67 |
> |
api::Cout << "Location: http://computers.douglasthrift.net/bender.xml\r\n\r\n"; |
68 |
|
} |
69 |
|
} |
70 |
|
|
71 |
< |
void Bender::bend(const string& path, const string& agent) |
71 |
> |
void Bender::bend(const ext::String& path, const ext::String& agent) |
72 |
|
{ |
73 |
|
Matcher matcher; |
74 |
|
|
76 |
|
{ |
77 |
|
bend(path); |
78 |
|
} |
79 |
< |
else if (agent == matcher(string("^Mozilla/5.0 \\(.*; rv:(\\d+)\\.(\\d+)(") |
77 |
< |
+ "\\.d+)?(\\w+)?\\) Gecko\\/.*")) |
79 |
> |
else if (agent == matcher("rv:(\\d+)\\.(\\d+).*\\) Gecko")) |
80 |
|
{ |
81 |
< |
int major, minor; |
80 |
< |
istringstream number(matcher[1] + ' ' + matcher[2]); |
81 |
> |
int major_(lexical_cast<int>(matcher[1])), minor_(lexical_cast<int>(matcher[2])); |
82 |
|
|
83 |
< |
number >> major; |
83 |
< |
number >> minor; |
84 |
< |
|
85 |
< |
if (major > 1 || (major == 1 && minor >= 5)) |
83 |
> |
if (major_ > 1 || (major_ == 1 && minor_ >= 5)) |
84 |
|
{ |
85 |
|
pass(path); |
86 |
|
} |
87 |
< |
else |
90 |
< |
{ |
91 |
< |
bend(path); |
92 |
< |
} |
93 |
< |
} |
94 |
< |
else |
95 |
< |
{ |
96 |
< |
bend(path); |
87 |
> |
else bend(path); |
88 |
|
} |
89 |
+ |
else bend(path); |
90 |
|
} |
91 |
|
|
92 |
< |
void Bender::bend(const string& path) |
92 |
> |
void Bender::bend(const ext::String& path) |
93 |
|
{ |
94 |
< |
ostringstream output; |
95 |
< |
XalanTransformer transformer; |
94 |
> |
_L<ext::String> args(1, path); |
95 |
> |
_S<api::Process> xslt("/usr/local/bin/xsltproc", args); |
96 |
> |
ios::String output; |
97 |
|
|
98 |
< |
if (transformer.transform(path.c_str(), output) == 0) |
98 |
> |
ios::ReadToWrite(*xslt.GetReader(), output); |
99 |
> |
|
100 |
> |
if (!output.IsEmpty()) |
101 |
|
{ |
102 |
< |
string type = "text/html; charset=ISO-8859-1"; |
102 |
> |
ext::String type("text/xml"); |
103 |
|
Matcher matcher("http-equiv=\"Content-Type\" content=\"(.*)\""); |
104 |
|
|
105 |
< |
if (matcher == output.str()) |
111 |
< |
{ |
112 |
< |
type = matcher[1]; |
113 |
< |
} |
105 |
> |
if (matcher == output) type = matcher[1]; |
106 |
|
|
107 |
< |
cout << "Content-Type: " << type << "\n\n" << output.str(); |
116 |
< |
} |
117 |
< |
else |
118 |
< |
{ |
119 |
< |
pass(path); |
107 |
> |
api::Cout << "Content-Type: " << type << "\r\n\r\n" << output; |
108 |
|
} |
109 |
+ |
else pass(path); |
110 |
|
} |
111 |
|
|
112 |
< |
void Bender::pass(const string& path) |
112 |
> |
void Bender::pass(const ext::String& path) |
113 |
|
{ |
114 |
< |
cout << "Content-Type: text/xml; charset=ISO-8859-1\n\n"; |
114 |
> |
api::Cout << "Content-Type: text/xml\r\n\r\n"; |
115 |
|
|
116 |
< |
ifstream fin(path.c_str()); |
128 |
< |
string line; |
116 |
> |
api::FileReader file(path); |
117 |
|
|
118 |
< |
do |
131 |
< |
{ |
132 |
< |
getline(fin, line); |
133 |
< |
|
134 |
< |
cout << line << '\n'; |
135 |
< |
} |
136 |
< |
while (fin.good()); |
118 |
> |
ios::ReadToWrite(file, api::Cout); |
119 |
|
} |