From 8cdc23a132dca5fb048121ded2360ef300a0e98b Mon Sep 17 00:00:00 2001
From: Seth Schoen <schoen@eff.org>
Date: Fri, 10 Aug 2012 17:57:17 -0700
Subject: [PATCH] set server on command line with --server, or CHOCOLATESERVER

---
 client-webserver/client.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/client-webserver/client.py b/client-webserver/client.py
index 049e1f6d1..1740c871f 100755
--- a/client-webserver/client.py
+++ b/client-webserver/client.py
@@ -26,11 +26,12 @@ from CONFIG import cert_file, chain_file
 # them by default
 allow_raw_ipv6_server = False
 
-opts = getopt.getopt(sys.argv[1:], "", ["text", "privkey=", "csr="])
+opts = getopt.getopt(sys.argv[1:], "", ["text", "privkey=", "csr=", "server="])
 
 curses = True
 csr = None
 privkey = None
+server = None
 for opt in opts[0]:
     if opt[0] == "--text":
         curses = False
@@ -38,6 +39,8 @@ for opt in opts[0]:
         csr = opt[1]
     if opt[0] == "--privkey":
         privkey = opt[1]
+    if opt[0] == "--server":
+        server = opt[1]
 names = opts[1]
 
 if curses:
@@ -154,10 +157,8 @@ def authenticate():
     Main call to do DV_SNI validation and deploy the trustify certificate
     TODO: This should be turned into a class...
     """
-    assert len(sys.argv) > 1 or "CHOCOLATESERVER" in os.environ, "Must specify server via command line or CHOCOLATESERVER environment variable."
-    if len(sys.argv) > 1:
-        server = sys.argv[1]
-    else:
+    assert server or "CHOCOLATESERVER" in os.environ, "Must specify server via command line or CHOCOLATESERVER environment variable."
+    if "CHOCOLATESERVER" in os.environ:
         server = os.environ["CHOCOLATESERVER"]
 
     assert is_hostname_sane(server), `server` + " is an impossible hostname"