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 if (!WriteConsoleW(GetStdHandle(STD_OUTPUT_HANDLE
), str
, len
, &count
, NULL
))
38 /* On Windows WriteConsoleW() fails if the output is redirected. So fall
39 * back to WriteFile() with OEM code page.
41 lenA
= WideCharToMultiByte(GetOEMCP(), 0, str
, len
,
43 strA
= HeapAlloc(GetProcessHeap(), 0, lenA
);
47 WideCharToMultiByte(GetOEMCP(), 0, str
, len
, strA
, lenA
, NULL
, NULL
);
48 WriteFile(GetStdHandle(STD_OUTPUT_HANDLE
), strA
, lenA
, &count
, FALSE
);
49 HeapFree(GetProcessHeap(), 0, strA
);
54 static int output_vprintf(const WCHAR
* fmt
, va_list va_args
)
59 len
= FormatMessageW(FORMAT_MESSAGE_FROM_STRING
, fmt
, 0, 0, str
, ARRAY_SIZE(str
), &va_args
);
60 if (len
== 0 && GetLastError() != ERROR_NO_WORK_DONE
)
61 WINE_FIXME("Could not format string: le=%lu, fmt=%s\n", GetLastError(), wine_dbgstr_w(fmt
));
63 output_write(str
, len
);
67 static int WINAPIV
output_printf(const WCHAR
* fmt
, ...)
71 va_start(arguments
, fmt
);
72 output_vprintf(fmt
, arguments
);
77 static int WINAPIV
output_string(int msg
, ...)
82 LoadStringW(GetModuleHandleW(NULL
), msg
, fmt
, ARRAY_SIZE(fmt
));
83 va_start(arguments
, msg
);
84 output_vprintf(fmt
, arguments
);
89 static BOOL
output_error_string(DWORD error
)
92 if (FormatMessageW(FORMAT_MESSAGE_FROM_SYSTEM
|
93 FORMAT_MESSAGE_IGNORE_INSERTS
| FORMAT_MESSAGE_ALLOCATE_BUFFER
,
94 NULL
, error
, 0, (LPWSTR
)&pBuffer
, 0, NULL
))
96 output_write(pBuffer
, lstrlenW(pBuffer
));
103 static BOOL
net_use(int argc
, const WCHAR
* argv
[])
105 USE_INFO_2
*buffer
, *connection
;
106 DWORD read
, total
, resume_handle
, rc
, i
;
107 WCHAR
* status
[STRING_RECONN
-STRING_OK
+1];
113 HMODULE hmod
= GetModuleHandleW(NULL
);
115 /* Load the status strings */
116 for (i
= 0; i
< ARRAY_SIZE(status
); i
++)
118 status
[i
] = HeapAlloc(GetProcessHeap(), 0, 1024 * sizeof(**status
));
119 LoadStringW(hmod
, STRING_OK
+i
, status
[i
], 1024);
123 rc
= NetUseEnum(NULL
, 2, (BYTE
**) &buffer
, 2048, &read
, &total
, &resume_handle
);
124 if (rc
!= ERROR_MORE_DATA
&& rc
!= ERROR_SUCCESS
)
131 output_string(STRING_NO_ENTRIES
);
135 output_string(STRING_USE_HEADER
);
136 for (i
= 0, connection
= buffer
; i
< read
; ++i
, ++connection
)
137 output_string(STRING_USE_ENTRY
, status
[connection
->ui2_status
], connection
->ui2_local
,
138 connection
->ui2_remote
, connection
->ui2_refcount
);
140 if (buffer
!= NULL
) NetApiBufferFree(buffer
);
141 } while (rc
== ERROR_MORE_DATA
);
143 /* Release the status strings */
144 for (i
= 0; i
< ARRAY_SIZE(status
); i
++)
145 HeapFree(GetProcessHeap(), 0, status
[i
]);
153 static BOOL
net_enum_services(void)
156 LPENUM_SERVICE_STATUS_PROCESSW services
;
157 DWORD size
, i
, count
, resume
;
158 BOOL success
= FALSE
;
160 SCManager
= OpenSCManagerW(NULL
, NULL
, SC_MANAGER_ALL_ACCESS
);
163 output_string(STRING_NO_SCM
);
167 EnumServicesStatusExW(SCManager
, SC_ENUM_PROCESS_INFO
, SERVICE_WIN32
, SERVICE_ACTIVE
, NULL
, 0, &size
, &count
, NULL
, NULL
);
168 if(GetLastError() != ERROR_MORE_DATA
)
170 output_error_string(GetLastError());
173 services
= HeapAlloc(GetProcessHeap(), 0, size
);
175 if(!EnumServicesStatusExW(SCManager
, SC_ENUM_PROCESS_INFO
, SERVICE_WIN32
, SERVICE_ACTIVE
, (LPBYTE
)services
, size
, &size
, &count
, &resume
, NULL
))
177 output_error_string(GetLastError());
180 output_string(STRING_RUNNING_HEADER
);
181 for(i
= 0; i
< count
; i
++)
183 output_printf(L
" %1\n", services
[i
].lpDisplayName
);
184 WINE_TRACE("service=%s state=%ld controls=%lx\n",
185 wine_dbgstr_w(services
[i
].lpServiceName
),
186 services
[i
].ServiceStatusProcess
.dwCurrentState
,
187 services
[i
].ServiceStatusProcess
.dwControlsAccepted
);
192 CloseServiceHandle(SCManager
);
196 static BOOL
StopService(SC_HANDLE SCManager
, SC_HANDLE serviceHandle
)
198 LPENUM_SERVICE_STATUSW dependencies
= NULL
;
199 DWORD buffer_size
= 0;
200 DWORD count
= 0, counter
;
202 SC_HANDLE dependent_serviceHandle
;
203 SERVICE_STATUS_PROCESS ssp
;
205 result
= EnumDependentServicesW(serviceHandle
, SERVICE_ACTIVE
, dependencies
, buffer_size
, &buffer_size
, &count
);
207 if(!result
&& (GetLastError() == ERROR_MORE_DATA
))
209 dependencies
= HeapAlloc(GetProcessHeap(), 0, buffer_size
);
210 if(EnumDependentServicesW(serviceHandle
, SERVICE_ACTIVE
, dependencies
, buffer_size
, &buffer_size
, &count
))
212 for(counter
= 0; counter
< count
; counter
++)
214 output_string(STRING_STOP_DEP
, dependencies
[counter
].lpDisplayName
);
215 dependent_serviceHandle
= OpenServiceW(SCManager
, dependencies
[counter
].lpServiceName
, SC_MANAGER_ALL_ACCESS
);
216 if(dependent_serviceHandle
)
218 result
= StopService(SCManager
, dependent_serviceHandle
);
219 CloseServiceHandle(dependent_serviceHandle
);
221 if(!result
) output_string(STRING_CANT_STOP
, dependencies
[counter
].lpDisplayName
);
226 if(result
) result
= ControlService(serviceHandle
, SERVICE_CONTROL_STOP
, (LPSERVICE_STATUS
)&ssp
);
227 HeapFree(GetProcessHeap(), 0, dependencies
);
231 static BOOL
net_service(int operation
, const WCHAR
* service_name
)
233 SC_HANDLE SCManager
, serviceHandle
;
235 WCHAR service_display_name
[4096];
238 SCManager
= OpenSCManagerW(NULL
, NULL
, SC_MANAGER_ALL_ACCESS
);
241 output_string(STRING_NO_SCM
);
244 serviceHandle
= OpenServiceW(SCManager
, service_name
, SC_MANAGER_ALL_ACCESS
);
247 output_string(STRING_NO_SVCHANDLE
);
248 CloseServiceHandle(SCManager
);
252 buffer_size
= ARRAY_SIZE(service_display_name
);
253 GetServiceDisplayNameW(SCManager
, service_name
, service_display_name
, &buffer_size
);
254 if (!service_display_name
[0]) lstrcpyW(service_display_name
, service_name
);
259 output_string(STRING_START_SVC
, service_display_name
);
260 result
= StartServiceW(serviceHandle
, 0, NULL
);
262 if(result
) output_string(STRING_START_SVC_SUCCESS
, service_display_name
);
265 if (!output_error_string(GetLastError()))
266 output_string(STRING_START_SVC_FAIL
, service_display_name
);
270 output_string(STRING_STOP_SVC
, service_display_name
);
271 result
= StopService(SCManager
, serviceHandle
);
273 if(result
) output_string(STRING_STOP_SVC_SUCCESS
, service_display_name
);
276 if (!output_error_string(GetLastError()))
277 output_string(STRING_STOP_SVC_FAIL
, service_display_name
);
282 CloseServiceHandle(serviceHandle
);
283 CloseServiceHandle(SCManager
);
287 static BOOL
arg_is(const WCHAR
* str1
, const WCHAR
* str2
)
289 return CompareStringW(LOCALE_USER_DEFAULT
, NORM_IGNORECASE
, str1
, -1, str2
, -1) == CSTR_EQUAL
;
292 int __cdecl
wmain(int argc
, const WCHAR
* argv
[])
296 output_string(STRING_USAGE
);
300 if(arg_is(argv
[1], L
"help"))
304 output_string(STRING_USAGE
);
308 output_string(STRING_USAGE
);
309 else if(arg_is(argv
[2], L
"start"))
310 output_string(STRING_START_USAGE
);
311 else if(arg_is(argv
[2], L
"stop"))
312 output_string(STRING_STOP_USAGE
);
314 output_string(STRING_USAGE
);
316 else if(arg_is(argv
[1], L
"start"))
320 output_string(STRING_START_USAGE
);
325 if (!net_enum_services())
328 else if(arg_is(argv
[2], L
"/help"))
329 output_string(STRING_START_USAGE
);
330 else if(!net_service(NET_START
, argv
[2]))
333 else if(arg_is(argv
[1], L
"stop"))
337 output_string(STRING_STOP_USAGE
);
340 if(arg_is(argv
[2], L
"/help"))
341 output_string(STRING_STOP_USAGE
);
342 else if(!net_service(NET_STOP
, argv
[2]))
345 else if(arg_is(argv
[1], L
"use"))
347 if(!net_use(argc
, argv
)) return 1;
350 output_string(STRING_USAGE
);