nl.nn.adapterframework.util.XmlEncodingUtils Maven / Gradle / Ivy

Go to download
/*
   Copyright 2023 WeAreFrank!

   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 nl.nn.adapterframework.util;

import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;

import org.apache.commons.io.ByteOrderMark;
import org.apache.commons.io.input.BOMInputStream;
import org.apache.commons.lang3.StringUtils;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

public class XmlEncodingUtils {
	static Logger log = LogManager.getLogger(XmlEncodingUtils.class);
	public static final char REPLACE_NON_XML_CHAR = 0x00BF; // Inverted question mark.
	public static String encodeChars(String string) {
		return encodeChars(string, false);
	}

	/**
	 * Translates special characters to xml equivalents
	 * like > and &. Please note that non valid xml chars
	 * are not changed, hence you might want to use
	 * replaceNonValidXmlCharacters() or stripNonValidXmlCharacters() too.
	 */
	public static String encodeChars(String string, boolean escapeNewLines) {
		if (string==null) {
			return null;
		}
		int length = string.length();
		char[] characters = new char[length];
		string.getChars(0, length, characters, 0);
		return encodeChars(characters, 0, length, escapeNewLines);
	}

	public static String encodeCharsAndReplaceNonValidXmlCharacters(String string) {
		return encodeChars(replaceNonValidXmlCharacters(string));
	}

	public static String encodeChars(char[] chars, int offset, int length) {
		return encodeChars(chars, offset, length, false);
	}

	/**
	 * Translates special characters to xml equivalents
	 * like > and &. Please note that non valid xml chars
	 * are not changed, hence you might want to use
	 * replaceNonValidXmlCharacters() or stripNonValidXmlCharacters() too.
	 */
	public static String encodeChars(char[] chars, int offset, int length, boolean escapeNewLines) {

		if (length<=0) {
			return "";
		}
		StringBuilder encoded = new StringBuilder(length);
		String escape;
		for (int i = 0; i < length; i++) {
			char c=chars[offset+i];
			escape = escapeChar(c, escapeNewLines);
			if (escape == null)
				encoded.append(c);
			else
				encoded.append(escape);
		}
		return encoded.toString();
	}

	/**
	 * Translates the five reserved XML characters (< > & " ') to their normal selves
	 */
	public static String decodeChars(String string) {
		StringBuilder decoded = new StringBuilder();

		boolean inEscape = false;
		int escapeStartPos = 0;

		for (int i = 0; i < string.length(); i++) {
			char cur=string.charAt(i);
			if (inEscape) {
				if ( cur == ';') {
					inEscape = false;
					String escapedString = string.substring(escapeStartPos, i + 1);
					char unEscape = unEscapeString(escapedString);
					if (unEscape == 0x0) {
						decoded.append(escapedString);
					}
					else {
						decoded.append(unEscape);
					}
				}
			} else {
				if (cur == '&') {
					inEscape = true;
					escapeStartPos = i;
				} else {
					decoded.append(cur);
				}
			}
		}
		if (inEscape) {
			decoded.append(string.substring(escapeStartPos));
		}
		return decoded.toString();
	}

	/**
	   * Conversion of special xml signs. Please note that non valid xml chars
	   * are not changed, hence you might want to use
	   * replaceNonValidXmlCharacters() or stripNonValidXmlCharacters() too.
	   **/
	private static String escapeChar(char c, boolean escapeNewLines) {
		switch (c) {
			case ('<') :
				return "<";
			case ('>') :
				return ">";
			case ('&') :
				return "&";
			case ('\"') :
				return """;
			case ('\'') :
				// return "'"; // apos does not work in Internet Explorer
				return "'";
			case ('\n')  :
				if(escapeNewLines)
					return "
";
		}
		return null;
	}

	private static char unEscapeString(String str) {
		if (str.equalsIgnoreCase("<"))
			return '<';
		else if (str.equalsIgnoreCase(">"))
			return '>';
		else if (str.equalsIgnoreCase("&"))
			return '&';
		else if (str.equalsIgnoreCase("""))
			return '\"';
		else if (str.equalsIgnoreCase("'") || str.equalsIgnoreCase("'"))
			return '\'';
		else
			return 0x0;
	}

	/**
	 * Replaces non-unicode-characters by '0x00BF' (inverted question mark).
	 */
	public static int replaceNonPrintableCharacters(char[] buf, int offset, int len) {
		if (len<0) {
			return len;
		}
		int c;
		int charCount = 0;
		int counter = 0;
		int readPos = 0;
		int writePos = 0;
		while(readPos0 && log.isDebugEnabled()) log.debug("replaced ["+counter+"] non valid xml characters to ["+ REPLACE_NON_XML_CHAR+"] in char array of length ["+len+"]");
		return writePos>0 ? writePos : readPos;
	}

	/**
	 * Replaces non-unicode-characters by '0x00BF' (inverted question mark)
	 * appended with #, the character number and ;.
	 */
	public static String replaceNonValidXmlCharacters(String string) {
		return replaceNonValidXmlCharacters(string, REPLACE_NON_XML_CHAR, true, true);
	}

	public static String replaceNonValidXmlCharacters(String string, char to, boolean appendCharNum, boolean allowUnicodeSupplementaryCharacters) {
		if (string==null) {
			return null;
		}
		int length = string.length();
		StringBuilder encoded = new StringBuilder(length);
		int c;
		int counter = 0;
		for (int i = 0; i < length; i += Character.charCount(c)) {
			c=string.codePointAt(i);
			if (isPrintableUnicodeChar(c, allowUnicodeSupplementaryCharacters)) {
				encoded.appendCodePoint(c);
			} else {
				if (appendCharNum) {
					encoded.append(to).append("#").append(c).append(";");
				} else {
					encoded.append(to);
				}
				counter++;
			}
		}
		if (counter>0) {
			if (log.isDebugEnabled()) log.debug("replaced ["+counter+"] non valid xml characters to ["+to+"] in string of length ["+length+"]");
		}
		return encoded.toString();
	}

	public static String stripNonValidXmlCharacters(String string, boolean allowUnicodeSupplementaryCharacters) {
		int length = string.length();
		StringBuilder encoded = new StringBuilder(length);
		int c;
		int counter = 0;
		for (int i = 0; i < length; i += Character.charCount(c)) {
			c=string.codePointAt(i);
			if (isPrintableUnicodeChar(c,
					allowUnicodeSupplementaryCharacters)) {
				encoded.appendCodePoint(c);
			} else {
				counter++;
			}
		}
		if (counter>0) {
			if (log.isDebugEnabled()) log.debug("stripped ["+counter+"] non valid xml characters in string of length ["+length+"]");
		}
		return encoded.toString();
	}

	public static boolean isPrintableUnicodeChar(int c) {
		return isPrintableUnicodeChar(c, false);
	}

	public static boolean isPrintableUnicodeChar(int c, boolean allowUnicodeSupplementaryCharacters) {
		return (c == 0x0009)
			|| (c == 0x000A)
			|| (c == 0x000D)
			|| (c >= 0x0020 && c <= 0xD7FF)
			|| (c >= 0xE000 && c <= 0xFFFD)
			|| (allowUnicodeSupplementaryCharacters && (c >= 0x00010000 && c <= 0x0010FFFF));
	}

	/**
	 * Reads binary XML data and uses the XML declaration encoding to turn it into character data.
	 */
	public static String readXml(InputStream inputStream, String defaultCharset) throws IOException {
		BOMInputStream bOMInputStream = new BOMInputStream(inputStream, ByteOrderMark.UTF_8, ByteOrderMark.UTF_16LE, ByteOrderMark.UTF_16BE);
		ByteOrderMark bom = bOMInputStream.getBOM();
		String charsetName = bom == null ? defaultCharset : bom.getCharsetName();
		return readXml(StreamUtil.streamToBytes(bOMInputStream), charsetName);
	}

	public static String readXml(byte[] source, String defaultEncoding) throws UnsupportedEncodingException {
		String charset = StringUtils.isEmpty(defaultEncoding) ? StreamUtil.DEFAULT_INPUT_STREAM_ENCODING : defaultEncoding;
		int length = source.length;

		String firstPart = new String(source, 0, length<100?length:100, charset);
		if (StringUtils.isEmpty(firstPart)) {
			return null;
		}

		if (firstPart.startsWith("")+2;
			if (endPos < 2) {
				throw new IllegalArgumentException("no valid xml declaration in string ["+firstPart+"]");
			}

			String declaration=firstPart.substring(6,endPos-2);
			log.debug("parsed declaration [{}]", declaration);
			final String encodingTarget= "encoding=\"";
			int encodingStart=declaration.indexOf(encodingTarget);
			if (encodingStart>0) {
				encodingStart+=encodingTarget.length();
				log.debug("encoding-declaration ["+declaration.substring(encodingStart)+"]");
				int encodingEnd=declaration.indexOf("\"", encodingStart);
				if (encodingEnd > 0) {
					charset=declaration.substring(encodingStart, encodingEnd);
					log.debug("parsed charset []", charset);
				} else {
					log.warn("no end in encoding attribute in declaration [{}]", declaration);
				}
			} else {
				log.warn("no encoding attribute in declaration [{}]", declaration);
			}
		}
		return new String(source, charset);
	}

}