先看该驱动的初始化函数:int snullnet_init()
函数分析:
1、为网络设备分配设备结构:net_device,该过由alloc_netdev实现;
2、注册设备,由register_netdev实现;
alloc_netdev函数的第一个参数是网络设备的私有结构体,由于一个设备驱动程序可以被多个网络设备使用,所以不同的设备是用这个私有结构体来区分的;网络设备通过netdev_priv函数可以从网络接口的net_device结构得到该私有结构体的指针;下面我们看看LDD中的网络接口的私有结构体(struct snull_priv):
点击(此处)折叠或打开
- struct snull_priv {
- struct net_device_stats stats;//网络统计信息,内核已经定义了该结构体
- int status; //网络接口状态
- struct snull_packet *ppool; //缓冲池,驱动需要该结构时候,直接从这个队列中取出,用完后归还
- struct snull_packet *rx_queue;//接收到的数据包的队列
- int rx_int_enabled; //接收中端是否使能
- int tx_packetlen; //要发送的数据包的长度
- u8 *tx_packetdata; //要发送的数据
- struct sk_buff *skb; //这个可以没有的,貌似我修改的程序中,木有用到
- spinlock_t lock; //保护该结构体并发访问的自旋锁
- struct napi_struct *snull_napi;//使用napi时候,要用到的结构,在linux2.6.24中加入
- };
void snull_init(struct net_device *dev)完成以下任务:
1、调用ether_setup;
这里不得不说一下的是,ether_setup是内核为方便初始化以太网接口而定义的,其他的网络也有类似的函数,比如fddi_setup用于设置FDDI网络接口等等;其实如果你想加入一个以前不存在的接口类型的话,你可以定义自己的接口初始化函数;也可以用一个类似的函数初始化,然后再修改一些域就可以了
2、设置两个主要的操作函数集;
- dev->netdev_ops = &snulldev_ops;//网络接口管理函数
- dev->header_ops = &snull_header_ops;//数据帧头部处理函数
既然提到了上面的两个操作集合,我们接下来就说说他们吧。
hdader_ops是header_ops结构的实例:
在本驱动中,我们定义了create、rebuild连个操作;
create操作在发送数据包之前被调用,用来组装帧头部;
rebuild在进行地址解析(如IPv4中的ARP协议)之后,重新组装帧头部;
netdev_ops是net_device_ops结构的实例;
该结构体定义了N多的操作,本驱动实现了其中的一部分下面一一对他们进行讲解:
snull_open设备UP时被调用,本驱动用它完成下面任务:
1、设置接口的硬件地址;
2、调用netif_start_queue函数,该调用等于告诉内核,该接口可以接收上层数据包了;
snull_release:设备进入DOWN状态时被调用,本驱动使用来实现:
1、调用netif_stop_queue函数,告诉内核,我以后不接收数据包了;
snull_chenge_mtu:这个没有歧义,不再说明;
snull_stats:获取设备的统计信息,本驱动直接返回私有结构中的net_device_ops内嵌结构的指针;
snull_tx:设备的发送函数,完成设备的数据发送工作;这个函数比较重要,但是根LDD中没有区别,可以参考上面的解释;
下面我们就主要看看设备接收数据的过程:
该驱动使用了两种方法来展示linux中处理网络接口数据接收问题的方法,由于该驱动模拟了接收中断,所以下面两种方法都是中断服务函数;
方法一:void snull_regular_interrupt(...),解析如下:
- pkt = priv->rx_queue;
然后,驱动调用了snull_rx函数,这个函数利用要传递的数据,调用dev_alloc_skb函数来初始化sk_buff结构,初始化sk_buff时期,最值得关注的一个细节是:
- skb->protocol = eth_type_trans(skb, dev);
最终,函数调用了netif_rx(skb)函数,这个函数由内核提供;至此,驱动完成了数据向上层传递的任务,其他工作由内核完成;
建议大家空闲的时候,跟踪一下netif_rx这个函数,以便更好的理解linux网络数据的接收;
方法二:void snull_napi_interrupt(...),解析如下:
- snull_rx_ints(dev, 0);//禁止接收中断
- napi_enable(priv->snull_napi);
- napi_schedule(priv->snull_napi);
这个主要涉及到网络的napi功能;该功能简单来说就是在发生网络数据接收中断时,关闭接口的中断,通过轮循的方法处理以后到来的网络数据,直到所有的数据处理完毕(这个不准确,但是更容易理解);数据处理完毕再打开接收中断;napi接口以后有空再讲,现在暂且不提;
至此,一个具备基本功能的网络驱动程序就已经勾画出来了,自己在这里记录一下,以便以后复习;下面粘贴完整的源文件:snullnet.c
--->文件名:snullnet.c
点击(此处)折叠或打开
- #include <linux/module.h>
- #include <linux/init.h>
- #include <linux/moduleparam.h>
- #include <linux/netdevice.h>
- #include <linux/etherdevice.h>
- #include <linux/kernel.h>
- #include <linux/interrupt.h>
- #include <linux/spinlock.h>
- #include <linux/sched.h>
- #include <linux/slab.h>
- #include <linux/errno.h>
- #include <linux/types.h>
- #include <linux/in.h>
- #include <linux/ip.h>
- #include <linux/tcp.h>
- #include <linux/skbuff.h>
- #include <asm/checksum.h>
- #define SNULL_RX_INTR 1UL
- #define SNULL_TX_INTR 2UL
- #define SNULL_TIMEOUT 5
- MODULE_AUTHOR("xishuai");
- MODULE_LICENSE("GPLv3");
- int pool_size = 8;
- module_param(pool_size, int, 0);
- int use_napi = 0;
- module_param(use_napi, int, 0);
- struct snull_packet {
- struct snull_packet *next;
- struct net_device *dev;
- int datalen;
- u8 data[ETH_DATA_LEN];
- };
- struct snull_priv {
- struct net_device_stats stats;
- int status;
- struct snull_packet *ppool;
- struct snull_packet *rx_queue;
- int rx_int_enabled;
- int tx_packetlen;
- u8 *tx_packetdata;
- struct sk_buff *skb;
- spinlock_t lock;
- struct napi_struct *snull_napi;
- };
- static struct net_device *snull_dev[2];
- static void(*snull_interrupt)(int, void*, struct pt_regs *);
- static int snull_open(struct net_device *dev)
- {
- printk(KERN_ALERT "snull_open\n");
- memcpy(dev->dev_addr, "\0SNUL0", ETH_ALEN);
- if (dev == snull_dev[1]) {
- dev->dev_addr[ETH_ALEN -1]++;
- }
- netif_start_queue(dev);//允许上层调用ndo_start_xmit函数
- return 0;
- }
- static int snull_release(struct net_device *dev)
- {
- printk("snullnet:snull_release\n");
- netif_stop_queue(dev);//禁止上层调用本设备的发送函数
- return 0;
- }
- static int snull_config(struct net_device *dev, struct ifmap *map)
- {
- printk("snullnet:snull_config\n");
- if(dev->flags & IFF_UP) return -EBUSY;
- if(map->base_addr != dev->base_addr) {
- printk(KERN_WARNING "snull: Can't change I/O address\n");
- return -EOPNOTSUPP;
- }
-
- if(map->irq != dev->irq) {
- dev->irq = map->irq;
- }
- /*忽略其他域的改变*/
- return 0;
- }
- static void snull_enqueue_buf(struct net_device *dev, struct snull_packet *pkt)
- {
- unsigned long flags;
- struct snull_priv *priv = netdev_priv(dev);
- printk("snullnet:snull_enqueue_buf\n");
- spin_lock_irqsave(&priv->lock, flags);
- pkt->next = priv->rx_queue;
- priv->rx_queue = pkt;
- spin_unlock_irqrestore(&priv->lock, flags);
- }
- static struct snull_packet *snull_get_tx_buffer(struct net_device *dev)
- {
- struct snull_priv *priv = netdev_priv(dev);
- unsigned long flags;
- struct snull_packet *pkt;
- printk("snullnet:snull_get_tx_buffer\n");
- spin_lock_irqsave(&priv->lock, flags);
- pkt = priv->ppool;
- priv->ppool = pkt->next;
- if (priv->ppool == NULL) {
- printk(KERN_INFO "pool empty\n");
- netif_stop_queue(dev);
- }
- spin_unlock_irqrestore(&priv->lock, flags);
- return pkt;
- }
- //发送上层数据,本函数吃力底层硬件细节
- static void snull_hw_tx(char *buf, int len, struct net_device *dev)
- {
- struct iphdr *ih;//
- struct net_device *dest;
- struct snull_priv *priv;
- u32 *saddr, *daddr;
- struct snull_packet *tx_buffer;
- ih = (struct iphdr *)(buf + sizeof(struct ethhdr));
- saddr = &ih->saddr;
- daddr = &ih->daddr;
-
- //驱动中修改了IP层的头部,这在正常的驱动中是不应该的
- ((u8 *)saddr)[2] ^= 1;
- ((u8 *)daddr)[2] ^= 1;
- ih->check = 0; //要重新计算校验和
- ih->check = ip_fast_csum((unsigned char *)ih, ih->ihl);
- //准备发送
- dest = snull_dev[dev == snull_dev[0] ? 1 : 0];
- priv = netdev_priv(dest);
- tx_buffer = snull_get_tx_buffer(dev);
- tx_buffer->datalen = len;
- memcpy(tx_buffer->data, buf, len);
- snull_enqueue_buf(dest, tx_buffer);
- if (priv->rx_int_enabled) {
- priv->status |= SNULL_RX_INTR;
- snull_interrupt(0, dest, NULL);
- }
- priv = netdev_priv(dev);
- priv->tx_packetlen = len;
- priv->tx_packetdata = buf;
- priv->status |= SNULL_TX_INTR;
- snull_interrupt(0, dev, NULL);//原来的代码里面有个模拟丢包的代码段
- }
- static int snull_tx(struct sk_buff *skb, struct net_device *dev)
- {
- int len;
- char *data, shortpkt[ETH_ZLEN];
- struct snull_priv *priv = netdev_priv(dev);
- printk("snullnet:snull_tx\n");
- data = skb->data;
- len = skb->len;
- if (len < ETH_ZLEN) {
- memset(shortpkt, 0, ETH_ZLEN);
- memcpy(shortpkt, skb->data, skb->len);
- len = ETH_ZLEN;
- data = shortpkt;
- }
- dev->trans_start = jiffies;
- priv->skb = skb;
- snull_hw_tx(data, len, dev);//实际的发送过程,设备相关
- return 0;
- }
- static void snull_tx_timeout (struct net_device *dev)
- {
- struct snull_priv *priv = netdev_priv(dev);
- //模拟传输中断
- priv->status = SNULL_TX_INTR;
- snull_interrupt(0, dev, NULL);
- priv->stats.tx_errors++;
- netif_wake_queue(dev);
- return;
- }
- static int snull_ioctl(struct net_device *dev, struct ifreq *rq, int cmd)
- {
- printk("snullnet:snull_ioctl\n");
- return 0;
- }
- static struct net_device_stats *snull_stats(struct net_device *dev)
- {
- struct snull_priv *priv = netdev_priv(dev);
- return &priv->stats;
- }
- static int snull_create_header(struct sk_buff *skb, struct net_device *dev,
- unsigned short type, const void *daddr,
- const void *saddr, unsigned len)
- {
- struct ethhdr *eth = (struct ethhdr *)skb_push(skb, ETH_HLEN);
-
- printk("snullnet:snull_create_header\n");
- eth->h_proto = htons(type);
- memcpy(eth->h_source, saddr ? saddr : dev->dev_addr, dev->addr_len);
- memcpy(eth->h_dest, daddr ? daddr : dev->dev_addr, dev->addr_len);
- eth->h_dest[ETH_ALEN - 1] ^= 1;
- return (dev->hard_header_len);
- }
- static int snull_rebuild_header(struct sk_buff *skb)
- {
- struct ethhdr *eth = (struct ethhdr *)skb->data;
- struct net_device *dev = skb->dev;
- printk("snullnet:snull_rebuild_header\n");
- memcpy(eth->h_source, dev->dev_addr, dev->addr_len);
- memcpy(eth->h_dest, dev->dev_addr, dev->addr_len);
- eth->h_dest[ETH_ALEN - 1] ^= 1;
- return 0;
- }
- static int snull_change_mtu(struct net_device *dev, int new_mtu)
- {
- unsigned long flags;
- struct snull_priv *priv = netdev_priv(dev);
- spinlock_t *lock = &priv->lock;
- printk("snullnet:snull_change_mtu\n");
- if((new_mtu < 68) || (new_mtu > 1500)) return -EINVAL;
- spin_lock_irqsave(lock, flags);
- dev->mtu = new_mtu;
- spin_unlock_irqrestore(lock, flags);
- return 0;
- }
- static struct header_ops snull_header_ops = {
- .create = snull_create_header,
- .rebuild = snull_rebuild_header,
- };
- static struct net_device_ops snulldev_ops = {
- .ndo_open = snull_open,
- .ndo_stop = snull_release,
- .ndo_set_config = snull_config,
- .ndo_start_xmit = snull_tx,
- .ndo_do_ioctl = snull_ioctl,
- .ndo_get_stats = snull_stats,
- .ndo_change_mtu = snull_change_mtu,
- .ndo_tx_timeout = snull_tx_timeout,
- };
- static void snull_rx_ints(struct net_device *dev, int enable)
- {
- struct snull_priv *priv = netdev_priv(dev);
- priv->rx_int_enabled = enable;
- }
- static void snull_setup_pool(struct net_device *dev)
- {
-
- struct snull_priv *priv = netdev_priv(dev);
- int i;
- struct snull_packet *pkt;
- printk("snullnet:snull_setup_pool\n");
- priv->ppool = NULL;
- for(i = 0; i < pool_size; i++) {
- pkt = kmalloc(sizeof (struct snull_packet), GFP_KERNEL);
- if(NULL == pkt) {
- printk(KERN_NOTICE "Ran out of memory allocating packet pool\n");
- return;
- }
- pkt->dev = dev;
- pkt->next = priv->ppool;
- priv->ppool = pkt;
- }
- }
- static void snull_release_buffer(struct snull_packet *pkt)
- {
- unsigned long flags;
- struct snull_priv *priv = netdev_priv(pkt->dev);
- spin_lock_irqsave(&priv->lock, flags);
- pkt->next = priv->ppool;//牛逼,还整个结构提复用池
- priv->ppool = pkt;
- spin_unlock_irqrestore(&priv->lock, flags);
- if (netif_queue_stopped(pkt->dev) && pkt->next == NULL) {
- netif_wake_queue(pkt->dev);
- }
- }
- static int snull_poll(struct napi_struct *napi, int work_limit)
- {
- int nworked = 0;
- struct snull_priv *priv = netdev_priv(napi->dev);
- struct sk_buff *skb;
- struct snull_packet *pkt;
- unsigned long flags;
- printk("snullnet:work_limit = %d\n", work_limit);
- while(nworked < work_limit && priv->rx_queue) {
- spin_lock_irqsave(&priv->lock, flags);
- pkt = priv->rx_queue;
- priv->rx_queue = priv->rx_queue->next;
- spin_unlock_irqrestore(&priv->lock, flags);
- skb = dev_alloc_skb(pkt->datalen + 2);
- if (!skb) {
- //没有分配到内存
- priv->stats.rx_dropped++;
- snull_release_buffer(pkt);
- continue;
- }
- skb_reserve(skb, 2);
- memcpy(skb_put(skb, pkt->datalen), pkt->data, pkt->datalen);
- skb->dev = napi->dev;
- skb->protocol = eth_type_trans(skb, napi->dev);
- //skb->ip_summed = CHECKSUM_UNNECTSSARY;
- netif_receive_skb(skb);
- nworked++;
- priv->stats.rx_packets++;
- priv->stats.rx_bytes += pkt->datalen;
- snull_release_buffer(pkt);
- }
- //该次的poll过程结束
- if (! priv->rx_queue) {
- //已经处理完所有数据包
- napi_complete(napi);
- snull_rx_ints(napi->dev, 1);
- //return 0;
- }
-
- return nworked;
- }
- static void snull_init(struct net_device *dev)
- {
- struct snull_priv *priv;
- printk("snullnet:snull_init\n");
- priv = netdev_priv(dev);
- ether_setup(dev);
- dev->netdev_ops = &snulldev_ops;
- dev->header_ops = &snull_header_ops;
- if (use_napi) {
- priv->snull_napi = kmalloc(sizeof (struct napi_struct), GFP_KERNEL);
- netif_napi_add(dev, priv->snull_napi, snull_poll, 16);
- }
- dev->flags |= IFF_NOARP;
- //dev->features |=
- //dev->hard_header_cache =
- memset(priv, 0, sizeof(struct snull_priv));
- spin_lock_init(&priv->lock);
- snull_rx_ints(dev, 1);
- snull_setup_pool(dev);
- }
- static void snull_rx(struct net_device *dev, struct snull_packet *pkt)
- {
- struct sk_buff *skb;
- struct snull_priv *priv = netdev_priv(dev);
- printk("snullnet:snull_rx\n");
- skb = dev_alloc_skb(pkt->datalen + 2);
- if(!skb) {
- priv->stats.rx_dropped++;
- goto out;
- }
- skb_reserve(skb, 2);
- memcpy(skb_put(skb, pkt->datalen), pkt->data, pkt->datalen);
- skb->dev = dev;
- skb->protocol = eth_type_trans(skb, dev);
- skb->ip_summed = CHECKSUM_UNNECESSARY;
- priv->stats.rx_packets++;
- priv->stats.rx_bytes += pkt->datalen;
- netif_rx(skb);
- out:
- return;
- }
- static void snull_napi_interrupt(int irq, void *dev_id, struct pt_regs *regs)
- {
- int statusword;
- struct snull_priv *priv;
- struct net_device *dev = (struct net_device *)dev_id;
-
- printk("snull_napi_interrupt\n");
- if (!dev) return;
- priv = netdev_priv(dev);
- spin_lock(&priv->lock);
- statusword = priv->status;
- priv->status = 0; return;
- if (statusword & SNULL_RX_INTR) {
- snull_rx_ints(dev, 0);//禁止接收中断
- napi_enable(priv->snull_napi);
- napi_schedule(priv->snull_napi);
- }
- if (statusword & SNULL_TX_INTR) {
- priv->stats.tx_packets++;
- priv->stats.tx_bytes += priv->tx_packetlen;
- kfree_skb(priv->skb);
- }
- spin_unlock(&priv->lock);
- return ;
- }
- static void snull_regular_interrupt(int irq, \
- void *dev_id, struct pt_regs *regs) {
- int statusword;
- struct snull_priv *priv;
- struct snull_packet *pkt = NULL;
- struct net_device *dev = (struct net_device *)dev_id;
- if (!dev) return;
- priv = netdev_priv(dev);
- spin_lock(&priv->lock);
- statusword = priv->status;
- priv->status = 0;
- if (statusword & SNULL_RX_INTR) {
- pkt = priv->rx_queue;
- if (pkt) {
- priv->rx_queue = pkt->next;
- snull_rx(dev, pkt);
- }
- }
- if (statusword & SNULL_TX_INTR) {
- priv->stats.tx_packets++;
- priv->stats.tx_bytes += priv->tx_packetlen;
- dev_kfree_skb(priv->skb);
- }
-
- spin_unlock(&priv->lock);
- if (pkt) snull_release_buffer(pkt);
- return;
- }
- static void snull_free_pool(struct net_device *dev)
- {
- int i;
- struct snull_priv *priv = netdev_priv(dev);
- struct snull_packet *pkt;
- for(i = 0; i < pool_size; i++) {
- pkt = priv->ppool;
- priv->ppool = pkt->next;
- kfree(pkt);
- }
- }
- static void snullnet_clean(void)
- {
- int i;
- struct snull_priv *priv;
- printk(KERN_ALERT "snullnet exiting");
- for(i = 0;i < 2; i++) {
- if(snull_dev[i]) {
- unregister_netdev(snull_dev[i]);
- snull_free_pool(snull_dev[i]);
- priv = netdev_priv(snull_dev[i]);
- kfree(priv->snull_napi);
- free_netdev(snull_dev[i]);
- }
- }
- }
- static int __init snullnet_init(void)
- {
- int i, ret = 0;
- printk(KERN_ALERT "snullnet initing");
-
- snull_interrupt = use_napi ? snull_napi_interrupt : \
- snull_regular_interrupt;
- for(i = 0; i < 2; i++) {//分配设备
- snull_dev[i] = alloc_netdev(sizeof(struct snull_priv), "sn%d",
- snull_init);
- }
- if(!snull_dev[0] || !snull_dev[1]) {
- ret = -ENOMEM;
- goto out;
- }
-
- for(i = 0; i < 2; i++) {//注册设备
- int ret = 0;
- if((ret = register_netdev(snull_dev[i]))) {
- printk(KERN_ALERT "snull: error %i registering device \
- \"%s\"\n", ret, snull_dev[i]->name);
- goto out;
- }
- }
- return 0;
- out:
- snullnet_clean();
- return ret;
- }
- module_init(snullnet_init);
- module_exit(snullnet_clean);