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
|
package main
import (
"fmt"
"log"
"strings"
"time"
)
var (
verbose *bool
)
func main() {
log.Println("---")
log.Println("[i] Parse the config")
config := parseConfig()
// build the bots
log.Println("---")
log.Println("[i] Build the bots")
for i := 0; i < 2; i++ {
bot := buildBot(config, "bots/warrior.asm")
config.Bots = append(config.Bots, bot)
}
// bot2 := buildBot(config, "bots/warrior.asm")
// config.Bots = append(config.Bots, bot2)
// bot3 := buildBot(config, "bots/warrior.asm")
// config.Bots = append(config.Bots, bot3)
// initialize the arena
log.Println("---")
log.Println("[i] Initialize the Arena")
r2p := initArena(config)
randomOffsets := getRandomOffsets(config)
// place each bot in the arena
log.Println("---")
log.Println("[i] Place the bots")
for bot := 0; bot < len(config.Bots); bot++ {
// Place the bot in the arena
log.Printf("[i] Placing bot %d", bot)
address := randomOffsets[bot]
placeBot(r2p, config.Bots[bot], address)
// store the initial address of the bot int the struct field
config.Bots[bot].Addr = address
// define the instruction point and the stack pointer
log.Printf("[i] setting up the PC and SP for bot %d", bot)
_ = r2cmd(r2p, fmt.Sprintf("aer PC=%d", address))
_ = r2cmd(r2p, fmt.Sprintf("aer SP=SP+%d", address))
// dump the registers of the user for being able to switch inbetween them
initialRegisers := strings.Replace(r2cmd(r2p, "aerR"), "\n", ";", -1)
config.Bots[bot].Regs = initialRegisers
// print the instruction point and the stack pointer
botStackPointer := r2cmd(r2p, "aerR~esp[2]")
log.Printf("[i] bot %d esp = %s", bot, botStackPointer)
botInstructionPointer := r2cmd(r2p, "aerR~eip[2]")
log.Printf("[i] bot %d eip = %s", bot, botInstructionPointer)
}
// handle errors in esil
_ = r2cmd(r2p, "e cmd.esil.todo=f theend=1")
_ = r2cmd(r2p, "e cmd.esil.trap=f theend=1")
_ = r2cmd(r2p, "e cmd.esil.intr=f theend=1")
_ = r2cmd(r2p, "e cmd.esil.ioer=f theend=1")
_ = r2cmd(r2p, "f theend=0")
fmt.Println(r2cmd(r2p, fmt.Sprintf("b %d", config.Memsize)))
// start the competition
i := 0
for true {
// Step, then print the users registers
registers := stepIn(r2p)
config.Bots[i].Regs = registers
fmt.Println(user(r2p, i, registers, config))
// switch players
i = switchPlayer(r2p, i, config)
// sleepti
time.Sleep(100 * time.Millisecond)
}
r2p.Close()
}
|
