summaryrefslogtreecommitdiff
path: root/go/diffie-hellman/diffie_hellman.go
blob: 0e3df06d198e58f0905f826055888a609af63924 (plain)
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
package diffiehellman

import (
	"crypto/rand"
	"math/big"
)

var one = big.NewInt(1)

// PrivateKey greater then 1 and less then p
func PrivateKey(p *big.Int) *big.Int {
	priv, _ := rand.Int(rand.Reader, p) // [0..p)
	if priv.Cmp(one) <= 0 {
		return PrivateKey(p)
	}
	return priv
}

func PublicKey(a, p *big.Int, g int64) *big.Int {
	return new(big.Int).Exp(big.NewInt(g), a, p)
}

func SecretKey(a, B, p *big.Int) *big.Int {
	return new(big.Int).Exp(B, a, p)
}

func NewPair(p *big.Int, g int64) (*big.Int, *big.Int) {
	priv := PrivateKey(p)
	pub := PublicKey(priv, p, g)
	return priv, pub
}