nmode's Git Repositories - signal-cli/blob - lib/src/main/java/org/asamk/signal/manager/util/MessageCacheUtils.java

   1 package org.asamk.signal.manager.util;
   2
   3 import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope;
   4 import org.whispersystems.signalservice.api.push.ServiceId;
   5 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
   6 import org.whispersystems.signalservice.api.util.UuidUtil;
   7
   8 import java.io.DataInputStream;
   9 import java.io.DataOutputStream;
  10 import java.io.File;
  11 import java.io.FileInputStream;
  12 import java.io.FileOutputStream;
  13 import java.io.IOException;
  14 import java.util.Optional;
  15
  16 public class MessageCacheUtils {
  17
  18     public static SignalServiceEnvelope loadEnvelope(File file) throws IOException {
  19         try (var f = new FileInputStream(file)) {
  20             var in = new DataInputStream(f);
  21             var version = in.readInt();
  22             if (version > 8) {
  23                 // Unsupported envelope version
  24                 return null;
  25             }
  26             var type = in.readInt();
  27             var source = in.readUTF();
  28             ServiceId sourceServiceId = null;
  29             if (version >= 3) {
  30                 sourceServiceId = ServiceId.parseOrNull(in.readUTF());
  31             }
  32             var sourceDevice = in.readInt();
  33             if (version == 1) {
  34                 // read legacy relay field
  35                 in.readUTF();
  36             }
  37             String destinationUuid = null;
  38             if (version >= 5) {
  39                 destinationUuid = in.readUTF();
  40             }
  41             var timestamp = in.readLong();
  42             byte[] content = null;
  43             var contentLen = in.readInt();
  44             if (contentLen > 0) {
  45                 content = new byte[contentLen];
  46                 in.readFully(content);
  47             }
  48             var legacyMessageLen = in.readInt();
  49             if (legacyMessageLen > 0) {
  50                 byte[] legacyMessage = new byte[legacyMessageLen];
  51                 in.readFully(legacyMessage);
  52             }
  53             long serverReceivedTimestamp = 0;
  54             String uuid = null;
  55             if (version >= 2) {
  56                 serverReceivedTimestamp = in.readLong();
  57                 uuid = in.readUTF();
  58                 if ("".equals(uuid)) {
  59                     uuid = null;
  60                 }
  61             }
  62             long serverDeliveredTimestamp = 0;
  63             if (version >= 4) {
  64                 serverDeliveredTimestamp = in.readLong();
  65             }
  66             boolean isUrgent = true;
  67             if (version >= 6) {
  68                 isUrgent = in.readBoolean();
  69             }
  70             boolean isStory = true;
  71             if (version >= 7) {
  72                 isStory = in.readBoolean();
  73             }
  74             String updatedPni = null;
  75             if (version >= 8) {
  76                 updatedPni = in.readUTF();
  77             }
  78             Optional<SignalServiceAddress> addressOptional = sourceServiceId == null
  79                     ? Optional.empty()
  80                     : Optional.of(new SignalServiceAddress(sourceServiceId, source));
  81             return new SignalServiceEnvelope(type,
  82                     addressOptional,
  83                     sourceDevice,
  84                     timestamp,
  85                     content,
  86                     serverReceivedTimestamp,
  87                     serverDeliveredTimestamp,
  88                     uuid,
  89                     destinationUuid == null ? UuidUtil.UNKNOWN_UUID.toString() : destinationUuid,
  90                     isUrgent,
  91                     updatedPni == null ? "" : updatedPni,
  92                     isStory);
  93         }
  94     }
  95
  96     public static void storeEnvelope(SignalServiceEnvelope envelope, File file) throws IOException {
  97         try (var f = new FileOutputStream(file)) {
  98             try (var out = new DataOutputStream(f)) {
  99                 out.writeInt(8); // version
 100                 out.writeInt(envelope.getType());
 101                 out.writeUTF(""); // legacy number
 102                 out.writeUTF(envelope.getSourceUuid().isPresent() ? envelope.getSourceUuid().get() : "");
 103                 out.writeInt(envelope.getSourceDevice());
 104                 out.writeUTF(envelope.getDestinationUuid() == null ? "" : envelope.getDestinationUuid());
 105                 out.writeLong(envelope.getTimestamp());
 106                 if (envelope.hasContent()) {
 107                     out.writeInt(envelope.getContent().length);
 108                     out.write(envelope.getContent());
 109                 } else {
 110                     out.writeInt(0);
 111                 }
 112                 out.writeInt(0); // legacy message
 113                 out.writeLong(envelope.getServerReceivedTimestamp());
 114                 var uuid = envelope.getServerGuid();
 115                 out.writeUTF(uuid == null ? "" : uuid);
 116                 out.writeLong(envelope.getServerDeliveredTimestamp());
 117                 out.writeBoolean(envelope.isUrgent());
 118                 out.writeBoolean(envelope.isStory());
 119                 out.writeUTF(envelope.getUpdatedPni() == null ? "" : envelope.getUpdatedPni());
 120             }
 121         }
 122     }
 123 }