USBIP源码分析

简介

在普通的电脑上，想使用USB设备，必须将插入到主机。USBIP却可以通过网络，让主机访问其他主机上的外部设备，而用户程序完全感知不到区别。

usbip的文章在这里：https://pdfs.semanticscholar.org/c7c4/cb054d75810fdb0a2affa11c288b7687267f.pdf

本文基于linux内核4.4.3，分析USBIP这部分的源码。先介绍整体结构，然后分两部分介绍USBIP源码

USBIP整体架构

从体系机构的角度上来说，USB的设备和总线都是通过Host分出，host叫做主机控制器，一个主机控制器会有一个root hub，然后root hub再通过接口分出多个hub，最后，一个hub的一个端口都可以用来连接一个外部设备或是宁一个hub，形成一个以host为跟的树状结构。而Host最终是作为一个PCI的设备，连接在PCI总线上。

在linux内核中，USB驱动的架构可以分成3层。

• 最靠近用户层的是USB设备驱动，这是每个USB设备有独有的，在它之上可以直接与VFS交互，比如键盘和鼠标这些设备，USB设备驱动就足以处理外部设备的交互任务，因为他们发送过来的就只有是一些很简单的数据，但是如果是一个插在USB上的网卡等设备，则需要在USB设备驱动之上再覆盖一层网卡等驱动，用来解析USB总线传输过来的数据。它与下层的USB Core通过urb交换数据。
• USB Core则是承上启下的一层，USB设备驱动通过urb的抽次昂概念来和底层交互，USB Core就用来解析urb，同时，还会包含Hub驱动等等一些USB驱动框架的其他主要部分。
• 最下层的是主机控制器驱动，前面看到USB外部设备最终通过Host和总线、CPU交互，也就是说最底层还需要有一个主机控制器的驱动，它主要负责Host的相关工作，将USB设备的信息通过主机控制器向上传递软件层。这部分会有一个PCI驱动，然后控制Host硬件

理解了USB驱动的框架，USBIP的架构就比价容易了，主要部分也是两个，读取设备的主机端，设置一个虚拟的主机控制器接口VHCI，它不操纵底层的主机控制器，而是将上层的消息通过网络转发到另一个主机，在另一侧，实现一个USB设备驱动，它不是将USB Core的内容向上传递，同样是通过网络发送出去，叫做Stub端。

在Linux内核中，USBIP的源码写在drivers/usb/usbip目录下，在这些文件中，以stub开头的都是server端的代码，vhci开头的是client端的代码，其余是公共部分的代码。

下面从Stub和VHCI的角度来分析。

Stub端

主要结构体

usbip_common中有一个usbip_device结构，这是stub和vhci两边设备从后向出来的公共部分。这里有3个内核线程，一个socket，以及和eh相关的等待队列和操作集合。usbip_common中其他几个结构就不再叙述，比较简单

struct usbip_device {
    enum usbip_side side;
    enum usbip_device_status status;

    /* lock for status */
    spinlock_t lock;

    struct socket *tcp_socket;//用来通信的socket

    struct task_struct *tcp_rx;//收消息的线程
    struct task_struct *tcp_tx;//发送消息的线程

    unsigned long event;//记录事件
    struct task_struct *eh;//内核线程
    wait_queue_head_t eh_waitq;//eh等待队列

    struct eh_ops {
        void (*shutdown)(struct usbip_device *);
        void (*reset)(struct usbip_device *);
        void (*unusable)(struct usbip_device *);
    } eh_ops;
};

stub.h中定义了关键的结构体，他们的内容和含义如下:

stub_device是stub端的设备抽象，代表以一个外部设备

struct stub_device {
    struct usb_interface *interface;//接口描述符指针
    struct usb_device *udev;
    struct usbip_device ud;//对于stub和vhci两端的设备都做了抽象，用来表示两端交互中都需要的内容
    __u32 devid;
    spinlock_t priv_lock;
    struct list_head priv_init;//urb初始队列
    struct list_head priv_tx;//urb被提交之后
    struct list_head priv_free;//urb的内容被发送给了chci
    struct list_head unlink_tx;//unlink请求队列
    struct list_head unlink_free;//unlink请求被处理
    wait_queue_head_t tx_waitq;//等待队列
};

