1// Protocol Buffers - Google's data interchange format 2// Copyright 2014 Google Inc. All rights reserved. 3// https://developers.google.com/protocol-buffers/ 4// 5// Redistribution and use in source and binary forms, with or without 6// modification, are permitted provided that the following conditions are 7// met: 8// 9// * Redistributions of source code must retain the above copyright 10// notice, this list of conditions and the following disclaimer. 11// * Redistributions in binary form must reproduce the above 12// copyright notice, this list of conditions and the following disclaimer 13// in the documentation and/or other materials provided with the 14// distribution. 15// * Neither the name of Google Inc. nor the names of its 16// contributors may be used to endorse or promote products derived from 17// this software without specific prior written permission. 18// 19// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 20// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 21// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 22// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 23// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 24// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 25// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 26// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 27// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 28// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 29// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 30 31#include "protobuf.h" 32 33// ----------------------------------------------------------------------------- 34// Global map from upb {msg,enum}defs to wrapper Descriptor/EnumDescriptor 35// instances. 36// ----------------------------------------------------------------------------- 37 38// This is a hash table from def objects (encoded by converting pointers to 39// Ruby integers) to MessageDef/EnumDef instances (as Ruby values). 40VALUE upb_def_to_ruby_obj_map; 41 42VALUE cError; 43VALUE cParseError; 44 45void add_def_obj(const void* def, VALUE value) { 46 rb_hash_aset(upb_def_to_ruby_obj_map, ULL2NUM((intptr_t)def), value); 47} 48 49VALUE get_def_obj(const void* def) { 50 return rb_hash_aref(upb_def_to_ruby_obj_map, ULL2NUM((intptr_t)def)); 51} 52 53// ----------------------------------------------------------------------------- 54// Utilities. 55// ----------------------------------------------------------------------------- 56 57// Raises a Ruby error if |status| is not OK, using its error message. 58void check_upb_status(const upb_status* status, const char* msg) { 59 if (!upb_ok(status)) { 60 rb_raise(rb_eRuntimeError, "%s: %s\n", msg, upb_status_errmsg(status)); 61 } 62} 63 64// String encodings: we look these up once, at load time, and then cache them 65// here. 66rb_encoding* kRubyStringUtf8Encoding; 67rb_encoding* kRubyStringASCIIEncoding; 68rb_encoding* kRubyString8bitEncoding; 69 70// Ruby-interned string: "descriptor". We use this identifier to store an 71// instance variable on message classes we create in order to link them back to 72// their descriptors. 73// 74// We intern this once at module load time then use the interned identifier at 75// runtime in order to avoid the cost of repeatedly interning in hot paths. 76const char* kDescriptorInstanceVar = "descriptor"; 77ID descriptor_instancevar_interned; 78 79// ----------------------------------------------------------------------------- 80// Initialization/entry point. 81// ----------------------------------------------------------------------------- 82 83// This must be named "Init_protobuf_c" because the Ruby module is named 84// "protobuf_c" -- the VM looks for this symbol in our .so. 85void Init_protobuf_c() { 86 VALUE google = rb_define_module("Google"); 87 VALUE protobuf = rb_define_module_under(google, "Protobuf"); 88 VALUE internal = rb_define_module_under(protobuf, "Internal"); 89 90 descriptor_instancevar_interned = rb_intern(kDescriptorInstanceVar); 91 DescriptorPool_register(protobuf); 92 Descriptor_register(protobuf); 93 FieldDescriptor_register(protobuf); 94 OneofDescriptor_register(protobuf); 95 EnumDescriptor_register(protobuf); 96 MessageBuilderContext_register(internal); 97 OneofBuilderContext_register(internal); 98 EnumBuilderContext_register(internal); 99 Builder_register(internal); 100 RepeatedField_register(protobuf); 101 Map_register(protobuf); 102 103 cError = rb_const_get(protobuf, rb_intern("Error")); 104 cParseError = rb_const_get(protobuf, rb_intern("ParseError")); 105 106 rb_define_singleton_method(protobuf, "deep_copy", 107 Google_Protobuf_deep_copy, 1); 108 109 kRubyStringUtf8Encoding = rb_utf8_encoding(); 110 kRubyStringASCIIEncoding = rb_usascii_encoding(); 111 kRubyString8bitEncoding = rb_ascii8bit_encoding(); 112 113 upb_def_to_ruby_obj_map = rb_hash_new(); 114 rb_gc_register_address(&upb_def_to_ruby_obj_map); 115} 116