滴水逆向联盟

标题: vC++实现遍历桌面和快速启动里的所有快捷方式，判断快捷方式是不是浏览器，如果是... [打印本页]

---

作者: 大灰狼    时间: 2014-10-16 08:50
标题: vC++实现遍历桌面和快速启动里的所有快捷方式，判断快捷方式是不是浏览器，如果是...

• #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)==0)     
•                     {     
•                         m_pIShellLink->SetArguments(L"");     
•                         m_pIPersistFile->Save(T2COLE(linkName),TRUE);     
•                     }     
•                 }     
•                 else     
•                 {     
•                     if(temp.Find(Brower)!=-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;     
• }   
  

---

欢迎光临 滴水逆向联盟 (http://www.dtdebug.com/)

Powered by Discuz! X3.2