third_party/tlslite/test/twistedserver.py

   1
   2 from twisted.internet.protocol import Protocol, Factory
   3 from twisted.internet import reactor
   4 from twisted.protocols.policies import WrappingFactory
   5 from twisted.protocols.basic import LineReceiver
   6 from twisted.python import log
   7 from twisted.python.failure import Failure
   8 import sys
   9 from tlslite.api import *
  10
  11 s = open("./serverX509Cert.pem").read()
  12 x509 = X509()
  13 x509.parse(s)
  14 certChain = X509CertChain([x509])
  15
  16 s = open("./serverX509Key.pem").read()
  17 privateKey = parsePEMKey(s, private=True)
  18
  19 verifierDB = VerifierDB("verifierDB")
  20 verifierDB.open()
  21
  22 class Echo(LineReceiver):
  23   def connectionMade(self):
  24       self.transport.write("Welcome to the echo server!\r\n")
  25
  26   def lineReceived(self, line):
  27       self.transport.write(line + "\r\n")
  28
  29 class Echo1(Echo):
  30   def connectionMade(self):
  31       if not self.transport.tlsStarted:
  32           self.transport.setServerHandshakeOp(certChain=certChain,
  33                                               privateKey=privateKey,
  34                                               verifierDB=verifierDB)
  35       else:
  36           Echo.connectionMade(self)
  37
  38   def connectionLost(self, reason):
  39       pass #Handle any TLS exceptions here
  40
  41 class Echo2(Echo):
  42   def lineReceived(self, data):
  43       if data == "STARTTLS":
  44           self.transport.setServerHandshakeOp(certChain=certChain,
  45                                               privateKey=privateKey,
  46                                               verifierDB=verifierDB)
  47       else:
  48           Echo.lineReceived(self, data)
  49
  50   def connectionLost(self, reason):
  51       pass #Handle any TLS exceptions here
  52
  53 factory = Factory()
  54 factory.protocol = Echo1
  55 #factory.protocol = Echo2
  56
  57 wrappingFactory = WrappingFactory(factory)
  58 wrappingFactory.protocol = TLSTwistedProtocolWrapper
  59
  60 log.startLogging(sys.stdout)
  61 reactor.listenTCP(1079, wrappingFactory)
  62 reactor.run()