library/cpp/messagebus/use_after_free_checker.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

#pragma once 
 
#include <util/system/platform.h> 
#include <util/system/types.h>
 
class TUseAfterFreeChecker { 
private: 
    ui64 Magic; 

public: 
    TUseAfterFreeChecker(); 
    ~TUseAfterFreeChecker(); 
    void CheckNotFreed() const; 
}; 
 
// check twice: in constructor and in destructor 
class TUseAfterFreeCheckerGuard { 
private: 
    const TUseAfterFreeChecker& Check; 

public: 
    TUseAfterFreeCheckerGuard(const TUseAfterFreeChecker& check) 
        : Check(check) 
    { 
        Check.CheckNotFreed(); 
    } 
 
    ~TUseAfterFreeCheckerGuard() { 
        Check.CheckNotFreed(); 
    } 
};