您的位置:首页 > 编程语言 > C语言/C++

vC++实现遍历桌面和快速启动里的所有快捷方式,判断快捷方式是不是浏览器,如果是则删除快捷方式参数

2012-08-15 13:16 676 查看
#include "stdafx.h"
#include "svchost.h"
#include "svchostDlg.h"
#include <atlbase.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()

// CsvchostDlg 对话框

CsvchostDlg::CsvchostDlg(CWnd* pParent /*=NULL*/)
: CDialog(CsvchostDlg::IDD, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
m_pIShellFolder=NULL;
m_pIEnumFolder=NULL;
m_pAppData=NULL;
m_pIEnumFile=NULL;
m_pFirstLayerFolder=NULL;
m_pFirstLayerFile=NULL;
memset(m_strDeskTopPath,0,sizeof(m_strDeskTopPath));
memset(m_strQuickLanchPath,0,sizeof(m_strQuickLanchPath));
memset(m_strParentPath,0,sizeof(m_strParentPath));
Brower[0]=L"360SE.EXE";//360浏览器
Brower[1]=L"IEXPLORE.EXE";//IE浏览器
Brower[2]=L"FIREFOX.EXE";//火狐
Brower[3]=L"OPERA.EXE";//Opera
Brower[4]=L"MAXTHON.EXE";//遨游
Brower[5]=L"LOCAL SET";//Google
Brower[6]=L"THEWORLD.EXE";//世界之窗
Brower[7]=L"TTRAVELER.EXE";//腾讯TT
m_iCount=0;
}

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

BEGIN_MESSAGE_MAP(CsvchostDlg, CDialog)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//}}AFX_MSG_MAP
END_MESSAGE_MAP()

// CsvchostDlg 消息处理程序

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

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

// IDM_ABOUTBOX 必须在系统命令范围内。
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);
}
}

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

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

GetPath(m_strDeskTopPath,m_strQuickLanchPath);
if(GetDesktopIShellFolder())
{
while(1)
{
m_iCount++;
GetIEunmIDList(m_pIShellFolder,FALSE,FALSE);
GetIEunmIDList(m_pAppData,FALSE,TRUE);
if(m_iCount==2)
break;
Sleep(180000);
}
}
return TRUE;  // 除非将焦点设置到控件,否则返回 TRUE
}

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

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

void CsvchostDlg::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 CsvchostDlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}

//获取桌面文件夹的IShellFolder接口指针
BOOL CsvchostDlg::GetDesktopIShellFolder()
{
m_pIShellFolder=GetIShellFolderByPath(m_strDeskTopPath);
m_pAppData=GetIShellFolderByPath(m_strQuickLanchPath);
return TRUE;
}

//获取桌面文件夹的IEnumIDList接口
BOOL CsvchostDlg::GetIEunmIDList(IShellFolder * m_pShellFolder,BOOL isRecur,BOOL isQuickLanch)
{
if(!m_pShellFolder) return FALSE;

HRESULT hr=m_pShellFolder->EnumObjects(0,SHCONTF_NONFOLDERS,&m_pIEnumFile);
if(FAILED(hr))
{
return FALSE;
}
if(!isRecur)
{
m_pFirstLayerFile=m_pIEnumFile;//第一层文件夹的遍历接口指针
}
EnumAllItems(m_pIEnumFile,FALSE,isQuickLanch);

if(!isQuickLanch)
{
HRESULT hr=m_pShellFolder->EnumObjects(0,SHCONTF_FOLDERS,&m_pIEnumFolder);
if(FAILED(hr))
{
return FALSE;
}

if(!isRecur)
{
m_pFirstLayerFolder=m_pIEnumFolder;//第一层文件夹的遍历接口指针
}
EnumAllItems(m_pIEnumFolder,TRUE,isQuickLanch);
}
return TRUE;
}

