1/* Copyright (C) 2017 The Android Open Source Project
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
3 *
4 * This file implements interfaces from the file jvmti.h. This implementation
5 * is licensed under the same terms as the file jvmti.h.  The
6 * copyright and license information for the file jvmti.h follows.
7 *
8 * Copyright (c) 2003, 2011, Oracle and/or its affiliates. All rights reserved.
9 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
10 *
11 * This code is free software; you can redistribute it and/or modify it
12 * under the terms of the GNU General Public License version 2 only, as
13 * published by the Free Software Foundation.  Oracle designates this
14 * particular file as subject to the "Classpath" exception as provided
15 * by Oracle in the LICENSE file that accompanied this code.
16 *
17 * This code is distributed in the hope that it will be useful, but WITHOUT
18 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
19 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
20 * version 2 for more details (a copy is included in the LICENSE file that
21 * accompanied this code).
22 *
23 * You should have received a copy of the GNU General Public License version
24 * 2 along with this work; if not, write to the Free Software Foundation,
25 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
26 *
27 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
28 * or visit www.oracle.com if you need additional information or have any
29 * questions.
30 */
31
32#ifndef ART_RUNTIME_OPENJDKJVMTI_TI_CLASS_DEFINITION_H_
33#define ART_RUNTIME_OPENJDKJVMTI_TI_CLASS_DEFINITION_H_
34
35#include "art_jvmti.h"
36
37namespace openjdkjvmti {
38
39// A struct that stores data needed for redefining/transforming classes. This structure should only
40// even be accessed from a single thread and must not survive past the completion of the
41// redefinition/retransformation function that created it.
42class ArtClassDefinition {
43 public:
44  ArtClassDefinition()
45      : klass_(nullptr),
46        loader_(nullptr),
47        name_(),
48        protection_domain_(nullptr),
49        dex_len_(0),
50        dex_data_(nullptr),
51        original_dex_file_memory_(nullptr),
52        original_dex_file_(),
53        redefined_(false) {}
54
55  jvmtiError Init(ArtJvmTiEnv* env, jclass klass);
56  jvmtiError Init(ArtJvmTiEnv* env, const jvmtiClassDefinition& def);
57
58  ArtClassDefinition(ArtClassDefinition&& o) = default;
59  ArtClassDefinition& operator=(ArtClassDefinition&& o) = default;
60
61  void SetNewDexData(ArtJvmTiEnv* env, jint new_dex_len, unsigned char* new_dex_data) {
62    DCHECK(IsInitialized());
63    if (new_dex_data == nullptr) {
64      return;
65    } else if (new_dex_data != dex_data_.get() || new_dex_len != dex_len_) {
66      dex_len_ = new_dex_len;
67      dex_data_ = MakeJvmtiUniquePtr(env, new_dex_data);
68    }
69  }
70
71  art::ArraySlice<const unsigned char> GetNewOriginalDexFile() const {
72    DCHECK(IsInitialized());
73    if (redefined_) {
74      return original_dex_file_;
75    } else {
76      return art::ArraySlice<const unsigned char>();
77    }
78  }
79
80  bool IsModified() const;
81
82  bool IsInitialized() const {
83    return klass_ != nullptr;
84  }
85
86  jclass GetClass() const {
87    DCHECK(IsInitialized());
88    return klass_;
89  }
90
91  jobject GetLoader() const {
92    DCHECK(IsInitialized());
93    return loader_;
94  }
95
96  const std::string& GetName() const {
97    DCHECK(IsInitialized());
98    return name_;
99  }
100
101  jobject GetProtectionDomain() const {
102    DCHECK(IsInitialized());
103    return protection_domain_;
104  }
105
106  art::ArraySlice<const unsigned char> GetDexData() const {
107    DCHECK(IsInitialized());
108    return art::ArraySlice<const unsigned char>(dex_data_.get(), dex_len_);
109  }
110
111 private:
112  jvmtiError InitCommon(ArtJvmTiEnv* env, jclass klass);
113
114  jclass klass_;
115  jobject loader_;
116  std::string name_;
117  jobject protection_domain_;
118  jint dex_len_;
119  JvmtiUniquePtr<unsigned char> dex_data_;
120  JvmtiUniquePtr<unsigned char> original_dex_file_memory_;
121  art::ArraySlice<const unsigned char> original_dex_file_;
122  bool redefined_;
123
124  DISALLOW_COPY_AND_ASSIGN(ArtClassDefinition);
125};
126
127}  // namespace openjdkjvmti
128
129#endif  // ART_RUNTIME_OPENJDKJVMTI_TI_CLASS_DEFINITION_H_
130