Fixed the use of the provided fragments

tls-attacker · Jul 7, 2023 · 7b25103 · 7b25103
1 parent aaf0042
commit 7b25103
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/TLS-Core/src/main/java/de/rub/nds/tlsattacker/core/layer/impl/DtlsFragmentLayer.java b/TLS-Core/src/main/java/de/rub/nds/tlsattacker/core/layer/impl/DtlsFragmentLayer.java
@@ -114,12 +114,16 @@ public LayerProcessingResult<DtlsHandshakeMessageFragment> sendData(
                 fragments = getEnoughFragments(context, data.length);
             } else {
                 // use the provided fragments
-                fragments.add(getUnprocessedConfiguredContainers().get(0));
-                if (context.getConfig().isCreateFragmentsDynamically()) {
-                    fragments.addAll(
-                            getEnoughFragments(
-                                    context,
-                                    data.length - fragments.get(0).getMaxFragmentLengthConfig()));
+                int dataToBeSent = data.length;
+                List<DtlsHandshakeMessageFragment> givenFragments =
+                        getUnprocessedConfiguredContainers();
+                while (dataToBeSent > 0 && givenFragments.size() > 0) {
+                    DtlsHandshakeMessageFragment nextFragment = givenFragments.remove(0);
+                    fragments.add(nextFragment);
+                    dataToBeSent -= nextFragment.getMaxFragmentLengthConfig();
+                }
+                if (dataToBeSent > 0 && context.getConfig().isCreateFragmentsDynamically()) {
+                    fragments.addAll(getEnoughFragments(context, dataToBeSent));
                 }
             }
             fragments =