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 15 by Douglas Thrift, 2003-11-09T16:51:42-08:00 vs.
Revision 907 by douglas, 2007-04-29T11:27:53-07:00

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

Diff Legend

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