xref: /aosp_15_r20/external/grpc-grpc/third_party/upb/upb/message/compat.c (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2023 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 #include "upb/message/compat.h"
9 
10 #include <stddef.h>
11 #include <stdint.h>
12 
13 #include "upb/message/internal/extension.h"
14 #include "upb/message/message.h"
15 #include "upb/mini_table/extension.h"
16 
17 // Must be last.
18 #include "upb/port/def.inc"
19 
upb_Message_ExtensionByIndex(const upb_Message * msg,size_t index)20 const upb_MiniTableExtension* upb_Message_ExtensionByIndex(
21     const upb_Message* msg, size_t index) {
22   size_t count;
23   const upb_Extension* ext = UPB_PRIVATE(_upb_Message_Getexts)(msg, &count);
24 
25   UPB_ASSERT(index < count);
26   return ext[index].ext;
27 }
28 
upb_Message_FindExtensionByNumber(const upb_Message * msg,uint32_t field_number)29 const upb_Extension* upb_Message_FindExtensionByNumber(const upb_Message* msg,
30                                                        uint32_t field_number) {
31   size_t count;
32   const upb_Extension* ext = UPB_PRIVATE(_upb_Message_Getexts)(msg, &count);
33 
34   while (count--) {
35     if (upb_MiniTableExtension_Number(ext->ext) == field_number) return ext;
36     ext++;
37   }
38   return NULL;
39 }
40