LibNewDel.cpp

[詳解]

//=============================================================================
/// @file
/// 生成消滅演算子ライブラリ実装ファイル
///
/// 生成消滅演算子ライブラリ実装ファイルです。
///
/// $Id: LibNewDel.cpp 245 2019-03-20 15:03:41Z admin $
/// $Date: 2019-03-21 00:03:41 +0900 (2019/03/21 (木)) $
/// $Author: admin $
///
/// @attention なし

#pragma managed( push, off )

//=============================================================================
// インクルードファイル
#include <LibNewDel.h>
#include <CSyncAuto.h>
#include <CProcess.h>
#include <LibLogOut.h>
#include <LibUtility.h>

//=============================================================================
// インクルード実装ファイル
#include <LibCommon.hpp>

//=============================================================================
// 生成消滅演算子ライブラリ名前空間
namespace LibNewDel {
    //=========================================================================
    // 生成消滅演算子ライブラリクラス
    //=========================================================================
    // 構築子と解体子
    //-------------------------------------------------------------------------
    // コンストラクタ
    CLibNewDel::CLibNewDel() noexcept
        // 基底クラスコンストラクタ
        : CMapping()
        // メンバ変数初期化
        , m_psMappingInfo( nullptr )
        , m_bLogOut( false )
    {
        // コンストラクタ実行通知
        NotifyConstructor();

        // 処理ブロック
        bool result = false;
        do {
            // マッピングメモリを作成する
            if ( !Create( sizeof( SMappingInfo ), OBJECT_NAME ) ) {
                // 失敗！
                break;
            }
            // 同期処理ブロック
            else do {
                CSyncAuto cSyncAuto( *this );

                // 共有メモリ情報構造体ポインタを取得する
                m_psMappingInfo = reinterpret_cast< SMappingInfo* >( GetView() );
                if ( nullptr == m_psMappingInfo ) {
                    // 致命的エラー！
                    NotifyFatalError();
                    break;
                }

                // プロセス情報を巡回する
                DWORD dwProcessId = ::GetCurrentProcessId();
                for ( int nIndex = 0; PROCESS_MAX > nIndex; ++nIndex ) {
                    // プロセスIDを調べる
                    if ( 0 == m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId ) {
                        // プロセスIDを設定する
                        m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId = dwProcessId;

                        // 終了！
                        break;
                    }
                }

                // 参照カウントをインクリメントする
                ++m_psMappingInfo->m_nReffer;

                // 成功！
                result = true;
            } while ( false );
        } while ( false );

        // 実行結果を調べる
        if ( !result ) {
            // マッピングをクローズする
            CMapping::Close();
        }
    }

    //-------------------------------------------------------------------------
    // デストラクタ
    CLibNewDel::~CLibNewDel() noexcept {
        // デストラクタ実行通知
        NotifyDestructor();

        // 同期処理ブロック
        do {
            CSyncAuto cSyncAuto( *this );

            // プロセス情報を更新する
            UpdateProcessInfo( true );
        } while ( false );
    }

