@novnc/novnc
Version:
An HTML5 VNC client
363 lines (359 loc) • 13.6 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
exports["default"] = void 0;
var Log = _interopRequireWildcard(require("../util/logging.js"));
var _inflator = _interopRequireDefault(require("../inflator.js"));
function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": e }; }
function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof(e) && "function" != typeof e) return { "default": e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n["default"] = e, t && t.set(e, n), n; }
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
function _classCallCheck(a, n) { if (!(a instanceof n)) throw new TypeError("Cannot call a class as a function"); }
function _defineProperties(e, r) { for (var t = 0; t < r.length; t++) { var o = r[t]; o.enumerable = o.enumerable || !1, o.configurable = !0, "value" in o && (o.writable = !0), Object.defineProperty(e, _toPropertyKey(o.key), o); } }
function _createClass(e, r, t) { return r && _defineProperties(e.prototype, r), t && _defineProperties(e, t), Object.defineProperty(e, "prototype", { writable: !1 }), e; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : i + ""; }
function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); } /*
* noVNC: HTML5 VNC client
* Copyright (C) 2019 The noVNC authors
* (c) 2012 Michael Tinglof, Joe Balaz, Les Piech (Mercuri.ca)
* Licensed under MPL 2.0 (see LICENSE.txt)
*
* See README.md for usage and integration instructions.
*
*/
var TightDecoder = exports["default"] = /*#__PURE__*/function () {
function TightDecoder() {
_classCallCheck(this, TightDecoder);
this._ctl = null;
this._filter = null;
this._numColors = 0;
this._palette = new Uint8Array(1024); // 256 * 4 (max palette size * max bytes-per-pixel)
this._len = 0;
this._zlibs = [];
for (var i = 0; i < 4; i++) {
this._zlibs[i] = new _inflator["default"]();
}
}
return _createClass(TightDecoder, [{
key: "decodeRect",
value: function decodeRect(x, y, width, height, sock, display, depth) {
if (this._ctl === null) {
if (sock.rQwait("TIGHT compression-control", 1)) {
return false;
}
this._ctl = sock.rQshift8();
// Reset streams if the server requests it
for (var i = 0; i < 4; i++) {
if (this._ctl >> i & 1) {
this._zlibs[i].reset();
Log.Info("Reset zlib stream " + i);
}
}
// Figure out filter
this._ctl = this._ctl >> 4;
}
var ret;
if (this._ctl === 0x08) {
ret = this._fillRect(x, y, width, height, sock, display, depth);
} else if (this._ctl === 0x09) {
ret = this._jpegRect(x, y, width, height, sock, display, depth);
} else if (this._ctl === 0x0A) {
ret = this._pngRect(x, y, width, height, sock, display, depth);
} else if ((this._ctl & 0x08) == 0) {
ret = this._basicRect(this._ctl, x, y, width, height, sock, display, depth);
} else {
throw new Error("Illegal tight compression received (ctl: " + this._ctl + ")");
}
if (ret) {
this._ctl = null;
}
return ret;
}
}, {
key: "_fillRect",
value: function _fillRect(x, y, width, height, sock, display, depth) {
if (sock.rQwait("TIGHT", 3)) {
return false;
}
var pixel = sock.rQshiftBytes(3);
display.fillRect(x, y, width, height, pixel, false);
return true;
}
}, {
key: "_jpegRect",
value: function _jpegRect(x, y, width, height, sock, display, depth) {
var data = this._readData(sock);
if (data === null) {
return false;
}
display.imageRect(x, y, width, height, "image/jpeg", data);
return true;
}
}, {
key: "_pngRect",
value: function _pngRect(x, y, width, height, sock, display, depth) {
throw new Error("PNG received in standard Tight rect");
}
}, {
key: "_basicRect",
value: function _basicRect(ctl, x, y, width, height, sock, display, depth) {
if (this._filter === null) {
if (ctl & 0x4) {
if (sock.rQwait("TIGHT", 1)) {
return false;
}
this._filter = sock.rQshift8();
} else {
// Implicit CopyFilter
this._filter = 0;
}
}
var streamId = ctl & 0x3;
var ret;
switch (this._filter) {
case 0:
// CopyFilter
ret = this._copyFilter(streamId, x, y, width, height, sock, display, depth);
break;
case 1:
// PaletteFilter
ret = this._paletteFilter(streamId, x, y, width, height, sock, display, depth);
break;
case 2:
// GradientFilter
ret = this._gradientFilter(streamId, x, y, width, height, sock, display, depth);
break;
default:
throw new Error("Illegal tight filter received (ctl: " + this._filter + ")");
}
if (ret) {
this._filter = null;
}
return ret;
}
}, {
key: "_copyFilter",
value: function _copyFilter(streamId, x, y, width, height, sock, display, depth) {
var uncompressedSize = width * height * 3;
var data;
if (uncompressedSize === 0) {
return true;
}
if (uncompressedSize < 12) {
if (sock.rQwait("TIGHT", uncompressedSize)) {
return false;
}
data = sock.rQshiftBytes(uncompressedSize);
} else {
data = this._readData(sock);
if (data === null) {
return false;
}
this._zlibs[streamId].setInput(data);
data = this._zlibs[streamId].inflate(uncompressedSize);
this._zlibs[streamId].setInput(null);
}
var rgbx = new Uint8Array(width * height * 4);
for (var i = 0, j = 0; i < width * height * 4; i += 4, j += 3) {
rgbx[i] = data[j];
rgbx[i + 1] = data[j + 1];
rgbx[i + 2] = data[j + 2];
rgbx[i + 3] = 255; // Alpha
}
display.blitImage(x, y, width, height, rgbx, 0, false);
return true;
}
}, {
key: "_paletteFilter",
value: function _paletteFilter(streamId, x, y, width, height, sock, display, depth) {
if (this._numColors === 0) {
if (sock.rQwait("TIGHT palette", 1)) {
return false;
}
var numColors = sock.rQpeek8() + 1;
var paletteSize = numColors * 3;
if (sock.rQwait("TIGHT palette", 1 + paletteSize)) {
return false;
}
this._numColors = numColors;
sock.rQskipBytes(1);
sock.rQshiftTo(this._palette, paletteSize);
}
var bpp = this._numColors <= 2 ? 1 : 8;
var rowSize = Math.floor((width * bpp + 7) / 8);
var uncompressedSize = rowSize * height;
var data;
if (uncompressedSize === 0) {
return true;
}
if (uncompressedSize < 12) {
if (sock.rQwait("TIGHT", uncompressedSize)) {
return false;
}
data = sock.rQshiftBytes(uncompressedSize);
} else {
data = this._readData(sock);
if (data === null) {
return false;
}
this._zlibs[streamId].setInput(data);
data = this._zlibs[streamId].inflate(uncompressedSize);
this._zlibs[streamId].setInput(null);
}
// Convert indexed (palette based) image data to RGB
if (this._numColors == 2) {
this._monoRect(x, y, width, height, data, this._palette, display);
} else {
this._paletteRect(x, y, width, height, data, this._palette, display);
}
this._numColors = 0;
return true;
}
}, {
key: "_monoRect",
value: function _monoRect(x, y, width, height, data, palette, display) {
// Convert indexed (palette based) image data to RGB
// TODO: reduce number of calculations inside loop
var dest = this._getScratchBuffer(width * height * 4);
var w = Math.floor((width + 7) / 8);
var w1 = Math.floor(width / 8);
for (var _y = 0; _y < height; _y++) {
var dp = void 0,
sp = void 0,
_x = void 0;
for (_x = 0; _x < w1; _x++) {
for (var b = 7; b >= 0; b--) {
dp = (_y * width + _x * 8 + 7 - b) * 4;
sp = (data[_y * w + _x] >> b & 1) * 3;
dest[dp] = palette[sp];
dest[dp + 1] = palette[sp + 1];
dest[dp + 2] = palette[sp + 2];
dest[dp + 3] = 255;
}
}
for (var _b = 7; _b >= 8 - width % 8; _b--) {
dp = (_y * width + _x * 8 + 7 - _b) * 4;
sp = (data[_y * w + _x] >> _b & 1) * 3;
dest[dp] = palette[sp];
dest[dp + 1] = palette[sp + 1];
dest[dp + 2] = palette[sp + 2];
dest[dp + 3] = 255;
}
}
display.blitImage(x, y, width, height, dest, 0, false);
}
}, {
key: "_paletteRect",
value: function _paletteRect(x, y, width, height, data, palette, display) {
// Convert indexed (palette based) image data to RGB
var dest = this._getScratchBuffer(width * height * 4);
var total = width * height * 4;
for (var i = 0, j = 0; i < total; i += 4, j++) {
var sp = data[j] * 3;
dest[i] = palette[sp];
dest[i + 1] = palette[sp + 1];
dest[i + 2] = palette[sp + 2];
dest[i + 3] = 255;
}
display.blitImage(x, y, width, height, dest, 0, false);
}
}, {
key: "_gradientFilter",
value: function _gradientFilter(streamId, x, y, width, height, sock, display, depth) {
// assume the TPIXEL is 3 bytes long
var uncompressedSize = width * height * 3;
var data;
if (uncompressedSize === 0) {
return true;
}
if (uncompressedSize < 12) {
if (sock.rQwait("TIGHT", uncompressedSize)) {
return false;
}
data = sock.rQshiftBytes(uncompressedSize);
} else {
data = this._readData(sock);
if (data === null) {
return false;
}
this._zlibs[streamId].setInput(data);
data = this._zlibs[streamId].inflate(uncompressedSize);
this._zlibs[streamId].setInput(null);
}
var rgbx = new Uint8Array(4 * width * height);
var rgbxIndex = 0,
dataIndex = 0;
var left = new Uint8Array(3);
for (var _x2 = 0; _x2 < width; _x2++) {
for (var c = 0; c < 3; c++) {
var prediction = left[c];
var value = data[dataIndex++] + prediction;
rgbx[rgbxIndex++] = value;
left[c] = value;
}
rgbx[rgbxIndex++] = 255;
}
var upperIndex = 0;
var upper = new Uint8Array(3),
upperleft = new Uint8Array(3);
for (var _y2 = 1; _y2 < height; _y2++) {
left.fill(0);
upperleft.fill(0);
for (var _x3 = 0; _x3 < width; _x3++) {
for (var _c = 0; _c < 3; _c++) {
upper[_c] = rgbx[upperIndex++];
var _prediction = left[_c] + upper[_c] - upperleft[_c];
if (_prediction < 0) {
_prediction = 0;
} else if (_prediction > 255) {
_prediction = 255;
}
var _value = data[dataIndex++] + _prediction;
rgbx[rgbxIndex++] = _value;
upperleft[_c] = upper[_c];
left[_c] = _value;
}
rgbx[rgbxIndex++] = 255;
upperIndex++;
}
}
display.blitImage(x, y, width, height, rgbx, 0, false);
return true;
}
}, {
key: "_readData",
value: function _readData(sock) {
if (this._len === 0) {
if (sock.rQwait("TIGHT", 3)) {
return null;
}
var _byte;
_byte = sock.rQshift8();
this._len = _byte & 0x7f;
if (_byte & 0x80) {
_byte = sock.rQshift8();
this._len |= (_byte & 0x7f) << 7;
if (_byte & 0x80) {
_byte = sock.rQshift8();
this._len |= _byte << 14;
}
}
}
if (sock.rQwait("TIGHT", this._len)) {
return null;
}
var data = sock.rQshiftBytes(this._len, false);
this._len = 0;
return data;
}
}, {
key: "_getScratchBuffer",
value: function _getScratchBuffer(size) {
if (!this._scratchBuffer || this._scratchBuffer.length < size) {
this._scratchBuffer = new Uint8Array(size);
}
return this._scratchBuffer;
}
}]);
}();