Bloom Filter for Replay Attacks

Sources/ShadowSwift/DarkStar.swift

@@ -163,18 +163,25 @@ public struct DarkStar
 
 
 
-    static public func handleTheirEphemeralPublicKey(connection: Connection, bloomFilter: BloomFilter<Data>) -> P256.KeyAgreement.PublicKey?
+    static public func handleTheirEphemeralPublicKey(connection: Connection, bloomFilter: BloomFilter<Data>?) -> P256.KeyAgreement.PublicKey?
     {
         // Receive their ephemeral key
         guard let theirEphemeralPublicKeyData = connection.read(size: P256KeySize) else
         {
             return nil
         }
 
-        // TODO: See if theirEphemeralPublicKeyData is in the BloomFilter, return nil if it is.
-
-        // TODO: If it's not in a BloomFilter, add it to the BloomFilter and Save the BloomFilter
-
+        if let serverBloomFilter = bloomFilter // Server
+        {
+            // TODO: See if theirEphemeralPublicKeyData is in the BloomFilter, return nil if it is.
+
+            // TODO: If it's not in a BloomFilter, add it to the BloomFilter and Save the BloomFilter
+        }
+        else // Client
+        {
+
+        }
+
         guard let theirEphemeralPublicKey = try? P256.KeyAgreement.PublicKey(compactRepresentation: theirEphemeralPublicKeyData) else
         {
             return nil

Sources/ShadowSwift/DarkStarClient.swift

@@ -176,7 +176,8 @@ public class DarkStarClient
         guard DarkStarClient.handleClientConfirmationCode(connection: connection, theirPublicKey: serverPersistentPublicKey, myPrivateKey: clientEphemeralPrivateKey, endpoint: endpoint, serverPersistentPublicKey: serverPersistentPublicKey, clientEphemeralPublicKey: clientEphemeralPublicKey) else {return nil}
 
         // Receive server ephemeral key
-        guard let serverEphemeralPublicKey = DarkStar.handleTheirEphemeralPublicKey(connection: connection) else {return nil}
+        guard let serverEphemeralPublicKey = DarkStar.handleTheirEphemeralPublicKey(connection: connection, bloomFilter: nil) else
+        {return nil}
 
         // Create shared key
         guard let clientToServerSharedKey = DarkStarClient.createClientToServerSharedKey(clientEphemeralPrivateKey: clientEphemeralPrivateKey, serverEphemeralPublicKey: serverEphemeralPublicKey, serverPersistentPublicKey: serverPersistentPublicKey, serverEndpoint: endpoint) else {return nil}

Sources/ShadowSwift/DarkStarServer.swift

@@ -154,7 +154,7 @@ public class DarkStarServer
         let serverPersistentPublicKey = serverPersistentPrivateKey.publicKey
 
         // Receive client ephemeral key
-        guard let clientEphemeralPublicKey = DarkStar.handleTheirEphemeralPublicKey(connection: connection) else
+        guard let clientEphemeralPublicKey = DarkStar.handleTheirEphemeralPublicKey(connection: connection, bloomFilter: bloomFilter) else
         {
             let transport = TransmissionToTransportConnection({return connection})
 

Sources/ShadowSwift/DarkStarServerConnection.swift

@@ -110,7 +110,7 @@ open class DarkStarServerConnection: Transport.Connection
             return nil
         }
 
-        guard let server = DarkStarServer(serverPersistentPrivateKey: serverPersistentPrivateKey, endpoint: endpoint, connection: connection) else
+        guard let server = DarkStarServer(serverPersistentPrivateKey: serverPersistentPrivateKey, endpoint: endpoint, connection: connection, bloomFilter: bloomFilter) else
         {
             logger.error("Failed to init DarkStarServer")
             return nil