ReactOS  0.4.13-dev-479-gec9c8fd
policy.c
Go to the documentation of this file.
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 
106 {
108  HRESULT hr;
109 
110  hr = CoCreateInstance(&CLSID_NetFwAuthorizedApplication, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
111  &IID_INetFwAuthorizedApplication, (void**)&app);
112  ok(hr == S_OK, "got: %08x\n", hr);
113  if(hr == S_OK)
114  {
116  static WCHAR empty[] = {0};
117  BSTR bstr;
118 
120  {
121  ok(0, "Failed to get filename\n");
122  SysFreeString( image );
123  return;
124  }
125 
126  hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, NULL);
127  ok(hr == E_POINTER, "got: %08x\n", hr);
128 
129  hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
130  ok(hr == S_OK || hr == HRESULT_FROM_WIN32(ERROR_NOT_ENOUGH_MEMORY), "got: %08x\n", hr);
131  ok(!bstr, "got: %s\n", wine_dbgstr_w(bstr));
132 
133  hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, NULL);
134  ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
135 
136  hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, empty);
137  ok(hr == E_INVALIDARG || hr == HRESULT_FROM_WIN32(ERROR_PATH_NOT_FOUND), "got: %08x\n", hr);
138 
139  hr = INetFwAuthorizedApplication_put_ProcessImageFileName(app, image);
140  ok(hr == S_OK, "got: %08x\n", hr);
141 
142  hr = INetFwAuthorizedApplication_get_ProcessImageFileName(app, &bstr);
143  ok(hr == S_OK, "got: %08x\n", hr);
144  ok(!lstrcmpiW(bstr,image), "got: %s\n", wine_dbgstr_w(bstr));
145  SysFreeString( bstr );
146 
147  SysFreeString( image );
148  INetFwAuthorizedApplication_Release(app);
149  }
150 }
151 
152 START_TEST(policy)
153 {
154  INetFwMgr *manager;
155  HRESULT hr;
156 
158 
159  hr = CoCreateInstance(&CLSID_NetFwMgr, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
160  &IID_INetFwMgr, (void**)&manager);
161  if(FAILED(hr))
162  {
163  win_skip("NetFwMgr object is not supported: %08x\n", hr);
164  CoUninitialize();
165  return;
166  }
167 
168  INetFwMgr_Release(manager);
169 
170  test_interfaces();
172 
173  CoUninitialize();
174 }
DWORD WINAPI GetModuleFileNameW(HINSTANCE hModule, LPWSTR lpFilename, DWORD nSize)
Definition: loader.c:607
#define E_NOINTERFACE
Definition: winerror.h:2364
#define HRESULT_FROM_WIN32(x)
Definition: winerror.h:92
HRESULT hr
Definition: shlfolder.c:183
BSTR WINAPI SysAllocStringLen(const OLECHAR *str, unsigned int len)
Definition: oleaut.c:342
START_TEST(policy)
Definition: policy.c:152
static const WCHAR empty[]
Definition: main.c:49
#define wine_dbgstr_w
Definition: kernel32.h:34
#define ERROR_NOT_ENOUGH_MEMORY
Definition: dderror.h:7
OLECHAR * BSTR
Definition: compat.h:1934
int WINAPI lstrcmpiW(LPCWSTR lpString1, LPCWSTR lpString2)
Definition: lstring.c:194
static void test_NetFwAuthorizedApplication(void)
Definition: policy.c:105
static void test_interfaces(void)
Definition: policy.c:66
static void test_policy2_rules(INetFwPolicy2 *policy2)
Definition: policy.c:31
#define ok(value,...)
#define E_INVALIDARG
Definition: ddrawi.h:101
smooth NULL
Definition: ftsmooth.c:416
__wchar_t WCHAR
Definition: xmlstorage.h:180
LONG HRESULT
Definition: typedefs.h:77
#define MAX_PATH
Definition: compat.h:26
#define todo_wine
Definition: test.h:154
HRESULT WINAPI DECLSPEC_HOTPATCH CoCreateInstance(REFCLSID rclsid, LPUNKNOWN pUnkOuter, DWORD dwClsContext, REFIID iid, LPVOID *ppv)
Definition: compobj.c:3234
#define S_OK
Definition: intsafe.h:59
void WINAPI DECLSPEC_HOTPATCH CoUninitialize(void)
Definition: compobj.c:1991
void WINAPI DECLSPEC_HOTPATCH SysFreeString(BSTR str)
Definition: oleaut.c:274
HRESULT WINAPI CoInitialize(LPVOID lpReserved)
Definition: compobj.c:1897
#define ERROR_PATH_NOT_FOUND
Definition: winerror.h:106
#define E_POINTER
Definition: winerror.h:2365
#define win_skip
Definition: test.h:141