[Golang实现JVM第四篇] 整数加法和条件判断指令的实现

在上一篇中我们实现了一个能跑的解释器，支持了一些基本的栈操作指令。现在我们就可以开始实现"有点用"的数学运算和条件判断了。

github: https://github.com/wanghongfei/mini-jvm

局部变量表、程序计数器

由于JVM字节码是基于栈的指令集，因此一切操作都是以栈为基础的，也就是说计算1+1,那需要先在栈中压入两个1然后进行计算，如果是对象方法调用，那么对象的引用、方法参数都会事先被压入栈中。除栈外还有一个跟执行相关的重要结构就是局部变量表(Local Variable Table)，用来保存当前执行环境(如当前方法)下的局部变量，在JVM中用一个数组来表示，有专门的字节码指令用于向局部变量表的指定下标存取数据，例如storeX和loadX指令。值得注意的是这个数组大小是固定的，不需要动态扩容，因为在编译期javac就能够确定一个方法需要用大的局部变量表，然后会把这个数字写入到class文件的code属性的max_locals字段中。我们在解释字节码的时候可以直接用它来创建数组。

程序计数器比较简单，就是一个整数类型，永远指向下一条将要执行的字节码，具体到实现就是指向下一条字节码数组的下标。

现在我们的方法栈就有三个元素了，操作数栈、局部变量表、程序计数器：

// 方法栈的栈帧
type MethodStackFrame struct {
    // 本地变量表
    localVariablesTable []interface{}

    // 操作数栈
    opStack *OpStack

    // 程序计数器
    pc int
}

因为本篇暂不涉及方法调用，因此栈帧创建的问题可以先忽略，假定全局只有一个MethodStackFrame。

iload_n, istore_n, iconst_n指令

iload是一组指令，包含iload, iload0, iload1 ... ... iload3。开头的i表示操作数必须是整数，后边的数字表示需要将局部变量表的哪个槽位中的整数压到栈顶，即数组下标。而对于不带数字的iload指令，指令后面会紧跟着一个byte, 表示数组下标，例如要把下标为5的槽位中的整数压栈那么指令就会是iload 5两个字节。了解这些以后就很容易实现了：

        case bcode.Iload:
            // Load int from local variable
            // ilaod index
            index := codeAttr.Code[frame.pc + 1]
            frame.pc++  // (1)
            frame.opStack.Push(frame.localVariablesTable[index])
        case bcode.Iload0:
            // 将第1个slot中的值压栈
            frame.opStack.Push(frame.localVariablesTable[0])
        case bcode.Iload1:
            frame.opStack.Push(frame.localVariablesTable[1])
        case bcode.Iload2:
            frame.opStack.Push(frame.localVariablesTable[2])
        case bcode.Iload3:
            frame.opStack.Push(frame.localVariablesTable[3])

注意(1)的位置，因为不带数字的iload指令后面会跟着一个表示数组下标的字节，因此我们在取出这个下标后需要将程序计数器+1, 否则下次循环后就会取到数组下标而不是字节码了。此外在操作数组的时候其实并不需要检查下标是否越界，javac会保证生成的指令不会操作越界的下标。

istore指令也是类似的，表示将栈顶元素出栈，然后保存到局部变量表的指定槽位中，例如：

        case bcode.Istore1:
            // 将栈顶int型数值存入第二个本地变量
            top, _ := frame.opStack.PopInt()
            frame.localVariablesTable[1] = top
        case bcode.Istore2:
            // 将栈顶int型数值存入第3个本地变量
            top, _ := frame.opStack.PopInt()
            frame.localVariablesTable[2] = top

iconst指令有点不太一样，虽然也是将整数压栈，但是他不跟局部变量表交互，压栈的值直接在指令中体现，例如iconst_1就是把1压栈，iconst_2就是压入2，实现起来也非常简单：

case bcode.Iconst1:
            frame.opStack.Push(1)

iadd指令

iadd表示连续做两次出栈操作，然后将得到的两个整数相加，最后再把结果压回栈中。实现起来也非常简单：

case bcode.Iadd:
            // 取出栈顶2元素，相加，入栈
            op1, _ := frame.opStack.PopInt()
            op2, _ := frame.opStack.PopInt()
            sum := op1 + op2
            frame.opStack.Push(sum)

还是那句话，不需要在pop()前检查栈是否为空，因为编译器会保证不会非法操作栈，除非是我们的go代码出了问题，如果是后者的话就直接让程序崩溃方便及时发现问题。

有了iload, istore, iadd后，我们终于能计算1+1了，然而尴尬的是，计算后的结果是保存在局部变量表里的，看不见摸不着，不过可以在debug调试过程中看到这个值。下一篇会介绍如何实现方法调用，到时候就可以实现控制台输出的功能了。

ifeq, iflt, ifeq等判断指令

if_<cond>是代表一组指令，格式为if_<cond> byte1 byte2，也就是指令后面跟着两个字节，用来组成一个16位的有符号整数，此整数表示当条件(栈顶元素跟数字0做比较)成立时的要跳转到的目标字节码的offset, 注意这个offset是以当前if_<cond>指令的位置为基准的。例如，if_lt所在的offset是10，byte1 byte2组合后是5, 那么目标字节码的位置就是 10 + 5 = 15。这里如果用javap反编译的话输出结果会有一点误导人，javap会输出这条指令计算好偏移量后的数值而不是byte1 byte2本身的值,例如：

3: ifle          9

右侧的9表示 3 + 6 = 9，即ifle后面跟着的16位数字其实是6，而不是9。

我们拿ifle举例，他表示将栈顶元素value出栈并且跟0作比较，当value <= 0时条件成立。go代码如下：

case bcode.Ifle:
            // 当栈顶int型数值小于等于0时跳转
            err := i.bcodeIfCompZero(frame, codeAttr, func(op1 int, op2 int) bool {
                return op1 <= op2
            })

            if nil != err {
                return fmt.Errorf("failed to execute 'ifle': %w", err)
            }

bcodeIfCompZero()函数实现如下：

func (i *InterpretedExecutionEngine) bcodeIfCompZero(frame *MethodStackFrame, codeAttr *class.CodeAttr, gotoJudgeFunc func(int, int) bool) error {
    // 当栈顶int型数值小于0时跳转
    // 跳转的偏移量
    twoByteNum := codeAttr.Code[frame.pc + 1 : frame.pc + 1 + 2]
    var offset int16
    err := binary.Read(bytes.NewBuffer(twoByteNum), binary.BigEndian, &offset)
    if nil != err {
        return fmt.Errorf("failed to read offset for if_icmpgt: %w", err)
    }

    op, _ := frame.opStack.PopInt()
    if gotoJudgeFunc(op, 0) {
        frame.pc = frame.pc + int(offset) - 1

    } else {
        frame.pc += 2
    }

    return nil
}

有些这些指令我们就可以解释一些类似于：

int sum = 0;
if (sum > 0) {
  sum = 100;
}

编译后的字节码了。我们可以照葫芦画瓢，先写一段java代码编译一下，然后javap -verbose看看有没有不认识的指令，如果有就查规范，看看应该如何解释，就能够实现很多简单指令了。这里要注意有很多指令后面会携带一个_w后缀，例如整数自增指令iinc_w，表示对字节码后面跟着的字节进行加宽处理，例如原本的iinc byte1 byte2变成了iinc_w byte1 byte2 byte3，诸如此类，只要对照规范看清楚就OK了。

下一篇会介绍方法调用相关指令的实现。