核心源码:

// ApkInstallerDlg.cpp : 实现文件
//
 
#include "stdafx.h"
#include "ApkInstaller.h"
#include "ApkInstallerDlg.h"
#include "afxdialogex.h"
 
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
 
// CApkInstallerDlg 对话框
static void Tokenize(CString text, CString split, CStringArray &list)
{
    CString partStr;
    int position = 0;
    do
    {
        partStr = text.Tokenize(split, position);
        if (partStr != T(""))
        {
            int offset = partStr.Find(T("device"));
            list.Add(partStr.Mid(0,offset).Trim());
        }
    } while (!partStr.IsEmpty());
}
 
static void GT_UTF8ToGBK(CString &strUTF8)  
{  
    //获取转换为多字节后需要的缓冲区大小，创建多字节缓冲区
    UINT nLen = MultiByteToWideChar(CP_UTF8,NULL,strUTF8,-1,NULL,NULL);
    WCHAR *wszBuffer = new WCHAR[nLen+1];
    nLen = MultiByteToWideChar(CP_UTF8,NULL,strUTF8,-1,wszBuffer,nLen);
    wszBuffer[nLen] = 0;
 
    nLen = WideCharToMultiByte(CP_ACP,NULL,wszBuffer,-1,NULL,NULL,NULL,NULL);
    CHAR *szBuffer = new CHAR[nLen+1];
    nLen = WideCharToMultiByte(CP_ACP,NULL,wszBuffer,-1,szBuffer,nLen,NULL,NULL);
    szBuffer[nLen] = 0;  
 
    strUTF8 = szBuffer;
    //清理内存
    delete []szBuffer;
    delete []wszBuffer;
}
 
static CString ParseName(CString strProp,CString beginLabel,CString endLabel)
{
    int mPos = strProp.Find(beginLabel);
    if (mPos != -1)
    {
        mPos += beginLabel.GetLength();
        int mEnd = strProp.Find(endLabel,mPos);
        if(mEnd != -1)
        {
            return strProp.Mid(mPos,mEnd-mPos);
        }
    }
    return "";
}
 
