From 938f073e5cbe9c479fbc50ec652b8e0dd7de3661 Mon Sep 17 00:00:00 2001
From: Dmitry Timoshkov <dmitry@baikal.ru>
Date: Tue, 14 Jan 2014 17:18:07 +0900
Subject: [PATCH] taskschd: Implement ITaskService::Connect.

---
 dlls/taskschd/task.c            | 48 +++++++++++++++++++++++++++++++--
 dlls/taskschd/tests/scheduler.c |  5 ----
 2 files changed, 46 insertions(+), 7 deletions(-)

diff --git a/dlls/taskschd/task.c b/dlls/taskschd/task.c
index 76c0e9b9921..fd64bf9d3fb 100644
--- a/dlls/taskschd/task.c
+++ b/dlls/taskschd/task.c
@@ -27,6 +27,7 @@
 #include "taskschd.h"
 #include "taskschd_private.h"
 
+#include "wine/unicode.h"
 #include "wine/debug.h"
 
 WINE_DEFAULT_DEBUG_CHANNEL(taskschd);
@@ -36,6 +37,7 @@ typedef struct
     ITaskService ITaskService_iface;
     LONG ref;
     BOOL connected;
+    WCHAR comp_name[MAX_COMPUTERNAME_LENGTH + 1];
 } TaskService;
 
 static inline TaskService *impl_from_ITaskService(ITaskService *iface)
@@ -127,11 +129,53 @@ static HRESULT WINAPI TaskService_NewTask(ITaskService *iface, DWORD flags, ITas
     return E_NOTIMPL;
 }
 
+static inline BOOL is_variant_null(const VARIANT *var)
+{
+    return V_VT(var) == VT_EMPTY || V_VT(var) == VT_NULL;
+}
+
 static HRESULT WINAPI TaskService_Connect(ITaskService *iface, VARIANT server, VARIANT user, VARIANT domain, VARIANT password)
 {
-    FIXME("%p,%s,%s,%s,%s: stub\n", iface, debugstr_variant(&server), debugstr_variant(&user),
+    TaskService *task_svc = impl_from_ITaskService(iface);
+    WCHAR comp_name[MAX_COMPUTERNAME_LENGTH + 1];
+    DWORD len;
+
+    TRACE("%p,%s,%s,%s,%s\n", iface, debugstr_variant(&server), debugstr_variant(&user),
           debugstr_variant(&domain), debugstr_variant(&password));
-    return E_NOTIMPL;
+
+    if (!is_variant_null(&user) || !is_variant_null(&domain) || !is_variant_null(&password))
+        FIXME("user/domain/password are ignored\n");
+
+    len = sizeof(comp_name)/sizeof(comp_name[0]);
+    if (!GetComputerNameW(comp_name, &len))
+        return HRESULT_FROM_WIN32(GetLastError());
+
+    if (!is_variant_null(&server))
+    {
+        const WCHAR *server_name;
+
+        if (V_VT(&server) != VT_BSTR)
+        {
+            FIXME("server variant type %d is not supported\n", V_VT(&server));
+            return HRESULT_FROM_WIN32(ERROR_BAD_NETPATH);
+        }
+
+        /* skip UNC prefix if any */
+        server_name = V_BSTR(&server);
+        if (server_name[0] == '\\' && server_name[1] == '\\')
+            server_name += 2;
+
+        if (strcmpiW(server_name, comp_name))
+        {
+            FIXME("connection to remote server %s is not supported\n", debugstr_w(V_BSTR(&server)));
+            return HRESULT_FROM_WIN32(ERROR_BAD_NETPATH);
+        }
+    }
+
+    strcpyW(task_svc->comp_name, comp_name);
+    task_svc->connected = TRUE;
+
+    return S_OK;
 }
 
 static HRESULT WINAPI TaskService_get_Connected(ITaskService *iface, VARIANT_BOOL *connected)
diff --git a/dlls/taskschd/tests/scheduler.c b/dlls/taskschd/tests/scheduler.c
index adb748795f0..bf84c0d9fbf 100644
--- a/dlls/taskschd/tests/scheduler.c
+++ b/dlls/taskschd/tests/scheduler.c
@@ -67,30 +67,25 @@ todo_wine
     V_BSTR(&v_comp) = SysAllocString(comp_name);
 
     hr = ITaskService_Connect(service, v_comp, v_null, v_null, v_null);
-todo_wine
     ok(hr == S_OK || hr == E_ACCESSDENIED /* not an administrator */, "Connect error %#x\n", hr);
 
     SysFreeString(V_BSTR(&v_comp));
     V_BSTR(&v_comp) = SysAllocString(deadbeefW);
 
     hr = ITaskService_Connect(service, v_comp, v_null, v_null, v_null);
-todo_wine
     ok(hr == HRESULT_FROM_WIN32(ERROR_BAD_NETPATH), "expected ERROR_BAD_NETPATH, got %#x\n", hr);
 
     vbool = 0xdead;
     hr = ITaskService_get_Connected(service, &vbool);
     ok(hr == S_OK, "get_Connected error %#x\n", hr);
-todo_wine
     ok(vbool == VARIANT_TRUE, "expected VARIANT_TRUE, got %d\n", vbool);
 
     hr = ITaskService_Connect(service, v_null, v_null, v_null, v_null);
-todo_wine
     ok(hr == S_OK, "Connect error %#x\n", hr);
 
     vbool = 0xdead;
     hr = ITaskService_get_Connected(service, &vbool);
     ok(hr == S_OK, "get_Connected error %#x\n", hr);
-todo_wine
     ok(vbool == VARIANT_TRUE, "expected VARIANT_TRUE, got %d\n", vbool);
 
     hr = ITaskService_get_TargetServer(service, &bstr);