add test-f5-login.py script

Signed-off-by: Daniel Lenski <dlenski@gmail.com>

diff --git a/test-f5-login.py b/test-f5-login.py
new file mode 100755 (executable)
index 0000000..de988ae
--- /dev/null
+++ b/test-f5-login.py
@@ -0,0 +1,98 @@
+#!/usr/bin/python3
+
+from __future__ import print_function
+from sys import stderr, version_info, platform
+if (version_info >= (3, 0)):
+    from urllib.parse import urlparse, urlencode
+    raw_input = input
+    import http.client as httplib
+else:
+    from urlparse import urlparse
+    from urllib import urlencode
+    import httplib
+import sys
+import requests
+import argparse
+import getpass
+from shlex import quote
+
+p = argparse.ArgumentParser()
+p.add_argument('-v','--verbose', default=0, action='count')
+p.add_argument('endpoint', help='F5 server (or complete URL, e.g. https://f5.vpn.com/my.policy)')
+p.add_argument('extra', nargs='*', help='Extra field to pass to include in the login query string (e.g. "foo=bar")')
+g = p.add_argument_group('Login credentials')
+g.add_argument('-u','--username', help='Username (will prompt if unspecified)')
+g.add_argument('-p','--password', help='Password (will prompt if unspecified)')
+g.add_argument('-c','--cert', help='PEM file containing client certificate (and optionally private key)')
+g.add_argument('--key', help='PEM file containing client private key (if not included in same file as certificate)')
+p.add_argument('--no-verify', dest='verify', action='store_false', default=True, help='Ignore invalid server certificate')
+args = p.parse_args()
+
+if args.verbose > 1:
+    httplib.HTTPConnection.debuglevel = 1
+
+extra = dict(x.split('=', 1) for x in args.extra)
+endpoint = urlparse(('https://' if '//' not in args.endpoint else '') + args.endpoint, 'https:')
+
+if args.cert and args.key:
+    cert = (args.cert, args.key)
+elif args.cert:
+    cert = (args.cert, None)
+elif args.key:
+    p.error('--key specified without --cert')
+else:
+    cert = None
+
+s = requests.Session()
+s.cert = cert
+s.verify = args.verify
+#s.headers['User-Agent'] = 'Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:75.0) Gecko/20100101 Firefox/75.0'
+
+print("Initial GET / to populate LastMRH_Session and MRHSession cookies...", file=stderr)
+res = s.get(endpoint.geturl(), allow_redirects=False)
+assert any(c.value for c in s.cookies if c.name=='MRHSession') and any(c.value for c in s.cookies if c.name=='LastMRH_Session')
+print("GET /my.policy to update MRHSession cookie...", file=stderr)
+res = s.get(endpoint._replace(path='/my.policy').geturl(), allow_redirects=False, headers={'Referer': res.url})
+#print("GET /vdesk/timeoutagent-i.php to update TIN cookie (probably unnecessary).")
+#res = s.get(endpoint._replace(path='/vdesk/timeoutagent-i.php').geturl(), allow_redirects=False)
+
+# Send login credentials
+if args.username == None:
+    args.username = raw_input('Username: ')
+if args.password == None:
+    args.password = getpass.getpass('Password: ')
+data=dict(username=args.username, password=args.password,
+          **extra)
+print("POST /my.policy to submit login credentials...", file=stderr)
+res = s.post(endpoint._replace(path='/my.policy').geturl(), data=data, headers={'Referer': res.url})
+
+res.raise_for_status()
+
+# Build openconnect --cookie argument from the result:
+url = urlparse(res.url)
+if any(c.name=='MRHSession' for c in s.cookies) and url.path.startswith('/vdesk/'):
+    cookie = next(c.value for c in s.cookies if c.name=='MRHSession')
+    if args.verbose:
+        if cert:
+            cert_and_key = ' \\\n        ' + ' '.join('%s "%s"' % (opt, quote(fn)) for opt, fn in zip(('-c','-k'), cert) if fn)
+        else:
+            cert_and_key = ''
+
+        print('''
+Extracted connection cookie. Use this to connect:
+
+    echo %s | openconnect --protocol=f5%s --cookie-on-stdin %s
+
+''' % (quote(cookie), cert_and_key, quote(endpoint.netloc)), file=stderr)
+
+    varvals = {
+        'HOST': quote(url.netloc),
+        'COOKIE': quote(cookie),
+    }
+    print('\n'.join('%s=%s' % pair for pair in varvals.items()))
+
+# Just print the result
+else:
+    if args.verbose:
+        print(res.headers, file=stderr)
+    print(res.text)