//函数名称：AssociateApkFile
//函数功能：关联apk文件和图标
//输入参数：strType 文件类型  .apk .mpk等
//     strPath 程序路径名 与文件关联的可执行程序的带路径的文件名
//返 回 值：NULL
//函数说明：
static void AssociateFile(CString strType, CString strPath) {
    //HKEY_LOCAL_MACHINE/Software/Classes/等于HKEY_CLASSES_ROOT
    HKEY hkey;
    CString strKeyName("ITHTW");
    CString strResult("");
    LPBYTE OwnerGet = new BYTE[80]; //定义用户姓名
    memset(OwnerGet, 0, 80);
    DWORD dType1 = REG_SZ; //定义数据类型
    DWORD dLength = 80; //定义数据长度
    CString strValue("");
    CString strRegPath("");
    strRegPath = strType + "\\"; // .apk
    // 设置路径"HKEY_CLASS_ROOT\\.apk"的键值，关联.apk文件用
    if (RegOpenKey(HKEY_CLASSES_ROOT, strRegPath, &hkey) == ERROR_SUCCESS) {
        RegQueryValueEx(hkey, NULL, NULL, &dType1, OwnerGet, &dLength);
        strResult = (LPTSTR) OwnerGet;
        if (-1 == strResult.Find(CString("ITHTW"))) {
            strValue.Format("ITHTW");
            RegSetValue(hkey, _T(""), REG_SZ, strValue,
                strValue.GetLength() + 1);
            RegSetValueEx(hkey, "", 0, REG_SZ,
                (const BYTE*) strValue.GetBuffer(strValue.GetLength()),
                strValue.GetLength() + 1);
            RegCloseKey(hkey);
        }
    } else {
        if (RegCreateKey(HKEY_CLASSES_ROOT, strType, &hkey) != ERROR_SUCCESS) {
            return;
        } else {
            strValue.Format("ITHTW");
            RegSetValue(hkey, _T(""), REG_SZ, strValue,
                strValue.GetLength() + 1);
            RegSetValueEx(hkey, "", 0, REG_SZ,
                (const BYTE*) strValue.GetBuffer(strValue.GetLength()),
                strValue.GetLength() + 1);
            RegCloseKey(hkey);
        }
    }
    // 设置"SOFTWARE\\Classes\\.apk"的键值
    memset(OwnerGet, 0, 80);
    if (strResult.IsEmpty())
        strResult.Empty();
    strValue.Empty();
    strRegPath = CString("SOFTWARE\\Classes\\") + strType + "\\"; // SOFTWARE\\Classes\\.apk
    if ( RegOpenKey( HKEY_LOCAL_MACHINE, strRegPath, &hkey ) == ERROR_SUCCESS )
    {
        RegQueryValueEx(hkey, NULL, NULL, &dType1, OwnerGet, &dLength);
        strResult = (LPTSTR) OwnerGet;
        if (-1 == strResult.Find("ITHTW")) {
            // 该关键字下没有我的软件相关名,修改之
            strValue.Format("ITHTW");
            RegSetValue(hkey, _T(""), REG_SZ, strValue,
                strValue.GetLength() + 1);
            RegSetValueEx(hkey, "", 0, REG_SZ,
                (const BYTE*) strValue.GetBuffer(strValue.GetLength()),
                strValue.GetLength() + 1);
            RegCloseKey(hkey);
        }
    }
    else
    {
        if ( RegCreateKey( HKEY_LOCAL_MACHINE, strRegPath, &hkey ) != ERROR_SUCCESS )
        {
            return;
        }
        else
        {
            strValue.Format("ITHTW");
            RegSetValue(hkey, _T(""), REG_SZ, strValue, strValue.GetLength()+1);
            RegSetValueEx(hkey, "", 0, REG_SZ, (const BYTE*)strValue.GetBuffer(strValue.GetLength()), strValue.GetLength()+1);
            RegCloseKey(hkey);
        }
    }
    // 设置路径"HKEY_CURRENT_USER\\Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\.apk\\UserChoice"的值
    memset(OwnerGet, 0, 80);
    if (strResult.IsEmpty())
        strResult.Empty();
    strValue.Empty();
    strRegPath =
        CString(
        "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\")
        + strType + "\\UserChoice";
    if (RegOpenKey(HKEY_CURRENT_USER, strRegPath, &hkey) == ERROR_SUCCESS) {
        RegQueryValueEx(hkey, NULL, NULL, &dType1, OwnerGet, &dLength);
        strResult = (LPTSTR) OwnerGet;
        if (-1 == strResult.Find("ITHTW")) {
            // 该关键字下没有我的软件相关名,修改之
            strValue.Format("ITHTW");
            RegSetValue(hkey, "Progid", REG_SZ, strValue,
                strValue.GetLength() + 1);
            RegSetValueEx(hkey, "Progid", 0, REG_SZ,
                (const BYTE*) strValue.GetBuffer(strValue.GetLength()),
                strValue.GetLength() + 1);
            RegCloseKey(hkey);
        }
    } else {
        if (RegCreateKey(HKEY_CURRENT_USER, strRegPath, &hkey)
            != ERROR_SUCCESS) {
                return;
        } else {
            strValue.Format("ITHTW");
            RegSetValue(hkey, "Progid", REG_SZ, strValue,
                strValue.GetLength() + 1);
            RegSetValueEx(hkey, "Progid", 0, REG_SZ,
                (const BYTE*) strValue.GetBuffer(strValue.GetLength()),
                strValue.GetLength() + 1);
            RegCloseKey(hkey);
        }
    }
    memset(OwnerGet, 0, 80);
    if (strResult.IsEmpty())
        strResult.Empty();
    strValue.Empty();
    strRegPath =
        CString(
        "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\")
        + strType;
    if (RegOpenKey(HKEY_CURRENT_USER, strRegPath, &hkey) == ERROR_SUCCESS) {
        RegQueryValueEx(hkey, NULL, NULL, &dType1, OwnerGet, &dLength);
        strResult = (LPTSTR) OwnerGet;
        if (-1 == strResult.Find("ITHTW")) {
            // 该关键字下没有我的软件相关名,修改之
            strValue.Format("ITHTW");
            RegSetValue(hkey, "Progid", REG_SZ, strValue,
                strValue.GetLength() + 1);
            RegSetValueEx(hkey, "Progid", 0, REG_SZ,
                (const BYTE*) strValue.GetBuffer(strValue.GetLength()),
                strValue.GetLength() + 1);
            RegCloseKey(hkey);
        }
    } else {
        if (RegCreateKey(HKEY_CURRENT_USER, strRegPath, &hkey)
            != ERROR_SUCCESS) {
                return;
        } else {
            strValue.Format("ITHTW");
            RegSetValue(hkey, "Progid", REG_SZ, strValue,
                strValue.GetLength() + 1);
            RegSetValueEx(hkey, "Progid", 0, REG_SZ,
                (const BYTE*) strValue.GetBuffer(strValue.GetLength()),
                strValue.GetLength() + 1);
            RegCloseKey(hkey);
        }
    }
    // 通知系统，文件关联改变了
    SHChangeNotify(SHCNE_ASSOCCHANGED, SHCNF_IDLIST, NULL, NULL);
    SHChangeNotify(SHCNE_UPDATEIMAGE, SHCNF_DWORD, NULL, NULL);
}
 
CApkInstallerDlg::CApkInstallerDlg(CWnd* pParent /*=NULL*/)
    : CDialogEx(CApkInstallerDlg::IDD, pParent)
{
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
    memset(szCommand,0,sizeof(szCommand));
    memset(szPath,0,sizeof(szPath));
    hEvent = CreateEvent(NULL,FALSE,TRUE,NULL);
}
 
void CApkInstallerDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
}
 
BEGIN_MESSAGE_MAP(CApkInstallerDlg, CDialogEx)
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_WM_DROPFILES()
    ON_BN_CLICKED(IDC_BTN_INSTALL, &CApkInstallerDlg::OnBnClickedBtnInstall)
    ON_BN_CLICKED(IDC_BTN_UNINSTALL, &CApkInstallerDlg::OnBnClickedBtnUninstall)
    ON_BN_CLICKED(IDC_BTN_REG, &CApkInstallerDlg::OnBnClickedBtnReg)
END_MESSAGE_MAP()
 
 
// CApkInstallerDlg 消息处理程序
 
CString ReadDeviceName(CString cmd)
{
    CApkInstallerDlg* dlg = (CApkInstallerDlg*)AfxGetApp()->GetMainWnd();
    char buffer[10240] = {0};
    DWORD dwRead = 0;
    HANDLE hReadPipe = NULL;
    dlg->ProcessTask(&hReadPipe,cmd.GetBuffer());
    cmd.ReleaseBuffer();
    CString text;
    while (ReadFile(hReadPipe,buffer,10240,&dwRead,0))
    {
        CString temp(buffer,dwRead);
        temp = temp.Trim();
        text += temp;
    }
    return text;
}
 
void CApkInstallerDlg::ReadDevices(void)
{
    _tcscpy_s(szCommand,T("adb devices"));
    //开始读取
    char buffer[10240] = {0};
    DWORD dwRead = 0;
    HANDLE hReadPipe = NULL;
    ProcessTask(&hReadPipe,szCommand);
    CString text;
    while (ReadFile(hReadPipe,buffer,10240,&dwRead,0))
    {
        CString temp(buffer,dwRead);
        temp = temp.Trim();
        text += temp;
    }
    Tokenize(text,T("\n"),devices);
    CComboBox *comboBox = (CComboBox*)GetDlgItem(IDC_COMBO_DEVICES);
    for (int i=1;i<devices.GetCount();i++)//过滤第一list信息
    {
        CString prop = ReadDeviceName("adb -s "+devices.GetAt(i)+" shell getprop");
        CString manufacturer = ParseName(prop,"&#91;ro.product.manufacturer&#93;: &#91;","&#93;");
        CString mode = ParseName(prop,"&#91;ro.product.model&#93;: &#91;","&#93;");
        comboBox->AddString(manufacturer+mode);
    }
    if(comboBox->GetCount() > 0)
        comboBox->SetCurSel(0);
}
 
BOOL CApkInstallerDlg::OnInitDialog()
{
    CDialogEx::OnInitDialog();
 
    // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);            // 设置大图标
    SetIcon(m_hIcon, FALSE);        // 设置小图标
 
    // TODO: 在此添加额外的初始化代码
 
    ReadDevices();
    SetDlgItemText(IDC_EDIT_LOG,T("日志"));
    if(_tcslen(szPath) == 0)
        SetDlgItemText(IDC_EDIT_FILE,T("文件路径"));
    else
    {
        SetDlgItemText(IDC_EDIT_FILE,szPath);
        if (CheckApk(szPath))
        {
            //获取包名
            SetSzCommand(T("aapt d badging "),CMD_GET_READ_INFO);
            _tcscat_s(szCommand,szPath);
            AfxBeginThread(ReadThread,this);
        }
    }
    return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}
 
// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。
 
void CApkInstallerDlg::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
    {
        CDialogEx::OnPaint();
        CRect rt;
        GetDlgItem(IDC_STATIC_ICO)->GetWindowRect(&rt);
        CPoint p(rt.left,rt.top);
        ::ScreenToClient(m_hWnd,&p);
 
        CDC *pDC = GetDC();
        Graphics graphics(pDC->m_hDC);
        Image *img = NULL;
        ImageFromID(IDB_PNG_ANDROID,img);
        graphics.DrawImage(img,p.x-20,p.y);
        FontFamily fontFamily(L"微软雅黑");
        Gdiplus::Font font(&fontFamily, 11, FontStyleRegular, UnitPoint);
        PointF pointF((REAL)(p.x+10), (REAL)p.y+30);
        SolidBrush solidBrush(Color(255, 0, 0, 0)); 
        //TCHAR string[MAX_PATH] = T("安装器");
        WCHAR string[MAX_PATH] = L"安装器";
        graphics.DrawString(string, (INT)wcslen(string), &font, pointF,&solidBrush);
        if(img != NULL)
            delete img;
        ReleaseDC(pDC);
    }
}
 
