70e189fbc9
Use a seek offset that fits within the file size. This was missed in presubmit because the FILE based stdio tests aren't run in overlay mode; fullbuild is not tested in presubmit. WRITE_SIZE == 11, so using a value of 42 for offseto would cause the expression `WRITE_SIZE - offseto` to evaluate to -31 as an unsigned 64b integer (18446744073709551585ULL). Fixes #86928
72 lines
2.9 KiB
C++
72 lines
2.9 KiB
C++
//===-- Unittests for ftell -----------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "src/stdio/fclose.h"
|
|
#include "src/stdio/fopen.h"
|
|
#include "src/stdio/fread.h"
|
|
#include "src/stdio/fseek.h"
|
|
#include "src/stdio/fseeko.h"
|
|
#include "src/stdio/ftell.h"
|
|
#include "src/stdio/ftello.h"
|
|
#include "src/stdio/fwrite.h"
|
|
#include "src/stdio/setvbuf.h"
|
|
#include "test/UnitTest/Test.h"
|
|
|
|
#include <stdio.h>
|
|
|
|
class LlvmLibcFTellTest : public LIBC_NAMESPACE::testing::Test {
|
|
protected:
|
|
void test_with_bufmode(int bufmode) {
|
|
constexpr char FILENAME[] = "testdata/ftell.test";
|
|
// We will set a special buffer to the file so that we guarantee buffering.
|
|
constexpr size_t BUFFER_SIZE = 1024;
|
|
char buffer[BUFFER_SIZE];
|
|
::FILE *file = LIBC_NAMESPACE::fopen(FILENAME, "w+");
|
|
ASSERT_FALSE(file == nullptr);
|
|
ASSERT_EQ(LIBC_NAMESPACE::setvbuf(file, buffer, bufmode, BUFFER_SIZE), 0);
|
|
|
|
// Include few '\n' chars to test when |bufmode| is _IOLBF.
|
|
constexpr char CONTENT[] = "12\n345\n6789";
|
|
constexpr size_t WRITE_SIZE = sizeof(CONTENT) - 1;
|
|
ASSERT_EQ(WRITE_SIZE, LIBC_NAMESPACE::fwrite(CONTENT, 1, WRITE_SIZE, file));
|
|
// The above write should have buffered the written data and not have
|
|
// trasferred it to the underlying stream. But, ftell operation should
|
|
// still return the correct effective offset.
|
|
ASSERT_EQ(size_t(LIBC_NAMESPACE::ftell(file)), WRITE_SIZE);
|
|
|
|
off_t offseto = 5;
|
|
ASSERT_EQ(0, LIBC_NAMESPACE::fseeko(file, offseto, SEEK_SET));
|
|
ASSERT_EQ(LIBC_NAMESPACE::ftello(file), offseto);
|
|
ASSERT_EQ(0, LIBC_NAMESPACE::fseeko(file, -offseto, SEEK_END));
|
|
ASSERT_EQ(size_t(LIBC_NAMESPACE::ftello(file)),
|
|
size_t(WRITE_SIZE - offseto));
|
|
|
|
long offset = 5;
|
|
ASSERT_EQ(0, LIBC_NAMESPACE::fseek(file, offset, SEEK_SET));
|
|
ASSERT_EQ(LIBC_NAMESPACE::ftell(file), offset);
|
|
ASSERT_EQ(0, LIBC_NAMESPACE::fseek(file, -offset, SEEK_END));
|
|
ASSERT_EQ(size_t(LIBC_NAMESPACE::ftell(file)), size_t(WRITE_SIZE - offset));
|
|
|
|
ASSERT_EQ(0, LIBC_NAMESPACE::fseek(file, 0, SEEK_SET));
|
|
constexpr size_t READ_SIZE = WRITE_SIZE / 2;
|
|
char data[READ_SIZE];
|
|
// Reading a small amount will actually read out much more data and
|
|
// buffer it. But, ftell should return the correct effective offset.
|
|
ASSERT_EQ(READ_SIZE, LIBC_NAMESPACE::fread(data, 1, READ_SIZE, file));
|
|
ASSERT_EQ(size_t(LIBC_NAMESPACE::ftell(file)), READ_SIZE);
|
|
|
|
ASSERT_EQ(0, LIBC_NAMESPACE::fclose(file));
|
|
}
|
|
};
|
|
|
|
TEST_F(LlvmLibcFTellTest, TellWithFBF) { test_with_bufmode(_IOFBF); }
|
|
|
|
TEST_F(LlvmLibcFTellTest, TellWithNBF) { test_with_bufmode(_IONBF); }
|
|
|
|
TEST_F(LlvmLibcFTellTest, TellWithLBF) { test_with_bufmode(_IOLBF); }
|