| /* |
| * PURPOSE: Load a DLL and run an entry point with the specified parameters |
| * |
| * Copyright 2002 Alberto Massari |
| * Copyright 2001-2003 Aric Stewart for CodeWeavers |
| * Copyright 2003 Mike McCormack for CodeWeavers |
| * |
| * 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 |
| * |
| */ |
| |
| /* |
| * |
| * rundll32 dllname,entrypoint [arguments] |
| * |
| * Documentation for this utility found on KB Q164787 |
| * |
| */ |
| |
| #include <stdio.h> |
| #include <string.h> |
| #include <stdlib.h> |
| |
| /* Exclude rarely-used stuff from Windows headers */ |
| #define WIN32_LEAN_AND_MEAN |
| #include "windows.h" |
| #include "wine/winbase16.h" |
| #include "wine/unicode.h" |
| #include "wine/debug.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(rundll32); |
| |
| |
| /* |
| * Control_RunDLL has these parameters |
| */ |
| typedef void (WINAPI *EntryPointW)(HWND hWnd, HINSTANCE hInst, LPWSTR lpszCmdLine, int nCmdShow); |
| typedef void (WINAPI *EntryPointA)(HWND hWnd, HINSTANCE hInst, LPSTR lpszCmdLine, int nCmdShow); |
| |
| /* |
| * Control_RunDLL needs to have a window. So lets make us a very |
| * simple window class. |
| */ |
| static const TCHAR *szTitle = "rundll32"; |
| static const TCHAR *szWindowClass = "class_rundll32"; |
| |
| static HINSTANCE16 (WINAPI *pLoadLibrary16)(LPCSTR libname); |
| static FARPROC16 (WINAPI *pGetProcAddress16)(HMODULE16 hModule, LPCSTR name); |
| static void (WINAPI *pRunDLL_CallEntry16)( FARPROC proc, HWND hwnd, HINSTANCE inst, |
| LPCSTR cmdline, INT cmdshow ); |
| |
| static ATOM MyRegisterClass(HINSTANCE hInstance) |
| { |
| WNDCLASSEX wcex; |
| |
| wcex.cbSize = sizeof(WNDCLASSEX); |
| |
| wcex.style = CS_HREDRAW | CS_VREDRAW; |
| wcex.lpfnWndProc = DefWindowProc; |
| wcex.cbClsExtra = 0; |
| wcex.cbWndExtra = 0; |
| wcex.hInstance = hInstance; |
| wcex.hIcon = NULL; |
| wcex.hCursor = LoadCursor(NULL, IDC_ARROW); |
| wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1); |
| wcex.lpszMenuName = NULL; |
| wcex.lpszClassName = szWindowClass; |
| wcex.hIconSm = NULL; |
| |
| return RegisterClassEx(&wcex); |
| } |
| |
| static HINSTANCE16 load_dll16( LPCWSTR dll ) |
| { |
| HINSTANCE16 ret = 0; |
| DWORD len = WideCharToMultiByte( CP_ACP, 0, dll, -1, NULL, 0, NULL, NULL ); |
| char *dllA = HeapAlloc( GetProcessHeap(), 0, len ); |
| WideCharToMultiByte( CP_ACP, 0, dll, -1, dllA, len, NULL, NULL ); |
| pLoadLibrary16 = (void *)GetProcAddress( GetModuleHandleA("kernel32.dll"), "LoadLibrary16" ); |
| if (pLoadLibrary16) ret = pLoadLibrary16( dllA ); |
| HeapFree( GetProcessHeap(), 0, dllA ); |
| return ret; |
| } |
| |
| static FARPROC16 get_entry_point16( HINSTANCE16 inst, LPCWSTR entry ) |
| { |
| FARPROC16 ret = 0; |
| DWORD len = WideCharToMultiByte( CP_ACP, 0, entry, -1, NULL, 0, NULL, NULL ); |
| char *entryA = HeapAlloc( GetProcessHeap(), 0, len ); |
| WideCharToMultiByte( CP_ACP, 0, entry, -1, entryA, len, NULL, NULL ); |
| pGetProcAddress16 = (void *)GetProcAddress( GetModuleHandleA("kernel32.dll"), "GetProcAddress16" ); |
| if (pGetProcAddress16) ret = pGetProcAddress16( inst, entryA ); |
| HeapFree( GetProcessHeap(), 0, entryA ); |
| return ret; |
| } |
| |
| static void *get_entry_point32( HMODULE module, LPCWSTR entry, BOOL *unicode ) |
| { |
| void *ret; |
| DWORD len = WideCharToMultiByte( CP_ACP, 0, entry, -1, NULL, 0, NULL, NULL ); |
| char *entryA = HeapAlloc( GetProcessHeap(), 0, len + 1 ); |
| WideCharToMultiByte( CP_ACP, 0, entry, -1, entryA, len, NULL, NULL ); |
| |
| /* first try the W version */ |
| *unicode = TRUE; |
| strcat( entryA, "W" ); |
| if (!(ret = GetProcAddress( module, entryA ))) |
| { |
| /* now the A version */ |
| *unicode = FALSE; |
| entryA[strlen(entryA)-1] = 'A'; |
| if (!(ret = GetProcAddress( module, entryA ))) |
| { |
| /* now the version without suffix */ |
| entryA[strlen(entryA)-1] = 0; |
| ret = GetProcAddress( module, entryA ); |
| } |
| } |
| HeapFree( GetProcessHeap(), 0, entryA ); |
| return ret; |
| } |
| |
| static LPWSTR GetNextArg(LPWSTR *cmdline) |
| { |
| LPWSTR s; |
| LPWSTR arg,d; |
| int in_quotes,bcount,len=0; |
| |
| /* count the chars */ |
| bcount=0; |
| in_quotes=0; |
| s=*cmdline; |
| while (1) { |
| if (*s==0 || ((*s=='\t' || *s==' ') && !in_quotes)) { |
| /* end of this command line argument */ |
| break; |
| } else if (*s=='\\') { |
| /* '\', count them */ |
| bcount++; |
| } else if ((*s=='"') && ((bcount & 1)==0)) { |
| /* unescaped '"' */ |
| in_quotes=!in_quotes; |
| bcount=0; |
| } else { |
| /* a regular character */ |
| bcount=0; |
| } |
| s++; |
| len++; |
| } |
| arg=HeapAlloc(GetProcessHeap(), 0, (len+1)*sizeof(WCHAR)); |
| if (!arg) |
| return NULL; |
| |
| bcount=0; |
| in_quotes=0; |
| d=arg; |
| s=*cmdline; |
| while (*s) { |
| if ((*s=='\t' || *s==' ') && !in_quotes) { |
| /* end of this command line argument */ |
| break; |
| } else if (*s=='\\') { |
| /* '\\' */ |
| *d++=*s++; |
| bcount++; |
| } else if (*s=='"') { |
| /* '"' */ |
| if ((bcount & 1)==0) { |
| /* Preceded by an even number of '\', this is half that |
| * number of '\', plus a quote which we erase. |
| */ |
| d-=bcount/2; |
| in_quotes=!in_quotes; |
| s++; |
| } else { |
| /* Preceded by an odd number of '\', this is half that |
| * number of '\' followed by a '"' |
| */ |
| d=d-bcount/2-1; |
| *d++='"'; |
| s++; |
| } |
| bcount=0; |
| } else { |
| /* a regular character */ |
| *d++=*s++; |
| bcount=0; |
| } |
| } |
| *d=0; |
| *cmdline=s; |
| |
| /* skip the remaining spaces */ |
| while (**cmdline=='\t' || **cmdline==' ') { |
| (*cmdline)++; |
| } |
| |
| return arg; |
| } |
| |
| int WINAPI WinMain(HINSTANCE instance, HINSTANCE hOldInstance, LPSTR szCmdArgs, int nCmdShow) |
| { |
| HWND hWnd; |
| LPWSTR szCmdLine; |
| LPWSTR szDllName,szEntryPoint; |
| void *entry_point; |
| BOOL unicode, win16; |
| STARTUPINFOW info; |
| HMODULE hDll; |
| |
| hWnd=NULL; |
| hDll=NULL; |
| szDllName=NULL; |
| |
| /* Initialize the rundll32 class */ |
| MyRegisterClass( NULL ); |
| hWnd = CreateWindow(szWindowClass, szTitle, |
| WS_OVERLAPPEDWINDOW|WS_VISIBLE, |
| CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, NULL, NULL); |
| |
| /* Skip the rundll32.exe path */ |
| szCmdLine=GetCommandLineW(); |
| WINE_TRACE("CmdLine=%s\n",wine_dbgstr_w(szCmdLine)); |
| szDllName=GetNextArg(&szCmdLine); |
| if (!szDllName || *szDllName==0) |
| goto CLEANUP; |
| HeapFree(GetProcessHeap(),0,szDllName); |
| |
| /* Get the dll name and API EntryPoint */ |
| szDllName=GetNextArg(&szCmdLine); |
| if (!szDllName || *szDllName==0) |
| goto CLEANUP; |
| WINE_TRACE("DllName=%s\n",wine_dbgstr_w(szDllName)); |
| if ((szEntryPoint = strchrW(szDllName, ',' ))) |
| *szEntryPoint++=0; |
| else |
| szEntryPoint = GetNextArg(&szCmdLine); |
| WINE_TRACE("EntryPoint=%s\n",wine_dbgstr_w(szEntryPoint)); |
| |
| /* Load the library */ |
| hDll=LoadLibraryW(szDllName); |
| if (hDll) |
| { |
| win16 = FALSE; |
| entry_point = get_entry_point32( hDll, szEntryPoint, &unicode ); |
| } |
| else |
| { |
| HINSTANCE16 dll = load_dll16( szDllName ); |
| if (dll <= 32) |
| { |
| /* Windows has a MessageBox here... */ |
| WINE_ERR("Unable to load %s\n",wine_dbgstr_w(szDllName)); |
| goto CLEANUP; |
| } |
| win16 = TRUE; |
| unicode = FALSE; |
| entry_point = get_entry_point16( dll, szEntryPoint ); |
| } |
| |
| if (!entry_point) |
| { |
| /* Windows has a MessageBox here... */ |
| WINE_ERR( "Unable to find the entry point %s in %s\n", |
| wine_dbgstr_w(szEntryPoint), wine_dbgstr_w(szDllName) ); |
| goto CLEANUP; |
| } |
| |
| GetStartupInfoW( &info ); |
| if (!(info.dwFlags & STARTF_USESHOWWINDOW)) info.wShowWindow = SW_SHOWDEFAULT; |
| |
| if (unicode) |
| { |
| EntryPointW pEntryPointW = entry_point; |
| |
| WINE_TRACE( "Calling %s (%p,%p,%s,%d)\n", wine_dbgstr_w(szEntryPoint), |
| hWnd, instance, wine_dbgstr_w(szCmdLine), info.wShowWindow ); |
| |
| pEntryPointW( hWnd, instance, szCmdLine, info.wShowWindow ); |
| } |
| else |
| { |
| DWORD len = WideCharToMultiByte( CP_ACP, 0, szCmdLine, -1, NULL, 0, NULL, NULL ); |
| char *cmdline = HeapAlloc( GetProcessHeap(), 0, len ); |
| WideCharToMultiByte( CP_ACP, 0, szCmdLine, -1, cmdline, len, NULL, NULL ); |
| |
| WINE_TRACE( "Calling %s (%p,%p,%s,%d)\n", wine_dbgstr_w(szEntryPoint), |
| hWnd, instance, wine_dbgstr_a(cmdline), info.wShowWindow ); |
| |
| if (win16) |
| { |
| HMODULE shell = LoadLibraryA( "shell32.dll" ); |
| if (shell) pRunDLL_CallEntry16 = (void *)GetProcAddress( shell, (LPCSTR)122 ); |
| if (pRunDLL_CallEntry16) |
| pRunDLL_CallEntry16( entry_point, hWnd, instance, cmdline, info.wShowWindow ); |
| } |
| else |
| { |
| EntryPointA pEntryPointA = entry_point; |
| pEntryPointA( hWnd, instance, cmdline, info.wShowWindow ); |
| } |
| HeapFree( GetProcessHeap(), 0, cmdline ); |
| } |
| |
| CLEANUP: |
| if (hWnd) |
| DestroyWindow(hWnd); |
| if (hDll) |
| FreeLibrary(hDll); |
| HeapFree(GetProcessHeap(),0,szDllName); |
| return 0; /* rundll32 always returns 0! */ |
| } |
