// FileExistDlg.h : ヘッダーファイル  
//
#pragma once

// CFileExistDlg ダイアログ
class CFileExistDlg : public CDialog  
{
// ・
// ・
// ・
protected:
    virtual void OnOK();  
    virtual void OnCancel();  

    void PreEndDialog();                //ダイアログ終了前処理
    BOOL GetSID(PSID *ppSid);           //現在のユーザのセキュリティ識別子取得
    BOOL WriteEvent(WORD wType,         //イベントログへの書込み
                    WORD wCategory,  
                    DWORD dwEventID,  
                    WORD wNumStrings,  
                    DWORD dwDataSize,  
                    LPCTSTR *lpStrings,  
                    LPVOID lpRawData);  

public:
    afx_msg void OnBnClickedBtnExec();  //検索ボタンクリック時

protected:
    DECLARE_MESSAGE_MAP()
};

// FileExistDlg.cpp : 実装ファイル  
//
#include "stdafx.h"
#include "FileExist.h"
#include "FileExistDlg.h"
#include "..\\MyApp\\MyAppMsg.h"

#define     EVENT_SRC_NAME     "MyApp" //イベントソース名


// CFileExistDlg ダイアログ  
// ・
// ・
// ・

BEGIN_MESSAGE_MAP(CFileExistDlg, CDialog)
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    //}}AFX_MSG_MAP
    ON_BN_CLICKED(IDC_BTN_EXEC, &CFileExistDlg::OnBnClickedBtnExec)  
END_MESSAGE_MAP()

// CFileExistDlg 初期化
BOOL CFileExistDlg::OnInitDialog()
{
    CDialog::OnInitDialog();
// ・
// ・

    if (! WriteEvent(EVENTLOG_INFORMATION_TYPE,  
                     APP_ACTION,
                     APP_OPEN,
                     0,
                     0,
                     NULL,
                     NULL)
    )
        TRACE(_T("Can't Write it in the Eventlog\n"));

    return TRUE;  
}

// ・
// ・
// ・

//検索ボタンクリック
void CFileExistDlg::OnBnClickedBtnExec()
{
    CString     strPath;  
    CString     strResult;  
    DWORD       dwAttri   = 0;  
    int         nExistFlg = 0; //1:あり   0:エラー    -1:なし

    //パス取得
    GetDlgItem(IDC_ED_FILEPATH)->GetWindowTextW(strPath);

    if (strPath.IsEmpty())  
        return;

    //ファイル検索
    dwAttri = ::GetFileAttributes(strPath);  
    if (dwAttri == (DWORD)-1)  
    {
        nExistFlg = -1;  
    } else  
    {
        dwAttri & FILE_ATTRIBUTE_DIRECTORY ? nExistFlg = -1 : nExistFlg = 1;  
    }

    //結果表示
    strResult = strPath;  
    nExistFlg == 1 ? strResult += _T("が見つかりました")  
                   : strResult += _T("がみつかりません");
    GetDlgItem(IDC_STC_RESULT)->SetWindowTextW(strResult);

    //イベントログへの書込み
    LPCTSTR     pInstStr = (PCWSTR)strPath;  

    if (! WriteEvent(nExistFlg == 1 ? EVENTLOG_AUDIT_SUCCESS : EVENTLOG_AUDIT_FAILURE,
                     USR_ACTION,
                     nExistFlg == 1 ? FILE_FIND : FILE_NOT_FOUND,  
                     1,
                     sizeof(DWORD),
                     &pInstStr,
                     &dwAttri)
    )
        TRACE(_T("Can't Write Eventlog\n"));
}

//OnOKオーバーライド
void CFileExistDlg::OnOK()
{
    PreEndDialog();
    CDialog::OnOK();
}

//OnCancelオーバーライド
void CFileExistDlg::OnCancel()
{
    PreEndDialog();
    CDialog::OnCancel();
}

//ダイアログ終了前処理 (ダイアログ終了をイベントログへ書込む)
void CFileExistDlg::PreEndDialog()
{
    if (! WriteEvent(EVENTLOG_INFORMATION_TYPE,  
                     APP_ACTION,
                     APP_CLOSE,
                     0,
                     0,
                     NULL,
                     NULL)
    )
        TRACE( _T("Can't Write it in the Eventlog\n") );
}

//セキュリティ識別子(SID)取得
BOOL CFileExistDlg::GetSID(PSID *ppSid)  
{
    CString          strUser;  
    DWORD            dwUserSize   = 0;  
    BOOL             bRet         = FALSE;  
    DWORD            dwSidSize    = 0;  
    DWORD            dwDomainSize = 0;  
    CString          strDomain;  
    PSID             pSid         = NULL;  
    SID_NAME_USE     snu;  

    //ユーザ名取得
    ::GetUserName(NULL, &dwUserSize);  
    if ( ::GetLastError() != ERROR_INSUFFICIENT_BUFFER )  
        return FALSE;  

    bRet = ::GetUserName(strUser.GetBuffer(dwUserSize), &dwUserSize);  
    strUser.ReleaseBuffer();

    if (! bRet)  
        return FALSE;  

    //SID取得
    ::LookupAccountName(NULL, strUser, NULL, &dwSidSize, NULL, &dwDomainSize, &snu);  
    if ( ::GetLastError() != ERROR_INSUFFICIENT_BUFFER )  
        return FALSE;  

    pSid = (PSID) new BYTE[dwSidSize];  

    bRet = ::LookupAccountName(NULL, strUser, pSid, &dwSidSize,  
                               strDomain.GetBuffer(dwDomainSize), &dwDomainSize, &snu);
    strDomain.ReleaseBuffer();

    if (! bRet)  
        delete [] pSid;  
    else
        *ppSid = pSid;  

    return bRet;  
}

//イベントログへの書込み
BOOL CFileExistDlg::WriteEvent(WORD wType,  
                               WORD wCategory,  
                               DWORD dwEventID,  
                               WORD wNumStrings,  
                               DWORD dwDataSize,  
                               LPCTSTR *lpStrings,  
                               LPVOID lpRawData)  
{
    PSID       pSid   = NULL;  
    HANDLE     hEvent = 0;  
    BOOL       bRet   = FALSE;  

    do
    {
        //セキュリティ識別子取得
        if (! GetSID(&pSid) )  
            break;

        //イベントソースのハンドル取得
        hEvent = ::RegisterEventSource( NULL, _T(EVENT_SRC_NAME) );  
        if (! hEvent)  
            break;

        //イベントログへの書込み
        if (! ::ReportEvent(hEvent, wType, wCategory, dwEventID, pSid,  
                            wNumStrings, dwDataSize, lpStrings, lpRawData)  
        )
            break;

        bRet = TRUE;  
    } while(0);  

    if (pSid)   { delete [] pSid;                      pSid = NULL;}  
    if (hEvent) { ::DeregisterEventSource(hEvent);     hEvent = 0;}  

    return bRet;  
}