mirror of
https://github.com/reactos/reactos.git
synced 2024-12-28 10:04:49 +00:00
[SCHTASKS] Sync with Wine Staging 3.3. CORE-14434
This commit is contained in:
Amine Khaldi
parent
457c843251
commit
4ce8df2ba6
3 changed files with 348 additions and 9 deletions
|
|
@ -2,5 +2,5 @@
|
|||
add_executable(schtasks schtasks.c)
|
||||
target_link_libraries(schtasks wine)
|
||||
set_module_type(schtasks win32cui UNICODE)
|
||||
add_importlibs(schtasks msvcrt kernel32 ntdll)
|
||||
add_importlibs(schtasks oleaut32 ole32 msvcrt kernel32 ntdll)
|
||||
add_cd_file(TARGET schtasks DESTINATION reactos/system32 FOR all)
|
||||
|
|
|
|||
|
|
@ -16,18 +16,357 @@
|
|||
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
||||
*/
|
||||
|
||||
#define COBJMACROS
|
||||
|
||||
#include "initguid.h"
|
||||
#include "taskschd.h"
|
||||
|
||||
#include "wine/debug.h"
|
||||
#include "wine/unicode.h"
|
||||
|
||||
WINE_DEFAULT_DEBUG_CHANNEL(schtasks);
|
||||
|
||||
int wmain(int argc, WCHAR *argv[])
|
||||
{
|
||||
int i;
|
||||
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};
|
||||
|
||||
WINE_FIXME("stub:");
|
||||
for (i = 0; i < argc; i++)
|
||||
WINE_FIXME(" %s", wine_dbgstr_w(argv[i]));
|
||||
WINE_FIXME("\n");
|
||||
static ITaskFolder *get_tasks_root_folder(void)
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
||||
static IRegisteredTask *get_registered_task(const WCHAR *name)
|
||||
{
|
||||
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, ®istered_task);
|
||||
SysFreeString(str);
|
||||
ITaskFolder_Release(root);
|
||||
if (FAILED(hres)) {
|
||||
FIXME("GetTask failed: %08x\n", hres);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
return registered_task;
|
||||
}
|
||||
|
||||
static BSTR read_file_to_bstr(const WCHAR *file_name)
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
||||
static int change_command(int argc, WCHAR *argv[])
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
||||
static int create_command(int argc, WCHAR *argv[])
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
||||
static int delete_command(int argc, WCHAR *argv[])
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
||||
int wmain(int argc, WCHAR *argv[])
|
||||
{
|
||||
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;
|
||||
}
|
||||
|
|
|
|||
|
|
@ -228,7 +228,7 @@ ReactOS shares the following programs with Winehq.
|
|||
|
||||
reactos/base/applications/cmdutils/cscript # Synced to WineStaging-3.3
|
||||
reactos/base/applications/cmdutils/reg # Synced to WineStaging-3.3
|
||||
reactos/base/applications/cmdutils/schtasks # Synced to WineStaging-2.9
|
||||
reactos/base/applications/cmdutils/schtasks # Synced to WineStaging-3.3
|
||||
reactos/base/applications/cmdutils/taskkill # Synced to Wine-3.0
|
||||
reactos/base/applications/cmdutils/wmic # Synced to WineStaging-2.9
|
||||
reactos/base/applications/cmdutils/wscript # Synced to WineStaging-3.3
|
||||
|
|
|
|||
Loading…
Reference in a new issue