1 /*
   2  * Copyright (c) 2005, 2019, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  *
  23  */
  24 
  25 #include "precompiled.hpp"
  26 #include "classfile/symbolTable.hpp"
  27 #include "interpreter/bytecodeStream.hpp"
  28 #include "memory/universe.hpp"
  29 #include "oops/fieldStreams.inline.hpp"
  30 #include "prims/jvmtiClassFileReconstituter.hpp"
  31 #include "runtime/handles.inline.hpp"
  32 #include "runtime/signature.hpp"
  33 #include "utilities/bytes.hpp"
  34 
  35 // FIXME: add Deprecated attribute
  36 // FIXME: fix Synthetic attribute
  37 // FIXME: per Serguei, add error return handling for ConstantPool::copy_cpool_bytes()
  38 
  39 JvmtiConstantPoolReconstituter::JvmtiConstantPoolReconstituter(InstanceKlass* ik) {
  40   set_error(JVMTI_ERROR_NONE);
  41   _ik = ik;
  42   _cpool = constantPoolHandle(Thread::current(), ik->constants());
  43   _symmap = new SymbolHashMap();
  44   _classmap = new SymbolHashMap();
  45   _cpool_size = _cpool->hash_entries_to(_symmap, _classmap);
  46   if (_cpool_size == 0) {
  47     set_error(JVMTI_ERROR_OUT_OF_MEMORY);
  48   } else if (_cpool_size < 0) {
  49     set_error(JVMTI_ERROR_INTERNAL);
  50   }
  51 }
  52 
  53 // Write the field information portion of ClassFile structure
  54 // JVMSpec|     u2 fields_count;
  55 // JVMSpec|     field_info fields[fields_count];
  56 void JvmtiClassFileReconstituter::write_field_infos() {
  57   HandleMark hm(thread());
  58   Array<AnnotationArray*>* fields_anno = ik()->fields_annotations();
  59   Array<AnnotationArray*>* fields_type_anno = ik()->fields_type_annotations();
  60 
  61   // Compute the real number of Java fields
  62   int java_fields = ik()->java_fields_count();
  63 
  64   write_u2(java_fields);
  65   for (JavaFieldStream fs(ik()); !fs.done(); fs.next()) {
  66     AccessFlags access_flags = fs.access_flags();
  67     int name_index = fs.name_index();
  68     int signature_index = fs.signature_index();
  69     int initial_value_index = fs.initval_index();
  70     guarantee(name_index != 0 && signature_index != 0, "bad constant pool index for field");
  71     // int offset = ik()->field_offset( index );
  72     int generic_signature_index = fs.generic_signature_index();
  73     AnnotationArray* anno = fields_anno == NULL ? NULL : fields_anno->at(fs.index());
  74     AnnotationArray* type_anno = fields_type_anno == NULL ? NULL : fields_type_anno->at(fs.index());
  75 
  76     // JVMSpec|   field_info {
  77     // JVMSpec|         u2 access_flags;
  78     // JVMSpec|         u2 name_index;
  79     // JVMSpec|         u2 descriptor_index;
  80     // JVMSpec|         u2 attributes_count;
  81     // JVMSpec|         attribute_info attributes[attributes_count];
  82     // JVMSpec|   }
  83 
  84     write_u2(access_flags.as_int() & JVM_RECOGNIZED_FIELD_MODIFIERS);
  85     write_u2(name_index);
  86     write_u2(signature_index);
  87     int attr_count = 0;
  88     if (initial_value_index != 0) {
  89       ++attr_count;
  90     }
  91     if (access_flags.is_synthetic()) {
  92       // ++attr_count;
  93     }
  94     if (generic_signature_index != 0) {
  95       ++attr_count;
  96     }
  97     if (anno != NULL) {
  98       ++attr_count;     // has RuntimeVisibleAnnotations attribute
  99     }
 100     if (type_anno != NULL) {
 101       ++attr_count;     // has RuntimeVisibleTypeAnnotations attribute
 102     }
 103 
 104     write_u2(attr_count);
 105 
 106     if (initial_value_index != 0) {
 107       write_attribute_name_index("ConstantValue");
 108       write_u4(2); //length always 2
 109       write_u2(initial_value_index);
 110     }
 111     if (access_flags.is_synthetic()) {
 112       // write_synthetic_attribute();
 113     }
 114     if (generic_signature_index != 0) {
 115       write_signature_attribute(generic_signature_index);
 116     }
 117     if (anno != NULL) {
 118       write_annotations_attribute("RuntimeVisibleAnnotations", anno);
 119     }
 120     if (type_anno != NULL) {
 121       write_annotations_attribute("RuntimeVisibleTypeAnnotations", type_anno);
 122     }
 123   }
 124 }
 125 
 126 // Write Code attribute
 127 // JVMSpec|   Code_attribute {
 128 // JVMSpec|     u2 attribute_name_index;
 129 // JVMSpec|     u4 attribute_length;
 130 // JVMSpec|     u2 max_stack;
 131 // JVMSpec|     u2 max_locals;
 132 // JVMSpec|     u4 code_length;
 133 // JVMSpec|     u1 code[code_length];
 134 // JVMSpec|     u2 exception_table_length;
 135 // JVMSpec|     {       u2 start_pc;
 136 // JVMSpec|             u2 end_pc;
 137 // JVMSpec|             u2  handler_pc;
 138 // JVMSpec|             u2  catch_type;
 139 // JVMSpec|     }       exception_table[exception_table_length];
 140 // JVMSpec|     u2 attributes_count;
 141 // JVMSpec|     attribute_info attributes[attributes_count];
 142 // JVMSpec|   }
 143 void JvmtiClassFileReconstituter::write_code_attribute(const methodHandle& method) {
 144   ConstMethod* const_method = method->constMethod();
 145   u2 line_num_cnt = 0;
 146   int stackmap_len = 0;
 147   int local_variable_table_length = 0;
 148   int local_variable_type_table_length = 0;
 149 
 150   // compute number and length of attributes
 151   int attr_count = 0;
 152   int attr_size = 0;
 153   if (const_method->has_linenumber_table()) {
 154     line_num_cnt = line_number_table_entries(method);
 155     if (line_num_cnt != 0) {
 156       ++attr_count;
 157       // Compute the complete size of the line number table attribute:
 158       //      LineNumberTable_attribute {
 159       //        u2 attribute_name_index;
 160       //        u4 attribute_length;
 161       //        u2 line_number_table_length;
 162       //        {  u2 start_pc;
 163       //           u2 line_number;
 164       //        } line_number_table[line_number_table_length];
 165       //      }
 166       attr_size += 2 + 4 + 2 + line_num_cnt * (2 + 2);
 167     }
 168   }
 169   if (method->has_stackmap_table()) {
 170     stackmap_len = method->stackmap_data()->length();
 171     if (stackmap_len != 0) {
 172       ++attr_count;
 173       // Compute the  size of the stack map table attribute (VM stores raw):
 174       //      StackMapTable_attribute {
 175       //        u2 attribute_name_index;
 176       //        u4 attribute_length;
 177       //        u2 number_of_entries;
 178       //        stack_map_frame_entries[number_of_entries];
 179       //      }
 180       attr_size += 2 + 4 + stackmap_len;
 181     }
 182   }
 183   if (method->has_localvariable_table()) {
 184     local_variable_table_length = method->localvariable_table_length();
 185     if (local_variable_table_length != 0) {
 186       ++attr_count;
 187       // Compute the size of the local variable table attribute (VM stores raw):
 188       // LocalVariableTable_attribute {
 189       //   u2 attribute_name_index;
 190       //   u4 attribute_length;
 191       //   u2 local_variable_table_length;
 192       //   {
 193       //     u2 start_pc;
 194       //     u2 length;
 195       //     u2 name_index;
 196       //     u2 descriptor_index;
 197       //     u2 index;
 198       //   }
 199       attr_size += 2 + 4 + 2 + local_variable_table_length * (2 + 2 + 2 + 2 + 2);
 200 
 201       // Local variables with generic signatures must have LVTT entries
 202       LocalVariableTableElement *elem = method->localvariable_table_start();
 203       for (int idx = 0; idx < local_variable_table_length; idx++) {
 204         if (elem[idx].signature_cp_index != 0) {
 205           local_variable_type_table_length++;
 206         }
 207       }
 208 
 209       if (local_variable_type_table_length != 0) {
 210         ++attr_count;
 211         // Compute the size of the local variable type table attribute (VM stores raw):
 212         // LocalVariableTypeTable_attribute {
 213         //   u2 attribute_name_index;
 214         //   u4 attribute_length;
 215         //   u2 local_variable_type_table_length;
 216         //   {
 217         //     u2 start_pc;
 218         //     u2 length;
 219         //     u2 name_index;
 220         //     u2 signature_index;
 221         //     u2 index;
 222         //   }
 223         attr_size += 2 + 4 + 2 + local_variable_type_table_length * (2 + 2 + 2 + 2 + 2);
 224       }
 225     }
 226   }
 227 
 228   ExceptionTable exception_table(method());
 229   int exception_table_length = exception_table.length();
 230   int code_size = const_method->code_size();
 231   int size =
 232     2+2+4 +                                // max_stack, max_locals, code_length
 233     code_size +                            // code
 234     2 +                                    // exception_table_length
 235     (2+2+2+2) * exception_table_length +   // exception_table
 236     2 +                                    // attributes_count
 237     attr_size;                             // attributes
 238 
 239   write_attribute_name_index("Code");
 240   write_u4(size);
 241   write_u2(method->verifier_max_stack());
 242   write_u2(method->max_locals());
 243   write_u4(code_size);
 244   copy_bytecodes(method, (unsigned char*)writeable_address(code_size));
 245   write_u2(exception_table_length);
 246   for (int index = 0; index < exception_table_length; index++) {
 247     write_u2(exception_table.start_pc(index));
 248     write_u2(exception_table.end_pc(index));
 249     write_u2(exception_table.handler_pc(index));
 250     write_u2(exception_table.catch_type_index(index));
 251   }
 252   write_u2(attr_count);
 253   if (line_num_cnt != 0) {
 254     write_line_number_table_attribute(method, line_num_cnt);
 255   }
 256   if (stackmap_len != 0) {
 257     write_stackmap_table_attribute(method, stackmap_len);
 258   }
 259   if (local_variable_table_length != 0) {
 260     write_local_variable_table_attribute(method, local_variable_table_length);
 261   }
 262   if (local_variable_type_table_length != 0) {
 263     write_local_variable_type_table_attribute(method, local_variable_type_table_length);
 264   }
 265 }
 266 
 267 // Write Exceptions attribute
 268 // JVMSpec|   Exceptions_attribute {
 269 // JVMSpec|     u2 attribute_name_index;
 270 // JVMSpec|     u4 attribute_length;
 271 // JVMSpec|     u2 number_of_exceptions;
 272 // JVMSpec|     u2 exception_index_table[number_of_exceptions];
 273 // JVMSpec|   }
 274 void JvmtiClassFileReconstituter::write_exceptions_attribute(ConstMethod* const_method) {
 275   CheckedExceptionElement* checked_exceptions = const_method->checked_exceptions_start();
 276   int checked_exceptions_length = const_method->checked_exceptions_length();
 277   int size =
 278     2 +                                    // number_of_exceptions
 279     2 * checked_exceptions_length;         // exception_index_table
 280 
 281   write_attribute_name_index("Exceptions");
 282   write_u4(size);
 283   write_u2(checked_exceptions_length);
 284   for (int index = 0; index < checked_exceptions_length; index++) {
 285     write_u2(checked_exceptions[index].class_cp_index);
 286   }
 287 }
 288 
 289 // Write SourceFile attribute
 290 // JVMSpec|   SourceFile_attribute {
 291 // JVMSpec|     u2 attribute_name_index;
 292 // JVMSpec|     u4 attribute_length;
 293 // JVMSpec|     u2 sourcefile_index;
 294 // JVMSpec|   }
 295 void JvmtiClassFileReconstituter::write_source_file_attribute() {
 296   assert(ik()->source_file_name() != NULL, "caller must check");
 297 
 298   write_attribute_name_index("SourceFile");
 299   write_u4(2);  // always length 2
 300   write_u2(symbol_to_cpool_index(ik()->source_file_name()));
 301 }
 302 
 303 // Write SourceDebugExtension attribute
 304 // JSR45|   SourceDebugExtension_attribute {
 305 // JSR45|       u2 attribute_name_index;
 306 // JSR45|       u4 attribute_length;
 307 // JSR45|       u1 debug_extension[attribute_length];
 308 // JSR45|   }
 309 void JvmtiClassFileReconstituter::write_source_debug_extension_attribute() {
 310   assert(ik()->source_debug_extension() != NULL, "caller must check");
 311 
 312   write_attribute_name_index("SourceDebugExtension");
 313   int len = (int)strlen(ik()->source_debug_extension());
 314   write_u4(len);
 315   u1* ext = (u1*)ik()->source_debug_extension();
 316   for (int i=0; i<len; i++) {
 317     write_u1(ext[i]);
 318   }
 319 }
 320 
 321 // Write (generic) Signature attribute
 322 // JVMSpec|   Signature_attribute {
 323 // JVMSpec|     u2 attribute_name_index;
 324 // JVMSpec|     u4 attribute_length;
 325 // JVMSpec|     u2 signature_index;
 326 // JVMSpec|   }
 327 void JvmtiClassFileReconstituter::write_signature_attribute(u2 generic_signature_index) {
 328   write_attribute_name_index("Signature");
 329   write_u4(2);  // always length 2
 330   write_u2(generic_signature_index);
 331 }
 332 
 333 // Compute the number of entries in the InnerClasses attribute
 334 u2 JvmtiClassFileReconstituter::inner_classes_attribute_length() {
 335   InnerClassesIterator iter(ik());
 336   return iter.length();
 337 }
 338 
 339 // Write an annotation attribute.  The VM stores them in raw form, so all we need
 340 // to do is add the attrubute name and fill in the length.
 341 // JSR202|   *Annotations_attribute {
 342 // JSR202|     u2 attribute_name_index;
 343 // JSR202|     u4 attribute_length;
 344 // JSR202|     ...
 345 // JSR202|   }
 346 void JvmtiClassFileReconstituter::write_annotations_attribute(const char* attr_name,
 347                                                               AnnotationArray* annos) {
 348   u4 length = annos->length();
 349   write_attribute_name_index(attr_name);
 350   write_u4(length);
 351   memcpy(writeable_address(length), annos->adr_at(0), length);
 352 }
 353 
 354 //  BootstrapMethods_attribute {
 355 //    u2 attribute_name_index;
 356 //    u4 attribute_length;
 357 //    u2 num_bootstrap_methods;
 358 //    {   u2 bootstrap_method_ref;
 359 //        u2 num_bootstrap_arguments;
 360 //        u2 bootstrap_arguments[num_bootstrap_arguments];
 361 //    } bootstrap_methods[num_bootstrap_methods];
 362 //  }
 363 void JvmtiClassFileReconstituter::write_bootstrapmethod_attribute() {
 364   Array<u2>* operands = cpool()->operands();
 365   write_attribute_name_index("BootstrapMethods");
 366   int num_bootstrap_methods = ConstantPool::operand_array_length(operands);
 367 
 368   // calculate length of attribute
 369   int length = sizeof(u2); // num_bootstrap_methods
 370   for (int n = 0; n < num_bootstrap_methods; n++) {
 371     u2 num_bootstrap_arguments = cpool()->operand_argument_count_at(n);
 372     length += sizeof(u2); // bootstrap_method_ref
 373     length += sizeof(u2); // num_bootstrap_arguments
 374     length += sizeof(u2) * num_bootstrap_arguments; // bootstrap_arguments[num_bootstrap_arguments]
 375   }
 376   write_u4(length);
 377 
 378   // write attribute
 379   write_u2(num_bootstrap_methods);
 380   for (int n = 0; n < num_bootstrap_methods; n++) {
 381     u2 bootstrap_method_ref = cpool()->operand_bootstrap_method_ref_index_at(n);
 382     u2 num_bootstrap_arguments = cpool()->operand_argument_count_at(n);
 383     write_u2(bootstrap_method_ref);
 384     write_u2(num_bootstrap_arguments);
 385     for (int arg = 0; arg < num_bootstrap_arguments; arg++) {
 386       u2 bootstrap_argument = cpool()->operand_argument_index_at(n, arg);
 387       write_u2(bootstrap_argument);
 388     }
 389   }
 390 }
 391 
 392 //  NestHost_attribute {
 393 //    u2 attribute_name_index;
 394 //    u4 attribute_length;
 395 //    u2 host_class_index;
 396 //  }
 397 void JvmtiClassFileReconstituter::write_nest_host_attribute() {
 398   int length = sizeof(u2);
 399   int host_class_index = ik()->nest_host_index();
 400 
 401   write_attribute_name_index("NestHost");
 402   write_u4(length);
 403   write_u2(host_class_index);
 404 }
 405 
 406 //  NestMembers_attribute {
 407 //    u2 attribute_name_index;
 408 //    u4 attribute_length;
 409 //    u2 number_of_classes;
 410 //    u2 classes[number_of_classes];
 411 //  }
 412 void JvmtiClassFileReconstituter::write_nest_members_attribute() {
 413   Array<u2>* nest_members = ik()->nest_members();
 414   int number_of_classes = nest_members->length();
 415   int length = sizeof(u2) * (1 + number_of_classes);
 416 
 417   write_attribute_name_index("NestMembers");
 418   write_u4(length);
 419   write_u2(number_of_classes);
 420   for (int i = 0; i < number_of_classes; i++) {
 421     u2 class_cp_index = nest_members->at(i);
 422     write_u2(class_cp_index);
 423   }
 424 }
 425 
 426 
 427 // Write InnerClasses attribute
 428 // JVMSpec|   InnerClasses_attribute {
 429 // JVMSpec|     u2 attribute_name_index;
 430 // JVMSpec|     u4 attribute_length;
 431 // JVMSpec|     u2 number_of_classes;
 432 // JVMSpec|     {  u2 inner_class_info_index;
 433 // JVMSpec|        u2 outer_class_info_index;
 434 // JVMSpec|        u2 inner_name_index;
 435 // JVMSpec|        u2 inner_class_access_flags;
 436 // JVMSpec|     } classes[number_of_classes];
 437 // JVMSpec|   }
 438 void JvmtiClassFileReconstituter::write_inner_classes_attribute(int length) {
 439   InnerClassesIterator iter(ik());
 440   guarantee(iter.length() != 0 && iter.length() == length,
 441             "caller must check");
 442   u2 entry_count = length / InstanceKlass::inner_class_next_offset;
 443   u4 size = 2 + entry_count * (2+2+2+2);
 444 
 445   write_attribute_name_index("InnerClasses");
 446   write_u4(size);
 447   write_u2(entry_count);
 448   for (; !iter.done(); iter.next()) {
 449     write_u2(iter.inner_class_info_index());
 450     write_u2(iter.outer_class_info_index());
 451     write_u2(iter.inner_name_index());
 452     write_u2(iter.inner_access_flags());
 453   }
 454 }
 455 
 456 // Write Synthetic attribute
 457 // JVMSpec|   Synthetic_attribute {
 458 // JVMSpec|     u2 attribute_name_index;
 459 // JVMSpec|     u4 attribute_length;
 460 // JVMSpec|   }
 461 void JvmtiClassFileReconstituter::write_synthetic_attribute() {
 462   write_attribute_name_index("Synthetic");
 463   write_u4(0); //length always zero
 464 }
 465 
 466 // Compute size of LineNumberTable
 467 u2 JvmtiClassFileReconstituter::line_number_table_entries(const methodHandle& method) {
 468   // The line number table is compressed so we don't know how big it is until decompressed.
 469   // Decompression is really fast so we just do it twice.
 470   u2 num_entries = 0;
 471   CompressedLineNumberReadStream stream(method->compressed_linenumber_table());
 472   while (stream.read_pair()) {
 473     num_entries++;
 474   }
 475   return num_entries;
 476 }
 477 
 478 // Write LineNumberTable attribute
 479 // JVMSpec|   LineNumberTable_attribute {
 480 // JVMSpec|     u2 attribute_name_index;
 481 // JVMSpec|     u4 attribute_length;
 482 // JVMSpec|     u2 line_number_table_length;
 483 // JVMSpec|     {  u2 start_pc;
 484 // JVMSpec|        u2 line_number;
 485 // JVMSpec|     } line_number_table[line_number_table_length];
 486 // JVMSpec|   }
 487 void JvmtiClassFileReconstituter::write_line_number_table_attribute(const methodHandle& method,
 488                                                                     u2 num_entries) {
 489 
 490   write_attribute_name_index("LineNumberTable");
 491   write_u4(2 + num_entries * (2 + 2));
 492   write_u2(num_entries);
 493 
 494   CompressedLineNumberReadStream stream(method->compressed_linenumber_table());
 495   while (stream.read_pair()) {
 496     write_u2(stream.bci());
 497     write_u2(stream.line());
 498   }
 499 }
 500 
 501 // Write LocalVariableTable attribute
 502 // JVMSpec|   LocalVariableTable_attribute {
 503 // JVMSpec|     u2 attribute_name_index;
 504 // JVMSpec|     u4 attribute_length;
 505 // JVMSpec|     u2 local_variable_table_length;
 506 // JVMSpec|     {  u2 start_pc;
 507 // JVMSpec|       u2 length;
 508 // JVMSpec|       u2 name_index;
 509 // JVMSpec|       u2 descriptor_index;
 510 // JVMSpec|       u2 index;
 511 // JVMSpec|     } local_variable_table[local_variable_table_length];
 512 // JVMSpec|   }
 513 void JvmtiClassFileReconstituter::write_local_variable_table_attribute(const methodHandle& method, u2 num_entries) {
 514     write_attribute_name_index("LocalVariableTable");
 515     write_u4(2 + num_entries * (2 + 2 + 2 + 2 + 2));
 516     write_u2(num_entries);
 517 
 518     assert(method->localvariable_table_length() == num_entries, "just checking");
 519 
 520     LocalVariableTableElement *elem = method->localvariable_table_start();
 521     for (int j=0; j<method->localvariable_table_length(); j++) {
 522       write_u2(elem->start_bci);
 523       write_u2(elem->length);
 524       write_u2(elem->name_cp_index);
 525       write_u2(elem->descriptor_cp_index);
 526       write_u2(elem->slot);
 527       elem++;
 528     }
 529 }
 530 
 531 // Write LocalVariableTypeTable attribute
 532 // JVMSpec|   LocalVariableTypeTable_attribute {
 533 // JVMSpec|     u2 attribute_name_index;
 534 // JVMSpec|     u4 attribute_length;
 535 // JVMSpec|     u2 local_variable_type_table_length;
 536 // JVMSpec|     { u2 start_pc;
 537 // JVMSpec|       u2 length;
 538 // JVMSpec|       u2 name_index;
 539 // JVMSpec|       u2 signature_index;
 540 // JVMSpec|       u2 index;
 541 // JVMSpec|     } local_variable_type_table[local_variable_type_table_length];
 542 // JVMSpec|   }
 543 void JvmtiClassFileReconstituter::write_local_variable_type_table_attribute(const methodHandle& method, u2 num_entries) {
 544     write_attribute_name_index("LocalVariableTypeTable");
 545     write_u4(2 + num_entries * (2 + 2 + 2 + 2 + 2));
 546     write_u2(num_entries);
 547 
 548     LocalVariableTableElement *elem = method->localvariable_table_start();
 549     for (int j=0; j<method->localvariable_table_length(); j++) {
 550       if (elem->signature_cp_index > 0) {
 551         // Local variable has a generic signature - write LVTT attribute entry
 552         write_u2(elem->start_bci);
 553         write_u2(elem->length);
 554         write_u2(elem->name_cp_index);
 555         write_u2(elem->signature_cp_index);
 556         write_u2(elem->slot);
 557         num_entries--;
 558       }
 559       elem++;
 560     }
 561     assert(num_entries == 0, "just checking");
 562 }
 563 
 564 // Write stack map table attribute
 565 // JSR-202|   StackMapTable_attribute {
 566 // JSR-202|     u2 attribute_name_index;
 567 // JSR-202|     u4 attribute_length;
 568 // JSR-202|     u2 number_of_entries;
 569 // JSR-202|     stack_map_frame_entries[number_of_entries];
 570 // JSR-202|   }
 571 void JvmtiClassFileReconstituter::write_stackmap_table_attribute(const methodHandle& method,
 572                                                                  int stackmap_len) {
 573 
 574   write_attribute_name_index("StackMapTable");
 575   write_u4(stackmap_len);
 576   memcpy(
 577     writeable_address(stackmap_len),
 578     (void*)(method->stackmap_data()->adr_at(0)),
 579     stackmap_len);
 580 }
 581 
 582 // Write one method_info structure
 583 // JVMSpec|   method_info {
 584 // JVMSpec|     u2 access_flags;
 585 // JVMSpec|     u2 name_index;
 586 // JVMSpec|     u2 descriptor_index;
 587 // JVMSpec|     u2 attributes_count;
 588 // JVMSpec|     attribute_info attributes[attributes_count];
 589 // JVMSpec|   }
 590 void JvmtiClassFileReconstituter::write_method_info(const methodHandle& method) {
 591   AccessFlags access_flags = method->access_flags();
 592   ConstMethod* const_method = method->constMethod();
 593   u2 generic_signature_index = const_method->generic_signature_index();
 594   AnnotationArray* anno = method->annotations();
 595   AnnotationArray* param_anno = method->parameter_annotations();
 596   AnnotationArray* default_anno = method->annotation_default();
 597   AnnotationArray* type_anno = method->type_annotations();
 598 
 599   // skip generated default interface methods
 600   if (method->is_overpass()) {
 601     return;
 602   }
 603 
 604   write_u2(access_flags.get_flags() & JVM_RECOGNIZED_METHOD_MODIFIERS);
 605   write_u2(const_method->name_index());
 606   write_u2(const_method->signature_index());
 607 
 608   // write attributes in the same order javac does, so we can test with byte for
 609   // byte comparison
 610   int attr_count = 0;
 611   if (const_method->code_size() != 0) {
 612     ++attr_count;     // has Code attribute
 613   }
 614   if (const_method->has_checked_exceptions()) {
 615     ++attr_count;     // has Exceptions attribute
 616   }
 617   if (default_anno != NULL) {
 618     ++attr_count;     // has AnnotationDefault attribute
 619   }
 620   // Deprecated attribute would go here
 621   if (access_flags.is_synthetic()) { // FIXME
 622     // ++attr_count;
 623   }
 624   if (generic_signature_index != 0) {
 625     ++attr_count;
 626   }
 627   if (anno != NULL) {
 628     ++attr_count;     // has RuntimeVisibleAnnotations attribute
 629   }
 630   if (param_anno != NULL) {
 631     ++attr_count;     // has RuntimeVisibleParameterAnnotations attribute
 632   }
 633   if (type_anno != NULL) {
 634     ++attr_count;     // has RuntimeVisibleTypeAnnotations attribute
 635   }
 636 
 637   write_u2(attr_count);
 638   if (const_method->code_size() > 0) {
 639     write_code_attribute(method);
 640   }
 641   if (const_method->has_checked_exceptions()) {
 642     write_exceptions_attribute(const_method);
 643   }
 644   if (default_anno != NULL) {
 645     write_annotations_attribute("AnnotationDefault", default_anno);
 646   }
 647   // Deprecated attribute would go here
 648   if (access_flags.is_synthetic()) {
 649     // write_synthetic_attribute();
 650   }
 651   if (generic_signature_index != 0) {
 652     write_signature_attribute(generic_signature_index);
 653   }
 654   if (anno != NULL) {
 655     write_annotations_attribute("RuntimeVisibleAnnotations", anno);
 656   }
 657   if (param_anno != NULL) {
 658     write_annotations_attribute("RuntimeVisibleParameterAnnotations", param_anno);
 659   }
 660   if (type_anno != NULL) {
 661     write_annotations_attribute("RuntimeVisibleTypeAnnotations", type_anno);
 662   }
 663 }
 664 
 665 // Write the class attributes portion of ClassFile structure
 666 // JVMSpec|     u2 attributes_count;
 667 // JVMSpec|     attribute_info attributes[attributes_count];
 668 void JvmtiClassFileReconstituter::write_class_attributes() {
 669   u2 inner_classes_length = inner_classes_attribute_length();
 670   Symbol* generic_signature = ik()->generic_signature();
 671   AnnotationArray* anno = ik()->class_annotations();
 672   AnnotationArray* type_anno = ik()->class_type_annotations();
 673 
 674   int attr_count = 0;
 675   if (generic_signature != NULL) {
 676     ++attr_count;
 677   }
 678   if (ik()->source_file_name() != NULL) {
 679     ++attr_count;
 680   }
 681   if (ik()->source_debug_extension() != NULL) {
 682     ++attr_count;
 683   }
 684   if (inner_classes_length > 0) {
 685     ++attr_count;
 686   }
 687   if (anno != NULL) {
 688     ++attr_count;     // has RuntimeVisibleAnnotations attribute
 689   }
 690   if (type_anno != NULL) {
 691     ++attr_count;     // has RuntimeVisibleTypeAnnotations attribute
 692   }
 693   if (cpool()->operands() != NULL) {
 694     ++attr_count;
 695   }
 696   if (ik()->nest_host_index() != 0) {
 697     ++attr_count;
 698   }
 699   if (ik()->nest_members() != Universe::the_empty_short_array()) {
 700     ++attr_count;
 701   }
 702 
 703   write_u2(attr_count);
 704 
 705   if (generic_signature != NULL) {
 706     write_signature_attribute(symbol_to_cpool_index(generic_signature));
 707   }
 708   if (ik()->source_file_name() != NULL) {
 709     write_source_file_attribute();
 710   }
 711   if (ik()->source_debug_extension() != NULL) {
 712     write_source_debug_extension_attribute();
 713   }
 714   if (inner_classes_length > 0) {
 715     write_inner_classes_attribute(inner_classes_length);
 716   }
 717   if (anno != NULL) {
 718     write_annotations_attribute("RuntimeVisibleAnnotations", anno);
 719   }
 720   if (type_anno != NULL) {
 721     write_annotations_attribute("RuntimeVisibleTypeAnnotations", type_anno);
 722   }
 723   if (cpool()->operands() != NULL) {
 724     write_bootstrapmethod_attribute();
 725   }
 726   if (ik()->nest_host_index() != 0) {
 727     write_nest_host_attribute();
 728   }
 729   if (ik()->nest_members() != Universe::the_empty_short_array()) {
 730     write_nest_members_attribute();
 731   }
 732 }
 733 
 734 // Write the method information portion of ClassFile structure
 735 // JVMSpec|     u2 methods_count;
 736 // JVMSpec|     method_info methods[methods_count];
 737 void JvmtiClassFileReconstituter::write_method_infos() {
 738   HandleMark hm(thread());
 739   Array<Method*>* methods = ik()->methods();
 740   int num_methods = methods->length();
 741   int num_overpass = 0;
 742 
 743   // count the generated default interface methods
 744   // these will not be re-created by write_method_info
 745   // and should not be included in the total count
 746   for (int index = 0; index < num_methods; index++) {
 747     Method* method = methods->at(index);
 748     if (method->is_overpass()) {
 749       num_overpass++;
 750     }
 751   }
 752 
 753   write_u2(num_methods - num_overpass);
 754   if (JvmtiExport::can_maintain_original_method_order()) {
 755     int index;
 756     int original_index;
 757     intArray method_order(num_methods, num_methods, 0);
 758 
 759     // invert the method order mapping
 760     for (index = 0; index < num_methods; index++) {
 761       original_index = ik()->method_ordering()->at(index);
 762       assert(original_index >= 0 && original_index < num_methods,
 763              "invalid original method index");
 764       method_order.at_put(original_index, index);
 765     }
 766 
 767     // write in original order
 768     for (original_index = 0; original_index < num_methods; original_index++) {
 769       index = method_order.at(original_index);
 770       methodHandle method(thread(), methods->at(index));
 771       write_method_info(method);
 772     }
 773   } else {
 774     // method order not preserved just dump the method infos
 775     for (int index = 0; index < num_methods; index++) {
 776       methodHandle method(thread(), methods->at(index));
 777       write_method_info(method);
 778     }
 779   }
 780 }
 781 
 782 void JvmtiClassFileReconstituter::write_class_file_format() {
 783   ReallocMark();
 784 
 785   // JVMSpec|   ClassFile {
 786   // JVMSpec|           u4 magic;
 787   write_u4(0xCAFEBABE);
 788 
 789   // JVMSpec|           u2 minor_version;
 790   // JVMSpec|           u2 major_version;
 791   write_u2(ik()->minor_version());
 792   u2 major = ik()->major_version();
 793   write_u2(major);
 794 
 795   // JVMSpec|           u2 constant_pool_count;
 796   // JVMSpec|           cp_info constant_pool[constant_pool_count-1];
 797   write_u2(cpool()->length());
 798   copy_cpool_bytes(writeable_address(cpool_size()));
 799 
 800   // JVMSpec|           u2 access_flags;
 801   write_u2(ik()->access_flags().get_flags() & JVM_RECOGNIZED_CLASS_MODIFIERS);
 802 
 803   // JVMSpec|           u2 this_class;
 804   // JVMSpec|           u2 super_class;
 805   write_u2(class_symbol_to_cpool_index(ik()->name()));
 806   Klass* super_class = ik()->super();
 807   write_u2(super_class == NULL? 0 :  // zero for java.lang.Object
 808                 class_symbol_to_cpool_index(super_class->name()));
 809 
 810   // JVMSpec|           u2 interfaces_count;
 811   // JVMSpec|           u2 interfaces[interfaces_count];
 812   Array<InstanceKlass*>* interfaces =  ik()->local_interfaces();
 813   int num_interfaces = interfaces->length();
 814   write_u2(num_interfaces);
 815   for (int index = 0; index < num_interfaces; index++) {
 816     HandleMark hm(thread());
 817     InstanceKlass* iik = interfaces->at(index);
 818     write_u2(class_symbol_to_cpool_index(iik->name()));
 819   }
 820 
 821   // JVMSpec|           u2 fields_count;
 822   // JVMSpec|           field_info fields[fields_count];
 823   write_field_infos();
 824 
 825   // JVMSpec|           u2 methods_count;
 826   // JVMSpec|           method_info methods[methods_count];
 827   write_method_infos();
 828 
 829   // JVMSpec|           u2 attributes_count;
 830   // JVMSpec|           attribute_info attributes[attributes_count];
 831   // JVMSpec|   } /* end ClassFile 8?
 832   write_class_attributes();
 833 }
 834 
 835 address JvmtiClassFileReconstituter::writeable_address(size_t size) {
 836   size_t used_size = _buffer_ptr - _buffer;
 837   if (size + used_size >= _buffer_size) {
 838     // compute the new buffer size: must be at least twice as big as before
 839     // plus whatever new is being used; then convert to nice clean block boundary
 840     size_t new_buffer_size = (size + _buffer_size*2 + 1) / initial_buffer_size
 841                                                          * initial_buffer_size;
 842 
 843     // VM goes belly-up if the memory isn't available, so cannot do OOM processing
 844     _buffer = REALLOC_RESOURCE_ARRAY(u1, _buffer, _buffer_size, new_buffer_size);
 845     _buffer_size = new_buffer_size;
 846     _buffer_ptr = _buffer + used_size;
 847   }
 848   u1* ret_ptr = _buffer_ptr;
 849   _buffer_ptr += size;
 850   return ret_ptr;
 851 }
 852 
 853 void JvmtiClassFileReconstituter::write_attribute_name_index(const char* name) {
 854   TempNewSymbol sym = SymbolTable::probe(name, (int)strlen(name));
 855   assert(sym != NULL, "attribute name symbol not found");
 856   u2 attr_name_index = symbol_to_cpool_index(sym);
 857   assert(attr_name_index != 0, "attribute name symbol not in constant pool");
 858   write_u2(attr_name_index);
 859 }
 860 
 861 void JvmtiClassFileReconstituter::write_u1(u1 x) {
 862   *writeable_address(1) = x;
 863 }
 864 
 865 void JvmtiClassFileReconstituter::write_u2(u2 x) {
 866   Bytes::put_Java_u2(writeable_address(2), x);
 867 }
 868 
 869 void JvmtiClassFileReconstituter::write_u4(u4 x) {
 870   Bytes::put_Java_u4(writeable_address(4), x);
 871 }
 872 
 873 void JvmtiClassFileReconstituter::write_u8(u8 x) {
 874   Bytes::put_Java_u8(writeable_address(8), x);
 875 }
 876 
 877 void JvmtiClassFileReconstituter::copy_bytecodes(const methodHandle& mh,
 878                                                  unsigned char* bytecodes) {
 879   // use a BytecodeStream to iterate over the bytecodes. JVM/fast bytecodes
 880   // and the breakpoint bytecode are converted to their original bytecodes.
 881 
 882   BytecodeStream bs(mh);
 883 
 884   unsigned char* p = bytecodes;
 885   Bytecodes::Code code;
 886   bool is_rewritten = mh->method_holder()->is_rewritten();
 887 
 888   while ((code = bs.next()) >= 0) {
 889     assert(Bytecodes::is_java_code(code), "sanity check");
 890     assert(code != Bytecodes::_breakpoint, "sanity check");
 891 
 892     // length of bytecode (mnemonic + operands)
 893     address bcp = bs.bcp();
 894     int     len = bs.instruction_size();
 895     assert(len > 0, "length must be > 0");
 896 
 897     // copy the bytecodes
 898     *p = (unsigned char) (bs.is_wide()? Bytecodes::_wide : code);
 899     if (len > 1) {
 900       memcpy(p+1, bcp+1, len-1);
 901     }
 902 
 903     // During linking the get/put and invoke instructions are rewritten
 904     // with an index into the constant pool cache. The original constant
 905     // pool index must be returned to caller.  Rewrite the index.
 906     if (is_rewritten && len > 1) {
 907       bool is_wide = false;
 908       switch (code) {
 909       case Bytecodes::_getstatic       :  // fall through
 910       case Bytecodes::_putstatic       :  // fall through
 911       case Bytecodes::_getfield        :  // fall through
 912       case Bytecodes::_putfield        :  // fall through
 913       case Bytecodes::_invokevirtual   :  // fall through
 914       case Bytecodes::_invokespecial   :  // fall through
 915       case Bytecodes::_invokestatic    :  // fall through
 916       case Bytecodes::_invokedynamic   :  // fall through
 917       case Bytecodes::_invokeinterface : {
 918         assert(len == 3 ||
 919                (code == Bytecodes::_invokeinterface && len == 5) ||
 920                (code == Bytecodes::_invokedynamic   && len == 5),
 921                "sanity check");
 922 
 923         int cpci = Bytes::get_native_u2(bcp+1);
 924         bool is_invokedynamic = (code == Bytecodes::_invokedynamic);
 925         ConstantPoolCacheEntry* entry;
 926         if (is_invokedynamic) {
 927           cpci = Bytes::get_native_u4(bcp+1);
 928           entry = mh->constants()->invokedynamic_cp_cache_entry_at(cpci);
 929         } else {
 930         // cache cannot be pre-fetched since some classes won't have it yet
 931           entry = mh->constants()->cache()->entry_at(cpci);
 932         }
 933         int i = entry->constant_pool_index();
 934         assert(i < mh->constants()->length(), "sanity check");
 935         Bytes::put_Java_u2((address)(p+1), (u2)i);     // java byte ordering
 936         if (is_invokedynamic)  *(p+3) = *(p+4) = 0;
 937         break;
 938       }
 939       case Bytecodes::_ldc_w:
 940         is_wide = true; // fall through
 941       case Bytecodes::_ldc: {
 942         if (bs.raw_code() == Bytecodes::_fast_aldc || bs.raw_code() == Bytecodes::_fast_aldc_w) {
 943           int cpci = is_wide ? Bytes::get_native_u2(bcp+1) : (u1)(*(bcp+1));
 944           int i = mh->constants()->object_to_cp_index(cpci);
 945           assert(i < mh->constants()->length(), "sanity check");
 946           if (is_wide) {
 947             Bytes::put_Java_u2((address)(p+1), (u2)i);     // java byte ordering
 948           } else {
 949             *(p+1) = (u1)i;
 950           }
 951         }
 952         break;
 953         }
 954       default:
 955         break;
 956       }
 957     }
 958 
 959     p += len;
 960   }
 961 }