//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CApkInstallerDlg::OnQueryDragIcon()
{
    return static_cast<HCURSOR>(m_hIcon);
}
 
 
void CApkInstallerDlg::OnDropFiles(HDROP hDropInfo)
{
    // TODO: 在此添加消息处理程序代码和/或调用默认值
    TCHAR file_name[MAX_PATH] = {0};
    ::DragQueryFile(hDropInfo,0, file_name, MAX_PATH);
    SetDlgItemText(IDC_EDIT_FILE,file_name);
    if (CheckApk(file_name))
    {
        //获取包名
        SetSzCommand(T("aapt d badging "),CMD_GET_READ_INFO);
        _tcscat_s(szCommand,file_name);
        AfxBeginThread(ReadThread,this);
    }
    CDialogEx::OnDropFiles(hDropInfo);
}
 
 
UINT CApkInstallerDlg::ReadThread(LPVOID pVoid)
{
    try{
        CApkInstallerDlg *dlg = (CApkInstallerDlg*)pVoid;
        WaitForSingleObject(dlg->hEvent,INFINITE);
        dlg->SetWindowText(T("处理中,请稍后..."));
         
        //开始读取
        char buffer[10240] = {0};
        DWORD dwRead = 0;
        HANDLE hReadPipe = NULL;
        dlg->ProcessTask(&hReadPipe,dlg->szCommand);
        CString text;
        while (ReadFile(hReadPipe,buffer,10240,&dwRead,0))
        {
            CString temp(buffer,dwRead);
            GT_UTF8ToGBK(temp);
            text += temp;
        }
        text.Trim();
        if (text.GetLength() > 0)
        {
             if (dlg->nCommand == CMD_INSTALL_APP)
            {
                dlg->SetDlgItemText(IDC_EDIT_LOG,text);
            }
            else if (dlg->nCommand == CMD_GET_READ_INFO)
            {
                dlg->SetDlgItemText(IDC_EDIT_LOG,text);
                CString temp = T("package: name='");
                int startIndex = text.Find(temp);
                int endIndex = text.Find(T("'"),startIndex+temp.GetLength());
                int nCount = endIndex - startIndex - temp.GetLength();
                if(startIndex >= 0 && nCount > 0)
                    dlg->packAge=text.Mid(startIndex+temp.GetLength(),nCount);
            }
            else if (dlg->nCommand == CMD_UNINSTALL)
            {
                if(hReadPipe != NULL)
                    CloseHandle(hReadPipe);
                hReadPipe = NULL;
                dlg->ProcessTask(&hReadPipe,dlg->szCommand);
                if (ReadFile(hReadPipe,buffer,10240,&dwRead,0))
                {
                    CString new_text(buffer,dwRead);
                    new_text = new_text.Trim();
                    dlg->SetDlgItemText(IDC_EDIT_LOG,new_text);
                }
            }
        }
        else
        {
            //控制台数据读取失败
            strcpy_s(buffer,"数据读取失败");
        }
        CloseHandle(hReadPipe);
        dlg->SetWindowText(T("处理完成"));
        SetEvent(dlg->hEvent);
    }catch(CException *e)
    {
        TCHAR msg[512];
        e->GetErrorMessage(msg,512);
        AfxMessageBox(msg);
    }
    return 0;
}
 
 
UINT CApkInstallerDlg::ProcessTask(HANDLE* hReadPipe,TCHAR* strCommand)
{
    STARTUPINFO si;
    PROCESS_INFORMATION pi;
    si.cb = sizeof(STARTUPINFO);
    GetStartupInfo(&si);
 
 
    SECURITY_ATTRIBUTES sa = {0};
    sa.nLength = sizeof(SECURITY_ATTRIBUTES);
    sa.bInheritHandle = TRUE;
    sa.lpSecurityDescriptor = NULL;
 
    HANDLE hWritePipe = NULL;
    if(!CreatePipe(hReadPipe,&hWritePipe,&sa,0))
    {
        AfxMessageBox(T("创建管道失败"));
        return 0;
    }
 
    si.hStdOutput = hWritePipe;
    si.hStdError = hWritePipe;
    si.wShowWindow = SW_HIDE;
    si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
    if (CreateProcess (NULL,strCommand,NULL,NULL,TRUE,NULL,NULL,NULL,&si,&pi))
    {
        //等待进程执行完毕
        //WaitForSingleObject(pi.hProcess,INFINITE);
        //关闭进程和主线程句柄
        CloseHandle (pi.hProcess);
        CloseHandle (pi.hThread);
    }
    //关闭写入句柄
    CloseHandle(hWritePipe);
    return 0;
}
 
