granicus.if.org Git - icinga2/blob - lib/icinga/user.cpp

   1 /* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
   2
   3 #include "icinga/user.hpp"
   4 #include "icinga/user-ti.cpp"
   5 #include "icinga/usergroup.hpp"
   6 #include "icinga/notification.hpp"
   7 #include "icinga/usergroup.hpp"
   8 #include "base/objectlock.hpp"
   9 #include "base/exception.hpp"
  10
  11 using namespace icinga;
  12
  13 REGISTER_TYPE(User);
  14
  15 void User::OnConfigLoaded()
  16 {
  17         ObjectImpl<User>::OnConfigLoaded();
  18
  19         SetTypeFilter(FilterArrayToInt(GetTypes(), Notification::GetTypeFilterMap(), ~0));
  20         SetStateFilter(FilterArrayToInt(GetStates(), Notification::GetStateFilterMap(), ~0));
  21 }
  22
  23 void User::OnAllConfigLoaded()
  24 {
  25         ObjectImpl<User>::OnAllConfigLoaded();
  26
  27         UserGroup::EvaluateObjectRules(this);
  28
  29         Array::Ptr groups = GetGroups();
  30
  31         if (groups) {
  32                 groups = groups->ShallowClone();
  33
  34                 ObjectLock olock(groups);
  35
  36                 for (const String& name : groups) {
  37                         UserGroup::Ptr ug = UserGroup::GetByName(name);
  38
  39                         if (ug)
  40                                 ug->ResolveGroupMembership(this, true);
  41                 }
  42         }
  43 }
  44
  45 void User::Stop(bool runtimeRemoved)
  46 {
  47         ObjectImpl<User>::Stop(runtimeRemoved);
  48
  49         Array::Ptr groups = GetGroups();
  50
  51         if (groups) {
  52                 ObjectLock olock(groups);
  53
  54                 for (const String& name : groups) {
  55                         UserGroup::Ptr ug = UserGroup::GetByName(name);
  56
  57                         if (ug)
  58                                 ug->ResolveGroupMembership(this, false);
  59                 }
  60         }
  61 }
  62
  63 void User::AddGroup(const String& name)
  64 {
  65         boost::mutex::scoped_lock lock(m_UserMutex);
  66
  67         Array::Ptr groups = GetGroups();
  68
  69         if (groups && groups->Contains(name))
  70                 return;
  71
  72         if (!groups)
  73                 groups = new Array();
  74
  75         groups->Add(name);
  76 }
  77
  78 TimePeriod::Ptr User::GetPeriod() const
  79 {
  80         return TimePeriod::GetByName(GetPeriodRaw());
  81 }
  82
  83 void User::ValidateStates(const Lazy<Array::Ptr>& lvalue, const ValidationUtils& utils)
  84 {
  85         ObjectImpl<User>::ValidateStates(lvalue, utils);
  86
  87         int filter = FilterArrayToInt(lvalue(), Notification::GetStateFilterMap(), 0);
  88
  89         if (filter == -1 || (filter & ~(StateFilterUp | StateFilterDown | StateFilterOK | StateFilterWarning | StateFilterCritical | StateFilterUnknown)) != 0)
  90                 BOOST_THROW_EXCEPTION(ValidationError(this, { "states" }, "State filter is invalid."));
  91 }
  92
  93 void User::ValidateTypes(const Lazy<Array::Ptr>& lvalue, const ValidationUtils& utils)
  94 {
  95         ObjectImpl<User>::ValidateTypes(lvalue, utils);
  96
  97         int filter = FilterArrayToInt(lvalue(), Notification::GetTypeFilterMap(), 0);
  98
  99         if (filter == -1 || (filter & ~(NotificationDowntimeStart | NotificationDowntimeEnd | NotificationDowntimeRemoved |
 100                 NotificationCustom | NotificationAcknowledgement | NotificationProblem | NotificationRecovery |
 101                 NotificationFlappingStart | NotificationFlappingEnd)) != 0)
 102                 BOOST_THROW_EXCEPTION(ValidationError(this, { "types" }, "Type filter is invalid."));
 103 }