4#include "Stroika/Frameworks/StroikaPreComp.h"
9#include "Stroika/Foundation/Characters/FloatConversion.h"
11#include "Stroika/Foundation/Characters/String2Int.h"
13#include "Stroika/Foundation/Containers/Common.h"
14#include "Stroika/Foundation/DataExchange/BadFormatException.h"
20#include "Stroika/Foundation/Execution/Throw.h"
23#include "Stroika/Foundation/IO/Network/HTTP/ClientErrorException.h"
24#include "Stroika/Foundation/IO/Network/HTTP/Headers.h"
26#include "Stroika/Foundation/IO/Network/HTTP/Methods.h"
30#include "Connection.h"
38using namespace Stroika::Foundation::Memory;
39using namespace Stroika::Foundation::Traversal;
40using namespace Stroika::Foundation::Time;
42using namespace Stroika::Frameworks;
58 const Headers& defaultResponseHeaders,
const optional<bool> autoComputeETagResponse)
59 :
Message{
Request{socketStream}, Response{socket, socketStream, defaultResponseHeaders}, socket.GetPeerAddress ()}
60 , fMsgHeaderInTextStream{HTTP::MessageStartTextInputStreamBinaryAdapter::New (rwRequest ().GetInputStream ())}
62 if (autoComputeETagResponse) {
63 this->rwResponse ().autoComputeETag = *autoComputeETagResponse;
67Connection::MyMessage_::ReadHeadersResult Connection::MyMessage_::ReadHeaders (
68#
if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
69 const function<
void (
const String&)>& logMsg
73#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
74 logMsg (
"Starting ReadHeaders_"sv);
80 if (not fMsgHeaderInTextStream.AssureHeaderSectionAvailable ()) {
81#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
82 logMsg (
"got fMsgHeaderInTextStream.AssureHeaderSectionAvailable INCOMPLETE"sv);
84 if (fMsgHeaderInTextStream.IsAtEOF (Streams::eDontBlock) ==
true) {
85 return ReadHeadersResult::eIncompleteDeadEnd;
87 return ReadHeadersResult::eIncompleteButMoreMayBeAvailable;
93 Request& updatableRequest = this->rwRequest ();
96 String line = fMsgHeaderInTextStream.ReadLine ();
98#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
99 logMsg (
"got EOF from src stream reading headers(incomplete)"sv);
101 return ReadHeadersResult::eIncompleteDeadEnd;
104 if (tokens.size () < 3) {
105 DbgTrace (
"tokens={}, line='{}', fMsgHeaderInTextStream={}"_f, tokens, line, fMsgHeaderInTextStream.ToString ());
110 if (tokens[1].empty ()) {
112 DbgTrace (
"tokens={}, line='{}'"_f, tokens, line);
116 if (updatableRequest.
httpMethod ().empty ()) {
118 DbgTrace (
"tokens={}, line='{}'"_f, tokens, line);
124 static const String kCRLF_{
"\r\n"sv};
125 String line = fMsgHeaderInTextStream.ReadLine ();
126 if (line == kCRLF_ or line.empty ()) {
131 size_t i = line.
find (
':');
132 if (i == string::npos) {
139 updatableRequest.
rwHeaders ().Add (hdr, value);
142#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
143 logMsg (
"ReadHeaders completed normally"sv);
145 return ReadHeadersResult::eCompleteGood;
167#if qStroika_Framework_WebServer_Connection_TrackExtraStats
168 if (fMostRecentMessage) {
169 sb <<
", mostRecentMessage: " << *fMostRecentMessage;
171 if (fHandlingThread) {
173 sb <<
", handlingThread: " << fHandlingThread;
176 sb <<
", thread: " << fHandlingThread;
196 const optional<Headers>& defaultGETResponseHeaders,
const optional<bool> autoComputeETagResponse)
197 :
Connection{s, Options{.fInterceptorChain = interceptorChain,
198 .fDefaultResponseHeaders = defaultResponseHeaders,
199 .fDefaultGETResponseHeaders = defaultGETResponseHeaders,
200 .fAutoComputeETagResponse = autoComputeETagResponse}}
205 :
socket{[qStroika_Foundation_Common_Property_ExtraCaptureStuff] ([[maybe_unused]] const auto* property) -> ConnectionOrientedStreamSocket::Ptr {
207 AssertExternallySynchronizedMutex::ReadContext declareContext{*thisObj};
208 return thisObj->fSocket_;
210 ,
request{[qStroika_Foundation_Common_Property_ExtraCaptureStuff] ([[maybe_unused]]
const auto* property) ->
const Request& {
212 AssertExternallySynchronizedMutex::ReadContext declareContext{*thisObj};
213 return thisObj->fMessage_->request ();
215 ,
response{[qStroika_Foundation_Common_Property_ExtraCaptureStuff] ([[maybe_unused]]
const auto* property) ->
const Response& {
217 AssertExternallySynchronizedMutex::ReadContext declareContext{*thisObj};
218 return thisObj->fMessage_->response ();
220 ,
rwResponse{[qStroika_Foundation_Common_Property_ExtraCaptureStuff] ([[maybe_unused]]
const auto* property) -> Response& {
222 AssertExternallySynchronizedMutex::WriteContext declareContext{*thisObj};
223 return thisObj->fMessage_->rwResponse ();
225 ,
stats{[qStroika_Foundation_Common_Property_ExtraCaptureStuff] ([[maybe_unused]]
const auto* property) -> Stats {
230#if qStroika_Framework_WebServer_Connection_TrackExtraStats
231 Stats2Capture_ statsCapturedDuringMessageProcessing = thisObj->fExtraStats_.
load ();
234 .fSocketID = uniqueID,
235 .fCreatedAt = createdAt,
236#if qStroika_Framework_WebServer_Connection_TrackExtraStats
237 .fMostRecentMessage = statsCapturedDuringMessageProcessing.fMessageStart
239 statsCapturedDuringMessageProcessing.fMessageCompleted}
240 : optional<Range<TimePointSeconds>>{},
241 .fHandlingThread = statsCapturedDuringMessageProcessing.fHandlingThread,
242 .fRemotePeerAddress = statsCapturedDuringMessageProcessing.fPeer,
243 .fRequestWebMethod = statsCapturedDuringMessageProcessing.fWebMethod,
244 .fRequestURI = statsCapturedDuringMessageProcessing.fRequestURI,
250 [qStroika_Foundation_Common_Property_ExtraCaptureStuff] ([[maybe_unused]]
const auto* property) -> optional<HTTP::KeepAlive> {
252 AssertExternallySynchronizedMutex::ReadContext declareContext{*thisObj};
253 return thisObj->fRemaining_;
255 [qStroika_Foundation_Common_Property_ExtraCaptureStuff] ([[maybe_unused]]
auto* property,
const auto&
remainingConnectionLimits) {
257 AssertExternallySynchronizedMutex::WriteContext declareContext{*thisObj};
260 , fInterceptorChain_{options.fInterceptorChain}
261 , fDefaultResponseHeaders_{options.fDefaultResponseHeaders}
262 , fDefaultGETResponseHeaders_{options.fDefaultGETResponseHeaders}
263 , fAutoComputeETagResponse_{options.fAutoComputeETagResponse}
264 , fSupportedCompressionEncodings_{options.fSupportedCompressionEncodings}
266 , fConnectionStartedAt_{Time::GetTickCount ()}
268 Require (s !=
nullptr);
269#if USE_NOISY_TRACE_IN_THIS_MODULE_
270 DbgTrace (
"Created connection for socket {}"_f, s);
272 fSocketStream_ = SocketStream::New (fSocket_);
273#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
275 String socketName =
"{}-{}"_f((
long)DateTime::Now ().As<time_t> (), (
int)s.GetNativeSocket ());
276 fSocketStream_ = Streams::LoggingInputOutputStream<byte>::New (
282 Streams::TextToBinary::Writer::Format::eUTF8WithoutBOM);
287Connection::~Connection ()
289#if USE_NOISY_TRACE_IN_THIS_MODULE_
290 DbgTrace (
"Destroying connection for socket {}, message={}"_f, fSocket_,
static_cast<const void*
> (fMessage_.get ()));
292 AssertExternallySynchronizedMutex::WriteContext declareContext{*
this};
293#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
294 WriteLogConnectionMsg_ (L
"DestroyingConnection");
296 if (fMessage_ !=
nullptr) {
297 if (not fMessage_->response ().responseCompleted ()) {
298 IgnoreExceptionsForCall (fMessage_->rwResponse ().Abort ());
300 Require (fMessage_->response ().responseCompleted ());
313 DbgTrace (
"Exception ignored closing socket: {}"_f, current_exception ());
319 AssertExternallySynchronizedMutex::WriteContext declareContext{*
this};
321#if USE_NOISY_TRACE_IN_THIS_MODULE_
324 fMessage_ = make_unique<MyMessage_> (fSocket_, fSocketStream_, fDefaultResponseHeaders_, fAutoComputeETagResponse_);
325#if qStroika_Foundation_Debug_AssertExternallySynchronizedMutex_Enabled
326 fMessage_->SetAssertExternallySynchronizedMutexContext (GetSharedContext ());
329 auto readHeaders = [&] () -> optional<ReadAndProcessResult> {
330 switch (fMessage_->ReadHeaders (
331#
if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
332 [
this] (
const String& i) ->
void { WriteLogConnectionMsg_ (i); }
335 case MyMessage_::eIncompleteDeadEnd: {
336 DbgTrace (
"ReadHeaders failed (socket {}) - incomplete data read from client."_f,
340 case MyMessage_::eIncompleteButMoreMayBeAvailable: {
341 DbgTrace (
"ReadHeaders failed - incomplete header (most likely a DOS attack)."_f);
342 return ReadAndProcessResult::eTryAgainLater;
344 case MyMessage_::eCompleteGood: {
354 if (
auto r = readHeaders ()) {
359#if qStroika_Foundation_Debug_AssertionsChecked
360 [[maybe_unused]]
auto&& cleanup =
Finally ([&] ()
noexcept { Ensure (fMessage_->response ().responseCompleted ()); });
363#if qStroika_Framework_WebServer_Connection_TrackExtraStats
364 [[maybe_unused]]
auto&& cleanup2 =
365 Finally ([&] ()
noexcept { fExtraStats_.
rwget ().rwref ().fMessageCompleted = Time::GetTickCount (); });
366 fExtraStats_.
store (Stats2Capture_{.fMessageStart = Time::GetTickCount (),
368 .fWebMethod = fMessage_->request ().httpMethod (),
369 .fRequestURI = fMessage_->request ().url (),
370 .fHandlingThread = std::this_thread::get_id ()});
373 auto applyDefaultsToResponseHeaders = [&] () ->
void {
374 if (fDefaultGETResponseHeaders_ and fMessage_->request ().httpMethod () == HTTP::Methods::kGet) {
375 fMessage_->rwResponse ().rwHeaders () += *fDefaultGETResponseHeaders_;
378 fMessage_->rwResponse ().rwHeaders ().date = DateTime::Now ();
383 if (optional<HTTP::ContentEncodings> acceptEncoding = fMessage_->request ().headers ().acceptEncoding) {
384 optional<HTTP::ContentEncodings> oBodyEncoding = fMessage_->rwResponse ().bodyEncoding ();
386 fMessage_->rwResponse ().bodyEncoding = [&] () {
388 auto bc = *oBodyEncoding;
389 bc += contentEncoding2Add;
393 return HTTP::ContentEncodings{contentEncoding2Add};
397 bool needBodyEncoding = not oBodyEncoding.has_value ();
400 if (needBodyEncoding and acceptEncoding->Contains (ce) and
401 (fSupportedCompressionEncodings_ == nullopt or fSupportedCompressionEncodings_->Contains (ce))) {
403 needBodyEncoding =
false;
406 if constexpr (DataExchange::Compression::Deflate::kSupported) {
409 if constexpr (DataExchange::Compression::GZip::kSupported) {
412 if constexpr (DataExchange::Compression::ZStd::kSupported) {
418 if (
auto requestedINoneMatch = this->
request ().headers ().ifNoneMatch ()) {
419 if (this->
response ().autoComputeETag ()) {
420 this->
rwResponse ().automaticTransferChunkSize =
421 Response::kNoChunkedTransfer;
425 applyDefaultsToResponseHeaders ();
430 auto applyKeepAliveLogic = [&] () ->
bool {
431 bool thisMessageKeepAlive = fMessage_->request ().keepAliveRequested;
432 if (thisMessageKeepAlive) {
436 if (
auto keepAliveValue = fMessage_->request ().headers ().keepAlive ()) {
441 if (oRemaining->fMessages) {
442 if (oRemaining->fMessages == 0u) {
443 thisMessageKeepAlive =
false;
446 oRemaining->fMessages = *oRemaining->fMessages - 1u;
449 if (oRemaining->fTimeout) {
450 if (fConnectionStartedAt_ + *oRemaining->fTimeout < Time::GetTickCount ()) {
451 thisMessageKeepAlive =
false;
458 this->
rwResponse ().rwHeaders ().connection = thisMessageKeepAlive ? Headers::eKeepAlive : Headers::eClose;
459 return thisMessageKeepAlive;
461 bool thisMessageKeepAlive = applyKeepAliveLogic ();
467 auto invokeInterceptorChain = [&] () {
468#if USE_NOISY_TRACE_IN_THIS_MODULE_
471#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
472 WriteLogConnectionMsg_ (
"Handing request {} to interceptor chain"_f(
request ()));
478#if USE_NOISY_TRACE_IN_THIS_MODULE_
479 DbgTrace (
"Interceptor-Chain caught exception handling message: {}"_f, current_exception ());
481#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
482 WriteLogConnectionMsg_ (
"Interceptor-Chain caught exception handling message: {}"_f(current_exception ()));
486 invokeInterceptorChain ();
488 auto assureRequestFullyRead = [&] () {
489 if (thisMessageKeepAlive) {
502 if (
request ().headers ().contentLength ()) {
503#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
504 WriteLogConnectionMsg_ (L
"msg is keepalive, and have content length, so making sure we read all of request body");
506#if USE_NOISY_TRACE_IN_THIS_MODULE_
510 (void)fMessage_->rwRequest ().GetBody ();
514 assureRequestFullyRead ();
520 auto completeResponse = [&] () {
522 if (
auto requestedINoneMatch = this->
request ().headers ().ifNoneMatch ()) {
523 if (
auto actualETag = this->
response ().headers ().ETag ()) {
524 bool ctm = this->
response ().chunkedTransferMode ();
526 DbgTrace (
"Warning - disregarding ifNoneMatch request (though it matched) - cuz in chunked transfer mode"_f);
528 if (requestedINoneMatch->fETags.Contains (*actualETag) and not ctm) {
529 DbgTrace (
"Updating OK response to NotModified (due to ETag match)"_f);
530 this->
rwResponse ().status = HTTP::StatusCodes::kNotModified;
536 thisMessageKeepAlive =
false;
538#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
539 WriteLogConnectionMsg_ (L
"Did GetResponse ().End ()");
544 return thisMessageKeepAlive ? eTryAgainLater : eClose;
547 DbgTrace (
"ReadAndProcessMessage Exception caught ({}), so returning ReadAndProcessResult::eClose"_f, current_exception ());
549 return Connection::ReadAndProcessResult::eClose;
553#if qStroika_Framework_WebServer_Connection_DetailedMessagingLog
554void Connection::WriteLogConnectionMsg_ (
const String& msg)
const
556 String useMsg = DateTime::Now ().Format () +
" -- "sv + msg.
Trim ();
557 fLogConnectionState_.WriteLn (useMsg);
563 AssertExternallySynchronizedMutex::ReadContext declareContext{*
this};
566 sb <<
"Socket: "sv << fSocket_;
567 if (not abbreviatedOutput) {
568 sb <<
", Message: "sv << fMessage_;
569 sb <<
", Remaining: "sv << fRemaining_;
571 sb <<
", Connection-Started-At: "sv << fConnectionStartedAt_;
#define AssertNotReached()
time_point< RealtimeClock, DurationSeconds > TimePointSeconds
TimePointSeconds is a simpler approach to chrono::time_point, which doesn't require using templates e...
Similar to String, but intended to more efficiently construct a String. Mutable type (String is large...
String is like std::u32string, except it is much easier to use, often much more space efficient,...
nonvirtual size_t length() const noexcept
nonvirtual String SubString(SZ from) const
nonvirtual String Trim(bool(*shouldBeTrimmed)(Character)=Character::IsWhitespace) const
nonvirtual Containers::Sequence< String > Tokenize() const
nonvirtual size_t find(Character c, size_t startAt=0) const
A generalization of a vector: a container whose elements are keyed by the natural numbers.
nonvirtual WritableReference rwget()
get a read-write smart pointer to the underlying Synchronized<> object, holding the full lock the who...
nonvirtual void store(const T &v)
nonvirtual T load() const
nonvirtual optional< IO::Network::SocketAddress > GetPeerAddress() const
nonvirtual void Close() const
ClientErrorException is to capture exceptions caused by a bad (e.g ill-formed) request.
Common::Property< String > httpMethod
typically HTTP::Methods::kGet
Common::Property< Headers & > rwHeaders
Common::Property< URI > url
Common::Property< String > httpVersion
nonvirtual PlatformNativeHandle GetNativeSocket() const
static URI ParseRelative(const String &rawRelativeURL)
A Connection object represents the state (and socket) for an ongoing, active, HTTP Connection,...
Common::Property< optional< HTTP::KeepAlive > > remainingConnectionLimits
const Common::ReadOnlyProperty< Stats > stats
retrieve stats about this connection, like threads used, start/end times. NB: INTERNALLY SYNCRONIZED
const Common::ReadOnlyProperty< ConnectionOrientedStreamSocket::Ptr > socket
nonvirtual ReadAndProcessResult ReadAndProcessMessage() noexcept
const Common::ReadOnlyProperty< const Request & > request
Common::ReadOnlyProperty< Response & > rwResponse
const Common::ReadOnlyProperty< const Response & > response
nonvirtual String ToString(bool abbreviatedOutput=true) const
nonvirtual void HandleMessage(Message &m) const
this represents a HTTP request object for the WebServer module
CONTAINER::value_type * End(CONTAINER &c)
For a contiguous container (such as a vector or basic_string) - find the pointer to the end of the co...
void Throw(T &&e2Throw)
identical to builtin C++ 'throw' except that it does helpful, type dependent DbgTrace() messages firs...
auto Finally(FUNCTION &&f) -> Private_::FinallySentry< FUNCTION >
filesystem::path GetTemporary()
constexpr bool IsOK(Status s)
several status codes considered OK, so check if it is among them
Ptr New(const Streams::OutputStream::Ptr< byte > &src, const Characters::CodeCvt<> &char2OutputConverter)
Content coding values indicate an encoding transformation that has been or can be applied to an entit...
static const ContentEncoding kDeflate
static const ContentEncoding kZStd
probably fastest/best, but NYI in Stroika as of 2024-06-20
static const ContentEncoding kGZip
optional< URI > fRequestURI
last requested URI (always relative uri)
nonvirtual String ToString() const
Socket::PlatformNativeHandle fSocketID
optional< SocketAddress > fRemotePeerAddress
the address of the client which is talking to the server
TimePointSeconds fCreatedAt
optional< String > fRequestWebMethod
last request