• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 #include <cstddef>
2 #include <cstdint>
3 #include <cstring>
4 #include <string>
5 #include <utility>
6 
7 #include "google/protobuf/message.h"
8 #include "google/protobuf/message_lite.h"
9 #include "google/protobuf/repeated_field.h"
10 #include "google/protobuf/repeated_ptr_field.h"
11 #include "rust/cpp_kernel/strings.h"
12 
13 extern "C" {
14 #define expose_repeated_field_methods(ty, rust_ty)                             \
15   google::protobuf::RepeatedField<ty>* proto2_rust_RepeatedField_##rust_ty##_new() {     \
16     return new google::protobuf::RepeatedField<ty>();                                    \
17   }                                                                            \
18   void proto2_rust_RepeatedField_##rust_ty##_free(                             \
19       google::protobuf::RepeatedField<ty>* r) {                                          \
20     delete r;                                                                  \
21   }                                                                            \
22   void proto2_rust_RepeatedField_##rust_ty##_add(google::protobuf::RepeatedField<ty>* r, \
23                                                  ty val) {                     \
24     r->Add(val);                                                               \
25   }                                                                            \
26   size_t proto2_rust_RepeatedField_##rust_ty##_size(                           \
27       google::protobuf::RepeatedField<ty>* r) {                                          \
28     return r->size();                                                          \
29   }                                                                            \
30   ty proto2_rust_RepeatedField_##rust_ty##_get(google::protobuf::RepeatedField<ty>* r,   \
31                                                size_t index) {                 \
32     return r->Get(index);                                                      \
33   }                                                                            \
34   void proto2_rust_RepeatedField_##rust_ty##_set(google::protobuf::RepeatedField<ty>* r, \
35                                                  size_t index, ty val) {       \
36     return r->Set(index, val);                                                 \
37   }                                                                            \
38   void proto2_rust_RepeatedField_##rust_ty##_copy_from(                        \
39       const google::protobuf::RepeatedField<ty>* src, google::protobuf::RepeatedField<ty>* dst) {  \
40     dst->CopyFrom(*src);                                                       \
41   }                                                                            \
42   void proto2_rust_RepeatedField_##rust_ty##_clear(                            \
43       google::protobuf::RepeatedField<ty>* r) {                                          \
44     r->Clear();                                                                \
45   }                                                                            \
46   void proto2_rust_RepeatedField_##rust_ty##_reserve(                          \
47       google::protobuf::RepeatedField<ty>* r, size_t additional) {                       \
48     r->Reserve(r->size() + additional);                                        \
49   }
50 
51 expose_repeated_field_methods(int32_t, i32);
52 expose_repeated_field_methods(uint32_t, u32);
53 expose_repeated_field_methods(float, f32);
54 expose_repeated_field_methods(double, f64);
55 expose_repeated_field_methods(bool, bool);
56 expose_repeated_field_methods(uint64_t, u64);
57 expose_repeated_field_methods(int64_t, i64);
58 #undef expose_repeated_field_methods
59 
60 #define expose_repeated_ptr_field_methods(ty)                        \
61   google::protobuf::RepeatedPtrField<std::string>*                             \
62       proto2_rust_RepeatedField_##ty##_new() {                       \
63     return new google::protobuf::RepeatedPtrField<std::string>();              \
64   }                                                                  \
65   void proto2_rust_RepeatedField_##ty##_free(                        \
66       google::protobuf::RepeatedPtrField<std::string>* r) {                    \
67     delete r;                                                        \
68   }                                                                  \
69   void proto2_rust_RepeatedField_##ty##_add(                         \
70       google::protobuf::RepeatedPtrField<std::string>* r, std::string* val) {  \
71     r->AddAllocated(val);                                            \
72   }                                                                  \
73   size_t proto2_rust_RepeatedField_##ty##_size(                      \
74       google::protobuf::RepeatedPtrField<std::string>* r) {                    \
75     return r->size();                                                \
76   }                                                                  \
77   google::protobuf::rust::PtrAndLen proto2_rust_RepeatedField_##ty##_get(      \
78       google::protobuf::RepeatedPtrField<std::string>* r, size_t index) {      \
79     const std::string& s = r->Get(index);                            \
80     return google::protobuf::rust::PtrAndLen{s.data(), s.size()};              \
81   }                                                                  \
82   void proto2_rust_RepeatedField_##ty##_set(                         \
83       google::protobuf::RepeatedPtrField<std::string>* r, size_t index,        \
84       std::string* val) {                                            \
85     *r->Mutable(index) = std::move(*val);                            \
86     delete val;                                                      \
87   }                                                                  \
88   void proto2_rust_RepeatedField_##ty##_copy_from(                   \
89       const google::protobuf::RepeatedPtrField<std::string>* src,              \
90       google::protobuf::RepeatedPtrField<std::string>* dst) {                  \
91     dst->CopyFrom(*src);                                             \
92   }                                                                  \
93   void proto2_rust_RepeatedField_##ty##_clear(                       \
94       google::protobuf::RepeatedPtrField<std::string>* r) {                    \
95     r->Clear();                                                      \
96   }                                                                  \
97   void proto2_rust_RepeatedField_##ty##_reserve(                     \
98       google::protobuf::RepeatedPtrField<std::string>* r, size_t additional) { \
99     r->Reserve(r->size() + additional);                              \
100   }
101 
102 expose_repeated_ptr_field_methods(ProtoString);
103 expose_repeated_ptr_field_methods(ProtoBytes);
104 #undef expose_repeated_field_methods
105 
106 #undef expose_repeated_ptr_field_methods
107 
108 }  // extern "C"
109