source: TI12-security/trunk/python/ndg.security.server/ndg/security/server/SessionMgr/server-config.tac @ 1778

#!/usr/bin/env python
"""NDG Security Attribute Authority .tac file 

This file enables the Session Manager web service to be 
called under the Twisted framework

NERC Data Grid Project

P J Kershaw 23/11/06

Copyright (C) 2006 CCLRC & NERC

This software may be distributed under the terms of the Q Public License,
version 1.0 or later.
"""
import socket

from ZSI.twisted.WSresource import WSResource
from twisted.application import service, internet
from twisted.web.server import Site
from twisted.web.resource import Resource

from SessionMgr_services_server import SessionMgrService
from ndg.security.server.SessionMgr import SessionMgr

from ndg.security.common.SessionCookie import SessionCookie
from datetime import datetime, timedelta

class SessionMgrServiceSub(SessionMgrService, WSResource):
     def __init__(self):
         WSResource.__init__(self)
         
         # Initialize Attribute Authority class
         self.__sm = SessionMgr()

     def soap_addUser(self, ps, **kw):
         #import pdb;pdb.set_trace()
         request, response = SessionMgrService.soap_addUser(self, ps)
         return request, response

     def soap_connect(self, ps, **kw):
         #import pdb;pdb.set_trace()
         request, response = SessionMgrService.soap_connect(self, ps)
         response.set_element_proxyCert('PROXY CERT')
         
         dtExpiry = datetime.utcnow() + timedelta(seconds=60*60)
         cookie = SessionCookie(ndgID1='9'*64, 
                                ndgID2='0'*64,
                                dtExpiry=dtExpiry)
         
         response.set_element_cookie(cookie)
         return request, response

     def soap_disconnect(self, ps, **kw):
         #import pdb;pdb.set_trace()
         request, response = SessionMgrService.soap_disconnect(self, ps)
         return request, response

     def soap_reqAuthorisation(self, ps, **kw):
         #import pdb;pdb.set_trace()
         request, response = SessionMgrService.soap_reqAuthorisation(self, ps)
         response.set_element_attCert('ATTRIBUTE CERTIFICATE')
         response.set_element_statusCode('AcessGranted')
         return request, response

     def soap_getX509Cert(self, ps, **kw):
         #import pdb;pdb.set_trace()
         request, response = SessionMgrService.soap_getX509Cert(self, ps)
         response.set_element_x509Cert('X.509 Cert.')
         return request, response

portNum = 5000
hostname = socket.gethostname()

root = Resource()
root.putChild('SessionManager', SessionMgrServiceSub())
siteFactory = Site(root)
application = service.Application("ndgSecurityContainer")

# Try SSL
from twisted.internet import ssl
kw = {}
priKeyFilePath = '../../../../../Tests/Junk2-key.pem'
certFilePath = '../../../../../Tests/Junk2-cert.pem'
ctxFactory = ssl.DefaultOpenSSLContextFactory(priKeyFilePath, certFilePath)
port = internet.SSLServer(portNum, siteFactory, ctxFactory)

#port = internet.TCPServer(portNum, siteFactory)#, interface=hostname)
port.setServiceParent(application)