mirror of
https://github.com/tun2proxy/tun2proxy.git
synced 2025-05-27 04:07:11 +00:00
publish version 0.6.0
This commit is contained in:
parent
e3cc5ea1ce
commit
2ade72e79d
3 changed files with 15 additions and 4 deletions
|
@ -4,6 +4,7 @@ import os
|
|||
import subprocess
|
||||
import time
|
||||
import unittest
|
||||
import psutil
|
||||
|
||||
import dotenv
|
||||
import requests
|
||||
|
@ -33,6 +34,14 @@ def get_tool_path():
|
|||
default = default[0] if len(default) > 0 else 'tun2proxy-bin'
|
||||
return os.environ.get('TOOL_PATH', default)
|
||||
|
||||
def sudo_kill_process_and_children(proc):
|
||||
try:
|
||||
for child in psutil.Process(proc.pid).children(recursive=True):
|
||||
if child.name() == 'tun2proxy-bin':
|
||||
subprocess.run(['sudo', 'kill', str(child.pid)])
|
||||
subprocess.run(['sudo', 'kill', str(proc.pid)])
|
||||
except psutil.NoSuchProcess:
|
||||
pass
|
||||
|
||||
class Tun2ProxyTest(unittest.TestCase):
|
||||
@staticmethod
|
||||
|
@ -49,6 +58,7 @@ class Tun2ProxyTest(unittest.TestCase):
|
|||
except Exception as e:
|
||||
raise e
|
||||
finally:
|
||||
sudo_kill_process_and_children(p)
|
||||
p.terminate()
|
||||
p.wait()
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue