gvpe/src/sockinfo.C

/*
    sockinfo.C -- socket address management
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.
 
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
 
    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#include "config.h"

#include <arpa/inet.h>
#include <netdb.h>

#include "gettext.h"

#include "sockinfo.h"
#include "slog.h"

// all ipv4-based protocols
#define PROTv4 (PROT_UDPv4 | PROT_TCPv4 | PROT_ICMPv4 | PROT_IPv4)

void sockinfo::set (const sockaddr_in *sa, u8 prot_)
{
  host = sa->sin_addr.s_addr;
  port = prot_ & (PROT_IPv4 | PROT_ICMPv4) ? 0 : sa->sin_port;
  prot = prot_;
}

void sockinfo::set (const char *hostname, u16 port_, u8 prot_)
{
  prot = prot_;
  host = 0;
  port = htons (port_);

  if (prot & PROTv4
      && hostname)
    {
      struct hostent *he = gethostbyname (hostname);

      if (he
          && he->h_addrtype == AF_INET && he->h_length == 4 && he->h_addr_list[0])
        {
          //sa->sin_family = he->h_addrtype;
          memcpy (&host, he->h_addr_list[0], 4);
        }
      else
        slog (L_NOTICE, _("unable to resolve host '%s'"), hostname);
    }
}

void
sockinfo::set (const conf_node *conf, u8 prot_)
{
 set (conf->hostname,
        prot_ == PROT_UDPv4 ? conf->udp_port
      : prot_ == PROT_TCPv4 ? conf->tcp_port
      : 0,
      prot_);
}

const sockaddr *
sockinfo::sav4() const
{
  static sockaddr_in sa;

  sa.sin_family      = AF_INET;
  sa.sin_port        = port;
  sa.sin_addr.s_addr = host;

  return (const sockaddr *)&sa;
}

static char hostport[10 + 15 + 1 + 5 + 1]; // proto / IPv4 : port

const char *
sockinfo::ntoa () const
{
  in_addr ia = { host };

  sprintf (hostport, "%.15s", inet_ntoa (ia));

  return hostport;
}

sockinfo::operator const char *() const
{
  in_addr ia = { host };

  sprintf (hostport, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff);

  return hostport;
}

u8
sockinfo::supported_protocols (conf_node *conf)
{
  u8 protocols = prot;

  if (prot & (PROT_UDPv4 | PROT_TCPv4))
    protocols |= PROT_IPv4 | PROT_ICMPv4;

  if (conf
      && prot & PROTv4
      && conf->protocols & PROT_UDPv4
      && conf->udp_port)
    protocols |= PROT_UDPv4;

  if (conf
      && prot & PROTv4
      && conf->protocols & PROT_TCPv4
      && conf->tcp_port)
    protocols |= PROT_TCPv4;

  return protocols;
}

bool
sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
{
  if (prot_ == prot)
    return true;

  if (prot & PROTv4
      && prot_ & PROTv4)
    {
      if (prot_ & (PROT_IPv4 | PROT_ICMPv4))
        {
          prot = prot_;
          port = 0;
          return true;
        }

      if (conf
          && prot_ & PROT_UDPv4
          && conf->protocols & PROT_UDPv4
          && conf->udp_port)
        {
          prot = prot_;
          port = htons (conf->udp_port);
          return true;
        }

      if (conf
          && prot_ & PROT_TCPv4
          && conf->protocols & PROT_TCPv4
          && conf->tcp_port)
        {
          prot = prot_;
          port = htons (conf->tcp_port);
          return true;
        }
    }

  return false;
}

bool
operator == (const sockinfo &a, const sockinfo &b)
{
  return a.host == b.host && a.port == b.port && a.prot == b.prot;
}

bool
operator < (const sockinfo &a, const sockinfo &b)
{
  return a.host < b.host
         || (a.host == b.host && (a.port < b.port
                                  || (a.port == b.port && a.prot < b.prot)));
}

Revision:	1.9
Committed:	Tue Apr 8 02:00:54 2003 UTC (21 years, 1 month ago) by pcg
Content type:	text/plain
Branch:	MAIN
CVS Tags:	VPE_0_9, VPE_1_0
Changes since 1.8:	+11 -8 lines
Log Message:	* empty log message *
#	User	Rev	Content
1	pcg	1.1	/*
2	pcg	1.4	sockinfo.C -- socket address management
3	pcg	1.1
4			This program is free software; you can redistribute it and/or modify
5			it under the terms of the GNU General Public License as published by
6			the Free Software Foundation; either version 2 of the License, or
7			(at your option) any later version.
8
9			This program is distributed in the hope that it will be useful,
10			but WITHOUT ANY WARRANTY; without even the implied warranty of
11			MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12			GNU General Public License for more details.
13
14			You should have received a copy of the GNU General Public License
15			along with this program; if not, write to the Free Software
16			Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17			*/
18
19			#include "config.h"
20
21	pcg	1.2	#include <arpa/inet.h>
22			#include <netdb.h>
23	pcg	1.1
24			#include "gettext.h"
25
26			#include "sockinfo.h"
27			#include "slog.h"
28
29	pcg	1.9	// all ipv4-based protocols
30			#define PROTv4 (PROT_UDPv4 \| PROT_TCPv4 \| PROT_ICMPv4 \| PROT_IPv4)
31
32	pcg	1.1	void sockinfo::set (const sockaddr_in *sa, u8 prot_)
33			{
34			host = sa->sin_addr.s_addr;
35	pcg	1.9	port = prot_ & (PROT_IPv4 \| PROT_ICMPv4) ? 0 : sa->sin_port;
36	pcg	1.1	prot = prot_;
37			}
38
39	pcg	1.8	void sockinfo::set (const char *hostname, u16 port_, u8 prot_)
40	pcg	1.1	{
41	pcg	1.3	prot = prot_;
42	pcg	1.1	host = 0;
43	pcg	1.8	port = htons (port_);
44	pcg	1.1
45	pcg	1.9	if (prot & PROTv4
46	pcg	1.8	&& hostname)
47	pcg	1.1	{
48	pcg	1.8	struct hostent *he = gethostbyname (hostname);
49	pcg	1.1
50			if (he
51			&& he->h_addrtype == AF_INET && he->h_length == 4 && he->h_addr_list[0])
52			{
53			//sa->sin_family = he->h_addrtype;
54			memcpy (&host, he->h_addr_list[0], 4);
55			}
56			else
57	pcg	1.8	slog (L_NOTICE, _("unable to resolve host '%s'"), hostname);
58	pcg	1.1	}
59			}
60
61	pcg	1.8	void
62			sockinfo::set (const conf_node *conf, u8 prot_)
63			{
64			set (conf->hostname,
65			prot_ == PROT_UDPv4 ? conf->udp_port
66			: prot_ == PROT_TCPv4 ? conf->tcp_port
67			: 0,
68			prot_);
69			}
70	pcg	1.1
71			const sockaddr *
72			sockinfo::sav4() const
73			{
74			static sockaddr_in sa;
75
76			sa.sin_family = AF_INET;
77			sa.sin_port = port;
78			sa.sin_addr.s_addr = host;
79
80			return (const sockaddr *)&sa;
81			}
82
83	pcg	1.8	static char hostport[10 + 15 + 1 + 5 + 1]; // proto / IPv4 : port
84	pcg	1.1
85			const char *
86			sockinfo::ntoa () const
87			{
88			in_addr ia = { host };
89
90			sprintf (hostport, "%.15s", inet_ntoa (ia));
91
92			return hostport;
93			}
94
95			sockinfo::operator const char *() const
96			{
97			in_addr ia = { host };
98
99	pcg	1.4	sprintf (hostport, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff);
100	pcg	1.1
101			return hostport;
102	pcg	1.4	}
103
104	pcg	1.5	u8
105			sockinfo::supported_protocols (conf_node *conf)
106			{
107			u8 protocols = prot;
108
109			if (prot & (PROT_UDPv4 \| PROT_TCPv4))
110	pcg	1.9	protocols \|= PROT_IPv4 \| PROT_ICMPv4;
111	pcg	1.5
112			if (conf
113	pcg	1.9	&& prot & PROTv4
114	pcg	1.5	&& conf->protocols & PROT_UDPv4
115			&& conf->udp_port)
116			protocols \|= PROT_UDPv4;
117
118			if (conf
119	pcg	1.9	&& prot & PROTv4
120	pcg	1.5	&& conf->protocols & PROT_TCPv4
121			&& conf->tcp_port)
122			protocols \|= PROT_TCPv4;
123
124			return protocols;
125			}
126
127			bool
128			sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
129			{
130			if (prot_ == prot)
131			return true;
132
133	pcg	1.9	if (prot & PROTv4
134			&& prot_ & PROTv4)
135	pcg	1.5	{
136	pcg	1.9	if (prot_ & (PROT_IPv4 \| PROT_ICMPv4))
137	pcg	1.5	{
138			prot = prot_;
139			port = 0;
140			return true;
141			}
142
143			if (conf
144			&& prot_ & PROT_UDPv4
145			&& conf->protocols & PROT_UDPv4
146			&& conf->udp_port)
147			{
148			prot = prot_;
149	pcg	1.6	port = htons (conf->udp_port);
150	pcg	1.5	return true;
151			}
152
153			if (conf
154			&& prot_ & PROT_TCPv4
155			&& conf->protocols & PROT_TCPv4
156			&& conf->tcp_port)
157			{
158			prot = prot_;
159	pcg	1.6	port = htons (conf->tcp_port);
160	pcg	1.5	return true;
161			}
162			}
163
164			return false;
165			}
166
167	pcg	1.4	bool
168			operator == (const sockinfo &a, const sockinfo &b)
169			{
170			return a.host == b.host && a.port == b.port && a.prot == b.prot;
171			}
172
173			bool
174			operator < (const sockinfo &a, const sockinfo &b)
175			{
176			return a.host < b.host
177			\|\| (a.host == b.host && (a.port < b.port
178			\|\| (a.port == b.port && a.prot < b.prot)));
179	pcg	1.1	}
180