整理一些常用的代码示例

已于 2023-05-17 20:10:33 修改
阅读量1.2k 收藏 3
点赞数 2
分类专栏： UEFI 文章标签： UEFI 学习
于 2023-05-17 19:59:42 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u011397314/article/details/130733362
版权

关于PCI_ROOT_BRIDGE_IO_PROTOCOL:

作用：该协议用于与PCI总线上的根桥（Root Bridge）进行通信，根桥是PCI总线的起始点，它负责转发数据和控制信号。
功能：EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL提供了访问PCI总线上的根桥所需的各种功能和方法，包括内存读写、I/O读写、配置空间读写等。
使用场景：主要用于与PCI总线上的根桥进行通信和配置，例如初始化PCI设备、读取和修改PCI设备的配置寄存器等。

#include <Uefi.h>
#include <Library/UefiLib.h>
#include <Protocol/PciRootBridgeIo.h>

EFI_STATUS EnumeratePciDevices()
{
    EFI_STATUS Status;
    EFI_HANDLE *HandleBuffer;
    UINTN HandleCount;

    // 获取PCI根桥IO协议的句柄
    Status = gBS->LocateHandleBuffer(
        ByProtocol,
        &gEfiPciRootBridgeIoProtocolGuid,
        NULL,
        &HandleCount,
        &HandleBuffer);

    if (EFI_ERROR(Status))
    {
        // 错误处理
        return Status;
    }

    UINTN Index;
    EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL *PciRootBridgeIo;

    // 遍历所有PCI根桥IO协议
    for (Index = 0; Index < HandleCount; Index++)
    {
        // 获取PCI根桥IO协议实例
        Status = gBS->HandleProtocol(
            HandleBuffer[Index],
            &gEfiPciRootBridgeIoProtocolGuid,
            (VOID **)&PciRootBridgeIo);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        // 获取PCI根桥上的所有PCI设备
        UINT64 Segment;
        Status = PciRootBridgeIo->Configuration->GetSegment(PciRootBridgeIo->Configuration, &Segment);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        UINT64 Bus;
        for (Bus = 0; Bus <= 0xFF; Bus++)
        {
            UINT64 Device;
            for (Device = 0; Device <= 0x1F; Device++)
            {
                UINT64 Function;
                for (Function = 0; Function <= 0x7; Function++)
                {
                    // 构建PCI设备地址
                    EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL_PCI_ADDRESS PciAddress;
                    PciAddress.Bus = (UINT32)Bus;
                    PciAddress.Device = (UINT32)Device;
                    PciAddress.Function = (UINT32)Function;

                    // 读取设备的Vendor ID和Device ID
                    UINT16 VendorId, DeviceId;
                    Status = PciRootBridgeIo->Pci.Read(
                        PciRootBridgeIo,
                        EfiPciWidthUint16,
                        PciAddress,
                        offsetof(PCI_DEVICE_HEADER, VendorId),
                        1,
                        &VendorId);

                    if (EFI_ERROR(Status))
                    {
                        // 错误处理
                        continue;
                    }

                    Status = PciRootBridgeIo->Pci.Read(
                        PciRootBridgeIo,
                        EfiPciWidthUint16,
                        PciAddress,
                        offsetof(PCI_DEVICE_HEADER, DeviceId),
                        1,
                        &DeviceId);

                    if (EFI_ERROR(Status))
                    {
                        // 错误处理
                        continue;
                    }

                    // 打印设备的Vendor ID和Device ID
                    Print(L"PCI Device: Bus %02X, Device %02X, Function %X\n", Bus, Device, Function);
                    Print(L"Vendor ID: %04X\n", VendorId);
                    Print(L"Device ID: %04X\n\n", DeviceId);
                }
            }
        }
    }
 // 释放句柄缓冲区
    gBS->FreePool(HandleBuffer);

    return EFI_SUCCESS;
}

关于EFI_PCI_IO_PROTOCOL：

EFI_PCI_IO_PROTOCOL是用于与具体设备通信的协议，用于访问PCI总线上的具体设备。
在使用时，首先使用EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL来初始化和配置PCI总线上的根桥，然后使用EFI_PCI_IO_PROTOCOL来与具体设备进行通信和控制。
EFI_PCI_ROOT_BRIDGE_IO_PROTOCOL提供了用于访问配置空间的方法，可以用于获取和修改设备的配置信息。
EFI_PCI_IO_PROTOCOL提供了更多的操作方法，可以进行内存读写、I/O读写、数据传输等操作。

