yv1ing/ShotRDP
1
0
Fork 0
mirror of https://github.com/yv1ing/ShotRDP.git synced 2025-09-16 15:10:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

整理代码

Browse Source
This commit is contained in:
yv1ing
2025-02-20 12:31:02 +08:00
parent 84362607c2
commit 71b5229198
17 changed files with 14 additions and 183 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
grdp/protocol/nla/cssp_test.go
View File

@@ -1,16 +0,0 @@
package nla_test
import (
"encoding/hex"
"testing"
"ShotRDP/grdp/protocol/nla"
)
func TestEncodeDERTRequest(t *testing.T) {
ntlm := nla.NewNTLMv2("", "", "")
result := nla.EncodeDERTRequest([]nla.Message{ntlm.GetNegotiateMessage()}, "", "")
if hex.EncodeToString(result) != "302fa003020102a12830263024a02204204e544c4d53535000010000003582086000000000000000000000000000000000" {
t.Error("not equal")
}
}

32
grdp/protocol/nla/encode_test.go
View File

@@ -1,32 +0,0 @@
package nla_test
import (
"encoding/hex"
"testing"
"ShotRDP/grdp/protocol/nla"
)
func TestNTOWFv2(t *testing.T) {
res := hex.EncodeToString(nla.NTOWFv2("", "", ""))
expected := "f4c1a15dd59d4da9bd595599220d971a"
if res != expected {
t.Error(res, "not equal to", expected)
}
res = hex.EncodeToString(nla.NTOWFv2("user", "pwd", "dom"))
expected = "652feb8208b3a8a6264c9c5d5b820979"
if res != expected {
t.Error(res, "not equal to", expected)
}
}
func TestRC4K(t *testing.T) {
key, _ := hex.DecodeString("55638e834ce774c100637f197bc0683f")
src, _ := hex.DecodeString("177d16086dd3f06fa8d594e3bad005b7")
res := hex.EncodeToString(nla.RC4K(key, src))
expected := "f5ab375222707a492bd5a90705d96d1d"
if res != expected {
t.Error(res, "not equal to", expected)
}
}

1
grdp/protocol/nla/ntlm.go
View File

@@ -10,6 +10,7 @@ import (
"ShotRDP/grdp/core"
"ShotRDP/grdp/glog"
"github.com/lunixbochs/struc"
)

62
grdp/protocol/nla/ntlm_test.go
View File

@@ -1,62 +0,0 @@
package nla_test
import (
"bytes"
"encoding/hex"
"testing"
"ShotRDP/grdp/protocol/nla"
"github.com/lunixbochs/struc"
)
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.ComputeResponse(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")
}
}
func TestSIGNKEY(t *testing.T) {
exportedSessionKey, _ := hex.DecodeString("be32c3c56ea6683200a35329d67880c3")
result := hex.EncodeToString(nla.SIGNKEY(exportedSessionKey, true))
expected := "79b4f9a4113230f378a0af99f784adae"
if result != expected {
t.Error(result, "not equal to", expected)
}
}