1. /*

  2.  * @(#)Connection.java	1.85 03/01/23

  3.  *

  4.  * Copyright 2003 Sun Microsystems, Inc. All rights reserved.

  5.  * SUN PROPRIETARY/CONFIDENTIAL. Use is subject to license terms.

  6.  */

  7. 

  8. /*

  9.  * Licensed Materials - Property of IBM

  10.  * RMI-IIOP v1.0

  11.  * Copyright IBM Corp. 1998 1999  All Rights Reserved

  12.  *

  13.  * US Government Users Restricted Rights - Use, duplication or

  14.  * disclosure restricted by GSA ADP Schedule Contract with IBM Corp.

  15.  */

  16. 

  17. package com.sun.corba.se.internal.iiop;

  18. 

  19. import java.io.IOException;

  20. import java.net.Socket;

  21. import java.net.UnknownHostException;

  22. import java.net.InetAddress;

  23. import org.omg.CORBA.SystemException;

  24. import org.omg.CORBA.CompletionStatus;

  25. import org.omg.CORBA.OBJECT_NOT_EXIST;

  26. import org.omg.CORBA.MARSHAL;

  27. import org.omg.CORBA.INTERNAL;

  28. import org.omg.CORBA.DATA_CONVERSION;

  29. 

  30. import org.omg.CORBA.portable.*;

  31. 

  32. import com.sun.org.omg.SendingContext.CodeBase;

  33. import com.sun.corba.se.internal.core.ServerGIOP;

  34. import com.sun.corba.se.internal.core.EndPoint;

  35. import com.sun.corba.se.internal.core.IOR;

  36. import com.sun.corba.se.internal.core.MarshalOutputStream;

  37. import com.sun.corba.se.internal.orbutil.MinorCodes; //d11638

  38. import com.sun.corba.se.internal.orbutil.ORBUtility; //d11638

  39. import com.sun.corba.se.internal.core.GIOPVersion;

  40. import com.sun.corba.se.internal.iiop.messages.MessageBase;

  41. import com.sun.corba.se.internal.iiop.messages.LocateRequestMessage;

  42. import com.sun.corba.se.internal.iiop.messages.LocateReplyMessage;

  43. import com.sun.corba.se.internal.core.CodeSetComponentInfo;

  44. import com.sun.corba.se.internal.core.OSFCodeSetRegistry;

  45. 

  46. /**

  47.  * Common connection base class.

  48.  */

  49. abstract public class Connection

  50.     implements

  51.         com.sun.corba.se.connection.Connection

  52. {

  53. 

  54.     // Connection close states

  55.     // REVISIT - rather than define these "intermediate" constants,

  56.     // just use the one from MinorCodes directly where these are

  57.     // referenced.

  58.     public static final int CONN_ABORT = MinorCodes.CONN_ABORT;

  59.     public static final int CONN_REBIND = MinorCodes.CONN_REBIND;

  60. 

  61.     protected ORB orb;

  62.     protected Socket socket;    // The socket used for this connection.

  63.     protected long timeStamp = 0;

  64.     protected boolean isServer = false;

  65.     protected ConnectionTable connectionTable = null;

  66. 

  67.     // Negotiated code sets for char and wchar data

  68.     protected CodeSetComponentInfo.CodeSetContext codeSetContext = null;

  69. 

  70.     void dprint(String msg) {

  71. 	ORBUtility.dprint(this, msg);

  72.     }

  73. 

  74.     public ORB getORB() {

  75.         return orb;

  76.     }

  77. 

  78.     public Socket getSocket() {

  79. 	return socket;

  80.     }

  81. 

  82.     abstract public IIOPInputStream invoke(IIOPOutputStream s)

  83. 	throws SystemException;

  84. 

  85.     abstract public void delete();

  86. 

  87.     abstract public java.io.InputStream getInputStream();

  88. 

  89.     abstract public ServerGIOP getServerGIOP();

  90. 

  91.     abstract public IIOPInputStream send(IIOPOutputStream s, boolean oneWay);

  92. 

  93.     abstract public void sendReply(IIOPOutputStream s) throws Exception;

  94. 

  95. 

  96.     // All of Following abstract required for Connection cleanup

  97. 

  98.     abstract public void cleanUp() throws java.lang.Exception;

  99. 

  100.     abstract public boolean isBusy();

  101. 

  102.     abstract public void requestBegins();

  103.     abstract public void requestEnds(IIOPInputStream request);

  104. 

  105.     abstract public void print();

  106. 

  107.     abstract public void setConnection(Socket _socket, ConnectionTable ctab)

  108. 	throws java.lang.Exception;

  109.     abstract public void abortConnection();

  110. 

  111.     // Indicates whether or not ServiceContexts have even been exchange yet

  112.     abstract public boolean isPostInitialContexts();

  113. 

  114.     // Sets to true the state that ServiceContexts have indeed been exchanged

  115.     // once already

  116.     abstract public void setPostInitialContexts();

  117. 

  118.     public IOR locate(int id, byte [] key, IOR ior)

  119.     {

  120.         LocateRequestMessage msg;

  121.         IIOPOutputStream os;

  122.         IIOPInputStream is;

  123. 

  124.         GIOPVersion requestVersion =

  125.             GIOPVersion.chooseRequestVersion(orb, ior);

  126.         msg = MessageBase.createLocateRequest(orb, requestVersion, id, key);

  127. 

  128.         // This chooses the right buffering strategy for the locate msg.

  129.         // locate msgs 1.0 & 1.1 :=> grow, 1.2 :=> stream

  130.         //os = orb.newOutputStream(this);

  131.         os = com.sun.corba.se.internal.iiop.IIOPOutputStream.

  132.                 createIIOPOutputStreamForLocateMsg(

  133.                         requestVersion, orb, this);

  134.         os.setMessage(msg);

  135.         msg.write(os);

  136. 

  137.         is = send(os, false);

  138. 

  139.         LocateReplyMessage reply;

  140.         reply = (LocateReplyMessage) is.getMessage();

  141.         switch (reply.getReplyStatus()) {

  142. 	    case LocateReplyMessage.UNKNOWN_OBJECT:

  143.             throw new OBJECT_NOT_EXIST( MinorCodes.LOCATE_UNKNOWN_OBJECT,

  144. 					CompletionStatus.COMPLETED_NO );

  145. 

  146. 	    case LocateReplyMessage.OBJECT_HERE:

  147.             return null;

  148. 

  149.         case LocateReplyMessage.OBJECT_FORWARD:

  150. 	    case LocateReplyMessage.OBJECT_FORWARD_PERM: {

  151.             /*

  152.             IOR ior;

  153. 

  154.             ior = new IOR(orb);

  155.             ior.read(is);

  156. 

  157.             return ior;

  158.             */

  159.             return reply.getIOR();

  160. 	    }

  161.         }

  162. 

  163.         throw new INTERNAL( MinorCodes.BAD_LOCATE_REQUEST_STATUS,

  164.             		    CompletionStatus.COMPLETED_NO );

  165.     }

  166. 

  167.     void shutdown() {

  168.         try {

  169.             socket.close();

  170.         } catch (IOException ioex) {}

  171.     }

  172. 

  173.     public void stampTime() {

  174. 	connectionTable.stampTime(this);

  175.     }

  176. 

  177.     // Sets this connection's code base IOR.  This is done after

  178.     // getting the IOR out of the SendingContext service context.

  179.     // Our ORBs always send this, but it's optional in CORBA.

  180.     public abstract void setCodeBaseIOR(IOR codeBase);

  181. 

  182.     abstract IOR getCodeBaseIOR();

  183.     abstract CodeBase getCodeBase();

  184. 

  185.     public synchronized CodeSetComponentInfo.CodeSetContext getCodeSetContext() {

  186.         // Needs to be synchronized for the following case when the client

  187.         // doesn't send the code set context twice, and we have two threads

  188.         // in ServerDelegate processCodeSetContext.

  189.         //

  190.         // Thread A checks to see if there is a context, there is none, so

  191.         //     it calls setCodeSetContext, getting the synch lock.

  192.         // Thread B checks to see if there is a context.  If we didn't synch,

  193.         //     it might decide to outlaw wchar/wstring.

  194.         return codeSetContext;

  195.     }

  196. 

  197.     public synchronized void setCodeSetContext(CodeSetComponentInfo.CodeSetContext csc) {

  198.         // Check whether or not we should set this.  Technically,

  199.         // someone should only send a duplicate code set service context,

  200.         // but this makes sure we always use the first one we got.

  201.         if (codeSetContext == null) {

  202.             

  203.             if (OSFCodeSetRegistry.lookupEntry(csc.getCharCodeSet()) == null ||

  204.                 OSFCodeSetRegistry.lookupEntry(csc.getWCharCodeSet()) == null) {

  205.                 // If the client says it's negotiated a code set that

  206.                 // isn't a fallback and we never said we support, then

  207.                 // it has a bug.

  208.                 throw new DATA_CONVERSION(MinorCodes.BAD_CODESETS_FROM_CLIENT,

  209.                                           CompletionStatus.COMPLETED_NO);

  210.             }

  211. 

  212.             codeSetContext = csc;

  213.         }

  214.     }

  215. 

  216.     final boolean isServer()

  217.     {

  218.         return isServer;

  219.     }

  220. }