222 lines
4.5 KiB
C#
222 lines
4.5 KiB
C#
|
//
|
||
|
// Author:
|
||
|
// Jb Evain (jbevain@gmail.com)
|
||
|
//
|
||
|
// Copyright (c) 2008 - 2015 Jb Evain
|
||
|
// Copyright (c) 2008 - 2011 Novell, Inc.
|
||
|
//
|
||
|
// Licensed under the MIT/X11 license.
|
||
|
//
|
||
|
|
||
|
using MonoFN.Collections.Generic;
|
||
|
using System;
|
||
|
using System.Diagnostics;
|
||
|
using System.Threading;
|
||
|
|
||
|
namespace MonoFN.Cecil {
|
||
|
|
||
|
public struct CustomAttributeArgument {
|
||
|
|
||
|
readonly TypeReference type;
|
||
|
readonly object value;
|
||
|
|
||
|
public TypeReference Type {
|
||
|
get { return type; }
|
||
|
}
|
||
|
|
||
|
public object Value {
|
||
|
get { return value; }
|
||
|
}
|
||
|
|
||
|
public CustomAttributeArgument (TypeReference type, object value)
|
||
|
{
|
||
|
Mixin.CheckType (type);
|
||
|
this.type = type;
|
||
|
this.value = value;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public struct CustomAttributeNamedArgument {
|
||
|
|
||
|
readonly string name;
|
||
|
readonly CustomAttributeArgument argument;
|
||
|
|
||
|
public string Name {
|
||
|
get { return name; }
|
||
|
}
|
||
|
|
||
|
public CustomAttributeArgument Argument {
|
||
|
get { return argument; }
|
||
|
}
|
||
|
|
||
|
public CustomAttributeNamedArgument (string name, CustomAttributeArgument argument)
|
||
|
{
|
||
|
Mixin.CheckName (name);
|
||
|
this.name = name;
|
||
|
this.argument = argument;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public interface ICustomAttribute {
|
||
|
|
||
|
TypeReference AttributeType { get; }
|
||
|
|
||
|
bool HasFields { get; }
|
||
|
bool HasProperties { get; }
|
||
|
bool HasConstructorArguments { get; }
|
||
|
Collection<CustomAttributeNamedArgument> Fields { get; }
|
||
|
Collection<CustomAttributeNamedArgument> Properties { get; }
|
||
|
Collection<CustomAttributeArgument> ConstructorArguments { get; }
|
||
|
}
|
||
|
|
||
|
[DebuggerDisplay ("{AttributeType}")]
|
||
|
public sealed class CustomAttribute : ICustomAttribute {
|
||
|
|
||
|
internal CustomAttributeValueProjection projection;
|
||
|
readonly internal uint signature;
|
||
|
internal bool resolved;
|
||
|
MethodReference constructor;
|
||
|
byte [] blob;
|
||
|
internal Collection<CustomAttributeArgument> arguments;
|
||
|
internal Collection<CustomAttributeNamedArgument> fields;
|
||
|
internal Collection<CustomAttributeNamedArgument> properties;
|
||
|
|
||
|
public MethodReference Constructor {
|
||
|
get { return constructor; }
|
||
|
set { constructor = value; }
|
||
|
}
|
||
|
|
||
|
public TypeReference AttributeType {
|
||
|
get { return constructor.DeclaringType; }
|
||
|
}
|
||
|
|
||
|
public bool IsResolved {
|
||
|
get { return resolved; }
|
||
|
}
|
||
|
|
||
|
public bool HasConstructorArguments {
|
||
|
get {
|
||
|
Resolve ();
|
||
|
|
||
|
return !arguments.IsNullOrEmpty ();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public Collection<CustomAttributeArgument> ConstructorArguments {
|
||
|
get {
|
||
|
Resolve ();
|
||
|
|
||
|
if (arguments == null)
|
||
|
Interlocked.CompareExchange (ref arguments, new Collection<CustomAttributeArgument> (), null);
|
||
|
|
||
|
return arguments;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public bool HasFields {
|
||
|
get {
|
||
|
Resolve ();
|
||
|
|
||
|
return !fields.IsNullOrEmpty ();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public Collection<CustomAttributeNamedArgument> Fields {
|
||
|
get {
|
||
|
Resolve ();
|
||
|
|
||
|
if (fields == null)
|
||
|
Interlocked.CompareExchange (ref fields, new Collection<CustomAttributeNamedArgument> (), null);
|
||
|
|
||
|
return fields;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public bool HasProperties {
|
||
|
get {
|
||
|
Resolve ();
|
||
|
|
||
|
return !properties.IsNullOrEmpty ();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public Collection<CustomAttributeNamedArgument> Properties {
|
||
|
get {
|
||
|
Resolve ();
|
||
|
|
||
|
if (properties == null)
|
||
|
Interlocked.CompareExchange (ref properties, new Collection<CustomAttributeNamedArgument> (), null);
|
||
|
|
||
|
return properties;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
internal bool HasImage {
|
||
|
get { return constructor != null && constructor.HasImage; }
|
||
|
}
|
||
|
|
||
|
internal ModuleDefinition Module {
|
||
|
get { return constructor.Module; }
|
||
|
}
|
||
|
|
||
|
internal CustomAttribute (uint signature, MethodReference constructor)
|
||
|
{
|
||
|
this.signature = signature;
|
||
|
this.constructor = constructor;
|
||
|
this.resolved = false;
|
||
|
}
|
||
|
|
||
|
public CustomAttribute (MethodReference constructor)
|
||
|
{
|
||
|
this.constructor = constructor;
|
||
|
this.resolved = true;
|
||
|
}
|
||
|
|
||
|
public CustomAttribute (MethodReference constructor, byte [] blob)
|
||
|
{
|
||
|
this.constructor = constructor;
|
||
|
this.resolved = false;
|
||
|
this.blob = blob;
|
||
|
}
|
||
|
|
||
|
public byte [] GetBlob ()
|
||
|
{
|
||
|
if (blob != null)
|
||
|
return blob;
|
||
|
|
||
|
if (!HasImage)
|
||
|
throw new NotSupportedException ();
|
||
|
|
||
|
return Module.Read (ref blob, this, (attribute, reader) => reader.ReadCustomAttributeBlob (attribute.signature));
|
||
|
}
|
||
|
|
||
|
void Resolve ()
|
||
|
{
|
||
|
if (resolved || !HasImage)
|
||
|
return;
|
||
|
|
||
|
lock (Module.SyncRoot) {
|
||
|
if (resolved)
|
||
|
return;
|
||
|
|
||
|
Module.Read (this, (attribute, reader) => {
|
||
|
try {
|
||
|
reader.ReadCustomAttributeSignature (attribute);
|
||
|
resolved = true;
|
||
|
}
|
||
|
catch (ResolutionException) {
|
||
|
if (arguments != null)
|
||
|
arguments.Clear ();
|
||
|
if (fields != null)
|
||
|
fields.Clear ();
|
||
|
if (properties != null)
|
||
|
properties.Clear ();
|
||
|
|
||
|
resolved = false;
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|