forked from ONLYOFFICE/nssm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
utf8.cpp
126 lines (98 loc) · 3.41 KB
/
utf8.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#include "nssm.h"
static unsigned long cp;
void setup_utf8() {
#ifdef UNICODE
/*
Ensure we write in UTF-8 mode, so that non-ASCII characters don't get
mangled. If we were compiled in ANSI mode it won't work.
*/
cp = GetConsoleOutputCP();
SetConsoleOutputCP(CP_UTF8);
_setmode(_fileno(stdout), _O_U8TEXT);
_setmode(_fileno(stderr), _O_U8TEXT);
#endif
}
void unsetup_utf8() {
if (cp) SetConsoleOutputCP(cp);
}
/*
Conversion functions.
to_utf8/16() converts a string which may be either utf8 or utf16 to
the desired format. If no conversion is needed a new string is still
allocated and the old string is copied.
from_utf8/16() converts a string which IS in the specified format to
whichever format is needed according to whether UNICODE is defined.
It simply wraps the appropriate to_utf8/16() function.
Therefore the caller must ALWAYS free the destination pointer after a
successful (return code 0) call to one of these functions.
The length pointer is optional. Pass NULL if you don't care about
the length of the converted string.
Both the destination and the length, if supplied, will be zeroed if
no conversion was done.
*/
int to_utf8(const wchar_t *utf16, char **utf8, unsigned long *utf8len) {
*utf8 = 0;
if (utf8len) *utf8len = 0;
int size = WideCharToMultiByte(CP_UTF8, 0, utf16, -1, NULL, 0, NULL, NULL);
if (! size) return 1;
*utf8 = (char *) HeapAlloc(GetProcessHeap(), 0, size);
if (! *utf8) return 2;
if (! WideCharToMultiByte(CP_UTF8, 0, utf16, -1, *utf8, size, NULL, NULL)) {
HeapFree(GetProcessHeap(), 0, *utf8);
*utf8 = 0;
return 3;
}
if (utf8len) *utf8len = (unsigned long) strlen(*utf8);
return 0;
}
int to_utf8(const char *ansi, char **utf8, unsigned long *utf8len) {
*utf8 = 0;
if (utf8len) *utf8len = 0;
size_t len = strlen(ansi);
int size = (int) len + 1;
*utf8 = (char *) HeapAlloc(GetProcessHeap(), 0, size);
if (! *utf8) return 2;
if (utf8len) *utf8len = (unsigned long) len;
memmove(*utf8, ansi, size);
return 0;
}
int to_utf16(const char *utf8, wchar_t **utf16, unsigned long *utf16len) {
*utf16 = 0;
if (utf16len) *utf16len = 0;
int size = MultiByteToWideChar(CP_UTF8, 0, utf8, -1, NULL, 0);
if (! size) return 1;
*utf16 = (wchar_t *) HeapAlloc(GetProcessHeap(), 0, size * sizeof(wchar_t));
if (! *utf16) return 2;
if (! MultiByteToWideChar(CP_UTF8, 0, utf8, -1, *utf16, size)) {
HeapFree(GetProcessHeap(), 0, *utf16);
*utf16 = 0;
return 3;
}
if (utf16len) *utf16len = (unsigned long) wcslen(*utf16);
return 0;
}
int to_utf16(const wchar_t *unicode, wchar_t **utf16, unsigned long *utf16len) {
*utf16 = 0;
if (utf16len) *utf16len = 0;
size_t len = wcslen(unicode);
int size = ((int) len + 1) * sizeof(wchar_t);
*utf16 = (wchar_t *) HeapAlloc(GetProcessHeap(), 0, size);
if (! *utf16) return 2;
if (utf16len) *utf16len = (unsigned long) len;
memmove(*utf16, unicode, size);
return 0;
}
int from_utf8(const char *utf8, TCHAR **buffer, unsigned long *buflen) {
#ifdef UNICODE
return to_utf16(utf8, buffer, buflen);
#else
return to_utf8(utf8, buffer, buflen);
#endif
}
int from_utf16(const wchar_t *utf16, TCHAR **buffer, unsigned long *buflen) {
#ifdef UNICODE
return to_utf16(utf16, buffer, buflen);
#else
return to_utf8(utf16, buffer, buflen);
#endif
}