ViewVC Help

View File | Revision Log | Show Annotations | Download File | View Changeset | Root Listing

root/repos/Smersh/Smersh.cpp

Comparing Smersh/Smersh.cpp (file contents):
Revision 165 by Douglas Thrift, 2004-06-18T01:38:55-07:00 vs.
Revision 242 by Douglas Thrift, 2004-09-11T21:06:02-07:00

#	Line 10 | Line 10
10		#include "Daemon.hpp"
11		#include "Redirector.hpp"
12
13	+	#include <menes-api/exename.hpp>
14	+	#include <menes-app/application.hpp>
15	+
16		string program;
17		bool debug(false);
18
19	<	int main(int argc, char* argv[])
19	>	struct SmershCommand : public app::Application
20		{
21	<	program = argv[0];
19	<
20	<	int port(8080);
21	<	bool daemon(false), redirector(false);
22	<	string redirect;
23	<
24	<	for (int index(1); index < argc; ++index)
21	>	virtual int Run(const app::ArgumentList& args)
22		{
23	<	string arg(argv[index]);
24	<	Matcher matcher;
23	>	program = api::GetExecutableName();
24	>
25	>	int port(54321);
26	>	bool fork(false), daemon(false), redirector(false);
27	>	string redirect, log("smersh.log");
28
29	<	if (arg == "-D")
29	>	for (size_t index(0); index < args.GetSize(); ++index)
30		{
31	<	if (!debug) debug = true;
31	>	string arg(args[index]);
32	>	Matcher matcher;
33	>
34	>	if (arg == "-daemon")
35	>	{
36	>	if (!daemon) daemon = true;
37	>	}
38	>	else if (arg == matcher("^-redirector=(.+)$"))
39	>	{
40	>	if (!redirector) redirector = true;
41	>
42	>	redirect = matcher[1];
43	>	}
44	>	else if (arg == matcher("^-port=([0-9]+)$"))
45	>	{
46	>	port = lexical_cast<int>(matcher[1]);
47	>	}
48	>	else if (arg == "-fork")
49	>	{
50	>	if (!fork) fork = true;
51	>	}
52	>	else if (arg == matcher("^-log=(.+)$"))
53	>	{
54	>	log = matcher[1];
55	>	}
56	>	else if (arg == "-D")
57	>	{
58	>	if (!debug) debug = true;
59	>	}
60	>	else
61	>	{
62	>	cout << "Usage: " << program
63	>	<< " [-daemon|-redirector=redirect] [-port=port] [-fork] "
64	>	<< "[-log=log] [-D]\n";
65	>
66	>	return 1;
67	>	}
68		}
69	<	else if (arg == "-daemon")
69	>
70	>	if (daemon)
71		{
72	<	if (!daemon) daemon = true;
72	>	Daemon daemon(port, fork, log);
73		}
74	<	else if (arg == matcher("^-redirector=(.+)$"))
74	>	else if (redirector)
75		{
76	<	if (!redirector) redirector = true;
40	<
41	<	redirect = matcher[1];
76	>	Redirector redirector(port, fork, log, redirect);
77		}
78	<	else if (arg == matcher("^-port=([0-9]+)$"))
78	>	else
79		{
80	<	port = lexical_cast<int>(matcher[1]);
80	>	Smersh smersh;
81		}
82	+
83	+	return 0;
84		}
85	+	} smersh;
86	+
87	+	Smersh::Smersh(istream& sin, ostream& sout, const Environment& env)
88	+	{
89	+	parse(sin, env);
90	+	smersh(sout, env);
91	+	}
92	+
93	+	Environment Smersh::env;
94	+
95	+	void Smersh::parse(istream& sin, const Environment& env)
96	+	{
97	+	stringstream query(env.get("QUERY_STRING"));
98
99	<	if (daemon)
99	>	if (env.get("REQUEST_METHOD") == "POST")
100		{
101	<	Daemon daemon(port);
101	>	streamsize length(lexical_cast<streamsize>(env.get("CONTENT_LENGTH")));
102	>	char* content(new char[length]);
103	>
104	>	sin.read(content, length);
105	>	query.write(content, length);
106	>
107	>	delete [] content;
108		}
109	<	else if (redirector)
109	>	if (query.str() == "") return;
110	>
111	>	do
112		{
113	<	Redirector redirector(port, redirect);
113	>	string name, value;
114	>
115	>	getline(query, name, '=');
116	>	getline(query, value, '&');
117	>
118	>	cgi.insert(pair<string, string>(name, value));
119		}
120	<	else
120	>	while (query.good());
121	>	}
122	>
123	>	void Smersh::smersh(ostream& sout, const Environment& env)
124	>	{
125	>	if (&env == &Smersh::env) sout << "Content-Type: text/html; charset=UTF-8\r"
126	>	<< "\n\r\n";
127	>
128	>	vector<Person> people(1);
129	>
130	>	for (multimap<string, string>::iterator itor(cgi.lower_bound("sn")); itor
131	>	!= cgi.upper_bound("sn"); ++itor)
132		{
133	<	Smersh smersh;
133	>	string sn(itor->second);
134	>
135	>	for (string::size_type index(0); index < sn.length(); ++index)
136	>	{
137	>	while (!isalnum(sn[index]) && index < sn.length())
138	>	{
139	>	if (sn[index] == '%' && index + 2 < sn.length())
140	>	{
141	>	istringstream code(sn.substr(index + 1, 2));
142	>	unsigned short character;
143	>
144	>	code.setf(ios_base::hex, ios_base::basefield);
145	>
146	>	code >> character;
147	>
148	>	sn.replace(index, 3, 1, character);
149	>
150	>	if (isalnum(sn[index])) break;
151	>	}
152	>
153	>	sn.erase(index, 1);
154	>	}
155	>	}
156	>
157	>	if (debug) cerr << "sn = " << sn << '\n';
158	>
159	>	Person person(sn);
160	>
161	>	people.push_back(person);
162	>
163	>	if (person.isMultiple())
164	>	{
165	>	people.insert(people.end(), person.beginMultiple(),
166	>	person.endMultiple());
167	>	person.clearMultiple();
168	>	}
169		}
170
171	<	return 0;
172	<	}
171	>	if (people.size() > 1)
172	>	{
173	>	people.erase(people.begin());
174
175	<	Smersh::Smersh(ostream& sout)
66	<	{
67	<	string method = sgetenv("REQUEST_METHOD");
175	>	sort(people.begin(), people.end());
176
177	<	parse(method);
177	>	people.erase(unique(people.begin(), people.end()), people.end());
178	>	}
179
180	<	sout << "Content-Type: text/plain\n\n" << (cgi.find("sn") != cgi.end() ?
72	<	cgi.find("sn")->second : "Unknown") << '\n';
73	<	}
180	>	sout << "<html><head><title>Hello, " << flush;
181
182	<	void Smersh::parse(const string& method)
76	<	{
77	<	string query(sgetenv("QUERY_STRING"));
182	>	output(sout, people);
183
184	<	if (method == "POST") getline(cin, query);
185	<	if (query == "") return;
184	>	sout << "!</title></head><body><p style=\"font-variant: small-caps\"><stro"
185	>	<< "ng><font face=\"Comic Sans MS\" size=\"4\">Hello, " << flush;
186
187	<	istringstream input(query);
187	>	output(sout, people);
188
189	<	do
189	>	sout << "!</font></strong></p></body></html>\r\n";
190	>	}
191	>
192	>	void Smersh::output(ostream& sout, const vector<Person>& people)
193	>	{
194	>	for (vector<Person>::const_iterator person(people.begin()); person !=
195	>	people.end(); ++person)
196		{
197	<	string name, value;
197	>	sout << *person;
198
199	<	getline(input, name, '=');
200	<	getline(input, value, '&');
199	>	if (person + 2 == people.end())
200	>	{
201	>	sout << (people.size() > 2 ? "," : "") << " and/or ";
202	>	}
203	>	else if (person + 1 != people.end())
204	>	{
205	>	sout << ", ";
206	>	}
207
208	<	cgi.insert(pair<string, string>(name, value));
208	>	sout << flush;
209		}
93	–	while (input.good());
210		}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines