1 package org
.asamk
.signal
.manager
.storage
.protocol
;
3 import com
.fasterxml
.jackson
.core
.JsonGenerator
;
4 import com
.fasterxml
.jackson
.core
.JsonParser
;
5 import com
.fasterxml
.jackson
.databind
.DeserializationContext
;
6 import com
.fasterxml
.jackson
.databind
.JsonDeserializer
;
7 import com
.fasterxml
.jackson
.databind
.JsonNode
;
8 import com
.fasterxml
.jackson
.databind
.JsonSerializer
;
9 import com
.fasterxml
.jackson
.databind
.SerializerProvider
;
11 import org
.asamk
.signal
.manager
.util
.Utils
;
12 import org
.slf4j
.Logger
;
13 import org
.slf4j
.LoggerFactory
;
14 import org
.whispersystems
.libsignal
.SignalProtocolAddress
;
15 import org
.whispersystems
.libsignal
.state
.SessionRecord
;
16 import org
.whispersystems
.signalservice
.api
.SignalServiceSessionStore
;
17 import org
.whispersystems
.signalservice
.api
.push
.SignalServiceAddress
;
18 import org
.whispersystems
.signalservice
.api
.util
.UuidUtil
;
20 import java
.io
.IOException
;
21 import java
.util
.ArrayList
;
22 import java
.util
.Base64
;
23 import java
.util
.LinkedList
;
24 import java
.util
.List
;
26 class JsonSessionStore
implements SignalServiceSessionStore
{
28 private final static Logger logger
= LoggerFactory
.getLogger(JsonSessionStore
.class);
30 private final List
<SessionInfo
> sessions
= new ArrayList
<>();
32 private SignalServiceAddressResolver resolver
;
34 public JsonSessionStore() {
37 public void setResolver(final SignalServiceAddressResolver resolver
) {
38 this.resolver
= resolver
;
41 private SignalServiceAddress
resolveSignalServiceAddress(String identifier
) {
42 if (resolver
!= null) {
43 return resolver
.resolveSignalServiceAddress(identifier
);
45 return Utils
.getSignalServiceAddressFromIdentifier(identifier
);
50 public synchronized SessionRecord
loadSession(SignalProtocolAddress address
) {
51 var serviceAddress
= resolveSignalServiceAddress(address
.getName());
52 for (var info
: sessions
) {
53 if (info
.address
.matches(serviceAddress
) && info
.deviceId
== address
.getDeviceId()) {
55 return new SessionRecord(info
.sessionRecord
);
56 } catch (IOException e
) {
57 logger
.warn("Failed to load session, resetting session: {}", e
.getMessage());
58 final var sessionRecord
= new SessionRecord();
59 info
.sessionRecord
= sessionRecord
.serialize();
65 return new SessionRecord();
68 public synchronized List
<SessionInfo
> getSessions() {
73 public synchronized List
<Integer
> getSubDeviceSessions(String name
) {
74 var serviceAddress
= resolveSignalServiceAddress(name
);
76 var deviceIds
= new LinkedList
<Integer
>();
77 for (var info
: sessions
) {
78 if (info
.address
.matches(serviceAddress
) && info
.deviceId
!= 1) {
79 deviceIds
.add(info
.deviceId
);
87 public synchronized void storeSession(SignalProtocolAddress address
, SessionRecord
record) {
88 var serviceAddress
= resolveSignalServiceAddress(address
.getName());
89 for (var info
: sessions
) {
90 if (info
.address
.matches(serviceAddress
) && info
.deviceId
== address
.getDeviceId()) {
91 if (!info
.address
.getUuid().isPresent() || !info
.address
.getNumber().isPresent()) {
92 info
.address
= serviceAddress
;
94 info
.sessionRecord
= record.serialize();
99 sessions
.add(new SessionInfo(serviceAddress
, address
.getDeviceId(), record.serialize()));
103 public synchronized boolean containsSession(SignalProtocolAddress address
) {
104 var serviceAddress
= resolveSignalServiceAddress(address
.getName());
105 for (var info
: sessions
) {
106 if (info
.address
.matches(serviceAddress
) && info
.deviceId
== address
.getDeviceId()) {
114 public synchronized void deleteSession(SignalProtocolAddress address
) {
115 var serviceAddress
= resolveSignalServiceAddress(address
.getName());
116 sessions
.removeIf(info
-> info
.address
.matches(serviceAddress
) && info
.deviceId
== address
.getDeviceId());
120 public synchronized void deleteAllSessions(String name
) {
121 var serviceAddress
= resolveSignalServiceAddress(name
);
122 deleteAllSessions(serviceAddress
);
125 public synchronized void deleteAllSessions(SignalServiceAddress serviceAddress
) {
126 sessions
.removeIf(info
-> info
.address
.matches(serviceAddress
));
130 public void archiveSession(final SignalProtocolAddress address
) {
131 final var sessionRecord
= loadSession(address
);
132 if (sessionRecord
== null) {
135 sessionRecord
.archiveCurrentState();
136 storeSession(address
, sessionRecord
);
139 public void archiveAllSessions() {
140 for (var info
: sessions
) {
142 final var sessionRecord
= new SessionRecord(info
.sessionRecord
);
143 sessionRecord
.archiveCurrentState();
144 info
.sessionRecord
= sessionRecord
.serialize();
145 } catch (IOException ignored
) {
150 public static class JsonSessionStoreDeserializer
extends JsonDeserializer
<JsonSessionStore
> {
153 public JsonSessionStore
deserialize(
154 JsonParser jsonParser
, DeserializationContext deserializationContext
155 ) throws IOException
{
156 JsonNode node
= jsonParser
.getCodec().readTree(jsonParser
);
158 var sessionStore
= new JsonSessionStore();
160 if (node
.isArray()) {
161 for (var session
: node
) {
162 var sessionName
= session
.hasNonNull("name") ? session
.get("name").asText() : null;
163 if (UuidUtil
.isUuid(sessionName
)) {
164 // Ignore sessions that were incorrectly created with UUIDs as name
168 var uuid
= session
.hasNonNull("uuid") ? UuidUtil
.parseOrNull(session
.get("uuid").asText()) : null;
169 final var serviceAddress
= uuid
== null
170 ? Utils
.getSignalServiceAddressFromIdentifier(sessionName
)
171 : new SignalServiceAddress(uuid
, sessionName
);
172 final var deviceId
= session
.get("deviceId").asInt();
173 final var record = Base64
.getDecoder().decode(session
.get("record").asText());
174 var sessionInfo
= new SessionInfo(serviceAddress
, deviceId
, record);
175 sessionStore
.sessions
.add(sessionInfo
);
183 public static class JsonSessionStoreSerializer
extends JsonSerializer
<JsonSessionStore
> {
186 public void serialize(
187 JsonSessionStore jsonSessionStore
, JsonGenerator json
, SerializerProvider serializerProvider
188 ) throws IOException
{
189 json
.writeStartArray();
190 for (var sessionInfo
: jsonSessionStore
.sessions
) {
191 json
.writeStartObject();
192 if (sessionInfo
.address
.getNumber().isPresent()) {
193 json
.writeStringField("name", sessionInfo
.address
.getNumber().get());
195 if (sessionInfo
.address
.getUuid().isPresent()) {
196 json
.writeStringField("uuid", sessionInfo
.address
.getUuid().get().toString());
198 json
.writeNumberField("deviceId", sessionInfo
.deviceId
);
199 json
.writeStringField("record", Base64
.getEncoder().encodeToString(sessionInfo
.sessionRecord
));
200 json
.writeEndObject();
202 json
.writeEndArray();