uboot/drivers/net/sandbox.c
<<
>>
Prefs
   1/*
   2 * Copyright (c) 2015 National Instruments
   3 *
   4 * (C) Copyright 2015
   5 * Joe Hershberger <joe.hershberger@ni.com>
   6 *
   7 * SPDX-License-Identifier:     GPL-2.0
   8 */
   9
  10#include <common.h>
  11#include <dm.h>
  12#include <malloc.h>
  13#include <net.h>
  14#include <asm/test.h>
  15
  16DECLARE_GLOBAL_DATA_PTR;
  17
  18/**
  19 * struct eth_sandbox_priv - memory for sandbox mock driver
  20 *
  21 * fake_host_hwaddr: MAC address of mocked machine
  22 * fake_host_ipaddr: IP address of mocked machine
  23 * recv_packet_buffer: buffer of the packet returned as received
  24 * recv_packet_length: length of the packet returned as received
  25 */
  26struct eth_sandbox_priv {
  27        uchar fake_host_hwaddr[ARP_HLEN];
  28        struct in_addr fake_host_ipaddr;
  29        uchar *recv_packet_buffer;
  30        int recv_packet_length;
  31};
  32
  33static bool disabled[8] = {false};
  34static bool skip_timeout;
  35
  36/*
  37 * sandbox_eth_disable_response()
  38 *
  39 * index - The alias index (also DM seq number)
  40 * disable - If non-zero, ignore sent packets and don't send mock response
  41 */
  42void sandbox_eth_disable_response(int index, bool disable)
  43{
  44        disabled[index] = disable;
  45}
  46
  47/*
  48 * sandbox_eth_skip_timeout()
  49 *
  50 * When the first packet read is attempted, fast-forward time
  51 */
  52void sandbox_eth_skip_timeout(void)
  53{
  54        skip_timeout = true;
  55}
  56
  57static int sb_eth_start(struct udevice *dev)
  58{
  59        struct eth_sandbox_priv *priv = dev_get_priv(dev);
  60
  61        debug("eth_sandbox: Start\n");
  62
  63        fdtdec_get_byte_array(gd->fdt_blob, dev->of_offset, "fake-host-hwaddr",
  64                              priv->fake_host_hwaddr, ARP_HLEN);
  65        priv->recv_packet_buffer = net_rx_packets[0];
  66        return 0;
  67}
  68
  69static int sb_eth_send(struct udevice *dev, void *packet, int length)
  70{
  71        struct eth_sandbox_priv *priv = dev_get_priv(dev);
  72        struct ethernet_hdr *eth = packet;
  73
  74        debug("eth_sandbox: Send packet %d\n", length);
  75
  76        if (dev->seq >= 0 && dev->seq < ARRAY_SIZE(disabled) &&
  77            disabled[dev->seq])
  78                return 0;
  79
  80        if (ntohs(eth->et_protlen) == PROT_ARP) {
  81                struct arp_hdr *arp = packet + ETHER_HDR_SIZE;
  82
  83                if (ntohs(arp->ar_op) == ARPOP_REQUEST) {
  84                        struct ethernet_hdr *eth_recv;
  85                        struct arp_hdr *arp_recv;
  86
  87                        /* store this as the assumed IP of the fake host */
  88                        priv->fake_host_ipaddr = net_read_ip(&arp->ar_tpa);
  89                        /* Formulate a fake response */
  90                        eth_recv = (void *)priv->recv_packet_buffer;
  91                        memcpy(eth_recv->et_dest, eth->et_src, ARP_HLEN);
  92                        memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
  93                               ARP_HLEN);
  94                        eth_recv->et_protlen = htons(PROT_ARP);
  95
  96                        arp_recv = (void *)priv->recv_packet_buffer +
  97                                ETHER_HDR_SIZE;
  98                        arp_recv->ar_hrd = htons(ARP_ETHER);
  99                        arp_recv->ar_pro = htons(PROT_IP);
 100                        arp_recv->ar_hln = ARP_HLEN;
 101                        arp_recv->ar_pln = ARP_PLEN;
 102                        arp_recv->ar_op = htons(ARPOP_REPLY);
 103                        memcpy(&arp_recv->ar_sha, priv->fake_host_hwaddr,
 104                               ARP_HLEN);
 105                        net_write_ip(&arp_recv->ar_spa, priv->fake_host_ipaddr);
 106                        memcpy(&arp_recv->ar_tha, &arp->ar_sha, ARP_HLEN);
 107                        net_copy_ip(&arp_recv->ar_tpa, &arp->ar_spa);
 108
 109                        priv->recv_packet_length = ETHER_HDR_SIZE +
 110                                ARP_HDR_SIZE;
 111                }
 112        } else if (ntohs(eth->et_protlen) == PROT_IP) {
 113                struct ip_udp_hdr *ip = packet + ETHER_HDR_SIZE;
 114
 115                if (ip->ip_p == IPPROTO_ICMP) {
 116                        struct icmp_hdr *icmp = (struct icmp_hdr *)&ip->udp_src;
 117
 118                        if (icmp->type == ICMP_ECHO_REQUEST) {
 119                                struct ethernet_hdr *eth_recv;
 120                                struct ip_udp_hdr *ipr;
 121                                struct icmp_hdr *icmpr;
 122
 123                                /* reply to the ping */
 124                                memcpy(priv->recv_packet_buffer, packet,
 125                                       length);
 126                                eth_recv = (void *)priv->recv_packet_buffer;
 127                                ipr = (void *)priv->recv_packet_buffer +
 128                                        ETHER_HDR_SIZE;
 129                                icmpr = (struct icmp_hdr *)&ipr->udp_src;
 130                                memcpy(eth_recv->et_dest, eth->et_src,
 131                                       ARP_HLEN);
 132                                memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
 133                                       ARP_HLEN);
 134                                ipr->ip_sum = 0;
 135                                ipr->ip_off = 0;
 136                                net_copy_ip((void *)&ipr->ip_dst, &ip->ip_src);
 137                                net_write_ip((void *)&ipr->ip_src,
 138                                             priv->fake_host_ipaddr);
 139                                ipr->ip_sum = compute_ip_checksum(ipr,
 140                                        IP_HDR_SIZE);
 141
 142                                icmpr->type = ICMP_ECHO_REPLY;
 143                                icmpr->checksum = 0;
 144                                icmpr->checksum = compute_ip_checksum(icmpr,
 145                                        ICMP_HDR_SIZE);
 146
 147                                priv->recv_packet_length = length;
 148                        }
 149                }
 150        }
 151
 152        return 0;
 153}
 154
 155static int sb_eth_recv(struct udevice *dev, int flags, uchar **packetp)
 156{
 157        struct eth_sandbox_priv *priv = dev_get_priv(dev);
 158
 159        if (skip_timeout) {
 160                sandbox_timer_add_offset(11000UL);
 161                skip_timeout = false;
 162        }
 163
 164        if (priv->recv_packet_length) {
 165                int lcl_recv_packet_length = priv->recv_packet_length;
 166
 167                debug("eth_sandbox: received packet %d\n",
 168                      priv->recv_packet_length);
 169                priv->recv_packet_length = 0;
 170                *packetp = priv->recv_packet_buffer;
 171                return lcl_recv_packet_length;
 172        }
 173        return 0;
 174}
 175
 176static void sb_eth_stop(struct udevice *dev)
 177{
 178        debug("eth_sandbox: Stop\n");
 179}
 180
 181static int sb_eth_write_hwaddr(struct udevice *dev)
 182{
 183        struct eth_pdata *pdata = dev_get_platdata(dev);
 184
 185        debug("eth_sandbox %s: Write HW ADDR - %pM\n", dev->name,
 186              pdata->enetaddr);
 187        return 0;
 188}
 189
 190static const struct eth_ops sb_eth_ops = {
 191        .start                  = sb_eth_start,
 192        .send                   = sb_eth_send,
 193        .recv                   = sb_eth_recv,
 194        .stop                   = sb_eth_stop,
 195        .write_hwaddr           = sb_eth_write_hwaddr,
 196};
 197
 198static int sb_eth_remove(struct udevice *dev)
 199{
 200        return 0;
 201}
 202
 203static int sb_eth_ofdata_to_platdata(struct udevice *dev)
 204{
 205        struct eth_pdata *pdata = dev_get_platdata(dev);
 206
 207        pdata->iobase = dev_get_addr(dev);
 208        return 0;
 209}
 210
 211static const struct udevice_id sb_eth_ids[] = {
 212        { .compatible = "sandbox,eth" },
 213        { }
 214};
 215
 216U_BOOT_DRIVER(eth_sandbox) = {
 217        .name   = "eth_sandbox",
 218        .id     = UCLASS_ETH,
 219        .of_match = sb_eth_ids,
 220        .ofdata_to_platdata = sb_eth_ofdata_to_platdata,
 221        .remove = sb_eth_remove,
 222        .ops    = &sb_eth_ops,
 223        .priv_auto_alloc_size = sizeof(struct eth_sandbox_priv),
 224        .platdata_auto_alloc_size = sizeof(struct eth_pdata),
 225};
 226