stub_priv被赋值给urb->priv字段，主要是给stub端来管理urb结构体

struct stub_priv {
    unsigned long seqnum;//给每个urb都有一个序列号
    struct list_head list;
    struct stub_device *sdev;
    struct urb *urb;
    int unlinking;//是否被unlink
};

stub_unlink表示一个urb的unlink请求

struct stub_unlink {
    unsigned long seqnum;//和stub_priv对于的序列号
    struct list_head list;
    __u32 status;//unlink是否成功
};

stub_main中有一个全局变量的数组，用来管理所有的设备

static struct bus_id_priv busid_table[MAX_BUSID];

其中一个设备用bus_id_priv来表示

struct bus_id_priv {
    char name[BUSID_SIZE];
    char status;
    int interf_count;
    struct stub_device *sdev;
    struct usb_device *udev;
    char shutdown_busid;
};

模块的初始化函数

先看stub_main中最后几行，有模块的的初始和卸载函数

module_init(usbip_host_init);
module_exit(usbip_host_exit);

usbip_host_init中主要函数有4个，做的工作如下

1、初始化全局变量busid_table
2、分配一个slab用来做stub_priv的分配工作
3、注册一个usb驱动，这里是stub的初始化，这里注册的驱动也是位于设备端，USB核心上层的USB驱动
4、创建两个sysfs文件，这两个文件的操作就在上面，rebind_store和store_match_busid、show_match_busid。

下面的退出函数usbip_host_exit同理，只是做了这几个函数的清理工作

先看看它创建的两个sysfs文件。这是给用户态的接口。一个是match_busid

static DRIVER_ATTR(match_busid, S_IRUSR | S_IWUSR, show_match_busid,store_match_busid);

涉及的函数是show_match_busid和store_match_busid，show_match_busid用来输出名字，store_match_busid则用来增加或是删除busid_table中的条目

另一个文件则是用来设置设备绑定的驱动。

这几个文件的操作都是围绕busid_table和它的几个操作函数开展开，也比较简单。整个文件的内容也分析完了。下面就还有一个关键的usb驱动，也就是Stub驱动，很显然，主要的工作都是通过这个驱动来开展的。

Stub驱动

stub_dev文件最底部有这个驱动

struct usb_device_driver stub_driver = {
    .name        = "usbip-host",
    .probe        = stub_probe,
    .disconnect    = stub_disconnect,
#ifdef CONFIG_PM
    .suspend    = stub_suspend,
    .resume        = stub_resume,
#endif
    .supports_autosuspend    =    ,
};

从probe函数开始，这里只说一些和Stub驱动自身逻辑相关的重要部分，proe函数通过stub_device_alloc分配了一个stub_device函数。而这个stub_device_alloc函数中还有一个usbip_start_eh函数。用来创建了一个内核线程eh。

int usbip_start_eh(struct usbip_device *ud)
{
    init_waitqueue_head(&ud->eh_waitq);
    ud->event = ;

    ud->eh = kthread_run(event_handler_loop, ud, "usbip_eh");
    if (IS_ERR(ud->eh)) {
        pr_warn("Unable to start control thread\n");
        return PTR_ERR(ud->eh);
    }

    return ;
}
EXPORT_SYMBOL_GPL(usbip_start_eh);

前面看到usbip_device结构中有3个内核线程，这是其中一个，二且eh_waitq就是用来给eh睡眠的，出发事件写在usbip_event_happened函数中，只是检查usbip_device上是否有事件产生，event字段会否为0。

好，接着回到probe函数，里面有一个stub_add_files函数。可以知道这个函数同样创建了几个sysfs文件，他们为dev_attr_usbip_status、dev_attr_usbip_sockfd、dev_attr_usbip_debug。

