xref: /linux/drivers/net/ethernet/amd/xgbe/xgbe-selftest.c (revision 862a64c83faf7708e7e79498193ff5270543a68d)

// SPDX-License-Identifier: (GPL-2.0-or-later OR BSD-3-Clause)
/*
 * Copyright (c) 2014-2025, Advanced Micro Devices, Inc.
 * Copyright (c) 2014, Synopsys, Inc.
 * All rights reserved
 *
 * Author: Raju Rangoju <Raju.Rangoju@amd.com>
 */
#include <linux/crc32.h>
#include <linux/ip.h>
#include <linux/udp.h>
#include <net/tcp.h>
#include <net/udp.h>
#include <net/checksum.h>
#include <net/selftests.h>

#include "xgbe.h"
#include "xgbe-common.h"

#define XGBE_LOOPBACK_NONE	0
#define XGBE_LOOPBACK_MAC	1

struct xgbe_test {
	char name[ETH_GSTRING_LEN];
	int lb;
	int (*fn)(struct xgbe_prv_data *pdata);
};

static u8 xgbe_test_id;

static int xgbe_test_loopback_validate(struct sk_buff *skb,
				       struct net_device *ndev,
				       struct packet_type *pt,
				       struct net_device *orig_ndev)
{
	struct net_test_priv *tdata = pt->af_packet_priv;
	const unsigned char *dst = tdata->packet->dst;
	const unsigned char *src = tdata->packet->src;
	struct netsfhdr *hdr;
	struct ethhdr *eh;
	struct tcphdr *th;
	struct udphdr *uh;
	struct iphdr *ih;

	skb = skb_unshare(skb, GFP_ATOMIC);
	if (!skb)
		goto out;

	if (skb_linearize(skb))
		goto out;

	if (skb_headlen(skb) < (NET_TEST_PKT_SIZE - ETH_HLEN))
		goto out;

	eh = (struct ethhdr *)skb_mac_header(skb);
	if (dst) {
		if (!ether_addr_equal_unaligned(eh->h_dest, dst))
			goto out;
	}
	if (src) {
		if (!ether_addr_equal_unaligned(eh->h_source, src))
			goto out;
	}

	ih = ip_hdr(skb);

	if (tdata->packet->tcp) {
		if (ih->protocol != IPPROTO_TCP)
			goto out;

		th = (struct tcphdr *)((u8 *)ih + 4 * ih->ihl);
		if (th->dest != htons(tdata->packet->dport))
			goto out;

		hdr = (struct netsfhdr *)((u8 *)th + sizeof(*th));
	} else {
		if (ih->protocol != IPPROTO_UDP)
			goto out;

		uh = (struct udphdr *)((u8 *)ih + 4 * ih->ihl);
		if (uh->dest != htons(tdata->packet->dport))
			goto out;

		hdr = (struct netsfhdr *)((u8 *)uh + sizeof(*uh));
	}

	if (hdr->magic != cpu_to_be64(NET_TEST_PKT_MAGIC))
		goto out;
	if (tdata->packet->id != hdr->id)
		goto out;

	tdata->ok = true;
	complete(&tdata->comp);
out:
	kfree_skb(skb);
	return 0;
}

static int __xgbe_test_loopback(struct xgbe_prv_data *pdata,
				struct net_packet_attrs *attr)
{
	struct net_test_priv *tdata;
	struct sk_buff *skb = NULL;
	int ret = 0;

	tdata = kzalloc(sizeof(*tdata), GFP_KERNEL);
	if (!tdata)
		return -ENOMEM;

	tdata->ok = false;
	init_completion(&tdata->comp);

	tdata->pt.type = htons(ETH_P_IP);
	tdata->pt.func = xgbe_test_loopback_validate;
	tdata->pt.dev = pdata->netdev;
	tdata->pt.af_packet_priv = tdata;
	tdata->packet = attr;

	dev_add_pack(&tdata->pt);

	skb = net_test_get_skb(pdata->netdev, xgbe_test_id, attr);
	if (!skb) {
		ret = -ENOMEM;
		goto cleanup;
	}

	xgbe_test_id++;
	ret = dev_direct_xmit(skb, attr->queue_mapping);
	if (ret)
		goto cleanup;

	if (!attr->timeout)
		attr->timeout = NET_LB_TIMEOUT;

	wait_for_completion_timeout(&tdata->comp, attr->timeout);
	ret = tdata->ok ? 0 : -ETIMEDOUT;

	if (ret)
		netdev_err(pdata->netdev, "Response timedout: ret %d\n", ret);
cleanup:
	dev_remove_pack(&tdata->pt);
	kfree(tdata);
	return ret;
}

static int xgbe_test_mac_loopback(struct xgbe_prv_data *pdata)
{
	struct net_packet_attrs attr = {};

	attr.dst = pdata->netdev->dev_addr;
	return __xgbe_test_loopback(pdata, &attr);
}

static const struct xgbe_test xgbe_selftests[] = {
	{
		.name = "MAC Loopback   ",
		.lb = XGBE_LOOPBACK_MAC,
		.fn = xgbe_test_mac_loopback,
	},
};

void xgbe_selftest_run(struct net_device *dev,
		       struct ethtool_test *etest, u64 *buf)
{
	struct xgbe_prv_data *pdata = netdev_priv(dev);
	int count = xgbe_selftest_get_count(pdata);
	int i, ret;

	memset(buf, 0, sizeof(*buf) * count);
	xgbe_test_id = 0;

	if (etest->flags != ETH_TEST_FL_OFFLINE) {
		netdev_err(pdata->netdev, "Only offline tests are supported\n");
		etest->flags |= ETH_TEST_FL_FAILED;
		return;
	} else if (!netif_carrier_ok(dev)) {
		netdev_err(pdata->netdev,
			   "Invalid link, cannot execute tests\n");
		etest->flags |= ETH_TEST_FL_FAILED;
		return;
	}

	/* Wait for queues drain */
	msleep(200);

	for (i = 0; i < count; i++) {
		ret = 0;

		switch (xgbe_selftests[i].lb) {
		case XGBE_LOOPBACK_MAC:
			ret = xgbe_enable_mac_loopback(pdata);
			break;
		case XGBE_LOOPBACK_NONE:
			break;
		default:
			ret = -EOPNOTSUPP;
			break;
		}

		/*
		 * First tests will always be MAC / PHY loopback.
		 * If any of them is not supported we abort earlier.
		 */
		if (ret) {
			netdev_err(pdata->netdev, "Loopback not supported\n");
			etest->flags |= ETH_TEST_FL_FAILED;
			break;
		}

		ret = xgbe_selftests[i].fn(pdata);
		if (ret && (ret != -EOPNOTSUPP))
			etest->flags |= ETH_TEST_FL_FAILED;
		buf[i] = ret;

		switch (xgbe_selftests[i].lb) {
		case XGBE_LOOPBACK_MAC:
			xgbe_disable_mac_loopback(pdata);
			break;
		default:
			break;
		}
	}
}

void xgbe_selftest_get_strings(struct xgbe_prv_data *pdata, u8 *data)
{
	u8 *p = data;
	int i;

	for (i = 0; i < xgbe_selftest_get_count(pdata); i++)
		ethtool_puts(&p, xgbe_selftests[i].name);
}

int xgbe_selftest_get_count(struct xgbe_prv_data *pdata)
{
	return ARRAY_SIZE(xgbe_selftests);
}