Protocol Buffers - Google's data interchange format (grpc依赖)
https://developers.google.com/protocol-buffers/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
188 lines
6.2 KiB
188 lines
6.2 KiB
// Protocol Buffers - Google's data interchange format |
|
// Copyright 2023 Google LLC. All rights reserved. |
|
// https://developers.google.com/protocol-buffers/ |
|
// |
|
// Redistribution and use in source and binary forms, with or without |
|
// modification, are permitted provided that the following conditions are |
|
// met: |
|
// |
|
// * Redistributions of source code must retain the above copyright |
|
// notice, this list of conditions and the following disclaimer. |
|
// * Redistributions in binary form must reproduce the above |
|
// copyright notice, this list of conditions and the following disclaimer |
|
// in the documentation and/or other materials provided with the |
|
// distribution. |
|
// * Neither the name of Google LLC nor the names of its |
|
// contributors may be used to endorse or promote products derived from |
|
// this software without specific prior written permission. |
|
// |
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
|
|
#include "upb/upbc/common.h" |
|
|
|
#include <assert.h> |
|
#include <stdint.h> |
|
#include <stdlib.h> |
|
|
|
#include <string> |
|
|
|
#include "absl/strings/ascii.h" |
|
#include "absl/strings/str_cat.h" |
|
#include "absl/strings/str_replace.h" |
|
#include "absl/strings/string_view.h" |
|
#include "absl/strings/substitute.h" |
|
#include "upb/upb/mini_table/field.h" |
|
#include "upb/upb/mini_table/internal/field.h" |
|
#include "upb/upb/reflection/def.hpp" |
|
|
|
// Must be last |
|
#include "upb/upb/port/def.inc" |
|
|
|
namespace upbc { |
|
|
|
std::string StripExtension(absl::string_view fname) { |
|
size_t lastdot = fname.find_last_of('.'); |
|
if (lastdot == std::string::npos) { |
|
return std::string(fname); |
|
} |
|
return std::string(fname.substr(0, lastdot)); |
|
} |
|
|
|
std::string ToCIdent(absl::string_view str) { |
|
return absl::StrReplaceAll(str, {{".", "_"}, {"/", "_"}, {"-", "_"}}); |
|
} |
|
|
|
std::string ToPreproc(absl::string_view str) { |
|
return absl::AsciiStrToUpper(ToCIdent(str)); |
|
} |
|
|
|
void EmitFileWarning(absl::string_view name, Output& output) { |
|
output( |
|
"/* This file was generated by upbc (the upb compiler) from the input\n" |
|
" * file:\n" |
|
" *\n" |
|
" * $0\n" |
|
" *\n" |
|
" * Do not edit -- your changes will be discarded when the file is\n" |
|
" * regenerated. */\n\n", |
|
name); |
|
} |
|
|
|
std::string MessageName(upb::MessageDefPtr descriptor) { |
|
return ToCIdent(descriptor.full_name()); |
|
} |
|
|
|
std::string FileLayoutName(upb::FileDefPtr file) { |
|
return ToCIdent(file.name()) + "_upb_file_layout"; |
|
} |
|
|
|
std::string CApiHeaderFilename(upb::FileDefPtr file) { |
|
return StripExtension(file.name()) + ".upb.h"; |
|
} |
|
|
|
std::string MiniTableHeaderFilename(upb::FileDefPtr file) { |
|
return StripExtension(file.name()) + ".upb_minitable.h"; |
|
} |
|
|
|
std::string MessageInit(absl::string_view full_name) { |
|
return ToCIdent(full_name) + "_msg_init"; |
|
} |
|
|
|
std::string EnumInit(upb::EnumDefPtr descriptor) { |
|
return ToCIdent(descriptor.full_name()) + "_enum_init"; |
|
} |
|
|
|
std::string FieldInitializer(upb::FieldDefPtr field, |
|
const upb_MiniTableField* field64, |
|
const upb_MiniTableField* field32) { |
|
return absl::Substitute( |
|
"{$0, $1, $2, $3, $4, $5}", field64->number, |
|
ArchDependentSize(field32->offset, field64->offset), |
|
ArchDependentSize(field32->presence, field64->presence), |
|
field64->UPB_PRIVATE(submsg_index) == kUpb_NoSub |
|
? "kUpb_NoSub" |
|
: absl::StrCat(field64->UPB_PRIVATE(submsg_index)).c_str(), |
|
field64->UPB_PRIVATE(descriptortype), GetModeInit(field32, field64)); |
|
} |
|
|
|
std::string ArchDependentSize(int64_t size32, int64_t size64) { |
|
if (size32 == size64) return absl::StrCat(size32); |
|
return absl::Substitute("UPB_SIZE($0, $1)", size32, size64); |
|
} |
|
|
|
// Returns the field mode as a string initializer. |
|
// |
|
// We could just emit this as a number (and we may yet go in that direction) but |
|
// for now emitting symbolic constants gives this better readability and |
|
// debuggability. |
|
std::string GetModeInit(const upb_MiniTableField* field32, |
|
const upb_MiniTableField* field64) { |
|
std::string ret; |
|
uint8_t mode32 = field32->mode; |
|
switch (mode32 & kUpb_FieldMode_Mask) { |
|
case kUpb_FieldMode_Map: |
|
ret = "(int)kUpb_FieldMode_Map"; |
|
break; |
|
case kUpb_FieldMode_Array: |
|
ret = "(int)kUpb_FieldMode_Array"; |
|
break; |
|
case kUpb_FieldMode_Scalar: |
|
ret = "(int)kUpb_FieldMode_Scalar"; |
|
break; |
|
default: |
|
break; |
|
} |
|
|
|
if (mode32 & kUpb_LabelFlags_IsPacked) { |
|
absl::StrAppend(&ret, " | (int)kUpb_LabelFlags_IsPacked"); |
|
} |
|
|
|
if (mode32 & kUpb_LabelFlags_IsExtension) { |
|
absl::StrAppend(&ret, " | (int)kUpb_LabelFlags_IsExtension"); |
|
} |
|
|
|
if (mode32 & kUpb_LabelFlags_IsAlternate) { |
|
absl::StrAppend(&ret, " | (int)kUpb_LabelFlags_IsAlternate"); |
|
} |
|
|
|
absl::StrAppend(&ret, " | ((int)", GetFieldRep(field32, field64), |
|
" << kUpb_FieldRep_Shift)"); |
|
return ret; |
|
} |
|
|
|
std::string GetFieldRep(const upb_MiniTableField* field32, |
|
const upb_MiniTableField* field64) { |
|
switch (_upb_MiniTableField_GetRep(field32)) { |
|
case kUpb_FieldRep_1Byte: |
|
return "kUpb_FieldRep_1Byte"; |
|
break; |
|
case kUpb_FieldRep_4Byte: { |
|
if (_upb_MiniTableField_GetRep(field64) == kUpb_FieldRep_4Byte) { |
|
return "kUpb_FieldRep_4Byte"; |
|
} else { |
|
assert(_upb_MiniTableField_GetRep(field64) == kUpb_FieldRep_8Byte); |
|
return "UPB_SIZE(kUpb_FieldRep_4Byte, kUpb_FieldRep_8Byte)"; |
|
} |
|
break; |
|
} |
|
case kUpb_FieldRep_StringView: |
|
return "kUpb_FieldRep_StringView"; |
|
break; |
|
case kUpb_FieldRep_8Byte: |
|
return "kUpb_FieldRep_8Byte"; |
|
break; |
|
} |
|
UPB_UNREACHABLE(); |
|
} |
|
|
|
} // namespace upbc
|
|
|