mainfrm.cpp

visual c++ 实例编程

// MainFrm.cpp : implementation of the CMainFrame class
//

#include "stdafx.h"
#include "DiskUtil.h"
#include "Rename.h"

#include "MainFrm.h"
#include "LeftView.h"
#include "DiskUtilView.h"
#include "General.h"
#include <afxcmn.h>
#include "connect.h"
#include <winsock2.h>
#include <Nspapi.h>
#include <svcguid.h>


#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

HLOCAL hBufferTmp;
char *BufferTmp;
BOOL Read=FALSE;
LPBYTE BufferNet;
DWORD BytesToFill, NumFloppiesNet, NumHardsNet, Int13ExtNet;
DWORD NetEvent;
SOCKET SocketNum;
BOOL BufferRead=FALSE;
extern void CallNet(void);
extern HANDLE hVxD;
extern HMODULE hLib;
extern HLOCAL hMemDrivePacket, hMemPackets, hMemBuffer, hMemBufferFatEntries;
extern DRIVEPACKET gDrivePacket, ScannedDrive, ParentDrivePacket, *DrivePacket;
extern int gSectorsPerCluster, gCurrentIndex;
extern LPDWORD BufferFatEntries;
extern CListCtrl *ListView;
extern CYLHEADSECT CylHeadSect;
extern BOOL LoadSectors(DRIVEPACKET *DrivePacket, CYLHEADSECT *CylHeadSect, LPBYTE Buffer);
extern BOOL WriteSectors(DRIVEPACKET *DrivePacket, CYLHEADSECT *CylHeadSect, LPBYTE Buffer);
extern DWORD gNumFatEntries;
extern CTreeCtrl *DiskTree;
extern void EnumTreeRoot(void);
extern BOOL Refreshing;
extern HTREEITEM hPrevHD;
extern LPBYTE Buffer;
extern BYTE gListViewType;
extern DLLFUNC *DllThunk32Write;
extern DWORD gGotoSector;
extern HANDLE hDisk[];
extern BOOL gOSWin95;
extern DWORD NumFloppies, NumHards;
extern DWORD CountFatEntries(DWORD StartCluster, WORD FatType, LPBYTE Buffer);
extern void StoreFatEntries(DWORD StartCluster, WORD FatType, LPDWORD FatEntries, DWORD NumFatEntries, LPBYTE Buffer);
extern void OccupyFATCluster(DWORD NextCluster, DWORD CurrentFreeCluster, WORD FatType, LPBYTE pFat);
extern DWORD FindAFreeFATEntry(WORD FatType, LPBYTE pFat);
extern DWORD GetNextCluster(DWORD CurrentCluster, WORD FatType, LPBYTE Buffer);
extern BOOL WriteDirBuffer(LPDWORD FatBuffer, DRIVEPACKET *DrivePacket, DWORD NumFatEntries, LPBYTE Buffer);
extern BOOL ReadDirBuffer(LPDWORD FatBuffer, DRIVEPACKET *DrivePacket, DWORD NumFatEntries, LPBYTE Buffer);

extern CYLHEADSECT CylHeadSect;
extern DLLFUNC *DllThunk32;
extern bool ItsFile;
extern TV_INSERTSTRUCT TvIns;
extern TV_ITEM TvI;

CStatusBar  m_wndStatusBar;
CStatusBarCtrl *m_wndStatusBarCtrl;

BOOL gbPrevSector=FALSE, gbNextSector=FALSE, MenuThere=FALSE;
extern HMENU Menu;
char RenFileName[20];
char RenFileExt[10], NetMachine[50];
extern SOCKET SocketNum;
#define MW_CONNECTED (WM_APP+101)

/////////////////////////////////////////////////////////////////////////////
// CMainFrame

