summaryrefslogtreecommitdiffstats
path: root/lib/sanitizer_common/sanitizer_flags.cc
blob: cdeeb78d7a5d5b99b4f7c58d36ecec965e5dad7e (plain)
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
//===-- sanitizer_flags.cc ------------------------------------------------===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// This file is a part of ThreadSanitizer/AddressSanitizer runtime.
//
//===----------------------------------------------------------------------===//

#include "sanitizer_flags.h"

#include "sanitizer_common.h"
#include "sanitizer_libc.h"

namespace __sanitizer {

static char *GetFlagValue(const char *env, const char *name) {
  if (env == 0)
    return 0;
  const char *pos = internal_strstr(env, name);
  const char *end;
  if (pos == 0)
    return 0;
  pos += internal_strlen(name);
  if (pos[0] != '=') {
    end = pos;
  } else {
    pos += 1;
    if (pos[0] == '"') {
      pos += 1;
      end = internal_strchr(pos, '"');
    } else if (pos[0] == '\'') {
      pos += 1;
      end = internal_strchr(pos, '\'');
    } else {
      end = internal_strchr(pos, ' ');
    }
    if (end == 0)
      end = pos + internal_strlen(pos);
  }
  int len = end - pos;
  char *f = (char*)InternalAlloc(len + 1);
  internal_memcpy(f, pos, len);
  f[len] = '\0';
  return f;
}

void ParseFlag(const char *env, bool *flag, const char *name) {
  char *val = GetFlagValue(env, name);
  if (val == 0)
    return;
  if (0 == internal_strcmp(val, "0") ||
      0 == internal_strcmp(val, "no") ||
      0 == internal_strcmp(val, "false"))
    *flag = false;
  if (0 == internal_strcmp(val, "1") ||
      0 == internal_strcmp(val, "yes") ||
      0 == internal_strcmp(val, "true"))
    *flag = true;
  InternalFree(val);
}

void ParseFlag(const char *env, int *flag, const char *name) {
  char *val = GetFlagValue(env, name);
  if (val == 0)
    return;
  *flag = internal_atoll(val);
  InternalFree(val);
}

void ParseFlag(const char *env, const char **flag, const char *name) {
  const char *val = GetFlagValue(env, name);
  if (val == 0)
    return;
  *flag = val;
}

}  // namespace __sanitizer
OpenPOWER on IntegriCloud