aboutsummaryrefslogtreecommitdiffstats
path: root/packages/sol-cov/src/instructions.ts
blob: 40987dbe52137c82ed791ef2fb976a2c879e070f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import { constants } from './constants';

const isPush = (inst: number) => inst >= constants.PUSH1 && inst <= constants.PUSH32;

const pushDataLength = (inst: number) => inst - constants.PUSH1 + 1;

const instructionLength = (inst: number) => (isPush(inst) ? pushDataLength(inst) + 1 : 1);

export const getPcToInstructionIndexMapping = (bytecode: Uint8Array) => {
    const result: {
        [programCounter: number]: number;
    } = {};
    let byteIndex = 0;
    let instructionIndex = 0;
    while (byteIndex < bytecode.length) {
        const instruction = bytecode[byteIndex];
        const length = instructionLength(instruction);
        result[byteIndex] = instructionIndex;
        byteIndex += length;
        instructionIndex += 1;
    }
    return result;
};