void CApkInstallerDlg::SetSzCommand(TCHAR* wstr,CMD_TYPE nType)
{
    memset(szCommand,0,sizeof(szCommand));
    _tcscpy_s(szCommand,wstr);
    this->nCommand = nType;
}
 
 
void CApkInstallerDlg::OnBnClickedBtnUninstall()
{
    // TODO: 在此添加控件通知处理程序代码
    try
    {
        TCHAR file_name[512] = {0};
        GetDlgItemText(IDC_EDIT_FILE,file_name,512);
        if (CheckApk(file_name))
        {
            SetDlgItemText(IDC_EDIT_LOG,T("正在卸载:")+packAge);
            //卸载
            SetSzCommand(T("adb -s "),CMD_UNINSTALL);
            CComboBox *comboBox = (CComboBox*)GetDlgItem(IDC_COMBO_DEVICES);
            int cur = comboBox->GetCurSel();
            if (cur >= 0 && cur < devices.GetCount()-1)
            {
                CString device = devices.GetAt(cur+1);
                _tcscat_s(szCommand,device);
                _tcscat_s(szCommand,T(" uninstall "));
                _tcscat_s(szCommand,packAge);
                AfxBeginThread(ReadThread,this);
            }
        }
    }
    catch (CException* e)
    {
        TCHAR msg&#91;512&#93;;
        e->GetErrorMessage(msg,512);
        MessageBox(msg);
    }
}
 
void CApkInstallerDlg::OnBnClickedBtnInstall()
{
    // TODO: 在此添加控件通知处理程序代码
    try
    {
        TCHAR file_name[512] = {0};
        GetDlgItemText(IDC_EDIT_FILE,file_name,512);
        if (CheckApk(file_name))
        {
            SetDlgItemText(IDC_EDIT_LOG,T("正在安装:")+CString(file_name));
            SetSzCommand(T("adb -s "),CMD_INSTALL_APP);
            CComboBox *comboBox = (CComboBox*)GetDlgItem(IDC_COMBO_DEVICES);
            int cur = comboBox->GetCurSel();
            if (cur >= 0 && cur < devices.GetCount()-1)
            {
                CString device = devices.GetAt(cur+1);
                _tcscat_s(szCommand,device);
                _tcscat_s(szCommand,T(" install "));
                _tcscat_s(szCommand,file_name);
                AfxBeginThread(ReadThread,this);
            }
        }
    }
    catch (CException* e)
    {
        TCHAR msg&#91;512&#93;;
        e->GetErrorMessage(msg,512);
        MessageBox(msg);
    }
}
 
 
BOOL CApkInstallerDlg::CheckApk(TCHAR* path)
{
    try
    {
        TCHAR* exist = _tcsstr(path,T("apk"));
        if (exist == NULL)
        {
            MessageBox(T(""));
            return FALSE;
        }
        CFileFind finder;
        if(!finder.FindFile(path))
        {
            MessageBox(T("指定文件不存在"));
            return FALSE;
        }
    }
    catch (CException* e)
    {
        TCHAR msg[512];
        e->GetErrorMessage(msg,512);
        MessageBox(msg);
    }
    return TRUE;
}
 
void CApkInstallerDlg::OnBnClickedBtnReg()
{
    // TODO: 在此添加控件通知处理程序代码
    TCHAR szName[512];
    GetModuleFileName(NULL,szName,sizeof(szName)*sizeof(TCHAR));
    AssociateFile(".apk",szName);
}

项目源码下载:ApkInstaller
编译好的APP:Release

本文固定链接: http://www.ithtw.com/1528.html

#include "stdafx.h"
#include <windows.h>
#include <stdio.h>
#include <iostream>
 
#ifndef _X_PE_H_
#define _X_PE_H_
//获取PE头基地址
#define SIZE_OF_NT_SIGNATURE sizeof(DWORD)
#define NTSIGNATURE(a) ((LPVOID)((BYTE *)a + ((PIMAGE_DOS_HEADER)a)->e_lfanew - 1))
#define PEFHDROFFSET(a) ((LPVOID)((BYTE *)a + ((PIMAGE_DOS_HEADER)a)->e_lfanew - 1 + SIZE_OF_NT_SIGNATURE))
#define OPTHDROFFSET(a) ((LPVOID)((BYTE *)a + ((PIMAGE_DOS_HEADER)a)->e_lfanew - 1 + SIZE_OF_NT_SIGNATURE + sizeof(IMAGE_FILE_HEADER)))
DWORD WINAPI ImageFileType (LPVOID lpFile);
#endif
 
