schtasks.c File Reference

#include "initguid.h"
#include "taskschd.h"
#include "wine/debug.h"
#include "wine/unicode.h"

Include dependency graph for schtasks.c:

Go to the source code of this file.

Macros
#define	COBJMACROS

Functions
	WINE_DEFAULT_DEBUG_CHANNEL (schtasks)
static ITaskFolder *	get_tasks_root_folder (void)
static IRegisteredTask *	get_registered_task (const WCHAR *name)
static BSTR	read_file_to_bstr (const WCHAR *file_name)
static int	change_command (int argc, WCHAR *argv[])
static int	create_command (int argc, WCHAR *argv[])
static int	delete_command (int argc, WCHAR *argv[])
int	wmain (int argc, WCHAR *argv[])

Variables
static const WCHAR	change_optW [] = {'/','c','h','a','n','g','e',0}
static const WCHAR	create_optW [] = {'/','c','r','e','a','t','e',0}
static const WCHAR	delete_optW [] = {'/','d','e','l','e','t','e',0}
static const WCHAR	enable_optW [] = {'/','e','n','a','b','l','e',0}
static const WCHAR	f_optW [] = {'/','f',0}
static const WCHAR	ru_optW [] = {'/','r','u',0}
static const WCHAR	tn_optW [] = {'/','t','n',0}
static const WCHAR	tr_optW [] = {'/','t','r',0}
static const WCHAR	xml_optW [] = {'/','x','m','l',0}

Macro Definition Documentation

COBJMACROS

#define COBJMACROS

Definition at line 19 of file schtasks.c.

Function Documentation

change_command()

static int change_command ( int  argc, WCHAR *  argv[]  )

static

Definition at line 136 of file schtasks.c.

Referenced by wmain().

{
    BOOL have_option = FALSE, enable = FALSE;
    const WCHAR *task_name = NULL;
    IRegisteredTask *task;
    HRESULT hres;

    while (argc) {
        if(!strcmpiW(argv[0], tn_optW)) {
            if (argc < 2) {
                FIXME("Missing /tn value\n");
                return 1;
            }

            if (task_name) {
                FIXME("Duplicated /tn argument\n");
                return 1;
            }

            task_name = argv[1];
            argc -= 2;
            argv += 2;
        }else if (!strcmpiW(argv[0], enable_optW)) {
            enable = TRUE;
            have_option = TRUE;
            argc--;
            argv++;
        }else if (!strcmpiW(argv[0], tr_optW)) {
            if (argc < 2) {
                FIXME("Missing /tr value\n");
                return 1;
            }

            FIXME("Unsupported /tr option %s\n", debugstr_w(argv[1]));
            have_option = TRUE;
            argc -= 2;
            argv += 2;
        }else {
            FIXME("Unsupported arguments %s\n", debugstr_w(argv[0]));
            return 1;
        }
    }

    if (!task_name) {
        FIXME("Missing /tn option\n");
        return 1;
    }

    if (!have_option) {
        FIXME("Missing change options\n");
        return 1;
    }

    task = get_registered_task(task_name);
    if (!task)
        return 1;

    if (enable) {
        hres = IRegisteredTask_put_Enabled(task, VARIANT_TRUE);
        if (FAILED(hres)) {
            IRegisteredTask_Release(task);
            FIXME("put_Enabled failed: %08x\n", hres);
            return 1;
        }
    }

    IRegisteredTask_Release(task);
    return 0;
}

create_command()

static int create_command ( int  argc, WCHAR *  argv[]  )

static

Definition at line 206 of file schtasks.c.

Referenced by wmain().

