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 #include "natupnp.h"
  31
  32 static void test_policy2_rules(INetFwPolicy2 *policy2)
  33 {
  34     HRESULT hr;
  35     INetFwRules *rules, *rules2;
  36     INetFwServiceRestriction *restriction;
  37
  38     hr = INetFwPolicy2_QueryInterface(policy2, &IID_INetFwRules, (void**)&rules);
  39     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  40
  41     hr = INetFwPolicy2_get_Rules(policy2, &rules);
  42     ok(hr == S_OK, "got %08x\n", hr);
  43
  44     hr = INetFwPolicy2_get_Rules(policy2, &rules2);
  45     ok(hr == S_OK, "got %08x\n", hr);
  46     ok(rules == rules2, "Different pointers\n");
  47
  48     hr = INetFwPolicy2_get_ServiceRestriction(policy2, &restriction);
  49     todo_wine ok(hr == S_OK, "got %08x\n", hr);
  50     if(hr == S_OK)
  51     {
  52         INetFwRules *rules3;
  53
  54         hr = INetFwServiceRestriction_get_Rules(restriction, &rules3);
  55         ok(hr == S_OK, "got %08x\n", hr);
  56         ok(rules != rules3, "same pointers\n");
  57
  58         if(rules3)
  59             INetFwRules_Release(rules3);
  60         INetFwServiceRestriction_Release(restriction);
  61     }
  62
  63     hr = INetFwRules_get__NewEnum(rules, NULL);
  64     ok(hr == E_POINTER, "got %08x\n", hr);
  65
  66     INetFwRules_Release(rules);
  67     INetFwRules_Release(rules2);
  68 }
  69
  70 static void test_interfaces(void)
  71 {
  72     INetFwMgr *manager;
  73     INetFwPolicy *policy;
  74     INetFwPolicy2 *policy2;
  75     HRESULT hr;
  76
  77     hr = CoCreateInstance(&CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
  78             &IID_INetFwMgr, (void**)&manager);
  79     ok(hr == S_OK, "NetFwMgr create failed: %08x\n", hr);
  80
  81     hr = INetFwMgr_QueryInterface(manager, &IID_INetFwPolicy, (void**)&policy);
  82     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  83
  84     hr = INetFwMgr_QueryInterface(manager, &IID_INetFwPolicy2, (void**)&policy2);
  85     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  86
  87     hr = INetFwMgr_get_LocalPolicy(manager, &policy);
  88     ok(hr == S_OK, "got 0x%08x\n", hr);
  89
  90     hr = INetFwPolicy_QueryInterface(policy, &IID_INetFwPolicy2, (void**)&policy2);
  91     ok(hr == E_NOINTERFACE, "got 0x%08x\n", hr);
  92
  93     INetFwPolicy_Release(policy);
  94
  95     hr = CoCreateInstance(&CLSID_NetFwPolicy2, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
  96             &IID_INetFwPolicy2, (void**)&policy2);
  97     if(hr == S_OK)
  98     {
  99         test_policy2_rules(policy2);
 100
 101         INetFwPolicy2_Release(policy2);
 102     }
 103     else
 104         win_skip("NetFwPolicy2 object is not supported: %08x\n", hr);
 105
 106     INetFwMgr_Release(manager);
 107 }
 108
 109 static void test_NetFwAuthorizedApplication(void)
 110 {
 111     INetFwAuthorizedApplication *app;
 112     static WCHAR empty[] = L"";
 113     UNIVERSAL_NAME_INFOW *info;
 114     WCHAR fullpath[MAX_PATH];
 115     WCHAR netpath[MAX_PATH];
 116     WCHAR image[MAX_PATH];
 117     HRESULT hr;
 118     BSTR bstr;
 119     DWORD sz;
 120
 121     hr = CoCreateInstance(&CLSID_NetFwAuthorizedApplication, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 122             &IID_INetFwAuthorizedApplication, (void**)&app);
 123     ok(hr == S_OK, "got: %08x\n", hr);
 124
 125     hr = GetModuleFileNameW(NULL, image, ARRAY_SIZE(image));
 126     ok(hr, "GetModuleFileName failed: %u\n", GetLastError());
 127
 128     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, NULL);
 129     ok(hr == E_POINTER, "got: %08x\n", hr);
 130
 131     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
 132     ok(hr == S_OK || hr == HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY), "got: %08x\n", hr);
 133     ok(!bstr, "got: %s\n",  wine_dbgstr_w(bstr));
 134
 135     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, NULL);
 136     ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
 137
 138     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, empty);
 139     ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
 140
 141     bstr = SysAllocString(image);
 142     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, bstr);
 143     ok(hr == S_OK, "got: %08x\n", hr);
 144     SysFreeString(bstr);
 145
 146     GetFullPathNameW(image, ARRAY_SIZE(fullpath), fullpath, NULL);
 147     GetLongPathNameW(fullpath, fullpath, ARRAY_SIZE(fullpath));
 148
 149     info = (UNIVERSAL_NAME_INFOW *)&netpath;
 150     sz = sizeof(netpath);
 151     hr = WNetGetUniversalNameW(image, UNIVERSAL_NAME_INFO_LEVEL, info, &sz);
 152     if (hr != NO_ERROR)
 153     {
 154         info->lpUniversalName = netpath + sizeof(*info)/sizeof(WCHAR);
 155         lstrcpyW(info->lpUniversalName, fullpath);
 156     }
 157
 158     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
 159     ok(hr == S_OK, "got: %08x\n", hr);
 160     ok(!lstrcmpW(bstr,info->lpUniversalName), "expected %s, got %s\n",
 161         wine_dbgstr_w(info->lpUniversalName), wine_dbgstr_w(bstr));
 162     SysFreeString(bstr);
 163
 164     INetFwAuthorizedApplication_Release(app);
 165 }
 166
 167 static void test_IUPnPNAT(void)
 168 {
 169     IUPnPNAT *nat;
 170     IStaticPortMappingCollection *static_ports;
 171     IDynamicPortMappingCollection *dync_ports;
 172     INATEventManager *manager;
 173     IProvideClassInfo *provider;
 174     HRESULT hr;
 175
 176     hr = CoCreateInstance(&CLSID_UPnPNAT, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER, &IID_IUPnPNAT, (void**)&nat);
 177     ok(hr == S_OK, "got: %08x\n", hr);
 178
 179     hr = IUPnPNAT_QueryInterface(nat, &IID_IProvideClassInfo, (void**)&provider);
 180     ok(hr == E_NOINTERFACE, "got: %08x\n", hr);
 181
 182     hr = IUPnPNAT_get_StaticPortMappingCollection(nat, &static_ports);
 183     ok(hr == S_OK, "got: %08x\n", hr);
 184     if(hr == S_OK && static_ports)
 185         IStaticPortMappingCollection_Release(static_ports);
 186
 187     hr = IUPnPNAT_get_DynamicPortMappingCollection(nat, &dync_ports);
 188     ok(hr == S_OK || hr == E_NOTIMPL /* Windows 8.1 */, "got: %08x\n", hr);
 189     if(hr == S_OK && dync_ports)
 190         IDynamicPortMappingCollection_Release(dync_ports);
 191
 192     hr = IUPnPNAT_get_NATEventManager(nat, &manager);
 193     todo_wine ok(hr == S_OK, "got: %08x\n", hr);
 194     if(hr == S_OK && manager)
 195         INATEventManager_Release(manager);
 196
 197     IUPnPNAT_Release(nat);
 198 }
 199
 200 START_TEST(policy)
 201 {
 202     INetFwMgr *manager;
 203     HRESULT hr;
 204
 205     CoInitialize(NULL);
 206
 207     hr = CoCreateInstance(&CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 208             &IID_INetFwMgr, (void**)&manager);
 209     if(FAILED(hr))
 210     {
 211         win_skip("NetFwMgr object is not supported: %08x\n", hr);
 212         CoUninitialize();
 213         return;
 214     }
 215
 216     INetFwMgr_Release(manager);
 217
 218     test_interfaces();
 219     test_NetFwAuthorizedApplication();
 220     test_IUPnPNAT();
 221
 222     CoUninitialize();
 223 }