trunk/Search/Page.cpp

/* ============================================================================
 * Douglas Thrift's Search Engine License
 *
 * Copyright (C) 2002-2004, Douglas Thrift. All Rights Reserved.
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 *    this list of conditions and the following disclaimer in the documentation
 *    and/or other materials provided with the distribution.
 *
 * 3. The end-user documentation included with the redistribution, if any, must
 *    include the following acknowledgment:
 *
 *       "This product includes software developed by Douglas Thrift
 *       (http://computers.douglasthrift.net/searchengine/)."
 *
 *    Alternately, this acknowledgment may appear in the software itself, if
 *    and wherever such third-party acknowledgments normally appear.
 *
 * 4. The names "Douglas Thrift" and "Douglas Thrift's Search Engine" must not
 *    be used to endorse or promote products derived from this software without
 *    specific prior written permission.  For written permission, please visit
 *    http://www.douglasthrift.net/contact.cgi for contact information.
 *
 * 5. Products derived from this software may not be called "Douglas Thrift's
 *    Search Engine", nor may "Douglas Thrift's Search Engine" appear in their
 *    name, without prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
 * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
 * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 * ============================================================================
 */
// Douglas Thrift's Search Engine Page
//
// Douglas Thrift
//
// $Id$

#include "Page.hpp"

istream& operator>>(istream& is, Page& data)
{
        string line;
#ifdef _OpenSSL_
        bool set = false;
#endif
        
        getline(is, line);
        
        if (line == "   <page>")
        {
                do
                {
                        getline(is, line);

                        if (line.find("         <address>") == 0)
                        {
                                unsigned start = line.find("<address>") + 9;
                                unsigned finish = line.find("</address>", start);

                                string address = line.substr(start, finish - start);

                                entities(address, "&lt;", '<');
                                entities(address, "&gt;", '>');
                                entities(address, "&amp;", '&');

                                data.setAddress(address);
                        }
                        else if (line.find("            <port>") == 0)
                        {
                                unsigned start = line.find("<port>") + 6;
                                unsigned finish = line.find("</port>", start);

                                istringstream buffer(line.substr(start, finish - start));
                                unsigned port;

                                buffer >> port;

                                data.setPort(port);

#ifdef _OpenSSL_
                                set = true;
#endif
                        }
#ifdef _OpenSSL_
                        else if (line.find("            <tls>") == 0)
                        {
                                unsigned start = line.find("<tls>") + 5;
                                unsigned finish = line.find("</tls>", start);

                                istringstream buffer(line.substr(start, finish - start));
                                bool tls;

                                buffer.setf(ios_base::boolalpha);
                                buffer >> tls;

                                data.setTls(tls);

                                if (!set && tls) data.setPort(443);
                        }
#endif
                        else if (line.find("            <path>") == 0)
                        {
                                unsigned start = line.find("<path>") + 6;
                                unsigned finish = line.find("</path>", start);

                                string path = line.substr(start, finish - start);

                                entities(path, "&lt;", '<');
                                entities(path, "&gt;", '>');
                                entities(path, "&amp;", '&');

                                data.setPath(path);
                        }
                        else if (line.find("            <size>") == 0)
                        {
                                unsigned start = line.find("<size>") + 6;
                                unsigned finish = line.find("</size>", start);

                                istringstream buffer(line.substr(start, finish - start));
                                unsigned size;

                                buffer >> size;

                                data.setSize(size);
                        }
                        else if (line.find("            <title>") == 0)
                        {
                                unsigned start = line.find("<title>") + 7;
                                unsigned finish = line.find("</title>", start);

                                string title = line.substr(start, finish - start);

                                while (finish == string::npos)
                                {
                                        getline(is, line);
                                        finish = line.find("</title>");
                                        title += '\n' + line.substr(0, finish - 0);
                                }

                                entities(title, "&lt;", '<');
                                entities(title, "&gt;", '>');
                                entities(title, "&amp;", '&');

                                data.setTitle(title);
                        }
                        else if (line.find("            <description>") == 0)
                        {
                                unsigned start = line.find("<description>") + 13;
                                unsigned finish = line.find("</description>", start);

                                string description = line.substr(start, finish - start);

                                entities(description, "&lt;", '<');
                                entities(description, "&gt;", '>');
                                entities(description, "&amp;", '&');

                                data.setDescription(description);
                        }
                        else if (line.find("            <text>") == 0)
                        {
                                unsigned start = line.find("<text>") + 6;
                                unsigned finish = line.find("</text>", start);

                                string text = line.substr(start, finish - start);

                                while (finish == string::npos)
                                {
                                        getline(is, line);
                                        finish = line.find("</text>");
                                        text += '\n' + line.substr(0, finish - 0);
                                }

                                entities(text, "&lt;", '<');
                                entities(text, "&gt;", '>');
                                entities(text, "&amp;", '&');

                                data.setText(text);
                        }
                        else if (line.find("            <heading>") == 0)
                        {
                                unsigned start = line.find("<heading>") + 9;
                                unsigned finish = line.find("</heading>", start);

                                string heading = line.substr(start, finish - start);

                                while (finish == string::npos)
                                {
                                        getline(is, line);
                                        finish = line.find("</heading>");
                                        heading += line.substr(0, finish - 0);
                                }

                                entities(heading, "&lt;", '<');
                                entities(heading, "&gt;", '>');
                                entities(heading, "&amp;", '&');

                                data.headings.push_back(heading);
                        }
                }
                while (line != "        </page>");
        }

        return is;
}