using namespace std;
//PE文件类型判断
DWORD WINAPI ImageFileType (LPVOID lpFile)
{
    /* 首先出现的是DOS文件标志 */
    if (*(USHORT *)lpFile == IMAGE_DOS_SIGNATURE)
    {
        /* 由DOS头部决定PE文件头部的位置 */
        if (LOWORD (*(DWORD *)NTSIGNATURE (lpFile)) ==
            IMAGE_OS2_SIGNATURE ||
            LOWORD (*(DWORD *)NTSIGNATURE (lpFile)) ==
            IMAGE_OS2_SIGNATURE_LE)
            return (DWORD)LOWORD(*(DWORD *)NTSIGNATURE (lpFile));
        else if (*(DWORD *)NTSIGNATURE (lpFile) ==
            IMAGE_NT_SIGNATURE)
            return IMAGE_NT_SIGNATURE;
        else
            return IMAGE_DOS_SIGNATURE;
    }
    else
        /* 不明文件种类 */
        return 0;
}
int main()
{
    //业务信息参考
    //http://www.vckbase.com/document/viewdoc/?id=1334
 
    CHAR szBuf[2048];
    FILE* pFile;
    fopen_s(&pFile,"F:\\SVN\\Debug\\LGT.exe","r");//这里加载一个PE文件
    int iRead=fread(szBuf,1,2048,pFile);
 
    //MZ-DOS头
    IMAGE_DOS_HEADER* dos;
    dos=(IMAGE_DOS_HEADER*)szBuf;
    cout.setf(ios::hex,ios::basefield);//设置十六进制显示数值
    cout.setf(ios::showbase|ios::uppercase);//设置0x头和大写
    cout<<"dos->e_magic="<<dos->e_magic<<endl;
    cout<<"dos->e_lfanew="<<dos->e_lfanew<<endl;
    cout<<"NTSIGNATURE="<<*(DWORD *)NTSIGNATURE(szBuf)<<endl;
    cout<<"ImageFileType="<<ImageFileType(szBuf)<<" is ";
    switch(ImageFileType(szBuf))
    {
    case IMAGE_DOS_SIGNATURE://MZ
        cout<<"IMAGE_DOS_SIGNATURE"<<endl;
        break;
    case IMAGE_OS2_SIGNATURE://NE
        cout<<"IMAGE_OS2_SIGNATURE"<<endl;
        break;
    case IMAGE_OS2_SIGNATURE_LE:
        cout<<"IMAGE_OS2_SIGNATURE_LE"<<endl;
        break;
    case IMAGE_NT_SIGNATURE:
        cout<<"IMAGE_NT_SIGNATURE"<<endl;
    }
 
    cout<<"PEFHDROFFSET="<<PEFHDROFFSET(szBuf)<<endl;
    //PE必选头
    PIMAGE_FILE_HEADER header =(PIMAGE_FILE_HEADER)PEFHDROFFSET(szBuf);
    cout<<"Machine="<<header->Machine<<endl;
    cout<<"区段数NumberOfSections="<<header->NumberOfSections<<endl;
    cout<<"建立时间TimeDateStamp="<<header->TimeDateStamp<<endl;
    cout<<"PointerToSymbolTable="<<header->PointerToSymbolTable<<endl;
    cout<<"可选头部大小SizeOfOptionalHeader="<<header->SizeOfOptionalHeader<<endl;
    cout<<"特征Characteristics="<<header->Characteristics<<endl;
 
    //PE可选头
    PIMAGE_OPTIONAL_HEADER opHeader=(PIMAGE_OPTIONAL_HEADER)OPTHDROFFSET(szBuf);
    cout<<"魔术字Magic="<<opHeader->Magic<<endl;//0x010B为.EXE，0x0107为ROM映像
    cout<<"连接器主版本MajorLinkerVersion="<<(short)opHeader->MajorLinkerVersion<<endl;
    cout<<"连接器副版本MinorLinkerVersion="<<(short)opHeader->MinorLinkerVersion<<endl;
    cout<<"可执行代码大小SizeOfCode="<<opHeader->SizeOfCode<<endl;
    cout<<"已初始化的数据大小SizeOfInitializedData="<<opHeader->SizeOfInitializedData<<endl;
    cout<<"未初始化的数据大小SizeOfUninitializedData="<<opHeader->SizeOfUninitializedData<<endl;
    cout<<"入口点AddressOfEntryPoint="<<opHeader->AddressOfEntryPoint<<endl;
    cout<<"代码基址(.text段)BaseOfCode="<<opHeader->BaseOfCode<<endl;
    cout<<"数据基址(.bss段)BaseOfData="<<opHeader->BaseOfData<<endl;
    cout<<"映像基址ImageBase"<<opHeader->ImageBase<<endl;//进程映像地址空间中的首选基地址,Windows NT的Microsoft Win32 SDK链接器将这个值默认设为0x00400000，但是你可以使用-BASE:linker开关改变这个值。 
    cout<<"区段对齐SectionAlignment="<<opHeader->SectionAlignment<<endl;
    cout<<"FileAlignment="<<opHeader->FileAlignment<<endl;
    cout<<"NT主版本号MajorOperatingSystemVersion="<<opHeader->MajorOperatingSystemVersion<<endl;
    cout<<"NT副版本号MinorOperatingSystemVersion="<<opHeader->MinorOperatingSystemVersion<<endl;
    cout<<"应用程序主版本号MajorImageVersion="<<opHeader->MajorImageVersion<<endl;
    cout<<"应用程序副版本号MinorImageVersion="<<opHeader->MinorImageVersion<<endl;
    cout<<"Win32子系统主版本号MajorSubsystemVersion="<<opHeader->MajorSubsystemVersion<<endl;
    cout<<"Win32子系统副版本号MinorSubsystemVersion="<<opHeader->MinorSubsystemVersion<<endl;
    cout<<"SizeOfImage="<<opHeader->SizeOfImage<<endl;
    cout<<"头大小SizeOfHeaders="<<opHeader->SizeOfHeaders<<endl;
    cout<<"校验和CheckSum="<<opHeader->CheckSum<<endl;//私有算法
    cout<<"子系统Subsystem="<<opHeader->Subsystem<<endl;
    cout<<"DllCharacteristics="<<opHeader->DllCharacteristics<<endl;
    cout<<"SizeOfStackReserve="<<opHeader->SizeOfStackReserve<<endl;
    cout<<"SizeOfStackCommit="<<opHeader->SizeOfStackCommit<<endl;
    cout<<"SizeOfHeapReserve="<<opHeader->SizeOfHeapReserve<<endl;
    cout<<"SizeOfHeapCommit="<<opHeader->SizeOfHeapCommit<<endl;
    cout<<"LoaderFlags="<<opHeader->LoaderFlags<<endl;
    cout<<"NumberOfRvaAndSizes="<<opHeader->NumberOfRvaAndSizes<<endl;
    cout<<"导出表RVA:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].VirtualAddress<<endl;
    cout<<"导出表大小:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_EXPORT].Size<<endl;
    cout<<"导入表RVA:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].VirtualAddress<<endl;
    cout<<"导入表大小:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_IMPORT].Size<<endl;
    cout<<"资源RVA:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].VirtualAddress<<endl;
    cout<<"资源大小:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_RESOURCE].Size<<endl;
    cout<<"TLSRVA:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_TLS].VirtualAddress<<endl;
    cout<<"TLS大小:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_TLS].Size<<endl;
    cout<<"调试RVA:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].VirtualAddress<<endl;
    cout<<"调试:"<<opHeader->DataDirectory[IMAGE_DIRECTORY_ENTRY_DEBUG].Size<<endl;
 
    cout<<sizeof(IMAGE_OPTIONAL_HEADER)<<endl;
    PIMAGE_SECTION_HEADER ps=(PIMAGE_SECTION_HEADER)((BYTE*)opHeader+sizeof(IMAGE_OPTIONAL_HEADER));
 
    //打印所有的段信息
    for (int i=0;i<header->NumberOfSections;i++)
    {
        cout<<endl;
        cout<<"Name="<<ps[i].Name<<endl;
        cout<<"PhysicalAddress="<<ps[i].Misc.PhysicalAddress<<endl;
        cout<<"VirtualSize="<<ps[i].Misc.VirtualSize<<endl;
        cout<<"VirtualAddress="<<ps[i].VirtualAddress<<endl;
        cout<<"SizeOfRawData="<<ps[i].SizeOfRawData<<endl;
        cout<<"PointerToRawData="<<ps[i].PointerToRawData<<endl;
        cout<<"PointerToRelocations="<<ps[i].PointerToRelocations<<endl;
        cout<<"PointerToLinenumbers="<<ps[i].PointerToLinenumbers<<endl;
        cout<<"NumberOfRelocations="<<ps[i].NumberOfRelocations<<endl;
        cout<<"NumberOfLinenumbers="<<ps[i].NumberOfLinenumbers<<endl;
        cout<<"Characteristics="<<ps[i].Characteristics<<endl;
    }
    system("pause");
    return 0;
}

