Tiny210 driver virtual network card driver

virt_net.c driver:

// Reference drivers\net\cs89x0.c

#include "linux/module.h"

#include "linux/errno.h"

#include "linux/netdevice.h"

#include "linux/etherdevice.h"

#include "linux/kernel.h"

#include "linux/types.h"

#include "linux/fcntl.h"

#include "linux/interrupt.h"

#include "linux/ioport.h"

#include "linux/in.h"

#include "linux/skbuff.h"

#include "linux/slab.h"

#include "linux/spinlock.h"

#include "linux/string.h"

#include "linux/init.h"

#include "linux/bitops.h"

#include "linux/delay.h"

#include "linux/ip.h"

#include "asm/system.h"

#include "asm/io.h"

#include "asm/irq.h"

static struct net_device *vnet_dev;

static void emulator_rx_packet(struct sk_buff *skb, struct net_device *dev)

{

    // Refer to LDD3 

    unsigned char *type;

    struct iphdr *ih;

    __be32 *saddr, *daddr, tmp;

    unsigned char    tmp_dev_addr[ETH_ALEN];

    struct ethhdr *ethhdr;

    struct sk_buff *rx_skb;

    // Read/save data from hardware

    // Swap the mac addresses of "source/destination" 

    ethhdr = (struct ethhdr *)skb->data;

    memcpy(tmp_dev_addr, ethhdr->h_dest, ETH_ALEN);

    memcpy(ethhdr->h_dest, ethhdr->h_source, ETH_ALEN);

    memcpy(ethhdr->h_source, tmp_dev_addr, ETH_ALEN);

    // Swap the "source/destination" IP addresses     

    ih = (struct iphdr *)(skb->data + sizeof(struct ethhdr));

    saddr = &ih->saddr;

    daddr = &ih->daddr;

    tmp = *saddr;

    *saddr = *daddr;

    *daddr = tmp;

    //((u8 *)saddr)[2] ^= 1; // change the third octet (class C) 

    //((u8 *)daddr)[2] ^= 1;

    type = skb->data + sizeof(struct ethhdr) + sizeof(struct iphdr);

    //printk("tx package type = x\n", *type);

    // Modify the type, originally 0x8 means ping

    *type = 0; // 0 means reply 

    ih->check = 0;           // and rebuild the checksum (ip needs it) 

    ih->check = ip_fast_csum((unsigned char *)ih,ih->ihl);

    //Construct a sk_buff

    rx_skb = dev_alloc_skb(skb->len + 2);

    skb_reserve(rx_skb, 2); // align IP on 16B boundary     

    memcpy(skb_put(rx_skb, skb->len), skb->data, skb->len);

    // Write metadata, and then pass to the receive level 

    rx_skb->dev = dev;

    rx_skb->protocol = eth_type_trans(rx_skb, dev);

    rx_skb->ip_summed = CHECKSUM_UNNECESSARY; // don't check it 

    dev->stats.rx_packets++;

    dev->stats.rx_bytes += skb->len;

    // Submit sk_buff

    netif_rx(rx_skb);

}

static int virt_net_send_packet(struct sk_buff *skb, struct net_device *dev)

{

    static int cnt = 0;

    printk("virt_net_send_packet cnt = %d\n", ++cnt);

    // For the real network card, send the data in skb through the network card 

    netif_stop_queue(dev); // Stop the queue of the network card 

    // ...... // Write skb data to the network card 

    //Construct a fake sk_buff and report it 

    emulator_rx_packet(skb, dev);

    dev_kfree_skb (skb); // release skb 

    netif_wake_queue(dev); //After all data is sent out, wake up the queue of the network card 

    // Update statistics 

    dev->stats.tx_packets++;

    dev->stats.tx_bytes += skb->len;

    return 0;

}

static const struct net_device_ops virt_netdev_ops = {

    .ndo_start_xmit = virt_net_send_packet,

};

static int virt_net_init(void)

{

    // 1. Allocate a net_device structure 

    vnet_dev = alloc_netdev(0, "vnet%d", ether_setup);  // alloc_etherdev 

    // 2. Setup 

    vnet_dev->netdev_ops    = &virt_netdev_ops;

    // Set the MAC address 

    vnet_dev->dev_addr[0] = 0x08;

    vnet_dev->dev_addr[1] = 0x89;

    vnet_dev->dev_addr[2] = 0x89;

    vnet_dev->dev_addr[3] = 0x89;

    vnet_dev->dev_addr[4] = 0x89;

    vnet_dev->dev_addr[5] = 0x11;

    // Set the following two items to ping successfully 

    vnet_dev->flags           |= IFF_NOARP;

    vnet_dev->features        |= NETIF_F_NO_CSUM;    

    // 3. Registration 

    //register_netdevice(vnet_dev);

    register_netdev(vnet_dev);

    return 0;

}

static void virt_net_exit(void)

{

    unregister_netdev(vnet_dev);

    free_netdev(vnet_dev);

}

module_init(virt_net_init);

module_exit(virt_net_exit);

MODULE_AUTHOR("thisway.diy@163.com,17653039@qq.com");

MODULE_LICENSE("GPL");