Merge changes Ie28ff622,I0179b42e
* changes:
init: accept -1 or 'unlimited' for an infinite rlimit
Fix two ParseInt edge cases
diff --git a/base/include/android-base/parseint.h b/base/include/android-base/parseint.h
index 55f1ed3..5bfa47e 100644
--- a/base/include/android-base/parseint.h
+++ b/base/include/android-base/parseint.h
@@ -33,6 +33,14 @@
template <typename T>
bool ParseUint(const char* s, T* out, T max = std::numeric_limits<T>::max(),
bool allow_suffixes = false) {
+ while (isspace(*s)) {
+ s++;
+ }
+
+ if (s[0] == '-') {
+ return false;
+ }
+
int base = (s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) ? 16 : 10;
errno = 0;
char* end;
@@ -79,6 +87,10 @@
bool ParseInt(const char* s, T* out,
T min = std::numeric_limits<T>::min(),
T max = std::numeric_limits<T>::max()) {
+ while (isspace(*s)) {
+ s++;
+ }
+
int base = (s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) ? 16 : 10;
errno = 0;
char* end;
diff --git a/base/parseint_test.cpp b/base/parseint_test.cpp
index 8f9ed77..4afa32e 100644
--- a/base/parseint_test.cpp
+++ b/base/parseint_test.cpp
@@ -25,8 +25,14 @@
ASSERT_TRUE(android::base::ParseInt("123", &i));
ASSERT_EQ(123, i);
+ i = 0;
+ EXPECT_TRUE(android::base::ParseInt(" 123", &i));
+ EXPECT_EQ(123, i);
ASSERT_TRUE(android::base::ParseInt("-123", &i));
ASSERT_EQ(-123, i);
+ i = 0;
+ EXPECT_TRUE(android::base::ParseInt(" -123", &i));
+ EXPECT_EQ(-123, i);
short s = 0;
ASSERT_TRUE(android::base::ParseInt("1234", &s));
@@ -49,7 +55,11 @@
ASSERT_TRUE(android::base::ParseUint("123", &i));
ASSERT_EQ(123u, i);
+ i = 0u;
+ EXPECT_TRUE(android::base::ParseUint(" 123", &i));
+ EXPECT_EQ(123u, i);
ASSERT_FALSE(android::base::ParseUint("-123", &i));
+ EXPECT_FALSE(android::base::ParseUint(" -123", &i));
unsigned short s = 0u;
ASSERT_TRUE(android::base::ParseUint("1234", &s));
@@ -63,6 +73,10 @@
ASSERT_FALSE(android::base::ParseUint<unsigned short>("x", nullptr));
ASSERT_FALSE(android::base::ParseUint<unsigned short>("123x", nullptr));
ASSERT_TRUE(android::base::ParseUint<unsigned short>("1234", nullptr));
+
+ unsigned long long int lli;
+ EXPECT_FALSE(android::base::ParseUint("-123", &lli));
+ EXPECT_FALSE(android::base::ParseUint(" -123", &lli));
}
TEST(parseint, no_implicit_octal) {
@@ -79,10 +93,16 @@
int i = 0;
ASSERT_TRUE(android::base::ParseInt("0x123", &i));
ASSERT_EQ(0x123, i);
+ i = 0;
+ EXPECT_TRUE(android::base::ParseInt(" 0x123", &i));
+ EXPECT_EQ(0x123, i);
unsigned int u = 0u;
ASSERT_TRUE(android::base::ParseUint("0x123", &u));
ASSERT_EQ(0x123u, u);
+ u = 0u;
+ EXPECT_TRUE(android::base::ParseUint(" 0x123", &u));
+ EXPECT_EQ(0x123u, u);
}
TEST(parseint, string) {
diff --git a/init/README.md b/init/README.md
index b0a73b9..b45da21 100644
--- a/init/README.md
+++ b/init/README.md
@@ -506,6 +506,7 @@
_resource_ is best specified using its text representation ('cpu', 'rtio', etc
or 'RLIM_CPU', 'RLIM_RTIO', etc). It also may be specified as the int value
that the resource enum corresponds to.
+ _cur_ and _max_ can be 'unlimited' or '-1' to indicate an infinite rlimit.
`start <service>`
> Start a service running if it is not already running.
diff --git a/init/rlimit_parser.cpp b/init/rlimit_parser.cpp
index fe1d6a7..1e0754a 100644
--- a/init/rlimit_parser.cpp
+++ b/init/rlimit_parser.cpp
@@ -65,12 +65,18 @@
}
rlimit limit;
- if (!ParseUint(args[2], &limit.rlim_cur)) {
+ if (args[2] == "-1" || args[2] == "unlimited") {
+ limit.rlim_cur = RLIM_INFINITY;
+ } else if (!ParseUint(args[2], &limit.rlim_cur)) {
return Error() << "Could not parse soft limit '" << args[2] << "'";
}
- if (!ParseUint(args[3], &limit.rlim_max)) {
+
+ if (args[3] == "-1" || args[3] == "unlimited") {
+ limit.rlim_max = RLIM_INFINITY;
+ } else if (!ParseUint(args[3], &limit.rlim_max)) {
return Error() << "Could not parse hard limit '" << args[3] << "'";
}
+
return {resource, limit};
}
diff --git a/init/rlimit_parser_test.cpp b/init/rlimit_parser_test.cpp
index f3f9eb4..659ba8a 100644
--- a/init/rlimit_parser_test.cpp
+++ b/init/rlimit_parser_test.cpp
@@ -49,58 +49,58 @@
TEST(rlimit, RlimitSuccess) {
const std::vector<std::pair<std::vector<std::string>, std::pair<int, rlimit>>>
- inputs_and_results = {
- {{"cpu", "10", "10"}, {0, {10, 10}}},
- {{"fsize", "10", "10"}, {1, {10, 10}}},
- {{"data", "10", "10"}, {2, {10, 10}}},
- {{"stack", "10", "10"}, {3, {10, 10}}},
- {{"core", "10", "10"}, {4, {10, 10}}},
- {{"rss", "10", "10"}, {5, {10, 10}}},
- {{"nproc", "10", "10"}, {6, {10, 10}}},
- {{"nofile", "10", "10"}, {7, {10, 10}}},
- {{"memlock", "10", "10"}, {8, {10, 10}}},
- {{"as", "10", "10"}, {9, {10, 10}}},
- {{"locks", "10", "10"}, {10, {10, 10}}},
- {{"sigpending", "10", "10"}, {11, {10, 10}}},
- {{"msgqueue", "10", "10"}, {12, {10, 10}}},
- {{"nice", "10", "10"}, {13, {10, 10}}},
- {{"rtprio", "10", "10"}, {14, {10, 10}}},
- {{"rttime", "10", "10"}, {15, {10, 10}}},
+ inputs_and_results = {
+ {{"cpu", "10", "10"}, {0, {10, 10}}},
+ {{"fsize", "10", "10"}, {1, {10, 10}}},
+ {{"data", "10", "10"}, {2, {10, 10}}},
+ {{"stack", "10", "10"}, {3, {10, 10}}},
+ {{"core", "10", "10"}, {4, {10, 10}}},
+ {{"rss", "10", "10"}, {5, {10, 10}}},
+ {{"nproc", "10", "10"}, {6, {10, 10}}},
+ {{"nofile", "10", "10"}, {7, {10, 10}}},
+ {{"memlock", "10", "10"}, {8, {10, 10}}},
+ {{"as", "10", "10"}, {9, {10, 10}}},
+ {{"locks", "10", "10"}, {10, {10, 10}}},
+ {{"sigpending", "10", "10"}, {11, {10, 10}}},
+ {{"msgqueue", "10", "10"}, {12, {10, 10}}},
+ {{"nice", "10", "10"}, {13, {10, 10}}},
+ {{"rtprio", "10", "10"}, {14, {10, 10}}},
+ {{"rttime", "10", "10"}, {15, {10, 10}}},
- {{"RLIM_CPU", "10", "10"}, {0, {10, 10}}},
- {{"RLIM_FSIZE", "10", "10"}, {1, {10, 10}}},
- {{"RLIM_DATA", "10", "10"}, {2, {10, 10}}},
- {{"RLIM_STACK", "10", "10"}, {3, {10, 10}}},
- {{"RLIM_CORE", "10", "10"}, {4, {10, 10}}},
- {{"RLIM_RSS", "10", "10"}, {5, {10, 10}}},
- {{"RLIM_NPROC", "10", "10"}, {6, {10, 10}}},
- {{"RLIM_NOFILE", "10", "10"}, {7, {10, 10}}},
- {{"RLIM_MEMLOCK", "10", "10"}, {8, {10, 10}}},
- {{"RLIM_AS", "10", "10"}, {9, {10, 10}}},
- {{"RLIM_LOCKS", "10", "10"}, {10, {10, 10}}},
- {{"RLIM_SIGPENDING", "10", "10"}, {11, {10, 10}}},
- {{"RLIM_MSGQUEUE", "10", "10"}, {12, {10, 10}}},
- {{"RLIM_NICE", "10", "10"}, {13, {10, 10}}},
- {{"RLIM_RTPRIO", "10", "10"}, {14, {10, 10}}},
- {{"RLIM_RTTIME", "10", "10"}, {15, {10, 10}}},
+ {{"RLIM_CPU", "10", "10"}, {0, {10, 10}}},
+ {{"RLIM_FSIZE", "10", "10"}, {1, {10, 10}}},
+ {{"RLIM_DATA", "10", "10"}, {2, {10, 10}}},
+ {{"RLIM_STACK", "10", "10"}, {3, {10, 10}}},
+ {{"RLIM_CORE", "10", "10"}, {4, {10, 10}}},
+ {{"RLIM_RSS", "10", "10"}, {5, {10, 10}}},
+ {{"RLIM_NPROC", "10", "10"}, {6, {10, 10}}},
+ {{"RLIM_NOFILE", "10", "10"}, {7, {10, 10}}},
+ {{"RLIM_MEMLOCK", "10", "10"}, {8, {10, 10}}},
+ {{"RLIM_AS", "10", "10"}, {9, {10, 10}}},
+ {{"RLIM_LOCKS", "10", "10"}, {10, {10, 10}}},
+ {{"RLIM_SIGPENDING", "10", "10"}, {11, {10, 10}}},
+ {{"RLIM_MSGQUEUE", "10", "10"}, {12, {10, 10}}},
+ {{"RLIM_NICE", "10", "10"}, {13, {10, 10}}},
+ {{"RLIM_RTPRIO", "10", "10"}, {14, {10, 10}}},
+ {{"RLIM_RTTIME", "10", "10"}, {15, {10, 10}}},
- {{"0", "10", "10"}, {0, {10, 10}}},
- {{"1", "10", "10"}, {1, {10, 10}}},
- {{"2", "10", "10"}, {2, {10, 10}}},
- {{"3", "10", "10"}, {3, {10, 10}}},
- {{"4", "10", "10"}, {4, {10, 10}}},
- {{"5", "10", "10"}, {5, {10, 10}}},
- {{"6", "10", "10"}, {6, {10, 10}}},
- {{"7", "10", "10"}, {7, {10, 10}}},
- {{"8", "10", "10"}, {8, {10, 10}}},
- {{"9", "10", "10"}, {9, {10, 10}}},
- {{"10", "10", "10"}, {10, {10, 10}}},
- {{"11", "10", "10"}, {11, {10, 10}}},
- {{"12", "10", "10"}, {12, {10, 10}}},
- {{"13", "10", "10"}, {13, {10, 10}}},
- {{"14", "10", "10"}, {14, {10, 10}}},
- {{"15", "10", "10"}, {15, {10, 10}}},
- };
+ {{"0", "10", "10"}, {0, {10, 10}}},
+ {{"1", "10", "10"}, {1, {10, 10}}},
+ {{"2", "10", "10"}, {2, {10, 10}}},
+ {{"3", "10", "10"}, {3, {10, 10}}},
+ {{"4", "10", "10"}, {4, {10, 10}}},
+ {{"5", "10", "10"}, {5, {10, 10}}},
+ {{"6", "10", "10"}, {6, {10, 10}}},
+ {{"7", "10", "10"}, {7, {10, 10}}},
+ {{"8", "10", "10"}, {8, {10, 10}}},
+ {{"9", "10", "10"}, {9, {10, 10}}},
+ {{"10", "10", "10"}, {10, {10, 10}}},
+ {{"11", "10", "10"}, {11, {10, 10}}},
+ {{"12", "unlimited", "10"}, {12, {RLIM_INFINITY, 10}}},
+ {{"13", "-1", "10"}, {13, {RLIM_INFINITY, 10}}},
+ {{"14", "10", "unlimited"}, {14, {10, RLIM_INFINITY}}},
+ {{"15", "10", "-1"}, {15, {10, RLIM_INFINITY}}},
+ };
for (const auto& [input, expected_result] : inputs_and_results) {
TestRlimitSuccess(input, expected_result);
@@ -109,12 +109,16 @@
TEST(rlimit, RlimitFailure) {
const std::vector<std::pair<std::vector<std::string>, std::string>> inputs_and_results = {
- {{"-4", "10", "10"}, "Resource '-4' below the minimum resource value '0'"},
- {{"100", "10", "10"}, "Resource '100' over the maximum resource value '16'"},
- {{"bad_string", "10", "10"}, "Could not parse resource 'bad_string'"},
- {{"RLIM_", "10", "10"}, "Could not parse resource 'RLIM_'"},
- {{"cpu", "abc", "10"}, "Could not parse soft limit 'abc'"},
- {{"cpu", "10", "abc"}, "Could not parse hard limit 'abc'"},
+ {{"-4", "10", "10"}, "Resource '-4' below the minimum resource value '0'"},
+ {{"100", "10", "10"}, "Resource '100' over the maximum resource value '16'"},
+ {{"bad_string", "10", "10"}, "Could not parse resource 'bad_string'"},
+ {{"RLIM_", "10", "10"}, "Could not parse resource 'RLIM_'"},
+ {{"cpu", "abc", "10"}, "Could not parse soft limit 'abc'"},
+ {{"cpu", "10", "abc"}, "Could not parse hard limit 'abc'"},
+ {{"cpu", "unlimit", "10"}, "Could not parse soft limit 'unlimit'"},
+ {{"cpu", "10", "unlimit"}, "Could not parse hard limit 'unlimit'"},
+ {{"cpu", "-2", "10"}, "Could not parse soft limit '-2'"},
+ {{"cpu", "10", "-2"}, "Could not parse hard limit '-2'"},
};
for (const auto& [input, expected_result] : inputs_and_results) {