ViewVC Help

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

root/repos/HostUpdate/HostUpdate.cpp

Comparing HostUpdate/HostUpdate.cpp (file contents):
Revision 13 by Douglas Thrift, 2003-11-07T12:17:57-08:00 vs.
Revision 24 by Douglas Thrift, 2003-11-11T12:33:38-08:00

#	Line 7 | Line 7
7		#include "HostUpdate.hpp"
8		#include "Host.hpp"
9
10	+	#ifdef _WIN32
11	+	#include <windows.h>
12	+	#endif
13	+
14		int main(int argc, char* argv[])
15		{
16		HostUpdate update;
#	Line 25 | Line 29 | HostUpdate::HostUpdate()
29
30		delete [] hosts;
31
32	<	chdir("hosts");
33	<
30	<	CgiEnvironment env = cgi.getEnvironment();
32	>	string agent = sgetenv("HTTP_USER_AGENT");
33	>	string method = sgetenv("REQUEST_METHOD");
34
35	<	string agent = env.getUserAgent();
36	<	string method = env.getRequestMethod();
35	>	parse(method);
36	>	mode();
37
38		if (agent.find("Host Update/") == 0 && method == "POST" && agent.find(" (")
39		!= string::npos && agent.find(") libwww-perl/") != string::npos)
40		{
41	<	update(env, agent);
41	>	update(agent);
42		}
43		else
44		{
45	<	display(env);
45	>	display();
46		}
47		}
48
49	<	HostUpdate::~HostUpdate()
49	>	void HostUpdate::parse(const string& method)
50		{
51	<	//
51	>	string query;
52	>
53	>	if (method == "POST")
54	>	{
55	>	getline(cin, query);
56	>	}
57	>	else
58	>	{
59	>	query = sgetenv("QUERY_STRING");
60	>	}
61	>
62	>	if (query == "") return;
63	>
64	>	istringstream input(query);
65	>
66	>	do
67	>	{
68	>	string name, value;
69	>
70	>	getline(input, name, '=');
71	>	getline(input, value, '&');
72	>
73	>	cgi.insert(pair<string, string>(name, value));
74	>	}
75	>	while (input.good());
76		}
77
78	<	void HostUpdate::update(CgiEnvironment& env, const string& agent)
78	>	void HostUpdate::mode()
79	>	{
80	>	host = false, name = false, address = false, platform = false, since =
81	>	false;
82	>
83	>	for (multimap<string, string>::iterator itor = cgi.find("mode"); itor !=
84	>	cgi.upper_bound("mode") && itor != cgi.end(); itor++)
85	>	{
86	>	string mode = itor->second;
87	>
88	>	for (string::iterator itor = mode.begin(); itor != mode.end(); itor++)
89	>	{
90	>	char mode = *itor;
91	>
92	>	switch (mode)
93	>	{
94	>	case 'h':
95	>	if (!host) host = true;
96	>	break;
97	>	case 'n':
98	>	if (!name) name = true;
99	>	break;
100	>	case 'a':
101	>	if (!address) address = true;
102	>	break;
103	>	case 'p':
104	>	if (!platform) platform = true;
105	>	break;
106	>	case 's':
107	>	if (!since) since = true;
108	>	break;
109	>	default:
110	>	break;
111	>	}
112	>	}
113	>	}
114	>
115	>	if (!host && !name && !address && !platform && !since) host = true, name =
116	>	true, address = true, platform = true, since = true;
117	>	}
118	>
119	>	void HostUpdate::update(const string& agent)
120		{
121		cout << "Content-Type: text/plain\n\n";
122
123	<	vector<FormEntry> entries;
123	>	multimap<string, string>::iterator itor = cgi.find("host");
124
125	<	if (!cgi.getElement("host", entries)) return;
126	<	if (entries[0].isEmpty()) return;
125	>	if (itor == cgi.end()) return;
126	>	if (itor->second == "") return;
127
128	<	string host = entries[0].getStrippedValue(), name = env.getRemoteHost(),
129	<	address = env.getRemoteAddr();
128	>	string host = itor->second, name = sgetenv("REMOTE_HOST"), address =
129	>	sgetenv("REMOTE_ADDR");
130
131	<	if (name == address) name = "";
131	>	string::size_type begin = agent.find('(') + 1, end = agent.rfind(')');
132
133	<	string::size_type begin = agent.find('(') + 1, end = agent.find(')',
66	<	begin);
67	<	string platform = agent.substr(begin, end - begin);
133	>	if (begin >= end) return;
134
135	<	cout << "host=" << host << '\n'
70	<	<< "name=" << name << '\n'
71	<	<< "address=" << address << '\n'
72	<	<< "platform=" << platform << '\n';
135	>	string platform = agent.substr(begin, end - begin);
136
137		Host client(host, name, address, platform), saved(host);
138
139		if (client != ++saved) client--;
140	+
141	+	display(client);
142		}
143
144	<	void HostUpdate::display(CgiEnvironment& env)
144	>	void HostUpdate::display()
145		{
146	<	cout << "Status: 401\nWWW-Authenticate: Basic realm=\"Host Update\"\n\n";
146	>	cout << "Content-Type: text/plain\n\n";
147	>
148	>	{
149	>	bool request = false;
150	>
151	>	for (multimap<string, string>::iterator itor = cgi.find("host"); itor !=
152	>	cgi.upper_bound("host") && itor != cgi.end(); itor++)
153	>	{
154	>	if (itor->second != "")
155	>	{
156	>	Host host(itor->second);
157	>
158	>	display(++host);
159	>
160	>	if (!request) request = true;
161	>	}
162	>	}
163	>
164	>	if (request) return;
165	>	}
166	>
167	>	set<Host> hosts;
168	>
169	>	#ifndef _WIN32
170	>	DIR* dir = opendir("hosts");
171	>	struct dirent* ent;
172	>
173	>	while ((ent = readdir(dir)) != NULL)
174	>	{
175	>	string file = ent->d_name;
176	>
177	>	if (file == "." || file == "..") continue;
178	>
179	>	Host host(file);
180	>
181	>	hosts.insert(++host);
182	>	}
183	>
184	>	closedir(dir);
185	>	#else
186	>	WIN32_FIND_DATA found;
187	>	HANDLE find = FindFirstFile("hosts\\*", &found);
188	>
189	>	do
190	>	{
191	>	string file = found.cFileName;
192	>
193	>	if (file == "." || file == "..") continue;
194	>
195	>	Host host(file);
196	>
197	>	hosts.insert(++host);
198	>	}
199	>	while (FindNextFile(find, &found) != 0);
200	>
201	>	FindClose(find);
202	>	#endif
203	>
204	>	for (set<Host>::iterator itor = hosts.begin(); itor != hosts.end(); itor++)
205	>	{
206	>	display(*itor);
207	>	}
208	>	}
209	>
210	>	void HostUpdate::display(const Host& host)
211	>	{
212	>	if (this->host) cout << "host=" << host.getHost() << '\n';
213	>	if (name) cout << "name=" << host.getName() << '\n';
214	>	if (address) cout << "address=" << host.getAddress() << '\n';
215	>	if (platform) cout << "platform=" << host.getPlatform() << '\n';
216	>
217	>	string name = string("hosts") + slash + host.getHost();
218	>	struct stat file;
219	>
220	>	if (since && stat(name.c_str(), &file) == 0)
221	>	{
222	>	char since[20];
223	>
224	>	strftime(since, 20, "%m/%d/%Y %H:%M:%S", gmtime(&file.st_mtime));
225	>
226	>	cout << "since=" << since << " GMT\n";
227	>	}
228	>	else if (since)
229	>	{
230	>	char since[20];
231	>	time_t* now = new time_t;
232	>
233	>	time(now);
234	>	strftime(since, 20, "%m/%d/%Y %H:%M:%S", gmtime(now));
235	>
236	>	delete now;
237	>
238	>	cout << "since=" << since << " GMT\n";
239	>	}
240		}

Diff Legend

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