Fixup deserialize_bs58_transaction, and make a few error types more targeted (#10171)

automerge
This commit is contained in:
Tyera Eulberg
2020-05-21 18:30:02 -06:00
committed by GitHub
parent 18be7a7966
commit 12a3b1ba6a
2 changed files with 21 additions and 18 deletions

View File

@ -663,11 +663,15 @@ fn get_tpu_addr(cluster_info: &ClusterInfo) -> Result<SocketAddr> {
} }
fn verify_pubkey(input: String) -> Result<Pubkey> { fn verify_pubkey(input: String) -> Result<Pubkey> {
input.parse().map_err(|_e| Error::invalid_request()) input
.parse()
.map_err(|e| Error::invalid_params(format!("{:?}", e)))
} }
fn verify_signature(input: &str) -> Result<Signature> { fn verify_signature(input: &str) -> Result<Signature> {
input.parse().map_err(|_e| Error::invalid_request()) input
.parse()
.map_err(|e| Error::invalid_params(format!("{:?}", e)))
} }
#[derive(Clone)] #[derive(Clone)]
@ -1524,21 +1528,24 @@ impl RpcSol for RpcSolImpl {
} }
fn deserialize_bs58_transaction(bs58_transaction: String) -> Result<(Vec<u8>, Transaction)> { fn deserialize_bs58_transaction(bs58_transaction: String) -> Result<(Vec<u8>, Transaction)> {
let wire_transaction = bs58::decode(bs58_transaction).into_vec().unwrap(); let wire_transaction = bs58::decode(bs58_transaction)
.into_vec()
.map_err(|e| Error::invalid_params(format!("{:?}", e)))?;
if wire_transaction.len() >= PACKET_DATA_SIZE { if wire_transaction.len() >= PACKET_DATA_SIZE {
info!( let err = format!(
"transaction too large: {} bytes (max: {} bytes)", "transaction too large: {} bytes (max: {} bytes)",
wire_transaction.len(), wire_transaction.len(),
PACKET_DATA_SIZE PACKET_DATA_SIZE
); );
return Err(Error::invalid_request()); info!("{}", err);
return Err(Error::invalid_params(&err));
} }
bincode::config() bincode::config()
.limit(PACKET_DATA_SIZE as u64) .limit(PACKET_DATA_SIZE as u64)
.deserialize(&wire_transaction) .deserialize(&wire_transaction)
.map_err(|err| { .map_err(|err| {
info!("transaction deserialize error: {:?}", err); info!("transaction deserialize error: {:?}", err);
Error::invalid_request() Error::invalid_params(&err.to_string())
}) })
.map(|transaction| (wire_transaction, transaction)) .map(|transaction| (wire_transaction, transaction))
} }
@ -1552,7 +1559,7 @@ pub mod tests {
replay_stage::tests::create_test_transactions_and_populate_blockstore, replay_stage::tests::create_test_transactions_and_populate_blockstore,
}; };
use bincode::deserialize; use bincode::deserialize;
use jsonrpc_core::{MetaIoHandler, Output, Response, Value}; use jsonrpc_core::{ErrorCode, MetaIoHandler, Output, Response, Value};
use solana_ledger::{ use solana_ledger::{
blockstore::entries_to_test_shreds, blockstore::entries_to_test_shreds,
blockstore_processor::fill_blockstore_slot_with_ticks, blockstore_processor::fill_blockstore_slot_with_ticks,
@ -2634,13 +2641,9 @@ pub mod tests {
let req = r#"{"jsonrpc":"2.0","id":1,"method":"sendTransaction","params":["37u9WtQpcm6ULa3Vmu7ySnANv"]}"#; let req = r#"{"jsonrpc":"2.0","id":1,"method":"sendTransaction","params":["37u9WtQpcm6ULa3Vmu7ySnANv"]}"#;
let res = io.handle_request_sync(req, meta); let res = io.handle_request_sync(req, meta);
let expected = let json: Value = serde_json::from_str(&res.unwrap()).unwrap();
r#"{"jsonrpc":"2.0","error":{"code":-32600,"message":"Invalid request"},"id":1}"#; let error = &json["error"];
let expected: Response = assert_eq!(error["code"], ErrorCode::InvalidParams.code());
serde_json::from_str(expected).expect("expected response deserialization");
let result: Response = serde_json::from_str(&res.expect("actual response"))
.expect("actual response deserialization");
assert_eq!(expected, result);
} }
#[test] #[test]
@ -2661,7 +2664,7 @@ pub mod tests {
let bad_pubkey = "a1b2c3d4"; let bad_pubkey = "a1b2c3d4";
assert_eq!( assert_eq!(
verify_pubkey(bad_pubkey.to_string()), verify_pubkey(bad_pubkey.to_string()),
Err(Error::invalid_request()) Err(Error::invalid_params("WrongSize"))
); );
} }
@ -2675,7 +2678,7 @@ pub mod tests {
let bad_signature = "a1b2c3d4"; let bad_signature = "a1b2c3d4";
assert_eq!( assert_eq!(
verify_signature(&bad_signature.to_string()), verify_signature(&bad_signature.to_string()),
Err(Error::invalid_request()) Err(Error::invalid_params("WrongSize"))
); );
} }

View File

@ -120,14 +120,14 @@ fn test_rpc_invalid_requests() {
let json = post_rpc(req, &leader_data); let json = post_rpc(req, &leader_data);
let the_error = json["error"]["message"].as_str().unwrap(); let the_error = json["error"]["message"].as_str().unwrap();
assert_eq!(the_error, "Invalid request"); assert_eq!(the_error, "Invalid");
// test invalid get_account_info request // test invalid get_account_info request
let req = json_req!("getAccountInfo", json!(["invalid9999"])); let req = json_req!("getAccountInfo", json!(["invalid9999"]));
let json = post_rpc(req, &leader_data); let json = post_rpc(req, &leader_data);
let the_error = json["error"]["message"].as_str().unwrap(); let the_error = json["error"]["message"].as_str().unwrap();
assert_eq!(the_error, "Invalid request"); assert_eq!(the_error, "Invalid");
// test invalid get_account_info request // test invalid get_account_info request
let req = json_req!("getAccountInfo", json!([bob_pubkey.to_string()])); let req = json_req!("getAccountInfo", json!([bob_pubkey.to_string()]));