Home | History | Annotate | Download | only in tests
      1 //===-- sanitizer_printf_test.cc ------------------------------------------===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 //
     10 // Tests for sanitizer_printf.cc
     11 //
     12 //===----------------------------------------------------------------------===//
     13 #include "sanitizer_common/sanitizer_common.h"
     14 #include "sanitizer_common/sanitizer_libc.h"
     15 #include "gtest/gtest.h"
     16 
     17 #include <string.h>
     18 #include <limits.h>
     19 
     20 namespace __sanitizer {
     21 
     22 TEST(Printf, Basic) {
     23   char buf[1024];
     24   uptr len = internal_snprintf(buf, sizeof(buf),
     25       "a%db%zdc%ue%zuf%xh%zxq%pe%sr",
     26       (int)-1, (long)-2, // NOLINT
     27       (unsigned)-4, (unsigned long)5, // NOLINT
     28       (unsigned)10, (unsigned long)11, // NOLINT
     29       (void*)0x123, "_string_");
     30   EXPECT_EQ(len, strlen(buf));
     31   void *ptr;
     32   if (sizeof(ptr) == 4) {
     33     EXPECT_STREQ("a-1b-2c4294967292e5fahbq"
     34                  "0x00000123e_string_r", buf);
     35   } else {
     36     EXPECT_STREQ("a-1b-2c4294967292e5fahbq"
     37                  "0x000000000123e_string_r", buf);
     38   }
     39 }
     40 
     41 TEST(Printf, OverflowStr) {
     42   char buf[] = "123456789";
     43   uptr len = internal_snprintf(buf, 4, "%s", "abcdef");  // NOLINT
     44   EXPECT_EQ(len, (uptr)6);
     45   EXPECT_STREQ("abc", buf);
     46   EXPECT_EQ(buf[3], 0);
     47   EXPECT_EQ(buf[4], '5');
     48   EXPECT_EQ(buf[5], '6');
     49   EXPECT_EQ(buf[6], '7');
     50   EXPECT_EQ(buf[7], '8');
     51   EXPECT_EQ(buf[8], '9');
     52   EXPECT_EQ(buf[9], 0);
     53 }
     54 
     55 TEST(Printf, OverflowInt) {
     56   char buf[] = "123456789";
     57   internal_snprintf(buf, 4, "%d", -123456789);  // NOLINT
     58   EXPECT_STREQ("-12", buf);
     59   EXPECT_EQ(buf[3], 0);
     60   EXPECT_EQ(buf[4], '5');
     61   EXPECT_EQ(buf[5], '6');
     62   EXPECT_EQ(buf[6], '7');
     63   EXPECT_EQ(buf[7], '8');
     64   EXPECT_EQ(buf[8], '9');
     65   EXPECT_EQ(buf[9], 0);
     66 }
     67 
     68 TEST(Printf, OverflowUint) {
     69   char buf[] = "123456789";
     70   uptr val;
     71   if (sizeof(val) == 4) {
     72     val = (uptr)0x12345678;
     73   } else {
     74     val = (uptr)0x123456789ULL;
     75   }
     76   internal_snprintf(buf, 4, "a%zx", val);  // NOLINT
     77   EXPECT_STREQ("a12", buf);
     78   EXPECT_EQ(buf[3], 0);
     79   EXPECT_EQ(buf[4], '5');
     80   EXPECT_EQ(buf[5], '6');
     81   EXPECT_EQ(buf[6], '7');
     82   EXPECT_EQ(buf[7], '8');
     83   EXPECT_EQ(buf[8], '9');
     84   EXPECT_EQ(buf[9], 0);
     85 }
     86 
     87 TEST(Printf, OverflowPtr) {
     88   char buf[] = "123456789";
     89   void *p;
     90   if (sizeof(p) == 4) {
     91     p = (void*)0x1234567;
     92   } else {
     93     p = (void*)0x123456789ULL;
     94   }
     95   internal_snprintf(buf, 4, "%p", p);  // NOLINT
     96   EXPECT_STREQ("0x0", buf);
     97   EXPECT_EQ(buf[3], 0);
     98   EXPECT_EQ(buf[4], '5');
     99   EXPECT_EQ(buf[5], '6');
    100   EXPECT_EQ(buf[6], '7');
    101   EXPECT_EQ(buf[7], '8');
    102   EXPECT_EQ(buf[8], '9');
    103   EXPECT_EQ(buf[9], 0);
    104 }
    105 
    106 template<typename T>
    107 static void TestMinMax(const char *fmt, T min, T max) {
    108   char buf[1024];
    109   uptr len = internal_snprintf(buf, sizeof(buf), fmt, min, max);
    110   char buf2[1024];
    111   snprintf(buf2, sizeof(buf2), fmt, min, max);
    112   EXPECT_EQ(len, strlen(buf));
    113   EXPECT_STREQ(buf2, buf);
    114 }
    115 
    116 TEST(Printf, MinMax) {
    117   TestMinMax<int>("%d-%d", INT_MIN, INT_MAX);  // NOLINT
    118   TestMinMax<long>("%zd-%zd", LONG_MIN, LONG_MAX);  // NOLINT
    119   TestMinMax<unsigned>("%u-%u", 0, UINT_MAX);  // NOLINT
    120   TestMinMax<unsigned long>("%zu-%zu", 0, ULONG_MAX);  // NOLINT
    121   TestMinMax<unsigned>("%x-%x", 0, UINT_MAX);  // NOLINT
    122   TestMinMax<unsigned long>("%zx-%zx", 0, ULONG_MAX);  // NOLINT
    123 }
    124 
    125 }  // namespace __sanitizer
    126