1//===-- sanitizer_flags_test.cc -------------------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file is a part of ThreadSanitizer/AddressSanitizer runtime.
11//
12//===----------------------------------------------------------------------===//
13#include "sanitizer_common/sanitizer_common.h"
14#include "sanitizer_common/sanitizer_flags.h"
15#include "sanitizer_common/sanitizer_libc.h"
16#include "gtest/gtest.h"
17
18#include <string.h>
19
20namespace __sanitizer {
21
22static const char kFlagName[] = "flag_name";
23
24template <typename T>
25static void TestFlag(T start_value, const char *env, T final_value) {
26  T flag = start_value;
27  ParseFlag(env, &flag, kFlagName, "flag description");
28  EXPECT_EQ(final_value, flag);
29}
30
31static void TestStrFlag(const char *start_value, const char *env,
32                        const char *final_value) {
33  const char *flag = start_value;
34  ParseFlag(env, &flag, kFlagName, "flag description");
35  EXPECT_EQ(0, internal_strcmp(final_value, flag));
36}
37
38TEST(SanitizerCommon, BooleanFlags) {
39  TestFlag(true, "--flag_name", true);
40  TestFlag(false, "flag_name", false);
41  TestFlag(false, "--flag_name=1", true);
42  TestFlag(true, "asdas flag_name=0 asdas", false);
43  TestFlag(true, "    --flag_name=0   ", false);
44  TestFlag(false, "flag_name=yes", true);
45  TestFlag(false, "flag_name=true", true);
46  TestFlag(true, "flag_name=no", false);
47  TestFlag(true, "flag_name=false", false);
48}
49
50TEST(SanitizerCommon, IntFlags) {
51  TestFlag(-11, 0, -11);
52  TestFlag(-11, "flag_name", -11);
53  TestFlag(-11, "--flag_name=", 0);
54  TestFlag(-11, "--flag_name=42", 42);
55  TestFlag(-11, "--flag_name=-42", -42);
56}
57
58TEST(SanitizerCommon, StrFlags) {
59  TestStrFlag("zzz", 0, "zzz");
60  TestStrFlag("zzz", "flag_name", "zzz");
61  TestStrFlag("zzz", "--flag_name=", "");
62  TestStrFlag("", "--flag_name=abc", "abc");
63  TestStrFlag("", "--flag_name='abc zxc'", "abc zxc");
64  TestStrFlag("", "--flag_name='abc zxcc'", "abc zxcc");
65  TestStrFlag("", "--flag_name=\"abc qwe\" asd", "abc qwe");
66  TestStrFlag("", "other_flag_name=zzz", "");
67}
68
69static void TestTwoFlags(const char *env, bool expected_flag1,
70                         const char *expected_flag2) {
71  bool flag1 = !expected_flag1;
72  const char *flag2 = "";
73  ParseFlag(env, &flag1, "flag1", "flag1 description");
74  ParseFlag(env, &flag2, "flag2", "flag2 description");
75  EXPECT_EQ(expected_flag1, flag1);
76  EXPECT_EQ(0, internal_strcmp(flag2, expected_flag2));
77}
78
79TEST(SanitizerCommon, MultipleFlags) {
80  TestTwoFlags("flag1=1 flag2='zzz'", true, "zzz");
81  TestTwoFlags("flag2='qxx' flag1=0", false, "qxx");
82  TestTwoFlags("flag1=false:flag2='zzz'", false, "zzz");
83  TestTwoFlags("flag2=qxx:flag1=yes", true, "qxx");
84}
85
86}  // namespace __sanitizer
87