umibdlg.cpp

最适合在学校成批地建立删除学生用户或学生文件夹

// UMIBDlg.cpp : implementation file
//

#include "stdafx.h"
#include "UMIB.h"
#include "UMIBDlg.h"
#include "UsersOperatingDlg.h"
#include "CreateDirsDlg.h"

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

char g_szCurrentUser[32]="";
char g_szDomain[32]="";
char g_szPrefix[4]="C";
char g_szYear[5]="00";
char g_szMainDir[32]="";
UINT g_nClassFrom=1;
UINT g_nClassTo=13;
UINT g_nNumFrom=1;
UINT g_nNumTo=70;
UINT g_nOperation=1;
UINT g_nPasOption=1;
BOOL g_bIsContinue=FALSE;
HWND g_hShowWnd=NULL;
HWND g_hExitWnd=NULL;
BOOL GetCurrentUserAndDomain(PTSTR szUser, PDWORD pcchUser, PTSTR szDomain, PDWORD pcchDomain) ;
/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

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

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

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

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

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CUMIBDlg dialog

CUMIBDlg::CUMIBDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CUMIBDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CUMIBDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CUMIBDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CUMIBDlg)
		// NOTE: the ClassWizard will add DDX and DDV calls here
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CUMIBDlg, CDialog)
	//{{AFX_MSG_MAP(CUMIBDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_ADD_USERS, OnAddUsers)
	ON_BN_CLICKED(IDC_CREATE_DIRS, OnCreateDirs)
	ON_BN_CLICKED(IDC_CHANGE_PAS, OnChangePas)
	ON_BN_CLICKED(IDC_DELETE_USERS, OnDeleteUsers)
	ON_BN_CLICKED(IDC_EXPIRE_ALL_USERS_PAS, OnExpireAllUsersPas)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CUMIBDlg message handlers

BOOL CUMIBDlg::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
	
	// TODO: Add extra initialization here
	((CButton*)GetDlgItem(IDC_ADD_USERS))->SetCheck(1);
	unsigned long nUserLen=32;
	unsigned long nDomainLen=32;
	if(!GetCurrentUserAndDomain(g_szCurrentUser,&nUserLen,g_szDomain,&nDomainLen))
	{
		AfxMessageBox("不能取得当前的域名和用户名!");
		return FALSE;
		DestroyWindow();
	}
	return TRUE;  // return TRUE  unless you set the focus to a control
}

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

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CUMIBDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		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;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CUMIBDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CUMIBDlg::OnAddUsers() 
{
	g_nOperation=1;
}

void CUMIBDlg::OnCreateDirs() 
{
	g_nOperation=2;	
}

void CUMIBDlg::OnChangePas() 
{
	g_nOperation=3;	
}

void CUMIBDlg::OnExpireAllUsersPas() 
{
	g_nOperation=4;	
}

void CUMIBDlg::OnDeleteUsers() 
{
	g_nOperation=5;	
}

void CUMIBDlg::OnOK() 
{
	CUsersOperatingDlg OperatingUsersDlg;
	CCreateDirsDlg CreDirsDlg;
	switch(g_nOperation)
	{
	case 1:
	case 3:
	case 4:
	case 5:
		OperatingUsersDlg.DoModal();
		break;
	case 2:
		CreDirsDlg.DoModal();
		break;
	}
		
}

BOOL GetCurrentUserAndDomain(PTSTR szUser, PDWORD pcchUser, 
      PTSTR szDomain, PDWORD pcchDomain) 
{

   BOOL         fSuccess = FALSE;
   HANDLE       hToken   = NULL;
   PTOKEN_USER  ptiUser  = NULL;
   DWORD        cbti     = 0;
   SID_NAME_USE snu;

   __try 
   {

      // Get the calling thread's access token.
      if (!OpenThreadToken(GetCurrentThread(), TOKEN_QUERY, TRUE,
            &hToken)) 
	  {

         if (GetLastError() != ERROR_NO_TOKEN)
            __leave;

         // Retry against process token if no thread token exists.
         if (!OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, 
               &hToken))
            __leave;
      }

      // Obtain the size of the user information in the token.
      if (GetTokenInformation(hToken, TokenUser, NULL, 0, &cbti)) 
	  {

         // Call should have failed due to zero-length buffer.
         __leave;
   
      } 
	  else 
	  {

         // Call should have failed due to zero-length buffer.
         if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
            __leave;
      }

      // Allocate buffer for user information in the token.
      ptiUser = (PTOKEN_USER) HeapAlloc(GetProcessHeap(), 0, cbti);
      if (!ptiUser)
         __leave;

      // Retrieve the user information from the token.
      if (!GetTokenInformation(hToken, TokenUser, ptiUser, cbti, &cbti))
         __leave;

      // Retrieve user name and domain name based on user's SID.
      if (!LookupAccountSid(NULL, ptiUser->User.Sid, szUser, pcchUser, 
            szDomain, pcchDomain, &snu))
         __leave;
      
      fSuccess = TRUE;

   } 
   __finally 
   {

      // Free resources.
      if (hToken)
         CloseHandle(hToken);

      if (ptiUser)
         HeapFree(GetProcessHeap(), 0, ptiUser);
   }

   return fSuccess;
}