    //=========================================================================
    // 公開関数
    //-------------------------------------------------------------------------
    // 生成演算子実行通知関数
    void CLibNewDel::NotifyOperatorNew( void* pAddr, size_t uSize, wchar_t const* pszFuncName ) noexcept {
        // 同期処理ブロック
        do {
            CSyncAuto cSyncAuto( *this );

            // ログ出力許可フラグを調べる
            if ( m_bLogOut ) {
                // プロセス情報を更新する
                UpdateProcessInfo();
            }

            // 生成消滅カウントをインクリメントする
            ++m_psMappingInfo->m_nCount;

            // トータルサイズに加算する
            m_psMappingInfo->m_nTotalSize += uSize;

            // 最大サイズを更新する
            if ( m_psMappingInfo->m_nMaxSize < m_psMappingInfo->m_nTotalSize ) {
                m_psMappingInfo->m_nMaxSize = m_psMappingInfo->m_nTotalSize;
            }

            // ログ出力する
            LogOutOperatorNewDelete( m_psMappingInfo->m_nCount, pAddr, uSize, m_psMappingInfo->m_nTotalSize, m_psMappingInfo->m_nMaxSize, pszFuncName );

            // プロセス情報を巡回する
            DWORD dwProcessId = ::GetCurrentProcessId();
            for ( int nIndex = 0;; ++nIndex ) {
                // インデックスを調べる
                if ( PROCESS_MAX <= nIndex ) {
                    // 致命的エラー！
                    NotifyFatalError();
                    break;
                }
                // プロセスIDを調べる
                else if ( dwProcessId == m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId ) {
                    // プロセス単位の生成消滅演算子カウントをインクリメントする
                    ++m_psMappingInfo->m_sProcess[ nIndex ].m_nCount;

                    // プロセス単位の生成消滅メモリサイズに加算する
                    m_psMappingInfo->m_sProcess[ nIndex ].m_nTotalSize += uSize;
                    break;
                }
            }
        } while ( false );
    }

    //-------------------------------------------------------------------------
    // 消滅演算子実行通知関数
    void CLibNewDel::NotifyOperatorDelete( void* pAddr, size_t uSize, wchar_t const* pszFuncName ) noexcept {
        // 同期処理ブロック
        do {
            CSyncAuto cSyncAuto( *this );

            // ログ出力許可フラグを調べる
            if ( m_bLogOut ) {
                // プロセス情報を更新する
                UpdateProcessInfo();
            }

            // トータルサイズから減算する
            m_psMappingInfo->m_nTotalSize -= uSize;

            // ログ出力する
            LogOutOperatorNewDelete( m_psMappingInfo->m_nCount, pAddr, uSize, m_psMappingInfo->m_nTotalSize, m_psMappingInfo->m_nMaxSize, pszFuncName );

            // 生成消滅カウントを調べる
            if ( 0 == m_psMappingInfo->m_nCount ) {
                // 致命的エラー！
                NotifyFatalError();
                break;
            }

            // 生成消滅カウントをデクリメントする
            --m_psMappingInfo->m_nCount;

            // プロセス情報を巡回する
            DWORD dwProcessId = ::GetCurrentProcessId();
            for ( int nIndex = 0;; ++nIndex ) {
                // インデックスを調べる
                if ( PROCESS_MAX <= nIndex ) {
                    // 致命的エラー！
                    NotifyFatalError();
                    break;
                }
                // プロセスIDを調べる
                else if ( dwProcessId == m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId ) {
                    // プロセス単位の生成消滅演算子カウントをデクリメントする
                    --m_psMappingInfo->m_sProcess[ nIndex ].m_nCount;

                    // プロセス単位の生成消滅メモリサイズから減算する
                    m_psMappingInfo->m_sProcess[ nIndex ].m_nTotalSize -= uSize;
                    break;
                }
            }
        } while ( false );
    }