这里主要两个示例合二为一，读取配置空间以及BAR中的内容（网卡的mac地址）：

#include <Uefi.h>
#include <Protocol/PciIo.h>

EFI_STATUS GetMacAddressFromPci()
{
    EFI_STATUS Status;
    EFI_HANDLE *HandleBuffer;
    UINTN HandleCount;

    // 获取所有PCI设备的句柄
    Status = gBS->LocateHandleBuffer(
        ByProtocol,
        &gEfiPciIoProtocolGuid,
        NULL,
        &HandleCount,
        &HandleBuffer);

    if (EFI_ERROR(Status))
    {
        // 错误处理
        return Status;
    }

    UINTN Index;
    EFI_PCI_IO_PROTOCOL *PciIo;

    // 遍历所有PCI设备
    for (Index = 0; Index < HandleCount; Index++)
    {
        // 获取PCI IO Protocol
        Status = gBS->HandleProtocol(
            HandleBuffer[Index],
            &gEfiPciIoProtocolGuid,
            (VOID **)&PciIo);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        // 判断是否为网卡设备（可根据设备的Vendor ID和Device ID进行判断）
        UINT16 VendorId, DeviceId;
        Status = PciIo->Pci.Read(
            PciIo,
            EfiPciIoWidthUint16,
            0x00, // Vendor ID寄存器地址
            1,    // 寄存器数量
            &VendorId);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        Status = PciIo->Pci.Read(
            PciIo,
            EfiPciIoWidthUint16,
            0x02, // Device ID寄存器地址
            1,    // 寄存器数量
            &DeviceId);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        if (VendorId == 0x10EC && DeviceId == 0x8168)
        {
            // 根据具体的Vendor ID和Device ID判断是否为目标网卡设备

            // 读取MAC地址
            UINT8 MacAddress[6];
            Status = PciIo->Pci.Read(
                PciIo,
                EfiPciIoWidthUint8,
                0x00, // MAC地址起始寄存器地址
                6,    // 寄存器数量
                MacAddress);

            if (EFI_ERROR(Status))
            {
                // 错误处理
                continue;
            }

            // 打印MAC地址
            Print(L"MAC Address: %02x:%02x:%02x:%02x:%02x:%02x\n",
                  MacAddress[0], MacAddress[1], MacAddress[2],
                  MacAddress[3], MacAddress[4], MacAddress[5]);
        }
    }

    // 释放句柄缓冲区
    gBS->FreePool(HandleBuffer);

    return EFI_SUCCESS;
}

获取BDF：

#include <Uefi.h>
#include <Protocol/PciIo.h>

EFI_STATUS GetPciBdf()
{
    EFI_STATUS Status;
    EFI_HANDLE *HandleBuffer;
    UINTN HandleCount;

    // 获取所有PCI设备的句柄
    Status = gBS->LocateHandleBuffer(
        ByProtocol,
        &gEfiPciIoProtocolGuid,
        NULL,
        &HandleCount,
        &HandleBuffer);

    if (EFI_ERROR(Status))
    {
        // 错误处理
        return Status;
    }

    UINTN Index;
    EFI_PCI_IO_PROTOCOL *PciIo;

    // 遍历所有PCI设备
    for (Index = 0; Index < HandleCount; Index++)
    {
        // 获取PCI IO Protocol
        Status = gBS->HandleProtocol(
            HandleBuffer[Index],
            &gEfiPciIoProtocolGuid,
            (VOID **)&PciIo);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        // 获取BDF信息
        UINTN SegmentNumber;
        UINTN BusNumber;
        UINTN DeviceNumber;
        UINTN FunctionNumber;

        Status = PciIo->GetLocation(
            PciIo,
            &SegmentNumber,
            &BusNumber,
            &DeviceNumber,
            &FunctionNumber);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        // 打印BDF信息
        Print(L"BDF: %04x:%02x:%02x.%x\n",
              SegmentNumber, BusNumber, DeviceNumber, FunctionNumber);
    }

    // 释放句柄缓冲区
    gBS->FreePool(HandleBuffer);

    return EFI_SUCCESS;
}

