AimRT/_deps/boost-src/libs/nowide/test/test_stdio.cpp
2025-01-12 20:41:24 +08:00

143 lines
4.5 KiB
C++

//
// Copyright (c) 2012 Artyom Beilis (Tonkikh)
// Copyright (c) 2019 Alexander Grund
//
// Distributed under the Boost Software License, Version 1.0.
// https://www.boost.org/LICENSE_1_0.txt
#include <boost/nowide/cstdio.hpp>
#include <boost/nowide/convert.hpp>
#include "test.hpp"
#include <cstdlib>
#include <cstring>
#include <iostream>
bool file_exists(const std::string& filename)
{
#ifdef BOOST_WINDOWS
FILE* f = boost::nowide::detail::wfopen(boost::nowide::widen(filename).c_str(), L"r");
#else
FILE* f = std::fopen(filename.c_str(), "r");
#endif
bool result = false;
if(f)
{
std::fclose(f);
result = true;
}
return result;
}
void create_test_file(const std::string& filename)
{
#ifdef BOOST_WINDOWS
FILE* f = boost::nowide::detail::wfopen(boost::nowide::widen(filename).c_str(), L"w");
#else
FILE* f = std::fopen(filename.c_str(), "w");
#endif
TEST(f);
TEST(std::fputs("test\n", f) >= 0);
std::fclose(f);
}
#ifdef BOOST_MSVC
#include <crtdbg.h> // For _CrtSetReportMode
void noop_invalid_param_handler(const wchar_t*, const wchar_t*, const wchar_t*, unsigned, uintptr_t)
{} // LCOV_EXCL_LINE
#endif
// coverity[root_function]
void test_main(int, char** argv, char**)
{
const std::string prefix = argv[0];
const std::string filename = prefix + "\xd7\xa9-\xd0\xbc-\xce\xbd.txt";
#ifdef BOOST_MSVC
// Prevent abort on freopen(NULL, ...)
_set_invalid_parameter_handler(noop_invalid_param_handler);
#endif
std::cout << " -- fopen - existing file" << std::endl;
{
create_test_file(filename);
FILE* f = boost::nowide::fopen(filename.c_str(), "r");
TEST(f);
char buf[16];
TEST(std::fgets(buf, 16, f) != 0);
TEST(strcmp(buf, "test\n") == 0);
std::fclose(f);
}
std::cout << " -- remove" << std::endl;
{
create_test_file(filename);
TEST(file_exists(filename));
TEST(boost::nowide::remove(filename.c_str()) == 0);
TEST(!file_exists(filename));
}
std::cout << " -- fopen non-existing file" << std::endl;
{
boost::nowide::remove(filename.c_str());
TEST(!file_exists(filename));
TEST(boost::nowide::fopen(filename.c_str(), "r") == nullptr);
TEST(!file_exists(filename));
}
std::cout << " -- freopen" << std::endl;
{
create_test_file(filename);
FILE* f = boost::nowide::fopen(filename.c_str(), "r+");
TEST(f);
std::cout << " -- Can read & write" << std::endl;
{
char buf[32];
TEST(std::fgets(buf, 32, f) != 0);
TEST(strcmp(buf, "test\n") == 0);
TEST(std::fseek(f, 0, SEEK_END) == 0);
TEST(std::fputs("foobar\n", f) >= 0);
}
// Reopen in read mode
// Note that changing the mode is not possibly on all implementations
// E.g. MSVC disallows NULL completely as the file parameter
FILE* f2 = boost::nowide::freopen(nullptr, "r", f);
if(!f2)
f2 = boost::nowide::freopen(filename.c_str(), "r", f);
std::cout << " -- no write possible" << std::endl;
{
TEST(f2 == f);
TEST(std::fputs("not-written\n", f) < 0);
TEST(std::fseek(f, 0, SEEK_SET) == 0);
char buf[32];
TEST(std::fgets(buf, 32, f) != 0);
TEST(strcmp(buf, "test\n") == 0);
TEST(std::fgets(buf, 32, f) != 0);
TEST(strcmp(buf, "foobar\n") == 0);
}
std::cout << " -- Reopen different file" << std::endl;
const std::string filename2 = filename + ".1.txt";
TEST(boost::nowide::freopen(filename2.c_str(), "w", f) == f);
{
char buf[32];
TEST(std::fputs("baz\n", f) >= 0);
std::fclose(f);
f = boost::nowide::fopen(filename2.c_str(), "r");
TEST(f);
TEST(std::fgets(buf, 32, f) != 0);
TEST(strcmp(buf, "baz\n") == 0);
}
std::fclose(f);
boost::nowide::remove(filename2.c_str());
}
std::cout << " -- rename" << std::endl;
{
create_test_file(filename);
const std::string filename2 = filename + ".1.txt";
boost::nowide::remove(filename2.c_str());
TEST(file_exists(filename));
TEST(!file_exists(filename2));
TEST(boost::nowide::rename(filename.c_str(), filename2.c_str()) == 0);
TEST(!file_exists(filename));
TEST(file_exists(filename2));
TEST(boost::nowide::remove(filename.c_str()) < 0);
TEST(boost::nowide::remove(filename2.c_str()) == 0);
}
}