Index: llvm/include/llvm/DebugInfo/MSF/MSFCommon.h =================================================================== --- llvm/include/llvm/DebugInfo/MSF/MSFCommon.h +++ llvm/include/llvm/DebugInfo/MSF/MSFCommon.h @@ -108,14 +108,27 @@ return L.SB->BlockSize; } -inline uint32_t getNumFpmIntervals(const MSFLayout &L, - bool IncludeUnusedFpmData = false) { - if (IncludeUnusedFpmData) - return divideCeil(L.SB->NumBlocks, L.SB->BlockSize); +inline uint32_t getNumFpmIntervals(uint32_t BlockSize, uint32_t NumBlocks, + bool IncludeUnusedFpmData, bool AltFpm) { + if (IncludeUnusedFpmData) { + // This calculation determines how many times a number of the form + // BlockSize * k + N appears in the range [0, NumBlocks). We only need to + // do this when unused data is included, since the number of blocks dwarfs + // the number of fpm blocks. + int N = AltFpm ? 2 : 1; + return divideCeil(NumBlocks - N, BlockSize); + } // We want the minimum number of intervals required, where each interval can // represent BlockSize * 8 blocks. - return divideCeil(L.SB->NumBlocks, 8 * L.SB->BlockSize); + return divideCeil(NumBlocks, 8 * BlockSize); +} + +inline uint32_t getNumFpmIntervals(const MSFLayout &L, + bool IncludeUnusedFpmData = false, + bool AltFpm = false) { + return getNumFpmIntervals(L.SB->BlockSize, L.SB->NumBlocks, + IncludeUnusedFpmData, AltFpm); } Error validateSuperBlock(const SuperBlock &SB); Index: llvm/lib/DebugInfo/MSF/MSFCommon.cpp =================================================================== --- llvm/lib/DebugInfo/MSF/MSFCommon.cpp +++ llvm/lib/DebugInfo/MSF/MSFCommon.cpp @@ -64,7 +64,8 @@ bool IncludeUnusedFpmData, bool AltFpm) { MSFStreamLayout FL; - uint32_t NumFpmIntervals = getNumFpmIntervals(Msf, IncludeUnusedFpmData); + uint32_t NumFpmIntervals = + getNumFpmIntervals(Msf, IncludeUnusedFpmData, AltFpm); support::ulittle32_t FpmBlock = Msf.SB->FreeBlockMapBlock; assert(FpmBlock == 1 || FpmBlock == 2); if (AltFpm) { Index: llvm/unittests/DebugInfo/MSF/MSFCommonTest.cpp =================================================================== --- llvm/unittests/DebugInfo/MSF/MSFCommonTest.cpp +++ llvm/unittests/DebugInfo/MSF/MSFCommonTest.cpp @@ -46,12 +46,47 @@ EXPECT_EQ(1u, getNumFpmIntervals(L, true)); SB.NumBlocks = SB.BlockSize; EXPECT_EQ(1u, getNumFpmIntervals(L, true)); - SB.NumBlocks = SB.BlockSize + 1; - EXPECT_EQ(2u, getNumFpmIntervals(L, true)); SB.NumBlocks = SB.BlockSize * 8; EXPECT_EQ(8u, getNumFpmIntervals(L, true)); - SB.NumBlocks = SB.BlockSize * 8 + 1; - EXPECT_EQ(9u, getNumFpmIntervals(L, true)); + + // The FPM is going to look like this: + // | 0 | 1 | 2 | ... | 4096 | 4097 | 4098 | ... | + // | SB | FPM0 | FPM1 | Data | Data | FPM0 | FPM1 | ... | + // + // So when there are up to 4097 blocks (last index 4096), the final blocks + // are data blocks. When there are 4098 blocks (last index 4097), there is + // one terminating FPM block, and when there are 4099 blocks, there are two + // terminating FPM blocks. Make sure all these cases are handled. + + // With 4096 or 4097 blocks, the last block is a data block so we only have + // 1 interval. + for (uint32_t I : {4096, 4097}) { + // 1 FPM0 interval + EXPECT_EQ(1U, getNumFpmIntervals(4096, I, true, false)); + EXPECT_EQ(1U, getNumFpmIntervals(4096, I, false, false)); + + // 1 FPM1 interval + EXPECT_EQ(1U, getNumFpmIntervals(4096, I, true, true)); + EXPECT_EQ(1U, getNumFpmIntervals(4096, I, false, true)); + } + + // With 4098 blocks, the last block belongs to FPM0 so we should have 2 FPM0 + // intervals. + EXPECT_EQ(2U, getNumFpmIntervals(4096, 4098, true, false)); + EXPECT_EQ(1U, getNumFpmIntervals(4096, 4098, false, false)); + + // And 1 FPM1 interval. + EXPECT_EQ(1U, getNumFpmIntervals(4096, 4098, true, true)); + EXPECT_EQ(1U, getNumFpmIntervals(4096, 4098, false, true)); + + // With 4099 blocks, the last block belongs to FPM1 so we should have 2 + // FPM0 intervals. + EXPECT_EQ(2U, getNumFpmIntervals(4096, 4099, true, false)); + EXPECT_EQ(1U, getNumFpmIntervals(4096, 4099, false, false)); + + // And 2 FPM1 intervals. + EXPECT_EQ(2U, getNumFpmIntervals(4096, 4099, true, true)); + EXPECT_EQ(1U, getNumFpmIntervals(4096, 4099, false, true)); } TEST(MSFCommonTest, FpmStreamLayout) { @@ -95,7 +130,7 @@ // 2. When we are including unused FPM data, there should be one FPM block // at every BlockSize interval in the file, even if entire FPM blocks are // unused. - SB.NumBlocks = SB.BlockSize * 8 + 1; + SB.NumBlocks = SB.BlockSize * 8 + 3; SL = getFpmStreamLayout(L, true, false); EXPECT_EQ(SB.BlockSize * 9, SL.Length); EXPECT_EQ(9u, SL.Blocks.size());