diff options
author | Rodolfo Alonso Hernandez <rodolfo.alonso.hernandez@intel.com> | 2018-07-10 07:48:43 +0000 |
---|---|---|
committer | Gerrit Code Review <gerrit@opnfv.org> | 2018-07-10 07:48:43 +0000 |
commit | 9fc8dd0d24489c8362ab3b20f0e62888e896c283 (patch) | |
tree | 46f2d5c21fe978de0c39c7d0a42c7553e97c1452 /yardstick/common | |
parent | 9fd6661f0480ee94c742611ce9c478e30ca5e416 (diff) | |
parent | c70be07682566ae05bafbf05a7e8c85ae4e1297e (diff) |
Merge "Add send socket commands function"
Diffstat (limited to 'yardstick/common')
-rw-r--r-- | yardstick/common/utils.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/yardstick/common/utils.py b/yardstick/common/utils.py index f9fe0e336..85cecc714 100644 --- a/yardstick/common/utils.py +++ b/yardstick/common/utils.py @@ -527,3 +527,25 @@ def wait_until_true(predicate, timeout=60, sleep=1, exception=None): if exception and issubclass(exception, Exception): raise exception # pylint: disable=raising-bad-type raise exceptions.WaitTimeout + + +def send_socket_command(host, port, command): + """Send a string command to a specific port in a host + + :param host: (str) ip or hostname of the host + :param port: (int) port number + :param command: (str) command to send + :return: 0 if success, error number if error + """ + sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) + ret = 0 + try: + err_number = sock.connect_ex((host, int(port))) + if err_number != 0: + return err_number + sock.sendall(six.b(command)) + except Exception: # pylint: disable=broad-except + ret = 1 + finally: + sock.close() + return ret |