--- gvpe/src/sockinfo.C	2003/03/28 04:05:10	1.1
+++ gvpe/src/sockinfo.C	2003/04/06 04:31:51	1.7
@@ -1,5 +1,5 @@
 /*
-    sockinfo.c -- socket address management
+    sockinfo.C -- socket address management
     
     This program is free software; you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by
@@ -18,7 +18,8 @@
 
 #include "config.h"
 
-#include "arpa/inet.h"
+#include <arpa/inet.h>
+#include <netdb.h>
 
 #include "gettext.h"
 
@@ -33,12 +34,16 @@
 }
 
 void
-sockinfo::set (const conf_node *conf)
+sockinfo::set (const conf_node *conf, u8 prot_)
 {
+  prot = prot_;
   host = 0;
-  port = htons (conf->udp_port);
+  port = prot == PROT_UDPv4 ? htons (conf->udp_port)
+       : prot == PROT_TCPv4 ? htons (conf->tcp_port)
+       : 0;
 
-  if (conf->hostname)
+  if (prot & (PROT_UDPv4 | PROT_TCPv4 | PROT_IPv4)
+      && conf->hostname)
     {
       struct hostent *he = gethostbyname (conf->hostname);
 
@@ -47,8 +52,6 @@
         {
           //sa->sin_family = he->h_addrtype;
           memcpy (&host, he->h_addr_list[0], 4);
-
-          prot = PROT_UDPv4 | PROT_IPv4;
         }
       else
         slog (L_NOTICE, _("unable to resolve host '%s'"), conf->hostname);
@@ -68,7 +71,7 @@
   return (const sockaddr *)&sa;
 }
 
-static char hostport[15 + 1 + 5 + 1]; // IPv4 : port
+static char hostport[10 + 15 + 1 + 5 + 1]; // protype / IPv4 : port
 
 const char *
 sockinfo::ntoa () const
@@ -84,8 +87,85 @@
 {
   in_addr ia = { host };
 
-  sprintf (hostport, "%.15s:%d", inet_ntoa (ia), ntohs (port) & 0xffff);
+  sprintf (hostport, "%s/%.15s:%d", strprotocol (prot), inet_ntoa (ia), ntohs (port) & 0xffff);
 
   return hostport;
 }
 
+u8
+sockinfo::supported_protocols (conf_node *conf)
+{
+  u8 protocols = prot;
+
+  if (prot & (PROT_UDPv4 | PROT_TCPv4))
+    protocols |= PROT_IPv4;
+
+  if (conf
+      && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4)
+      && conf->protocols & PROT_UDPv4
+      && conf->udp_port)
+    protocols |= PROT_UDPv4;
+
+  if (conf
+      && prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4)
+      && conf->protocols & PROT_TCPv4
+      && conf->tcp_port)
+    protocols |= PROT_TCPv4;
+
+  return protocols;
+}
+
+bool
+sockinfo::upgrade_protocol (u8 prot_, conf_node *conf)
+{
+  if (prot_ == prot)
+    return true;
+
+  if (prot & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4)
+      && prot_ & (PROT_IPv4 | PROT_UDPv4 | PROT_TCPv4))
+    {
+      if (prot_ & PROT_IPv4)
+        {
+          prot = prot_;
+          port = 0;
+          return true;
+        }
+
+      if (conf
+          && prot_ & PROT_UDPv4
+          && conf->protocols & PROT_UDPv4
+          && conf->udp_port)
+        {
+          prot = prot_;
+          port = htons (conf->udp_port);
+          return true;
+        }
+
+      if (conf
+          && prot_ & PROT_TCPv4
+          && conf->protocols & PROT_TCPv4
+          && conf->tcp_port)
+        {
+          prot = prot_;
+          port = htons (conf->tcp_port);
+          return true;
+        }
+    }
+
+  return false;
+}
+
+bool
+operator == (const sockinfo &a, const sockinfo &b)
+{
+  return a.host == b.host && a.port == b.port && a.prot == b.prot;
+}
+
+bool
+operator < (const sockinfo &a, const sockinfo &b)
+{
+  return a.host < b.host
+         || (a.host == b.host && (a.port < b.port
+                                  || (a.port == b.port && a.prot < b.prot)));
+}
+