home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / fnb101.zip / Lib / site-packages / Fnorb / orb / IIOPConnection.py < prev    next >
Text File  |  1999-06-28  |  5KB  |  175 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/IIOPConnection.py,v $
  31. # Version:      @(#)$RCSfile: IIOPConnection.py,v $ $Revision: 1.8 $
  32. #
  33. #############################################################################
  34. """ IIOPConnection class. """
  35.  
  36.  
  37. # Standard/built-in modules.
  38. import select, socket, struct
  39.  
  40. # Fnorb modules.
  41. import CORBA, Connection
  42.  
  43.  
  44. class IIOPConnection(Connection.Connection):
  45.     """ IIOPConnection class. """
  46.  
  47.     def __init__(self, _socket=None):
  48.     """ Provide an IIOP connection to a remote object!
  49.  
  50.     '_socket' is a socket that is *already* connected.
  51.  
  52.     """
  53.     # If we are given a "here's one I prepared earlier" socket, then we
  54.     # use that.
  55.     #
  56.     if _socket is not None:
  57.         self.__socket = _socket
  58.         self.__connected = 1
  59.  
  60.         # The 'TCP_NODELAY' option disables the TCP Nagle algorithm which
  61.         # prevents the transmission of small packets (like for example
  62.         # 'CloseConnection' messages).
  63.         #
  64.         # fixme: We don't use symbolic constants from the 'socket' module
  65.         # as they don't appear to be present on all platforms (eg. Linux!).
  66.         # Is this just an ommision or is there a reason?!?
  67.         self.__socket.setsockopt(6, # IPPROTO_TCP
  68.                      1, # TCP_NODELAY
  69.                      1) # Option on!
  70.  
  71.     # Otherwise, we will create a socket when the 'connect' method is
  72.     # called.
  73.     else:
  74.         self.__connected = 0
  75.  
  76.     return
  77.  
  78.     #########################################################################
  79.     # Connection interface.
  80.     #########################################################################
  81.  
  82.     def connect(self, address):
  83.     """ Connect to the remote object. """
  84.  
  85.     # fixme: Need a better address abstraction than just tuples.
  86.     (host, port) = address
  87.  
  88.     # Create a socket and connect to the remote object.
  89.     try:
  90.         self.__socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  91.         self.__socket.connect(host, port)
  92.  
  93.         # Set the socket by default to NON-blocking mode.
  94.         self.__socket.setblocking(0)
  95.  
  96.         # Connected!
  97.         self.__connected = 1
  98.  
  99.     except socket.error:
  100.         raise CORBA.COMM_FAILURE() # System exception.
  101.  
  102.     return
  103.  
  104.     def disconnect(self):
  105.     """ Disconnect from the remote server. """
  106.  
  107.     # Make sure that we are actually connected!
  108.     if self.__connected:
  109.         try:
  110.         # Close the connection.
  111.         self.__socket.close()
  112.  
  113.         except socket.error:
  114.         pass
  115.  
  116.         # Disconnected.
  117.         self.__connected = 0
  118.  
  119.     return
  120.  
  121.     def send(self, data, block=0):
  122.     """ Send as much of the data as we can. """
  123.  
  124.     try:
  125.         n = self.__socket.send(data)
  126.  
  127.     except socket.error:
  128.         raise CORBA.COMM_FAILURE(0, CORBA.COMPLETED_MAYBE)
  129.  
  130.     return n
  131.  
  132.     def recv(self, n):
  133.     """ Receive at most 'n' bytes of data. """
  134.     
  135.     try:
  136.         # Block until there is some data present on the socket.
  137.         #
  138.         # We do a 'select' instead of a blocking read, because 'select'
  139.         # can be interrupted by closing the socket. This is useful in
  140.         # the presence of threads when we need to shut down the read
  141.         # thread.
  142.         (iwtd, owtd, ewtd) = select.select([self.__socket], [], [])
  143.  
  144.         # Read the data.
  145.         data = self.__socket.recv(n)
  146.  
  147.     # fixme: The 'AttributeError' is caught because it seems that on
  148.     # Windows platforms this exception is raised when the socket has been
  149.     # closed by the peer?
  150.     except (select.error, socket.error, ValueError, AttributeError):
  151.         raise CORBA.COMM_FAILURE(0, CORBA.COMPLETED_MAYBE)
  152.     
  153.     return data
  154.  
  155.     def is_connected(self):
  156.     """ Are we connected to the remote address? """
  157.  
  158.     return self.__connected
  159.  
  160.     def blocking(self, blocking):
  161.     """ Set the connection to blocking (1) or non-blocking (0). """
  162.  
  163.     self.__socket.setblocking(blocking)
  164.     return
  165.  
  166.     def handle(self):
  167.     """ Return my underlying I/O handle.
  168.  
  169.     In this case, my I/O handle is the file descriptor of my socket. 
  170.  
  171.     """
  172.     return self.__socket.fileno()
  173.  
  174. #############################################################################
  175.