Encrypting should be OK.

[crypto_lab2.git] / labo2-fsharp / CryptoFile / API.fs

diff --git a/labo2-fsharp/CryptoFile/API.fs b/labo2-fsharp/CryptoFile/API.fs
index 2cf2167..e4a7008 100644 (file)
--- a/labo2-fsharp/CryptoFile/API.fs
+++ b/labo2-fsharp/CryptoFile/API.fs
@@ -1,20 +1,26 @@
 namespace CryptoFile
 open System.IO
 
-type internal Metadata (d : (string * string) list) =
+type internal Metadata (d: (string * string) list) =
     new (stream : Stream, size: int) =
         let binaryReader = new BinaryReader (stream)
         new Metadata ([])
+
     member this.WriteTo (stream : Stream) =
-        let binaryWriter = new BinaryWriter (stream)
-        List.iter (fun (key : string, value : string) -> binaryWriter.Write key; binaryWriter.Write value) d
+        let memoryStream = new MemoryStream () // To know the serialized meta-data size before writtent them to 'stream'.
+        let memoryWriter = new BinaryWriter (memoryStream)
+        List.iter (fun (key : string, value : string) -> memoryWriter.Write key; memoryWriter.Write value) d
+        (new BinaryWriter (stream)).Write (int memoryStream.Length)
+        memoryStream.CopyTo (stream)
 
 module API =
-    let internal filename = "filename"
-    let internal creationTimeKey = "file-creation-time"
+    module internal MetaData =
+        let filename = "filename"
+        let creationTimeKey = "file-creation-time"
 
     let generatKeysPair : Key * Key = Crypto.generateRSAKeysPair
 
+    // Encrypt a given file
     let encryptFile (inputFilePath : string) (outputFilePath : string) (signaturePrivKey: Key) (cryptPubKey : Key)  =
         let keyAES, keyMAC, iv = Crypto.rand 32, Crypto.rand 32, Crypto.rand 16
         let fileInfo = new FileInfo (inputFilePath)
@@ -22,24 +28,30 @@ module API =
         use outputStream = new FileStream (outputFilePath, FileMode.Create, FileAccess.Write)
         let writer = new BinaryWriter (outputStream)
 
-        ignore <| writer.Seek (8 + 32 + 256, SeekOrigin.Current) // Skips file-content-size, mac and signature. They will be written later.
+        writer.Seek (32 + 256, SeekOrigin.Current) |> ignore // Skips mac and signature. They will be written later.
 
         Crypto.encryptRSA cryptPubKey (Array.append keyAES <| Array.append keyMAC iv) |> writer.Write
 
-        printfn "pos: %A" outputStream.Position
-
-        use cryptoStream = Crypto.encryptAES keyAES iv outputStream
+        let (hmacStream, hmac) = Crypto.HMACStream keyMAC outputStream
+        use cryptoStream = Crypto.encryptAES keyAES iv hmacStream
         let cryptoWriter = new BinaryWriter (cryptoStream)
 
-        // Write metadata.
-        let metaData = new Metadata ([filename, fileInfo.Name; creationTimeKey, fileInfo.CreationTimeUtc.Ticks.ToString ()])
-        let metaDataStream = new MemoryStream ()
-        metaData.WriteTo metaDataStream
-        cryptoWriter.Write (int metaDataStream.Length)
-        printfn "meta size: %A" (int metaDataStream.Length)
-        metaDataStream.Position <- 0L
-        metaDataStream.CopyTo cryptoStream
+        // Write the file metadata.
+        let metaData = new Metadata ([MetaData.filename, fileInfo.Name
+                                      MetaData.creationTimeKey, fileInfo.CreationTimeUtc.Ticks.ToString ()])
+        metaData.WriteTo cryptoStream
+
+        // Write the content of the file.
+        inputStream.CopyTo cryptoStream
+        cryptoStream.FlushFinalBlock ()
+
+        // Write the HMAC at the begining of the file.
+        outputStream.Position <- 0L
+        writer.Write hmac.Hash
+
+        // Write the signature.
+        Crypto.signRSA signaturePrivKey hmac.Hash |> writer.Write
         ()
 
     let decryptFile (sourceFilePath : string) (targetDirPath : string) (signaturePubKey: Key) (decryptPrivKey : Key) =
-        ()
+        ()
\ No newline at end of file