mirror of
https://github.com/oven-sh/bun
synced 2026-02-16 22:01:47 +00:00
76545140af
### What does this PR do? handle socket upgrade in NodeHTTP.cpp ### How did you verify your code works? Run the test added with asan it should catch the bug --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
108 lines
3.5 KiB
C++
108 lines
3.5 KiB
C++
/*
|
|
* Authored by Alex Hultman, 2018-2020.
|
|
* Intellectual property of third-party.
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef UWS_WEBSOCKETCONTEXTDATA_H
|
|
#define UWS_WEBSOCKETCONTEXTDATA_H
|
|
|
|
#include "Loop.h"
|
|
#include "AsyncSocket.h"
|
|
|
|
#include "MoveOnlyFunction.h"
|
|
#include <string_view>
|
|
#include <vector>
|
|
|
|
#include "WebSocketProtocol.h"
|
|
#include "TopicTree.h"
|
|
#include "WebSocketData.h"
|
|
|
|
namespace uWS {
|
|
|
|
/* Type queued up when publishing */
|
|
struct TopicTreeMessage {
|
|
std::string message;
|
|
/*OpCode*/ int opCode;
|
|
bool compress;
|
|
};
|
|
struct TopicTreeBigMessage {
|
|
std::string_view message;
|
|
/*OpCode*/ int opCode;
|
|
bool compress;
|
|
};
|
|
|
|
template <bool, bool, typename> struct WebSocket;
|
|
|
|
/* todo: this looks identical to WebSocketBehavior, why not just std::move that entire thing in? */
|
|
|
|
template <bool SSL, typename USERDATA>
|
|
struct WebSocketContextData {
|
|
private:
|
|
|
|
public:
|
|
/* This one points to the App's shared topicTree */
|
|
TopicTree<TopicTreeMessage, TopicTreeBigMessage> *topicTree;
|
|
|
|
/* The callbacks for this context */
|
|
MoveOnlyFunction<void(WebSocket<SSL, true, USERDATA> *)> openHandler = nullptr;
|
|
MoveOnlyFunction<void(WebSocket<SSL, true, USERDATA> *, std::string_view, OpCode)> messageHandler = nullptr;
|
|
MoveOnlyFunction<void(WebSocket<SSL, true, USERDATA> *)> drainHandler = nullptr;
|
|
MoveOnlyFunction<void(WebSocket<SSL, true, USERDATA> *, std::string_view, int, int)> subscriptionHandler = nullptr;
|
|
MoveOnlyFunction<void(WebSocket<SSL, true, USERDATA> *, int, std::string_view)> closeHandler = nullptr;
|
|
MoveOnlyFunction<void(WebSocket<SSL, true, USERDATA> *, std::string_view)> pingHandler = nullptr;
|
|
MoveOnlyFunction<void(WebSocket<SSL, true, USERDATA> *, std::string_view)> pongHandler = nullptr;
|
|
|
|
/* Settings for this context */
|
|
size_t maxPayloadLength = 0;
|
|
|
|
/* We do need these for async upgrade */
|
|
CompressOptions compression;
|
|
|
|
/* There needs to be a maxBackpressure which will force close everything over that limit */
|
|
size_t maxBackpressure = 0;
|
|
bool closeOnBackpressureLimit;
|
|
bool resetIdleTimeoutOnSend;
|
|
bool sendPingsAutomatically;
|
|
unsigned short maxLifetime;
|
|
|
|
/* These are calculated on creation */
|
|
std::pair<unsigned short, unsigned short> idleTimeoutComponents;
|
|
|
|
/* This is run once on start-up */
|
|
void calculateIdleTimeoutComponents(unsigned short idleTimeout) {
|
|
unsigned short margin = 4;
|
|
/* 4, 8 or 16 seconds margin based on idleTimeout */
|
|
while ((int) idleTimeout - margin * 2 >= margin * 2 && margin < 16) {
|
|
margin = (unsigned short) (margin << 1);
|
|
}
|
|
idleTimeoutComponents = {
|
|
idleTimeout - (sendPingsAutomatically ? margin : 0), /* reduce normal idleTimeout if it is extended by ping-timeout */
|
|
margin /* ping-timeout - also used for end() timeout */
|
|
};
|
|
}
|
|
|
|
~WebSocketContextData() {
|
|
|
|
}
|
|
|
|
WebSocketContextData(TopicTree<TopicTreeMessage, TopicTreeBigMessage> *topicTree) : topicTree(topicTree) {
|
|
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif // UWS_WEBSOCKETCONTEXTDATA_H
|