127 lines
4.8 KiB
Go
127 lines
4.8 KiB
Go
package pblite
|
|
|
|
import (
|
|
"google.golang.org/protobuf/proto"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
|
|
"go.mau.fi/mautrix-gmessages/libgm/crypto"
|
|
"go.mau.fi/mautrix-gmessages/libgm/gmproto"
|
|
"go.mau.fi/mautrix-gmessages/libgm/routes"
|
|
)
|
|
|
|
type DevicePair struct {
|
|
Mobile *gmproto.Device `json:"mobile,omitempty"`
|
|
Browser *gmproto.Device `json:"browser,omitempty"`
|
|
}
|
|
|
|
type RequestData struct {
|
|
RequestID string `json:"requestId,omitempty"`
|
|
Timestamp int64 `json:"timestamp,omitempty"`
|
|
Action gmproto.ActionType `json:"action,omitempty"`
|
|
Bool1 bool `json:"bool1,omitempty"`
|
|
Bool2 bool `json:"bool2,omitempty"`
|
|
EncryptedData []byte `json:"requestData,omitempty"`
|
|
RawDecrypted []byte `json:"-,omitempty"`
|
|
Decrypted proto.Message `json:"decrypted,omitempty"`
|
|
Bool3 bool `json:"bool3,omitempty"`
|
|
}
|
|
|
|
type Response struct {
|
|
ResponseID string `json:"responseId,omitempty"`
|
|
BugleRoute gmproto.BugleRoute `json:"bugleRoute,omitempty"`
|
|
StartExecute string `json:"startExecute,omitempty"`
|
|
MessageType gmproto.MessageType `json:"eventType,omitempty"`
|
|
FinishExecute string `json:"finishExecute,omitempty"`
|
|
MillisecondsTaken string `json:"millisecondsTaken,omitempty"`
|
|
Devices *DevicePair `json:"devices,omitempty"`
|
|
Data RequestData `json:"data,omitempty"`
|
|
SignatureId string `json:"signatureId,omitempty"`
|
|
Timestamp string `json:"timestamp"`
|
|
}
|
|
|
|
func DecryptInternalMessage(internalMessage *gmproto.InternalMessage, cryptor *crypto.AESCTRHelper) (*Response, error) {
|
|
var resp *Response
|
|
switch internalMessage.Data.BugleRoute {
|
|
case gmproto.BugleRoute_PairEvent:
|
|
decodedData := &gmproto.PairEvents{}
|
|
decodeErr := proto.Unmarshal(internalMessage.Data.ProtobufData, decodedData)
|
|
if decodeErr != nil {
|
|
return nil, decodeErr
|
|
}
|
|
resp = newResponseFromPairEvent(internalMessage.GetData(), decodedData)
|
|
case gmproto.BugleRoute_DataEvent:
|
|
internalRequestData := &gmproto.InternalRequestData{}
|
|
decodeErr := proto.Unmarshal(internalMessage.Data.ProtobufData, internalRequestData)
|
|
if decodeErr != nil {
|
|
return nil, decodeErr
|
|
}
|
|
if internalRequestData.EncryptedData != nil {
|
|
decryptedBytes, err := cryptor.Decrypt(internalRequestData.EncryptedData)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
responseStruct := routes.Routes[internalRequestData.GetAction()].ResponseStruct
|
|
deserializedData := responseStruct.ProtoReflect().New().Interface()
|
|
err = proto.Unmarshal(decryptedBytes, deserializedData)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
resp = newResponseFromDataEvent(internalMessage.GetData(), internalRequestData, decryptedBytes, deserializedData)
|
|
} else {
|
|
resp = newResponseFromDataEvent(internalMessage.GetData(), internalRequestData, nil, nil)
|
|
}
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func newResponseFromPairEvent(internalMsg *gmproto.InternalMessageData, data *gmproto.PairEvents) *Response {
|
|
resp := &Response{
|
|
ResponseID: internalMsg.GetResponseID(),
|
|
BugleRoute: internalMsg.GetBugleRoute(),
|
|
StartExecute: internalMsg.GetStartExecute(),
|
|
MessageType: internalMsg.GetMessageType(),
|
|
FinishExecute: internalMsg.GetFinishExecute(),
|
|
MillisecondsTaken: internalMsg.GetMillisecondsTaken(),
|
|
Devices: &DevicePair{
|
|
Mobile: internalMsg.GetMobile(),
|
|
Browser: internalMsg.GetBrowser(),
|
|
},
|
|
Data: RequestData{
|
|
Decrypted: data,
|
|
},
|
|
Timestamp: internalMsg.GetTimestamp(),
|
|
SignatureId: internalMsg.GetSignatureID(),
|
|
}
|
|
|
|
return resp
|
|
}
|
|
|
|
func newResponseFromDataEvent(internalMsg *gmproto.InternalMessageData, internalRequestData *gmproto.InternalRequestData, rawData []byte, decrypted protoreflect.ProtoMessage) *Response {
|
|
resp := &Response{
|
|
ResponseID: internalMsg.GetResponseID(),
|
|
BugleRoute: internalMsg.GetBugleRoute(),
|
|
StartExecute: internalMsg.GetStartExecute(),
|
|
MessageType: internalMsg.GetMessageType(),
|
|
FinishExecute: internalMsg.GetFinishExecute(),
|
|
MillisecondsTaken: internalMsg.GetMillisecondsTaken(),
|
|
Devices: &DevicePair{
|
|
Mobile: internalMsg.GetMobile(),
|
|
Browser: internalMsg.GetBrowser(),
|
|
},
|
|
Data: RequestData{
|
|
RequestID: internalRequestData.GetSessionID(),
|
|
Timestamp: internalRequestData.GetTimestamp(),
|
|
Action: internalRequestData.GetAction(),
|
|
Bool1: internalRequestData.GetBool1(),
|
|
Bool2: internalRequestData.GetBool2(),
|
|
EncryptedData: internalRequestData.GetEncryptedData(),
|
|
Decrypted: decrypted,
|
|
RawDecrypted: rawData,
|
|
Bool3: internalRequestData.GetBool3(),
|
|
},
|
|
SignatureId: internalMsg.GetSignatureID(),
|
|
Timestamp: internalMsg.GetTimestamp(),
|
|
}
|
|
|
|
return resp
|
|
}
|