C++实现Linux下弹出U盘的方法

最新推荐文章于 2021-05-23 15:06:40 发布

hai0808

最新推荐文章于 2021-05-23 15:06:40 发布

阅读量1.2k

点赞数

分类专栏：其他

其他专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一个使用C++编写的程序，该程序通过系统调用和SCSI指令控制USB存储设备，实现允许移除介质及启动停止设备的功能，并强制内核重新读取分区表。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

转载http://www.jb51.net/article/69806.htm

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include <fcntl.h>
#include <string.h>
#include <linux/fd.h>
#include <sys/mount.h>
#include <scsi/scsi.h>
#include <scsi/sg.h>
#include <scsi/scsi_ioctl.h>
int main(int argc, char *argv[])
{
int fd = -1;
char *device;
if (argc != 2)
{
    printf("usage: usb-s /dev/sde1");
    return -1;
}
device = strdup(argv[1]);
if ((fd = open(device, O_RDONLY|O_NONBLOCK)) < 0)
{
    printf("open device %s failed!\n", device);
    free(device);
    return -1;
}
int status, k;
sg_io_hdr_t io_hdr;
unsigned char allowRmBlk[6] = {ALLOW_MEDIUM_REMOVAL, 0, 0, 0, 0, 0};
unsigned char startStop1Blk[6] = {START_STOP, 0, 0, 0, 1, 0};
unsigned char startStop2Blk[6] = {START_STOP, 0, 0, 0, 2, 0};
unsigned char inqBuff[2];
unsigned char sense_buffer[32];
if ((ioctl(fd, SG_GET_VERSION_NUM, &k) < 0) || (k < 30000)) {
   printf("not an sg device, or old sg driver\n");
   goto out;
}
memset(&io_hdr, 0, sizeof(sg_io_hdr_t));
io_hdr.interface_id = 'S';
io_hdr.cmd_len = 6;
io_hdr.mx_sb_len = sizeof(sense_buffer);
io_hdr.dxfer_direction = SG_DXFER_NONE;
io_hdr.dxfer_len = 0;
io_hdr.dxferp = inqBuff;
io_hdr.sbp = sense_buffer;
io_hdr.timeout = 10000;
io_hdr.cmdp = allowRmBlk;
status = ioctl(fd, SG_IO, (void *)&io_hdr);
if (status < 0)
{
   goto out;
}
io_hdr.cmdp = startStop1Blk;
status = ioctl(fd, SG_IO, (void *)&io_hdr);
if (status < 0)
{
   goto out;
}
io_hdr.cmdp = startStop2Blk;
status = ioctl(fd, SG_IO, (void *)&io_hdr);
if (status < 0)
{
   goto out;
}
/* force kernel to reread partition table when new disc inserted */
status = ioctl(fd, BLKRRPART);
out:
close(fd);
free(device);
return 0;
}