Decode address to unicode

sorz · Nov 21, 2016 · 628fa23 · 628fa23
1 parent fc2cb66
commit 628fa23
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sstpd/address.py b/sstpd/address.py
@@ -20,6 +20,8 @@ def _next_host(self):
 
 
     def register(self, address):
+        if isinstance(address, str):
+            address = address.decode()
         addr = ipaddress.ip_address(address)
         if addr in self._pool:
             raise RegisteredException()
@@ -45,6 +47,8 @@ def apply(self):
 
 
     def unregister(self, address):
+        if isinstance(address, str):
+            address = address.decode()
         addr = ipaddress.ip_address(address)
         try:
             self._pool.remove(addr)