    //=========================================================================
    // 限定公開関数
    //-------------------------------------------------------------------------
    // プロセス情報更新関数
    bool CLibNewDel::UpdateProcessInfo( bool bCurrent ) noexcept {
        // 同期処理ブロック
        bool result = false;
        do {
            CSyncAuto cSyncAuto( *this );

            // プロセス情報を巡回する
            DWORD dwProcessId = ::GetCurrentProcessId();
            for ( int nIndex = 0;; ++nIndex ) {
                // インデックスを調べる
                if ( PROCESS_MAX <= nIndex ) {
                    // 成功！
                    result = true;
                    break;
                }
                // プロセスIDを調べる
                else if ( 0 == m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId ) {
                    // 継続！
                    continue;
                }
                // プロセスIDを調べる
                else if ( dwProcessId == m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId ) {
                    // カレントプロセス削除フラグを調べる
                    if ( !bCurrent ) {
                        // 継続！
                        continue;
                    }
                }
                else {
                    // プロセスをオープンする
                    CProcess cProcess;
                    if ( cProcess.Open( m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId ) ) {
                        // プロセス実行状態を調べる
                        if ( !cProcess.Wait( 0 ) ) {
                            // プロセスをクローズする
                            cProcess.Close();

                            // 継続！
                            continue;
                        }

                        // プロセスをクローズする
                        cProcess.Close();
                    }
                }

                // 生成消滅カウントを減算する
                m_psMappingInfo->m_nCount -= m_psMappingInfo->m_sProcess[ nIndex ].m_nCount;

                // トータルメモリサイズを減算する
                m_psMappingInfo->m_nTotalSize -= m_psMappingInfo->m_sProcess[ nIndex ].m_nTotalSize;

                // 参照カウントを調べる
                if ( 0 == m_psMappingInfo->m_nReffer ) {
                    // 致命的エラー！
                    NotifyFatalError();
                    break;
                }

                // 参照カウントをデクリメントする
                wchar_t const* pszMessage = L"";
                if ( 0 == --m_psMappingInfo->m_nReffer ) {
                    // 生成消滅演算子情報更新ログを出力する
                    pszMessage = L", All Processes Termination";
                }

                // プロセス終了生成消滅演算子ログを出力する
                LogOutNoIndent( L"NewDel#%d, Total = %lld, Max = %lld, By Terminated Process[%04X], Count = %d, Size = %lld%s."
                    , m_psMappingInfo->m_nCount
                    , m_psMappingInfo->m_nTotalSize
                    , m_psMappingInfo->m_nMaxSize
                    , m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId
                    , m_psMappingInfo->m_sProcess[ nIndex ].m_nCount
                    , m_psMappingInfo->m_sProcess[ nIndex ].m_nTotalSize
                    , pszMessage );

                // プロセス情報をクリアする
                m_psMappingInfo->m_sProcess[ nIndex ].m_dwProcessId = 0;
                m_psMappingInfo->m_sProcess[ nIndex ].m_nCount      = 0;
                m_psMappingInfo->m_sProcess[ nIndex ].m_nTotalSize  = 0;
            }
        } while ( false );

        // 実行結果を返す
        return result;
    }

    //-------------------------------------------------------------------------
    // 生成消滅演算子ログ出力関数
    void CLibNewDel::LogOutOperatorNewDelete( int nCount, void* pAddr, size_t uSize, LONGLONG nTotalSize, LONGLONG nMaxSize, wchar_t const* pszFuncName ) noexcept {
        // ログ出力許可フラグを調べる
        if ( m_bLogOut ) {
            // インデントなしでログ出力する
            LogOutNoIndent( L"NewDel#%d, Total = %lld, Max = %lld, Addr = 0x%p, Size = %zu, By %s!%s.", nCount, nTotalSize, nMaxSize, pAddr, uSize, LogOutExeFile(), pszFuncName );
        }
    }

    //=========================================================================
    // 静的公開関数
    //-------------------------------------------------------------------------
    // プロジェクト種別取得関数
    wchar_t const* CLibNewDel::GetProjectType() noexcept {
        // プロジェクト種別文字列を返す
        return _PROJECT_TYPE;
    }

    //-------------------------------------------------------------------------
    // インスタンスハンドル取得関数
    HINSTANCE CLibNewDel::GetInstanceHandle() noexcept {
#if defined( _WINDLL ) && !defined( _MANAGED )

        // インスタンスハンドルを返す
        return s_hInstance;

#       else

        // 実行モジュールのインスタンスハンドルを返す
        return ::GetModuleHandleW( nullptr );

#       endif
    }

    //=========================================================================
    // 静的限定公開文字列定数
    wchar_t const* const CLibNewDel::OBJECT_NAME = L"LibNewDel::CLibNewDel";

    //=========================================================================
    // 静的限定公開変数
    CLibNewDel CLibNewDel::s_cInstance;
}

#pragma managed( pop )