ostream& operator<<(ostream& os, Page& data)
{
        string address = data.getAddress();

        entities(address, '&', "&amp;");
        entities(address, '<', "&lt;");
        entities(address, '>', "&gt;");

        os << " <page>\n" << "          <address>" << address << "</address>\n";

#ifndef _OpenSSL_
        if (data.getPort() != 80)
#else
        if ((data.getPort() != 80 && !data.getTls()) || (data.getPort() != 443 &&
                data.getTls()))
#endif
        {
                os << "         <port>" << data.getPort() << "</port>\n";
        }

#ifdef _OpenSSL_
        os.setf(ios::boolalpha);
        os << "         <tls>" << data.getTls() << "</tls>\n";
#endif

        string path = data.getPath();

        entities(path, '&', "&amp;");
        entities(path, '<', "&lt;");
        entities(path, '>', "&gt;");

        os << "         <path>" << path << "</path>\n";

        os << "         <size>" << data.getSize() << "</size>\n";

        if(data.getTitle() != "")
        {
                string title = data.getTitle();

                entities(title, '&', "&amp;");
                entities(title, '<', "&lt;");
                entities(title, '>', "&gt;");

                os << "         <title>" << title << "</title>\n";
        }

        if(data.getDescription() != "")
        {
                string description = data.getDescription();

                entities(description, '&', "&amp;");
                entities(description, '<', "&lt;");
                entities(description, '>', "&gt;");

                os << "         <description>" << description << "</description>\n";
        }

        string text = data.getText();

        entities(text, '&', "&amp;");
        entities(text, '<', "&lt;");
        entities(text, '>', "&gt;");

        os << "         <text>" << text << "</text>\n";

        for (size_t index(0); index < data.getHeadings().size(); index++)
        {
                string heading = data.getHeadings()[index];

                entities(heading, '&', "&amp;");
                entities(heading, '<', "&lt;");
                entities(heading, '>', "&gt;");

                os << "         <heading>" << heading << "</heading>\n";
        }

        os << " </page>";

        return os;
}
Revision:	348
Committed:	2004-05-26T17:44:17-07:00 (21 years ago) by Douglas Thrift
File size:	8062 byte(s)
Log Message:	Lots of C++ifying!
#	Content
1	/* ============================================================================
2	* Douglas Thrift's Search Engine License
3	*
4	* Copyright (C) 2002-2004, Douglas Thrift. All Rights Reserved.
5	* Redistribution and use in source and binary forms, with or without
6	* modification, are permitted provided that the following conditions are met:
7	*
8	* 1. Redistributions of source code must retain the above copyright notice,
9	* this list of conditions and the following disclaimer.
10	*
11	* 2. Redistributions in binary form must reproduce the above copyright notice,
12	* this list of conditions and the following disclaimer in the documentation
13	* and/or other materials provided with the distribution.
14	*
15	* 3. The end-user documentation included with the redistribution, if any, must
16	* include the following acknowledgment:
17	*
18	* "This product includes software developed by Douglas Thrift
19	* (http://computers.douglasthrift.net/searchengine/)."
20	*
21	* Alternately, this acknowledgment may appear in the software itself, if
22	* and wherever such third-party acknowledgments normally appear.
23	*
24	* 4. The names "Douglas Thrift" and "Douglas Thrift's Search Engine" must not
25	* be used to endorse or promote products derived from this software without
26	* specific prior written permission. For written permission, please visit
27	* http://www.douglasthrift.net/contact.cgi for contact information.
28	*
29	* 5. Products derived from this software may not be called "Douglas Thrift's
30	* Search Engine", nor may "Douglas Thrift's Search Engine" appear in their
31	* name, without prior written permission.
32	*
33	* THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
34	* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
35	* FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
36	* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
37	* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
38	* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
39	* OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
40	* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
41	* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
42	* EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
43	* ============================================================================
44	*/
45	// Douglas Thrift's Search Engine Page
46	//
47	// Douglas Thrift
48	//
49	// $Id$
50
51	#include "Page.hpp"
52
53	istream& operator>>(istream& is, Page& data)
54	{
55	string line;
56	#ifdef _OpenSSL_
57	bool set = false;
58	#endif
59
60	getline(is, line);
61
62	if (line == " <page>")
63	{
64	do
65	{
66	getline(is, line);
67
68	if (line.find(" <address>") == 0)
69	{
70	unsigned start = line.find("<address>") + 9;
71	unsigned finish = line.find("</address>", start);
72
73	string address = line.substr(start, finish - start);
74
75	entities(address, "<", '<');
76	entities(address, ">", '>');
77	entities(address, "&", '&');
78
79	data.setAddress(address);
80	}
81	else if (line.find(" <port>") == 0)
82	{
83	unsigned start = line.find("<port>") + 6;
84	unsigned finish = line.find("</port>", start);
85
86	istringstream buffer(line.substr(start, finish - start));
87	unsigned port;
88
89	buffer >> port;
90
91	data.setPort(port);
92
93	#ifdef _OpenSSL_
94	set = true;
95	#endif
96	}
97	#ifdef _OpenSSL_
98	else if (line.find(" <tls>") == 0)
99	{
100	unsigned start = line.find("<tls>") + 5;
101	unsigned finish = line.find("</tls>", start);
102
103	istringstream buffer(line.substr(start, finish - start));
104	bool tls;
105
106	buffer.setf(ios_base::boolalpha);
107	buffer >> tls;
108
109	data.setTls(tls);
110
111	if (!set && tls) data.setPort(443);
112	}
113	#endif
114	else if (line.find(" <path>") == 0)
115	{
116	unsigned start = line.find("<path>") + 6;
117	unsigned finish = line.find("</path>", start);
118
119	string path = line.substr(start, finish - start);
120
121	entities(path, "<", '<');
122	entities(path, ">", '>');
123	entities(path, "&", '&');
124
125	data.setPath(path);
126	}
127	else if (line.find(" <size>") == 0)
128	{
129	unsigned start = line.find("<size>") + 6;
130	unsigned finish = line.find("</size>", start);
131
132	istringstream buffer(line.substr(start, finish - start));
133	unsigned size;
134
135	buffer >> size;
136
137	data.setSize(size);
138	}
139	else if (line.find(" <title>") == 0)
140	{
141	unsigned start = line.find("<title>") + 7;
142	unsigned finish = line.find("</title>", start);
143
144	string title = line.substr(start, finish - start);
145
146	while (finish == string::npos)
147	{
148	getline(is, line);
149	finish = line.find("</title>");
150	title += '\n' + line.substr(0, finish - 0);
151	}
152
153	entities(title, "<", '<');
154	entities(title, ">", '>');
155	entities(title, "&", '&');
156
157	data.setTitle(title);
158	}
159	else if (line.find(" <description>") == 0)
160	{
161	unsigned start = line.find("<description>") + 13;
162	unsigned finish = line.find("</description>", start);
163
164	string description = line.substr(start, finish - start);
165
166	entities(description, "<", '<');
167	entities(description, ">", '>');
168	entities(description, "&", '&');
169
170	data.setDescription(description);
171	}
172	else if (line.find(" <text>") == 0)
173	{
174	unsigned start = line.find("<text>") + 6;
175	unsigned finish = line.find("</text>", start);
176
177	string text = line.substr(start, finish - start);
178
179	while (finish == string::npos)
180	{
181	getline(is, line);
182	finish = line.find("</text>");
183	text += '\n' + line.substr(0, finish - 0);
184	}
185
186	entities(text, "<", '<');
187	entities(text, ">", '>');
188	entities(text, "&", '&');
189
190	data.setText(text);
191	}
192	else if (line.find(" <heading>") == 0)
193	{
194	unsigned start = line.find("<heading>") + 9;
195	unsigned finish = line.find("</heading>", start);
196
197	string heading = line.substr(start, finish - start);
198
199	while (finish == string::npos)
200	{
201	getline(is, line);
202	finish = line.find("</heading>");
203	heading += line.substr(0, finish - 0);
204	}
205
206	entities(heading, "<", '<');
207	entities(heading, ">", '>');
208	entities(heading, "&", '&');
209
210	data.headings.push_back(heading);
211	}
212	}
213	while (line != " </page>");
214	}
215
216	return is;
217	}
218
219	ostream& operator<<(ostream& os, Page& data)
220	{
221	string address = data.getAddress();
222
223	entities(address, '&', "&");
224	entities(address, '<', "<");
225	entities(address, '>', ">");
226
227	os << " <page>\n" << " <address>" << address << "</address>\n";
228
229	#ifndef _OpenSSL_
230	if (data.getPort() != 80)
231	#else
232	if ((data.getPort() != 80 && !data.getTls()) \|\| (data.getPort() != 443 &&
233	data.getTls()))
234	#endif
235	{
236	os << " <port>" << data.getPort() << "</port>\n";
237	}
238
239	#ifdef _OpenSSL_
240	os.setf(ios::boolalpha);
241	os << " <tls>" << data.getTls() << "</tls>\n";
242	#endif
243
244	string path = data.getPath();
245
246	entities(path, '&', "&");
247	entities(path, '<', "<");
248	entities(path, '>', ">");
249
250	os << " <path>" << path << "</path>\n";
251
252	os << " <size>" << data.getSize() << "</size>\n";
253
254	if(data.getTitle() != "")
255	{
256	string title = data.getTitle();
257
258	entities(title, '&', "&");
259	entities(title, '<', "<");
260	entities(title, '>', ">");
261
262	os << " <title>" << title << "</title>\n";
263	}
264
265	if(data.getDescription() != "")
266	{
267	string description = data.getDescription();
268
269	entities(description, '&', "&");
270	entities(description, '<', "<");
271	entities(description, '>', ">");
272
273	os << " <description>" << description << "</description>\n";
274	}
275
276	string text = data.getText();
277
278	entities(text, '&', "&");
279	entities(text, '<', "<");
280	entities(text, '>', ">");
281
282	os << " <text>" << text << "</text>\n";
283
284	for (size_t index(0); index < data.getHeadings().size(); index++)
285	{
286	string heading = data.getHeadings()[index];
287
288	entities(heading, '&', "&");
289	entities(heading, '<', "<");
290	entities(heading, '>', ">");
291
292	os << " <heading>" << heading << "</heading>\n";
293	}
294
295	os << " </page>";
296
297	return os;
298	}