当然还有另外一种读网卡mac地址的方式：

#include <Uefi.h>
#include <Protocol/SimpleNetwork.h>

EFI_STATUS GetMacAddress()
{
    EFI_STATUS Status;
    EFI_HANDLE *HandleBuffer;
    UINTN HandleCount;

    // 获取所有网络设备的句柄
    Status = gBS->LocateHandleBuffer(
        ByProtocol,
        &gEfiSimpleNetworkProtocolGuid,
        NULL,
        &HandleCount,
        &HandleBuffer);

    if (EFI_ERROR(Status))
    {
        // 错误处理
        return Status;
    }

    UINTN Index;
    EFI_SIMPLE_NETWORK_PROTOCOL *Snp;

    // 遍历所有网络设备
    for (Index = 0; Index < HandleCount; Index++)
    {
        // 获取Simple Network Protocol
        Status = gBS->HandleProtocol(
            HandleBuffer[Index],
            &gEfiSimpleNetworkProtocolGuid,
            (VOID **)&Snp);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        EFI_MAC_ADDRESS MacAddress;

        // 获取MAC地址
        Status = Snp->GetMacAddress(Snp, &MacAddress);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        // 打印MAC地址
        Print(L"MAC Address: %02x:%02x:%02x:%02x:%02x:%02x\n",
              MacAddress.Addr[0], MacAddress.Addr[1], MacAddress.Addr[2],
              MacAddress.Addr[3], MacAddress.Addr[4], MacAddress.Addr[5]);
    }

    // 释放句柄缓冲区
    gBS->FreePool(HandleBuffer);

    return EFI_SUCCESS;
}

然后就是文件系统，比如读取USB设备中的文件：

#include <Uefi.h>
#include <Protocol/SimpleFileSystem.h>
#include <Protocol/LoadedImage.h>

EFI_STATUS ReadFileFromUSB()
{
    EFI_STATUS Status;
    EFI_HANDLE *HandleBuffer;
    UINTN HandleCount;

    // 获取所有文件系统协议的句柄
    Status = gBS->LocateHandleBuffer(
        ByProtocol,
        &gEfiSimpleFileSystemProtocolGuid,
        NULL,
        &HandleCount,
        &HandleBuffer);

    if (EFI_ERROR(Status))
    {
        // 错误处理
        return Status;
    }

    UINTN Index;
    EFI_SIMPLE_FILE_SYSTEM_PROTOCOL *FileSystem;
    EFI_FILE_PROTOCOL *RootDir;

    // 遍历所有文件系统
    for (Index = 0; Index < HandleCount; Index++)
    {
        // 获取文件系统协议
        Status = gBS->HandleProtocol(
            HandleBuffer[Index],
            &gEfiSimpleFileSystemProtocolGuid,
            (VOID **)&FileSystem);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        // 打开根目录
        Status = FileSystem->OpenVolume(FileSystem, &RootDir);
        if (EFI_ERROR(Status))
        {
            // 错误处理
            continue;
        }

        // 打开目标文件
        EFI_FILE_PROTOCOL *FileHandle;
        Status = RootDir->Open(
            RootDir,
            &FileHandle,
            L"filename.txt", // 文件名
            EFI_FILE_MODE_READ, // 打开模式
            0); // 属性

        if (EFI_ERROR(Status))
        {
            // 错误处理
            RootDir->Close(RootDir);
            continue;
        }

        // 读取文件内容
        UINTN BufferSize = 1024; // 缓冲区大小
        UINT8 Buffer[BufferSize];
        Status = FileHandle->Read(
            FileHandle,
            &BufferSize,
            Buffer);

        if (EFI_ERROR(Status))
        {
            // 错误处理
            FileHandle->Close(FileHandle);
            RootDir->Close(RootDir);
            continue;
        }

        // 在此处对读取的文件内容进行处理或打印

        // 关闭文件和根目录
        FileHandle->Close(FileHandle);
        RootDir->Close(RootDir);
    }

    // 释放句柄缓冲区
    gBS->FreePool(HandleBuffer);

    return EFI_SUCCESS;
}

