Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit fe878cad authored by Sowmini Varadhan's avatar Sowmini Varadhan Committed by David S. Miller
Browse files

tools: test case for TPACKET_V3/TX_RING support 



Add a test case and sample code for (TPACKET_V3, PACKET_TX_RING)

Signed-off-by: default avatarSowmini Varadhan <sowmini.varadhan@oracle.com>
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
parent 7f953ab2

tools/testing/selftests/net/psock_tpacket.c

+74 −17
Original line number Diff line number Diff line
@@ -311,20 +311,33 @@ static inline void __v2_tx_user_ready(struct tpacket2_hdr *hdr) 
	__sync_synchronize();

}



static inline int __v1_v2_tx_kernel_ready(void *base, int version)

static inline int __v3_tx_kernel_ready(struct tpacket3_hdr *hdr)

{

	return !(hdr->tp_status & (TP_STATUS_SEND_REQUEST | TP_STATUS_SENDING));

}



static inline void __v3_tx_user_ready(struct tpacket3_hdr *hdr)

{

	hdr->tp_status = TP_STATUS_SEND_REQUEST;

	__sync_synchronize();

}



static inline int __tx_kernel_ready(void *base, int version)

{

	switch (version) {

	case TPACKET_V1:

		return __v1_tx_kernel_ready(base);

	case TPACKET_V2:

		return __v2_tx_kernel_ready(base);

	case TPACKET_V3:

		return __v3_tx_kernel_ready(base);

	default:

		bug_on(1);

		return 0;

	}

}



static inline void __v1_v2_tx_user_ready(void *base, int version)

static inline void __tx_user_ready(void *base, int version)

{

	switch (version) {

	case TPACKET_V1:
@@ -333,6 +346,9 @@ static inline void __v1_v2_tx_user_ready(void *base, int version) 
	case TPACKET_V2:

		__v2_tx_user_ready(base);

		break;

	case TPACKET_V3:

		__v3_tx_user_ready(base);

		break;

	}

}
@@ -348,7 +364,22 @@ static void __v1_v2_set_packet_loss_discard(int sock) 
	}

}



static void walk_v1_v2_tx(int sock, struct ring *ring)

static inline void *get_next_frame(struct ring *ring, int n)

{

	uint8_t *f0 = ring->rd[0].iov_base;



	switch (ring->version) {

	case TPACKET_V1:

	case TPACKET_V2:

		return ring->rd[n].iov_base;

	case TPACKET_V3:

		return f0 + (n * ring->req3.tp_frame_size);

	default:

		bug_on(1);

	}

}



static void walk_tx(int sock, struct ring *ring)

{

	struct pollfd pfd;

	int rcv_sock, ret;
@@ -360,9 +391,19 @@ static void walk_v1_v2_tx(int sock, struct ring *ring) 
		.sll_family = PF_PACKET,

		.sll_halen = ETH_ALEN,

	};

	int nframes;



	/* TPACKET_V{1,2} sets up the ring->rd* related variables based

	 * on frames (e.g., rd_num is tp_frame_nr) whereas V3 sets these

	 * up based on blocks (e.g, rd_num is  tp_block_nr)

	 */

	if (ring->version <= TPACKET_V2)

		nframes = ring->rd_num;

	else

		nframes = ring->req3.tp_frame_nr;



	bug_on(ring->type != PACKET_TX_RING);

	bug_on(ring->rd_num < NUM_PACKETS);

	bug_on(nframes < NUM_PACKETS);



	rcv_sock = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_ALL));

	if (rcv_sock == -1) {
@@ -388,10 +429,11 @@ static void walk_v1_v2_tx(int sock, struct ring *ring) 
	create_payload(packet, &packet_len);



	while (total_packets > 0) {

		while (__v1_v2_tx_kernel_ready(ring->rd[frame_num].iov_base,

					       ring->version) &&

		void *next = get_next_frame(ring, frame_num);



		while (__tx_kernel_ready(next, ring->version) &&

		       total_packets > 0) {

			ppd.raw = ring->rd[frame_num].iov_base;

			ppd.raw = next;



			switch (ring->version) {

			case TPACKET_V1:
@@ -413,14 +455,27 @@ static void walk_v1_v2_tx(int sock, struct ring *ring) 
				       packet_len);

				total_bytes += ppd.v2->tp_h.tp_snaplen;

				break;

			case TPACKET_V3: {

				struct tpacket3_hdr *tx = next;



				tx->tp_snaplen = packet_len;

				tx->tp_len = packet_len;

				tx->tp_next_offset = 0;



				memcpy((uint8_t *)tx + TPACKET3_HDRLEN -

				       sizeof(struct sockaddr_ll), packet,

				       packet_len);

				total_bytes += tx->tp_snaplen;

				break;

			}

			}



			status_bar_update();

			total_packets--;



			__v1_v2_tx_user_ready(ppd.raw, ring->version);

			__tx_user_ready(next, ring->version);



			frame_num = (frame_num + 1) % ring->rd_num;

			frame_num = (frame_num + 1) % nframes;

		}



		poll(&pfd, 1, 1);
@@ -460,7 +515,7 @@ static void walk_v1_v2(int sock, struct ring *ring) 
	if (ring->type == PACKET_RX_RING)

		walk_v1_v2_rx(sock, ring);

	else

		walk_v1_v2_tx(sock, ring);

		walk_tx(sock, ring);

}



static uint64_t __v3_prev_block_seq_num = 0;
@@ -583,7 +638,7 @@ static void walk_v3(int sock, struct ring *ring) 
	if (ring->type == PACKET_RX_RING)

		walk_v3_rx(sock, ring);

	else

		bug_on(1);

		walk_tx(sock, ring);

}



static void __v1_v2_fill(struct ring *ring, unsigned int blocks)
@@ -602,12 +657,13 @@ static void __v1_v2_fill(struct ring *ring, unsigned int blocks) 
	ring->flen = ring->req.tp_frame_size;

}



static void __v3_fill(struct ring *ring, unsigned int blocks)

static void __v3_fill(struct ring *ring, unsigned int blocks, int type)

{

	if (type == PACKET_RX_RING) {

		ring->req3.tp_retire_blk_tov = 64;

		ring->req3.tp_sizeof_priv = 0;

		ring->req3.tp_feature_req_word = TP_FT_REQ_FILL_RXHASH;



	}

	ring->req3.tp_block_size = getpagesize() << 2;

	ring->req3.tp_frame_size = TPACKET_ALIGNMENT << 7;

	ring->req3.tp_block_nr = blocks;
@@ -641,7 +697,7 @@ static void setup_ring(int sock, struct ring *ring, int version, int type) 
		break;



	case TPACKET_V3:

		__v3_fill(ring, blocks);

		__v3_fill(ring, blocks, type);

		ret = setsockopt(sock, SOL_PACKET, type, &ring->req3,

				 sizeof(ring->req3));

		break;
@@ -796,6 +852,7 @@ int main(void) 
	ret |= test_tpacket(TPACKET_V2, PACKET_TX_RING);



	ret |= test_tpacket(TPACKET_V3, PACKET_RX_RING);

	ret |= test_tpacket(TPACKET_V3, PACKET_TX_RING);



	if (ret)

		return 1;