use crate::c_api::buffer::*;
use crate::c_api::utils::*;
use crate::shortint;
use crate::shortint::ciphertext::Degree;
use std::os::raw::c_int;
pub struct ShortintCiphertext(pub(in crate::c_api) shortint::Ciphertext);
pub struct ShortintCompressedCiphertext(pub(in crate::c_api) shortint::CompressedCiphertext);
#[no_mangle]
pub unsafe extern "C" fn shortint_ciphertext_set_degree(
ciphertext: *mut ShortintCiphertext,
degree: u64,
) -> c_int {
catch_panic(|| {
let ciphertext = get_mut_checked(ciphertext).unwrap();
let inner_ct = &mut ciphertext.0;
inner_ct.degree = Degree::new(degree);
})
}
#[no_mangle]
pub unsafe extern "C" fn shortint_ciphertext_get_degree(
ciphertext: *const ShortintCiphertext,
result: *mut u64,
) -> c_int {
catch_panic(|| {
check_ptr_is_non_null_and_aligned(result).unwrap();
let ciphertext = get_ref_checked(ciphertext).unwrap();
let inner_ct = &ciphertext.0;
*result = inner_ct.degree.get();
})
}
#[no_mangle]
pub unsafe extern "C" fn shortint_serialize_ciphertext(
ciphertext: *const ShortintCiphertext,
result: *mut DynamicBuffer,
) -> c_int {
catch_panic(|| {
check_ptr_is_non_null_and_aligned(result).unwrap();
let ciphertext = get_ref_checked(ciphertext).unwrap();
let buffer: DynamicBuffer = bincode::serialize(&ciphertext.0).unwrap().into();
*result = buffer;
})
}
#[no_mangle]
pub unsafe extern "C" fn shortint_deserialize_ciphertext(
buffer_view: DynamicBufferView,
result: *mut *mut ShortintCiphertext,
) -> c_int {
catch_panic(|| {
check_ptr_is_non_null_and_aligned(result).unwrap();
*result = std::ptr::null_mut();
let ciphertext = bincode::deserialize(buffer_view.as_slice()).unwrap();
let heap_allocated_ciphertext = Box::new(ShortintCiphertext(ciphertext));
*result = Box::into_raw(heap_allocated_ciphertext);
})
}
#[no_mangle]
pub unsafe extern "C" fn shortint_decompress_ciphertext(
compressed_ciphertext: *const ShortintCompressedCiphertext,
result: *mut *mut ShortintCiphertext,
) -> c_int {
catch_panic(|| {
check_ptr_is_non_null_and_aligned(result).unwrap();
*result = std::ptr::null_mut();
let compressed_ciphertext = get_ref_checked(compressed_ciphertext).unwrap();
let ciphertext = compressed_ciphertext.0.decompress();
let heap_allocated_ciphertext = Box::new(ShortintCiphertext(ciphertext));
*result = Box::into_raw(heap_allocated_ciphertext);
})
}
#[no_mangle]
pub unsafe extern "C" fn shortint_serialize_compressed_ciphertext(
ciphertext: *const ShortintCompressedCiphertext,
result: *mut DynamicBuffer,
) -> c_int {
catch_panic(|| {
check_ptr_is_non_null_and_aligned(result).unwrap();
let ciphertext = get_ref_checked(ciphertext).unwrap();
let buffer: DynamicBuffer = bincode::serialize(&ciphertext.0).unwrap().into();
*result = buffer;
})
}
#[no_mangle]
pub unsafe extern "C" fn shortint_deserialize_compressed_ciphertext(
buffer_view: DynamicBufferView,
result: *mut *mut ShortintCompressedCiphertext,
) -> c_int {
catch_panic(|| {
check_ptr_is_non_null_and_aligned(result).unwrap();
*result = std::ptr::null_mut();
let ciphertext = bincode::deserialize(buffer_view.as_slice()).unwrap();
let heap_allocated_ciphertext = Box::new(ShortintCompressedCiphertext(ciphertext));
*result = Box::into_raw(heap_allocated_ciphertext);
})
}