/*
 * 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! */
}