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.*;
  30 
  31 import javax.activation.*;
  32 import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.MimeUtility;
  33 import com.sun.xml.internal.messaging.saaj.packaging.mime.util.ASCIIUtility;
  34 import com.sun.xml.internal.messaging.saaj.packaging.mime.internet.ContentType;
  35 
  36 /**
  37  * JAF data content handler for text/plain --> String
  38  *
  39  */
  40 public class StringDataContentHandler implements DataContentHandler {
  41     private static ActivationDataFlavor myDF = new ActivationDataFlavor(
  42         java.lang.String.class,
  43         "text/plain",
  44         "Text String");
  45 
  46     protected ActivationDataFlavor getDF() {
  47         return myDF;
  48     }
  49 
  50     /**
  51      * Return the DataFlavors for this <code>DataContentHandler</code>.
  52      *
  53      * @return The DataFlavors
  54      */
  55     public DataFlavor[] getTransferDataFlavors() {
  56         return new DataFlavor[] { getDF() };
  57     }
  58 
  59     /**
  60      * Return the Transfer Data of type DataFlavor from InputStream.
  61      *
  62      * @param df The DataFlavor
  63      * @param ds The DataSource corresponding to the data
  64      * @return String object
  65      */
  66     public Object getTransferData(DataFlavor df, DataSource ds)
  67                         throws IOException {
  68         // use myDF.equals to be sure to get ActivationDataFlavor.equals,
  69         // which properly ignores Content-Type parameters in comparison
  70         if (getDF().equals(df))
  71             return getContent(ds);
  72         else
  73             return null;
  74     }
  75 
  76     public Object getContent(DataSource ds) throws IOException {
  77         String enc = null;
  78         InputStreamReader is = null;
  79 
  80         try {
  81             enc = getCharset(ds.getContentType());
  82             is = new InputStreamReader(ds.getInputStream(), enc);
  83         } catch (IllegalArgumentException iex) {
  84             /*
  85              * An unknown charset of the form ISO-XXX-XXX will cause
  86              * the JDK to throw an IllegalArgumentException.  The
  87              * JDK will attempt to create a classname using this string,
  88              * but valid classnames must not contain the character '-',
  89              * and this results in an IllegalArgumentException, rather than
  90              * the expected UnsupportedEncodingException.  Yikes.
  91              */
  92             throw new UnsupportedEncodingException(enc);
  93         }
  94 
  95         try {
  96             int pos = 0;
  97             int count;
  98             char buf[] = new char[1024];
  99 
 100             while ((count = is.read(buf, pos, buf.length - pos)) != -1) {
 101                 pos += count;
 102                 if (pos >= buf.length) {
 103                     int size = buf.length;
 104                     if (size < 256*1024)
 105                         size += size;
 106                     else
 107                         size += 256*1024;
 108                     char tbuf[] = new char[size];
 109                     System.arraycopy(buf, 0, tbuf, 0, pos);
 110                     buf = tbuf;
 111                 }
 112             }
 113             return new String(buf, 0, pos);
 114         } finally {
 115             try {
 116                 is.close();
 117             } catch (IOException ex) { }
 118         }
 119     }
 120 
 121     /**
 122      * Write the object to the output stream, using the specified MIME type.
 123      */
 124     public void writeTo(Object obj, String type, OutputStream os)
 125                         throws IOException {
 126         if (!(obj instanceof String))
 127             throw new IOException("\"" + getDF().getMimeType() +
 128                 "\" DataContentHandler requires String object, " +
 129                 "was given object of type " + obj.getClass().toString());
 130 
 131         String enc = null;
 132         OutputStreamWriter osw = null;
 133 
 134         try {
 135             enc = getCharset(type);
 136             osw = new OutputStreamWriter(os, enc);
 137         } catch (IllegalArgumentException iex) {
 138             /*
 139              * An unknown charset of the form ISO-XXX-XXX will cause
 140              * the JDK to throw an IllegalArgumentException.  The
 141              * JDK will attempt to create a classname using this string,
 142              * but valid classnames must not contain the character '-',
 143              * and this results in an IllegalArgumentException, rather than
 144              * the expected UnsupportedEncodingException.  Yikes.
 145              */
 146             throw new UnsupportedEncodingException(enc);
 147         }
 148 
 149         String s = (String)obj;
 150         osw.write(s, 0, s.length());
 151         osw.flush();
 152     }
 153 
 154     private String getCharset(String type) {
 155         try {
 156             ContentType ct = new ContentType(type);
 157             String charset = ct.getParameter("charset");
 158             if (charset == null)
 159                 // If the charset parameter is absent, use US-ASCII.
 160                 charset = "us-ascii";
 161             return MimeUtility.javaCharset(charset);
 162         } catch (Exception ex) {
 163             return null;
 164         }
 165     }
 166 
 167 }
--- EOF ---