[HNETCFG_WINETEST] Sync with Wine Staging 3.9. CORE-14656
[reactos.git] / modules / rostests / winetests / hnetcfg / 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 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 {
115 BSTR image = SysAllocStringLen( NULL, MAX_PATH );
116 static WCHAR empty[] = {0};
117 BSTR bstr;
118
119 if (!GetModuleFileNameW( NULL, image, MAX_PATH ))
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
157 CoInitialize(NULL);
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();
171 test_NetFwAuthorizedApplication();
172
173 CoUninitialize();
174 }