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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/* Copyright 2013 10gen Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "mongo/util/options_parser/constraints.h"
#include <pcrecpp.h>
#include "mongo/base/status.h"
#include "mongo/bson/util/builder.h"
namespace mongo {
namespace optionenvironment {
Status NumericKeyConstraint::check(const Environment& env) {
Value val;
Status s = env.get(_key, &val);
if (s == ErrorCodes::NoSuchKey) {
// Key not set, skipping numeric constraint check
return Status::OK();
}
// The code that controls whether a type is "compatible" is contained in the Value
// class, so if that handles compatibility between numeric types then this will too.
long intVal;
if (val.get(&intVal).isOK()) {
if (intVal < _min || intVal > _max) {
StringBuilder sb;
sb << "Error: Attempting to set " << _key << " to value: " <<
intVal << " which is out of range: (" <<
_min << "," << _max << ")";
return Status(ErrorCodes::BadValue, sb.str());
}
}
else {
StringBuilder sb;
sb << "Error: " << _key << " is of type: " << val.typeToString() <<
" but must be of a numeric type.";
return Status(ErrorCodes::BadValue, sb.str());
}
return Status::OK();
}
Status ImmutableKeyConstraint::check(const Environment& env) {
Value env_value;
Status ret = env.get(_key, &env_value);
if (ret.isOK()) {
if (_value.isEmpty()) {
_value = env_value;
}
else {
if (!_value.equal(env_value)) {
StringBuilder sb;
sb << "Error: " << _key << " is immutable once set";
return Status(ErrorCodes::BadValue, sb.str());
}
}
}
return ret;
}
Status MutuallyExclusiveKeyConstraint::check(const Environment& env) {
Value env_value;
Status ret = env.get(_key, &env_value);
if (ret.isOK()) {
ret = env.get(_otherKey, &env_value);
if (ret.isOK()) {
StringBuilder sb;
sb << _otherKey << " is not allowed when " << _key << " is specified";
return Status(ErrorCodes::BadValue, sb.str());
}
}
return Status::OK();
}
Status RequiresOtherKeyConstraint::check(const Environment& env) {
Value env_value;
Status ret = env.get(_key, &env_value);
if (ret.isOK()) {
ret = env.get(_otherKey, &env_value);
if (!ret.isOK()) {
StringBuilder sb;
sb << _otherKey << " is required when " << _key << " is specified";
return Status(ErrorCodes::BadValue, sb.str());
}
}
return Status::OK();
}
Status StringFormatKeyConstraint::check(const Environment& env) {
Value value;
Status ret = env.get(_key, &value);
if (ret.isOK()) {
std::string stringVal;
ret = value.get(&stringVal);
if (!ret.isOK()) {
StringBuilder sb;
sb << _key << " could not be read as a string: " << ret.reason();
return Status(ErrorCodes::BadValue, sb.str());
}
pcrecpp::RE re(_regexFormat);
if (!re.FullMatch(stringVal)) {
StringBuilder sb;
sb << _key << " must be a string of the format: " << _displayFormat;
return Status(ErrorCodes::BadValue, sb.str());
}
}
return Status::OK();
}
} // namespace optionenvironment
} // namespace mongo
|