git.sur5r.net Git - u-boot/blob - drivers/net/sandbox.c

   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
  15 DECLARE_GLOBAL_DATA_PTR;
  16
  17 /**
  18  * struct eth_sandbox_priv - memory for sandbox mock driver
  19  *
  20  * fake_host_hwaddr: MAC address of mocked machine
  21  * fake_host_ipaddr: IP address of mocked machine
  22  * recv_packet_buffer: buffer of the packet returned as received
  23  * recv_packet_length: length of the packet returned as received
  24  */
  25 struct eth_sandbox_priv {
  26         uchar fake_host_hwaddr[ARP_HLEN];
  27         IPaddr_t fake_host_ipaddr;
  28         uchar *recv_packet_buffer;
  29         int recv_packet_length;
  30 };
  31
  32 static int sb_eth_start(struct udevice *dev)
  33 {
  34         struct eth_sandbox_priv *priv = dev_get_priv(dev);
  35
  36         debug("eth_sandbox: Start\n");
  37
  38         fdtdec_get_byte_array(gd->fdt_blob, dev->of_offset, "fake-host-hwaddr",
  39                               priv->fake_host_hwaddr, ARP_HLEN);
  40         priv->recv_packet_buffer = net_rx_packets[0];
  41         return 0;
  42 }
  43
  44 static int sb_eth_send(struct udevice *dev, void *packet, int length)
  45 {
  46         struct eth_sandbox_priv *priv = dev_get_priv(dev);
  47         struct ethernet_hdr *eth = packet;
  48
  49         debug("eth_sandbox: Send packet %d\n", length);
  50
  51         if (ntohs(eth->et_protlen) == PROT_ARP) {
  52                 struct arp_hdr *arp = packet + ETHER_HDR_SIZE;
  53
  54                 if (ntohs(arp->ar_op) == ARPOP_REQUEST) {
  55                         struct ethernet_hdr *eth_recv;
  56                         struct arp_hdr *arp_recv;
  57
  58                         /* store this as the assumed IP of the fake host */
  59                         priv->fake_host_ipaddr = NetReadIP(&arp->ar_tpa);
  60                         /* Formulate a fake response */
  61                         eth_recv = (void *)priv->recv_packet_buffer;
  62                         memcpy(eth_recv->et_dest, eth->et_src, ARP_HLEN);
  63                         memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
  64                                ARP_HLEN);
  65                         eth_recv->et_protlen = htons(PROT_ARP);
  66
  67                         arp_recv = (void *)priv->recv_packet_buffer +
  68                                 ETHER_HDR_SIZE;
  69                         arp_recv->ar_hrd = htons(ARP_ETHER);
  70                         arp_recv->ar_pro = htons(PROT_IP);
  71                         arp_recv->ar_hln = ARP_HLEN;
  72                         arp_recv->ar_pln = ARP_PLEN;
  73                         arp_recv->ar_op = htons(ARPOP_REPLY);
  74                         memcpy(&arp_recv->ar_sha, priv->fake_host_hwaddr,
  75                                ARP_HLEN);
  76                         NetWriteIP(&arp_recv->ar_spa, priv->fake_host_ipaddr);
  77                         memcpy(&arp_recv->ar_tha, &arp->ar_sha, ARP_HLEN);
  78                         NetCopyIP(&arp_recv->ar_tpa, &arp->ar_spa);
  79
  80                         priv->recv_packet_length = ETHER_HDR_SIZE +
  81                                 ARP_HDR_SIZE;
  82                 }
  83         } else if (ntohs(eth->et_protlen) == PROT_IP) {
  84                 struct ip_udp_hdr *ip = packet + ETHER_HDR_SIZE;
  85
  86                 if (ip->ip_p == IPPROTO_ICMP) {
  87                         struct icmp_hdr *icmp = (struct icmp_hdr *)&ip->udp_src;
  88
  89                         if (icmp->type == ICMP_ECHO_REQUEST) {
  90                                 struct ethernet_hdr *eth_recv;
  91                                 struct ip_udp_hdr *ipr;
  92                                 struct icmp_hdr *icmpr;
  93
  94                                 /* reply to the ping */
  95                                 memcpy(priv->recv_packet_buffer, packet,
  96                                        length);
  97                                 eth_recv = (void *)priv->recv_packet_buffer;
  98                                 ipr = (void *)priv->recv_packet_buffer +
  99                                         ETHER_HDR_SIZE;
 100                                 icmpr = (struct icmp_hdr *)&ipr->udp_src;
 101                                 memcpy(eth_recv->et_dest, eth->et_src,
 102                                        ARP_HLEN);
 103                                 memcpy(eth_recv->et_src, priv->fake_host_hwaddr,
 104                                        ARP_HLEN);
 105                                 ipr->ip_sum = 0;
 106                                 ipr->ip_off = 0;
 107                                 NetCopyIP((void *)&ipr->ip_dst, &ip->ip_src);
 108                                 NetWriteIP((void *)&ipr->ip_src,
 109                                            priv->fake_host_ipaddr);
 110                                 ipr->ip_sum = compute_ip_checksum(ipr,
 111                                         IP_HDR_SIZE);
 112
 113                                 icmpr->type = ICMP_ECHO_REPLY;
 114                                 icmpr->checksum = 0;
 115                                 icmpr->checksum = compute_ip_checksum(icmpr,
 116                                         ICMP_HDR_SIZE);
 117
 118                                 priv->recv_packet_length = length;
 119                         }
 120                 }
 121         }
 122
 123         return 0;
 124 }
 125
 126 static int sb_eth_recv(struct udevice *dev, uchar **packetp)
 127 {
 128         struct eth_sandbox_priv *priv = dev_get_priv(dev);
 129
 130         if (priv->recv_packet_length) {
 131                 int lcl_recv_packet_length = priv->recv_packet_length;
 132
 133                 debug("eth_sandbox: received packet %d\n",
 134                       priv->recv_packet_length);
 135                 priv->recv_packet_length = 0;
 136                 *packetp = priv->recv_packet_buffer;
 137                 return lcl_recv_packet_length;
 138         }
 139         return 0;
 140 }
 141
 142 static void sb_eth_stop(struct udevice *dev)
 143 {
 144         debug("eth_sandbox: Stop\n");
 145 }
 146
 147 static int sb_eth_write_hwaddr(struct udevice *dev)
 148 {
 149         struct eth_pdata *pdata = dev_get_platdata(dev);
 150
 151         debug("eth_sandbox %s: Write HW ADDR - %pM\n", dev->name,
 152               pdata->enetaddr);
 153         return 0;
 154 }
 155
 156 static const struct eth_ops sb_eth_ops = {
 157         .start                  = sb_eth_start,
 158         .send                   = sb_eth_send,
 159         .recv                   = sb_eth_recv,
 160         .stop                   = sb_eth_stop,
 161         .write_hwaddr           = sb_eth_write_hwaddr,
 162 };
 163
 164 static int sb_eth_remove(struct udevice *dev)
 165 {
 166         return 0;
 167 }
 168
 169 static int sb_eth_ofdata_to_platdata(struct udevice *dev)
 170 {
 171         struct eth_pdata *pdata = dev_get_platdata(dev);
 172
 173         pdata->iobase = dev_get_addr(dev);
 174         return 0;
 175 }
 176
 177 static const struct udevice_id sb_eth_ids[] = {
 178         { .compatible = "sandbox,eth" },
 179         { }
 180 };
 181
 182 U_BOOT_DRIVER(eth_sandbox) = {
 183         .name   = "eth_sandbox",
 184         .id     = UCLASS_ETH,
 185         .of_match = sb_eth_ids,
 186         .ofdata_to_platdata = sb_eth_ofdata_to_platdata,
 187         .remove = sb_eth_remove,
 188         .ops    = &sb_eth_ops,
 189         .priv_auto_alloc_size = sizeof(struct eth_sandbox_priv),
 190         .platdata_auto_alloc_size = sizeof(struct eth_pdata),
 191 };