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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// Copyright 2019 The Chromium OS 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 TestRealConfigWithUseCCacheFlag(t *testing.T) {
resetGlobals()
defer resetGlobals()
ConfigName = "cros.hardened"
UseCCache = "false"
cfg, err := getRealConfig()
if err != nil {
t.Fatal(err)
}
if cfg.useCCache {
t.Fatal("UseCCache: Expected false got true")
}
UseCCache = "true"
cfg, err = getRealConfig()
if err != nil {
t.Fatal(err)
}
if !cfg.useCCache {
t.Fatal("UseCCache: Expected true got false")
}
UseCCache = "invalid"
_, err = getRealConfig()
if err == nil {
t.Fatalf("UseCCache: Expected an error, got none")
}
}
func TestRealConfigWithConfigNameFlag(t *testing.T) {
resetGlobals()
defer resetGlobals()
UseCCache = "false"
ConfigName = "cros.hardened"
cfg, err := getRealConfig()
if err != nil {
t.Fatal(err)
}
if !isSysrootHardened(cfg) || cfg.isHostWrapper {
t.Fatalf("ConfigName: Expected sysroot hardened config. Got: %#v", cfg)
}
ConfigName = "cros.nonhardened"
cfg, err = getRealConfig()
if err != nil {
t.Fatal(err)
}
if isSysrootHardened(cfg) || cfg.isHostWrapper {
t.Fatalf("ConfigName: Expected sysroot non hardened config. Got: %#v", cfg)
}
ConfigName = "cros.host"
cfg, err = getRealConfig()
if err != nil {
t.Fatal(err)
}
if !cfg.isHostWrapper {
t.Fatalf("ConfigName: Expected clang host config. Got: %#v", cfg)
}
ConfigName = "invalid"
_, err = getRealConfig()
if err == nil {
t.Fatalf("ConfigName: Expected an error, got none")
}
}
func TestRealConfigWithOldWrapperPath(t *testing.T) {
resetGlobals()
defer resetGlobals()
UseCCache = "false"
ConfigName = "cros.hardened"
OldWrapperPath = "somepath"
cfg, err := getRealConfig()
if err != nil {
t.Fatal(err)
}
if cfg.oldWrapperPath != "somepath" {
t.Fatalf("OldWrapperPath: Expected somepath, got %s", cfg.oldWrapperPath)
}
}
func isSysrootHardened(cfg *config) bool {
for _, arg := range cfg.commonFlags {
if arg == "-pie" {
return true
}
}
return false
}
func resetGlobals() {
// Set all global variables to a defined state.
OldWrapperPath = ""
ConfigName = "unknown"
UseCCache = "unknown"
}
|
