]> granicus.if.org Git - icinga2/blobdiff - lib/icinga/usergroup.cpp
Correct current_concurrent_checks to actually running checks
[icinga2] / lib / icinga / usergroup.cpp
index f122b231c715c7aa8f96ee1ad8aea250a6ff1529..913882d1cfe9b2c6038ec989846d509fc9f34d56 100644 (file)
@@ -1,24 +1,7 @@
-/******************************************************************************
- * Icinga 2                                                                   *
- * Copyright (C) 2012-2015 Icinga Development Team (http://www.icinga.org)    *
- *                                                                            *
- * This program is free software; you can redistribute it and/or              *
- * modify it under the terms of the GNU General Public License                *
- * as published by the Free Software Foundation; either version 2             *
- * of the License, or (at your option) any later version.                     *
- *                                                                            *
- * This program is distributed in the hope that it will be useful,            *
- * but WITHOUT ANY WARRANTY; without even the implied warranty of             *
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              *
- * GNU General Public License for more details.                               *
- *                                                                            *
- * You should have received a copy of the GNU General Public License          *
- * along with this program; if not, write to the Free Software Foundation     *
- * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA.             *
- ******************************************************************************/
+/* Icinga 2 | (c) 2012 Icinga GmbH | GPLv2+ */
 
 #include "icinga/usergroup.hpp"
-#include "icinga/usergroup.tcpp"
+#include "icinga/usergroup-ti.cpp"
 #include "config/objectrule.hpp"
 #include "config/configitem.hpp"
 #include "base/configtype.hpp"
 #include "base/logger.hpp"
 #include "base/context.hpp"
 #include "base/workqueue.hpp"
-#include <boost/foreach.hpp>
 
 using namespace icinga;
 
 REGISTER_TYPE(UserGroup);
 
-INITIALIZE_ONCE(&UserGroup::RegisterObjectRuleHandler);
-
-void UserGroup::RegisterObjectRuleHandler(void)
-{
+INITIALIZE_ONCE([]() {
        ObjectRule::RegisterType("UserGroup");
-}
+});
 
 bool UserGroup::EvaluateObjectRule(const User::Ptr& user, const ConfigItem::Ptr& group)
 {
-       String group_name = group->GetName();
+       String groupName = group->GetName();
 
-       CONTEXT("Evaluating rule for group '" + group_name + "'");
+       CONTEXT("Evaluating rule for group '" + groupName + "'");
 
-       ScriptFrame frame;
+       ScriptFrame frame(true);
        if (group->GetScope())
                group->GetScope()->CopyTo(frame.Locals);
        frame.Locals->Set("user", user);
@@ -54,10 +33,12 @@ bool UserGroup::EvaluateObjectRule(const User::Ptr& user, const ConfigItem::Ptr&
                return false;
 
        Log(LogDebug, "UserGroup")
-           << "Assigning membership for group '" << group_name << "' to user '" << user->GetName() << "'";
+               << "Assigning membership for group '" << groupName << "' to user '" << user->GetName() << "'";
 
        Array::Ptr groups = user->GetGroups();
-       groups->Add(group_name);
+
+       if (groups && !groups->Contains(groupName))
+               groups->Add(groupName);
 
        return true;
 }
@@ -66,7 +47,7 @@ void UserGroup::EvaluateObjectRules(const User::Ptr& user)
 {
        CONTEXT("Evaluating group membership for user '" + user->GetName() + "'");
 
-       BOOST_FOREACH(const ConfigItem::Ptr& group, ConfigItem::GetItems("UserGroup"))
+       for (const ConfigItem::Ptr& group : ConfigItem::GetItems(UserGroup::TypeInstance))
        {
                if (!group->GetFilter())
                        continue;
@@ -75,7 +56,7 @@ void UserGroup::EvaluateObjectRules(const User::Ptr& user)
        }
 }
 
-std::set<User::Ptr> UserGroup::GetMembers(void) const
+std::set<User::Ptr> UserGroup::GetMembers() const
 {
        boost::mutex::scoped_lock lock(m_UserGroupMutex);
        return m_Members;
@@ -99,8 +80,8 @@ bool UserGroup::ResolveGroupMembership(const User::Ptr& user, bool add, int rsta
 
        if (add && rstack > 20) {
                Log(LogWarning, "UserGroup")
-                   << "Too many nested groups for group '" << GetName() << "': User '"
-                   << user->GetName() << "' membership assignment failed.";
+                       << "Too many nested groups for group '" << GetName() << "': User '"
+                       << user->GetName() << "' membership assignment failed.";
 
                return false;
        }
@@ -110,7 +91,7 @@ bool UserGroup::ResolveGroupMembership(const User::Ptr& user, bool add, int rsta
        if (groups && groups->GetLength() > 0) {
                ObjectLock olock(groups);
 
-               BOOST_FOREACH(const String& name, groups) {
+               for (const String& name : groups) {
                        UserGroup::Ptr group = UserGroup::GetByName(name);
 
                        if (group && !group->ResolveGroupMembership(user, add, rstack + 1))