trezor-link
Version:
Trezor Link for browser
269 lines (196 loc) • 11.7 kB
JavaScript
;
function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
Object.defineProperty(exports, "__esModule", {
value: true
});
exports.buildOne = buildOne;
exports.buildAndSend = buildAndSend;
var _regenerator = _interopRequireDefault(require("@babel/runtime/regenerator"));
var ProtoBuf = _interopRequireWildcard(require("protobufjs-old-fixed-webpack"));
function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function _getRequireWildcardCache() { return cache; }; return cache; }
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || _typeof(obj) !== "object" && typeof obj !== "function") { return { "default": obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj["default"] = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }
function _createForOfIteratorHelper(o, allowArrayLike) { var it; if (typeof Symbol === "undefined" || o[Symbol.iterator] == null) { if (Array.isArray(o) || (it = _unsupportedIterableToArray(o)) || allowArrayLike && o && typeof o.length === "number") { if (it) o = it; var i = 0; var F = function F() {}; return { s: F, n: function n() { if (i >= o.length) return { done: true }; return { done: false, value: o[i++] }; }, e: function e(_e) { throw _e; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var normalCompletion = true, didErr = false, err; return { s: function s() { it = o[Symbol.iterator](); }, n: function n() { var step = it.next(); normalCompletion = step.done; return step; }, e: function e(_e2) { didErr = true; err = _e2; }, f: function f() { try { if (!normalCompletion && it["return"] != null) it["return"](); } finally { if (didErr) throw err; } } }; }
function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && Symbol.iterator in Object(iter)) return Array.from(iter); }
function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) { arr2[i] = arr[i]; } return arr2; }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
var HEADER_SIZE = 1 + 1 + 4 + 2;
var MESSAGE_HEADER_BYTE = 0x23;
var BUFFER_SIZE = 63; // Sends more buffers to device.
function sendBuffers(_x, _x2) {
return _sendBuffers.apply(this, arguments);
} // already built PB message
function _sendBuffers() {
_sendBuffers = _asyncToGenerator( /*#__PURE__*/_regenerator["default"].mark(function _callee(sender, buffers) {
var _iterator, _step, buffer;
return _regenerator["default"].wrap(function _callee$(_context) {
while (1) {
switch (_context.prev = _context.next) {
case 0:
// eslint-disable-next-line prefer-const
_iterator = _createForOfIteratorHelper(buffers);
_context.prev = 1;
_iterator.s();
case 3:
if ((_step = _iterator.n()).done) {
_context.next = 9;
break;
}
buffer = _step.value;
_context.next = 7;
return sender(buffer);
case 7:
_context.next = 3;
break;
case 9:
_context.next = 14;
break;
case 11:
_context.prev = 11;
_context.t0 = _context["catch"](1);
_iterator.e(_context.t0);
case 14:
_context.prev = 14;
_iterator.f();
return _context.finish(14);
case 17:
case "end":
return _context.stop();
}
}
}, _callee, null, [[1, 11, 14, 17]]);
}));
return _sendBuffers.apply(this, arguments);
}
var BuiltMessage = /*#__PURE__*/function () {
function BuiltMessage(messages, // Builders, generated by reading config
name, // Name of the message
data) // data as "pure" object, from trezor.js
{
_classCallCheck(this, BuiltMessage);
var Builder = messages.messagesByName[name];
if (Builder == null) {
throw new Error("The message name ".concat(name, " is not found."));
} // cleans up stuff from angular and remove "null" that crashes in builder
cleanupInput(data);
if (data) {
this.message = new Builder(data);
} else {
this.message = new Builder();
} // protobuf lib doesn't know how to work with "(wire_type)" option.
// NOTE: round brackets are valid protobuf syntax for custom user declared option
// messages: `TxAckInput`, `TxAckOutput`, `TxAckPrevInput`, `TxAckPrevOutput`, `TxAckPrevMeta`, `TxAckPrevExtraData`
if (typeof this.message.$type.options["(wire_type)"] === "number") {
this.type = this.message.$type.options["(wire_type)"];
} else {
this.type = messages.messageTypes["MessageType_".concat(name)];
}
} // encodes into "raw" data, but it can be too long and needs to be split into
// smaller buffers
_createClass(BuiltMessage, [{
key: "_encodeLong",
value: function _encodeLong(addTrezorHeaders) {
var headerSize = HEADER_SIZE; // should be 8
var bytes = new Uint8Array(this.message.encodeAB());
var fullSize = (addTrezorHeaders ? headerSize : headerSize - 2) + bytes.length;
var encodedByteBuffer = new ProtoBuf.ByteBuffer(fullSize); // first encode header
if (addTrezorHeaders) {
// 2*1 byte
encodedByteBuffer.writeByte(MESSAGE_HEADER_BYTE);
encodedByteBuffer.writeByte(MESSAGE_HEADER_BYTE);
} // 2 bytes
encodedByteBuffer.writeUint16(this.type); // 4 bytes (so 8 in total)
encodedByteBuffer.writeUint32(bytes.length); // then put in the actual message
encodedByteBuffer.append(bytes); // and convert to uint8 array
// (it can still be too long to send though)
var encoded = new Uint8Array(encodedByteBuffer.buffer);
return encoded;
} // encodes itself and splits into "nice" chunks
}, {
key: "encode",
value: function encode() {
var bytes = this._encodeLong(true);
var result = [];
var size = BUFFER_SIZE; // How many pieces will there actually be
var count = Math.floor((bytes.length - 1) / size) + 1; // slice and dice
for (var i = 0; i < count; i++) {
var slice = bytes.subarray(i * size, (i + 1) * size);
var newArray = new Uint8Array(size);
newArray.set(slice);
result.push(newArray.buffer);
}
return result;
} // encodes itself into one long arraybuffer
}, {
key: "encodeOne",
value: function encodeOne() {
var bytes = this._encodeLong(false);
return Buffer.from(_toConsumableArray(bytes));
}
}]);
return BuiltMessage;
}(); // Removes $$hashkey from angular and remove nulls
function cleanupInput(message) {
delete message.$$hashKey;
for (var key in message) {
var value = message[key];
if (value == null) {
delete message[key];
} else {
if (Array.isArray(value)) {
value.forEach(function (i) {
if (typeof i === "object") {
cleanupInput(i);
}
});
}
if (typeof value === "object") {
cleanupInput(value);
}
}
}
} // Builds buffers to send.
// messages: Builders, generated by reading config
// name: Name of the message
// data: Data to serialize, exactly as given by trezor.js
// Returning buffers that will be sent to Trezor
function buildBuffers(messages, name, data) {
var message = new BuiltMessage(messages, name, data);
var encoded = message.encode();
return encoded;
} // Sends message to device.
// Resolves iff everything gets sent
function buildOne(messages, name, data) {
var message = new BuiltMessage(messages, name, data);
return message.encodeOne();
} // Sends message to device.
// Resolves iff everything gets sent
function buildAndSend(_x3, _x4, _x5, _x6) {
return _buildAndSend.apply(this, arguments);
}
function _buildAndSend() {
_buildAndSend = _asyncToGenerator( /*#__PURE__*/_regenerator["default"].mark(function _callee2(messages, sender, name, data) {
var buffers;
return _regenerator["default"].wrap(function _callee2$(_context2) {
while (1) {
switch (_context2.prev = _context2.next) {
case 0:
buffers = buildBuffers(messages, name, data);
return _context2.abrupt("return", sendBuffers(sender, buffers));
case 2:
case "end":
return _context2.stop();
}
}
}, _callee2);
}));
return _buildAndSend.apply(this, arguments);
}