forked from badfortrains/spotcontrol
/
keys.go
115 lines (97 loc) · 2.69 KB
/
keys.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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package spotcontrol
import (
"crypto/hmac"
"crypto/rand"
"crypto/sha1"
"log"
"math/big"
)
type PrivateKeys struct {
privateKey *big.Int
publicKey *big.Int
generator *big.Int
prime *big.Int
}
type SharedKeys struct {
challenge []byte
sendKey []byte
recvKey []byte
}
func randomVec(count int) []byte {
c := count
b := make([]byte, c)
_, err := rand.Read(b)
if err != nil {
log.Fatal("error:", err)
}
return b
}
func powm(base, exp, modulus *big.Int) *big.Int {
exp2 := big.NewInt(0).SetBytes(exp.Bytes())
base2 := big.NewInt(0).SetBytes(base.Bytes())
modulus2 := big.NewInt(0).SetBytes(modulus.Bytes())
zero := big.NewInt(0)
result := big.NewInt(1)
temp := new(big.Int)
for zero.Cmp(exp2) != 0 {
if temp.Rem(exp2, big.NewInt(2)).Cmp(zero) != 0 {
result = result.Mul(result, base2)
result = result.Rem(result, modulus2)
}
exp2 = exp2.Rsh(exp2, 1)
base2 = base2.Mul(base2, base2)
base2 = base2.Rem(base2, modulus2)
}
return result
}
func GenerateKeys() PrivateKeys {
private := new(big.Int)
private.SetBytes(randomVec(95))
return generateKeysFromPrivate(private)
}
func generateKeysFromPrivate(private *big.Int) PrivateKeys {
DH_GENERATOR := big.NewInt(0x2)
DH_PRIME := new(big.Int)
DH_PRIME.SetBytes([]byte{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xc9,
0x0f, 0xda, 0xa2, 0x21, 0x68, 0xc2, 0x34, 0xc4, 0xc6,
0x62, 0x8b, 0x80, 0xdc, 0x1c, 0xd1, 0x29, 0x02, 0x4e,
0x08, 0x8a, 0x67, 0xcc, 0x74, 0x02, 0x0b, 0xbe, 0xa6,
0x3b, 0x13, 0x9b, 0x22, 0x51, 0x4a, 0x08, 0x79, 0x8e,
0x34, 0x04, 0xdd, 0xef, 0x95, 0x19, 0xb3, 0xcd, 0x3a,
0x43, 0x1b, 0x30, 0x2b, 0x0a, 0x6d, 0xf2, 0x5f, 0x14,
0x37, 0x4f, 0xe1, 0x35, 0x6d, 0x6d, 0x51, 0xc2, 0x45,
0xe4, 0x85, 0xb5, 0x76, 0x62, 0x5e, 0x7e, 0xc6, 0xf4,
0x4c, 0x42, 0xe9, 0xa6, 0x3a, 0x36, 0x20, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff})
return PrivateKeys{
privateKey: private,
publicKey: powm(DH_GENERATOR, private, DH_PRIME),
generator: DH_GENERATOR,
prime: DH_PRIME,
}
}
func (p *PrivateKeys) addRemoteKey(remote []byte, clientPacket []byte, serverPacket []byte) SharedKeys {
remote_be := new(big.Int)
remote_be.SetBytes(remote)
shared_key := powm(remote_be, p.privateKey, p.prime)
data := make([]byte, 0, 100)
mac := hmac.New(sha1.New, shared_key.Bytes())
for i := 1; i < 6; i++ {
mac.Write(clientPacket)
mac.Write(serverPacket)
mac.Write([]byte{uint8(i)})
data = append(data, mac.Sum(nil)...)
mac.Reset()
}
mac = hmac.New(sha1.New, data[0:0x14])
mac.Write(clientPacket)
mac.Write(serverPacket)
return SharedKeys{
challenge: mac.Sum(nil),
sendKey: data[0x14:0x34],
recvKey: data[0x34:0x54],
}
}
func (p *PrivateKeys) pubKey() []byte {
return p.publicKey.Bytes()
}