ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/gvpe/src/sockinfo.C
(Generate patch)

Comparing gvpe/src/sockinfo.C (file contents):
Revision 1.3 by pcg, Wed Apr 2 05:15:00 2003 UTC vs.
Revision 1.29 by root, Tue Dec 4 13:49:16 2012 UTC

1/* 1/*
2 sockinfo.c -- socket address management 2 sockinfo.C -- socket address management
3 Copyright (C) 2003-2008 Marc Lehmann <gvpe@schmorp.de>
3 4
5 This file is part of GVPE.
6
4 This program is free software; you can redistribute it and/or modify 7 GVPE is free software; you can redistribute it and/or modify it
5 it under the terms of the GNU General Public License as published by 8 under the terms of the GNU General Public License as published by the
6 the Free Software Foundation; either version 2 of the License, or 9 Free Software Foundation; either version 3 of the License, or (at your
7 (at your option) any later version. 10 option) any later version.
8 11
9 This program is distributed in the hope that it will be useful, 12 This program is distributed in the hope that it will be useful, but
10 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
12 GNU General Public License for more details. 15 Public License for more details.
13 16
14 You should have received a copy of the GNU General Public License 17 You should have received a copy of the GNU General Public License along
15 along with this program; if not, write to the Free Software 18 with this program; if not, see <http://www.gnu.org/licenses/>.
16 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 19
20 Additional permission under GNU GPL version 3 section 7
21
22 If you modify this Program, or any covered work, by linking or
23 combining it with the OpenSSL project's OpenSSL library (or a modified
24 version of that library), containing parts covered by the terms of the
25 OpenSSL or SSLeay licenses, the licensors of this Program grant you
26 additional permission to convey the resulting work. Corresponding
27 Source for a non-source form of such a combination shall include the
28 source code for the parts of OpenSSL used as well as that of the
29 covered work.
17*/ 30*/
18 31
19#include "config.h" 32#include "config.h"
20 33
34// for gethostbyname
35#include <netdb.h>
36
37// for inet_aton
38#include <sys/socket.h>
39#include <netinet/in.h>
21#include <arpa/inet.h> 40#include <arpa/inet.h>
22#include <netdb.h>
23 41
24#include "gettext.h" 42#include "gettext.h"
25 43
26#include "sockinfo.h" 44#include "sockinfo.h"
27#include "slog.h" 45#include "slog.h"
28 46
47#include <cstring>
48#include <cstdio>
49
50// all ipv4-based protocols
51#define PROTv4 (PROT_UDPv4 | PROT_TCPv4 | PROT_ICMPv4 | PROT_IPv4 | PROT_DNSv4)
52
53void
29void sockinfo::set (const sockaddr_in *sa, u8 prot_) 54sockinfo::set (const sockaddr_in *sa, u8 prot_)
30{ 55{
31 host = sa->sin_addr.s_addr; 56 host = sa->sin_addr.s_addr;
32 port = prot_ == PROT_IPv4 ? 0 : sa->sin_port; 57 port = prot_ & (PROT_IPv4 | PROT_ICMPv4) ? 0 : sa->sin_port;
33 prot = prot_; 58 prot = prot_;
59}
60
61void
62sockinfo::set (const char *hostname, u16 port_, u8 prot_)
63{
64 prot = prot_;
65 host = 0;
66 port = htons (port_);
67
68 if (prot & PROTv4
69 && hostname)
70 {
71 struct in_addr in;
72 struct hostent *he;
73
74 if (inet_aton (hostname, &in))
75 {
76 host = in.s_addr;
77 }
78 else if ((he = gethostbyname (hostname))
79 && he->h_addrtype == AF_INET
80 && he->h_length == 4
81 && he->h_addr_list[0])
82 {
83 //sa->sin_family = he->h_addrtype;
84 memcpy (&host, he->h_addr_list[0], 4);
85 }
86 else
87 slog (L_NOTICE, _("unable to resolve host '%s'"), hostname);
88 }
34} 89}
35 90
36void 91void
37sockinfo::set (const conf_node *conf, u8 prot_) 92sockinfo::set (const conf_node *conf, u8 prot_)
38{ 93{
39 prot = prot_; 94 if (prot_ == PROT_DNSv4)
40 host = 0; 95 {
96 host = htonl (conf->id); port = 0; prot = prot_;
97 }
98 else
99 set (conf->hostname,
41 port = prot_ == PROT_UDPv4 ? htons (conf->udp_port) 100 prot_ == PROT_UDPv4 ? conf->udp_port
42 : prot_ == PROT_TCPv4 ? htons (conf->tcp_port) 101 : prot_ == PROT_TCPv4 ? conf->tcp_port
102 : prot_ == PROT_DNSv4 ? conf->dns_port
43 : 0; 103 : 0,
44 104 prot_);
45 if (conf->hostname)
46 {
47 struct hostent *he = gethostbyname (conf->hostname);
48
49 if (he
50 && he->h_addrtype == AF_INET && he->h_length == 4 && he->h_addr_list[0])
51 {
52 //sa->sin_family = he->h_addrtype;
53 memcpy (&host, he->h_addr_list[0], 4);
54
55 }
56 else
57 slog (L_NOTICE, _("unable to resolve host '%s'"), conf->hostname);
58 }
59} 105}
60
61 106
62const sockaddr * 107const sockaddr *
63sockinfo::sav4() const 108sockinfo::sav4() const
64{ 109{
65 static sockaddr_in sa; 110 static sockaddr_in sa;
69 sa.sin_addr.s_addr = host; 114 sa.sin_addr.s_addr = host;
70 115
71 return (const sockaddr *)&sa; 116 return (const sockaddr *)&sa;
72} 117}
73 118
74static char hostport[15 + 1 + 5 + 1]; // IPv4 : port 119static char hostport[10 + 15 + 1 + 5 + 1]; // proto / IPv4 : port
75 120
76const char * 121const char *
77sockinfo::ntoa () const 122sockinfo::ntoa () const
78{ 123{
79 in_addr ia = { host }; 124 in_addr ia = { host };
81 sprintf (hostport, "%.15s", inet_ntoa (ia)); 126 sprintf (hostport, "%.15s", inet_ntoa (ia));
82 127
83 return hostport; 128 return hostport;
84} 129}
85 130
131bool
132sockinfo::valid () const
133{
134 return (prot & THISNODE->protocols) && host;
135}
136
86sockinfo::operator const char *() const 137sockinfo::operator const char *() const
87{ 138{
88 in_addr ia = { host }; 139 in_addr ia = { host };
89 140
90 sprintf (hostport, "%.15s:%d", inet_ntoa (ia), ntohs (port) & 0xffff); 141 sprintf (hostport, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff);
91 142
92 return hostport; 143 return hostport;
93} 144}
94 145
146u8
147sockinfo::supported_protocols (conf_node *conf)
148{
149 u8 protocols = prot;
150
151 if (prot & (PROT_IPv4 | PROT_ICMPv4 | PROT_UDPv4 | PROT_TCPv4))
152 protocols |= PROT_IPv4 | PROT_ICMPv4;
153
154 if (conf && prot & PROTv4)
155 {
156 if (conf->protocols & PROT_UDPv4 && conf->udp_port)
157 protocols |= PROT_UDPv4;
158
159 if (conf->protocols & PROT_TCPv4 && conf->tcp_port)
160 protocols |= PROT_TCPv4;
161
162 if (conf->protocols & PROT_DNSv4 && conf->dns_port)
163 protocols |= PROT_DNSv4;
164 }
165
166 return protocols;
167}
168
169bool
170sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
171{
172 if (prot_ == prot)
173 return true;
174
175 if (prot & PROTv4
176 && prot_ & PROTv4)
177 {
178 if (prot_ & (PROT_IPv4 | PROT_ICMPv4))
179 {
180 prot = prot_;
181 port = 0;
182 return true;
183 }
184
185 if (conf
186 && prot_ & PROT_UDPv4
187 && conf->protocols & PROT_UDPv4
188 && conf->udp_port)
189 {
190 prot = prot_;
191 port = htons (conf->udp_port);
192 return true;
193 }
194
195 if (conf
196 && prot_ & PROT_TCPv4
197 && conf->protocols & PROT_TCPv4
198 && conf->tcp_port)
199 {
200 prot = prot_;
201 port = htons (conf->tcp_port);
202 return true;
203 }
204 }
205
206 return false;
207}
208
209bool
210operator == (const sockinfo &a, const sockinfo &b)
211{
212 return a.host == b.host && a.port == b.port && a.prot == b.prot;
213}
214
215bool
216operator < (const sockinfo &a, const sockinfo &b)
217{
218 return a.host < b.host
219 || (a.host == b.host && (a.port < b.port
220 || (a.port == b.port && a.prot < b.prot)));
221}
222

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines