rusk: persisting driver code signature

Author: miloszm

rusk/src/lib/http/rusk.rs

@@ -220,7 +220,11 @@ impl Rusk {
         }
 
         let mut driver_store = self.driver_store.write();
-        driver_store.store_bytecode(&contract_id, data)?;
+        driver_store.store_bytecode_and_signature(
+            &contract_id,
+            data,
+            signature.to_bytes(),
+        )?;
         let mut instance_cache = self.instance_cache.write();
         instance_cache.remove(&contract_id);
         Ok(ResponseData::new(UPLOAD_DRIVER_RESPONSE.to_string()))

rusk/src/lib/node/driverstore.rs

@@ -4,10 +4,14 @@
 //
 // Copyright (c) DUSK NETWORK. All rights reserved.
 
+use dusk_bytes::Serializable;
 use std::path::{Path, PathBuf};
 use std::{fs, io};
 
 use dusk_core::abi::ContractId;
+use dusk_core::signatures::bls::Signature;
+
+const SIGNATURE_FILE_EXTENSION: &str = "sign";
 
 pub struct DriverStore {
     store_path: Option<PathBuf>,
@@ -27,13 +31,27 @@ impl DriverStore {
         self.contract_path(contract_id).map(fs::read).transpose()
     }
 
-    pub fn store_bytecode(
+    pub fn get_signature(
+        &self,
+        contract_id: &ContractId,
+    ) -> io::Result<Option<Vec<u8>>> {
+        self.contract_path(contract_id)
+            .map(|path| fs::read(path.with_extension(SIGNATURE_FILE_EXTENSION)))
+            .transpose()
+    }
+
+    pub fn store_bytecode_and_signature(
         &mut self,
         contract_id: &ContractId,
         bytecode: &[u8],
+        signature_bytes: [u8; Signature::SIZE],
     ) -> io::Result<()> {
         if let Some(path) = self.contract_path(contract_id) {
-            fs::write(path, bytecode)?;
+            fs::write(&path, bytecode)?;
+            fs::write(
+                path.with_extension(SIGNATURE_FILE_EXTENSION),
+                signature_bytes,
+            )?;
         }
         Ok(())
     }