1 /*

     2  * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 

    26 package com.sun.xml.internal.messaging.saaj.soap;

    27 

    28 import java.awt.datatransfer.DataFlavor;

    29 import java.io.IOException;

    30 import java.io.OutputStream;

    31 import java.io.InputStream;

    32 

    33 import javax.activation.*;

    34 import javax.xml.transform.Source;

    35 

    36 import com.sun.xml.internal.messaging.saaj.util.FastInfosetReflection;

    37 

    38 /**

    39  * JAF data handler for Fast Infoset content

    40  *

    41  * @author Santiago Pericas-Geertsen

    42  */

    43 public class FastInfosetDataContentHandler implements DataContentHandler {

    44     public static final String STR_SRC = "com.sun.xml.internal.org.jvnet.fastinfoset.FastInfosetSource";

    45 

    46     public FastInfosetDataContentHandler() {

    47     }

    48 

    49     /**

    50      * return the DataFlavors for this <code>DataContentHandler</code>

    51      * @return The DataFlavors.

    52      */

    53     public DataFlavor[] getTransferDataFlavors() { // throws Exception;

    54         DataFlavor flavors[] = new DataFlavor[1];

    55         flavors[0] = new ActivationDataFlavor(

    56                 FastInfosetReflection.getFastInfosetSource_class(),

    57                 "application/fastinfoset", "Fast Infoset");

    58         return flavors;

    59     }

    60 

    61     /**

    62      * return the Transfer Data of type DataFlavor from InputStream

    63      * @param df The DataFlavor.

    64      * @param ins The InputStream corresponding to the data.

    65      * @return The constructed Object.

    66      */

    67     public Object getTransferData(DataFlavor flavor, DataSource dataSource)

    68         throws IOException

    69     {

    70         if (flavor.getMimeType().startsWith("application/fastinfoset")) {

    71             try {

    72                 if (flavor.getRepresentationClass().getName().equals(STR_SRC)) {

    73                     return FastInfosetReflection.FastInfosetSource_new(

    74                         dataSource.getInputStream());

    75                 }

    76             }

    77             catch (Exception e) {

    78                 throw new IOException(e.getMessage());

    79             }

    80         }

    81         return null;

    82     }

    83 

    84     public Object getContent(DataSource dataSource) throws IOException {

    85         try {

    86             return FastInfosetReflection.FastInfosetSource_new(

    87                 dataSource.getInputStream());

    88         }

    89         catch (Exception e) {

    90             throw new IOException(e.getMessage());

    91         }

    92     }

    93 

    94     /**

    95      * construct an object from a byte stream

    96      * (similar semantically to previous method, we are deciding

    97      *  which one to support)

    98      */

    99     public void writeTo(Object obj, String mimeType, OutputStream os)

   100         throws IOException

   101     {

   102         if (!mimeType.equals("application/fastinfoset")) {

   103             throw new IOException("Invalid content type \"" + mimeType

   104                 + "\" for FastInfosetDCH");

   105         }

   106 

   107         try {

   108             InputStream is = FastInfosetReflection.FastInfosetSource_getInputStream(

   109                 (Source) obj);

   110 

   111             int n; byte[] buffer = new byte[4096];

   112             while ((n = is.read(buffer)) != -1) {

   113                 os.write(buffer, 0, n);

   114             }

   115         }

   116         catch (Exception ex) {

   117             throw new IOException(

   118                 "Error copying FI source to output stream " + ex.getMessage());

   119         }

   120     }

   121 }