2530 lines
84 KiB
C
2530 lines
84 KiB
C
/*
|
|
* Ntdll environment functions
|
|
*
|
|
* Copyright 1996, 1998 Alexandre Julliard
|
|
* Copyright 2003 Eric Pouech
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#if 0
|
|
#pragma makedep unix
|
|
#endif
|
|
|
|
#include "config.h"
|
|
|
|
#include <assert.h>
|
|
#include <errno.h>
|
|
#include <locale.h>
|
|
#include <langinfo.h>
|
|
#include <fcntl.h>
|
|
#include <stdarg.h>
|
|
#include <sys/types.h>
|
|
#include <sys/ioctl.h>
|
|
#ifdef HAVE_SYS_STAT_H
|
|
# include <sys/stat.h>
|
|
#endif
|
|
#ifdef HAVE_SYS_PRCTL_H
|
|
# include <sys/prctl.h>
|
|
#endif
|
|
#ifdef HAVE_PWD_H
|
|
# include <pwd.h>
|
|
#endif
|
|
#ifdef HAVE_UNISTD_H
|
|
# include <unistd.h>
|
|
#endif
|
|
#ifdef __APPLE__
|
|
# include <CoreFoundation/CFLocale.h>
|
|
# include <CoreFoundation/CFString.h>
|
|
#endif
|
|
|
|
#include "ntstatus.h"
|
|
#define WIN32_NO_STATUS
|
|
#include "windef.h"
|
|
#include "winternl.h"
|
|
#include "winbase.h"
|
|
#include "winnls.h"
|
|
#include "wine/condrv.h"
|
|
#include "wine/debug.h"
|
|
#include "unix_private.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(environ);
|
|
|
|
PEB *peb = NULL;
|
|
USHORT *uctable = NULL, *lctable = NULL;
|
|
SIZE_T startup_info_size = 0;
|
|
BOOL is_prefix_bootstrap = FALSE;
|
|
|
|
static const WCHAR bootstrapW[] = {'W','I','N','E','B','O','O','T','S','T','R','A','P','M','O','D','E'};
|
|
|
|
int main_argc = 0;
|
|
char **main_argv = NULL;
|
|
char **main_envp = NULL;
|
|
WCHAR **main_wargv = NULL;
|
|
|
|
static LCID user_lcid, system_lcid;
|
|
static LANGID user_ui_language, system_ui_language;
|
|
|
|
static char system_locale[LOCALE_NAME_MAX_LENGTH];
|
|
static char user_locale[LOCALE_NAME_MAX_LENGTH];
|
|
|
|
/* system directory with trailing backslash */
|
|
const WCHAR system_dir[] = {'\\','?','?','\\','C',':','\\','w','i','n','d','o','w','s','\\',
|
|
's','y','s','t','e','m','3','2','\\',0};
|
|
|
|
static struct
|
|
{
|
|
USHORT *data;
|
|
USHORT *dbcs;
|
|
USHORT *mbtable;
|
|
void *wctable;
|
|
} unix_cp;
|
|
|
|
enum nls_section_type
|
|
{
|
|
NLS_SECTION_SORTKEYS = 9,
|
|
NLS_SECTION_CASEMAP = 10,
|
|
NLS_SECTION_CODEPAGE = 11,
|
|
NLS_SECTION_NORMALIZE = 12
|
|
};
|
|
|
|
static char *get_nls_file_path( ULONG type, ULONG id )
|
|
{
|
|
const char *dir = build_dir ? build_dir : data_dir;
|
|
const char *name = NULL;
|
|
char *path, tmp[16];
|
|
|
|
switch (type)
|
|
{
|
|
case NLS_SECTION_SORTKEYS: name = "sortdefault"; break;
|
|
case NLS_SECTION_CASEMAP: name = "l_intl"; break;
|
|
case NLS_SECTION_CODEPAGE: name = tmp; sprintf( tmp, "c_%03u", id ); break;
|
|
case NLS_SECTION_NORMALIZE:
|
|
switch (id)
|
|
{
|
|
case NormalizationC: name = "normnfc"; break;
|
|
case NormalizationD: name = "normnfd"; break;
|
|
case NormalizationKC: name = "normnfkc"; break;
|
|
case NormalizationKD: name = "normnfkd"; break;
|
|
case 13: name = "normidna"; break;
|
|
}
|
|
break;
|
|
}
|
|
if (!name) return NULL;
|
|
if (!(path = malloc( strlen(dir) + strlen(name) + 10 ))) return NULL;
|
|
sprintf( path, "%s/nls/%s.nls", dir, name );
|
|
return path;
|
|
}
|
|
|
|
static void *read_nls_file( ULONG type, ULONG id )
|
|
{
|
|
char *path = get_nls_file_path( type, id );
|
|
struct stat st;
|
|
void *data, *ret = NULL;
|
|
int fd;
|
|
|
|
if ((fd = open( path, O_RDONLY )) != -1)
|
|
{
|
|
fstat( fd, &st );
|
|
if ((data = malloc( st.st_size )) && st.st_size > 0x1000 &&
|
|
read( fd, data, st.st_size ) == st.st_size)
|
|
{
|
|
ret = data;
|
|
}
|
|
else
|
|
{
|
|
free( data );
|
|
data = NULL;
|
|
}
|
|
close( fd );
|
|
}
|
|
else ERR( "failed to load %u/%u\n", type, id );
|
|
free( path );
|
|
return ret;
|
|
}
|
|
|
|
static NTSTATUS open_nls_data_file( ULONG type, ULONG id, HANDLE *file )
|
|
{
|
|
static const WCHAR sortdirW[] = {'\\','?','?','\\','C',':','\\','w','i','n','d','o','w','s','\\',
|
|
'g','l','o','b','a','l','i','z','a','t','i','o','n','\\',
|
|
's','o','r','t','i','n','g','\\',0};
|
|
|
|
NTSTATUS status = STATUS_OBJECT_NAME_NOT_FOUND;
|
|
OBJECT_ATTRIBUTES attr;
|
|
UNICODE_STRING valueW;
|
|
WCHAR buffer[ARRAY_SIZE(sortdirW) + 16];
|
|
char *p, *path = get_nls_file_path( type, id );
|
|
|
|
if (!path) return STATUS_OBJECT_NAME_NOT_FOUND;
|
|
|
|
/* try to open file in system dir */
|
|
wcscpy( buffer, type == NLS_SECTION_SORTKEYS ? sortdirW : system_dir );
|
|
p = strrchr( path, '/' ) + 1;
|
|
ascii_to_unicode( buffer + wcslen(buffer), p, strlen(p) + 1 );
|
|
init_unicode_string( &valueW, buffer );
|
|
InitializeObjectAttributes( &attr, &valueW, 0, 0, NULL );
|
|
|
|
status = open_unix_file( file, path, GENERIC_READ, &attr, 0, FILE_SHARE_READ,
|
|
FILE_OPEN, FILE_SYNCHRONOUS_IO_ALERT, NULL, 0 );
|
|
free( path );
|
|
if (status != STATUS_NO_SUCH_FILE) return status;
|
|
|
|
if ((status = nt_to_unix_file_name( &attr, &path, FILE_OPEN ))) return status;
|
|
status = open_unix_file( file, path, GENERIC_READ, &attr, 0, FILE_SHARE_READ,
|
|
FILE_OPEN, FILE_SYNCHRONOUS_IO_ALERT, NULL, 0 );
|
|
free( path );
|
|
return status;
|
|
}
|
|
|
|
static NTSTATUS get_nls_section_name( ULONG type, ULONG id, WCHAR name[32] )
|
|
{
|
|
char buffer[32];
|
|
|
|
switch (type)
|
|
{
|
|
case NLS_SECTION_SORTKEYS:
|
|
if (id) return STATUS_INVALID_PARAMETER_1;
|
|
strcpy( buffer, "\\NLS\\NlsSectionSORTDEFAULT" );
|
|
break;
|
|
case NLS_SECTION_CASEMAP:
|
|
if (id) return STATUS_UNSUCCESSFUL;
|
|
strcpy( buffer, "\\NLS\\NlsSectionLANG_INTL" );
|
|
break;
|
|
case NLS_SECTION_CODEPAGE:
|
|
sprintf( buffer, "\\NLS\\NlsSectionCP%03u", id);
|
|
break;
|
|
case NLS_SECTION_NORMALIZE:
|
|
sprintf( buffer, "\\NLS\\NlsSectionNORM%08x", id);
|
|
break;
|
|
default:
|
|
return STATUS_INVALID_PARAMETER_1;
|
|
}
|
|
ascii_to_unicode( name, buffer, strlen(buffer) + 1 );
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
static int get_utf16( const WCHAR *src, unsigned int srclen, unsigned int *ch )
|
|
{
|
|
if (IS_HIGH_SURROGATE( src[0] ))
|
|
{
|
|
if (srclen <= 1) return 0;
|
|
if (!IS_LOW_SURROGATE( src[1] )) return 0;
|
|
*ch = 0x10000 + ((src[0] & 0x3ff) << 10) + (src[1] & 0x3ff);
|
|
return 2;
|
|
}
|
|
if (IS_LOW_SURROGATE( src[0] )) return 0;
|
|
*ch = src[0];
|
|
return 1;
|
|
}
|
|
|
|
|
|
#ifdef __APPLE__
|
|
|
|
/* The Apple filesystem enforces NFD so we need the compose tables to put it back into NFC */
|
|
|
|
struct norm_table
|
|
{
|
|
WCHAR name[13]; /* 00 file name */
|
|
USHORT checksum[3]; /* 1a checksum? */
|
|
USHORT version[4]; /* 20 Unicode version */
|
|
USHORT form; /* 28 normalization form */
|
|
USHORT len_factor; /* 2a factor for length estimates */
|
|
USHORT unknown1; /* 2c */
|
|
USHORT decomp_size; /* 2e decomposition hash size */
|
|
USHORT comp_size; /* 30 composition hash size */
|
|
USHORT unknown2; /* 32 */
|
|
USHORT classes; /* 34 combining classes table offset */
|
|
USHORT props_level1; /* 36 char properties table level 1 offset */
|
|
USHORT props_level2; /* 38 char properties table level 2 offset */
|
|
USHORT decomp_hash; /* 3a decomposition hash table offset */
|
|
USHORT decomp_map; /* 3c decomposition character map table offset */
|
|
USHORT decomp_seq; /* 3e decomposition character sequences offset */
|
|
USHORT comp_hash; /* 40 composition hash table offset */
|
|
USHORT comp_seq; /* 42 composition character sequences offset */
|
|
/* BYTE[] combining class values */
|
|
/* BYTE[0x2200] char properties index level 1 */
|
|
/* BYTE[] char properties index level 2 */
|
|
/* WORD[] decomposition hash table */
|
|
/* WORD[] decomposition character map */
|
|
/* WORD[] decomposition character sequences */
|
|
/* WORD[] composition hash table */
|
|
/* WORD[] composition character sequences */
|
|
};
|
|
|
|
static struct norm_table *nfc_table;
|
|
|
|
static void init_unix_codepage(void)
|
|
{
|
|
nfc_table = read_nls_file( NLS_SECTION_NORMALIZE, NormalizationC );
|
|
}
|
|
|
|
static void put_utf16( WCHAR *dst, unsigned int ch )
|
|
{
|
|
if (ch >= 0x10000)
|
|
{
|
|
ch -= 0x10000;
|
|
dst[0] = 0xd800 | (ch >> 10);
|
|
dst[1] = 0xdc00 | (ch & 0x3ff);
|
|
}
|
|
else dst[0] = ch;
|
|
}
|
|
|
|
static BYTE rol( BYTE val, BYTE count )
|
|
{
|
|
return (val << count) | (val >> (8 - count));
|
|
}
|
|
|
|
|
|
static BYTE get_char_props( const struct norm_table *info, unsigned int ch )
|
|
{
|
|
const BYTE *level1 = (const BYTE *)((const USHORT *)info + info->props_level1);
|
|
const BYTE *level2 = (const BYTE *)((const USHORT *)info + info->props_level2);
|
|
BYTE off = level1[ch / 128];
|
|
|
|
if (!off || off >= 0xfb) return rol( off, 5 );
|
|
return level2[(off - 1) * 128 + ch % 128];
|
|
}
|
|
|
|
static BYTE get_combining_class( const struct norm_table *info, unsigned int c )
|
|
{
|
|
const BYTE *classes = (const BYTE *)((const USHORT *)info + info->classes);
|
|
BYTE class = get_char_props( info, c ) & 0x3f;
|
|
|
|
if (class == 0x3f) return 0;
|
|
return classes[class];
|
|
}
|
|
|
|
#define HANGUL_SBASE 0xac00
|
|
#define HANGUL_LBASE 0x1100
|
|
#define HANGUL_VBASE 0x1161
|
|
#define HANGUL_TBASE 0x11a7
|
|
#define HANGUL_LCOUNT 19
|
|
#define HANGUL_VCOUNT 21
|
|
#define HANGUL_TCOUNT 28
|
|
#define HANGUL_NCOUNT (HANGUL_VCOUNT * HANGUL_TCOUNT)
|
|
#define HANGUL_SCOUNT (HANGUL_LCOUNT * HANGUL_NCOUNT)
|
|
|
|
static unsigned int compose_hangul( unsigned int ch1, unsigned int ch2 )
|
|
{
|
|
if (ch1 >= HANGUL_LBASE && ch1 < HANGUL_LBASE + HANGUL_LCOUNT)
|
|
{
|
|
int lindex = ch1 - HANGUL_LBASE;
|
|
int vindex = ch2 - HANGUL_VBASE;
|
|
if (vindex >= 0 && vindex < HANGUL_VCOUNT)
|
|
return HANGUL_SBASE + (lindex * HANGUL_VCOUNT + vindex) * HANGUL_TCOUNT;
|
|
}
|
|
if (ch1 >= HANGUL_SBASE && ch1 < HANGUL_SBASE + HANGUL_SCOUNT)
|
|
{
|
|
int sindex = ch1 - HANGUL_SBASE;
|
|
if (!(sindex % HANGUL_TCOUNT))
|
|
{
|
|
int tindex = ch2 - HANGUL_TBASE;
|
|
if (tindex > 0 && tindex < HANGUL_TCOUNT) return ch1 + tindex;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static unsigned int compose_chars( const struct norm_table *info, unsigned int ch1, unsigned int ch2 )
|
|
{
|
|
const USHORT *table = (const USHORT *)info + info->comp_hash;
|
|
const WCHAR *chars = (const USHORT *)info + info->comp_seq;
|
|
unsigned int hash, start, end, i, len, ch[3];
|
|
|
|
hash = (ch1 + 95 * ch2) % info->comp_size;
|
|
start = table[hash];
|
|
end = table[hash + 1];
|
|
while (start < end)
|
|
{
|
|
for (i = 0; i < 3; i++, start += len) len = get_utf16( chars + start, end - start, ch + i );
|
|
if (ch[0] == ch1 && ch[1] == ch2) return ch[2];
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static unsigned int compose_string( const struct norm_table *info, WCHAR *str, unsigned int srclen )
|
|
{
|
|
unsigned int i, ch, comp, len, start_ch = 0, last_starter = srclen;
|
|
BYTE class, prev_class = 0;
|
|
|
|
for (i = 0; i < srclen; i += len)
|
|
{
|
|
if (!(len = get_utf16( str + i, srclen - i, &ch ))) return 0;
|
|
class = get_combining_class( info, ch );
|
|
if (last_starter == srclen || (prev_class && prev_class >= class) ||
|
|
(!(comp = compose_hangul( start_ch, ch )) &&
|
|
!(comp = compose_chars( info, start_ch, ch ))))
|
|
{
|
|
if (!class)
|
|
{
|
|
last_starter = i;
|
|
start_ch = ch;
|
|
}
|
|
prev_class = class;
|
|
}
|
|
else
|
|
{
|
|
int comp_len = 1 + (comp >= 0x10000);
|
|
int start_len = 1 + (start_ch >= 0x10000);
|
|
|
|
if (comp_len != start_len)
|
|
memmove( str + last_starter + comp_len, str + last_starter + start_len,
|
|
(i - (last_starter + start_len)) * sizeof(WCHAR) );
|
|
memmove( str + i + comp_len - start_len, str + i + len, (srclen - i - len) * sizeof(WCHAR) );
|
|
srclen += comp_len - start_len - len;
|
|
start_ch = comp;
|
|
i = last_starter;
|
|
len = comp_len;
|
|
prev_class = 0;
|
|
put_utf16( str + i, comp );
|
|
}
|
|
}
|
|
return srclen;
|
|
}
|
|
|
|
#elif defined(__ANDROID__) /* Android always uses UTF-8 */
|
|
|
|
static void init_unix_codepage(void) { }
|
|
|
|
#else /* __APPLE__ || __ANDROID__ */
|
|
|
|
/* charset to codepage map, sorted by name */
|
|
static const struct { const char *name; UINT cp; } charset_names[] =
|
|
{
|
|
{ "ANSIX341968", 20127 },
|
|
{ "BIG5", 950 },
|
|
{ "BIG5HKSCS", 950 },
|
|
{ "CP1250", 1250 },
|
|
{ "CP1251", 1251 },
|
|
{ "CP1252", 1252 },
|
|
{ "CP1253", 1253 },
|
|
{ "CP1254", 1254 },
|
|
{ "CP1255", 1255 },
|
|
{ "CP1256", 1256 },
|
|
{ "CP1257", 1257 },
|
|
{ "CP1258", 1258 },
|
|
{ "CP932", 932 },
|
|
{ "CP936", 936 },
|
|
{ "CP949", 949 },
|
|
{ "CP950", 950 },
|
|
{ "EUCJP", 20932 },
|
|
{ "EUCKR", 949 },
|
|
{ "GB18030", 936 /* 54936 */ },
|
|
{ "GB2312", 936 },
|
|
{ "GBK", 936 },
|
|
{ "IBM037", 37 },
|
|
{ "IBM1026", 1026 },
|
|
{ "IBM424", 20424 },
|
|
{ "IBM437", 437 },
|
|
{ "IBM500", 500 },
|
|
{ "IBM850", 850 },
|
|
{ "IBM852", 852 },
|
|
{ "IBM855", 855 },
|
|
{ "IBM857", 857 },
|
|
{ "IBM860", 860 },
|
|
{ "IBM861", 861 },
|
|
{ "IBM862", 862 },
|
|
{ "IBM863", 863 },
|
|
{ "IBM864", 864 },
|
|
{ "IBM865", 865 },
|
|
{ "IBM866", 866 },
|
|
{ "IBM869", 869 },
|
|
{ "IBM874", 874 },
|
|
{ "IBM875", 875 },
|
|
{ "ISO88591", 28591 },
|
|
{ "ISO885913", 28603 },
|
|
{ "ISO885915", 28605 },
|
|
{ "ISO88592", 28592 },
|
|
{ "ISO88593", 28593 },
|
|
{ "ISO88594", 28594 },
|
|
{ "ISO88595", 28595 },
|
|
{ "ISO88596", 28596 },
|
|
{ "ISO88597", 28597 },
|
|
{ "ISO88598", 28598 },
|
|
{ "ISO88599", 28599 },
|
|
{ "KOI8R", 20866 },
|
|
{ "KOI8U", 21866 },
|
|
{ "TIS620", 28601 },
|
|
{ "UTF8", CP_UTF8 }
|
|
};
|
|
|
|
static void init_unix_cptable( USHORT *ptr )
|
|
{
|
|
unix_cp.data = ptr;
|
|
ptr += ptr[0];
|
|
unix_cp.wctable = ptr + ptr[0] + 1;
|
|
unix_cp.mbtable = ++ptr;
|
|
ptr += 256;
|
|
if (*ptr++) ptr += 256; /* glyph table */
|
|
if (*ptr) unix_cp.dbcs = ptr + 1; /* dbcs ranges */
|
|
}
|
|
|
|
static void init_unix_codepage(void)
|
|
{
|
|
char charset_name[16];
|
|
const char *name;
|
|
size_t i, j;
|
|
int min = 0, max = ARRAY_SIZE(charset_names) - 1;
|
|
|
|
setlocale( LC_CTYPE, "" );
|
|
if (!(name = nl_langinfo( CODESET ))) return;
|
|
|
|
/* remove punctuation characters from charset name */
|
|
for (i = j = 0; name[i] && j < sizeof(charset_name)-1; i++)
|
|
{
|
|
if (name[i] >= '0' && name[i] <= '9') charset_name[j++] = name[i];
|
|
else if (name[i] >= 'A' && name[i] <= 'Z') charset_name[j++] = name[i];
|
|
else if (name[i] >= 'a' && name[i] <= 'z') charset_name[j++] = name[i] + ('A' - 'a');
|
|
}
|
|
charset_name[j] = 0;
|
|
|
|
while (min <= max)
|
|
{
|
|
int pos = (min + max) / 2;
|
|
int res = strcmp( charset_names[pos].name, charset_name );
|
|
if (!res)
|
|
{
|
|
if (charset_names[pos].cp != CP_UTF8)
|
|
{
|
|
void *data = read_nls_file( NLS_SECTION_CODEPAGE, charset_names[pos].cp );
|
|
if (data) init_unix_cptable( data );
|
|
}
|
|
return;
|
|
}
|
|
if (res > 0) max = pos - 1;
|
|
else min = pos + 1;
|
|
}
|
|
ERR( "unrecognized charset '%s'\n", name );
|
|
}
|
|
|
|
#endif /* __APPLE__ || __ANDROID__ */
|
|
|
|
|
|
static inline SIZE_T get_env_length( const WCHAR *env )
|
|
{
|
|
const WCHAR *end = env;
|
|
while (*end) end += wcslen(end) + 1;
|
|
return end + 1 - env;
|
|
}
|
|
|
|
|
|
#define STARTS_WITH(var,str) (!strncmp( var, str, sizeof(str) - 1 ))
|
|
|
|
/***********************************************************************
|
|
* is_special_env_var
|
|
*
|
|
* Check if an environment variable needs to be handled specially when
|
|
* passed through the Unix environment (i.e. prefixed with "WINE").
|
|
*/
|
|
static BOOL is_special_env_var( const char *var )
|
|
{
|
|
return (STARTS_WITH( var, "PATH=" ) ||
|
|
STARTS_WITH( var, "PWD=" ) ||
|
|
STARTS_WITH( var, "HOME=" ) ||
|
|
STARTS_WITH( var, "TEMP=" ) ||
|
|
STARTS_WITH( var, "TMP=" ) ||
|
|
STARTS_WITH( var, "QT_" ) ||
|
|
STARTS_WITH( var, "VK_" ));
|
|
}
|
|
|
|
/* check if an environment variable changes dynamically in every new process */
|
|
static BOOL is_dynamic_env_var( const char *var )
|
|
{
|
|
return (STARTS_WITH( var, "WINEDLLOVERRIDES=" ) ||
|
|
STARTS_WITH( var, "WINEDATADIR=" ) ||
|
|
STARTS_WITH( var, "WINEHOMEDIR=" ) ||
|
|
STARTS_WITH( var, "WINEBUILDDIR=" ) ||
|
|
STARTS_WITH( var, "WINECONFIGDIR=" ) ||
|
|
STARTS_WITH( var, "WINEDLLDIR" ) ||
|
|
STARTS_WITH( var, "WINEUNIXCP=" ) ||
|
|
STARTS_WITH( var, "WINELOCALE=" ) ||
|
|
STARTS_WITH( var, "WINEUSERLOCALE=" ) ||
|
|
STARTS_WITH( var, "WINEUSERNAME=" ) ||
|
|
STARTS_WITH( var, "WINEPRELOADRESERVE=" ) ||
|
|
STARTS_WITH( var, "WINELOADERNOEXEC=" ) ||
|
|
STARTS_WITH( var, "WINESERVERSOCKET=" ));
|
|
}
|
|
|
|
static unsigned int decode_utf8_char( unsigned char ch, const char **str, const char *strend )
|
|
{
|
|
/* number of following bytes in sequence based on first byte value (for bytes above 0x7f) */
|
|
static const char utf8_length[128] =
|
|
{
|
|
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0x80-0x8f */
|
|
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0x90-0x9f */
|
|
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0xa0-0xaf */
|
|
0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0, /* 0xb0-0xbf */
|
|
0,0,1,1,1,1,1,1,1,1,1,1,1,1,1,1, /* 0xc0-0xcf */
|
|
1,1,1,1,1,1,1,1,1,1,1,1,1,1,1,1, /* 0xd0-0xdf */
|
|
2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2, /* 0xe0-0xef */
|
|
3,3,3,3,3,0,0,0,0,0,0,0,0,0,0,0 /* 0xf0-0xff */
|
|
};
|
|
|
|
/* first byte mask depending on UTF-8 sequence length */
|
|
static const unsigned char utf8_mask[4] = { 0x7f, 0x1f, 0x0f, 0x07 };
|
|
|
|
unsigned int len = utf8_length[ch - 0x80];
|
|
unsigned int res = ch & utf8_mask[len];
|
|
const char *end = *str + len;
|
|
|
|
if (end > strend)
|
|
{
|
|
*str = end;
|
|
return ~0;
|
|
}
|
|
switch (len)
|
|
{
|
|
case 3:
|
|
if ((ch = end[-3] ^ 0x80) >= 0x40) break;
|
|
res = (res << 6) | ch;
|
|
(*str)++;
|
|
if (res < 0x10) break;
|
|
case 2:
|
|
if ((ch = end[-2] ^ 0x80) >= 0x40) break;
|
|
res = (res << 6) | ch;
|
|
if (res >= 0x110000 >> 6) break;
|
|
(*str)++;
|
|
if (res < 0x20) break;
|
|
if (res >= 0xd800 >> 6 && res <= 0xdfff >> 6) break;
|
|
case 1:
|
|
if ((ch = end[-1] ^ 0x80) >= 0x40) break;
|
|
res = (res << 6) | ch;
|
|
(*str)++;
|
|
if (res < 0x80) break;
|
|
return res;
|
|
}
|
|
return ~0;
|
|
}
|
|
|
|
|
|
/******************************************************************
|
|
* ntdll_umbstowcs
|
|
*/
|
|
DWORD ntdll_umbstowcs( const char *src, DWORD srclen, WCHAR *dst, DWORD dstlen )
|
|
{
|
|
DWORD reslen;
|
|
|
|
if (unix_cp.data)
|
|
{
|
|
DWORD i;
|
|
|
|
if (unix_cp.dbcs)
|
|
{
|
|
for (i = dstlen; srclen && i; i--, srclen--, src++, dst++)
|
|
{
|
|
USHORT off = unix_cp.dbcs[(unsigned char)*src];
|
|
if (off && srclen > 1)
|
|
{
|
|
src++;
|
|
srclen--;
|
|
*dst = unix_cp.dbcs[off + (unsigned char)*src];
|
|
}
|
|
else *dst = unix_cp.mbtable[(unsigned char)*src];
|
|
}
|
|
reslen = dstlen - i;
|
|
}
|
|
else
|
|
{
|
|
reslen = min( srclen, dstlen );
|
|
for (i = 0; i < reslen; i++) dst[i] = unix_cp.mbtable[(unsigned char)src[i]];
|
|
}
|
|
}
|
|
else /* utf-8 */
|
|
{
|
|
reslen = 0;
|
|
RtlUTF8ToUnicodeN( dst, dstlen * sizeof(WCHAR), &reslen, src, srclen );
|
|
reslen /= sizeof(WCHAR);
|
|
#ifdef __APPLE__ /* work around broken Mac OS X filesystem that enforces NFD */
|
|
if (reslen && nfc_table) reslen = compose_string( nfc_table, dst, reslen );
|
|
#endif
|
|
}
|
|
return reslen;
|
|
}
|
|
|
|
|
|
/******************************************************************
|
|
* ntdll_wcstoumbs
|
|
*/
|
|
int ntdll_wcstoumbs( const WCHAR *src, DWORD srclen, char *dst, DWORD dstlen, BOOL strict )
|
|
{
|
|
DWORD i, reslen;
|
|
|
|
if (unix_cp.data)
|
|
{
|
|
if (unix_cp.dbcs)
|
|
{
|
|
const unsigned short *uni2cp = unix_cp.wctable;
|
|
for (i = dstlen; srclen && i; i--, srclen--, src++)
|
|
{
|
|
unsigned short ch = uni2cp[*src];
|
|
if (ch >> 8)
|
|
{
|
|
if (strict && unix_cp.dbcs[unix_cp.dbcs[ch >> 8] + (ch & 0xff)] != *src) return -1;
|
|
if (i == 1) break; /* do not output a partial char */
|
|
i--;
|
|
*dst++ = ch >> 8;
|
|
}
|
|
else
|
|
{
|
|
if (unix_cp.mbtable[ch] != *src) return -1;
|
|
*dst++ = (char)ch;
|
|
}
|
|
}
|
|
reslen = dstlen - i;
|
|
}
|
|
else
|
|
{
|
|
const unsigned char *uni2cp = unix_cp.wctable;
|
|
reslen = min( srclen, dstlen );
|
|
for (i = 0; i < reslen; i++)
|
|
{
|
|
unsigned char ch = uni2cp[src[i]];
|
|
if (strict && unix_cp.mbtable[ch] != src[i]) return -1;
|
|
dst[i] = ch;
|
|
}
|
|
}
|
|
}
|
|
else /* utf-8 */
|
|
{
|
|
char *end;
|
|
unsigned int val;
|
|
|
|
for (end = dst + dstlen; srclen; srclen--, src++)
|
|
{
|
|
WCHAR ch = *src;
|
|
|
|
if (ch < 0x80) /* 0x00-0x7f: 1 byte */
|
|
{
|
|
if (dst > end - 1) break;
|
|
*dst++ = ch;
|
|
continue;
|
|
}
|
|
if (ch < 0x800) /* 0x80-0x7ff: 2 bytes */
|
|
{
|
|
if (dst > end - 2) break;
|
|
dst[1] = 0x80 | (ch & 0x3f);
|
|
ch >>= 6;
|
|
dst[0] = 0xc0 | ch;
|
|
dst += 2;
|
|
continue;
|
|
}
|
|
if (!get_utf16( src, srclen, &val ))
|
|
{
|
|
if (strict) return -1;
|
|
val = 0xfffd;
|
|
}
|
|
if (val < 0x10000) /* 0x800-0xffff: 3 bytes */
|
|
{
|
|
if (dst > end - 3) break;
|
|
dst[2] = 0x80 | (val & 0x3f);
|
|
val >>= 6;
|
|
dst[1] = 0x80 | (val & 0x3f);
|
|
val >>= 6;
|
|
dst[0] = 0xe0 | val;
|
|
dst += 3;
|
|
}
|
|
else /* 0x10000-0x10ffff: 4 bytes */
|
|
{
|
|
if (dst > end - 4) break;
|
|
dst[3] = 0x80 | (val & 0x3f);
|
|
val >>= 6;
|
|
dst[2] = 0x80 | (val & 0x3f);
|
|
val >>= 6;
|
|
dst[1] = 0x80 | (val & 0x3f);
|
|
val >>= 6;
|
|
dst[0] = 0xf0 | val;
|
|
dst += 4;
|
|
src++;
|
|
srclen--;
|
|
}
|
|
}
|
|
reslen = dstlen - (end - dst);
|
|
}
|
|
return reslen;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* ntdll_get_build_dir
|
|
*/
|
|
const char *ntdll_get_build_dir(void)
|
|
{
|
|
return build_dir;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* ntdll_get_data_dir
|
|
*/
|
|
const char *ntdll_get_data_dir(void)
|
|
{
|
|
return data_dir;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* build_envp
|
|
*
|
|
* Build the environment of a new child process.
|
|
*/
|
|
char **build_envp( const WCHAR *envW )
|
|
{
|
|
static const char * const unix_vars[] = { "PATH", "TEMP", "TMP", "HOME" };
|
|
char **envp;
|
|
char *env, *p;
|
|
int count = 1, length, lenW;
|
|
unsigned int i;
|
|
|
|
lenW = get_env_length( envW );
|
|
if (!(env = malloc( lenW * 3 ))) return NULL;
|
|
length = ntdll_wcstoumbs( envW, lenW, env, lenW * 3, FALSE );
|
|
|
|
for (p = env; *p; p += strlen(p) + 1, count++)
|
|
{
|
|
if (is_dynamic_env_var( p )) continue;
|
|
if (is_special_env_var( p )) length += 4; /* prefix it with "WINE" */
|
|
}
|
|
|
|
for (i = 0; i < ARRAY_SIZE( unix_vars ); i++)
|
|
{
|
|
if (!(p = getenv(unix_vars[i]))) continue;
|
|
length += strlen(unix_vars[i]) + strlen(p) + 2;
|
|
count++;
|
|
}
|
|
|
|
if ((envp = malloc( count * sizeof(*envp) + length )))
|
|
{
|
|
char **envptr = envp;
|
|
char *dst = (char *)(envp + count);
|
|
|
|
/* some variables must not be modified, so we get them directly from the unix env */
|
|
for (i = 0; i < ARRAY_SIZE( unix_vars ); i++)
|
|
{
|
|
if (!(p = getenv( unix_vars[i] ))) continue;
|
|
*envptr++ = strcpy( dst, unix_vars[i] );
|
|
strcat( dst, "=" );
|
|
strcat( dst, p );
|
|
dst += strlen(dst) + 1;
|
|
}
|
|
|
|
/* now put the Windows environment strings */
|
|
for (p = env; *p; p += strlen(p) + 1)
|
|
{
|
|
if (*p == '=') continue; /* skip drive curdirs, this crashes some unix apps */
|
|
if (is_dynamic_env_var( p )) continue;
|
|
if (is_special_env_var( p )) /* prefix it with "WINE" */
|
|
{
|
|
*envptr++ = strcpy( dst, "WINE" );
|
|
strcat( dst, p );
|
|
}
|
|
else
|
|
{
|
|
*envptr++ = strcpy( dst, p );
|
|
}
|
|
dst += strlen(dst) + 1;
|
|
}
|
|
*envptr = 0;
|
|
}
|
|
free( env );
|
|
return envp;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* set_process_name
|
|
*
|
|
* Change the process name in the ps output.
|
|
*/
|
|
static void set_process_name( const char *name )
|
|
{
|
|
char *p;
|
|
|
|
#ifdef HAVE_SETPROCTITLE
|
|
setproctitle("-%s", name );
|
|
#endif
|
|
if ((p = strrchr( name, '\\' ))) name = p + 1;
|
|
if ((p = strrchr( name, '/' ))) name = p + 1;
|
|
#ifdef HAVE_SETPROGNAME
|
|
setprogname( name );
|
|
#endif
|
|
#ifdef HAVE_PRCTL
|
|
#ifndef PR_SET_NAME
|
|
# define PR_SET_NAME 15
|
|
#endif
|
|
prctl( PR_SET_NAME, name );
|
|
#endif
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* rebuild_argv
|
|
*
|
|
* Build the main argv by removing argv[0].
|
|
*/
|
|
static void rebuild_argv(void)
|
|
{
|
|
BOOL shift_strings = FALSE;
|
|
int i;
|
|
|
|
#ifndef HAVE_SETPROCTITLE
|
|
for (i = 1; i < main_argc; i++)
|
|
if (main_argv[i - 1] + strlen( main_argv[i - 1] ) + 1 != main_argv[i]) break;
|
|
shift_strings = (i == main_argc);
|
|
#endif
|
|
if (shift_strings)
|
|
{
|
|
int offset = main_argv[1] - main_argv[0];
|
|
char *end = main_argv[main_argc - 1] + strlen(main_argv[main_argc - 1]) + 1;
|
|
memmove( main_argv[0], main_argv[1], end - main_argv[1] );
|
|
memset( end - offset, 0, offset );
|
|
for (i = 1; i < main_argc; i++) main_argv[i - 1] = main_argv[i] - offset;
|
|
}
|
|
else memmove( main_argv, main_argv + 1, (main_argc - 1) * sizeof(main_argv[0]) );
|
|
|
|
main_argv[--main_argc] = NULL;
|
|
set_process_name( main_argv[0] );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* prepend_argv
|
|
*
|
|
* Prepend values to the main argv, removing the original argv[0].
|
|
*/
|
|
static void prepend_argv( const char **args, int count )
|
|
{
|
|
char **new_argv;
|
|
char *p, *end;
|
|
BOOL write_strings = FALSE;
|
|
int i, total = 0, new_argc = main_argc + count - 1;
|
|
|
|
for (i = 0; i < count; i++) total += strlen(args[i]) + 1;
|
|
for (i = 1; i < main_argc; i++) total += strlen(main_argv[i]) + 1;
|
|
|
|
new_argv = malloc( (new_argc + 1) * sizeof(*new_argv) + total );
|
|
p = (char *)(new_argv + new_argc + 1);
|
|
for (i = 0; i < count; i++)
|
|
{
|
|
new_argv[i] = p;
|
|
strcpy( p, args[i] );
|
|
p += strlen(p) + 1;
|
|
}
|
|
for (i = 1; i < main_argc; i++)
|
|
{
|
|
new_argv[count + i - 1] = p;
|
|
strcpy( p, main_argv[i] );
|
|
p += strlen(p) + 1;
|
|
}
|
|
new_argv[new_argc] = NULL;
|
|
|
|
/* copy what we can over the original argv */
|
|
|
|
#ifndef HAVE_SETPROCTITLE
|
|
for (i = 1; i < main_argc; i++)
|
|
if (main_argv[i - 1] + strlen(main_argv[i - 1]) + 1 != main_argv[i]) break;
|
|
write_strings = (i == main_argc);
|
|
#endif
|
|
if (write_strings)
|
|
{
|
|
p = main_argv[0];
|
|
end = main_argv[main_argc - 1] + strlen(main_argv[main_argc - 1]) + 1;
|
|
|
|
for (i = 0; i < min( new_argc, main_argc ) && p < end; i++)
|
|
{
|
|
int len = min( end - p - 1, strlen(new_argv[i]) );
|
|
|
|
memcpy( p, new_argv[i], len );
|
|
main_argv[i] = p;
|
|
p += len;
|
|
*p++ = 0;
|
|
}
|
|
memset( p, 0, end - p );
|
|
main_argv[i] = NULL;
|
|
}
|
|
else
|
|
{
|
|
memcpy( main_argv, new_argv, min( new_argc, main_argc ) * sizeof(*new_argv) );
|
|
main_argv[min( new_argc, main_argc )] = NULL;
|
|
}
|
|
|
|
main_argc = new_argc;
|
|
main_argv = new_argv;
|
|
set_process_name( main_argv[0] );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* build_wargv
|
|
*
|
|
* Build the Unicode argv array, replacing argv[0] by the image name.
|
|
*/
|
|
static WCHAR **build_wargv( const WCHAR *image )
|
|
{
|
|
int argc;
|
|
WCHAR *p, **wargv;
|
|
DWORD total = wcslen(image) + 1;
|
|
|
|
for (argc = 1; main_argv[argc]; argc++) total += strlen(main_argv[argc]) + 1;
|
|
|
|
wargv = malloc( total * sizeof(WCHAR) + (argc + 1) * sizeof(*wargv) );
|
|
p = (WCHAR *)(wargv + argc + 1);
|
|
wargv[0] = p;
|
|
wcscpy( p, image );
|
|
total -= wcslen( p ) + 1;
|
|
p += wcslen( p ) + 1;
|
|
for (argc = 1; main_argv[argc]; argc++)
|
|
{
|
|
DWORD reslen = ntdll_umbstowcs( main_argv[argc], strlen(main_argv[argc]) + 1, p, total );
|
|
wargv[argc] = p;
|
|
p += reslen;
|
|
total -= reslen;
|
|
}
|
|
wargv[argc] = NULL;
|
|
return wargv;
|
|
}
|
|
|
|
|
|
/* Unix format is: lang[_country][.charset][@modifier]
|
|
* Windows format is: lang[-script][-country][_modifier] */
|
|
static BOOL unix_to_win_locale( const char *unix_name, char *win_name )
|
|
{
|
|
static const char sep[] = "_.@";
|
|
char buffer[LOCALE_NAME_MAX_LENGTH];
|
|
char *p, *country = NULL, *modifier = NULL;
|
|
|
|
if (!unix_name || !unix_name[0] || !strcmp( unix_name, "C" ))
|
|
{
|
|
unix_name = getenv( "LC_ALL" );
|
|
if (!unix_name || !unix_name[0]) return FALSE;
|
|
}
|
|
|
|
if (strlen( unix_name ) >= LOCALE_NAME_MAX_LENGTH) return FALSE;
|
|
strcpy( buffer, unix_name );
|
|
if (!(p = strpbrk( buffer, sep )))
|
|
{
|
|
if (!strcmp( buffer, "POSIX" ) || !strcmp( buffer, "C" ))
|
|
strcpy( win_name, "en-US" );
|
|
else
|
|
strcpy( win_name, buffer );
|
|
return TRUE;
|
|
}
|
|
|
|
if (*p == '_')
|
|
{
|
|
*p++ = 0;
|
|
country = p;
|
|
p = strpbrk( p, sep + 1 );
|
|
}
|
|
if (p && *p == '.')
|
|
{
|
|
*p++ = 0;
|
|
/* charset, ignore */
|
|
p = strchr( p, '@' );
|
|
}
|
|
if (p)
|
|
{
|
|
*p++ = 0;
|
|
modifier = p;
|
|
}
|
|
|
|
/* rebuild a Windows name */
|
|
|
|
strcpy( win_name, buffer );
|
|
if (modifier)
|
|
{
|
|
if (!strcmp( modifier, "latin" )) strcat( win_name, "-Latn" );
|
|
else if (!strcmp( modifier, "euro" )) {} /* ignore */
|
|
else return FALSE;
|
|
}
|
|
if (country)
|
|
{
|
|
p = win_name + strlen(win_name);
|
|
*p++ = '-';
|
|
strcpy( p, country );
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
/******************************************************************
|
|
* init_locale
|
|
*/
|
|
static void init_locale(void)
|
|
{
|
|
setlocale( LC_ALL, "" );
|
|
if (!unix_to_win_locale( setlocale( LC_CTYPE, NULL ), system_locale )) system_locale[0] = 0;
|
|
if (!unix_to_win_locale( setlocale( LC_MESSAGES, NULL ), user_locale )) user_locale[0] = 0;
|
|
|
|
#ifdef __APPLE__
|
|
if (!system_locale[0])
|
|
{
|
|
CFLocaleRef locale = CFLocaleCopyCurrent();
|
|
CFStringRef lang = CFLocaleGetValue( locale, kCFLocaleLanguageCode );
|
|
CFStringRef country = CFLocaleGetValue( locale, kCFLocaleCountryCode );
|
|
CFStringRef locale_string;
|
|
|
|
if (country)
|
|
locale_string = CFStringCreateWithFormat(NULL, NULL, CFSTR("%@-%@"), lang, country);
|
|
else
|
|
locale_string = CFStringCreateCopy(NULL, lang);
|
|
|
|
CFStringGetCString(locale_string, system_locale, sizeof(system_locale), kCFStringEncodingUTF8);
|
|
CFRelease(locale);
|
|
CFRelease(locale_string);
|
|
}
|
|
if (!user_locale[0])
|
|
{
|
|
/* Retrieve the preferred language as chosen in System Preferences. */
|
|
CFArrayRef preferred_langs = CFLocaleCopyPreferredLanguages();
|
|
if (preferred_langs && CFArrayGetCount( preferred_langs ))
|
|
{
|
|
CFStringRef preferred_lang = CFArrayGetValueAtIndex( preferred_langs, 0 );
|
|
CFDictionaryRef components = CFLocaleCreateComponentsFromLocaleIdentifier( NULL, preferred_lang );
|
|
if (components)
|
|
{
|
|
CFStringRef lang = CFDictionaryGetValue( components, kCFLocaleLanguageCode );
|
|
CFStringRef country = CFDictionaryGetValue( components, kCFLocaleCountryCode );
|
|
CFLocaleRef locale = NULL;
|
|
CFStringRef locale_string;
|
|
|
|
if (!country)
|
|
{
|
|
locale = CFLocaleCopyCurrent();
|
|
country = CFLocaleGetValue( locale, kCFLocaleCountryCode );
|
|
}
|
|
if (country)
|
|
locale_string = CFStringCreateWithFormat( NULL, NULL, CFSTR("%@-%@"), lang, country );
|
|
else
|
|
locale_string = CFStringCreateCopy( NULL, lang );
|
|
CFStringGetCString( locale_string, user_locale, sizeof(user_locale), kCFStringEncodingUTF8 );
|
|
CFRelease( locale_string );
|
|
if (locale) CFRelease( locale );
|
|
CFRelease( components );
|
|
}
|
|
}
|
|
if (preferred_langs) CFRelease( preferred_langs );
|
|
}
|
|
#endif
|
|
setlocale( LC_NUMERIC, "C" ); /* FIXME: oleaut32 depends on this */
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* init_environment
|
|
*/
|
|
void init_environment( int argc, char *argv[], char *envp[] )
|
|
{
|
|
USHORT *case_table;
|
|
|
|
init_unix_codepage();
|
|
init_locale();
|
|
|
|
if ((case_table = read_nls_file( NLS_SECTION_CASEMAP, 0 )))
|
|
{
|
|
uctable = case_table + 2;
|
|
lctable = case_table + case_table[1] + 2;
|
|
}
|
|
|
|
main_argc = argc;
|
|
main_argv = argv;
|
|
main_envp = envp;
|
|
}
|
|
|
|
|
|
static const char overrides_help_message[] =
|
|
"Syntax:\n"
|
|
" WINEDLLOVERRIDES=\"entry;entry;entry...\"\n"
|
|
" where each entry is of the form:\n"
|
|
" module[,module...]={native|builtin}[,{b|n}]\n"
|
|
"\n"
|
|
" Only the first letter of the override (native or builtin)\n"
|
|
" is significant.\n\n"
|
|
"Example:\n"
|
|
" WINEDLLOVERRIDES=\"comdlg32=n,b;shell32,shlwapi=b\"\n";
|
|
|
|
/*************************************************************************
|
|
* get_initial_environment
|
|
*
|
|
* Return the initial environment.
|
|
*/
|
|
static WCHAR *get_initial_environment( SIZE_T *pos, SIZE_T *size )
|
|
{
|
|
char **e;
|
|
WCHAR *env, *ptr, *end;
|
|
|
|
/* estimate needed size */
|
|
*size = 1;
|
|
for (e = main_envp; *e; e++) *size += strlen(*e) + 1;
|
|
|
|
if (!(env = malloc( *size * sizeof(WCHAR) ))) return NULL;
|
|
ptr = env;
|
|
end = env + *size - 1;
|
|
for (e = main_envp; *e && ptr < end; e++)
|
|
{
|
|
char *str = *e;
|
|
|
|
/* skip Unix special variables and use the Wine variants instead */
|
|
if (!strncmp( str, "WINE", 4 ))
|
|
{
|
|
if (is_special_env_var( str + 4 )) str += 4;
|
|
else if (!strcmp( str, "WINEDLLOVERRIDES=help" ))
|
|
{
|
|
MESSAGE( overrides_help_message );
|
|
exit(0);
|
|
}
|
|
}
|
|
else if (is_special_env_var( str )) continue; /* skip it */
|
|
|
|
if (is_dynamic_env_var( str )) continue;
|
|
ptr += ntdll_umbstowcs( str, strlen(str) + 1, ptr, end - ptr );
|
|
}
|
|
*pos = ptr - env;
|
|
return env;
|
|
}
|
|
|
|
|
|
static WCHAR *find_env_var( WCHAR *env, SIZE_T size, const WCHAR *name, SIZE_T namelen )
|
|
{
|
|
WCHAR *p = env;
|
|
|
|
while (p < env + size)
|
|
{
|
|
if (!wcsnicmp( p, name, namelen ) && p[namelen] == '=') return p;
|
|
p += wcslen(p) + 1;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
static WCHAR *get_env_var( WCHAR *env, SIZE_T size, const WCHAR *name, SIZE_T namelen )
|
|
{
|
|
WCHAR *ret = NULL, *var = find_env_var( env, size, name, namelen );
|
|
|
|
if (var)
|
|
{
|
|
var += namelen + 1; /* skip name */
|
|
if ((ret = malloc( (wcslen(var) + 1) * sizeof(WCHAR) ))) wcscpy( ret, var );
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/* set an environment variable, replacing it if it exists */
|
|
static void set_env_var( WCHAR **env, SIZE_T *pos, SIZE_T *size,
|
|
const WCHAR *name, SIZE_T namelen, const WCHAR *value )
|
|
{
|
|
WCHAR *p;
|
|
SIZE_T len;
|
|
|
|
/* remove existing string */
|
|
if ((p = find_env_var( *env, *pos, name, namelen )))
|
|
{
|
|
len = wcslen(p) + 1;
|
|
memmove( p, p + len, (*pos - (p + len - *env)) * sizeof(WCHAR) );
|
|
*pos -= len;
|
|
}
|
|
|
|
if (!value) return;
|
|
len = wcslen( value );
|
|
if (*pos + namelen + len + 3 > *size)
|
|
{
|
|
*size = max( *size * 2, *pos + namelen + len + 3 );
|
|
*env = realloc( *env, *size * sizeof(WCHAR) );
|
|
}
|
|
memcpy( *env + *pos, name, namelen * sizeof(WCHAR) );
|
|
(*env)[*pos + namelen] = '=';
|
|
memcpy( *env + *pos + namelen + 1, value, (len + 1) * sizeof(WCHAR) );
|
|
*pos += namelen + len + 2;
|
|
}
|
|
|
|
static void append_envW( WCHAR **env, SIZE_T *pos, SIZE_T *size, const char *name, const WCHAR *value )
|
|
{
|
|
WCHAR nameW[32];
|
|
|
|
ascii_to_unicode( nameW, name, strlen(name) + 1 );
|
|
set_env_var( env, pos, size, nameW, wcslen(nameW), value );
|
|
}
|
|
|
|
static void append_envA( WCHAR **env, SIZE_T *pos, SIZE_T *size, const char *name, const char *value )
|
|
{
|
|
if (value)
|
|
{
|
|
SIZE_T len = strlen(value) + 1;
|
|
WCHAR *valueW = malloc( len * sizeof(WCHAR) );
|
|
ntdll_umbstowcs( value, len, valueW, len );
|
|
append_envW( env, pos, size, name, valueW );
|
|
free( valueW );
|
|
}
|
|
else append_envW( env, pos, size, name, NULL );
|
|
}
|
|
|
|
/* set an environment variable for one of the wine path variables */
|
|
static void add_path_var( WCHAR **env, SIZE_T *pos, SIZE_T *size, const char *name, const char *path )
|
|
{
|
|
WCHAR *nt_name = NULL;
|
|
|
|
if (path && unix_to_nt_file_name( path, &nt_name )) return;
|
|
append_envW( env, pos, size, name, nt_name );
|
|
free( nt_name );
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* add_dynamic_environment
|
|
*
|
|
* Add the environment variables that can differ between processes.
|
|
*/
|
|
static void add_dynamic_environment( WCHAR **env, SIZE_T *pos, SIZE_T *size )
|
|
{
|
|
const char *overrides = getenv( "WINEDLLOVERRIDES" );
|
|
DWORD i;
|
|
char str[22];
|
|
|
|
add_path_var( env, pos, size, "WINEDATADIR", data_dir );
|
|
add_path_var( env, pos, size, "WINEHOMEDIR", home_dir );
|
|
add_path_var( env, pos, size, "WINEBUILDDIR", build_dir );
|
|
add_path_var( env, pos, size, "WINECONFIGDIR", config_dir );
|
|
for (i = 0; dll_paths[i]; i++)
|
|
{
|
|
sprintf( str, "WINEDLLDIR%u", i );
|
|
add_path_var( env, pos, size, str, dll_paths[i] );
|
|
}
|
|
sprintf( str, "WINEDLLDIR%u", i );
|
|
append_envW( env, pos, size, str, NULL );
|
|
append_envA( env, pos, size, "WINEUSERNAME", user_name );
|
|
append_envA( env, pos, size, "WINEDLLOVERRIDES", overrides );
|
|
if (unix_cp.data)
|
|
{
|
|
sprintf( str, "%u", unix_cp.data[1] );
|
|
append_envA( env, pos, size, "WINEUNIXCP", str );
|
|
}
|
|
else append_envW( env, pos, size, "WINEUNIXCP", NULL );
|
|
append_envA( env, pos, size, "WINELOCALE", system_locale );
|
|
append_envA( env, pos, size, "WINEUSERLOCALE",
|
|
strcmp( user_locale, system_locale ) ? user_locale : NULL );
|
|
append_envA( env, pos, size, "SystemDrive", "C:" );
|
|
append_envA( env, pos, size, "SystemRoot", "C:\\windows" );
|
|
}
|
|
|
|
|
|
static WCHAR *expand_value( WCHAR *env, SIZE_T size, const WCHAR *src, SIZE_T src_len )
|
|
{
|
|
SIZE_T len, retlen = src_len, count = 0;
|
|
const WCHAR *var;
|
|
WCHAR *ret;
|
|
|
|
ret = malloc( retlen * sizeof(WCHAR) );
|
|
while (src_len)
|
|
{
|
|
if (*src != '%')
|
|
{
|
|
for (len = 0; len < src_len; len++) if (src[len] == '%') break;
|
|
var = src;
|
|
src += len;
|
|
src_len -= len;
|
|
}
|
|
else /* we are at the start of a variable */
|
|
{
|
|
for (len = 1; len < src_len; len++) if (src[len] == '%') break;
|
|
if (len < src_len)
|
|
{
|
|
if ((var = find_env_var( env, size, src + 1, len - 1 )))
|
|
{
|
|
src += len + 1; /* skip the variable name */
|
|
src_len -= len + 1;
|
|
var += len;
|
|
len = wcslen(var);
|
|
}
|
|
else
|
|
{
|
|
var = src; /* copy original name instead */
|
|
len++;
|
|
src += len;
|
|
src_len -= len;
|
|
}
|
|
}
|
|
else /* unfinished variable name, ignore it */
|
|
{
|
|
var = src;
|
|
src += len;
|
|
src_len = 0;
|
|
}
|
|
}
|
|
if (len >= retlen - count)
|
|
{
|
|
retlen *= 2;
|
|
ret = realloc( ret, retlen * sizeof(WCHAR) );
|
|
}
|
|
memcpy( ret + count, var, len * sizeof(WCHAR) );
|
|
count += len;
|
|
}
|
|
ret[count] = 0;
|
|
return ret;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* add_registry_variables
|
|
*
|
|
* Set environment variables by enumerating the values of a key;
|
|
* helper for add_registry_environment().
|
|
* Note that Windows happily truncates the value if it's too big.
|
|
*/
|
|
static void add_registry_variables( WCHAR **env, SIZE_T *pos, SIZE_T *size, HANDLE key )
|
|
{
|
|
static const WCHAR pathW[] = {'P','A','T','H'};
|
|
NTSTATUS status;
|
|
DWORD index = 0, info_size, namelen, datalen;
|
|
WCHAR *data, *value, *p;
|
|
WCHAR buffer[offsetof(KEY_VALUE_FULL_INFORMATION, Name[1024]) / sizeof(WCHAR)];
|
|
KEY_VALUE_FULL_INFORMATION *info = (KEY_VALUE_FULL_INFORMATION *)buffer;
|
|
|
|
for (;;)
|
|
{
|
|
status = NtEnumerateValueKey( key, index++, KeyValueFullInformation,
|
|
buffer, sizeof(buffer) - sizeof(WCHAR), &info_size );
|
|
if (status != STATUS_SUCCESS && status != STATUS_BUFFER_OVERFLOW) break;
|
|
|
|
value = data = buffer + info->DataOffset / sizeof(WCHAR);
|
|
datalen = info->DataLength / sizeof(WCHAR);
|
|
namelen = info->NameLength / sizeof(WCHAR);
|
|
|
|
if (datalen && !data[datalen - 1]) datalen--; /* don't count terminating null if any */
|
|
if (!datalen) continue;
|
|
data[datalen] = 0;
|
|
if (info->Type == REG_EXPAND_SZ) value = expand_value( *env, *pos, data, datalen );
|
|
|
|
/* PATH is magic */
|
|
if (namelen == 4 && !wcsnicmp( info->Name, pathW, 4 ) && (p = find_env_var( *env, *pos, pathW, 4 )))
|
|
{
|
|
static const WCHAR sepW[] = {';',0};
|
|
WCHAR *newpath = malloc( (wcslen(p) - 3 + wcslen(value)) * sizeof(WCHAR) );
|
|
wcscpy( newpath, p + 5 );
|
|
wcscat( newpath, sepW );
|
|
wcscat( newpath, value );
|
|
if (value != data) free( value );
|
|
value = newpath;
|
|
}
|
|
|
|
set_env_var( env, pos, size, info->Name, namelen, value );
|
|
if (value != data) free( value );
|
|
}
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* get_registry_value
|
|
*/
|
|
static WCHAR *get_registry_value( WCHAR *env, SIZE_T pos, HKEY key, const WCHAR *name )
|
|
{
|
|
WCHAR buffer[offsetof(KEY_VALUE_PARTIAL_INFORMATION, Data[1024 * sizeof(WCHAR)])];
|
|
KEY_VALUE_PARTIAL_INFORMATION *info = (KEY_VALUE_PARTIAL_INFORMATION *)buffer;
|
|
DWORD len, size = sizeof(buffer) - sizeof(WCHAR);
|
|
WCHAR *ret = NULL;
|
|
UNICODE_STRING nameW;
|
|
|
|
init_unicode_string( &nameW, name );
|
|
if (NtQueryValueKey( key, &nameW, KeyValuePartialInformation, buffer, size, &size )) return NULL;
|
|
if (size <= offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data )) return NULL;
|
|
len = size - offsetof( KEY_VALUE_PARTIAL_INFORMATION, Data );
|
|
|
|
if (info->Type == REG_EXPAND_SZ)
|
|
{
|
|
ret = expand_value( env, pos, (WCHAR *)info->Data, len / sizeof(WCHAR) );
|
|
}
|
|
else
|
|
{
|
|
ret = malloc( len + sizeof(WCHAR) );
|
|
memcpy( ret, info->Data, len );
|
|
ret[len / sizeof(WCHAR)] = 0;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* add_registry_environment
|
|
*
|
|
* Set the environment variables specified in the registry.
|
|
*/
|
|
static void add_registry_environment( WCHAR **env, SIZE_T *pos, SIZE_T *size )
|
|
{
|
|
static const WCHAR syskeyW[] = {'\\','R','e','g','i','s','t','r','y',
|
|
'\\','M','a','c','h','i','n','e',
|
|
'\\','S','y','s','t','e','m',
|
|
'\\','C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t',
|
|
'\\','C','o','n','t','r','o','l',
|
|
'\\','S','e','s','s','i','o','n',' ','M','a','n','a','g','e','r',
|
|
'\\','E','n','v','i','r','o','n','m','e','n','t',0};
|
|
static const WCHAR profileW[] = {'\\','R','e','g','i','s','t','r','y',
|
|
'\\','M','a','c','h','i','n','e','\\',
|
|
'S','o','f','t','w','a','r','e','\\',
|
|
'M','i','c','r','o','s','o','f','t','\\',
|
|
'W','i','n','d','o','w','s',' ','N','T','\\',
|
|
'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
|
|
'P','r','o','f','i','l','e','L','i','s','t',0};
|
|
static const WCHAR computerW[] = {'\\','R','e','g','i','s','t','r','y',
|
|
'\\','M','a','c','h','i','n','e',
|
|
'\\','S','y','s','t','e','m',
|
|
'\\','C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t',
|
|
'\\','C','o','n','t','r','o','l',
|
|
'\\','C','o','m','p','u','t','e','r','N','a','m','e',
|
|
'\\','A','c','t','i','v','e','C','o','m','p','u','t','e','r','N','a','m','e',0};
|
|
static const WCHAR curversionW[] = {'\\','R','e','g','i','s','t','r','y',
|
|
'\\','M','a','c','h','i','n','e',
|
|
'\\','S','o','f','t','w','a','r','e',
|
|
'\\','M','i','c','r','o','s','o','f','t',
|
|
'\\','W','i','n','d','o','w','s',
|
|
'\\','C','u','r','r','e','n','t','V','e','r','s','i','o','n',0};
|
|
OBJECT_ATTRIBUTES attr;
|
|
UNICODE_STRING nameW;
|
|
WCHAR *value;
|
|
HANDLE key;
|
|
|
|
InitializeObjectAttributes( &attr, &nameW, 0, 0, NULL );
|
|
init_unicode_string( &nameW, syskeyW );
|
|
if (!NtOpenKey( &key, KEY_READ, &attr ))
|
|
{
|
|
add_registry_variables( env, pos, size, key );
|
|
NtClose( key );
|
|
}
|
|
if (!open_hkcu_key( "Environment", &key ))
|
|
{
|
|
add_registry_variables( env, pos, size, key );
|
|
NtClose( key );
|
|
}
|
|
if (!open_hkcu_key( "Volatile Environment", &key ))
|
|
{
|
|
add_registry_variables( env, pos, size, key );
|
|
NtClose( key );
|
|
}
|
|
|
|
/* set the user profile variables */
|
|
init_unicode_string( &nameW, profileW );
|
|
if (!NtOpenKey( &key, KEY_READ, &attr ))
|
|
{
|
|
static const WCHAR progdataW[] = {'P','r','o','g','r','a','m','D','a','t','a',0};
|
|
static const WCHAR allusersW[] = {'A','L','L','U','S','E','R','S','P','R','O','F','I','L','E',0};
|
|
static const WCHAR publicW[] = {'P','U','B','L','I','C',0};
|
|
if ((value = get_registry_value( *env, *pos, key, progdataW )))
|
|
{
|
|
set_env_var( env, pos, size, allusersW, wcslen(allusersW), value );
|
|
set_env_var( env, pos, size, progdataW, wcslen(progdataW), value );
|
|
free( value );
|
|
}
|
|
if ((value = get_registry_value( *env, *pos, key, publicW )))
|
|
{
|
|
set_env_var( env, pos, size, publicW, wcslen(publicW), value );
|
|
free( value );
|
|
}
|
|
NtClose( key );
|
|
}
|
|
|
|
/* set the ProgramFiles variables */
|
|
init_unicode_string( &nameW, curversionW );
|
|
if (!NtOpenKey( &key, KEY_READ | KEY_WOW64_64KEY, &attr ))
|
|
{
|
|
static const WCHAR progdirW[] = {'P','r','o','g','r','a','m','F','i','l','e','s','D','i','r',0};
|
|
static const WCHAR progdirx86W[] = {'P','r','o','g','r','a','m','F','i','l','e','s','D','i','r',' ','(','x','8','6',')',0};
|
|
static const WCHAR progfilesW[] = {'P','r','o','g','r','a','m','F','i','l','e','s',0};
|
|
static const WCHAR prog6432W[] = {'P','r','o','g','r','a','m','W','6','4','3','2',0};
|
|
static const WCHAR progx86W[] = {'P','r','o','g','r','a','m','F','i','l','e','s','(','x','8','6',')',0};
|
|
static const WCHAR commondirW[] = {'C','o','m','m','o','n','F','i','l','e','s','D','i','r',0};
|
|
static const WCHAR commondirx86W[] = {'C','o','m','m','o','n','F','i','l','e','s','D','i','r',' ','(','x','8','6',')',0};
|
|
static const WCHAR commonfilesW[] = {'C','o','m','m','o','n','P','r','o','g','r','a','m','F','i','l','e','s',0};
|
|
static const WCHAR common6432W[] = {'C','o','m','m','o','n','P','r','o','g','r','a','m','W','6','4','3','2',0};
|
|
static const WCHAR commonx86W[] = {'C','o','m','m','o','n','P','r','o','g','r','a','m','F','i','l','e','s','(','x','8','6',')',0};
|
|
|
|
if ((value = get_registry_value( *env, *pos, key, progdirx86W )))
|
|
{
|
|
set_env_var( env, pos, size, progx86W, wcslen(progx86W), value );
|
|
free( value );
|
|
if ((value = get_registry_value( *env, *pos, key, progdirW )))
|
|
set_env_var( env, pos, size, prog6432W, wcslen(prog6432W), value );
|
|
}
|
|
else
|
|
{
|
|
if ((value = get_registry_value( *env, *pos, key, progdirW )))
|
|
set_env_var( env, pos, size, progfilesW, wcslen(progfilesW), value );
|
|
}
|
|
free( value );
|
|
|
|
if ((value = get_registry_value( *env, *pos, key, commondirx86W )))
|
|
{
|
|
set_env_var( env, pos, size, commonx86W, wcslen(commonx86W), value );
|
|
free( value );
|
|
if ((value = get_registry_value( *env, *pos, key, commondirW )))
|
|
set_env_var( env, pos, size, common6432W, wcslen(common6432W), value );
|
|
}
|
|
else
|
|
{
|
|
if ((value = get_registry_value( *env, *pos, key, commondirW )))
|
|
set_env_var( env, pos, size, commonfilesW, wcslen(commonfilesW), value );
|
|
}
|
|
free( value );
|
|
NtClose( key );
|
|
}
|
|
|
|
/* set the computer name */
|
|
init_unicode_string( &nameW, computerW );
|
|
if (!NtOpenKey( &key, KEY_READ, &attr ))
|
|
{
|
|
static const WCHAR computernameW[] = {'C','O','M','P','U','T','E','R','N','A','M','E',0};
|
|
if ((value = get_registry_value( *env, *pos, key, computernameW )))
|
|
{
|
|
set_env_var( env, pos, size, computernameW, wcslen(computernameW), value );
|
|
free( value );
|
|
}
|
|
NtClose( key );
|
|
}
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* get_initial_console
|
|
*
|
|
* Return the initial console handles.
|
|
*/
|
|
static void get_initial_console( RTL_USER_PROCESS_PARAMETERS *params )
|
|
{
|
|
int output_fd = -1;
|
|
|
|
wine_server_fd_to_handle( 0, GENERIC_READ|SYNCHRONIZE, OBJ_INHERIT, ¶ms->hStdInput );
|
|
wine_server_fd_to_handle( 1, GENERIC_WRITE|SYNCHRONIZE, OBJ_INHERIT, ¶ms->hStdOutput );
|
|
wine_server_fd_to_handle( 2, GENERIC_WRITE|SYNCHRONIZE, OBJ_INHERIT, ¶ms->hStdError );
|
|
|
|
/* mark tty handles for kernelbase, see init_console */
|
|
if (params->hStdInput && isatty(0))
|
|
{
|
|
params->ConsoleHandle = CONSOLE_HANDLE_SHELL;
|
|
params->hStdInput = (HANDLE)((UINT_PTR)params->hStdInput | 1);
|
|
}
|
|
if (params->hStdError && isatty(2))
|
|
{
|
|
params->ConsoleHandle = CONSOLE_HANDLE_SHELL;
|
|
params->hStdError = (HANDLE)((UINT_PTR)params->hStdError | 1);
|
|
output_fd = 2;
|
|
}
|
|
if (params->hStdOutput && isatty(1))
|
|
{
|
|
params->ConsoleHandle = CONSOLE_HANDLE_SHELL;
|
|
params->hStdOutput = (HANDLE)((UINT_PTR)params->hStdOutput | 1);
|
|
output_fd = 1;
|
|
}
|
|
|
|
if (output_fd != -1)
|
|
{
|
|
struct winsize size;
|
|
if (!ioctl( output_fd, TIOCGWINSZ, &size ))
|
|
{
|
|
params->dwXCountChars = size.ws_col;
|
|
params->dwYCountChars = size.ws_row;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* get_initial_directory
|
|
*
|
|
* Get the current directory at startup.
|
|
*/
|
|
static WCHAR *get_initial_directory(void)
|
|
{
|
|
static const WCHAR backslashW[] = {'\\',0};
|
|
static const WCHAR windows_dir[] = {'\\','?','?','\\','C',':','\\','w','i','n','d','o','w','s','\\',0};
|
|
const char *pwd;
|
|
char *cwd;
|
|
int size;
|
|
WCHAR *ret = NULL;
|
|
|
|
/* try to get it from the Unix cwd */
|
|
|
|
for (size = 1024; ; size *= 2)
|
|
{
|
|
if (!(cwd = malloc( size ))) break;
|
|
if (getcwd( cwd, size )) break;
|
|
free( cwd );
|
|
if (errno == ERANGE) continue;
|
|
cwd = NULL;
|
|
break;
|
|
}
|
|
|
|
/* try to use PWD if it is valid, so that we don't resolve symlinks */
|
|
|
|
pwd = getenv( "PWD" );
|
|
if (cwd)
|
|
{
|
|
struct stat st1, st2;
|
|
|
|
if (!pwd || stat( pwd, &st1 ) == -1 ||
|
|
(!stat( cwd, &st2 ) && (st1.st_dev != st2.st_dev || st1.st_ino != st2.st_ino)))
|
|
pwd = cwd;
|
|
}
|
|
|
|
if (pwd)
|
|
{
|
|
if (!unix_to_nt_file_name( pwd, &ret ))
|
|
{
|
|
ULONG len = wcslen( ret );
|
|
if (len && ret[len - 1] != '\\')
|
|
{
|
|
/* add trailing backslash */
|
|
WCHAR *tmp = malloc( (len + 2) * sizeof(WCHAR) );
|
|
wcscpy( tmp, ret );
|
|
wcscat( tmp, backslashW );
|
|
free( ret );
|
|
ret = tmp;
|
|
}
|
|
free( cwd );
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
/* still not initialized */
|
|
MESSAGE("Warning: could not find DOS drive for current working directory '%s', "
|
|
"starting in the Windows directory.\n", cwd ? cwd : "" );
|
|
ret = malloc( sizeof(windows_dir) );
|
|
wcscpy( ret, windows_dir );
|
|
free( cwd );
|
|
return ret;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* build_command_line
|
|
*
|
|
* Build the command line of a process from the argv array.
|
|
*
|
|
* We must quote and escape characters so that the argv array can be rebuilt
|
|
* from the command line:
|
|
* - spaces and tabs must be quoted
|
|
* 'a b' -> '"a b"'
|
|
* - quotes must be escaped
|
|
* '"' -> '\"'
|
|
* - if '\'s are followed by a '"', they must be doubled and followed by '\"',
|
|
* resulting in an odd number of '\' followed by a '"'
|
|
* '\"' -> '\\\"'
|
|
* '\\"' -> '\\\\\"'
|
|
* - '\'s are followed by the closing '"' must be doubled,
|
|
* resulting in an even number of '\' followed by a '"'
|
|
* ' \' -> '" \\"'
|
|
* ' \\' -> '" \\\\"'
|
|
* - '\'s that are not followed by a '"' can be left as is
|
|
* 'a\b' == 'a\b'
|
|
* 'a\\b' == 'a\\b'
|
|
*/
|
|
static WCHAR *build_command_line( WCHAR **wargv )
|
|
{
|
|
int len;
|
|
WCHAR **arg, *ret;
|
|
LPWSTR p;
|
|
|
|
len = 1;
|
|
for (arg = wargv; *arg; arg++) len += 3 + 2 * wcslen( *arg );
|
|
if (!(ret = malloc( len * sizeof(WCHAR) ))) return NULL;
|
|
|
|
p = ret;
|
|
for (arg = wargv; *arg; arg++)
|
|
{
|
|
BOOL has_space, has_quote;
|
|
int i, bcount;
|
|
WCHAR *a;
|
|
|
|
/* check for quotes and spaces in this argument (first arg is always quoted) */
|
|
has_space = (arg == wargv) || !**arg || wcschr( *arg, ' ' ) || wcschr( *arg, '\t' );
|
|
has_quote = wcschr( *arg, '"' ) != NULL;
|
|
|
|
/* now transfer it to the command line */
|
|
if (has_space) *p++ = '"';
|
|
if (has_quote || has_space)
|
|
{
|
|
bcount = 0;
|
|
for (a = *arg; *a; a++)
|
|
{
|
|
if (*a == '\\') bcount++;
|
|
else
|
|
{
|
|
if (*a == '"') /* double all the '\\' preceding this '"', plus one */
|
|
for (i = 0; i <= bcount; i++) *p++ = '\\';
|
|
bcount = 0;
|
|
}
|
|
*p++ = *a;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
wcscpy( p, *arg );
|
|
p += wcslen( p );
|
|
}
|
|
if (has_space)
|
|
{
|
|
/* Double all the '\' preceding the closing quote */
|
|
for (i = 0; i < bcount; i++) *p++ = '\\';
|
|
*p++ = '"';
|
|
}
|
|
*p++ = ' ';
|
|
}
|
|
if (p > ret) p--; /* remove last space */
|
|
*p = 0;
|
|
if (p - ret >= 32767)
|
|
{
|
|
ERR( "command line too long (%u)\n", (DWORD)(p - ret) );
|
|
NtTerminateProcess( GetCurrentProcess(), 1 );
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* run_wineboot
|
|
*/
|
|
static void run_wineboot( WCHAR *env, SIZE_T size )
|
|
{
|
|
static const WCHAR eventW[] = {'\\','K','e','r','n','e','l','O','b','j','e','c','t','s',
|
|
'\\','_','_','w','i','n','e','b','o','o','t','_','e','v','e','n','t',0};
|
|
static const WCHAR appnameW[] = {'\\','?','?','\\','C',':','\\','w','i','n','d','o','w','s',
|
|
'\\','s','y','s','t','e','m','3','2','\\','w','i','n','e','b','o','o','t','.','e','x','e',0};
|
|
static const WCHAR cmdlineW[] = {'"','C',':','\\','w','i','n','d','o','w','s','\\',
|
|
's','y','s','t','e','m','3','2','\\','w','i','n','e','b','o','o','t','.','e','x','e','"',
|
|
' ','-','-','i','n','i','t',0};
|
|
RTL_USER_PROCESS_PARAMETERS params = { sizeof(params), sizeof(params) };
|
|
PS_ATTRIBUTE_LIST ps_attr;
|
|
PS_CREATE_INFO create_info;
|
|
HANDLE process, thread, handles[2];
|
|
UNICODE_STRING nameW;
|
|
OBJECT_ATTRIBUTES attr;
|
|
LARGE_INTEGER timeout;
|
|
NTSTATUS status;
|
|
int count = 1;
|
|
|
|
init_unicode_string( &nameW, eventW );
|
|
InitializeObjectAttributes( &attr, &nameW, OBJ_OPENIF, 0, NULL );
|
|
status = NtCreateEvent( &handles[0], EVENT_ALL_ACCESS, &attr, NotificationEvent, 0 );
|
|
if (status == STATUS_OBJECT_NAME_EXISTS) goto wait;
|
|
if (status)
|
|
{
|
|
ERR( "failed to create wineboot event, expect trouble\n" );
|
|
return;
|
|
}
|
|
|
|
env[size] = 0;
|
|
params.Flags = PROCESS_PARAMS_FLAG_NORMALIZED;
|
|
params.Environment = env;
|
|
params.EnvironmentSize = size;
|
|
init_unicode_string( ¶ms.CurrentDirectory.DosPath, system_dir + 4 );
|
|
init_unicode_string( ¶ms.ImagePathName, appnameW + 4 );
|
|
init_unicode_string( ¶ms.CommandLine, cmdlineW );
|
|
init_unicode_string( ¶ms.WindowTitle, appnameW + 4 );
|
|
init_unicode_string( &nameW, appnameW );
|
|
|
|
ps_attr.TotalLength = sizeof(ps_attr);
|
|
ps_attr.Attributes[0].Attribute = PS_ATTRIBUTE_IMAGE_NAME;
|
|
ps_attr.Attributes[0].Size = sizeof(appnameW) - sizeof(WCHAR);
|
|
ps_attr.Attributes[0].ValuePtr = (WCHAR *)appnameW;
|
|
ps_attr.Attributes[0].ReturnLength = NULL;
|
|
|
|
wine_server_fd_to_handle( 2, GENERIC_WRITE | SYNCHRONIZE, OBJ_INHERIT, ¶ms.hStdError );
|
|
|
|
if (NtCurrentTeb64() && !NtCurrentTeb64()->TlsSlots[WOW64_TLS_FILESYSREDIR])
|
|
{
|
|
NtCurrentTeb64()->TlsSlots[WOW64_TLS_FILESYSREDIR] = TRUE;
|
|
status = NtCreateUserProcess( &process, &thread, PROCESS_ALL_ACCESS, THREAD_ALL_ACCESS,
|
|
NULL, NULL, 0, THREAD_CREATE_FLAGS_CREATE_SUSPENDED, ¶ms,
|
|
&create_info, &ps_attr );
|
|
NtCurrentTeb64()->TlsSlots[WOW64_TLS_FILESYSREDIR] = FALSE;
|
|
}
|
|
else
|
|
status = NtCreateUserProcess( &process, &thread, PROCESS_ALL_ACCESS, THREAD_ALL_ACCESS,
|
|
NULL, NULL, 0, THREAD_CREATE_FLAGS_CREATE_SUSPENDED, ¶ms,
|
|
&create_info, &ps_attr );
|
|
NtClose( params.hStdError );
|
|
|
|
if (status)
|
|
{
|
|
ERR( "failed to start wineboot %x\n", status );
|
|
NtClose( handles[0] );
|
|
return;
|
|
}
|
|
NtResumeThread( thread, NULL );
|
|
NtClose( thread );
|
|
handles[count++] = process;
|
|
|
|
wait:
|
|
timeout.QuadPart = (ULONGLONG)5 * 60 * 1000 * -10000;
|
|
if (NtWaitForMultipleObjects( count, handles, TRUE, FALSE, &timeout ) == WAIT_TIMEOUT)
|
|
ERR( "boot event wait timed out\n" );
|
|
while (count) NtClose( handles[--count] );
|
|
}
|
|
|
|
|
|
static inline void copy_unicode_string( WCHAR **src, WCHAR **dst, UNICODE_STRING *str, UINT len )
|
|
{
|
|
str->Buffer = *dst;
|
|
str->Length = len;
|
|
str->MaximumLength = len + sizeof(WCHAR);
|
|
memcpy( *dst, *src, len );
|
|
(*dst)[len / sizeof(WCHAR)] = 0;
|
|
*src += len / sizeof(WCHAR);
|
|
*dst += len / sizeof(WCHAR) + 1;
|
|
}
|
|
|
|
static inline void put_unicode_string( WCHAR *src, WCHAR **dst, UNICODE_STRING *str )
|
|
{
|
|
copy_unicode_string( &src, dst, str, wcslen(src) * sizeof(WCHAR) );
|
|
}
|
|
|
|
static inline WCHAR *get_dos_path( WCHAR *nt_path )
|
|
{
|
|
if (nt_path[4] && nt_path[5] == ':') return nt_path + 4; /* skip the \??\ prefix */
|
|
nt_path[1] = '\\'; /* change \??\ to \\?\ */
|
|
return nt_path;
|
|
}
|
|
|
|
static inline const WCHAR *get_params_string( const RTL_USER_PROCESS_PARAMETERS *params,
|
|
const UNICODE_STRING *str )
|
|
{
|
|
if (params->Flags & PROCESS_PARAMS_FLAG_NORMALIZED) return str->Buffer;
|
|
return (const WCHAR *)((const char *)params + (UINT_PTR)str->Buffer);
|
|
}
|
|
|
|
static inline DWORD append_string( void **ptr, const RTL_USER_PROCESS_PARAMETERS *params,
|
|
const UNICODE_STRING *str )
|
|
{
|
|
const WCHAR *buffer = get_params_string( params, str );
|
|
memcpy( *ptr, buffer, str->Length );
|
|
*ptr = (WCHAR *)*ptr + str->Length / sizeof(WCHAR);
|
|
return str->Length;
|
|
}
|
|
|
|
#ifdef _WIN64
|
|
static inline void dup_unicode_string( const UNICODE_STRING *src, WCHAR **dst, UNICODE_STRING32 *str )
|
|
#else
|
|
static inline void dup_unicode_string( const UNICODE_STRING *src, WCHAR **dst, UNICODE_STRING64 *str )
|
|
#endif
|
|
{
|
|
if (!src->Buffer) return;
|
|
str->Buffer = PtrToUlong( *dst );
|
|
str->Length = src->Length;
|
|
str->MaximumLength = src->MaximumLength;
|
|
memcpy( *dst, src->Buffer, src->MaximumLength );
|
|
*dst += src->MaximumLength / sizeof(WCHAR);
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* get_dword_option
|
|
*/
|
|
static ULONG get_dword_option( HANDLE key, const WCHAR *name, ULONG defval )
|
|
{
|
|
UNICODE_STRING str;
|
|
ULONG size;
|
|
WCHAR buffer[64];
|
|
KEY_VALUE_PARTIAL_INFORMATION *info = (KEY_VALUE_PARTIAL_INFORMATION *)buffer;
|
|
|
|
init_unicode_string( &str, name );
|
|
size = sizeof(buffer) - sizeof(WCHAR);
|
|
if (NtQueryValueKey( key, &str, KeyValuePartialInformation, buffer, size, &size )) return defval;
|
|
if (info->Type != REG_DWORD) return defval;
|
|
return *(ULONG *)info->Data;
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* load_global_options
|
|
*/
|
|
static void load_global_options( const UNICODE_STRING *image )
|
|
{
|
|
static const WCHAR optionsW[] = {'M','a','c','h','i','n','e','\\','S','o','f','t','w','a','r','e','\\',
|
|
'M','i','c','r','o','s','o','f','t','\\','W','i','n','d','o','w','s',' ','N','T','\\',
|
|
'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
|
|
'I','m','a','g','e',' ','F','i','l','e',' ','E','x','e','c','u','t','i','o','n',' ','O','p','t','i','o','n','s',0};
|
|
static const WCHAR sessionW[] = {'M','a','c','h','i','n','e','\\','S','y','s','t','e','m','\\',
|
|
'C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t','\\',
|
|
'C','o','n','t','r','o','l','\\','S','e','s','s','i','o','n',' ','M','a','n','a','g','e','r',0};
|
|
static const WCHAR globalflagW[] = {'G','l','o','b','a','l','F','l','a','g',0};
|
|
static const WCHAR critsectionW[] = {'C','r','i','t','i','c','a','l','S','e','c','t','i','o','n','T','i','m','e','o','u','t',0};
|
|
static const WCHAR heapreserveW[] = {'H','e','a','p','S','e','g','m','e','n','t','R','e','s','e','r','v','e',0};
|
|
static const WCHAR heapcommitW[] = {'H','e','a','p','S','e','g','m','e','n','t','C','o','m','m','i','t',0};
|
|
static const WCHAR heapdecommittotalW[] = {'H','e','a','p','D','e','C','o','m','m','i','t','T','o','t','a','l','F','r','e','e','T','h','r','e','s','h','o','l','d',0};
|
|
static const WCHAR heapdecommitblockW[] = {'H','e','a','p','D','e','C','o','m','m','i','t','F','r','e','e','B','l','o','c','k','T','h','r','e','s','h','o','l','d',0};
|
|
OBJECT_ATTRIBUTES attr;
|
|
UNICODE_STRING nameW;
|
|
HANDLE key;
|
|
ULONG i;
|
|
|
|
InitializeObjectAttributes( &attr, &nameW, OBJ_CASE_INSENSITIVE, 0, NULL );
|
|
init_unicode_string( &nameW, sessionW );
|
|
if (!NtOpenKey( &key, KEY_QUERY_VALUE, &attr ))
|
|
{
|
|
peb->NtGlobalFlag = get_dword_option( key, globalflagW, 0 );
|
|
peb->CriticalSectionTimeout.QuadPart = get_dword_option( key, critsectionW, 30 * 24 * 60 * 60 ) * (ULONGLONG)-10000000;
|
|
peb->HeapSegmentReserve = get_dword_option( key, heapreserveW, 0x100000 );
|
|
peb->HeapSegmentCommit = get_dword_option( key, heapcommitW, 0x10000 );
|
|
peb->HeapDeCommitTotalFreeThreshold = get_dword_option( key, heapdecommittotalW, 0x10000 );
|
|
peb->HeapDeCommitFreeBlockThreshold = get_dword_option( key, heapdecommitblockW, 0x1000 );
|
|
NtClose( key );
|
|
}
|
|
init_unicode_string( &nameW, optionsW );
|
|
if (!NtOpenKey( &key, KEY_QUERY_VALUE, &attr ))
|
|
{
|
|
attr.RootDirectory = key;
|
|
for (i = image->Length / sizeof(WCHAR); i; i--) if (image->Buffer[i - 1] == '\\') break;
|
|
nameW.Buffer = image->Buffer + i;
|
|
nameW.Length = image->Length - i * sizeof(WCHAR);
|
|
if (!NtOpenKey( &key, KEY_QUERY_VALUE, &attr ))
|
|
{
|
|
peb->NtGlobalFlag = get_dword_option( key, globalflagW, peb->NtGlobalFlag );
|
|
NtClose( key );
|
|
}
|
|
NtClose( attr.RootDirectory );
|
|
}
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* build_wow64_parameters
|
|
*/
|
|
static void *build_wow64_parameters( const RTL_USER_PROCESS_PARAMETERS *params )
|
|
{
|
|
#ifdef _WIN64
|
|
RTL_USER_PROCESS_PARAMETERS32 *wow64_params = NULL;
|
|
#else
|
|
RTL_USER_PROCESS_PARAMETERS64 *wow64_params = NULL;
|
|
#endif
|
|
NTSTATUS status;
|
|
WCHAR *dst;
|
|
SIZE_T size = (sizeof(*wow64_params)
|
|
+ params->CurrentDirectory.DosPath.MaximumLength
|
|
+ params->DllPath.MaximumLength
|
|
+ params->ImagePathName.MaximumLength
|
|
+ params->CommandLine.MaximumLength
|
|
+ params->WindowTitle.MaximumLength
|
|
+ params->Desktop.MaximumLength
|
|
+ params->ShellInfo.MaximumLength
|
|
+ params->RuntimeInfo.MaximumLength
|
|
+ params->EnvironmentSize);
|
|
|
|
status = NtAllocateVirtualMemory( NtCurrentProcess(), (void **)&wow64_params, 0, &size,
|
|
MEM_COMMIT, PAGE_READWRITE );
|
|
assert( !status );
|
|
|
|
wow64_params->AllocationSize = size;
|
|
wow64_params->Size = size;
|
|
wow64_params->Flags = params->Flags;
|
|
wow64_params->DebugFlags = params->DebugFlags;
|
|
wow64_params->ConsoleHandle = HandleToULong( params->ConsoleHandle );
|
|
wow64_params->ConsoleFlags = params->ConsoleFlags;
|
|
wow64_params->hStdInput = HandleToULong( params->hStdInput );
|
|
wow64_params->hStdOutput = HandleToULong( params->hStdOutput );
|
|
wow64_params->hStdError = HandleToULong( params->hStdError );
|
|
wow64_params->dwX = params->dwX;
|
|
wow64_params->dwY = params->dwY;
|
|
wow64_params->dwXSize = params->dwXSize;
|
|
wow64_params->dwYSize = params->dwYSize;
|
|
wow64_params->dwXCountChars = params->dwXCountChars;
|
|
wow64_params->dwYCountChars = params->dwYCountChars;
|
|
wow64_params->dwFillAttribute = params->dwFillAttribute;
|
|
wow64_params->dwFlags = params->dwFlags;
|
|
wow64_params->wShowWindow = params->wShowWindow;
|
|
|
|
dst = (WCHAR *)(wow64_params + 1);
|
|
dup_unicode_string( ¶ms->CurrentDirectory.DosPath, &dst, &wow64_params->CurrentDirectory.DosPath );
|
|
dup_unicode_string( ¶ms->DllPath, &dst, &wow64_params->DllPath );
|
|
dup_unicode_string( ¶ms->ImagePathName, &dst, &wow64_params->ImagePathName );
|
|
dup_unicode_string( ¶ms->CommandLine, &dst, &wow64_params->CommandLine );
|
|
dup_unicode_string( ¶ms->WindowTitle, &dst, &wow64_params->WindowTitle );
|
|
dup_unicode_string( ¶ms->Desktop, &dst, &wow64_params->Desktop );
|
|
dup_unicode_string( ¶ms->ShellInfo, &dst, &wow64_params->ShellInfo );
|
|
dup_unicode_string( ¶ms->RuntimeInfo, &dst, &wow64_params->RuntimeInfo );
|
|
|
|
wow64_params->Environment = PtrToUlong( dst );
|
|
wow64_params->EnvironmentSize = params->EnvironmentSize;
|
|
memcpy( dst, params->Environment, params->EnvironmentSize );
|
|
return wow64_params;
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* init_peb
|
|
*/
|
|
static void init_peb( RTL_USER_PROCESS_PARAMETERS *params, void *module )
|
|
{
|
|
peb->ImageBaseAddress = module;
|
|
peb->ProcessParameters = params;
|
|
peb->OSMajorVersion = 6;
|
|
peb->OSMinorVersion = 1;
|
|
peb->OSBuildNumber = 0x1db1;
|
|
peb->OSPlatformId = VER_PLATFORM_WIN32_NT;
|
|
peb->ImageSubSystem = main_image_info.SubSystemType;
|
|
peb->ImageSubSystemMajorVersion = main_image_info.MajorSubsystemVersion;
|
|
peb->ImageSubSystemMinorVersion = main_image_info.MinorSubsystemVersion;
|
|
|
|
load_global_options( ¶ms->ImagePathName );
|
|
|
|
if (NtCurrentTeb()->WowTebOffset)
|
|
{
|
|
void *wow64_params = build_wow64_parameters( params );
|
|
#ifdef _WIN64
|
|
PEB32 *wow64_peb = (PEB32 *)((char *)peb + page_size);
|
|
#else
|
|
PEB64 *wow64_peb = (PEB64 *)((char *)peb - page_size);
|
|
#endif
|
|
wow64_peb->ImageBaseAddress = PtrToUlong( peb->ImageBaseAddress );
|
|
wow64_peb->ProcessParameters = PtrToUlong( wow64_params );
|
|
wow64_peb->NumberOfProcessors = peb->NumberOfProcessors;
|
|
wow64_peb->NtGlobalFlag = peb->NtGlobalFlag;
|
|
wow64_peb->CriticalSectionTimeout.QuadPart = peb->CriticalSectionTimeout.QuadPart;
|
|
wow64_peb->HeapSegmentReserve = peb->HeapSegmentReserve;
|
|
wow64_peb->HeapSegmentCommit = peb->HeapSegmentCommit;
|
|
wow64_peb->HeapDeCommitTotalFreeThreshold = peb->HeapDeCommitTotalFreeThreshold;
|
|
wow64_peb->HeapDeCommitFreeBlockThreshold = peb->HeapDeCommitFreeBlockThreshold;
|
|
wow64_peb->OSMajorVersion = peb->OSMajorVersion;
|
|
wow64_peb->OSMinorVersion = peb->OSMinorVersion;
|
|
wow64_peb->OSBuildNumber = peb->OSBuildNumber;
|
|
wow64_peb->OSPlatformId = peb->OSPlatformId;
|
|
wow64_peb->ImageSubSystem = peb->ImageSubSystem;
|
|
wow64_peb->ImageSubSystemMajorVersion = peb->ImageSubSystemMajorVersion;
|
|
wow64_peb->ImageSubSystemMinorVersion = peb->ImageSubSystemMinorVersion;
|
|
wow64_peb->SessionId = peb->SessionId;
|
|
}
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* build_initial_params
|
|
*
|
|
* Build process parameters from scratch, for processes without a parent.
|
|
*/
|
|
static RTL_USER_PROCESS_PARAMETERS *build_initial_params( void **module )
|
|
{
|
|
static const WCHAR valueW[] = {'1',0};
|
|
static const WCHAR pathW[] = {'P','A','T','H'};
|
|
RTL_USER_PROCESS_PARAMETERS *params = NULL;
|
|
SIZE_T size, env_pos, env_size;
|
|
WCHAR *dst, *image, *cmdline, *path, *bootstrap;
|
|
WCHAR *env = get_initial_environment( &env_pos, &env_size );
|
|
WCHAR *curdir = get_initial_directory();
|
|
NTSTATUS status;
|
|
|
|
/* store the initial PATH value */
|
|
path = get_env_var( env, env_pos, pathW, 4 );
|
|
add_dynamic_environment( &env, &env_pos, &env_size );
|
|
add_registry_environment( &env, &env_pos, &env_size );
|
|
bootstrap = get_env_var( env, env_pos, bootstrapW, ARRAY_SIZE(bootstrapW) );
|
|
set_env_var( &env, &env_pos, &env_size, bootstrapW, ARRAY_SIZE(bootstrapW), valueW );
|
|
is_prefix_bootstrap = TRUE;
|
|
env[env_pos] = 0;
|
|
run_wineboot( env, env_pos );
|
|
|
|
/* reload environment now that wineboot has run */
|
|
set_env_var( &env, &env_pos, &env_size, pathW, 4, path ); /* reset PATH */
|
|
free( path );
|
|
set_env_var( &env, &env_pos, &env_size, bootstrapW, ARRAY_SIZE(bootstrapW), bootstrap );
|
|
is_prefix_bootstrap = !!bootstrap;
|
|
free( bootstrap );
|
|
add_registry_environment( &env, &env_pos, &env_size );
|
|
env[env_pos++] = 0;
|
|
|
|
status = load_main_exe( NULL, main_argv[1], curdir, &image, module );
|
|
if (!status)
|
|
{
|
|
if (main_image_info.ImageCharacteristics & IMAGE_FILE_DLL) status = STATUS_INVALID_IMAGE_FORMAT;
|
|
if (main_image_info.Machine != current_machine) status = STATUS_INVALID_IMAGE_FORMAT;
|
|
}
|
|
|
|
if (status) /* try launching it through start.exe */
|
|
{
|
|
static const char *args[] = { "start.exe", "/exec" };
|
|
free( image );
|
|
if (*module) NtUnmapViewOfSection( GetCurrentProcess(), *module );
|
|
load_start_exe( &image, module );
|
|
prepend_argv( args, 2 );
|
|
}
|
|
else rebuild_argv();
|
|
|
|
main_wargv = build_wargv( get_dos_path( image ));
|
|
cmdline = build_command_line( main_wargv );
|
|
|
|
TRACE( "image %s cmdline %s dir %s\n",
|
|
debugstr_w(main_wargv[0]), debugstr_w(cmdline), debugstr_w(curdir) );
|
|
|
|
size = (sizeof(*params)
|
|
+ MAX_PATH * sizeof(WCHAR) /* curdir */
|
|
+ (wcslen( cmdline ) + 1) * sizeof(WCHAR) /* command line */
|
|
+ (wcslen( main_wargv[0] ) + 1) * sizeof(WCHAR) * 2 /* image path + window title */
|
|
+ env_pos * sizeof(WCHAR));
|
|
|
|
status = NtAllocateVirtualMemory( NtCurrentProcess(), (void **)¶ms, 0, &size,
|
|
MEM_COMMIT, PAGE_READWRITE );
|
|
assert( !status );
|
|
|
|
params->AllocationSize = size;
|
|
params->Size = size;
|
|
params->Flags = PROCESS_PARAMS_FLAG_NORMALIZED;
|
|
params->wShowWindow = 1; /* SW_SHOWNORMAL */
|
|
|
|
params->CurrentDirectory.DosPath.Buffer = (WCHAR *)(params + 1);
|
|
wcscpy( params->CurrentDirectory.DosPath.Buffer, get_dos_path( curdir ));
|
|
params->CurrentDirectory.DosPath.Length = wcslen(params->CurrentDirectory.DosPath.Buffer) * sizeof(WCHAR);
|
|
params->CurrentDirectory.DosPath.MaximumLength = MAX_PATH * sizeof(WCHAR);
|
|
dst = params->CurrentDirectory.DosPath.Buffer + MAX_PATH;
|
|
|
|
put_unicode_string( main_wargv[0], &dst, ¶ms->ImagePathName );
|
|
put_unicode_string( cmdline, &dst, ¶ms->CommandLine );
|
|
put_unicode_string( main_wargv[0], &dst, ¶ms->WindowTitle );
|
|
free( image );
|
|
free( cmdline );
|
|
free( curdir );
|
|
|
|
params->Environment = dst;
|
|
params->EnvironmentSize = env_pos * sizeof(WCHAR);
|
|
memcpy( dst, env, env_pos * sizeof(WCHAR) );
|
|
free( env );
|
|
|
|
get_initial_console( params );
|
|
|
|
return params;
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* init_startup_info
|
|
*/
|
|
void init_startup_info(void)
|
|
{
|
|
WCHAR *src, *dst, *env, *image;
|
|
void *module = NULL;
|
|
NTSTATUS status;
|
|
SIZE_T size, info_size, env_size, env_pos;
|
|
RTL_USER_PROCESS_PARAMETERS *params = NULL;
|
|
startup_info_t *info;
|
|
|
|
if (!startup_info_size)
|
|
{
|
|
params = build_initial_params( &module );
|
|
init_peb( params, module );
|
|
return;
|
|
}
|
|
|
|
info = malloc( startup_info_size );
|
|
|
|
SERVER_START_REQ( get_startup_info )
|
|
{
|
|
wine_server_set_reply( req, info, startup_info_size );
|
|
status = wine_server_call( req );
|
|
info_size = reply->info_size;
|
|
env_size = (wine_server_reply_size( reply ) - info_size) / sizeof(WCHAR);
|
|
}
|
|
SERVER_END_REQ;
|
|
assert( !status );
|
|
|
|
env = malloc( env_size * sizeof(WCHAR) );
|
|
memcpy( env, (char *)info + info_size, env_size * sizeof(WCHAR) );
|
|
env_pos = env_size - 1;
|
|
add_dynamic_environment( &env, &env_pos, &env_size );
|
|
is_prefix_bootstrap = !!find_env_var( env, env_pos, bootstrapW, ARRAY_SIZE(bootstrapW) );
|
|
env[env_pos++] = 0;
|
|
|
|
size = (sizeof(*params)
|
|
+ MAX_PATH * sizeof(WCHAR) /* curdir */
|
|
+ info->dllpath_len + sizeof(WCHAR)
|
|
+ info->imagepath_len + sizeof(WCHAR)
|
|
+ info->cmdline_len + sizeof(WCHAR)
|
|
+ info->title_len + sizeof(WCHAR)
|
|
+ info->desktop_len + sizeof(WCHAR)
|
|
+ info->shellinfo_len + sizeof(WCHAR)
|
|
+ info->runtime_len + sizeof(WCHAR)
|
|
+ env_pos * sizeof(WCHAR));
|
|
|
|
status = NtAllocateVirtualMemory( NtCurrentProcess(), (void **)¶ms, 0, &size,
|
|
MEM_COMMIT, PAGE_READWRITE );
|
|
assert( !status );
|
|
|
|
params->AllocationSize = size;
|
|
params->Size = size;
|
|
params->Flags = PROCESS_PARAMS_FLAG_NORMALIZED;
|
|
params->DebugFlags = info->debug_flags;
|
|
params->ConsoleHandle = wine_server_ptr_handle( info->console );
|
|
params->ConsoleFlags = info->console_flags;
|
|
params->hStdInput = wine_server_ptr_handle( info->hstdin );
|
|
params->hStdOutput = wine_server_ptr_handle( info->hstdout );
|
|
params->hStdError = wine_server_ptr_handle( info->hstderr );
|
|
params->dwX = info->x;
|
|
params->dwY = info->y;
|
|
params->dwXSize = info->xsize;
|
|
params->dwYSize = info->ysize;
|
|
params->dwXCountChars = info->xchars;
|
|
params->dwYCountChars = info->ychars;
|
|
params->dwFillAttribute = info->attribute;
|
|
params->dwFlags = info->flags;
|
|
params->wShowWindow = info->show;
|
|
|
|
src = (WCHAR *)(info + 1);
|
|
dst = (WCHAR *)(params + 1);
|
|
|
|
/* curdir is special */
|
|
copy_unicode_string( &src, &dst, ¶ms->CurrentDirectory.DosPath, info->curdir_len );
|
|
params->CurrentDirectory.DosPath.MaximumLength = MAX_PATH * sizeof(WCHAR);
|
|
dst = params->CurrentDirectory.DosPath.Buffer + MAX_PATH;
|
|
|
|
if (info->dllpath_len) copy_unicode_string( &src, &dst, ¶ms->DllPath, info->dllpath_len );
|
|
copy_unicode_string( &src, &dst, ¶ms->ImagePathName, info->imagepath_len );
|
|
copy_unicode_string( &src, &dst, ¶ms->CommandLine, info->cmdline_len );
|
|
copy_unicode_string( &src, &dst, ¶ms->WindowTitle, info->title_len );
|
|
copy_unicode_string( &src, &dst, ¶ms->Desktop, info->desktop_len );
|
|
copy_unicode_string( &src, &dst, ¶ms->ShellInfo, info->shellinfo_len );
|
|
if (info->runtime_len)
|
|
{
|
|
/* runtime info isn't a real string */
|
|
params->RuntimeInfo.MaximumLength = params->RuntimeInfo.Length = info->runtime_len;
|
|
params->RuntimeInfo.Buffer = dst;
|
|
memcpy( dst, src, info->runtime_len );
|
|
src += (info->runtime_len + 1) / sizeof(WCHAR);
|
|
dst += (info->runtime_len + 1) / sizeof(WCHAR);
|
|
}
|
|
assert( (char *)src == (char *)info + info_size );
|
|
|
|
params->Environment = dst;
|
|
params->EnvironmentSize = env_pos * sizeof(WCHAR);
|
|
memcpy( dst, env, env_pos * sizeof(WCHAR) );
|
|
free( env );
|
|
free( info );
|
|
|
|
status = load_main_exe( params->ImagePathName.Buffer, NULL,
|
|
params->CommandLine.Buffer, &image, &module );
|
|
if (status)
|
|
{
|
|
MESSAGE( "wine: failed to start %s\n", debugstr_us(¶ms->ImagePathName) );
|
|
NtTerminateProcess( GetCurrentProcess(), status );
|
|
}
|
|
rebuild_argv();
|
|
main_wargv = build_wargv( get_dos_path( image ));
|
|
free( image );
|
|
init_peb( params, module );
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* create_startup_info
|
|
*/
|
|
void *create_startup_info( const UNICODE_STRING *nt_image, const RTL_USER_PROCESS_PARAMETERS *params,
|
|
DWORD *info_size )
|
|
{
|
|
startup_info_t *info;
|
|
UNICODE_STRING dos_image = *nt_image;
|
|
DWORD size;
|
|
void *ptr;
|
|
|
|
dos_image.Buffer = get_dos_path( nt_image->Buffer );
|
|
dos_image.Length = nt_image->Length - (dos_image.Buffer - nt_image->Buffer) * sizeof(WCHAR);
|
|
|
|
size = sizeof(*info);
|
|
size += params->CurrentDirectory.DosPath.Length;
|
|
size += params->DllPath.Length;
|
|
size += dos_image.Length;
|
|
size += params->CommandLine.Length;
|
|
size += params->WindowTitle.Length;
|
|
size += params->Desktop.Length;
|
|
size += params->ShellInfo.Length;
|
|
size += params->RuntimeInfo.Length;
|
|
size = (size + 1) & ~1;
|
|
*info_size = size;
|
|
|
|
if (!(info = calloc( size, 1 ))) return NULL;
|
|
|
|
info->debug_flags = params->DebugFlags;
|
|
info->console_flags = params->ConsoleFlags;
|
|
info->console = wine_server_obj_handle( params->ConsoleHandle );
|
|
info->hstdin = wine_server_obj_handle( params->hStdInput );
|
|
info->hstdout = wine_server_obj_handle( params->hStdOutput );
|
|
info->hstderr = wine_server_obj_handle( params->hStdError );
|
|
info->x = params->dwX;
|
|
info->y = params->dwY;
|
|
info->xsize = params->dwXSize;
|
|
info->ysize = params->dwYSize;
|
|
info->xchars = params->dwXCountChars;
|
|
info->ychars = params->dwYCountChars;
|
|
info->attribute = params->dwFillAttribute;
|
|
info->flags = params->dwFlags;
|
|
info->show = params->wShowWindow;
|
|
|
|
ptr = info + 1;
|
|
info->curdir_len = append_string( &ptr, params, ¶ms->CurrentDirectory.DosPath );
|
|
info->dllpath_len = append_string( &ptr, params, ¶ms->DllPath );
|
|
info->imagepath_len = append_string( &ptr, params, &dos_image );
|
|
info->cmdline_len = append_string( &ptr, params, ¶ms->CommandLine );
|
|
info->title_len = append_string( &ptr, params, ¶ms->WindowTitle );
|
|
info->desktop_len = append_string( &ptr, params, ¶ms->Desktop );
|
|
info->shellinfo_len = append_string( &ptr, params, ¶ms->ShellInfo );
|
|
info->runtime_len = append_string( &ptr, params, ¶ms->RuntimeInfo );
|
|
return info;
|
|
}
|
|
|
|
|
|
/**************************************************************************
|
|
* NtGetNlsSectionPtr (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI NtGetNlsSectionPtr( ULONG type, ULONG id, void *unknown, void **ptr, SIZE_T *size )
|
|
{
|
|
UNICODE_STRING nameW;
|
|
OBJECT_ATTRIBUTES attr;
|
|
WCHAR name[32];
|
|
HANDLE handle, file;
|
|
NTSTATUS status;
|
|
|
|
if ((status = get_nls_section_name( type, id, name ))) return status;
|
|
|
|
init_unicode_string( &nameW, name );
|
|
InitializeObjectAttributes( &attr, &nameW, 0, 0, NULL );
|
|
if ((status = NtOpenSection( &handle, SECTION_MAP_READ, &attr )))
|
|
{
|
|
if ((status = open_nls_data_file( type, id, &file ))) return status;
|
|
attr.Attributes = OBJ_OPENIF | OBJ_PERMANENT;
|
|
status = NtCreateSection( &handle, SECTION_MAP_READ, &attr, NULL, PAGE_READONLY, SEC_COMMIT, file );
|
|
NtClose( file );
|
|
if (status == STATUS_OBJECT_NAME_EXISTS) status = STATUS_SUCCESS;
|
|
}
|
|
if (!status)
|
|
{
|
|
*ptr = NULL;
|
|
*size = 0;
|
|
status = NtMapViewOfSection( handle, GetCurrentProcess(), ptr, 0, 0, NULL, size,
|
|
ViewShare, 0, PAGE_READONLY );
|
|
}
|
|
NtClose( handle );
|
|
return status;
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* NtQueryDefaultLocale (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI NtQueryDefaultLocale( BOOLEAN user, LCID *lcid )
|
|
{
|
|
*lcid = user ? user_lcid : system_lcid;
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* NtSetDefaultLocale (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI NtSetDefaultLocale( BOOLEAN user, LCID lcid )
|
|
{
|
|
if (user) user_lcid = lcid;
|
|
else
|
|
{
|
|
system_lcid = lcid;
|
|
system_ui_language = LANGIDFROMLCID(lcid); /* there is no separate call to set it */
|
|
}
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* NtQueryDefaultUILanguage (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI NtQueryDefaultUILanguage( LANGID *lang )
|
|
{
|
|
*lang = user_ui_language;
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* NtSetDefaultUILanguage (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI NtSetDefaultUILanguage( LANGID lang )
|
|
{
|
|
user_ui_language = lang;
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
|
|
/**********************************************************************
|
|
* NtQueryInstallUILanguage (NTDLL.@)
|
|
*/
|
|
NTSTATUS WINAPI NtQueryInstallUILanguage( LANGID *lang )
|
|
{
|
|
*lang = system_ui_language;
|
|
return STATUS_SUCCESS;
|
|
}
|
|
|
|
WCHAR WINAPI RtlUpcaseUnicodeChar( WCHAR wch )
|
|
{
|
|
return ntdll_towupper( wch );
|
|
}
|
|
|
|
WCHAR WINAPI RtlDowncaseUnicodeChar( WCHAR wch )
|
|
{
|
|
return ntdll_towlower( wch );
|
|
}
|
|
|
|
NTSTATUS WINAPI RtlUTF8ToUnicodeN( WCHAR *dst, DWORD dstlen, DWORD *reslen, const char *src, DWORD srclen )
|
|
{
|
|
unsigned int res, len;
|
|
NTSTATUS status = STATUS_SUCCESS;
|
|
const char *srcend = src + srclen;
|
|
WCHAR *dstend;
|
|
|
|
if (!src) return STATUS_INVALID_PARAMETER_4;
|
|
if (!reslen) return STATUS_INVALID_PARAMETER;
|
|
|
|
dstlen /= sizeof(WCHAR);
|
|
dstend = dst + dstlen;
|
|
if (!dst)
|
|
{
|
|
for (len = 0; src < srcend; len++)
|
|
{
|
|
unsigned char ch = *src++;
|
|
if (ch < 0x80) continue;
|
|
if ((res = decode_utf8_char( ch, &src, srcend )) > 0x10ffff)
|
|
status = STATUS_SOME_NOT_MAPPED;
|
|
else
|
|
if (res > 0xffff) len++;
|
|
}
|
|
*reslen = len * sizeof(WCHAR);
|
|
return status;
|
|
}
|
|
|
|
while ((dst < dstend) && (src < srcend))
|
|
{
|
|
unsigned char ch = *src++;
|
|
if (ch < 0x80) /* special fast case for 7-bit ASCII */
|
|
{
|
|
*dst++ = ch;
|
|
continue;
|
|
}
|
|
if ((res = decode_utf8_char( ch, &src, srcend )) <= 0xffff)
|
|
{
|
|
*dst++ = res;
|
|
}
|
|
else if (res <= 0x10ffff) /* we need surrogates */
|
|
{
|
|
res -= 0x10000;
|
|
*dst++ = 0xd800 | (res >> 10);
|
|
if (dst == dstend) break;
|
|
*dst++ = 0xdc00 | (res & 0x3ff);
|
|
}
|
|
else
|
|
{
|
|
*dst++ = 0xfffd;
|
|
status = STATUS_SOME_NOT_MAPPED;
|
|
}
|
|
}
|
|
if (src < srcend) status = STATUS_BUFFER_TOO_SMALL; /* overflow */
|
|
*reslen = (dstlen - (dstend - dst)) * sizeof(WCHAR);
|
|
return status;
|
|
}
|