fix on the SignedProtoMessage

pfouto · Mar 6, 2023 · 7761287 · 7761287
1 parent a5c3bc3
commit 7761287
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/main/java/pt/unl/fct/di/novasys/babel/generic/signed/SignedMessageSerializer.java b/src/main/java/pt/unl/fct/di/novasys/babel/generic/signed/SignedMessageSerializer.java
@@ -14,7 +14,7 @@ public void serialize(T msg, ByteBuf out) throws IOException {
         if(msg.serializedMessage == null || msg.serializedMessage.length == 0) {
         	ByteBuf serial = Unpooled.buffer();
         	serial.writeShort(msg.getId());
-        	this.serializeBody(msg, out);
+        	this.serializeBody(msg, serial);
         	msg.serializedMessage = ByteBufUtil.getBytes(serial);
         }
 

diff --git a/src/main/java/pt/unl/fct/di/novasys/babel/generic/signed/SignedProtoMessage.java b/src/main/java/pt/unl/fct/di/novasys/babel/generic/signed/SignedProtoMessage.java
@@ -39,7 +39,7 @@ public final void signMessage(PrivateKey key) throws NoSuchAlgorithmException, I
 			} else {
 				ByteBuf b = Unpooled.buffer();
 				serializer.serializeBody(this, b);
-				this.serializedMessage = ByteBufUtil.getBytes(b);
+				this.serializedMessage = ByteBufUtil.getBytes(b.slice());
 			}
 		}