0
0
mirror of https://github.com/OpenVPN/openvpn3.git synced 2024-09-20 04:02:15 +02:00
openvpn3/openvpn/common/wstring.hpp
Leonard Ossa 3646265d15 Refactor nested namespace to C++17 style
Signed-off-by: Leonard Ossa <leonard.ossa@openvpn.com>
2024-07-03 10:20:11 +00:00

120 lines
4.1 KiB
C++

// OpenVPN -- An application to securely tunnel IP networks
// over a single port, with support for SSL/TLS-based
// session authentication and key exchange,
// packet encryption, packet authentication, and
// packet compression.
//
// Copyright (C) 2012-2024 OpenVPN Inc.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License Version 3
// as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program in the COPYING file.
// If not, see <http://www.gnu.org/licenses/>.
#pragma once
#ifdef _WIN32
#include <string>
#include <vector>
#include <memory>
namespace openvpn::wstring {
/**
* @brief Convert a UTF-8 string to UTF-16
*
* @param str The UTF-8 string to be converted
* @return std::wstring The converted UTF-16 string
*/
inline std::wstring from_utf8(const std::string &str)
{
std::wstring wStr; // enable RVO
if (str.empty())
return wStr;
const int str_size = static_cast<int>(str.size());
const auto reqSize = ::MultiByteToWideChar(CP_UTF8, 0, str.data(), str_size, nullptr, 0);
if (reqSize == 0)
throw std::runtime_error("MultiByteToWideChar(1) failed with code: [" + std::to_string(::GetLastError()) + "]");
wStr.resize(reqSize, L'\0'); // Allocate space
if (::MultiByteToWideChar(CP_UTF8, 0, str.data(), str_size, wStr.data(), reqSize) == 0)
throw std::runtime_error("MultiByteToWideChar(2) failed with code: [" + std::to_string(::GetLastError()) + "]");
return wStr;
}
/**
* @brief Convert a UTF-16 string to UTF-8
*
* @param wstr The UTF-16 string to be converted
* @return std::string The converted UTF-8 string
*/
inline std::string to_utf8(const std::wstring &wstr)
{
std::string str; // For RVO
if (wstr.empty())
return str;
const int wstr_size = static_cast<int>(wstr.size());
const auto reqSize = ::WideCharToMultiByte(CP_UTF8, 0, wstr.data(), wstr_size, nullptr, 0, nullptr, nullptr);
if (reqSize == 0)
throw std::runtime_error("WideCharToMultiByte(1) failed with code: [" + std::to_string(::GetLastError()) + "]");
str.resize(reqSize, '\0'); // Allocate space
if (::WideCharToMultiByte(CP_UTF8, 0, wstr.data(), wstr_size, str.data(), reqSize, nullptr, nullptr) == 0)
throw std::runtime_error("WideCharToMultiByte(2) failed with code: [" + std::to_string(::GetLastError()) + "]");
return str;
}
/**
* @brief Split a wide character string into a C array of it's letters
*
* @param wstr The wide string to be split
* @return std::unique_ptr<wchar_t[]> Smart pointer to the generated array
*/
inline std::unique_ptr<wchar_t[]> to_wchar_t(const std::wstring &wstr)
{
const size_t len = wstr.length();
std::unique_ptr<wchar_t[]> ret(new wchar_t[len + 1]);
size_t i;
for (i = 0; i < len; ++i)
ret[i] = wstr[i];
ret[i] = L'\0';
return ret;
}
/**
* @brief Convert a UTF-8 string vector to a UTF-16 MULTI_SZ string
*
* MULTI_SZ is a format used in the Windows Registry. It's a buffer
* containing multiple NUL terminated strings concatenated with each
* other, with an extra NUL at the end to signal termination of the
* MULTI_SZ string itself.
*
* @param strvec The vector of strings to convert
* @return std::wstring The converted MULTI_SZ string
*/
inline std::wstring pack_string_vector(const std::vector<std::string> &strvec)
{
if (strvec.empty())
{
return std::wstring(2, L'\0'); // empty MULTI_SZ
}
std::wstring ret;
for (auto &s : strvec)
{
ret += from_utf8(s);
ret += L'\0';
}
ret += L'\0';
return ret;
}
#endif // #ifdef _WIN32
}