test_test.go

package srp6ago

import (
	"testing"

	"github.com/stretchr/testify/require"
)

// @refs https://datatracker.ietf.org/doc/html/rfc5054#appendix-B
func Test_RFC5054b1024Sha1(t *testing.T) {
	//I := "alice"
	//P := "password123"
	s := MustHex2Bytes("BEB25379 D1A8581E B5A72767 3A2441EE")
	k := MustHex2Bytes("7556AA04 5AEF2CDD 07ABAF0F 665C3E81 8913186F")
	v := MustHex2Bytes(`
7E273DE8 696FFC4F 4E337D05 B4B375BE B0DDE156 9E8FA00A 9886D812
9BADA1F1 822223CA 1A605B53 0E379BA4 729FDC59 F105B478 7E5186F5
C671085A 1447B52A 48CF1970 B4FB6F84 00BBF4CE BFBB1681 52E08AB5
EA53D15C 1AFF87B2 B9DA6E04 E058AD51 CC72BFC9 033B564E 26480D78
E955A5E2 9E7AB245 DB2BE315 E2099AFB`)
	//a := MustHex2Bytes(`
	//60975527 035CF2AD 1989806F 0407210B C81EDC04 E2762A56 AFD529DD DA2D4393`)
	b := MustHex2Bytes(`
E487CB59 D31AC550 471E81F0 0F6928E0 1DDA08E9 74A004F4 9E61F5D1 05284D20`)
	A := MustHex2Bytes(`
61D5E490 F6F1B795 47B0704C 436F523D D0E560F0 C64115BB 72557EC4
4352E890 3211C046 92272D8B 2D1A5358 A2CF1B6E 0BFCF99F 921530EC
8E393561 79EAE45E 42BA92AE ACED8251 71E1E8B9 AF6D9C03 E1327F44
BE087EF0 6530E69F 66615261 EEF54073 CA11CF58 58F0EDFD FE15EFEA
B349EF5D 76988A36 72FAC47B 0769447B`)
	B := MustHex2Bytes(`
BD0C6151 2C692C0C B6D041FA 01BB152D 4916A1E7 7AF46AE1 05393011
BAF38964 DC46A067 0DD125B9 5A981652 236F99D9 B681CBF8 7837EC99
6C6DA044 53728610 D0C6DDB5 8B318885 D7D82C7F 8DEB75CE 7BD4FBAA
37089E6F 9C6059F3 88838E7A 00030B33 1EB76840 910440B1 B27AAEAE
EB4012B7 D7665238 A8E3FB00 4B117B58`)
	u := MustHex2Bytes("CE38B959 3487DA98 554ED47D 70A7AE5F 462EF019")
	S := MustHex2Bytes(`
B0DC82BA BCF30674 AE450C02 87745E79 90A3381F 63B387AA F271A10D
233861E3 59B48220 F7C4693C 9AE12B0A 6F67809F 0876E2D0 13800D6C
41BB59B6 D5979B5C 00A172B4 A2A5903A 0BDCAF8A 709585EB 2AFAFA8F
3499B200 210DCC1F 10EB3394 3CD67FC8 8A2F39A4 BE5BEC4E C0A3212D
C346D7E4 74B29EDE 8A469FFE CA686E5A`)
	m1 := MustHex2Bytes("B46A7838 46B7E569 FF8F9B44 AB8D88ED EB085A65")
	m2 := MustHex2Bytes("0B0A6AD3 024E79B5 CAD04042 ABB3A3F5 92D20C17")

	t.Run("it works", func(t *testing.T) {
		server := NewServer(v, s, RFC5054b1024Sha1)
		server.set_b(b)

		server = MustMarshalServer(t, server)
		require.Equal(t, k, server.e.k().Bytes(), "k")
		require.Equal(t, b, server.b.Bytes(), "b")
		require.Equal(t, v, server.v.Bytes(), "v")

		pub, err := server.PublicKey()
		require.NoError(t, err)
		require.Equal(t, B, pub, "B")

		err = server.SetClientPublicKey(A)
		require.NoError(t, err)

		server = MustMarshalServer(t, server)
		require.Equal(t, u, server.u.Bytes(), "u")
		require.Equal(t, S, server.S, "S")
		require.Equal(t, S, server.SecretKey(), "SecretKey")
		require.Equal(t, m1, server.m1, "m1")
		require.Equal(t, m2, server.m2, "m2")
		require.Equal(t, m2, server.Proof(), "m2")

		server = MustMarshalServer(t, server)
		require.Equal(t, true, server.IsProofValid(m1), "valid proof")
		require.Equal(t, false, server.IsProofValid([]byte{1, 2, 3}), "invalid proof")
	})
}

func MustMarshalServer(t *testing.T, server *Server) *Server {
	s, err := UnmarshalServer(server.Marshal())
	require.NoError(t, err)
	return s
}