dhe.go

package main

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

func main() {
	p, _ := rand.Prime(rand.Reader, 256)
	g, _ := rand.Int(rand.Reader, p)
	a, _ := rand.Int(rand.Reader, p.Sub(p, big.NewInt(1)))
	A := g.Exp(g, a, p)
	// Alice - send(g, p, A) -> Bob

	b, _ := rand.Int(rand.Reader, p.Sub(p, big.NewInt(1)))
	B := g.Exp(g, b, p)
	// Bob - send(g, p, B) -> Bob

	// Bob side
	Kb := A.Exp(A, b, p)

	// Alice side
	Ka := B.Exp(B, a, p)

	fmt.Println("Ka:", Ka.Text(16))
	fmt.Println("Kb:", Kb.Text(16))
}