Ashim Kapoor: handle %i,%f,%d,%x,%X better (#694)
This commit is contained in:
parent
a919f4541a
commit
3da779725d
|
@ -756,6 +756,7 @@ int CheckBufferOverrun::count(const std::string &input_string)
|
|||
int on_on_next = 0;
|
||||
std::string digits_string = "";
|
||||
int digits = 0;
|
||||
int check_for_i_d_x_f = 0;
|
||||
|
||||
for (std::string::size_type i = 0; i != input_string.size(); i++)
|
||||
{
|
||||
|
@ -767,18 +768,20 @@ int CheckBufferOverrun::count(const std::string &input_string)
|
|||
}
|
||||
switch (input_string[i])
|
||||
{
|
||||
|
||||
case 'f':
|
||||
case 'x':
|
||||
case 'X':
|
||||
case 'd':
|
||||
case 'i':
|
||||
check_for_i_d_x_f = 1;
|
||||
case 'c':
|
||||
case 'e':
|
||||
case 'E':
|
||||
case 'f':
|
||||
case 'g':
|
||||
case 'o':
|
||||
case 's':
|
||||
case 'u':
|
||||
case 'x':
|
||||
case 'X':
|
||||
case 'p':
|
||||
case 'n':
|
||||
if (flag == 0) on_on_next = 1;
|
||||
|
@ -797,13 +800,17 @@ int CheckBufferOverrun::count(const std::string &input_string)
|
|||
|
||||
if (on_on_next == 1 && flag == 0)
|
||||
{
|
||||
std::cout << digits_string;
|
||||
|
||||
digits_string = digits_string.substr(1, digits_string.size());
|
||||
digits += abs(atoi(digits_string.c_str()));
|
||||
if (check_for_i_d_x_f == 1) digits += std::max(abs(atoi(digits_string.c_str())), 1);
|
||||
else
|
||||
digits += abs(atoi(digits_string.c_str()));
|
||||
|
||||
digits_string = "";
|
||||
|
||||
check_for_i_d_x_f = 0;
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
|
@ -811,5 +818,3 @@ int CheckBufferOverrun::count(const std::string &input_string)
|
|||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
|
|
@ -790,10 +790,14 @@ private:
|
|||
std::string str1 = "Hello";
|
||||
std::string str2 = "Hello %2.2d %1d";
|
||||
std::string str3 = "Test \" ";
|
||||
std::string str4 = "Try %s";
|
||||
std::string str5 = "%d";
|
||||
|
||||
ASSERT_EQUALS(6, CheckBufferOverrun::count(str1));
|
||||
ASSERT_EQUALS(11, CheckBufferOverrun::count(str2));
|
||||
ASSERT_EQUALS(8, CheckBufferOverrun::count(str3));
|
||||
ASSERT_EQUALS(5, CheckBufferOverrun::count(str4));
|
||||
ASSERT_EQUALS(2, CheckBufferOverrun::count(str5));
|
||||
}
|
||||
|
||||
void strncpy1()
|
||||
|
@ -844,5 +848,3 @@ private:
|
|||
REGISTER_TEST(TestBufferOverrun)
|
||||
|
||||
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue