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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
// Copyright Epic Games, Inc. All Rights Reserved.
#include "casstore.h"
#include <zencore/fmtutils.h>
#include <zencore/logging.h>
#include <gsl/gsl-lite.hpp>
namespace zen {
HttpCasService::HttpCasService(CasStore& Store) : m_CasStore(Store)
{
m_Router.AddPattern("cas", "([0-9A-Fa-f]{40})");
m_Router.RegisterRoute(
"{cas}",
[this](HttpRouterRequest& Req) {
IoHash Hash = IoHash::FromHexString(Req.GetCapture(1));
ZEN_DEBUG("CAS request for {}", Hash);
HttpServerRequest& ServerRequest = Req.ServerRequest();
switch (ServerRequest.RequestVerb())
{
case HttpVerb::kGet:
case HttpVerb::kHead:
{
if (IoBuffer Value = m_CasStore.FindChunk(Hash))
{
return ServerRequest.WriteResponse(HttpResponseCode::OK, HttpContentType::kBinary, Value);
}
return ServerRequest.WriteResponse(HttpResponseCode::NotFound);
}
break;
case HttpVerb::kPut:
{
IoBuffer Payload = ServerRequest.ReadPayload();
IoHash PayloadHash = IoHash::HashBuffer(Payload.Data(), Payload.Size());
// URI hash must match content hash
if (PayloadHash != Hash)
{
return ServerRequest.WriteResponse(HttpResponseCode::BadRequest);
}
m_CasStore.InsertChunk(Payload, PayloadHash);
return ServerRequest.WriteResponse(HttpResponseCode::OK);
}
break;
default:
break;
}
},
HttpVerb::kGet | HttpVerb::kPut | HttpVerb::kHead);
}
const char*
HttpCasService::BaseUri() const
{
return "/cas/";
}
void
HttpCasService::HandleRequest(zen::HttpServerRequest& Request)
{
if (Request.RelativeUri().empty())
{
// Root URI request
switch (Request.RequestVerb())
{
case HttpVerb::kPut:
case HttpVerb::kPost:
{
IoBuffer Payload = Request.ReadPayload();
IoHash PayloadHash = IoHash::HashBuffer(Payload.Data(), Payload.Size());
ZEN_DEBUG("CAS POST request for {} ({} bytes)", PayloadHash, Payload.Size());
auto InsertResult = m_CasStore.InsertChunk(Payload, PayloadHash);
if (InsertResult.New)
{
return Request.WriteResponse(HttpResponseCode::Created);
}
else
{
return Request.WriteResponse(HttpResponseCode::OK);
}
}
break;
case HttpVerb::kGet:
case HttpVerb::kHead:
break;
default:
break;
}
}
else
{
m_Router.HandleRequest(Request);
}
}
} // namespace zen
|