// Win32Project2.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <iostream>
#include <wtypes.h>
#include<tlhelp32.h>
#include <psapi.h>
#pragma   comment(lib,"psapi.lib")
using namespace std;
#define ProcessBasicInformation 0 

typedef struct
{
    DWORD ExitStatus;
    DWORD PebBaseAddress;
    DWORD AffinityMask;
    DWORD BasePriority;
    ULONG UniqueProcessId;
    ULONG InheritedFromUniqueProcessId;
}   PROCESS_BASIC_INFORMATION;

typedef LONG(__stdcall *PROCNTQSIP)(HANDLE, UINT, PVOID, ULONG, PULONG);
//
DWORD GetParentProcessID(DWORD dwProcessId);

int _tmain(int argc, _TCHAR* argv[])
{
    //  输入文件名
    HANDLE   hSnapShot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    PROCESSENTRY32   procentry;//=sizeof(PROCESSENTRY32); 
    BOOL   bFlag = Process32First(hSnapShot, &procentry);
    DWORD       parentprocessid;
    while (bFlag)
    {
        if (_stricmp(procentry.szExeFile, "QQExternal.exe") == 0) //你的程序
        {
            DWORD       processid = procentry.th32ProcessID;         //找到 
            cout << procentry.th32ProcessID << endl;
            cout << GetParentProcessID(procentry.th32ProcessID) << endl;
            parentprocessid = GetParentProcessID(procentry.th32ProcessID);
        }
        bFlag = Process32Next(hSnapShot, &procentry);
    }
    cout << parentprocessid << endl;


    HANDLE processHandle = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, parentprocessid);
    char processName[MAX_PATH] = { 0 };
    GetModuleBaseName(processHandle, 0, processName, MAX_PATH);
    char  processFullPath[MAX_PATH] = { 0 };
    GetModuleFileNameEx(processHandle, 0, processFullPath, MAX_PATH);


    cout << processName << endl;
    cout << processFullPath << endl;
    //cout << Environment::CommandLine<< endl;
    system("pause");
    return 0;
}

DWORD GetParentProcessID(DWORD dwProcessId)
{
    LONG                        status;
    DWORD                        dwParentPID = (DWORD)-1;
    HANDLE                        hProcess;
    PROCESS_BASIC_INFORMATION    pbi;

    PROCNTQSIP NtQueryInformationProcess = (PROCNTQSIP)GetProcAddress(
        GetModuleHandle("ntdll"), "NtQueryInformationProcess");

    if (NULL == NtQueryInformationProcess)
    {
        return (DWORD)-1;
    }
    // Get process handle
    hProcess = OpenProcess(PROCESS_QUERY_INFORMATION, FALSE, dwProcessId);
    if (!hProcess)
    {
        return (DWORD)-1;
    }

    // Retrieve information
    status = NtQueryInformationProcess(hProcess,
        ProcessBasicInformation,
        (PVOID)&pbi,
        sizeof(PROCESS_BASIC_INFORMATION),
        NULL
        );

    // Copy parent Id on success
    if (!status)
    {
        dwParentPID = pbi.InheritedFromUniqueProcessId;
    }

    CloseHandle(hProcess);

    return dwParentPID;

}

