utf8_helper.cpp
author Volker Birk <vb@pep-project.org>
Sun, 31 May 2015 19:31:19 +0200
changeset 39 6fc5cc7a167c
parent 38 d314ed57180c
child 40 dfb59114debe
permissions -rw-r--r--
safer
     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             return utf8_string((wstring) (wchar_t *) _bstr_t(bstr, true), norm);
    53         }
    54 
    55         wstring utf16_string(string str)
    56         {
    57             if (str.size() == 0)
    58                 return wstring();
    59 
    60             wstring result;
    61 
    62             int size = MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, str.c_str(), -1, NULL, 0);
    63             assert(size);
    64             if (size) {
    65                 wchar_t * buf = new wchar_t[size];
    66                 MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, str.c_str(), -1, buf, size);
    67                 result = buf;
    68                 delete[] buf;
    69             }
    70             else
    71                 throw out_of_range("input string is not valid while converting UTF-8 to UTF-16.");
    72 
    73             return result;
    74         }
    75 
    76         _bstr_t utf16_bstr(string str)
    77         {
    78             wstring wstr = utf16_string(str);
    79             return _bstr_t(wstr.c_str());
    80         }
    81 
    82         CComSafeArray<BSTR> string_array(const ::stringlist_t *stringlist)
    83         {
    84             int len = ::stringlist_length(stringlist);
    85 
    86             if (len = 0)
    87                 return CComSafeArray<BSTR>((ULONG)0);
    88 
    89             CComSafeArray<BSTR> sa_string_list((LONG) len);
    90             LONG n = 0;
    91             for (const ::stringlist_t *k = stringlist; k && k->value; k = k->next) {
    92                 if (k->value) {
    93                     HRESULT _result = sa_string_list.SetAt(n, utf16_bstr(k->value).Detach(), false);
    94                     assert(_result == S_OK);
    95                     if (_result == E_OUTOFMEMORY)
    96                         throw std::bad_alloc();
    97                     ++n;
    98                 }
    99             }
   100 
   101             return sa_string_list;
   102         }
   103 
   104         ::stringlist_t * new_stringlist(const SAFEARRAY * safearray)
   105         {
   106             if (safearray == NULL)
   107                 return NULL;
   108 
   109             CComSafeArray<BSTR> sa(safearray);
   110             int n_strings = 0;
   111             ::stringlist_t *_stringlist = ::new_stringlist((const char *) NULL);
   112             assert(_stringlist);
   113             if (_stringlist == NULL)
   114                 throw std::bad_alloc();
   115 
   116             n_strings = sa.GetUpperBound() - sa.GetLowerBound() + 1;
   117             ::stringlist_t *k = _stringlist;
   118             for (int i = 0, j = sa.GetLowerBound(); i < n_strings; ++i, ++j) {
   119                 k = ::stringlist_add(k, utf8_string(sa.GetAt(j)).c_str());
   120                 assert(k);
   121                 if (k == NULL) {
   122                     ::free_stringlist(_stringlist);
   123                     throw std::bad_alloc();
   124                 }
   125             }
   126 
   127             return _stringlist;
   128         }
   129 
   130     }
   131 }