(libusb) usb口自动刷新，程序员经验分享

/**
* @brief Hotplug::Hotplug
* 构造的时候init
*/
Hotplug::Hotplug() {
libusb_init(&context);
}

/**
* @brief Hotplug::Hotplug
* @param vendor
* @param product
* 委托构造
*/
Hotplug::Hotplug(uint32_t vendor, uint32_t product, uint32_t deviceClass) : Hotplug() {
/// data
{
m_regiest_vendor = vendor;
m_regiest_product = product;
m_regiest_deviceClass = deviceClass;
}

/// find
{
libusb_device **m_deviceList;
libusb_get_device_list(nullptr, &m_deviceList);
for (int i = 0; m_deviceList[i] != nullptr; i += 1) {
auto device = m_deviceList[i];
struct libusb_device_descriptor descript;
int ret = libusb_get_device_descriptor(device, &descript);
if (ret < 0) {
sprintf(
str,
“Error libusb_get_device_descriptor idx = %d res = %d\n”, i,
ret);
}

if (descript.idVendor == vendor && descript.idProduct == product) {
const int isOpen =
libusb_open(m_deviceList[i], &m_deviceHandle);
if (LIBUSB_SUCCESS == isOpen) {
break;
}
}
}
} /// find

/// test
{
Register_arrived();
Register_left();
}
}

/**
* @brief Hotplug::~Hotplug
* 析构的时候注销和释放句柄
*/
Hotplug::~Hotplug() {
libusb_hotplug_deregister_callback(context, usb_arrived_handle);
libusb_hotplug_deregister_callback(context, usb_left_handle);
libusb_exit(context);
}

/**
* @brief Hotplug::Register_arrived
* @return
* 注册热拔插arrive
*/
int Hotplug::Register_arrived() {
int res = libusb_hotplug_register_callback(
context, LIBUSB_HOTPLUG_EVENT_DEVICE_ARRIVED, LIBUSB_HOTPLUG_NO_FLAGS,
m_regiest_vendor, m_regiest_product, m_regiest_deviceClass,
Hotplug::usb_arrived_callback, NULL, &usb_arrived_handle);
return res;
}

/**
* @brief Hotplug::Register_left
* @return
* 注册热拔插left
*/
int Hotplug::Register_left() {
int res = libusb_hotplug_register_callback(
context, LIBUSB_HOTPLUG_EVENT_DEVICE_LEFT, LIBUSB_HOTPLUG_NO_FLAGS,
m_regiest_vendor, m_regiest_product, m_regiest_deviceClass,
Hotplug::usb_left_callback, NULL, &usb_left_handle);
return res;
}
} // namespace USB

code包

WIDGET_Main.h

#ifndef FORM_H_27453073957
#define FORM_H_27453073957

#include
#include
#include

#include “THREAD_TimerUsb.h”

namespace Ui {
class MainWidget;
}

class MainWidget : public QWidget {
Q_OBJECT

private:
QScopedPointerUi::MainWidget ui;

private:
TimerUsb m_usb;

public:
explicit MainWidget(QWidget *parent = nullptr);
~MainWidget();

private:
void show_usbDeviceMsgAll();
void start_work();
};

#endif // FORM_H_27453073957

WIDGET_Main.cpp

#include “WIDGET_Main.h”

#include
#include
#include
#include

#include “ui_WIDGET_Main.h”
#include “usb/USB_Reset.h”

QWidget *g_widgetDisplayBoard = nullptr;
QtMessageHandler g_oldMessageHandler = nullptr;
QMutex g_dispalyMutex;

/**
* debug 重定向
*/
void newDebugHandlerFunc(QtMsgType type, const QMessageLogContext &context, const QString &msg) {
QMutexLocker locker(&g_dispalyMutex);
if (g_widgetDisplayBoard) {
dynamic_cast<QTextEdit *>(g_widgetDisplayBoard)->append(msg);
} else {
g_oldMessageHandler(type, context, msg);
}
}

/**
* @brief MainWidget::MainWidget
* @param parent
* 1. debug重定向
* 2. connect
*/
MainWidget::MainWidget(QWidget *parent) : QWidget(parent), ui(new Ui::MainWidget{}) {
ui->setupUi(this);
setWindowTitle(“usb controller”);

/// debug -> widget
{
g_widgetDisplayBoard = ui->textEdit;
g_oldMessageHandler = qInstallMessageHandler(newDebugHandlerFunc);
}

/// connect
{
connect(ui->btn_showAll, &QPushButton::clicked, this,
&MainWidget::show_usbDeviceMsgAll);
connect(ui->btn_clear, &QPushButton::clicked, ui->textEdit,
&QTextEdit::clear);
connect(ui->btn_start, &QPushButton::clicked, this,
&MainWidget::start_work);

connect(&m_usb, &TimerUsb::signal_message, this,
[](const QString &msg) { qInfo() << msg; });
}

/// before exe
{ show_usbDeviceMsgAll(); }
}

/**
* @brief MainWidget::~MainWidget
* 将线程安全关闭
*/
MainWidget::~MainWidget() {
if (m_usb.isRunning()) {
m_usb.quit();
m_usb.wait();
}
}

/**
* @brief MainWidget::show_usbDeviceMsgAll
* 展示所有usb设备的信息
* 因为设计的是通用库
* 因此返回的是std::vectorstd::string
*/
void MainWidget::show_usbDeviceMsgAll() {
for (auto &&s : USB::Reset().Show_device()) {
qDebug() << s.c_str();
}
}