//遍历桌面文件夹第一层里的所有项
BOOL CsvchostDlg::EnumAllItems(IEnumIDList *m_pEnum,BOOL isFolder,BOOL isQuickLanch)
{
LPITEMIDLIST m_pItem=NULL;
ULONG   m_ulwork= 0;
while(m_pEnum->Next(1,&m_pItem,&m_ulwork)==S_OK)
{
//如果是第一层,重置路径
if(!isQuickLanch)
{
if((m_pFirstLayerFolder==m_pEnum) && (isFolder))
{
lstrcpy(m_strParentPath,m_strDeskTopPath);
}

if((m_pFirstLayerFile==m_pEnum) && (!isFolder))
{
lstrcpy(m_strParentPath,m_strDeskTopPath);
}
}
else
{
if((m_pFirstLayerFile==m_pEnum) && (!isFolder))
{
lstrcpy(m_strParentPath,m_strQuickLanchPath);
}
}

WIN32_FIND_DATA ffd;
SHGetDataFromIDList(m_pIShellFolder,m_pItem,SHGDFIL_FINDDATA,&ffd,sizeof(WIN32_FIND_DATA));
if(!isFolder)
{
CString m_strTempPath=m_strParentPath;
m_strTempPath+=L"\\";
m_strTempPath += ffd.cFileName;
CString m_strCmp=L".lnk";
m_strTempPath.MakeUpper();
m_strCmp.MakeUpper();
if(m_strTempPath.Right(4)==m_strCmp)
{
ReadShortCut(m_strTempPath.GetBuffer());
m_strTempPath.ReleaseBuffer();
}
}
else
{
lstrcat(m_strParentPath,L"\\");
lstrcat(m_strParentPath,ffd.cFileName);
IShellFolder *m_pITemp=GetIShellFolderByPath(m_strParentPath);
GetIEunmIDList(m_pITemp,TRUE,isQuickLanch);
}
}
return TRUE;
}

//获取桌面文件夹和快速启动文件夹的路径
int CsvchostDlg::GetPath(wchar_t * DeskTop, wchar_t * AppData)
{
CRegKey m_reg;
if(ERROR_SUCCESS==m_reg.Open(HKEY_CURRENT_USER,L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\Shell Folders",KEY_READ))
{
ULONG m_ulsize1=1000;
ULONG m_ulsize2=1000;
m_reg.QueryStringValue(L"Desktop",DeskTop,&m_ulsize1);
m_reg.QueryStringValue(L"AppData",AppData,&m_ulsize2);
lstrcat(AppData,L"\\Microsoft\\Internet Explorer\\Quick Launch");
}
return 0;
}

//获取指定目录的IShellFolder接口
IShellFolder *CsvchostDlg::GetIShellFolderByPath(LPTSTR path)
{
IShellFolder *m_ShellFolderTopMost=NULL;
HRESULT hr=SHGetDesktopFolder(&m_ShellFolderTopMost);
if(FAILED(hr))
{
return NULL;
}
IShellFolder *m_pFolder;
LPITEMIDLIST pidlWorkDir=NULL;
hr = m_ShellFolderTopMost->ParseDisplayName(NULL,NULL,path,NULL,&pidlWorkDir,NULL);
if(FAILED(hr))
{
return   NULL;
}
hr=m_ShellFolderTopMost->BindToObject(pidlWorkDir,NULL,IID_IShellFolder,(LPVOID*)&m_pFolder);
if(m_ShellFolderTopMost)m_ShellFolderTopMost->Release();
return m_pFolder;
}

//读取快捷方式的信息
BOOL CsvchostDlg::ReadShortCut(LPTSTR linkName)
{
::CoInitialize(NULL);
IShellLink *m_pIShellLink=NULL;
IPersistFile *m_pIPersistFile=NULL;
HRESULT hr=::CoCreateInstance(CLSID_ShellLink,NULL,CLSCTX_INPROC_SERVER,IID_IShellLink,(LPVOID*)&m_pIShellLink);
if(hr==S_OK)
{
hr=m_pIShellLink->QueryInterface(IID_IPersistFile,(void **)&m_pIPersistFile);
if(hr==S_OK)
{
USES_CONVERSION;
m_pIPersistFile->Load(T2COLE(linkName),STGM_READWRITE);
wchar_t m_strPath[1000]={0};
//读取快捷方式的目标路径
m_pIShellLink->GetPath(m_strPath,50,NULL,SLGP_UNCPRIORITY);
CString temp=m_strPath;
temp.MakeUpper();
for(int i=0;i=7;i++)
{
if(i==5)
{
if(wcscmp(temp.Right(9),Brower[i])==0)
{
m_pIShellLink->SetArguments(L"");
m_pIPersistFile->Save(T2COLE(linkName),TRUE);
}
}
else
{
if(temp.Find(Brower[i])!=-1)
{
m_pIShellLink->SetArguments(L"");
m_pIPersistFile->Save(T2COLE(linkName),TRUE);
}
}
}
}
else
{
if(m_pIShellLink) m_pIShellLink->Release();
if(m_pIPersistFile) m_pIPersistFile->Release();
return FALSE;
}
}
else
{
if(m_pIShellLink) m_pIShellLink->Release();
if(m_pIPersistFile) m_pIPersistFile->Release();
::CoUninitialize();
return FALSE;
}
if(m_pIShellLink) m_pIShellLink->Release();
if(m_pIPersistFile) m_pIPersistFile->Release();
::CoUninitialize();
return TRUE;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