Package com.cloud.bridge.service

Source Code of com.cloud.bridge.service.S3RestServlet

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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 com.cloud.bridge.service;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.security.SignatureException;
import java.sql.SQLException;
import java.util.Enumeration;

import javax.inject.Inject;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.xml.bind.DatatypeConverter;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;

import org.apache.axis2.AxisFault;
import org.apache.log4j.Logger;
import org.springframework.web.context.support.SpringBeanAutowiringSupport;
import org.w3c.dom.Document;
import org.w3c.dom.NamedNodeMap;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

import com.cloud.bridge.io.MultiPartDimeInputStream;
import com.cloud.bridge.model.SAcl;
import com.cloud.bridge.model.UserCredentialsVO;
import com.cloud.bridge.persist.dao.CloudStackConfigurationDao;
import com.cloud.bridge.persist.dao.UserCredentialsDao;
import com.cloud.bridge.service.controller.s3.S3BucketAction;
import com.cloud.bridge.service.controller.s3.S3ObjectAction;
import com.cloud.bridge.service.controller.s3.ServiceProvider;
import com.cloud.bridge.service.controller.s3.ServletAction;
import com.cloud.bridge.service.core.s3.S3AccessControlList;
import com.cloud.bridge.service.core.s3.S3AuthParams;
import com.cloud.bridge.service.core.s3.S3Engine;
import com.cloud.bridge.service.core.s3.S3Grant;
import com.cloud.bridge.service.core.s3.S3MetaDataEntry;
import com.cloud.bridge.service.core.s3.S3PutObjectRequest;
import com.cloud.bridge.service.core.s3.S3PutObjectResponse;
import com.cloud.bridge.service.exception.InvalidBucketName;
import com.cloud.bridge.service.exception.PermissionDeniedException;
import com.cloud.bridge.util.ConfigurationHelper;
import com.cloud.bridge.util.HeaderParam;
import com.cloud.bridge.util.RestAuth;
import com.cloud.bridge.util.S3SoapAuth;
import com.cloud.utils.db.DB;
import com.cloud.utils.db.Transaction;
import com.cloud.utils.db.TransactionLegacy;
public class S3RestServlet extends HttpServlet {
    private static final long serialVersionUID = -6168996266762804877L;
    public static final String ENABLE_S3_API="enable.s3.api";
    private static boolean isS3APIEnabled = false;

