main_test.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestRiscvTests(t *testing.T) {
	files, err := ioutil.ReadDir("testdata")
	assert.NoError(t, err)
	for _, file := range files {
		if strings.HasSuffix(file.Name(), ".dump") {
			continue
		}
		// if file.Name() != "rv64uf-p-fcvt_w" {
		// 	continue
		// }
		// var err error
		// debugFile, err = os.Create("trace.txt")
		// assert.NoError(t, err)
		// defer debugFile.Close()
		// DEBUG = true
		t.Run(file.Name(), func(t *testing.T) {
			mem := make([]byte, 0x10000)
			entry, err := loadElf(filepath.Join("testdata", file.Name()), mem)
			assert.NoError(t, err)
			cpu := NewCPU(mem, entry, nil, nil)
			for {
				cpu.step()
				exitcode := mem[0x1000]
				if exitcode != 0 {
					assert.Equal(t, uint8(1), exitcode, "failing exitcode %d recieved", exitcode)
					return
				}
			}
		})
	}
}

func TestLinux(t *testing.T) {
	var err error
	debugFile, err = os.Create("trace.txt")
	assert.NoError(t, err)
	defer debugFile.Close()
	DEBUG = false

	mem := make([]byte, 0x10000000)
	entry, err := loadElf(filepath.Join("linux", "fw_payload.elf"), mem)
	assert.NoError(t, err)
	rootfs, err := ioutil.ReadFile(filepath.Join("linux", "rootfs.img"))
	assert.NoError(t, err)
	cpu := NewCPU(mem, entry, rootfs, nil)
	start := time.Now()

	defer func() {
		end := time.Now()
		err := recover()
		fmt.Printf("finished at cycle: %d -- pc==%x\n", cpu.count, cpu.pc)
		fmt.Printf("failed with: %v\n", err)
		fmt.Printf("%0.4f MHz\n", float64(cpu.count*1000)/float64(end.UnixNano()-start.UnixNano()))

		assert.GreaterOrEqual(t, cpu.count, uint64(128439363))
	}()

	for {
		cpu.step()
		if cpu.wfi {
			return
		}
	}
}