mirror of https://github.com/Icinga/icinga2.git
74 lines
2.0 KiB
C++
74 lines
2.0 KiB
C++
#include "i2-icinga.h"
|
|
|
|
using namespace icinga;
|
|
|
|
string AuthenticationComponent::GetName(void) const
|
|
{
|
|
return "authenticationcomponent";
|
|
}
|
|
|
|
void AuthenticationComponent::Start(void)
|
|
{
|
|
m_AuthenticationEndpoint = make_shared<VirtualEndpoint>();
|
|
m_AuthenticationEndpoint->RegisterMethodHandler("auth::SetIdentity", bind_weak(&AuthenticationComponent::IdentityMessageHandler, shared_from_this()));
|
|
m_AuthenticationEndpoint->RegisterMethodSource("auth::Welcome");
|
|
|
|
EndpointManager::Ptr mgr = GetEndpointManager();
|
|
mgr->OnNewEndpoint += bind_weak(&AuthenticationComponent::NewEndpointHandler, shared_from_this());
|
|
mgr->ForeachEndpoint(bind(&AuthenticationComponent::NewEndpointHandler, this, _1));
|
|
mgr->RegisterEndpoint(m_AuthenticationEndpoint);
|
|
}
|
|
|
|
void AuthenticationComponent::Stop(void)
|
|
{
|
|
EndpointManager::Ptr mgr = GetEndpointManager();
|
|
|
|
if (mgr)
|
|
mgr->UnregisterEndpoint(m_AuthenticationEndpoint);
|
|
}
|
|
|
|
int AuthenticationComponent::NewEndpointHandler(const NewEndpointEventArgs& neea)
|
|
{
|
|
if (neea.Endpoint->IsLocal() || neea.Endpoint->HasIdentity())
|
|
return 0;
|
|
|
|
neea.Endpoint->AddAllowedMethodSinkPrefix("auth::");
|
|
neea.Endpoint->AddAllowedMethodSourcePrefix("auth::");
|
|
|
|
neea.Endpoint->RegisterMethodSink("auth::SetIdentity");
|
|
|
|
JsonRpcRequest request;
|
|
request.SetMethod("auth::SetIdentity");
|
|
|
|
IdentityMessage params;
|
|
params.SetIdentity("keks");
|
|
request.SetParams(params);
|
|
|
|
GetEndpointManager()->SendUnicastRequest(m_AuthenticationEndpoint, neea.Endpoint, request);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int AuthenticationComponent::IdentityMessageHandler(const NewRequestEventArgs& nrea)
|
|
{
|
|
Message params;
|
|
if (!nrea.Request.GetParams(¶ms))
|
|
return 0;
|
|
|
|
IdentityMessage identityMessage = params;
|
|
|
|
string identity;
|
|
if (!identityMessage.GetIdentity(&identity))
|
|
return 0;
|
|
|
|
nrea.Sender->SetIdentity(identity);
|
|
|
|
/* there's no authentication for now, just tell them it's ok to send messages */
|
|
JsonRpcRequest request;
|
|
request.SetMethod("auth::Welcome");
|
|
|
|
GetEndpointManager()->SendUnicastRequest(m_AuthenticationEndpoint, nrea.Sender, request);
|
|
|
|
return 0;
|
|
}
|