108 lines
3.4 KiB
C++
108 lines
3.4 KiB
C++
/*
|
|
* Copyright (C) 2015 The Android Open Source Project
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#ifndef ANDROID_BASE_PARSEINT_H
|
|
#define ANDROID_BASE_PARSEINT_H
|
|
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include <limits>
|
|
#include <string>
|
|
|
|
namespace android {
|
|
namespace base {
|
|
|
|
// Parses the unsigned decimal integer in the string 's' and sets 'out' to
|
|
// that value. Optionally allows the caller to define a 'max' beyond which
|
|
// otherwise valid values will be rejected. Returns boolean success; 'out'
|
|
// is untouched if parsing fails.
|
|
template <typename T>
|
|
bool ParseUint(const char* s, T* out, T max = std::numeric_limits<T>::max(),
|
|
bool allow_suffixes = false) {
|
|
int base = (s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) ? 16 : 10;
|
|
errno = 0;
|
|
char* end;
|
|
unsigned long long int result = strtoull(s, &end, base);
|
|
if (errno != 0 || end == s) return false;
|
|
if (*end != '\0') {
|
|
const char* suffixes = "bkmgtpe";
|
|
const char* suffix;
|
|
if (!allow_suffixes || (suffix = strchr(suffixes, tolower(*end))) == nullptr) return false;
|
|
#if __clang__ // TODO: win32 still builds with GCC :-(
|
|
if (__builtin_mul_overflow(result, 1ULL << (10 * (suffix - suffixes)), &result)) return false;
|
|
#endif
|
|
}
|
|
if (max < result) {
|
|
return false;
|
|
}
|
|
*out = static_cast<T>(result);
|
|
return true;
|
|
}
|
|
|
|
// TODO: string_view
|
|
template <typename T>
|
|
bool ParseUint(const std::string& s, T* out, T max = std::numeric_limits<T>::max(),
|
|
bool allow_suffixes = false) {
|
|
return ParseUint(s.c_str(), out, max, allow_suffixes);
|
|
}
|
|
|
|
template <typename T>
|
|
bool ParseByteCount(const char* s, T* out, T max = std::numeric_limits<T>::max()) {
|
|
return ParseUint(s, out, max, true);
|
|
}
|
|
|
|
// TODO: string_view
|
|
template <typename T>
|
|
bool ParseByteCount(const std::string& s, T* out, T max = std::numeric_limits<T>::max()) {
|
|
return ParseByteCount(s.c_str(), out, max);
|
|
}
|
|
|
|
// Parses the signed decimal integer in the string 's' and sets 'out' to
|
|
// that value. Optionally allows the caller to define a 'min' and 'max
|
|
// beyond which otherwise valid values will be rejected. Returns boolean
|
|
// success; 'out' is untouched if parsing fails.
|
|
template <typename T>
|
|
bool ParseInt(const char* s, T* out,
|
|
T min = std::numeric_limits<T>::min(),
|
|
T max = std::numeric_limits<T>::max()) {
|
|
int base = (s[0] == '0' && (s[1] == 'x' || s[1] == 'X')) ? 16 : 10;
|
|
errno = 0;
|
|
char* end;
|
|
long long int result = strtoll(s, &end, base);
|
|
if (errno != 0 || s == end || *end != '\0') {
|
|
return false;
|
|
}
|
|
if (result < min || max < result) {
|
|
return false;
|
|
}
|
|
*out = static_cast<T>(result);
|
|
return true;
|
|
}
|
|
|
|
// TODO: string_view
|
|
template <typename T>
|
|
bool ParseInt(const std::string& s, T* out,
|
|
T min = std::numeric_limits<T>::min(),
|
|
T max = std::numeric_limits<T>::max()) {
|
|
return ParseInt(s.c_str(), out, min, max);
|
|
}
|
|
|
|
} // namespace base
|
|
} // namespace android
|
|
|
|
#endif // ANDROID_BASE_PARSEINT_H
|