rlp uses elastic array
This commit is contained in:
parent
182f356467
commit
8cf437b349
@ -19,6 +19,7 @@ rocksdb = "0.2.1"
|
|||||||
lazy_static = "0.1.*"
|
lazy_static = "0.1.*"
|
||||||
secp256k1 = "0.5.1"
|
secp256k1 = "0.5.1"
|
||||||
rust-crypto = "0.2.34"
|
rust-crypto = "0.2.34"
|
||||||
|
elastic-array = "0.4"
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
json-tests = { path = "json-tests" }
|
json-tests = { path = "json-tests" }
|
||||||
|
@ -42,6 +42,7 @@ extern crate time;
|
|||||||
extern crate crypto as rcrypto;
|
extern crate crypto as rcrypto;
|
||||||
extern crate secp256k1;
|
extern crate secp256k1;
|
||||||
extern crate arrayvec;
|
extern crate arrayvec;
|
||||||
|
extern crate elastic_array;
|
||||||
|
|
||||||
pub mod macros;
|
pub mod macros;
|
||||||
pub mod error;
|
pub mod error;
|
||||||
|
29
src/rlp.rs
29
src/rlp.rs
@ -33,6 +33,7 @@
|
|||||||
use std::fmt;
|
use std::fmt;
|
||||||
use std::cell::Cell;
|
use std::cell::Cell;
|
||||||
use std::error::Error as StdError;
|
use std::error::Error as StdError;
|
||||||
|
use elastic_array::*;
|
||||||
use bytes::{ToBytes, FromBytes, FromBytesError};
|
use bytes::{ToBytes, FromBytes, FromBytesError};
|
||||||
use vector::InsertSlice;
|
use vector::InsertSlice;
|
||||||
|
|
||||||
@ -758,7 +759,7 @@ impl Decoder for BasicDecoder {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug, Copy, Clone)]
|
||||||
struct ListInfo {
|
struct ListInfo {
|
||||||
position: usize,
|
position: usize,
|
||||||
current: usize,
|
current: usize,
|
||||||
@ -777,7 +778,7 @@ impl ListInfo {
|
|||||||
|
|
||||||
/// Appendable rlp encoder.
|
/// Appendable rlp encoder.
|
||||||
pub struct RlpStream {
|
pub struct RlpStream {
|
||||||
unfinished_lists: Vec<ListInfo>,
|
unfinished_lists: ElasticArray16<ListInfo>,
|
||||||
encoder: BasicEncoder,
|
encoder: BasicEncoder,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -785,7 +786,7 @@ impl RlpStream {
|
|||||||
/// Initializes instance of empty `RlpStream`.
|
/// Initializes instance of empty `RlpStream`.
|
||||||
pub fn new() -> RlpStream {
|
pub fn new() -> RlpStream {
|
||||||
RlpStream {
|
RlpStream {
|
||||||
unfinished_lists: vec![],
|
unfinished_lists: ElasticArray16::new(),
|
||||||
encoder: BasicEncoder::new(),
|
encoder: BasicEncoder::new(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -846,7 +847,7 @@ impl RlpStream {
|
|||||||
},
|
},
|
||||||
_ => {
|
_ => {
|
||||||
// reserve at least double size of the len
|
// reserve at least double size of the len
|
||||||
self.encoder.bytes.reserve(len * 2);
|
//self.encoder.bytes.reserve(len * 2);
|
||||||
self.unfinished_lists.push(ListInfo::new(position, len));
|
self.unfinished_lists.push(ListInfo::new(position, len));
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
@ -882,7 +883,7 @@ impl RlpStream {
|
|||||||
/// Appends raw (pre-serialised) RLP data. Use with caution. Chainable.
|
/// Appends raw (pre-serialised) RLP data. Use with caution. Chainable.
|
||||||
pub fn append_raw<'a>(&'a mut self, bytes: &[u8], item_count: usize) -> &'a mut RlpStream {
|
pub fn append_raw<'a>(&'a mut self, bytes: &[u8], item_count: usize) -> &'a mut RlpStream {
|
||||||
// push raw items
|
// push raw items
|
||||||
self.encoder.bytes.extend(bytes);
|
self.encoder.bytes.append_slice(bytes);
|
||||||
|
|
||||||
// try to finish and prepend the length
|
// try to finish and prepend the length
|
||||||
self.note_appended(item_count);
|
self.note_appended(item_count);
|
||||||
@ -937,7 +938,7 @@ impl RlpStream {
|
|||||||
/// panic! if stream is not finished.
|
/// panic! if stream is not finished.
|
||||||
pub fn out(self) -> Vec<u8> {
|
pub fn out(self) -> Vec<u8> {
|
||||||
match self.is_finished() {
|
match self.is_finished() {
|
||||||
true => self.encoder.out(),
|
true => self.encoder.out().to_vec(),
|
||||||
false => panic!()
|
false => panic!()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -985,7 +986,7 @@ pub fn encode<E>(object: &E) -> Vec<u8> where E: Encodable
|
|||||||
{
|
{
|
||||||
let mut encoder = BasicEncoder::new();
|
let mut encoder = BasicEncoder::new();
|
||||||
object.encode(&mut encoder);
|
object.encode(&mut encoder);
|
||||||
encoder.out()
|
encoder.out().to_vec()
|
||||||
}
|
}
|
||||||
|
|
||||||
pub trait Encodable {
|
pub trait Encodable {
|
||||||
@ -1038,12 +1039,12 @@ impl Encodable for Vec<u8> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
struct BasicEncoder {
|
struct BasicEncoder {
|
||||||
bytes: Vec<u8>,
|
bytes: ElasticArray1024<u8>,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl BasicEncoder {
|
impl BasicEncoder {
|
||||||
fn new() -> BasicEncoder {
|
fn new() -> BasicEncoder {
|
||||||
BasicEncoder { bytes: vec![] }
|
BasicEncoder { bytes: ElasticArray1024::new() }
|
||||||
}
|
}
|
||||||
|
|
||||||
/// inserts list prefix at given position
|
/// inserts list prefix at given position
|
||||||
@ -1062,7 +1063,7 @@ impl BasicEncoder {
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// get encoded value
|
/// get encoded value
|
||||||
fn out(self) -> Vec<u8> {
|
fn out(self) -> ElasticArray1024<u8> {
|
||||||
self.bytes
|
self.bytes
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -1073,17 +1074,17 @@ impl Encoder for BasicEncoder {
|
|||||||
// just 0
|
// just 0
|
||||||
0 => self.bytes.push(0x80u8),
|
0 => self.bytes.push(0x80u8),
|
||||||
// byte is its own encoding
|
// byte is its own encoding
|
||||||
1 if bytes[0] < 0x80 => self.bytes.extend(bytes),
|
1 if bytes[0] < 0x80 => self.bytes.append_slice(bytes),
|
||||||
// (prefix + length), followed by the string
|
// (prefix + length), followed by the string
|
||||||
len @ 1 ... 55 => {
|
len @ 1 ... 55 => {
|
||||||
self.bytes.push(0x80u8 + len as u8);
|
self.bytes.push(0x80u8 + len as u8);
|
||||||
self.bytes.extend(bytes);
|
self.bytes.append_slice(bytes);
|
||||||
}
|
}
|
||||||
// (prefix + length of length), followed by the length, followd by the string
|
// (prefix + length of length), followed by the length, followd by the string
|
||||||
len => {
|
len => {
|
||||||
self.bytes.push(0xb7 + len.to_bytes_len() as u8);
|
self.bytes.push(0xb7 + len.to_bytes_len() as u8);
|
||||||
self.bytes.extend(len.to_bytes());
|
self.bytes.append_slice(&len.to_bytes());
|
||||||
self.bytes.extend(bytes);
|
self.bytes.append_slice(bytes);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user