1 /*
2  * Copyright (C) 2016 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef ART_OPENJDKJVMTI_TI_HEAP_H_
18 #define ART_OPENJDKJVMTI_TI_HEAP_H_
19 
20 #include "jvmti.h"
21 
22 namespace openjdkjvmti {
23 
24 class ObjectTagTable;
25 
26 class HeapUtil {
27  public:
HeapUtil(ObjectTagTable * tags)28   explicit HeapUtil(ObjectTagTable* tags) : tags_(tags) {
29   }
30 
31   jvmtiError GetLoadedClasses(jvmtiEnv* env, jint* class_count_ptr, jclass** classes_ptr);
32 
33   jvmtiError IterateOverInstancesOfClass(jvmtiEnv* env,
34                                          jclass klass,
35                                          jvmtiHeapObjectFilter filter,
36                                          jvmtiHeapObjectCallback cb,
37                                          const void* user_data);
38 
39   jvmtiError IterateThroughHeap(jvmtiEnv* env,
40                                 jint heap_filter,
41                                 jclass klass,
42                                 const jvmtiHeapCallbacks* callbacks,
43                                 const void* user_data);
44 
45   jvmtiError FollowReferences(jvmtiEnv* env,
46                               jint heap_filter,
47                               jclass klass,
48                               jobject initial_object,
49                               const jvmtiHeapCallbacks* callbacks,
50                               const void* user_data);
51 
52   static jvmtiError ForceGarbageCollection(jvmtiEnv* env);
53 
GetTags()54   ObjectTagTable* GetTags() {
55     return tags_;
56   }
57 
58   static void Register();
59   static void Unregister();
60 
61  private:
62   ObjectTagTable* tags_;
63 };
64 
65 class HeapExtensions {
66  public:
67   static jvmtiError JNICALL GetObjectHeapId(jvmtiEnv* env, jlong tag, jint* heap_id, ...);
68   static jvmtiError JNICALL GetHeapName(jvmtiEnv* env, jint heap_id, char** heap_name, ...);
69 
70   static jvmtiError JNICALL IterateThroughHeapExt(jvmtiEnv* env,
71                                                   jint heap_filter,
72                                                   jclass klass,
73                                                   const jvmtiHeapCallbacks* callbacks,
74                                                   const void* user_data);
75 };
76 
77 }  // namespace openjdkjvmti
78 
79 #endif  // ART_OPENJDKJVMTI_TI_HEAP_H_
80