IMPLEMENT_DYNCREATE(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
	//{{AFX_MSG_MAP(CMainFrame)
	ON_WM_CREATE()
	ON_WM_CLOSE()
	ON_COMMAND(ID_REFRESH, OnRefresh)
	ON_UPDATE_COMMAND_UI(ID_REFRESH, OnUpdateRefresh)
	ON_COMMAND(ID_UNDELETE, OnUndelete)
	ON_COMMAND(ID_RENAME, OnRename)
	ON_COMMAND(ID_CONNECT, OnConnect)
	//}}AFX_MSG_MAP
	ON_UPDATE_COMMAND_UI_RANGE(AFX_ID_VIEW_MINIMUM, AFX_ID_VIEW_MAXIMUM, OnUpdateViewStyles)
	ON_COMMAND_RANGE(AFX_ID_VIEW_MINIMUM, AFX_ID_VIEW_MAXIMUM, OnViewStyle)
END_MESSAGE_MAP()

static UINT indicators[] =
{
	ID_SEPARATOR,           // status line indicator
//	ID_INDICATOR_CAPS,
//	ID_INDICATOR_NUM,
};

/////////////////////////////////////////////////////////////////////////////
// CMainFrame construction/destruction

CMainFrame::CMainFrame()
{
	// TODO: add member initialization code here
	
}

CMainFrame::~CMainFrame()
{
}

int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
	if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	if (!m_wndToolBar.CreateEx(this, TBSTYLE_FLAT, WS_CHILD | WS_VISIBLE | CBRS_TOP
		| CBRS_GRIPPER | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC) ||
		!m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
	{
		TRACE0("Failed to create toolbar\n");
		return -1;      // fail to create
	}

	if (!m_wndStatusBar.Create(this) /*||
		!m_wndStatusBar.SetIndicators(indicators,
		  sizeof(indicators)/sizeof(UINT))*/)
	{
		TRACE0("Failed to create status bar\n");
		return -1;      // fail to create
	}
	CStatusBarCtrl &ab = m_wndStatusBar.GetStatusBarCtrl();
	m_wndStatusBarCtrl = &ab;
	int WidthStat[]={200, -1};
	m_wndStatusBarCtrl->SetParts(2, WidthStat);

	// TODO: Delete these three lines if you don't want the toolbar to
	//  be dockable
	m_wndToolBar.EnableDocking(CBRS_ALIGN_ANY);
	EnableDocking(CBRS_ALIGN_ANY);
	DockControlBar(&m_wndToolBar);

	WORD wVersionRequested;
	WSADATA wsaData;
	int err;
	wVersionRequested = MAKEWORD( 2, 2 );

	err = WSAStartup( wVersionRequested, &wsaData );
	if ( err != 0 ) {
		AfxMessageBox("WSAStartup error");
		return FALSE;
	}
	if ( LOBYTE( wsaData.wVersion ) != 2 ||
			HIBYTE( wsaData.wVersion ) != 2 ) {
		AfxMessageBox("WSAStartup version error");
		WSACleanup( );
		return FALSE; 
	}
	return 0;
}

BOOL CMainFrame::OnCreateClient(LPCREATESTRUCT /*lpcs*/,
	CCreateContext* pContext)
{
	// create splitter window
	if (!m_wndSplitter.CreateStatic(this, 1, 2))
		return FALSE;

	if (!m_wndSplitter.CreateView(0, 0, RUNTIME_CLASS(CLeftView), CSize(100, 100), pContext) ||
		!m_wndSplitter.CreateView(0, 1, RUNTIME_CLASS(CDiskUtilView), CSize(100, 100), pContext))
	{
		m_wndSplitter.DestroyWindow();
		return FALSE;
	}

	return TRUE;
}

BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
	if( !CFrameWnd::PreCreateWindow(cs) )
		return FALSE;
	// TODO: Modify the Window class or styles here by modifying
	//  the CREATESTRUCT cs

	return TRUE;
}

/////////////////////////////////////////////////////////////////////////////
// CMainFrame diagnostics

#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
	CFrameWnd::AssertValid();
}

void CMainFrame::Dump(CDumpContext& dc) const
{
	CFrameWnd::Dump(dc);
}

#endif //_DEBUG

/////////////////////////////////////////////////////////////////////////////
// CMainFrame message handlers

CDiskUtilView* CMainFrame::GetRightPane()
{
	CWnd* pWnd = m_wndSplitter.GetPane(0, 1);
	CDiskUtilView* pView = DYNAMIC_DOWNCAST(CDiskUtilView, pWnd);
	return pView;
}

