Prevent printing binary in RPC errors (#1604)

## Issue Addressed

#1566 

## Proposed Changes

Prevents printing binary characters in the RPC error response from peers.
This commit is contained in:
Age Manning 2020-09-10 04:43:22 +00:00
parent b19cf02d2d
commit d79366c503
3 changed files with 6 additions and 4 deletions

1
Cargo.lock generated
View File

@ -1521,6 +1521,7 @@ dependencies = [
"lru",
"parking_lot 0.11.0",
"rand 0.7.3",
"regex",
"serde",
"serde_derive",
"sha2 0.9.1",

View File

@ -37,6 +37,7 @@ tiny-keccak = "2.0.2"
environment = { path = "../../lighthouse/environment" }
# TODO: Remove rand crate for mainnet
rand = "0.7.3"
regex = "1.3.9"
[dependencies.libp2p]
#version = "0.23.0"

View File

@ -1,6 +1,7 @@
//! Available RPC methods types and ids.
use crate::types::EnrBitfield;
use regex::bytes::Regex;
use serde::Serialize;
use ssz_derive::{Decode, Encode};
use ssz_types::{
@ -42,10 +43,9 @@ impl Deref for ErrorType {
impl ToString for ErrorType {
fn to_string(&self) -> String {
match std::str::from_utf8(self.0.deref()) {
Ok(s) => s.to_string(),
Err(_) => format!("{:?}", self.0.deref()), // Display raw bytes if not a UTF-8 string
}
#[allow(clippy::invalid_regex)]
let re = Regex::new("\\p{C}").expect("Regex is valid");
String::from_utf8_lossy(&re.replace_all(self.0.deref(), &b""[..])).to_string()
}
}