Skip to content

Commit a9e6ec8

Browse files
authored
examples: add encrypted-net-messages example (#332)
1 parent 3aaec4c commit a9e6ec8

File tree

3 files changed

+74
-1
lines changed

3 files changed

+74
-1
lines changed
Lines changed: 56 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,56 @@
1+
package main
2+
3+
import (
4+
"flag"
5+
"io/ioutil"
6+
"log"
7+
"os"
8+
9+
dem "github.com/markus-wa/demoinfocs-golang/v2/pkg/demoinfocs"
10+
"github.com/markus-wa/demoinfocs-golang/v2/pkg/demoinfocs/events"
11+
)
12+
13+
func checkErr(err error) {
14+
if err != nil {
15+
panic(err)
16+
}
17+
}
18+
19+
func main() {
20+
fl := new(flag.FlagSet)
21+
22+
demPathPtr := fl.String("demo", "", "Demo file `path`")
23+
infoPathPtr := fl.String("info", "", "Info file `path`")
24+
25+
err := fl.Parse(os.Args[1:])
26+
checkErr(err)
27+
28+
demPath := *demPathPtr
29+
infoPath := *infoPathPtr
30+
31+
infoF, err := os.Open(infoPath)
32+
checkErr(err)
33+
34+
b, err := ioutil.ReadAll(infoF)
35+
checkErr(err)
36+
37+
k, err := dem.MatchInfoDecryptionKey(b)
38+
checkErr(err)
39+
40+
f, err := os.Open(demPath)
41+
checkErr(err)
42+
43+
defer f.Close()
44+
45+
cfg := dem.DefaultParserConfig
46+
cfg.NetMessageDecryptionKey = k
47+
48+
p := dem.NewParserWithConfig(f, cfg)
49+
50+
p.RegisterEventHandler(func(message events.ChatMessage) {
51+
log.Println(message)
52+
})
53+
54+
err = p.ParseToEnd()
55+
checkErr(err)
56+
}
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
package main
2+
3+
import (
4+
"os"
5+
"testing"
6+
)
7+
8+
// Just make sure the example runs
9+
func TestEncryptedNetMessages(t *testing.T) {
10+
if testing.Short() {
11+
t.Skip("skipping test")
12+
}
13+
14+
os.Args = []string{"cmd", "-demo", "../../test/cs-demos/match730_003528806449641685104_1453182610_271.dem", "-info", "../../test/cs-demos/match730_003528806449641685104_1453182610_271.dem.info"}
15+
16+
main()
17+
}

test/cs-demos

Submodule cs-demos updated from a335c3d to 6ec2aac

0 commit comments

Comments
 (0)