send.py

#!/usr/bin/env python2
import sys
import socket

from scapy.all import srp1, hexdump, get_if_hwaddr
from scapy.all import Ether, IP
from scapy.fields import *

from headers import *


def send_program(insns):
    """
    Sends a packet containing a Tiny RISC-V program to load balancer for execution
    :param insns: a list of Tiny RISC-V instructions
    """
    iface = get_if()

    pkt = Ether(src=get_if_hwaddr(iface), dst=LOAD_BALANCER_MAC)
    pkt = pkt / IP(proto=PROTO_RAW_PROGRAM, dst=LOAD_BALANCER_IP)
    pkt = make_program(pkt, insns)
    hexdump(pkt)
    pkt.show2()
    response = srp1(pkt, iface=iface, verbose=False)
    return response


def main():
    send_program([
        AddI(dst=1, src=0, imm=42),      # r1 = 42
        AddI(dst=2, src=0, imm=31),      # r2 = 31
        AddI(dst=3, src=0, imm=-1),      # r3 = -1 (4294967295)
        Add(dst=4, src=1, target=1),     # r4 = 84
        Sub(dst=5, src=1, target=2),     # r5 = 11
        And(dst=6, src=1, target=2),     # r6 = 10
        Or(dst=7, src=1, target=2),      # r7 = 63
        Xor(dst=8, src=1, target=2),     # r8 = 53
        AndI(dst=9, src=1, imm=8),       # r9 = 8
        Slt(dst=10, src=3, target=2),    # r10 = 1
        Sltu(dst=11, src=2, target=3),   # r11 = 1
        Sra(dst=12, src=3, target=1),    # r12 = -1 (4294967295)
        Srl(dst=13, src=3, target=1),    # r13 = 4194303
        Sll(dst=14, src=1, target=1),    # r14 = 43008
        Mul(dst=15, src=1, target=2),    # r15 = 1302
        OrI(dst=16, src=1, imm=31),      # r16 = 63
        XorI(dst=17, src=1, imm=31),     # r17 = 53
        SltI(dst=18, src=3, imm=100),    # r18 = 1
        SltIu(dst=19, src=2, imm=-100),  # r19 = 1
        SraI(dst=20, src=3, imm=10),     # r20 = -1 (4294967295)
        SrlI(dst=21, src=3, imm=14),     # r21 = 262143
        SllI(dst=22, src=1, imm=5),      # r22 = 1344
        Lui(dst=23, imm=0xabcde),        # r23 = 2882396160
        Auipc(dst=24, imm=0xabcde),      # r24 = 2882396252
        Jal(dst=25, imm=4),              # r25 = 100
        AddI(dst=26, src=25, imm=8),     # r26 = 108 (want to point to pc after Jr)
        Jr(src=26),
        Jalr(dst=27, src=26, imm=4),     # r27 = 112
        Sw(dst=9, src=15, imm=4),        # mem[12] = 1302
        Lw(dst=28, src=9, imm=4),        # r28 = mem[12] = 1302
        AddI(dst=29, src=1, imm=50),     # r29 = 92
        Beq(src=13, target=13, imm=8),
        AddI(dst=29, src=0, imm=0),      # skipped by previous Beq
        AddI(dst=29, src=29, imm=2),     # r29 = 94 (Beq success)
        Bne(src=13, target=14, imm=8),
        AddI(dst=29, src=0, imm=0),      # skipped by previous Bne
        AddI(dst=29, src=29, imm=3),     # r29 = 97 (Bne success)
        Blt(src=3, target=2, imm=8),
        AddI(dst=29, src=0, imm=0),      # skipped by previous Blt
        AddI(dst=29, src=29, imm=5),     # r29 = 102 (Blt success)
        Bge(src=2, target=3, imm=8),
        AddI(dst=29, src=0, imm=0),      # skipped by previous Bge
        AddI(dst=29, src=29, imm=7),     # r29 = 109 (Bge success)
        Bltu(src=2, target=3, imm=8),
        AddI(dst=29, src=0, imm=0),      # skipped by previous Bltu
        AddI(dst=29, src=29, imm=11),    # r29 = 120 (Bltu success)
        Bgeu(src=3, target=2, imm=8),
        AddI(dst=29, src=0, imm=0),      # skipped by previous Bgeu
        AddI(dst=29, src=29, imm=13),    # r29 = 133 (Bgeu success)
        Nop(),
        Lw(dst=30, src=1, imm=2),        # r30 = # of times this program has been run
        AddI(dst=30, src=30, imm=1),
        Sw(dst=1, src=30, imm=2),
    ])


if __name__ == '__main__':
    main()