package chat.server.im;
|
|
import java.lang.reflect.Method;
|
import java.nio.charset.StandardCharsets;
|
import java.text.ParseException;
|
import java.text.SimpleDateFormat;
|
import java.util.Base64;
|
import java.util.Date;
|
import java.util.List;
|
|
import org.apache.log4j.Logger;
|
import org.json.JSONObject;
|
|
import com.google.protobuf.GeneratedMessage;
|
import com.google.protobuf.InvalidProtocolBufferException;
|
import com.xiaoleilu.hutool.json.JSONArray;
|
|
import chat.module.entity.MessageRecord;
|
import chat.security.AES;
|
import chat.server.moquette.message.MqttPublishMessage;
|
import chat.user.Session;
|
import cn.wildfirechat.proto.WFCMessage;
|
import cn.wildfirechat.proto.WFCMessage.IMHttpWrapper;
|
import frame.util.ContentBuilder;
|
import frame.variant.translator.Translator;
|
import io.netty.buffer.ByteBuf;
|
import io.netty.handler.codec.http.FullHttpRequest;
|
|
public class DataPool {
|
|
private static Logger logger;
|
private static WFCMessageParseTable messageParseTable;
|
private RequestType requestType;
|
private FullHttpRequest httpRequest;
|
private JSONObject jsonObject;
|
private String body;
|
private String secret;
|
private byte[] rawData;
|
private byte[] data;
|
private byte[] srcBytes;
|
private IMHttpWrapper httpWrapper;
|
private int messageId;
|
private String phone;
|
|
public String getPhone() {
|
return phone;
|
}
|
|
public void setPhone(String phone) {
|
this.phone = phone;
|
}
|
|
static {
|
logger = Logger.getLogger(IMDispatcher.class);
|
messageParseTable = WFCMessageParseTable.getInstance();
|
}
|
|
//short
|
public DataPool(RequestType requestType, Session session, FullHttpRequest httpRequest) {
|
//1.
|
this.requestType = requestType;
|
this.httpRequest = httpRequest;
|
//2.
|
if (session != null) {
|
secret = session.getSecret();
|
}
|
|
//3.
|
body = readHttpRequestJSON();
|
|
if (body == null) {
|
return;
|
}
|
|
if (body.startsWith("{")) {
|
jsonObject = new JSONObject(body);
|
}
|
else if (secret != null) {
|
httpWrapper = createHttpWrapper(body, secret);
|
}
|
}
|
|
//long
|
public DataPool(RequestType requestType, Session session, MqttPublishMessage message) throws Exception {
|
//1.
|
this.requestType = requestType;
|
this.messageId = message.variableHeader().packetId();
|
|
if (session != null) {
|
secret = session.getSecret();
|
}
|
|
//2.
|
// ByteBuf payload = message.payload();
|
// byte[] content = readBytesAndRewind(payload);
|
//
|
// if (content == null) {
|
// return;
|
// }
|
//
|
// data = AES.AESDecrypt(content, secret, true);
|
//
|
// if (secret != null) {
|
// httpWrapper = createHttpWrapper(new String(data), secret);
|
// }
|
|
ByteBuf payload = message.payload();
|
byte[] content = readBytesAndRewind(payload);
|
|
if (content == null) {
|
return;
|
}
|
|
content = AES.AESDecrypt(content, secret, true);
|
|
if (content == null) {
|
return;
|
}
|
|
if (secret != null) {
|
try {
|
body = null;
|
byte[] bytes = Base64.getDecoder().decode(new String(content));
|
bytes = AES.AESDecrypt(bytes, secret, true);
|
httpWrapper = WFCMessage.IMHttpWrapper.parseFrom(bytes);
|
} catch(Exception e) {
|
httpWrapper = null;
|
srcBytes = content;
|
}
|
}
|
|
// if (body == null) {
|
// return;
|
// }
|
//
|
// if (body.startsWith("{")) {
|
// jsonObject = new JSONObject(body);
|
// }
|
// else if (secret != null) {
|
// httpWrapper = createHttpWrapper(body, secret);
|
// }
|
}
|
|
@SuppressWarnings("unchecked")
|
public <T> T getWFCMessage(Class<? extends GeneratedMessage> clazz) {
|
try {
|
byte[] bytes = null;
|
|
if (httpWrapper != null) {
|
bytes = httpWrapper.getData().toByteArray();
|
}
|
else if (srcBytes != null) {
|
bytes = srcBytes;
|
}
|
else if (body != null) {
|
bytes = body.getBytes();
|
}
|
|
if (bytes == null) {
|
return null;
|
}
|
|
Method method = messageParseTable.getMethod(clazz);
|
|
T result = (T) method.invoke(clazz, bytes);
|
return result;
|
}
|
catch (Exception e) {
|
if (logger.isDebugEnabled())
|
e.printStackTrace();
|
return null;
|
}
|
}
|
|
protected String readHttpRequestJSON() {
|
String result = null;
|
try {
|
byte[] bytes = readBytesAndRewind(httpRequest.content());
|
result = new String(bytes, StandardCharsets.UTF_8);
|
}
|
|
catch (Exception e) {
|
e.printStackTrace();
|
}
|
|
return result;
|
}
|
|
protected String readPublishMessageJSON(MqttPublishMessage message) {
|
String result = null;
|
try {
|
ByteBuf payload = message.payload();
|
byte[] content = readBytesAndRewind(payload);
|
|
if (content == null) {
|
return null;
|
}
|
|
byte[] clearContent = AES.AESDecrypt(content, secret, true);
|
|
if (clearContent == null) {
|
return null;
|
}
|
|
srcBytes = clearContent;
|
result = new String(clearContent);
|
}
|
catch (Exception e) {
|
e.printStackTrace();
|
}
|
|
return result;
|
}
|
|
public static byte[] readBytesAndRewind(ByteBuf payload) {
|
byte[] payloadContent = new byte[payload.readableBytes()];
|
int mark = payload.readerIndex();
|
payload.readBytes(payloadContent);
|
payload.readerIndex(mark);
|
|
return payloadContent;
|
}
|
|
public Object getObject(String name) {
|
if (jsonObject != null) {
|
return jsonObject.get(name);
|
}
|
|
return null;
|
}
|
|
public String getString(String name) {
|
try {
|
if (jsonObject != null) {
|
return jsonObject.getString(name);
|
}
|
|
|
}
|
catch (Exception e) {
|
return null;
|
}
|
return null;
|
}
|
|
public Integer getInteger(String name, Integer defaultValue) {
|
Object value = getObject(name);
|
return Translator.toInteger(value, defaultValue);
|
}
|
|
|
|
public Date getDate(String string) throws ParseException {
|
// TODO Auto-generated method stub
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
|
return sdf.parse(string);
|
}
|
|
public JSONArray getJSONArray(String name) {
|
Object obj = getObject(name);
|
|
if (obj == null) {
|
return null;
|
}
|
|
return (JSONArray) obj;
|
}
|
|
public String getHeader(String name) {
|
return httpRequest.headers().get(name);
|
}
|
|
public MessageRecord getMessageRecord() {
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
public List<String> getList(String string) {
|
// TODO Auto-generated method stub
|
return null;
|
}
|
|
public String getBody() {
|
return body;
|
}
|
|
public void decode() {
|
data = AES.AESDecrypt(rawData, secret, true);
|
}
|
|
public byte[] getRawData() {
|
return rawData;
|
}
|
|
public byte[] getData() {
|
return data;
|
}
|
|
public IMHttpWrapper getHttpWrapper() {
|
return httpWrapper;
|
}
|
|
public static IMHttpWrapper createHttpWrapper(String content, String secret) {
|
WFCMessage.IMHttpWrapper result = null;
|
|
try {
|
byte[] bytes = Base64.getDecoder().decode(content);
|
bytes = AES.AESDecrypt(bytes, secret, true);
|
result = WFCMessage.IMHttpWrapper.parseFrom(bytes);
|
}
|
catch(Exception e) {}
|
|
return result;
|
}
|
|
public String getRequestURI() {
|
if (httpRequest == null) {
|
return null;
|
}
|
|
return httpRequest.uri();
|
}
|
|
public String getRequestTopic() {
|
if (httpWrapper == null) {
|
return null;
|
}
|
|
return httpWrapper.getRequest();
|
}
|
|
public RequestType getRequestType() {
|
return requestType;
|
}
|
|
public FullHttpRequest getHttpRequest() {
|
return httpRequest;
|
}
|
|
public int getMessageId() {
|
return messageId;
|
}
|
|
public void setMessageId(int messageId) {
|
this.messageId = messageId;
|
}
|
|
public JSONObject getJsonObject() {
|
return jsonObject;
|
}
|
|
public void setJsonObject(JSONObject jsonObject) {
|
this.jsonObject = jsonObject;
|
}
|
|
@Override
|
public String toString() {
|
ContentBuilder builder = new ContentBuilder();
|
builder.append("requestType: " + requestType);
|
builder.append("jsonObject: " + jsonObject);
|
builder.append("body: " + body);
|
builder.append("rawData: " + rawData);
|
builder.append("httpWrapper: " + httpWrapper);
|
builder.append("messageId: " + messageId);
|
|
return builder.toString();
|
}
|
|
|
}
|
