// Copyright 2019 The ChromiumOS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package main

import (
	"testing"
)

func TestAddPieFlags(t *testing.T) {
	withTestContext(t, func(ctx *testContext) {
		initPieConfig(ctx.cfg)
		cmd := ctx.must(callCompiler(ctx, ctx.cfg,
			ctx.newCommand(gccX86_64, mainCc)))
		if err := verifyArgOrder(cmd, "-pie", mainCc); err != nil {
			t.Error(err)
		}
		if err := verifyArgOrder(cmd, "-fPIE", mainCc); err != nil {
			t.Error(err)
		}
	})
}

func TestOmitPieFlagsWhenNoPieArgGiven(t *testing.T) {
	withTestContext(t, func(ctx *testContext) {
		initPieConfig(ctx.cfg)
		cmd := ctx.must(callCompiler(ctx, ctx.cfg,
			ctx.newCommand(gccX86_64, "-nopie", mainCc)))
		if err := verifyArgCount(cmd, 0, "-nopie"); err != nil {
			t.Error(err)
		}
		if err := verifyArgCount(cmd, 0, "-pie"); err != nil {
			t.Error(err)
		}
		if err := verifyArgCount(cmd, 0, "-fPIE"); err != nil {
			t.Error(err)
		}

		cmd = ctx.must(callCompiler(ctx, ctx.cfg,
			ctx.newCommand(gccX86_64, "-fno-pie", mainCc)))
		if err := verifyArgCount(cmd, 0, "-pie"); err != nil {
			t.Error(err)
		}
		if err := verifyArgCount(cmd, 0, "-fPIE"); err != nil {
			t.Error(err)
		}
	})
}

func TestOmitPieFlagsWhenKernelDefined(t *testing.T) {
	withTestContext(t, func(ctx *testContext) {
		initPieConfig(ctx.cfg)
		cmd := ctx.must(callCompiler(ctx, ctx.cfg,
			ctx.newCommand(gccX86_64, "-D__KERNEL__", mainCc)))
		if err := verifyArgCount(cmd, 0, "-pie"); err != nil {
			t.Error(err)
		}
		if err := verifyArgCount(cmd, 0, "-fPIE"); err != nil {
			t.Error(err)
		}
	})
}

func TestAddPieFlagsForEabiEvenIfNoPieGiven(t *testing.T) {
	withTestContext(t, func(ctx *testContext) {
		initPieConfig(ctx.cfg)
		cmd := ctx.must(callCompiler(ctx, ctx.cfg,
			ctx.newCommand(gccX86_64Eabi, "-nopie", mainCc)))
		if err := verifyArgCount(cmd, 0, "-nopie"); err != nil {
			t.Error(err)
		}
		if err := verifyArgCount(cmd, 1, "-pie"); err != nil {
			t.Error(err)
		}
		if err := verifyArgCount(cmd, 1, "-fPIE"); err != nil {
			t.Error(err)
		}
	})
}

func initPieConfig(cfg *config) {
	cfg.commonFlags = []string{"-fPIE", "-pie"}
}