[英]How can I put a custom header for websocket handshake with boost beast
[英]How to read the header of a bad request in beast::websocket?
我有一個應該響應 GET 請求的 websocket 服務器(監聽環回接口)。 這是我如何做到的
ws_.async_accept_ex(
[self = shared_from_this()](websocket::response_type& res) //
{
if (res.result_int() == 400) { // bad request, assume HTTP GET
osstream response;
/* invoke request handlers */
for (auto& do_handle : self->handlers) {
do_handle({ "GET" }, response);
/* if responded, assing response to the body*/
if (response.tellp()) {
res.content_length(response.str().size());
res.body() = std::move(response.str());
break;
}
}
}
},
net::bind_executor(strand_,
[self = shared_from_this()](beast::error_code ec) {
self->on_accept(ec);
}));
但我還需要能夠處理位置(如localhost:2019/some_location
)。
有辦法嗎?
解決方法:手動讀取header,檢查是否升級
beast::flat_buffer buffer_;
websocket::request_type req_;
/*
...
...
*/
http::async_read(ws_.next_layer(), buffer_, req_,
[self=shared_from_this()](beast::error_code ec, size_t)
{
if(ec){
fail(ec, "[http] read");
return;
}
if(websocket::is_upgrade(self->req_)){
self->ws_.accept(self->req_,ec);
self->on_accept(ec);
}
else{
osstream response;
websocket::response_type res;
vector<string_view> req_args
{ "GET", self->req_.base().target().begin() };
/* invoke request handlers */
for(auto &do_handle : self->handlers){
do_handle(req_args, response);
/* if responded, assing response to the body*/
if(response.tellp()){
res.content_length(response.str().size());
res.body() = std::move(response.str());
break;
}}
http::write(self->ws_.next_layer(), res);
}
});
WebSocket 升級請求的請求目標必須采用原始格式。 “localhost:2019/some_location”是絕對形式的,因此是非法的。 在我看來,您想要的是能夠專門處理不是 WebSocket 升級的 HTTP 請求,而對於實際的 WebSocket 升級,讓 websocket 流通過執行握手來處理它。
這在文檔中有介紹: https : //www.boost.org/doc/libs/1_69_0/libs/beast/doc/html/beast/using_websocket/handshaking_servers.html#beast.using_websocket.handshaking_servers.passing_http_requests
同樣在示例中: https : //github.com/boostorg/beast/blob/06efddd8b851610b5b3a5832ac87f1c52b838d9b/example/advanced/server/advanced_server.cpp#L665
tl;dr:使用beast::http::async_read
自己閱讀請求,看看它是否是使用beast::websocket::is_upgrade
的 websocket 升級。 如果是升級,構建beast::websocket::stream
並使用請求調用async_accept
,否則按照您想要的方式處理 HTTP 請求並使用beast::http::async_write
自己發回響應。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.