123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277 |
- /* Distributed under the OSI-approved BSD 3-Clause License. See accompanying
- file Copyright.txt or https://cmake.org/licensing#kwsys for details. */
- #ifdef __osf__
- #define _OSF_SOURCE
- #define _POSIX_C_SOURCE 199506L
- #define _XOPEN_SOURCE_EXTENDED
- #endif
- #include "kwsysPrivate.h"
- #include KWSYS_HEADER(Encoding.hxx)
- #include KWSYS_HEADER(Encoding.h)
- // Work-around CMake dependency scanning limitation. This must
- // duplicate the above list of headers.
- #if 0
- #include "Encoding.h.in"
- #include "Encoding.hxx.in"
- #endif
- #include <stdlib.h>
- #include <string.h>
- #include <vector>
- #ifdef _MSC_VER
- #pragma warning(disable : 4786)
- #endif
- // Windows API.
- #if defined(_WIN32)
- #include <windows.h>
- #include <ctype.h>
- #include <shellapi.h>
- #endif
- namespace KWSYS_NAMESPACE {
- Encoding::CommandLineArguments Encoding::CommandLineArguments::Main(
- int argc, char const* const* argv)
- {
- #ifdef _WIN32
- (void)argc;
- (void)argv;
- int ac;
- LPWSTR* w_av = CommandLineToArgvW(GetCommandLineW(), &ac);
- std::vector<std::string> av1(ac);
- std::vector<char const*> av2(ac);
- for (int i = 0; i < ac; i++) {
- av1[i] = ToNarrow(w_av[i]);
- av2[i] = av1[i].c_str();
- }
- LocalFree(w_av);
- return CommandLineArguments(ac, &av2[0]);
- #else
- return CommandLineArguments(argc, argv);
- #endif
- }
- Encoding::CommandLineArguments::CommandLineArguments(int ac,
- char const* const* av)
- {
- this->argv_.resize(ac + 1);
- for (int i = 0; i < ac; i++) {
- this->argv_[i] = strdup(av[i]);
- }
- this->argv_[ac] = KWSYS_NULLPTR;
- }
- Encoding::CommandLineArguments::CommandLineArguments(int ac,
- wchar_t const* const* av)
- {
- this->argv_.resize(ac + 1);
- for (int i = 0; i < ac; i++) {
- this->argv_[i] = kwsysEncoding_DupToNarrow(av[i]);
- }
- this->argv_[ac] = KWSYS_NULLPTR;
- }
- Encoding::CommandLineArguments::~CommandLineArguments()
- {
- for (size_t i = 0; i < this->argv_.size(); i++) {
- free(argv_[i]);
- }
- }
- Encoding::CommandLineArguments::CommandLineArguments(
- const CommandLineArguments& other)
- {
- this->argv_.resize(other.argv_.size());
- for (size_t i = 0; i < this->argv_.size(); i++) {
- this->argv_[i] = other.argv_[i] ? strdup(other.argv_[i]) : KWSYS_NULLPTR;
- }
- }
- Encoding::CommandLineArguments& Encoding::CommandLineArguments::operator=(
- const CommandLineArguments& other)
- {
- if (this != &other) {
- size_t i;
- for (i = 0; i < this->argv_.size(); i++) {
- free(this->argv_[i]);
- }
- this->argv_.resize(other.argv_.size());
- for (i = 0; i < this->argv_.size(); i++) {
- this->argv_[i] = other.argv_[i] ? strdup(other.argv_[i]) : KWSYS_NULLPTR;
- }
- }
- return *this;
- }
- int Encoding::CommandLineArguments::argc() const
- {
- return static_cast<int>(this->argv_.size() - 1);
- }
- char const* const* Encoding::CommandLineArguments::argv() const
- {
- return &this->argv_[0];
- }
- #if KWSYS_STL_HAS_WSTRING
- std::wstring Encoding::ToWide(const std::string& str)
- {
- std::wstring wstr;
- #if defined(_WIN32)
- const int wlength = MultiByteToWideChar(
- KWSYS_ENCODING_DEFAULT_CODEPAGE, 0, str.data(), int(str.size()), NULL, 0);
- if (wlength > 0) {
- wchar_t* wdata = new wchar_t[wlength];
- int r = MultiByteToWideChar(KWSYS_ENCODING_DEFAULT_CODEPAGE, 0, str.data(),
- int(str.size()), wdata, wlength);
- if (r > 0) {
- wstr = std::wstring(wdata, wlength);
- }
- delete[] wdata;
- }
- #else
- size_t pos = 0;
- size_t nullPos = 0;
- do {
- if (pos < str.size() && str.at(pos) != '\0') {
- wstr += ToWide(str.c_str() + pos);
- }
- nullPos = str.find('\0', pos);
- if (nullPos != std::string::npos) {
- pos = nullPos + 1;
- wstr += wchar_t('\0');
- }
- } while (nullPos != std::string::npos);
- #endif
- return wstr;
- }
- std::string Encoding::ToNarrow(const std::wstring& str)
- {
- std::string nstr;
- #if defined(_WIN32)
- int length =
- WideCharToMultiByte(KWSYS_ENCODING_DEFAULT_CODEPAGE, 0, str.c_str(),
- int(str.size()), NULL, 0, NULL, NULL);
- if (length > 0) {
- char* data = new char[length];
- int r =
- WideCharToMultiByte(KWSYS_ENCODING_DEFAULT_CODEPAGE, 0, str.c_str(),
- int(str.size()), data, length, NULL, NULL);
- if (r > 0) {
- nstr = std::string(data, length);
- }
- delete[] data;
- }
- #else
- size_t pos = 0;
- size_t nullPos = 0;
- do {
- if (pos < str.size() && str.at(pos) != '\0') {
- nstr += ToNarrow(str.c_str() + pos);
- }
- nullPos = str.find(wchar_t('\0'), pos);
- if (nullPos != std::string::npos) {
- pos = nullPos + 1;
- nstr += '\0';
- }
- } while (nullPos != std::string::npos);
- #endif
- return nstr;
- }
- std::wstring Encoding::ToWide(const char* cstr)
- {
- std::wstring wstr;
- size_t length = kwsysEncoding_mbstowcs(KWSYS_NULLPTR, cstr, 0) + 1;
- if (length > 0) {
- std::vector<wchar_t> wchars(length);
- if (kwsysEncoding_mbstowcs(&wchars[0], cstr, length) > 0) {
- wstr = &wchars[0];
- }
- }
- return wstr;
- }
- std::string Encoding::ToNarrow(const wchar_t* wcstr)
- {
- std::string str;
- size_t length = kwsysEncoding_wcstombs(KWSYS_NULLPTR, wcstr, 0) + 1;
- if (length > 0) {
- std::vector<char> chars(length);
- if (kwsysEncoding_wcstombs(&chars[0], wcstr, length) > 0) {
- str = &chars[0];
- }
- }
- return str;
- }
- #if defined(_WIN32)
- // Convert local paths to UNC style paths
- std::wstring Encoding::ToWindowsExtendedPath(std::string const& source)
- {
- std::wstring wsource = Encoding::ToWide(source);
- // Resolve any relative paths
- DWORD wfull_len;
- /* The +3 is a workaround for a bug in some versions of GetFullPathNameW that
- * won't return a large enough buffer size if the input is too small */
- wfull_len = GetFullPathNameW(wsource.c_str(), 0, NULL, NULL) + 3;
- std::vector<wchar_t> wfull(wfull_len);
- GetFullPathNameW(wsource.c_str(), wfull_len, &wfull[0], NULL);
- /* This should get the correct size without any extra padding from the
- * previous size workaround. */
- wfull_len = static_cast<DWORD>(wcslen(&wfull[0]));
- if (wfull_len >= 2 && isalpha(wfull[0]) &&
- wfull[1] == L':') { /* C:\Foo\bar\FooBar.txt */
- return L"\\\\?\\" + std::wstring(&wfull[0]);
- } else if (wfull_len >= 2 && wfull[0] == L'\\' &&
- wfull[1] == L'\\') { /* Starts with \\ */
- if (wfull_len >= 4 && wfull[2] == L'?' &&
- wfull[3] == L'\\') { /* Starts with \\?\ */
- if (wfull_len >= 8 && wfull[4] == L'U' && wfull[5] == L'N' &&
- wfull[6] == L'C' &&
- wfull[7] == L'\\') { /* \\?\UNC\Foo\bar\FooBar.txt */
- return std::wstring(&wfull[0]);
- } else if (wfull_len >= 6 && isalpha(wfull[4]) &&
- wfull[5] == L':') { /* \\?\C:\Foo\bar\FooBar.txt */
- return std::wstring(&wfull[0]);
- } else if (wfull_len >= 5) { /* \\?\Foo\bar\FooBar.txt */
- return L"\\\\?\\UNC\\" + std::wstring(&wfull[4]);
- }
- } else if (wfull_len >= 4 && wfull[2] == L'.' &&
- wfull[3] == L'\\') { /* Starts with \\.\ a device name */
- if (wfull_len >= 6 && isalpha(wfull[4]) &&
- wfull[5] == L':') { /* \\.\C:\Foo\bar\FooBar.txt */
- return L"\\\\?\\" + std::wstring(&wfull[4]);
- } else if (wfull_len >=
- 5) { /* \\.\Foo\bar\ Device name is left unchanged */
- return std::wstring(&wfull[0]);
- }
- } else if (wfull_len >= 3) { /* \\Foo\bar\FooBar.txt */
- return L"\\\\?\\UNC\\" + std::wstring(&wfull[2]);
- }
- }
- // If this case has been reached, then the path is invalid. Leave it
- // unchanged
- return Encoding::ToWide(source);
- }
- #endif
- #endif // KWSYS_STL_HAS_WSTRING
- } // namespace KWSYS_NAMESPACE
|