#include "rust/cpp_kernel/cpp_api.h" #include #include #include #include "google/protobuf/map.h" #include "google/protobuf/repeated_field.h" extern "C" { #define expose_repeated_field_methods(ty, rust_ty) \ google::protobuf::RepeatedField* __pb_rust_RepeatedField_##rust_ty##_new() { \ return new google::protobuf::RepeatedField(); \ } \ void __pb_rust_RepeatedField_##rust_ty##_free( \ google::protobuf::RepeatedField* r) { \ delete r; \ } \ void __pb_rust_RepeatedField_##rust_ty##_add(google::protobuf::RepeatedField* r, \ ty val) { \ r->Add(val); \ } \ size_t __pb_rust_RepeatedField_##rust_ty##_size( \ google::protobuf::RepeatedField* r) { \ return r->size(); \ } \ ty __pb_rust_RepeatedField_##rust_ty##_get(google::protobuf::RepeatedField* r, \ size_t index) { \ return r->Get(index); \ } \ void __pb_rust_RepeatedField_##rust_ty##_set(google::protobuf::RepeatedField* r, \ size_t index, ty val) { \ return r->Set(index, val); \ } \ void __pb_rust_RepeatedField_##rust_ty##_copy_from( \ const google::protobuf::RepeatedField* src, google::protobuf::RepeatedField* dst) { \ dst->CopyFrom(*src); \ } \ void __pb_rust_RepeatedField_##rust_ty##_clear( \ google::protobuf::RepeatedField* r) { \ r->Clear(); \ } expose_repeated_field_methods(int32_t, i32); expose_repeated_field_methods(uint32_t, u32); expose_repeated_field_methods(float, f32); expose_repeated_field_methods(double, f64); expose_repeated_field_methods(bool, bool); expose_repeated_field_methods(uint64_t, u64); expose_repeated_field_methods(int64_t, i64); #undef expose_repeated_field_methods #define expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \ value_ty, rust_value_ty, ffi_value_ty, \ to_cpp_value, to_ffi_value) \ google::protobuf::Map* \ __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_new() { \ return new google::protobuf::Map(); \ } \ void __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_free( \ google::protobuf::Map* m) { \ delete m; \ } \ void __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_clear( \ google::protobuf::Map* m) { \ m->clear(); \ } \ size_t __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_size( \ const google::protobuf::Map* m) { \ return m->size(); \ } \ void __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_insert( \ google::protobuf::Map* m, ffi_key_ty key, ffi_value_ty value) { \ auto cpp_key = to_cpp_key; \ auto cpp_value = to_cpp_value; \ (*m)[cpp_key] = cpp_value; \ } \ bool __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_get( \ const google::protobuf::Map* m, ffi_key_ty key, \ ffi_value_ty* value) { \ auto cpp_key = to_cpp_key; \ auto it = m->find(cpp_key); \ if (it == m->end()) { \ return false; \ } \ auto& cpp_value = it->second; \ *value = to_ffi_value; \ return true; \ } \ bool __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_remove( \ google::protobuf::Map* m, ffi_key_ty key, ffi_value_ty* value) { \ auto cpp_key = to_cpp_key; \ auto num_removed = m->erase(cpp_key); \ return num_removed > 0; \ } #define expose_scalar_map_methods_for_key_type(key_ty, rust_key_ty, \ ffi_key_ty, to_cpp_key) \ expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \ int32_t, i32, int32_t, value, cpp_value); \ expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \ uint32_t, u32, uint32_t, value, cpp_value); \ expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \ float, f32, float, value, cpp_value); \ expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \ double, f64, double, value, cpp_value); \ expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, bool, \ bool, bool, value, cpp_value); \ expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \ uint64_t, u64, uint64_t, value, cpp_value); \ expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \ int64_t, i64, int64_t, value, cpp_value); \ expose_scalar_map_methods( \ key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, std::string, Bytes, \ google::protobuf::rust_internal::PtrAndLen, std::string(value.ptr, value.len), \ google::protobuf::rust_internal::PtrAndLen(cpp_value.data(), cpp_value.size())); \ expose_scalar_map_methods( \ key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, std::string, ProtoStr, \ google::protobuf::rust_internal::PtrAndLen, std::string(value.ptr, value.len), \ google::protobuf::rust_internal::PtrAndLen(cpp_value.data(), cpp_value.size())); expose_scalar_map_methods_for_key_type(int32_t, i32, int32_t, key); expose_scalar_map_methods_for_key_type(uint32_t, u32, uint32_t, key); expose_scalar_map_methods_for_key_type(bool, bool, bool, key); expose_scalar_map_methods_for_key_type(uint64_t, u64, uint64_t, key); expose_scalar_map_methods_for_key_type(int64_t, i64, int64_t, key); expose_scalar_map_methods_for_key_type(std::string, ProtoStr, google::protobuf::rust_internal::PtrAndLen, std::string(key.ptr, key.len)); #undef expose_scalar_map_methods #undef expose_map_methods }