src/jdk.jfr/share/classes/jdk/jfr/internal/instrument/SocketInputStreamInstrumentor.java
author alanb
Sat, 16 Mar 2019 19:44:12 +0000
changeset 54155 b5a73f22b2bd
parent 50113 caf115bb98ad
child 57268 adcdd45830a0
permissions -rw-r--r--
8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl Reviewed-by: chegar Contributed-by: alan.bateman@oracle.com, michael.x.mcmahon@oracle.com
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     1
/*
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
     2
 * Copyright (c) 2013, 2019, Oracle and/or its affiliates. All rights reserved.
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     3
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     4
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     5
 * This code is free software; you can redistribute it and/or modify it
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     6
 * under the terms of the GNU General Public License version 2 only, as
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     7
 * published by the Free Software Foundation.  Oracle designates this
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     8
 * particular file as subject to the "Classpath" exception as provided
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
     9
 * by Oracle in the LICENSE file that accompanied this code.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    10
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    11
 * This code is distributed in the hope that it will be useful, but WITHOUT
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    12
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    13
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    14
 * version 2 for more details (a copy is included in the LICENSE file that
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    15
 * accompanied this code).
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    16
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    17
 * You should have received a copy of the GNU General Public License version
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    18
 * 2 along with this work; if not, write to the Free Software Foundation,
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    19
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    20
 *
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    21
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    22
 * or visit www.oracle.com if you need additional information or have any
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    23
 * questions.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    24
 */
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    25
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    26
package jdk.jfr.internal.instrument;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    27
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    28
import java.io.IOException;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    29
import java.net.InetAddress;
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    30
import java.net.Socket;
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    31
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    32
import jdk.jfr.events.SocketReadEvent;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    33
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    34
/**
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    35
 * See {@link JITracer} for an explanation of this code.
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    36
 */
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    37
@JIInstrumentationTarget("java.net.Socket$SocketInputStream")
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    38
final class SocketInputStreamInstrumentor {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    39
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    40
    private SocketInputStreamInstrumentor() {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    41
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    42
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    43
    @SuppressWarnings("deprecation")
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    44
    @JIInstrumentationMethod
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    45
    public int read(byte b[], int off, int length) throws IOException {
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    46
        SocketReadEvent event = SocketReadEvent.EVENT.get();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    47
        if (!event.isEnabled()) {
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    48
            return read(b, off, length);
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    49
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    50
        int bytesRead = 0;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    51
        try {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    52
            event.begin();
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    53
            bytesRead = read(b, off, length);
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    54
        } finally {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    55
            event.end();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    56
            if (event.shouldCommit()) {
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    57
                InetAddress remote = parent.getInetAddress();
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    58
                event.host = remote.getHostName();
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    59
                event.address = remote.getHostAddress();
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    60
                event.port = parent.getPort();
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    61
                if (bytesRead < 0) {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    62
                    event.endOfStream = true;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    63
                } else {
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    64
                    event.bytesRead = bytesRead;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    65
                }
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    66
                event.timeout = parent.getSoTimeout();
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    67
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    68
                event.commit();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    69
                event.reset();
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    70
            }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    71
        }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    72
        return bytesRead;
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    73
    }
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    74
54155
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    75
    // private field in java.net.Socket$SocketInputStream
b5a73f22b2bd 8220493: Prepare Socket/ServerSocket for alternative platform SocketImpl
alanb
parents: 50113
diff changeset
    76
    private Socket parent;
50113
caf115bb98ad 8199712: Flight Recorder
egahlin
parents:
diff changeset
    77
}