mirror of
https://github.com/veracrypt/VeraCrypt.git
synced 2025-11-11 11:08:02 -06:00
Add original TrueCrypt 7.1a sources
This commit is contained in:
367
src/Platform/StringConverter.cpp
Normal file
367
src/Platform/StringConverter.cpp
Normal file
@@ -0,0 +1,367 @@
|
||||
/*
|
||||
Copyright (c) 2008-2009 TrueCrypt Developers Association. All rights reserved.
|
||||
|
||||
Governed by the TrueCrypt License 3.0 the full text of which is contained in
|
||||
the file License.txt included in TrueCrypt binary and source code distribution
|
||||
packages.
|
||||
*/
|
||||
|
||||
#ifdef __GNUC__
|
||||
# include <cxxabi.h>
|
||||
#endif
|
||||
#include <locale>
|
||||
#include <typeinfo>
|
||||
#include "Buffer.h"
|
||||
#include "Exception.h"
|
||||
#include "ForEach.h"
|
||||
#include "StringConverter.h"
|
||||
#include "SystemException.h"
|
||||
|
||||
namespace TrueCrypt
|
||||
{
|
||||
void StringConverter::Erase (string &str)
|
||||
{
|
||||
for (size_t i = 0; i < str.size(); ++i)
|
||||
{
|
||||
str[i] = ' ';
|
||||
}
|
||||
}
|
||||
|
||||
void StringConverter::Erase (wstring &str)
|
||||
{
|
||||
for (size_t i = 0; i < str.size(); ++i)
|
||||
{
|
||||
str[i] = ' ';
|
||||
}
|
||||
}
|
||||
|
||||
wstring StringConverter::FromNumber (double number)
|
||||
{
|
||||
wstringstream s;
|
||||
s << number;
|
||||
return s.str();
|
||||
}
|
||||
|
||||
wstring StringConverter::FromNumber (int32 number)
|
||||
{
|
||||
wstringstream s;
|
||||
s << number;
|
||||
return s.str();
|
||||
}
|
||||
|
||||
wstring StringConverter::FromNumber (uint32 number)
|
||||
{
|
||||
wstringstream s;
|
||||
s << number;
|
||||
return s.str();
|
||||
}
|
||||
|
||||
wstring StringConverter::FromNumber (int64 number)
|
||||
{
|
||||
wstringstream s;
|
||||
s << number;
|
||||
return s.str();
|
||||
}
|
||||
|
||||
wstring StringConverter::FromNumber (uint64 number)
|
||||
{
|
||||
wstringstream s;
|
||||
s << number;
|
||||
return s.str();
|
||||
}
|
||||
|
||||
string StringConverter::GetTrailingNumber (const string &str)
|
||||
{
|
||||
size_t start = str.find_last_not_of ("0123456789");
|
||||
if (start == string::npos)
|
||||
return str;
|
||||
|
||||
string s = str.substr (start + 1);
|
||||
if (s.empty ())
|
||||
throw ParameterIncorrect (SRC_POS);
|
||||
|
||||
return s;
|
||||
}
|
||||
|
||||
string StringConverter::GetTypeName (const type_info &typeInfo)
|
||||
{
|
||||
try
|
||||
{
|
||||
#ifdef _MSC_VER
|
||||
// type_info::name() leaks memory as of MS VC++ 8.0
|
||||
string rawName (typeInfo.raw_name());
|
||||
|
||||
size_t cut1 = (rawName.find (".?A") != string::npos) ? 4 : string::npos;
|
||||
size_t cut2 = rawName.find ("@");
|
||||
size_t cut3 = rawName.find ("@@");
|
||||
|
||||
if (cut1 == string::npos || cut2 == string::npos || cut3 == string::npos)
|
||||
return typeInfo.name();
|
||||
|
||||
return rawName.substr (cut2 + 1, cut3 - cut2 - 1) + "::" + rawName.substr (cut1, cut2 - cut1);
|
||||
|
||||
#elif defined (__GNUC__)
|
||||
int status;
|
||||
char *name = abi::__cxa_demangle (typeInfo.name(), nullptr, nullptr, &status);
|
||||
|
||||
if (name)
|
||||
{
|
||||
string s (name);
|
||||
free (name);
|
||||
return s;
|
||||
}
|
||||
#endif
|
||||
}
|
||||
catch (...) { }
|
||||
|
||||
return typeInfo.name();
|
||||
}
|
||||
|
||||
wstring StringConverter::QuoteSpaces (const wstring &str)
|
||||
{
|
||||
if (str.find (L' ') == string::npos)
|
||||
return str;
|
||||
|
||||
wstring escaped (L"'");
|
||||
foreach (wchar_t c, str)
|
||||
{
|
||||
if (c == L'\'')
|
||||
escaped += L'\'';
|
||||
escaped += c;
|
||||
}
|
||||
return escaped + L'\'';
|
||||
}
|
||||
|
||||
vector <string> StringConverter::Split (const string &str, const string &separators, bool returnEmptyFields)
|
||||
{
|
||||
vector <string> elements;
|
||||
|
||||
if (!returnEmptyFields)
|
||||
{
|
||||
size_t p = 0;
|
||||
while ((p = str.find_first_not_of (separators, p)) != string::npos)
|
||||
{
|
||||
size_t end = str.find_first_of (separators, p);
|
||||
if (end == string::npos)
|
||||
{
|
||||
elements.push_back (str.substr (p));
|
||||
break;
|
||||
}
|
||||
|
||||
elements.push_back (str.substr (p, end - p));
|
||||
p = end;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
string element;
|
||||
elements.push_back (element);
|
||||
foreach (char c, str)
|
||||
{
|
||||
if (separators.find (c) != string::npos)
|
||||
{
|
||||
element.erase();
|
||||
elements.push_back (element);
|
||||
}
|
||||
else
|
||||
{
|
||||
elements.back() += c;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return elements;
|
||||
}
|
||||
|
||||
string StringConverter::StripTrailingNumber (const string &str)
|
||||
{
|
||||
size_t start = str.find_last_not_of ("0123456789");
|
||||
if (start == string::npos)
|
||||
return "";
|
||||
|
||||
return str.substr (0, start + 1);
|
||||
}
|
||||
|
||||
wstring StringConverter::ToExceptionString (const exception &ex)
|
||||
{
|
||||
const SystemException *sysEx = dynamic_cast <const SystemException *> (&ex);
|
||||
if (sysEx)
|
||||
return ToWide (sysEx->what()) + L": " + sysEx->SystemText() + L": " + sysEx->GetSubject();
|
||||
|
||||
if (ex.what() && !string (ex.what()).empty())
|
||||
return ToWide (GetTypeName (typeid (ex)) + ": " + ex.what());
|
||||
|
||||
return ToWide (GetTypeName (typeid (ex)));
|
||||
}
|
||||
|
||||
string StringConverter::ToLower (const string &str)
|
||||
{
|
||||
string s;
|
||||
foreach (char c, str)
|
||||
s += tolower (c, locale());
|
||||
return s;
|
||||
}
|
||||
|
||||
string StringConverter::ToSingle (const wstring &wstr, bool noThrow)
|
||||
{
|
||||
string str;
|
||||
ToSingle (wstr, str, noThrow);
|
||||
return str;
|
||||
}
|
||||
|
||||
void StringConverter::ToSingle (const wstring &wstr, string &str, bool noThrow)
|
||||
{
|
||||
try
|
||||
{
|
||||
mbstate_t mbState;
|
||||
Memory::Zero (&mbState, sizeof (mbState));
|
||||
const wchar_t *src = wstr.c_str();
|
||||
|
||||
size_t size = wcsrtombs (nullptr, &src, 0, &mbState);
|
||||
if (size == (size_t) -1)
|
||||
throw StringConversionFailed (SRC_POS, wstr);
|
||||
|
||||
vector <char> buf (size + 1);
|
||||
Memory::Zero (&mbState, sizeof (mbState));
|
||||
|
||||
if ((size = wcsrtombs (&buf[0], &src, buf.size(), &mbState)) == (size_t) -1)
|
||||
throw StringConversionFailed (SRC_POS, wstr);
|
||||
|
||||
str.clear();
|
||||
str.insert (0, &buf.front(), size);
|
||||
Memory::Erase (&buf.front(), buf.size());
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
if (!noThrow)
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
uint32 StringConverter::ToUInt32 (const string &str)
|
||||
{
|
||||
uint32 n;
|
||||
stringstream ss (str);
|
||||
|
||||
ss >> n;
|
||||
if (ss.fail() || n == 0xffffFFFFU)
|
||||
throw ParameterIncorrect (SRC_POS);
|
||||
|
||||
return n;
|
||||
}
|
||||
|
||||
uint32 StringConverter::ToUInt32 (const wstring &str)
|
||||
{
|
||||
uint32 n;
|
||||
wstringstream ss (str);
|
||||
|
||||
ss >> n;
|
||||
if (ss.fail() || n == 0xffffFFFFU)
|
||||
throw ParameterIncorrect (SRC_POS);
|
||||
|
||||
return n;
|
||||
}
|
||||
|
||||
uint64 StringConverter::ToUInt64 (const string &str)
|
||||
{
|
||||
uint64 n;
|
||||
stringstream ss (str);
|
||||
|
||||
ss >> n;
|
||||
if (ss.fail() || n == 0xffffFFFFffffFFFFULL)
|
||||
throw ParameterIncorrect (SRC_POS);
|
||||
|
||||
return n;
|
||||
}
|
||||
|
||||
uint64 StringConverter::ToUInt64 (const wstring &str)
|
||||
{
|
||||
uint64 n;
|
||||
wstringstream ss (str);
|
||||
|
||||
ss >> n;
|
||||
if (ss.fail() || n == 0xffffFFFFffffFFFFULL)
|
||||
throw ParameterIncorrect (SRC_POS);
|
||||
|
||||
return n;
|
||||
}
|
||||
|
||||
string StringConverter::ToUpper (const string &str)
|
||||
{
|
||||
string s;
|
||||
foreach (char c, str)
|
||||
s += toupper (c, locale());
|
||||
return s;
|
||||
}
|
||||
|
||||
wstring StringConverter::ToWide (const string &str, bool noThrow)
|
||||
{
|
||||
try
|
||||
{
|
||||
mbstate_t mbState;
|
||||
Memory::Zero (&mbState, sizeof (mbState));
|
||||
const char *src = str.c_str();
|
||||
|
||||
size_t size = mbsrtowcs (nullptr, &src, 0, &mbState);
|
||||
if (size == (size_t) -1)
|
||||
throw StringConversionFailed (SRC_POS);
|
||||
|
||||
vector <wchar_t> buf (size + 1);
|
||||
Memory::Zero (&mbState, sizeof (mbState));
|
||||
|
||||
if ((size = mbsrtowcs (&buf[0], &src, buf.size(), &mbState)) == (size_t) -1)
|
||||
throw StringConversionFailed (SRC_POS);
|
||||
|
||||
wstring s;
|
||||
s.insert (s.begin(), buf.begin(), buf.begin() + size);
|
||||
return s;
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
if (noThrow)
|
||||
return L"";
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
void StringConverter::ToWideBuffer (const wstring &str, wchar_t *buffer, size_t bufferSize)
|
||||
{
|
||||
if (str.length() < 1)
|
||||
{
|
||||
buffer[0] = 0;
|
||||
return;
|
||||
}
|
||||
|
||||
BufferPtr (
|
||||
(byte *) buffer,
|
||||
bufferSize).CopyFrom (
|
||||
ConstBufferPtr ((byte *) (wstring (str).c_str()),
|
||||
(str.length() + 1) * sizeof (wchar_t)
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
string StringConverter::Trim (const string &str)
|
||||
{
|
||||
size_t start = 0;
|
||||
size_t end = str.size();
|
||||
if (end < 1)
|
||||
return str;
|
||||
|
||||
foreach (char c, str)
|
||||
{
|
||||
if (c > ' ')
|
||||
break;
|
||||
++start;
|
||||
}
|
||||
|
||||
foreach_reverse (char c, str)
|
||||
{
|
||||
if (c > ' ')
|
||||
break;
|
||||
--end;
|
||||
}
|
||||
|
||||
return str.substr (start, end - start);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user