about summary refs log tree commit diff
path: root/src/r2pipe.go
blob: 013478aed5c92155e1c893a0f37b9678e3105f11 (plain)
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
package main

import (
	"fmt"
	"log"
	"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)

		// 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))

		// 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 {

		// clear the screen
		registers := stepIn(r2p)

		fmt.Println(user(r2p, i, registers, config))

		i = switchPlayer(i, config)

		time.Sleep(2 * time.Second)
	}

	r2p.Close()
}