仿RW读取PCI信息
1、预期分析
使用工具:visual studio
预期实现功能:
- 获取pci信息显示
- 获取下拉框,并刷新显示的PCI信息
- 显示模式:8byte,16byte,32byte
- 获取一个帮助界面
- 建立info和summary界面,并实现切换,读取数据到summary
- 设置一个按钮,打开一个界面实现打开RWapp(自动获取路径)
2、程序设计思路
3、MFC设计
- MFC设计框图
- 主对话框设计
4、展示
-
8bit界面
注:这里info和summary子对话框需要样式改为child,否则会弹出主界面 -
16bit界面
-
32bit界面
-
下拉框
-
help界面
-
打开外部程序RW
5 、代码
MFCApplication5Dlg.cpp
// MFCApplication5Dlg.cpp: 实现文件
//
#include"windows.h"
//#include"winbase.h"
#include"shellapi.h"
#include "pch.h"
#include "framework.h"
#include "MFCApplication5.h"
#include "MFCApplication5Dlg.h"
#include "afxdialogex.h"
//以下为需要用到的额外头文件
#include<iostream>
//存储读取信息的函数
#include"IO.h"
#include"string"
//访问次级对话框
#include"my_tab2.h"
#include"helpList.h"
#include"info.h"
//连接动态链接库
#pragma comment (lib,"IO64.lib")
//声明一个次级对话框变量
my_tab2 m_page2;
helpList help_page;
info info_page;
//用于记录按钮的活动状态,初始时为0
int button_active = 0;
//创建数组用于遍历所有可能的PCI设备,并存储总数
int bdfs[256 * 16 * 8][3], total_pci = 0;
//定义nuion结构体,方便分割8位15进制数据
typedef union _INT_2_CHAR_
{
struct _Data_16
{
UINT16 FIRST_16;
UINT16 SECOND_16;
}HIGH_LOW_16;
struct _Data_8
{
UINT8 FIRST_8;
UINT8 SECOND_8;
UINT8 THIRD_8;
UINT8 FOUTH_8;
}HIGH_LOW_8;
UINT32 Data32;
}INT_CHAR;
//声明结构体变量
INT_CHAR data_poor;
#ifdef _DEBUG
#define new DEBUG_NEW
#endif
// 用于应用程序“关于”菜单项的 CAboutDlg 对话框
class CAboutDlg : public CDialogEx
{
public:
CAboutDlg();
// 对话框数据
#ifdef AFX_DESIGN_TIME
enum { IDD = IDD_ABOUTBOX };
#endif
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持
// 实现
protected:
DECLARE_MESSAGE_MAP()
public:
};
CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX)
{
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()
// CMFCApplication5Dlg 对话框
CMFCApplication5Dlg::CMFCApplication5Dlg(CWnd* pParent /*=nullptr*/)
: CDialogEx(IDD_MFCAPPLICATION5_DIALOG, pParent)
{
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}
void CMFCApplication5Dlg::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
DDX_Control(pDX, IDC_LIST1, my_list);
DDX_Control(pDX, IDC_COMBO1, m_cmb_comm);
DDX_Control(pDX, IDC_TAB1, my_tab1);
DDX_Control(pDX, IDC_BUTTON4, helpButton);
}
//这里保存了所有的交互事件,并指向事件发生时的处理程序,为组件添加处理程序时会自动创建
BEGIN_MESSAGE_MAP(CMFCApplication5Dlg, CDialogEx)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
//处理按钮被单击的事件
ON_BN_CLICKED(IDC_BUTTON1, &CMFCApplication5Dlg::OnBnClickedButton1)
ON_BN_CLICKED(IDC_BUTTON2, &CMFCApplication5Dlg::OnBnClickedButton2)
ON_BN_CLICKED(IDC_BUTTON3, &CMFCApplication5Dlg::OnBnClickedButton3)
//处理下拉选择框变动时的事件
ON_CBN_SELCHANGE(IDC_COMBO1, &CMFCApplication5Dlg::OnCbnSelchangeCombo1)
//处理列表发送变化时的事件
//ON_NOTIFY(LVN_ITEMCHANGED, IDC_LIST1, &CMFCApplication5Dlg::OnLvnItemchangedList1)
ON_BN_CLICKED(IDC_BUTTON4, &CMFCApplication5Dlg::OnBnClickedButton4)
ON_BN_CLICKED(IDC_BUTTON5, &CMFCApplication5Dlg::OnBnClickedButton5)
ON_NOTIFY(TCN_SELCHANGE, IDC_TAB1, &CMFCApplication5Dlg::OnTcnSelchangeTab1)
END_MESSAGE_MAP()
// CMFCApplication5Dlg 消息处理程序
BOOL CMFCApplication5Dlg::OnInitDialog()
{
CDialogEx::OnInitDialog();
// 将“关于...”菜单项添加到系统菜单中。
// IDM_ABOUTBOX 必须在系统命令范围内。
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
if (pSysMenu != nullptr)
{
BOOL bNameValid;
CString strAboutMenu;
bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
ASSERT(bNameValid);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
}
// 设置此对话框的图标。 当应用程序主窗口不是对话框时,框架将自动
// 执行此操作
SetIcon(m_hIcon, TRUE); // 设置大图标
SetIcon(m_hIcon, FALSE); // 设置小图标
// TODO: 在此添加额外的初始化代码
//以下为启动程序时面板中的初始化内容
//创建帮助界面次级对话框
help_page.Create(IDD_DIALOG4);//创建一个非模态对话框
//在tab组件中插入一个分页
my_tab1.InsertItem(0, _T("info"), LVCFMT_CENTER);
my_tab1.InsertItem(1, _T("Summary"));
//将次级对话框与tab组件联系起来
m_page2.Create(IDD_DIALOG1, GetDlgItem(IDC_TAB1));
info_page.Create(IDD_DIALOG2, GetDlgItem(IDC_TAB1));
//定义一个变量用来保存组件的坐标
CRect rs;
my_tab1.GetClientRect(&rs);
//调整子对话框在父窗口中的位置
rs.top += 20;
rs.bottom -= 0;
rs.left -= 20;
rs.right -= 0;
//设置子对话框尺寸并移动到指定位置
m_page2.MoveWindow(&rs);
info_page.MoveWindow(&rs);
//pDialog[0] = &m_page2;
//pDialog[0]-> ShowWindow(SW_SHOW);
//分别设置隐藏和显示
info_page.ShowWindow(false);
m_page2.ShowWindow(true);
//设置默认选项卡
my_tab1.SetCurSel(1);
//获取原表的样式
DWORD dwStyle = my_list.GetExtendedStyle();
//整行选取
dwStyle |= LVS_EX_FULLROWSELECT;
//添加网格线
dwStyle |= LVS_EX_GRIDLINES;
//为每行添加选择框
//dwStyle |= LVS_EX_CHECKBOXES;
//设置背景板颜色
//my_list.SetBkColor(RGB(255, 0, 0));
//设置表格颜色
//my_list.SetTextBkColor(RGB(0, 100, 10));
//设置文本颜色
my_list.SetTextColor(RGB(0, 100, 10));
//重新设定列表样式
my_list.SetExtendedStyle(dwStyle);
//重置下拉选项框
m_cmb_comm.ResetContent();
//遍历所有可能的PCI设备,如果存在则将在主要信息存入bdfs数组
for (int b = 0; b < 256; b++)
{
for (int d = 0; d < 32; d++)
{
for (int f = 0; f < 8; f++)
{
if (ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, 0)) != 0xFFFFFFFF)
{
bdfs[total_pci][0] = b;
bdfs[total_pci][1] = d;
bdfs[total_pci][2] = f;
//每有一个存在的设备则总数加1
total_pci++;
}
}
}
}
//根据bfds数组中的数据为下拉选项框添加存在的设备
for (int i = 0; i < total_pci; i++)
{
TCHAR s[1000] = { 0 };
_stprintf_s(s, TEXT("Bus % 02X, Device % 02X, Function % 02X"), bdfs[i][0], bdfs[i][1], bdfs[i][2]);
m_cmb_comm.AddString(s);
}
//设定默认选项
m_cmb_comm.SetCurSel(0);
//调用函数设定表格样式(button_active为按钮活动状态,初始为0,即8字节)
CMFCApplication5Dlg::clear_item(button_active);
//将第“0”个设备的PCI8字节信息写入表格中
CMFCApplication5Dlg::write_item(button_active, bdfs[0][0], bdfs[0][1], bdfs[0][2]);
//将第“0”个设备的Device ID等信息写入次级对话框中
CMFCApplication5Dlg::write_tab(bdfs[0][0], bdfs[0][1], bdfs[0][2]);
return TRUE; // 除非将焦点设置到控件,否则返回 TRUE
}
void CMFCApplication5Dlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialogEx::OnSysCommand(nID, lParam);
}
}
// 如果向对话框添加最小化按钮,则需要下面的代码
// 来绘制该图标。 对于使用文档/视图模型的 MFC 应用程序,
// 这将由框架自动完成。
void CMFCApplication5Dlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // 用于绘制的设备上下文
SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
// 使图标在工作区矩形中居中
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// 绘制图标
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialogEx::OnPaint();
}
}
//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CMFCApplication5Dlg::OnQueryDragIcon()
{
return static_cast<HCURSOR>(m_hIcon);
}
//下拉选项框发生变动时的事件处理程序
void CMFCApplication5Dlg::OnCbnSelchangeCombo1()
{
// TODO: 在此添加控件通知处理程序代码
//首先传递当前按钮活动状态,设定表格的格式
CMFCApplication5Dlg::clear_item(button_active);
//捕捉下拉选择框的选项
int info = m_cmb_comm.GetCurSel();
//获取对应的Bus、Device、Function参数
int b = bdfs[info][0], d = bdfs[info][1], f = bdfs[info][2];
//传递参数将对应信息写入表格
CMFCApplication5Dlg::write_item(button_active, b, d, f);
//传递参数将对应信息写入次级对话框
CMFCApplication5Dlg::write_tab(b, d, f);
}
//void CMFCApplication5Dlg::OnLvnItemchangedList1(NMHDR* pNMHDR, LRESULT* pResult)
//{
// LPNMLISTVIEW pNMLV = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
// // TODO: 在此添加控件通知处理程序代码
// *pResult = 0;
//}
//当按钮1被单击时的事件处理程序
void CMFCApplication5Dlg::OnBnClickedButton1()
{
// TODO: 在此添加控件通知处理程序代码
//设定按钮活动状态
button_active = 0;
//改变表格格式
CMFCApplication5Dlg::clear_item(button_active);
//捕捉选择框中的选项
int info = m_cmb_comm.GetCurSel();
//获取对应的参数
int b = bdfs[info][0], d = bdfs[info][1], f = bdfs[info][2];
//将对应的信息写入表格
CMFCApplication5Dlg::write_item(button_active, b, d, f);
}
//当按钮2被单击时的事件处理程序
void CMFCApplication5Dlg::OnBnClickedButton2()
{
// TODO: 在此添加控件通知处理程序代码
//改变按钮活动状态
button_active = 1;
//根据按钮活动状态设定表格格式
CMFCApplication5Dlg::clear_item(button_active);
//捕捉选择框选项
int info = m_cmb_comm.GetCurSel();
//获取设备对应的参数
int b = bdfs[info][0], d = bdfs[info][1], f = bdfs[info][2];
//将对应的信息写入表格
CMFCApplication5Dlg::write_item(button_active, b, d, f);
}
//当按钮3被单击时的事件处理程序
void CMFCApplication5Dlg::OnBnClickedButton3()
{
// TODO: 在此添加控件通知处理程序代码
button_active = 2;
// TODO: 在此添加控件通知处理程序代码
//根据按钮活动状态设定表格格式
CMFCApplication5Dlg::clear_item(button_active);
//捕捉选择框中的选项
int info = m_cmb_comm.GetCurSel();
//获取对应的参数
int b = bdfs[info][0], d = bdfs[info][1], f = bdfs[info][2];
//将对应的信息写入表格
CMFCApplication5Dlg::write_item(button_active, b, d, f);
}
//自定义函数,用于设定表格的格式,由活动按钮决定
void CMFCApplication5Dlg::clear_item(int a)
{
my_list.DeleteAllItems();
for (int i = 0; i < 17; i++)
{
my_list.DeleteColumn(0);
}
CRect mRect;
my_list.GetWindowRect(&mRect);
int width = mRect.Width();
my_list.InsertColumn(0, _T("0"), LVCFMT_CENTER, 30);
switch (a)
{
//当第一个按钮活动时,表格为17x17(连同索引)
case 0:
{
for (int i = 0; i < 16; i++)
{
TCHAR s[3] = { 0 };
_stprintf_s(s, TEXT("0%X"), i);
my_list.InsertColumn(i + 1, s, LVCFMT_CENTER, (width - 30) / 16);
}
for (int i = 0; i < 16; i++)
{
TCHAR s[3] = { 0 };
_stprintf_s(s, TEXT("%X0"), i);
my_list.InsertItem(i, s, LVCFMT_CENTER);
}
break;
}
//当第二个按钮活动时,表格为17x9(连同索引)
case 1:
{
for (int i = 0; i < 16; i += 2)
{
TCHAR s[5] = { 0 };
_stprintf_s(s, TEXT("0%X0%X"), i + 1, i);
my_list.InsertColumn(i / 2 + 1, s, LVCFMT_CENTER, (width - 30) / 8);
}
for (int i = 0; i < 16; i++)
{
TCHAR s[5] = { 0 };
_stprintf_s(s, TEXT("%X0"), i);
my_list.InsertItem(i, s, LVCFMT_CENTER);
}
break;
}
//当第三个按钮活动时,表格为17x5(连同索引)
case 2:
{
for (int i = 0; i < 16; i += 4)
{
TCHAR s[9] = { 0 };
_stprintf_s(s, TEXT("0%X0%X0%X0%X"), i + 3, i + 2, i + 1, i);
my_list.InsertColumn(i / 4 + 1, s, LVCFMT_CENTER, (width - 30) / 4);
}
for (int i = 0; i < 16; i++)
{
TCHAR s[9] = { 0 };
_stprintf_s(s, TEXT("%X0"), i);
my_list.InsertItem(i, s, LVCFMT_CENTER);
}
break;
}
};
}
//以下为自定义函数,用于写入对应设备及对应格式的信息,由活动按钮和选择框共同决定
//参数b、d、f用于读取设备信息
void CMFCApplication5Dlg::write_item(int a, int b, int d, int f)
{
//参数a为按钮活动状态
switch (a)
{
//写入8字节信息
case 0:
{
//用于访问要写入的列
int j = 0;
for (int i = 0; i <= 252; i++)
{
TCHAR s[5] = { 0 };
if (i % 4 == 0)
{
//调用ReaPCIConfig函数读取信息并存入结构体
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, i));
//获取第一个8位信息,将其转换为相应格式,并写入表格
_stprintf_s(s, TEXT("%02X"), data_poor.HIGH_LOW_8.FIRST_8);
my_list.SetItemText(i / 16, j + 1, s);
//获取第二个8位信息,将其转换为相应格式,并写入表格
_stprintf_s(s, TEXT("%02X"), data_poor.HIGH_LOW_8.SECOND_8);
my_list.SetItemText(i / 16, j + 2, s);
//获取第三个8位信息,将其转换为相应格式,并写入表格
_stprintf_s(s, TEXT("%02X"), data_poor.HIGH_LOW_8.THIRD_8);
my_list.SetItemText(i / 16, j + 3, s);
//获取第四个8位信息,将其转换为相应格式,并写入表格
_stprintf_s(s, TEXT("%02X"), data_poor.HIGH_LOW_8.FOUTH_8);
my_list.SetItemText(i / 16, j + 4, s);
//每循环一次,将列数加4
j += 4;
}
//如果已经写到了行尾,则重置变量j
if (j == 16)
j = 0;
}
break;
};
//写入16字节信息
case 1:
{
//用于访问要写入的列
int j = 0;
for (int i = 0; i <= 252; i++)
{
TCHAR s[9] = { 0 };
if (i % 4 == 0)
{
//调用函数获取信息
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, i));
//写入第一个16位信息
_stprintf_s(s, TEXT("%04X"), data_poor.HIGH_LOW_16.FIRST_16);
my_list.SetItemText(i / 16, j + 1, s);
//写入第二个16位信息
_stprintf_s(s, TEXT("%04X"), data_poor.HIGH_LOW_16.SECOND_16);
my_list.SetItemText(i / 16, j + 2, s);
//每循环一次,列数加2
j += 2;
}
//如果已经写到行尾,则重置
if (j == 8)
j = 0;
}
break;
};
//用于写入32位数据
case 2:
{
//用于访问要写入的列
int j = 0;
for (int i = 0; i <= 252; i++)
{
TCHAR s[17] = { 0 };
if (i % 4 == 0)
{
//调用函数获取信息
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, i));
//将32位信息写入表格
_stprintf_s(s, TEXT("%08X"), data_poor.Data32);
my_list.SetItemText(i / 16, j + 1, s);
//每写入一条列数加1
j++;
}
//如果已经写到行尾,则重置
if (j == 4)
j = 0;
}
break;
}
};
}
//以下为自定义函数,用于将部分设备信息写入次级对话框,由选择框决定(与按钮活动状态无关)
void CMFCApplication5Dlg::write_tab(int b, int d, int f)
{
TCHAR s[17] = { 0 };
//每个设备信息的第一个32字节数据
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, 0));
_stprintf_s(s, TEXT("0x%08X"), data_poor.Data32);
m_page2.SetDlgItemText(IDC_STATIC_1r, s);
//设备信息的第3个32字节数据的末尾两位
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, 11));
_stprintf_s(s, TEXT("0x%02X"), data_poor.HIGH_LOW_8.FIRST_8);
m_page2.SetDlgItemText(IDC_STATIC_2r, s);
//设备信息的第3个32字节数据的前6位
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, 11));
_stprintf_s(s, TEXT("0x%04X%02X"), data_poor.HIGH_LOW_16.SECOND_16, data_poor.HIGH_LOW_8.THIRD_8);
m_page2.SetDlgItemText(IDC_STATIC_3r, s);
//设备信息的第4个32字节数据的末尾两位
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, 15));
_stprintf_s(s, TEXT("0x%02X"), data_poor.HIGH_LOW_8.FIRST_8);
m_page2.SetDlgItemText(IDC_STATIC_4r, s);
//设备信息的第12个32字节数据
data_poor.Data32 = ReadPCIConfig(MAKE_CONFIG_ADDRESS(b, d, f, 44));
_stprintf_s(s, TEXT("0x%08X"), data_poor.Data32);
m_page2.SetDlgItemText(IDC_STATIC_5r, s);
}
//控制my_tab1表的显示和隐藏
void CMFCApplication5Dlg::OnTcnSelchangeTab1(NMHDR* pNMHDR, LRESULT* pResult)
{
// TODO: 在此添加控件通知处理程序代码
int CurSel = my_tab1.GetCurSel();
switch (CurSel)
{
case 0:
info_page.ShowWindow(true);
m_page2.ShowWindow(false);
break;
case 1:
info_page.ShowWindow(false);
m_page2.ShowWindow(true);
break;
default:
;
}
*pResult = 0;
}
//帮助界面
void CMFCApplication5Dlg::OnBnClickedButton4()
{
// TODO: 在此添加控件通知处理程序代码
//help_page.Create(IDD_DIALOG4);//在这里创建,按钮只能打开一次help界面
help_page.ShowWindow(true);//显示非模态对话框
}
//按钮实现打开外部RW程序
void CMFCApplication5Dlg::OnBnClickedButton5()
{
// TODO: 在此添加控件通知处理程序代码
//方式一
//绝对路径打开
// WinExec("C:\\Users\\1\\Desktop\\RW\\Win64\\Portable\\Rw.exe", SW_SHOWNORMAL); // 以普通的方式打开Test.exe
//方式二
//ShellExecute(NULL,L"open",L".\\res\\help.chm",NULL,NULL,SW_SHOWMAXIMIZED);
//获取当前目录,拼目录打开RW
CString path;
GetModuleFileName(NULL, path.GetBufferSetLength(MAX_PATH + 1), MAX_PATH);
path.ReleaseBuffer();
int pos = path.ReverseFind('\\');
path = path.Left(pos);
path = path + L"\\RW\\Win64\\Portable\\Rw.exe";
ShellExecute(NULL, L"Open", path, NULL, NULL, SW_SHOWMAXIMIZED);
}
my_tab2.cpp(summery子对话框)
// my_tab2.cpp: 实现文件
//
#include "pch.h"
#include "MFCApplication5.h"
#include "afxdialogex.h"
#include "my_tab2.h"
// my_tab2 对话框
IMPLEMENT_DYNAMIC(my_tab2, CDialogEx)
my_tab2::my_tab2(CWnd* pParent /*=nullptr*/)
: CDialogEx(IDD_DIALOG1, pParent)
{
}
my_tab2::~my_tab2()
{
}
void my_tab2::DoDataExchange(CDataExchange* pDX)
{
CDialogEx::DoDataExchange(pDX);
SetDlgItemText(IDC_STATIC_1, _T("Device/Vendor ID"));
SetDlgItemText(IDC_STATIC_2, _T("Revision ID"));
SetDlgItemText(IDC_STATIC_3, _T("Class Code"));
SetDlgItemText(IDC_STATIC_4, _T("Cacheline Size"));
SetDlgItemText(IDC_STATIC_5, _T("Subsystem ID"));
}
BEGIN_MESSAGE_MAP(my_tab2, CDialogEx)
END_MESSAGE_MAP()
// my_tab2 消息处理程序
注:这里需要使用到IO.h和IO.dll和IO.64lib文件,因为属于内部文件,这里需要自己去找文件了,还有,这个代码不能拿去直接跑呀,需要先建立mfc界面
最后,读到这儿,就写完了,嘿嘿!
大胆尝试吧,代码写错了可以再改,大不了重头来过