llvm-project/libcxx/test/support/platform_support.h
Ryan Prichard b4722cc4c9 [libc++][test] Fix infinite loop when mkstemp fails
mkstemp is guaranteed to make at least TMP_MAX attempts to create the
random file, and if it can't, it fails with EEXIST. get_temp_file_name
shouldn't call mkstemp again if it fails with anything other than
EEXIST. A single mkstemp call seems sufficient.

On Android, I've seen mkstemp fail with:
 - EROFS (because cwd wasn't set to a writable filesystem)
 - EACCES (because cwd pointed to a dir owned by root, but the test
   program was running as the shell user instead)

Previously, get_temp_file_name would run forever in these situations.

See D4962 and "llvm-svn: 229035"

Reviewed By: #libc, ldionne

Differential Revision: https://reviews.llvm.org/D130214
2022-07-22 16:20:47 -07:00

115 lines
3.1 KiB
C++

//===----------------------------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
// Define a bunch of macros that can be used in the tests instead of
// implementation defined assumptions:
// - locale names
// - floating point number string output
#ifndef PLATFORM_SUPPORT_H
#define PLATFORM_SUPPORT_H
#include "test_macros.h"
// locale names
#define LOCALE_en_US "en_US"
#define LOCALE_en_US_UTF_8 "en_US.UTF-8"
#define LOCALE_fr_FR_UTF_8 "fr_FR.UTF-8"
#ifdef __linux__
# define LOCALE_fr_CA_ISO8859_1 "fr_CA.ISO-8859-1"
# define LOCALE_cs_CZ_ISO8859_2 "cs_CZ.ISO-8859-2"
#elif defined(_WIN32)
# define LOCALE_fr_CA_ISO8859_1 "fr-CA"
# define LOCALE_cs_CZ_ISO8859_2 "cs-CZ"
#else
# define LOCALE_fr_CA_ISO8859_1 "fr_CA.ISO8859-1"
# define LOCALE_cs_CZ_ISO8859_2 "cs_CZ.ISO8859-2"
#endif
#define LOCALE_ja_JP_UTF_8 "ja_JP.UTF-8"
#define LOCALE_ru_RU_UTF_8 "ru_RU.UTF-8"
#define LOCALE_zh_CN_UTF_8 "zh_CN.UTF-8"
#include <stdio.h>
#include <stdlib.h>
#include <codecvt>
#include <locale>
#include <string>
#if defined(_WIN32)
# include <io.h> // _mktemp_s
# include <fcntl.h> // _O_EXCL, ...
# include <sys/stat.h> // _S_IREAD, ...
#else
# include <unistd.h> // close
#endif
#if defined(_CS_GNU_LIBC_VERSION)
# include <string.h> // strverscmp
#endif
#if defined(_NEWLIB_VERSION) && defined(__STRICT_ANSI__)
// Newlib provides this, but in the header it's under __STRICT_ANSI__
extern "C" {
int mkstemp(char*);
}
#endif
inline
std::string get_temp_file_name()
{
#if defined(_WIN32)
while (true) {
char Name[] = "libcxx.XXXXXX";
if (_mktemp_s(Name, sizeof(Name)) != 0) abort();
int fd = _open(Name, _O_RDWR | _O_CREAT | _O_EXCL, _S_IREAD | _S_IWRITE);
if (fd != -1) {
_close(fd);
return Name;
}
if (errno == EEXIST)
continue;
abort();
}
#else
std::string Name = "libcxx.XXXXXX";
int FD = mkstemp(&Name[0]);
if (FD == -1) {
perror("mkstemp");
abort();
}
close(FD);
return Name;
#endif
}
_LIBCPP_SUPPRESS_DEPRECATED_PUSH
#ifdef _LIBCPP_HAS_OPEN_WITH_WCHAR
inline
std::wstring get_wide_temp_file_name()
{
return std::wstring_convert<std::codecvt_utf8_utf16<wchar_t> >().from_bytes(
get_temp_file_name());
}
#endif // _LIBCPP_HAS_OPEN_WITH_WCHAR
_LIBCPP_SUPPRESS_DEPRECATED_POP
#if defined(_CS_GNU_LIBC_VERSION)
inline bool glibc_version_less_than(char const* version) {
std::string test_version = std::string("glibc ") + version;
size_t n = confstr(_CS_GNU_LIBC_VERSION, nullptr, (size_t)0);
char *current_version = new char[n];
confstr(_CS_GNU_LIBC_VERSION, current_version, n);
bool result = strverscmp(current_version, test_version.c_str()) < 0;
delete[] current_version;
return result;
}
#endif // _CS_GNU_LIBC_VERSION
#endif // PLATFORM_SUPPORT_H