+VOID\r
+S3Verification (\r
+ VOID\r
+ )\r
+{\r
+#if defined (MDE_CPU_X64)\r
+ if (FeaturePcdGet (PcdSmmSmramRequire) && mS3Supported) {\r
+ DEBUG ((EFI_D_ERROR,\r
+ "%a: S3Resume2Pei doesn't support X64 PEI + SMM yet.\n", __FUNCTION__));\r
+ DEBUG ((EFI_D_ERROR,\r
+ "%a: Please disable S3 on the QEMU command line (see the README),\n",\r
+ __FUNCTION__));\r
+ DEBUG ((EFI_D_ERROR,\r
+ "%a: or build OVMF with \"OvmfPkgIa32X64.dsc\".\n", __FUNCTION__));\r
+ ASSERT (FALSE);\r
+ CpuDeadLoop ();\r
+ }\r
+#endif\r
+}\r
+\r
+\r
+/**\r
+ Fetch the boot CPU count and the possible CPU count from QEMU, and expose\r
+ them to UefiCpuPkg modules. Set the mMaxCpuCount variable.\r
+**/\r
+VOID\r
+MaxCpuCountInitialization (\r
+ VOID\r
+ )\r
+{\r
+ UINT16 BootCpuCount;\r
+ RETURN_STATUS PcdStatus;\r
+\r
+ //\r
+ // Try to fetch the boot CPU count.\r
+ //\r
+ QemuFwCfgSelectItem (QemuFwCfgItemSmpCpuCount);\r
+ BootCpuCount = QemuFwCfgRead16 ();\r
+ if (BootCpuCount == 0) {\r
+ //\r
+ // QEMU doesn't report the boot CPU count. (BootCpuCount == 0) will let\r
+ // MpInitLib count APs up to (PcdCpuMaxLogicalProcessorNumber - 1), or\r
+ // until PcdCpuApInitTimeOutInMicroSeconds elapses (whichever is reached\r
+ // first).\r
+ //\r
+ DEBUG ((DEBUG_WARN, "%a: boot CPU count unavailable\n", __FUNCTION__));\r
+ mMaxCpuCount = PcdGet32 (PcdCpuMaxLogicalProcessorNumber);\r
+ } else {\r
+ //\r
+ // We will expose BootCpuCount to MpInitLib. MpInitLib will count APs up to\r
+ // (BootCpuCount - 1) precisely, regardless of timeout.\r
+ //\r
+ // Now try to fetch the possible CPU count.\r
+ //\r
+ UINTN CpuHpBase;\r
+ UINT32 CmdData2;\r
+\r
+ CpuHpBase = ((mHostBridgeDevId == INTEL_Q35_MCH_DEVICE_ID) ?\r
+ ICH9_CPU_HOTPLUG_BASE : PIIX4_CPU_HOTPLUG_BASE);\r
+\r
+ //\r
+ // If only legacy mode is available in the CPU hotplug register block, or\r
+ // the register block is completely missing, then the writes below are\r
+ // no-ops.\r
+ //\r
+ // 1. Switch the hotplug register block to modern mode.\r
+ //\r
+ IoWrite32 (CpuHpBase + QEMU_CPUHP_W_CPU_SEL, 0);\r
+ //\r
+ // 2. Select a valid CPU for deterministic reading of\r
+ // QEMU_CPUHP_R_CMD_DATA2.\r
+ //\r
+ // CPU#0 is always valid; it is the always present and non-removable\r
+ // BSP.\r
+ //\r
+ IoWrite32 (CpuHpBase + QEMU_CPUHP_W_CPU_SEL, 0);\r
+ //\r
+ // 3. Send a command after which QEMU_CPUHP_R_CMD_DATA2 is specified to\r
+ // read as zero, and which does not invalidate the selector. (The\r
+ // selector may change, but it must not become invalid.)\r
+ //\r
+ // Send QEMU_CPUHP_CMD_GET_PENDING, as it will prove useful later.\r
+ //\r
+ IoWrite8 (CpuHpBase + QEMU_CPUHP_W_CMD, QEMU_CPUHP_CMD_GET_PENDING);\r
+ //\r
+ // 4. Read QEMU_CPUHP_R_CMD_DATA2.\r
+ //\r
+ // If the register block is entirely missing, then this is an unassigned\r
+ // IO read, returning all-bits-one.\r
+ //\r
+ // If only legacy mode is available, then bit#0 stands for CPU#0 in the\r
+ // "CPU present bitmap". CPU#0 is always present.\r
+ //\r
+ // Otherwise, QEMU_CPUHP_R_CMD_DATA2 is either still reserved (returning\r
+ // all-bits-zero), or it is specified to read as zero after the above\r
+ // steps. Both cases confirm modern mode.\r
+ //\r
+ CmdData2 = IoRead32 (CpuHpBase + QEMU_CPUHP_R_CMD_DATA2);\r
+ DEBUG ((DEBUG_VERBOSE, "%a: CmdData2=0x%x\n", __FUNCTION__, CmdData2));\r
+ if (CmdData2 != 0) {\r
+ //\r
+ // QEMU doesn't support the modern CPU hotplug interface. Assume that the\r
+ // possible CPU count equals the boot CPU count (precluding hotplug).\r
+ //\r
+ DEBUG ((DEBUG_WARN, "%a: modern CPU hotplug interface unavailable\n",\r
+ __FUNCTION__));\r
+ mMaxCpuCount = BootCpuCount;\r
+ } else {\r
+ //\r
+ // Grab the possible CPU count from the modern CPU hotplug interface.\r
+ //\r
+ UINT32 Present, Possible, Selected;\r
+\r
+ Present = 0;\r
+ Possible = 0;\r
+\r
+ //\r
+ // We've sent QEMU_CPUHP_CMD_GET_PENDING last; this ensures\r
+ // QEMU_CPUHP_RW_CMD_DATA can now be read usefully. However,\r
+ // QEMU_CPUHP_CMD_GET_PENDING may have selected a CPU with actual pending\r
+ // hotplug events; therefore, select CPU#0 forcibly.\r
+ //\r
+ IoWrite32 (CpuHpBase + QEMU_CPUHP_W_CPU_SEL, Possible);\r
+\r
+ do {\r
+ UINT8 CpuStatus;\r
+\r
+ //\r
+ // Read the status of the currently selected CPU. This will help with a\r
+ // sanity check against "BootCpuCount".\r
+ //\r
+ CpuStatus = IoRead8 (CpuHpBase + QEMU_CPUHP_R_CPU_STAT);\r
+ if ((CpuStatus & QEMU_CPUHP_STAT_ENABLED) != 0) {\r
+ ++Present;\r
+ }\r
+ //\r
+ // Attempt to select the next CPU.\r
+ //\r
+ ++Possible;\r
+ IoWrite32 (CpuHpBase + QEMU_CPUHP_W_CPU_SEL, Possible);\r
+ //\r
+ // If the selection is successful, then the following read will return\r
+ // the selector (which we know is positive at this point). Otherwise,\r
+ // the read will return 0.\r
+ //\r
+ Selected = IoRead32 (CpuHpBase + QEMU_CPUHP_RW_CMD_DATA);\r
+ ASSERT (Selected == Possible || Selected == 0);\r
+ } while (Selected > 0);\r
+\r
+ //\r
+ // Sanity check: fw_cfg and the modern CPU hotplug interface should\r
+ // return the same boot CPU count.\r
+ //\r
+ if (BootCpuCount != Present) {\r
+ DEBUG ((DEBUG_WARN, "%a: QEMU v2.7 reset bug: BootCpuCount=%d "\r
+ "Present=%u\n", __FUNCTION__, BootCpuCount, Present));\r
+ //\r
+ // The handling of QemuFwCfgItemSmpCpuCount, across CPU hotplug plus\r
+ // platform reset (including S3), was corrected in QEMU commit\r
+ // e3cadac073a9 ("pc: fix FW_CFG_NB_CPUS to account for -device added\r
+ // CPUs", 2016-11-16), part of release v2.8.0.\r
+ //\r
+ BootCpuCount = (UINT16)Present;\r
+ }\r
+\r
+ mMaxCpuCount = Possible;\r
+ }\r
+ }\r
+\r
+ DEBUG ((DEBUG_INFO, "%a: BootCpuCount=%d mMaxCpuCount=%u\n", __FUNCTION__,\r
+ BootCpuCount, mMaxCpuCount));\r
+ ASSERT (BootCpuCount <= mMaxCpuCount);\r
+\r
+ PcdStatus = PcdSet32S (PcdCpuBootLogicalProcessorNumber, BootCpuCount);\r
+ ASSERT_RETURN_ERROR (PcdStatus);\r
+ PcdStatus = PcdSet32S (PcdCpuMaxLogicalProcessorNumber, mMaxCpuCount);\r
+ ASSERT_RETURN_ERROR (PcdStatus);\r
+}\r
+\r
+\r