ceph/src/boost/libs/nowide/test/test_stdio.cpp

   1 //
   2 //  Copyright (c) 2012 Artyom Beilis (Tonkikh)
   3 //  Copyright (c) 2019 Alexander Grund
   4 //
   5 //  Distributed under the Boost Software License, Version 1.0. (See
   6 //  accompanying file LICENSE_1_0.txt or copy at
   7 //  http://www.boost.org/LICENSE_1_0.txt)
   8 //
   9
  10 #include <boost/nowide/cstdio.hpp>
  11
  12 #include <boost/nowide/convert.hpp>
  13 #include <cstdlib>
  14 #include <cstring>
  15 #include <iostream>
  16
  17 #include "test.hpp"
  18
  19 bool file_exists(const std::string& filename)
  20 {
  21 #ifdef BOOST_WINDOWS
  22     FILE* f = boost::nowide::detail::wfopen(boost::nowide::widen(filename).c_str(), L"r");
  23 #else
  24     FILE* f = std::fopen(filename.c_str(), "r");
  25 #endif
  26     bool result = false;
  27     if(f)
  28     {
  29         std::fclose(f);
  30         result = true;
  31     }
  32     return result;
  33 }
  34
  35 void create_test_file(const std::string& filename)
  36 {
  37 #ifdef BOOST_WINDOWS
  38     FILE* f = boost::nowide::detail::wfopen(boost::nowide::widen(filename).c_str(), L"w");
  39 #else
  40     FILE* f = std::fopen(filename.c_str(), "w");
  41 #endif
  42     TEST(f);
  43     TEST(std::fputs("test\n", f) >= 0);
  44     std::fclose(f);
  45 }
  46
  47 #if BOOST_MSVC
  48 #include <crtdbg.h> // For _CrtSetReportMode
  49 void noop_invalid_param_handler(const wchar_t*, const wchar_t*, const wchar_t*, unsigned, uintptr_t)
  50 {}
  51 #endif
  52
  53 void test_main(int, char** argv, char**)
  54 {
  55     const std::string prefix = argv[0];
  56     const std::string filename = prefix + "\xd7\xa9-\xd0\xbc-\xce\xbd.txt";
  57 #if BOOST_MSVC
  58     // Prevent abort on freopen(NULL, ...)
  59     _set_invalid_parameter_handler(noop_invalid_param_handler);
  60 #endif
  61
  62     std::cout << " -- fopen - existing file" << std::endl;
  63     {
  64         create_test_file(filename);
  65         FILE* f = boost::nowide::fopen(filename.c_str(), "r");
  66         TEST(f);
  67         char buf[16];
  68         TEST(std::fgets(buf, 16, f) != 0);
  69         TEST(strcmp(buf, "test\n") == 0);
  70         std::fclose(f);
  71     }
  72     std::cout << " -- remove" << std::endl;
  73     {
  74         create_test_file(filename);
  75         TEST(file_exists(filename));
  76         TEST(boost::nowide::remove(filename.c_str()) == 0);
  77         TEST(!file_exists(filename));
  78     }
  79     std::cout << " -- fopen non-existing file" << std::endl;
  80     {
  81         boost::nowide::remove(filename.c_str());
  82         TEST(!file_exists(filename));
  83         TEST(boost::nowide::fopen(filename.c_str(), "r") == NULL);
  84         TEST(!file_exists(filename));
  85     }
  86     std::cout << " -- freopen" << std::endl;
  87     {
  88         create_test_file(filename);
  89         FILE* f = boost::nowide::fopen(filename.c_str(), "r+");
  90         TEST(f);
  91         std::cout << " -- Can read & write" << std::endl;
  92         {
  93             char buf[32];
  94             TEST(std::fgets(buf, 32, f) != 0);
  95             TEST(strcmp(buf, "test\n") == 0);
  96             TEST(std::fseek(f, 0, SEEK_END) == 0);
  97             TEST(std::fputs("foobar\n", f) >= 0);
  98         }
  99         // Reopen in read mode
 100         // Note that changing the mode is not possibly on all implementations
 101         // E.g. MSVC disallows NULL completely as the file parameter
 102         FILE* f2 = boost::nowide::freopen(NULL, "r", f);
 103         if(!f2)
 104             f2 = boost::nowide::freopen(filename.c_str(), "r", f);
 105         std::cout << " -- no write possible" << std::endl;
 106         {
 107             TEST(f2 == f);
 108             TEST(std::fputs("not-written\n", f) < 0);
 109             TEST(std::fseek(f, 0, SEEK_SET) == 0);
 110             char buf[32];
 111             TEST(std::fgets(buf, 32, f) != 0);
 112             TEST(strcmp(buf, "test\n") == 0);
 113             TEST(std::fgets(buf, 32, f) != 0);
 114             TEST(strcmp(buf, "foobar\n") == 0);
 115         }
 116         std::cout << " -- Reopen different file" << std::endl;
 117         const std::string filename2 = filename + ".1.txt";
 118         TEST(boost::nowide::freopen(filename2.c_str(), "w", f) == f);
 119         {
 120             char buf[32];
 121             TEST(std::fputs("baz\n", f) >= 0);
 122             std::fclose(f);
 123             f = boost::nowide::fopen(filename2.c_str(), "r");
 124             TEST(f);
 125             TEST(std::fgets(buf, 32, f) != 0);
 126             TEST(strcmp(buf, "baz\n") == 0);
 127         }
 128         std::fclose(f);
 129         boost::nowide::remove(filename2.c_str());
 130     }
 131     std::cout << " -- rename" << std::endl;
 132     {
 133         create_test_file(filename);
 134         const std::string filename2 = filename + ".1.txt";
 135         boost::nowide::remove(filename2.c_str());
 136         TEST(file_exists(filename));
 137         TEST(!file_exists(filename2));
 138         TEST(boost::nowide::rename(filename.c_str(), filename2.c_str()) == 0);
 139         TEST(!file_exists(filename));
 140         TEST(file_exists(filename2));
 141         TEST(boost::nowide::remove(filename.c_str()) < 0);
 142         TEST(boost::nowide::remove(filename2.c_str()) == 0);
 143     }
 144 }