diff options
author | Maciej Żenczykowski <maze@google.com> | 2020-05-27 11:06:15 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2020-05-27 11:06:15 +0000 |
commit | 5447b97eb453423e60ff684050084643251a30b3 (patch) | |
tree | 4e89fe4575db7e202bccb2e259e84d4df9520638 | |
parent | b547349334125a4395a45b2b936611673388c0ed (diff) | |
parent | a1197aaead58f2b78c829d1615833aecdc32f2c0 (diff) | |
download | tests-5447b97eb453423e60ff684050084643251a30b3.tar.gz |
net-test: attempt to use namespaces on devices with ADB/TCP am: a1197aaead
Change-Id: I4c2eaae8cfd3f33bcd364a9b45371453ba3741cf
-rwxr-xr-x | net/test/all_tests.py | 5 | ||||
-rw-r--r-- | net/test/namespace.py | 17 |
2 files changed, 22 insertions, 0 deletions
diff --git a/net/test/all_tests.py b/net/test/all_tests.py index 5f6c583..17d9701 100755 --- a/net/test/all_tests.py +++ b/net/test/all_tests.py @@ -18,6 +18,8 @@ import importlib import sys import unittest +import namespace + test_modules = [ 'anycast_test', 'bpf_test', @@ -46,6 +48,9 @@ test_modules = [ ] if __name__ == '__main__': + # Check whether ADB over TCP is occupying TCP port 5555. + if namespace.HasEstablishedTcpSessionOnPort(5555): + namespace.IfPossibleEnterNewNetworkNamespace() # First, run InjectTests on all modules, to ensure that any parameterized # tests in those modules are injected. for name in test_modules: diff --git a/net/test/namespace.py b/net/test/namespace.py index f7fbaf4..85db654 100644 --- a/net/test/namespace.py +++ b/net/test/namespace.py @@ -19,8 +19,11 @@ import ctypes import ctypes.util import os +import socket import net_test +import sock_diag +import tcp_test # //include/linux/fs.h MNT_FORCE = 1 # Attempt to forcibily umount @@ -146,3 +149,17 @@ def IfPossibleEnterNewNetworkNamespace(): print 'succeeded.' return True + + +def HasEstablishedTcpSessionOnPort(port): + sd = sock_diag.SockDiag() + + sock_id = sd._EmptyInetDiagSockId() + sock_id.sport = port + + states = 1 << tcp_test.TCP_ESTABLISHED + + matches = sd.DumpAllInetSockets(socket.IPPROTO_TCP, "", + sock_id=sock_id, states=states) + + return len(matches) > 0 |