repos/HostUpdate/Host.cpp

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

#include "Host.hpp"

Host::Host(const string& host, const string& name, const string& address, const
        string& platform)
{
        memset(&this->address, 0, sizeof(in_addr));

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