143 lines
4.4 KiB
Java
143 lines
4.4 KiB
Java
/*
|
|
* Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved.
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
*
|
|
* This code is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License version 2 only, as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
|
* or visit www.oracle.com if you need additional information or have any
|
|
* questions.
|
|
*/
|
|
|
|
|
|
import java.io.UnsupportedEncodingException;
|
|
import java.util.Arrays;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.logging.ConsoleHandler;
|
|
import java.util.logging.Level;
|
|
import java.util.logging.Logger;
|
|
|
|
/**
|
|
* Utility methods for parsing raw JDP packets.
|
|
*
|
|
* @author Alex Schenkman
|
|
*/
|
|
public class JdpTestUtil {
|
|
|
|
static final int HEADER_SIZE = 4 + 2; // magic + protocol version
|
|
|
|
/**
|
|
* Reads two bytes, starting at the given position,
|
|
* and converts them into an int.
|
|
*
|
|
* @param data
|
|
* @param pos
|
|
* @return
|
|
*/
|
|
static int decode2ByteInt(byte[] data, int pos) {
|
|
return (((data[pos] & 0xFF) << 8) | (data[pos + 1] & 0xFF));
|
|
}
|
|
|
|
/**
|
|
* Reads four bytes, starting at the given position,
|
|
* and converts them into an int.
|
|
*
|
|
* @param data
|
|
* @param pos
|
|
* @return
|
|
*/
|
|
static int decode4ByteInt(byte[] data, int pos) {
|
|
int result = data[pos + 3] & 0xFF;
|
|
result = result | ((data[pos + 2] & 0xFF) << 8);
|
|
result = result | ((data[pos + 1] & 0xFF) << 16);
|
|
result = result | ((data[pos] & 0xFF) << 24);
|
|
return result;
|
|
}
|
|
|
|
/**
|
|
* Reads an entry from the given byte array, starting at the given position.
|
|
* This is an internal function used by @see readRawPayload(byte[] rawPayload, int size).
|
|
* <p/>
|
|
* The format of an entry is:
|
|
* 2 bytes with the size of the following string.
|
|
* n bytes of characters.
|
|
*
|
|
* @param data
|
|
* @param pos
|
|
* @return
|
|
* @throws UnsupportedEncodingException
|
|
*/
|
|
static String decodeEntry(byte[] data, int pos)
|
|
throws UnsupportedEncodingException {
|
|
|
|
int size = JdpTestUtil.decode2ByteInt(data, pos);
|
|
pos = pos + 2;
|
|
byte[] raw = Arrays.copyOfRange(data, pos, pos + size);
|
|
return new String(raw, "UTF-8");
|
|
}
|
|
|
|
/**
|
|
* Builds a Map with the payload, from the raw data.
|
|
*
|
|
* @param rawData
|
|
* @return
|
|
* @throws UnsupportedEncodingException
|
|
*/
|
|
static Map<String, String> readPayload(byte[] rawData)
|
|
throws UnsupportedEncodingException {
|
|
|
|
int totalSize = rawData.length;
|
|
int payloadSize = totalSize - HEADER_SIZE;
|
|
byte[] rawPayload = Arrays.copyOfRange(rawData, HEADER_SIZE, HEADER_SIZE + payloadSize);
|
|
Map<String, String> payload = readRawPayload(rawPayload, payloadSize);
|
|
return payload;
|
|
}
|
|
|
|
/**
|
|
* Builds a map from the payload's raw data.
|
|
* This is an internal function used by @see readPayload(byte[] rawData)
|
|
*
|
|
* @param rawPayload
|
|
* @param size
|
|
* @return
|
|
* @throws UnsupportedEncodingException
|
|
*/
|
|
static Map<String, String> readRawPayload(byte[] rawPayload, int size)
|
|
throws UnsupportedEncodingException {
|
|
|
|
String key, value;
|
|
Map<String, String> payload = new HashMap<String, String>();
|
|
|
|
for (int pos = 0; pos < size; ) {
|
|
key = decodeEntry(rawPayload, pos);
|
|
pos = pos + 2 + key.length();
|
|
value = decodeEntry(rawPayload, pos);
|
|
pos = pos + 2 + value.length();
|
|
|
|
payload.put(key, value);
|
|
}
|
|
return payload;
|
|
}
|
|
|
|
static void enableConsoleLogging(Logger log, Level level) throws SecurityException {
|
|
ConsoleHandler handler = new ConsoleHandler();
|
|
handler.setLevel(level);
|
|
log.addHandler(handler);
|
|
log.setLevel(level);
|
|
}
|
|
|
|
}
|