一个 EXE 在运行过程中（被）改名了，需要准确地获取它的文件名。

尝试

原本以为这是一个非常简单的 CASE，直接用 GetModuleFileName 不就行了吗？结果还真不如我所想。无论程序运行过程中被改名成什么样子，GetModuleFileName 返回的都是 EXE 开始运行时的名字。然后又尝试了 GetProcessImageFileName，也是如此，直到最后找到了 QueryFullProcessImageName。

示例代码

#include <Windows.h>
#include <Psapi.h>
#include <stdio.h>

#pragma comment(lib, "Psapi.lib")

void OutputSelfpath()
{
	char szFile[MAX_PATH] = {0};
	GetModuleFileName(NULL, szFile, MAX_PATH);
	printf("GetModuleFileName:\n\r%s\n\n", szFile);

	memset(szFile, 0, MAX_PATH);

	HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS, FALSE, GetCurrentProcessId());
	if (!hProcess)
	{
		printf("OpenProcess failed!\n");
	}
	else
	{
		DWORD dwRet = GetProcessImageFileName(hProcess, szFile, MAX_PATH);
		if (dwRet)
		{
			printf("GetProcessImageFileName:\n\r%s\n\n", szFile);
		}
		else
		{
			printf("GetProcessImageFileName failed!\n");
		}

        memset(szFile, 0, MAX_PATH);
		DWORD dwSize = MAX_PATH;
		if (QueryFullProcessImageName(hProcess, 0, szFile, &dwSize))
		{
			printf("QueryFullProcessImageName:\n\r%s\n\n", szFile);
		}
		else
		{
			printf("QueryFullProcessImageName failed\n", szFile);
		}
	}
}

int main()
{
	const char* pszFile = "ConsoleTest.exe";
	const char* pszNewFile = "ConsoleTest_bak.exe";
	remove(pszNewFile);

	OutputSelfpath();

	int nRet = rename(pszFile, pszNewFile);

	if (0 != nRet)
	{
		printf("rename file failed!\n");
	}
	else
	{
		printf("################### after rename ###################\n\n");
		OutputSelfpath();
	}

	system("pause");
	return 0;
}

运行结果

思考

现象上讲就是如此了，这几个 API 的本质区别是什么呢？待续。

http://mazhuang.org/2014/10/09/path-after-rename/

//a.h
#include "b.h"
class A{
#pragma push_macro("aaaa")
#undef aaaa
  void fuckyou( int aaaa);
#pragma pop_macro("aaaa")
};

//b.h

#define aaaa (*__daylight())

转载:http://bbs.csdn.net/topics/390842294

使用Duilib创建窗口的时候，假如有这样一个需求：

1.窗口大小规定

2.不可拖拽改变窗口大小

3.双击标题栏禁止最大化

4.禁止拖拽窗口到屏幕上边沿最大化

针对以上四点,给出解决办法:

1.设置size=”宽,高”

2.sizebox=”0,0,0,0″

3.在HandleMessage中屏蔽掉WM_NCLBUTTONDBLCLK

case WM_NCLBUTTONDBLCLK:
  //不做处理
{
    return 0;
}
break;

4.要修改窗口风格

//修改窗口风格
    SetWindowLong(m_hWnd,GWL_STYLE,GetWindowLong(m_hWnd,GWL_STYLE) & ~WS_MAXIMIZEBOX & ~WS_THICKFRAME );

设置完成后:

//unistd.h 
#ifndef _UNISTD_H
#define _UNISTD_H 
#include <io.h> 
#include <process.h> 
#endif /* _UNISTD_H */

#include <string.h>

/*
 * Find the first occurrence of the byte string s in byte string l.
 */

const void *
memmem(const void *l, size_t l_len, const void *s, size_t s_len)
{
	register char *cur, *last;
	const char *cl = (const char *)l;
	const char *cs = (const char *)s;

	/* we need something to compare */
	if (l_len == 0 || s_len == 0)
		return NULL;

	/* "s" must be smaller or equal to "l" */
	if (l_len < s_len)
		return NULL;

	/* special case where s_len == 1 */
	if (s_len == 1)
		return memchr(l, (int)*cs, l_len);

	/* the last position where its possible to find "s" in "l" */
	last = (char *)cl + l_len - s_len;

	for (cur = (char *)cl; cur <= last; cur++)
		if (cur[0] == cs[0] && memcmp(cur, cs, s_len) == 0)
			return cur;

	return NULL;
}

//mman.h
/*
 * sys/mman.h
 * mman-win32
 */

#ifndef _SYS_MMAN_H_
#define _SYS_MMAN_H_

#ifndef _WIN32_WINNT		// Allow use of features specific to Windows XP or later.                   
#define _WIN32_WINNT 0x0501	// Change this to the appropriate value to target other versions of Windows.
#endif						

/* All the headers include this file. */
#ifndef _MSC_VER
#include <_mingw.h>
#endif

#include <sys/types.h>

