mirror of
https://github.com/reactos/reactos.git
synced 2024-11-04 22:00:55 +00:00
375 lines
13 KiB
C
375 lines
13 KiB
C
/*
|
|
* ReactOS winfile
|
|
*
|
|
* network.c
|
|
*
|
|
* Copyright (C) 2002 Robert Dickenson <robd@reactos.org>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*/
|
|
|
|
#define WIN32_LEAN_AND_MEAN // Exclude rarely-used stuff from Windows headers
|
|
#include <windows.h>
|
|
#include <commctrl.h>
|
|
#include <stdlib.h>
|
|
#include <malloc.h>
|
|
#include <memory.h>
|
|
#include <tchar.h>
|
|
#include <process.h>
|
|
#include <stdio.h>
|
|
|
|
#include "main.h"
|
|
#include "network.h"
|
|
|
|
#include "trace.h"
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// Global Variables:
|
|
//
|
|
|
|
static HANDLE hNetworkMonitorThreadEvent = NULL; // When this event becomes signaled then we run the monitor thread
|
|
|
|
static HMODULE hMPR;
|
|
static BOOL bNetAvailable = FALSE;
|
|
|
|
typedef DWORD (WINAPI *WNetCloseEnum_Ptr)(HANDLE);
|
|
typedef DWORD (WINAPI *WNetConnectionDialog_Ptr)(HWND, DWORD);
|
|
typedef DWORD (WINAPI *WNetDisconnectDialog_Ptr)(HWND, DWORD);
|
|
typedef DWORD (WINAPI *WNetConnectionDialog1_Ptr)(LPCONNECTDLGSTRUCT);
|
|
typedef DWORD (WINAPI *WNetDisconnectDialog1_Ptr)(LPDISCDLGSTRUCT);
|
|
typedef DWORD (WINAPI *WNetEnumResourceA_Ptr)(HANDLE, LPDWORD, LPVOID, LPDWORD);
|
|
typedef DWORD (WINAPI *WNetOpenEnumA_Ptr)(DWORD, DWORD, DWORD, LPNETRESOURCE, LPHANDLE);
|
|
|
|
static WNetCloseEnum_Ptr pWNetCloseEnum;
|
|
static WNetConnectionDialog_Ptr pWNetConnectionDialog;
|
|
static WNetDisconnectDialog_Ptr pWNetDisconnectDialog;
|
|
static WNetConnectionDialog1_Ptr pWNetConnectionDialog1;
|
|
static WNetDisconnectDialog1_Ptr pWNetDisconnectDialog1;
|
|
static WNetEnumResourceA_Ptr pWNetEnumResource;
|
|
static WNetOpenEnumA_Ptr pWNetOpenEnum;
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
// Local module support methods
|
|
//
|
|
|
|
static BOOL CheckNetworkAvailable(void)
|
|
{
|
|
|
|
hMPR = LoadLibrary(_T("MPR.DLL"));
|
|
if (hMPR) {
|
|
pWNetCloseEnum = (WNetCloseEnum_Ptr)(FARPROC)GetProcAddress(hMPR, "WNetCloseEnum");
|
|
pWNetConnectionDialog = (WNetConnectionDialog_Ptr)(FARPROC)GetProcAddress(hMPR, "WNetConnectionDialog");
|
|
pWNetDisconnectDialog = (WNetDisconnectDialog_Ptr)(FARPROC)GetProcAddress(hMPR, "WNetDisconnectDialog");
|
|
pWNetConnectionDialog1 = (WNetConnectionDialog1_Ptr)(FARPROC)GetProcAddress(hMPR, "WNetConnectionDialog1");
|
|
pWNetDisconnectDialog1 = (WNetDisconnectDialog1_Ptr)(FARPROC)GetProcAddress(hMPR, "WNetDisconnectDialog1");
|
|
pWNetEnumResource = (WNetEnumResourceA_Ptr)(FARPROC)GetProcAddress(hMPR, "WNetEnumResourceA");
|
|
pWNetOpenEnum = (WNetOpenEnumA_Ptr)(FARPROC)GetProcAddress(hMPR, "WNetOpenEnumA");
|
|
// FreeLibrary(hMPR);
|
|
bNetAvailable = TRUE;
|
|
}
|
|
return (hMPR != NULL);
|
|
}
|
|
|
|
|
|
static LRESULT CALLBACK EnumNetConnectionsProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
DWORD WNetOpenEnum(
|
|
DWORD dwScope, // scope of enumeration
|
|
DWORD dwType, // resource types to list
|
|
DWORD dwUsage, // resource usage to list
|
|
LPNETRESOURCE lpNetResource, // resource structure
|
|
LPHANDLE lphEnum // enumeration handle buffer
|
|
);
|
|
|
|
result = WNetOpenEnum(RESOURCE_CONNECTED, RESOURCETYPE_DISK, RESOURCEUSAGE_ALL, NULL, &EnumNetConnectionsProc);
|
|
|
|
*/
|
|
DWORD MapNetworkDrives(HWND hWnd, BOOL connect)
|
|
{
|
|
DWORD result = 0L;
|
|
|
|
if (!bNetAvailable) return result;
|
|
#if 1
|
|
if (connect) {
|
|
pWNetConnectionDialog(hWnd, RESOURCETYPE_DISK);
|
|
} else {
|
|
pWNetDisconnectDialog(hWnd, RESOURCETYPE_DISK);
|
|
}
|
|
#else
|
|
if (connect) {
|
|
NETRESOURCE netResouce;
|
|
CONNECTDLGSTRUCT connectDlg;
|
|
|
|
//netResouce.dwScope;
|
|
//netResouce.dwType;
|
|
netResouce.dwDisplayType = 0;
|
|
//netResouce.dwUsage;
|
|
//netResouce.lpLocalName;
|
|
//netResouce.lpRemoteName;
|
|
//netResouce.lpComment;
|
|
//netResouce.lpProvider;
|
|
|
|
//connectDlg.cbStructure;
|
|
connectDlg.hwndOwner = hWnd;
|
|
connectDlg.lpConnRes = &netResouce;
|
|
//connectDlg.dwFlags;
|
|
//connectDlg.dwDevNum;
|
|
|
|
result = WNetConnectionDialog1(&connectDlg);
|
|
} else {
|
|
DISCDLGSTRUCT disconnectDlg;
|
|
//disconnectDlg.cbStructure;
|
|
disconnectDlg.hwndOwner = hWnd;
|
|
//disconnectDlg.lpLocalName;
|
|
//disconnectDlg.lpRemoteName;
|
|
//disconnectDlg.dwFlags;
|
|
result = pWNetDisconnectDialog1(&disconnectDlg);
|
|
}
|
|
#endif
|
|
return result;
|
|
}
|
|
|
|
////////////////////////////////////
|
|
static void NetErrorHandler(HWND hwnd, DWORD dwResult, LPTSTR str)
|
|
{
|
|
TRACE(_T("NetErrorHandler(0x%08X) %s\n"), dwResult, str);
|
|
}
|
|
|
|
static void DisplayStruct(HDC hdc, LPNETRESOURCE lpnrLocal)
|
|
{
|
|
LPTSTR str = NULL;
|
|
//TRACE(_T("DisplayStruct(%p)\n"), lpnrLocal);
|
|
|
|
switch (lpnrLocal->dwScope) {
|
|
case RESOURCE_CONNECTED: str = _T("Enumerate currently connected resources. The dwUsage member cannot be specified."); break;
|
|
case RESOURCE_GLOBALNET: str = _T("Enumerate all resources on the network. The dwUsage member is specified."); break;
|
|
case RESOURCE_REMEMBERED: str = _T("Enumerate remembered (persistent) connections. The dwUsage member cannot be specified."); break;
|
|
default: str = _T("Unknown Scope."); break;
|
|
}
|
|
//TRACE(_T(" %s\n"), str);
|
|
|
|
switch (lpnrLocal->dwType) {
|
|
case RESOURCETYPE_ANY: str = _T("All resources."); break;
|
|
case RESOURCETYPE_DISK: str = _T("Disk resources."); break;
|
|
case RESOURCETYPE_PRINT: str = _T("Print resources."); break;
|
|
default: str = _T("Unknown Type."); break;
|
|
}
|
|
//TRACE(_T(" %s\n"), str);
|
|
|
|
switch (lpnrLocal->dwDisplayType) {
|
|
case RESOURCEDISPLAYTYPE_DOMAIN: str = _T("The object should be displayed as a domain."); break;
|
|
case RESOURCEDISPLAYTYPE_SERVER: str = _T("The object should be displayed as a server."); break;
|
|
case RESOURCEDISPLAYTYPE_SHARE: str = _T("The object should be displayed as a share."); break;
|
|
case RESOURCEDISPLAYTYPE_GENERIC: str = _T("The method used to display the object does not matter."); break;
|
|
default: str = _T("Unknown DisplayType."); break;
|
|
}
|
|
//TRACE(_T(" %s\n"), str);
|
|
|
|
// switch (lpnrLocal->dwUsage ) {
|
|
// case RESOURCEUSAGE_CONNECTABLE: str = _T("The resource is a connectable resource; the name pointed to by the lpRemoteName member can be passed to the WNetAddConnection function to make a network connection."); break;
|
|
// case RESOURCEUSAGE_CONTAINER: str = _T("The resource is a container resource; the name pointed to by the lpRemoteName member can be passed to the WNetOpenEnum function to enumerate the resources in the container."); break;
|
|
// default: str = _T("Unknown Usage."); break;
|
|
// }
|
|
//TRACE(_T("\tLocalName: %s\tRemoteName: %s"), lpnrLocal->lpLocalName, lpnrLocal->lpRemoteName);
|
|
//TRACE(_T("\tComment: %s\tProvider: %s\n"), lpnrLocal->lpComment, lpnrLocal->lpProvider);
|
|
}
|
|
|
|
////////////////////////////////////
|
|
|
|
static BOOL WINAPI EnumerateFunc(HWND hwnd, HDC hdc, LPNETRESOURCE lpnr)
|
|
{
|
|
DWORD dwResult;
|
|
DWORD dwResultEnum;
|
|
HANDLE hEnum;
|
|
DWORD cbBuffer = 16384; // 16K is a good size
|
|
DWORD cEntries = -1; // enumerate all possible entries
|
|
LPNETRESOURCE lpnrLocal; // pointer to enumerated structures
|
|
DWORD i;
|
|
|
|
if (!bNetAvailable) return FALSE;
|
|
|
|
// Call the WNetOpenEnum function to begin the enumeration.
|
|
dwResult = pWNetOpenEnum(RESOURCE_GLOBALNET, // all network resources
|
|
// RESOURCETYPE_ANY, // all resources
|
|
RESOURCETYPE_DISK, // disk resources only, exlude printers
|
|
0, // enumerate all resources
|
|
lpnr, // NULL first time the function is called
|
|
&hEnum); // handle to the resource
|
|
|
|
if (dwResult != NO_ERROR) {
|
|
// Process errors with an application-defined error handler.
|
|
NetErrorHandler(hwnd, dwResult, (LPTSTR)_T("WNetOpenEnum"));
|
|
return FALSE;
|
|
}
|
|
|
|
// Call the GlobalAlloc function to allocate resources.
|
|
lpnrLocal = (LPNETRESOURCE)GlobalAlloc(GPTR, cbBuffer);
|
|
|
|
do {
|
|
// Initialize the buffer.
|
|
ZeroMemory(lpnrLocal, cbBuffer);
|
|
|
|
// Call the WNetEnumResource function to continue the enumeration.
|
|
dwResultEnum = pWNetEnumResource(hEnum, // resource handle
|
|
&cEntries, // defined locally as -1
|
|
lpnrLocal, // LPNETRESOURCE
|
|
&cbBuffer); // buffer size
|
|
|
|
// If the call succeeds, loop through the structures.
|
|
if (dwResultEnum == NO_ERROR) {
|
|
for (i = 0; i < cEntries; i++) {
|
|
// Call an application-defined function to display the contents of the NETRESOURCE structures.
|
|
DisplayStruct(hdc, &lpnrLocal[i]);
|
|
|
|
// If the NETRESOURCE structure represents a container resource, call the EnumerateFunc function recursively.
|
|
if (RESOURCEUSAGE_CONTAINER == (lpnrLocal[i].dwUsage & RESOURCEUSAGE_CONTAINER))
|
|
if (!EnumerateFunc(hwnd, hdc, &lpnrLocal[i])) {
|
|
//TextOut(hdc, 10, 10, _T("EnumerateFunc returned FALSE."), 29);
|
|
TRACE(_T("EnumerateFunc returned FALSE.\n"));
|
|
}
|
|
}
|
|
}
|
|
// Process errors.
|
|
else if (dwResultEnum != ERROR_NO_MORE_ITEMS) {
|
|
NetErrorHandler(hwnd, dwResultEnum, (LPTSTR)_T("WNetEnumResource"));
|
|
break;
|
|
}
|
|
}
|
|
// End do.
|
|
|
|
while (dwResultEnum != ERROR_NO_MORE_ITEMS);
|
|
|
|
// Call the GlobalFree function to free the memory.
|
|
GlobalFree((HGLOBAL)lpnrLocal);
|
|
|
|
// Call WNetCloseEnum to end the enumeration.
|
|
dwResult = pWNetCloseEnum(hEnum);
|
|
|
|
if (dwResult != NO_ERROR) {
|
|
// Process errors.
|
|
NetErrorHandler(hwnd, dwResult, (LPTSTR)_T("WNetCloseEnum"));
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
/*
|
|
|
|
DWORD WNetConnectionDialog(
|
|
HWND hwnd, // handle to window owning dialog box
|
|
DWORD dwType // resource type
|
|
);
|
|
|
|
|
|
DWORD WNetAddConnection(
|
|
LPCTSTR lpRemoteName, // network device name
|
|
LPCTSTR lpPassword, // password
|
|
LPCTSTR lpLocalName // local device name
|
|
);
|
|
|
|
|
|
DWORD WNetOpenEnum(
|
|
DWORD dwScope, // scope of enumeration
|
|
DWORD dwType, // resource types to list
|
|
DWORD dwUsage, // resource usage to list
|
|
LPNETRESOURCE lpNetResource, // resource structure
|
|
LPHANDLE lphEnum // enumeration handle buffer
|
|
);
|
|
*/
|
|
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void NetworkMonitorThreadProc(void *lpParameter)
|
|
{
|
|
// ULONG OldProcessorUsage = 0;
|
|
// ULONG OldProcessCount = 0;
|
|
HWND hWnd = (HWND)lpParameter;
|
|
|
|
// Create the event
|
|
hNetworkMonitorThreadEvent = CreateEvent(NULL, TRUE, TRUE, _T("Winfile Network Monitor Event"));
|
|
|
|
// If we couldn't create the event then exit the thread
|
|
if (!hNetworkMonitorThreadEvent)
|
|
return;
|
|
|
|
while (1) {
|
|
DWORD dwWaitVal;
|
|
|
|
// Wait on the event
|
|
dwWaitVal = WaitForSingleObject(hNetworkMonitorThreadEvent, INFINITE);
|
|
|
|
// If the wait failed then the event object must have been
|
|
// closed and the task manager is exiting so exit this thread
|
|
if (dwWaitVal == WAIT_FAILED) {
|
|
// CloseHandle(hNetworkMonitorThreadEvent); // Should we close the event object handle or not ???
|
|
// hNetworkMonitorThreadEvent = NULL; // if we do then check what happens when main thread tries to delete it also....
|
|
return;
|
|
}
|
|
|
|
if (dwWaitVal == WAIT_OBJECT_0) {
|
|
// Reset our event
|
|
ResetEvent(hNetworkMonitorThreadEvent);
|
|
|
|
|
|
if ( EnumerateFunc(hWnd, NULL, NULL) ) {
|
|
|
|
}
|
|
|
|
#if 0
|
|
TCHAR text[260];
|
|
if ((ULONG)SendMessage(hProcessPageListCtrl, LVM_GETITEMCOUNT, 0, 0) != PerfDataGetProcessCount())
|
|
SendMessage(hProcessPageListCtrl, LVM_SETITEMCOUNT, PerfDataGetProcessCount(), /*LVSICF_NOINVALIDATEALL|*/LVSICF_NOSCROLL);
|
|
if (IsWindowVisible(hProcessPage))
|
|
InvalidateRect(hProcessPageListCtrl, NULL, FALSE);
|
|
if (OldProcessorUsage != PerfDataGetProcessorUsage()) {
|
|
OldProcessorUsage = PerfDataGetProcessorUsage();
|
|
wsprintf(text, _T("CPU Usage: %3d%%"), OldProcessorUsage);
|
|
SendMessage(hStatusWnd, SB_SETTEXT, 1, (LPARAM)text);
|
|
}
|
|
if (OldProcessCount != PerfDataGetProcessCount()) {
|
|
OldProcessCount = PerfDataGetProcessCount();
|
|
wsprintf(text, _T("Processes: %d"), OldProcessCount);
|
|
SendMessage(hStatusWnd, SB_SETTEXT, 0, (LPARAM)text);
|
|
}
|
|
#endif
|
|
}
|
|
}
|
|
}
|
|
|
|
BOOL CreateNetworkMonitorThread(HWND hWnd)
|
|
{
|
|
CheckNetworkAvailable();
|
|
_beginthread(NetworkMonitorThreadProc, 0, hWnd);
|
|
return TRUE;
|
|
}
|
|
|
|
void SignalNetworkMonitorEvent(void)
|
|
{
|
|
SetEvent(hNetworkMonitorThreadEvent);
|
|
}
|
|
|
|
BOOL DestryNetworkMonitorThread(void)
|
|
{
|
|
CloseHandle(hNetworkMonitorThreadEvent);
|
|
hNetworkMonitorThreadEvent = NULL;
|
|
return TRUE;
|
|
}
|
|
|