1 /* 2 * Copyright (C) 2014 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 #include <stdio_ext.h> 18 19 #include <gtest/gtest.h> 20 21 #include <errno.h> 22 #include <fcntl.h> 23 #include <limits.h> 24 #include <math.h> 25 #include <stdio.h> 26 #include <sys/types.h> 27 #include <sys/stat.h> 28 #include <unistd.h> 29 #include <wchar.h> 30 #include <locale.h> 31 32 #include <android-base/file.h> 33 34 #include "utils.h" 35 TEST(stdio_ext,__fbufsize)36 TEST(stdio_ext, __fbufsize) { 37 FILE* fp = fopen("/proc/version", "r"); 38 39 // Initially, there's no buffer in case the first thing you do is disable buffering. 40 ASSERT_EQ(0U, __fbufsize(fp)); 41 42 // A read forces a buffer to be created. 43 char buf[128]; 44 fgets(buf, sizeof(buf), fp); 45 ASSERT_EQ(1024U, __fbufsize(fp)); 46 47 ASSERT_EQ(0, setvbuf(fp, buf, _IOFBF, 1)); 48 ASSERT_EQ(1U, __fbufsize(fp)); 49 50 ASSERT_EQ(0, setvbuf(fp, buf, _IOFBF, 8)); 51 ASSERT_EQ(8U, __fbufsize(fp)); 52 53 fclose(fp); 54 } 55 TEST(stdio_ext,__flbf)56 TEST(stdio_ext, __flbf) { 57 FILE* fp = fopen("/proc/version", "r"); 58 59 ASSERT_FALSE(__flbf(fp)); 60 61 char buf[128]; 62 ASSERT_EQ(0, setvbuf(fp, buf, _IOLBF, sizeof(buf))); 63 64 ASSERT_TRUE(__flbf(fp)); 65 66 fclose(fp); 67 } 68 TEST(stdio_ext,__fpending)69 TEST(stdio_ext, __fpending) { 70 FILE* fp = fopen("/dev/null", "w"); 71 ASSERT_EQ(0U, __fpending(fp)); 72 ASSERT_EQ('x', fputc('x', fp)); 73 ASSERT_EQ(1U, __fpending(fp)); 74 ASSERT_EQ('y', fputc('y', fp)); 75 ASSERT_EQ(2U, __fpending(fp)); 76 fflush(fp); 77 ASSERT_EQ(0U, __fpending(fp)); 78 fclose(fp); 79 } 80 TEST(stdio_ext,__freadahead)81 TEST(stdio_ext, __freadahead) { 82 #if defined(__GLIBC__) 83 GTEST_SKIP() << "glibc doesn't have __freadahead"; 84 #else 85 FILE* fp = tmpfile(); 86 ASSERT_NE(EOF, fputs("hello", fp)); 87 rewind(fp); 88 89 ASSERT_EQ('h', fgetc(fp)); 90 ASSERT_EQ(4u, __freadahead(fp)); 91 92 ASSERT_EQ('H', ungetc('H', fp)); 93 ASSERT_EQ(5u, __freadahead(fp)); 94 95 fclose(fp); 96 #endif 97 } 98 TEST(stdio_ext,__fpurge)99 TEST(stdio_ext, __fpurge) { 100 FILE* fp = tmpfile(); 101 102 ASSERT_EQ('a', fputc('a', fp)); 103 ASSERT_EQ(1U, __fpending(fp)); 104 __fpurge(fp); 105 ASSERT_EQ(0U, __fpending(fp)); 106 107 ASSERT_EQ('b', fputc('b', fp)); 108 ASSERT_EQ('\n', fputc('\n', fp)); 109 ASSERT_EQ(2U, __fpending(fp)); 110 111 rewind(fp); 112 113 char buf[16]; 114 char* s = fgets(buf, sizeof(buf), fp); 115 ASSERT_TRUE(s != nullptr); 116 ASSERT_STREQ("b\n", s); 117 118 fclose(fp); 119 } 120 TEST(stdio_ext,_flushlbf)121 TEST(stdio_ext, _flushlbf) { 122 FILE* fp = fopen("/dev/null", "w"); 123 124 char buf[128]; 125 ASSERT_EQ(0, setvbuf(fp, buf, _IOLBF, sizeof(buf))); 126 127 ASSERT_EQ('a', fputc('a', fp)); 128 ASSERT_EQ(1U, __fpending(fp)); 129 130 _flushlbf(); 131 132 ASSERT_EQ(0U, __fpending(fp)); 133 134 fclose(fp); 135 } 136 TEST(stdio_ext,__freadable__fwritable)137 TEST(stdio_ext, __freadable__fwritable) { 138 FILE* fp; 139 140 // Read-only. 141 fp = fopen("/dev/null", "r"); 142 ASSERT_TRUE(__freadable(fp)); 143 ASSERT_FALSE(__fwritable(fp)); 144 fclose(fp); 145 146 // Write-only. 147 fp = fopen("/dev/null", "w"); 148 ASSERT_FALSE(__freadable(fp)); 149 ASSERT_TRUE(__fwritable(fp)); 150 fclose(fp); 151 152 // Append (aka write-only). 153 fp = fopen("/dev/null", "a"); 154 ASSERT_FALSE(__freadable(fp)); 155 ASSERT_TRUE(__fwritable(fp)); 156 fclose(fp); 157 158 // The three read-write modes. 159 for (auto read_write_mode : {"r+", "w+", "a+"}) { 160 fp = fopen("/dev/null", read_write_mode); 161 ASSERT_TRUE(__freadable(fp)); 162 ASSERT_TRUE(__fwritable(fp)); 163 fclose(fp); 164 } 165 } 166 TEST(stdio_ext,__freading__fwriting)167 TEST(stdio_ext, __freading__fwriting) { 168 FILE* fp; 169 170 // Append (aka write-only). Never reading. Always writing. 171 fp = fopen("/dev/zero", "a"); 172 ASSERT_FALSE(__freading(fp)); // Not reading initially. 173 ASSERT_TRUE(__fwriting(fp)); // Writing initially. 174 ASSERT_TRUE(fputc('x', fp) != EOF); 175 ASSERT_FALSE(__freading(fp)); // Not reading after write. 176 ASSERT_TRUE(__fwriting(fp)); // Still writing after write. 177 fclose(fp); 178 179 // Write-only. Never reading. Always writing. 180 fp = fopen("/dev/zero", "w"); 181 ASSERT_FALSE(__freading(fp)); // Not reading initially. 182 ASSERT_TRUE(__fwriting(fp)); // Writing initially. 183 ASSERT_TRUE(fputc('x', fp) != EOF); 184 ASSERT_FALSE(__freading(fp)); // Not reading after write. 185 ASSERT_TRUE(__fwriting(fp)); // Still writing after write. 186 fclose(fp); 187 188 // Read-only. Always reading. Never writing. 189 fp = fopen("/dev/zero", "r"); 190 ASSERT_TRUE(__freading(fp)); // Reading initially. 191 ASSERT_FALSE(__fwriting(fp)); // Not writing initially. 192 ASSERT_TRUE(fgetc(fp) == 0); 193 ASSERT_TRUE(__freading(fp)); // Still reading after read. 194 ASSERT_FALSE(__fwriting(fp)); // Still not writing after read. 195 fclose(fp); 196 197 // The three read-write modes. 198 for (auto read_write_mode : {"r+", "w+", "a+"}) { 199 fp = fopen("/dev/zero", read_write_mode); 200 ASSERT_FALSE(__freading(fp)); // Not reading initially. 201 ASSERT_FALSE(__fwriting(fp)); // Not writing initially. 202 ASSERT_TRUE(fgetc(fp) == 0); 203 ASSERT_TRUE(__freading(fp)); // Reading after read. 204 ASSERT_FALSE(__fwriting(fp)); // Not writing after read. 205 ASSERT_TRUE(fputc('x', fp) != EOF); 206 ASSERT_FALSE(__freading(fp)); // Not reading after write. 207 ASSERT_TRUE(__fwriting(fp)); // Writing after write. 208 fclose(fp); 209 } 210 } 211 TEST(stdio_ext,__fseterr)212 TEST(stdio_ext, __fseterr) { 213 #if defined(__GLIBC__) 214 GTEST_SKIP() << "glibc doesn't have __fseterr, but gnulib will use it"; 215 #else 216 FILE* fp = fopen("/dev/null", "w"); 217 218 ASSERT_FALSE(ferror(fp)); 219 __fseterr(fp); 220 ASSERT_TRUE(ferror(fp)); 221 clearerr(fp); 222 ASSERT_FALSE(ferror(fp)); 223 224 fclose(fp); 225 #endif 226 } 227 TEST(stdio_ext,__fsetlocking)228 TEST(stdio_ext, __fsetlocking) { 229 FILE* fp = fopen("/proc/version", "r"); 230 ASSERT_EQ(FSETLOCKING_INTERNAL, __fsetlocking(fp, FSETLOCKING_QUERY)); 231 ASSERT_EQ(FSETLOCKING_INTERNAL, __fsetlocking(fp, FSETLOCKING_BYCALLER)); 232 ASSERT_EQ(FSETLOCKING_BYCALLER, __fsetlocking(fp, FSETLOCKING_QUERY)); 233 ASSERT_EQ(FSETLOCKING_BYCALLER, __fsetlocking(fp, FSETLOCKING_INTERNAL)); 234 ASSERT_EQ(FSETLOCKING_INTERNAL, __fsetlocking(fp, FSETLOCKING_QUERY)); 235 fclose(fp); 236 } 237 LockingByCallerHelper(std::atomic<pid_t> * pid)238 static void LockingByCallerHelper(std::atomic<pid_t>* pid) { 239 *pid = gettid(); 240 flockfile(stdout); 241 funlockfile(stdout); 242 } 243 TEST(stdio_ext,__fsetlocking_BYCALLER)244 TEST(stdio_ext, __fsetlocking_BYCALLER) { 245 // Check if users can use flockfile/funlockfile to protect stdio operations. 246 int old_state = __fsetlocking(stdout, FSETLOCKING_BYCALLER); 247 flockfile(stdout); 248 pthread_t thread; 249 std::atomic<pid_t> pid(0); 250 ASSERT_EQ(0, pthread_create(&thread, nullptr, 251 reinterpret_cast<void* (*)(void*)>(LockingByCallerHelper), &pid)); 252 WaitUntilThreadSleep(pid); 253 funlockfile(stdout); 254 255 ASSERT_EQ(0, pthread_join(thread, nullptr)); 256 __fsetlocking(stdout, old_state); 257 } 258