anna/IM-server-new
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

wildfire chat server source initial submit

Browse Source
This commit is contained in:
heavyrain2012
2019-03-16 17:04:23 +08:00
parent e59217bd10
commit cab3ca0916
1466 changed files with 528862 additions and 15919 deletions
Download Patch File Download Diff File Expand all files Collapse all files
client/src/main/java/cn/wildfirechat/client/IMClient.java
+234
View File
@@ -0,0 +1,234 @@
package cn.wildfirechat.client;
import cn.wildfirechat.proto.WFCMessage;
import com.google.protobuf.ByteString;
import com.google.protobuf.InvalidProtocolBufferException;
import com.xiaoleilu.loServer.pojos.InputRoute;
import io.moquette.spi.impl.security.AES;
import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.StringEntity;
import org.apache.http.impl.client.HttpClientBuilder;
import org.fusesource.hawtbuf.Buffer;
import org.fusesource.hawtbuf.UTF8Buffer;
import org.fusesource.mqtt.client.*;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.Base64;
public class IMClient implements Listener {
private final String userId;
private final String token;
private final String clientId;
private final String host;
private final int port;
protected String mqttServerIp;
protected long mqttServerPort;
private static byte[] commonSecret= {0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x78,0x79,0x7A,0x7B,0x7C,0x7D,0x7E,0x7F};
protected String privateSecret;
private long messageHead;
public IMClient(String userId, String token, String clientId, String host, int port) {
this.userId = userId;
this.token = token;
this.clientId = clientId;
this.host = host;
this.port = port;
AES.init(commonSecret);
}
private transient MQTT mqtt = new MQTT();
private transient CallbackConnection connection = null;
public void connect() {
if(route(userId, token)) {
try {
mqtt.setHost("tcp://" + mqttServerIp + ":" + mqttServerPort);
mqtt.setVersion("1.1");
mqtt.setKeepAlive((short)180);
mqtt.setClientId(clientId);
mqtt.setConnectAttemptsMax(100);
mqtt.setReconnectAttemptsMax(100);
mqtt.setUserName(userId);
byte[] password = AES.AESEncrypt(token, privateSecret);
mqtt.setPassword(new UTF8Buffer(password));
connection = mqtt.callbackConnection();
connection.listener(this);
connection.connect(new Callback<byte[]>() {
@Override
public void onSuccess(byte[] value) {
if (value != null) {
try {
WFCMessage.ConnectAckPayload ackPayload = WFCMessage.ConnectAckPayload.parseFrom(value);
System.out.println(ackPayload.getMsgHead());
messageHead = ackPayload.getMsgHead();
} catch (InvalidProtocolBufferException e) {
e.printStackTrace();
}
}
System.out.println("on connect success");
}
@Override
public void onFailure(Throwable value) {
System.out.println("on connect failure");
}
});
} catch (Exception e) {
e.printStackTrace();
}
}
}
public void sendMessage(WFCMessage.Conversation conversation, WFCMessage.MessageContent messageContent) {
WFCMessage.Message message = WFCMessage.Message.newBuilder().setConversation(conversation).setContent(messageContent).setFromUser(userId).build();
byte[] data = message.toByteArray();
data = AES.AESEncrypt(data, privateSecret);
connection.publish("MS", data, QoS.AT_LEAST_ONCE, false, new Callback<byte[]>() {
@Override
public void onSuccess(byte[] value) {
}
@Override
public void onFailure(Throwable value) {
}
});
}
protected boolean route(String userId, String token) {
HttpPost httpPost;
try{
HttpClient httpClient = HttpClientBuilder.create().build();
httpPost = new HttpPost("http://" + host + ":" + port + "/route");
InputRoute inputRoute = new InputRoute();
inputRoute.setUserId(userId);
inputRoute.setClientId(clientId);
inputRoute.setToken(token);
WFCMessage.IMHttpWrapper request = WFCMessage.IMHttpWrapper.newBuilder().setClientId(clientId).setToken(token).setRequest("ROUTE").setData(ByteString.copyFrom(token.getBytes())).build();
byte[] data = AES.AESEncrypt(request.toByteArray(), commonSecret);
data = Base64.getEncoder().encode(data);
StringEntity entity = new StringEntity(new String(data), Charset.forName("UTF-8"));
entity.setContentEncoding("UTF-8");
entity.setContentType("application/json");
httpPost.setEntity(entity);
HttpResponse response = httpClient.execute(httpPost);
if(response != null){
if (response.getStatusLine().getStatusCode() != 200) {
System.out.println("Http response error {" + response.getStatusLine().getStatusCode() + "}");
return false;
}
HttpEntity resEntity = response.getEntity();
if(resEntity != null){
try {
byte[] bytes = new byte[resEntity.getContent().available()];
resEntity.getContent().read(bytes);
if (bytes[0] == 0) {
byte[] bytes1 = new byte[bytes.length -1];
for (int i = 0; i < bytes1.length; i++) {
bytes1[i] = bytes[i+1];
}
byte[] rawData = AES.AESDecrypt(bytes1, "", true);
WFCMessage.RouteResponse routeResponse = WFCMessage.RouteResponse.parseFrom(rawData);
mqttServerIp = routeResponse.getHost();
mqttServerPort = routeResponse.getLongPort();
routeResponse.getShortPort();
return true;
} else {
System.out.println("the route failure:" + bytes[0]);
return false;
}
} catch (IOException e) {
e.printStackTrace();
} catch (UnsupportedOperationException e) {
e.printStackTrace();
}
}
} else {
System.out.println("Http response nil");
}
}catch(Exception ex){
ex.printStackTrace();
}
return false;
}
@Override
public void onConnected() {
System.out.println("onConnected");
}
@Override
public void onDisconnected() {
System.out.println("onDisconnected");
}
@Override
public void onPublish(UTF8Buffer topic, Buffer body, Runnable ack) {
System.out.println("onPublish" + topic.toString());
ack.run();
if (topic.toString().equals("MN")) {
try {
WFCMessage.NotifyMessage notifyMessage = WFCMessage.NotifyMessage.parseFrom(body.toByteArray());
WFCMessage.PullMessageRequest request = WFCMessage.PullMessageRequest.newBuilder().setId(messageHead).setType(notifyMessage.getType()).build();
connection.publish("MP", request.toByteArray(), QoS.AT_LEAST_ONCE, false, new Callback<byte[]>(){
@Override
public void onSuccess(byte[] value) {
try {
byte[] data = AES.AESDecrypt(value, privateSecret, true);
try {
WFCMessage.PullMessageResult result = WFCMessage.PullMessageResult.parseFrom(data);
} catch (InvalidProtocolBufferException e) {
e.printStackTrace();
}
System.out.println("onSuccess");
} catch (Exception e) {
e.printStackTrace();
}
}
@Override
public void onFailure(Throwable value) {
System.out.println("onFailure");
}
});
} catch (Exception e) {
e.printStackTrace();
}
}
}
@Override
public void onFailure(Throwable value) {
System.out.println("onDisconnected" + value.toString());
}
public static void main(String[] args) {
IMClient client = new IMClient("-C-3-3KK","hN0AF2XX6+o1vf6UIC4Fo5O6E+EWC2nmnZd7xZi4SlU=", "1234", "im.liyufan.win", 80);
client.connect();
try {
Thread.sleep(1000000);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
}
client/src/main/java/cn/wildfirechat/proto/ProtoConstants.java
+125
View File
@@ -0,0 +1,125 @@
package cn.wildfirechat.proto;
public class ProtoConstants {
//message Conversation -> type
public interface ConversationType {
int ConversationType_Private = 0;
int ConversationType_Group = 1;
int ConversationType_ChatRoom = 2;
int ConversationType_Channel = 3;
int ConversationType_Thing = 4;
}
//message GroupInfo -> type
public interface GroupType {
//member can add quit change group name and portrait, owner can do all the operations
int GroupType_Normal = 0;
//every member can add quit change group name and portrait, no one can kickoff others
int GroupType_Free = 1;
//member can only quit, owner can do all the operations
int GroupType_Restricted = 2;
}
//message GroupMember -> type
public interface GroupMemberType {
int GroupMemberType_Normal = 0;
int GroupMemberType_Manager = 1;
int GroupMemberType_Owner = 2;
int GroupMemberType_Silent = 3;
int GroupMemberType_Removed = 4;
}
//message FriendRequest -> status
public interface FriendRequestStatus {
int RequestStatus_Sent = 0;
int RequestStatus_Accepted = 1;
int RequestStatus_Rejected = 2;
}
//message UploadDeviceTokenRequest -> platform
public interface Platform {
int Platform_UNSET = 0;
int Platform_iOS = 1;
int Platform_Android = 2;
int Platform_Windows = 3;
int Platform_OSX = 4;
int Platform_WEB = 5;
}
//message NotifyMessage & PullMessageRequest -> type
public interface PullType {
int Pull_Normal = 0;
int Pull_ChatRoom = 1;
int Pull_Group = 2;
}
//message UserResult -> code
public interface UserResultCode {
int Success = 0;
int NotFound = 1;
int NotModified = 2;
}
//message ChatroomInfo -> state
public interface ChatroomState {
int Chatroom_State_Normal = 0;
int Chatroom_State_NotStart = 1;
int Chatroom_State_End = 2;
}
//message MessageContent -> contentType
public interface ContentType {
int Unknown = 0;
int Text = 1;
int Image = 2;
int Voice = 3;
int Location = 4;
int Video = 5;
int RichMedia = 6;
int Custom = 7;
}
//ModifyGroupInfoRequest -> type
public interface ModifyGroupInfoType {
int Modify_Group_Name = 0;
int Modify_Group_Portrait = 1;
int Modify_Group_Extra = 2;
}
//ModifyGroupInfoRequest -> type
public interface PersistFlag {
int Not_Persist = 0;
int Persist = 1;
int Persist_And_Count = 3;
int Transparent = 4;
}
//ModifyChannelInfoRequest -> type
public interface ModifyChannelInfoType {
int Modify_Channel_Name = 0;
int Modify_Channel_Portrait = 1;
int Modify_Channel_Desc = 2;
int Modify_Channel_Extra = 3;
int Modify_Channel_Secret = 4;
int Modify_Channel_Callback = 5;
int Modify_Channel_OnlyCallback = 6;
}
//Channel -> status
public interface ChannelStatus {
int Channel_Status_Public = 0;
int Channel_Status_Private = 1;
int Channel_Status_Destoryed = 2;
}
public interface UserType {
int UserType_Normal = 0;
int UserType_Robot = 1;
int UserType_Device = 2;
}
}
client/src/main/java/cn/wildfirechat/proto/WFCMessage.java
+56388
View File
File diff suppressed because it is too large Load Diff