ftpdlg.cpp

一个电视台专用的信息管理软件源代码

// FtpDlg.cpp : implementation file
//

#include "stdafx.h"
#include "tvnews.h"
#include "FtpDlg.h"

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

/////////////////////////////////////////////////////////////////////////////
// CFtpDlg dialog



/*char szLocDirectory[MAX_PATH];
char szFtpDirectory[MAX_PATH];



char szAppName[256];
char szFtpName[256];
char szFtpUser[20];
char szFtpPassword[20];
*/

CFtpDlg::CFtpDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CFtpDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CFtpDlg)
	m_Server = _T("");
	m_PWD = _T("");
	m_PASV = FALSE;
	m_Port = 0;
	m_UID = _T("");
	m_Message = _T("");
	//}}AFX_DATA_INIT
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
	m_bIsLogon = false;
	m_filename = "";
	m_filepath = "";
}


void CFtpDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CFtpDlg)
	DDX_Control(pDX, IDC_COMBOBOXEX_LOCAL, m_ctrlLocal);
	DDX_Control(pDX, IDC_COMBOBOXEX_SERVER, m_ctrlServer);
	DDX_Control(pDX, IDC_LIST_SERVERFILE, m_ctrlServerFile);
	DDX_Control(pDX, IDC_LIST_LOCALFILE, m_ctrlLocalFile);
	DDX_Text(pDX, IDC_EDIT_SERVER, m_Server);
	DDX_Text(pDX, IDC_EDIT_PWD, m_PWD);
	DDX_Check(pDX, IDC_CHECK_PASV, m_PASV);
	DDX_Text(pDX, IDC_EDIT_PORT, m_Port);
	DDX_Text(pDX, IDC_EDIT_UID, m_UID);
	DDX_Text(pDX, IDC_EDIT_MESSAGE, m_Message);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CFtpDlg, CDialog)
	//{{AFX_MSG_MAP(CFtpDlg)
	ON_BN_CLICKED(IDC_BUTTON_CONNECT, OnButtonConnect)
	ON_NOTIFY(NM_DBLCLK, IDC_LIST_LOCALFILE, OnDblclkListLocalfile)
	ON_NOTIFY(NM_DBLCLK, IDC_LIST_SERVERFILE, OnDblclkListServerfile)
	ON_CBN_SELCHANGE(IDC_COMBOBOXEX_LOCAL, OnSelchangeComboboxexLocal)
    //}}AFX_MSG_MAP

	ON_MESSAGE(WM_SETFILE,CFtpDlg::OnInsertLocFile)
	ON_MESSAGE(WM_SETREDRAWFLAG,OnSetLocRedrawFlag)
	ON_MESSAGE(WM_SETDIRECTORYDISPLAY,OnSetLocDirectoryDisplay)
	ON_MESSAGE(WM_REDISPLAYFILE,OnReDisplayLocFile)
	ON_MESSAGE(WM_SETDIR,OnSetLocDir)
	
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CFtpDlg message handlers

BOOL CFtpDlg::OnInitDialog()
{
	CDialog::OnInitDialog();
	
	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon

	//创建ImageList
    HIMAGELIST hImageList;
	//定义文件信息变量
	SHFILEINFO shFi;
	//获取文件图标信息
	hImageList = (HIMAGELIST)SHGetFileInfo("D:\\",0,&shFi,sizeof(shFi),
		SHGFI_SYSICONINDEX|SHGFI_SMALLICON);
	//将文件图象列表存入m_ctImageList
	m_ctImageList.m_hImageList = hImageList;
	

	// TODO: Add extra initialization here
	//设定客户端文件显示列表框的栏标题
	SetFileColumns(&m_ctrlLocalFile);
	SetServerFileColumns(&m_ctrlServerFile);

	SetImageList(&m_ctrlLocalFile,&m_ctrlLocal);
	SetImageList(&m_ctrlServerFile,NULL);

	m_ctrlLocalFile.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES);



	//m_ctrlLocalFile.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES);
	//获取本地驱动器信息 

	GetLocDir();

	int i=m_ctrlLocal.FindStringExact(-1,"D:\\");

	m_ctrlLocal.SetCurSel(i);
	::SetCurrentDirectory("D:\\");
	GetLocCurrentDirectory();
	BeginFindLocFile();	
	//初始化，为了调试，设定初试值
	char host[100];
	gethostname(host,100);
	hostent * ht=gethostbyname(host);
