X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdeModulePkg%2FBus%2FPci%2FNvmExpressDxe%2FNvmExpressHci.c;h=a173504cdf4d3a46004df0c62a94816f8e705ed2;hp=157e10127aa76675ef38b981253ed5d0849f3093;hb=05bf4747dd8e412d10fccbe35346aa2597b4167b;hpb=eb290d0257814701eefd93ee71d46efb6dcc320b diff --git a/MdeModulePkg/Bus/Pci/NvmExpressDxe/NvmExpressHci.c b/MdeModulePkg/Bus/Pci/NvmExpressDxe/NvmExpressHci.c index 157e10127a..a173504cdf 100644 --- a/MdeModulePkg/Bus/Pci/NvmExpressDxe/NvmExpressHci.c +++ b/MdeModulePkg/Bus/Pci/NvmExpressDxe/NvmExpressHci.c @@ -2,7 +2,7 @@ NvmExpressDxe driver is used to manage non-volatile memory subsystem which follows NVM Express specification. - Copyright (c) 2013, Intel Corporation. All rights reserved.
+ Copyright (c) 2013 - 2016, Intel Corporation. All rights reserved.
This program and the accompanying materials are licensed and made available under the terms and conditions of the BSD License which accompanies this distribution. The full text of the license may be found at @@ -33,21 +33,23 @@ ReadNvmeControllerCapabilities ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT64 Data; PciIo = Private->PciIo; Status = PciIo->Mem.Read ( PciIo, - EfiPciIoWidthUint64, + EfiPciIoWidthUint32, NVME_BAR, NVME_CAP_OFFSET, - 1, - Cap + 2, + &Data ); if (EFI_ERROR(Status)) { return Status; } + WriteUnaligned64 ((UINT64*)Cap, Data); return EFI_SUCCESS; } @@ -69,6 +71,7 @@ ReadNvmeControllerConfiguration ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT32 Data; PciIo = Private->PciIo; Status = PciIo->Mem.Read ( @@ -77,13 +80,14 @@ ReadNvmeControllerConfiguration ( NVME_BAR, NVME_CC_OFFSET, 1, - Cc + &Data ); if (EFI_ERROR(Status)) { return Status; } + WriteUnaligned32 ((UINT32*)Cc, Data); return EFI_SUCCESS; } @@ -105,15 +109,17 @@ WriteNvmeControllerConfiguration ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT32 Data; PciIo = Private->PciIo; + Data = ReadUnaligned32 ((UINT32*)Cc); Status = PciIo->Mem.Write ( PciIo, EfiPciIoWidthUint32, NVME_BAR, NVME_CC_OFFSET, 1, - Cc + &Data ); if (EFI_ERROR(Status)) { @@ -149,6 +155,7 @@ ReadNvmeControllerStatus ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT32 Data; PciIo = Private->PciIo; Status = PciIo->Mem.Read ( @@ -157,13 +164,14 @@ ReadNvmeControllerStatus ( NVME_BAR, NVME_CSTS_OFFSET, 1, - Csts + &Data ); if (EFI_ERROR(Status)) { return Status; } + WriteUnaligned32 ((UINT32*)Csts, Data); return EFI_SUCCESS; } @@ -185,6 +193,7 @@ ReadNvmeAdminQueueAttributes ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT32 Data; PciIo = Private->PciIo; Status = PciIo->Mem.Read ( @@ -193,13 +202,14 @@ ReadNvmeAdminQueueAttributes ( NVME_BAR, NVME_AQA_OFFSET, 1, - Aqa + &Data ); if (EFI_ERROR(Status)) { return Status; } + WriteUnaligned32 ((UINT32*)Aqa, Data); return EFI_SUCCESS; } @@ -221,15 +231,17 @@ WriteNvmeAdminQueueAttributes ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT32 Data; PciIo = Private->PciIo; + Data = ReadUnaligned32 ((UINT32*)Aqa); Status = PciIo->Mem.Write ( PciIo, EfiPciIoWidthUint32, NVME_BAR, NVME_AQA_OFFSET, 1, - Aqa + &Data ); if (EFI_ERROR(Status)) { @@ -260,21 +272,23 @@ ReadNvmeAdminSubmissionQueueBaseAddress ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT64 Data; PciIo = Private->PciIo; Status = PciIo->Mem.Read ( PciIo, - EfiPciIoWidthUint64, + EfiPciIoWidthUint32, NVME_BAR, NVME_ASQ_OFFSET, - 1, - Asq + 2, + &Data ); if (EFI_ERROR(Status)) { return Status; } + WriteUnaligned64 ((UINT64*)Asq, Data); return EFI_SUCCESS; } @@ -296,22 +310,25 @@ WriteNvmeAdminSubmissionQueueBaseAddress ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT64 Data; PciIo = Private->PciIo; + Data = ReadUnaligned64 ((UINT64*)Asq); + Status = PciIo->Mem.Write ( PciIo, - EfiPciIoWidthUint64, + EfiPciIoWidthUint32, NVME_BAR, NVME_ASQ_OFFSET, - 1, - Asq + 2, + &Data ); if (EFI_ERROR(Status)) { return Status; } - DEBUG ((EFI_D_INFO, "Asq.Asqb: %lx\n", Asq->Asqb)); + DEBUG ((EFI_D_INFO, "Asq: %lx\n", *Asq)); return EFI_SUCCESS; } @@ -334,21 +351,24 @@ ReadNvmeAdminCompletionQueueBaseAddress ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT64 Data; PciIo = Private->PciIo; + Status = PciIo->Mem.Read ( PciIo, - EfiPciIoWidthUint64, + EfiPciIoWidthUint32, NVME_BAR, NVME_ACQ_OFFSET, - 1, - Acq + 2, + &Data ); if (EFI_ERROR(Status)) { return Status; } + WriteUnaligned64 ((UINT64*)Acq, Data); return EFI_SUCCESS; } @@ -370,22 +390,25 @@ WriteNvmeAdminCompletionQueueBaseAddress ( { EFI_PCI_IO_PROTOCOL *PciIo; EFI_STATUS Status; + UINT64 Data; PciIo = Private->PciIo; + Data = ReadUnaligned64 ((UINT64*)Acq); + Status = PciIo->Mem.Write ( PciIo, - EfiPciIoWidthUint64, + EfiPciIoWidthUint32, NVME_BAR, NVME_ACQ_OFFSET, - 1, - Acq + 2, + &Data ); if (EFI_ERROR(Status)) { return Status; } - DEBUG ((EFI_D_INFO, "Acq.Acqb: %lxh\n", Acq->Acqb)); + DEBUG ((EFI_D_INFO, "Acq: %lxh\n", *Acq)); return EFI_SUCCESS; } @@ -407,6 +430,8 @@ NvmeDisableController ( NVME_CC Cc; NVME_CSTS Csts; EFI_STATUS Status; + UINT32 Index; + UINT8 Timeout; // // Read Controller Configuration Register. @@ -427,19 +452,35 @@ NvmeDisableController ( return Status; } - gBS->Stall(10000); - // - // Check if the controller is reset + // Cap.To specifies max delay time in 500ms increments for Csts.Rdy to transition from 1 to 0 after + // Cc.Enable transition from 1 to 0. Loop produces a 1 millisecond delay per itteration, up to 500 * Cap.To. // - Status = ReadNvmeControllerStatus (Private, &Csts); + if (Private->Cap.To == 0) { + Timeout = 1; + } else { + Timeout = Private->Cap.To; + } - if (EFI_ERROR(Status)) { - return Status; + for(Index = (Timeout * 500); Index != 0; --Index) { + gBS->Stall(1000); + + // + // Check if the controller is initialized + // + Status = ReadNvmeControllerStatus (Private, &Csts); + + if (EFI_ERROR(Status)) { + return Status; + } + + if (Csts.Rdy == 0) { + break; + } } - if (Csts.Rdy != 0) { - return EFI_DEVICE_ERROR; + if (Index == 0) { + Status = EFI_DEVICE_ERROR; } DEBUG ((EFI_D_INFO, "NVMe controller is disabled with status [%r].\n", Status)); @@ -468,14 +509,15 @@ NvmeEnableController ( UINT8 Timeout; // - // Enable the controller + // Enable the controller. + // CC.AMS, CC.MPS and CC.CSS are all set to 0. // ZeroMem (&Cc, sizeof (NVME_CC)); Cc.En = 1; Cc.Iosqes = 6; Cc.Iocqes = 4; - Status = WriteNvmeControllerConfiguration (Private, &Cc); + Status = WriteNvmeControllerConfiguration (Private, &Cc); if (EFI_ERROR(Status)) { return Status; } @@ -531,17 +573,16 @@ NvmeIdentifyController ( IN VOID *Buffer ) { - NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; - NVM_EXPRESS_COMMAND Command; - NVM_EXPRESS_RESPONSE Response; + EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; + EFI_NVM_EXPRESS_COMMAND Command; + EFI_NVM_EXPRESS_COMPLETION Completion; EFI_STATUS Status; - ZeroMem (&CommandPacket, sizeof(NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); - ZeroMem (&Command, sizeof(NVM_EXPRESS_COMMAND)); - ZeroMem (&Response, sizeof(NVM_EXPRESS_RESPONSE)); + ZeroMem (&CommandPacket, sizeof(EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); + ZeroMem (&Command, sizeof(EFI_NVM_EXPRESS_COMMAND)); + ZeroMem (&Completion, sizeof(EFI_NVM_EXPRESS_COMPLETION)); - Command.Cdw0.Opcode = NVME_ADMIN_IDENTIFY_OPC; - Command.Cdw0.Cid = Private->Cid[0]++; + Command.Cdw0.Opcode = NVME_ADMIN_IDENTIFY_CMD; // // According to Nvm Express 1.1 spec Figure 38, When not used, the field shall be cleared to 0h. // For the Identify command, the Namespace Identifier is only used for the Namespace data structure. @@ -549,11 +590,11 @@ NvmeIdentifyController ( Command.Nsid = 0; CommandPacket.NvmeCmd = &Command; - CommandPacket.NvmeResponse = &Response; + CommandPacket.NvmeCompletion = &Completion; CommandPacket.TransferBuffer = Buffer; CommandPacket.TransferLength = sizeof (NVME_ADMIN_CONTROLLER_DATA); CommandPacket.CommandTimeout = NVME_GENERIC_TIMEOUT; - CommandPacket.QueueId = NVME_ADMIN_QUEUE; + CommandPacket.QueueType = NVME_ADMIN_QUEUE; // // Set bit 0 (Cns bit) to 1 to identify a controller // @@ -563,7 +604,6 @@ NvmeIdentifyController ( Status = Private->Passthru.PassThru ( &Private->Passthru, NVME_CONTROLLER_ID, - 0, &CommandPacket, NULL ); @@ -589,25 +629,24 @@ NvmeIdentifyNamespace ( IN VOID *Buffer ) { - NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; - NVM_EXPRESS_COMMAND Command; - NVM_EXPRESS_RESPONSE Response; + EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; + EFI_NVM_EXPRESS_COMMAND Command; + EFI_NVM_EXPRESS_COMPLETION Completion; EFI_STATUS Status; - ZeroMem (&CommandPacket, sizeof(NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); - ZeroMem (&Command, sizeof(NVM_EXPRESS_COMMAND)); - ZeroMem (&Response, sizeof(NVM_EXPRESS_RESPONSE)); + ZeroMem (&CommandPacket, sizeof(EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); + ZeroMem (&Command, sizeof(EFI_NVM_EXPRESS_COMMAND)); + ZeroMem (&Completion, sizeof(EFI_NVM_EXPRESS_COMPLETION)); - CommandPacket.NvmeCmd = &Command; - CommandPacket.NvmeResponse = &Response; + CommandPacket.NvmeCmd = &Command; + CommandPacket.NvmeCompletion = &Completion; - Command.Cdw0.Opcode = NVME_ADMIN_IDENTIFY_OPC; - Command.Cdw0.Cid = Private->Cid[0]++; + Command.Cdw0.Opcode = NVME_ADMIN_IDENTIFY_CMD; Command.Nsid = NamespaceId; CommandPacket.TransferBuffer = Buffer; CommandPacket.TransferLength = sizeof (NVME_ADMIN_NAMESPACE_DATA); CommandPacket.CommandTimeout = NVME_GENERIC_TIMEOUT; - CommandPacket.QueueId = NVME_ADMIN_QUEUE; + CommandPacket.QueueType = NVME_ADMIN_QUEUE; // // Set bit 0 (Cns bit) to 1 to identify a namespace // @@ -617,7 +656,6 @@ NvmeIdentifyNamespace ( Status = Private->Passthru.PassThru ( &Private->Passthru, NamespaceId, - 0, &CommandPacket, NULL ); @@ -639,40 +677,57 @@ NvmeCreateIoCompletionQueue ( IN NVME_CONTROLLER_PRIVATE_DATA *Private ) { - NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; - NVM_EXPRESS_COMMAND Command; - NVM_EXPRESS_RESPONSE Response; + EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; + EFI_NVM_EXPRESS_COMMAND Command; + EFI_NVM_EXPRESS_COMPLETION Completion; EFI_STATUS Status; NVME_ADMIN_CRIOCQ CrIoCq; + UINT32 Index; + UINT16 QueueSize; + + Status = EFI_SUCCESS; + + for (Index = 1; Index < NVME_MAX_QUEUES; Index++) { + ZeroMem (&CommandPacket, sizeof(EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); + ZeroMem (&Command, sizeof(EFI_NVM_EXPRESS_COMMAND)); + ZeroMem (&Completion, sizeof(EFI_NVM_EXPRESS_COMPLETION)); + ZeroMem (&CrIoCq, sizeof(NVME_ADMIN_CRIOCQ)); + + CommandPacket.NvmeCmd = &Command; + CommandPacket.NvmeCompletion = &Completion; + + Command.Cdw0.Opcode = NVME_ADMIN_CRIOCQ_CMD; + CommandPacket.TransferBuffer = Private->CqBufferPciAddr[Index]; + CommandPacket.TransferLength = EFI_PAGE_SIZE; + CommandPacket.CommandTimeout = NVME_GENERIC_TIMEOUT; + CommandPacket.QueueType = NVME_ADMIN_QUEUE; + + if (Index == 1) { + QueueSize = NVME_CCQ_SIZE; + } else { + if (Private->Cap.Mqes > NVME_ASYNC_CCQ_SIZE) { + QueueSize = NVME_ASYNC_CCQ_SIZE; + } else { + QueueSize = Private->Cap.Mqes; + } + } - ZeroMem (&CommandPacket, sizeof(NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); - ZeroMem (&Command, sizeof(NVM_EXPRESS_COMMAND)); - ZeroMem (&Response, sizeof(NVM_EXPRESS_RESPONSE)); - ZeroMem (&CrIoCq, sizeof(NVME_ADMIN_CRIOCQ)); - - CommandPacket.NvmeCmd = &Command; - CommandPacket.NvmeResponse = &Response; - - Command.Cdw0.Opcode = NVME_ADMIN_CRIOCQ_OPC; - Command.Cdw0.Cid = Private->Cid[0]++; - CommandPacket.TransferBuffer = Private->CqBufferPciAddr[1]; - CommandPacket.TransferLength = EFI_PAGE_SIZE; - CommandPacket.CommandTimeout = NVME_GENERIC_TIMEOUT; - CommandPacket.QueueId = NVME_ADMIN_QUEUE; - - CrIoCq.Qid = NVME_IO_QUEUE; - CrIoCq.Qsize = NVME_CCQ_SIZE; - CrIoCq.Pc = 1; - CopyMem (&CommandPacket.NvmeCmd->Cdw10, &CrIoCq, sizeof (NVME_ADMIN_CRIOCQ)); - CommandPacket.NvmeCmd->Flags = CDW10_VALID | CDW11_VALID; - - Status = Private->Passthru.PassThru ( - &Private->Passthru, - 0, - 0, - &CommandPacket, - NULL - ); + CrIoCq.Qid = Index; + CrIoCq.Qsize = QueueSize; + CrIoCq.Pc = 1; + CopyMem (&CommandPacket.NvmeCmd->Cdw10, &CrIoCq, sizeof (NVME_ADMIN_CRIOCQ)); + CommandPacket.NvmeCmd->Flags = CDW10_VALID | CDW11_VALID; + + Status = Private->Passthru.PassThru ( + &Private->Passthru, + 0, + &CommandPacket, + NULL + ); + if (EFI_ERROR (Status)) { + break; + } + } return Status; } @@ -691,42 +746,59 @@ NvmeCreateIoSubmissionQueue ( IN NVME_CONTROLLER_PRIVATE_DATA *Private ) { - NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; - NVM_EXPRESS_COMMAND Command; - NVM_EXPRESS_RESPONSE Response; + EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET CommandPacket; + EFI_NVM_EXPRESS_COMMAND Command; + EFI_NVM_EXPRESS_COMPLETION Completion; EFI_STATUS Status; NVME_ADMIN_CRIOSQ CrIoSq; + UINT32 Index; + UINT16 QueueSize; + + Status = EFI_SUCCESS; + + for (Index = 1; Index < NVME_MAX_QUEUES; Index++) { + ZeroMem (&CommandPacket, sizeof(EFI_NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); + ZeroMem (&Command, sizeof(EFI_NVM_EXPRESS_COMMAND)); + ZeroMem (&Completion, sizeof(EFI_NVM_EXPRESS_COMPLETION)); + ZeroMem (&CrIoSq, sizeof(NVME_ADMIN_CRIOSQ)); + + CommandPacket.NvmeCmd = &Command; + CommandPacket.NvmeCompletion = &Completion; + + Command.Cdw0.Opcode = NVME_ADMIN_CRIOSQ_CMD; + CommandPacket.TransferBuffer = Private->SqBufferPciAddr[Index]; + CommandPacket.TransferLength = EFI_PAGE_SIZE; + CommandPacket.CommandTimeout = NVME_GENERIC_TIMEOUT; + CommandPacket.QueueType = NVME_ADMIN_QUEUE; + + if (Index == 1) { + QueueSize = NVME_CSQ_SIZE; + } else { + if (Private->Cap.Mqes > NVME_ASYNC_CSQ_SIZE) { + QueueSize = NVME_ASYNC_CSQ_SIZE; + } else { + QueueSize = Private->Cap.Mqes; + } + } - ZeroMem (&CommandPacket, sizeof(NVM_EXPRESS_PASS_THRU_COMMAND_PACKET)); - ZeroMem (&Command, sizeof(NVM_EXPRESS_COMMAND)); - ZeroMem (&Response, sizeof(NVM_EXPRESS_RESPONSE)); - ZeroMem (&CrIoSq, sizeof(NVME_ADMIN_CRIOSQ)); - - CommandPacket.NvmeCmd = &Command; - CommandPacket.NvmeResponse = &Response; - - Command.Cdw0.Opcode = NVME_ADMIN_CRIOSQ_OPC; - Command.Cdw0.Cid = Private->Cid[0]++; - CommandPacket.TransferBuffer = Private->SqBufferPciAddr[1]; - CommandPacket.TransferLength = EFI_PAGE_SIZE; - CommandPacket.CommandTimeout = NVME_GENERIC_TIMEOUT; - CommandPacket.QueueId = NVME_ADMIN_QUEUE; - - CrIoSq.Qid = NVME_IO_QUEUE; - CrIoSq.Qsize = NVME_CSQ_SIZE; - CrIoSq.Pc = 1; - CrIoSq.Cqid = NVME_IO_QUEUE; - CrIoSq.Qprio = 0; - CopyMem (&CommandPacket.NvmeCmd->Cdw10, &CrIoSq, sizeof (NVME_ADMIN_CRIOSQ)); - CommandPacket.NvmeCmd->Flags = CDW10_VALID | CDW11_VALID; - - Status = Private->Passthru.PassThru ( - &Private->Passthru, - 0, - 0, - &CommandPacket, - NULL - ); + CrIoSq.Qid = Index; + CrIoSq.Qsize = QueueSize; + CrIoSq.Pc = 1; + CrIoSq.Cqid = Index; + CrIoSq.Qprio = 0; + CopyMem (&CommandPacket.NvmeCmd->Cdw10, &CrIoSq, sizeof (NVME_ADMIN_CRIOSQ)); + CommandPacket.NvmeCmd->Flags = CDW10_VALID | CDW11_VALID; + + Status = Private->Passthru.PassThru ( + &Private->Passthru, + 0, + &CommandPacket, + NULL + ); + if (EFI_ERROR (Status)) { + break; + } + } return Status; } @@ -751,7 +823,8 @@ NvmeControllerInit ( NVME_AQA Aqa; NVME_ASQ Asq; NVME_ACQ Acq; - + UINT8 Sn[21]; + UINT8 Mn[41]; // // Save original PCI attributes and enable this controller. // @@ -775,7 +848,7 @@ NvmeControllerInit ( ); if (!EFI_ERROR (Status)) { - Supports &= EFI_PCI_DEVICE_ENABLE; + Supports &= (UINT64)EFI_PCI_DEVICE_ENABLE; Status = PciIo->Attributes ( PciIo, EfiPciIoAttributeOperationEnable, @@ -809,6 +882,17 @@ NvmeControllerInit ( Private->Cid[0] = 0; Private->Cid[1] = 0; + Private->Cid[2] = 0; + Private->Pt[0] = 0; + Private->Pt[1] = 0; + Private->Pt[2] = 0; + Private->SqTdbl[0].Sqt = 0; + Private->SqTdbl[1].Sqt = 0; + Private->SqTdbl[2].Sqt = 0; + Private->CqHdbl[0].Cqh = 0; + Private->CqHdbl[1].Cqh = 0; + Private->CqHdbl[2].Cqh = 0; + Private->AsyncSqHead = 0; Status = NvmeDisableController (Private); @@ -819,24 +903,25 @@ NvmeControllerInit ( // // set number of entries admin submission & completion queues. // - Aqa.Asqs = NVME_ASQ_SIZE; - Aqa.Acqs = NVME_ACQ_SIZE; + Aqa.Asqs = NVME_ASQ_SIZE; + Aqa.Rsvd1 = 0; + Aqa.Acqs = NVME_ACQ_SIZE; + Aqa.Rsvd2 = 0; // // Address of admin submission queue. // - Asq.Rsvd1 = 0; - Asq.Asqb = (UINT64)(UINTN)(Private->BufferPciAddr) >> 12; + Asq = (UINT64)(UINTN)(Private->BufferPciAddr) & ~0xFFF; // // Address of admin completion queue. // - Acq.Rsvd1 = 0; - Acq.Acqb = (UINT64)(UINTN)(Private->BufferPciAddr + EFI_PAGE_SIZE) >> 12; + Acq = (UINT64)(UINTN)(Private->BufferPciAddr + EFI_PAGE_SIZE) & ~0xFFF; // // Address of I/O submission & completion queue. // + ZeroMem (Private->Buffer, EFI_PAGES_TO_SIZE (6)); Private->SqBuffer[0] = (NVME_SQ *)(UINTN)(Private->Buffer); Private->SqBufferPciAddr[0] = (NVME_SQ *)(UINTN)(Private->BufferPciAddr); Private->CqBuffer[0] = (NVME_CQ *)(UINTN)(Private->Buffer + 1 * EFI_PAGE_SIZE); @@ -845,14 +930,20 @@ NvmeControllerInit ( Private->SqBufferPciAddr[1] = (NVME_SQ *)(UINTN)(Private->BufferPciAddr + 2 * EFI_PAGE_SIZE); Private->CqBuffer[1] = (NVME_CQ *)(UINTN)(Private->Buffer + 3 * EFI_PAGE_SIZE); Private->CqBufferPciAddr[1] = (NVME_CQ *)(UINTN)(Private->BufferPciAddr + 3 * EFI_PAGE_SIZE); + Private->SqBuffer[2] = (NVME_SQ *)(UINTN)(Private->Buffer + 4 * EFI_PAGE_SIZE); + Private->SqBufferPciAddr[2] = (NVME_SQ *)(UINTN)(Private->BufferPciAddr + 4 * EFI_PAGE_SIZE); + Private->CqBuffer[2] = (NVME_CQ *)(UINTN)(Private->Buffer + 5 * EFI_PAGE_SIZE); + Private->CqBufferPciAddr[2] = (NVME_CQ *)(UINTN)(Private->BufferPciAddr + 5 * EFI_PAGE_SIZE); DEBUG ((EFI_D_INFO, "Private->Buffer = [%016X]\n", (UINT64)(UINTN)Private->Buffer)); - DEBUG ((EFI_D_INFO, "Admin Submission Queue size (Aqa.Asqs) = [%08X]\n", Aqa.Asqs)); - DEBUG ((EFI_D_INFO, "Admin Completion Queue size (Aqa.Acqs) = [%08X]\n", Aqa.Acqs)); - DEBUG ((EFI_D_INFO, "Admin Submission Queue (SqBuffer[0]) = [%016X]\n", Private->SqBuffer[0])); - DEBUG ((EFI_D_INFO, "Admin Completion Queue (CqBuffer[0]) = [%016X]\n", Private->CqBuffer[0])); - DEBUG ((EFI_D_INFO, "I/O Submission Queue (SqBuffer[1]) = [%016X]\n", Private->SqBuffer[1])); - DEBUG ((EFI_D_INFO, "I/O Completion Queue (CqBuffer[1]) = [%016X]\n", Private->CqBuffer[1])); + DEBUG ((EFI_D_INFO, "Admin Submission Queue size (Aqa.Asqs) = [%08X]\n", Aqa.Asqs)); + DEBUG ((EFI_D_INFO, "Admin Completion Queue size (Aqa.Acqs) = [%08X]\n", Aqa.Acqs)); + DEBUG ((EFI_D_INFO, "Admin Submission Queue (SqBuffer[0]) = [%016X]\n", Private->SqBuffer[0])); + DEBUG ((EFI_D_INFO, "Admin Completion Queue (CqBuffer[0]) = [%016X]\n", Private->CqBuffer[0])); + DEBUG ((EFI_D_INFO, "Sync I/O Submission Queue (SqBuffer[1]) = [%016X]\n", Private->SqBuffer[1])); + DEBUG ((EFI_D_INFO, "Sync I/O Completion Queue (CqBuffer[1]) = [%016X]\n", Private->CqBuffer[1])); + DEBUG ((EFI_D_INFO, "Async I/O Submission Queue (SqBuffer[2]) = [%016X]\n", Private->SqBuffer[2])); + DEBUG ((EFI_D_INFO, "Async I/O Completion Queue (CqBuffer[2]) = [%016X]\n", Private->CqBuffer[2])); // // Program admin queue attributes. @@ -887,40 +978,65 @@ NvmeControllerInit ( } // - // Create one I/O completion queue. + // Allocate buffer for Identify Controller data // - Status = NvmeCreateIoCompletionQueue (Private); - if (EFI_ERROR(Status)) { - return Status; + if (Private->ControllerData == NULL) { + Private->ControllerData = (NVME_ADMIN_CONTROLLER_DATA *)AllocateZeroPool (sizeof(NVME_ADMIN_CONTROLLER_DATA)); + + if (Private->ControllerData == NULL) { + return EFI_OUT_OF_RESOURCES; + } } // - // Create one I/O Submission queue. + // Get current Identify Controller Data // - Status = NvmeCreateIoSubmissionQueue (Private); + Status = NvmeIdentifyController (Private, Private->ControllerData); + if (EFI_ERROR(Status)) { - return Status; + FreePool(Private->ControllerData); + Private->ControllerData = NULL; + return EFI_NOT_FOUND; } // - // Allocate buffer for Identify Controller data + // Dump NvmExpress Identify Controller Data // - Private->ControllerData = (NVME_ADMIN_CONTROLLER_DATA *)AllocateZeroPool (sizeof(NVME_ADMIN_CONTROLLER_DATA)); + CopyMem (Sn, Private->ControllerData->Sn, sizeof (Private->ControllerData->Sn)); + Sn[20] = 0; + CopyMem (Mn, Private->ControllerData->Mn, sizeof (Private->ControllerData->Mn)); + Mn[40] = 0; + DEBUG ((EFI_D_INFO, " == NVME IDENTIFY CONTROLLER DATA ==\n")); + DEBUG ((EFI_D_INFO, " PCI VID : 0x%x\n", Private->ControllerData->Vid)); + DEBUG ((EFI_D_INFO, " PCI SSVID : 0x%x\n", Private->ControllerData->Ssvid)); + DEBUG ((EFI_D_INFO, " SN : %a\n", Sn)); + DEBUG ((EFI_D_INFO, " MN : %a\n", Mn)); + DEBUG ((EFI_D_INFO, " FR : 0x%x\n", *((UINT64*)Private->ControllerData->Fr))); + DEBUG ((EFI_D_INFO, " RAB : 0x%x\n", Private->ControllerData->Rab)); + DEBUG ((EFI_D_INFO, " IEEE : 0x%x\n", *(UINT32*)Private->ControllerData->Ieee_oui)); + DEBUG ((EFI_D_INFO, " AERL : 0x%x\n", Private->ControllerData->Aerl)); + DEBUG ((EFI_D_INFO, " SQES : 0x%x\n", Private->ControllerData->Sqes)); + DEBUG ((EFI_D_INFO, " CQES : 0x%x\n", Private->ControllerData->Cqes)); + DEBUG ((EFI_D_INFO, " NN : 0x%x\n", Private->ControllerData->Nn)); - if (Private->ControllerData == NULL) { - return EFI_OUT_OF_RESOURCES; + // + // Create two I/O completion queues. + // One for blocking I/O, one for non-blocking I/O. + // + Status = NvmeCreateIoCompletionQueue (Private); + if (EFI_ERROR(Status)) { + return Status; } // - // Get current Identify Controller Data + // Create two I/O Submission queues. + // One for blocking I/O, one for non-blocking I/O. // - Status = NvmeIdentifyController (Private, Private->ControllerData); - + Status = NvmeCreateIoSubmissionQueue (Private); if (EFI_ERROR(Status)) { - FreePool(Private->ControllerData); - Private->ControllerData = NULL; - return EFI_NOT_FOUND; + return Status; } + return Status; }