blob: bc63125bdeeb5808064c2bd102e5a6c930d20d49 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
#include "extension_manager.h"
#include "extension_final.h"
#include "extension_whoami.h"
namespace NMVP::NOIDC {
TExtensionManager::TExtensionManager(const TActorId sender,
const TOpenIdConnectSettings& settings,
const TCrackedPage& protectedPage,
const TString authHeader)
: Settings(settings)
, AuthHeader(std::move(authHeader))
{
ExtensionCtx = MakeIntrusive<TExtensionContext>();
ExtensionCtx->Params = MakeHolder<TProxiedResponseParams>();
ExtensionCtx->Params->ProtectedPage = MakeHolder<TCrackedPage>(protectedPage);
ExtensionCtx->Sender = sender;
Timeout = settings.DefaultRequestTimeout;
}
void TExtensionManager::SetExtensionTimeout(TDuration timeout) {
Timeout = timeout;
}
void TExtensionManager::SetRequest(NHttp::THttpIncomingRequestPtr request) {
ExtensionCtx->Params->Request = std::move(request);
}
void TExtensionManager::SetOverrideResponse(NHttp::TEvHttpProxy::TEvHttpIncomingResponse::TPtr event) {
ExtensionCtx->Params->HeadersOverride = MakeHolder<NHttp::THeadersBuilder>();
ExtensionCtx->Params->ResponseError = event->Get()->GetError();
if (!event->Get()->Response)
return;
auto& response = event->Get()->Response;
ExtensionCtx->Params->StatusOverride = response->Status;
auto headers = NHttp::THeaders(response->Headers);
for (const auto& header : headers.Headers) {
ExtensionCtx->Params->HeadersOverride->Set(header.first, header.second);
}
ExtensionCtx->Params->MessageOverride = response->Message;
ExtensionCtx->Params->BodyOverride = response->Body;
}
void TExtensionManager::AddExtensionWhoami() {
auto ext = std::make_unique<TExtensionWhoami>(Settings, AuthHeader, Timeout);
AddExtension(std::move(ext));
}
void TExtensionManager::AddExtensionFinal() {
auto ext = std::make_unique<TExtensionFinal>(Settings);
AddExtension(std::move(ext));
}
void TExtensionManager::AddExtension(std::unique_ptr<IExtension> ext) {
ExtensionCtx->Steps.push(std::move(ext));
}
bool TExtensionManager::NeedExtensionWhoami(const NHttp::THttpIncomingRequestPtr& request) const {
if (!Settings.EnabledExtensionWhoami() || request->Method == "OPTIONS") {
return false;
}
TCrackedPage page(request);
auto path = TStringBuf(page.Url).Before('?');
for (const auto& whoamiPath : Settings.WHOAMI_PATHS) {
if (path.EndsWith(whoamiPath)) {
return true;
}
}
return false;
}
void TExtensionManager::ArrangeExtensions(const NHttp::THttpIncomingRequestPtr& request) {
if (NeedExtensionWhoami(request)) {
AddExtensionWhoami();
}
AddExtensionFinal();
}
void TExtensionManager::StartExtensionProcess(NHttp::THttpIncomingRequestPtr request,
NHttp::TEvHttpProxy::TEvHttpIncomingResponse::TPtr event) {
SetRequest(std::move(request));
SetOverrideResponse(std::move(event));
const auto step = ExtensionCtx->Steps.Next();
step->Execute(std::move(ExtensionCtx));
}
} // NMVP::NOIDC
|
