io.symcpe.wraith.Utils Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of wraith-engine Show documentation
Show all versions of wraith-engine Show documentation
An efficient event correlation library
/**
* Copyright 2016 Symantec Corporation.
*
* 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 io.symcpe.wraith;
import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.ObjectOutputStream;
import java.lang.reflect.Field;
import java.nio.ByteBuffer;
import java.nio.charset.Charset;
import java.util.AbstractMap;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Base64;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import io.symcpe.wraith.conditions.ConditionSerializer;
/**
* Utils class for wraith-crux
*
* @author ambud_sharma
*/
public class Utils {
public static final Charset UTF8 = Charset.forName("utf-8");
public static final String PROP_NAMING_MAP = "naming.map";
public static final Map CLASSNAME_FORWARD_MAP = new HashMap<>();
public static final Map CLASSNAME_REVERSE_MAP = new HashMap<>();
static {
String map = System.getProperty(PROP_NAMING_MAP);
InputStream stream;
try {
if (map == null) {
stream = ConditionSerializer.class.getClassLoader().getResourceAsStream("naming.default");
System.out.println("Loading default naming convention");
} else {
stream = new FileInputStream(new File(map));
System.out.println("Found naming map configuration:" + map);
}
if (stream != null) {
List lines = Utils.readAllLinesFromStream(stream);
for (String line : lines) {
String[] entry = line.split("=");
CLASSNAME_FORWARD_MAP.put(entry[0], entry[1]);
CLASSNAME_REVERSE_MAP.put(entry[1], entry[0]);
}
}else {
System.out.println("Couldn't load the default naming resource");
}
} catch (IOException e) {
throw new RuntimeException(e);
}
}
protected Utils() {
}
public static byte[] eventToBytes(Event event) {
try {
ByteArrayOutputStream bytes = new ByteArrayOutputStream();
ObjectOutputStream stream = new ObjectOutputStream(bytes);
stream.writeObject(event);
return bytes.toByteArray();
} catch (IOException e) {
// should be an unreachable state
return null;
}
}
public static List readAllLinesFromStream(InputStream stream) throws IOException {
List lines = new ArrayList<>();
BufferedReader reader = new BufferedReader(new InputStreamReader(stream, UTF8));
try {
String temp = null;
while ((temp = reader.readLine()) != null) {
if(temp.trim().isEmpty()) {
continue;
}
lines.add(temp);
}
} finally {
reader.close();
}
return lines;
}
public static void addDeclaredAndInheritedFields(Class c, Collection fields) {
fields.addAll(Arrays.asList(c.getDeclaredFields()));
Class superClass = c.getSuperclass();
if (superClass != null) {
addDeclaredAndInheritedFields(superClass, fields);
}
}
/**
* @param timestamp
* @param aggregationWindow
* @param ruleActionId
* @param aggregationKey
* @return
*/
public static String createMapKey(long timestamp, int aggregationWindow, String ruleActionId,
String aggregationKey) {
String ts = intToString(floorTs(timestamp, aggregationWindow));
return new StringBuilder(ruleActionId.length() + 1 + ts.length() + 1 + aggregationKey.length())
.append(ruleActionId).append(Constants.KEY_SEPARATOR).append(ts).append(Constants.KEY_SEPARATOR)
.append(aggregationKey).toString();
}
/**
* @param timestamp
* @param aggregationWindow
* @return
*/
public static int floorTs(long timestamp, int aggregationWindow) {
return (int) (timestamp / (1000 * aggregationWindow)) * aggregationWindow;
}
public static short bytesToShort(byte[] data) {
return (short) (((data[0] << 8)) | ((data[1] & 0xFF)));
}
public static byte[] shortToBytes(short s) {
return new byte[] { (byte) ((s & 0xFF00) >> 8), (byte) (s & 0x00FF) };
}
public static byte[] combineRuleAction(short ruleId, short actionId) {
return ByteBuffer.allocate(4).putShort(ruleId).putShort(actionId).array();
}
public static Entry separateRuleAction(byte[] ruleAction) {
ByteBuffer wrap = ByteBuffer.wrap(ruleAction);
return new AbstractMap.SimpleEntry(wrap.getShort(), wrap.getShort());
}
public static byte[] combineRuleActionTs(short ruleId, short actionId, int ts) {
return ByteBuffer.allocate(8).putShort(ruleId).putShort(actionId).putInt(ts).array();
}
public static Entry separateRuleActionTs(byte[] data) {
ByteBuffer wrap = ByteBuffer.wrap(data);
return new AbstractMap.SimpleEntry(new short[] { wrap.getShort(), wrap.getShort() },
wrap.getInt());
}
public static String combineRuleActionIdTs(short ruleId, short actionId, int ts) {
return Base64.getEncoder().encodeToString(combineRuleActionTs(ruleId, actionId, ts));
}
public static Entry separateRuleActionIdTs(String data) {
return separateRuleActionTs(Base64.getDecoder().decode(data));
}
public static byte[] intToBytes(int val) {
return ByteBuffer.allocate(4).putInt(val).array();
}
public static int byteToInt(byte[] val) {
return ByteBuffer.wrap(val).getInt();
}
public static String intToString(int val) {
return Integer.toHexString(val);
}
public static int stringToInt(String val) {
return Integer.parseInt(val, 16);
}
public static String longToString(long val) {
return Long.toHexString(val);
}
public static long stringToLong(String val) {
return Long.parseLong(val, 16);
}
public static String combineRuleActionId(short ruleId, short actionId) {
String str = Base64.getEncoder().encodeToString(combineRuleAction(ruleId, actionId));
return str;
}
public static Entry separateRuleActionId(String ruleActionId) {
return separateRuleAction(Base64.getDecoder().decode(ruleActionId));
}
/**
* @param key
* @return
*/
public static String[] splitMapKey(String key) {
return key.split("\\" + Constants.KEY_SEPARATOR);
}
/**
* @param v
* @return
*/
public static String concat(String... v) {
int size = 0;
for (String vx : v) {
size += vx.length();
}
StringBuilder builder = new StringBuilder(size);
for (String vx : v) {
builder.append(vx);
}
return builder.toString();
}
}