git.delta.rocks / jrsonnet / refs/commits / 0d5cefbba5bd

difftreelog

source

bindings/jsonnet/src/lib.rs7.4 KiBsourcehistory
1#[cfg(feature = "interop")]2pub mod interop;34pub mod import;5pub mod native;6pub mod val_extract;7pub mod val_make;8pub mod val_modify;9pub mod vars_tlas;1011use std::{12	alloc::Layout,13	ffi::{CStr, CString},14	os::raw::{c_char, c_double, c_int, c_uint},15	path::PathBuf,16};1718use import::NativeImportResolver;19use jrsonnet_evaluator::{EvaluationState, IStr, ManifestFormat, Val};2021/// WASM stub22#[cfg(target_arch = "wasm32")]23#[no_mangle]24pub extern "C" fn _start() {}2526#[no_mangle]27pub extern "C" fn jsonnet_version() -> &'static [u8; 8] {28	b"v0.16.0\0"29}3031#[no_mangle]32pub extern "C" fn jsonnet_make() -> *mut EvaluationState {33	let state = EvaluationState::default();34	state.with_stdlib();35	state.settings_mut().import_resolver = Box::new(NativeImportResolver::default());36	Box::into_raw(Box::new(state))37}3839/// # Safety40#[no_mangle]41#[allow(clippy::boxed_local)]42pub unsafe extern "C" fn jsonnet_destroy(vm: *mut EvaluationState) {43	Box::from_raw(vm);44}4546#[no_mangle]47pub extern "C" fn jsonnet_max_stack(vm: &EvaluationState, v: c_uint) {48	vm.settings_mut().max_stack = v as usize;49}5051// jrsonnet currently have no GC, so these functions is no-op52#[no_mangle]53pub extern "C" fn jsonnet_gc_min_objects(_vm: &EvaluationState, _v: c_uint) {}54#[no_mangle]55pub extern "C" fn jsonnet_gc_growth_trigger(_vm: &EvaluationState, _v: c_double) {}5657#[no_mangle]58pub extern "C" fn jsonnet_string_output(vm: &EvaluationState, v: c_int) {59	match v {60		1 => vm.set_manifest_format(ManifestFormat::String),61		0 => vm.set_manifest_format(ManifestFormat::Json {62			padding: 4,63			#[cfg(feature = "exp-preserve-order")]64			preserve_order: false,65		}),66		_ => panic!("incorrect output format"),67	}68}6970/// # Safety71///72/// This function is most definitely broken, but it works somehow, see TODO inside73#[no_mangle]74pub unsafe extern "C" fn jsonnet_realloc(75	_vm: &EvaluationState,76	buf: *mut u8,77	sz: usize,78) -> *mut u8 {79	if buf.is_null() {80		assert!(sz != 0);81		return std::alloc::alloc(Layout::from_size_align(sz, std::mem::align_of::<u8>()).unwrap());82	}83	// TODO: Somehow store size of allocation, because its real size is probally not 16 :D84	// OR (Alternative way of fixing this TODO)85	// TODO: Standard allocator uses malloc, and it doesn't uses allocation size,86	// TODO: so it should work in normal cases. Maybe force allocator for this library?87	let old_layout = Layout::from_size_align(16, std::mem::align_of::<u8>()).unwrap();88	if sz == 0 {89		std::alloc::dealloc(buf, old_layout);90		return std::ptr::null_mut();91	}92	std::alloc::realloc(buf, old_layout, sz)93}9495/// # Safety96#[no_mangle]97#[allow(clippy::boxed_local)]98pub unsafe extern "C" fn jsonnet_json_destroy(_vm: &EvaluationState, v: *mut Val) {99	Box::from_raw(v);100}101102#[no_mangle]103pub extern "C" fn jsonnet_max_trace(vm: &EvaluationState, v: c_uint) {104	vm.set_max_trace(v as usize)105}106107/// # Safety108///109/// This function is safe, if received v is a pointer to normal C string110#[no_mangle]111pub unsafe extern "C" fn jsonnet_evaluate_file(112	vm: &EvaluationState,113	filename: *const c_char,114	error: &mut c_int,115) -> *const c_char {116	vm.run_in_state(|| {117		let filename = CStr::from_ptr(filename);118		match vm119			.evaluate_file_raw_nocwd(&PathBuf::from(filename.to_str().unwrap()))120			.and_then(|v| vm.with_tla(v))121			.and_then(|v| vm.manifest(v))122		{123			Ok(v) => {124				*error = 0;125				CString::new(&*v as &str).unwrap().into_raw()126			}127			Err(e) => {128				*error = 1;129				let out = vm.stringify_err(&e);130				CString::new(&out as &str).unwrap().into_raw()131			}132		}133	})134}135136/// # Safety137///138/// This function is safe, if received v is a pointer to normal C string139#[no_mangle]140pub unsafe extern "C" fn jsonnet_evaluate_snippet(141	vm: &EvaluationState,142	filename: *const c_char,143	snippet: *const c_char,144	error: &mut c_int,145) -> *const c_char {146	vm.run_in_state(|| {147		let filename = CStr::from_ptr(filename);148		let snippet = CStr::from_ptr(snippet);149		match vm150			.evaluate_snippet_raw(151				PathBuf::from(filename.to_str().unwrap()).into(),152				snippet.to_str().unwrap().into(),153			)154			.and_then(|v| vm.with_tla(v))155			.and_then(|v| vm.manifest(v))156		{157			Ok(v) => {158				*error = 0;159				CString::new(&*v as &str).unwrap().into_raw()160			}161			Err(e) => {162				*error = 1;163				let out = vm.stringify_err(&e);164				CString::new(&out as &str).unwrap().into_raw()165			}166		}167	})168}169170fn multi_to_raw(multi: Vec<(IStr, IStr)>) -> *const c_char {171	let mut out = Vec::new();172	for (i, (k, v)) in multi.iter().enumerate() {173		if i != 0 {174			out.push(0);175		}176		out.extend_from_slice(k.as_bytes());177		out.push(0);178		out.extend_from_slice(v.as_bytes());179	}180	out.push(0);181	out.push(0);182	let v = out.as_ptr();183	std::mem::forget(out);184	v as *const c_char185}186187/// # Safety188#[no_mangle]189pub unsafe extern "C" fn jsonnet_evaluate_file_multi(190	vm: &EvaluationState,191	filename: *const c_char,192	error: &mut c_int,193) -> *const c_char {194	vm.run_in_state(|| {195		let filename = CStr::from_ptr(filename);196		match vm197			.evaluate_file_raw_nocwd(&PathBuf::from(filename.to_str().unwrap()))198			.and_then(|v| vm.with_tla(v))199			.and_then(|v| vm.manifest_multi(v))200		{201			Ok(v) => {202				*error = 0;203				multi_to_raw(v)204			}205			Err(e) => {206				*error = 1;207				let out = vm.stringify_err(&e);208				CString::new(&out as &str).unwrap().into_raw()209			}210		}211	})212}213214/// # Safety215#[no_mangle]216pub unsafe extern "C" fn jsonnet_evaluate_snippet_multi(217	vm: &EvaluationState,218	filename: *const c_char,219	snippet: *const c_char,220	error: &mut c_int,221) -> *const c_char {222	vm.run_in_state(|| {223		let filename = CStr::from_ptr(filename);224		let snippet = CStr::from_ptr(snippet);225		match vm226			.evaluate_snippet_raw(227				PathBuf::from(filename.to_str().unwrap()).into(),228				snippet.to_str().unwrap().into(),229			)230			.and_then(|v| vm.with_tla(v))231			.and_then(|v| vm.manifest_multi(v))232		{233			Ok(v) => {234				*error = 0;235				multi_to_raw(v)236			}237			Err(e) => {238				*error = 1;239				let out = vm.stringify_err(&e);240				CString::new(&out as &str).unwrap().into_raw()241			}242		}243	})244}245246fn stream_to_raw(multi: Vec<IStr>) -> *const c_char {247	let mut out = Vec::new();248	for (i, v) in multi.iter().enumerate() {249		if i != 0 {250			out.push(0);251		}252		out.extend_from_slice(v.as_bytes());253	}254	out.push(0);255	out.push(0);256	let v = out.as_ptr();257	std::mem::forget(out);258	v as *const c_char259}260261/// # Safety262#[no_mangle]263pub unsafe extern "C" fn jsonnet_evaluate_file_stream(264	vm: &EvaluationState,265	filename: *const c_char,266	error: &mut c_int,267) -> *const c_char {268	vm.run_in_state(|| {269		let filename = CStr::from_ptr(filename);270		match vm271			.evaluate_file_raw_nocwd(&PathBuf::from(filename.to_str().unwrap()))272			.and_then(|v| vm.with_tla(v))273			.and_then(|v| vm.manifest_stream(v))274		{275			Ok(v) => {276				*error = 0;277				stream_to_raw(v)278			}279			Err(e) => {280				*error = 1;281				let out = vm.stringify_err(&e);282				CString::new(&out as &str).unwrap().into_raw()283			}284		}285	})286}287288/// # Safety289#[no_mangle]290pub unsafe extern "C" fn jsonnet_evaluate_snippet_stream(291	vm: &EvaluationState,292	filename: *const c_char,293	snippet: *const c_char,294	error: &mut c_int,295) -> *const c_char {296	vm.run_in_state(|| {297		let filename = CStr::from_ptr(filename);298		let snippet = CStr::from_ptr(snippet);299		match vm300			.evaluate_snippet_raw(301				PathBuf::from(filename.to_str().unwrap()).into(),302				snippet.to_str().unwrap().into(),303			)304			.and_then(|v| vm.with_tla(v))305			.and_then(|v| vm.manifest_stream(v))306		{307			Ok(v) => {308				*error = 0;309				stream_to_raw(v)310			}311			Err(e) => {312				*error = 1;313				let out = vm.stringify_err(&e);314				CString::new(&out as &str).unwrap().into_raw()315			}316		}317	})318}