utf8_helper.cpp
author Markus Schaber <markus@pep-security.net>
Tue, 04 Oct 2016 23:07:42 +0200
branchkeysync
changeset 178 9f30e2a5f72c
parent 47 a364a7231af7
child 269 26c8597fe860
permissions -rw-r--r--
COM-30 COM-24: Refactorings
Drop unused enum.
     1 #include "stdafx.h"
     2 #include "utf8_helper.h"
     3 
     4 using namespace ATL;
     5 using namespace std;
     6 
     7 namespace pEp {
     8     namespace utility {
     9 
    10         string utf8_string(wstring wstr, NORM_FORM norm)
    11         {
    12             if (wstr.size() == 0)
    13                 return string();
    14 
    15             wstring _wstr_normalized;
    16 
    17             if (norm == NormalizationOther)
    18                 _wstr_normalized = wstr;
    19             else {
    20                 int size = NormalizeString(norm, wstr.c_str(), -1, NULL, 0);
    21                 assert(size > 0);
    22                 if (size > 0) {
    23                     wchar_t *buf = new wchar_t[size];
    24                     NormalizeString(norm, wstr.c_str(), -1, buf, size);
    25                     _wstr_normalized = buf;
    26                     delete[] buf;
    27                 }
    28                 else
    29                     throw out_of_range("input wstring is not valid while normalizing.");
    30             }
    31             string result;
    32 
    33             int size = WideCharToMultiByte(CP_UTF8, WC_ERR_INVALID_CHARS, _wstr_normalized.c_str(), -1, NULL, 0, NULL, NULL);
    34             assert(size);
    35             if (size) {
    36                 char *buf = new char[size];
    37                 WideCharToMultiByte(CP_UTF8, WC_ERR_INVALID_CHARS, _wstr_normalized.c_str(), -1, buf, size, NULL, NULL);
    38                 result = buf;
    39                 delete[] buf;
    40             }
    41             else
    42                 throw out_of_range("input wstring is not valid while converting UTF-16 to UTF-8.");
    43 
    44             return result;
    45         }
    46 
    47         string utf8_string(BSTR bstr, NORM_FORM norm)
    48         {
    49             if (bstr == NULL)
    50                 return "";
    51             
    52             _bstr_t _bstr(bstr);
    53             wstring wstr = (wchar_t *) _bstr;
    54             _bstr.Detach();
    55 
    56             return utf8_string(wstr, norm);
    57         }
    58 
    59         wstring utf16_string(string str)
    60         {
    61             if (str.size() == 0)
    62                 return wstring();
    63 
    64             wstring result;
    65 
    66             int size = MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, str.c_str(), -1, NULL, 0);
    67             assert(size);
    68             if (size) {
    69                 wchar_t * buf = new wchar_t[size];
    70                 MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, str.c_str(), -1, buf, size);
    71                 result = buf;
    72                 delete[] buf;
    73             }
    74             else
    75                 throw out_of_range("input string is not valid while converting UTF-8 to UTF-16.");
    76 
    77             return result;
    78         }
    79 
    80         BSTR utf16_bstr(string str)
    81         {
    82             wstring wstr = utf16_string(str);
    83             return _bstr_t(wstr.c_str()).Detach();
    84         }
    85 
    86         LPSAFEARRAY string_array(const ::stringlist_t *stringlist)
    87         {
    88             int len = ::stringlist_length(stringlist);
    89 
    90             if (len == 0)
    91                 return NULL;
    92 
    93             CComSafeArray<BSTR> sa_string_list((LONG) len);
    94             LONG n = 0;
    95             for (const ::stringlist_t *k = stringlist; k && k->value; k = k->next) {
    96                 if (k->value) {
    97                     HRESULT _result = sa_string_list.SetAt(n, utf16_bstr(k->value), false);
    98                     assert(_result == S_OK);
    99                     if (_result == E_OUTOFMEMORY)
   100                         throw std::bad_alloc();
   101                     ++n;
   102                 }
   103             }
   104 
   105             return sa_string_list.Detach();
   106         }
   107 
   108         ::stringlist_t * new_stringlist(const SAFEARRAY * safearray)
   109         {
   110             if (safearray == NULL)
   111                 return NULL;
   112 
   113             CComSafeArray<BSTR> sa(safearray);
   114             int n_strings = 0;
   115             ::stringlist_t *_stringlist = ::new_stringlist((const char *) NULL);
   116             assert(_stringlist);
   117             if (_stringlist == NULL)
   118                 throw std::bad_alloc();
   119 
   120             n_strings = sa.GetUpperBound() - sa.GetLowerBound() + 1;
   121             ::stringlist_t *k = _stringlist;
   122             for (int i = 0, j = sa.GetLowerBound(); i < n_strings; ++i, ++j) {
   123                 k = ::stringlist_add(k, utf8_string(sa.GetAt(j)).c_str());
   124                 assert(k);
   125                 if (k == NULL) {
   126                     ::free_stringlist(_stringlist);
   127                     throw std::bad_alloc();
   128                 }
   129             }
   130 
   131             return _stringlist;
   132         }
   133 
   134     }
   135 }