mono/metadata/custom-attrs.c

   1 /*
   2  * custom-attrs.c: Custom attributes.
   3  *
   4  * Author:
   5  *   Paolo Molaro (lupus@ximian.com)
   6  *
   7  * Copyright 2001-2003 Ximian, Inc (http://www.ximian.com)
   8  * Copyright 2004-2009 Novell, Inc (http://www.novell.com)
   9  * Copyright 2011 Rodrigo Kumpera
  10  * Copyright 2016 Microsoft
  11  *
  12  * Licensed under the MIT license. See LICENSE file in the project root for full license information.
  13  */
  14 #include <config.h>
  15 #include "mono/metadata/assembly.h"
  16 #include "mono/metadata/gc-internals.h"
  17 #include "mono/metadata/mono-endian.h"
  18 #include "mono/metadata/object-internals.h"
  19 #include "mono/metadata/custom-attrs-internals.h"
  20 #include "mono/metadata/sre-internals.h"
  21 #include "mono/metadata/reflection-internals.h"
  22 #include "mono/metadata/tabledefs.h"
  23 #include "mono/metadata/tokentype.h"
  24 #include "mono/metadata/verify-internals.h"
  25 #include "mono/utils/checked-build.h"
  26
  27
  28 #define CHECK_ADD4_OVERFLOW_UN(a, b) ((guint32)(0xFFFFFFFFU) - (guint32)(b) < (guint32)(a))
  29 #define CHECK_ADD8_OVERFLOW_UN(a, b) ((guint64)(0xFFFFFFFFFFFFFFFFUL) - (guint64)(b) < (guint64)(a))
  30
  31 #if SIZEOF_VOID_P == 4
  32 #define CHECK_ADDP_OVERFLOW_UN(a,b) CHECK_ADD4_OVERFLOW_UN(a, b)
  33 #else
  34 #define CHECK_ADDP_OVERFLOW_UN(a,b) CHECK_ADD8_OVERFLOW_UN(a, b)
  35 #endif
  36
  37 #define ADDP_IS_GREATER_OR_OVF(a, b, c) (((a) + (b) > (c)) || CHECK_ADDP_OVERFLOW_UN (a, b))
  38 #define ADD_IS_GREATER_OR_OVF(a, b, c) (((a) + (b) > (c)) || CHECK_ADD4_OVERFLOW_UN (a, b))
  39
  40 static gboolean type_is_reference (MonoType *type);
  41
  42 static GENERATE_GET_CLASS_WITH_CACHE (custom_attribute_typed_argument, System.Reflection, CustomAttributeTypedArgument);
  43 static GENERATE_GET_CLASS_WITH_CACHE (custom_attribute_named_argument, System.Reflection, CustomAttributeNamedArgument);
  44
  45 /*
  46  * LOCKING: Acquires the loader lock.
  47  */
  48 static MonoCustomAttrInfo*
  49 lookup_custom_attr (MonoImage *image, gpointer member)
  50 {
  51         MONO_REQ_GC_NEUTRAL_MODE;
  52
  53         MonoCustomAttrInfo* res;
  54
  55         res = (MonoCustomAttrInfo *)mono_image_property_lookup (image, member, MONO_PROP_DYNAMIC_CATTR);
  56
  57         if (!res)
  58                 return NULL;
  59
  60         res = (MonoCustomAttrInfo *)g_memdup (res, MONO_SIZEOF_CUSTOM_ATTR_INFO + sizeof (MonoCustomAttrEntry) * res->num_attrs);
  61         res->cached = 0;
  62         return res;
  63 }
  64
  65 static gboolean
  66 custom_attr_visible (MonoImage *image, MonoReflectionCustomAttr *cattr)
  67 {
  68         MONO_REQ_GC_UNSAFE_MODE;
  69
  70         /* FIXME: Need to do more checks */
  71         if (cattr->ctor->method && (cattr->ctor->method->klass->image != image)) {
  72                 int visibility = mono_class_get_flags (cattr->ctor->method->klass) & TYPE_ATTRIBUTE_VISIBILITY_MASK;
  73
  74                 if ((visibility != TYPE_ATTRIBUTE_PUBLIC) && (visibility != TYPE_ATTRIBUTE_NESTED_PUBLIC))
  75                         return FALSE;
  76         }
  77
  78         return TRUE;
  79 }
  80
  81 static gboolean
  82 type_is_reference (MonoType *type)
  83 {
  84         switch (type->type) {
  85         case MONO_TYPE_BOOLEAN:
  86         case MONO_TYPE_CHAR:
  87         case MONO_TYPE_U:
  88         case MONO_TYPE_I:
  89         case MONO_TYPE_U1:
  90         case MONO_TYPE_I1:
  91         case MONO_TYPE_U2:
  92         case MONO_TYPE_I2:
  93         case MONO_TYPE_U4:
  94         case MONO_TYPE_I4:
  95         case MONO_TYPE_U8:
  96         case MONO_TYPE_I8:
  97         case MONO_TYPE_R8:
  98         case MONO_TYPE_R4:
  99         case MONO_TYPE_VALUETYPE:
 100                 return FALSE;
 101         default:
 102                 return TRUE;
 103         }
 104 }
 105
 106 static void
 107 free_param_data (MonoMethodSignature *sig, void **params) {
 108         int i;
 109         for (i = 0; i < sig->param_count; ++i) {
 110                 if (!type_is_reference (sig->params [i]))
 111                         g_free (params [i]);
 112         }
 113 }
 114
 115 /*
 116  * Find the field index in the metadata FieldDef table.
 117  */
 118 static guint32
 119 find_field_index (MonoClass *klass, MonoClassField *field) {
 120         int i;
 121
 122         int fcount = mono_class_get_field_count (klass);
 123         for (i = 0; i < fcount; ++i) {
 124                 if (field == &klass->fields [i])
 125                         return mono_class_get_first_field_idx (klass) + 1 + i;
 126         }
 127         return 0;
 128 }
 129
 130 /*
 131  * Find the property index in the metadata Property table.
 132  */
 133 static guint32
 134 find_property_index (MonoClass *klass, MonoProperty *property)
 135 {
 136         int i;
 137         MonoClassPropertyInfo *info = mono_class_get_property_info (klass);
 138
 139         for (i = 0; i < info->count; ++i) {
 140                 if (property == &info->properties [i])
 141                         return info->first + 1 + i;
 142         }
 143         return 0;
 144 }
 145
 146 /*
 147  * Find the event index in the metadata Event table.
 148  */
 149 static guint32
 150 find_event_index (MonoClass *klass, MonoEvent *event)
 151 {
 152         int i;
 153         MonoClassEventInfo *info = mono_class_get_event_info (klass);
 154
 155         for (i = 0; i < info->count; ++i) {
 156                 if (event == &info->events [i])
 157                         return info->first + 1 + i;
 158         }
 159         return 0;
 160 }
 161
 162 /*
 163  * Load the type with name @n on behalf of image @image.  On failure sets @error and returns NULL.
 164  * The @is_enum flag only affects the error message that's displayed on failure.
 165  */
 166 static MonoType*
 167 cattr_type_from_name (char *n, MonoImage *image, gboolean is_enum, MonoError *error)
 168 {
 169         MonoError inner_error;
 170         MonoType *t = mono_reflection_type_from_name_checked (n, image, &inner_error);
 171         if (!t) {
 172                 mono_error_set_type_load_name (error, g_strdup(n), NULL,
 173                                                "Could not load %s %s while decoding custom attribute: %s",
 174                                                is_enum ? "enum type": "type",
 175                                                n,
 176                                                mono_error_get_message (&inner_error));
 177                 mono_error_cleanup (&inner_error);
 178                 return NULL;
 179         }
 180         return t;
 181 }
 182
 183 static MonoClass*
 184 load_cattr_enum_type (MonoImage *image, const char *p, const char **end, MonoError *error)
 185 {
 186         char *n;
 187         MonoType *t;
 188         int slen = mono_metadata_decode_value (p, &p);
 189
 190         mono_error_init (error);
 191
 192         n = (char *)g_memdup (p, slen + 1);
 193         n [slen] = 0;
 194         t = cattr_type_from_name (n, image, TRUE, error);
 195         g_free (n);
 196         return_val_if_nok (error, NULL);
 197         p += slen;
 198         *end = p;
 199         return mono_class_from_mono_type (t);
 200 }
 201
 202 static void*
 203 load_cattr_value (MonoImage *image, MonoType *t, const char *p, const char **end, MonoError *error)
 204 {
 205         int slen, type = t->type;
 206         MonoClass *tklass = t->data.klass;
 207
 208         mono_error_init (error);
 209
 210 handle_enum:
 211         switch (type) {
 212         case MONO_TYPE_U1:
 213         case MONO_TYPE_I1:
 214         case MONO_TYPE_BOOLEAN: {
 215                 MonoBoolean *bval = (MonoBoolean *)g_malloc (sizeof (MonoBoolean));
 216                 *bval = *p;
 217                 *end = p + 1;
 218                 return bval;
 219         }
 220         case MONO_TYPE_CHAR:
 221         case MONO_TYPE_U2:
 222         case MONO_TYPE_I2: {
 223                 guint16 *val = (guint16 *)g_malloc (sizeof (guint16));
 224                 *val = read16 (p);
 225                 *end = p + 2;
 226                 return val;
 227         }
 228 #if SIZEOF_VOID_P == 4
 229         case MONO_TYPE_U:
 230         case MONO_TYPE_I:
 231 #endif
 232         case MONO_TYPE_R4:
 233         case MONO_TYPE_U4:
 234         case MONO_TYPE_I4: {
 235                 guint32 *val = (guint32 *)g_malloc (sizeof (guint32));
 236                 *val = read32 (p);
 237                 *end = p + 4;
 238                 return val;
 239         }
 240 #if SIZEOF_VOID_P == 8
 241         case MONO_TYPE_U: /* error out instead? this should probably not happen */
 242         case MONO_TYPE_I:
 243 #endif
 244         case MONO_TYPE_U8:
 245         case MONO_TYPE_I8: {
 246                 guint64 *val = (guint64 *)g_malloc (sizeof (guint64));
 247                 *val = read64 (p);
 248                 *end = p + 8;
 249                 return val;
 250         }
 251         case MONO_TYPE_R8: {
 252                 double *val = (double *)g_malloc (sizeof (double));
 253                 readr8 (p, val);
 254                 *end = p + 8;
 255                 return val;
 256         }
 257         case MONO_TYPE_VALUETYPE:
 258                 if (t->data.klass->enumtype) {
 259                         type = mono_class_enum_basetype (t->data.klass)->type;
 260                         goto handle_enum;
 261                 } else {
 262                         MonoClass *k =  t->data.klass;
 263
 264                         if (mono_is_corlib_image (k->image) && strcmp (k->name_space, "System") == 0 && strcmp (k->name, "DateTime") == 0){
 265                                 guint64 *val = (guint64 *)g_malloc (sizeof (guint64));
 266                                 *val = read64 (p);
 267                                 *end = p + 8;
 268                                 return val;
 269                         }
 270                 }
 271                 g_error ("generic valutype %s not handled in custom attr value decoding", t->data.klass->name);
 272                 break;
 273
 274         case MONO_TYPE_STRING:
 275                 if (*p == (char)0xFF) {
 276                         *end = p + 1;
 277                         return NULL;
 278                 }
 279                 slen = mono_metadata_decode_value (p, &p);
 280                 *end = p + slen;
 281                 return mono_string_new_len_checked (mono_domain_get (), p, slen, error);
 282         case MONO_TYPE_CLASS: {
 283                 MonoReflectionType *rt;
 284                 char *n;
 285                 MonoType *t;
 286                 if (*p == (char)0xFF) {
 287                         *end = p + 1;
 288                         return NULL;
 289                 }
 290 handle_type:
 291                 slen = mono_metadata_decode_value (p, &p);
 292                 n = (char *)g_memdup (p, slen + 1);
 293                 n [slen] = 0;
 294                 t = cattr_type_from_name (n, image, FALSE, error);
 295                 g_free (n);
 296                 return_val_if_nok (error, NULL);
 297                 *end = p + slen;
 298
 299                 rt = mono_type_get_object_checked (mono_domain_get (), t, error);
 300                 if (!mono_error_ok (error))
 301                         return NULL;
 302
 303                 return rt;
 304         }
 305         case MONO_TYPE_OBJECT: {
 306                 char subt = *p++;
 307                 MonoObject *obj;
 308                 MonoClass *subc = NULL;
 309                 void *val;
 310
 311                 if (subt == 0x50) {
 312                         goto handle_type;
 313                 } else if (subt == 0x0E) {
 314                         type = MONO_TYPE_STRING;
 315                         goto handle_enum;
 316                 } else if (subt == 0x1D) {
 317                         MonoType simple_type = {{0}};
 318                         int etype = *p;
 319                         p ++;
 320
 321                         type = MONO_TYPE_SZARRAY;
 322                         if (etype == 0x50) {
 323                                 tklass = mono_defaults.systemtype_class;
 324                         } else if (etype == 0x55) {
 325                                 tklass = load_cattr_enum_type (image, p, &p, error);
 326                                 if (!mono_error_ok (error))
 327                                         return NULL;
 328                         } else {
 329                                 if (etype == 0x51)
 330                                         /* See Partition II, Appendix B3 */
 331                                         etype = MONO_TYPE_OBJECT;
 332                                 simple_type.type = (MonoTypeEnum)etype;
 333                                 tklass = mono_class_from_mono_type (&simple_type);
 334                         }
 335                         goto handle_enum;
 336                 } else if (subt == 0x55) {
 337                         char *n;
 338                         MonoType *t;
 339                         slen = mono_metadata_decode_value (p, &p);
 340                         n = (char *)g_memdup (p, slen + 1);
 341                         n [slen] = 0;
 342                         t = cattr_type_from_name (n, image, FALSE, error);
 343                         g_free (n);
 344                         return_val_if_nok (error, NULL);
 345                         p += slen;
 346                         subc = mono_class_from_mono_type (t);
 347                 } else if (subt >= MONO_TYPE_BOOLEAN && subt <= MONO_TYPE_R8) {
 348                         MonoType simple_type = {{0}};
 349                         simple_type.type = (MonoTypeEnum)subt;
 350                         subc = mono_class_from_mono_type (&simple_type);
 351                 } else {
 352                         g_error ("Unknown type 0x%02x for object type encoding in custom attr", subt);
 353                 }
 354                 val = load_cattr_value (image, &subc->byval_arg, p, end, error);
 355                 obj = NULL;
 356                 if (mono_error_ok (error)) {
 357                         obj = mono_object_new_checked (mono_domain_get (), subc, error);
 358                         g_assert (!subc->has_references);
 359                         if (mono_error_ok (error))
 360                                 mono_gc_memmove_atomic ((char*)obj + sizeof (MonoObject), val, mono_class_value_size (subc, NULL));
 361                 }
 362
 363                 g_free (val);
 364                 return obj;
 365         }
 366         case MONO_TYPE_SZARRAY: {
 367                 MonoArray *arr;
 368                 guint32 i, alen, basetype;
 369                 alen = read32 (p);
 370                 p += 4;
 371                 if (alen == 0xffffffff) {
 372                         *end = p;
 373                         return NULL;
 374                 }
 375                 arr = mono_array_new_checked (mono_domain_get(), tklass, alen, error);
 376                 return_val_if_nok (error, NULL);
 377                 basetype = tklass->byval_arg.type;
 378                 if (basetype == MONO_TYPE_VALUETYPE && tklass->enumtype)
 379                         basetype = mono_class_enum_basetype (tklass)->type;
 380                 switch (basetype)
 381                 {
 382                         case MONO_TYPE_U1:
 383                         case MONO_TYPE_I1:
 384                         case MONO_TYPE_BOOLEAN:
 385                                 for (i = 0; i < alen; i++) {
 386                                         MonoBoolean val = *p++;
 387                                         mono_array_set (arr, MonoBoolean, i, val);
 388                                 }
 389                                 break;
 390                         case MONO_TYPE_CHAR:
 391                         case MONO_TYPE_U2:
 392                         case MONO_TYPE_I2:
 393                                 for (i = 0; i < alen; i++) {
 394                                         guint16 val = read16 (p);
 395                                         mono_array_set (arr, guint16, i, val);
 396                                         p += 2;
 397                                 }
 398                                 break;
 399                         case MONO_TYPE_R4:
 400                         case MONO_TYPE_U4:
 401                         case MONO_TYPE_I4:
 402                                 for (i = 0; i < alen; i++) {
 403                                         guint32 val = read32 (p);
 404                                         mono_array_set (arr, guint32, i, val);
 405                                         p += 4;
 406                                 }
 407                                 break;
 408                         case MONO_TYPE_R8:
 409                                 for (i = 0; i < alen; i++) {
 410                                         double val;
 411                                         readr8 (p, &val);
 412                                         mono_array_set (arr, double, i, val);
 413                                         p += 8;
 414                                 }
 415                                 break;
 416                         case MONO_TYPE_U8:
 417                         case MONO_TYPE_I8:
 418                                 for (i = 0; i < alen; i++) {
 419                                         guint64 val = read64 (p);
 420                                         mono_array_set (arr, guint64, i, val);
 421                                         p += 8;
 422                                 }
 423                                 break;
 424                         case MONO_TYPE_CLASS:
 425                         case MONO_TYPE_OBJECT:
 426                         case MONO_TYPE_STRING:
 427                         case MONO_TYPE_SZARRAY:
 428                                 for (i = 0; i < alen; i++) {
 429                                         MonoObject *item = (MonoObject *)load_cattr_value (image, &tklass->byval_arg, p, &p, error);
 430                                         if (!mono_error_ok (error))
 431                                                 return NULL;
 432                                         mono_array_setref (arr, i, item);
 433                                 }
 434                                 break;
 435                         default:
 436                                 g_error ("Type 0x%02x not handled in custom attr array decoding", basetype);
 437                 }
 438                 *end=p;
 439                 return arr;
 440         }
 441         default:
 442                 g_error ("Type 0x%02x not handled in custom attr value decoding", type);
 443         }
 444         return NULL;
 445 }
 446
 447 static MonoObject*
 448 load_cattr_value_boxed (MonoDomain *domain, MonoImage *image, MonoType *t, const char* p, const char** end, MonoError *error)
 449 {
 450         mono_error_init (error);
 451
 452         gboolean is_ref = type_is_reference (t);
 453
 454         void *val = load_cattr_value (image, t, p, end, error);
 455         if (!is_ok (error)) {
 456                 if (is_ref)
 457                         g_free (val);
 458                 return NULL;
 459         }
 460
 461         if (is_ref)
 462                 return (MonoObject*)val;
 463
 464         MonoObject *boxed = mono_value_box_checked (domain, mono_class_from_mono_type (t), val, error);
 465         g_free (val);
 466         return boxed;
 467 }
 468
 469 static MonoObject*
 470 create_cattr_typed_arg (MonoType *t, MonoObject *val, MonoError *error)
 471 {
 472         static MonoMethod *ctor;
 473         MonoObject *retval;
 474         void *params [2], *unboxed;
 475
 476         mono_error_init (error);
 477
 478         if (!ctor)
 479                 ctor = mono_class_get_method_from_name (mono_class_get_custom_attribute_typed_argument_class (), ".ctor", 2);
 480
 481         params [0] = mono_type_get_object_checked (mono_domain_get (), t, error);
 482         return_val_if_nok (error, NULL);
 483
 484         params [1] = val;
 485         retval = mono_object_new_checked (mono_domain_get (), mono_class_get_custom_attribute_typed_argument_class (), error);
 486         return_val_if_nok (error, NULL);
 487         unboxed = mono_object_unbox (retval);
 488
 489         mono_runtime_invoke_checked (ctor, unboxed, params, error);
 490         return_val_if_nok (error, NULL);
 491
 492         return retval;
 493 }
 494
 495 static MonoObject*
 496 create_cattr_named_arg (void *minfo, MonoObject *typedarg, MonoError *error)
 497 {
 498         static MonoMethod *ctor;
 499         MonoObject *retval;
 500         void *unboxed, *params [2];
 501
 502         mono_error_init (error);
 503
 504         if (!ctor)
 505                 ctor = mono_class_get_method_from_name (mono_class_get_custom_attribute_named_argument_class (), ".ctor", 2);
 506
 507         params [0] = minfo;
 508         params [1] = typedarg;
 509         retval = mono_object_new_checked (mono_domain_get (), mono_class_get_custom_attribute_named_argument_class (), error);
 510         return_val_if_nok (error, NULL);
 511
 512         unboxed = mono_object_unbox (retval);
 513
 514         mono_runtime_invoke_checked (ctor, unboxed, params, error);
 515         return_val_if_nok (error, NULL);
 516
 517         return retval;
 518 }
 519
 520
 521 MonoCustomAttrInfo*
 522 mono_custom_attrs_from_builders (MonoImage *alloc_img, MonoImage *image, MonoArray *cattrs)
 523 {
 524         MONO_REQ_GC_UNSAFE_MODE;
 525
 526         int i, index, count, not_visible;
 527         MonoCustomAttrInfo *ainfo;
 528         MonoReflectionCustomAttr *cattr;
 529
 530         if (!cattrs)
 531                 return NULL;
 532         /* FIXME: check in assembly the Run flag is set */
 533
 534         count = mono_array_length (cattrs);
 535
 536         /* Skip nonpublic attributes since MS.NET seems to do the same */
 537         /* FIXME: This needs to be done more globally */
 538         not_visible = 0;
 539         for (i = 0; i < count; ++i) {
 540                 cattr = (MonoReflectionCustomAttr*)mono_array_get (cattrs, gpointer, i);
 541                 if (!custom_attr_visible (image, cattr))
 542                         not_visible ++;
 543         }
 544
 545         int num_attrs = count - not_visible;
 546         ainfo = (MonoCustomAttrInfo *)mono_image_g_malloc0 (alloc_img, MONO_SIZEOF_CUSTOM_ATTR_INFO + sizeof (MonoCustomAttrEntry) * num_attrs);
 547
 548         ainfo->image = image;
 549         ainfo->num_attrs = num_attrs;
 550         ainfo->cached = alloc_img != NULL;
 551         index = 0;
 552         for (i = 0; i < count; ++i) {
 553                 cattr = (MonoReflectionCustomAttr*)mono_array_get (cattrs, gpointer, i);
 554                 if (custom_attr_visible (image, cattr)) {
 555                         unsigned char *saved = (unsigned char *)mono_image_alloc (image, mono_array_length (cattr->data));
 556                         memcpy (saved, mono_array_addr (cattr->data, char, 0), mono_array_length (cattr->data));
 557                         ainfo->attrs [index].ctor = cattr->ctor->method;
 558                         g_assert (cattr->ctor->method);
 559                         ainfo->attrs [index].data = saved;
 560                         ainfo->attrs [index].data_size = mono_array_length (cattr->data);
 561                         index ++;
 562                 }
 563         }
 564         g_assert (index == num_attrs && count == num_attrs + not_visible);
 565
 566         return ainfo;
 567 }
 568
 569
 570 static MonoObject*
 571 create_custom_attr (MonoImage *image, MonoMethod *method, const guchar *data, guint32 len, MonoError *error)
 572 {
 573         const char *p = (const char*)data;
 574         const char *named;
 575         guint32 i, j, num_named;
 576         MonoObject *attr;
 577         void *params_buf [32];
 578         void **params = NULL;
 579         MonoMethodSignature *sig;
 580
 581         mono_error_init (error);
 582
 583         mono_class_init (method->klass);
 584
 585         if (!mono_verifier_verify_cattr_content (image, method, data, len, NULL)) {
 586                 mono_error_set_generic_error (error, "System.Reflection", "CustomAttributeFormatException", "Binary format of the specified custom attribute was invalid.");
 587                 return NULL;
 588         }
 589
 590         if (len == 0) {
 591                 attr = mono_object_new_checked (mono_domain_get (), method->klass, error);
 592                 if (!mono_error_ok (error)) return NULL;
 593
 594                 mono_runtime_invoke_checked (method, attr, NULL, error);
 595                 if (!mono_error_ok (error))
 596                         return NULL;
 597
 598                 return attr;
 599         }
 600
 601         if (len < 2 || read16 (p) != 0x0001) /* Prolog */
 602                 return NULL;
 603
 604         /*g_print ("got attr %s\n", method->klass->name);*/
 605
 606         sig = mono_method_signature (method);
 607         if (sig->param_count < 32) {
 608                 params = params_buf;
 609                 memset (params, 0, sizeof (void*) * sig->param_count);
 610         } else {
 611                 /* Allocate using GC so it gets GC tracking */
 612                 params = (void **)mono_gc_alloc_fixed (sig->param_count * sizeof (void*), MONO_GC_DESCRIPTOR_NULL, MONO_ROOT_SOURCE_REFLECTION, "custom attribute parameters");
 613         }
 614
 615         /* skip prolog */
 616         p += 2;
 617         for (i = 0; i < mono_method_signature (method)->param_count; ++i) {
 618                 params [i] = load_cattr_value (image, mono_method_signature (method)->params [i], p, &p, error);
 619                 if (!mono_error_ok (error))
 620                         goto fail;
 621         }
 622
 623         named = p;
 624         attr = mono_object_new_checked (mono_domain_get (), method->klass, error);
 625         if (!mono_error_ok (error)) goto fail;
 626
 627         MonoObject *exc = NULL;
 628         mono_runtime_try_invoke (method, attr, params, &exc, error);
 629         if (!mono_error_ok (error))
 630                 goto fail;
 631         if (exc) {
 632                 mono_error_set_exception_instance (error, (MonoException*)exc);
 633                 goto fail;
 634         }
 635
 636         num_named = read16 (named);
 637         named += 2;
 638         for (j = 0; j < num_named; j++) {
 639                 gint name_len;
 640                 char *name, named_type, data_type;
 641                 named_type = *named++;
 642                 data_type = *named++; /* type of data */
 643                 if (data_type == MONO_TYPE_SZARRAY)
 644                         data_type = *named++;
 645                 if (data_type == MONO_TYPE_ENUM) {
 646                         gint type_len;
 647                         char *type_name;
 648                         type_len = mono_metadata_decode_blob_size (named, &named);
 649                         type_name = (char *)g_malloc (type_len + 1);
 650                         memcpy (type_name, named, type_len);
 651                         type_name [type_len] = 0;
 652                         named += type_len;
 653                         /* FIXME: lookup the type and check type consistency */
 654                         g_free (type_name);
 655                 }
 656                 name_len = mono_metadata_decode_blob_size (named, &named);
 657                 name = (char *)g_malloc (name_len + 1);
 658                 memcpy (name, named, name_len);
 659                 name [name_len] = 0;
 660                 named += name_len;
 661                 if (named_type == 0x53) {
 662                         MonoClassField *field;
 663                         void *val;
 664
 665                         /* how this fail is a blackbox */
 666                         field = mono_class_get_field_from_name (mono_object_class (attr), name);
 667                         if (!field) {
 668                                 mono_error_set_generic_error (error, "System.Reflection", "CustomAttributeFormatException", "Could not find a field with name %s", name);
 669                                 g_free (name);
 670                                 goto fail;
 671                         }
 672
 673                         val = load_cattr_value (image, field->type, named, &named, error);
 674                         if (!mono_error_ok (error)) {
 675                                 g_free (name);
 676                                 if (!type_is_reference (field->type))
 677                                         g_free (val);
 678                                 goto fail;
 679                         }
 680
 681                         mono_field_set_value (attr, field, val);
 682                         if (!type_is_reference (field->type))
 683                                 g_free (val);
 684                 } else if (named_type == 0x54) {
 685                         MonoProperty *prop;
 686                         void *pparams [1];
 687                         MonoType *prop_type;
 688
 689                         prop = mono_class_get_property_from_name (mono_object_class (attr), name);
 690
 691                         if (!prop) {
 692                                 mono_error_set_generic_error (error, "System.Reflection", "CustomAttributeFormatException", "Could not find a property with name %s", name);
 693                                 g_free (name);
 694                                 goto fail;
 695                         }
 696
 697                         if (!prop->set) {
 698                                 mono_error_set_generic_error (error, "System.Reflection", "CustomAttributeFormatException", "Could not find the setter for %s", name);
 699                                 g_free (name);
 700                                 goto fail;
 701                         }
 702
 703                         /* can we have more that 1 arg in a custom attr named property? */
 704                         prop_type = prop->get? mono_method_signature (prop->get)->ret :
 705                              mono_method_signature (prop->set)->params [mono_method_signature (prop->set)->param_count - 1];
 706
 707                         pparams [0] = load_cattr_value (image, prop_type, named, &named, error);
 708                         if (!mono_error_ok (error)) {
 709                                 g_free (name);
 710                                 if (!type_is_reference (prop_type))
 711                                         g_free (pparams [0]);
 712                                 goto fail;
 713                         }
 714
 715
 716                         mono_property_set_value_checked (prop, attr, pparams, error);
 717                         if (!type_is_reference (prop_type))
 718                                 g_free (pparams [0]);
 719                         if (!is_ok (error)) {
 720                                 g_free (name);
 721                                 goto fail;
 722                         }
 723                 }
 724                 g_free (name);
 725         }
 726
 727         free_param_data (method->signature, params);
 728         if (params != params_buf)
 729                 mono_gc_free_fixed (params);
 730
 731         return attr;
 732
 733 fail:
 734         free_param_data (method->signature, params);
 735         if (params != params_buf)
 736                 mono_gc_free_fixed (params);
 737         return NULL;
 738 }
 739
 740 /*
 741  * mono_reflection_create_custom_attr_data_args:
 742  *
 743  *   Create an array of typed and named arguments from the cattr blob given by DATA.
 744  * TYPED_ARGS and NAMED_ARGS will contain the objects representing the arguments,
 745  * NAMED_ARG_INFO will contain information about the named arguments.
 746  */
 747 void
 748 mono_reflection_create_custom_attr_data_args (MonoImage *image, MonoMethod *method, const guchar *data, guint32 len, MonoArray **typed_args, MonoArray **named_args, CattrNamedArg **named_arg_info, MonoError *error)
 749 {
 750         MonoArray *typedargs, *namedargs;
 751         MonoClass *attrklass;
 752         MonoDomain *domain;
 753         const char *p = (const char*)data;
 754         const char *named;
 755         guint32 i, j, num_named;
 756         CattrNamedArg *arginfo = NULL;
 757
 758         *typed_args = NULL;
 759         *named_args = NULL;
 760         *named_arg_info = NULL;
 761
 762         mono_error_init (error);
 763
 764         if (!mono_verifier_verify_cattr_content (image, method, data, len, NULL)) {
 765                 mono_error_set_generic_error (error, "System.Reflection", "CustomAttributeFormatException", "Binary format of the specified custom attribute was invalid.");
 766                 return;
 767         }
 768
 769         mono_class_init (method->klass);
 770
 771         domain = mono_domain_get ();
 772
 773         if (len < 2 || read16 (p) != 0x0001) /* Prolog */
 774                 return;
 775
 776         typedargs = mono_array_new_checked (domain, mono_get_object_class (), mono_method_signature (method)->param_count, error);
 777         return_if_nok (error);
 778
 779         /* skip prolog */
 780         p += 2;
 781         for (i = 0; i < mono_method_signature (method)->param_count; ++i) {
 782                 MonoObject *obj;
 783
 784                 obj = load_cattr_value_boxed (domain, image, mono_method_signature (method)->params [i], p, &p, error);
 785                 return_if_nok (error);
 786                 mono_array_setref (typedargs, i, obj);
 787         }
 788
 789         named = p;
 790         num_named = read16 (named);
 791         namedargs = mono_array_new_checked (domain, mono_get_object_class (), num_named, error);
 792         return_if_nok (error);
 793         named += 2;
 794         attrklass = method->klass;
 795
 796         arginfo = g_new0 (CattrNamedArg, num_named);
 797         *named_arg_info = arginfo;
 798
 799         for (j = 0; j < num_named; j++) {
 800                 gint name_len;
 801                 char *name, named_type, data_type;
 802                 named_type = *named++;
 803                 data_type = *named++; /* type of data */
 804                 if (data_type == MONO_TYPE_SZARRAY)
 805                         data_type = *named++;
 806                 if (data_type == MONO_TYPE_ENUM) {
 807                         gint type_len;
 808                         char *type_name;
 809                         type_len = mono_metadata_decode_blob_size (named, &named);
 810                         if (ADDP_IS_GREATER_OR_OVF ((const guchar*)named, type_len, data + len))
 811                                 goto fail;
 812
 813                         type_name = (char *)g_malloc (type_len + 1);
 814                         memcpy (type_name, named, type_len);
 815                         type_name [type_len] = 0;
 816                         named += type_len;
 817                         /* FIXME: lookup the type and check type consistency */
 818                         g_free (type_name);
 819                 }
 820                 name_len = mono_metadata_decode_blob_size (named, &named);
 821                 if (ADDP_IS_GREATER_OR_OVF ((const guchar*)named, name_len, data + len))
 822                         goto fail;
 823                 name = (char *)g_malloc (name_len + 1);
 824                 memcpy (name, named, name_len);
 825                 name [name_len] = 0;
 826                 named += name_len;
 827                 if (named_type == 0x53) {
 828                         MonoObject *obj;
 829                         MonoClassField *field = mono_class_get_field_from_name (attrklass, name);
 830
 831                         if (!field) {
 832                                 g_free (name);
 833                                 goto fail;
 834                         }
 835
 836                         arginfo [j].type = field->type;
 837                         arginfo [j].field = field;
 838
 839                         obj = load_cattr_value_boxed (domain, image, field->type, named, &named, error);
 840                         if (!is_ok (error)) {
 841                                 g_free (name);
 842                                 return;
 843                         }
 844                         mono_array_setref (namedargs, j, obj);
 845
 846                 } else if (named_type == 0x54) {
 847                         MonoObject *obj;
 848                         MonoType *prop_type;
 849                         MonoProperty *prop = mono_class_get_property_from_name (attrklass, name);
 850
 851                         if (!prop || !prop->set) {
 852                                 g_free (name);
 853                                 goto fail;
 854                         }
 855
 856                         prop_type = prop->get? mono_method_signature (prop->get)->ret :
 857                              mono_method_signature (prop->set)->params [mono_method_signature (prop->set)->param_count - 1];
 858
 859                         arginfo [j].type = prop_type;
 860                         arginfo [j].prop = prop;
 861
 862                         obj = load_cattr_value_boxed (domain, image, prop_type, named, &named, error);
 863                         if (!is_ok (error)) {
 864                                 g_free (name);
 865                                 return;
 866                         }
 867                         mono_array_setref (namedargs, j, obj);
 868                 }
 869                 g_free (name);
 870         }
 871
 872         *typed_args = typedargs;
 873         *named_args = namedargs;
 874         return;
 875 fail:
 876         mono_error_set_generic_error (error, "System.Reflection", "CustomAttributeFormatException", "Binary format of the specified custom attribute was invalid.");
 877         g_free (arginfo);
 878         *named_arg_info = NULL;
 879 }
 880
 881 static gboolean
 882 reflection_resolve_custom_attribute_data (MonoReflectionMethod *ref_method, MonoReflectionAssembly *assembly, gpointer data, guint32 len, MonoArray **ctor_args, MonoArray **named_args, MonoError *error)
 883 {
 884         MonoDomain *domain;
 885         MonoArray *typedargs, *namedargs;
 886         MonoImage *image;
 887         MonoMethod *method;
 888         CattrNamedArg *arginfo = NULL;
 889         int i;
 890
 891         mono_error_init (error);
 892
 893         *ctor_args = NULL;
 894         *named_args = NULL;
 895
 896         if (len == 0)
 897                 return TRUE;
 898
 899         image = assembly->assembly->image;
 900         method = ref_method->method;
 901         domain = mono_object_domain (ref_method);
 902
 903         if (!mono_class_init (method->klass)) {
 904                 mono_error_set_for_class_failure (error, method->klass);
 905                 goto leave;
 906         }
 907
 908         mono_reflection_create_custom_attr_data_args (image, method, (const guchar *)data, len, &typedargs, &namedargs, &arginfo, error);
 909         if (!is_ok (error))
 910                 goto leave;
 911
 912         if (!typedargs || !namedargs)
 913                 goto leave;
 914
 915         for (i = 0; i < mono_method_signature (method)->param_count; ++i) {
 916                 MonoObject *obj = mono_array_get (typedargs, MonoObject*, i);
 917                 MonoObject *typedarg;
 918
 919                 typedarg = create_cattr_typed_arg (mono_method_signature (method)->params [i], obj, error);
 920                 if (!is_ok (error))
 921                         goto leave;
 922                 mono_array_setref (typedargs, i, typedarg);
 923         }
 924
 925         for (i = 0; i < mono_array_length (namedargs); ++i) {
 926                 MonoObject *obj = mono_array_get (namedargs, MonoObject*, i);
 927                 MonoObject *typedarg, *namedarg, *minfo;
 928
 929                 if (arginfo [i].prop) {
 930                         minfo = (MonoObject*)mono_property_get_object_checked (domain, NULL, arginfo [i].prop, error);
 931                         if (!minfo)
 932                                 goto leave;
 933                 } else {
 934                         minfo = (MonoObject*)mono_field_get_object_checked (domain, NULL, arginfo [i].field, error);
 935                         if (!is_ok (error))
 936                                 goto leave;
 937                 }
 938
 939                 typedarg = create_cattr_typed_arg (arginfo [i].type, obj, error);
 940                 if (!is_ok (error))
 941                         goto leave;
 942                 namedarg = create_cattr_named_arg (minfo, typedarg, error);
 943                 if (!is_ok (error))
 944                         goto leave;
 945
 946                 mono_array_setref (namedargs, i, namedarg);
 947         }
 948
 949         *ctor_args = typedargs;
 950         *named_args = namedargs;
 951
 952 leave:
 953         g_free (arginfo);
 954         return mono_error_ok (error);
 955 }
 956
 957 void
 958 ves_icall_System_Reflection_CustomAttributeData_ResolveArgumentsInternal (MonoReflectionMethod *ref_method, MonoReflectionAssembly *assembly, gpointer data, guint32 len, MonoArray **ctor_args, MonoArray **named_args)
 959 {
 960         MonoError error;
 961         (void) reflection_resolve_custom_attribute_data (ref_method, assembly, data, len, ctor_args, named_args, &error);
 962         mono_error_set_pending_exception (&error);
 963 }
 964
 965 static MonoObject*
 966 create_custom_attr_data (MonoImage *image, MonoCustomAttrEntry *cattr, MonoError *error)
 967 {
 968         static MonoMethod *ctor;
 969
 970         MonoDomain *domain;
 971         MonoObject *attr;
 972         void *params [4];
 973
 974         mono_error_init (error);
 975
 976         g_assert (image->assembly);
 977
 978         if (!ctor)
 979                 ctor = mono_class_get_method_from_name (mono_defaults.customattribute_data_class, ".ctor", 4);
 980
 981         domain = mono_domain_get ();
 982         attr = mono_object_new_checked (domain, mono_defaults.customattribute_data_class, error);
 983         return_val_if_nok (error, NULL);
 984         params [0] = mono_method_get_object_checked (domain, cattr->ctor, NULL, error);
 985         return_val_if_nok (error, NULL);
 986         params [1] = mono_assembly_get_object_checked (domain, image->assembly, error);
 987         return_val_if_nok (error, NULL);
 988         params [2] = (gpointer)&cattr->data;
 989         params [3] = &cattr->data_size;
 990
 991         mono_runtime_invoke_checked (ctor, attr, params, error);
 992         return_val_if_nok (error, NULL);
 993         return attr;
 994 }
 995
 996 static MonoArray*
 997 mono_custom_attrs_construct_by_type (MonoCustomAttrInfo *cinfo, MonoClass *attr_klass, MonoError *error)
 998 {
 999         MonoArray *result;
1000         MonoObject *attr;
1001         int i, n;
1002
1003         mono_error_init (error);
1004
1005         for (i = 0; i < cinfo->num_attrs; ++i) {
1006                 MonoCustomAttrEntry *centry = &cinfo->attrs[i];
1007                 if (!centry->ctor) {
1008                         /* The cattr type is not finished yet */
1009                         /* We should include the type name but cinfo doesn't contain it */
1010                         mono_error_set_type_load_name (error, NULL, NULL, "Custom attribute constructor is null because the custom attribute type is not finished yet.");
1011                         return NULL;
1012                 }
1013         }
1014
1015         n = 0;
1016         if (attr_klass) {
1017                 for (i = 0; i < cinfo->num_attrs; ++i) {
1018                         MonoMethod *ctor = cinfo->attrs[i].ctor;
1019                         g_assert (ctor);
1020                         if (mono_class_is_assignable_from (attr_klass, ctor->klass))
1021                                 n++;
1022                 }
1023         } else {
1024                 n = cinfo->num_attrs;
1025         }
1026
1027         result = mono_array_new_cached (mono_domain_get (), mono_defaults.attribute_class, n, error);
1028         return_val_if_nok (error, NULL);
1029         n = 0;
1030         for (i = 0; i < cinfo->num_attrs; ++i) {
1031                 MonoCustomAttrEntry *centry = &cinfo->attrs [i];
1032                 if (!attr_klass || mono_class_is_assignable_from (attr_klass, centry->ctor->klass)) {
1033                         attr = create_custom_attr (cinfo->image, centry->ctor, centry->data, centry->data_size, error);
1034                         if (!mono_error_ok (error))
1035                                 return result;
1036                         mono_array_setref (result, n, attr);
1037                         n ++;
1038                 }
1039         }
1040         return result;
1041 }
1042
1043 MonoArray*
1044 mono_custom_attrs_construct (MonoCustomAttrInfo *cinfo)
1045 {
1046         MonoError error;
1047         MonoArray *result = mono_custom_attrs_construct_by_type (cinfo, NULL, &error);
1048         mono_error_assert_ok (&error); /*FIXME proper error handling*/
1049
1050         return result;
1051 }
1052
1053 static MonoArray*
1054 mono_custom_attrs_data_construct (MonoCustomAttrInfo *cinfo, MonoError *error)
1055 {
1056         MonoArray *result;
1057         MonoObject *attr;
1058         int i;
1059
1060         mono_error_init (error);
1061         result = mono_array_new_checked (mono_domain_get (), mono_defaults.customattribute_data_class, cinfo->num_attrs, error);
1062         return_val_if_nok (error, NULL);
1063         for (i = 0; i < cinfo->num_attrs; ++i) {
1064                 attr = create_custom_attr_data (cinfo->image, &cinfo->attrs [i], error);
1065                 return_val_if_nok (error, NULL);
1066                 mono_array_setref (result, i, attr);
1067         }
1068         return result;
1069 }
1070
1071 /**
1072  * mono_custom_attrs_from_index:
1073  *
1074  * Returns: NULL if no attributes are found or if a loading error occurs.
1075  */
1076 MonoCustomAttrInfo*
1077 mono_custom_attrs_from_index (MonoImage *image, guint32 idx)
1078 {
1079         MonoError error;
1080         MonoCustomAttrInfo *result = mono_custom_attrs_from_index_checked (image, idx, FALSE, &error);
1081         mono_error_cleanup (&error);
1082         return result;
1083 }
1084 /**
1085  * mono_custom_attrs_from_index_checked:
1086  *
1087  * Returns: NULL if no attributes are found.  On error returns NULL and sets @error.
1088  */
1089 MonoCustomAttrInfo*
1090 mono_custom_attrs_from_index_checked (MonoImage *image, guint32 idx, gboolean ignore_missing, MonoError *error)
1091 {
1092         guint32 mtoken, i, len;
1093         guint32 cols [MONO_CUSTOM_ATTR_SIZE];
1094         MonoTableInfo *ca;
1095         MonoCustomAttrInfo *ainfo;
1096         GList *tmp, *list = NULL;
1097         const char *data;
1098         MonoCustomAttrEntry* attr;
1099
1100         mono_error_init (error);
1101
1102         ca = &image->tables [MONO_TABLE_CUSTOMATTRIBUTE];
1103
1104         i = mono_metadata_custom_attrs_from_index (image, idx);
1105         if (!i)
1106                 return NULL;
1107         i --;
1108         while (i < ca->rows) {
1109                 if (mono_metadata_decode_row_col (ca, i, MONO_CUSTOM_ATTR_PARENT) != idx)
1110                         break;
1111                 list = g_list_prepend (list, GUINT_TO_POINTER (i));
1112                 ++i;
1113         }
1114         len = g_list_length (list);
1115         if (!len)
1116                 return NULL;
1117         ainfo = (MonoCustomAttrInfo *)g_malloc0 (MONO_SIZEOF_CUSTOM_ATTR_INFO + sizeof (MonoCustomAttrEntry) * len);
1118         ainfo->num_attrs = len;
1119         ainfo->image = image;
1120         for (i = len, tmp = list; i != 0; --i, tmp = tmp->next) {
1121                 mono_metadata_decode_row (ca, GPOINTER_TO_UINT (tmp->data), cols, MONO_CUSTOM_ATTR_SIZE);
1122                 mtoken = cols [MONO_CUSTOM_ATTR_TYPE] >> MONO_CUSTOM_ATTR_TYPE_BITS;
1123                 switch (cols [MONO_CUSTOM_ATTR_TYPE] & MONO_CUSTOM_ATTR_TYPE_MASK) {
1124                 case MONO_CUSTOM_ATTR_TYPE_METHODDEF:
1125                         mtoken |= MONO_TOKEN_METHOD_DEF;
1126                         break;
1127                 case MONO_CUSTOM_ATTR_TYPE_MEMBERREF:
1128                         mtoken |= MONO_TOKEN_MEMBER_REF;
1129                         break;
1130                 default:
1131                         g_error ("Unknown table for custom attr type %08x", cols [MONO_CUSTOM_ATTR_TYPE]);
1132                         break;
1133                 }
1134                 attr = &ainfo->attrs [i - 1];
1135                 attr->ctor = mono_get_method_checked (image, mtoken, NULL, NULL, error);
1136                 if (!attr->ctor) {
1137                         g_warning ("Can't find custom attr constructor image: %s mtoken: 0x%08x due to: %s", image->name, mtoken, mono_error_get_message (error));
1138                         if (ignore_missing) {
1139                                 mono_error_cleanup (error);
1140                                 mono_error_init (error);
1141                         } else {
1142                                 g_list_free (list);
1143                                 g_free (ainfo);
1144                                 return NULL;
1145                         }
1146                 }
1147
1148                 if (!mono_verifier_verify_cattr_blob (image, cols [MONO_CUSTOM_ATTR_VALUE], NULL)) {
1149                         /*FIXME raising an exception here doesn't make any sense*/
1150                         g_warning ("Invalid custom attribute blob on image %s for index %x", image->name, idx);
1151                         g_list_free (list);
1152                         g_free (ainfo);
1153                         return NULL;
1154                 }
1155                 data = mono_metadata_blob_heap (image, cols [MONO_CUSTOM_ATTR_VALUE]);
1156                 attr->data_size = mono_metadata_decode_value (data, &data);
1157                 attr->data = (guchar*)data;
1158         }
1159         g_list_free (list);
1160
1161         return ainfo;
1162 }
1163
1164 MonoCustomAttrInfo*
1165 mono_custom_attrs_from_method (MonoMethod *method)
1166 {
1167         MonoError error;
1168         MonoCustomAttrInfo* result = mono_custom_attrs_from_method_checked  (method, &error);
1169         mono_error_cleanup (&error); /* FIXME want a better API that doesn't swallow the error */
1170         return result;
1171 }
1172
1173 MonoCustomAttrInfo*
1174 mono_custom_attrs_from_method_checked (MonoMethod *method, MonoError *error)
1175 {
1176         guint32 idx;
1177
1178         mono_error_init (error);
1179
1180         /*
1181          * An instantiated method has the same cattrs as the generic method definition.
1182          *
1183          * LAMESPEC: The .NET SRE throws an exception for instantiations of generic method builders
1184          *           Note that this stanza is not necessary for non-SRE types, but it's a micro-optimization
1185          */
1186         if (method->is_inflated)
1187                 method = ((MonoMethodInflated *) method)->declaring;
1188
1189         if (method_is_dynamic (method) || image_is_dynamic (method->klass->image))
1190                 return lookup_custom_attr (method->klass->image, method);
1191
1192         if (!method->token)
1193                 /* Synthetic methods */
1194                 return NULL;
1195
1196         idx = mono_method_get_index (method);
1197         idx <<= MONO_CUSTOM_ATTR_BITS;
1198         idx |= MONO_CUSTOM_ATTR_METHODDEF;
1199         return mono_custom_attrs_from_index_checked (method->klass->image, idx, FALSE, error);
1200 }
1201
1202 MonoCustomAttrInfo*
1203 mono_custom_attrs_from_class (MonoClass *klass)
1204 {
1205         MonoError error;
1206         MonoCustomAttrInfo *result = mono_custom_attrs_from_class_checked (klass, &error);
1207         mono_error_cleanup (&error);
1208         return result;
1209 }
1210
1211 MonoCustomAttrInfo*
1212 mono_custom_attrs_from_class_checked (MonoClass *klass, MonoError *error)
1213 {
1214         guint32 idx;
1215
1216         mono_error_init (error);
1217
1218         if (mono_class_is_ginst (klass))
1219                 klass = mono_class_get_generic_class (klass)->container_class;
1220
1221         if (image_is_dynamic (klass->image))
1222                 return lookup_custom_attr (klass->image, klass);
1223
1224         if (klass->byval_arg.type == MONO_TYPE_VAR || klass->byval_arg.type == MONO_TYPE_MVAR) {
1225                 idx = mono_metadata_token_index (klass->sizes.generic_param_token);
1226                 idx <<= MONO_CUSTOM_ATTR_BITS;
1227                 idx |= MONO_CUSTOM_ATTR_GENERICPAR;
1228         } else {
1229                 idx = mono_metadata_token_index (klass->type_token);
1230                 idx <<= MONO_CUSTOM_ATTR_BITS;
1231                 idx |= MONO_CUSTOM_ATTR_TYPEDEF;
1232         }
1233         return mono_custom_attrs_from_index_checked (klass->image, idx, FALSE, error);
1234 }
1235
1236 MonoCustomAttrInfo*
1237 mono_custom_attrs_from_assembly (MonoAssembly *assembly)
1238 {
1239         MonoError error;
1240         MonoCustomAttrInfo *result = mono_custom_attrs_from_assembly_checked (assembly, FALSE, &error);
1241         mono_error_cleanup (&error);
1242         return result;
1243 }
1244
1245 MonoCustomAttrInfo*
1246 mono_custom_attrs_from_assembly_checked (MonoAssembly *assembly, gboolean ignore_missing, MonoError *error)
1247 {
1248         guint32 idx;
1249
1250         mono_error_init (error);
1251
1252         if (image_is_dynamic (assembly->image))
1253                 return lookup_custom_attr (assembly->image, assembly);
1254         idx = 1; /* there is only one assembly */
1255         idx <<= MONO_CUSTOM_ATTR_BITS;
1256         idx |= MONO_CUSTOM_ATTR_ASSEMBLY;
1257         return mono_custom_attrs_from_index_checked (assembly->image, idx, ignore_missing, error);
1258 }
1259
1260 static MonoCustomAttrInfo*
1261 mono_custom_attrs_from_module (MonoImage *image, MonoError *error)
1262 {
1263         guint32 idx;
1264
1265         if (image_is_dynamic (image))
1266                 return lookup_custom_attr (image, image);
1267         idx = 1; /* there is only one module */
1268         idx <<= MONO_CUSTOM_ATTR_BITS;
1269         idx |= MONO_CUSTOM_ATTR_MODULE;
1270         return mono_custom_attrs_from_index_checked (image, idx, FALSE, error);
1271 }
1272
1273 MonoCustomAttrInfo*
1274 mono_custom_attrs_from_property (MonoClass *klass, MonoProperty *property)
1275 {
1276         MonoError error;
1277         MonoCustomAttrInfo * result = mono_custom_attrs_from_property_checked (klass, property, &error);
1278         mono_error_cleanup (&error);
1279         return result;
1280 }
1281
1282 MonoCustomAttrInfo*
1283 mono_custom_attrs_from_property_checked (MonoClass *klass, MonoProperty *property, MonoError *error)
1284 {
1285         guint32 idx;
1286
1287         if (image_is_dynamic (klass->image)) {
1288                 property = mono_metadata_get_corresponding_property_from_generic_type_definition (property);
1289                 return lookup_custom_attr (klass->image, property);
1290         }
1291         idx = find_property_index (klass, property);
1292         idx <<= MONO_CUSTOM_ATTR_BITS;
1293         idx |= MONO_CUSTOM_ATTR_PROPERTY;
1294         return mono_custom_attrs_from_index_checked (klass->image, idx, FALSE, error);
1295 }
1296
1297 MonoCustomAttrInfo*
1298 mono_custom_attrs_from_event (MonoClass *klass, MonoEvent *event)
1299 {
1300         MonoError error;
1301         MonoCustomAttrInfo * result = mono_custom_attrs_from_event_checked (klass, event, &error);
1302         mono_error_cleanup (&error);
1303         return result;
1304 }
1305
1306 MonoCustomAttrInfo*
1307 mono_custom_attrs_from_event_checked (MonoClass *klass, MonoEvent *event, MonoError *error)
1308 {
1309         guint32 idx;
1310
1311         if (image_is_dynamic (klass->image)) {
1312                 event = mono_metadata_get_corresponding_event_from_generic_type_definition (event);
1313                 return lookup_custom_attr (klass->image, event);
1314         }
1315         idx = find_event_index (klass, event);
1316         idx <<= MONO_CUSTOM_ATTR_BITS;
1317         idx |= MONO_CUSTOM_ATTR_EVENT;
1318         return mono_custom_attrs_from_index_checked (klass->image, idx, FALSE, error);
1319 }
1320
1321 MonoCustomAttrInfo*
1322 mono_custom_attrs_from_field (MonoClass *klass, MonoClassField *field)
1323 {
1324         MonoError error;
1325         MonoCustomAttrInfo * result = mono_custom_attrs_from_field_checked (klass, field, &error);
1326         mono_error_cleanup (&error);
1327         return result;
1328 }
1329
1330 MonoCustomAttrInfo*
1331 mono_custom_attrs_from_field_checked (MonoClass *klass, MonoClassField *field, MonoError *error)
1332 {
1333         guint32 idx;
1334         mono_error_init (error);
1335
1336         if (image_is_dynamic (klass->image)) {
1337                 field = mono_metadata_get_corresponding_field_from_generic_type_definition (field);
1338                 return lookup_custom_attr (klass->image, field);
1339         }
1340         idx = find_field_index (klass, field);
1341         idx <<= MONO_CUSTOM_ATTR_BITS;
1342         idx |= MONO_CUSTOM_ATTR_FIELDDEF;
1343         return mono_custom_attrs_from_index_checked (klass->image, idx, FALSE, error);
1344 }
1345
1346 /**
1347  * mono_custom_attrs_from_param:
1348  * @method: handle to the method that we want to retrieve custom parameter information from
1349  * @param: parameter number, where zero represent the return value, and one is the first parameter in the method
1350  *
1351  * The result must be released with mono_custom_attrs_free().
1352  *
1353  * Returns: the custom attribute object for the specified parameter, or NULL if there are none.
1354  */
1355 MonoCustomAttrInfo*
1356 mono_custom_attrs_from_param (MonoMethod *method, guint32 param)
1357 {
1358         MonoError error;
1359         MonoCustomAttrInfo *result = mono_custom_attrs_from_param_checked (method, param, &error);
1360         mono_error_cleanup (&error);
1361         return result;
1362 }
1363
1364 /**
1365  * mono_custom_attrs_from_param_checked:
1366  * @method: handle to the method that we want to retrieve custom parameter information from
1367  * @param: parameter number, where zero represent the return value, and one is the first parameter in the method
1368  * @error: set on error
1369  *
1370  * The result must be released with mono_custom_attrs_free().
1371  *
1372  * Returns: the custom attribute object for the specified parameter, or NULL if there are none.  On failure returns NULL and sets @error.
1373  */
1374 MonoCustomAttrInfo*
1375 mono_custom_attrs_from_param_checked (MonoMethod *method, guint32 param, MonoError *error)
1376 {
1377         MonoTableInfo *ca;
1378         guint32 i, idx, method_index;
1379         guint32 param_list, param_last, param_pos, found;
1380         MonoImage *image;
1381         MonoReflectionMethodAux *aux;
1382
1383         mono_error_init (error);
1384
1385         /*
1386          * An instantiated method has the same cattrs as the generic method definition.
1387          *
1388          * LAMESPEC: The .NET SRE throws an exception for instantiations of generic method builders
1389          *           Note that this stanza is not necessary for non-SRE types, but it's a micro-optimization
1390          */
1391         if (method->is_inflated)
1392                 method = ((MonoMethodInflated *) method)->declaring;
1393
1394         if (image_is_dynamic (method->klass->image)) {
1395                 MonoCustomAttrInfo *res, *ainfo;
1396                 int size;
1397
1398                 aux = (MonoReflectionMethodAux *)g_hash_table_lookup (((MonoDynamicImage*)method->klass->image)->method_aux_hash, method);
1399                 if (!aux || !aux->param_cattr)
1400                         return NULL;
1401
1402                 /* Need to copy since it will be freed later */
1403                 ainfo = aux->param_cattr [param];
1404                 if (!ainfo)
1405                         return NULL;
1406                 size = MONO_SIZEOF_CUSTOM_ATTR_INFO + sizeof (MonoCustomAttrEntry) * ainfo->num_attrs;
1407                 res = (MonoCustomAttrInfo *)g_malloc0 (size);
1408                 memcpy (res, ainfo, size);
1409                 return res;
1410         }
1411
1412         image = method->klass->image;
1413         method_index = mono_method_get_index (method);
1414         if (!method_index)
1415                 return NULL;
1416         ca = &image->tables [MONO_TABLE_METHOD];
1417
1418         param_list = mono_metadata_decode_row_col (ca, method_index - 1, MONO_METHOD_PARAMLIST);
1419         if (method_index == ca->rows) {
1420                 ca = &image->tables [MONO_TABLE_PARAM];
1421                 param_last = ca->rows + 1;
1422         } else {
1423                 param_last = mono_metadata_decode_row_col (ca, method_index, MONO_METHOD_PARAMLIST);
1424                 ca = &image->tables [MONO_TABLE_PARAM];
1425         }
1426         found = FALSE;
1427         for (i = param_list; i < param_last; ++i) {
1428                 param_pos = mono_metadata_decode_row_col (ca, i - 1, MONO_PARAM_SEQUENCE);
1429                 if (param_pos == param) {
1430                         found = TRUE;
1431                         break;
1432                 }
1433         }
1434         if (!found)
1435                 return NULL;
1436         idx = i;
1437         idx <<= MONO_CUSTOM_ATTR_BITS;
1438         idx |= MONO_CUSTOM_ATTR_PARAMDEF;
1439         return mono_custom_attrs_from_index_checked (image, idx, FALSE, error);
1440 }
1441
1442 gboolean
1443 mono_custom_attrs_has_attr (MonoCustomAttrInfo *ainfo, MonoClass *attr_klass)
1444 {
1445         int i;
1446         for (i = 0; i < ainfo->num_attrs; ++i) {
1447                 MonoCustomAttrEntry *centry = &ainfo->attrs[i];
1448                 if (centry->ctor == NULL)
1449                         continue;
1450                 MonoClass *klass = centry->ctor->klass;
1451                 if (klass == attr_klass || mono_class_has_parent (klass, attr_klass) || (MONO_CLASS_IS_INTERFACE (attr_klass) && mono_class_is_assignable_from (attr_klass, klass)))
1452                         return TRUE;
1453         }
1454         return FALSE;
1455 }
1456
1457 MonoObject*
1458 mono_custom_attrs_get_attr (MonoCustomAttrInfo *ainfo, MonoClass *attr_klass)
1459 {
1460         MonoError error;
1461         MonoObject *res = mono_custom_attrs_get_attr_checked (ainfo, attr_klass, &error);
1462         mono_error_assert_ok (&error); /*FIXME proper error handling*/
1463         return res;
1464 }
1465
1466 MonoObject*
1467 mono_custom_attrs_get_attr_checked (MonoCustomAttrInfo *ainfo, MonoClass *attr_klass, MonoError *error)
1468 {
1469         int i;
1470         MonoCustomAttrEntry *centry = NULL;
1471
1472         g_assert (attr_klass != NULL);
1473
1474         mono_error_init (error);
1475
1476         for (i = 0; i < ainfo->num_attrs; ++i) {
1477                 centry = &ainfo->attrs[i];
1478                 if (centry->ctor == NULL)
1479                         continue;
1480                 MonoClass *klass = centry->ctor->klass;
1481                 if (attr_klass == klass || mono_class_is_assignable_from (attr_klass, klass))
1482                         break;
1483         }
1484         if (centry == NULL)
1485                 return NULL;
1486
1487         return create_custom_attr (ainfo->image, centry->ctor, centry->data, centry->data_size, error);
1488 }
1489
1490 /*
1491  * mono_reflection_get_custom_attrs_info:
1492  * @obj: a reflection object handle
1493  *
1494  * Return the custom attribute info for attributes defined for the
1495  * reflection handle @obj. The objects.
1496  *
1497  * FIXME this function leaks like a sieve for SRE objects.
1498  */
1499 MonoCustomAttrInfo*
1500 mono_reflection_get_custom_attrs_info (MonoObject *obj)
1501 {
1502         MonoError error;
1503         MonoCustomAttrInfo *result = mono_reflection_get_custom_attrs_info_checked (obj, &error);
1504         mono_error_assert_ok (&error);
1505         return result;
1506 }
1507
1508 /**
1509  * mono_reflection_get_custom_attrs_info_checked:
1510  * @obj: a reflection object handle
1511  * @error: set on error
1512  *
1513  * Return the custom attribute info for attributes defined for the
1514  * reflection handle @obj. The objects.
1515  *
1516  * On failure returns NULL and sets @error.
1517  *
1518  * FIXME this function leaks like a sieve for SRE objects.
1519  */
1520 MonoCustomAttrInfo*
1521 mono_reflection_get_custom_attrs_info_checked (MonoObject *obj, MonoError *error)
1522 {
1523         MonoClass *klass;
1524         MonoCustomAttrInfo *cinfo = NULL;
1525
1526         mono_error_init (error);
1527
1528         klass = obj->vtable->klass;
1529         if (klass == mono_defaults.runtimetype_class) {
1530                 MonoType *type = mono_reflection_type_get_handle ((MonoReflectionType *)obj, error);
1531                 return_val_if_nok (error, NULL);
1532                 klass = mono_class_from_mono_type (type);
1533                 /*We cannot mono_class_init the class from which we'll load the custom attributes since this must work with broken types.*/
1534                 cinfo = mono_custom_attrs_from_class_checked (klass, error);
1535                 return_val_if_nok (error, NULL);
1536         } else if (strcmp ("Assembly", klass->name) == 0 || strcmp ("MonoAssembly", klass->name) == 0) {
1537                 MonoReflectionAssembly *rassembly = (MonoReflectionAssembly*)obj;
1538                 cinfo = mono_custom_attrs_from_assembly_checked (rassembly->assembly, FALSE, error);
1539                 return_val_if_nok (error, NULL);
1540         } else if (strcmp ("Module", klass->name) == 0 || strcmp ("MonoModule", klass->name) == 0) {
1541                 MonoReflectionModule *module = (MonoReflectionModule*)obj;
1542                 cinfo = mono_custom_attrs_from_module (module->image, error);
1543                 return_val_if_nok (error, NULL);
1544         } else if (strcmp ("MonoProperty", klass->name) == 0) {
1545                 MonoReflectionProperty *rprop = (MonoReflectionProperty*)obj;
1546                 cinfo = mono_custom_attrs_from_property_checked (rprop->property->parent, rprop->property, error);
1547                 return_val_if_nok (error, NULL);
1548         } else if (strcmp ("MonoEvent", klass->name) == 0) {
1549                 MonoReflectionMonoEvent *revent = (MonoReflectionMonoEvent*)obj;
1550                 cinfo = mono_custom_attrs_from_event_checked (revent->event->parent, revent->event, error);
1551                 return_val_if_nok (error, NULL);
1552         } else if (strcmp ("MonoField", klass->name) == 0) {
1553                 MonoReflectionField *rfield = (MonoReflectionField*)obj;
1554                 cinfo = mono_custom_attrs_from_field_checked (rfield->field->parent, rfield->field, error);
1555                 return_val_if_nok (error, NULL);
1556         } else if ((strcmp ("MonoMethod", klass->name) == 0) || (strcmp ("MonoCMethod", klass->name) == 0)) {
1557                 MonoReflectionMethod *rmethod = (MonoReflectionMethod*)obj;
1558                 cinfo = mono_custom_attrs_from_method_checked (rmethod->method, error);
1559                 return_val_if_nok (error, NULL);
1560         } else if (strcmp ("ParameterInfo", klass->name) == 0 || strcmp ("MonoParameterInfo", klass->name) == 0) {
1561                 MonoReflectionParameter *param = (MonoReflectionParameter*)obj;
1562                 MonoClass *member_class = mono_object_class (param->MemberImpl);
1563                 if (mono_class_is_reflection_method_or_constructor (member_class)) {
1564                         MonoReflectionMethod *rmethod = (MonoReflectionMethod*)param->MemberImpl;
1565                         cinfo = mono_custom_attrs_from_param_checked (rmethod->method, param->PositionImpl + 1, error);
1566                         return_val_if_nok (error, NULL);
1567                 } else if (mono_is_sr_mono_property (member_class)) {
1568                         MonoReflectionProperty *prop = (MonoReflectionProperty *)param->MemberImpl;
1569                         MonoMethod *method;
1570                         if (!(method = prop->property->get))
1571                                 method = prop->property->set;
1572                         g_assert (method);
1573
1574                         cinfo = mono_custom_attrs_from_param_checked (method, param->PositionImpl + 1, error);
1575                         return_val_if_nok (error, NULL);
1576                 }
1577 #ifndef DISABLE_REFLECTION_EMIT
1578                 else if (mono_is_sre_method_on_tb_inst (member_class)) {/*XXX This is a workaround for Compiler Context*/
1579                         // FIXME: Is this still needed ?
1580                         g_assert_not_reached ();
1581                 } else if (mono_is_sre_ctor_on_tb_inst (member_class)) { /*XX This is a workaround for Compiler Context*/
1582                         // FIXME: Is this still needed ?
1583                         g_assert_not_reached ();
1584                 }
1585 #endif
1586                 else {
1587                         char *type_name = mono_type_get_full_name (member_class);
1588                         mono_error_set_not_supported (error,
1589                                                       "Custom attributes on a ParamInfo with member %s are not supported",
1590                                                       type_name);
1591                         g_free (type_name);
1592                         return NULL;
1593                 }
1594         } else if (strcmp ("AssemblyBuilder", klass->name) == 0) {
1595                 MonoReflectionAssemblyBuilder *assemblyb = (MonoReflectionAssemblyBuilder*)obj;
1596                 cinfo = mono_custom_attrs_from_builders (NULL, assemblyb->assembly.assembly->image, assemblyb->cattrs);
1597         } else if (strcmp ("TypeBuilder", klass->name) == 0) {
1598                 MonoReflectionTypeBuilder *tb = (MonoReflectionTypeBuilder*)obj;
1599                 cinfo = mono_custom_attrs_from_builders (NULL, &tb->module->dynamic_image->image, tb->cattrs);
1600         } else if (strcmp ("ModuleBuilder", klass->name) == 0) {
1601                 MonoReflectionModuleBuilder *mb = (MonoReflectionModuleBuilder*)obj;
1602                 cinfo = mono_custom_attrs_from_builders (NULL, &mb->dynamic_image->image, mb->cattrs);
1603         } else if (strcmp ("ConstructorBuilder", klass->name) == 0) {
1604                 MonoReflectionCtorBuilder *cb = (MonoReflectionCtorBuilder*)obj;
1605                 cinfo = mono_custom_attrs_from_builders (NULL, cb->mhandle->klass->image, cb->cattrs);
1606         } else if (strcmp ("MethodBuilder", klass->name) == 0) {
1607                 MonoReflectionMethodBuilder *mb = (MonoReflectionMethodBuilder*)obj;
1608                 cinfo = mono_custom_attrs_from_builders (NULL, mb->mhandle->klass->image, mb->cattrs);
1609         } else if (strcmp ("FieldBuilder", klass->name) == 0) {
1610                 MonoReflectionFieldBuilder *fb = (MonoReflectionFieldBuilder*)obj;
1611                 cinfo = mono_custom_attrs_from_builders (NULL, &((MonoReflectionTypeBuilder*)fb->typeb)->module->dynamic_image->image, fb->cattrs);
1612         } else if (strcmp ("MonoGenericClass", klass->name) == 0) {
1613                 MonoReflectionGenericClass *gclass = (MonoReflectionGenericClass*)obj;
1614                 cinfo = mono_reflection_get_custom_attrs_info_checked ((MonoObject*)gclass->generic_type, error);
1615                 return_val_if_nok (error, NULL);
1616         } else { /* handle other types here... */
1617                 g_error ("get custom attrs not yet supported for %s", klass->name);
1618         }
1619
1620         return cinfo;
1621 }
1622
1623 /*
1624  * mono_reflection_get_custom_attrs_by_type:
1625  * @obj: a reflection object handle
1626  *
1627  * Return an array with all the custom attributes defined of the
1628  * reflection handle @obj. If @attr_klass is non-NULL, only custom attributes
1629  * of that type are returned. The objects are fully build. Return NULL if a loading error
1630  * occurs.
1631  */
1632 MonoArray*
1633 mono_reflection_get_custom_attrs_by_type (MonoObject *obj, MonoClass *attr_klass, MonoError *error)
1634 {
1635         MonoArray *result;
1636         MonoCustomAttrInfo *cinfo;
1637
1638         mono_error_init (error);
1639
1640         cinfo = mono_reflection_get_custom_attrs_info_checked (obj, error);
1641         return_val_if_nok (error, NULL);
1642         if (cinfo) {
1643                 result = mono_custom_attrs_construct_by_type (cinfo, attr_klass, error);
1644                 if (!cinfo->cached)
1645                         mono_custom_attrs_free (cinfo);
1646                 if (!result)
1647                         return NULL;
1648         } else {
1649                 result = mono_array_new_cached (mono_domain_get (), mono_defaults.attribute_class, 0, error);
1650         }
1651
1652         return result;
1653 }
1654
1655 /*
1656  * mono_reflection_get_custom_attrs:
1657  * @obj: a reflection object handle
1658  *
1659  * Return an array with all the custom attributes defined of the
1660  * reflection handle @obj. The objects are fully build. Return NULL if a loading error
1661  * occurs.
1662  */
1663 MonoArray*
1664 mono_reflection_get_custom_attrs (MonoObject *obj)
1665 {
1666         MonoError error;
1667
1668         return mono_reflection_get_custom_attrs_by_type (obj, NULL, &error);
1669 }
1670
1671 /*
1672  * mono_reflection_get_custom_attrs_data:
1673  * @obj: a reflection obj handle
1674  *
1675  * Returns an array of System.Reflection.CustomAttributeData,
1676  * which include information about attributes reflected on
1677  * types loaded using the Reflection Only methods
1678  */
1679 MonoArray*
1680 mono_reflection_get_custom_attrs_data (MonoObject *obj)
1681 {
1682         MonoError error;
1683         MonoArray* result;
1684         result = mono_reflection_get_custom_attrs_data_checked (obj, &error);
1685         mono_error_cleanup (&error);
1686         return result;
1687 }
1688
1689 /*
1690  * mono_reflection_get_custom_attrs_data_checked:
1691  * @obj: a reflection obj handle
1692  * @error: set on error
1693  *
1694  * Returns an array of System.Reflection.CustomAttributeData,
1695  * which include information about attributes reflected on
1696  * types loaded using the Reflection Only methods
1697  */
1698 MonoArray*
1699 mono_reflection_get_custom_attrs_data_checked (MonoObject *obj, MonoError *error)
1700 {
1701         MonoArray *result;
1702         MonoCustomAttrInfo *cinfo;
1703
1704         mono_error_init (error);
1705
1706         cinfo = mono_reflection_get_custom_attrs_info_checked (obj, error);
1707         return_val_if_nok (error, NULL);
1708         if (cinfo) {
1709                 result = mono_custom_attrs_data_construct (cinfo, error);
1710                 if (!cinfo->cached)
1711                         mono_custom_attrs_free (cinfo);
1712                 return_val_if_nok (error, NULL);
1713         } else
1714                 result = mono_array_new_checked (mono_domain_get (), mono_defaults.customattribute_data_class, 0, error);
1715
1716         return result;
1717 }
1718
1719 static gboolean
1720 custom_attr_class_name_from_methoddef (MonoImage *image, guint32 method_token, const gchar **nspace, const gchar **class_name)
1721 {
1722         /* mono_get_method_from_token () */
1723         g_assert (mono_metadata_token_table (method_token) == MONO_TABLE_METHOD);
1724         guint32 type_token = mono_metadata_typedef_from_method (image, method_token);
1725         if (!type_token) {
1726                 /* Bad method token (could not find corresponding typedef) */
1727                 return FALSE;
1728         }
1729         type_token |= MONO_TOKEN_TYPE_DEF;
1730         {
1731                 /* mono_class_create_from_typedef () */
1732                 MonoTableInfo *tt = &image->tables [MONO_TABLE_TYPEDEF];
1733                 guint32 cols [MONO_TYPEDEF_SIZE];
1734                 guint tidx = mono_metadata_token_index (type_token);
1735
1736                 if (mono_metadata_token_table (type_token) != MONO_TABLE_TYPEDEF || tidx > tt->rows) {
1737                         /* "Invalid typedef token %x", type_token */
1738                         return FALSE;
1739                 }
1740
1741                 mono_metadata_decode_row (tt, tidx - 1, cols, MONO_TYPEDEF_SIZE);
1742
1743                 if (class_name)
1744                         *class_name = mono_metadata_string_heap (image, cols [MONO_TYPEDEF_NAME]);
1745                 if (nspace)
1746                         *nspace = mono_metadata_string_heap (image, cols [MONO_TYPEDEF_NAMESPACE]);
1747                 return TRUE;
1748         }
1749 }
1750
1751
1752 /**
1753  * custom_attr_class_name_from_method_token:
1754  * @image: The MonoImage
1755  * @method_token: a token for a custom attr constructor in @image
1756  * @assembly_token: out argment set to the assembly ref token of the custom attr
1757  * @nspace: out argument set to namespace (a string in the string heap of @image) of the custom attr
1758  * @class_name: out argument set to the class name of the custom attr.
1759  *
1760  * Given an @image and a @method_token (which is assumed to be a
1761  * constructor), fills in the out arguments with the assembly ref (if
1762  * a methodref) and the namespace and class name of the custom
1763  * attribute.
1764  *
1765  * Returns: TRUE on success, FALSE otherwise.
1766  *
1767  * LOCKING: does not take locks
1768  */
1769 static gboolean
1770 custom_attr_class_name_from_method_token (MonoImage *image, guint32 method_token, guint32 *assembly_token, const gchar **nspace, const gchar **class_name)
1771 {
1772         /* This only works with method tokens constructed from a
1773          * custom attr token, which can only be methoddef or
1774          * memberref */
1775         g_assert (mono_metadata_token_table (method_token) == MONO_TABLE_METHOD
1776                   || mono_metadata_token_table  (method_token) == MONO_TABLE_MEMBERREF);
1777
1778         if (mono_metadata_token_table (method_token) == MONO_TABLE_MEMBERREF) {
1779                 /* method_from_memberref () */
1780                 guint32 cols[6];
1781                 guint32 nindex, class_index;
1782
1783                 int idx = mono_metadata_token_index (method_token);
1784
1785                 mono_metadata_decode_row (&image->tables [MONO_TABLE_MEMBERREF], idx-1, cols, 3);
1786                 nindex = cols [MONO_MEMBERREF_CLASS] >> MONO_MEMBERREF_PARENT_BITS;
1787                 class_index = cols [MONO_MEMBERREF_CLASS] & MONO_MEMBERREF_PARENT_MASK;
1788                 if (class_index == MONO_MEMBERREF_PARENT_TYPEREF) {
1789                         guint32 type_token = MONO_TOKEN_TYPE_REF | nindex;
1790                         /* mono_class_from_typeref_checked () */
1791                         {
1792                                 guint32 cols [MONO_TYPEREF_SIZE];
1793                                 MonoTableInfo  *t = &image->tables [MONO_TABLE_TYPEREF];
1794
1795                                 mono_metadata_decode_row (t, (type_token&0xffffff)-1, cols, MONO_TYPEREF_SIZE);
1796
1797                                 if (class_name)
1798                                         *class_name = mono_metadata_string_heap (image, cols [MONO_TYPEREF_NAME]);
1799                                 if (nspace)
1800                                         *nspace = mono_metadata_string_heap (image, cols [MONO_TYPEREF_NAMESPACE]);
1801                                 if (assembly_token)
1802                                         *assembly_token = cols [MONO_TYPEREF_SCOPE];
1803                                 return TRUE;
1804                         }
1805                 } else if (class_index == MONO_MEMBERREF_PARENT_METHODDEF) {
1806                         guint32 methoddef_token = MONO_TOKEN_METHOD_DEF | nindex;
1807                         if (assembly_token)
1808                                 *assembly_token = 0;
1809                         return custom_attr_class_name_from_methoddef (image, methoddef_token, nspace, class_name);
1810                 } else {
1811                         /* Attributes can't be generic, so it won't be
1812                          * a typespec, and they're always
1813                          * constructors, so it won't be a moduleref */
1814                         g_assert_not_reached ();
1815                 }
1816         } else {
1817                 /* must be MONO_TABLE_METHOD */
1818                 if (assembly_token)
1819                         *assembly_token = 0;
1820                 return custom_attr_class_name_from_methoddef (image, method_token, nspace, class_name);
1821         }
1822 }
1823
1824 /**
1825  * mono_assembly_metadata_foreach_custom_attr:
1826  * @assembly: the assembly to iterate over
1827  * @func: the function to call for each custom attribute
1828  * @user_data: passed to @func
1829  *
1830  * Calls @func for each custom attribute type on the given assembly until @func returns TRUE.
1831  * Everything is done using low-level metadata APIs, so it is safe to use during assembly loading.
1832  *
1833  */
1834 void
1835 mono_assembly_metadata_foreach_custom_attr (MonoAssembly *assembly, MonoAssemblyMetadataCustomAttrIterFunc func, gpointer user_data)
1836 {
1837         MonoImage *image;
1838         guint32 mtoken, i;
1839         guint32 cols [MONO_CUSTOM_ATTR_SIZE];
1840         MonoTableInfo *ca;
1841         guint32 idx;
1842
1843         /*
1844          * This might be called during assembly loading, so do everything using the low-level
1845          * metadata APIs.
1846          */
1847
1848         image = assembly->image;
1849         g_assert (!image_is_dynamic (image));
1850         idx = 1; /* there is only one assembly */
1851         idx <<= MONO_CUSTOM_ATTR_BITS;
1852         idx |= MONO_CUSTOM_ATTR_ASSEMBLY;
1853
1854         /* Inlined from mono_custom_attrs_from_index_checked () */
1855         ca = &image->tables [MONO_TABLE_CUSTOMATTRIBUTE];
1856         i = mono_metadata_custom_attrs_from_index (image, idx);
1857         if (!i)
1858                 return;
1859         i --;
1860         gboolean stop_iterating = FALSE;
1861         while (!stop_iterating && i < ca->rows) {
1862                 if (mono_metadata_decode_row_col (ca, i, MONO_CUSTOM_ATTR_PARENT) != idx)
1863                         break;
1864                 mono_metadata_decode_row (ca, i, cols, MONO_CUSTOM_ATTR_SIZE);
1865                 i ++;
1866                 mtoken = cols [MONO_CUSTOM_ATTR_TYPE] >> MONO_CUSTOM_ATTR_TYPE_BITS;
1867                 switch (cols [MONO_CUSTOM_ATTR_TYPE] & MONO_CUSTOM_ATTR_TYPE_MASK) {
1868                 case MONO_CUSTOM_ATTR_TYPE_METHODDEF:
1869                         mtoken |= MONO_TOKEN_METHOD_DEF;
1870                         break;
1871                 case MONO_CUSTOM_ATTR_TYPE_MEMBERREF:
1872                         mtoken |= MONO_TOKEN_MEMBER_REF;
1873                         break;
1874                 default:
1875                         g_warning ("Unknown table for custom attr type %08x", cols [MONO_CUSTOM_ATTR_TYPE]);
1876                         continue;
1877                 }
1878
1879                 const char *nspace = NULL;
1880                 const char *name = NULL;
1881                 guint32 assembly_token = 0;
1882
1883                 if (!custom_attr_class_name_from_method_token (image, mtoken, &assembly_token, &nspace, &name))
1884                         continue;
1885
1886                 stop_iterating = func (image, assembly_token, nspace, name, mtoken, user_data);
1887         }
1888 }