home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / fnb101.zip / Lib / site-packages / Fnorb / orb / GIOPConnectionHandler.py < prev    next >
Text File  |  1999-06-28  |  6KB  |  193 lines
  1. #!/usr/bin/env python
  2. #############################################################################
  3. # Copyright (C) DSTC Pty Ltd (ACN 052 372 577) 1997, 1998, 1999
  4. # All Rights Reserved.
  5. #
  6. # The software contained on this media is the property of the DSTC Pty
  7. # Ltd.  Use of this software is strictly in accordance with the
  8. # license agreement in the accompanying LICENSE.HTML file.  If your
  9. # distribution of this software does not contain a LICENSE.HTML file
  10. # then you have no rights to use this software in any manner and
  11. # should contact DSTC at the address below to determine an appropriate
  12. # licensing arrangement.
  13. # 
  14. #      DSTC Pty Ltd
  15. #      Level 7, GP South
  16. #      Staff House Road
  17. #      University of Queensland
  18. #      St Lucia, 4072
  19. #      Australia
  20. #      Tel: +61 7 3365 4310
  21. #      Fax: +61 7 3365 4311
  22. #      Email: enquiries@dstc.edu.au
  23. # 
  24. # This software is being provided "AS IS" without warranty of any
  25. # kind.  In no event shall DSTC Pty Ltd be liable for damage of any
  26. # kind arising out of or in connection with the use or performance of
  27. # this software.
  28. #
  29. # Project:      Fnorb
  30. # File:         $Source: /units/arch/src/Fnorb/orb/RCS/GIOPConnectionHandler.py,v $
  31. # Version:      @(#)$RCSfile: GIOPConnectionHandler.py,v $ $Revision: 1.5 $
  32. #
  33. #############################################################################
  34. """ GIOPConnectionHandler class. """
  35.  
  36.  
  37. # Standard/built-in modules.
  38. import string
  39.  
  40. # Fnorb modules.
  41. import CORBA, OctetStream
  42.  
  43.  
  44. class GIOPConnectionHandlerListener:
  45.     """ Listener interface for the GIOPConnectionHandler class. """
  46.  
  47.     def message_received(self, reply_header, cursor):
  48.     """ Called when a complete GIOP message has been received. """
  49.  
  50.     pass
  51.  
  52.     def message_sent(self):
  53.     """ Called when a complete GIOP message has been sent. """
  54.     
  55.     pass
  56.  
  57.  
  58. class GIOPConnectionHandler:
  59.     """  GIOPConnectionHandler class. """
  60.     
  61.     # Maximum size of the buffer used for reading from connections (we
  62.     # deliberately do *not* make this figure a power of two to make sure that
  63.     # memory allocations will be as efficient as possible).
  64.     BUFSIZE = 8000
  65.  
  66.     def __init__(self, listener, connection):
  67.     """ Provide an interface to a remote object!
  68.  
  69.     'listener'   is the listener that is using this handler.
  70.     'connection' is the connection to handle!
  71.  
  72.     """
  73.     self.__listener = listener
  74.     self.__connection = connection
  75.  
  76.     # Details of the current incoming GIOP message.
  77.     self.__incoming = []
  78.     self.__incoming_size = 0
  79.     self.__message_size = 0
  80.  
  81.     # Details of the current outgoing GIOP message.
  82.     self.__outgoing = None
  83.     self.__outgoing_size = 0
  84.  
  85.     return
  86.  
  87.     def pseudo__del__(self):
  88.     """ Pseudo destructor to remove circular references. """
  89.  
  90.     # Clean up the reference to the listener.
  91.     del self.__listener
  92.  
  93.     return
  94.  
  95.     #########################################################################
  96.     # GIOPConnectionHandler interface.
  97.     #########################################################################
  98.  
  99.     def send(self, message):
  100.     """ Perform a single 'send' operation on the connection. """
  101.  
  102.     if self.__outgoing is None:
  103.         # Extract the string of bytes from the message.
  104.         self.__outgoing = message
  105.  
  106.     # Extract the string of bytes from the message.
  107.     data = self.__outgoing.data()
  108.  
  109.     # Starting a new message.
  110.     if self.__outgoing_size == 0:
  111.         n = self.__connection.send(data)
  112.         self.__outgoing_size = self.__outgoing_size + n
  113.  
  114.     # Continuing transmission of a previous message.
  115.         elif self.__outgoing_size > 0 and self.__outgoing_size < len(data):
  116.         n = self.__connection.send(data[self.__outgoing_size:])
  117.         self.__outgoing_size = self.__outgoing_size + n
  118.  
  119.     # If the entire message has been sent.
  120.     if self.__outgoing_size > 0 and self.__outgoing_size == len(data):
  121.         # Get ready to start the next message.
  122.         self.__outgoing = None
  123.         self.__outgoing_size = 0
  124.  
  125.         # Message sent!
  126.         self.__listener.message_sent()
  127.  
  128.     return n
  129.  
  130.     def recv(self):
  131.     """ Perform a single 'recv' operation on the connection. """
  132.     
  133.     # Read the GIOP message header.
  134.     if self.__incoming_size < 12:
  135.         # Get the next chunk of the GIOP header.
  136.         chunk = self.__connection.recv(12 - self.__incoming_size)
  137.         # Zero length reads not allowed!
  138.         if len(chunk) == 0:
  139.         raise CORBA.COMM_FAILURE(0, CORBA.COMPLETED_MAYBE)
  140.  
  141.         # Add the chunk to the incoming message.
  142.         self.__incoming_size = self.__incoming_size + len(chunk)
  143.         self.__incoming.append(chunk)
  144.  
  145.         # Have we got the entire header now?
  146.         if self.__incoming_size == 12:
  147.         # Join the chunks together.
  148.         data = string.join(self.__incoming, '')
  149.  
  150.         # Unmarshal the header.
  151.         octet_stream = OctetStream.GIOPMessage(data)
  152.  
  153.         # Get the size of the message body.
  154.         self.__message_size = octet_stream.header().message_size
  155.  
  156.     # Read the message body.
  157.     elif self.__incoming_size >= 12 and \
  158.          self.__incoming_size < self.__message_size + 12:
  159.  
  160.         # Work out the number of bytes remaining.
  161.         remaining = (self.__message_size + 12) - self.__incoming_size
  162.  
  163.         # Calculate the size of the buffer.
  164.         bufsize = min(remaining, GIOPConnectionHandler.BUFSIZE)
  165.  
  166.         # Get the next chunk of the message.
  167.         chunk = self.__connection.recv(bufsize)
  168.         if len(chunk) == 0:
  169.         raise CORBA.COMM_FAILURE(0, CORBA.COMPLETED_MAYBE)
  170.             
  171.         # Add the chunk to the incoming message.
  172.         self.__incoming_size = self.__incoming_size + len(chunk)
  173.         self.__incoming.append(chunk)
  174.  
  175.     # Has the entire message been read?
  176.     if self.__incoming_size >= 12 and \
  177.        self.__incoming_size == self.__message_size + 12:
  178.  
  179.         # Join the chunks together.
  180.         data = string.join(self.__incoming, '')
  181.  
  182.         # Reset ready for the next message.
  183.         self.__incoming = []
  184.         self.__incoming_size = 0
  185.         self.__message_size = 0
  186.  
  187.         # Message received!
  188.         self.__listener.message_received(OctetStream.GIOPMessage(data))
  189.  
  190.     return
  191.  
  192. #############################################################################
  193.