revm_precompile/
secp256r1.rs1use crate::{
10 crypto, u64_to_address, Precompile, PrecompileError, PrecompileId, PrecompileOutput,
11 PrecompileResult,
12};
13use primitives::{alloy_primitives::B512, Bytes, B256};
14
15pub const P256VERIFY_ADDRESS: u64 = 256;
17
18pub const P256VERIFY_BASE_GAS_FEE: u64 = 3450;
20
21pub const P256VERIFY_BASE_GAS_FEE_OSAKA: u64 = 6900;
23
24pub fn precompiles() -> impl Iterator<Item = Precompile> {
26 [P256VERIFY].into_iter()
27}
28
29pub const P256VERIFY: Precompile = Precompile::new(
31 PrecompileId::P256Verify,
32 u64_to_address(P256VERIFY_ADDRESS),
33 p256_verify,
34);
35
36pub const P256VERIFY_OSAKA: Precompile = Precompile::new(
38 PrecompileId::P256Verify,
39 u64_to_address(P256VERIFY_ADDRESS),
40 p256_verify_osaka,
41);
42
43pub fn p256_verify(input: &[u8], gas_limit: u64) -> PrecompileResult {
53 p256_verify_inner(input, gas_limit, P256VERIFY_BASE_GAS_FEE)
54}
55
56pub fn p256_verify_osaka(input: &[u8], gas_limit: u64) -> PrecompileResult {
66 p256_verify_inner(input, gas_limit, P256VERIFY_BASE_GAS_FEE_OSAKA)
67}
68
69fn p256_verify_inner(input: &[u8], gas_limit: u64, gas_cost: u64) -> PrecompileResult {
70 if gas_cost > gas_limit {
71 return Err(PrecompileError::OutOfGas);
72 }
73 let result = if verify_impl(input) {
74 B256::with_last_byte(1).into()
75 } else {
76 Bytes::new()
77 };
78 Ok(PrecompileOutput::new(gas_cost, result))
79}
80
81pub fn verify_impl(input: &[u8]) -> bool {
84 if input.len() != 160 {
85 return false;
86 }
87
88 let msg = <&B256>::try_from(&input[..32]).unwrap();
90 let sig = <&B512>::try_from(&input[32..96]).unwrap();
92 let pk = <&B512>::try_from(&input[96..160]).unwrap();
94
95 crypto().secp256r1_verify_signature(&msg.0, &sig.0, &pk.0)
96}
97
98pub(crate) fn verify_signature(msg: [u8; 32], sig: [u8; 64], pk: [u8; 64]) -> Option<()> {
99 cfg_if::cfg_if! {
100 if #[cfg(feature = "p256-aws-lc-rs")] {
101 use aws_lc_rs::{digest, signature::{self, UnparsedPublicKey}};
102
103 let digest = digest::Digest::import_less_safe(&msg, &digest::SHA256).ok()?;
105
106 let mut pubkey_bytes = [0u8; 65];
108 pubkey_bytes[0] = 0x04;
109 pubkey_bytes[1..].copy_from_slice(&pk);
110
111 let public_key = UnparsedPublicKey::new(&signature::ECDSA_P256_SHA256_FIXED, &pubkey_bytes);
112
113 public_key.verify_digest(&digest, &sig).ok()
114 } else {
115 use p256::{
116 ecdsa::{signature::hazmat::PrehashVerifier, Signature, VerifyingKey},
117 EncodedPoint,
118 };
119
120 let signature = Signature::from_slice(&sig).ok()?;
122 let encoded_point = EncodedPoint::from_untagged_bytes(&pk.into());
124 let public_key = VerifyingKey::from_encoded_point(&encoded_point).ok()?;
126
127 public_key.verify_prehash(&msg, &signature).ok()
128 }
129 }
130}
131
132#[cfg(test)]
133mod test {
134 use super::*;
135 use crate::PrecompileError;
136 use primitives::hex::FromHex;
137 use rstest::rstest;
138
139 #[rstest]
140 #[case::ok_1("4cee90eb86eaa050036147a12d49004b6b9c72bd725d39d4785011fe190f0b4da73bd4903f0ce3b639bbbf6e8e80d16931ff4bcf5993d58468e8fb19086e8cac36dbcd03009df8c59286b162af3bd7fcc0450c9aa81be5d10d312af6c66b1d604aebd3099c618202fcfe16ae7770b0c49ab5eadf74b754204a3bb6060e44eff37618b065f9832de4ca6ca971a7a1adc826d0f7c00181a5fb2ddf79ae00b4e10e", true)]
142 #[case::ok_2("3fec5769b5cf4e310a7d150508e82fb8e3eda1c2c94c61492d3bd8aea99e06c9e22466e928fdccef0de49e3503d2657d00494a00e764fd437bdafa05f5922b1fbbb77c6817ccf50748419477e843d5bac67e6a70e97dde5a57e0c983b777e1ad31a80482dadf89de6302b1988c82c29544c9c07bb910596158f6062517eb089a2f54c9a0f348752950094d3228d3b940258c75fe2a413cb70baa21dc2e352fc5", true)]
143 #[case::ok_3("e775723953ead4a90411a02908fd1a629db584bc600664c609061f221ef6bf7c440066c8626b49daaa7bf2bcc0b74be4f7a1e3dcf0e869f1542fe821498cbf2de73ad398194129f635de4424a07ca715838aefe8fe69d1a391cfa70470795a80dd056866e6e1125aff94413921880c437c9e2570a28ced7267c8beef7e9b2d8d1547d76dfcf4bee592f5fefe10ddfb6aeb0991c5b9dbbee6ec80d11b17c0eb1a", true)]
144 #[case::ok_4("b5a77e7a90aa14e0bf5f337f06f597148676424fae26e175c6e5621c34351955289f319789da424845c9eac935245fcddd805950e2f02506d09be7e411199556d262144475b1fa46ad85250728c600c53dfd10f8b3f4adf140e27241aec3c2da3a81046703fccf468b48b145f939efdbb96c3786db712b3113bb2488ef286cdcef8afe82d200a5bb36b5462166e8ce77f2d831a52ef2135b2af188110beaefb1", true)]
145 #[case::ok_5("858b991cfd78f16537fe6d1f4afd10273384db08bdfc843562a22b0626766686f6aec8247599f40bfe01bec0e0ecf17b4319559022d4d9bf007fe929943004eb4866760dedf31b7c691f5ce665f8aae0bda895c23595c834fecc2390a5bcc203b04afcacbb4280713287a2d0c37e23f7513fab898f2c1fefa00ec09a924c335d9b629f1d4fb71901c3e59611afbfea354d101324e894c788d1c01f00b3c251b2", true)]
146 #[case::fail_wrong_msg_1("3cee90eb86eaa050036147a12d49004b6b9c72bd725d39d4785011fe190f0b4da73bd4903f0ce3b639bbbf6e8e80d16931ff4bcf5993d58468e8fb19086e8cac36dbcd03009df8c59286b162af3bd7fcc0450c9aa81be5d10d312af6c66b1d604aebd3099c618202fcfe16ae7770b0c49ab5eadf74b754204a3bb6060e44eff37618b065f9832de4ca6ca971a7a1adc826d0f7c00181a5fb2ddf79ae00b4e10e", false)]
147 #[case::fail_wrong_msg_2("afec5769b5cf4e310a7d150508e82fb8e3eda1c2c94c61492d3bd8aea99e06c9e22466e928fdccef0de49e3503d2657d00494a00e764fd437bdafa05f5922b1fbbb77c6817ccf50748419477e843d5bac67e6a70e97dde5a57e0c983b777e1ad31a80482dadf89de6302b1988c82c29544c9c07bb910596158f6062517eb089a2f54c9a0f348752950094d3228d3b940258c75fe2a413cb70baa21dc2e352fc5", false)]
148 #[case::fail_wrong_msg_3("f775723953ead4a90411a02908fd1a629db584bc600664c609061f221ef6bf7c440066c8626b49daaa7bf2bcc0b74be4f7a1e3dcf0e869f1542fe821498cbf2de73ad398194129f635de4424a07ca715838aefe8fe69d1a391cfa70470795a80dd056866e6e1125aff94413921880c437c9e2570a28ced7267c8beef7e9b2d8d1547d76dfcf4bee592f5fefe10ddfb6aeb0991c5b9dbbee6ec80d11b17c0eb1a", false)]
149 #[case::fail_wrong_msg_4("c5a77e7a90aa14e0bf5f337f06f597148676424fae26e175c6e5621c34351955289f319789da424845c9eac935245fcddd805950e2f02506d09be7e411199556d262144475b1fa46ad85250728c600c53dfd10f8b3f4adf140e27241aec3c2da3a81046703fccf468b48b145f939efdbb96c3786db712b3113bb2488ef286cdcef8afe82d200a5bb36b5462166e8ce77f2d831a52ef2135b2af188110beaefb1", false)]
150 #[case::fail_wrong_msg_5("958b991cfd78f16537fe6d1f4afd10273384db08bdfc843562a22b0626766686f6aec8247599f40bfe01bec0e0ecf17b4319559022d4d9bf007fe929943004eb4866760dedf31b7c691f5ce665f8aae0bda895c23595c834fecc2390a5bcc203b04afcacbb4280713287a2d0c37e23f7513fab898f2c1fefa00ec09a924c335d9b629f1d4fb71901c3e59611afbfea354d101324e894c788d1c01f00b3c251b2", false)]
151 #[case::fail_short_input_1("4cee90eb86eaa050036147a12d49004b6a", false)]
152 #[case::fail_short_input_2("4cee90eb86eaa050036147a12d49004b6a958b991cfd78f16537fe6d1f4afd10273384db08bdfc843562a22b0626766686f6aec8247599f40bfe01bec0e0ecf17b4319559022d4d9bf007fe929943004eb4866760dedf319", false)]
153 #[case::fail_long_input("4cee90eb86eaa050036147a12d49004b6b9c72bd725d39d4785011fe190f0b4da73bd4903f0ce3b639bbbf6e8e80d16931ff4bcf5993d58468e8fb19086e8cac36dbcd03009df8c59286b162af3bd7fcc0450c9aa81be5d10d312af6c66b1d604aebd3099c618202fcfe16ae7770b0c49ab5eadf74b754204a3bb6060e44eff37618b065f9832de4ca6ca971a7a1adc826d0f7c00181a5fb2ddf79ae00b4e10e00", false)]
154 #[case::fail_invalid_sig("4cee90eb86eaa050036147a12d49004b6b9c72bd725d39d4785011fe190f0b4dffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff4aebd3099c618202fcfe16ae7770b0c49ab5eadf74b754204a3bb6060e44eff37618b065f9832de4ca6ca971a7a1adc826d0f7c00181a5fb2ddf79ae00b4e10e", false)]
155 #[case::fail_invalid_pubkey("4cee90eb86eaa050036147a12d49004b6b9c72bd725d39d4785011fe190f0b4da73bd4903f0ce3b639bbbf6e8e80d16931ff4bcf5993d58468e8fb19086e8cac36dbcd03009df8c59286b162af3bd7fcc0450c9aa81be5d10d312af6c66b1d6000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000", false)]
156 fn test_sig_verify(#[case] input: &str, #[case] expect_success: bool) {
157 let input = Bytes::from_hex(input).unwrap();
158 let target_gas = 3_500u64;
159 let outcome = p256_verify(&input, target_gas).unwrap();
160 assert_eq!(outcome.gas_used, 3_450u64);
161 let expected_result = if expect_success {
162 B256::with_last_byte(1).into()
163 } else {
164 Bytes::new()
165 };
166 assert_eq!(outcome.bytes, expected_result);
167 }
168
169 #[rstest]
170 fn test_not_enough_gas_errors() {
171 let input = Bytes::from_hex("4cee90eb86eaa050036147a12d49004b6b9c72bd725d39d4785011fe190f0b4da73bd4903f0ce3b639bbbf6e8e80d16931ff4bcf5993d58468e8fb19086e8cac36dbcd03009df8c59286b162af3bd7fcc0450c9aa81be5d10d312af6c66b1d604aebd3099c618202fcfe16ae7770b0c49ab5eadf74b754204a3bb6060e44eff37618b065f9832de4ca6ca971a7a1adc826d0f7c00181a5fb2ddf79ae00b4e10e").unwrap();
172 let target_gas = 2_500u64;
173 let result = p256_verify(&input, target_gas);
174
175 assert!(result.is_err());
176 assert_eq!(result.err(), Some(PrecompileError::OutOfGas));
177 }
178
179 #[rstest]
180 #[case::ok_1("b5a77e7a90aa14e0bf5f337f06f597148676424fae26e175c6e5621c34351955289f319789da424845c9eac935245fcddd805950e2f02506d09be7e411199556d262144475b1fa46ad85250728c600c53dfd10f8b3f4adf140e27241aec3c2da3a81046703fccf468b48b145f939efdbb96c3786db712b3113bb2488ef286cdcef8afe82d200a5bb36b5462166e8ce77f2d831a52ef2135b2af188110beaefb1", true)]
181 #[case::fail_1("b5a77e7a90aa14e0bf5f337f06f597148676424fae26e175c6e5621c34351955289f319789da424845c9eac935245fcddd805950e2f02506d09be7e411199556d262144475b1fa46ad85250728c600c53dfd10f8b3f4adf140e27241aec3c2daaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaef8afe82d200a5bb36b5462166e8ce77f2d831a52ef2135b2af188110beaefb1", false)]
182 fn test_verify_impl(#[case] input: &str, #[case] expect_success: bool) {
183 let input = Bytes::from_hex(input).unwrap();
184 let result = verify_impl(&input);
185
186 assert_eq!(result, expect_success);
187 }
188}