1 /*
  2  * Copyright (c) 2022, 2023, 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/classLoaderData.hpp"
 27 #include "classfile/classLoaderDataGraph.hpp"
 28 #include "classfile/classPrinter.hpp"
 29 #include "memory/iterator.hpp"
 30 #include "memory/resourceArea.hpp"
 31 #include "oops/instanceKlass.hpp"
 32 #include "oops/klass.hpp"
 33 #include "oops/method.hpp"
 34 #include "oops/symbol.hpp"
 35 #include "utilities/ostream.hpp"
 36 
 37 class ClassPrinter::KlassPrintClosure : public LockedClassesDo {
 38   const char* _class_name_pattern;
 39   const char* _method_name_pattern;
 40   const char* _method_signature_pattern;
 41   bool _always_print_class_name;
 42   int _flags;
 43   outputStream* _st;
 44   int _num;
 45   bool _has_printed_methods;
 46 public:
 47   KlassPrintClosure(const char* class_name_pattern,
 48                     const char* method_name_pattern,
 49                     const char* method_signature_pattern,
 50                     bool always_print_class_name,
 51                     int flags, outputStream* st)
 52     : _class_name_pattern(class_name_pattern),
 53       _method_name_pattern(method_name_pattern),
 54       _method_signature_pattern(method_signature_pattern),
 55       _always_print_class_name(always_print_class_name),
 56       _flags(flags), _st(st), _num(0), _has_printed_methods(false)
 57   {
 58     if (has_mode(_flags, PRINT_METHOD_HANDLE)) {
 59       _flags |= (PRINT_METHOD_NAME | PRINT_BYTECODE);
 60     }
 61     if (has_mode(_flags, PRINT_DYNAMIC)) {
 62       _flags |= (PRINT_METHOD_NAME | PRINT_BYTECODE);
 63     }
 64     if (has_mode(_flags, PRINT_BYTECODE_ADDR)) {
 65       _flags |= (PRINT_METHOD_NAME | PRINT_BYTECODE);
 66     }
 67     if (has_mode(_flags, PRINT_BYTECODE)) {
 68       _flags |= (PRINT_METHOD_NAME);
 69     }
 70   }
 71 
 72   virtual void do_klass(Klass* k) {
 73     if (!k->is_instance_klass()) {
 74       return;
 75     }
 76     print_instance_klass(InstanceKlass::cast(k));
 77   }
 78 
 79   static bool match(const char* pattern, Symbol* sym) {
 80     return (pattern == nullptr || sym->is_star_match(pattern));
 81   }
 82 
 83   void print_klass_name(InstanceKlass* ik) {
 84     _st->print("[%3d] " INTPTR_FORMAT " class %s ", _num++, p2i(ik), ik->name()->as_C_string());
 85     ik->class_loader_data()->print_value_on(_st);
 86     _st->cr();
 87   }
 88 
 89   void print_instance_klass(InstanceKlass* ik) {
 90     if (ik->is_loaded() && ik->name()->is_star_match(_class_name_pattern)) {
 91       ResourceMark rm;
 92       if (_has_printed_methods) {
 93         // We have printed some methods in the previous class.
 94         // Print a new line to separate the two classes
 95         _st->cr();
 96       }
 97       _has_printed_methods = false;
 98       if (_always_print_class_name) {
 99         print_klass_name(ik);
100       }
101 
102       if (has_mode(_flags, ClassPrinter::PRINT_METHOD_NAME)) {
103         bool print_codes = has_mode(_flags, ClassPrinter::PRINT_BYTECODE);
104         int len = ik->methods()->length();
105         int num_methods_printed = 0;
106 
107         for (int index = 0; index < len; index++) {
108           Method* m = ik->methods()->at(index);
109           if (match(_method_name_pattern, m->name()) &&
110               match(_method_signature_pattern, m->signature())) {
111             if (print_codes && num_methods_printed++ > 0) {
112               _st->cr();
113             }
114 
115             if (_has_printed_methods == false) {
116               if (!_always_print_class_name) {
117                 print_klass_name(ik);
118               }
119               _has_printed_methods = true;
120             }
121             print_method(m);
122           }
123         }
124       }
125     }
126   }
127 
128   void print_method(Method* m) {
129     bool print_codes = has_mode(_flags, ClassPrinter::PRINT_BYTECODE);
130     _st->print_cr(INTPTR_FORMAT " %smethod %s : %s", p2i(m),
131                   m->is_static() ? "static " : "",
132                   m->name()->as_C_string(), m->signature()->as_C_string());
133     if (print_codes) {
134       m->print_codes_on(_st, _flags);
135     }
136   }
137 };
138 
139 void ClassPrinter::print_flags_help(outputStream* os) {
140   os->print_cr("flags (bitmask):");
141   os->print_cr("   0x%02x  - print names of methods", PRINT_METHOD_NAME);
142   os->print_cr("   0x%02x  - print bytecodes", PRINT_BYTECODE);
143   os->print_cr("   0x%02x  - print the address of bytecodes", PRINT_BYTECODE_ADDR);
144   os->print_cr("   0x%02x  - print info for invokedynamic", PRINT_DYNAMIC);
145   os->print_cr("   0x%02x  - print info for invokehandle",  PRINT_METHOD_HANDLE);
146   os->cr();
147 }
148 
149 void ClassPrinter::print_classes(const char* class_name_pattern, int flags, outputStream* os) {
150   KlassPrintClosure closure(class_name_pattern, nullptr, nullptr, true, flags, os);
151   ClassLoaderDataGraph::classes_do(&closure);
152 }
153 
154 void ClassPrinter::print_methods(const char* class_name_pattern,
155                                  const char* method_pattern, int flags, outputStream* os) {
156   ResourceMark rm;
157   const char* method_name_pattern;
158   const char* method_signature_pattern;
159 
160   const char* colon = strchr(method_pattern, ':');
161   if (colon == nullptr) {
162     method_name_pattern = method_pattern;
163     method_signature_pattern = nullptr;
164   } else {
165     ptrdiff_t name_pat_len = colon - method_pattern;
166     assert(name_pat_len >= 0, "sanity");
167     char* buf = NEW_RESOURCE_ARRAY(char, name_pat_len + 1);
168     strncpy(buf, method_pattern, name_pat_len);
169     buf[name_pat_len] = 0;
170 
171     method_name_pattern = buf;
172     method_signature_pattern = colon + 1;
173   }
174 
175   KlassPrintClosure closure(class_name_pattern, method_name_pattern, method_signature_pattern,
176                             false, flags | PRINT_METHOD_NAME, os);
177   ClassLoaderDataGraph::classes_do(&closure);
178 }