其中比较关键的是dev_attr_usbip_sockfd，当写入数据为-1时执行关闭操作，另一条分支则会通过写入数据打开socket，赋值给tcp_socket字段，接着创建了两个内核线程。

看看这个stub_dev文件，剩下的代码都是处理断开和释放的函数了，这里就不分析，剩下的两个文件stub_rx和stub_tx可以知道这是和内核线程相关的了。所以说剩下的主要内容就是这两个内核线程。

rx和tx两个内核线程和stub_device中的5个链表

先看rx，主函数如下，沿着这个逻辑往下看

int stub_rx_loop(void *data)
{
    struct usbip_device *ud = data;

    while (!kthread_should_stop()) {
        if (usbip_event_happened(ud))
            break;

        stub_rx_pdu(ud);
    }

    return ;
}

stub_rx_pdu函数中，调用usbip_recv收一个消息，usbip_header_correct_endian转换成小端，然后根据收到的消息来做不同的处理，如果是unlink消息则调用stub_recv_cmd_unlink，如果是urb的提交消息则调用stub_recv_cmd_submit。

说一下priv_init、priv_tx、priv_free这3个队列。stub中的urb通过stub_priv来进行管理。通过stub_priv，给每个urb分配了序列号，同时，通过stub_priv中的list字段，将urb连接到上述3个队列中，当urb被提交给USB Core，直到完成前，放在priv_init，提交完成后，还需要通过网络发送给vhci，发送之前放在priv_tx，发送之后放在priv_free。

所以说，stub_recv_cmd_unlink的行为就是遍历priv_init，找到那些还没有完成的urb，调用usb_unlink_urb让USB Core去取消他们。

stub_recv_cmd_submit函数，调用了stub_priv_alloc，这个函数的末尾将urb加入到了priv_init中。最后调用usb_submit_urb提交urb，注意这里的回调函数，是stub_complete。这个函数在urb执行完后执行下面的语句，将urb移入priv_tx

list_move_tail(&priv->list, &sdev->priv_tx);

然后，调用了wake_up，唤醒tx内核线程

wake_up(&sdev->tx_waitq);

接着看rx线程，主要是两个函数stub_send_ret_submit和stub_send_ret_unlink，因为这是一个发送消息的线程，所以发送的消息有两种，一个是USB设备的交互内容，一种是vhci发送的unlink消息的回复。

stub_send_ret_submit处理的就是USB设备的返回内容，dequeue_from_priv_tx将urb从priv_tx取下，放入priv_free。stub_send_ret_submit最后再将内容封装成usbip协议规定的样子，通过网络发送出去。

下面再介绍剩下的两个队列，unlink_tx和unlink_free，这是用来处理unlink的两个队列，一个stub_unlink结构表示一个unlink请求，unlink_tx存放还没有被回复的unlink请求，而unlink_free则是存放已经回复了的。

所以stub_send_ret_unlink中的dequeue_from_unlink_tx用来完成unlink的队列转换，剩下的代码就会通过网络发送回复消息

这样，stub这边的几个函数就都已经完了

VHCI端

这边的代码和上面已经非常相似了，抓住3个点

1、模块的初始化时注册的驱动程序，在usb_add_hcd中会调用reset和start函数，接着抓住urb的几个操作函数就可以了

static struct hc_driver vhci_hc_driver = {
    .description    = driver_name,
    .product_desc    = driver_desc,
    .hcd_priv_size    = sizeof(struct vhci_hcd),

    .flags        = HCD_USB2,

    .start        = vhci_start,
    .stop        = vhci_stop,

    .urb_enqueue    = vhci_urb_enqueue,
    .urb_dequeue    = vhci_urb_dequeue,

    .get_frame_number = vhci_get_frame_number,

    .hub_status_data = vhci_hub_status,
    .hub_control    = vhci_hub_control,
    .bus_suspend    = vhci_bus_suspend,
    .bus_resume    = vhci_bus_resume,
};

2、创建的几个sysfs文件。

3、3个内核线程的工作

就不赘述了