Pithos_GSS » Java GSS CLI Client

#!/usr/bin/env python

from mechanize import Browser

from getpass import getpass

import sys, re

class ShibError(Exception):

    def __init__(self, value):

        self.parameter = value

    def __str__(self):

        return repr(self.parameter)

def getGSSNonce(edupersonPrincipalName):

    bot = Browser()

    response = bot.open("http://gss.grnet.gr/gss/nonce?user=%s" % edupersonPrincipalName)

    return response.read().rstrip()

def getGSSToken(edupersonPrincipalName, nonce):

    bot = Browser()

    response = bot.open("https://gss.grnet.gr/gss/token?user=%s&nonce=%s" % (edupersonPrincipalName,nonce))

    return response.read().rstrip()

def shibOpen(url,username="",password=""):

    # Create our www bot and send it to the website

    bot = Browser()

    bot.open(url)

    # Check if we were redirected to the WAYF

    if not re.match(r'https://wayf\.grnet\.gr',bot.geturl()):

        raise ShibError("We weren't redirected to WAYF, exiting...\n")

    sys.stderr.write("Hit the WAYF server\n")

    # Fill in the form ;-)

    bot.select_form(name="IdPList")

    bot["user_idp"] = ['http://www.grnet.gr/aai:admin.grnet.gr']

    # Submit the form. This will almost certainly

    # fail with a 401.

    try:

        response = bot.submit()

    except:

        bot.add_password(bot.geturl(),username,password)

        response = bot.reload()

    # Check if we were redirected back to the SP

    if not re.match(r'.*/SSO\?shire=', bot.geturl()):

        raise ShibError("The IdP didn't send us back to the SP, something's wrong\n")

    sys.stderr.write("Back at the SP\n")

    # Select the first form. The form has no name, only id='shibboleth'

    bot.select_form(nr=0)

    response = bot.submit()

    return response

if __name__ == "__main__":

    if len(sys.argv) != 2:

        print "Usage: %s <edupersonPrincipalName>" % sys.argv[0]

        sys.exit(1)

    nonce = getGSSNonce(sys.argv[1])

    sys.stderr.write("Got nonce: %s\n" % nonce)

    response = shibOpen("https://gss.grnet.gr/gss/login?nonce=%s" % nonce, sys.argv[1].split("@")[0], getpass())

    print response.read()

    print getGSSToken(sys.argv[1],nonce)