mirror of
				https://github.com/Icinga/icinga2.git
				synced 2025-11-03 21:25:56 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			104 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			104 lines
		
	
	
		
			2.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
 | 
						|
 | 
						|
#include "icinga/user.hpp"
 | 
						|
#include "icinga/user-ti.cpp"
 | 
						|
#include "icinga/usergroup.hpp"
 | 
						|
#include "icinga/notification.hpp"
 | 
						|
#include "icinga/usergroup.hpp"
 | 
						|
#include "base/objectlock.hpp"
 | 
						|
#include "base/exception.hpp"
 | 
						|
 | 
						|
using namespace icinga;
 | 
						|
 | 
						|
REGISTER_TYPE(User);
 | 
						|
 | 
						|
void User::OnConfigLoaded()
 | 
						|
{
 | 
						|
	ObjectImpl<User>::OnConfigLoaded();
 | 
						|
 | 
						|
	SetTypeFilter(FilterArrayToInt(GetTypes(), Notification::GetTypeFilterMap(), ~0));
 | 
						|
	SetStateFilter(FilterArrayToInt(GetStates(), Notification::GetStateFilterMap(), ~0));
 | 
						|
}
 | 
						|
 | 
						|
void User::OnAllConfigLoaded()
 | 
						|
{
 | 
						|
	ObjectImpl<User>::OnAllConfigLoaded();
 | 
						|
 | 
						|
	UserGroup::EvaluateObjectRules(this);
 | 
						|
 | 
						|
	Array::Ptr groups = GetGroups();
 | 
						|
 | 
						|
	if (groups) {
 | 
						|
		groups = groups->ShallowClone();
 | 
						|
 | 
						|
		ObjectLock olock(groups);
 | 
						|
 | 
						|
		for (const String& name : groups) {
 | 
						|
			UserGroup::Ptr ug = UserGroup::GetByName(name);
 | 
						|
 | 
						|
			if (ug)
 | 
						|
				ug->ResolveGroupMembership(this, true);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void User::Stop(bool runtimeRemoved)
 | 
						|
{
 | 
						|
	ObjectImpl<User>::Stop(runtimeRemoved);
 | 
						|
 | 
						|
	Array::Ptr groups = GetGroups();
 | 
						|
 | 
						|
	if (groups) {
 | 
						|
		ObjectLock olock(groups);
 | 
						|
 | 
						|
		for (const String& name : groups) {
 | 
						|
			UserGroup::Ptr ug = UserGroup::GetByName(name);
 | 
						|
 | 
						|
			if (ug)
 | 
						|
				ug->ResolveGroupMembership(this, false);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
void User::AddGroup(const String& name)
 | 
						|
{
 | 
						|
	boost::mutex::scoped_lock lock(m_UserMutex);
 | 
						|
 | 
						|
	Array::Ptr groups = GetGroups();
 | 
						|
 | 
						|
	if (groups && groups->Contains(name))
 | 
						|
		return;
 | 
						|
 | 
						|
	if (!groups)
 | 
						|
		groups = new Array();
 | 
						|
 | 
						|
	groups->Add(name);
 | 
						|
}
 | 
						|
 | 
						|
TimePeriod::Ptr User::GetPeriod() const
 | 
						|
{
 | 
						|
	return TimePeriod::GetByName(GetPeriodRaw());
 | 
						|
}
 | 
						|
 | 
						|
void User::ValidateStates(const Lazy<Array::Ptr>& lvalue, const ValidationUtils& utils)
 | 
						|
{
 | 
						|
	ObjectImpl<User>::ValidateStates(lvalue, utils);
 | 
						|
 | 
						|
	int filter = FilterArrayToInt(lvalue(), Notification::GetStateFilterMap(), 0);
 | 
						|
 | 
						|
	if (filter == -1 || (filter & ~(StateFilterUp | StateFilterDown | StateFilterOK | StateFilterWarning | StateFilterCritical | StateFilterUnknown)) != 0)
 | 
						|
		BOOST_THROW_EXCEPTION(ValidationError(this, { "states" }, "State filter is invalid."));
 | 
						|
}
 | 
						|
 | 
						|
void User::ValidateTypes(const Lazy<Array::Ptr>& lvalue, const ValidationUtils& utils)
 | 
						|
{
 | 
						|
	ObjectImpl<User>::ValidateTypes(lvalue, utils);
 | 
						|
 | 
						|
	int filter = FilterArrayToInt(lvalue(), Notification::GetTypeFilterMap(), 0);
 | 
						|
 | 
						|
	if (filter == -1 || (filter & ~(NotificationDowntimeStart | NotificationDowntimeEnd | NotificationDowntimeRemoved |
 | 
						|
		NotificationCustom | NotificationAcknowledgement | NotificationProblem | NotificationRecovery |
 | 
						|
		NotificationFlappingStart | NotificationFlappingEnd)) != 0)
 | 
						|
		BOOST_THROW_EXCEPTION(ValidationError(this, { "types" }, "Type filter is invalid."));
 | 
						|
}
 |