Y
yanglihao2006
// autoRunDlg.cpp : implementation file
//
#include "stdafx.h"
#include "autoRun.h"
#include "autoRunDlg.h"
#include "tchar.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// CAboutDlg dialog used for App About
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// Dialog Data
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV
support
// Implementation
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}
void CAboutDlg:oDataExchange(CDataExchange* pDX)
{
CDialog:oDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CautoRunDlg dialog
CautoRunDlg::CautoRunDlg(CWnd* pParent /*=NULL*/)
: CDialog(CautoRunDlg::IDD, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CautoRunDlg:oDataExchange(CDataExchange* pDX)
{
CDialog:oDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CautoRunDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_BN_CLICKED(IDC_BUTTON1, &CautoRunDlg::OnBnClickedButton1)
END_MESSAGE_MAP()
// CautoRunDlg message handlers
BOOL CautoRunDlg::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
return TRUE; // return TRUE unless you set the focus to a control
}
void CautoRunDlg::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 CautoRunDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND,
reinterpret_cast<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 function to obtain the cursor to display
while the user drags
// the minimized window.
HCURSOR CautoRunDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
void CautoRunDlg::OnBnClickedButton1()
{
TCHAR szFilePath[MAX_PATH+1];
GetModuleFileName(NULL,szFilePath,MAX_PATH);
HKEY hKey;
HKEY RootKey=HKEY_CURRENT_USER;
TCHAR *KeyTree =_T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\
\Run");
if(RegOpenKeyEx(RootKey,KeyTree,0,KEY_READ,&hKey)==ERROR_SUCCESS)
{
if(RegSetValueEx(hKey,_T("example"),NULL,REG_SZ,
(byte*)szFilePath,CString(szFilePath).GetLength())!=ERROR_SUCCESS)
{
AfxMessageBox(_T("error.1"));
}
RegCloseKey(hKey);
}
else
{
AfxMessageBox(_T("error.1"));
}
}
//
#include "stdafx.h"
#include "autoRun.h"
#include "autoRunDlg.h"
#include "tchar.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// CAboutDlg dialog used for App About
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// Dialog Data
enum { IDD = IDD_ABOUTBOX };
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV
support
// Implementation
protected:
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
}
void CAboutDlg:oDataExchange(CDataExchange* pDX)
{
CDialog:oDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
END_MESSAGE_MAP()
// CautoRunDlg dialog
CautoRunDlg::CautoRunDlg(CWnd* pParent /*=NULL*/)
: CDialog(CautoRunDlg::IDD, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CautoRunDlg:oDataExchange(CDataExchange* pDX)
{
CDialog:oDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CautoRunDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
ON_BN_CLICKED(IDC_BUTTON1, &CautoRunDlg::OnBnClickedButton1)
END_MESSAGE_MAP()
// CautoRunDlg message handlers
BOOL CautoRunDlg::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
return TRUE; // return TRUE unless you set the focus to a control
}
void CautoRunDlg::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 CautoRunDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND,
reinterpret_cast<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 function to obtain the cursor to display
while the user drags
// the minimized window.
HCURSOR CautoRunDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
void CautoRunDlg::OnBnClickedButton1()
{
TCHAR szFilePath[MAX_PATH+1];
GetModuleFileName(NULL,szFilePath,MAX_PATH);
HKEY hKey;
HKEY RootKey=HKEY_CURRENT_USER;
TCHAR *KeyTree =_T("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\
\Run");
if(RegOpenKeyEx(RootKey,KeyTree,0,KEY_READ,&hKey)==ERROR_SUCCESS)
{
if(RegSetValueEx(hKey,_T("example"),NULL,REG_SZ,
(byte*)szFilePath,CString(szFilePath).GetLength())!=ERROR_SUCCESS)
{
AfxMessageBox(_T("error.1"));
}
RegCloseKey(hKey);
}
else
{
AfxMessageBox(_T("error.1"));
}
}