// Copyright 2019 syzkaller project authors. All rights reserved.
// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.

// syz-showprio visualizes the call to call priorities from the prog package.
package main

import (
	"flag"
	"fmt"
	"os"
	"runtime"
	"strings"

	"github.com/google/syzkaller/pkg/db"
	"github.com/google/syzkaller/pkg/mgrconfig"
	"github.com/google/syzkaller/prog"
)

var (
	flagOS     = flag.String("os", runtime.GOOS, "target os")
	flagArch   = flag.String("arch", runtime.GOARCH, "target arch")
	flagEnable = flag.String("enable", "", "comma-separated list of enabled syscalls")
	flagCorpus = flag.String("corpus", "", "name of the corpus file")
)

func main() {
	flag.Parse()
	target, err := prog.GetTarget(*flagOS, *flagArch)
	if err != nil {
		fmt.Fprintf(os.Stderr, "%v", err)
		os.Exit(1)
	}
	if *flagEnable == "" {
		fmt.Fprintf(os.Stderr, "no syscalls enabled")
		os.Exit(1)
	}
	enabled := strings.Split(*flagEnable, ",")
	_, err = mgrconfig.ParseEnabledSyscalls(target, enabled, nil)
	if err != nil {
		fmt.Fprintf(os.Stderr, "failed to parse enabled syscalls: %v", err)
		os.Exit(1)
	}
	corpus, err := db.ReadCorpus(*flagCorpus, target)
	if err != nil {
		fmt.Fprintf(os.Stderr, "failed to read corpus: %v", err)
		os.Exit(1)
	}
	showPriorities(enabled, target.CalculatePriorities(corpus), target)
}

func showPriorities(calls []string, prios [][]float32, target *prog.Target) {
	printLine(append([]string{"CALLS"}, calls...))
	for _, callRow := range calls {
		line := []string{callRow}
		for _, callCol := range calls {
			val := prios[target.SyscallMap[callRow].ID][target.SyscallMap[callCol].ID]
			line = append(line, fmt.Sprintf("%.2f", val))
		}
		printLine(line)
	}
}

func printLine(values []string) {
	fmt.Printf("|")
	for _, val := range values {
		fmt.Printf("%-20v|", val)
	}
	fmt.Printf("\n")
}