Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

246 lines
5.6 KiB

/*++
Copyright (c) 1989 Microsoft Corporation
Module Name:
startup.c
Abstract:
This module contains the startup code for an NT Application
Author:
Steve Wood (stevewo) 22-Aug-1989
Environment:
User Mode only
Revision History:
--*/
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
//
// User mode process entry point.
//
int
__cdecl
main(
int argc,
char *argv[],
char *envp[],
ULONG DebugParameter OPTIONAL
);
VOID
NtProcessStartup(
PPEB Peb
)
{
int argc;
char **argv;
char **envp;
char **dst;
char *nullPtr = NULL;
PCH s, d;
LPWSTR ws,wd;
ULONG n, DebugParameter;
PRTL_USER_PROCESS_PARAMETERS ProcessParameters;
PUNICODE_STRING p;
ANSI_STRING AnsiString;
ULONG NumberOfArgPointers;
ULONG NumberOfEnvPointers;
ULONG TotalNumberOfPointers;
NTSTATUS Status;
ASSERT( Peb != NULL );
ProcessParameters = RtlNormalizeProcessParams( Peb->ProcessParameters );
DebugParameter = 0;
argc = 0;
argv = &nullPtr;
envp = &nullPtr;
NumberOfEnvPointers = 1;
NumberOfArgPointers = 1;
Status = STATUS_SUCCESS;
if (ARGUMENT_PRESENT( ProcessParameters )) {
//
// Compute how many pointers are needed to pass argv[] and envp[]
//
//
// Now extract the arguments from the process command line.
// using whitespace as separator characters.
//
p = &ProcessParameters->CommandLine;
if (p->Buffer == NULL || p->Length == 0) {
p = &ProcessParameters->ImagePathName;
if (p->Buffer == NULL || p->Length == 0) {
goto SkipInit;
}
}
Status = RtlUnicodeStringToAnsiString( &AnsiString, p, TRUE );
if (! NT_SUCCESS(Status)) {
goto SkipInit;
}
s = AnsiString.Buffer;
n = AnsiString.Length;
while (*s) {
//
// Skip over any white space.
//
while (*s && *s <= ' ') {
s++;
}
//
// Copy token to next white space separator and null terminate
//
if (*s) {
NumberOfArgPointers++;
while (*s > ' ') {
s++;
}
}
}
RtlFreeAnsiString(&AnsiString);
NumberOfArgPointers++;
ws = ProcessParameters->Environment;
if (ws != NULL) {
while (*ws) {
NumberOfEnvPointers++;
while (*ws++) {
;
}
}
}
NumberOfEnvPointers++;
}
//
// both counters also have a trailing pointer to NULL, so count this twice for each
//
TotalNumberOfPointers = NumberOfArgPointers + NumberOfEnvPointers + 4;
if (ARGUMENT_PRESENT( ProcessParameters )) {
DebugParameter = ProcessParameters->DebugFlags;
NtCurrentTeb()->LastStatusValue = STATUS_SUCCESS;
dst = RtlAllocateHeap( Peb->ProcessHeap, 0, TotalNumberOfPointers * sizeof( PCH ) );
if (! dst) {
Status = NtCurrentTeb()->LastStatusValue;
if (NT_SUCCESS(Status)) {
Status = STATUS_NO_MEMORY;
}
goto SkipInit;
}
argv = dst;
*dst = NULL;
//
// Now extract the arguments from the process command line.
// using whitespace as separator characters.
//
p = &ProcessParameters->CommandLine;
if (p->Buffer == NULL || p->Length == 0) {
p = &ProcessParameters->ImagePathName;
if (p->Buffer == NULL || p->Length == 0) {
goto SkipInit;
}
}
Status = RtlUnicodeStringToAnsiString( &AnsiString, p, TRUE );
if (! NT_SUCCESS(Status)) {
goto SkipInit;
}
s = AnsiString.Buffer;
n = AnsiString.Length;
NtCurrentTeb()->LastStatusValue = STATUS_SUCCESS;
d = RtlAllocateHeap( Peb->ProcessHeap, 0, n+2 );
if (! d) {
Status = NtCurrentTeb()->LastStatusValue;
if (NT_SUCCESS(Status)) {
Status = STATUS_NO_MEMORY;
}
RtlFreeAnsiString (&AnsiString);
goto SkipInit;
}
while (*s) {
//
// Skip over any white space.
//
while (*s && *s <= ' ') {
s++;
}
//
// Copy token to next white space separator and null terminate
//
if (*s) {
*dst++ = d;
argc++;
while (*s > ' ') {
*d++ = *s++;
}
*d++ = '\0';
}
}
*dst++ = NULL;
RtlFreeAnsiString (&AnsiString);
envp = dst;
ws = ProcessParameters->Environment;
if (ws != NULL) {
while (*ws) {
*dst++ = (char *)ws;
while (*ws++) {
;
}
}
}
*dst++ = NULL;
}
SkipInit:
if (DebugParameter != 0) {
DbgBreakPoint();
}
if (NT_SUCCESS(Status)) {
Status = main( argc, argv, envp, DebugParameter );
}
NtTerminateProcess( NtCurrentProcess(), Status );
}