1 /*
   2  * Copyright (c) 2012, 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 
  27 #include "memory/universe.hpp"
  28 #include "oops/oop.inline.hpp"
  29 
  30 #include "classfile/symbolTable.hpp"
  31 #include "classfile/classLoaderData.hpp"
  32 
  33 #include "prims/whitebox.hpp"
  34 #include "prims/wbtestmethods/parserTests.hpp"
  35 
  36 #include "runtime/interfaceSupport.hpp"
  37 #include "runtime/os.hpp"
  38 #include "utilities/debug.hpp"
  39 #include "utilities/macros.hpp"
  40 
  41 #if INCLUDE_ALL_GCS
  42 #include "gc_implementation/g1/concurrentMark.hpp"
  43 #include "gc_implementation/g1/g1CollectedHeap.inline.hpp"
  44 #include "gc_implementation/g1/heapRegionRemSet.hpp"
  45 #endif // INCLUDE_ALL_GCS
  46 
  47 #ifdef INCLUDE_NMT
  48 #include "services/memTracker.hpp"
  49 #endif // INCLUDE_NMT
  50 
  51 #include "compiler/compileBroker.hpp"
  52 
  53 bool WhiteBox::_used = false;
  54 
  55 WB_ENTRY(jlong, WB_GetObjectAddress(JNIEnv* env, jobject o, jobject obj))
  56   return (jlong)(void*)JNIHandles::resolve(obj);
  57 WB_END
  58 
  59 WB_ENTRY(jint, WB_GetHeapOopSize(JNIEnv* env, jobject o))
  60   return heapOopSize;
  61 WB_END
  62 
  63 
  64 class WBIsKlassAliveClosure : public KlassClosure {
  65     Symbol* _name;
  66     bool _found;
  67 public:
  68     WBIsKlassAliveClosure(Symbol* name) : _name(name), _found(false) {}
  69 
  70     void do_klass(Klass* k) {
  71       if (_found) return;
  72       Symbol* ksym = k->name();
  73       if (ksym->fast_compare(_name) == 0) {
  74         _found = true;
  75       }
  76     }
  77 
  78     bool found() const {
  79         return _found;
  80     }
  81 };
  82 
  83 WB_ENTRY(jboolean, WB_IsClassAlive(JNIEnv* env, jobject target, jstring name))
  84   Handle h_name = JNIHandles::resolve(name);
  85   if (h_name.is_null()) return false;
  86   Symbol* sym = java_lang_String::as_symbol(h_name, CHECK_false);
  87   TempNewSymbol tsym(sym); // Make sure to decrement reference count on sym on return
  88 
  89   WBIsKlassAliveClosure closure(sym);
  90   ClassLoaderDataGraph::classes_do(&closure);
  91 
  92   return closure.found();
  93 WB_END
  94 
  95 #if INCLUDE_ALL_GCS
  96 WB_ENTRY(jboolean, WB_G1IsHumongous(JNIEnv* env, jobject o, jobject obj))
  97   G1CollectedHeap* g1 = G1CollectedHeap::heap();
  98   oop result = JNIHandles::resolve(obj);
  99   const HeapRegion* hr = g1->heap_region_containing(result);
 100   return hr->isHumongous();
 101 WB_END
 102 
 103 WB_ENTRY(jlong, WB_G1NumFreeRegions(JNIEnv* env, jobject o))
 104   G1CollectedHeap* g1 = G1CollectedHeap::heap();
 105   size_t nr = g1->free_regions();
 106   return (jlong)nr;
 107 WB_END
 108 
 109 WB_ENTRY(jboolean, WB_G1InConcurrentMark(JNIEnv* env, jobject o))
 110   G1CollectedHeap* g1 = G1CollectedHeap::heap();
 111   ConcurrentMark* cm = g1->concurrent_mark();
 112   return cm->concurrent_marking_in_progress();
 113 WB_END
 114 
 115 WB_ENTRY(jint, WB_G1RegionSize(JNIEnv* env, jobject o))
 116   return (jint)HeapRegion::GrainBytes;
 117 WB_END
 118 #endif // INCLUDE_ALL_GCS
 119 
 120 #ifdef INCLUDE_NMT
 121 // Keep track of the 3 allocations in NMTAllocTest so we can free them later
 122 // on and verify that they're not visible anymore
 123 static void* nmtMtTest1 = NULL, *nmtMtTest2 = NULL, *nmtMtTest3 = NULL;
 124 
 125 // Alloc memory using the test memory type so that we can use that to see if
 126 // NMT picks it up correctly
 127 WB_ENTRY(jboolean, WB_NMTAllocTest(JNIEnv* env))
 128   void *mem;
 129 
 130   if (!MemTracker::is_on() || MemTracker::shutdown_in_progress()) {
 131     return false;
 132   }
 133 
 134   // Allocate 2 * 128k + 256k + 1024k and free the 1024k one to make sure we track
 135   // everything correctly. Total should be 512k held alive.
 136   nmtMtTest1 = os::malloc(128 * 1024, mtTest);
 137   mem = os::malloc(1024 * 1024, mtTest);
 138   nmtMtTest2 = os::malloc(256 * 1024, mtTest);
 139   os::free(mem, mtTest);
 140   nmtMtTest3 = os::malloc(128 * 1024, mtTest);
 141 
 142   return true;
 143 WB_END
 144 
 145 // Free the memory allocated by NMTAllocTest
 146 WB_ENTRY(jboolean, WB_NMTFreeTestMemory(JNIEnv* env))
 147 
 148   if (nmtMtTest1 == NULL || nmtMtTest2 == NULL || nmtMtTest3 == NULL) {
 149     return false;
 150   }
 151 
 152   os::free(nmtMtTest1, mtTest);
 153   nmtMtTest1 = NULL;
 154   os::free(nmtMtTest2, mtTest);
 155   nmtMtTest2 = NULL;
 156   os::free(nmtMtTest3, mtTest);
 157   nmtMtTest3 = NULL;
 158 
 159   return true;
 160 WB_END
 161 
 162 // Block until the current generation of NMT data to be merged, used to reliably test the NMT feature
 163 WB_ENTRY(jboolean, WB_NMTWaitForDataMerge(JNIEnv* env))
 164 
 165   if (!MemTracker::is_on() || MemTracker::shutdown_in_progress()) {
 166     return false;
 167   }
 168 
 169   return MemTracker::wbtest_wait_for_data_merge();
 170 WB_END
 171 
 172 #endif // INCLUDE_NMT
 173 
 174 static jmethodID reflected_method_to_jmid(JavaThread* thread, JNIEnv* env, jobject method) {
 175   assert(method != NULL, "method should not be null");
 176   ThreadToNativeFromVM ttn(thread);
 177   return env->FromReflectedMethod(method);
 178 }
 179 
 180 WB_ENTRY(void, WB_DeoptimizeAll(JNIEnv* env, jobject o))
 181   MutexLockerEx mu(Compile_lock);
 182   CodeCache::mark_all_nmethods_for_deoptimization();
 183   VM_Deoptimize op;
 184   VMThread::execute(&op);
 185 WB_END
 186 
 187 WB_ENTRY(jint, WB_DeoptimizeMethod(JNIEnv* env, jobject o, jobject method))
 188   jmethodID jmid = reflected_method_to_jmid(thread, env, method);
 189   MutexLockerEx mu(Compile_lock);
 190   methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));
 191   int result = 0;
 192   nmethod* code = mh->code();
 193   if (code != NULL) {
 194     code->mark_for_deoptimization();
 195     ++result;
 196   }
 197   result += CodeCache::mark_for_deoptimization(mh());
 198   if (result > 0) {
 199     VM_Deoptimize op;
 200     VMThread::execute(&op);
 201   }
 202   return result;
 203 WB_END
 204 
 205 WB_ENTRY(jboolean, WB_IsMethodCompiled(JNIEnv* env, jobject o, jobject method))
 206   jmethodID jmid = reflected_method_to_jmid(thread, env, method);
 207   MutexLockerEx mu(Compile_lock);
 208   methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));
 209   nmethod* code = mh->code();
 210   if (code == NULL) {
 211     return JNI_FALSE;
 212   }
 213   return (code->is_alive() && !code->is_marked_for_deoptimization());
 214 WB_END
 215 
 216 WB_ENTRY(jboolean, WB_IsMethodCompilable(JNIEnv* env, jobject o, jobject method))
 217   jmethodID jmid = reflected_method_to_jmid(thread, env, method);
 218   MutexLockerEx mu(Compile_lock);
 219   methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));
 220   return !mh->is_not_compilable();
 221 WB_END
 222 
 223 WB_ENTRY(jboolean, WB_IsMethodQueuedForCompilation(JNIEnv* env, jobject o, jobject method))
 224   jmethodID jmid = reflected_method_to_jmid(thread, env, method);
 225   MutexLockerEx mu(Compile_lock);
 226   methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));
 227   return mh->queued_for_compilation();
 228 WB_END
 229 
 230 WB_ENTRY(jint, WB_GetMethodCompilationLevel(JNIEnv* env, jobject o, jobject method))
 231   jmethodID jmid = reflected_method_to_jmid(thread, env, method);
 232   methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));
 233   nmethod* code = mh->code();
 234   return (code != NULL ? code->comp_level() : CompLevel_none);
 235 WB_END
 236 
 237 
 238 WB_ENTRY(void, WB_MakeMethodNotCompilable(JNIEnv* env, jobject o, jobject method))
 239   jmethodID jmid = reflected_method_to_jmid(thread, env, method);
 240   methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));
 241   mh->set_not_compilable();
 242 WB_END
 243 
 244 WB_ENTRY(jboolean, WB_SetDontInlineMethod(JNIEnv* env, jobject o, jobject method, jboolean value))
 245   jmethodID jmid = reflected_method_to_jmid(thread, env, method);
 246   methodHandle mh(THREAD, Method::checked_resolve_jmethod_id(jmid));
 247   bool result = mh->dont_inline();
 248   mh->set_dont_inline(value == JNI_TRUE);
 249   return result;
 250 WB_END
 251 
 252 WB_ENTRY(jint, WB_GetCompileQueuesSize(JNIEnv* env, jobject o))
 253   return CompileBroker::queue_size(CompLevel_full_optimization) /* C2 */ +
 254          CompileBroker::queue_size(CompLevel_full_profile) /* C1 */;
 255 WB_END
 256 
 257 //Some convenience methods to deal with objects from java
 258 int WhiteBox::offset_for_field(const char* field_name, oop object,
 259     Symbol* signature_symbol) {
 260   assert(field_name != NULL && strlen(field_name) > 0, "Field name not valid");
 261   Thread* THREAD = Thread::current();
 262 
 263   //Get the class of our object
 264   Klass* arg_klass = object->klass();
 265   //Turn it into an instance-klass
 266   InstanceKlass* ik = InstanceKlass::cast(arg_klass);
 267 
 268   //Create symbols to look for in the class
 269   TempNewSymbol name_symbol = SymbolTable::lookup(field_name, (int) strlen(field_name),
 270       THREAD);
 271 
 272   //To be filled in with an offset of the field we're looking for
 273   fieldDescriptor fd;
 274 
 275   Klass* res = ik->find_field(name_symbol, signature_symbol, &fd);
 276   if (res == NULL) {
 277     tty->print_cr("Invalid layout of %s at %s", ik->external_name(),
 278         name_symbol->as_C_string());
 279     fatal("Invalid layout of preloaded class");
 280   }
 281 
 282   //fetch the field at the offset we've found
 283   int dest_offset = fd.offset();
 284 
 285   return dest_offset;
 286 }
 287 
 288 
 289 const char* WhiteBox::lookup_jstring(const char* field_name, oop object) {
 290   int offset = offset_for_field(field_name, object,
 291       vmSymbols::string_signature());
 292   oop string = object->obj_field(offset);
 293   if (string == NULL) {
 294     return NULL;
 295   }
 296   const char* ret = java_lang_String::as_utf8_string(string);
 297   return ret;
 298 }
 299 
 300 bool WhiteBox::lookup_bool(const char* field_name, oop object) {
 301   int offset =
 302       offset_for_field(field_name, object, vmSymbols::bool_signature());
 303   bool ret = (object->bool_field(offset) == JNI_TRUE);
 304   return ret;
 305 }
 306 
 307 
 308 #define CC (char*)
 309 
 310 static JNINativeMethod methods[] = {
 311   {CC"getObjectAddress",   CC"(Ljava/lang/Object;)J", (void*)&WB_GetObjectAddress  },
 312   {CC"getHeapOopSize",     CC"()I",                   (void*)&WB_GetHeapOopSize    },
 313   {CC"isClassAlive0",      CC"(Ljava/lang/String;)Z", (void*)&WB_IsClassAlive      },
 314   {CC"parseCommandLine",
 315       CC"(Ljava/lang/String;[Lsun/hotspot/parser/DiagnosticCommand;)[Ljava/lang/Object;",
 316       (void*) &WB_ParseCommandLine
 317   },
 318 #if INCLUDE_ALL_GCS
 319   {CC"g1InConcurrentMark", CC"()Z",                   (void*)&WB_G1InConcurrentMark},
 320   {CC"g1IsHumongous",      CC"(Ljava/lang/Object;)Z", (void*)&WB_G1IsHumongous     },
 321   {CC"g1NumFreeRegions",   CC"()J",                   (void*)&WB_G1NumFreeRegions  },
 322   {CC"g1RegionSize",       CC"()I",                   (void*)&WB_G1RegionSize      },
 323 #endif // INCLUDE_ALL_GCS
 324 #ifdef INCLUDE_NMT
 325   {CC"NMTAllocTest",       CC"()Z",                   (void*)&WB_NMTAllocTest      },
 326   {CC"NMTFreeTestMemory",  CC"()Z",                   (void*)&WB_NMTFreeTestMemory },
 327   {CC"NMTWaitForDataMerge",CC"()Z",                   (void*)&WB_NMTWaitForDataMerge},
 328 #endif // INCLUDE_NMT
 329   {CC"deoptimizeAll",      CC"()V",                   (void*)&WB_DeoptimizeAll     },
 330   {CC"deoptimizeMethod",   CC"(Ljava/lang/reflect/Method;)I",
 331                                                       (void*)&WB_DeoptimizeMethod  },
 332   {CC"isMethodCompiled",   CC"(Ljava/lang/reflect/Method;)Z",
 333                                                       (void*)&WB_IsMethodCompiled  },
 334   {CC"isMethodCompilable", CC"(Ljava/lang/reflect/Method;)Z",
 335                                                       (void*)&WB_IsMethodCompilable},
 336   {CC"isMethodQueuedForCompilation",
 337       CC"(Ljava/lang/reflect/Method;)Z",              (void*)&WB_IsMethodQueuedForCompilation},
 338   {CC"makeMethodNotCompilable",
 339       CC"(Ljava/lang/reflect/Method;)V",              (void*)&WB_MakeMethodNotCompilable},
 340   {CC"setDontInlineMethod",
 341       CC"(Ljava/lang/reflect/Method;Z)Z",             (void*)&WB_SetDontInlineMethod},
 342   {CC"getMethodCompilationLevel",
 343       CC"(Ljava/lang/reflect/Method;)I",              (void*)&WB_GetMethodCompilationLevel},
 344   {CC"getCompileQueuesSize",
 345       CC"()I",                                        (void*)&WB_GetCompileQueuesSize},
 346 };
 347 
 348 #undef CC
 349 
 350 JVM_ENTRY(void, JVM_RegisterWhiteBoxMethods(JNIEnv* env, jclass wbclass))
 351   {
 352     if (WhiteBoxAPI) {
 353       // Make sure that wbclass is loaded by the null classloader
 354       instanceKlassHandle ikh = instanceKlassHandle(JNIHandles::resolve(wbclass)->klass());
 355       Handle loader(ikh->class_loader());
 356       if (loader.is_null()) {
 357         ThreadToNativeFromVM ttnfv(thread); // can't be in VM when we call JNI
 358         jint result = env->RegisterNatives(wbclass, methods, sizeof(methods)/sizeof(methods[0]));
 359         if (result == 0) {
 360           WhiteBox::set_used();
 361         }
 362       }
 363     }
 364   }
 365 JVM_END