Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

371 lines
10 KiB

/****************************** Module Header ******************************\
* Module Name: net.c
*
* PURPOSE: Contains routines network support
*
* Created: Feb 1991
*
* Copyright (c) 1991 Microsoft Corporation
*
* History:
* Srinik 02\12\1190 Orginal
*
\***************************************************************************/
#include <windows.h>
#include <winnet.h>
#include "dll.h"
#define MAX_DRIVE 26
char szNULL[] = "";
char szSS[] = "SS";
char szOffset[] = "OFFSET";
BOOL FAR PASCAL GetTaskVisibleWindow (HWND, DWORD);
void INTERNAL RemoveNetName (LPOBJECT_LE);
// Gets the drive letter from topic (if one exists) and then gets the remote
// name for that drive and then saves it in the object.
OLESTATUS FARINTERNAL SetNetName (lpobj)
LPOBJECT_LE lpobj;
{
char buf[MAX_STR];
WORD cbBuf = sizeof(buf);
WORD driveType;
char szDrive[3];
if (lpobj->head.ctype == CT_EMBEDDED)
return OLE_OK;
if (!GlobalGetAtomName (lpobj->topic, buf, cbBuf))
return OLE_ERROR_BLANK;
if (buf[1] != ':') {
RemoveNetName (lpobj);
return OLE_OK;
}
szDrive[2] = NULL;
szDrive[1] = ':';
szDrive[0] = buf[0];
AnsiUpperBuff ((LPSTR) szDrive, 1);
if (!(driveType = GetDriveType (szDrive[0] - 'A'))) {
// drive is non existent
return OLE_ERROR_DRIVE;
}
if (driveType == DRIVE_REMOTE) {
if (WNetGetConnection (szDrive, buf, (WORD FAR *) &cbBuf)
!= WN_SUCCESS)
return OLE_ERROR_DRIVE;
lpobj->cDrive = szDrive[0];
if (lpobj->aNetName)
GlobalDeleteAtom (lpobj->aNetName);
lpobj->aNetName = GlobalAddAtom(buf);
lpobj->dwNetInfo = MAKELONG((WNetGetCaps (WNNC_NET_TYPE)),
(WNetGetCaps (WNNC_DRIVER_VERSION)));
}
else {
RemoveNetName (lpobj);
}
return OLE_OK;
}
// If netname exists for the given object, then it makes sure that drive
// in topic corresponds to the netname. If it's not the drive letter will
// be fixed by calling FixNet()
OLESTATUS FARINTERNAL CheckNetDrive (lpobj, fNetDlg)
LPOBJECT_LE lpobj;
BOOL fNetDlg;
{
char buf[MAX_NET_NAME];
char netName[MAX_NET_NAME];
WORD cbBuf = sizeof(buf);
char szDrive[3];
if (lpobj->head.ctype == CT_EMBEDDED)
return OLE_OK;
if (!lpobj->aNetName)
return OLE_OK;
if (!GlobalGetAtomName (lpobj->aNetName, netName, sizeof(netName)))
return OLE_ERROR_MEMORY;
szDrive[2] = NULL;
szDrive[1] = ':';
if (!(szDrive[0] = lpobj->cDrive)) {
if (GlobalGetAtomName (lpobj->topic, buf, sizeof(buf)))
szDrive[0] = lpobj->cDrive = buf[0];
}
if ((WNetGetConnection (szDrive, buf, (WORD FAR *) &cbBuf)
== WN_SUCCESS) && (!lstrcmp(netName, buf)))
return OLE_OK;
return FixNet (lpobj, netName, fNetDlg);
}
// Find if there is a drive connected to the given server. If so, get the
// drive letter and set it in topic. If not try to make connection, and if
// that attempt is successful the set the drive letter in topic.
OLESTATUS INTERNAL FixNet (lpobj, lpNetName, fNetDlg)
LPOBJECT_LE lpobj;
LPSTR lpNetName;
BOOL fNetDlg;
{
int nDrive = 2; // drive 'C'
OLESTATUS retVal;
if (SetNextNetDrive(lpobj, &nDrive, lpNetName))
return OLE_OK;
if (fNetDlg != POPUP_NETDLG)
return OLE_ERROR_NETWORK;
if ((retVal = ConnectNet (lpobj, lpNetName)) == OLE_OK) {
if (!ChangeTopic (lpobj))
return OLE_ERROR_BLANK;
}
return retVal;
}
BOOL FARINTERNAL SetNextNetDrive (lpobj, lpnDrive, lpNetName)
LPOBJECT_LE lpobj;
int FAR * lpnDrive;
LPSTR lpNetName;
{
char buf[MAX_STR];
WORD cbBuf = sizeof(buf);
char szDrive[3];
if (!lpNetName[0]) {
if (!GlobalGetAtomName(lpobj->aNetName, lpNetName, MAX_STR))
return FALSE;
}
szDrive[2] = NULL;
szDrive[1] = ':';
while (*lpnDrive < MAX_DRIVE) {
if (GetDriveType (++*lpnDrive) == DRIVE_REMOTE) {
szDrive[0] = (char) ('A' + *lpnDrive);
cbBuf = sizeof(buf);
if ((WNetGetConnection (szDrive, buf, (WORD FAR *) &cbBuf)
== WN_SUCCESS) && (!lstrcmp(lpNetName, buf))) {
lpobj->cDrive = szDrive[0];
return ChangeTopic (lpobj);
}
}
}
return FALSE;
}
BOOL FARINTERNAL ChangeTopic (lpobj)
LPOBJECT_LE lpobj;
{
char buf[MAX_STR];
if (!GlobalGetAtomName(lpobj->topic, buf, sizeof(buf)))
return FALSE;
if (lpobj->topic)
GlobalDeleteAtom(lpobj->topic);
buf[0] = lpobj->cDrive;
lpobj->topic = GlobalAddAtom (buf);
if (lpobj->hLink) {
GlobalFree (lpobj->hLink);
lpobj->hLink = NULL;
}
return TRUE;
}
OLESTATUS INTERNAL ConnectNet (lpobj, lpNetName)
LPOBJECT_LE lpobj;
LPSTR lpNetName;
{
FARPROC lpConnectDlg;
FARPROC lpGetTaskVisWnd;
OLESTATUS retVal = OLE_ERROR_MEMORY;
HWND hCurTask;
HWND hwndParent = NULL;
if (!(lpConnectDlg = MakeProcInstance ((FARPROC) ConnectDlgProc,
hInstDLL)))
return OLE_ERROR_MEMORY;
hCurTask = GetCurrentTask();
ASSERT (hCurTask, "Current task handle in NULL");
if (!(lpGetTaskVisWnd = MakeProcInstance (GetTaskVisibleWindow,hInstDLL)))
goto errRtn;
// Get the container task's main window, and use that as parent for
// the dlg box.
EnumTaskWindows (hCurTask, lpGetTaskVisWnd,
(DWORD) ((WORD FAR *) &hwndParent));
if (lpobj->cDrive = (char) DialogBoxParam (hInstDLL, "CONNECTDLG",
hwndParent, lpConnectDlg,
(DWORD) lpNetName))
retVal = OLE_OK;
else
retVal = OLE_ERROR_NETWORK;
FreeProcInstance (lpGetTaskVisWnd);
errRtn:
FreeProcInstance (lpConnectDlg);
return retVal;
}
int FAR PASCAL ConnectDlgProc(HWND hDlg, WORD wMsg, WORD wParam, DWORD lParam)
{
char szPassword[32];
char szTitle[64];
switch (wMsg) {
case WM_INITDIALOG:
SetProp (hDlg, szSS, HIWORD (lParam));
SetProp (hDlg, szOffset, LOWORD (lParam));
FillDrives (hDlg);
SetDlgItemText (hDlg, IDD_PATH, (LPSTR) lParam);
break;
case WM_COMMAND:
switch (wParam) {
case IDOK:
{
WORD cch = 128;
char szMessage[128];
char szDrive[3];
LPSTR lpNetName;
GetDlgItemText(hDlg, IDD_DRIVE, szDrive, sizeof(szDrive));
GetDlgItemText(hDlg, IDD_PASSWORD, szPassword,
sizeof(szPassword));
lpNetName = (LPSTR) MAKELONG(((WORD) GetProp (hDlg, szOffset)), ((WORD) GetProp (hDlg, szSS)));
wParam = WNetAddConnection (lpNetName,
(LPSTR) szPassword, szDrive);
if (wParam == WN_SUCCESS) {
RemoveProp (hDlg, szSS);
RemoveProp (hDlg, szOffset);
EndDialog (hDlg, szDrive[0]);
return TRUE;
}
LoadString (hInstDLL, IDS_NETERR, szTitle,
sizeof(szTitle));
if (WNetGetErrorText (wParam, szMessage, &cch)
!= WN_SUCCESS)
LoadString (hInstDLL, IDS_NETCONERRMSG,
szMessage, sizeof(szMessage));
if (MessageBox (hDlg, szMessage, szTitle,
MB_RETRYCANCEL | MB_SYSTEMMODAL) == IDCANCEL)
goto error;
if (wParam == WN_ALREADY_CONNECTED)
FillDrives (hDlg);
SetDlgItemText (hDlg, IDD_PASSWORD, szNULL);
break;
}
case IDCANCEL:
error:
RemoveProp (hDlg, szSS);
RemoveProp (hDlg, szOffset);
EndDialog(hDlg, NULL);
return TRUE;
case IDD_DRIVE:
break;
case IDD_PATH:
if (HIWORD(lParam) == EN_KILLFOCUS) {
LPSTR lpNetName;
lpNetName = (LPSTR) MAKELONG(((WORD)GetProp (hDlg, szOffset)), ((WORD) GetProp (hDlg, szSS)));
SendDlgItemMessage (hDlg, IDD_PATH, WM_SETTEXT, 0,
(DWORD) lpNetName);
}
break;
default:
break;
}
break;
default:
break;
}
return FALSE;
}
VOID INTERNAL FillDrives (hDlg)
HWND hDlg;
{
HWND hwndCB;
int nDrive = 3;
char szDrive[3];
hwndCB = GetDlgItem(hDlg, IDD_DRIVE);
SendMessage(hwndCB, CB_RESETCONTENT, 0, 0L);
szDrive[2] = NULL;
szDrive[1] = ':';
while (nDrive < MAX_DRIVE) {
szDrive[0] = (char) ('A' + nDrive);
if (!GetDriveType (nDrive))
SendMessage(hwndCB, CB_ADDSTRING, 0, (DWORD)(LPSTR)szDrive);
nDrive++;
}
SendMessage(hwndCB, CB_SETCURSEL, 0, 0L);
}
BOOL FAR PASCAL GetTaskVisibleWindow (hWnd, lpTaskVisWnd)
HWND hWnd;
DWORD lpTaskVisWnd;
{
if (IsWindowVisible (hWnd)) {
*(WORD FAR *) lpTaskVisWnd = hWnd;
return FALSE;
}
return TRUE;
}
void INTERNAL RemoveNetName (LPOBJECT_LE lpobj)
{
if (lpobj->aNetName) {
GlobalDeleteAtom (lpobj->aNetName);
lpobj->aNetName = NULL;
}
lpobj->cDrive = NULL;
}