#pragma once

#ifdef PSWMOUSEHOOKDLL_API_EXPORTS
#define PSWMOUSEHOOKDLL_API __declspec(dllexport)
#else
#define PSWMOUSEHOOKDLL_API __declspec(dllimport)
#endif

/*
Winty:2009-09-29
调用SetHook()设置Hook，但无需卸载Hook，DLLMain中有清理。
*/
//My Declaration================================
#define MAXCOUNT 200 //密码最大长度
#define DWDATA_PSW_NOTIFY 1 //COPYDATASTRUCT的dwData自定义值

//密码信息结构体，用于发送
typedef struct tagPswNotify
{
    char psw[MAXCOUNT];//password/text
    POINT pt;//鼠标位置
    HWND hWnd;//控件句柄
} PSWNOTIFY , *PPSWNOTIFY;

/*extern表示这里只是变量声明，变量定义在cpp文件中*/
extern PSWMOUSEHOOKDLL_API BOOL g_bReadySend;//WM_COPYDATA互斥的标志，因为WM_COPYDATA不能重叠
extern PSWMOUSEHOOKDLL_API HWND g_hWnd; //接收消息的窗体句柄
extern PSWMOUSEHOOKDLL_API HHOOK g_hhk; //钩子句柄
extern PSWMOUSEHOOKDLL_API BOOL g_bView;//是否需要查看密码

//鼠标钩子过程
PSWMOUSEHOOKDLL_API LRESULT CALLBACK MouseProc(int nCode, WPARAM wParam, LPARAM lParam );
//设置钩子(可在窗口的OnInitDialog()中调用)
PSWMOUSEHOOKDLL_API void SetHook(HWND hWnd);

#include "stdafx.h"

#define PSWMOUSEHOOKDLL_API_EXPORTS
#include "PasswordViewerMouseHookDLL.h"

#include <stdio.h>

//进程间共享数据,还要在链接器选项里添加:"/SECTION:.MyShare,RWS"
//或: #pragma   comment(linker,"/SECTION:.MyShare,RWS") 
//查看结果:dumpbin /headers *.DLL
#pragma data_seg(".MyShare")
HHOOK g_hhk = NULL;/*Hook句柄*/
HWND  g_hWnd = NULL;/*接收消息的窗口句柄*/
BOOL  g_bReadySend = TRUE;/*用于同步COPYDATA消息*/
#pragma data_seg()
#pragma   comment(linker,"/SECTION:.MyShare,RWS")

BOOL APIENTRY DllMain( HANDLE hModule,
                       DWORD  ul_reason_for_call,
                       LPVOID lpReserved
                     )
{
    switch (ul_reason_for_call)
    {
        case DLL_PROCESS_ATTACH:
            break;
        case DLL_PROCESS_DETACH:
            if(g_hhk != NULL)
            {
                UnhookWindowsHookEx(g_hhk);
                g_hhk = NULL;
                g_hWnd = NULL;
            }
            break;
        case DLL_THREAD_ATTACH:
        case DLL_THREAD_DETACH:
            break;

    }
    return TRUE;
}

LRESULT CALLBACK MouseProc(
                           int nCode,      // hook code
                           WPARAM wParam,  // message identifier
                           LPARAM lParam   // mouse coordinates
)
{
    if(nCode == HC_ACTION)
    {
        PMOUSEHOOKSTRUCT pMouseHookStruct
            = reinterpret_cast<PMOUSEHOOKSTRUCT>(lParam);
        LONG x = pMouseHookStruct->pt.x;
        LONG y = pMouseHookStruct->pt.y;
        HWND hWnd = pMouseHookStruct->hwnd;

        HWND hWndFromPoint = ::WindowFromPoint(pMouseHookStruct->pt);

        if(hWndFromPoint != g_hWnd  &&  g_bReadySend)//不能获取显示密码的文本框的内容
        {
            g_bReadySend = FALSE;
            char psw[MAXCOUNT];
            ::GetWindowText(hWndFromPoint , psw , MAXCOUNT);
            PSWNOTIFY pswNotify;
            strcpy(pswNotify.psw , psw);
            pswNotify.pt.x = x;
            pswNotify.pt.y = y;
            pswNotify.hWnd = hWndFromPoint;

            COPYDATASTRUCT cd;
            cd.lpData = &pswNotify;
            cd.cbData = sizeof(PSWNOTIFY);
            cd.dwData = DWDATA_PSW_NOTIFY;
            ::SendMessage(g_hWnd, WM_COPYDATA, NULL , (LPARAM)(LPVOID)&cd);
        }
    }

    return CallNextHookEx(g_hhk , nCode , wParam ,lParam);
}

void SetHook(HWND hWnd)
{
    g_hWnd = hWnd;
    g_hhk = SetWindowsHookEx(WH_MOUSE ,
                             MouseProc ,
                             GetModuleHandle("PasswordViewerMouseHookDLL") ,
                             NULL);
}

#pragma comment(linker,"/manifestdependency:\"type='win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='x86' publicKeyToken='6595b64144ccf1df' language='*'\"")

BOOL CPasswordViewerDlg::OnCopyData(CWnd* pWnd, COPYDATASTRUCT* pCopyDataStruct)
{
    if(pCopyDataStruct->dwData == DWDATA_PSW_NOTIFY && !g_bReadySend)
    {
        PPSWNOTIFY pPswNotify = (PPSWNOTIFY)pCopyDataStruct->lpData;

        CString strMousePosition;
        strMousePosition.Format("(%3d,%3d)" , pPswNotify->pt.x ,pPswNotify->pt.y);
        m_strMousePosition = strMousePosition;
        CString strPsw;
        strPsw.Format("%s" , pPswNotify->psw);
        m_strPsw = strPsw;
        CString strHWnd;
        strHWnd.Format("%p" , pPswNotify->hWnd);
        m_strHWnd = strHWnd;
        UpdateData(FALSE);

        g_bReadySend = TRUE;

        return TRUE;
    }
    else
    {
        CString str(_T("未发现窗口"));
        m_strPsw = str;

        UpdateData(FALSE);

        return CDialog::OnCopyData(pWnd, pCopyDataStruct);
    }   
}