From 0c91fe7e130527353f2cbd2ff1c4d443d0d6c4be Mon Sep 17 00:00:00 2001
From: Roman <zajic@zajic.net>
Date: Thu, 23 Jan 2025 11:40:36 +0800
Subject: [PATCH] fix: add padding for dispersal data

---
 src/steps/da.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/src/steps/da.py b/src/steps/da.py
index bbf1379..c534e75 100644
--- a/src/steps/da.py
+++ b/src/steps/da.py
@@ -4,9 +4,29 @@ from src.env_vars import NOMOS_EXECUTOR
 from src.steps.common import StepsCommon
 
 
+def add_padding(orig_bytes):
+    block_size = 31
+    """
+    Pads a list of bytes (integers in [0..255]) using a PKCS#7-like scheme:
+    - The value of each padded byte is the number of bytes padded.
+    - If the original data is already a multiple of the block size,
+      an additional full block of bytes (each the block size) is added.
+    """
+    original_len = len(orig_bytes)
+    padding_needed = block_size - (original_len % block_size)
+    # If the data is already a multiple of block_size, add a full block of padding
+    if padding_needed == 0:
+        padding_needed = block_size
+
+    # Each padded byte will be equal to padding_needed
+    padded_bytes = orig_bytes + [padding_needed] * padding_needed
+    return padded_bytes
+
+
 def prepare_dispersal_request(data, app_id, index):
     data_bytes = data.encode("utf-8")
-    dispersal_data = {"data": list(data_bytes), "metadata": {"app_id": app_id, "index": index}}
+    padded_bytes = add_padding(list(data_bytes))
+    dispersal_data = {"data": padded_bytes, "metadata": {"app_id": app_id, "index": index}}
     return dispersal_data