1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * Copyright (c) 2019 Synopsys, Inc. and/or its affiliates. 4 * stmmac Selftests Support 5 * 6 * Author: Jose Abreu <joabreu@synopsys.com> 7 * 8 * Ported from stmmac by: 9 * Copyright (C) 2021 Oleksij Rempel <o.rempel@pengutronix.de> 10 */ 11 12 #include <linux/phy.h> 13 #include <net/selftests.h> 14 #include <net/tcp.h> 15 #include <net/udp.h> 16 17 struct net_packet_attrs { 18 const unsigned char *src; 19 const unsigned char *dst; 20 u32 ip_src; 21 u32 ip_dst; 22 bool tcp; 23 u16 sport; 24 u16 dport; 25 int timeout; 26 int size; 27 int max_size; 28 u8 id; 29 u16 queue_mapping; 30 }; 31 32 struct net_test_priv { 33 struct net_packet_attrs *packet; 34 struct packet_type pt; 35 struct completion comp; 36 int double_vlan; 37 int vlan_id; 38 int ok; 39 }; 40 41 struct netsfhdr { 42 __be32 version; 43 __be64 magic; 44 u8 id; 45 } __packed; 46 47 static u8 net_test_next_id; 48 49 #define NET_TEST_PKT_SIZE (sizeof(struct ethhdr) + sizeof(struct iphdr) + \ 50 sizeof(struct netsfhdr)) 51 #define NET_TEST_PKT_MAGIC 0xdeadcafecafedeadULL 52 #define NET_LB_TIMEOUT msecs_to_jiffies(200) 53 54 static struct sk_buff *net_test_get_skb(struct net_device *ndev, 55 struct net_packet_attrs *attr) 56 { 57 struct sk_buff *skb = NULL; 58 struct udphdr *uhdr = NULL; 59 struct tcphdr *thdr = NULL; 60 struct netsfhdr *shdr; 61 struct ethhdr *ehdr; 62 struct iphdr *ihdr; 63 int iplen, size; 64 65 size = attr->size + NET_TEST_PKT_SIZE; 66 67 if (attr->tcp) 68 size += sizeof(struct tcphdr); 69 else 70 size += sizeof(struct udphdr); 71 72 if (attr->max_size && attr->max_size > size) 73 size = attr->max_size; 74 75 skb = netdev_alloc_skb(ndev, size); 76 if (!skb) 77 return NULL; 78 79 prefetchw(skb->data); 80 81 ehdr = skb_push(skb, ETH_HLEN); 82 skb_reset_mac_header(skb); 83 84 skb_set_network_header(skb, skb->len); 85 ihdr = skb_put(skb, sizeof(*ihdr)); 86 87 skb_set_transport_header(skb, skb->len); 88 if (attr->tcp) 89 thdr = skb_put(skb, sizeof(*thdr)); 90 else 91 uhdr = skb_put(skb, sizeof(*uhdr)); 92 93 eth_zero_addr(ehdr->h_dest); 94 95 if (attr->src) 96 ether_addr_copy(ehdr->h_source, attr->src); 97 if (attr->dst) 98 ether_addr_copy(ehdr->h_dest, attr->dst); 99 100 ehdr->h_proto = htons(ETH_P_IP); 101 102 if (attr->tcp) { 103 thdr->source = htons(attr->sport); 104 thdr->dest = htons(attr->dport); 105 thdr->doff = sizeof(struct tcphdr) / 4; 106 thdr->check = 0; 107 } else { 108 uhdr->source = htons(attr->sport); 109 uhdr->dest = htons(attr->dport); 110 uhdr->len = htons(sizeof(*shdr) + sizeof(*uhdr) + attr->size); 111 if (attr->max_size) 112 uhdr->len = htons(attr->max_size - 113 (sizeof(*ihdr) + sizeof(*ehdr))); 114 uhdr->check = 0; 115 } 116 117 ihdr->ihl = 5; 118 ihdr->ttl = 32; 119 ihdr->version = 4; 120 if (attr->tcp) 121 ihdr->protocol = IPPROTO_TCP; 122 else 123 ihdr->protocol = IPPROTO_UDP; 124 iplen = sizeof(*ihdr) + sizeof(*shdr) + attr->size; 125 if (attr->tcp) 126 iplen += sizeof(*thdr); 127 else 128 iplen += sizeof(*uhdr); 129 130 if (attr->max_size) 131 iplen = attr->max_size - sizeof(*ehdr); 132 133 ihdr->tot_len = htons(iplen); 134 ihdr->frag_off = 0; 135 ihdr->saddr = htonl(attr->ip_src); 136 ihdr->daddr = htonl(attr->ip_dst); 137 ihdr->tos = 0; 138 ihdr->id = 0; 139 ip_send_check(ihdr); 140 141 shdr = skb_put(skb, sizeof(*shdr)); 142 shdr->version = 0; 143 shdr->magic = cpu_to_be64(NET_TEST_PKT_MAGIC); 144 attr->id = net_test_next_id; 145 shdr->id = net_test_next_id++; 146 147 if (attr->size) 148 skb_put(skb, attr->size); 149 if (attr->max_size && attr->max_size > skb->len) 150 skb_put(skb, attr->max_size - skb->len); 151 152 skb->csum = 0; 153 skb->ip_summed = CHECKSUM_PARTIAL; 154 if (attr->tcp) { 155 thdr->check = ~tcp_v4_check(skb->len, ihdr->saddr, 156 ihdr->daddr, 0); 157 skb->csum_start = skb_transport_header(skb) - skb->head; 158 skb->csum_offset = offsetof(struct tcphdr, check); 159 } else { 160 udp4_hwcsum(skb, ihdr->saddr, ihdr->daddr); 161 } 162 163 skb->protocol = htons(ETH_P_IP); 164 skb->pkt_type = PACKET_HOST; 165 skb->dev = ndev; 166 167 return skb; 168 } 169 170 static int net_test_loopback_validate(struct sk_buff *skb, 171 struct net_device *ndev, 172 struct packet_type *pt, 173 struct net_device *orig_ndev) 174 { 175 struct net_test_priv *tpriv = pt->af_packet_priv; 176 const unsigned char *src = tpriv->packet->src; 177 const unsigned char *dst = tpriv->packet->dst; 178 struct netsfhdr *shdr; 179 struct ethhdr *ehdr; 180 struct udphdr *uhdr; 181 struct tcphdr *thdr; 182 struct iphdr *ihdr; 183 184 skb = skb_unshare(skb, GFP_ATOMIC); 185 if (!skb) 186 goto out; 187 188 if (skb_linearize(skb)) 189 goto out; 190 if (skb_headlen(skb) < (NET_TEST_PKT_SIZE - ETH_HLEN)) 191 goto out; 192 193 ehdr = (struct ethhdr *)skb_mac_header(skb); 194 if (dst) { 195 if (!ether_addr_equal_unaligned(ehdr->h_dest, dst)) 196 goto out; 197 } 198 199 if (src) { 200 if (!ether_addr_equal_unaligned(ehdr->h_source, src)) 201 goto out; 202 } 203 204 ihdr = ip_hdr(skb); 205 if (tpriv->double_vlan) 206 ihdr = (struct iphdr *)(skb_network_header(skb) + 4); 207 208 if (tpriv->packet->tcp) { 209 if (ihdr->protocol != IPPROTO_TCP) 210 goto out; 211 212 thdr = (struct tcphdr *)((u8 *)ihdr + 4 * ihdr->ihl); 213 if (thdr->dest != htons(tpriv->packet->dport)) 214 goto out; 215 216 shdr = (struct netsfhdr *)((u8 *)thdr + sizeof(*thdr)); 217 } else { 218 if (ihdr->protocol != IPPROTO_UDP) 219 goto out; 220 221 uhdr = (struct udphdr *)((u8 *)ihdr + 4 * ihdr->ihl); 222 if (uhdr->dest != htons(tpriv->packet->dport)) 223 goto out; 224 225 shdr = (struct netsfhdr *)((u8 *)uhdr + sizeof(*uhdr)); 226 } 227 228 if (shdr->magic != cpu_to_be64(NET_TEST_PKT_MAGIC)) 229 goto out; 230 if (tpriv->packet->id != shdr->id) 231 goto out; 232 233 tpriv->ok = true; 234 complete(&tpriv->comp); 235 out: 236 kfree_skb(skb); 237 return 0; 238 } 239 240 static int __net_test_loopback(struct net_device *ndev, 241 struct net_packet_attrs *attr) 242 { 243 struct net_test_priv *tpriv; 244 struct sk_buff *skb = NULL; 245 int ret = 0; 246 247 tpriv = kzalloc(sizeof(*tpriv), GFP_KERNEL); 248 if (!tpriv) 249 return -ENOMEM; 250 251 tpriv->ok = false; 252 init_completion(&tpriv->comp); 253 254 tpriv->pt.type = htons(ETH_P_IP); 255 tpriv->pt.func = net_test_loopback_validate; 256 tpriv->pt.dev = ndev; 257 tpriv->pt.af_packet_priv = tpriv; 258 tpriv->packet = attr; 259 dev_add_pack(&tpriv->pt); 260 261 skb = net_test_get_skb(ndev, attr); 262 if (!skb) { 263 ret = -ENOMEM; 264 goto cleanup; 265 } 266 267 ret = dev_direct_xmit(skb, attr->queue_mapping); 268 if (ret < 0) { 269 goto cleanup; 270 } else if (ret > 0) { 271 ret = -ENETUNREACH; 272 goto cleanup; 273 } 274 275 if (!attr->timeout) 276 attr->timeout = NET_LB_TIMEOUT; 277 278 wait_for_completion_timeout(&tpriv->comp, attr->timeout); 279 ret = tpriv->ok ? 0 : -ETIMEDOUT; 280 281 cleanup: 282 dev_remove_pack(&tpriv->pt); 283 kfree(tpriv); 284 return ret; 285 } 286 287 static int net_test_netif_carrier(struct net_device *ndev) 288 { 289 return netif_carrier_ok(ndev) ? 0 : -ENOLINK; 290 } 291 292 static int net_test_phy_phydev(struct net_device *ndev) 293 { 294 return ndev->phydev ? 0 : -EOPNOTSUPP; 295 } 296 297 static int net_test_phy_loopback_enable(struct net_device *ndev) 298 { 299 if (!ndev->phydev) 300 return -EOPNOTSUPP; 301 302 return phy_loopback(ndev->phydev, true); 303 } 304 305 static int net_test_phy_loopback_disable(struct net_device *ndev) 306 { 307 if (!ndev->phydev) 308 return -EOPNOTSUPP; 309 310 return phy_loopback(ndev->phydev, false); 311 } 312 313 static int net_test_phy_loopback_udp(struct net_device *ndev) 314 { 315 struct net_packet_attrs attr = { }; 316 317 attr.dst = ndev->dev_addr; 318 return __net_test_loopback(ndev, &attr); 319 } 320 321 static int net_test_phy_loopback_udp_mtu(struct net_device *ndev) 322 { 323 struct net_packet_attrs attr = { }; 324 325 attr.dst = ndev->dev_addr; 326 attr.max_size = ndev->mtu; 327 return __net_test_loopback(ndev, &attr); 328 } 329 330 static int net_test_phy_loopback_tcp(struct net_device *ndev) 331 { 332 struct net_packet_attrs attr = { }; 333 334 attr.dst = ndev->dev_addr; 335 attr.tcp = true; 336 return __net_test_loopback(ndev, &attr); 337 } 338 339 static const struct net_test { 340 char name[ETH_GSTRING_LEN]; 341 int (*fn)(struct net_device *ndev); 342 } net_selftests[] = { 343 { 344 .name = "Carrier ", 345 .fn = net_test_netif_carrier, 346 }, { 347 .name = "PHY dev is present ", 348 .fn = net_test_phy_phydev, 349 }, { 350 /* This test should be done before all PHY loopback test */ 351 .name = "PHY internal loopback, enable ", 352 .fn = net_test_phy_loopback_enable, 353 }, { 354 .name = "PHY internal loopback, UDP ", 355 .fn = net_test_phy_loopback_udp, 356 }, { 357 .name = "PHY internal loopback, MTU ", 358 .fn = net_test_phy_loopback_udp_mtu, 359 }, { 360 .name = "PHY internal loopback, TCP ", 361 .fn = net_test_phy_loopback_tcp, 362 }, { 363 /* This test should be done after all PHY loopback test */ 364 .name = "PHY internal loopback, disable", 365 .fn = net_test_phy_loopback_disable, 366 }, 367 }; 368 369 void net_selftest(struct net_device *ndev, struct ethtool_test *etest, u64 *buf) 370 { 371 int count = net_selftest_get_count(); 372 int i; 373 374 memset(buf, 0, sizeof(*buf) * count); 375 net_test_next_id = 0; 376 377 if (etest->flags != ETH_TEST_FL_OFFLINE) { 378 netdev_err(ndev, "Only offline tests are supported\n"); 379 etest->flags |= ETH_TEST_FL_FAILED; 380 return; 381 } 382 383 384 for (i = 0; i < count; i++) { 385 buf[i] = net_selftests[i].fn(ndev); 386 if (buf[i] && (buf[i] != -EOPNOTSUPP)) 387 etest->flags |= ETH_TEST_FL_FAILED; 388 } 389 } 390 EXPORT_SYMBOL_GPL(net_selftest); 391 392 int net_selftest_get_count(void) 393 { 394 return ARRAY_SIZE(net_selftests); 395 } 396 EXPORT_SYMBOL_GPL(net_selftest_get_count); 397 398 void net_selftest_get_strings(u8 *data) 399 { 400 u8 *p = data; 401 int i; 402 403 for (i = 0; i < net_selftest_get_count(); i++) { 404 snprintf(p, ETH_GSTRING_LEN, "%2d. %s", i + 1, 405 net_selftests[i].name); 406 p += ETH_GSTRING_LEN; 407 } 408 } 409 EXPORT_SYMBOL_GPL(net_selftest_get_strings); 410 411 MODULE_LICENSE("GPL v2"); 412 MODULE_AUTHOR("Oleksij Rempel <o.rempel@pengutronix.de>"); 413