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.6 by pcg, Sat Apr 5 18:03:20 2003 UTC vs.
Revision 1.12 by pcg, Fri Apr 2 14:42:45 2004 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)
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_;
34} 37}
35 38
36void 39void sockinfo::set (const char *hostname, u16 port_, u8 prot_)
37sockinfo::set (const conf_node *conf, u8 prot_)
38{ 40{
39 prot = prot_; 41 prot = prot_;
40 host = 0; 42 host = 0;
41 port = prot == PROT_UDPv4 ? htons (conf->udp_port) 43 port = htons (port_);
42 : prot == PROT_TCPv4 ? htons (conf->tcp_port)
43 : 0;
44 44
45 if (prot && conf->hostname) 45 if (prot & PROTv4
46 && hostname)
46 { 47 {
47 struct hostent *he = gethostbyname (conf->hostname); 48 struct hostent *he = gethostbyname (hostname);
48 49
49 if (he 50 if (he
50 && he->h_addrtype == AF_INET && he->h_length == 4 && he->h_addr_list[0]) 51 && he->h_addrtype == AF_INET && he->h_length == 4 && he->h_addr_list[0])
51 { 52 {
52 //sa->sin_family = he->h_addrtype; 53 //sa->sin_family = he->h_addrtype;
53 memcpy (&host, he->h_addr_list[0], 4); 54 memcpy (&host, he->h_addr_list[0], 4);
54
55 } 55 }
56 else 56 else
57 slog (L_NOTICE, _("unable to resolve host '%s'"), conf->hostname); 57 slog (L_NOTICE, _("unable to resolve host '%s'"), hostname);
58 } 58 }
59} 59}
60 60
61void
62sockinfo::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 : prot_ == PROT_DNSv4 ? conf->dns_port
68 : 0,
69 prot_);
70}
61 71
62const sockaddr * 72const sockaddr *
63sockinfo::sav4() const 73sockinfo::sav4() const
64{ 74{
65 static sockaddr_in sa; 75 static sockaddr_in sa;
… …
69 sa.sin_addr.s_addr = host; 79 sa.sin_addr.s_addr = host;
70 80
71 return (const sockaddr *)&sa; 81 return (const sockaddr *)&sa;
72} 82}
73 83
74static char hostport[10 + 15 + 1 + 5 + 1]; // protype / IPv4 : port 84static char hostport[10 + 15 + 1 + 5 + 1]; // proto / IPv4 : port
75 85
76const char * 86const char *
77sockinfo::ntoa () const 87sockinfo::ntoa () const
78{ 88{
79 in_addr ia = { host }; 89 in_addr ia = { host };
… …
96sockinfo::supported_protocols (conf_node *conf) 106sockinfo::supported_protocols (conf_node *conf)
97{ 107{
98 u8 protocols = prot; 108 u8 protocols = prot;
99 109
100 if (prot & (PROT_UDPv4 | PROT_TCPv4)) 110 if (prot & (PROT_UDPv4 | PROT_TCPv4))
101 protocols |= PROT_IPv4; 111 protocols |= PROT_IPv4 | PROT_ICMPv4;
102 112
103 if (conf 113 if (conf
104 && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4) 114 && prot & PROTv4
105 && conf->protocols & PROT_UDPv4 115 && conf->protocols & PROT_UDPv4
106 && conf->udp_port) 116 && conf->udp_port)
107 protocols |= PROT_UDPv4; 117 protocols |= PROT_UDPv4;
108 118
109 if (conf 119 if (conf
110 && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4) 120 && prot & PROTv4
111 && conf->protocols & PROT_TCPv4 121 && conf->protocols & PROT_TCPv4
112 && conf->tcp_port) 122 && conf->tcp_port)
113 protocols |= PROT_TCPv4; 123 protocols |= PROT_TCPv4;
124
125 if (conf
126 && prot & PROTv4
127 && conf->protocols & PROT_DNSv4)
128 protocols |= PROT_DNSv4;
114 129
115 return protocols; 130 return protocols;
116} 131}
117 132
118bool 133bool
119sockinfo::upgrade_protocol (u8 prot_, conf_node *conf) 134sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
120{ 135{
121 if (prot_ == prot) 136 if (prot_ == prot)
122 return true; 137 return true;
123 138
124 if (prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4) 139 if (prot & PROTv4
125 && prot_ & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4)) 140 && prot_ & PROTv4)
126 { 141 {
127 if (prot_ & PROT_IPv4) 142 if (prot_ & (PROT_IPv4 | PROT_ICMPv4))
128 { 143 {
129 prot = prot_; 144 prot = prot_;
130 port = 0; 145 port = 0;
131 return true; 146 return true;
132 } 147 }

Diff Legend

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