STEPBYSTEP设计一个USB调试助手之九:基于异步传输的简单USB测试工具

//#include //#include #include #include #include #include "libusb.h"
#define TRANSFER_SIZE 64    /* 一次传输的大小 */#define VID 0x1999#define PID 0x0101#define USB_ITF_ID 0#define USB_IN_EP 0x81#define USB_OUT_EP 0x01
static void* usb_event_thread(void *arg);  /* USB事件线程处理函数 */static void* usb_handle_thread(void *arg); /* USB业务线程处理函数 */pthread_t  s_usb_event_thread;  /* USB事件处理线程句柄 */pthread_t  s_usb_handle_thread; /* USB业务处理线程句柄 */
libusb_device_handle *s_opened_handle = NULL; /* USB打开的设备句柄 */
struct libusb_transfer* s_tx_transfer = NULL; /* 发送传输         */struct libusb_transfer* s_rx_transfer = NULL; /* 接收传输         */static uint8_t s_tx_buffer[TRANSFER_SIZE]; /* 发送数据 */static uint8_t s_rx_buffer[TRANSFER_SIZE]; /* 接收数据 */static int s_tx_busy = 0;       /* 发送忙标志 */static int s_rx_busy = 0;       /* 接收忙标志 */static int s_tx_len = 0;static int s_rx_len = 0;sem_t s_sem;
int16_t vid = VID;int16_t pid = PID;int16_t itf = USB_ITF_ID;int16_t in_ep = USB_IN_EP;int16_t out_ep = USB_OUT_EP;
#define MAX_TX_LEN (sizeof(s_tx_buffer)*100)
int usb_test_main(int argc, char *argv[]){    int r;    /* 解析参数 */    if(argc != 6)    {        printf("usage:usb_test.exe vid pid itf inep outep");    }
    sscanf("%*s %#x %#x %d %#x %#x",&vid,&pid,&itf,&in_ep,&out_ep);    for(size_t i=0 ;i<sizeof(s_tx_buffer); i++)    {        s_tx_buffer[i] = i;    }
    r = libusb_init_context(/*ctx=*/NULL, /*options=*/NULL, /*num_options=*/0);    if (r < 0)    {        printf("failed to init context %d\r\n",r);        return r;    }
    s_opened_handle = libusb_open_device_with_vid_pid(NULL, vid, pid);    if (s_opened_handle == NULL)    {        printf("open dev err\r\n");        libusb_exit(NULL);        return r;    }
    r = libusb_claim_interface(s_opened_handle,itf);    if (r < 0)    {        printf("failed to claim interface %d\r\n",r);        libusb_close(s_opened_handle);        libusb_exit(NULL);        return r;    }
    sem_init(&s_sem, 0, 0);
    /* 创建usb事件处理线程 */    r = pthread_create(&s_usb_event_thread,0,usb_event_thread,0);    if (r != 0)    {        printf("failed to create usb event thread:%d\r\n",r);        libusb_release_interface(s_opened_handle,itf);        libusb_close(s_opened_handle);        libusb_exit(NULL);        return r;    }
    /* 创建usb业务处理线程 */    r = pthread_create(&s_usb_handle_thread,0,usb_handle_thread,0);    if (r != 0)    {        printf("failed to create usb handle thread:%d\r\n",r);        libusb_release_interface(s_opened_handle,itf);        libusb_close(s_opened_handle);        libusb_exit(NULL);        return r;    }
    /* 等待线程结束 */    void *res;    pthread_join(s_usb_event_thread,&res);    pthread_join(s_usb_handle_thread,&res);    sem_destroy(&s_sem);
    if(s_tx_len != s_rx_len)    {        printf("err send_len:%d,rx_len:%d!\r\n",s_tx_len,s_rx_len);    }    else    {        printf("ok send_len:%d,rx_len:%d!\r\n",s_tx_len,s_rx_len);    }
    libusb_release_interface(s_opened_handle,USB_ITF_ID);    libusb_close(s_opened_handle);    libusb_exit(NULL);
    return 0;}
void tx_cb(struct libusb_transfer *transfer){    if (transfer->status == LIBUSB_TRANSFER_COMPLETED)    {        /* 成功 */        printf("tx_cb ok\r\n");    }    else    {        /* 失败 */        printf("tx_cb err %d\r\n",transfer->status);        //libusb_submit_transfer(transfer);    }    s_tx_len += transfer->actual_length;    libusb_free_transfer(transfer);    s_tx_busy = 0;}
void rx_cb(struct libusb_transfer *transfer){    if (transfer->status == LIBUSB_TRANSFER_COMPLETED)    {        /* 成功 */        printf("rx_cb ok\r\n");    }    else    {        /* 失败 */        printf("rx_cb err %d\r\n",transfer->status);        //libusb_submit_transfer(transfer);    }
    if(transfer->actual_length > 0)    {        printf("rx len %d\r\n",transfer->actual_length);    }    s_rx_len += transfer->actual_length;    libusb_free_transfer(transfer);
    s_rx_busy = 0;}
static void* usb_event_thread(void *arg){    while(1)    {        if(0 == sem_trywait(&s_sem))        {            return 0;        }        libusb_handle_events(0);        const struct timespec interval=        {            .tv_nsec = 1000000,            .tv_sec = 0,        };        pthread_delay_np(&interval);    }    return 0;}
static void* usb_handle_thread(void *arg){    while(1)    {        int rc;        if(s_rx_len >= MAX_TX_LEN)        {            sem_post(&s_sem);            return 0;        }
        /* 发送处理 */        s_tx_transfer = libusb_alloc_transfer(0);        libusb_fill_bulk_transfer(s_tx_transfer,s_opened_handle,out_ep,s_tx_buffer,sizeof(s_tx_buffer),&tx_cb,0,100);        rc = libusb_submit_transfer(s_tx_transfer);        if(rc < 0)        {            s_tx_busy = 0;            libusb_free_transfer(s_tx_transfer);            s_tx_transfer = 0;        }
        /* 接收处理 */        s_rx_transfer = libusb_alloc_transfer(0);        libusb_fill_bulk_transfer(s_rx_transfer,s_opened_handle,in_ep,s_rx_buffer,sizeof(s_rx_buffer),&rx_cb,0,100);        rc = libusb_submit_transfer(s_rx_transfer);        if(rc < 0)        {            s_rx_busy = 0;            libusb_free_transfer(s_rx_transfer);            s_rx_transfer = 0;        }
        const struct timespec interval=        {            .tv_nsec = 1000000,            .tv_sec = 0,        };        pthread_delay_np(&interval);    }
    return 0;}

#ifndef USB_TEST_H#define USB_TEST_H
#if defined(__cplusplus)extern "C" {#endif
int usb_test_main(int argc, char *argv[]);
#if defined(__cplusplus)}#endif
#endif // USB_TEST_H

#include #include "usb_test.h"
int main(int argc, char *argv[]){    QCoreApplication a(argc, argv);    usb_test_main(argc, argv);    return a.exec();}