/*
* 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);
}
}
}