08-ssl-fake-cacert.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. #!/usr/bin/env python3
  2. from mosq_test_helper import *
  3. port = mosq_test.get_lib_port()
  4. if sys.version < '2.7':
  5. print("WARNING: SSL not supported on Python 2.6")
  6. exit(0)
  7. sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  8. sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  9. ssock = ssl.wrap_socket(sock, ca_certs="../ssl/all-ca.crt",
  10. keyfile="../ssl/server.key", certfile="../ssl/server.crt",
  11. server_side=True, cert_reqs=ssl.CERT_REQUIRED)
  12. ssock.settimeout(10)
  13. ssock.bind(('', port))
  14. ssock.listen(5)
  15. client_args = sys.argv[1:]
  16. env = dict(os.environ)
  17. env['LD_LIBRARY_PATH'] = '../../lib:../../lib/cpp'
  18. try:
  19. pp = env['PYTHONPATH']
  20. except KeyError:
  21. pp = ''
  22. env['PYTHONPATH'] = '../../lib/python:'+pp
  23. client = mosq_test.start_client(filename=sys.argv[1].replace('/', '-'), cmd=client_args, env=env, port=port)
  24. try:
  25. (conn, address) = ssock.accept()
  26. conn.close()
  27. except ssl.SSLError:
  28. # Expected error due to ca certs not matching.
  29. pass
  30. except mosq_test.TestError:
  31. pass
  32. finally:
  33. time.sleep(1.0)
  34. try:
  35. client.terminate()
  36. except OSError:
  37. pass
  38. client.wait()
  39. ssock.close()
  40. if client.returncode == 0:
  41. exit(0)
  42. else:
  43. exit(1)