1 //===-- sanitizer_libc_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 // Tests for sanitizer_libc.h.
10 //===----------------------------------------------------------------------===//
11
12 #include "sanitizer_common/sanitizer_common.h"
13 #include "sanitizer_common/sanitizer_libc.h"
14 #include "sanitizer_common/sanitizer_platform.h"
15 #include "gtest/gtest.h"
16
17 #if SANITIZER_LINUX || SANITIZER_MAC
18 # define SANITIZER_TEST_HAS_STAT_H 1
19 # include <sys/stat.h>
20 #else
21 # define SANITIZER_TEST_HAS_STAT_H 0
22 #endif
23
24 // A regression test for internal_memmove() implementation.
TEST(SanitizerCommon,InternalMemmoveRegression)25 TEST(SanitizerCommon, InternalMemmoveRegression) {
26 char src[] = "Hello World";
27 char *dest = src + 6;
28 __sanitizer::internal_memmove(dest, src, 5);
29 EXPECT_EQ(dest[0], src[0]);
30 EXPECT_EQ(dest[4], src[4]);
31 }
32
TEST(SanitizerCommon,mem_is_zero)33 TEST(SanitizerCommon, mem_is_zero) {
34 size_t size = 128;
35 char *x = new char[size];
36 memset(x, 0, size);
37 for (size_t pos = 0; pos < size; pos++) {
38 x[pos] = 1;
39 for (size_t beg = 0; beg < size; beg++) {
40 for (size_t end = beg; end < size; end++) {
41 // fprintf(stderr, "pos %zd beg %zd end %zd \n", pos, beg, end);
42 if (beg <= pos && pos < end)
43 EXPECT_FALSE(__sanitizer::mem_is_zero(x + beg, end - beg));
44 else
45 EXPECT_TRUE(__sanitizer::mem_is_zero(x + beg, end - beg));
46 }
47 }
48 x[pos] = 0;
49 }
50 delete [] x;
51 }
52
53 struct stat_and_more {
54 struct stat st;
55 unsigned char z;
56 };
57
TEST(SanitizerCommon,FileOps)58 TEST(SanitizerCommon, FileOps) {
59 const char *str1 = "qwerty";
60 uptr len1 = internal_strlen(str1);
61 const char *str2 = "zxcv";
62 uptr len2 = internal_strlen(str2);
63
64 u32 uid = GetUid();
65 char temp_filename[128];
66 #if SANITIZER_ANDROID
67 // I don't know a way to query temp directory location on Android without
68 // going through Java interfaces. The code below is not ideal, but should
69 // work. May require "adb root", but it is needed for almost any use of ASan
70 // on Android already.
71 internal_snprintf(temp_filename, sizeof(temp_filename),
72 "%s/sanitizer_common.tmp.%d",
73 GetEnv("EXTERNAL_STORAGE"), uid);
74 #else
75 internal_snprintf(temp_filename, sizeof(temp_filename),
76 "/tmp/sanitizer_common.tmp.%d", uid);
77 #endif
78 uptr openrv = OpenFile(temp_filename, true);
79 EXPECT_FALSE(internal_iserror(openrv));
80 fd_t fd = openrv;
81 EXPECT_EQ(len1, internal_write(fd, str1, len1));
82 EXPECT_EQ(len2, internal_write(fd, str2, len2));
83 internal_close(fd);
84
85 openrv = OpenFile(temp_filename, false);
86 EXPECT_FALSE(internal_iserror(openrv));
87 fd = openrv;
88 uptr fsize = internal_filesize(fd);
89 EXPECT_EQ(len1 + len2, fsize);
90
91 #if SANITIZER_TEST_HAS_STAT_H
92 struct stat st1, st2, st3;
93 EXPECT_EQ(0u, internal_stat(temp_filename, &st1));
94 EXPECT_EQ(0u, internal_lstat(temp_filename, &st2));
95 EXPECT_EQ(0u, internal_fstat(fd, &st3));
96 EXPECT_EQ(fsize, (uptr)st3.st_size);
97
98 // Verify that internal_fstat does not write beyond the end of the supplied
99 // buffer.
100 struct stat_and_more sam;
101 memset(&sam, 0xAB, sizeof(sam));
102 EXPECT_EQ(0u, internal_fstat(fd, &sam.st));
103 EXPECT_EQ(0xAB, sam.z);
104 EXPECT_NE(0xAB, sam.st.st_size);
105 EXPECT_NE(0, sam.st.st_size);
106 #endif
107
108 char buf[64] = {};
109 EXPECT_EQ(len1, internal_read(fd, buf, len1));
110 EXPECT_EQ(0, internal_memcmp(buf, str1, len1));
111 EXPECT_EQ((char)0, buf[len1 + 1]);
112 internal_memset(buf, 0, len1);
113 EXPECT_EQ(len2, internal_read(fd, buf, len2));
114 EXPECT_EQ(0, internal_memcmp(buf, str2, len2));
115 internal_close(fd);
116 }
117
118