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

//go:build linux
// +build linux

package build

import (
	"bytes"
	"os"
	"strings"
	"sync"
	"testing"
	"text/template"

	"github.com/google/syzkaller/pkg/osutil"
)

func TestElfBinarySignature(t *testing.T) {
	t.Parallel()
	enumerateFlags(t, nil, []string{"-g", "-O1", "-O2", "-no-pie", "-static"})
}

func TestQueryLinuxCompiler(t *testing.T) {
	const goodDir = "./testdata/linux_compiler_ok"
	const expectedCompiler = "gcc (Debian 10.2.1-6+build2) 10.2.1 20210110, GNU ld (GNU Binutils for Debian) 2.35.2"
	ret, err := queryLinuxCompiler(goodDir)
	if err != nil {
		t.Fatalf("error: %v", err)
	}
	if ret != expectedCompiler {
		t.Fatalf("got: %T, expected: %T", ret, expectedCompiler)
	}
	const badDir = "./testingData/non_existing_folder"
	_, err = queryLinuxCompiler(badDir)
	if err == nil {
		t.Fatalf("Expected an error, got none")
	}
}

func enumerateFlags(t *testing.T, flags, allFlags []string) {
	if len(allFlags) != 0 {
		enumerateFlags(t, flags, allFlags[1:])
		enumerateFlags(t, append(flags, allFlags[0]), allFlags[1:])
		return
	}
	t.Run(strings.Join(flags, "-"), func(t *testing.T) {
		t.Parallel()
		sign1, sign2 := "", ""
		var wg sync.WaitGroup
		wg.Add(2)
		go func() {
			sign1 = sign(t, flags, false, false)
			wg.Done()
		}()
		go func() {
			sign2 = sign(t, flags, false, true)
			wg.Done()
		}()
		sign3 := sign(t, flags, true, false)
		wg.Wait()
		if sign1 != sign2 {
			t.Errorf("signature has changed after a comment-only change")
		}
		if sign1 == sign3 {
			t.Errorf("signature has not changed after a change")
		}
	})
}

func sign(t *testing.T, flags []string, changed, comment bool) string {
	buf := new(bytes.Buffer)
	if err := srcTemplate.Execute(buf, SrcParams{Changed: changed, Comment: comment}); err != nil {
		t.Fatal(err)
	}
	src := buf.Bytes()
	bin, err := osutil.TempFile("syz-build-test")
	if err != nil {
		t.Fatal(err)
	}
	defer os.Remove(bin)
	cmd := osutil.Command("gcc", append(flags, "-pthread", "-o", bin, "-x", "c", "-")...)
	cmd.Stdin = buf
	out, err := cmd.CombinedOutput()
	if err != nil {
		t.Fatalf("compiler failed: %v\n%s\n\n%s", err, src, out)
	}
	sign, err := elfBinarySignature(bin)
	if err != nil {
		t.Fatal(err)
	}
	return sign
}

type SrcParams struct {
	Changed bool
	Comment bool
}

var srcTemplate = template.Must(template.New("").Parse(`
#include <stdio.h>
#include <pthread.h>

int main() {
	int x = {{if .Changed}}0{{else}}1{{end}};
	{{if .Comment}}
	// Some comment goes here.
	// It affects line numbers in debug info.
	{{end}}
	printf("%d %p\n", x, pthread_create);
}
`))