blob: 75b050aa6069a645c35dd82707bd7073fad7741a (
plain) (
blame)
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
58
59
60
61
62
63
64
65
66
67
68
69
|
// Copyright (c) 2021 Tulir Asokan
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
// Package keys contains a utility struct for elliptic curve keypairs.
package keys
import (
"go.mau.fi/libsignal/ecc"
"golang.org/x/crypto/curve25519"
"go.mau.fi/whatsmeow/util/randbytes"
)
type KeyPair struct {
Pub *[32]byte
Priv *[32]byte
}
var _ ecc.ECPublicKeyable
func NewKeyPairFromPrivateKey(priv [32]byte) *KeyPair {
var kp KeyPair
kp.Priv = &priv
var pub [32]byte
curve25519.ScalarBaseMult(&pub, kp.Priv)
kp.Pub = &pub
return &kp
}
func NewKeyPair() *KeyPair {
priv := *(*[32]byte)(randbytes.Make(32))
priv[0] &= 248
priv[31] &= 127
priv[31] |= 64
return NewKeyPairFromPrivateKey(priv)
}
func (kp *KeyPair) CreateSignedPreKey(keyID uint32) *PreKey {
newKey := NewPreKey(keyID)
newKey.Signature = kp.Sign(&newKey.KeyPair)
return newKey
}
func (kp *KeyPair) Sign(keyToSign *KeyPair) *[64]byte {
pubKeyForSignature := make([]byte, 33)
pubKeyForSignature[0] = ecc.DjbType
copy(pubKeyForSignature[1:], keyToSign.Pub[:])
signature := ecc.CalculateSignature(ecc.NewDjbECPrivateKey(*kp.Priv), pubKeyForSignature)
return &signature
}
type PreKey struct {
KeyPair
KeyID uint32
Signature *[64]byte
}
func NewPreKey(keyID uint32) *PreKey {
return &PreKey{
KeyPair: *NewKeyPair(),
KeyID: keyID,
}
}
|