//	char * temp=ht->h_addr_list[0];
//	struct   in_addr in;  
//	memcpy((void*)&in,temp,sizeof(in));
//	m_Server=inet_ntoa(in);
	m_Server="192.168.135.204";
	m_Port=21;
	m_UID="111";
	m_PWD="111";
	UpdateData(false);


	return TRUE;  // 
}


void CFtpDlg::OnButtonConnect() 
{
	int ndx=0;	
	CString str;
	m_ctrlServerFile.DeleteAllItems ();	
	//将控件中的数据更新到对应的变量中
	UpdateData(TRUE);
	m_Message=m_Message+"连接服务器:"+m_Server+"\r\n";
	m_Message=m_Message+"登录用户名:"+m_UID+"\r\n";
	UpdateData(false);

	m_bIsLogon = ftp.LogOnToServer(m_Server,m_Port,m_UID,m_PWD,"","","","",1080,0);
	if(m_bIsLogon == false) 
	{
		m_Message=m_Message+m_UID+"登录失败。"+"\r\n";
		UpdateData(false);

		return;
	}
	else
	{
		ftp.List ();
		while(ftp.GetLine(ndx))
		{
			AddLine(ftp.m_strLine );
			str="/";
//			strcpy(szFtpDirectory,str);
		    SetServerDir(str);
		}

		ftp.FTPcommand(""); 
	}
	
}

void CFtpDlg::SetFileColumns(CListCtrl *pListCtrl)
{
   	static struct
	{
		LPSTR pszText;
		int uiFormat;
	}columns[]={
		_T("文件名"),LVCFMT_LEFT,
		_T("日期"),LVCFMT_CENTER,
		_T("文件大小"),LVCFMT_CENTER
	};
	for(int i=0,j=120;i<sizeof(columns)/sizeof(columns[0]);i++,j-=8)
		pListCtrl->InsertColumn(i,columns[i].pszText,columns[i].uiFormat,j);
}

void CFtpDlg::SetServerFileColumns(CListCtrl *pListCtrl)
{
   	static struct
	{
		LPSTR pszText;
		int uiFormat;
	}columns[]={
		_T("文件名"),LVCFMT_LEFT,
		_T("文件大小"),LVCFMT_CENTER,
		_T("日期"),LVCFMT_CENTER,
		_T("权限"),LVCFMT_CENTER
	};
//	for(int i=0,j=120;i<sizeof(columns)/sizeof(columns[0]);i++,j-=10)
//		pListCtrl->InsertColumn(i,columns[i].pszText,columns[i].uiFormat,j);
	pListCtrl->InsertColumn(0,columns[0].pszText,columns[0].uiFormat,120);
	pListCtrl->InsertColumn(1,columns[1].pszText,columns[1].uiFormat,80);
	pListCtrl->InsertColumn(2,columns[2].pszText,columns[2].uiFormat,80);
	pListCtrl->InsertColumn(3,columns[3].pszText,columns[3].uiFormat,75);
}

BOOL CFtpDlg::SetImageList(CListCtrl* pListCtrl,CComboBoxEx* pCombo)
{
	pListCtrl->SetImageList(&m_ctImageList,LVSIL_SMALL);
	if(!pCombo)
		return TRUE;
	pCombo->SetImageList(&m_ctImageList);
	return TRUE;
}

void CFtpDlg::GetLocDir()
{
	int iIcon,iIconSel;
	char lpBuffer[MAX_PATH];
	char* ptr;
	CString str;
	m_iDirCount=0;
	::GetLogicalDriveStrings(MAX_PATH,lpBuffer);
	ptr=lpBuffer;
	while(*ptr)
	{
		m_iDirCount++;
		str=ptr;
		GetFileIcon(str,&iIcon,&iIconSel);
		InsertLocDir(str,iIcon,iIconSel);
		ptr+=strlen(ptr)+1;
	}
}


void CFtpDlg::GetLocCurrentDirectory()
{
    CString str;
	m_ctrlLocal.GetWindowText(str);
//	strcpy(szLocDirectory,str);
}

BOOL CFtpDlg::BeginFindLocFile()
{
	m_ctrlLocalFile.DeleteAllItems();

	FindLocFile();
		return TRUE;
}


