From 42d62fe28a9476b35eae8a2551463d6166347665 Mon Sep 17 00:00:00 2001
From: James Lu <james@overdrivenetworks.com>
Date: Sat, 23 Sep 2017 13:20:58 -0700
Subject: [PATCH] p10: pass IPv6 IPs on to supported servers

Closes #254.
---
 protocols/p10.py | 36 ++++++++++++++++++++++++++++++++----
 1 file changed, 32 insertions(+), 4 deletions(-)

diff --git a/protocols/p10.py b/protocols/p10.py
index d7bd093..b0fc364 100644
--- a/protocols/p10.py
+++ b/protocols/p10.py
@@ -27,7 +27,7 @@ def p10b64encode(num, length=2):
     """
     # Pack the given number as an unsigned int.
     sidbytes = struct.pack('>I', num)[1:]
-    sid = base64.b64encode(sidbytes, b'[]')[-2:]
+    sid = base64.b64encode(sidbytes, b'[]')[-length:]
     return sid.decode()  # Return a string, not bytes.
 
 class P10SIDGenerator():
@@ -241,6 +241,28 @@ class P10Protocol(IRCS2SProtocol):
                 ip = '0' + ip
             return ip
 
+    @staticmethod
+    def encode_p10_ipv6(ip):
+        """Encodes a P10 IPv6 address."""
+        # This method is documented briefly at https://github.com/evilnet/nefarious2/blob/47c8bea/doc/p10.txt#L723
+        # Basically, each address chunk is encoded into a 3-length word, with :: replaced by _
+        # e.g. '1:2::3' -> AABAAC_AAD
+        #      '::1' -> AAA_AAB
+        ipbits = []
+        for ipbit in ip.split(':'):
+            if not ipbit:
+                # split() creates an empty string between the two colons in "::" if it exists
+                # Use this to our advantage here.
+                ipbits.append('_')
+            else:
+                ipbit = int(ipbit, base=16)
+                ipbits.append(p10b64encode(ipbit, length=3))
+
+        encoded_ip = ''.join(ipbits)
+        if encoded_ip.startswith('_'):  # Special case for ::1, as "__AAA" is probably invalid
+            encoded_ip = 'AAA' + encoded_ip
+        return encoded_ip
+
     ### COMMANDS
 
     def spawn_client(self, nick, ident='null', host='null', realhost=None, modes=set(),
@@ -288,11 +310,14 @@ class P10Protocol(IRCS2SProtocol):
 
         # Encode IPs when sending
         if ip_address(ip).version == 4:
-            # Thanks to Jobe @ evilnet for the tips here! -GL
+            # Thanks to Jobe for the tips here!
             ip = b'\x00\x00' + socket.inet_aton(ip)
             b64ip = base64.b64encode(ip, b'[]')[2:].decode()
-        else:  # TODO: propagate IPv6 address, but only if uplink supports it
-            b64ip = 'AAAAAA'
+        else:  # Propagate IPv6 address, but only if uplink supports it
+            if '6' in self._flags:
+                b64ip = self.encode_p10_ipv6(ip)
+            else:
+                b64ip = 'AAAAAA'
 
         self._send_with_prefix(server, "N {nick} 1 {ts} {ident} {host} {modes} {ip} {uid} "
                    ":{realname}".format(ts=ts, host=host, nick=nick, ident=ident, uid=uid,
@@ -780,6 +805,8 @@ class P10Protocol(IRCS2SProtocol):
 
         desc = self.serverdata.get('serverdesc') or conf.conf['bot']['serverdesc']
 
+        self._flags = []
+
         # Enumerate modes, from https://github.com/evilnet/nefarious2/blob/master/doc/modes.txt
         p10_ircd = self.serverdata.get('p10_ircd', 'nefarious').lower()
         if p10_ircd == 'nefarious':
@@ -846,6 +873,7 @@ class P10Protocol(IRCS2SProtocol):
         sid = args[5][:2]
         sdesc = args[-1]
         self.servers[sid] = Server(self, source, servername, desc=sdesc)
+        self._flags = list(args[6])[1:]
 
         if self.uplink is None:
             # If we haven't already found our uplink, this is probably it.