#ifdef __cplusplus
extern "C" {
#endif

#define PROT_NONE       0
#define PROT_READ       1
#define PROT_WRITE      2
#define PROT_EXEC       4

#define MAP_FILE        0
#define MAP_SHARED      1
#define MAP_PRIVATE     2
#define MAP_TYPE        0xf
#define MAP_FIXED       0x10
#define MAP_ANONYMOUS   0x20
#define MAP_ANON        MAP_ANONYMOUS

#define MAP_FAILED      ((void *)-1)

/* Flags for msync. */
#define MS_ASYNC        1
#define MS_SYNC         2
#define MS_INVALIDATE   4

void*   mmap(void *addr, size_t len, int prot, int flags, int fildes, off_t off);
int     munmap(void *addr, size_t len);
int     mprotect(void *addr, size_t len, int prot);
int     msync(void *addr, size_t len, int flags);
int     mlock(const void *addr, size_t len);
int     munlock(const void *addr, size_t len);

#ifdef __cplusplus
};
#endif

#endif /*  _SYS_MMAN_H_ */

//mman.c

#include <windows.h>
#include <errno.h>
#include <io.h>

#include "mman.h"

#ifndef FILE_MAP_EXECUTE
#define FILE_MAP_EXECUTE    0x0020
#endif /* FILE_MAP_EXECUTE */

static int __map_mman_error(const DWORD err, const int deferr)
{
    if (err == 0)
        return 0;
    //TODO: implement
    return err;
}

static DWORD __map_mmap_prot_page(const int prot)
{
    DWORD protect = 0;
    
    if (prot == PROT_NONE)
        return protect;
        
    if ((prot & PROT_EXEC) != 0)
    {
        protect = ((prot & PROT_WRITE) != 0) ? 
                    PAGE_EXECUTE_READWRITE : PAGE_EXECUTE_READ;
    }
    else
    {
        protect = ((prot & PROT_WRITE) != 0) ?
                    PAGE_READWRITE : PAGE_READONLY;
    }
    
    return protect;
}

static DWORD __map_mmap_prot_file(const int prot)
{
    DWORD desiredAccess = 0;
    
    if (prot == PROT_NONE)
        return desiredAccess;
        
    if ((prot & PROT_READ) != 0)
        desiredAccess |= FILE_MAP_READ;
    if ((prot & PROT_WRITE) != 0)
        desiredAccess |= FILE_MAP_WRITE;
    if ((prot & PROT_EXEC) != 0)
        desiredAccess |= FILE_MAP_EXECUTE;
    
    return desiredAccess;
}

void* mmap(void *addr, size_t len, int prot, int flags, int fildes, off_t off)
{
    HANDLE fm, h;
    
    void * map = MAP_FAILED;
    
#ifdef _MSC_VER
#pragma warning(push)
#pragma warning(disable: 4293)
#endif

    const DWORD dwFileOffsetLow = (sizeof(off_t) <= sizeof(DWORD)) ? 
                    (DWORD)off : (DWORD)(off & 0xFFFFFFFFL);
    const DWORD dwFileOffsetHigh = (sizeof(off_t) <= sizeof(DWORD)) ?
                    (DWORD)0 : (DWORD)((off >> 32) & 0xFFFFFFFFL);
    const DWORD protect = __map_mmap_prot_page(prot);
    const DWORD desiredAccess = __map_mmap_prot_file(prot);

    const off_t maxSize = off + (off_t)len;

    const DWORD dwMaxSizeLow = (sizeof(off_t) <= sizeof(DWORD)) ? 
                    (DWORD)maxSize : (DWORD)(maxSize & 0xFFFFFFFFL);
    const DWORD dwMaxSizeHigh = (sizeof(off_t) <= sizeof(DWORD)) ?
                    (DWORD)0 : (DWORD)((maxSize >> 32) & 0xFFFFFFFFL);

#ifdef _MSC_VER
#pragma warning(pop)
#endif

    errno = 0;
    
    if (len == 0 
        /* Unsupported flag combinations */
        || (flags & MAP_FIXED) != 0
        /* Usupported protection combinations */
        || prot == PROT_EXEC)
    {
        errno = EINVAL;
        return MAP_FAILED;
    }
    
    h = ((flags & MAP_ANONYMOUS) == 0) ? 
                    (HANDLE)_get_osfhandle(fildes) : INVALID_HANDLE_VALUE;

    if ((flags & MAP_ANONYMOUS) == 0 && h == INVALID_HANDLE_VALUE)
    {
        errno = EBADF;
        return MAP_FAILED;
    }

    fm = CreateFileMapping(h, NULL, protect, dwMaxSizeHigh, dwMaxSizeLow, NULL);

    if (fm == NULL)
    {
        errno = __map_mman_error(GetLastError(), EPERM);
        return MAP_FAILED;
    }
  
    map = MapViewOfFile(fm, desiredAccess, dwFileOffsetHigh, dwFileOffsetLow, len);

    CloseHandle(fm);
  
    if (map == NULL)
    {
        errno = __map_mman_error(GetLastError(), EPERM);
        return MAP_FAILED;
    }

    return map;
}

int munmap(void *addr, size_t len)
{
    if (UnmapViewOfFile(addr))
        return 0;
        
    errno =  __map_mman_error(GetLastError(), EPERM);
    
    return -1;
}

int mprotect(void *addr, size_t len, int prot)
{
    DWORD newProtect = __map_mmap_prot_page(prot);
    DWORD oldProtect = 0;
    
    if (VirtualProtect(addr, len, newProtect, &oldProtect))
        return 0;
    
    errno =  __map_mman_error(GetLastError(), EPERM);
    
    return -1;
}

int msync(void *addr, size_t len, int flags)
{
    if (FlushViewOfFile(addr, len))
        return 0;
    
    errno =  __map_mman_error(GetLastError(), EPERM);
    
    return -1;
}

int mlock(const void *addr, size_t len)
{
    if (VirtualLock((LPVOID)addr, len))
        return 0;
        
    errno =  __map_mman_error(GetLastError(), EPERM);
    
    return -1;
}

int munlock(const void *addr, size_t len)
{
    if (VirtualUnlock((LPVOID)addr, len))
        return 0;
        
    errno =  __map_mman_error(GetLastError(), EPERM);
    
    return -1;
}