2 * Copyright 2007 Tim Schwartz
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.
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.
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
21 #include <wine/debug.h>
23 #include "resources.h"
25 WINE_DEFAULT_DEBUG_CHANNEL(net
);
27 #define NET_START 0001
30 static int output_write(const WCHAR
* str
, int len
)
33 ret
= WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE
), str
, len
, &count
, NULL
);
39 /* On Windows WriteConsoleW() fails if the output is redirected. So fall
40 * back to WriteFile(), assuming the console encoding is still the right
43 lenA
= WideCharToMultiByte(GetConsoleOutputCP(), 0, str
, len
,
45 strA
= HeapAlloc(GetProcessHeap(), 0, lenA
);
49 WideCharToMultiByte(GetConsoleOutputCP(), 0, str
, len
, strA
, lenA
,
51 WriteFile(GetStdHandle(STD_OUTPUT_HANDLE
), strA
, lenA
, &count
, FALSE
);
52 HeapFree(GetProcessHeap(), 0, strA
);
57 static int output_vprintf(const WCHAR
* fmt
, __ms_va_list va_args
)
62 SetLastError(NO_ERROR
);
63 len
= FormatMessageW(FORMAT_MESSAGE_FROM_STRING
, fmt
, 0, 0, str
, ARRAY_SIZE(str
), &va_args
);
64 if (len
== 0 && GetLastError() != NO_ERROR
)
65 WINE_FIXME("Could not format string: le=%u, fmt=%s\n", GetLastError(), wine_dbgstr_w(fmt
));
67 output_write(str
, len
);
71 static int WINAPIV
output_printf(const WCHAR
* fmt
, ...)
73 __ms_va_list arguments
;
75 __ms_va_start(arguments
, fmt
);
76 output_vprintf(fmt
, arguments
);
77 __ms_va_end(arguments
);
81 static int WINAPIV
output_string(int msg
, ...)
84 __ms_va_list arguments
;
86 LoadStringW(GetModuleHandleW(NULL
), msg
, fmt
, ARRAY_SIZE(fmt
));
87 __ms_va_start(arguments
, msg
);
88 output_vprintf(fmt
, arguments
);
89 __ms_va_end(arguments
);
93 static BOOL
output_error_string(DWORD error
)
96 if (FormatMessageW(FORMAT_MESSAGE_FROM_SYSTEM
|
97 FORMAT_MESSAGE_IGNORE_INSERTS
| FORMAT_MESSAGE_ALLOCATE_BUFFER
,
98 NULL
, error
, 0, (LPWSTR
)&pBuffer
, 0, NULL
))
100 output_write(pBuffer
, lstrlenW(pBuffer
));
107 static BOOL
net_use(int argc
, const WCHAR
* argv
[])
109 USE_INFO_2
*buffer
, *connection
;
110 DWORD read
, total
, resume_handle
, rc
, i
;
111 WCHAR
* status
[STRING_RECONN
-STRING_OK
+1];
117 HMODULE hmod
= GetModuleHandleW(NULL
);
119 /* Load the status strings */
120 for (i
= 0; i
< ARRAY_SIZE(status
); i
++)
122 status
[i
] = HeapAlloc(GetProcessHeap(), 0, 1024 * sizeof(**status
));
123 LoadStringW(hmod
, STRING_OK
+i
, status
[i
], 1024);
127 rc
= NetUseEnum(NULL
, 2, (BYTE
**) &buffer
, 2048, &read
, &total
, &resume_handle
);
128 if (rc
!= ERROR_MORE_DATA
&& rc
!= ERROR_SUCCESS
)
135 output_string(STRING_NO_ENTRIES
);
139 output_string(STRING_USE_HEADER
);
140 for (i
= 0, connection
= buffer
; i
< read
; ++i
, ++connection
)
141 output_string(STRING_USE_ENTRY
, status
[connection
->ui2_status
], connection
->ui2_local
,
142 connection
->ui2_remote
, connection
->ui2_refcount
);
144 if (buffer
!= NULL
) NetApiBufferFree(buffer
);
145 } while (rc
== ERROR_MORE_DATA
);
147 /* Release the status strings */
148 for (i
= 0; i
< ARRAY_SIZE(status
); i
++)
149 HeapFree(GetProcessHeap(), 0, status
[i
]);
157 static BOOL
net_enum_services(void)
159 static const WCHAR runningW
[]={' ',' ',' ',' ','%','1','\n',0};
161 LPENUM_SERVICE_STATUS_PROCESSW services
;
162 DWORD size
, i
, count
, resume
;
163 BOOL success
= FALSE
;
165 SCManager
= OpenSCManagerW(NULL
, NULL
, SC_MANAGER_ALL_ACCESS
);
168 output_string(STRING_NO_SCM
);
172 EnumServicesStatusExW(SCManager
, SC_ENUM_PROCESS_INFO
, SERVICE_WIN32
, SERVICE_ACTIVE
, NULL
, 0, &size
, &count
, NULL
, NULL
);
173 if(GetLastError() != ERROR_MORE_DATA
)
175 output_error_string(GetLastError());
178 services
= HeapAlloc(GetProcessHeap(), 0, size
);
180 if(!EnumServicesStatusExW(SCManager
, SC_ENUM_PROCESS_INFO
, SERVICE_WIN32
, SERVICE_ACTIVE
, (LPBYTE
)services
, size
, &size
, &count
, &resume
, NULL
))
182 output_error_string(GetLastError());
185 output_string(STRING_RUNNING_HEADER
);
186 for(i
= 0; i
< count
; i
++)
188 output_printf(runningW
, services
[i
].lpDisplayName
);
189 WINE_TRACE("service=%s state=%d controls=%x\n",
190 wine_dbgstr_w(services
[i
].lpServiceName
),
191 services
[i
].ServiceStatusProcess
.dwCurrentState
,
192 services
[i
].ServiceStatusProcess
.dwControlsAccepted
);
197 CloseServiceHandle(SCManager
);
201 static BOOL
StopService(SC_HANDLE SCManager
, SC_HANDLE serviceHandle
)
203 LPENUM_SERVICE_STATUSW dependencies
= NULL
;
204 DWORD buffer_size
= 0;
205 DWORD count
= 0, counter
;
207 SC_HANDLE dependent_serviceHandle
;
208 SERVICE_STATUS_PROCESS ssp
;
210 result
= EnumDependentServicesW(serviceHandle
, SERVICE_ACTIVE
, dependencies
, buffer_size
, &buffer_size
, &count
);
212 if(!result
&& (GetLastError() == ERROR_MORE_DATA
))
214 dependencies
= HeapAlloc(GetProcessHeap(), 0, buffer_size
);
215 if(EnumDependentServicesW(serviceHandle
, SERVICE_ACTIVE
, dependencies
, buffer_size
, &buffer_size
, &count
))
217 for(counter
= 0; counter
< count
; counter
++)
219 output_string(STRING_STOP_DEP
, dependencies
[counter
].lpDisplayName
);
220 dependent_serviceHandle
= OpenServiceW(SCManager
, dependencies
[counter
].lpServiceName
, SC_MANAGER_ALL_ACCESS
);
221 if(dependent_serviceHandle
)
223 result
= StopService(SCManager
, dependent_serviceHandle
);
224 CloseServiceHandle(dependent_serviceHandle
);
226 if(!result
) output_string(STRING_CANT_STOP
, dependencies
[counter
].lpDisplayName
);
231 if(result
) result
= ControlService(serviceHandle
, SERVICE_CONTROL_STOP
, (LPSERVICE_STATUS
)&ssp
);
232 HeapFree(GetProcessHeap(), 0, dependencies
);
236 static BOOL
net_service(int operation
, const WCHAR
* service_name
)
238 SC_HANDLE SCManager
, serviceHandle
;
240 WCHAR service_display_name
[4096];
243 SCManager
= OpenSCManagerW(NULL
, NULL
, SC_MANAGER_ALL_ACCESS
);
246 output_string(STRING_NO_SCM
);
249 serviceHandle
= OpenServiceW(SCManager
, service_name
, SC_MANAGER_ALL_ACCESS
);
252 output_string(STRING_NO_SVCHANDLE
);
253 CloseServiceHandle(SCManager
);
257 buffer_size
= ARRAY_SIZE(service_display_name
);
258 GetServiceDisplayNameW(SCManager
, service_name
, service_display_name
, &buffer_size
);
259 if (!service_display_name
[0]) lstrcpyW(service_display_name
, service_name
);
264 output_string(STRING_START_SVC
, service_display_name
);
265 result
= StartServiceW(serviceHandle
, 0, NULL
);
267 if(result
) output_string(STRING_START_SVC_SUCCESS
, service_display_name
);
270 if (!output_error_string(GetLastError()))
271 output_string(STRING_START_SVC_FAIL
, service_display_name
);
275 output_string(STRING_STOP_SVC
, service_display_name
);
276 result
= StopService(SCManager
, serviceHandle
);
278 if(result
) output_string(STRING_STOP_SVC_SUCCESS
, service_display_name
);
281 if (!output_error_string(GetLastError()))
282 output_string(STRING_STOP_SVC_FAIL
, service_display_name
);
287 CloseServiceHandle(serviceHandle
);
288 CloseServiceHandle(SCManager
);
292 static BOOL
arg_is(const WCHAR
* str1
, const WCHAR
* str2
)
294 return CompareStringW(LOCALE_USER_DEFAULT
, NORM_IGNORECASE
, str1
, -1, str2
, -1) == CSTR_EQUAL
;
297 int __cdecl
wmain(int argc
, const WCHAR
* argv
[])
299 static const WCHAR helpW
[]={'h','e','l','p',0};
300 static const WCHAR shelpW
[]={'/','h','e','l','p',0};
301 static const WCHAR startW
[]={'s','t','a','r','t',0};
302 static const WCHAR stopW
[]={'s','t','o','p',0};
303 static const WCHAR useW
[]={'u','s','e',0};
307 output_string(STRING_USAGE
);
311 if(arg_is(argv
[1], helpW
))
315 output_string(STRING_USAGE
);
319 output_string(STRING_USAGE
);
320 else if(arg_is(argv
[2], startW
))
321 output_string(STRING_START_USAGE
);
322 else if(arg_is(argv
[2], stopW
))
323 output_string(STRING_STOP_USAGE
);
325 output_string(STRING_USAGE
);
327 else if(arg_is(argv
[1], startW
))
331 output_string(STRING_START_USAGE
);
336 if (!net_enum_services())
339 else if(arg_is(argv
[2], shelpW
))
340 output_string(STRING_START_USAGE
);
341 else if(!net_service(NET_START
, argv
[2]))
344 else if(arg_is(argv
[1], stopW
))
348 output_string(STRING_STOP_USAGE
);
351 if(arg_is(argv
[2], shelpW
))
352 output_string(STRING_STOP_USAGE
);
353 else if(!net_service(NET_STOP
, argv
[2]))
356 else if(arg_is(argv
[1], useW
))
358 if(!net_use(argc
, argv
)) return 1;
361 output_string(STRING_USAGE
);