repos/HostUpdate/Host.cpp

// Host Update
//
// Douglas Thrift
//
// $Id$

#include "Host.hpp"

#ifndef _WIN32

static char slash = '/';

#else

static char slash = '\\';

#endif

Host::Host(const string& host, const string& name, const string& address, const
        string& platform)
{
#ifdef _WIN32
        if (count == 0)
        {
                if (WSAStartup(MAKEWORD(2, 0), &data) != 0)
                {
                        cerr << "Host(): WSAStartup()\n";
                        exit(1);
                }
        }

        count++;
#endif

        setHost(host);
        if (name != "") setName(name, address == "");
        if (address != "") setAddress(address, name == "");
        setPlatform(platform);
}

Host::~Host()
{
#ifdef _WIN32
        count--;

        if (count == 0)
        {
                if (WSACleanup() != 0)
                {
                        cerr << "~Host(): WSACleanup()\n";
                        exit(1);
                }
        }
#endif
}

void Host::setName(const string& name, bool lookup)
{
        this->name = name;

        if (lookup)
        {
                struct hostent* ent = gethostbyname(this->name.c_str());

                memcpy(&address, *(ent->h_addr_list), ent->h_length);
        }
}

void Host::setAddress(const string& address, bool lookup)
{
        in_addr_t value = inet_addr(address.c_str());

        if (value == INADDR_NONE)
        {
                cerr << "Host.setAddress(): INADDR_NONE\n";
                exit(1);
        }
        else
        {
                memcpy(&this->address, &value, sizeof(in_addr_t));
        }

        if (lookup)
        {
                struct hostent* ent = gethostbyaddr((char*)(&this->address),
                        sizeof(in_addr), AF_INET);

                if (ent != NULL) name = ent->h_name;
        }
}

bool Host::operator==(const Host& host) const
{
        if (this->host == host.host)
        {
                if (name == host.name)
                {
                        if (string(inet_ntoa(address)) == inet_ntoa(host.address))
                        {
                                return platform == host.platform;
                        }
                        else
                        {
                                return false;
                        }
                }
                else
                {
                        return false;
                }
        }
        else
        {
                return false;
        }
}

Host Host::operator++()
{
        string file = string("hosts") + slash + host;

        ifstream fin(file.c_str());

        if (fin.is_open())
        {
                getline(fin, name);

                string address;
                
                getline(fin, address);

                in_addr_t value = inet_addr(address.c_str());

                if (value == INADDR_NONE)
                {
                        cerr << "Host.operator++(): INADDR_NONE\n";
                        exit(1);
                }
                else
                {
                        memcpy(&this->address, &value, sizeof(in_addr_t));
                }

                getline(fin, platform);
                fin.close();
        }

        return *this;
}

Host Host::operator++(int)
{
        Host old = *this;

        ++*this;

        return old;
}

Host Host::operator--()
{
        string file = string("hosts") + slash + host;
        
        ofstream fout(file.c_str());

        fout << name << '\n'
                << inet_ntoa(address) << '\n'
                << platform << '\n';

        fout.close();

        return *this;
}

Host Host::operator--(int)
{
        Host old = *this;

        --*this;

        return old;
}

#ifdef _WIN32

unsigned Host::count = 0;
WSADATA Host::data;

