repos_ecs/LANShare/LANShareTest/LANShareTestDlg.cpp

// LANShareTestDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "LANShareTest.h"
#include "LANShareTestDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif


// 用于应用程序“关于”菜单项的 CAboutDlg 对话框

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// 对话框数据
	enum { IDD = IDD_ABOUTBOX };

	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()


// CLANShareTestDlg 对话框




CLANShareTestDlg::CLANShareTestDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CLANShareTestDlg::IDD, pParent)
{
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CLANShareTestDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CLANShareTestDlg, CDialog)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(BTN_OPENDIR, &CLANShareTestDlg::OnBnClickedOpendir)
	ON_BN_CLICKED(IDC_BUTTON2, &CLANShareTestDlg::OnBnClickedButton2)
	ON_BN_CLICKED(IDC_BUTTON4, &CLANShareTestDlg::OnBnClickedButton4)
	ON_BN_CLICKED(IDC_BUTTON3, &CLANShareTestDlg::OnBnClickedButton3)
END_MESSAGE_MAP()


// CLANShareTestDlg 消息处理程序

BOOL CLANShareTestDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// 将“关于...”菜单项添加到系统菜单中。

	// IDM_ABOUTBOX 必须在系统命令范围内。
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

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

	// 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
	//  执行此操作
	SetIcon(m_hIcon, TRUE);			// 设置大图标
	SetIcon(m_hIcon, FALSE);		// 设置小图标

	// TODO: 在此添加额外的初始化代码

	return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

void CLANShareTestDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CLANShareTestDlg::OnPaint()
{
	if (IsIconic())
	{
		CPaintDC dc(this); // 用于绘制的设备上下文

		SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

		// 使图标在工作区矩形中居中
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// 绘制图标
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CLANShareTestDlg::OnQueryDragIcon()
{
	return static_cast<HCURSOR>(m_hIcon);
}

static INT CALLBACK BrowseCallbackProc(HWND hwnd, UINT uMsg, LPARAM lp, LPARAM pData)
{
	switch (uMsg)
	{
	case BFFM_INITIALIZED:
		SendMessage(hwnd, BFFM_SETSELECTION, TRUE, NULL);
		break;
	case   BFFM_SELCHANGED:
		{
			TCHAR szFileName[MAX_PATH];
			LPITEMIDLIST pidlCurrent = (LPITEMIDLIST)lp;
			SHGetPathFromIDList(pidlCurrent, szFileName);
			SendMessage(hwnd, BFFM_SETSTATUSTEXT, 0, (LPARAM)szFileName);
		}
		break;
		// 这个不能让默认目录为桌面;
		// 	case BFFM_INITIALIZED:
		// 		{
		// 			// WParam is TRUE since you are passing a path.
		// 			// It would be FALSE if you were passing a pidl.
		// 			TCHAR szDir[MAX_PATH] = { 0 };
		// 			GetCurrentDirectory(sizeof(szDir) / sizeof(TCHAR), szDir);
		// 			SendMessage(hwnd, BFFM_SETSELECTION, TRUE, (LPARAM)szDir);
		// 		}
		// 		break;
	}
	return 0;
}

BOOL GetFilePath(OUT TCHAR *FilePath, IN HWND hParentWnd, IN TCHAR* InstructionString, IN UINT ulFlags)
{
	BROWSEINFO bInfo;
	LPITEMIDLIST pidl, pidlDesktop;
	ZeroMemory((PVOID)&bInfo, sizeof(BROWSEINFO));

	HRESULT hr = SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOP, &pidlDesktop);
	if ( hr != S_OK )
	{
		return FALSE;
	}

	bInfo.pidlRoot = pidlDesktop;
	bInfo.hwndOwner = hParentWnd;
	bInfo.pszDisplayName = FilePath;
	bInfo.lpszTitle = InstructionString;
	bInfo.ulFlags = ulFlags;
	bInfo.lpfn = BrowseCallbackProc;
	bInfo.lParam = 0;

	if ((pidl = ::SHBrowseForFolder(&bInfo)) == NULL)    
	{
		return FALSE;
	}

	if (::SHGetPathFromIDList(pidl, FilePath) == FALSE)   
	{
		return FALSE;
	}

	return TRUE;
}

void CLANShareTestDlg::OnBnClickedOpendir()
{
	TCHAR szImgSourceDirectory[MAX_PATH] = {0};
	if ( GetFilePath(szImgSourceDirectory, ::GetActiveWindow(), _T("请选择源相片的文件夹！"), BIF_USENEWUI | BIF_RETURNFSANCESTORS | BIF_RETURNONLYFSDIRS))
	{
		SetDlgItemText(ET_DIR, szImgSourceDirectory);
	}
}

void CLANShareTestDlg::OnBnClickedButton2()
{
	CString strDir;
	CString strShareName;

	GetDlgItemText(ET_DIR, strDir);
	GetDlgItemText(ET_SHARENAME, strShareName);
	if ( API_IsTheDirectoryShared(strDir, strShareName.GetString(), MAX_PATH) )
	{
		AfxMessageBox(_T("目录已共享"));
		return ;
	}

	if ( !LanShareDll::API_AddShareDirectory(strDir, strShareName) )
	{
		AfxMessageBox(_T("设置共享失败"));
		return ;
	}
	else
		AfxMessageBox(_T("设置共享成功"));

}

void CLANShareTestDlg::OnBnClickedButton4()
{
	CString strDir;
	CString strShareName;

	GetDlgItemText(ET_DIR, strDir);
	GetDlgItemText(ET_SHARENAME, strShareName);

	if ( !LanShareDll::API_SetNetShareSecurity(strShareName, _T("Everyone")))
	{
		AfxMessageBox(_T("设置共享权限失败"));
		return ;
	}else
		AfxMessageBox(_T("设置共享权限成功"));

	if ( !LanShareDll::API_AddAccessRights(strDir, _T("Everyone")))
	{
		AfxMessageBox(_T("设置安全权限失败"));
		return ;
	}else
		AfxMessageBox(_T("设置安全权限成功"));
}

void CLANShareTestDlg::OnBnClickedButton3()
{
	CString strDir;
	CString strShareName;

	GetDlgItemText(ET_DIR, strDir);
	GetDlgItemText(ET_SHARENAME, strShareName);
	if ( !LanShareDll::API_DelShareDirectory(strShareName))
	{
		AfxMessageBox(_T("删除共享失败"));
		return ;
	}else
		AfxMessageBox(_T("删除共享成功"));
}