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
|
/*
* Copyright (c) 2024 Vaughn Nugent
*
* Library: VNLib
* Package: ObjectCacheServer
* File: WellKnownEndpoint.cs
*
* WellKnownEndpoint.cs is part of ObjectCacheServer which is part of the larger
* VNLib collection of libraries and utilities.
*
* ObjectCacheServer is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* ObjectCacheServer 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see https://www.gnu.org/licenses/.
*/
using System;
using System.Net;
using VNLib.Hashing;
using VNLib.Hashing.IdentityUtility;
using VNLib.Plugins;
using VNLib.Plugins.Essentials;
using VNLib.Plugins.Essentials.Endpoints;
using VNLib.Plugins.Essentials.Extensions;
using VNLib.Plugins.Extensions.Loading;
using VNLib.Data.Caching.Extensions;
using VNLib.Data.Caching.Extensions.Clustering;
namespace VNLib.Data.Caching.ObjectCache.Server.Endpoints
{
/*
* The well-known endpoint is used to advertise the node's existence to
* the network. Clients need to know the endpoint layout to be able to
* connect and discover other nodes.
*/
internal sealed class WellKnownEndpoint : ResourceEndpointBase
{
//Store serialized advertisment
private readonly CacheNodeAdvertisment _advertisment;
private readonly ICacheAuthManager _keyStore;
//Loosen up security requirements
protected override ProtectionSettings EndpointProtectionSettings { get; } = new()
{
DisableSessionsRequired = true,
};
public WellKnownEndpoint(PluginBase plugin)
{
//Get the node config
ObjectCacheSystemState conf = plugin.GetOrCreateSingleton<ObjectCacheSystemState>();
//serialize the config, discovery may not be enabled
_advertisment = conf.NodeConfig.Advertisment;
_keyStore = conf.KeyStore;
InitPathAndLog(conf.ClusterConfig.WellKnownPath, plugin.Log);
}
protected override VfReturnType Get(HttpEntity entity)
{
string entropy = RandomHash.GetRandomBase32(16);
//Create jwt signed for the client
using JsonWebToken jwt = new();
jwt.WriteHeader(_keyStore.GetJwtHeader());
//Write the advertisment as the message body
jwt.InitPayloadClaim()
.AddClaim("sub", _advertisment)
.AddClaim("chl", entropy)
.AddClaim("iat", entity.RequestedTimeUtc.ToUnixTimeSeconds())
.CommitClaims();
_keyStore.SignJwt(jwt);
//Return the advertisment jwt
entity.CloseResponse(HttpStatusCode.OK, Net.Http.ContentType.Text, jwt.DataBuffer);
return VfReturnType.VirtualSkip;
}
}
}
|