Counter Strike : Global Offensive Source Code
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.

169 lines
6.4 KiB

  1. # Protocol Buffers - Google's data interchange format
  2. # Copyright 2008 Google Inc. All rights reserved.
  3. # http://code.google.com/p/protobuf/
  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. # This code is meant to work on Python 2.4 and above only.
  31. """Contains a metaclass and helper functions used to create
  32. protocol message classes from Descriptor objects at runtime.
  33. Recall that a metaclass is the "type" of a class.
  34. (A class is to a metaclass what an instance is to a class.)
  35. In this case, we use the GeneratedProtocolMessageType metaclass
  36. to inject all the useful functionality into the classes
  37. output by the protocol compiler at compile-time.
  38. The upshot of all this is that the real implementation
  39. details for ALL pure-Python protocol buffers are *here in
  40. this file*.
  41. """
  42. __author__ = '[email protected] (Will Robinson)'
  43. from google.protobuf.internal import api_implementation
  44. from google.protobuf import descriptor as descriptor_mod
  45. from google.protobuf import message
  46. _FieldDescriptor = descriptor_mod.FieldDescriptor
  47. if api_implementation.Type() == 'cpp':
  48. if api_implementation.Version() == 2:
  49. from google.protobuf.internal.cpp import cpp_message
  50. _NewMessage = cpp_message.NewMessage
  51. _InitMessage = cpp_message.InitMessage
  52. else:
  53. from google.protobuf.internal import cpp_message
  54. _NewMessage = cpp_message.NewMessage
  55. _InitMessage = cpp_message.InitMessage
  56. else:
  57. from google.protobuf.internal import python_message
  58. _NewMessage = python_message.NewMessage
  59. _InitMessage = python_message.InitMessage
  60. class GeneratedProtocolMessageType(type):
  61. """Metaclass for protocol message classes created at runtime from Descriptors.
  62. We add implementations for all methods described in the Message class. We
  63. also create properties to allow getting/setting all fields in the protocol
  64. message. Finally, we create slots to prevent users from accidentally
  65. "setting" nonexistent fields in the protocol message, which then wouldn't get
  66. serialized / deserialized properly.
  67. The protocol compiler currently uses this metaclass to create protocol
  68. message classes at runtime. Clients can also manually create their own
  69. classes at runtime, as in this example:
  70. mydescriptor = Descriptor(.....)
  71. class MyProtoClass(Message):
  72. __metaclass__ = GeneratedProtocolMessageType
  73. DESCRIPTOR = mydescriptor
  74. myproto_instance = MyProtoClass()
  75. myproto.foo_field = 23
  76. ...
  77. """
  78. # Must be consistent with the protocol-compiler code in
  79. # proto2/compiler/internal/generator.*.
  80. _DESCRIPTOR_KEY = 'DESCRIPTOR'
  81. def __new__(cls, name, bases, dictionary):
  82. """Custom allocation for runtime-generated class types.
  83. We override __new__ because this is apparently the only place
  84. where we can meaningfully set __slots__ on the class we're creating(?).
  85. (The interplay between metaclasses and slots is not very well-documented).
  86. Args:
  87. name: Name of the class (ignored, but required by the
  88. metaclass protocol).
  89. bases: Base classes of the class we're constructing.
  90. (Should be message.Message). We ignore this field, but
  91. it's required by the metaclass protocol
  92. dictionary: The class dictionary of the class we're
  93. constructing. dictionary[_DESCRIPTOR_KEY] must contain
  94. a Descriptor object describing this protocol message
  95. type.
  96. Returns:
  97. Newly-allocated class.
  98. """
  99. descriptor = dictionary[GeneratedProtocolMessageType._DESCRIPTOR_KEY]
  100. bases = _NewMessage(bases, descriptor, dictionary)
  101. superclass = super(GeneratedProtocolMessageType, cls)
  102. new_class = superclass.__new__(cls, name, bases, dictionary)
  103. setattr(descriptor, '_concrete_class', new_class)
  104. return new_class
  105. def __init__(cls, name, bases, dictionary):
  106. """Here we perform the majority of our work on the class.
  107. We add enum getters, an __init__ method, implementations
  108. of all Message methods, and properties for all fields
  109. in the protocol type.
  110. Args:
  111. name: Name of the class (ignored, but required by the
  112. metaclass protocol).
  113. bases: Base classes of the class we're constructing.
  114. (Should be message.Message). We ignore this field, but
  115. it's required by the metaclass protocol
  116. dictionary: The class dictionary of the class we're
  117. constructing. dictionary[_DESCRIPTOR_KEY] must contain
  118. a Descriptor object describing this protocol message
  119. type.
  120. """
  121. descriptor = dictionary[GeneratedProtocolMessageType._DESCRIPTOR_KEY]
  122. _InitMessage(descriptor, cls)
  123. superclass = super(GeneratedProtocolMessageType, cls)
  124. superclass.__init__(name, bases, dictionary)
  125. def ParseMessage(descriptor, byte_str):
  126. """Generate a new Message instance from this Descriptor and a byte string.
  127. Args:
  128. descriptor: Protobuf Descriptor object
  129. byte_str: Serialized protocol buffer byte string
  130. Returns:
  131. Newly created protobuf Message object.
  132. """
  133. class _ResultClass(message.Message):
  134. __metaclass__ = GeneratedProtocolMessageType
  135. DESCRIPTOR = descriptor
  136. new_msg = _ResultClass()
  137. new_msg.ParseFromString(byte_str)
  138. return new_msg