app-ethereum/tests/functional/lib/test.fixture.js

"use strict";

var _zemu = _interopRequireWildcard(require("@zondax/zemu"));

var _hwAppEth = _interopRequireDefault(require("@ledgerhq/hw-app-eth"));

var _utils = require("ethers/lib/utils");

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }

function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && 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; }

const transactionUploadDelay = 60000;

async function waitForAppScreen(sim, current_screen = null) {
  if (current_screen === null) current_screen = sim.getMainMenuSnapshot();
  await sim.waitUntilScreenIsNot(current_screen, transactionUploadDelay);
}

const sim_options_nano = { ..._zemu.DEFAULT_START_OPTIONS,
  logging: true,
  X11: true,
  startText: 'is ready'
};

const Resolve = require('path').resolve;

const NANOS_ELF_PATH = Resolve('elfs/ethereum_nanos.elf');
const NANOX_ELF_PATH = Resolve('elfs/ethereum_nanox.elf');
const NANOS_CLONE_ELF_PATH = Resolve("elfs/ethereum_classic_nanos.elf");
const NANOX_CLONE_ELF_PATH = Resolve("elfs/ethereum_classic_nanox.elf");
const nano_models = [{
  name: 'nanos',
  letter: 'S',
  path: NANOS_ELF_PATH,
  clone_path: NANOS_CLONE_ELF_PATH
}, {
  name: 'nanox',
  letter: 'X',
  path: NANOX_ELF_PATH,
  clone_path: NANOX_CLONE_ELF_PATH
}];
const TIMEOUT = 1000000; // useful to take an apdu as a hex string and convert its JS representation

function apdu_as_string(str) {
  let buffer = [];

  for (let i = 0; i < str.length; i += 2) {
    const str_extract = str.substring(i, i + 2);
    buffer[i / 2] = parseInt(str_extract, 16);
  }

  return {
    cla: buffer[0],
    ins: buffer[1],
    p1: buffer[2],
    p2: buffer[3],
    data: Buffer.from(buffer.slice(5))
  };
}

async function send_apdu(ts, apdu) {
  return ts.send(apdu.cla, apdu.ins, apdu.p1, apdu.p2, apdu.data);
} // Generates a serializedTransaction from a rawHexTransaction copy pasted from etherscan.


function txFromEtherscan(rawTx) {
  // Remove 0x prefix
  rawTx = rawTx.slice(2);
  let txType = rawTx.slice(0, 2);

  if (txType == "02" || txType == "01") {
    // Remove "02" prefix
    rawTx = rawTx.slice(2);
  } else {
    txType = "";
  }

  let decoded = _utils.RLP.decode("0x" + rawTx);

  if (txType != "") {
    decoded = decoded.slice(0, decoded.length - 3); // remove v, r, s
  } else {
    decoded[decoded.length - 1] = "0x"; // empty

    decoded[decoded.length - 2] = "0x"; // empty

    decoded[decoded.length - 3] = "0x01"; // chainID 1
  } // Encode back the data, drop the '0x' prefix


  let encoded = _utils.RLP.encode(decoded).slice(2); // Don't forget to prepend the txtype


  return txType + encoded;
}

function zemu(device, func, start_clone = false) {
  return async () => {
    jest.setTimeout(TIMEOUT);
    let elf_path;
    let lib_elf;

    if (start_clone) {
      elf_path = device.clone_path;
      lib_elf = {
        'Ethereum': device.path
      };
    } else {
      elf_path = device.path;
    }

    const sim = new _zemu.default(elf_path, lib_elf);

    try {
      await sim.start({ ...sim_options_nano,
        model: device.name
      });
      const transport = await sim.getTransport();
      await func(sim, new _hwAppEth.default(transport));
    } finally {
      await sim.close();
    }
  };
}

module.exports = {
  zemu,
  waitForAppScreen,
  sim_options_nano,
  nano_models,
  TIMEOUT,
  txFromEtherscan,
  apdu_as_string,
  send_apdu
};