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 import java.io.File;
18 import java.io.IOException;
19 import java.lang.reflect.Method;
20 
21 public class Main {
main(String[] args)22     public static void main(String[] args) throws Exception {
23         String name = System.getProperty("java.vm.name");
24         if (!"Dalvik".equals(name)) {
25             System.out.println("This test is not supported on " + name);
26             return;
27         }
28         testMethodTracing();
29     }
30 
testMethodTracing()31     private static void testMethodTracing() throws Exception {
32         File tempFile;
33         try {
34             tempFile = File.createTempFile("test", ".trace");
35         } catch (IOException e) {
36             System.setProperty("java.io.tmpdir", "/sdcard");
37             tempFile = File.createTempFile("test", ".trace");
38         }
39         tempFile.deleteOnExit();
40         String tempFileName = tempFile.getPath();
41 
42         if (VMDebug.getMethodTracingMode() != 0) {
43             VMDebug.stopMethodTracing();
44         }
45 
46         System.out.println("Confirm enable/disable");
47         System.out.println("status=" + VMDebug.getMethodTracingMode());
48         VMDebug.startMethodTracing(tempFileName, 0, 0, false, 0);
49         System.out.println("status=" + VMDebug.getMethodTracingMode());
50         VMDebug.stopMethodTracing();
51         System.out.println("status=" + VMDebug.getMethodTracingMode());
52         if (tempFile.length() == 0) {
53             System.out.println("ERROR: tracing output file is empty");
54         }
55 
56         System.out.println("Confirm sampling");
57         VMDebug.startMethodTracing(tempFileName, 0, 0, true, 1000);
58         System.out.println("status=" + VMDebug.getMethodTracingMode());
59         VMDebug.stopMethodTracing();
60         System.out.println("status=" + VMDebug.getMethodTracingMode());
61         if (tempFile.length() == 0) {
62             System.out.println("ERROR: sample tracing output file is empty");
63         }
64 
65         System.out.println("Test starting when already started");
66         VMDebug.startMethodTracing(tempFileName, 0, 0, false, 0);
67         System.out.println("status=" + VMDebug.getMethodTracingMode());
68         VMDebug.startMethodTracing(tempFileName, 0, 0, false, 0);
69         System.out.println("status=" + VMDebug.getMethodTracingMode());
70 
71         System.out.println("Test stopping when already stopped");
72         VMDebug.stopMethodTracing();
73         System.out.println("status=" + VMDebug.getMethodTracingMode());
74         VMDebug.stopMethodTracing();
75         System.out.println("status=" + VMDebug.getMethodTracingMode());
76 
77         System.out.println("Test tracing with empty filename");
78         try {
79             VMDebug.startMethodTracing("", 0, 0, false, 0);
80             System.out.println("Should have thrown an exception");
81         } catch (Exception e) {
82             System.out.println("Got expected exception");
83         }
84 
85         System.out.println("Test tracing with bogus (< 1024 && != 0) filesize");
86         try {
87             VMDebug.startMethodTracing(tempFileName, 1000, 0, false, 0);
88             System.out.println("Should have thrown an exception");
89         } catch (Exception e) {
90             System.out.println("Got expected exception");
91         }
92 
93         System.out.println("Test sampling with bogus (<= 0) interval");
94         try {
95             VMDebug.startMethodTracing(tempFileName, 0, 0, true, 0);
96             System.out.println("Should have thrown an exception");
97         } catch (Exception e) {
98             System.out.println("Got expected exception");
99         }
100 
101         tempFile.delete();
102     }
103 
104     private static class VMDebug {
105         private static final Method startMethodTracingMethod;
106         private static final Method stopMethodTracingMethod;
107         private static final Method getMethodTracingModeMethod;
108         static {
109             try {
110                 Class c = Class.forName("dalvik.system.VMDebug");
111                 startMethodTracingMethod = c.getDeclaredMethod("startMethodTracing", String.class,
112                         Integer.TYPE, Integer.TYPE, Boolean.TYPE, Integer.TYPE);
113                 stopMethodTracingMethod = c.getDeclaredMethod("stopMethodTracing");
114                 getMethodTracingModeMethod = c.getDeclaredMethod("getMethodTracingMode");
115             } catch (Exception e) {
116                 throw new RuntimeException(e);
117             }
118         }
119 
startMethodTracing(String filename, int bufferSize, int flags, boolean samplingEnabled, int intervalUs)120         public static void startMethodTracing(String filename, int bufferSize, int flags,
121                 boolean samplingEnabled, int intervalUs) throws Exception {
122             startMethodTracingMethod.invoke(null, filename, bufferSize, flags, samplingEnabled,
123                     intervalUs);
124         }
stopMethodTracing()125         public static void stopMethodTracing() throws Exception {
126             stopMethodTracingMethod.invoke(null);
127         }
getMethodTracingMode()128         public static int getMethodTracingMode() throws Exception {
129             return (int) getMethodTracingModeMethod.invoke(null);
130         }
131     }
132 }
133