make script work even if ONIONSHARE_PLATFORM isn't set

This commit is contained in:
Micah Lee 2014-05-21 14:56:09 -04:00
parent 4d2a2cec6d
commit ce93125c99

View File

@ -48,13 +48,19 @@ def download():
basename = os.path.basename(filename) basename = os.path.basename(filename)
return send_from_directory(dirname, basename, as_attachment=True) return send_from_directory(dirname, basename, as_attachment=True)
def get_platform():
if 'ONIONSHARE_PLATFORM' in os.environ:
return os.environ['ONIONSHARE_PLATFORM']
else:
return 'unknown'
def tails_open_port(port): def tails_open_port(port):
if os.environ['ONIONSHARE_PLATFORM'] == 'Tails': if get_platform() == 'Tails':
print 'Punching a hole in the firewall' print 'Punching a hole in the firewall'
subprocess.call(['/sbin/iptables', '-I', 'OUTPUT', '-o', 'lo', '-p', 'tcp', '--dport', str(port), '-j', 'ACCEPT']) subprocess.call(['/sbin/iptables', '-I', 'OUTPUT', '-o', 'lo', '-p', 'tcp', '--dport', str(port), '-j', 'ACCEPT'])
def tails_close_port(port): def tails_close_port(port):
if os.environ['ONIONSHARE_PLATFORM'] == 'Tails': if get_platform() == 'Tails':
print 'Closing hole in firewall' print 'Closing hole in firewall'
subprocess.call(['/sbin/iptables', '-I', 'OUTPUT', '-o', 'lo', '-p', 'tcp', '--dport', str(port), '-j', 'REJECT']) subprocess.call(['/sbin/iptables', '-I', 'OUTPUT', '-o', 'lo', '-p', 'tcp', '--dport', str(port), '-j', 'REJECT'])