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.4 by pcg, Sat Apr 5 02:32:40 2003 UTC vs.
Revision 1.14 by pcg, Thu Mar 3 07:24:57 2005 UTC

1/* 1/*
2 sockinfo.C -- socket address management 2 sockinfo.C -- socket address management
3 Copyright (C) 2003 Marc Lehmann <pcg@goof.com>
3 4
4 This program is free software; you can redistribute it and/or modify 5 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 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 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version. 8 (at your option) any later version.
… …
16 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 17 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17*/ 18*/
18 19
19#include "config.h" 20#include "config.h"
20 21
21#include <arpa/inet.h>
22#include <netdb.h> 22#include <netdb.h>
23 23
24#include "gettext.h" 24#include "gettext.h"
25 25
26#include "sockinfo.h" 26#include "sockinfo.h"
27#include "slog.h" 27#include "slog.h"
28 28
29// all ipv4-based protocols
30#define PROTv4 (PROT_UDPv4 | PROT_TCPv4 | PROT_ICMPv4 | PROT_IPv4 | PROT_DNSv4)
31
29void sockinfo::set (const sockaddr_in *sa, u8 prot_) 32void sockinfo::set (const sockaddr_in *sa, u8 prot_)
30{ 33{
31 host = sa->sin_addr.s_addr; 34 host = sa->sin_addr.s_addr;
32 port = prot_ == PROT_IPv4 ? 0 : sa->sin_port; 35 port = prot_ & (PROT_IPv4 | PROT_ICMPv4) ? 0 : sa->sin_port;
33 prot = prot_; 36 prot = prot_;
37}
38
39void sockinfo::set (const char *hostname, u16 port_, u8 prot_)
40{
41 prot = prot_;
42 host = 0;
43 port = htons (port_);
44
45 if (prot & PROTv4
46 && hostname)
47 {
48 struct hostent *he = gethostbyname (hostname);
49
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 slog (L_NOTICE, _("unable to resolve host '%s'"), hostname);
58 }
34} 59}
35 60
36void 61void
37sockinfo::set (const conf_node *conf, u8 prot_) 62sockinfo::set (const conf_node *conf, u8 prot_)
38{ 63{
39 prot = prot_; 64 if (prot_ == PROT_DNSv4)
40 host = 0; 65 set (conf->dns_hostname, conf->dns_hostname ? conf->dns_port : 0, prot_);
66 else
67 set (conf->hostname,
41 port = prot == PROT_UDPv4 ? htons (conf->udp_port) 68 prot_ == PROT_UDPv4 ? conf->udp_port
42 : prot == PROT_TCPv4 ? htons (conf->tcp_port) 69 : prot_ == PROT_TCPv4 ? conf->tcp_port
70 : prot_ == PROT_DNSv4 ? conf->dns_port
43 : 0; 71 : 0,
44 72 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} 73}
60
61 74
62const sockaddr * 75const sockaddr *
63sockinfo::sav4() const 76sockinfo::sav4() const
64{ 77{
65 static sockaddr_in sa; 78 static sockaddr_in sa;
… …
69 sa.sin_addr.s_addr = host; 82 sa.sin_addr.s_addr = host;
70 83
71 return (const sockaddr *)&sa; 84 return (const sockaddr *)&sa;
72} 85}
73 86
74static char hostport[10 + 15 + 1 + 5 + 1]; // protype / IPv4 : port 87static char hostport[10 + 15 + 1 + 5 + 1]; // proto / IPv4 : port
75 88
76const char * 89const char *
77sockinfo::ntoa () const 90sockinfo::ntoa () const
78{ 91{
79 in_addr ia = { host }; 92 in_addr ia = { host };
… …
88 in_addr ia = { host }; 101 in_addr ia = { host };
89 102
90 sprintf (hostport, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff); 103 sprintf (hostport, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff);
91 104
92 return hostport; 105 return hostport;
106}
107
108u8
109sockinfo::supported_protocols (conf_node *conf)
110{
111 u8 protocols = prot;
112
113 if (prot & (PROT_UDPv4 | PROT_TCPv4))
114 protocols |= PROT_IPv4 | PROT_ICMPv4;
115
116 if (conf
117 && prot & PROTv4
118 && conf->protocols & PROT_UDPv4
119 && conf->udp_port)
120 protocols |= PROT_UDPv4;
121
122 if (conf
123 && prot & PROTv4
124 && conf->protocols & PROT_TCPv4
125 && conf->tcp_port)
126 protocols |= PROT_TCPv4;
127
128 if (conf
129 && prot & PROTv4
130 && conf->protocols & PROT_DNSv4
131 && conf->dns_port)
132 protocols |= PROT_DNSv4;
133
134 return protocols;
135}
136
137bool
138sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
139{
140 if (prot_ == prot)
141 return true;
142
143 if (prot & PROTv4
144 && prot_ & PROTv4)
145 {
146 if (prot_ & (PROT_IPv4 | PROT_ICMPv4))
147 {
148 prot = prot_;
149 port = 0;
150 return true;
151 }
152
153 if (conf
154 && prot_ & PROT_UDPv4
155 && conf->protocols & PROT_UDPv4
156 && conf->udp_port)
157 {
158 prot = prot_;
159 port = htons (conf->udp_port);
160 return true;
161 }
162
163 if (conf
164 && prot_ & PROT_TCPv4
165 && conf->protocols & PROT_TCPv4
166 && conf->tcp_port)
167 {
168 prot = prot_;
169 port = htons (conf->tcp_port);
170 return true;
171 }
172
173#if ENABLE_DNS
174 if (conf
175 && prot_ & PROT_DNSv4
176 && conf->protocols & PROT_DNSv4
177 && conf->dns_port)
178 {
179 set (::conf.dns_forw_host, ::conf.dns_forw_port, prot_);
180 return true;
181 }
182#endif
183 }
184
185 return false;
93} 186}
94 187
95bool 188bool
96operator == (const sockinfo &a, const sockinfo &b) 189operator == (const sockinfo &a, const sockinfo &b)
97{ 190{

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines