add warp command in proxy

This commit is contained in:
Martin Michelsen
2022-03-28 22:30:13 -07:00
parent 04aef91c16
commit b61a9bcdcb
3 changed files with 29 additions and 9 deletions
+14 -9
View File
@@ -927,18 +927,19 @@ void ProxyServer::LinkedSession::on_server_input() {
}
}
void ProxyServer::LinkedSession::send_to_end(const string& data, bool to_server) {
string name = string_printf("ProxySession:%08" PRIX32 ":shell:%s",
this->license->serial_number, to_server ? "server" : "client");
void ProxyServer::LinkedSession::send_to_end(
const void* data, size_t size, bool to_server) {
size_t header_size = PSOCommandHeader::header_size(this->version);
if (data.size() < header_size) {
if (size < header_size) {
throw runtime_error("command is too small for header");
}
if (data.size() & 3) {
if (size & 3) {
throw runtime_error("command size is not a multiple of 4");
}
const auto* header = reinterpret_cast<const PSOCommandHeader*>(data.data());
const auto* header = reinterpret_cast<const PSOCommandHeader*>(data);
string name = string_printf("ProxySession:%08" PRIX32 ":shell:%s",
this->license->serial_number, to_server ? "server" : "client");
send_command(
to_server ? this->server_bev.get() : this->client_bev.get(),
@@ -946,11 +947,15 @@ void ProxyServer::LinkedSession::send_to_end(const string& data, bool to_server)
to_server ? this->server_output_crypt.get() : this->client_output_crypt.get(),
header->command(this->version),
header->flag(this->version),
data.data() + header_size,
data.size() - header_size,
reinterpret_cast<const uint8_t*>(data) + header_size,
size - header_size,
name.c_str());
}
void ProxyServer::LinkedSession::send_to_end(const string& data, bool to_server) {
this->send_to_end(data.data(), data.size(), to_server);
}
shared_ptr<ProxyServer::LinkedSession> ProxyServer::get_session() {
if (this->serial_number_to_session.empty()) {
throw runtime_error("no sessions exist");