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
|
/* 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.
*/
/* Usage Notes:
*
* this module, and the misc/win32/utf8.c modules must be
* compiled APR_EXPORT_STATIC and linked to an application with
* the /entry:wmainCRTStartup flag (which this module kindly
* provides to the developer who links to libaprapp-1.lib).
* This module becomes the true wmain entry point, and passes
* utf-8 reformatted argv and env arrays to the application's
* main() function as if nothing happened.
*
* This module is only compatible with Unicode operating systems.
* Mixed (Win9x backwards compatible) binaries should refer instead
* to the apr_startup.c module.
*
* _dbg_malloc/realloc is used in place of the usual API, in order
* to convince the MSVCRT that it created these entities. If we
* do not create them as _CRT_BLOCK entities, the crt will fault
* on an assert. We are not worrying about the crt's locks here,
* since we are single threaded [so far].
*/
#include "apr_general.h"
#include "ShellAPI.h"
#include "wchar.h"
#include "apr_arch_file_io.h"
#include "assert.h"
#include "apr_private.h"
#include "apr_arch_misc.h"
#pragma comment(linker,"/ENTRY:wmainCRTStartup")
extern int main(int argc, const char **argv, const char **env);
int wmain(int argc, const wchar_t **wargv, const wchar_t **wenv)
{
char **argv;
char **env;
int envc;
int i;
(void)apr_wastrtoastr(&argv, wargv, argc);
envc = 0;
while (wenv[envc]) {
envc++;
}
/* Initial environment stored as single heap block, but uses
* separate heap entry for every environment variable
* after first change.
*/
env = apr_malloc_dbg((envc + 1) * sizeof(char *), __FILE__, __LINE__);
for (i = 0; i < envc; i++) {
apr_size_t wcount;
apr_size_t envlen;
wcount = wcslen(wenv[i]) + 1;
envlen = (wcount - 1) * 3 + 1;
env[i] = apr_malloc_dbg(envlen, __FILE__, __LINE__);
(void)apr_conv_ucs2_to_utf8(wenv[i], &wcount, env[i], &envlen);
}
env[i] = NULL;
_environ = env;
/* MSVCRT will attempt to maintain the wide environment calls
* on _putenv(), which is bogus if we've passed a non-ascii
* string to _putenv(), since they use MultiByteToWideChar
* and breaking the implicit utf-8 assumption we've built.
*
* Reset _wenviron for good measure.
*/
if (_wenviron) {
_wenviron = NULL;
/* There is no need to free _wenviron because MSVCRT
* automatically free 'initial' environment block.
*/
}
apr_app_init_complete = 1;
return main(argc, argv, env);
}
|
