def __init__(self, initializer=None, age=None, **kwargs):
super(PathSpec, self).__init__(age=age, **kwargs)
# Instantiate from another PathSpec.
if isinstance(initializer, PathSpec):
# pylint: disable=protected-access
self.SetRawData(initializer._CopyRawData())
# pylint: enable=protected-access
self.age = initializer.age
# Allow initialization from a list of protobufs each representing a
# component.
elif isinstance(initializer, list):
for element in initializer:
self.last.SetRawData(element.GetRawData())
# Or we can initialize from a string.
elif isinstance(initializer, str):
self.ParseFromString(initializer)
# Legacy protocol buffer implementation.
elif isinstance(initializer, self.protobuf):
self.ParseFromString(initializer.SerializeToString())
elif initializer is not None:
raise rdfvalue.InitializeError("Unable to initialize")
def __init__(self,
initializer=None,
common_name=None,
private_key=None,
age=None):
super(CertificateSigningRequest,
self).__init__(initializer=initializer, age=age)
if self._value is None:
if isinstance(initializer, x509.CertificateSigningRequest):
self._value = initializer
elif isinstance(initializer, basestring):
self.ParseFromString(initializer)
elif common_name and private_key:
self._value = x509.CertificateSigningRequestBuilder(
).subject_name(
x509.Name([
x509.NameAttribute(oid.NameOID.COMMON_NAME,
unicode(common_name))
])).sign(private_key.GetRawPrivateKey(),
hashes.SHA256(),
backend=openssl.backend)
elif initializer is not None:
raise rdfvalue.InitializeError(
"Cannot initialize %s from %s." %
(self.__class__, initializer))
def __init__(self, initializer=None, age=None):
super(RDFX509Cert, self).__init__(initializer=initializer, age=age)
if self._value is None and initializer is not None:
if isinstance(initializer, x509.Certificate):
self._value = initializer
elif isinstance(initializer, basestring):
self.ParseFromString(initializer)
else:
raise rdfvalue.InitializeError("Cannot initialize %s from %s." %
(self.__class__, initializer))
def __init__(self, initializer=None, age=None, allow_prompt=None):
self.allow_prompt = allow_prompt
super(RSAPrivateKey, self).__init__(initializer=initializer, age=age)
if self._value is None and initializer is not None:
if isinstance(initializer, rsa.RSAPrivateKey):
self._value = initializer
elif isinstance(initializer, basestring):
self.ParseFromString(initializer)
else:
raise rdfvalue.InitializeError("Cannot initialize %s from %s." %
(self.__class__, initializer))
def __init__(self, initializer=None, age=None, **kwarg):
super(Dict, self).__init__(initializer=None, age=age)
# Support initializing from a mapping
if isinstance(initializer, dict):
self.FromDict(initializer)
# Can be initialized from kwargs (like a dict).
elif initializer is None:
self.FromDict(kwarg)
# Initialize from another Dict.
elif isinstance(initializer, Dict):
self.FromDict(initializer.ToDict())
self.age = initializer.age
else:
raise rdfvalue.InitializeError(
"Invalid initializer for ProtoDict.")
def __init__(self, initializer=None, age=None):
super(RDFValueArray, self).__init__(age=age)
if self.__class__ == initializer.__class__:
self.content = initializer.Copy().content
self.age = initializer.age
# Initialize from a serialized protobuf.
elif isinstance(initializer, str):
self.ParseFromString(initializer)
else:
try:
for item in initializer:
self.Append(item)
except TypeError:
if initializer is not None:
raise rdfvalue.InitializeError(
"%s can not be initialized from %s" %
(self.__class__.__name__, type(initializer)))
def DefineFromProtobuf(cls, protobuf):
"""Add type info definitions from an existing protobuf.
We support building this class by copying definitions from an annotated
protobuf using the semantic protobuf. This is ideal for interoperability
with other languages and non-semantic protobuf implementations. In that case
it might be easier to simply annotate the .proto file with the relevant
semantic information.
Args:
cls: The class to add fields descriptors to (i.e. the new semantic class).
protobuf: A generated proto2 protocol buffer class as produced by the
standard Google protobuf compiler.
"""
cls.recorded_rdf_deps = set()
# Parse message level options.
message_options = protobuf.DESCRIPTOR.GetOptions()
semantic_options = message_options.Extensions[semantic_pb2.semantic]
# Hack to avoid dependency loop.
# TODO(user): remove this hack
classes_dict = type_info.TypeInfoObject.classes
# Support message descriptions
if semantic_options.description and not cls.__doc__:
cls.__doc__ = semantic_options.description
cls.union_field = semantic_options.union_field or None
# We search through all the field descriptors and build type info
# descriptors from them.
for field in protobuf.DESCRIPTOR.fields:
type_descriptor = None
# Does this field have semantic options?
options = field.GetOptions().Extensions[semantic_pb2.sem_type]
kwargs = dict(
description=options.description,
name=field.name,
friendly_name=options.friendly_name,
field_number=field.number,
labels=list(options.label))
if field.has_default_value:
kwargs["default"] = field.default_value
# This field is a non-protobuf semantic value.
if options.type and field.type != TYPE_MESSAGE:
cls.recorded_rdf_deps.add(options.type)
rdf_type = getattr(rdfvalue, options.type, None)
if rdf_type:
if (CHECK_PROTOBUF_DEPENDENCIES and rdf_type not in cls.rdf_deps and
options.type not in cls.rdf_deps):
raise rdfvalue.InitializeError(
"%s.%s: field %s is of type %s, "
"but type is missing from its dependencies list" %
(cls.__module__, cls.__name__, field.name, options.type))
# Make sure that the field type is the same as what is required by the
# semantic type.
required_field_type = _SEMANTIC_PRIMITIVE_TO_FIELD_TYPE[
rdf_type.data_store_type]
if required_field_type != field.type:
raise rdfvalue.InitializeError(
("%s: .proto file uses incorrect field to store Semantic Value "
"%s: Should be %s") % (cls.__name__, field.name,
rdf_type.data_store_type))
type_descriptor = classes_dict["ProtoRDFValue"](
rdf_type=options.type, **kwargs)
# A semantic protobuf is already a semantic value so it is an error to
# specify it in two places.
elif options.type and field.type == TYPE_MESSAGE:
raise rdfvalue.InitializeError(
("%s: .proto file specified both Semantic Value type %s and "
"Semantic protobuf %s") % (cls.__name__, options.type,
field.message_type.name))
# Try to figure out what this field actually is from the descriptor.
elif field.type == TYPE_DOUBLE:
type_descriptor = classes_dict["ProtoDouble"](**kwargs)
elif field.type == TYPE_FLOAT:
type_descriptor = classes_dict["ProtoFloat"](**kwargs)
elif field.type == TYPE_BOOL:
type_descriptor = classes_dict["ProtoBoolean"](**kwargs)
elif field.type == TYPE_STRING:
type_descriptor = classes_dict["ProtoString"](**kwargs)
elif field.type == TYPE_BYTES:
type_descriptor = classes_dict["ProtoBinary"](**kwargs)
if options.dynamic_type:
# This may be a dynamic type. In this case the dynamic_type option
# names a method (which must exist) which should return the class of
# the embedded semantic value.
dynamic_cb = getattr(cls, options.dynamic_type, None)
if dynamic_cb is not None:
type_descriptor = classes_dict["ProtoDynamicEmbedded"](
dynamic_cb=dynamic_cb, **kwargs)
else:
logging.warning("Dynamic type specifies a non existant callback %s",
options.dynamic_type)
elif (field.type == TYPE_MESSAGE and options.dynamic_type and
field.message_type.name == "Any"):
dynamic_cb = getattr(cls, options.dynamic_type, None)
if dynamic_cb is not None:
type_descriptor = classes_dict["ProtoDynamicAnyValueEmbedded"](
dynamic_cb=dynamic_cb, **kwargs)
else:
logging.warning("Dynamic type specifies a non existant AnyValue "
"callback %s", options.dynamic_type)
elif field.type == TYPE_INT64 or field.type == TYPE_INT32:
type_descriptor = classes_dict["ProtoSignedInteger"](**kwargs)
elif field.type == TYPE_UINT32 or field.type == TYPE_UINT64:
type_descriptor = classes_dict["ProtoUnsignedInteger"](**kwargs)
# An embedded protocol buffer.
elif field.type == TYPE_MESSAGE and field.message_type:
# Refer to another protobuf. Note that the target does not need to be
# known at this time. It will be resolved using the late binding algorithm
# when it is known. Therefore this can actually also refer to this current
# protobuf (i.e. nested proto).
type_descriptor = classes_dict["ProtoEmbedded"](
nested=field.message_type.name, **kwargs)
cls.recorded_rdf_deps.add(field.message_type.name)
if CHECK_PROTOBUF_DEPENDENCIES:
found = False
for d in cls.rdf_deps:
if (hasattr(d, "__name__") and d.__name__ == field.message_type.name
or d == field.message_type.name):
found = True
if not found:
raise rdfvalue.InitializeError(
"%s.%s: TYPE_MESSAGE field %s is %s, "
"but type is missing from its dependencies list" %
(cls.__module__, cls.__name__, field.name,
field.message_type.name))
# TODO(user): support late binding here.
if type_descriptor.type:
# This traps the following problem:
# class Certificate(rdf_protodict.RDFValueArray):
# protobuf = jobs_pb2.BlobArray
#
# A primitive Protobuf definition like:
# message Certificate {
# ....
# };
# And a field like:
# optional Certificate csr = 1 [(sem_type) = {
# description: "A Certificate RDFValue with the CSR in it.",
# }];
# If we blindly allowed the Certificate RDFValue to be used, the
# semantic library will end up embedding a BlobArray protobuf, but the
# primitive library will still use Certificate.
# The name of the primitive protobuf the semantic type implements.
semantic_protobuf_primitive = type_descriptor.type.protobuf.__name__
# This is an error because the primitive library will use the protobuf
# named in the field, but the semantic library will implement a
# different protobuf.
if semantic_protobuf_primitive != field.message_type.name:
raise rdfvalue.InitializeError(
("%s.%s: Conflicting primitive (%s) and semantic protobuf %s "
"which implements primitive protobuf (%s)") %
(cls.__name__, field.name, field.message_type.name,
type_descriptor.type.__name__, semantic_protobuf_primitive))
elif field.enum_type: # It is an enum.
enum_desc = field.enum_type
enum_dict = {}
enum_descriptions = {}
enum_labels = {}
for enum_value in enum_desc.values:
enum_dict[enum_value.name] = enum_value.number
description = enum_value.GetOptions().Extensions[
semantic_pb2.description]
enum_descriptions[enum_value.name] = description
labels = [
label
for label in enum_value.GetOptions().Extensions[semantic_pb2.label]
]
enum_labels[enum_value.name] = labels
enum_dict = dict((x.name, x.number) for x in enum_desc.values)
type_descriptor = classes_dict["ProtoEnum"](
enum_name=enum_desc.name,
enum=enum_dict,
enum_descriptions=enum_descriptions,
enum_labels=enum_labels,
**kwargs)
# Attach the enum container to the class for easy reference:
setattr(cls, enum_desc.name, type_descriptor.enum_container)
# If we do not recognize the type descriptor we ignore this field.
if type_descriptor is not None:
# If the field is repeated, wrap it in a ProtoList.
if field.label == LABEL_REPEATED:
options = field.GetOptions().Extensions[semantic_pb2.sem_type]
type_descriptor = classes_dict["ProtoList"](
type_descriptor, labels=list(options.label))
try:
cls.AddDescriptor(type_descriptor)
except Exception:
logging.error("Failed to parse protobuf %s", cls)
raise
else:
logging.error("Unknown field type for %s - Ignoring.", field.name)