{
    const WCHAR *task_name = NULL, *xml_file = NULL;
    ITaskFolder *root = NULL;
    LONG flags = TASK_CREATE;
    IRegisteredTask *task;
    VARIANT empty;
    BSTR str, xml;
    HRESULT hres;

    while (argc) {
        if (!strcmpiW(argv[0], xml_optW)) {
            if (argc < 2) {
                FIXME("Missing /xml value\n");
                return 1;
            }

            if (xml_file) {
                FIXME("Duplicated /xml argument\n");
                return 1;
            }

            xml_file = argv[1];
            argc -= 2;
            argv += 2;
        }else if(!strcmpiW(argv[0], tn_optW)) {
            if (argc < 2) {
                FIXME("Missing /tn value\n");
                return 1;
            }

            if (task_name) {
                FIXME("Duplicated /tn argument\n");
                return 1;
            }

            task_name = argv[1];
            argc -= 2;
            argv += 2;
        }else if(!strcmpiW(argv[0], f_optW)) {
            flags = TASK_CREATE_OR_UPDATE;
            argc--;
            argv++;
        }else if (!strcmpiW(argv[0], ru_optW)) {
            if (argc < 2) {
                FIXME("Missing /ru value\n");
                return 1;
            }

            FIXME("Unsupported /ru option %s\n", debugstr_w(argv[1]));
            argc -= 2;
            argv += 2;
        }else {
            FIXME("Unsupported argument %s\n", debugstr_w(argv[0]));
            return 1;
        }
    }

    if (!task_name) {
        FIXME("Missing /tn argument\n");
        return 1;
    }

    if (!xml_file) {
        FIXME("Missing /xml argument\n");
        return E_FAIL;
    }

    xml = read_file_to_bstr(xml_file);
    if (!xml)
        return 1;

    root = get_tasks_root_folder();
    if (!root) {
        SysFreeString(xml);
        return 1;
    }

    V_VT(&empty) = VT_EMPTY;
    str = SysAllocString(task_name);
    hres = ITaskFolder_RegisterTask(root, str, xml, flags, empty, empty,
                                    TASK_LOGON_NONE, empty, &task);
    SysFreeString(str);
    SysFreeString(xml);
    ITaskFolder_Release(root);
    if (FAILED(hres))
        return 1;

    IRegisteredTask_Release(task);
    return 0;
}

delete_command()

static int delete_command ( int  argc, WCHAR *  argv[]  )

static

Definition at line 298 of file schtasks.c.

Referenced by wmain().

{
    const WCHAR *task_name = NULL;
    ITaskFolder *root = NULL;
    BSTR str;
    HRESULT hres;

    while (argc) {
        if (!strcmpiW(argv[0], f_optW)) {
            TRACE("force opt\n");
            argc--;
            argv++;
        }else if(!strcmpiW(argv[0], tn_optW)) {
            if (argc < 2) {
                FIXME("Missing /tn value\n");
                return 1;
            }

            if (task_name) {
                FIXME("Duplicated /tn argument\n");
                return 1;
            }

            task_name = argv[1];
            argc -= 2;
            argv += 2;
        }else {
            FIXME("Unsupported argument %s\n", debugstr_w(argv[0]));
            return 1;
        }
    }

    if (!task_name) {
        FIXME("Missing /tn argument\n");
        return 1;
    }

    root = get_tasks_root_folder();
    if (!root)
        return 1;

    str = SysAllocString(task_name);
    hres = ITaskFolder_DeleteTask(root, str, 0);
    SysFreeString(str);
    ITaskFolder_Release(root);
    if (FAILED(hres))
        return 1;

    return 0;
}

get_registered_task()

static IRegisteredTask* get_registered_task ( const WCHAR *  name )

static

Definition at line 68 of file schtasks.c.

Referenced by change_command().

{
    IRegisteredTask *registered_task;
    ITaskFolder *root;
    BSTR str;
    HRESULT hres;

    root = get_tasks_root_folder();
    if (!root)
        return NULL;

    str = SysAllocString(name);
    hres = ITaskFolder_GetTask(root, str, &registered_task);
    SysFreeString(str);
    ITaskFolder_Release(root);
    if (FAILED(hres)) {
        FIXME("GetTask failed: %08x\n", hres);
        return NULL;
    }

    return registered_task;
}

get_tasks_root_folder()

static ITaskFolder* get_tasks_root_folder ( void  )

static

Definition at line 39 of file schtasks.c.

Referenced by create_command(), delete_command(), and get_registered_task().

{
    ITaskService *service;
    ITaskFolder *root;
    VARIANT empty;
    HRESULT hres;

    hres = CoCreateInstance(&CLSID_TaskScheduler, NULL, CLSCTX_INPROC_SERVER,
                            &IID_ITaskService, (void**)&service);
    if (FAILED(hres))
        return NULL;

    V_VT(&empty) = VT_EMPTY;
    hres = ITaskService_Connect(service, empty, empty, empty, empty);
    if (FAILED(hres)) {
        FIXME("Connect failed: %08x\n", hres);
        return NULL;
    }

    hres = ITaskService_GetFolder(service, NULL, &root);
    ITaskService_Release(service);
    if (FAILED(hres)) {
        FIXME("GetFolder failed: %08x\n", hres);
        return NULL;
    }

    return root;
}