/**
* @brief MainWidget::start_work
* 检测线程启动
*/
void MainWidget::start_work() {
uint vendor = ui->edit_Vendor->text().toUInt();
double time = ui->edit_timeout->text().toDouble();
m_usb.Set_usbDeviceVendor(vendor);
m_usb.Set_timerInterval(time * 1000);
m_usb.Set_timerStart();

if (false == m_usb.isRunning()) {
qDebug() << “=== start before =";
m_usb.start();
qDebug() << "= start after ===”;
}
}

WIDGET_Main.ui

<?xml version="1.0" encoding="UTF-8"?> MainWidget 0 0 726 480 UsbForm 12 8746 12 目标Vendor 12 倒计时（秒） 12 2 18 设备usb列表 18 清空列表 18 开始

THREAD_TimerUsb.h

#ifndef THREADUSB_H_70403004403
#define THREADUSB_H_70403004403

#include
#include

class TimerUsb : public QThread {
Q_OBJECT
private:
uint32_t m_usbDeviceVendor = 0;

private:
QTimer m_timer;
int m_timerIntervalMsec = 1500;

public:
TimerUsb();

public:
void Set_usbDeviceVendor(uint32_t vendor);

public:
void Set_timerInterval(int msec = 1500);
void Set_timerStart();
void Set_timerStop();
bool Is_timerRunning();

protected:
void run() override;

signals:
void signal_message(const QString&);
};

#endif // THREADUSB_H_70403004403

THREAD_TimerUsb.cpp

#include “THREAD_TimerUsb.h”

#include
#include

#include “usb/USB_Reset.h”

#if 0
#define DEBUG_MODEL qInfo() <<
#else
#define DEBUG_MODEL emit this->signal_message
#endif

/**
* @brief ThreadUsb::ThreadUsb
* construct
*/
TimerUsb::TimerUsb() {
Set_timerInterval();
}

/**
* @brief ThreadUsb::Set_usbDeviceVendor
* @param vendor
* 根据 vendor 查询设备
*/
void TimerUsb::Set_usbDeviceVendor(uint32_t vendor) {
this->m_usbDeviceVendor = vendor;
}

/**
* @brief ThreadUsb::Set_timerInterval
* @param msec
* 设置reset间隔
*/
void TimerUsb::Set_timerInterval(int msec) {
this->m_timerIntervalMsec = msec;
m_timer.setInterval(m_timerIntervalMsec);
}

/**
* @brief TimerUsb::Set_timerStart
* 启动定时器，但不启动线程
*/
void TimerUsb::Set_timerStart() {
this->m_timer.start();
}

/**
* @brief TimerUsb::Set_timerStop
* 关闭定时器，但不关闭线程
*/
void TimerUsb::Set_timerStop() {
this->m_timer.stop();
}

/**
* @brief TimerUsb::Is_timerRunning
* @return
* 定时器是否运行
*/
bool TimerUsb::Is_timerRunning() {
return this->m_timer.isActive();
}

/**
* @brief ThreadUsb::run
* 定时器timeout一次，usb-reset一次
*/
void TimerUsb::run() {
USB::Reset usb;
libusb_device* device = nullptr;

/// 为了防止刚启动的时候没有获得
/// 高强度轮询获取
const size_t loopDeviceCount = 1e5 + 10;
const size_t loopContextPeriod = 1e3;
for (size_t i = 0; i < loopDeviceCount; i += 1) {
device = usb.Find_deviceByidVendor(this->m_usbDeviceVendor);
if (nullptr != device) {
break;
} else {
if (i % loopContextPeriod == 0) {
DEBUG_MODEL(“device is null & context resert”);
USB::Reset::Reset_context();
}
}
}
if (nullptr == device) {
DEBUG_MODEL(“libusb_device is null & Thread end!”);
return;
} else {
DEBUG_MODEL(“libusb_device require ok!”);
}

libusb_device_handle* handle = usb.Get_handleByOpenDevice(device);
if (handle == nullptr) {
DEBUG_MODEL(“libusb_device require is null & Thread end!”);
return ;
} else {
DEBUG_MODEL(“libusb_device_handle require ok!”);

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数网络安全工程师，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年网络安全全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上网络安全知识点，真正体系化！

由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新

如果你觉得这些内容对你有帮助，可以添加VX：vip204888 （备注网络安全获取）

给大家的福利

零基础入门

对于从来没有接触过网络安全的同学，我们帮你准备了详细的学习成长路线图。可以说是最科学最系统的学习路线，大家跟着这个大的方向学习准没问题。

同时每个成长路线对应的板块都有配套的视频提供：

因篇幅有限，仅展示部分资料

网络安全面试题

绿盟护网行动

还有大家最喜欢的黑客技术

网络安全源码合集+工具包

所有资料共282G，朋友们如果有需要全套《网络安全入门+黑客进阶学习资源包》，可以扫描下方二维码领取（如遇扫码问题，可以在评论区留言领取哦）~

一个人可以走的很快，但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎扫码加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

dnimg.cn/img_convert/9f3395407120bb0e1b5bf17bb6b6c743.png)

还有大家最喜欢的黑客技术

网络安全源码合集+工具包

所有资料共282G，朋友们如果有需要全套《网络安全入门+黑客进阶学习资源包》，可以扫描下方二维码领取（如遇扫码问题，可以在评论区留言领取哦）~

一个人可以走的很快，但一群人才能走的更远。不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎扫码加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！
[外链图片转存中…(img-6h9fas1o-1712464005331)]