Changeset 156 for etherws/trunk/etherws.py

Timestamp:

05/19/12 03:00:33 (12 years ago)

Author:

atzm

Message:

• server cert verification support

File:

• etherws/trunk/etherws.py (modified) (4 diffs)

etherws/trunk/etherws.py

@@ -43 +43 @@
 import os
 import sys
+import ssl
 import base64
 import hashlib
@@ -222 +223 @@
                         if passwd.startswith('{SHA}'):
                             users[name] = passwd[5:]
+            if not users:
+                raise RuntimeError('no valid users found')
         except TypeError:
             pass
@@ -266 +269 @@
     if args.debug:
         websocket.enableTrace(True)
+
+    if not args.insecure:
+        websocket._SSLSocketWrapper = \
+            lambda s: ssl.wrap_socket(s, cert_reqs=ssl.CERT_REQUIRED,
+                                      ca_certs=args.cacerts)
 
     passwd = None
@@ -308 +316 @@
     parser_client = subparsers.add_parser('client')
     parser_client.add_argument('--uri', action='store', required=True)
+    parser_client.add_argument('--insecure', action='store_true', default=False)
+    parser_client.add_argument('--cacerts', action='store')
     parser_client.add_argument('--user', action='store')