connectors/naming/src/org/apache/naming/handler/jndi/DirContextURLConnection.java - tomcat55 - Git at Google

 /*
  *  Copyright 1999-2004 The Apache Software Foundation
  *
  *  Licensed under the Apache License, Version 2.0 (the "License");
  *  you may not use this file except in compliance with the License.
  *  You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  *  Unless required by applicable law or agreed to in writing, software
  *  distributed under the License is distributed on an "AS IS" BASIS,
  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  *  See the License for the specific language governing permissions and
  *  limitations under the License.
  */

 package org.apache.naming.handler.jndi;

 import java.io.ByteArrayInputStream;
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URL;
 import java.net.URLConnection;
 import java.security.Permission;
 import java.util.Enumeration;
 import java.util.Vector;

 import javax.naming.NameClassPair;
 import javax.naming.NamingEnumeration;
 import javax.naming.NamingException;
 import javax.naming.directory.Attribute;
 import javax.naming.directory.Attributes;
 import javax.naming.directory.DirContext;

 import org.apache.naming.core.JndiPermission;
 import org.apache.naming.util.AttributeHelper;
 // import org.apache.naming.resources.Resource;
 // import org.apache.naming.resources.ResourceAttributes;

 /**
  * Connection to a JNDI directory context.
  * <p/>
  * Note: All the object attribute names are the WebDAV names, not the HTTP
  * names, so this class overrides some methods from URLConnection to do the
  * queries using the right names. Content handler is also not used; the
  * content is directly returned.
  *
  * @author <a href="mailto:remm@apache.org">Remy Maucherat</a>
  * @author Costin Manolache
  */
 public class DirContextURLConnection
     extends URLConnection
 {


     // ----------------------------------------------------------- Constructors

     /**
      * @param context The base context for the dynamic resources.
      *        For regular webapps, it should be a thread-bound context, probably
      *        a branch under java:
      *        For top-level apps, it can be either the InitialContext or a branch
      *        that is used for all content.
      *
      * The choice of the context affects the base of the URLs.
      */
     public DirContextURLConnection(DirContext context, URL url) {
         super(url);
         if (context == null)
             throw new IllegalArgumentException
                 ("Directory context can't be null");
         if (System.getSecurityManager() != null) {
             this.permission = new JndiPermission(url.toString());
 	}
         this.context = context;
     }


     // ----------------------------------------------------- Instance Variables


     /**
      * Directory context.
      */
     protected DirContext context;


     /**
      * Associated resource.
      */
     protected Object resource;


     /**
      * Associated DirContext.
      */
     protected DirContext collection;


     /**
      * Other unknown object.
      */
     protected Object object;


     /**
      * Attributes.
      */
     protected Attributes attributes;


     /**
      * Date.
      */
     protected long date;


     /**
      * Permission
      */
     protected Permission permission;


     // ------------------------------------------------------------- Properties


     /**
      * Connect to the DirContext, and retrive the bound object, as well as
      * its attributes. If no object is bound with the name specified in the
      * URL, then an IOException is thrown.
      *
      * @throws IOException Object not found
      */
     public void connect()
         throws IOException {

         if (!connected) {

             try {
                 // TODO: What is this ???  (costin)
                 date = System.currentTimeMillis();
                 String path = getURL().getFile();

                 /* This deals with a strange case, where the
                    name is prefixed by hostname and contextname.

                    A webapp should never use this - all resources
                    are local ( or a different mean should be used to
                    locate external res ).

                    The top-level handler must use the hostname + context to
                    locate files in a particular context.

                   if (context instanceof ProxyDirContext) {
                     ProxyDirContext proxyDirContext =
                         (ProxyDirContext) context;
                     String hostName = proxyDirContext.getHostName();
                     String contextName = proxyDirContext.getContextName();
                     if (hostName != null) {
                         if (!path.startsWith("/" + hostName + "/"))
                             return;
                         path = path.substring(hostName.length()+ 1);
                     }
                     if (contextName != null) {
                         if (!path.startsWith(contextName + "/")) {
                             return;
                         } else {
                             path = path.substring(contextName.length());
                         }
                     }
                 }
                 */
                 object = context.lookup(path);
                 attributes = context.getAttributes(path);
                 //                if (object instanceof Resource)
                 //  resource = (Resource) object;
                 if (object instanceof DirContext)
                     collection = (DirContext) object;
                 else
                     resource=object;
             } catch (NamingException e) {
                 // Object not found
             }

             connected = true;

         }

     }


     /**
      * Return the content length value.
      */
     public int getContentLength() {
         if (!connected) {
             // Try to connect (silently)
             try {
                 connect();
             } catch (IOException e) {
             }
         }

         if (attributes == null)
             return (-1);

         return (int)AttributeHelper.getContentLength( attributes );
     }

     /**
      * Return the content type value.
      */
     public String getContentType() {
         if (!connected) {
             // Try to connect (silently)
             try {
                 connect();
             } catch (IOException e) {
             }
         }

         if (attributes == null)
             return null;

         return AttributeHelper.getContentType(attributes);
     }


     /**
      * Return the last modified date.
      * TODO: it's broken
      */
     public long getDate() {
         return date;
     }


     /**
      * Return the last modified date.
      */
     public long getLastModified() {

         if (!connected) {
             // Try to connect (silently)
             try {
                 connect();
             } catch (IOException e) {
             }
         }

         if (attributes == null)
             return 0;

         return AttributeHelper.getLastModified( attributes );
     }


     /**
      * Returns the name of the specified header field.
      */
     public String getHeaderField(String name) {

         if (!connected) {
             // Try to connect (silently)
             try {
                 connect();
             } catch (IOException e) {
             }
         }

         if (attributes == null)
             return (null);

         Attribute attribute = attributes.get(name);
         try {
             return attribute.get().toString();
         } catch (Exception e) {
             // Shouldn't happen, unless the attribute has no value
         }

         return (null);

     }


     /**
      * Get object content.
      */
     public Object getContent()
         throws IOException {

         if (!connected)
             connect();

         if (resource != null)
             return getInputStream();
         if (collection != null)
             return collection;
         if (object != null)
             return object;

         throw new FileNotFoundException();

     }


     /**
      * Get object content.
      */
     public Object getContent(Class[] classes)
         throws IOException {

         Object object = getContent();

         for (int i = 0; i < classes.length; i++) {
             if (classes[i].isInstance(object))
                 return object;
         }

         return null;

     }


     /**
      * Get input stream.
      */
     public InputStream getInputStream()
         throws IOException {

         if (!connected)
             connect();

         if (resource == null) {
             throw new FileNotFoundException();
         } else {
             // Reopen resource
             try {
                 resource =  context.lookup(getURL().getFile());
             } catch (NamingException e) {
             }
         }

         //        return (resource.streamContent());
         return getInputStream(resource);

     }

     /** Try to extract content from a resource found in the directory
      *  Code from Resource and ProxyContext.
      */
     public static InputStream getInputStream( Object resource ) {
         if( resource instanceof InputStream )
             return (InputStream) resource;

         // Found in: ProxyDirContext.lookup ( strange, only in one ).
         return new ByteArrayInputStream(resource.toString().getBytes());
     }

     /**
      * Get the Permission for this URL
      */
     public Permission getPermission() {

         return permission;
     }


     // --------------------------------------------------------- Public Methods


     /**
      * List children of this collection. The names given are relative to this
      * URI's path. The full uri of the children is then : path + "/" + name.
      */
     public Enumeration list()
         throws IOException {

         if (!connected) {
             connect();
         }

         if ((resource == null) && (collection == null)) {
             throw new FileNotFoundException();
         }

         Vector result = new Vector();

         if (collection != null) {
             try {
                 NamingEnumeration enum = context.list(getURL().getFile());
                 while (enum.hasMoreElements()) {
                     NameClassPair ncp = (NameClassPair) enum.nextElement();
                     result.addElement(ncp.getName());
                 }
             } catch (NamingException e) {
                 // Unexpected exception
                 throw new FileNotFoundException();
             }
         }

         return result.elements();

     }


 }
	/*
	* Copyright 1999-2004 The Apache Software Foundation
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	package org.apache.naming.handler.jndi;

	import java.io.ByteArrayInputStream;
	import java.io.FileNotFoundException;
	import java.io.IOException;
	import java.io.InputStream;
	import java.net.URL;
	import java.net.URLConnection;
	import java.security.Permission;
	import java.util.Enumeration;
	import java.util.Vector;

	import javax.naming.NameClassPair;
	import javax.naming.NamingEnumeration;
	import javax.naming.NamingException;
	import javax.naming.directory.Attribute;
	import javax.naming.directory.Attributes;
	import javax.naming.directory.DirContext;

	import org.apache.naming.core.JndiPermission;
	import org.apache.naming.util.AttributeHelper;
	// import org.apache.naming.resources.Resource;
	// import org.apache.naming.resources.ResourceAttributes;

	/**
	* Connection to a JNDI directory context.
	* <p/>
	* Note: All the object attribute names are the WebDAV names, not the HTTP
	* names, so this class overrides some methods from URLConnection to do the
	* queries using the right names. Content handler is also not used; the
	* content is directly returned.
	*
	* @author <a href="mailto:remm@apache.org">Remy Maucherat</a>
	* @author Costin Manolache
	*/
	public class DirContextURLConnection
	extends URLConnection
	{


	// ----------------------------------------------------------- Constructors

	/**
	* @param context The base context for the dynamic resources.
	* For regular webapps, it should be a thread-bound context, probably
	* a branch under java:
	* For top-level apps, it can be either the InitialContext or a branch
	* that is used for all content.
	*
	* The choice of the context affects the base of the URLs.
	*/
	public DirContextURLConnection(DirContext context, URL url) {
	super(url);
	if (context == null)
	throw new IllegalArgumentException
	("Directory context can't be null");
	if (System.getSecurityManager() != null) {
	this.permission = new JndiPermission(url.toString());
	}
	this.context = context;
	}


	// ----------------------------------------------------- Instance Variables


	/**
	* Directory context.
	*/
	protected DirContext context;


	/**
	* Associated resource.
	*/
	protected Object resource;


	/**
	* Associated DirContext.
	*/
	protected DirContext collection;


	/**
	* Other unknown object.
	*/
	protected Object object;


	/**
	* Attributes.
	*/
	protected Attributes attributes;


	/**
	* Date.
	*/
	protected long date;


	/**
	* Permission
	*/
	protected Permission permission;


	// ------------------------------------------------------------- Properties


	/**
	* Connect to the DirContext, and retrive the bound object, as well as
	* its attributes. If no object is bound with the name specified in the
	* URL, then an IOException is thrown.
	*
	* @throws IOException Object not found
	*/
	public void connect()
	throws IOException {

	if (!connected) {

	try {
	// TODO: What is this ??? (costin)
	date = System.currentTimeMillis();
	String path = getURL().getFile();

	/* This deals with a strange case, where the
	name is prefixed by hostname and contextname.

	A webapp should never use this - all resources
	are local ( or a different mean should be used to
	locate external res ).

	The top-level handler must use the hostname + context to
	locate files in a particular context.

	if (context instanceof ProxyDirContext) {
	ProxyDirContext proxyDirContext =
	(ProxyDirContext) context;
	String hostName = proxyDirContext.getHostName();
	String contextName = proxyDirContext.getContextName();
	if (hostName != null) {
	if (!path.startsWith("/" + hostName + "/"))
	return;
	path = path.substring(hostName.length()+ 1);
	}
	if (contextName != null) {
	if (!path.startsWith(contextName + "/")) {
	return;
	} else {
	path = path.substring(contextName.length());
	}
	}
	}
	*/
	object = context.lookup(path);
	attributes = context.getAttributes(path);
	// if (object instanceof Resource)
	// resource = (Resource) object;
	if (object instanceof DirContext)
	collection = (DirContext) object;
	else
	resource=object;
	} catch (NamingException e) {
	// Object not found
	}

	connected = true;

	}

	}


	/**
	* Return the content length value.
	*/
	public int getContentLength() {
	if (!connected) {
	// Try to connect (silently)
	try {
	connect();
	} catch (IOException e) {
	}
	}

	if (attributes == null)
	return (-1);

	return (int)AttributeHelper.getContentLength( attributes );
	}

	/**
	* Return the content type value.
	*/
	public String getContentType() {
	if (!connected) {
	// Try to connect (silently)
	try {
	connect();
	} catch (IOException e) {
	}
	}

	if (attributes == null)
	return null;

	return AttributeHelper.getContentType(attributes);
	}


	/**
	* Return the last modified date.
	* TODO: it's broken
	*/
	public long getDate() {
	return date;
	}


	/**
	* Return the last modified date.
	*/
	public long getLastModified() {

	if (!connected) {
	// Try to connect (silently)
	try {
	connect();
	} catch (IOException e) {
	}
	}

	if (attributes == null)
	return 0;

	return AttributeHelper.getLastModified( attributes );
	}


	/**
	* Returns the name of the specified header field.
	*/
	public String getHeaderField(String name) {

	if (!connected) {
	// Try to connect (silently)
	try {
	connect();
	} catch (IOException e) {
	}
	}

	if (attributes == null)
	return (null);

	Attribute attribute = attributes.get(name);
	try {
	return attribute.get().toString();
	} catch (Exception e) {
	// Shouldn't happen, unless the attribute has no value
	}

	return (null);

	}


	/**
	* Get object content.
	*/
	public Object getContent()
	throws IOException {

	if (!connected)
	connect();

	if (resource != null)
	return getInputStream();
	if (collection != null)
	return collection;
	if (object != null)
	return object;

	throw new FileNotFoundException();

	}


	/**
	* Get object content.
	*/
	public Object getContent(Class[] classes)
	throws IOException {

	Object object = getContent();

	for (int i = 0; i < classes.length; i++) {
	if (classes[i].isInstance(object))
	return object;
	}

	return null;

	}


	/**
	* Get input stream.
	*/
	public InputStream getInputStream()
	throws IOException {

	if (!connected)
	connect();

	if (resource == null) {
	throw new FileNotFoundException();
	} else {
	// Reopen resource
	try {
	resource = context.lookup(getURL().getFile());
	} catch (NamingException e) {
	}
	}

	// return (resource.streamContent());
	return getInputStream(resource);

	}

	/** Try to extract content from a resource found in the directory
	* Code from Resource and ProxyContext.
	*/
	public static InputStream getInputStream( Object resource ) {
	if( resource instanceof InputStream )
	return (InputStream) resource;

	// Found in: ProxyDirContext.lookup ( strange, only in one ).
	return new ByteArrayInputStream(resource.toString().getBytes());
	}

	/**
	* Get the Permission for this URL
	*/
	public Permission getPermission() {

	return permission;
	}


	// --------------------------------------------------------- Public Methods


	/**
	* List children of this collection. The names given are relative to this
	* URI's path. The full uri of the children is then : path + "/" + name.
	*/
	public Enumeration list()
	throws IOException {

	if (!connected) {
	connect();
	}

	if ((resource == null) && (collection == null)) {
	throw new FileNotFoundException();
	}

	Vector result = new Vector();

	if (collection != null) {
	try {
	NamingEnumeration enum = context.list(getURL().getFile());
	while (enum.hasMoreElements()) {
	NameClassPair ncp = (NameClassPair) enum.nextElement();
	result.addElement(ncp.getName());
	}
	} catch (NamingException e) {
	// Unexpected exception
	throw new FileNotFoundException();
	}
	}

	return result.elements();

	}


	}