Merge pull request #101 from mdaus/bugfix/multiple-des-buffers-in-byt…

…estream Fixing NITF subheader length for multiple DES writes
ngageoint · Oct 22, 2018 · 047aea3 · 047aea3
2 parents ed27b46 + eb04970
commit 047aea3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/modules/c++/nitf/source/ByteProvider.cpp b/modules/c++/nitf/source/ByteProvider.cpp
@@ -243,14 +243,14 @@ void ByteProvider::getFileLayout(nitf::Record& inRecord,
 
     std::vector<size_t> desSubheaderLengths(numDESs);
     std::vector<size_t> desDataLengths(numDESs);
-
     for (size_t ii = 0; ii < numDESs; ++ii)
     {
         nitf::DESegment deSegment = record.getDataExtensions()[ii];
         nitf::DESubheader subheader = deSegment.getSubheader();
         nitf::Uint32 userSublen;
+        const size_t prevSize = byteStream->getSize();
         writer.writeDESubheader(subheader, userSublen, record.getVersion());
-        desSubheaderLengths[ii] = byteStream->getSize();
+        desSubheaderLengths[ii] = byteStream->getSize() - prevSize;
 
         // Write data
         const PtrAndLength& curData(desData[ii]);