再一个就是USB设备扫描：

#include <Uefi.h>
#include <Library/UefiLib.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Protocol/UsbIo.h>
#include <Protocol/Usb2HostController.h>

EFI_STATUS ScanUsbDevices()
{
    EFI_STATUS status;
    EFI_HANDLE *controllerHandles;
    UINTN controllerHandleCount;
    UINTN index;

    // 获取所有USB控制器的句柄
    status = gBS->LocateHandleBuffer(
        ByProtocol,
        &gEfiUsb2HcProtocolGuid,
        NULL,
        &controllerHandleCount,
        &controllerHandles
    );
    if (EFI_ERROR(status)) {
        Print(L"Failed to locate USB controllers. Error: %r\n", status);
        return status;
    }

    // 遍历所有USB控制器并扫描设备
    for (index = 0; index < controllerHandleCount; index++) {
        EFI_USB2_HC_PROTOCOL *usbHc;
        status = gBS->OpenProtocol(
            controllerHandles[index],
            &gEfiUsb2HcProtocolGuid,
            (VOID **)&usbHc,
            gImageHandle,
            NULL,
            EFI_OPEN_PROTOCOL_GET_PROTOCOL
        );
        if (EFI_ERROR(status)) {
            Print(L"Failed to open USB2_HC_PROTOCOL. Error: %r\n", status);
            continue;
        }

        // 扫描USB设备
        EFI_USB_DEVICE_DESCRIPTOR deviceDescriptor;
        EFI_USB_INTERFACE_DESCRIPTOR interfaceDescriptor;
        EFI_HANDLE deviceHandle = NULL;

        for (UINT8 port = 1; port <= usbHc->GetRootHubPortNumber(usbHc); port++) {
            status = usbHc->GetRootHubPortStatus(usbHc, port, &portStatus);
            if (EFI_ERROR(status)) {
                // 处理错误
                continue;
            }

            if (portStatus.PortStatus.PortConnected) {
                // 获取USB设备描述符
                status = usbHc->GetRootHubPortStatus(usbHc, port, &deviceDescriptor);
                if (EFI_ERROR(status)) {
                    // 处理错误
                    continue;
                }

                // 创建USB设备的句柄
                status = usbHc->CreateDevice(usbHc, &deviceHandle);
                if (EFI_ERROR(status)) {
                    // 处理错误
                    continue;
                }

                // 设置USB设备的配置
                status = usbHc->SetConfig(usbHc, deviceHandle, deviceConfig);
                if (EFI_ERROR(status)) {
                    // 处理错误
                    continue;
                }

                // 获取USB设备的接口描述符
                status = usbHc->GetInterfaceDescriptor(usbHc, deviceHandle, &interfaceDescriptor);
                if (EFI_ERROR(status)) {
                    // 处理错误
                    continue;
                }

                // 处理扫描到的USB设备

                // 关闭USB设备的句柄
                usbHc->DestroyDevice(usbHc, deviceHandle);
                deviceHandle = NULL;
            }
        }

上述代码可以说是一个模版，需要用到的时候直接参照就行，当然还有比如PCI设备的Capability部分怎么去获取等等

获取DEVICE PATH:

#include <Uefi.h>
#include <Library/UefiLib.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Library/DevicePathLib.h>

EFI_STATUS PrintDevicePath(EFI_HANDLE deviceHandle)
{
    EFI_STATUS status;
    EFI_DEVICE_PATH_PROTOCOL *devicePath;

    // 获取设备的Device Path
    status = gBS->HandleProtocol(
        deviceHandle,
        &gEfiDevicePathProtocolGuid,
        (VOID **)&devicePath
    );
    if (EFI_ERROR(status)) {
        Print(L"Failed to get Device Path Protocol. Error: %r\n", status);
        return status;
    }

    // 打印设备的Device Path
    CHAR16 *devicePathString = ConvertDevicePathToText(devicePath, FALSE, FALSE);
    if (devicePathString != NULL) {
        Print(L"Device Path: %s\n", devicePathString);
        FreePool(devicePathString);
    }

    return EFI_SUCCESS;
}

EFI_STATUS EFIAPI UefiMain(
    IN EFI_HANDLE ImageHandle,
    IN EFI_SYSTEM_TABLE *SystemTable
)
{
    EFI_STATUS status;
    EFI_HANDLE *deviceHandles;
    UINTN deviceHandleCount;
    UINTN index;

    // 获取所有设备的句柄
    status = gBS->LocateHandleBuffer(
        AllHandles,
        NULL,
        NULL,
        &deviceHandleCount,
        &deviceHandles
    );
    if (EFI_ERROR(status)) {
        Print(L"Failed to locate devices. Error: %r\n", status);
        return status;
    }

    // 遍历所有设备并打印Device Path
    for (index = 0; index < deviceHandleCount; index++) {
        status = PrintDevicePath(deviceHandles[index]);
        if (EFI_ERROR(status)) {
            // 处理错误
        }
    }

    // 释放内存
    FreePool(deviceHandles);

    return EFI_SUCCESS;
}

潇洒Anthony
关注
  • 2
    点赞
  • 踩
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
基于深度优先的PCIe设备遍历的递归实现
12-15
基于深度优先的PCIe设备遍历的递归实现，如果有看不懂或者疑惑的可以私信我与我探讨，本人也是UEFI小白，刚开始学，难免会有出错的地方，接受批评，乐意改正！
代码实例
03-22
NULL 博文链接：https://regular.iteye.com/blog/906130
代码示例
地推
10-24 1794
简单示例 package cn.tinyice.demo.thread; import java.util.Random; import java.util.concurrent.TimeUnit; /** * ThreadLocalDemo * * @author Tinyice */ public class ThreadLocalDemo { private static final ThreadLocal<Integer> threadLocal = new Threa...
jquery必须知道的一些常用特效方法及使用示例(整理)
12-12
实例： 代码如下: $(“button”).click(function(){ $(“#div1”).fadeIn(); $(“#div2”).fadeIn(“slow”); $(“#div3”).fadeIn(3000);}); 2.jQuery fadeOut() 方法用于淡出可见元素。 语法： $(selector)....
JS截取字符串常用方法整理及使用示例
12-09
本文主要介绍几种常用的字符串截取方法，包括`substring()`、`slice()`、`split()`以及`join()`，并提供使用示例来帮助理解它们的功能。 1. **substring()**： `substring()`方法用于从字符串中提取一部分，它接受...
Qt TCP相关的一些整理：服务端常见操作 socket 通信 network 示例代码
11-18
在Qt的服务端上，不单单会用到服务端本身的API，对连接上来的客户端，也需要进行数据交互，也要用到一些收发包相关的API操作； 1、相关的库和类 Qt下使用tcp协议，需要在工程的pro文件中添加： QT += network ...
最近在学习常用的Java开源项目，整理一下常用的代码.zip
最新发布
01-20
这些项目不仅展示了实际开发中的最佳实践，还提供了丰富的代码示例，帮助开发者理解和应用各种Java技术。以下是对"最近在学习常用的Java开源项目，整理一下常用的代码.zip"这个压缩包中可能包含的知识点的详细解释：...
示例代码
12-06
node实现restful
代码大全的例子
09-15
代码大全的例子代码大全的例子代码大全的例子代码大全的例子代码大全的例子代码大全的例子代码大全的例子代码大全的例子代码大全的例子
DataList代码示例
03-28
DataList基本的用法，其中包括编辑、删除、添加、更新、取消等基本操作的代码示例，另外还使用PagedataSource来实现了数据的分页。
Java8 Demo代码示例
02-28
Java8已经正式发布很久了 , 从各项新特性来讲 , Java8无疑是重大的版本升级 . Orcale公司于2014年3月18日发布Java8 , 新增了非常多的特性 . 但时至今日 , 大部分人在项目中依然使用着Java7及之前的新特性 . 所谓没有实践就不会深入详细的理解运用 , 故今天抛砖引玉 , 和大家一起探讨Java8在实际代码中的运用 .
js 操作select和option常用代码整理
12-11
JavaScript是Web开发中不可或缺的一部分，...这些代码示例涵盖了获取、添加、删除和修改选项，以及响应用户选择变化的基本操作。在实际项目中，开发者可以根据需求灵活运用这些方法，构建功能丰富的交互式下拉列表。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值