Skip to content

Commit

Permalink
Support of is_net_port and is_host_port externs.
Browse files Browse the repository at this point in the history
Signed-off-by: Y <[email protected]>
  • Loading branch information
komaljai committed Jul 31, 2024
1 parent 9d4e3c1 commit 2bbfcd5
Show file tree
Hide file tree
Showing 17 changed files with 1,665 additions and 2 deletions.
17 changes: 16 additions & 1 deletion backends/tc/ebpfCodeGen.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1987,14 +1987,29 @@ void ControlBodyTranslatorPNA::processMethod(const P4::ExternMethod *method) {

bool ControlBodyTranslatorPNA::preorder(const IR::AssignmentStatement *a) {
if (auto methodCallExpr = a->right->to<IR::MethodCallExpression>()) {
if (methodCallExpr->method->toString() == "is_net_port" ||
methodCallExpr->method->toString() == "is_host_port") {
builder->emitIndent();
if (methodCallExpr->method->toString() == "is_net_port") {
builder->append("bpf_p4tc_is_net_port(skb, ");
} else {
builder->append("bpf_p4tc_is_host_port(skb, ");
}
if (methodCallExpr->arguments->size() == 1) {
visit(methodCallExpr->arguments->at(0));
builder->append(");");
} else {
::error(ErrorType::ERR_UNEXPECTED, "Expected exactly 1 argument %1%",
methodCallExpr);
}
}
auto mi = P4::MethodInstance::resolve(methodCallExpr, control->program->refMap,
control->program->typeMap);
auto ext = mi->to<P4::ExternMethod>();
auto pnaControl = dynamic_cast<const EBPFControlPNA *>(control);
if (ext == nullptr) {
return false;
}

if (ext->originalExternType->name.name == "Register" && ext->method->type->name == "read") {
cstring name = EBPF::EBPFObject::externalName(ext->object);
auto reg = pnaControl->getRegister(name);
Expand Down
1 change: 0 additions & 1 deletion backends/tc/ebpfCodeGen.h
Original file line number Diff line number Diff line change
Expand Up @@ -354,7 +354,6 @@ class ControlBodyTranslatorPNA : public EBPF::ControlBodyTranslator {
const EBPF::EBPFTablePSA *table);
void processFunction(const P4::ExternFunction *function) override;
void processApply(const P4::ApplyMethod *method) override;
bool checkPnaPortMem(const IR::Member *m);
virtual cstring getParamName(const IR::PathExpression *);
bool preorder(const IR::AssignmentStatement *a) override;
void processMethod(const P4::ExternMethod *method) override;
Expand Down
21 changes: 21 additions & 0 deletions backends/tc/runtime/pna.h
Original file line number Diff line number Diff line change
Expand Up @@ -353,6 +353,27 @@ bpf_p4tc_ext_hash_base_16bit_complement(const void *data, const u32 data__sz,
return (base + (hash % max));
}

/* DPDK doesn't implement this yet */
/* Equivalent to PNA is_net_port */
extern bool
bpf_p4tc_is_net_port(struct __sk_buff *skb_ctx, const u32 ifindex) __ksym;

/* DPDK doesn't implement this yet */
/* Equivalent to PNA is_host_port */
extern bool
bpf_p4tc_is_host_port(struct __sk_buff *skb_ctx, const u32 ifindex) __ksym;

/* DPDK doesn't implement this yet */
/* Equivalent to PNA is_net_port */
extern bool
xdp_p4tc_is_net_port(struct xdp_md *xdp_ctx, const u32 ifindex) __ksym;

/* DPDK doesn't implement this yet */
/* Equivalent to PNA is_host_port */
extern bool
xdp_p4tc_is_host_port(struct xdp_md *xdp_ctx, const u32 ifindex) __ksym;


/* per extern specifics start */

/* in this case it is PNA so we have these helpers like below
Expand Down
128 changes: 128 additions & 0 deletions testdata/p4tc_samples/is_host_port.p4
Original file line number Diff line number Diff line change
@@ -0,0 +1,128 @@
/* -*- P4_16 -*- */

#include <core.p4>
#include <tc/pna.p4>

#define PORT_TABLE_SIZE 262144

/*
* Standard ethernet header
*/
header ethernet_t {
@tc_type ("macaddr") bit<48> dstAddr;
@tc_type ("macaddr") bit<48> srcAddr;
bit<16> etherType;
}

header ipv4_t {
bit<4> version;
bit<4> ihl;
bit<8> diffserv;
bit<16> totalLen;
bit<16> identification;
bit<3> flags;
bit<13> fragOffset;
bit<8> ttl;
bit<8> protocol;
bit<16> hdrChecksum;
bit<32> srcAddr;
bit<32> dstAddr;
}

struct my_ingress_headers_t {
ethernet_t ethernet;
ipv4_t ipv4;
}

/****** G L O B A L I N G R E S S M E T A D A T A *********/

struct my_ingress_metadata_t {
}

struct empty_metadata_t {
}

/*********************** P A R S E R **************************/

parser Ingress_Parser(
packet_in pkt,
out my_ingress_headers_t hdr,
inout my_ingress_metadata_t meta,
in pna_main_parser_input_metadata_t istd)
{
const bit<16> ETHERTYPE_IPV4 = 0x0800;

state start {
transition parse_ethernet;
}
state parse_ethernet {
pkt.extract(hdr.ethernet);
transition select(hdr.ethernet.etherType) {
ETHERTYPE_IPV4 : parse_ipv4;
default : reject;
}
}
state parse_ipv4 {
pkt.extract(hdr.ipv4);
transition accept;
}
}

/***************** M A T C H - A C T I O N *********************/

control ingress(
inout my_ingress_headers_t hdr,
inout my_ingress_metadata_t meta,
in pna_main_input_metadata_t istd,
inout pna_main_output_metadata_t ostd
)
{
action send_nh(@tc_type("dev") PortId_t port_id, @tc_type("macaddr") bit<48> dmac, @tc_type("macaddr") bit<48> smac) {
hdr.ethernet.srcAddr = smac;
hdr.ethernet.dstAddr = dmac;
send_to_port(port_id);
}
action drop() {
drop_packet();
}

@tc_acl("RUS:RXP") table nh_table {
key = {
hdr.ipv4.srcAddr : exact @tc_type ("ipv4");
}
actions = {
send_nh;
drop;
}
size = PORT_TABLE_SIZE;
const default_action = drop;
}

apply {
if (hdr.ipv4.isValid() && is_host_port(istd.input_port)) {
nh_table.apply();
}
}
}

/********************* D E P A R S E R ************************/

control Ingress_Deparser(
packet_out pkt,
inout my_ingress_headers_t hdr,
in my_ingress_metadata_t meta,
in pna_main_output_metadata_t ostd)
{
apply {
pkt.emit(hdr.ethernet);
pkt.emit(hdr.ipv4);
}
}

/************ F I N A L P A C K A G E ******************************/

PNA_NIC(
Ingress_Parser(),
ingress(),
Ingress_Deparser()
) main;
128 changes: 128 additions & 0 deletions testdata/p4tc_samples/is_net_port.p4
Original file line number Diff line number Diff line change
@@ -0,0 +1,128 @@
/* -*- P4_16 -*- */

#include <core.p4>
#include <tc/pna.p4>

#define PORT_TABLE_SIZE 262144

/*
* Standard ethernet header
*/
header ethernet_t {
@tc_type ("macaddr") bit<48> dstAddr;
@tc_type ("macaddr") bit<48> srcAddr;
bit<16> etherType;
}

header ipv4_t {
bit<4> version;
bit<4> ihl;
bit<8> diffserv;
bit<16> totalLen;
bit<16> identification;
bit<3> flags;
bit<13> fragOffset;
bit<8> ttl;
bit<8> protocol;
bit<16> hdrChecksum;
bit<32> srcAddr;
bit<32> dstAddr;
}

struct my_ingress_headers_t {
ethernet_t ethernet;
ipv4_t ipv4;
}

/****** G L O B A L I N G R E S S M E T A D A T A *********/

struct my_ingress_metadata_t {
}

struct empty_metadata_t {
}

/*********************** P A R S E R **************************/

parser Ingress_Parser(
packet_in pkt,
out my_ingress_headers_t hdr,
inout my_ingress_metadata_t meta,
in pna_main_parser_input_metadata_t istd)
{
const bit<16> ETHERTYPE_IPV4 = 0x0800;

state start {
transition parse_ethernet;
}
state parse_ethernet {
pkt.extract(hdr.ethernet);
transition select(hdr.ethernet.etherType) {
ETHERTYPE_IPV4 : parse_ipv4;
default : reject;
}
}
state parse_ipv4 {
pkt.extract(hdr.ipv4);
transition accept;
}
}

/***************** M A T C H - A C T I O N *********************/

control ingress(
inout my_ingress_headers_t hdr,
inout my_ingress_metadata_t meta,
in pna_main_input_metadata_t istd,
inout pna_main_output_metadata_t ostd
)
{
action send_nh(@tc_type("dev") PortId_t port_id, @tc_type("macaddr") bit<48> dmac, @tc_type("macaddr") bit<48> smac) {
hdr.ethernet.srcAddr = smac;
hdr.ethernet.dstAddr = dmac;
send_to_port(port_id);
}
action drop() {
drop_packet();
}

@tc_acl("RUS:RXP") table nh_table {
key = {
hdr.ipv4.srcAddr : exact @tc_type ("ipv4");
}
actions = {
send_nh;
drop;
}
size = PORT_TABLE_SIZE;
const default_action = drop;
}

apply {
if (hdr.ipv4.isValid() && is_net_port(istd.input_port)) {
nh_table.apply();
}
}
}

/********************* D E P A R S E R ************************/

control Ingress_Deparser(
packet_out pkt,
inout my_ingress_headers_t hdr,
in my_ingress_metadata_t meta,
in pna_main_output_metadata_t ostd)
{
apply {
pkt.emit(hdr.ethernet);
pkt.emit(hdr.ipv4);
}
}

/************ F I N A L P A C K A G E ******************************/

PNA_NIC(
Ingress_Parser(),
ingress(),
Ingress_Deparser()
) main;
63 changes: 63 additions & 0 deletions testdata/p4tc_samples_outputs/is_host_port.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,63 @@
{
"schema_version" : "1.0.0",
"pipeline_name" : "is_host_port",
"externs" : [],
"tables" : [
{
"name" : "ingress/nh_table",
"id" : 1,
"tentries" : 262144,
"permissions" : "0x18a6",
"nummask" : 8,
"keysize" : 32,
"keyfields" : [
{
"id" : 1,
"name" : "hdr.ipv4.srcAddr",
"type" : "ipv4",
"match_type" : "exact",
"bitwidth" : 32
}
],
"actions" : [
{
"id" : 1,
"name" : "ingress/send_nh",
"action_scope" : "TableAndDefault",
"annotations" : [],
"params" : [
{
"id" : 1,
"name" : "port_id",
"type" : "dev",
"bitwidth" : 32
},
{
"id" : 2,
"name" : "dmac",
"type" : "macaddr",
"bitwidth" : 48
},
{
"id" : 3,
"name" : "smac",
"type" : "macaddr",
"bitwidth" : 48
}
],
"default_hit_action" : false,
"default_miss_action" : false
},
{
"id" : 2,
"name" : "ingress/drop",
"action_scope" : "TableAndDefault",
"annotations" : [],
"params" : [],
"default_hit_action" : false,
"default_miss_action" : true
}
]
}
]
}
Empty file.
Loading

0 comments on commit 2bbfcd5

Please sign in to comment.