read_file_to_bstr()

static BSTR read_file_to_bstr ( const WCHAR *  file_name )

static

Definition at line 91 of file schtasks.c.

Referenced by create_command().

{
    LARGE_INTEGER file_size;
    DWORD read_size, size;
    unsigned char *data;
    HANDLE file;
    BOOL r = FALSE;
    BSTR ret;

    file = CreateFileW(file_name, GENERIC_READ, FILE_SHARE_READ, NULL,
                       OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    if (file == INVALID_HANDLE_VALUE) {
        FIXME("Could not open file\n");
        return NULL;
    }

    if (!GetFileSizeEx(file, &file_size) || !file_size.QuadPart) {
        FIXME("Empty file\n");
        CloseHandle(file);
        return NULL;
    }

    data = HeapAlloc(GetProcessHeap(), 0, file_size.QuadPart);
    if (data)
        r = ReadFile(file, data, file_size.QuadPart, &read_size, NULL);
    CloseHandle(file);
    if (!r) {
        FIXME("Read filed\n");
        HeapFree(GetProcessHeap(), 0, data);
        return NULL;
    }

    if (read_size > 2 && data[0] == 0xff && data[1] == 0xfe) { /* UTF-16 BOM */
        ret = SysAllocStringLen((const WCHAR *)(data + 2), (read_size - 2) / sizeof(WCHAR));
    }else {
        size = MultiByteToWideChar(CP_ACP, 0, (const char *)data, read_size, NULL, 0);
        ret = SysAllocStringLen(NULL, size);
        if (ret)
            MultiByteToWideChar(CP_ACP, 0, (const char *)data, read_size, ret, size);
    }
    HeapFree(GetProcessHeap(), 0, data);

    return ret;
}

WINE_DEFAULT_DEBUG_CHANNEL()

WINE_DEFAULT_DEBUG_CHANNEL

(

schtasks

)

wmain()

int wmain	(	int	argc,
		WCHAR *	argv[]
	)

Definition at line 349 of file schtasks.c.

{
    int i, ret = 0;

    for (i = 0; i < argc; i++)
        TRACE(" %s", wine_dbgstr_w(argv[i]));
    TRACE("\n");

    CoInitialize(NULL);

    if (argc < 2)
        FIXME("Print current tasks state\n");
    else if (!strcmpiW(argv[1], change_optW))
        ret = change_command(argc - 2, argv + 2);
    else if (!strcmpiW(argv[1], create_optW))
        ret = create_command(argc - 2, argv + 2);
    else if (!strcmpiW(argv[1], delete_optW))
        ret = delete_command(argc - 2, argv + 2);
    else
        FIXME("Unsupported command %s\n", debugstr_w(argv[1]));

    CoUninitialize();
    return ret;
}

Variable Documentation

change_optW

const WCHAR change_optW[] = {'/','c','h','a','n','g','e',0}

static

Definition at line 29 of file schtasks.c.

Referenced by wmain().

create_optW

const WCHAR create_optW[] = {'/','c','r','e','a','t','e',0}

static

Definition at line 30 of file schtasks.c.

Referenced by wmain().

delete_optW

const WCHAR delete_optW[] = {'/','d','e','l','e','t','e',0}

static

Definition at line 31 of file schtasks.c.

Referenced by wmain().

enable_optW

const WCHAR enable_optW[] = {'/','e','n','a','b','l','e',0}

static

Definition at line 32 of file schtasks.c.

Referenced by change_command().

f_optW

const WCHAR f_optW[] = {'/','f',0}

static

Definition at line 33 of file schtasks.c.

Referenced by create_command(), and delete_command().

ru_optW

const WCHAR ru_optW[] = {'/','r','u',0}

static

Definition at line 34 of file schtasks.c.

Referenced by create_command().

tn_optW

const WCHAR tn_optW[] = {'/','t','n',0}

static

Definition at line 35 of file schtasks.c.

Referenced by change_command(), create_command(), and delete_command().

tr_optW

const WCHAR tr_optW[] = {'/','t','r',0}

static

Definition at line 36 of file schtasks.c.

Referenced by change_command().

xml_optW

const WCHAR xml_optW[] = {'/','x','m','l',0}

static

Definition at line 37 of file schtasks.c.

Referenced by create_command().