master-thesis/netpfga/minip4/sw/CLI/P4_SWITCH_CLI.py

484 lines
16 KiB
Python
Executable file

#!/usr/bin/env python
#
# Copyright (c) 2017 Stephen Ibanez
# All rights reserved.
#
# This software was developed by Stanford University and the University of Cambridge Computer Laboratory
# under National Science Foundation under Grant No. CNS-0855268,
# the University of Cambridge Computer Laboratory under EPSRC INTERNET Project EP/H040536/1 and
# by the University of Cambridge Computer Laboratory under DARPA/AFRL contract FA8750-11-C-0249 ("MRC2"),
# as part of the DARPA MRC research programme.
#
# @NETFPGA_LICENSE_HEADER_START@
#
# Licensed to NetFPGA C.I.C. (NetFPGA) under one or more contributor
# license agreements. See the NOTICE file distributed with this work for
# additional information regarding copyright ownership. NetFPGA licenses this
# file to you under the NetFPGA Hardware-Software License, Version 1.0 (the
# "License"); you may not use this file except in compliance with the
# License. You may obtain a copy of the License at:
#
# http://www.netfpga-cic.org
#
# Unless required by applicable law or agreed to in writing, Work distributed
# under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
# CONDITIONS OF ANY KIND, either express or implied. See the License for the
# specific language governing permissions and limitations under the License.
#
# @NETFPGA_LICENSE_HEADER_END@
#
import os, sys, cmd, re
from collections import OrderedDict
from pprint import pprint
import p4_regs_api, p4_tables_api
# defines the table_*_add_entry command
# also defines convert_to_int()
from p4_px_tables import *
### Global Variables ###
P4_EXTERNS = p4_regs_api.P4_EXTERNS
P4_REGS = OrderedDict() # just the externs with a control interface
for extern_name, extern_dict in P4_EXTERNS.items():
if 'control_width' in extern_dict.keys() and extern_dict['control_width'] > 0:
P4_REGS[extern_name] = extern_dict
#p4_tables = p4_tables_api.p4_tables_info
PX_CAM_TABLES = p4_tables_api.PX_CAM_TABLES
PX_TCAM_TABLES = p4_tables_api.PX_TCAM_TABLES
PX_LPM_TABLES = p4_tables_api.PX_LPM_TABLES
class SimpleSumeSwitch(cmd.Cmd):
"""The SimpleSumeSwitch interactive command line tool"""
prompt = ">> "
intro = "The SimpleSumeSwitch interactive command line tool\n type help to see all commands"
##########################
### Register Functions ###
##########################
"""
List the registers defined in the SimpleSumeSwitch
"""
def do_list_regs(self, line):
for reg_name, reg_dict in P4_REGS.items():
print '-'*len(reg_name), '\n', reg_name, ':\n', '-'*len(reg_name)
pprint(reg_dict)
def help_list_regs(self):
print """
list_regs
DESCRIPTION: List the registers defined in the SimpleSumeSwitch and their relevant compile time information
"""
def do_reg_read(self, line):
fmat = r"(.*)\[(\d*)\]"
searchObj = re.search(fmat, line)
if searchObj is not None:
reg_name = searchObj.group(1)
index = int(searchObj.group(2))
else:
reg_name = line
index = 0
result = p4_regs_api.reg_read(reg_name, index)
print result
def help_reg_read(self):
print """
reg_read <REG_NAME>[<INDEX>]
DESCRIPTION: Read the current value of the provided register at the given index
"""
def complete_reg_read(self, text, line, begidx, endidx):
if not text:
completions = P4_REGS.keys()
else:
completions = [ r for r in P4_REGS.keys() if r.startswith(text)]
return completions
def do_reg_write(self, line):
fmat = r"(.*)\[(\d*)\]\s*(\d*)"
searchObj = re.search(fmat, line)
if searchObj is not None:
reg_name = searchObj.group(1)
index = int(searchObj.group(2))
val = int(searchObj.group(3))
else:
print >> sys.stderr, "ERROR: usage ..."
self.help_reg_write()
return
result = p4_regs_api.reg_write(reg_name, index, val)
print result
def help_reg_write(self):
print """
writeReg <REG_NAME>[<INDEX>] <VALUE>
DESCRIPTION: Write VALUE to the provided register at the given INDEX
"""
def complete_reg_write(self, text, line, begidx, endidx):
if not text:
completions = P4_REGS.keys()
else:
completions = [ r for r in P4_REGS.keys() if r.startswith(text)]
return completions
###########################
### CAM Table Functions ###
###########################
"""
List the CAM tables and some relevant info
"""
def do_list_cam_tables(self, line):
for table_name, table in PX_CAM_TABLES.items():
print '-'*len(table_name), '\n', table_name, ':\n', '-'*len(table_name)
pprint(table.info)
def help_list_cam_tables(self):
print """
list_cam_tables
DESCRIPTION: List the exact match tables defined in the SimpleSumeSwitch and their relevant compile time information
"""
"""
Read entry from a table
"""
def do_table_cam_read_entry(self, line):
args = line.split(' ')
try:
assert(len(args) >= 2)
except:
print >> sys.stderr, "ERROR: usage ... "
self.help_table_cam_read_entry()
return
table_name = args[0]
keys = map(convert_to_int, args[1:])
(found, val) = p4_tables_api.table_cam_read_entry(table_name, keys)
print "Entry found: ", found
print hex(int(val, 16))
def help_table_cam_read_entry(self):
print """
table_cam_read_entry <table_name> <keys>
DESCRIPTION: Read the entry in table corresponding to the given list of keys
PARAMS:
<table_name> : name of the table to read from
<keys> : space separated list of keys to look for in the table (must correspond to table's keys in the order defined in the P4 program)
"""
def complete_table_cam_read_entry(self, text, line, begidx, endidx):
if not text:
completions = PX_CAM_TABLES.keys()
else:
completions = [ t for t in PX_CAM_TABLES.keys() if t.startswith(text)]
return completions
"""
Add an entry to a table
"""
def do_table_cam_add_entry(self, line):
# defined in p4_px_tables.py
(table_name, keys, action_name, action_data) = parse_table_cam_add_entry(line)
p4_tables_api.table_cam_add_entry(table_name, keys, action_name, action_data)
def help_table_cam_add_entry(self):
# defined in p4_px_tables.py
print help_table_cam_add_entry()
def complete_table_cam_add_entry(self, text, line, begidx, endidx):
if not text:
if len(line.split()) == 1:
# table names
completions = PX_CAM_TABLES.keys()
elif len(line.split()) == 2:
# actions names
table_name = line.split()[1]
if table_name in PX_CAM_TABLES.keys():
# the table name is recognized
actions = PX_CAM_TABLES[table_name].actions
completions = [a['p4_name'] for a in actions]
else:
completions = []
else:
if len(line.split()) == 2:
# trying to complete table name
completions = [ t for t in PX_CAM_TABLES.keys() if t.startswith(text)]
elif len(line.split()) == 3:
# trying to complete action_name
table_name = line.split()[1]
if table_name in PX_CAM_TABLES.keys():
# the table name is recognized
actions = PX_CAM_TABLES[table_name].actions
completions = [ a['p4_name'] for a in actions if a['p4_name'].startswith(text)]
else:
completions = []
return completions
"""
Delete an entry from a table
"""
def do_table_cam_delete_entry(self, line):
args = line.split(' ')
if (len(args) < 2):
print >> sys.stderr, "ERROR: usage..."
self.help_table_cam_delete_entry()
return
table_name = args[0]
keys = map(convert_to_int, args[1:])
p4_tables_api.table_cam_delete_entry(table_name, keys)
def help_table_cam_delete_entry(self):
print """
table_cam_delete_entry <table_name> <keys>
DESCRIPTION: Delete the entry in the specified table with the given keys
PARAMS:
<table_name> : name of the table to delete an entry from
<keys> : space separated list of keys (must correspond to table's keys in the order defined in the P4 program)
"""
def complete_table_cam_delete_entry(self, text, line, begidx, endidx):
if not text:
completions = PX_CAM_TABLES.keys()
else:
completions = [ t for t in PX_CAM_TABLES.keys() if t.startswith(text)]
return completions
"""
Get the current number of entries in the table
"""
def do_table_cam_get_size(self, line):
table_name = line.strip()
print p4_tables_api.table_cam_get_size(table_name)
def help_table_cam_get_size(self):
print """
table_cam_get_size <table_name>
DESCRIPTION: Get the current number of entries in the specified table
"""
def complete_table_cam_get_size(self, text, line, begidx, endidx):
if not text:
completions = PX_CAM_TABLES.keys()
else:
completions = [ t for t in p4_tables.keys() if t.startswith(text)]
return completions
def do_EOF(self, line):
return True
############################
### TCAM Table Functions ###
############################
"""
List the TCAM tables and some relevant info
"""
def do_list_tcam_tables(self, line):
for table_name, table in PX_TCAM_TABLES.items():
print '-'*len(table_name), '\n', table_name, ':\n', '-'*len(table_name)
pprint(table.info)
def help_list_tcam_tables(self):
print """
list_tcam_tables
DESCRIPTION: List the ternary match tables defined in the SimpleSumeSwitch and their relevant compile time information
"""
def do_table_tcam_clean(self, line):
table_name = line.strip()
p4_tables_api.table_tcam_clean(table_name)
def help_table_tcam_clean(self):
print """
table_tcam_clean <table_name>
DESCRIPTION: performs table self-initialization, erasing and invalidating all stored rules
"""
def do_table_tcam_get_addr_size(self, line):
print p4_tables_api.table_tcam_get_addr_size()
def help_table_tcam_get_addr_size(self):
print """
table_tcam_get_addr_size
DESCRIPTION: returns the TCAM_ADDR_SIZE
"""
def do_table_tcam_set_log_level(self, line):
args = line.strip().split()
if (len(args) < 2):
print >> sys.stderr, "ERROR: usage..."
self.help_table_tcam_set_log_level()
return
table_name = args[0]
try:
msg_level = int(args[1], 0)
except:
print >> sys.stderr, "ERROR: msg_level must be valid int"
return
p4_tables_api.table_tcam_set_log_level(table_name, msg_level)
def help_table_tcam_set_log_level(self):
print """
table_tcam_set_log_level <table_name> <msg_level>
DESCRIPTION: Update the logging level of the table
"""
def do_table_tcam_add_entry(self, line):
# defined in p4_px_tables.py
(table_name, address, keys, masks, action_name, action_data) = parse_table_tcam_add_entry(line)
p4_tables_api.table_tcam_write_entry(table_name, address, keys, masks, action_name, action_data)
def help_table_tcam_add_entry(self):
# defined in p4_px_tables.py
print help_table_tcam_add_entry()
def do_table_tcam_erase_entry(self, line):
args = line.strip().split()
if (len(args) < 2):
print >> sys.stderr, "ERROR: usage..."
self.help_table_tcam_erase_entry()
return
table_name = args[0]
try:
address = int(args[1], 0)
except:
print >> sys.stderr, "ERROR: address must be valid int"
return
p4_tables_api.table_tcam_erase_entry(table_name, address)
def help_table_tcam_erase_entry(self):
print """
table_tcam_erase_entry <table_name> <address>
DESCRIPTION: invalidates(removes) an entry in the TCAM
"""
def do_table_tcam_verify_entry(self, line):
# defined in p4_px_tables.py
(table_name, address, keys, masks, action_name, action_data) = parse_table_tcam_add_entry(line)
rc = p4_tables_api.table_tcam_verify_entry(table_name, address, keys, masks, action_name, action_data)
print p4_tables_api.table_tcam_error_decode(rc)
def help_table_tcam_verify_entry(self):
print """
table_tcam_verify_entry <table_name> <address> <action_name> <key1/mask1 ... keyN/maskN> => <action_data>
DESCRIPTION: verifies whether an entry exists in TCAM
PARAMS:
<table_name> : name of the table to add an entry to
<address> : address in table at which to add the entry
<action_name> : name of the action to use in the entry (must be listed in the table's actions list)
<key/mask> : space separated list of key/mask to use as the entry key (must correspond to table's keys in the order defined in the P4 program)
<action_data> : space separated list of values to provide as input to the action
"""
###########################
### LPM Table Functions ###
###########################
"""
List the LPM tables and some relevant info
"""
def do_list_lpm_tables(self, line):
for table_name, table in PX_LPM_TABLES.items():
print '-'*len(table_name), '\n', table_name, ':\n', '-'*len(table_name)
pprint(table.info)
def help_list_lpm_tables(self):
print """
list_lpm_tables
DESCRIPTION: List the longest prefix match tables defined in the SimpleSumeSwitch and their relevant compile time information
"""
def do_table_lpm_get_addr_size(self, line):
print p4_tables_api.table_lpm_get_addr_size()
def help_table_lpm_get_addr_size(self):
print """
table_lpm_get_addr_size
DESCRIPTION: returns the LPM_ADDR_SIZE
"""
def do_table_lpm_set_log_level(self, line):
args = line.strip().split()
if (len(args) < 2):
print >> sys.stderr, "ERROR: usage..."
self.help_table_lpm_set_log_level()
return
table_name = args[0]
try:
msg_level = int(args[1], 0)
except:
print >> sys.stderr, "ERROR: msg_level must be valid int"
return
p4_tables_api.table_lpm_set_log_level(table_name, msg_level)
def help_table_lpm_set_log_level(self):
print """
table_lpm_set_log_level <table_name> <msg_level>
DESCRIPTION: Update the logging level of the table
"""
def do_table_lpm_load_dataset(self, line):
args = line.strip().split()
if (len(args) < 2):
print >> sys.stderr, "ERROR: usage..."
self.help_table_lpm_load_dataset()
return
table_name = args[0]
filename = args[1]
p4_tables_api.table_lpm_load_dataset(table_name, filename)
def help_table_lpm_load_dataset(self):
print """
table_lpm_load_dataset <table_name> <filename>
DESCRIPTION: Load new dataset into table
"""
def do_table_lpm_verify_dataset(self, line):
args = line.strip().split()
if (len(args) < 2):
print >> sys.stderr, "ERROR: usage..."
self.help_table_lpm_verify_dataset()
return
table_name = args[0]
filename = args[1]
rc = p4_tables_api.table_lpm_verify_dataset(table_name, filename)
print p4_tables_api.table_lpm_error_decode(rc)
def help_table_lpm_verify_dataset(self):
print """
table_lpm_verify_dataset <table_name> <filename>
DESCRIPTION: Verify that dataset is in table
"""
def do_table_lpm_set_active_lookup_bank(self, line):
args = line.strip().split()
if (len(args) < 2):
print >> sys.stderr, "ERROR: usage..."
self.help_table_lpm_set_log_level()
return
table_name = args[0]
try:
bank = int(args[1], 0)
except:
print >> sys.stderr, "ERROR: bank must be valid int"
return
p4_tables_api.table_lpm_set_active_lookup_bank(table_name, bank)
def help_table_lpm_set_active_lookup_bank(self):
print """
table_lpm_set_log_level <table_name> <bank>
DESCRIPTION: sets the active lookup bank
"""
if __name__ == '__main__':
if len(sys.argv) > 1:
SimpleSumeSwitch().onecmd(' '.join(sys.argv[1:]))
else:
SimpleSumeSwitch().cmdloop()