Fix the WebSocket client to work with none PVE servers.
[pve-client.git] / PVE / APIClient / Commands / lxc.pm
index afa7e99..e6ebb6c 100644 (file)
@@ -8,6 +8,7 @@ use IO::Select;
 use IO::Socket::SSL;
 use MIME::Base64;
 use Digest::SHA;
+use HTTP::Response;
 
 use PVE::Tools;
 use PVE::JSONSchema qw(get_standard_option);
@@ -20,7 +21,7 @@ my $CRLF = "\x0D\x0A";
 my $max_payload_size = 65536;
 
 my $build_web_socket_request = sub {
-    my ($path, $ticket, $termproxy) = @_;
+    my ($host, $path, $ticket, $termproxy) = @_;
 
     my $key = '';
     $key .= chr(int(rand(256))) for 1 .. 16;
@@ -35,6 +36,7 @@ my $build_web_socket_request = sub {
     my $request = "GET $path HTTP/1.1$CRLF"
        . "Upgrade: WebSocket$CRLF"
        . "Connection: Upgrade$CRLF"
+       . "Host: $host$CRLF"
        . "Sec-WebSocket-Key: $enckey$CRLF"
        . "Sec-WebSocket-Version: 13$CRLF"
        . "Sec-WebSocket-Protocol: binary$CRLF"
@@ -44,7 +46,6 @@ my $build_web_socket_request = sub {
     return ($request, $enckey);
 };
 
-# FIXME: share this code with websockt code in PVE/APIServer/AnyEvent.pm ?
 my $create_websockt_frame = sub {
     my ($payload) = @_;
 
@@ -66,9 +67,8 @@ my $create_websockt_frame = sub {
     return $string;
 };
 
-# FIXME: share this code with websockt code in PVE/APIServer/AnyEvent.pm ?
 my $parse_web_socket_frame = sub  {
-    my ($wsbuf_ref, $binary, $is_server) = @_;
+    my ($wsbuf_ref) = @_;
 
     my $wsbuf = $$wsbuf_ref;
 
@@ -90,11 +90,7 @@ my $parse_web_socket_frame = sub  {
        my $payload_len = unpack 'C', substr($wsbuf, 1, 1);
 
        my $masked = $payload_len & 0b10000000;
-       if ($is_server) {
-           die "received unmasked websocket frame from client\n" if !$masked;
-       } else {
-           die "received masked websocket frame from server\n" if $masked;
-       }
+       die "received masked websocket frame from server\n" if $masked;
 
        my $offset = 2;
        $payload_len = $payload_len & 0b01111111;
@@ -111,34 +107,12 @@ my $parse_web_socket_frame = sub  {
        die "received too large websocket frame (len = $payload_len)\n"
            if ($payload_len > $max_payload_size) || ($payload_len < 0);
 
-       my @mask = (0, 0, 0, 0);
-       if ($masked) {
-           last if $len < $offset + 4;
-
-           @mask = (unpack('C', substr($wsbuf, $offset+0, 1)),
-                    unpack('C', substr($wsbuf, $offset+1, 1)),
-                    unpack('C', substr($wsbuf, $offset+2, 1)),
-                    unpack('C', substr($wsbuf, $offset+3, 1)));
-
-           $offset += 4;
-       }
-
        last if $len < ($offset + $payload_len);
 
        my $data = substr($wsbuf, 0, $offset + $payload_len, ''); # now consume data
 
        my $frame_data = substr($data, $offset, $payload_len);
 
-       if ($masked) {
-           for (my $i = 0; $i < $payload_len; $i++) {
-               my $d = unpack('C', substr($frame_data, $i, 1));
-               my $n = $d ^ $mask[$i % 4];
-               substr($frame_data, $i, 1, pack('C', $n));
-           }
-       }
-
-       $frame_data = decode_base64($frame_data) if !$binary;
-
        $payload = '' if !defined($payload);
        $payload .= $frame_data;
 
@@ -174,7 +148,8 @@ __PACKAGE__->register_method ({
     code => sub {
        my ($param) = @_;
 
-       my $conn = PVE::APIClient::Config::get_remote_connection($param->{remote});
+       my $config = PVE::APIClient::Config->load();
+       my $conn = PVE::APIClient::Config->remote_conn($config, $param->{remote});
 
        # Get the real node from the resources endpoint
        my $resource_list = $conn->get("api2/json/cluster/resources", { type => 'vm'});
@@ -199,7 +174,7 @@ __PACKAGE__->register_method ({
        # WebSocket Handshake
 
        my ($request, $wskey) = $build_web_socket_request->(
-           "/$api_path/vncwebsocket", $conn->{ticket}, $termproxy);
+           $conn->{host}, "/$api_path/vncwebsocket", $conn->{ticket}, $termproxy);
 
        $web_socket->syswrite($request);
 
@@ -231,8 +206,8 @@ __PACKAGE__->register_method ({
 
        die "got invalid websocket reponse: $raw_response\n"
            if !(($response->code == 101) &&
-                ($response->header('connection') eq 'upgrade') &&
-                ($response->header('upgrade') eq 'websocket') &&
+                (lc $response->header('connection') eq 'upgrade') &&
+                (lc $response->header('upgrade') eq 'websocket') &&
                 ($response->header('sec-websocket-protocol') eq 'binary') &&
                 ($response->header('sec-websocket-accept') eq $wsaccept));
 
@@ -249,7 +224,7 @@ __PACKAGE__->register_method ({
            foreach my $fh (@ready) {
                if ($fh == $web_socket) {
                    my $nr = $wb_socket_read_available_bytes->();
-                   my ($payload, $req_close) = $parse_web_socket_frame->(\$wsbuf, 1);
+                   my ($payload, $req_close) = $parse_web_socket_frame->(\$wsbuf);
                    print "GOT: $payload\n" if defined($payload);
                    last if $req_close;
                    last if !$nr; # eos