#endif
Revision:	15
Committed:	2003-11-09T16:51:42-08:00 (21 years, 7 months ago) by Douglas Thrift
File size:	2584 byte(s)
Log Message:	Ah, much better; I'll wait to do binary another day.
#	User	Rev	Content
1	Douglas Thrift	11	// Host Update
2			//
3			// Douglas Thrift
4			//
5			// $Id$
6
7			#include "Host.hpp"
8
9	Douglas Thrift	15	#ifndef _WIN32
10
11			static char slash = '/';
12
13			#else
14
15			static char slash = '\\';
16
17			#endif
18
19	Douglas Thrift	12	Host::Host(const string& host, const string& name, const string& address, const
20			string& platform)
21	Douglas Thrift	11	{
22			#ifdef _WIN32
23			if (count == 0)
24			{
25			if (WSAStartup(MAKEWORD(2, 0), &data) != 0)
26			{
27			cerr << "Host(): WSAStartup()\n";
28			exit(1);
29			}
30			}
31
32			count++;
33			#endif
34
35			setHost(host);
36			if (name != "") setName(name, address == "");
37			if (address != "") setAddress(address, name == "");
38	Douglas Thrift	12	setPlatform(platform);
39	Douglas Thrift	11	}
40
41			Host::~Host()
42			{
43			#ifdef _WIN32
44			count--;
45
46			if (count == 0)
47			{
48			if (WSACleanup() != 0)
49			{
50			cerr << "~Host(): WSACleanup()\n";
51			exit(1);
52			}
53			}
54			#endif
55			}
56
57			void Host::setName(const string& name, bool lookup)
58			{
59			this->name = name;
60
61			if (lookup)
62			{
63			struct hostent* ent = gethostbyname(this->name.c_str());
64
65	Douglas Thrift	15	memcpy(&address, *(ent->h_addr_list), ent->h_length);
66	Douglas Thrift	11	}
67			}
68
69			void Host::setAddress(const string& address, bool lookup)
70			{
71			in_addr_t value = inet_addr(address.c_str());
72
73			if (value == INADDR_NONE)
74			{
75			cerr << "Host.setAddress(): INADDR_NONE\n";
76			exit(1);
77			}
78			else
79			{
80	Douglas Thrift	15	memcpy(&this->address, &value, sizeof(in_addr_t));
81	Douglas Thrift	11	}
82
83			if (lookup)
84			{
85	Douglas Thrift	15	struct hostent* ent = gethostbyaddr((char*)(&this->address),
86	Douglas Thrift	11	sizeof(in_addr), AF_INET);
87
88	Douglas Thrift	12	if (ent != NULL) name = ent->h_name;
89	Douglas Thrift	11	}
90			}
91
92	Douglas Thrift	13	bool Host::operator==(const Host& host) const
93	Douglas Thrift	11	{
94	Douglas Thrift	13	if (this->host == host.host)
95			{
96			if (name == host.name)
97			{
98	Douglas Thrift	15	if (string(inet_ntoa(address)) == inet_ntoa(host.address))
99	Douglas Thrift	13	{
100			return platform == host.platform;
101			}
102			else
103			{
104			return false;
105			}
106			}
107			else
108			{
109			return false;
110			}
111			}
112			else
113			{
114			return false;
115			}
116			}
117
118			Host Host::operator++()
119			{
120	Douglas Thrift	15	string file = string("hosts") + slash + host;
121	Douglas Thrift	11
122	Douglas Thrift	15	ifstream fin(file.c_str());
123
124	Douglas Thrift	13	if (fin.is_open())
125			{
126			getline(fin, name);
127	Douglas Thrift	15
128			string address;
129
130			getline(fin, address);
131
132			in_addr_t value = inet_addr(address.c_str());
133
134			if (value == INADDR_NONE)
135			{
136			cerr << "Host.operator++(): INADDR_NONE\n";
137			exit(1);
138			}
139			else
140			{
141			memcpy(&this->address, &value, sizeof(in_addr_t));
142			}
143
144	Douglas Thrift	13	getline(fin, platform);
145			fin.close();
146			}
147
148			return *this;
149	Douglas Thrift	11	}
150
151	Douglas Thrift	13	Host Host::operator++(int)
152	Douglas Thrift	11	{
153	Douglas Thrift	13	Host old = *this;
154
155			++*this;
156
157			return old;
158			}
159
160			Host Host::operator--()
161			{
162	Douglas Thrift	15	string file = string("hosts") + slash + host;
163
164			ofstream fout(file.c_str());
165	Douglas Thrift	11
166	Douglas Thrift	15	fout << name << '\n'
167			<< inet_ntoa(address) << '\n'
168			<< platform << '\n';
169	Douglas Thrift	11
170			fout.close();
171	Douglas Thrift	13
172			return *this;
173	Douglas Thrift	11	}
174
175	Douglas Thrift	13	Host Host::operator--(int)
176			{
177			Host old = *this;
178
179			--*this;
180
181			return old;
182			}
183
184	Douglas Thrift	11	#ifdef _WIN32
185
186			unsigned Host::count = 0;
187			WSADATA Host::data;
188
189			#endif