lib/core/CWindowsError_Windows.cc

/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the following additional limitation. Functionality enabled by the
 * files subject to the Elastic License 2.0 may only be used in production when
 * invoked by an Elasticsearch process with a license key installed that permits
 * use of machine learning features. You may not use this file except in
 * compliance with the Elastic License 2.0 and the foregoing additional
 * limitation.
 */
#include <core/CWindowsError.h>

#include <core/CStringUtils.h>
#include <core/WindowsSafe.h>

#include <cmath>
#include <ostream>

namespace {
const size_t BUFFER_SIZE(1024);
}

namespace ml {
namespace core {

CWindowsError::CWindowsError() : m_ErrorCode(GetLastError()) {
}

CWindowsError::CWindowsError(std::uint32_t errorCode) : m_ErrorCode(errorCode) {
}

std::uint32_t CWindowsError::errorCode() const {
    return m_ErrorCode;
}

std::string CWindowsError::errorString() const {
    char message[BUFFER_SIZE] = {'\0'};

    DWORD msgLen(FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS |
                                   FORMAT_MESSAGE_MAX_WIDTH_MASK,
                               0, m_ErrorCode, 0, message, BUFFER_SIZE, 0));
    if (msgLen == 0) {
        return "unknown error code (" + CStringUtils::typeToString(m_ErrorCode) + ')';
    }

    return message;
}

std::ostream& operator<<(std::ostream& os, const CWindowsError& windowsError) {
    char message[BUFFER_SIZE] = {'\0'};

    DWORD msgLen(FormatMessage(
        FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS | FORMAT_MESSAGE_MAX_WIDTH_MASK,
        0, windowsError.m_ErrorCode, 0, message, BUFFER_SIZE, 0));
    if (msgLen == 0) {
        os << "unknown error code (" << windowsError.m_ErrorCode << ')';
    } else {
        os << message;
    }

    return os;
}
}
}