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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
|
/*
* win32_xlate.c : Windows xlate stuff.
*
* ====================================================================
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
* ====================================================================
*/
/* prevent "empty compilation unit" warning on e.g. UNIX */
typedef int win32_xlate__dummy;
#ifdef WIN32
/* Define _WIN32_DCOM for CoInitializeEx(). */
#define _WIN32_DCOM
/* We must include windows.h ourselves or apr.h includes it for us with
many ignore options set. Including Winsock is required to resolve IPv6
compilation errors. APR_HAVE_IPV6 is only defined after including
apr.h, so we can't detect this case here. */
/* winsock2.h includes windows.h */
#include <winsock2.h>
#include <Ws2tcpip.h>
#include <mlang.h>
#include <apr.h>
#include <apr_errno.h>
#include <apr_portable.h>
#include "svn_pools.h"
#include "svn_string.h"
#include "svn_utf.h"
#include "private/svn_atomic.h"
#include "win32_xlate.h"
static svn_atomic_t com_initialized = 0;
/* Initializes COM and keeps COM available until process exit.
Implements svn_atomic__init_once init_func */
static svn_error_t *
initialize_com(void *baton, apr_pool_t* pool)
{
/* Try to initialize for apartment-threaded object concurrency. */
HRESULT hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
if (hr == RPC_E_CHANGED_MODE)
{
/* COM already initalized for multi-threaded object concurrency. We are
neutral to object concurrency so try to initalize it in the same way
for us, to keep an handle open. */
hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
}
if (FAILED(hr))
return svn_error_create(APR_EGENERAL, NULL, NULL);
return SVN_NO_ERROR;
}
typedef struct win32_xlate_t
{
UINT from_page_id;
UINT to_page_id;
} win32_xlate_t;
static apr_status_t
get_page_id_from_name(UINT *page_id_p, const char *page_name, apr_pool_t *pool)
{
IMultiLanguage * mlang = NULL;
HRESULT hr;
MIMECSETINFO page_info;
WCHAR ucs2_page_name[128];
svn_error_t *err;
if (page_name == SVN_APR_DEFAULT_CHARSET)
{
*page_id_p = CP_ACP;
return APR_SUCCESS;
}
else if (page_name == SVN_APR_LOCALE_CHARSET)
{
*page_id_p = CP_THREAD_ACP; /* Valid on Windows 2000+ */
return APR_SUCCESS;
}
else if (!strcmp(page_name, "UTF-8"))
{
*page_id_p = CP_UTF8;
return APR_SUCCESS;
}
/* Use codepage identifier nnn if the codepage name is in the form
of "CPnnn".
We need this code since apr_os_locale_encoding() and svn_cmdline_init()
generates such codepage names even if they are not valid IANA charset
name. */
if ((page_name[0] == 'c' || page_name[0] == 'C')
&& (page_name[1] == 'p' || page_name[1] == 'P'))
{
*page_id_p = atoi(page_name + 2);
return APR_SUCCESS;
}
err = svn_atomic__init_once(&com_initialized, initialize_com, NULL, pool);
if (err)
{
svn_error_clear(err);
return APR_EGENERAL;
}
hr = CoCreateInstance(&CLSID_CMultiLanguage, NULL, CLSCTX_INPROC_SERVER,
&IID_IMultiLanguage, (void **) &mlang);
if (FAILED(hr))
return APR_EGENERAL;
/* Convert page name to wide string. */
MultiByteToWideChar(CP_UTF8, 0, page_name, -1, ucs2_page_name,
sizeof(ucs2_page_name) / sizeof(ucs2_page_name[0]));
memset(&page_info, 0, sizeof(page_info));
hr = mlang->lpVtbl->GetCharsetInfo(mlang, ucs2_page_name, &page_info);
if (FAILED(hr))
{
mlang->lpVtbl->Release(mlang);
return APR_EINVAL;
}
if (page_info.uiInternetEncoding)
*page_id_p = page_info.uiInternetEncoding;
else
*page_id_p = page_info.uiCodePage;
mlang->lpVtbl->Release(mlang);
return APR_SUCCESS;
}
apr_status_t
svn_subr__win32_xlate_open(win32_xlate_t **xlate_p, const char *topage,
const char *frompage, apr_pool_t *pool)
{
UINT from_page_id, to_page_id;
apr_status_t apr_err = APR_SUCCESS;
win32_xlate_t *xlate;
apr_err = get_page_id_from_name(&to_page_id, topage, pool);
if (apr_err == APR_SUCCESS)
apr_err = get_page_id_from_name(&from_page_id, frompage, pool);
if (apr_err == APR_SUCCESS)
{
xlate = apr_palloc(pool, sizeof(*xlate));
xlate->from_page_id = from_page_id;
xlate->to_page_id = to_page_id;
*xlate_p = xlate;
}
return apr_err;
}
apr_status_t
svn_subr__win32_xlate_to_stringbuf(win32_xlate_t *handle,
const char *src_data,
apr_size_t src_length,
svn_stringbuf_t **dest,
apr_pool_t *pool)
{
WCHAR * wide_str;
int retval, wide_size;
if (src_length == 0)
{
*dest = svn_stringbuf_create_empty(pool);
return APR_SUCCESS;
}
retval = MultiByteToWideChar(handle->from_page_id, 0, src_data, src_length,
NULL, 0);
if (retval == 0)
return apr_get_os_error();
wide_size = retval;
/* Allocate temporary buffer for small strings on stack instead of heap. */
if (wide_size <= MAX_PATH)
{
wide_str = alloca(wide_size * sizeof(WCHAR));
}
else
{
wide_str = apr_palloc(pool, wide_size * sizeof(WCHAR));
}
retval = MultiByteToWideChar(handle->from_page_id, 0, src_data, src_length,
wide_str, wide_size);
if (retval == 0)
return apr_get_os_error();
retval = WideCharToMultiByte(handle->to_page_id, 0, wide_str, wide_size,
NULL, 0, NULL, NULL);
if (retval == 0)
return apr_get_os_error();
/* Ensure that buffer is enough to hold result string and termination
character. */
*dest = svn_stringbuf_create_ensure(retval + 1, pool);
(*dest)->len = retval;
retval = WideCharToMultiByte(handle->to_page_id, 0, wide_str, wide_size,
(*dest)->data, (*dest)->len, NULL, NULL);
if (retval == 0)
return apr_get_os_error();
(*dest)->len = retval;
return APR_SUCCESS;
}
#endif /* WIN32 */
|
