Sweden-Number/programs/rundll32/rundll32.c

285 lines
8.1 KiB
C

/*
* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 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/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 TCHAR *szTitle = "rundll32";
static TCHAR *szWindowClass = "class_rundll32";
static ATOM MyRegisterClass(HINSTANCE hInstance)
{
WNDCLASSEX wcex;
wcex.cbSize = sizeof(WNDCLASSEX);
wcex.style = CS_HREDRAW | CS_VREDRAW;
wcex.lpfnWndProc = (WNDPROC)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 LPWSTR GetNextArg(LPCWSTR *cmdline)
{
LPCWSTR 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) {
/* Preceeded 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 {
/* Preceeded 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 main(int argc, char* argv[])
{
HWND hWnd;
LPCWSTR szCmdLine;
LPWSTR szDllName,szEntryPoint;
char* szProcName;
HMODULE hDll;
EntryPointW pEntryPointW;
EntryPointA pEntryPointA;
int len;
hWnd=NULL;
hDll=NULL;
szDllName=NULL;
szProcName=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));
szEntryPoint=szDllName;
while (*szEntryPoint!=0 && *szEntryPoint!=0x002c /* ',' */)
szEntryPoint++;
if (*szEntryPoint==0)
goto CLEANUP;
*szEntryPoint++=0;
WINE_TRACE("EntryPoint=%s\n",wine_dbgstr_w(szEntryPoint));
/* Load the library */
hDll=LoadLibraryW(szDllName);
if (!hDll)
{
/* Windows has a MessageBox here... */
WINE_WARN("Unable to load %s\n",wine_dbgstr_w(szDllName));
goto CLEANUP;
}
/* Try the Unicode entrypoint. Note that GetProcAddress only takes ascii
* names.
*/
len=WideCharToMultiByte(CP_ACP,0,szEntryPoint,-1,NULL,0,NULL,NULL);
szProcName=HeapAlloc(GetProcessHeap(),0,len);
if (!szProcName)
goto CLEANUP;
WideCharToMultiByte(CP_ACP,0,szEntryPoint,-1,szProcName,len,NULL,NULL);
szProcName[len-1]=0x0057;
szProcName[len]=0;
pEntryPointW=(void*)GetProcAddress(hDll,szProcName);
if (pEntryPointW)
{
WCHAR* szArguments=NULL;
/* Make a copy of the arguments so they are writable */
len=lstrlenW(szCmdLine);
if (len>0)
{
szArguments=HeapAlloc(GetProcessHeap(),0,(len+1)*sizeof(WCHAR));
if (!szArguments)
goto CLEANUP;
lstrcpyW(szArguments,szCmdLine);
}
WINE_TRACE("Calling %s, arguments=%s\n",
szProcName,wine_dbgstr_w(szArguments));
pEntryPointW(hWnd,hDll,szArguments,SW_SHOWDEFAULT);
if (szArguments)
HeapFree(GetProcessHeap(),0,szArguments);
}
else
{
/* Then try to append 'A' and finally nothing */
szProcName[len-1]=0x0041;
pEntryPointA=(void*)GetProcAddress(hDll,szProcName);
if (!pEntryPointA)
{
szProcName[len-1]=0;
pEntryPointA=(void*)GetProcAddress(hDll,szProcName);
}
if (pEntryPointA)
{
char* szArguments=NULL;
/* Convert the command line to ascii */
WINE_TRACE("Calling %s, arguments=%s\n",
szProcName,wine_dbgstr_a(szArguments));
len=WideCharToMultiByte(CP_ACP,0,szCmdLine,-1,NULL,0,NULL,NULL);
if (len>1)
{
szArguments=HeapAlloc(GetProcessHeap(),0,len);
if (!szArguments)
goto CLEANUP;
WideCharToMultiByte(CP_ACP,0,szCmdLine,-1,szArguments,len,NULL,NULL);
}
pEntryPointA(hWnd,hDll,szArguments,SW_SHOWDEFAULT);
if (szArguments)
HeapFree(GetProcessHeap(),0,szArguments);
}
else
{
/* Windows has a MessageBox here... */
WINE_WARN("Unable to find the entry point: %s\n",szProcName);
}
}
CLEANUP:
if (hWnd)
DestroyWindow(hWnd);
if (hDll)
FreeLibrary(hDll);
if (szDllName)
HeapFree(GetProcessHeap(),0,szDllName);
if (szProcName)
HeapFree(GetProcessHeap(),0,szProcName);
return 0; /* rundll32 always returns 0! */
}