init: 自 zomaintain/backend/rdplib 平移独立成库; module path 从上游 github.com/nakagami/grdp 改为 git.zeroonesoft.cn/golib/rdplib

2026-09-19 18:55:43 +08:00
commit 1096fbce9d
94 changed files with 36790 additions and 0 deletions
@@ -0,0 +1,53 @@
+package nla_test
+
+import (
+	"bytes"
+	"encoding/hex"
+	"testing"
+
+	"github.com/lunixbochs/struc"
+	"git.zeroonesoft.cn/golib/rdplib/protocol/nla"
+)
+
+func TestNewNegotiateMessage(t *testing.T) {
+	ntlm := nla.NewNTLMv2("", "", "")
+	negoMsg := ntlm.GetNegotiateMessage()
+	buff := &bytes.Buffer{}
+	struc.Pack(buff, negoMsg)
+
+	result := hex.EncodeToString(buff.Bytes())
+	expected := "4e544c4d535350000100000035820860000000000000000000000000000000000000000000000000"
+
+	if result != expected {
+		t.Error(result, " not equals to", expected)
+	}
+}
+
+func TestNTLMv2_ComputeResponse(t *testing.T) {
+	ntlm := nla.NewNTLMv2("", "", "")
+
+	ResponseKeyNT, _ := hex.DecodeString("39e32c766260586a9036f1ceb04c3007")
+	ResponseKeyLM, _ := hex.DecodeString("39e32c766260586a9036f1ceb04c3007")
+	ServerChallenge, _ := hex.DecodeString("adcb9d1c8d4a5ed8")
+	ClienChallenge, _ := hex.DecodeString("1a78bed8e5d5efa7")
+	Timestamp, _ := hex.DecodeString("a02f44f01267d501")
+	ServerName, _ := hex.DecodeString("02001e00570049004e002d00460037005200410041004d004100500034004a00430001001e00570049004e002d00460037005200410041004d004100500034004a00430004001e00570049004e002d00460037005200410041004d004100500034004a00430003001e00570049004e002d00460037005200410041004d004100500034004a00430007000800a02f44f01267d50100000000")
+
+	NtChallengeResponse, LmChallengeResponse, SessionBaseKey := ntlm.ComputeResponseV2(ResponseKeyNT, ResponseKeyLM, ServerChallenge, ClienChallenge, Timestamp, ServerName)
+
+	ntChallRespExpected := "4e7316531937d2fc91e7230853844b890101000000000000a02f44f01267d5011a78bed8e5d5efa70000000002001e00570049004e002d00460037005200410041004d004100500034004a00430001001e00570049004e002d00460037005200410041004d004100500034004a00430004001e00570049004e002d00460037005200410041004d004100500034004a00430003001e00570049004e002d00460037005200410041004d004100500034004a00430007000800a02f44f01267d50100000000"
+	lmChallRespExpected := "d4dc6edc0c37dd70f69b5c4f05a615661a78bed8e5d5efa7"
+	sessBaseKeyExpected := "034009be89a0507b2bd6d28e966e1dab"
+
+	if hex.EncodeToString(NtChallengeResponse) != ntChallRespExpected {
+		t.Error("NtChallengeResponse incorrect")
+	}
+
+	if hex.EncodeToString(LmChallengeResponse) != lmChallRespExpected {
+		t.Error("LmChallengeResponse incorrect")
+	}
+
+	if hex.EncodeToString(SessionBaseKey) != sessBaseKeyExpected {
+		t.Error("SessionBaseKey incorrect")
+	}
+}