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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
package smb2
import (
"bytes"
"crypto/aes"
"encoding/hex"
"testing"
"github.com/cloudsoda/go-smb2/internal/crypto/cmac"
. "github.com/cloudsoda/go-smb2/internal/smb2"
)
func TestSign(t *testing.T) {
sessionKey, err := hex.DecodeString("726d4c454e63516446695457664e5042")
if err != nil {
t.Fatal(err)
}
pkt, err := hex.DecodeString("fe534d42400001000000000001007f00090000000000000003000000000000000000000000000000020000007bfba3f4041393e756a048c9092c4e52dc7037190900000048000900a1073005a0030a0100")
if err != nil {
t.Fatal(err)
}
signature, err := hex.DecodeString("041393e756a048c9092c4e52dc703719")
if err != nil {
t.Fatal(err)
}
signingKey := kdf(sessionKey, []byte("SMB2AESCMAC\x00"), []byte("SmbSign\x00"))
ciph, err := aes.NewCipher(signingKey)
if err != nil {
t.Fatal(err)
}
signer := cmac.New(ciph)
p := PacketCodec(pkt)
if !bytes.Equal(p.Signature(), signature) {
t.Error("fail")
}
p.SetSignature(zero[:])
signer.Reset()
signer.Write(pkt)
signer.Sum(pkt[:48])
if !bytes.Equal(p.Signature(), signature) {
t.Error("fail")
}
}
|
