Google Git
Sign in
pigweed / third_party / github / protocolbuffers / protobuf / e1d3008eaf7e3726dd7cb1574e8521b8ac401ade / . / rust / cpp_kernel / cpp_api.cc
blob: 05aec1f0aedf89ca0e017d7739bea30722c13f2b [file] [log] [blame]
Jakob Buchgraber97602922023-12-07 07:19:54 -0800[diff] [blame]1#include "rust/cpp_kernel/cpp_api.h"
2
3#include <cstdint>
4#include <string>
5
Jakob Buchgraber001be3c2023-11-16 01:36:28 -0800[diff] [blame]6#include "google/protobuf/map.h"
Protobuf Team Bote1bb7d62023-10-17 14:15:37 -0700[diff] [blame]7#include "google/protobuf/repeated_field.h"
8
9extern "C" {
10
Kevin King65cdac42023-10-23 14:50:42 -0700[diff] [blame]11#define expose_repeated_field_methods(ty, rust_ty) \
12 google::protobuf::RepeatedField<ty>* __pb_rust_RepeatedField_##rust_ty##_new() { \
13 return new google::protobuf::RepeatedField<ty>(); \
14 } \
15 void __pb_rust_RepeatedField_##rust_ty##_add(google::protobuf::RepeatedField<ty>* r, \
16 ty val) { \
17 r->Add(val); \
18 } \
19 size_t __pb_rust_RepeatedField_##rust_ty##_size( \
20 google::protobuf::RepeatedField<ty>* r) { \
21 return r->size(); \
22 } \
23 ty __pb_rust_RepeatedField_##rust_ty##_get(google::protobuf::RepeatedField<ty>* r, \
24 size_t index) { \
25 return r->Get(index); \
26 } \
27 void __pb_rust_RepeatedField_##rust_ty##_set(google::protobuf::RepeatedField<ty>* r, \
28 size_t index, ty val) { \
29 return r->Set(index, val); \
30 } \
31 void __pb_rust_RepeatedField_##rust_ty##_copy_from( \
32 google::protobuf::RepeatedField<ty> const& src, google::protobuf::RepeatedField<ty>& dst) { \
33 dst.CopyFrom(src); \
Protobuf Team Bote1bb7d62023-10-17 14:15:37 -0700[diff] [blame]34 }
35
36expose_repeated_field_methods(int32_t, i32);
37expose_repeated_field_methods(uint32_t, u32);
38expose_repeated_field_methods(float, f32);
39expose_repeated_field_methods(double, f64);
40expose_repeated_field_methods(bool, bool);
41expose_repeated_field_methods(uint64_t, u64);
42expose_repeated_field_methods(int64_t, i64);
43
44#undef expose_repeated_field_methods
Jakob Buchgraber001be3c2023-11-16 01:36:28 -0800[diff] [blame]45
Jakob Buchgraber97602922023-12-07 07:19:54 -0800[diff] [blame]46#define expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
47 value_ty, rust_value_ty, ffi_value_ty, \
48 to_cpp_value, to_ffi_value) \
49 google::protobuf::Map<key_ty, value_ty>* \
50 __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_new() { \
51 return new google::protobuf::Map<key_ty, value_ty>(); \
52 } \
53 void __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_clear( \
54 google::protobuf::Map<key_ty, value_ty>* m) { \
55 m->clear(); \
56 } \
57 size_t __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_size( \
58 google::protobuf::Map<key_ty, value_ty>* m) { \
59 return m->size(); \
60 } \
61 void __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_insert( \
62 google::protobuf::Map<key_ty, value_ty>* m, ffi_key_ty key, ffi_value_ty value) { \
63 auto cpp_key = to_cpp_key; \
64 auto cpp_value = to_cpp_value; \
65 (*m)[cpp_key] = cpp_value; \
66 } \
67 bool __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_get( \
68 google::protobuf::Map<key_ty, value_ty>* m, ffi_key_ty key, ffi_value_ty* value) { \
69 auto cpp_key = to_cpp_key; \
70 auto it = m->find(cpp_key); \
71 if (it == m->end()) { \
72 return false; \
73 } \
74 auto& cpp_value = it->second; \
75 *value = to_ffi_value; \
76 return true; \
77 } \
78 bool __pb_rust_Map_##rust_key_ty##_##rust_value_ty##_remove( \
79 google::protobuf::Map<key_ty, value_ty>* m, ffi_key_ty key, ffi_value_ty* value) { \
80 auto cpp_key = to_cpp_key; \
81 auto num_removed = m->erase(cpp_key); \
82 return num_removed > 0; \
Jakob Buchgraber001be3c2023-11-16 01:36:28 -0800[diff] [blame]83 }
84
Jakob Buchgraber97602922023-12-07 07:19:54 -0800[diff] [blame]85#define expose_scalar_map_methods_for_key_type(key_ty, rust_key_ty, \
86 ffi_key_ty, to_cpp_key) \
87 expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
88 int32_t, i32, int32_t, value, cpp_value); \
89 expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
90 uint32_t, u32, uint32_t, value, cpp_value); \
91 expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
92 float, f32, float, value, cpp_value); \
93 expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
94 double, f64, double, value, cpp_value); \
95 expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, bool, \
96 bool, bool, value, cpp_value); \
97 expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
98 uint64_t, u64, uint64_t, value, cpp_value); \
99 expose_scalar_map_methods(key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, \
100 int64_t, i64, int64_t, value, cpp_value); \
101 expose_scalar_map_methods( \
102 key_ty, rust_key_ty, ffi_key_ty, to_cpp_key, std::string, ProtoStr, \
103 google::protobuf::rust_internal::PtrAndLen, std::string(value.ptr, value.len), \
104 google::protobuf::rust_internal::PtrAndLen(cpp_value.data(), cpp_value.size()));
Jakob Buchgraber001be3c2023-11-16 01:36:28 -0800[diff] [blame]105
Jakob Buchgraber97602922023-12-07 07:19:54 -0800[diff] [blame]106expose_scalar_map_methods_for_key_type(int32_t, i32, int32_t, key);
107expose_scalar_map_methods_for_key_type(uint32_t, u32, uint32_t, key);
108expose_scalar_map_methods_for_key_type(bool, bool, bool, key);
109expose_scalar_map_methods_for_key_type(uint64_t, u64, uint64_t, key);
110expose_scalar_map_methods_for_key_type(int64_t, i64, int64_t, key);
111expose_scalar_map_methods_for_key_type(std::string, ProtoStr,
112 google::protobuf::rust_internal::PtrAndLen,
113 std::string(key.ptr, key.len));
Jakob Buchgraber001be3c2023-11-16 01:36:28 -0800[diff] [blame]114
115#undef expose_scalar_map_methods
116#undef expose_map_methods
Protobuf Team Bote1bb7d62023-10-17 14:15:37 -0700[diff] [blame]117}