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.5 by pcg, Sat Apr 5 17:54:22 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 (prot && 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[10 + 15 + 1 + 5 + 1]; // protype / 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, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff); 141 sprintf (hostport, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff);
95u8 146u8
96sockinfo::supported_protocols (conf_node *conf) 147sockinfo::supported_protocols (conf_node *conf)
97{ 148{
98 u8 protocols = prot; 149 u8 protocols = prot;
99 150
100 if (prot & (PROT_UDPv4 | PROT_TCPv4)) 151 if (prot & (PROT_IPv4 | PROT_ICMPv4 | PROT_UDPv4 | PROT_TCPv4))
101 protocols |= PROT_IPv4; 152 protocols |= PROT_IPv4 | PROT_ICMPv4;
102 153
103 if (conf 154 if (conf && prot & PROTv4)
104 && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4) 155 {
105 && conf->protocols & PROT_UDPv4 156 if (conf->protocols & PROT_UDPv4 && conf->udp_port)
106 && conf->udp_port)
107 protocols |= PROT_UDPv4; 157 protocols |= PROT_UDPv4;
108 158
109 if (conf 159 if (conf->protocols & PROT_TCPv4 && conf->tcp_port)
110 && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4)
111 && conf->protocols & PROT_TCPv4
112 && conf->tcp_port)
113 protocols |= PROT_TCPv4; 160 protocols |= PROT_TCPv4;
161
162 if (conf->protocols & PROT_DNSv4 && conf->dns_port)
163 protocols |= PROT_DNSv4;
164 }
114 165
115 return protocols; 166 return protocols;
116} 167}
117 168
118bool 169bool
119sockinfo::upgrade_protocol (u8 prot_, conf_node *conf) 170sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
120{ 171{
121 if (prot_ == prot) 172 if (prot_ == prot)
122 return true; 173 return true;
123 174
124 if (prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4)
125 && prot_ & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4))
126 {
127 if (prot_ & PROT_IPv4) 175 if (prot & PROTv4
176 && prot_ & PROTv4)
177 {
178 if (prot_ & (PROT_IPv4 | PROT_ICMPv4))
128 { 179 {
129 prot = prot_; 180 prot = prot_;
130 port = 0; 181 port = 0;
131 return true; 182 return true;
132 } 183 }
135 && prot_ & PROT_UDPv4 186 && prot_ & PROT_UDPv4
136 && conf->protocols & PROT_UDPv4 187 && conf->protocols & PROT_UDPv4
137 && conf->udp_port) 188 && conf->udp_port)
138 { 189 {
139 prot = prot_; 190 prot = prot_;
140 port = conf->udp_port; 191 port = htons (conf->udp_port);
141 return true; 192 return true;
142 } 193 }
143 194
144 if (conf 195 if (conf
145 && prot_ & PROT_TCPv4 196 && prot_ & PROT_TCPv4
146 && conf->protocols & PROT_TCPv4 197 && conf->protocols & PROT_TCPv4
147 && conf->tcp_port) 198 && conf->tcp_port)
148 { 199 {
149 prot = prot_; 200 prot = prot_;
150 port = conf->tcp_port; 201 port = htons (conf->tcp_port);
151 return true; 202 return true;
152 } 203 }
153 } 204 }
154 205
155 return false; 206 return false;

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines