97 lines
2.4 KiB
C++
97 lines
2.4 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
#include "td/utils/Status.h"
|
|
|
|
#include "td/utils/SliceBuilder.h"
|
|
|
|
#if TD_PORT_WINDOWS
|
|
#include "td/utils/port/wstring_convert.h"
|
|
#endif
|
|
|
|
#if TD_PORT_POSIX
|
|
#include "td/utils/port/thread_local.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include <cstring>
|
|
#endif
|
|
|
|
namespace td {
|
|
|
|
#if TD_PORT_POSIX
|
|
CSlice strerror_safe(int code) {
|
|
const size_t size = 1000;
|
|
|
|
static TD_THREAD_LOCAL char *buf;
|
|
init_thread_local<char[]>(buf, size);
|
|
|
|
#if !defined(__GLIBC__) || ((_POSIX_C_SOURCE >= 200112L || _XOPEN_SOURCE >= 600) && !_GNU_SOURCE)
|
|
strerror_r(code, buf, size);
|
|
return CSlice(buf, buf + std::strlen(buf));
|
|
#else
|
|
return CSlice(strerror_r(code, buf, size));
|
|
#endif
|
|
}
|
|
#endif
|
|
|
|
#if TD_PORT_WINDOWS
|
|
string winerror_to_string(int code) {
|
|
const size_t size = 1000;
|
|
wchar_t wbuf[size];
|
|
auto res_size = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, nullptr, code, 0, wbuf, size - 1, nullptr);
|
|
if (res_size == 0) {
|
|
return "Unknown Windows error";
|
|
}
|
|
while (res_size != 0 && (wbuf[res_size - 1] == '\n' || wbuf[res_size - 1] == '\r')) {
|
|
res_size--;
|
|
}
|
|
auto error_message = from_wstring(wbuf, res_size);
|
|
if (error_message.is_error()) {
|
|
return "Invalid Windows error";
|
|
}
|
|
return error_message.move_as_ok();
|
|
}
|
|
#endif
|
|
|
|
Status Status::move_as_error_prefix(Slice prefix) const {
|
|
CHECK(is_error());
|
|
return move_as_error_prefix_unsafe(prefix);
|
|
}
|
|
|
|
Status Status::move_as_error_prefix_unsafe(Slice prefix) const {
|
|
Info info = get_info();
|
|
switch (info.error_type) {
|
|
case ErrorType::General:
|
|
return Error(code(), PSLICE() << prefix << message());
|
|
case ErrorType::Os:
|
|
return Status(false, ErrorType::Os, code(), PSLICE() << prefix << message());
|
|
default:
|
|
UNREACHABLE();
|
|
return {};
|
|
}
|
|
}
|
|
|
|
Status Status::move_as_error_suffix(Slice suffix) const {
|
|
CHECK(is_error());
|
|
return move_as_error_suffix_unsafe(suffix);
|
|
}
|
|
|
|
Status Status::move_as_error_suffix_unsafe(Slice suffix) const {
|
|
Info info = get_info();
|
|
switch (info.error_type) {
|
|
case ErrorType::General:
|
|
return Error(code(), PSLICE() << message() << suffix);
|
|
case ErrorType::Os:
|
|
return Status(false, ErrorType::Os, code(), PSLICE() << message() << suffix);
|
|
default:
|
|
UNREACHABLE();
|
|
return {};
|
|
}
|
|
}
|
|
|
|
} // namespace td
|