1 /*
2  * Copyright (C) 2014 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 #include <jni.h>
18 #include <vector>
19 
20 #include "common_compiler_test.h"
21 #include "field_helper.h"
22 #include "mirror/art_field-inl.h"
23 #include "scoped_thread_state_change.h"
24 
25 namespace art {
26 
27 class ProxyTest : public CommonCompilerTest {
28  public:
29   // Generate a proxy class with the given name and interfaces. This is a simplification from what
30   // libcore does to fit to our test needs. We do not check for duplicated interfaces or methods and
31   // we do not declare exceptions.
GenerateProxyClass(ScopedObjectAccess & soa,jobject jclass_loader,const char * className,const std::vector<mirror::Class * > & interfaces)32   mirror::Class* GenerateProxyClass(ScopedObjectAccess& soa, jobject jclass_loader,
33                                     const char* className,
34                                     const std::vector<mirror::Class*>& interfaces)
35       SHARED_LOCKS_REQUIRED(Locks::mutator_lock_) {
36     mirror::Class* javaLangObject = class_linker_->FindSystemClass(soa.Self(), "Ljava/lang/Object;");
37     CHECK(javaLangObject != nullptr);
38 
39     jclass javaLangClass = soa.AddLocalReference<jclass>(mirror::Class::GetJavaLangClass());
40 
41     // Builds the interfaces array.
42     jobjectArray proxyClassInterfaces = soa.Env()->NewObjectArray(interfaces.size(), javaLangClass,
43                                                                   nullptr);
44     soa.Self()->AssertNoPendingException();
45     for (size_t i = 0; i < interfaces.size(); ++i) {
46       soa.Env()->SetObjectArrayElement(proxyClassInterfaces, i,
47                                        soa.AddLocalReference<jclass>(interfaces[i]));
48     }
49 
50     // Builds the method array.
51     jsize methods_count = 3;  // Object.equals, Object.hashCode and Object.toString.
52     for (mirror::Class* interface : interfaces) {
53       mirror::ObjectArray<mirror::ArtMethod>* virtual_methods = interface->GetVirtualMethods();
54       methods_count += (virtual_methods == nullptr) ? 0 : virtual_methods->GetLength();
55     }
56     jclass javaLangReflectArtMethod =
57         soa.AddLocalReference<jclass>(mirror::ArtMethod::GetJavaLangReflectArtMethod());
58     jobjectArray proxyClassMethods = soa.Env()->NewObjectArray(methods_count,
59                                                                javaLangReflectArtMethod, nullptr);
60     soa.Self()->AssertNoPendingException();
61 
62     // Fill the method array
63     mirror::ArtMethod* equalsMethod = javaLangObject->FindDeclaredVirtualMethod("equals",
64                                                                                 "(Ljava/lang/Object;)Z");
65     mirror::ArtMethod* hashCodeMethod = javaLangObject->FindDeclaredVirtualMethod("hashCode",
66                                                                                   "()I");
67     mirror::ArtMethod* toStringMethod = javaLangObject->FindDeclaredVirtualMethod("toString",
68                                                                                   "()Ljava/lang/String;");
69     CHECK(equalsMethod != nullptr);
70     CHECK(hashCodeMethod != nullptr);
71     CHECK(toStringMethod != nullptr);
72 
73     jsize array_index = 0;
74     // Adds Object methods.
75     soa.Env()->SetObjectArrayElement(proxyClassMethods, array_index++,
76                                      soa.AddLocalReference<jobject>(equalsMethod));
77     soa.Env()->SetObjectArrayElement(proxyClassMethods, array_index++,
78                                      soa.AddLocalReference<jobject>(hashCodeMethod));
79     soa.Env()->SetObjectArrayElement(proxyClassMethods, array_index++,
80                                      soa.AddLocalReference<jobject>(toStringMethod));
81 
82     // Now adds all interfaces virtual methods.
83     for (mirror::Class* interface : interfaces) {
84       mirror::ObjectArray<mirror::ArtMethod>* virtual_methods = interface->GetVirtualMethods();
85       if (virtual_methods != nullptr) {
86         for (int32_t mth_index = 0; mth_index < virtual_methods->GetLength(); ++mth_index) {
87           mirror::ArtMethod* method = virtual_methods->Get(mth_index);
88           soa.Env()->SetObjectArrayElement(proxyClassMethods, array_index++,
89                                            soa.AddLocalReference<jobject>(method));
90         }
91       }
92     }
93     CHECK_EQ(array_index, methods_count);
94 
95     // Builds an empty exception array.
96     jobjectArray proxyClassThrows = soa.Env()->NewObjectArray(0, javaLangClass, nullptr);
97     soa.Self()->AssertNoPendingException();
98 
99     mirror::Class* proxyClass = class_linker_->CreateProxyClass(soa,
100                                                                 soa.Env()->NewStringUTF(className),
101                                                                 proxyClassInterfaces, jclass_loader,
102                                                                 proxyClassMethods, proxyClassThrows);
103     soa.Self()->AssertNoPendingException();
104     return proxyClass;
105   }
106 };
107 
108 // Creates a proxy class and check ClassHelper works correctly.
TEST_F(ProxyTest,ProxyClassHelper)109 TEST_F(ProxyTest, ProxyClassHelper) {
110   ScopedObjectAccess soa(Thread::Current());
111   jobject jclass_loader = LoadDex("Interfaces");
112   StackHandleScope<4> hs(soa.Self());
113   Handle<mirror::ClassLoader> class_loader(
114       hs.NewHandle(soa.Decode<mirror::ClassLoader*>(jclass_loader)));
115 
116   Handle<mirror::Class> I(hs.NewHandle(
117       class_linker_->FindClass(soa.Self(), "LInterfaces$I;", class_loader)));
118   Handle<mirror::Class> J(hs.NewHandle(
119       class_linker_->FindClass(soa.Self(), "LInterfaces$J;", class_loader)));
120   ASSERT_TRUE(I.Get() != nullptr);
121   ASSERT_TRUE(J.Get() != nullptr);
122 
123   std::vector<mirror::Class*> interfaces;
124   interfaces.push_back(I.Get());
125   interfaces.push_back(J.Get());
126   Handle<mirror::Class> proxy_class(hs.NewHandle(
127       GenerateProxyClass(soa, jclass_loader, "$Proxy1234", interfaces)));
128   interfaces.clear();  // Don't least possibly stale objects in the array as good practice.
129   ASSERT_TRUE(proxy_class.Get() != nullptr);
130   ASSERT_TRUE(proxy_class->IsProxyClass());
131   ASSERT_TRUE(proxy_class->IsInitialized());
132 
133   EXPECT_EQ(2U, proxy_class->NumDirectInterfaces());  // Interfaces$I and Interfaces$J.
134   EXPECT_EQ(I.Get(), mirror::Class::GetDirectInterface(soa.Self(), proxy_class, 0));
135   EXPECT_EQ(J.Get(), mirror::Class::GetDirectInterface(soa.Self(), proxy_class, 1));
136   std::string temp;
137   const char* proxy_class_descriptor = proxy_class->GetDescriptor(&temp);
138   EXPECT_STREQ("L$Proxy1234;", proxy_class_descriptor);
139   EXPECT_EQ(nullptr, proxy_class->GetSourceFile());
140 }
141 
142 // Creates a proxy class and check FieldHelper works correctly.
TEST_F(ProxyTest,ProxyFieldHelper)143 TEST_F(ProxyTest, ProxyFieldHelper) {
144   ScopedObjectAccess soa(Thread::Current());
145   jobject jclass_loader = LoadDex("Interfaces");
146   StackHandleScope<9> hs(soa.Self());
147   Handle<mirror::ClassLoader> class_loader(
148       hs.NewHandle(soa.Decode<mirror::ClassLoader*>(jclass_loader)));
149 
150   Handle<mirror::Class> I(hs.NewHandle(
151       class_linker_->FindClass(soa.Self(), "LInterfaces$I;", class_loader)));
152   Handle<mirror::Class> J(hs.NewHandle(
153       class_linker_->FindClass(soa.Self(), "LInterfaces$J;", class_loader)));
154   ASSERT_TRUE(I.Get() != nullptr);
155   ASSERT_TRUE(J.Get() != nullptr);
156 
157   Handle<mirror::Class> proxyClass;
158   {
159     std::vector<mirror::Class*> interfaces;
160     interfaces.push_back(I.Get());
161     interfaces.push_back(J.Get());
162     proxyClass = hs.NewHandle(GenerateProxyClass(soa, jclass_loader, "$Proxy1234", interfaces));
163   }
164 
165   ASSERT_TRUE(proxyClass.Get() != nullptr);
166   ASSERT_TRUE(proxyClass->IsProxyClass());
167   ASSERT_TRUE(proxyClass->IsInitialized());
168 
169   Handle<mirror::ObjectArray<mirror::ArtField>> instance_fields(
170       hs.NewHandle(proxyClass->GetIFields()));
171   EXPECT_TRUE(instance_fields.Get() == nullptr);
172 
173   Handle<mirror::ObjectArray<mirror::ArtField>> static_fields(
174       hs.NewHandle(proxyClass->GetSFields()));
175   ASSERT_TRUE(static_fields.Get() != nullptr);
176   ASSERT_EQ(2, static_fields->GetLength());
177 
178   Handle<mirror::Class> interfacesFieldClass(
179       hs.NewHandle(class_linker_->FindSystemClass(soa.Self(), "[Ljava/lang/Class;")));
180   ASSERT_TRUE(interfacesFieldClass.Get() != nullptr);
181   Handle<mirror::Class> throwsFieldClass(
182       hs.NewHandle(class_linker_->FindSystemClass(soa.Self(), "[[Ljava/lang/Class;")));
183   ASSERT_TRUE(throwsFieldClass.Get() != nullptr);
184 
185   // Test "Class[] interfaces" field.
186   FieldHelper fh(hs.NewHandle(static_fields->Get(0)));
187   EXPECT_EQ("interfaces", std::string(fh.GetField()->GetName()));
188   EXPECT_EQ("[Ljava/lang/Class;", std::string(fh.GetField()->GetTypeDescriptor()));
189   EXPECT_EQ(interfacesFieldClass.Get(), fh.GetType());
190   EXPECT_EQ("L$Proxy1234;", std::string(fh.GetDeclaringClassDescriptor()));
191   EXPECT_FALSE(fh.GetField()->IsPrimitiveType());
192 
193   // Test "Class[][] throws" field.
194   fh.ChangeField(static_fields->Get(1));
195   EXPECT_EQ("throws", std::string(fh.GetField()->GetName()));
196   EXPECT_EQ("[[Ljava/lang/Class;", std::string(fh.GetField()->GetTypeDescriptor()));
197   EXPECT_EQ(throwsFieldClass.Get(), fh.GetType());
198   EXPECT_EQ("L$Proxy1234;", std::string(fh.GetDeclaringClassDescriptor()));
199   EXPECT_FALSE(fh.GetField()->IsPrimitiveType());
200 }
201 
202 }  // namespace art
203