dlls/hnetcfg/tests/policy.c

   1 /*
   2  * Copyright 2016 Alistair Leslie-Hughes
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  17  */
  18 #define COBJMACROS
  19 #include <stdio.h>
  20
  21 #include "windows.h"
  22 #include "ole2.h"
  23 #include "oleauto.h"
  24 #include "olectl.h"
  25 #include "dispex.h"
  26
  27 #include "wine/test.h"
  28
  29 #include "netfw.h"
  30
  31 static void test_policy2_rules(INetFwPolicy2 *policy2)
  32 {
  33     HRESULT hr;
  34     INetFwRules *rules, *rules2;
  35     INetFwServiceRestriction *restriction;
  36
  37     hr = INetFwPolicy2_QueryInterface(policy2, &IID_INetFwRules, (void**)&rules);
  38     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  39
  40     hr = INetFwPolicy2_get_Rules(policy2, &rules);
  41     ok(hr == S_OK, "got %08x\n", hr);
  42
  43     hr = INetFwPolicy2_get_Rules(policy2, &rules2);
  44     ok(hr == S_OK, "got %08x\n", hr);
  45     ok(rules == rules2, "Different pointers\n");
  46
  47     hr = INetFwPolicy2_get_ServiceRestriction(policy2, &restriction);
  48     todo_wine ok(hr == S_OK, "got %08x\n", hr);
  49     if(hr == S_OK)
  50     {
  51         INetFwRules *rules3;
  52
  53         hr = INetFwServiceRestriction_get_Rules(restriction, &rules3);
  54         ok(hr == S_OK, "got %08x\n", hr);
  55         ok(rules != rules3, "same pointers\n");
  56
  57         if(rules3)
  58             INetFwRules_Release(rules3);
  59         INetFwServiceRestriction_Release(restriction);
  60     }
  61
  62     INetFwRules_Release(rules);
  63     INetFwRules_Release(rules2);
  64 }
  65
  66 static void test_interfaces(void)
  67 {
  68     INetFwMgr *manager;
  69     INetFwPolicy *policy;
  70     INetFwPolicy2 *policy2;
  71     HRESULT hr;
  72
  73     hr = CoCreateInstance(&CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
  74             &IID_INetFwMgr, (void**)&manager);
  75     ok(hr == S_OK, "NetFwMgr create failed: %08x\n", hr);
  76
  77     hr = INetFwMgr_QueryInterface(manager, &IID_INetFwPolicy, (void**)&policy);
  78     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  79
  80     hr = INetFwMgr_QueryInterface(manager, &IID_INetFwPolicy2, (void**)&policy2);
  81     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  82
  83     hr = INetFwMgr_get_LocalPolicy(manager, &policy);
  84     ok(hr == S_OK, "got 0x%08x\n", hr);
  85
  86     hr = INetFwPolicy_QueryInterface(policy, &IID_INetFwPolicy2, (void**)&policy2);
  87     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  88
  89     INetFwPolicy_Release(policy);
  90
  91     hr = CoCreateInstance(&CLSID_NetFwPolicy2, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
  92             &IID_INetFwPolicy2, (void**)&policy2);
  93     if(hr == S_OK)
  94     {
  95         test_policy2_rules(policy2);
  96
  97         INetFwPolicy2_Release(policy2);
  98     }
  99     else
 100         win_skip("NetFwPolicy2 object is not supported: %08x\n", hr);
 101
 102     INetFwMgr_Release(manager);
 103 }
 104
 105 static void test_NetFwAuthorizedApplication(void)
 106 {
 107     INetFwAuthorizedApplication *app;
 108     static WCHAR empty[] = {0};
 109     UNIVERSAL_NAME_INFOW *info;
 110     WCHAR fullpath[MAX_PATH];
 111     WCHAR netpath[MAX_PATH];
 112     WCHAR image[MAX_PATH];
 113     HRESULT hr;
 114     BSTR bstr;
 115     DWORD sz;
 116
 117     hr = CoCreateInstance(&CLSID_NetFwAuthorizedApplication, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 118             &IID_INetFwAuthorizedApplication, (void**)&app);
 119     ok(hr == S_OK, "got: %08x\n", hr);
 120
 121     hr = GetModuleFileNameW(NULL, image, ARRAY_SIZE(image));
 122     ok(hr, "GetModuleFileName failed: %u\n", GetLastError());
 123
 124     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, NULL);
 125     ok(hr == E_POINTER, "got: %08x\n", hr);
 126
 127     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
 128     ok(hr == S_OK || hr == HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY), "got: %08x\n", hr);
 129     ok(!bstr, "got: %s\n",  wine_dbgstr_w(bstr));
 130
 131     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, NULL);
 132     ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
 133
 134     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, empty);
 135     ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
 136
 137     bstr = SysAllocString(image);
 138     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, bstr);
 139     ok(hr == S_OK, "got: %08x\n", hr);
 140     SysFreeString(bstr);
 141
 142     GetFullPathNameW(image, ARRAY_SIZE(fullpath), fullpath, NULL);
 143     GetLongPathNameW(fullpath, fullpath, ARRAY_SIZE(fullpath));
 144
 145     info = (UNIVERSAL_NAME_INFOW *)&netpath;
 146     sz = sizeof(netpath);
 147     hr = WNetGetUniversalNameW(image, UNIVERSAL_NAME_INFO_LEVEL, info, &sz);
 148     if (hr != NO_ERROR)
 149     {
 150         info->lpUniversalName = netpath + sizeof(*info)/sizeof(WCHAR);
 151         lstrcpyW(info->lpUniversalName, fullpath);
 152     }
 153
 154     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
 155     ok(hr == S_OK, "got: %08x\n", hr);
 156     ok(!lstrcmpW(bstr,info->lpUniversalName), "expected %s, got %s\n",
 157         wine_dbgstr_w(info->lpUniversalName), wine_dbgstr_w(bstr));
 158     SysFreeString(bstr);
 159
 160     INetFwAuthorizedApplication_Release(app);
 161 }
 162
 163 START_TEST(policy)
 164 {
 165     INetFwMgr *manager;
 166     HRESULT hr;
 167
 168     CoInitialize(NULL);
 169
 170     hr = CoCreateInstance(&CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 171             &IID_INetFwMgr, (void**)&manager);
 172     if(FAILED(hr))
 173     {
 174         win_skip("NetFwMgr object is not supported: %08x\n", hr);
 175         CoUninitialize();
 176         return;
 177     }
 178
 179     INetFwMgr_Release(manager);
 180
 181     test_interfaces();
 182     test_NetFwAuthorizedApplication();
 183
 184     CoUninitialize();
 185 }