Skip to content

Commit

Permalink
Secure API JSON RPC ID can be integer or string (#314)
Browse files Browse the repository at this point in the history
* Fix #312

* Fix tests
  • Loading branch information
quentinlesceller authored and yeastplume committed Jan 24, 2020
1 parent f09c91f commit 3091bd9
Show file tree
Hide file tree
Showing 3 changed files with 12 additions and 9 deletions.
15 changes: 9 additions & 6 deletions api/src/types.rs
Original file line number Diff line number Diff line change
Expand Up @@ -142,14 +142,15 @@ pub struct EncryptedRequest {
/// method
pub method: String,
/// id
pub id: u32,
#[serde(with = "secp_ser::string_or_u64")]
pub id: u64,
/// Body params, which includes nonce and encrypted request
pub params: EncryptedBody,
}

impl EncryptedRequest {
/// from json
pub fn from_json(id: u32, json_in: &Value, enc_key: &SecretKey) -> Result<Self, Error> {
pub fn from_json(id: u64, json_in: &Value, enc_key: &SecretKey) -> Result<Self, Error> {
Ok(EncryptedRequest {
jsonrpc: "2.0".to_owned(),
method: "encrypted_request_v3".to_owned(),
Expand Down Expand Up @@ -187,14 +188,15 @@ pub struct EncryptedResponse {
/// JSON RPC response
pub jsonrpc: String,
/// id
pub id: u32,
#[serde(with = "secp_ser::string_or_u64")]
pub id: u64,
/// result
pub result: HashMap<String, EncryptedBody>,
}

impl EncryptedResponse {
/// from json
pub fn from_json(id: u32, json_in: &Value, enc_key: &SecretKey) -> Result<Self, Error> {
pub fn from_json(id: u64, json_in: &Value, enc_key: &SecretKey) -> Result<Self, Error> {
let mut result_set = HashMap::new();
result_set.insert(
"Ok".to_string(),
Expand Down Expand Up @@ -245,14 +247,15 @@ pub struct EncryptionErrorResponse {
/// JSON RPC response
pub jsonrpc: String,
/// id
pub id: u32,
#[serde(with = "secp_ser::string_or_u64")]
pub id: u64,
/// error
pub error: EncryptionError,
}

impl EncryptionErrorResponse {
/// Create new response
pub fn new(id: u32, code: i32, message: &str) -> Self {
pub fn new(id: u64, code: i32, message: &str) -> Self {
EncryptionErrorResponse {
jsonrpc: "2.0".to_owned(),
id: id,
Expand Down
4 changes: 2 additions & 2 deletions controller/src/controller.rs
Original file line number Diff line number Diff line change
Expand Up @@ -475,7 +475,7 @@ impl OwnerV3Helpers {
pub fn decrypt_request(
key: Arc<Mutex<Option<SecretKey>>>,
req: &serde_json::Value,
) -> Result<(u32, serde_json::Value), serde_json::Value> {
) -> Result<(u64, serde_json::Value), serde_json::Value> {
let share_key_ref = key.lock();
let shared_key = share_key_ref.as_ref().unwrap();
let enc_req: EncryptedRequest = serde_json::from_value(req.clone()).map_err(|e| {
Expand All @@ -497,7 +497,7 @@ impl OwnerV3Helpers {
/// Encrypt a response
pub fn encrypt_response(
key: Arc<Mutex<Option<SecretKey>>>,
id: u32,
id: u64,
res: &serde_json::Value,
) -> Result<serde_json::Value, serde_json::Value> {
let share_key_ref = key.lock();
Expand Down
2 changes: 1 addition & 1 deletion tests/common/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -371,7 +371,7 @@ where

#[allow(dead_code)]
pub fn send_request_enc<OUT>(
sec_req_id: u32,
sec_req_id: u64,
internal_request_id: u32,
dest: &str,
req: &str,
Expand Down

0 comments on commit 3091bd9

Please sign in to comment.