include/libtorrent/invariant_check.hpp

   1 // Copyright Daniel Wallin 2004. Use, modification and distribution is
   2 // subject to the Boost Software License, Version 1.0. (See accompanying
   3 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
   4
   5 #ifndef TORRENT_INVARIANT_ACCESS_HPP_INCLUDED
   6 #define TORRENT_INVARIANT_ACCESS_HPP_INCLUDED
   7
   8 #include "libtorrent/assert.hpp"
   9
  10 namespace libtorrent
  11 {
  12
  13         class invariant_access
  14         {
  15         public:
  16                 template<class T>
  17                 static void check_invariant(T const& self)
  18                 {
  19                         self.check_invariant();
  20                 }
  21         };
  22
  23         template<class T>
  24         void check_invariant(T const& x)
  25         {
  26                 invariant_access::check_invariant(x);
  27         }
  28
  29         struct invariant_checker {};
  30
  31         template<class T>
  32         struct invariant_checker_impl : invariant_checker
  33         {
  34                 invariant_checker_impl(T const& self_)
  35                         : self(self_)
  36                 {
  37                         try
  38                         {
  39                                 check_invariant(self);
  40                         }
  41                         catch (...)
  42                         {
  43                                 TORRENT_ASSERT(false);
  44                         }
  45                 }
  46
  47                 ~invariant_checker_impl()
  48                 {
  49                         try
  50                         {
  51                                 check_invariant(self);
  52                         }
  53                         catch (...)
  54                         {
  55                                 TORRENT_ASSERT(false);
  56                         }
  57                 }
  58
  59                 T const& self;
  60         };
  61
  62         template<class T>
  63         invariant_checker_impl<T> make_invariant_checker(T const& x)
  64         {
  65                 return invariant_checker_impl<T>(x);
  66         }
  67 }
  68
  69 #if !defined NDEBUG && !defined TORRENT_DISABLE_INVARIANT_CHECKS
  70 #define INVARIANT_CHECK \
  71         invariant_checker const& _invariant_check = make_invariant_checker(*this); \
  72         (void)_invariant_check; \
  73         do {} while (false)
  74 #else
  75 #define INVARIANT_CHECK do {} while (false)
  76 #endif
  77
  78 #endif // TORRENT_INVARIANT_ACCESS_HPP_INCLUDED