• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2024 Google LLC.  All rights reserved.
3 //
4 // Use of this source code is governed by a BSD-style
5 // license that can be found in the LICENSE file or at
6 // https://developers.google.com/open-source/licenses/bsd
7 
8 use super::opaque_pointee::opaque_pointee;
9 use super::{upb_MessageValue, upb_MutableMessageValue, CType, RawArena};
10 use core::ptr::NonNull;
11 
12 opaque_pointee!(upb_Array);
13 pub type RawArray = NonNull<upb_Array>;
14 
15 extern "C" {
upb_Array_New(a: RawArena, r#type: CType) -> RawArray16     pub fn upb_Array_New(a: RawArena, r#type: CType) -> RawArray;
upb_Array_Size(arr: RawArray) -> usize17     pub fn upb_Array_Size(arr: RawArray) -> usize;
upb_Array_Set(arr: RawArray, i: usize, val: upb_MessageValue)18     pub fn upb_Array_Set(arr: RawArray, i: usize, val: upb_MessageValue);
upb_Array_Get(arr: RawArray, i: usize) -> upb_MessageValue19     pub fn upb_Array_Get(arr: RawArray, i: usize) -> upb_MessageValue;
upb_Array_Append(arr: RawArray, val: upb_MessageValue, arena: RawArena) -> bool20     pub fn upb_Array_Append(arr: RawArray, val: upb_MessageValue, arena: RawArena) -> bool;
upb_Array_Resize(arr: RawArray, size: usize, arena: RawArena) -> bool21     pub fn upb_Array_Resize(arr: RawArray, size: usize, arena: RawArena) -> bool;
upb_Array_Reserve(arr: RawArray, size: usize, arena: RawArena) -> bool22     pub fn upb_Array_Reserve(arr: RawArray, size: usize, arena: RawArena) -> bool;
upb_Array_MutableDataPtr(arr: RawArray) -> *mut core::ffi::c_void23     pub fn upb_Array_MutableDataPtr(arr: RawArray) -> *mut core::ffi::c_void;
upb_Array_DataPtr(arr: RawArray) -> *const core::ffi::c_void24     pub fn upb_Array_DataPtr(arr: RawArray) -> *const core::ffi::c_void;
upb_Array_GetMutable(arr: RawArray, i: usize) -> upb_MutableMessageValue25     pub fn upb_Array_GetMutable(arr: RawArray, i: usize) -> upb_MutableMessageValue;
26 }
27 
28 #[cfg(test)]
29 mod tests {
30     use super::super::Arena;
31     use super::*;
32     use googletest::gtest;
33 
34     #[gtest]
assert_array_linked()35     fn assert_array_linked() {
36         use crate::assert_linked;
37         assert_linked!(upb_Array_New);
38         assert_linked!(upb_Array_Size);
39         assert_linked!(upb_Array_Set);
40         assert_linked!(upb_Array_Get);
41         assert_linked!(upb_Array_Append);
42         assert_linked!(upb_Array_Resize);
43         assert_linked!(upb_Array_Reserve);
44         assert_linked!(upb_Array_MutableDataPtr);
45         assert_linked!(upb_Array_DataPtr);
46         assert_linked!(upb_Array_GetMutable);
47     }
48 
49     #[gtest]
array_ffi_test()50     fn array_ffi_test() {
51         // SAFETY: FFI unit test uses C API under expected patterns.
52         unsafe {
53             let arena = Arena::new();
54             let raw_arena = arena.raw();
55             let array = upb_Array_New(raw_arena, CType::Float);
56 
57             assert!(upb_Array_Append(array, upb_MessageValue { float_val: 7.0 }, raw_arena));
58             assert!(upb_Array_Append(array, upb_MessageValue { float_val: 42.0 }, raw_arena));
59             assert_eq!(upb_Array_Size(array), 2);
60             assert!(matches!(upb_Array_Get(array, 1), upb_MessageValue { float_val: 42.0 }));
61 
62             assert!(upb_Array_Resize(array, 3, raw_arena));
63             assert_eq!(upb_Array_Size(array), 3);
64             assert!(matches!(upb_Array_Get(array, 2), upb_MessageValue { float_val: 0.0 }));
65         }
66     }
67 }
68