Merge pull request #29 from ComposableFi/godilov/polkadot-signature

fix: polkadot full signature check

Author: godilov

vesting-backend/src/main.rs

@@ -295,14 +295,21 @@ pub async fn submit_signature(signature_params: web::Json<SignatureParams>) -> i
             return HttpResponse::Ok().body(serde_json::to_string(&ret).unwrap());
         }
     } else if post_params.network == 2 {
-        let msg = post_params.message_bytes.as_deref().unwrap_or_default();
-        let sig = post_params.signatures_bytes.as_deref().unwrap_or_default();
+        let msg = post_params.message.as_bytes();
+        let sig = post_params.signatures.as_bytes();
+
+        let msgb = post_params.message_bytes.as_deref().unwrap_or_default();
+        let sigb = post_params.signatures_bytes.as_deref().unwrap_or_default();
 
         let result = verify::verify_polkadot_signature(msg, sig, &post_params.pubkey);
         let result_ed = verify::verify_polkadot_signature_ed25519(msg, sig, &post_params.pubkey);
         let result_ec = verify::verify_polkadot_signature_ecdsa(msg, sig, &post_params.pubkey);
 
-        if result || result_ed || result_ec {
+        let resultb = verify::verify_polkadot_signature(msgb, sigb, &post_params.pubkey);
+        let result_edb = verify::verify_polkadot_signature_ed25519(msgb, sigb, &post_params.pubkey);
+        let result_ecb = verify::verify_polkadot_signature_ecdsa(msgb, sigb, &post_params.pubkey);
+
+        if result || result_ed || result_ec || resultb || result_edb || result_ecb {
             let mut dropped_item = DroppedItem {
                 address: post_params.pubkey.clone(),
                 solana_address: post_params.message.clone(),

vesting-backend/src/verify.rs

@@ -97,21 +97,18 @@ pub fn verify_polkadot_signature_ed25519(
     signature.verify(message, &public_key)
 }
 pub fn verify_polkadot_signature_ecdsa(message: &[u8], signature: &[u8], public_key: &str) -> bool {
-    // Convert the signature from bytes to the correct type
     let signature = EcdsaSignature::from_slice(signature);
     if signature.is_none() {
         return false;
     }
     let signature = signature.unwrap();
 
-    // Convert the public key from bytes to the correct type
     let public_key = EcdsaPublic::from_ss58check(public_key);
     if public_key.is_err() {
         return false;
     }
     let public_key = public_key.unwrap();
 
-    // Verify the signature
     signature.verify(message, &public_key)
 }