Skip to content

Commit

Permalink
可以发送群消息
Browse files Browse the repository at this point in the history
  • Loading branch information
test committed Mar 1, 2021
1 parent e577bb9 commit ff26332
Show file tree
Hide file tree
Showing 2 changed files with 186 additions and 14 deletions.
198 changes: 185 additions & 13 deletions Gorgeous/src/main/java/GorgeousEngine.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,12 @@
import org.whispersystems.libsignal.groups.state.SenderKeyRecord;
import org.whispersystems.libsignal.logging.Log;
import org.whispersystems.libsignal.protocol.CiphertextMessage;
import org.whispersystems.libsignal.protocol.SenderKeyDistributionMessage;
import org.whispersystems.libsignal.state.PreKeyBundle;
import org.whispersystems.libsignal.state.PreKeyRecord;
import org.whispersystems.libsignal.state.SignedPreKeyRecord;

import java.math.BigInteger;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
Expand Down Expand Up @@ -239,8 +239,7 @@ public void run() {
}

void Test() {
LinkedList<PreKeyRecord> unsentPreKeys = axolotlManager_.LoadUnSendPreKey();
FlushKeys(axolotlManager_.LoadLatestSignedPreKey(true), unsentPreKeys);
SendPing();
}

public String SendText(String jid, String content) {
Expand Down Expand Up @@ -536,7 +535,7 @@ void HandleRecvMessage(ProtocolTreeNode node) {

ProtocolTreeNode skMsgEncNode = GetEncNode(encNodes, "skmsg");
if (skMsgEncNode != null) {
HandleSenderKeyMessage(recepid, node);
plainText = HandleSenderKeyMessage(recepid, node);
}

retries_.remove(node.IqId());
Expand Down Expand Up @@ -760,7 +759,9 @@ void FlushKeys(SignedPreKeyRecord signedPreKeyRecord, List<PreKeyRecord> unsent
}

String AddTask(ProtocolTreeNode node, NodeCallback callback) {
registerHandleMap_.put(node.IqId(), new NodeHandleInfo(callback, node));
if (null != callback) {
registerHandleMap_.put(node.IqId(), new NodeHandleInfo(callback, node));
}
return noiseHandshake_.SendNode(node);
}

Expand Down Expand Up @@ -806,6 +807,9 @@ String JidNormalize(String jid) {
}

String SendSerialData(String jid, byte[] serialData, String messageType, String mediaType, String iqId) {
if (StringUtil.isEmpty(iqId)) {
iqId = GenerateIqId();
}
jid = JidNormalize(jid);
String recepid = jid;
int index =recepid.indexOf("@");
Expand All @@ -818,9 +822,6 @@ String SendSerialData(String jid, byte[] serialData, String messageType, String
} else if (axolotlManager_.ContainsSession(recepid)) {
return SendToContact(jid, serialData, messageType, mediaType, iqId);
} else {
if (StringUtil.isEmpty(iqId)) {
iqId = GenerateIqId();
}
LinkedList<String> jids = new LinkedList<>();
jids.add(jid);
String finalJid = jid;
Expand All @@ -830,17 +831,188 @@ String SendSerialData(String jid, byte[] serialData, String messageType, String
}
}


class HandleGetGroupInfo implements NodeCallback{
byte[] message_;
String messageType_;
String mediaType_;
String iqId_;
HandleGetGroupInfo(byte[] message, String messageType, String mediaType, String iqId) {
message_ = message;
messageType_ = messageType;
mediaType_ = mediaType;
iqId_ = iqId;
}
@Override
public void Run(ProtocolTreeNode srcNode, ProtocolTreeNode result) {
ProtocolTreeNode group = result.GetChild("group");
if (group == null) {
Log.e(TAG, "获取群失败:" + srcNode.toString());
return;
}
String selfJid = JidNormalize(envBuilder_.getFullphone());
List<String> sessionJids = new LinkedList<>();
LinkedList<ProtocolTreeNode> participants = group.GetChildren("participant");
for (ProtocolTreeNode participant : participants) {
String jid = participant.GetAttributeValue("jid");
if (jid.equals(selfJid)) {
continue;
}
sessionJids.add(jid);
}
EnsureSessionsAndSendToGroup(srcNode.GetAttributeValue("to"), sessionJids, message_, messageType_, mediaType_, iqId_);
}
}


void EnsureSessionsAndSendToGroup(String groupId, List<String> sessionJids, byte[] message, String messageType, String mediaType, String iqId) {
List<String> jidsNoSession = new LinkedList<>();
for (String jid : sessionJids) {
String recepid = jid;
int index =recepid.indexOf("@");
if (index != -1) {
recepid = recepid.substring(0, index);
}
if (!axolotlManager_.ContainsSession(recepid)) {
jidsNoSession.add(jid);
}
}
if (jidsNoSession.isEmpty()) {
try {
SendToGroupWithSessions(groupId, sessionJids, message, messageType, mediaType, iqId);
} catch (UntrustedIdentityException e) {
e.printStackTrace();
} catch (NoSessionException e) {
e.printStackTrace();
} catch (InvalidKeyException e) {
e.printStackTrace();
}
} else {
GetKeysFor(jidsNoSession, (srcNode, result) -> {
try {
SendToGroupWithSessions(groupId, jidsNoSession, message, messageType, mediaType, iqId);
} catch (UntrustedIdentityException e) {
e.printStackTrace();
} catch (NoSessionException e) {
e.printStackTrace();
} catch (InvalidKeyException e) {
e.printStackTrace();
}
});
}
}

void SendToGroupWithSessions(String groupId, List<String> sessionJids, byte[] message, String messageType, String mediaType, String iqId) throws UntrustedIdentityException, NoSessionException, InvalidKeyException {
ProtocolTreeNode partici = new ProtocolTreeNode("participants");
if (!sessionJids.isEmpty()) {
SenderKeyDistributionMessage senderKeyDistributionMessage = axolotlManager_.GroupCreateSKMsg(groupId);
ByteString senderKeySerial = ByteString.copyFrom(senderKeyDistributionMessage.serialize());
for (String jid : sessionJids) {
String recepid = jid;
int index =recepid.indexOf("@");
if (index != -1) {
recepid = recepid.substring(0, index);
}
WhatsMessage.WhatsAppMessage.Builder sendKeyMessage = SerializeSenderKeyDistributionMessageToProtobuf(groupId, senderKeySerial);
CiphertextMessage cipherText = axolotlManager_.Encrypt(recepid, sendKeyMessage.build().toByteArray());
ProtocolTreeNode encNode = new ProtocolTreeNode("enc");
encNode.AddAttribute(new StanzaAttribute("v", "2"));
encNode.SetData(cipherText.serialize());
if (!StringUtil.isEmpty(mediaType)) {
encNode.AddAttribute(new StanzaAttribute("mediatype", mediaType));
}
switch (cipherText.getType()) {
case CiphertextMessage.PREKEY_TYPE:
{
encNode.AddAttribute(new StanzaAttribute("type", "pkmsg"));
}
break;
case CiphertextMessage.SENDERKEY_TYPE:
{
encNode.AddAttribute(new StanzaAttribute("type", "skmsg"));
}
break;
default:
{
encNode.AddAttribute(new StanzaAttribute("type", "msg"));
}
}
ProtocolTreeNode participant = new ProtocolTreeNode("participant");
participant.AddAttribute(new StanzaAttribute("jid",jid));
participant.AddChild(encNode);
partici.AddChild(participant);
}
}

//组装消息
ProtocolTreeNode msg = new ProtocolTreeNode("message");
msg.AddAttribute(new StanzaAttribute("id", iqId));
msg.AddAttribute(new StanzaAttribute("type", messageType));
msg.AddAttribute(new StanzaAttribute("to", groupId));
msg.AddAttribute(new StanzaAttribute("t", String.valueOf(System.currentTimeMillis() / 1000)));

//添加group 消息
ProtocolTreeNode encNode = new ProtocolTreeNode("enc");
encNode.AddAttribute(new StanzaAttribute("v", "2"));
encNode.AddAttribute(new StanzaAttribute("type", "skmsg"));
if (!StringUtil.isEmpty(mediaType)) {
encNode.AddAttribute(new StanzaAttribute("mediatype", mediaType));
}
encNode.SetData(axolotlManager_.GroupEncrypt(groupId, message));
msg.AddChild(encNode);
msg.AddChild(partici);
AddTask(msg);
}

WhatsMessage.WhatsAppMessage.Builder SerializeSenderKeyDistributionMessageToProtobuf(String groupId, ByteString senderKeySerial) {
WhatsMessage.WhatsAppMessage.Builder builder = WhatsMessage.WhatsAppMessage.newBuilder();
WhatsMessage.WhatsAppSenderKeyDistributionMessage.Builder senderKeyBuilder = builder.getSenderKeyDistributionMessageBuilder();
senderKeyBuilder.setGroupId(groupId);
senderKeyBuilder.setAxolotlSenderKeyDistributionMessage(senderKeySerial);
return builder;
}

String SendToGroup(String jid, byte[] message, String messageType, String mediaType, String iqId) {
SenderKeyRecord senderKeyRecord = axolotlManager_.LoadSenderKey(jid);
if (null == senderKeyRecord) {
if (senderKeyRecord.isEmpty()) {
//获取群信息

InnerGetGroupInfo(jid, new HandleGetGroupInfo(message, messageType, mediaType,iqId));
} else {

try {
SendToGroupWithSessions(jid, new LinkedList<>(), message, messageType, mediaType, iqId);
} catch (UntrustedIdentityException e) {
e.printStackTrace();
} catch (NoSessionException e) {
e.printStackTrace();
} catch (InvalidKeyException e) {
e.printStackTrace();
}
}
return "";
return iqId;
}





String GetGroupInfo(String jid) {
return InnerGetGroupInfo(jid, null);
}

String InnerGetGroupInfo(String jid, NodeCallback callback) {
ProtocolTreeNode iq = new ProtocolTreeNode("iq");
iq.AddAttribute(new StanzaAttribute("id", GenerateIqId()));
iq.AddAttribute(new StanzaAttribute("type", "get"));
iq.AddAttribute(new StanzaAttribute("to", jid));
iq.AddAttribute(new StanzaAttribute("xmlns", "w:g2"));

ProtocolTreeNode query = new ProtocolTreeNode("query");
query.AddAttribute(new StanzaAttribute("request", "interactive"));
iq.AddChild(query);
return AddTask(iq, callback);
}


String SendToContact(String jid, byte[] message, String messageType, String mediaType, String iqId) {
String recepid = jid;
int index =recepid.indexOf("@");
Expand All @@ -854,7 +1026,7 @@ String SendToContact(String jid, byte[] message, String messageType, String medi
} catch (UntrustedIdentityException e) {
e.printStackTrace();
}
return "";
return iqId;
}

String SendEncMessage(String jid, byte[] cipherText, int encType, String messageType, String mediaType, String iqId) {
Expand Down
2 changes: 1 addition & 1 deletion Gorgeous/src/main/java/axolotl/store/SignalSenderKeyStore.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -49,6 +49,6 @@ public SenderKeyRecord loadSenderKey(SenderKeyName senderKeyName) {
catch (Exception e){
Log.e(TAG, e.getMessage());
}
return null;
return new SenderKeyRecord();
}
}

0 comments on commit ff26332

Please sign in to comment.