1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 
22 /*
23  * Copyright 2006 Sun Microsystems, Inc.  All rights reserved.
24  * Use is subject to license terms.
25  *
26  * ident	"%Z%%M%	%I%	%E% SMI"
27  */
28 
29 import org.opensolaris.os.dtrace.*;
30 
31 /**
32  * Regression for bug 6413280 lookupKernelFunction() and
33  * lookupUserFunction() truncate last character.
34  */
35 public class TestFunctionLookup {
36     static final String kernelLookupProgram = "sdt:::callout-start { " +
37            "@[((callout_t *)arg0)->c_func] = count(); }";
38     static final String userLookupProgram = "pid$target::f2:entry { " +
39            "@[arg0] = count(); }";
40 
41     public static void
42     main(String[] args)
43     {
44 	if (args.length != 1) {
45 	    System.err.println("usage: java TestFunctionLookup <command>");
46 	    System.exit(1);
47 	}
48 	String cmd = args[0];
49 
50 	Consumer consumer = new LocalConsumer();
51 	try {
52 	    consumer.open();
53 	    consumer.compile(kernelLookupProgram);
54 	    consumer.enable();
55 	    consumer.go();
56 	    Aggregate a;
57 	    Number address;
58 	    String f;
59 	    boolean done = false;
60 	    for (int i = 0; (i < 20) && !done; ++i) {
61 		Thread.currentThread().sleep(100);
62 		a = consumer.getAggregate();
63 		for (Aggregation agg : a.getAggregations()) {
64 		    for (Tuple tuple : agg.asMap().keySet()) {
65 			address = (Number)tuple.get(0).getValue();
66 			if (address instanceof Integer) {
67 			    int addr = (Integer)address;
68 			    f = consumer.lookupKernelFunction(addr);
69 			} else {
70 			    long addr = (Long)address;
71 			    f = consumer.lookupKernelFunction(addr);
72 			}
73 			if (f.equals("genunix`setrun")) {
74 			    System.out.println(f);
75 			    done = true;
76 			}
77 		    }
78 		}
79 	    }
80 	    consumer.close();
81 	} catch (Exception e) {
82 	    e.printStackTrace();
83 	    System.exit(1);
84 	}
85 
86 	consumer = new LocalConsumer();
87 	try {
88 	    consumer.open();
89 	    int pid = consumer.createProcess(cmd);
90 	    consumer.compile(userLookupProgram);
91 	    consumer.enable();
92 	    consumer.go();
93 	    Thread.currentThread().sleep(500);
94 	    Aggregate a = consumer.getAggregate();
95 	    Number address;
96 	    String f;
97 	    for (Aggregation agg : a.getAggregations()) {
98 		for (Tuple tuple : agg.asMap().keySet()) {
99 		    address = (Number)tuple.get(0).getValue();
100 		    if (address instanceof Integer) {
101 			int addr = (Integer)address;
102 			f = consumer.lookupUserFunction(pid, addr);
103 		    } else {
104 			long addr = (Long)address;
105 			f = consumer.lookupUserFunction(pid, addr);
106 		    }
107 		    System.out.println(f);
108 		}
109 	    }
110 	    consumer.close();
111 	} catch (Exception e) {
112 	    e.printStackTrace();
113 	    System.exit(1);
114 	}
115     }
116 }
117