/* (C) 2003-2007 Willem Jan Hengeveld <itsme@xs4all.nl>
 * Web: http://www.xs4all.nl/~itsme/
 *      http://wiki.xda-developers.com/
 *
 * $Id: pkill.cpp 1921 2008-07-31 17:08:31Z itsme $
 */
#include <windows.h>
#include <tchar.h>
#include <stdio.h>
#include <string.h>
#include <rapi.h>

#include "debug.h"
#include "args.h"
#include "ptrutils.h"

#include "ItsUtils.h"
#include "dllversion.h"

// todo: kill by handle does not seem to work.
void ITTerminateProcess(DWORD dwProcessHandle, char *szProcessName, bool bWait);
void ITTerminateThread(DWORD dwThreadHandle);
void usage()
{
    printf("(C) 2003-2008 Willem jan Hengeveld  itsme@xs4all.nl\n");
    printf("Usage: pkill [-w] [-t HANDLE] [ {-h HANDLE } | { processname } ]...\n");
    printf("    -w : wait until process has terminated\n");
}
int main( int argc, char *argv[])
{
    DebugStdOut();

    if (argc==1)
    {
        return 1;
    }

    CheckITSDll();


    bool bWaitForProcess= false;
    for (int i=1 ; i<argc ; i++) {
        if (strncmp(argv[i], "-w", 2)==0) {
            bWaitForProcess= true;
        }
        else if (strncmp(argv[i], "-h", 2)==0) {
            DWORD dwProcessHandle=0;
            if (HANDLEULOPTION(dwProcessHandle, DWORD)) {
                ITTerminateProcess(dwProcessHandle, NULL, bWaitForProcess);
            }
        }
        else if (strncmp(argv[i], "-t", 2)==0) {
            DWORD dwThreadHandle=0;
            if (HANDLEULOPTION(dwThreadHandle, DWORD)) {
                ITTerminateThread(dwThreadHandle);
            }
        }
        else {
            ITTerminateProcess(0, argv[i], bWaitForProcess);
        }
    }

    CeRapiUninit();

    return 0;
}

void ITTerminateProcess(DWORD dwProcessHandle, char *szProcessName, bool bWait)
{
    TerminateProcessParams*inbuf=NULL;
    DWORD insize= szProcessName ? PTR_DIFF(inbuf, inbuf->wszProcessName+strlen(szProcessName)+1)
            : PTR_DIFF(inbuf, inbuf->wszProcessName+1);
    inbuf= (TerminateProcessParams*)LocalAlloc(LPTR, insize);
    DWORD outsize=0;

    inbuf->dwProcessHandle= dwProcessHandle;
    inbuf->bWait= bWait;
    if (szProcessName) {
        _snwprintf(inbuf->wszProcessName, strlen(szProcessName), L"%hs", szProcessName);
        inbuf->wszProcessName[strlen(szProcessName)]= 0;
    }
    else {
        inbuf->wszProcessName[0]= 0;
    }

    HRESULT res= ItsutilsInvoke(L"ITTerminateProcess",
            insize, (BYTE*)inbuf,
            &outsize, NULL);
    if (res)
        error(res, "killing %08lx | %s", dwProcessHandle, szProcessName?szProcessName:"-");
    else
        debug("%08lx | %s killed\n", dwProcessHandle, szProcessName?szProcessName:"-");
}

void ITTerminateThread(DWORD dwThreadHandle)
{
    TerminateThreadParams inbuf;
    DWORD insize= sizeof(inbuf);
    DWORD outsize=0;

    inbuf.dwThreadHandle= dwThreadHandle;

    HRESULT res= ItsutilsInvoke(L"ITTerminateThread",
            insize, (BYTE*)&inbuf,
            &outsize, NULL);
    if (res)
        error(res, "killing thread %08lx", dwThreadHandle);
    else
        debug("thread %08lx killed\n", dwThreadHandle);
}