bionic/tests/stdio_ext_test.cpp
Elliott Hughes 8c4994bbc1 Implement __fsetlocking.
The old __isthreaded hack was never very useful on Android because all user
code runs in a VM where there are lots of threads running. But __fsetlocking
lets a caller say "I'll worry about the locking for this FILE*", which is
useful for the normal case where you don't share a FILE* between threads
so you don't need any locking.

Bug: 17154740
Bug: 18593728
Change-Id: I2a8dddc29d3edff39a3d7d793387f2253608a68d
2015-01-21 10:33:30 -08:00

143 lines
3.4 KiB
C++

/*
* Copyright (C) 2014 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.
*/
#include <stdio_ext.h>
#include <gtest/gtest.h>
#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <math.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <wchar.h>
#include <locale.h>
#include "TemporaryFile.h"
TEST(stdio_ext, __fbufsize) {
FILE* fp = fopen("/proc/version", "r");
// Initially, there's no buffer in case the first thing you do is disable buffering.
ASSERT_EQ(0U, __fbufsize(fp));
// A read forces a buffer to be created.
char buf[128];
fgets(buf, sizeof(buf), fp);
ASSERT_EQ(1024U, __fbufsize(fp));
ASSERT_EQ(0, setvbuf(fp, buf, _IOFBF, 1));
ASSERT_EQ(1U, __fbufsize(fp));
ASSERT_EQ(0, setvbuf(fp, buf, _IOFBF, 8));
ASSERT_EQ(8U, __fbufsize(fp));
fclose(fp);
}
TEST(stdio_ext, __flbf) {
FILE* fp = fopen("/proc/version", "r");
ASSERT_FALSE(__flbf(fp));
char buf[128];
ASSERT_EQ(0, setvbuf(fp, buf, _IOLBF, sizeof(buf)));
ASSERT_TRUE(__flbf(fp));
fclose(fp);
}
TEST(stdio_ext, __fpending) {
FILE* fp = fopen("/dev/null", "w");
ASSERT_EQ(0U, __fpending(fp));
ASSERT_EQ('x', fputc('x', fp));
ASSERT_EQ(1U, __fpending(fp));
ASSERT_EQ('y', fputc('y', fp));
ASSERT_EQ(2U, __fpending(fp));
fflush(fp);
ASSERT_EQ(0U, __fpending(fp));
fclose(fp);
}
TEST(stdio_ext, __fpurge) {
FILE* fp = tmpfile();
ASSERT_EQ('a', fputc('a', fp));
ASSERT_EQ(1U, __fpending(fp));
__fpurge(fp);
ASSERT_EQ(0U, __fpending(fp));
ASSERT_EQ('b', fputc('b', fp));
ASSERT_EQ('\n', fputc('\n', fp));
ASSERT_EQ(2U, __fpending(fp));
rewind(fp);
char buf[16];
char* s = fgets(buf, sizeof(buf), fp);
ASSERT_TRUE(s != NULL);
ASSERT_STREQ("b\n", s);
fclose(fp);
}
TEST(stdio_ext, _flushlbf) {
FILE* fp = fopen("/dev/null", "w");
char buf[128];
ASSERT_EQ(0, setvbuf(fp, buf, _IOLBF, sizeof(buf)));
ASSERT_EQ('a', fputc('a', fp));
ASSERT_EQ(1U, __fpending(fp));
_flushlbf();
ASSERT_EQ(0U, __fpending(fp));
fclose(fp);
}
TEST(stdio_ext, __freadable__fwritable) {
FILE* fp = fopen("/dev/null", "r");
ASSERT_TRUE(__freadable(fp));
ASSERT_FALSE(__fwritable(fp));
fclose(fp);
fp = fopen("/dev/null", "w");
ASSERT_FALSE(__freadable(fp));
ASSERT_TRUE(__fwritable(fp));
fclose(fp);
fp = fopen("/dev/null", "w+");
ASSERT_TRUE(__freadable(fp));
ASSERT_TRUE(__fwritable(fp));
fclose(fp);
}
TEST(stdio_ext, __fsetlocking) {
FILE* fp = fopen("/proc/version", "r");
ASSERT_EQ(FSETLOCKING_INTERNAL, __fsetlocking(fp, FSETLOCKING_QUERY));
ASSERT_EQ(FSETLOCKING_INTERNAL, __fsetlocking(fp, FSETLOCKING_BYCALLER));
ASSERT_EQ(FSETLOCKING_BYCALLER, __fsetlocking(fp, FSETLOCKING_QUERY));
ASSERT_EQ(FSETLOCKING_BYCALLER, __fsetlocking(fp, FSETLOCKING_INTERNAL));
ASSERT_EQ(FSETLOCKING_INTERNAL, __fsetlocking(fp, FSETLOCKING_QUERY));
fclose(fp);
}