Add duplicate signer check

Outblock · Aug 21, 2024 · 51d8954 · 51d8954
1 parent 123c01a
commit 51d8954
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Sources/Models/FlowTransaction.swift b/Sources/Models/FlowTransaction.swift
@@ -192,6 +192,14 @@ public extension Flow {
         }
 
         public mutating func addPayloadSignature(address: Address, keyIndex: Int, signature: Data) {
+
+            // Avoid same signer with multiple signatures
+            for sig in payloadSignatures {
+                if sig.keyIndex == keyIndex && sig.address == address {
+                    return
+                }
+            }
+
             payloadSignatures.append(
                 TransactionSignature(address: address,
                                      signerIndex: signers[address] ?? -1,
@@ -202,6 +210,14 @@ public extension Flow {
         }
 
         public mutating func addEnvelopeSignature(address: Address, keyIndex: Int, signature: Data) {
+
+            // Avoid same signer with multiple signatures
+            for sig in envelopeSignatures {
+                if sig.keyIndex == keyIndex && sig.address == address {
+                    return
+                }
+            }
+
             envelopeSignatures.append(
                 TransactionSignature(address: address,
                                      signerIndex: signers[address] ?? -1,