#![allow(clippy::missing_safety_doc)]
use std::cmp::max;
use std::collections::TryReserveError;
use std::ffi::CStr;
use std::os::raw::c_char;
use std::str::Utf8Error;
const INIT_SIZE: usize = 4096;
#[derive(Debug, PartialEq, Eq)]
pub enum JsonError {
InvalidState,
Utf8Error(Utf8Error),
Memory,
}
impl std::error::Error for JsonError {}
impl std::fmt::Display for JsonError {
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
match self {
JsonError::InvalidState => write!(f, "invalid state"),
JsonError::Utf8Error(ref e) => e.fmt(f),
JsonError::Memory => write!(f, "memory error"),
}
}
}
impl From<TryReserveError> for JsonError {
fn from(_: TryReserveError) -> Self {
JsonError::Memory
}
}
impl From<Utf8Error> for JsonError {
fn from(e: Utf8Error) -> Self {
JsonError::Utf8Error(e)
}
}
#[derive(Clone, Debug)]
enum Type {
Object,
Array,
}
#[derive(Debug, Clone, Copy, PartialEq)]
#[repr(C)]
enum State {
None = 0,
ObjectFirst,
ObjectNth,
ArrayFirst,
ArrayNth,
}
impl State {
fn from_u64(v: u64) -> Result<State, JsonError> {
let s = match v {
0 => State::None,
1 => State::ObjectFirst,
2 => State::ObjectNth,
3 => State::ArrayFirst,
4 => State::ArrayNth,
_ => {
return Err(JsonError::InvalidState);
}
};
Ok(s)
}
}
#[repr(C)]
pub struct JsonBuilderMark {
position: u64,
state_index: u64,
state: u64,
}
#[derive(Debug, Clone)]
pub struct JsonBuilder {
buf: String,
state: Vec<State>,
init_type: Type,
}
impl JsonBuilder {
pub fn try_new_object() -> Result<Self, JsonError> {
Self::try_new_object_with_capacity(INIT_SIZE)
}
pub fn try_new_object_with_capacity(capacity: usize) -> Result<Self, JsonError> {
let mut buf = String::new();
buf.try_reserve(capacity)?;
buf.push('{');
let mut state = Vec::new();
state.try_reserve(32)?;
state.extend_from_slice(&[State::None, State::ObjectFirst]);
Ok(Self {
buf,
state,
init_type: Type::Object,
})
}
pub fn try_new_array() -> Result<Self, JsonError> {
Self::try_new_array_with_capacity(INIT_SIZE)
}
pub fn try_new_array_with_capacity(capacity: usize) -> Result<Self, JsonError> {
let mut buf = String::new();
buf.try_reserve(capacity)?;
buf.push('[');
let mut state = Vec::new();
state.try_reserve(32)?;
state.extend_from_slice(&[State::None, State::ArrayFirst]);
Ok(Self {
buf,
state,
init_type: Type::Array,
})
}
pub fn push(&mut self, ch: char) -> Result<&mut Self, JsonError> {
if self.buf.capacity() == self.buf.len() {
self.buf.try_reserve(INIT_SIZE)?;
}
self.buf.push(ch);
Ok(self)
}
pub fn push_str(&mut self, s: &str) -> Result<&mut Self, JsonError> {
if self.buf.capacity() < self.buf.len() + s.len() {
self.buf.try_reserve(max(INIT_SIZE, s.len()))?;
}
self.buf.push_str(s);
Ok(self)
}
pub fn reset(&mut self) {
self.buf.truncate(0);
self.state.clear();
match self.init_type {
Type::Array => {
self.buf.push('[');
self.state
.extend_from_slice(&[State::None, State::ArrayFirst]);
}
Type::Object => {
self.buf.push('{');
self.state
.extend_from_slice(&[State::None, State::ObjectFirst]);
}
}
}
pub fn close(&mut self) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectFirst | State::ObjectNth => {
self.push('}')?;
self.pop_state();
Ok(self)
}
State::ArrayFirst | State::ArrayNth => {
self.push(']')?;
self.pop_state();
Ok(self)
}
State::None => {
debug_validate_fail!("invalid state");
Err(JsonError::InvalidState)
}
}
}
fn current_state(&self) -> State {
if self.state.is_empty() {
State::None
} else {
self.state[self.state.len() - 1]
}
}
fn push_state(&mut self, state: State) -> Result<(), JsonError> {
if self.state.len() == self.state.capacity() {
self.state.try_reserve(32)?;
}
self.state.push(state);
Ok(())
}
fn pop_state(&mut self) {
self.state.pop();
}
fn set_state(&mut self, state: State) {
let n = self.state.len() - 1;
self.state[n] = state;
}
pub fn get_mark(&self) -> JsonBuilderMark {
JsonBuilderMark {
position: self.buf.len() as u64,
state: self.current_state() as u64,
state_index: self.state.len() as u64,
}
}
pub fn restore_mark(&mut self, mark: &JsonBuilderMark) -> Result<(), JsonError> {
let state = State::from_u64(mark.state)?;
if mark.position < (self.buf.len() as u64) && mark.state_index < (self.state.len() as u64) {
self.buf.truncate(mark.position as usize);
self.state.truncate(mark.state_index as usize);
self.state[(mark.state_index as usize) - 1] = state;
}
Ok(())
}
pub fn open_object(&mut self, key: &str) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectFirst => {
self.push('"')?;
self.set_state(State::ObjectNth);
}
State::ObjectNth => {
self.push_str(",\"")?;
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push_str(key)?;
self.push_str("\":{")?;
self.push_state(State::ObjectFirst)?;
Ok(self)
}
pub fn start_object(&mut self) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ArrayFirst => {}
State::ArrayNth => {
self.push(',')?;
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('{')?;
self.set_state(State::ArrayNth);
self.push_state(State::ObjectFirst)?;
Ok(self)
}
pub fn open_array(&mut self, key: &str) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectFirst => {}
State::ObjectNth => {
self.push(',')?;
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":[")?;
self.set_state(State::ObjectNth);
self.push_state(State::ArrayFirst)?;
Ok(self)
}
pub fn append_string(&mut self, val: &str) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ArrayFirst => {
self.encode_string(val)?;
self.set_state(State::ArrayNth);
Ok(self)
}
State::ArrayNth => {
self.push(',')?;
self.encode_string(val)?;
Ok(self)
}
_ => {
debug_validate_fail!("invalid state");
Err(JsonError::InvalidState)
}
}
}
pub fn append_string_from_bytes(&mut self, val: &[u8]) -> Result<&mut Self, JsonError> {
match std::str::from_utf8(val) {
Ok(s) => self.append_string(s),
Err(_) => self.append_string(&try_string_from_bytes(val)?),
}
}
pub fn append_base64(&mut self, val: &[u8]) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ArrayFirst => {
self.push('"')?;
self.encode_base64(val)?;
self.push('"')?;
self.set_state(State::ArrayNth);
Ok(self)
}
State::ArrayNth => {
self.push(',')?;
self.push('"')?;
self.encode_base64(val)?;
self.push('"')?;
Ok(self)
}
_ => {
debug_validate_fail!("invalid state");
Err(JsonError::InvalidState)
}
}
}
pub fn append_hex(&mut self, val: &[u8]) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ArrayFirst => {
self.push('"')?;
for i in 0..val.len() {
self.push(HEX[(val[i] >> 4) as usize] as char)?;
self.push(HEX[(val[i] & 0xf) as usize] as char)?;
}
self.push('"')?;
self.set_state(State::ArrayNth);
Ok(self)
}
State::ArrayNth => {
self.push(',')?;
self.push('"')?;
for i in 0..val.len() {
self.push(HEX[(val[i] >> 4) as usize] as char)?;
self.push(HEX[(val[i] & 0xf) as usize] as char)?;
}
self.push('"')?;
Ok(self)
}
_ => {
debug_validate_fail!("invalid state");
Err(JsonError::InvalidState)
}
}
}
pub fn append_uint(&mut self, val: u64) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ArrayFirst => {
self.set_state(State::ArrayNth);
}
State::ArrayNth => {
self.push(',')?;
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push_str(&val.to_string())
}
pub fn append_float(&mut self, val: f64) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ArrayFirst => {
self.set_state(State::ArrayNth);
}
State::ArrayNth => {
self.push(',')?;
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push_float(val)?;
Ok(self)
}
pub fn set_object(&mut self, key: &str, js: &JsonBuilder) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":")?;
self.push_str(&js.buf)?;
Ok(self)
}
pub fn append_object(&mut self, js: &JsonBuilder) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ArrayFirst => {
self.set_state(State::ArrayNth);
}
State::ArrayNth => {
self.push(',')?;
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push_str(&js.buf)?;
Ok(self)
}
#[inline(always)]
pub fn set_string(&mut self, key: &str, val: &str) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":")?;
self.encode_string(val)?;
Ok(self)
}
pub fn set_formatted(&mut self, formatted: &str) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push_str(formatted)?;
Ok(self)
}
pub fn set_string_from_bytes(&mut self, key: &str, val: &[u8]) -> Result<&mut Self, JsonError> {
match std::str::from_utf8(val) {
Ok(s) => self.set_string(key, s),
Err(_) => self.set_string(key, &try_string_from_bytes(val)?),
}
}
pub fn set_string_from_bytes_limited(&mut self, key: &str, val: &[u8], limit: usize) -> Result<&mut Self, JsonError> {
let mut valtrunc = Vec::new();
let val = if val.len() > limit {
valtrunc.extend_from_slice(&val[..limit]);
valtrunc.extend_from_slice(b"[truncated]");
&valtrunc
} else {
val
};
match std::str::from_utf8(val) {
Ok(s) => self.set_string(key, s),
Err(_) => self.set_string(key, &try_string_from_bytes(val)?),
}
}
pub fn set_base64(&mut self, key: &str, val: &[u8]) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":\"")?;
self.encode_base64(val)?;
self.push('"')?;
Ok(self)
}
pub fn set_hex(&mut self, key: &str, val: &[u8]) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":\"")?;
for i in 0..val.len() {
self.push(HEX[(val[i] >> 4) as usize] as char)?;
self.push(HEX[(val[i] & 0xf) as usize] as char)?;
}
self.push('"')?;
Ok(self)
}
pub fn set_uint(&mut self, key: &str, val: u64) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":")?;
self.push_str(&val.to_string())?;
Ok(self)
}
pub fn set_int(&mut self, key: &str, val: i64) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":")?;
self.push_str(&val.to_string())?;
Ok(self)
}
pub fn set_float(&mut self, key: &str, val: f64) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
self.push_str("\":")?;
self.push_float(val)?;
Ok(self)
}
pub fn set_bool(&mut self, key: &str, val: bool) -> Result<&mut Self, JsonError> {
match self.current_state() {
State::ObjectNth => {
self.push(',')?;
}
State::ObjectFirst => {
self.set_state(State::ObjectNth);
}
_ => {
debug_validate_fail!("invalid state");
return Err(JsonError::InvalidState);
}
}
self.push('"')?;
self.push_str(key)?;
if val {
self.push_str("\":true")?;
} else {
self.push_str("\":false")?;
}
Ok(self)
}
pub fn capacity(&self) -> usize {
self.buf.capacity()
}
fn push_float(&mut self, val: f64) -> Result<(), JsonError> {
if val.is_nan() || val.is_infinite() {
self.push_str("null")?;
} else {
self.push_str(&val.to_string())?;
}
Ok(())
}
#[inline(always)]
fn encode_string(&mut self, val: &str) -> Result<(), JsonError> {
let mut buf = Vec::new();
buf.try_reserve(val.len() * 2 + 2)?;
buf.resize(val.len() * 2 + 2, 0);
let mut offset = 0;
let bytes = val.as_bytes();
buf[offset] = b'"';
offset += 1;
for &x in bytes.iter() {
if offset + 7 >= buf.capacity() {
buf.try_reserve(buf.capacity())?;
buf.resize(buf.capacity(), 0);
}
let escape = ESCAPED[x as usize];
if escape == 0 {
buf[offset] = x;
offset += 1;
} else if escape == b'u' {
buf[offset] = b'\\';
offset += 1;
buf[offset] = b'u';
offset += 1;
buf[offset] = b'0';
offset += 1;
buf[offset] = b'0';
offset += 1;
buf[offset] = HEX[(x >> 4 & 0xf) as usize];
offset += 1;
buf[offset] = HEX[(x & 0xf) as usize];
offset += 1;
} else {
buf[offset] = b'\\';
offset += 1;
buf[offset] = escape;
offset += 1;
}
}
buf[offset] = b'"';
offset += 1;
match std::str::from_utf8(&buf[0..offset]) {
Ok(s) => {
self.push_str(s)?;
}
Err(err) => {
let error = format!(
"\"UTF8-ERROR: what=[escaped string] error={} output={:02x?} input={:02x?}\"",
err,
&buf[0..offset],
val.as_bytes(),
);
self.push_str(&error)?;
}
}
Ok(())
}
fn encode_base64(&mut self, val: &[u8]) -> Result<&mut Self, JsonError> {
let encoded_len = 4 * ((val.len() + 2) / 3);
if self.buf.capacity() < self.buf.len() + encoded_len {
self.buf.try_reserve(encoded_len)?;
}
base64::encode_config_buf(val, base64::STANDARD, &mut self.buf);
Ok(self)
}
}
fn try_string_from_bytes(input: &[u8]) -> Result<String, JsonError> {
let mut out = String::new();
out.try_reserve(input.len() * 4)?;
for b in input.iter() {
if *b < 128 {
out.push(*b as char);
} else {
out.push_str(&format!("\\x{:02x}", *b));
}
}
return Ok(out);
}
#[no_mangle]
pub extern "C" fn jb_new_object() -> *mut JsonBuilder {
match JsonBuilder::try_new_object() {
Ok(js) => {
let boxed = Box::new(js);
Box::into_raw(boxed)
}
Err(_) => std::ptr::null_mut(),
}
}
#[no_mangle]
pub extern "C" fn jb_new_array() -> *mut JsonBuilder {
match JsonBuilder::try_new_array() {
Ok(js) => {
let boxed = Box::new(js);
Box::into_raw(boxed)
}
Err(_) => std::ptr::null_mut(),
}
}
#[no_mangle]
pub extern "C" fn jb_clone(js: &mut JsonBuilder) -> *mut JsonBuilder {
let clone = Box::new(js.clone());
Box::into_raw(clone)
}
#[no_mangle]
pub unsafe extern "C" fn jb_free(js: &mut JsonBuilder) {
let _ = Box::from_raw(js);
}
#[no_mangle]
pub extern "C" fn jb_capacity(jb: &mut JsonBuilder) -> usize {
jb.capacity()
}
#[no_mangle]
pub extern "C" fn jb_reset(jb: &mut JsonBuilder) {
jb.reset();
}
#[no_mangle]
pub unsafe extern "C" fn jb_open_object(js: &mut JsonBuilder, key: *const c_char) -> bool {
if let Ok(s) = CStr::from_ptr(key).to_str() {
js.open_object(s).is_ok()
} else {
false
}
}
#[no_mangle]
pub unsafe extern "C" fn jb_start_object(js: &mut JsonBuilder) -> bool {
js.start_object().is_ok()
}
#[no_mangle]
pub unsafe extern "C" fn jb_open_array(js: &mut JsonBuilder, key: *const c_char) -> bool {
if let Ok(s) = CStr::from_ptr(key).to_str() {
js.open_array(s).is_ok()
} else {
false
}
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_string(
js: &mut JsonBuilder, key: *const c_char, val: *const c_char,
) -> bool {
if val.is_null() {
return false;
}
if let Ok(key) = CStr::from_ptr(key).to_str() {
if let Ok(val) = CStr::from_ptr(val).to_str() {
return js.set_string(key, val).is_ok();
}
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_string_from_bytes(
js: &mut JsonBuilder, key: *const c_char, bytes: *const u8, len: u32,
) -> bool {
if bytes.is_null() || len == 0 {
return false;
}
if let Ok(key) = CStr::from_ptr(key).to_str() {
let val = std::slice::from_raw_parts(bytes, len as usize);
return js.set_string_from_bytes(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_base64(
js: &mut JsonBuilder, key: *const c_char, bytes: *const u8, len: u32,
) -> bool {
if bytes.is_null() || len == 0 {
return false;
}
if let Ok(key) = CStr::from_ptr(key).to_str() {
let val = std::slice::from_raw_parts(bytes, len as usize);
return js.set_base64(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_hex(
js: &mut JsonBuilder, key: *const c_char, bytes: *const u8, len: u32,
) -> bool {
if bytes.is_null() || len == 0 {
return false;
}
if let Ok(key) = CStr::from_ptr(key).to_str() {
let val = std::slice::from_raw_parts(bytes, len as usize);
return js.set_hex(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_formatted(js: &mut JsonBuilder, formatted: *const c_char) -> bool {
if let Ok(formatted) = CStr::from_ptr(formatted).to_str() {
return js.set_formatted(formatted).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_append_object(jb: &mut JsonBuilder, obj: &JsonBuilder) -> bool {
jb.append_object(obj).is_ok()
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_object(
js: &mut JsonBuilder, key: *const c_char, val: &mut JsonBuilder,
) -> bool {
if let Ok(key) = CStr::from_ptr(key).to_str() {
return js.set_object(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_append_string(js: &mut JsonBuilder, val: *const c_char) -> bool {
if val.is_null() {
return false;
}
if let Ok(val) = CStr::from_ptr(val).to_str() {
return js.append_string(val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_append_string_from_bytes(
js: &mut JsonBuilder, bytes: *const u8, len: u32,
) -> bool {
if bytes.is_null() || len == 0 {
return false;
}
let val = std::slice::from_raw_parts(bytes, len as usize);
return js.append_string_from_bytes(val).is_ok();
}
#[no_mangle]
pub unsafe extern "C" fn jb_append_base64(
js: &mut JsonBuilder, bytes: *const u8, len: u32,
) -> bool {
if bytes.is_null() || len == 0 {
return false;
}
let val = std::slice::from_raw_parts(bytes, len as usize);
return js.append_base64(val).is_ok();
}
#[no_mangle]
pub unsafe extern "C" fn jb_append_uint(js: &mut JsonBuilder, val: u64) -> bool {
return js.append_uint(val).is_ok();
}
#[no_mangle]
pub unsafe extern "C" fn jb_append_float(js: &mut JsonBuilder, val: f64) -> bool {
return js.append_float(val).is_ok();
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_uint(js: &mut JsonBuilder, key: *const c_char, val: u64) -> bool {
if let Ok(key) = CStr::from_ptr(key).to_str() {
return js.set_uint(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_int(js: &mut JsonBuilder, key: *const c_char, val: i64) -> bool {
if let Ok(key) = CStr::from_ptr(key).to_str() {
return js.set_int(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_float(js: &mut JsonBuilder, key: *const c_char, val: f64) -> bool {
if let Ok(key) = CStr::from_ptr(key).to_str() {
return js.set_float(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_set_bool(js: &mut JsonBuilder, key: *const c_char, val: bool) -> bool {
if let Ok(key) = CStr::from_ptr(key).to_str() {
return js.set_bool(key, val).is_ok();
}
return false;
}
#[no_mangle]
pub unsafe extern "C" fn jb_close(js: &mut JsonBuilder) -> bool {
js.close().is_ok()
}
#[no_mangle]
pub unsafe extern "C" fn jb_len(js: &JsonBuilder) -> usize {
js.buf.len()
}
#[no_mangle]
pub unsafe extern "C" fn jb_ptr(js: &mut JsonBuilder) -> *const u8 {
js.buf.as_ptr()
}
#[no_mangle]
pub unsafe extern "C" fn jb_get_mark(js: &mut JsonBuilder, mark: &mut JsonBuilderMark) {
let m = js.get_mark();
mark.position = m.position;
mark.state_index = m.state_index;
mark.state = m.state;
}
#[no_mangle]
pub unsafe extern "C" fn jb_restore_mark(js: &mut JsonBuilder, mark: &mut JsonBuilderMark) -> bool {
js.restore_mark(mark).is_ok()
}
#[cfg(test)]
mod test {
use super::*;
#[test]
fn test_try_reserve() {
let mut buf = String::new();
assert_eq!(buf.len(), 0);
assert_eq!(buf.capacity(), 0);
buf.try_reserve(1).unwrap();
assert_eq!(buf.len(), 0);
assert!(buf.capacity() >= 1);
}
#[test]
fn test_set_bool() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_bool("first", true).unwrap();
assert_eq!(jb.buf, r#"{"first":true"#);
jb.set_bool("second", false).unwrap();
assert_eq!(jb.buf, r#"{"first":true,"second":false"#);
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_bool("first", false).unwrap();
assert_eq!(jb.buf, r#"{"first":false"#);
jb.set_bool("second", true).unwrap();
assert_eq!(jb.buf, r#"{"first":false,"second":true"#);
}
#[test]
fn test_object_in_object() -> Result<(), JsonError> {
let mut js = JsonBuilder::try_new_object().unwrap();
js.open_object("object")?;
assert_eq!(js.current_state(), State::ObjectFirst);
assert_eq!(js.buf, r#"{"object":{"#);
js.set_string("one", "one")?;
assert_eq!(js.current_state(), State::ObjectNth);
assert_eq!(js.buf, r#"{"object":{"one":"one""#);
js.close()?;
assert_eq!(js.current_state(), State::ObjectNth);
assert_eq!(js.buf, r#"{"object":{"one":"one"}"#);
js.close()?;
assert_eq!(js.current_state(), State::None);
assert_eq!(js.buf, r#"{"object":{"one":"one"}}"#);
Ok(())
}
#[test]
fn test_empty_array_in_object() -> Result<(), JsonError> {
let mut js = JsonBuilder::try_new_object().unwrap();
js.open_array("array")?;
assert_eq!(js.current_state(), State::ArrayFirst);
js.close()?;
assert_eq!(js.current_state(), State::ObjectNth);
assert_eq!(js.buf, r#"{"array":[]"#);
js.close()?;
assert_eq!(js.buf, r#"{"array":[]}"#);
Ok(())
}
#[test]
#[cfg(not(feature = "debug-validate"))]
fn test_array_in_object() -> Result<(), JsonError> {
let mut js = JsonBuilder::try_new_object().unwrap();
assert_eq!(
js.append_string("will fail").err().unwrap(),
JsonError::InvalidState
);
js.open_array("array")?;
assert_eq!(js.current_state(), State::ArrayFirst);
js.append_string("one")?;
assert_eq!(js.current_state(), State::ArrayNth);
assert_eq!(js.buf, r#"{"array":["one""#);
js.append_string("two")?;
assert_eq!(js.current_state(), State::ArrayNth);
assert_eq!(js.buf, r#"{"array":["one","two""#);
js.append_uint(3)?;
assert_eq!(js.current_state(), State::ArrayNth);
assert_eq!(js.buf, r#"{"array":["one","two",3"#);
js.close()?;
assert_eq!(js.current_state(), State::ObjectNth);
assert_eq!(js.buf, r#"{"array":["one","two",3]"#);
js.close()?;
assert_eq!(js.current_state(), State::None);
assert_eq!(js.buf, r#"{"array":["one","two",3]}"#);
Ok(())
}
#[test]
fn basic_test() -> Result<(), JsonError> {
let mut js = JsonBuilder::try_new_object().unwrap();
assert_eq!(js.current_state(), State::ObjectFirst);
assert_eq!(js.buf, "{");
js.set_string("one", "one")?;
assert_eq!(js.current_state(), State::ObjectNth);
assert_eq!(js.buf, r#"{"one":"one""#);
js.set_string("two", "two")?;
assert_eq!(js.current_state(), State::ObjectNth);
assert_eq!(js.buf, r#"{"one":"one","two":"two""#);
js.close()?;
assert_eq!(js.current_state(), State::None);
assert_eq!(js.buf, r#"{"one":"one","two":"two"}"#);
Ok(())
}
#[test]
fn test_combine() -> Result<(), JsonError> {
let mut main = JsonBuilder::try_new_object().unwrap();
let mut obj = JsonBuilder::try_new_object().unwrap();
obj.close()?;
let mut array = JsonBuilder::try_new_array().unwrap();
array.append_string("one")?;
array.append_uint(2)?;
array.close()?;
main.set_object("object", &obj)?;
main.set_object("array", &array)?;
main.close()?;
assert_eq!(main.buf, r#"{"object":{},"array":["one",2]}"#);
Ok(())
}
#[test]
fn test_objects_in_array() -> Result<(), JsonError> {
let mut js = JsonBuilder::try_new_array()?;
assert_eq!(js.buf, r#"["#);
js.start_object()?;
assert_eq!(js.buf, r#"[{"#);
js.set_string("uid", "0")?;
assert_eq!(js.buf, r#"[{"uid":"0""#);
js.close()?;
assert_eq!(js.buf, r#"[{"uid":"0"}"#);
js.start_object()?;
assert_eq!(js.buf, r#"[{"uid":"0"},{"#);
js.set_string("username", "root")?;
assert_eq!(js.buf, r#"[{"uid":"0"},{"username":"root""#);
js.close()?;
assert_eq!(js.buf, r#"[{"uid":"0"},{"username":"root"}"#);
js.close()?;
assert_eq!(js.buf, r#"[{"uid":"0"},{"username":"root"}]"#);
Ok(())
}
#[test]
fn test_grow() -> Result<(), JsonError> {
let mut jb = JsonBuilder::try_new_object_with_capacity(1).unwrap();
assert!(jb.capacity() > 0);
let capacity = jb.capacity();
let mut value = String::new();
for i in 0..capacity {
value.push_str((i % 10).to_string().as_str());
}
jb.set_string("foo", &value)?;
assert!(jb.capacity() > capacity);
Ok(())
}
#[test]
fn test_reset() -> Result<(), JsonError> {
let mut jb = JsonBuilder::try_new_object().unwrap();
assert_eq!(jb.buf, "{");
jb.set_string("foo", "bar")?;
let cap = jb.capacity();
jb.reset();
assert_eq!(jb.buf, "{");
assert_eq!(jb.capacity(), cap);
Ok(())
}
#[test]
fn test_append_string_from_bytes() -> Result<(), JsonError> {
let mut jb = JsonBuilder::try_new_array().unwrap();
let s = &[0x41, 0x41, 0x41, 0x00];
jb.append_string_from_bytes(s)?;
assert_eq!(jb.buf, r#"["AAA\u0000""#);
let s = &[0x00, 0x01, 0x02, 0x03];
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_string_from_bytes(s)?;
assert_eq!(jb.buf, r#"["\u0000\u0001\u0002\u0003""#);
Ok(())
}
#[test]
fn test_set_string_from_bytes() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_string_from_bytes("first", &[]).unwrap();
assert_eq!(jb.buf, r#"{"first":"""#);
jb.set_string_from_bytes("second", &[]).unwrap();
assert_eq!(jb.buf, r#"{"first":"","second":"""#);
}
#[test]
fn test_append_string_from_bytes_grow() -> Result<(), JsonError> {
let s = &[0x00, 0x01, 0x02, 0x03];
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_string_from_bytes(s)?;
for i in 1..1000 {
let mut s = Vec::new();
s.resize(i, 0x41);
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_string_from_bytes(&s)?;
}
Ok(())
}
#[test]
fn test_invalid_utf8() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_string_from_bytes("invalid", &[0xf0, 0xf1, 0xf2])
.unwrap();
assert_eq!(jb.buf, r#"{"invalid":"\\xf0\\xf1\\xf2""#);
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_string_from_bytes(&[0xf0, 0xf1, 0xf2]).unwrap();
assert_eq!(jb.buf, r#"["\\xf0\\xf1\\xf2""#);
}
#[test]
fn test_marks() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_string("foo", "bar").unwrap();
assert_eq!(jb.buf, r#"{"foo":"bar""#);
assert_eq!(jb.current_state(), State::ObjectNth);
assert_eq!(jb.state.len(), 2);
let mark = jb.get_mark();
jb.open_array("bar").unwrap();
jb.start_object().unwrap();
assert_eq!(jb.buf, r#"{"foo":"bar","bar":[{"#);
assert_eq!(jb.current_state(), State::ObjectFirst);
assert_eq!(jb.state.len(), 4);
jb.restore_mark(&mark).unwrap();
assert_eq!(jb.buf, r#"{"foo":"bar""#);
assert_eq!(jb.current_state(), State::ObjectNth);
assert_eq!(jb.state.len(), 2);
}
#[test]
fn test_set_formatted() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_formatted("\"foo\":\"bar\"").unwrap();
assert_eq!(jb.buf, r#"{"foo":"bar""#);
jb.set_formatted("\"bar\":\"foo\"").unwrap();
assert_eq!(jb.buf, r#"{"foo":"bar","bar":"foo""#);
jb.close().unwrap();
assert_eq!(jb.buf, r#"{"foo":"bar","bar":"foo"}"#);
}
#[test]
fn test_set_float() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_float("one", 1.1).unwrap();
jb.set_float("two", 2.2).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"{"one":1.1,"two":2.2}"#);
}
#[test]
fn test_append_float() {
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_float(1.1).unwrap();
jb.append_float(2.2).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"[1.1,2.2]"#);
}
#[test]
fn test_set_nan() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_float("nan", f64::NAN).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"{"nan":null}"#);
}
#[test]
fn test_append_nan() {
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_float(f64::NAN).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"[null]"#);
}
#[test]
fn test_set_inf() {
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_float("inf", f64::INFINITY).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"{"inf":null}"#);
let mut jb = JsonBuilder::try_new_object().unwrap();
jb.set_float("inf", f64::NEG_INFINITY).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"{"inf":null}"#);
}
#[test]
fn test_append_inf() {
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_float(f64::INFINITY).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"[null]"#);
let mut jb = JsonBuilder::try_new_array().unwrap();
jb.append_float(f64::NEG_INFINITY).unwrap();
jb.close().unwrap();
assert_eq!(jb.buf, r#"[null]"#);
}
}
const QU: u8 = b'"';
const BS: u8 = b'\\';
const BB: u8 = b'b';
const TT: u8 = b't';
const NN: u8 = b'n';
const FF: u8 = b'f';
const RR: u8 = b'r';
const UU: u8 = b'u';
const __: u8 = 0;
static ESCAPED: [u8; 256] = [
UU, UU, UU, UU, UU, UU, UU, UU, BB, TT, NN, UU, FF, RR, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, UU, __, __, QU, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, BS, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, __, ];
pub static HEX: [u8; 16] = [
b'0', b'1', b'2', b'3', b'4', b'5', b'6', b'7', b'8', b'9', b'a', b'b', b'c', b'd', b'e', b'f',
];