Home | History | Annotate | Download | only in misc
      1 /*
      2  * Copyright (C) 2009 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 package sun.misc;
     18 
     19 import junit.framework.TestCase;
     20 
     21 import java.lang.reflect.Field;
     22 import java.util.concurrent.Callable;
     23 import java.util.concurrent.Executors;
     24 
     25 public class UnsafeTest extends TestCase {
     26 
     27     public void test_getUnsafeForbidden() {
     28         try {
     29             Unsafe.getUnsafe();
     30             fail();
     31         } catch (SecurityException expected) {
     32         }
     33     }
     34 
     35     /**
     36      * Regression for 2053217. We used to look one level higher than necessary
     37      * on the stack.
     38      */
     39     public void test_getUnsafeForbiddenWithSystemCaller() throws Exception {
     40         Callable<Object> callable = Executors.callable(new Runnable() {
     41             public void run() {
     42                 Unsafe.getUnsafe();
     43             }
     44         });
     45 
     46         try {
     47             callable.call();
     48             fail();
     49         } catch (SecurityException expected) {
     50         }
     51     }
     52 
     53     private class AllocateInstanceTestClass {
     54         public int i = 123;
     55         public String s = "hello";
     56         public Object getThis() { return AllocateInstanceTestClass.this; }
     57     }
     58 
     59     private static Unsafe getUnsafe() throws Exception {
     60         Class<?> unsafeClass = Class.forName("sun.misc.Unsafe");
     61         Field f = unsafeClass.getDeclaredField("theUnsafe");
     62         f.setAccessible(true);
     63         return (Unsafe) f.get(null);
     64     }
     65 
     66     public void test_allocateInstance() throws Exception {
     67         AllocateInstanceTestClass i = (AllocateInstanceTestClass)
     68                 getUnsafe().allocateInstance(AllocateInstanceTestClass.class);
     69         assertEquals(0, i.i);
     70         assertEquals(null, i.s);
     71         assertEquals(i, i.getThis());
     72     }
     73 }
     74