diff options
Diffstat (limited to 'win32/globals.c')
| -rw-r--r-- | win32/globals.c | 222 |
1 files changed, 222 insertions, 0 deletions
diff --git a/win32/globals.c b/win32/globals.c new file mode 100644 index 0000000000000..7c5cc95b52175 --- /dev/null +++ b/win32/globals.c @@ -0,0 +1,222 @@ +/*$Header: /p/tcsh/cvsroot/tcsh/win32/globals.c,v 1.11 2008/09/10 20:34:21 amold Exp $*/ +/*- + * Copyright (c) 1980, 1991 The Regents of the University of California. + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * 3. Neither the name of the University nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND + * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS + * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT + * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY + * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF + * SUCH DAMAGE. + */ +/* + * globals.c: The mem locations needed in the child are copied here. + * -amol + */ +#define WIN32_LEAN_AND_MEAN +#include <windows.h> +#include <stdlib.h> +#include <stdio.h> +#define STRSAFE_LIB +#define STRSAFE_NO_CCH_FUNCTIONS +#include <strsafe.h> + +extern unsigned long bookend1,bookend2; +extern char **environ; + +#define IMAGE_SIZEOF_NT_OPTIONAL32_HEADER 224 +#define IMAGE_SIZEOF_NT_OPTIONAL64_HEADER 240 + +#ifdef _WIN64 +#define IMAGE_SIZEOF_NT_OPTIONAL_HEADER IMAGE_SIZEOF_NT_OPTIONAL64_HEADER +#else +#define IMAGE_SIZEOF_NT_OPTIONAL_HEADER IMAGE_SIZEOF_NT_OPTIONAL32_HEADER +#endif + + +#undef dprintf +void +dprintf(char *format, ...) +{ /* } */ + va_list vl; + char putbuf[2048]; + DWORD err; + + err = GetLastError(); + { + va_start(vl, format); +#pragma warning(disable:4995) + wvsprintf(putbuf,format, vl); +#pragma warning(default:4995) + va_end(vl); + OutputDebugString(putbuf); + } + SetLastError(err); +} +/* + * This function is called by fork(). The process must copy + * whatever memory is needed in the child. hproc is a handle + * to the child process + * + */ +int fork_copy_user_mem(HANDLE hproc) { + + SIZE_T bytes,rc; + SIZE_T size; + void *low = &bookend1, *high= &bookend2; + + if(&bookend1 > &bookend2) { + low = &bookend2; + high = &bookend1; + } + + size =(char*)high - (char*)low; + + + rc =WriteProcessMemory(hproc,low,low, (DWORD)size, &bytes); + + if (!rc) { + rc = GetLastError(); + return -1; + } + if (size != bytes) { + //dprintf("size %d , wrote %d\n",size,bytes); + } + return 0; +} +/* + * Inspired by Microsoft KB article ID: Q90493 + * + * returns 0 (false) if app is non-gui, 1 otherwise. +*/ +#include <winnt.h> +#include <ntport.h> + +__inline BOOL wait_for_io(HANDLE hi, OVERLAPPED *pO) { + + DWORD bytes = 0; + if(GetLastError() != ERROR_IO_PENDING) + { + return FALSE; + } + + return GetOverlappedResult(hi,pO,&bytes,TRUE); +} +#define CHECK_IO(h,o) if(!wait_for_io(h,o)) {goto done;} + +int is_gui(char *exename) { + + HANDLE hImage; + + DWORD bytes; + OVERLAPPED overlap; + + ULONG ntSignature; + + struct DosHeader{ + IMAGE_DOS_HEADER doshdr; + DWORD extra[16]; + }; + + struct DosHeader dh; + IMAGE_OPTIONAL_HEADER optionalhdr; + + int retCode = 0; + + memset(&overlap,0,sizeof(overlap)); + + + hImage = CreateFile(exename, GENERIC_READ, FILE_SHARE_READ, NULL, + OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL| FILE_FLAG_OVERLAPPED, NULL); + if (INVALID_HANDLE_VALUE == hImage) { + return 0; + } + + ReadFile(hImage, &dh, sizeof(struct DosHeader), &bytes,&overlap); + CHECK_IO(hImage,&overlap); + + + if (IMAGE_DOS_SIGNATURE != dh.doshdr.e_magic) { + goto done; + } + + // read from the coffheaderoffset; + overlap.Offset = dh.doshdr.e_lfanew; + + ReadFile(hImage, &ntSignature, sizeof(ULONG), &bytes,&overlap); + CHECK_IO(hImage,&overlap); + + if (IMAGE_NT_SIGNATURE != ntSignature) { + goto done; + } + overlap.Offset = dh.doshdr.e_lfanew + sizeof(ULONG) + + sizeof(IMAGE_FILE_HEADER); + + ReadFile(hImage, &optionalhdr,IMAGE_SIZEOF_NT_OPTIONAL_HEADER, &bytes,&overlap); + CHECK_IO(hImage,&overlap); + + if (optionalhdr.Subsystem ==IMAGE_SUBSYSTEM_WINDOWS_GUI) + retCode = 1; +done: + CloseHandle(hImage); + return retCode; +} +int is_9x_gui(char *prog) { + + char *progpath; + DWORD dwret; + char *pathbuf; + char *pext; + + pathbuf=heap_alloc(MAX_PATH+1); + if(!pathbuf) + return 0; + + progpath=heap_alloc((MAX_PATH<<1)+1); + if(!progpath) + return 0; + + if (GetEnvironmentVariable("PATH",pathbuf,MAX_PATH) ==0) { + goto failed; + } + + pathbuf[MAX_PATH]=0; + + dwret = SearchPath(pathbuf,prog,".EXE",MAX_PATH<<1,progpath,&pext); + + if ( (dwret == 0) || (dwret > (MAX_PATH<<1) ) ) + goto failed; + + dprintf("progpath is %s\n",progpath); + dwret = is_gui(progpath); + + heap_free(pathbuf); + heap_free(progpath); + + return dwret; + +failed: + heap_free(pathbuf); + heap_free(progpath); + return 0; + + +} |