forked from btraceio/btrace
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathSocketTracker1.java
More file actions
110 lines (101 loc) · 3.48 KB
/
SocketTracker1.java
File metadata and controls
110 lines (101 loc) · 3.48 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/*
* Copyright (c) 2008, 2015, Oracle and/or its affiliates. All rights reserved.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* This code is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License version 2 only, as
* published by the Free Software Foundation. Oracle designates this
* particular file as subject to the Classpath exception as provided
* by Oracle in the LICENSE file that accompanied this code.
*
* This code is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* version 2 for more details (a copy is included in the LICENSE file that
* accompanied this code).
*
* You should have received a copy of the GNU General Public License version
* 2 along with this work; if not, write to the Free Software Foundation,
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
*
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
* or visit www.oracle.com if you need additional information or have any
* questions.
*/
package com.sun.btrace.samples;
import com.sun.btrace.annotations.*;
import static com.sun.btrace.BTraceUtils.*;
import static com.sun.btrace.BTraceUtils.Strings.*;
import java.net.*;
import com.sun.btrace.AnyType;
/**
* This example tracks all server socket creations
* and client socket accepts. Unlike SockerTracker.java,
* this script uses only public API classes and @OnProbe
* probes - which would be mapped to internal implementation
* classes by a XML descriptor at BTrace agent. For this
* sample, XML probe descriptor is "java.net.socket.xml".
*/
@BTrace public class SocketTracker1 {
@TLS private static int port = -1;
@TLS private static InetAddress inetAddr;
@TLS private static SocketAddress sockAddr;
@OnMethod(
clazz="java.net.ServerSocket",
method="<init>"
)
public static void onServerSocket(@Self ServerSocket self,
int p, int backlog, InetAddress bindAddr) {
port = p;
inetAddr = bindAddr;
}
@OnMethod(
clazz="java.net.ServerSocket",
method="<init>",
type="void (int, int, java.net.InetAddress)",
location=@Location(Kind.RETURN)
)
public static void onSockReturn() {
if (port != -1) {
println("server socket at " + port);
port = -1;
}
if (inetAddr != null) {
println("server socket at " + inetAddr);
inetAddr = null;
}
}
@OnProbe(
namespace="java.net.socket",
name="server-socket-creator"
)
public static void onSocket(@Return ServerSocket ssock) {
println("server socket at " + ssock);
}
@OnProbe(
namespace="java.net.socket",
name="bind"
)
public static void onBind(@Self Object self, SocketAddress addr, int backlog) {
sockAddr = addr;
}
@OnProbe(
namespace="java.net.socket",
name="bind-return"
)
public static void onBindReturn() {
if (sockAddr != null) {
println("server socket bind " + sockAddr);
sockAddr = null;
}
}
@OnProbe(
namespace="java.net.socket",
name="accept-return"
)
public static void onAcceptReturn(AnyType sock) {
if (sock != null) {
println("client socket accept " + sock);
}
}
}