/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (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.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is the Netscape Portable Runtime (NSPR).
*
* The Initial Developer of the Original Code is
* Netscape Communications Corporation.
* Portions created by the Initial Developer are Copyright (C) 1998-2000
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the MPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the MPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
/*
* ntmisc.c
*
*/
#include "primpl.h"
char *_PR_MD_GET_ENV(const char *name)
{
return getenv(name);
}
/*
** _PR_MD_PUT_ENV() -- add or change environment variable
**
**
*/
PRIntn _PR_MD_PUT_ENV(const char *name)
{
return(putenv(name));
}
/*
**************************************************************************
**************************************************************************
**
** Date and time routines
**
**************************************************************************
**************************************************************************
*/
/*
*-----------------------------------------------------------------------
*
* PR_Now --
*
* Returns the current time in microseconds since the epoch.
* The epoch is midnight January 1, 1970 GMT.
* The implementation is machine dependent. This is the
* implementation for Windows.
* Cf. time_t time(time_t *tp)
*
*-----------------------------------------------------------------------
*/
PR_IMPLEMENT(PRTime)
PR_Now(void)
{
PRTime prt;
FILETIME ft;
GetSystemTimeAsFileTime(&ft);
_PR_FileTimeToPRTime(&ft, &prt);
return prt;
}
/*
***********************************************************************
***********************************************************************
*
* Process creation routines
*
***********************************************************************
***********************************************************************
*/
/*
* Assemble the command line by concatenating the argv array.
* On success, this function returns 0 and the resulting command
* line is returned in *cmdLine. On failure, it returns -1.
*/
static int assembleCmdLine(char *const *argv, char **cmdLine)
{
char *const *arg;
char *p, *q;
int cmdLineSize;
int numBackslashes;
int i;
int argNeedQuotes;
/*
* Find out how large the command line buffer should be.
*/
cmdLineSize = 0;
for (arg = argv; *arg; arg++) {
/*
* \ and " need to be escaped by a \. In the worst case,
* every character is a \ or ", so the string of length
* may double. If we quote an argument, that needs two ".
* Finally, we need a space between arguments, and
* a null byte at the end of command line.
*/
cmdLineSize += 2 * strlen(*arg) /* \ and " need to be escaped */
+ 2 /* we quote every argument */
+ 1; /* space in between, or final null */
}
p = *cmdLine = PR_MALLOC(cmdLineSize);
if (p == NULL) {
return -1;
}
for (arg = argv; *arg; arg++) {
/* Add a space to separates the arguments */
if (arg != argv) {
*p++ = ' ';
}
q = *arg;
numBackslashes = 0;
argNeedQuotes = 0;
/* If the argument contains white space, it needs to be quoted. */
if (strpbrk(*arg, " \f\n\r\t\v")) {
argNeedQuotes = 1;
}
if (argNeedQuotes) {
*p++ = '"';
}
while (*q) {
if (*q == '\\') {
numBackslashes++;
q++;
} else if (*q == '"') {
if (numBackslashes) {
/*
* Double the backslashes since they are followed
* by a quote
*/
for (i = 0; i < 2 * numBackslashes; i++) {
*p++ = '\\';
}
numBackslashes = 0;
}
/* To escape the quote */
*p++ = '\\';
*p++ = *q++;
} else {
if (numBackslashes) {
/*
* Backslashes are not followed by a quote, so
* don't need to double the backslashes.
*/
for (i = 0; i < numBackslashes; i++) {
*p++ = '\\';
}
numBackslashes = 0;
}
*p++ = *q++;
}
}
/* Now we are at the end of this argument */
if (numBackslashes) {
/*
* Double the backslashes if we have a quote string
* delimiter at the end.
*/
if (argNeedQuotes) {
numBackslashes *= 2;
}
for (i = 0; i < numBackslashes; i++) {
*p++ = '\\';
}
}
if (argNeedQuotes) {
*p++ = '"';
}
}
*p = '\0';
return 0;
}
/*
* Assemble the environment block by concatenating the envp array
* (preserving the terminating null byte in each array element)
* and adding a null byte at the end.
*
* Returns 0 on success. The resulting environment block is returned
* in *envBlock. Note that if envp is NULL, a NULL pointer is returned
* in *envBlock. Returns -1 on failure.
*/
static int assembleEnvBlock(char **envp, char **envBlock)
{
char *p;
char *q;
char **env;
char *curEnv;
char *cwdStart, *cwdEnd;
int envBlockSize;
if (envp == NULL) {
*envBlock = NULL;
return 0;
}
curEnv = GetEnvironmentStrings();
cwdStart = curEnv;
while (*cwdStart) {
if (cwdStart[0] == '=' && cwdStart[1] != '\0'
&& cwdStart[2] == ':' && cwdStart[3] == '=') {
break;
}
cwdStart += strlen(cwdStart) + 1;
}
cwdEnd = cwdStart;
if (*cwdEnd) {
cwdEnd += strlen(cwdEnd) + 1;
while (*cwdEnd) {
if (cwdEnd[0] != '=' || cwdEnd[1] == '\0'
|| cwdEnd[2] != ':' || cwdEnd[3] != '=') {
break;
}
cwdEnd += strlen(cwdEnd) + 1;
}
}
envBlockSize = cwdEnd - cwdStart;
for (env = envp; *env; env++) {
envBlockSize += strlen(*env) + 1;
}
envBlockSize++;
p = *envBlock = PR_MALLOC(envBlockSize);
if (p == NULL) {
FreeEnvironmentStrings(curEnv);
return -1;
}
q = cwdStart;
while (q < cwdEnd) {
*p++ = *q++;
}
FreeEnvironmentStrings(curEnv);
for (env = envp; *env; env++) {
q = *env;
while (*q) {
*p++ = *q++;
}
*p++ = '\0';
}
*p = '\0';
return 0;
}
/*
* For qsort. We sort (case-insensitive) the environment strings
* before generating the environment block.
*/
static int compare(const void *arg1, const void *arg2)
{
return _stricmp(* (char**)arg1, * (char**)arg2);
}
PRProcess * _PR_CreateWindowsProcess(
const char *path,
char *const *argv,
char *const *envp,
const PRProcessAttr *attr)
{
STARTUPINFO startupInfo;
PROCESS_INFORMATION procInfo;
BOOL retVal;
char *cmdLine = NULL;
char *envBlock = NULL;
char **newEnvp = NULL;
const char *cwd = NULL; /* current working directory */
PRProcess *proc = NULL;
proc = PR_NEW(PRProcess);
if (!proc) {
PR_SetError(PR_OUT_OF_MEMORY_ERROR, 0);
goto errorExit;
}
if (assembleCmdLine(argv, &cmdLine) == -1) {
PR_SetError(PR_OUT_OF_MEMORY_ERROR, 0);
goto errorExit;
}
/*
* If attr->fdInheritBuffer is not NULL, we need to insert
* it into the envp array, so envp cannot be NULL.
*/
if ((envp == NULL) && attr && attr->fdInheritBuffer) {
envp = environ;
}
if (envp != NULL) {
int idx;
int numEnv;
int newEnvpSize;
numEnv = 0;
while (envp[numEnv]) {
numEnv++;
}
newEnvpSize = numEnv + 1; /* terminating null pointer */
if (attr && attr->fdInheritBuffer) {
newEnvpSize++;
}
newEnvp = (char **) PR_MALLOC(newEnvpSize * sizeof(char *));
for (idx = 0; idx < numEnv; idx++) {
newEnvp[idx] = envp[idx];
}
if (attr && attr->fdInheritBuffer) {
newEnvp[idx++] = attr->fdInheritBuffer;
}
newEnvp[idx] = NULL;
qsort((void *) newEnvp, (size_t) (newEnvpSize - 1),
sizeof(char *), compare);
}
if (assembleEnvBlock(newEnvp, &envBlock) == -1) {
PR_SetError(PR_OUT_OF_MEMORY_ERROR, 0);
goto errorExit;
}
ZeroMemory(&startupInfo, sizeof(startupInfo));
startupInfo.cb = sizeof(startupInfo);
if (attr) {
PRBool redirected = PR_FALSE;
/*
* XXX the default value for stdin, stdout, and stderr
* should probably be the console input and output, not
* those of the parent process.
*/
startupInfo.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
startupInfo.hStdOutput = GetStdHandle(STD_OUTPUT_HANDLE);
startupInfo.hStdError = GetStdHandle(STD_ERROR_HANDLE);
if (attr->stdinFd) {
startupInfo.hStdInput = (HANDLE) attr->stdinFd->secret->md.osfd;
redirected = PR_TRUE;
}
if (attr->stdoutFd) {
startupInfo.hStdOutput = (HANDLE) attr->stdoutFd->secret->md.osfd;
redirected = PR_TRUE;
}
if (attr->stderrFd) {
startupInfo.hStdError = (HANDLE) attr->stderrFd->secret->md.osfd;
redirected = PR_TRUE;
}
if (redirected) {
startupInfo.dwFlags |= STARTF_USESTDHANDLES;
}
cwd = attr->currentDirectory;
}
retVal = CreateProcess(NULL,
cmdLine,
NULL, /* security attributes for the new
* process */
NULL, /* security attributes for the primary
* thread in the new process */
TRUE, /* inherit handles */
0, /* creation flags */
envBlock, /* an environment block, consisting
* of a null-terminated block of
* null-terminated strings. Each
* string is in the form:
* name=value
* XXX: usually NULL */
cwd, /* current drive and directory */
&startupInfo,
&procInfo
);
if (retVal == FALSE) {
/* XXX what error code? */
PR_SetError(PR_UNKNOWN_ERROR, GetLastError());
goto errorExit;
}
CloseHandle(procInfo.hThread);
proc->md.handle = procInfo.hProcess;
proc->md.id = procInfo.dwProcessId;
PR_DELETE(cmdLine);
if (newEnvp) {
PR_DELETE(newEnvp);
}
if (envBlock) {
PR_DELETE(envBlock);
}
return proc;
errorExit:
if (cmdLine) {
PR_DELETE(cmdLine);
}
if (newEnvp) {
PR_DELETE(newEnvp);
}
if (envBlock) {
PR_DELETE(envBlock);
}
if (proc) {
PR_DELETE(proc);
}
return NULL;
} /* _PR_CreateWindowsProcess */
PRStatus _PR_DetachWindowsProcess(PRProcess *process)
{
CloseHandle(process->md.handle);
PR_DELETE(process);
return PR_SUCCESS;
}
/*
* XXX: This implementation is a temporary quick solution.
* It can be called by native threads only (not by fibers).
*/
PRStatus _PR_WaitWindowsProcess(PRProcess *process,
PRInt32 *exitCode)
{
DWORD dwRetVal;
dwRetVal = WaitForSingleObject(process->md.handle, INFINITE);
if (dwRetVal == WAIT_FAILED) {
PR_SetError(PR_UNKNOWN_ERROR, GetLastError());
return PR_FAILURE;
}
PR_ASSERT(dwRetVal == WAIT_OBJECT_0);
if (exitCode != NULL &&
GetExitCodeProcess(process->md.handle, exitCode) == FALSE) {
PR_SetError(PR_UNKNOWN_ERROR, GetLastError());
return PR_FAILURE;
}
CloseHandle(process->md.handle);
PR_DELETE(process);
return PR_SUCCESS;
}
PRStatus _PR_KillWindowsProcess(PRProcess *process)
{
/*
* On Unix, if a process terminates normally, its exit code is
* between 0 and 255. So here on Windows, we use the exit code
* 256 to indicate that the process is killed.
*/
if (TerminateProcess(process->md.handle, 256)) {
return PR_SUCCESS;
}
PR_SetError(PR_UNKNOWN_ERROR, GetLastError());
return PR_FAILURE;
}
PRStatus _MD_WindowsGetHostName(char *name, PRUint32 namelen)
{
PRIntn rv;
PRInt32 syserror;
rv = gethostname(name, (PRInt32) namelen);
if (0 == rv) {
return PR_SUCCESS;
}
syserror = WSAGetLastError();
PR_ASSERT(WSANOTINITIALISED != syserror);
_PR_MD_MAP_GETHOSTNAME_ERROR(syserror);
return PR_FAILURE;
}
PRStatus _MD_WindowsGetSysInfo(PRSysInfo cmd, char *name, PRUint32 namelen)
{
OSVERSIONINFO osvi;
PR_ASSERT((cmd == PR_SI_SYSNAME) || (cmd == PR_SI_RELEASE));
ZeroMemory(&osvi, sizeof(OSVERSIONINFO));
osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
if (! GetVersionEx (&osvi) ) {
_PR_MD_MAP_DEFAULT_ERROR(GetLastError());
return PR_FAILURE;
}
switch (osvi.dwPlatformId) {
case VER_PLATFORM_WIN32_NT:
if (PR_SI_SYSNAME == cmd)
(void)PR_snprintf(name, namelen, "Windows_NT");
else if (PR_SI_RELEASE == cmd)
(void)PR_snprintf(name, namelen, "%d.%d",osvi.dwMajorVersion,
osvi.dwMinorVersion);
break;
case VER_PLATFORM_WIN32_WINDOWS:
if (PR_SI_SYSNAME == cmd) {
if ((osvi.dwMajorVersion > 4) ||
((osvi.dwMajorVersion == 4) && (osvi.dwMinorVersion > 0)))
(void)PR_snprintf(name, namelen, "Windows_98");
else
(void)PR_snprintf(name, namelen, "Windows_95");
} else if (PR_SI_RELEASE == cmd) {
(void)PR_snprintf(name, namelen, "%d.%d",osvi.dwMajorVersion,
osvi.dwMinorVersion);
}
break;
default:
if (PR_SI_SYSNAME == cmd)
(void)PR_snprintf(name, namelen, "Windows_Unknown");
else if (PR_SI_RELEASE == cmd)
(void)PR_snprintf(name, namelen, "%d.%d",0,0);
break;
}
return PR_SUCCESS;
}
PRStatus _MD_WindowsGetReleaseName(char *name, PRUint32 namelen)
{
OSVERSIONINFO osvi;
ZeroMemory(&osvi, sizeof(OSVERSIONINFO));
osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
if (! GetVersionEx (&osvi) ) {
_PR_MD_MAP_DEFAULT_ERROR(GetLastError());
return PR_FAILURE;
}
switch (osvi.dwPlatformId) {
case VER_PLATFORM_WIN32_NT:
case VER_PLATFORM_WIN32_WINDOWS:
(void)PR_snprintf(name, namelen, "%d.%d",osvi.dwMajorVersion,
osvi.dwMinorVersion);
break;
default:
(void)PR_snprintf(name, namelen, "%d.%d",0,0);
break;
}
return PR_SUCCESS;
}
/*
**********************************************************************
*
* Memory-mapped files
*
**********************************************************************
*/
PRStatus _MD_CreateFileMap(PRFileMap *fmap, PRInt64 size)
{
DWORD dwHi, dwLo;
DWORD flProtect;
PRUint32 osfd;
osfd = ( fmap->fd == (PRFileDesc*)-1 )? -1 : fmap->fd->secret->md.osfd;
dwLo = (DWORD) (size & 0xffffffff);
dwHi = (DWORD) (((PRUint64) size >> 32) & 0xffffffff);
if (fmap->prot == PR_PROT_READONLY) {
flProtect = PAGE_READONLY;
fmap->md.dwAccess = FILE_MAP_READ;
} else if (fmap->prot == PR_PROT_READWRITE) {
flProtect = PAGE_READWRITE;
fmap->md.dwAccess = FILE_MAP_WRITE;
} else {
PR_ASSERT(fmap->prot == PR_PROT_WRITECOPY);
flProtect = PAGE_WRITECOPY;
fmap->md.dwAccess = FILE_MAP_COPY;
}
fmap->md.hFileMap = CreateFileMapping(
(HANDLE) osfd,
NULL,
flProtect,
dwHi,
dwLo,
NULL);
if (fmap->md.hFileMap == NULL) {
PR_SetError(PR_UNKNOWN_ERROR, GetLastError());
return PR_FAILURE;
}
return PR_SUCCESS;
}
PRInt32 _MD_GetMemMapAlignment(void)
{
SYSTEM_INFO info;
GetSystemInfo(&info);
return info.dwAllocationGranularity;
}
extern PRLogModuleInfo *_pr_shma_lm;
void * _MD_MemMap(
PRFileMap *fmap,
PROffset64 offset,
PRUint32 len)
{
DWORD dwHi, dwLo;
void *addr;
dwLo = (DWORD) (offset & 0xffffffff);
dwHi = (DWORD) (((PRUint64) offset >> 32) & 0xffffffff);
if ((addr = MapViewOfFile(fmap->md.hFileMap, fmap->md.dwAccess,
dwHi, dwLo, len)) == NULL) {
{
LPVOID lpMsgBuf;
FormatMessage(
FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
NULL,
GetLastError(),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
(LPTSTR) &lpMsgBuf,
0,
NULL
);
PR_LOG( _pr_shma_lm, PR_LOG_DEBUG, ("md_memmap(): %s", lpMsgBuf ));
}
PR_SetError(PR_UNKNOWN_ERROR, GetLastError());
}
return addr;
}
PRStatus _MD_MemUnmap(void *addr, PRUint32 len)
{
if (UnmapViewOfFile(addr)) {
return PR_SUCCESS;
} else {
PR_SetError(PR_UNKNOWN_ERROR, GetLastError());
return PR_FAILURE;
}
}
PRStatus _MD_CloseFileMap(PRFileMap *fmap)
{
CloseHandle(fmap->md.hFileMap);
PR_DELETE(fmap);
return PR_SUCCESS;
}
/*
***********************************************************************
*
* Atomic increment and decrement operations for x86 processors
*
* We don't use InterlockedIncrement and InterlockedDecrement
* because on NT 3.51 and Win95, they return a number with
* the same sign as the incremented/decremented result, rather
* than the result itself. On NT 4.0 these functions do return
* the incremented/decremented result.
*
* The result is returned in the eax register by the inline
* assembly code. We disable the harmless "no return value"
* warning (4035) for these two functions.
*
***********************************************************************
*/
#if defined(_M_IX86) || defined(_X86_)
#pragma warning(disable: 4035)
PRInt32 _PR_MD_ATOMIC_INCREMENT(PRInt32 *val)
{
#if defined(__GNUC__)
PRInt32 result;
asm volatile ("lock ; xadd %0, %1"
: "=r"(result), "=m"(*val)
: "0"(1), "m"(*val));
return result + 1;
#else
__asm
{
mov ecx, val
mov eax, 1
lock xadd dword ptr [ecx], eax
inc eax
}
#endif /* __GNUC__ */
}
#pragma warning(default: 4035)
#pragma warning(disable: 4035)
PRInt32 _PR_MD_ATOMIC_DECREMENT(PRInt32 *val)
{
#if defined(__GNUC__)
PRInt32 result;
asm volatile ("lock ; xadd %0, %1"
: "=r"(result), "=m"(*val)
: "0"(-1), "m"(*val));
//asm volatile("lock ; xadd %0, %1" : "=m" (val), "=a" (result) : "-1" (1));
return result - 1;
#else
__asm
{
mov ecx, val
mov eax, 0ffffffffh
lock xadd dword ptr [ecx], eax
dec eax
}
#endif /* __GNUC__ */
}
#pragma warning(default: 4035)
#pragma warning(disable: 4035)
PRInt32 _PR_MD_ATOMIC_ADD(PRInt32 *intp, PRInt32 val)
{
#if defined(__GNUC__)
PRInt32 result;
//asm volatile("lock ; xadd %1, %0" : "=m" (intp), "=a" (result) : "1" (val));
asm volatile ("lock ; xadd %0, %1"
: "=r"(result), "=m"(*intp)
: "0"(val), "m"(*intp));
return result + val;
#else
__asm
{
mov ecx, intp
mov eax, val
mov edx, eax
lock xadd dword ptr [ecx], eax
add eax, edx
}
#endif /* __GNUC__ */
}
#pragma warning(default: 4035)
#ifdef _PR_HAVE_ATOMIC_CAS
#pragma warning(disable: 4035)
void
PR_StackPush(PRStack *stack, PRStackElem *stack_elem)
{
#if defined(__GNUC__)
void **tos = (void **) stack;
void *tmp;
retry:
if (*tos == (void *) -1)
goto retry;
__asm__("lock xchg %0,%1"
: "=r" (tmp), "=m"(*tos)
: "0" (-1), "m"(*tos));
if (tmp == (void *) -1)
goto retry;
*(void **)stack_elem = tmp;
__asm__("" : : : "memory");
*tos = stack_elem;
#else
__asm
{
mov ebx, stack
mov ecx, stack_elem
retry: mov eax,[ebx]
cmp eax,-1
je retry
mov eax,-1
xchg dword ptr [ebx], eax
cmp eax,-1
je retry
mov [ecx],eax
mov [ebx],ecx
}
#endif /* __GNUC__ */
}
#pragma warning(default: 4035)
#pragma warning(disable: 4035)
PRStackElem *
PR_StackPop(PRStack *stack)
{
#if defined(__GNUC__)
void **tos = (void **) stack;
void *tmp;
retry:
if (*tos == (void *) -1)
goto retry;
__asm__("lock xchg %0,%1"
: "=r" (tmp), "=m"(*tos)
: "0" (-1), "m"(*tos));
if (tmp == (void *) -1)
goto retry;
if (tmp != (void *) 0)
{
void *next = *(void **)tmp;
*tos = next;
*(void **)tmp = 0;
}
else
*tos = tmp;
return tmp;
#else
__asm
{
mov ebx, stack
retry: mov eax,[ebx]
cmp eax,-1
je retry
mov eax,-1
xchg dword ptr [ebx], eax
cmp eax,-1
je retry
cmp eax,0
je empty
mov ecx,[eax]
mov [ebx],ecx
mov [eax],0
jmp done
empty:
mov [ebx],eax
done:
}
#endif /* __GNUC__ */
}
#pragma warning(default: 4035)
#endif /* _PR_HAVE_ATOMIC_CAS */
#endif /* x86 processors */
syntax highlighted by Code2HTML, v. 0.9.1