+ Use sys.platform.startswith("freebsd") instead of matching just freebsd10.

+ This means support for any freebsd version (note that I only tested 11-CURRENT and 9)
This commit is contained in:
Mike C 2015-04-27 02:31:58 +01:00
parent 1c5277b21f
commit b942f98f37
2 changed files with 4 additions and 4 deletions

View File

@ -8,9 +8,9 @@ if sys.platform == "linux2":
elif sys.platform == "darwin": elif sys.platform == "darwin":
from . import osx from . import osx
resolver = osx.Resolver resolver = osx.Resolver
elif sys.platform == "freebsd10": elif sys.platform.startswith("freebsd"):
from . import osx from . import osx
resolver = osx.Resolver resolver = osx.Resolver
elif sys.platform == "win32": elif sys.platform == "win32":
from . import windows from . import windows
resolver = windows.Resolver resolver = windows.Resolver

View File

@ -13,7 +13,7 @@ def lookup(address, port, s):
if "ESTABLISHED:ESTABLISHED" in i and spec in i: if "ESTABLISHED:ESTABLISHED" in i and spec in i:
s = i.split() s = i.split()
if len(s) > 4: if len(s) > 4:
if sys.platform == "freebsd10": if sys.platform.startswith("freebsd"):
# strip parentheses for FreeBSD pfctl # strip parentheses for FreeBSD pfctl
s = s[3][1:-1].split(":") s = s[3][1:-1].split(":")
else: else:
@ -21,4 +21,4 @@ def lookup(address, port, s):
if len(s) == 2: if len(s) == 2:
return s[0], int(s[1]) return s[0], int(s[1])
raise RuntimeError("Could not resolve original destination.") raise RuntimeError("Could not resolve original destination.")