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.8 by pcg, Mon Apr 7 01:12:56 2003 UTC vs.
Revision 1.16 by pcg, Sat Mar 5 19:13:15 2005 UTC

1/* 1/*
2 sockinfo.C -- socket address management 2 sockinfo.C -- socket address management
3 Copyright (C) 2003 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
5 it under the terms of the GNU General Public License as published by 8 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 9 the Free Software Foundation; either version 2 of the License, or
7 (at your option) any later version. 10 (at your 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,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of 13 but 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
12 GNU General Public License for more details. 15 GNU General 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
15 along with this program; if not, write to the Free Software 18 along with gvpe; if not, write to the Free Software
16 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 19 Foundation, Inc. 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17*/ 20*/
18 21
19#include "config.h" 22#include "config.h"
20 23
21#include <arpa/inet.h>
22#include <netdb.h> 24#include <netdb.h>
23 25
24#include "gettext.h" 26#include "gettext.h"
25 27
26#include "sockinfo.h" 28#include "sockinfo.h"
27#include "slog.h" 29#include "slog.h"
28 30
31// all ipv4-based protocols
32#define PROTv4 (PROT_UDPv4 | PROT_TCPv4 | PROT_ICMPv4 | PROT_IPv4 | PROT_DNSv4)
33
29void sockinfo::set (const sockaddr_in *sa, u8 prot_) 34void sockinfo::set (const sockaddr_in *sa, u8 prot_)
30{ 35{
31 host = sa->sin_addr.s_addr; 36 host = sa->sin_addr.s_addr;
32 port = prot_ == PROT_IPv4 ? 0 : sa->sin_port; 37 port = prot_ & (PROT_IPv4 | PROT_ICMPv4) ? 0 : sa->sin_port;
33 prot = prot_; 38 prot = prot_;
34} 39}
35 40
36void sockinfo::set (const char *hostname, u16 port_, u8 prot_) 41void sockinfo::set (const char *hostname, u16 port_, u8 prot_)
37{ 42{
38 prot = prot_; 43 prot = prot_;
39 host = 0; 44 host = 0;
40 port = htons (port_); 45 port = htons (port_);
41 46
42 if (prot & (PROT_UDPv4 | PROT_TCPv4 | PROT_IPv4) 47 if (prot & PROTv4
43 && hostname) 48 && hostname)
44 { 49 {
45 struct hostent *he = gethostbyname (hostname); 50 struct hostent *he = gethostbyname (hostname);
46 51
47 if (he 52 if (he
56} 61}
57 62
58void 63void
59sockinfo::set (const conf_node *conf, u8 prot_) 64sockinfo::set (const conf_node *conf, u8 prot_)
60{ 65{
66 if (prot_ == PROT_DNSv4)
67 {
68 host = 0x01010101; port = htons (conf->id); prot = prot_;
69 }
70 else
61 set (conf->hostname, 71 set (conf->hostname,
62 prot_ == PROT_UDPv4 ? conf->udp_port 72 prot_ == PROT_UDPv4 ? conf->udp_port
63 : prot_ == PROT_TCPv4 ? conf->tcp_port 73 : prot_ == PROT_TCPv4 ? conf->tcp_port
74 : prot_ == PROT_DNSv4 ? conf->dns_port
64 : 0, 75 : 0,
65 prot_); 76 prot_);
66} 77}
67 78
68const sockaddr * 79const sockaddr *
69sockinfo::sav4() const 80sockinfo::sav4() const
70{ 81{
102sockinfo::supported_protocols (conf_node *conf) 113sockinfo::supported_protocols (conf_node *conf)
103{ 114{
104 u8 protocols = prot; 115 u8 protocols = prot;
105 116
106 if (prot & (PROT_UDPv4 | PROT_TCPv4)) 117 if (prot & (PROT_UDPv4 | PROT_TCPv4))
107 protocols |= PROT_IPv4; 118 protocols |= PROT_IPv4 | PROT_ICMPv4;
108 119
109 if (conf 120 if (conf
110 && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4) 121 && prot & PROTv4
111 && conf->protocols & PROT_UDPv4 122 && conf->protocols & PROT_UDPv4
112 && conf->udp_port) 123 && conf->udp_port)
113 protocols |= PROT_UDPv4; 124 protocols |= PROT_UDPv4;
114 125
115 if (conf 126 if (conf
116 && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4) 127 && prot & PROTv4
117 && conf->protocols & PROT_TCPv4 128 && conf->protocols & PROT_TCPv4
118 && conf->tcp_port) 129 && conf->tcp_port)
119 protocols |= PROT_TCPv4; 130 protocols |= PROT_TCPv4;
120 131
132 if (conf
133 && prot & PROTv4
134 && conf->protocols & PROT_DNSv4
135 && conf->dns_port)
136 protocols |= PROT_DNSv4;
137
121 return protocols; 138 return protocols;
122} 139}
123 140
124bool 141bool
125sockinfo::upgrade_protocol (u8 prot_, conf_node *conf) 142sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
126{ 143{
127 if (prot_ == prot) 144 if (prot_ == prot)
128 return true; 145 return true;
129 146
130 if (prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4) 147 if (prot & PROTv4
131 && prot_ & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4)) 148 && prot_ & PROTv4)
132 { 149 {
133 if (prot_ & PROT_IPv4) 150 if (prot_ & (PROT_IPv4 | PROT_ICMPv4))
134 { 151 {
135 prot = prot_; 152 prot = prot_;
136 port = 0; 153 port = 0;
137 return true; 154 return true;
138 } 155 }
154 { 171 {
155 prot = prot_; 172 prot = prot_;
156 port = htons (conf->tcp_port); 173 port = htons (conf->tcp_port);
157 return true; 174 return true;
158 } 175 }
176
177#if ENABLE_DNS
178 if (conf
179 && prot_ & PROT_DNSv4
180 && conf->protocols & PROT_DNSv4
181 && conf->dns_port)
182 {
183 set (::conf.dns_forw_host, ::conf.dns_forw_port, prot_);
184 return true;
185 }
186#endif
159 } 187 }
160 188
161 return false; 189 return false;
162} 190}
163 191

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines