xref: /aosp_15_r20/external/pigweed/pw_protobuf/find.cc (revision 61c4878ac05f98d0ceed94b57d316916de578985)
1 // Copyright 2023 The Pigweed Authors
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4 // use this file except in compliance with the License. You may obtain a copy of
5 // the License at
6 //
7 //     https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 // License for the specific language governing permissions and limitations under
13 // the License.
14 
15 #include "pw_protobuf/find.h"
16 
17 #include "pw_protobuf/wire_format.h"
18 
19 namespace pw::protobuf::internal {
20 
AdvanceToField(Decoder & decoder,uint32_t field_number)21 Status AdvanceToField(Decoder& decoder, uint32_t field_number) {
22   if (!ValidFieldNumber(field_number)) {
23     return Status::InvalidArgument();
24   }
25 
26   Status status;
27 
28   while ((status = decoder.Next()).ok()) {
29     if (decoder.FieldNumber() == field_number) {
30       return OkStatus();
31     }
32   }
33 
34   // As this is a backend for the Find() APIs, remap OUT_OF_RANGE to NOT_FOUND.
35   return status.IsOutOfRange() ? Status::NotFound() : status;
36 }
37 
AdvanceToField(StreamDecoder & decoder,uint32_t field_number)38 Status AdvanceToField(StreamDecoder& decoder, uint32_t field_number) {
39   if (!ValidFieldNumber(field_number)) {
40     return Status::InvalidArgument();
41   }
42 
43   Status status;
44 
45   while ((status = decoder.Next()).ok()) {
46     PW_TRY_ASSIGN(uint32_t field, decoder.FieldNumber());
47     if (field == field_number) {
48       return OkStatus();
49     }
50   }
51 
52   // As this is a backend for the Find() APIs, remap OUT_OF_RANGE to NOT_FOUND.
53   return status.IsOutOfRange() ? Status::NotFound() : status;
54 }
55 
56 }  // namespace pw::protobuf::internal
57