void CFtpDlg::GetFileIcon(CString& fileName,int* iIcon,int* iIconSel)
{
	CString str=fileName;
	if(str.Right(1)!="\\")
		str+="\\";
	SHFILEINFO shFi;
	if(!SHGetFileInfo(str,0,&shFi,sizeof(shFi),SHGFI_ICON|SHGFI_SMALLICON|SHGFI_DISPLAYNAME))
	{
		AfxMessageBox("111");
		return;
	}
	*iIcon=shFi.iIcon;
	DestroyIcon(shFi.hIcon);
	if(iIconSel)
	{
		if(!SHGetFileInfo(str,0,&shFi,sizeof(shFi),SHGFI_ICON|SHGFI_SMALLICON|SHGFI_OPENICON|SHGFI_DISPLAYNAME))
		{
			AfxMessageBox("222");
			return;
		}
		*iIconSel=shFi.iIcon;
		DestroyIcon(shFi.hIcon);
	}
	return ;
}


int CFtpDlg::InsertLocDir(CString &str, int iIcon, int iIconSel)
{
   int nItem=m_ctrlLocal.GetCount();
	COMBOBOXEXITEM Combo;
	Combo.mask=CBEIF_IMAGE|CBEIF_TEXT|CBEIF_SELECTEDIMAGE;
	Combo.iImage=iIcon;
	Combo.iSelectedImage=iIconSel;
	Combo.pszText=(LPTSTR)(LPCTSTR)str;
	Combo.iItem=nItem;
	m_ctrlLocal.InsertItem(&Combo);
	return nItem;
}

void  CFtpDlg::FindLocFile()
{
	FILE_FTP_INFO* pInfo=new FILE_FTP_INFO;
	CFileFind fileFind;
	UINT i=0;
	CTime time;
	BOOL bContinue;
//	::SetCurrentDirectory(szLocDirectory);
	::PostMessage(this->m_hWnd ,WM_SETDIRECTORYDISPLAY,0,0);
	bContinue=fileFind.FindFile("*.*");
	CString str;
	while(bContinue)
	{
		bContinue=fileFind.FindNextFile();

		if(fileFind.IsHidden())
			continue;

		str = fileFind.GetFileName();
		strcpy(pInfo->szFileName,str);
		fileFind.GetLastWriteTime(time);
		str=time.Format("%x");
		strcpy(pInfo->szFileDate,str);

		if(fileFind.IsDirectory())
		{
			strcpy(pInfo->szFileSize,"");
			pInfo->nType=DIRECTORYICON;
			::SendMessage(this->m_hWnd,WM_SETFILE,(WPARAM)pInfo,
				(LPARAM)LOCFILE);
		}
		else
		{
			long i=fileFind.GetLength();
			if(i>1024*1024)
			{
				str.Format("%d",i/1048576);
				str+="MB";
			}
			else if(i>1024)
			{
				str.Format("%d",i/1024);
				str+="KB";
			}
			else
			{
				str.Format("%d",i);
				str+="B";
			}

			strcpy(pInfo->szFileSize,str);
			pInfo->nType=FILEICON;
			::SendMessage(this->m_hWnd,WM_SETFILE,(WPARAM)pInfo,
				(LPARAM)LOCFILE);
		}
	}
	::SendMessage(this->m_hWnd,WM_SETREDRAWFLAG,0,0);
	delete pInfo;
	fileFind.Close();
	//return 0L;
}


void CFtpDlg::AddLine(CString strLine)
{
	CListItem* pItem = new CListItem();
	if(!pItem->ParseLine(strLine))
	{
		delete pItem;
		return;
	}

	AddItem(pItem);
	delete pItem;
}

void CFtpDlg::SetServerDir (CString& str)
{
	m_ctrlServer.SetWindowText(str);
	m_ctrlServer.Invalidate();
}


BOOL CFtpDlg::AddItem(CListItem *pItem)
{
	//CListCtrl& list= GetListCtrl();
	// look for a match on the file name....
//	LVITEM lvfi;
	LV_ITEM lvi;
	lvi.lParam = (LPARAM)pItem;
	lvi.iImage = pItem->GetImage();
	lvi.mask = LVIF_TEXT|LVIF_IMAGE|LVIF_PARAM;

	int ndx = m_ctrlServerFile.GetItemCount ();
	lvi.pszText = pItem->m_strName.GetBuffer(0);
	pItem->m_strName.ReleaseBuffer();
	lvi.iItem = ndx;
	lvi.iSubItem = 0;

	ndx=m_ctrlServerFile.InsertItem( &lvi );
	lvi.mask =LVIF_TEXT;
	lvi.pszText = pItem->m_strDate.GetBuffer(0);
	pItem->m_strDate.ReleaseBuffer();
	lvi.iItem = ndx;
	lvi.iSubItem = 1;
	m_ctrlServerFile.SetItem( &lvi );