void CMainFrame::OnUpdateViewStyles(CCmdUI* pCmdUI)
{
	// TODO: customize or extend this code to handle choices on the
	// View menu.

	CDiskUtilView* pView = GetRightPane(); 

	// if the right-hand pane hasn't been created or isn't a view,
	// disable commands in our range

	if (pView == NULL)
		pCmdUI->Enable(FALSE);
	else
	{
		DWORD dwStyle = pView->GetStyle() & LVS_TYPEMASK;

		// if the command is ID_VIEW_LINEUP, only enable command
		// when we're in LVS_ICON or LVS_SMALLICON mode

		if (pCmdUI->m_nID == ID_VIEW_LINEUP)
		{
			if (dwStyle == LVS_ICON || dwStyle == LVS_SMALLICON)
				pCmdUI->Enable();
			else
				pCmdUI->Enable(FALSE);
		}
		else
		{
			// otherwise, use dots to reflect the style of the view
			pCmdUI->Enable();
			BOOL bChecked = FALSE;

			switch (pCmdUI->m_nID)
			{
			case ID_VIEW_DETAILS:
				bChecked = (dwStyle == LVS_REPORT);
				break;

			case ID_VIEW_SMALLICON:
				bChecked = (dwStyle == LVS_SMALLICON);
				break;

			case ID_VIEW_LARGEICON:
				bChecked = (dwStyle == LVS_ICON);
				break;

			case ID_VIEW_LIST:
				bChecked = (dwStyle == LVS_LIST);
				break;

			default:
				bChecked = FALSE;
				break;
			}

			pCmdUI->SetRadio(bChecked ? 1 : 0);
		}
	}
}


void CMainFrame::OnViewStyle(UINT nCommandID)
{
	// TODO: customize or extend this code to handle choices on the
	// View menu.
	CDiskUtilView* pView = GetRightPane();

	// if the right-hand pane has been created and is a CDiskUtilView,
	// process the menu commands...
	if (pView != NULL)
	{
		DWORD dwStyle = -1;

		switch (nCommandID)
		{
		case ID_VIEW_LINEUP:
			{
				// ask the list control to snap to grid
				CListCtrl& refListCtrl = pView->GetListCtrl();
				refListCtrl.Arrange(LVA_SNAPTOGRID);
			}
			break;

		// other commands change the style on the list control
		case ID_VIEW_DETAILS:
			dwStyle = LVS_REPORT;
			break;

		case ID_VIEW_SMALLICON:
			dwStyle = LVS_SMALLICON;
			break;

		case ID_VIEW_LARGEICON:
			dwStyle = LVS_ICON;
			break;

		case ID_VIEW_LIST:
			dwStyle = LVS_LIST;
			break;
		}

		// change the style; window will repaint automatically
		if (dwStyle != -1)
			pView->ModifyStyle(LVS_TYPEMASK, dwStyle);
	}
}

void CMainFrame::OnClose() 
{
	// TODO: Add your message handler code here and/or call default
		if(hMemBuffer){
			LocalUnlock(hMemBuffer);
			LocalFree(hMemBuffer);
		}
	if(hMemDrivePacket != NULL){
		LocalUnlock(hMemDrivePacket);
		LocalFree(hMemDrivePacket);
	}
	if(hMemPackets != NULL){
		LocalUnlock(hMemPackets);
		LocalFree(hMemPackets);
	}
	if(hMemBufferFatEntries){
		LocalUnlock(hMemBufferFatEntries);
		LocalFree(hMemBufferFatEntries);
	}
	if(hLib)
		FreeLibrary(hLib);
	if(!gOSWin95){
		for(int i=0; i<NumFloppies; i++){
			CloseHandle(hDisk[i]);
		}
		for(i=0x80; i<(NumHards+0x80); i++){
			CloseHandle(hDisk[i]);
		}
	}
	if(SocketNum)
		closesocket(SocketNum);
	OnDestroy();
}


void CMainFrame::OnRefresh() 
{
	// TODO: Add your command handler code here
	Refreshing = TRUE;
	LocalUnlock(hMemPackets);
	LocalUnlock(hMemDrivePacket);
	DiskTree->DeleteAllItems();
	EnumTreeRoot();
	Refreshing = FALSE;
	DiskTree->SelectItem(hPrevHD);
}



void CMainFrame::OnUpdateRefresh(CCmdUI* pCmdUI) 
{
	// TODO: Add your command update UI handler code here
	pCmdUI->Enable(MenuThere);	
	
}