aboutsummaryrefslogtreecommitdiff
path: root/cmd/parser.go
blob: 1b17dce642c3e0de6fad91593c228cd5a7a79039 (plain) (blame)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package main

import (
	"fmt"
	"os"
	"sort"
	"strconv"
	"strings"

	"github.com/pektezol/bitreader"
	"github.com/pektezol/demoparser/pkg/packets"
	"github.com/pektezol/demoparser/pkg/verification"
)

const littleEndian bool = true

func main() {
	fmt.Println("Portal 2 Run Validity Checker Tool")
	fmt.Println()
	if len(os.Args) != 2 {
		panic("specify file in command line arguments")
	}
	files, err := os.ReadDir(os.Args[1])
	if err != nil { // If it's not a directory
		file, err := os.Open(os.Args[1])
		if err != nil {
			panic(err)
		}
		reader := bitreader.NewReader(file, littleEndian)
		demoParserHandler(reader)
		defer file.Close()
		fmt.Println()
		fmt.Println("Total Ticks from SAR:", verification.Ticks)
		fmt.Println()
		if verification.IsContinuous(verification.ServerNumbers) {
			fmt.Println("Server Numbers: VALID")
			fmt.Println(verification.ServerNumbers)
		} else {
			fmt.Println("[!] Server Numbers: NOT VALID")
			fmt.Println(verification.ServerNumbers)
		}
		fmt.Scanln()
		return
	}
	sort.Slice(files, func(i, j int) bool {
		// Extract numeric parts from file names
		numA := extractNumber(files[i].Name())
		numB := extractNumber(files[j].Name())

		// Compare the extracted numbers for sorting
		return numA < numB
	})
	for _, fileinfo := range files { // If it is a directory
		file, err := os.Open(os.Args[1] + fileinfo.Name())
		if err != nil {
			panic(err)
		}
		reader := bitreader.NewReader(file, littleEndian)
		demoParserHandler(reader)
		defer file.Close()
	}
	fmt.Println()
	fmt.Println("Total Ticks from SAR:", verification.Ticks)
	fmt.Println()
	if verification.IsContinuous(verification.ServerNumbers) {
		fmt.Println("Server Numbers: VALID")
		fmt.Println(verification.ServerNumbers)
	} else {
		fmt.Println("[!] Server Numbers: NOT VALID")
		fmt.Println(verification.ServerNumbers)
	}
	fmt.Scanln()
}

func demoParserHandler(reader *bitreader.Reader) {
	packets.ParseHeaders(reader)
	for {
		packet := packets.ParsePackets(reader)
		if packet.PacketType == 7 {
			break
		}
	}
}

func extractNumber(filename string) int {
	// Split the filename by underscores
	parts := strings.Split(filename, "_")

	// Get the last part which should be the number
	lastPart := parts[len(parts)-1]

	// Remove any non-digit characters
	numStr := strings.TrimSuffix(lastPart, ".dem") // Assuming filenames end with ".txt"
	num, _ := strconv.Atoi(numStr)

	return num
}