github_mirrors/openappsec
3
0
Fork 1
mirror of https://github.com/openappsec/openappsec.git synced 2025-06-28 16:41:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

identifying playground clusters

Browse Source
This commit is contained in:
Roy Barda 2022-11-13 12:32:29 +00:00
parent af5056511e
commit 2b00b93046
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
components/security_apps/orchestration/k8s_policy_gen/k8s_policy_gen.cc
View File

@ -13,9 +13,12 @@
#include "k8s_policy_gen.h" #include "k8s_policy_gen.h"
#include <algorithm>
#include <cctype>
#include <map> #include <map>
#include <set> #include <set>
#include <string> #include <string>
#include <iostream>
#include <fstream> #include <fstream>
#include <streambuf> #include <streambuf>
#include <cereal/types/vector.hpp> #include <cereal/types/vector.hpp>
@ -854,9 +857,28 @@ private:
string token; string token;
map<string, string> practice_name_to_id_map; map<string, string> practice_name_to_id_map;
bool
isPlaygroundEnv()
{
string playground_variable = "PLAYGROUND";
const char* env_string = getenv(playground_variable.c_str());
if (env_string)
{
string env_value = env_string;
std::transform(env_value.begin(), env_value.end(), env_value.begin(),
[](unsigned char c){ return std::tolower(c); });
return env_value == "true";
}
return false;
}
bool bool
getClusterId() getClusterId()
{ {
string playground_uid = isPlaygroundEnv() ? "playground-" : "";
dbgTrace(D_K8S_POLICY) << "Getting cluster UID"; dbgTrace(D_K8S_POLICY) << "Getting cluster UID";
NamespaceData namespaces_data; NamespaceData namespaces_data;
bool res = messaging->sendObject( bool res = messaging->sendObject(
@ -885,7 +907,7 @@ private:
uid, uid,
EnvKeyAttr::LogSection::SOURCE EnvKeyAttr::LogSection::SOURCE
); );
Singleton::Consume<I_AgentDetails>::by<K8sPolicyGenerator::Impl>()->setClusterId(uid); Singleton::Consume<I_AgentDetails>::by<K8sPolicyGenerator::Impl>()->setClusterId(playground_uid + uid);
return true; return true;
} }
} }