Skip to content

Commit

Permalink
Fix incorrect field extraction arithmetic when multiple non-byte alig…
Browse files Browse the repository at this point in the history
…ned header fields are combined into single byte aligned field (#4301)
  • Loading branch information
usha1830 authored Dec 20, 2023
1 parent 7b84f38 commit 9bad3cb
Show file tree
Hide file tree
Showing 28 changed files with 1,420 additions and 803 deletions.
7 changes: 3 additions & 4 deletions backends/dpdk/dpdkArch.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -553,8 +553,8 @@ const IR::Node *AlignHdrMetaField::preorder(IR::Type_StructLike *st) {
fieldObj.headerStr = st->name.name;
fieldObj.modifiedWidth = size_sum_so_far;
fieldObj.fieldWidth = s->second.fieldWidth;
fieldObj.lsb = offset;
fieldObj.msb = offset + s->second.fieldWidth - 1;
fieldObj.lsb = size_sum_so_far - (offset + fieldObj.fieldWidth);
fieldObj.msb = fieldObj.lsb + s->second.fieldWidth - 1;
fieldObj.offset = offset;
structure->hdrFieldInfoList[s->first].push_back(fieldObj);
offset += s->second.fieldWidth;
Expand Down Expand Up @@ -620,8 +620,7 @@ const IR::Node *AlignHdrMetaField::preorder(IR::Member *m) {
two different headers have field with same name */
if (memVec.headerStr != hdrStrName) continue;
auto mem = new IR::Member(m->expr, IR::ID(memVec.modifiedName));
auto sliceMem =
new IR::Slice(mem->clone(), (memVec.offset + memVec.fieldWidth - 1), memVec.offset);
auto sliceMem = new IR::Slice(mem->clone(), memVec.msb, memVec.lsb);
return sliceMem;
}
}
Expand Down
134 changes: 134 additions & 0 deletions testdata/p4_16_samples/pna-dpdk-parser-wrong-arith.p4
Original file line number Diff line number Diff line change
@@ -0,0 +1,134 @@
/*
Copyright 2023 Intel Corporation
Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
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.
*/

#include <core.p4>
#include "dpdk/pna.p4"


typedef bit<48> EthernetAddress;

header ethernet_t {
EthernetAddress dstAddr;
EthernetAddress 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 user_metadata_t {
bit<1> ipv4_hdr_truncated;
}

struct headers_t {
ethernet_t ethernet;
ipv4_t ipv4;
}

control PreControlImpl(
in headers_t hdr,
inout user_metadata_t meta,
in pna_pre_input_metadata_t istd,
inout pna_pre_output_metadata_t ostd)
{
apply {
}
}

parser MainParserImpl(
packet_in pkt,
out headers_t hdr,
inout user_metadata_t user_meta,
in pna_main_parser_input_metadata_t istd)
{
state start {
user_meta.ipv4_hdr_truncated = 0;
pkt.extract(hdr.ethernet);
transition select(hdr.ethernet.etherType) {
0x0800: parse_ipv4;
default: accept;
}
}
state parse_ipv4 {
bit<4> ihl = pkt.lookahead<ipv4_t>().ihl;
transition select (ihl) {
0x0 &&& 0xC : parse_ipv4_ihl_too_small; //0-3
0x4 : parse_ipv4_ihl_too_small;
default : parse_ipv4_ok;
}
}
state parse_ipv4_ihl_too_small {
user_meta.ipv4_hdr_truncated = 1;
transition accept;
}
state parse_ipv4_ok {
pkt.extract(hdr.ipv4);
transition accept;
}
}

control MainControlImpl(
inout headers_t hdr, // from main parser
inout user_metadata_t user_meta, // from main parser, to "next block"
in pna_main_input_metadata_t istd,
inout pna_main_output_metadata_t ostd)
{
apply {
if (hdr.ethernet.isValid()) {
bit<8> tmp;
tmp = 0;
tmp[0:0] = user_meta.ipv4_hdr_truncated;
tmp[1:1] = (bit<1>) hdr.ipv4.isValid();
hdr.ethernet.srcAddr[7:0] = tmp;
}
}
}

control MainDeparserImpl(
packet_out pkt,
in headers_t hdr, // from main control
in user_metadata_t user_meta, // from main control
in pna_main_output_metadata_t ostd)
{
apply {
pkt.emit(hdr.ethernet);
pkt.emit(hdr.ipv4);
}
}

// BEGIN:Package_Instantiation_Example
PNA_NIC(
MainParserImpl(),
PreControlImpl(),
MainControlImpl(),
MainDeparserImpl()
// Hoping to make this optional parameter later, but not supported
// by p4c yet.
//, PreParserImpl()
) main;
// END:Package_Instantiation_Example
Original file line number Diff line number Diff line change
Expand Up @@ -122,38 +122,38 @@ action encap_one_tunnel_layer_ipv4 args instanceof encap_one_tunnel_layer_ipv4_a
mov h.mac.type 0x800
validate h.ipv4_0
mov m.MainControlT_tmp h.ipv4_0.version_ihl
and m.MainControlT_tmp 0xF0
and m.MainControlT_tmp 0xF
mov h.ipv4_0.version_ihl m.MainControlT_tmp
or h.ipv4_0.version_ihl 0x4
or h.ipv4_0.version_ihl 0x40
mov m.MainControlT_tmp_0 h.ipv4_0.version_ihl
and m.MainControlT_tmp_0 0xF
and m.MainControlT_tmp_0 0xF0
mov h.ipv4_0.version_ihl m.MainControlT_tmp_0
or h.ipv4_0.version_ihl 0x50
or h.ipv4_0.version_ihl 0x5
mov m.MainControlT_tmp_1 h.ipv4_0.dscp_ecn
and m.MainControlT_tmp_1 0xC0
and m.MainControlT_tmp_1 0x3
mov h.ipv4_0.dscp_ecn m.MainControlT_tmp_1
or h.ipv4_0.dscp_ecn 0x5
or h.ipv4_0.dscp_ecn 0x14
mov m.MainControlT_tmp_2 h.ipv4_0.dscp_ecn
and m.MainControlT_tmp_2 0x3F
and m.MainControlT_tmp_2 0xFC
mov h.ipv4_0.dscp_ecn m.MainControlT_tmp_2
or h.ipv4_0.dscp_ecn 0x0
mov h.ipv4_0.length 0x14
add h.ipv4_0.length m.local_metadata_L2_packet_len_bytes
mov h.ipv4_0.identification 0x0
mov m.MainControlT_tmp_3 h.ipv4_0.rsvd_df_mf_frag_off
and m.MainControlT_tmp_3 0xFFFE
and m.MainControlT_tmp_3 0x7FFF
mov h.ipv4_0.rsvd_df_mf_frag_off m.MainControlT_tmp_3
or h.ipv4_0.rsvd_df_mf_frag_off 0x0
mov m.MainControlT_tmp_4 h.ipv4_0.rsvd_df_mf_frag_off
and m.MainControlT_tmp_4 0xFFFD
and m.MainControlT_tmp_4 0xBFFF
mov h.ipv4_0.rsvd_df_mf_frag_off m.MainControlT_tmp_4
or h.ipv4_0.rsvd_df_mf_frag_off 0x0
mov m.MainControlT_tmp_5 h.ipv4_0.rsvd_df_mf_frag_off
and m.MainControlT_tmp_5 0xFFFB
and m.MainControlT_tmp_5 0xDFFF
mov h.ipv4_0.rsvd_df_mf_frag_off m.MainControlT_tmp_5
or h.ipv4_0.rsvd_df_mf_frag_off 0x0
mov m.MainControlT_tmp_6 h.ipv4_0.rsvd_df_mf_frag_off
and m.MainControlT_tmp_6 0x7
and m.MainControlT_tmp_6 0xE000
mov h.ipv4_0.rsvd_df_mf_frag_off m.MainControlT_tmp_6
or h.ipv4_0.rsvd_df_mf_frag_off 0x0
mov h.ipv4_0.ttl 0x40
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,86 @@
#include <core.p4>
#include <dpdk/pna.p4>

typedef bit<48> EthernetAddress;
header ethernet_t {
EthernetAddress dstAddr;
EthernetAddress 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 user_metadata_t {
bit<1> ipv4_hdr_truncated;
}

struct headers_t {
ethernet_t ethernet;
ipv4_t ipv4;
}

control PreControlImpl(in headers_t hdr, inout user_metadata_t meta, in pna_pre_input_metadata_t istd, inout pna_pre_output_metadata_t ostd) {
apply {
}
}

parser MainParserImpl(packet_in pkt, out headers_t hdr, inout user_metadata_t user_meta, in pna_main_parser_input_metadata_t istd) {
state start {
user_meta.ipv4_hdr_truncated = 1w0;
pkt.extract<ethernet_t>(hdr.ethernet);
transition select(hdr.ethernet.etherType) {
16w0x800: parse_ipv4;
default: accept;
}
}
state parse_ipv4 {
bit<4> ihl = (pkt.lookahead<ipv4_t>()).ihl;
transition select(ihl) {
4w0x0 &&& 4w0xc: parse_ipv4_ihl_too_small;
4w0x4: parse_ipv4_ihl_too_small;
default: parse_ipv4_ok;
}
}
state parse_ipv4_ihl_too_small {
user_meta.ipv4_hdr_truncated = 1w1;
transition accept;
}
state parse_ipv4_ok {
pkt.extract<ipv4_t>(hdr.ipv4);
transition accept;
}
}

control MainControlImpl(inout headers_t hdr, inout user_metadata_t user_meta, in pna_main_input_metadata_t istd, inout pna_main_output_metadata_t ostd) {
apply {
if (hdr.ethernet.isValid()) {
bit<8> tmp;
tmp = 8w0;
tmp[0:0] = user_meta.ipv4_hdr_truncated;
tmp[1:1] = (bit<1>)hdr.ipv4.isValid();
hdr.ethernet.srcAddr[7:0] = tmp;
}
}
}

control MainDeparserImpl(packet_out pkt, in headers_t hdr, in user_metadata_t user_meta, in pna_main_output_metadata_t ostd) {
apply {
pkt.emit<ethernet_t>(hdr.ethernet);
pkt.emit<ipv4_t>(hdr.ipv4);
}
}

PNA_NIC<headers_t, user_metadata_t, headers_t, user_metadata_t>(MainParserImpl(), PreControlImpl(), MainControlImpl(), MainDeparserImpl()) main;
Original file line number Diff line number Diff line change
@@ -0,0 +1,89 @@
#include <core.p4>
#include <dpdk/pna.p4>

typedef bit<48> EthernetAddress;
header ethernet_t {
EthernetAddress dstAddr;
EthernetAddress 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 user_metadata_t {
bit<1> ipv4_hdr_truncated;
}

struct headers_t {
ethernet_t ethernet;
ipv4_t ipv4;
}

control PreControlImpl(in headers_t hdr, inout user_metadata_t meta, in pna_pre_input_metadata_t istd, inout pna_pre_output_metadata_t ostd) {
apply {
}
}

parser MainParserImpl(packet_in pkt, out headers_t hdr, inout user_metadata_t user_meta, in pna_main_parser_input_metadata_t istd) {
@name("MainParserImpl.ihl") bit<4> ihl_0;
@name("MainParserImpl.tmp") ipv4_t tmp;
state start {
user_meta.ipv4_hdr_truncated = 1w0;
pkt.extract<ethernet_t>(hdr.ethernet);
transition select(hdr.ethernet.etherType) {
16w0x800: parse_ipv4;
default: accept;
}
}
state parse_ipv4 {
tmp = pkt.lookahead<ipv4_t>();
ihl_0 = tmp.ihl;
transition select(ihl_0) {
4w0x0 &&& 4w0xc: parse_ipv4_ihl_too_small;
4w0x4: parse_ipv4_ihl_too_small;
default: parse_ipv4_ok;
}
}
state parse_ipv4_ihl_too_small {
user_meta.ipv4_hdr_truncated = 1w1;
transition accept;
}
state parse_ipv4_ok {
pkt.extract<ipv4_t>(hdr.ipv4);
transition accept;
}
}

control MainControlImpl(inout headers_t hdr, inout user_metadata_t user_meta, in pna_main_input_metadata_t istd, inout pna_main_output_metadata_t ostd) {
@name("MainControlImpl.tmp") bit<8> tmp_0;
apply {
if (hdr.ethernet.isValid()) {
tmp_0 = 8w0;
tmp_0[0:0] = user_meta.ipv4_hdr_truncated;
tmp_0[1:1] = (bit<1>)hdr.ipv4.isValid();
hdr.ethernet.srcAddr[7:0] = tmp_0;
}
}
}

control MainDeparserImpl(packet_out pkt, in headers_t hdr, in user_metadata_t user_meta, in pna_main_output_metadata_t ostd) {
apply {
pkt.emit<ethernet_t>(hdr.ethernet);
pkt.emit<ipv4_t>(hdr.ipv4);
}
}

PNA_NIC<headers_t, user_metadata_t, headers_t, user_metadata_t>(MainParserImpl(), PreControlImpl(), MainControlImpl(), MainDeparserImpl()) main;
Loading

0 comments on commit 9bad3cb

Please sign in to comment.