proxmox-mirrors/pve-http-server
1
0
Fork 0
mirror of https://git.proxmox.com/git/pve-http-server synced 2025-05-30 08:18:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

websocket_proxy: pull out encoding into helper sub

Browse Source 
to allow adding other types of frames.

Signed-off-by: Fabian Grünbichler <f.gruenbichler@proxmox.com>
This commit is contained in:
Fabian Grünbichler 2020-03-06 11:20:28 +01:00 committed by Thomas Lamprecht
parent c55702c39b
commit c3bd22ec46
1 changed files with 28 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
PVE/APIServer/AnyEvent.pm
View File

@ -380,6 +380,33 @@ sub websocket_proxy {
die "websocket_proxy: missing port or socket\n"; die "websocket_proxy: missing port or socket\n";
} }
my $encode = sub {
my ($data, $opcode) = @_;
my $string;
my $payload;
if ($binary) {
$string = $opcode ? $opcode : "\x82"; # binary frame
$payload = $$data;
} else {
$string = $opcode ? $opcode : "\x81"; # text frame
$payload = encode_base64($$data, '');
}
my $payload_len = length($payload);
if ($payload_len <= 125) {
$string .= pack 'C', $payload_len;
} elsif ($payload_len <= 0xffff) {
$string .= pack 'C', 126;
$string .= pack 'n', $payload_len;
} else {
$string .= pack 'C', 127;
$string .= pack 'Q>', $payload_len;
}
$string .= $payload;
return $string;
};
tcp_connect $remhost, $remport, sub { tcp_connect $remhost, $remport, sub {
my ($fh) = @_ my ($fh) = @_
or die "connect to '$remhost:$remport' failed: $!"; or die "connect to '$remhost:$remport' failed: $!";
@ -414,28 +441,7 @@ sub websocket_proxy {
my $len = length($hdl->{rbuf}); my $len = length($hdl->{rbuf});
my $data = substr($hdl->{rbuf}, 0, $len > $max_payload_size ? $max_payload_size : $len, ''); my $data = substr($hdl->{rbuf}, 0, $len > $max_payload_size ? $max_payload_size : $len, '');
my $string; my $string = $encode->(\$data);
my $payload;
if ($binary) {
$string = "\x82"; # binary frame
$payload = $data;
} else {
$string = "\x81"; # text frame
$payload = encode_base64($data, '');
}
my $payload_len = length($payload);
if ($payload_len <= 125) {
$string .= pack 'C', $payload_len;
} elsif ($payload_len <= 0xffff) {
$string .= pack 'C', 126;
$string .= pack 'n', $payload_len;
} else {
$string .= pack 'C', 127;
$string .= pack 'Q>', $payload_len;
}
$string .= $payload;
$reqstate->{hdl}->push_write($string) if $reqstate->{hdl}; $reqstate->{hdl}->push_write($string) if $reqstate->{hdl};
}; };