From 6eb9a403558d4dfde1de36e5c6744b58dc26ea79 Mon Sep 17 00:00:00 2001
From: "Matthias P. Braendli" <matthias.braendli@mpb.li>
Date: Sun, 12 Feb 2017 17:51:48 +0100
Subject: Fix incorrect frame size bug due to missing puncturing padding

---
 src/ConvEncoder.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/ConvEncoder.cpp')

diff --git a/src/ConvEncoder.cpp b/src/ConvEncoder.cpp
index 74de34f..06b2e85 100644
--- a/src/ConvEncoder.cpp
+++ b/src/ConvEncoder.cpp
@@ -111,6 +111,7 @@ int ConvEncoder::process(Buffer* const dataIn, Buffer* dataOut)
             }
             ++in_offset;
         }
+
         for (unsigned pad_count = 0; pad_count < 3; ++pad_count) {
             out[out_offset] = 0;
             // For each 4-bit output word
@@ -138,5 +139,11 @@ int ConvEncoder::process(Buffer* const dataIn, Buffer* dataOut)
 
     PDEBUG(" Consume: %zu\n", in_offset);
     PDEBUG(" Return: %zu\n", out_offset);
+
+    if (out_offset != dataOut->getLength()) {
+        throw std::runtime_error("Assertion error: ConvEncoder output " +
+                std::to_string(out_offset) + " == " +
+                std::to_string(dataOut->getLength()) + " fail");
+    }
     return out_offset;
 }
-- 
cgit v1.2.3