Telnet服务器

开发平台：
Visual C++

DirTree.cpp：源码内容
							// DirTreeCtrl.cpp : implementation file
//
#include "stdafx.h"
#include "shlobj.h"
#include "DirTree.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CDirTree
//
// By Nicola Delfino 1999 delfinon@altavista.net
//
// 
// THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND. 
// THE AUTHOR DISCLAIMS ALL WARRANTIES, EITHER EXPRESS OR IMPLIED, 
// INCLUDING THE WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A 
// PARTICULAR PURPOSE. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR 
// ANY DAMAGES WHATSOEVER INCLUDING DIRECT, INDIRECT, INCIDENTAL, 
// CONSEQUENTIAL, LOSS OF BUSINESS PROFITS OR SPECIAL DAMAGES, 
// 
////////////////////////////////////////////////////////////////////////////
CDirTree::CDirTree()
{
}
CDirTree::~CDirTree()
{
}
BEGIN_MESSAGE_MAP(CDirTree, CTreeCtrl)
	//{{AFX_MSG_MAP(CDirTree)
	ON_NOTIFY_REFLECT(TVN_ITEMEXPANDED, OnItemexpanded)
	ON_WM_CREATE()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CDirTree message handlers
void CDirTree::Initialize(bool clear)
{
	LPITEMIDLIST itemlist;
	char buf[300];
	int desktopIcon;
	int mycomputerIcon;
    int ret;
    if (clear)
	    DeleteAllItems();
	SetImageList (NULL,0);
	SHGetSpecialFolderLocation (this->m_hWnd, CSIDL_DESKTOP, &itemlist);
	SHGetPathFromIDList(itemlist, buf ); 
	m_desktop_path=buf;
	m_img.Create( GetSystemMetrics(SM_CXSMICON), 
		          GetSystemMetrics(SM_CYSMICON), 
				  ILC_COLOR24, 50, 50);
	m_img.SetBkColor( GetSysColor(COLOR_WINDOW) );
    CBitmap bmp;
    CBitmap *old;
    CBitmap *newbmp;
    CDC dc;
    dc.CreateCompatibleDC(NULL);
    bmp.CreateBitmap (32,32,1,24,NULL);
    old=dc.SelectObject (&bmp);
    dc.FillSolidRect(0,0,31,31,GetSysColor(COLOR_WINDOW));
    newbmp=dc.SelectObject (old);
    dc.DeleteDC ();
    ret=m_img.Add(newbmp,COLORREF(0x00));
	desktopIcon=    m_img.Add(ExtractIcon( AfxGetApp()->m_hInstance, "shell32.dll", 34));
	mycomputerIcon= m_img.Add(ExtractIcon( AfxGetApp()->m_hInstance, "shell32.dll", 15));
	m_drives_root=AddItem_Core("Drives", TVI_ROOT, true, mycomputerIcon, mycomputerIcon);
//	m_desktop_root=AddItem(m_desktop_path, TVI_ROOT, true, desktopIcon, desktopIcon);
}
HTREEITEM CDirTree::AddItem(const char * path, HTREEITEM parent, bool isDir, int iconEx, int iconEx2)
{
    SHFILEINFO shinfo, shinfo_sel;
	SHGetFileInfo( path, NULL, 
				   &shinfo, 
				   sizeof(shinfo), 
				   SHGFI_DISPLAYNAME | SHGFI_ICON | SHGFI_SMALLICON);
	SHGetFileInfo( path, NULL, 
				   &shinfo_sel, 
				   sizeof(shinfo_sel), 
				   SHGFI_DISPLAYNAME |  SHGFI_ICON | SHGFI_OPENICON | SHGFI_SMALLICON);
	int icon    = iconEx!=-1 ? iconEx : m_img.Add(shinfo.hIcon);
    int iconsel = iconEx2!=-1 ? iconEx2 : m_img.Add(shinfo_sel.hIcon);
	SetImageList (&m_img,LVSIL_NORMAL);
	return AddItem_Core ( shinfo.szDisplayName, parent, isDir, icon, iconsel);
}
void CDirTree::OnItemexpanded(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
	TV_ITEM itm= pNMTreeView->itemNew;
	if (itm.state & TVIS_EXPANDED)
	{
		SetRedraw(FALSE);
		ExpandBranch(itm.hItem);
		SetRedraw(TRUE);
	}
	else
	{
		SetRedraw(FALSE);
		DeleteAllChild(itm.hItem);
		SetRedraw(TRUE);
	}
	
	*pResult = 0;
}
void CDirTree::DeleteAllChild(HTREEITEM itm, bool isEmpty)
{
	HTREEITEM child;
	child= GetChildItem(itm);
	while (child)
	{
		int img1, img2;
		
        if (GetChildItem(child))
            DeleteAllChild(GetChildItem(child), isEmpty);
        GetItemImage(child,img1,img2);
        if (img2!=0 && img2!=img1) 
		    {
            m_img.Remove(img2);
            
            RefreshTreeImages(m_drives_root, img2);
            RefreshTreeImages(m_desktop_root, img2);
            }
		if (img1!= 0)
            {
            m_img.Remove(img1);
            RefreshTreeImages(m_drives_root, img1);
            RefreshTreeImages(m_desktop_root, img1);
            }
		
        DeleteItem (child);
        SetImageList(&m_img,TVSIL_NORMAL);
		child= GetChildItem(itm);
	}
	if (isEmpty)
		InsertItem( "",itm);; 
}
void CDirTree::ExpandBranch(HTREEITEM parent)
{
    bool bOne=false;
	if(parent==m_drives_root)
	{
		char *ptr;
    	DeleteAllChild(parent,false);
		GetLogicalDriveStrings(500,m_drives_name);
		ptr=m_drives_name;
		while (*ptr)
		{
            bOne=true;
			AddItem(ptr, m_drives_root, true);
			ptr+=strlen(ptr)+1;
		}
	}
	else
	{
        CString fullpath,pathWildcard;
        fullpath=GetPathFromHere(fullpath, parent);
        DeleteAllChild(parent,false);
        pathWildcard=fullpath+"*.*";
        CFileFind finder;
        BOOL bWorking = finder.FindFile(pathWildcard);
        while (bWorking)
            {
            bWorking = finder.FindNextFile();
            if ( finder.IsDirectory() && !finder.IsDots() )
                {
                bOne=true;
                HTREEITEM itm=AddItem (fullpath+finder.GetFileName(), parent );
                InsertItem("",itm);
                }
            }
        
        bWorking = finder.FindFile(pathWildcard);
        while (bWorking)
            {
            bWorking = finder.FindNextFile();
            if ( !finder.IsDirectory() && MatchExtension(finder.GetFileName()) )
                {
                bOne=true;
                AddItem (fullpath+finder.GetFileName(), parent );
                }
            }
	}
    
    if (!bOne) InsertItem( "",parent);
}
HTREEITEM CDirTree::AddItem_Core(const char * name, HTREEITEM parent, bool isDir, int icon,int iconOpen)
{
	HTREEITEM hitm;
	hitm=InsertItem( name, icon, iconOpen, parent);
	if (isDir) InsertItem( "",hitm);
	return hitm;
}
CString CDirTree::GetPathFromHere(CString s, HTREEITEM itm)
{
    HTREEITEM parent=GetParentItem(itm);
    HTREEITEM itmChild=GetChildItem(itm);
    if (itm==m_drives_root) return "";
        
	if (parent==m_drives_root)
	{
		HTREEITEM child;
		char *ptr=m_drives_name;
		child=GetChildItem(parent);
		
		while (child)
		{
			if (child==itm)
			{
				s=CString(ptr)+s;
				return s;
			}
			ptr+=strlen(ptr)+1;
			child=GetNextItem(child, TVGN_NEXT);
		}
	}
	else if (itm==m_desktop_root)
	{
		s=m_desktop_path+'\'+s;
		return s;
	}
    else
    {
        if (GetItemText(itm)!= CString(""))
            s=GetItemText(itm)+'\'+s;
        s=GetPathFromHere(s,parent);
    }
    // remove the last '' if it isn't a directory
    if (!itmChild && s.GetLength()>0)
        s=s.Left (s.GetLength()-1);
	return s;
}
/* if image number is more than img number -- */
void CDirTree::RefreshTreeImages(HTREEITEM item,int img)
{
    int iStandard, iOpen;
    HTREEITEM itmCurrent;
       
    itmCurrent=GetChildItem(item);
    
    while (itmCurrent)
        {
        if (GetItemImage(itmCurrent,iStandard,iOpen))
            {
            if (iStandard>img) iStandard--;
            if (iOpen>img) iOpen--;
            SetItemImage(itmCurrent,iStandard, iOpen);
            }
        if (ItemHasChildren( itmCurrent)!=0)
            {
            RefreshTreeImages(itmCurrent, img);
            }
        itmCurrent=GetNextSiblingItem(itmCurrent);
        }
}
bool CDirTree::MatchExtension(CString file)
{
    //return file.Right(4)==CString(".exe");
	return TRUE;
}
CString CDirTree::GetFullPath(HTREEITEM item)
{
    if (GetItemText(item)==CString("")) return "";
    else return GetPathFromHere("",item);
}
bool CDirTree::SetPath(CString path_in)
{
    char path[MAX_PATH];
    char *delimiters="\";
    char *token;
    char *pathStart=path;
    HTREEITEM current=m_drives_root;
    strcpy(path,path_in);
    Expand(current,TVE_EXPAND );
    token=strtok (path,delimiters);
    while (token!=NULL)
        {
        current=Dir2Item(current, token);
        if (!current)
            {
            return false;
            }
        
        Expand(current,TVE_EXPAND );
        token=strtok(NULL,delimiters);
        }
    SelectItem(current);
    return true;
}
HTREEITEM CDirTree::Dir2Item(HTREEITEM current, char *dir)
{
    if (current==m_drives_root)
        {
        HTREEITEM child;
        CString szDir(dir);
        szDir+="\";
        szDir.MakeUpper();
		char *ptr=m_drives_name;
		child=GetChildItem(current);
		
		while (child)
		    {
			if (szDir==CString(ptr))
		    	{
				return child;
			    }
			ptr+=strlen(ptr)+1;
			child=GetNextItem(child, TVGN_NEXT);
		    }
        }
    else
        {
        HTREEITEM child;
        CString szChildText;
        CString szDir(dir);
        szDir.MakeUpper();
		child=GetChildItem(current);
		while (child)
		    {
            szChildText=GetItemText(child);
            szChildText.MakeUpper();
			if ( szChildText==szDir)
		    	{
				return child;
			    }
			child=GetNextItem(child, TVGN_NEXT);
		    }
        }
    return NULL;
}
int CDirTree::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if (CTreeCtrl::OnCreate(lpCreateStruct) == -1)
		return -1;
	
	Initialize();
	
	return 0;
}

						