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 netpath[MAX_PATH];
 111     WCHAR image[MAX_PATH];
 112     HRESULT hr;
 113     BSTR bstr;
 114     DWORD sz;
 115
 116     hr = CoCreateInstance(&CLSID_NetFwAuthorizedApplication, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 117             &IID_INetFwAuthorizedApplication, (void**)&app);
 118     ok(hr == S_OK, "got: %08x\n", hr);
 119
 120     hr = GetModuleFileNameW(NULL, image, ARRAY_SIZE(image));
 121     ok(hr, "GetModuleFileName failed: %u\n", GetLastError());
 122
 123     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, NULL);
 124     ok(hr == E_POINTER, "got: %08x\n", hr);
 125
 126     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
 127     ok(hr == S_OK || hr == HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY), "got: %08x\n", hr);
 128     ok(!bstr, "got: %s\n",  wine_dbgstr_w(bstr));
 129
 130     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, NULL);
 131     ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
 132
 133     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, empty);
 134     ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
 135
 136     bstr = SysAllocString(image);
 137     hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, bstr);
 138     ok(hr == S_OK, "got: %08x\n", hr);
 139     SysFreeString(bstr);
 140
 141     info = (UNIVERSAL_NAME_INFOW *)&netpath;
 142     sz = sizeof(netpath);
 143     hr = WNetGetUniversalNameW(image, UNIVERSAL_NAME_INFO_LEVEL, &info, &sz);
 144     if (hr != NO_ERROR)
 145     {
 146         info->lpUniversalName = netpath + sizeof(*info)/sizeof(WCHAR);
 147         lstrcpyW(info->lpUniversalName, image);
 148     }
 149
 150     hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
 151     ok(hr == S_OK, "got: %08x\n", hr);
 152     ok(!lstrcmpW(bstr,info->lpUniversalName), "expected %s, got %s\n",
 153         wine_dbgstr_w(info->lpUniversalName), wine_dbgstr_w(bstr));
 154     SysFreeString(bstr);
 155
 156     INetFwAuthorizedApplication_Release(app);
 157 }
 158
 159 START_TEST(policy)
 160 {
 161     INetFwMgr *manager;
 162     HRESULT hr;
 163
 164     CoInitialize(NULL);
 165
 166     hr = CoCreateInstance(&CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 167             &IID_INetFwMgr, (void**)&manager);
 168     if(FAILED(hr))
 169     {
 170         win_skip("NetFwMgr object is not supported: %08x\n", hr);
 171         CoUninitialize();
 172         return;
 173     }
 174
 175     INetFwMgr_Release(manager);
 176
 177     test_interfaces();
 178     test_NetFwAuthorizedApplication();
 179
 180     CoUninitialize();
 181 }