-
-
Notifications
You must be signed in to change notification settings - Fork 119
/
encrypted_message.go
57 lines (51 loc) · 1.17 KB
/
encrypted_message.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
package crypto
import "github.com/gotd/td/bin"
// EncryptedMessage of protocol.
type EncryptedMessage struct {
AuthKeyID [8]byte
MsgKey bin.Int128
EncryptedData []byte
}
// Decode implements bin.Decoder.
func (e *EncryptedMessage) Decode(b *bin.Buffer) error {
if err := b.ConsumeN(e.AuthKeyID[:], 8); err != nil {
return err
}
{
v, err := b.Int128()
if err != nil {
return err
}
e.MsgKey = v
}
// Consuming the rest of the buffer.
e.EncryptedData = append(e.EncryptedData[:0], make([]byte, b.Len())...)
if err := b.ConsumeN(e.EncryptedData, b.Len()); err != nil {
return err
}
return nil
}
// DecodeWithoutCopy is like Decode, but EncryptedData references to given buffer instead of
// copying.
func (e *EncryptedMessage) DecodeWithoutCopy(b *bin.Buffer) error {
if err := b.ConsumeN(e.AuthKeyID[:], 8); err != nil {
return err
}
{
v, err := b.Int128()
if err != nil {
return err
}
e.MsgKey = v
}
// Consuming the rest of the buffer.
e.EncryptedData = b.Buf
return nil
}
// Encode implements bin.Encoder.
func (e EncryptedMessage) Encode(b *bin.Buffer) error {
b.Put(e.AuthKeyID[:])
b.PutInt128(e.MsgKey)
b.Put(e.EncryptedData)
return nil
}