    public static final Logger logger = Logger.getLogger(S3RestServlet.class);
    @Inject CloudStackConfigurationDao csDao;
    @Inject UserCredentialsDao ucDao;

    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) {
        processRequest( req, resp, "GET" );
    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp)
    {
        // -> DIME requests are authenticated via the SOAP auth mechanism
        String type = req.getHeader( "Content-Type" );
        if ( null != type && type.equalsIgnoreCase( "application/dime" ))  
            processDimeRequest(req, resp);
        else processRequest( req, resp, "POST" );
    }

    @Override
    protected void doPut(HttpServletRequest req, HttpServletResponse resp) {
        processRequest( req, resp, "PUT" );
    }

    @Override
    protected void doHead(HttpServletRequest req, HttpServletResponse resp) {
        processRequest( req, resp, "HEAD" );
    }

    @Override
    protected void doOptions(HttpServletRequest req, HttpServletResponse resp) {
        processRequest( req, resp, "OPTIONS" );
    }

    @Override
    protected void doDelete( HttpServletRequest req, HttpServletResponse resp ) {
        processRequest( req, resp, "DELETE" );
    }

    @Override
    public void init( ServletConfig config ) throws ServletException {
        try{
          SpringBeanAutowiringSupport.processInjectionBasedOnServletContext(this, config.getServletContext());        
        
            ConfigurationHelper.preConfigureConfigPathFromServletContext(config.getServletContext());
            // check if API is enabled
            String value = csDao.getConfigValue(ENABLE_S3_API);
            if(value != null) {
                isS3APIEnabled = Boolean.valueOf(value);
            }
            logger.info("S3Engine :: Configuration value is : " + value);

        }catch(Exception e){
            throw new ServletException("Error initializing awsapi: " + e.getMessage());
        }

    }



    /**
     * POST requests do not get authenticated on entry.   The associated
     * access key and signature headers are embedded in the message not encoded
     * as HTTP headers.
     */
    private void processRequest( HttpServletRequest request, HttpServletResponse response, String method )
    {
        TransactionLegacy txn = TransactionLegacy.open("cloudbridge", TransactionLegacy.AWSAPI_DB, true);
        try {
            logRequest(request);

            // Our extensions to the S3 REST API for simple management actions
            // are conveyed with Request parameter "Action".
            // The present extensions are either to set up the user credentials
            // (see the cloud-bridge-register script for more detail) or
            // to report our version of this capability.
            // -> unauthenticated calls, should still be done over HTTPS
            String cloudAction = request.getParameter( "Action" );

            if(!isS3APIEnabled){
                throw new RuntimeException("Amazon S3 API is disabled.");
            }


            if (null != cloudAction)
            {
                if (cloudAction.equalsIgnoreCase( "SetUserKeys" )) {
                    setUserKeys(request, response);
                    return;
                }

                if (cloudAction.equalsIgnoreCase( "SetCertificate" ))
                    // At present a noop
                    return;

                if (cloudAction.equalsIgnoreCase( "CloudS3Version" )) {
                    cloudS3Version(request, response);
                    return;
                }
            }


            txn.start();
            // -> authenticated calls
            if ( !((method.equalsIgnoreCase( "POST" ) && !(request.getQueryString().equalsIgnoreCase("delete"))) ) ){
                S3AuthParams params = extractRequestHeaders( request );
                authenticateRequest( request, params );
            }

            ServletAction action = routeRequest(request);
            if ( action != null ) {
                action.execute(request, response);
            }
            else {
                response.setStatus(404);
                endResponse(response, "File not found");
            }
            txn.close();
        }
        catch( InvalidBucketName e) {
            logger.error("Unexpected exception " + e.getMessage(), e);
            response.setStatus(400);
            endResponse(response, "Invalid Bucket Name - " + e.toString());     
        }
        catch(PermissionDeniedException e) {
            logger.error("Unexpected exception " + e.getMessage(), e);
            response.setStatus(403);
            endResponse(response, "Access denied - " + e.toString());
        }
        catch(Throwable e) {
            logger.error("Unexpected exception " + e.getMessage(), e);
            response.setStatus(404);
            endResponse(response, "Bad request");

        } finally {

            try {
                response.flushBuffer();
            } catch (IOException e) {
                logger.error("Unexpected exception " + e.getMessage(), e);
            }
        }
    }

    /**
     * Provide an easy way to determine the version of the implementation running.
     *
     * This is an unauthenticated REST call.
     */
    private void cloudS3Version( HttpServletRequest request, HttpServletResponse response ) {
        String version = new String( "<?xml version=\"1.0\" encoding=\"utf-8\"?><CloudS3Version>1.04</CloudS3Version>" );          
        response.setStatus(200);
        endResponse(response, version);
    }

    /**
     * This request registers the user Cloud.com account holder to the S3 service.   The Cloud.com
     * account holder saves his API access and secret keys with the S3 service so that
     * each rest call he makes can be verified was originated from him.   The given API access
     * and secret key are saved into the "usercredentials" database table.  
     *
     * This is an unauthenticated REST call.   The only required parameters are 'accesskey' and
     * 'secretkey'.
     *
     * To verify that the given keys represent an existing account they are used to execute the
     * Cloud.com's listAccounts API function.   If the keys do not represent a valid account the
     * listAccounts function will fail.
     *
     * A user can call this REST function any number of times, on each call the Cloud.com secret
     * key is simply over writes any previously stored value.
     *
     * As with all REST calls HTTPS should be used to ensure their security.
     */
    @DB
    private void setUserKeys( HttpServletRequest request, HttpServletResponse response ) {
        String[] accessKey = null;
        String[] secretKey = null;

        try {
            // -> both these parameters are required
            accessKey = request.getParameterValues( "accesskey" );
            if ( null == accessKey || 0 == accessKey.length ) {
                response.sendError(530, "Missing accesskey parameter" );
                return;
            }

            secretKey = request.getParameterValues( "secretkey" );
            if ( null == secretKey || 0 == secretKey.length ) {
                response.sendError(530, "Missing secretkey parameter" );
                return;
            }
        } catch( Exception e ) {
            logger.error("SetUserKeys exception " + e.getMessage(), e);
            response.setStatus(500);
            endResponse(response, "SetUserKeys exception " + e.getMessage());
            return;
        }

        try {
            // -> use the keys to see if the account actually exists
            //ServiceProvider.getInstance().getEC2Engine().validateAccount( accessKey[0], secretKey[0] );
            //UserCredentialsDaoImpl credentialDao = new UserCredentialsDao();
            TransactionLegacy txn = TransactionLegacy.open(TransactionLegacy.AWSAPI_DB);
            txn.start();
            UserCredentialsVO user = new UserCredentialsVO(accessKey[0], secretKey[0]);
            user = ucDao.persist(user);
            txn.commit();
            txn.close();
            //credentialDao.setUserKeys( accessKey[0], secretKey[0] );

        } catch( Exception e ) {
            logger.error("SetUserKeys " + e.getMessage(), e);
            response.setStatus(401);
            endResponse(response, e.toString());
            return;
        }
        response.setStatus(200)
        endResponse(response, "User keys set successfully");
    }

    /**
     * We are using the S3AuthParams class to hide where the header values are coming
     * from so that the authenticateRequest call can be made from several places.
     */
    public static S3AuthParams extractRequestHeaders( HttpServletRequest request ) {
        S3AuthParams params = new S3AuthParams();

        Enumeration headers = request.getHeaderNames();
        if (null != headers)
        {
            while( headers.hasMoreElements())
            {
                HeaderParam oneHeader = new HeaderParam();
                String headerName = (String)headers.nextElement();
                oneHeader.setName( headerName );
                oneHeader.setValue( request.getHeader( headerName ));
                params.addHeader( oneHeader );
            }
        }
        return params;
    }

    public static void authenticateRequest( HttpServletRequest request, S3AuthParams params )
            throws InstantiationException, IllegalAccessException, ClassNotFoundException, SQLException
            {
        RestAuth auth          = new RestAuth(ServiceProvider.getInstance().getUseSubDomain());  
        String   AWSAccessKey  = null;
        String   signature     = null;
        String   authorization = null;

        // [A] Is it an annonymous request?
        if (null == (authorization = params.getHeader( "Authorization" ))) {
            UserContext.current().initContext();
            return;
        }

        // [B] Is it an authenticated request?
        int offset = authorization.indexOf( "AWS" );
        if (-1 != offset) {
            String temp = authorization.substring( offset+3 ).trim();
            offset = temp.indexOf( ":" );
            AWSAccessKey = temp.substring( 0, offset );
            signature    = temp.substring( offset+1 );
        }

        // [C] Calculate the signature from the request's headers
        auth.setDateHeader( request.getHeader( "Date" ));
        auth.setContentTypeHeader( request.getHeader( "Content-Type" ));
        auth.setContentMD5Header( request.getHeader( "Content-MD5" ));
        auth.setHostHeader( request.getHeader( "Host" ));
        auth.setQueryString( request.getQueryString());
        auth.addUriPath( request.getRequestURI());

        // -> are their any Amazon specific (i.e. 'x-amz-' ) headers?
        HeaderParam[] headers = params.getHeaders();
        for( int i=0; null != headers && i < headers.length; i++ )
        {
            String headerName = headers[i].getName();
            String ignoreCase = headerName.toLowerCase();
            if (ignoreCase.startsWith( "x-amz-" ))
                auth.addAmazonHeader( headerName + ":" + headers[i].getValue());
        }

        UserInfo info = ServiceProvider.getInstance().getUserInfo(AWSAccessKey);
        if (info == null) throw new PermissionDeniedException("Unable to authenticate access key: " + AWSAccessKey);

        try {
            if (auth.verifySignature( request.getMethod(), info.getSecretKey(), signature )) {
                UserContext.current().initContext(AWSAccessKey, info.getSecretKey(), AWSAccessKey, info.getDescription(), request);
                return;
            }


        } catch (SignatureException e) {
            throw new PermissionDeniedException(e);

        } catch (UnsupportedEncodingException e) {
            throw new PermissionDeniedException(e);
        }
        throw new PermissionDeniedException("Invalid signature");
            }



    private ServletAction routeRequest(HttpServletRequest request)
    {
        //  URL routing for S3 REST calls.
        String pathInfo = request.getPathInfo();
        String bucketName = null;
        String key = null;

        String serviceEndpoint = ServiceProvider.getInstance().getServiceEndpoint();
        String host            = request.getHeader("Host");

        // Check for unrecognized forms of URI information in request

        if ( ( pathInfo == null ) || ( pathInfo.indexOf('/') != 0 ) )
            if ( "POST".equalsIgnoreCase(request.getMethod()) )
                // Case where request is POST operation with no pathinfo
                // This is the POST alternative to PUT described at s3.amazonaws.com API doc page 141
            {
                return routePlainPostRequest (request);        
            }


        // Irrespective of whether the requester is using subdomain or full host naming of path expressions
        // to buckets, wherever the request is made up of a service endpoint followed by a /, in AWS S3 this always
        // conveys a ListAllMyBuckets command

        if  ( (serviceEndpoint.equalsIgnoreCase( host )) && (pathInfo.equalsIgnoreCase("/")) ) {
            request.setAttribute(S3Constants.BUCKET_ATTR_KEY, "/");
            return new S3BucketAction();   // for ListAllMyBuckets
        }

        // Because there is a leading / at position 0 of pathInfo, now subtract this to process the remainder 
        pathInfo = pathInfo.substring(1);

        if (ServiceProvider.getInstance().getUseSubDomain())

        {      
            // -> verify the format of the bucket name
            int endPos = host.indexOf( ServiceProvider.getInstance().getMasterDomain());
            if ( endPos > 0 )
            {
                bucketName = host.substring(0, endPos);
                S3Engine.verifyBucketName( bucketName, false );
                request.setAttribute(S3Constants.BUCKET_ATTR_KEY, bucketName);
            }
            else request.setAttribute(S3Constants.BUCKET_ATTR_KEY, "");

            if (pathInfo == null || pathInfo.equalsIgnoreCase("/"))
            {
                return new S3BucketAction();
            }
            else {
                String objectKey = pathInfo.substring(1);
                request.setAttribute(S3Constants.OBJECT_ATTR_KEY, objectKey);
                return new S3ObjectAction();
            }
        }

        else

        {

            int endPos = pathInfo.indexOf('/')// Subsequent / character?

            if (endPos < 1)
            {
                bucketName = pathInfo;
                S3Engine.verifyBucketName( bucketName, false );
                request.setAttribute(S3Constants.BUCKET_ATTR_KEY, bucketName);
                return new S3BucketAction();
            }
            else
            {
                bucketName = pathInfo.substring(0, endPos);
                key        = pathInfo.substring(endPos + 1);     
                S3Engine.verifyBucketName( bucketName, false );

                if (!key.isEmpty())
                {
                    request.setAttribute(S3Constants.BUCKET_ATTR_KEY, bucketName);
                    request.setAttribute(S3Constants.OBJECT_ATTR_KEY, pathInfo.substring(endPos + 1));
                    return new S3ObjectAction();
                }
                else {
                    request.setAttribute(S3Constants.BUCKET_ATTR_KEY, bucketName);
                    return new S3BucketAction();
                }
            }
        }
    }


    public static void endResponse(HttpServletResponse response, String content) {
        try {
            byte[] data = content.getBytes();
            response.setContentLength(data.length);
            OutputStream os = response.getOutputStream();
            os.write(data);
            os.close();
        } catch(Throwable e) {
            logger.error("Unexpected exception " + e.getMessage(), e);
        }
    }

    public static void writeResponse(HttpServletResponse response, String content) throws IOException {
        byte[] data = content.getBytes();
        OutputStream os = response.getOutputStream();
        os.write(data);
    }

    public static void writeResponse(HttpServletResponse response, InputStream is) throws IOException {
        byte[] data = new byte[4096];
        int length = 0;
        while((length = is.read(data)) > 0) {
            response.getOutputStream().write(data, 0, length);
        }
    }

    // Route for the case where request is POST operation with no pathinfo
    // This is the POST alternative to PUT described at s3.amazonaws.com API doc, Amazon Simple
    // Storage Service API Reference API Version 2006-03-01 page 141.
    // The purpose of the plain POST operation is to add an object to a specified bucket using HTML forms.

    private S3ObjectAction routePlainPostRequest (HttpServletRequest request)   
    { 
        // TODO - Remove the unnecessary fields below   
        // Obtain the mandatory fields from the HTML form or otherwise fail with a logger message
        String keyString = request.getParameter("key");
        String metatagString = request.getParameter("x-amz-meta-tag");
        String bucketString = request.getParameter("Bucket");
        String aclString = request.getParameter("acl");
        String fileString = request.getParameter("file");

        String accessKeyString = request.getParameter("AWSAccessKeyId");
        String signatureString = request.getParameter("Signature");

        // Obtain the discretionary fields from the HTML form
        String policyKeyString = request.getParameter("Policy");
        String metauuidString = request.getParameter("x-amz-meta-uuid");
        String redirectString = request.getParameter("redirect")

        // if none of the above are null then ...
        request.setAttribute(S3Constants.BUCKET_ATTR_KEY, bucketString);
        request.setAttribute(S3Constants.OBJECT_ATTR_KEY, keyString);
        request.setAttribute(S3Constants.PLAIN_POST_ACCESS_KEY, accessKeyString);
        request.setAttribute(S3Constants.PLAIN_POST_SIGNATURE, signatureString);

        // -> authenticated calls
        try {
            // S3AuthParams params = extractRequestHeaders( request );
            S3AuthParams params = new S3AuthParams();
            HeaderParam headerParam1 = new HeaderParam("accessKey", accessKeyString);
            params.addHeader(headerParam1);
            HeaderParam headerParam2 = new HeaderParam("secretKey", signatureString);
            params.addHeader(headerParam2);
            authenticateRequest( request, params );
        }
        catch (Exception e)
        { logger.warn("Authentication details insufficient"); }

        return new S3ObjectAction();

    }

    /**
     * A DIME request is really a SOAP request that we are dealing with, and so its
     * authentication is the SOAP authentication approach.   Since Axis2 does not handle
     * DIME messages we deal with them here.
     *
     * @param request
     * @param response
     */
    private void processDimeRequest(HttpServletRequest request, HttpServletResponse response) {
        S3PutObjectRequest  putRequest  = null;
        S3PutObjectResponse putResponse = null;
        int                 bytesRead   = 0;

        S3Engine engine = new S3Engine();

        try {  
            logRequest(request);

            MultiPartDimeInputStream ds = new MultiPartDimeInputStream( request.getInputStream());

            // -> the first stream MUST be the SOAP party
            if (ds.nextInputStream())
            {
                //logger.debug( "DIME msg [" + ds.getStreamType() + "," + ds.getStreamTypeFormat() + "," + ds.getStreamId() + "]" );
                byte[] buffer = new byte[8192];
                bytesRead = ds.read( buffer, 0, 8192 );
                //logger.debug( "DIME SOAP Bytes read: " + bytesRead );
                ByteArrayInputStream bis = new ByteArrayInputStream( buffer, 0, bytesRead );
                putRequest = toEnginePutObjectRequest( bis );
            }

            // -> we only need to support a DIME message with two bodyparts
            if (null != putRequest && ds.nextInputStream())
            {
                InputStream is = ds.getInputStream();
                putRequest.setData( is );
            }

            // -> need to do SOAP level auth here, on failure return the SOAP fault
            StringBuffer xml = new StringBuffer();
            String AWSAccessKey = putRequest.getAccessKey();
            UserInfo info = ServiceProvider.getInstance().getUserInfo(AWSAccessKey);
            try
            {   S3SoapAuth.verifySignature( putRequest.getSignature(), "PutObject", putRequest.getRawTimestamp(), AWSAccessKey, info.getSecretKey());    

            } catch( AxisFault e ) {
                String reason = e.toString();
                int start = reason.indexOf( ".AxisFault:" );
                if (-1 != start) reason = reason.substring( start+11 );

                xml.append( "<?xml version=\"1.0\" encoding=\"utf-8\"?>" );
                xml.append( "<soap:Envelope xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\" >\n" );
                xml.append( "<soap:Body>\n" );              
                xml.append( "<soap:Fault>\n" );
                xml.append( "<faultcode>" ).append( e.getFaultCode().toString()).append( "</faultcode>\n" );
                xml.append( "<faultstring>" ).append( reason ).append( "</faultstring>\n" );
                xml.append( "</soap:Fault>\n" );
                xml.append( "</soap:Body></soap:Envelope>" );

                endResponse(response, xml.toString());
                return;
            }

            // -> PutObject S3 Bucket Policy would be done in the engine.handleRequest() call
            UserContext.current().initContext( AWSAccessKey, info.getSecretKey(), AWSAccessKey, "S3 DIME request", request );
            putResponse = engine.handleRequest( putRequest );

            xml.append( "<?xml version=\"1.0\" encoding=\"utf-8\"?>" );
            xml.append( "<soap:Envelope xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\" xmlns:tns=\"http://s3.amazonaws.com/doc/2006-03-01/\">" );
            xml.append( "<soap:Body>" );
            xml.append( "<tns:PutObjectResponse>" );
            xml.append( "<tns:PutObjectResponse>" );
            xml.append( "<tns:ETag>\"").append( putResponse.getETag()).append( "\"</tns:ETag>" );
            xml.append( "<tns:LastModified>").append( DatatypeConverter.printDateTime(putResponse.getLastModified())).append( "</tns:LastModified>" );
            xml.append( "</tns:PutObjectResponse></tns:PutObjectResponse>" );
            xml.append( "</soap:Body></soap:Envelope>" );

            endResponse(response, xml.toString());
        }
        catch(PermissionDeniedException e) {
            logger.error("Unexpected exception " + e.getMessage(), e);
            response.setStatus(403);
            endResponse(response, "Access denied");    
        }
        catch(Throwable e)
        {
            logger.error("Unexpected exception " + e.getMessage(), e);
        }
        finally
        {
        }
    }


    /**
     * Convert the SOAP XML we extract from the DIME message into our local object.
     * Here Axis2 is not parsing the SOAP for us.   I tried to use the Amazon PutObject
     * parser but it keep throwing exceptions.
     *
     * @param putObjectInline
     * @return
     * @throws Exception
     */
    public static S3PutObjectRequest toEnginePutObjectRequest( InputStream is ) throws Exception
    {
        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
        dbf.setNamespaceAware( true );

        DocumentBuilder db       = dbf.newDocumentBuilder();
        Document        doc      = db.parse( is );
        Node            parent   = null;
        Node            contents = null;
        NodeList        children = null;
        String          temp     = null;
        String          element  = null;
        int             count    = 0;

        S3PutObjectRequest request = new S3PutObjectRequest();

        // [A] Pull out the simple nodes first
        NodeList part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "Bucket" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
                request.setBucketName( contents.getFirstChild().getNodeValue());
       
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "Key" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
                request.setKey( contents.getFirstChild().getNodeValue());
        }   
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "ContentLength" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
            {
                String length = contents.getFirstChild().getNodeValue();
                if (null != length) request.setContentLength( Long.decode( length ));
            }
        }   
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "AWSAccessKeyId" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
                request.setAccessKey( contents.getFirstChild().getNodeValue());
        }   
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "Signature" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
                request.setSignature( contents.getFirstChild().getNodeValue());
        }   
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "Timestamp" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
                request.setRawTimestamp( contents.getFirstChild().getNodeValue());
       
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "StorageClass" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
                request.setStorageClass( contents.getFirstChild().getNodeValue());
        }   
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "Credential" );
        if (null != part)
        {
            if (null != (contents = part.item( 0 )))
                request.setCredential( contents.getFirstChild().getNodeValue());
        }


        // [B] Get a list of all 'Metadata' elements
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "Metadata" );
        if (null != part)
        {
            count = part.getLength();
            S3MetaDataEntry[] metaEntry = new S3MetaDataEntry[ count ];

            for( int i=0; i < count; i++ )
            {
                parent = part.item(i);
                metaEntry[i] = new S3MetaDataEntry();

                // -> get a list of all the children elements of the 'Metadata' parent element
                if (null != (children = parent.getChildNodes()))
                {
                    int numChildren = children.getLength();
                    for( int j=0; j < numChildren; j++ )
                    {
                        contents = children.item( j );
                        element = contents.getNodeName().trim();
                        if ( element.endsWith( "Name" ))
                        {
                            temp = contents.getFirstChild().getNodeValue();
                            if (null != temp) metaEntry[i].setName( temp );
                        }
                        else if (element.endsWith( "Value" ))
                        {
                            temp = contents.getFirstChild().getNodeValue();
                            if (null != temp) metaEntry[i].setValue( temp );
                        }
                    }
                }
            }
            request.setMetaEntries( metaEntry );
        }

        // [C] Get a list of all Grant elements in an AccessControlList
        part = getElement( doc, "http://s3.amazonaws.com/doc/2006-03-01/", "Grant" );
        if (null != part)
        {
            S3AccessControlList engineAcl = new S3AccessControlList();

            count = part.getLength();
            for( int i=0; i < count; i++ )
            {
                parent = part.item(i);
                S3Grant engineGrant = new S3Grant();

                // -> get a list of all the children elements of the 'Grant' parent element
                if (null != (children = parent.getChildNodes()))
                {
                    int numChildren = children.getLength();
                    for( int j=0; j < numChildren; j++ )
                    {
                        contents = children.item( j );
                        element  = contents.getNodeName().trim();
                        if ( element.endsWith( "Grantee" ))
                        {
                            NamedNodeMap attbs = contents.getAttributes();
                            if (null != attbs)
                            {
                                Node type = attbs.getNamedItemNS( "http://www.w3.org/2001/XMLSchema-instance", "type" );
                                if ( null != type )
                                    temp = type.getFirstChild().getNodeValue().trim();
                                else temp = null;

                                if ( null != temp && temp.equalsIgnoreCase( "CanonicalUser" ))
                                {
                                    engineGrant.setGrantee(SAcl.GRANTEE_USER);
                                    engineGrant.setCanonicalUserID( getChildNodeValue( contents, "ID" ));
                                }
                                else throw new UnsupportedOperationException( "Missing http://www.w3.org/2001/XMLSchema-instance:type value" );
                            }
                        }
                        else if (element.endsWith( "Permission" ))
                        {
                            temp = contents.getFirstChild().getNodeValue().trim();
                            if (temp.equalsIgnoreCase("READ"        )) engineGrant.setPermission(SAcl.PERMISSION_READ);
                            else if (temp.equalsIgnoreCase("WRITE"       )) engineGrant.setPermission(SAcl.PERMISSION_WRITE);
                            else if (temp.equalsIgnoreCase("READ_ACP"    )) engineGrant.setPermission(SAcl.PERMISSION_READ_ACL);
                            else if (temp.equalsIgnoreCase("WRITE_ACP"   )) engineGrant.setPermission(SAcl.PERMISSION_WRITE_ACL);
                            else if (temp.equalsIgnoreCase("FULL_CONTROL")) engineGrant.setPermission(SAcl.PERMISSION_FULL);
                            else throw new UnsupportedOperationException( "Unsupported permission: " + temp );
                        }
                    }
                    engineAcl.addGrant( engineGrant );
                }
            }
            request.setAcl( engineAcl );
        }
        return request;
    }

    /**
     * Have to deal with XML with and without namespaces.
     */
    public static NodeList getElement( Document doc, String namespace, String tagName )
    {
        NodeList part = doc.getElementsByTagNameNS( namespace, tagName );
        if (null == part || 0 == part.getLength()) part = doc.getElementsByTagName( tagName );

        return part;
    }

    /**
     * Looking for the value of a specific child of the given parent node.
     *
     * @param parent
     * @param childName
     * @return
     */
    private static String getChildNodeValue( Node parent, String childName )
    {
        NodeList children = null;
        Node     element  = null;

        if (null != (children = parent.getChildNodes()))
        {
            int numChildren = children.getLength();
            for( int i=0; i < numChildren; i++ )
            {
                if (null != (element = children.item( i )))
                {
                    // -> name may have a namespace on it
                    String name = element.getNodeName().trim();
                    if ( name.endsWith( childName ))
                    {
                        String value = element.getFirstChild().getNodeValue();
                        if (null != value) value = value.trim();
                        return value;
                    }
                }
            }
        }
        return null;
    }

    private void logRequest(HttpServletRequest request) {
        if(logger.isInfoEnabled()) {
            logger.info("Request method: " + request.getMethod());
            logger.info("Request contextPath: " + request.getContextPath());
            logger.info("Request pathInfo: " + request.getPathInfo());
            logger.info("Request pathTranslated: " + request.getPathTranslated());
            logger.info("Request queryString: " + request.getQueryString());
            logger.info("Request requestURI: " + request.getRequestURI());
            logger.info("Request requestURL: " + request.getRequestURL());
            logger.info("Request servletPath: " + request.getServletPath());
            Enumeration headers = request.getHeaderNames();
            if(headers != null) {
                while(headers.hasMoreElements()) {
                    Object headerName = headers.nextElement();
                    logger.info("Request header " + headerName + ":" + request.getHeader((String)headerName));
                }
            }

            Enumeration params = request.getParameterNames();
            if(params != null) {
                while(params.hasMoreElements()) {
                    Object paramName = params.nextElement();
                    logger.info("Request parameter " + paramName + ":" +
                            request.getParameter((String)paramName));
                }
            }
            logger.info( "- End of request -" );
        }
    }
}
TOP

Related Classes of com.cloud.bridge.service.S3RestServlet

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.