WJBET v2.4.39版本的 MD5 值为:a1cc3f24c34b916fdf36d9b31dec8203
以下内容为反编译后的 OkHttpServerTransport.java 源代码,内容仅作参考
package io.grpc.okhttp;
import com.google.android.gms.common.internal.ServiceSpecificExtraArgs;
import com.google.common.base.Preconditions;
import com.google.common.util.concurrent.Futures;
import com.google.common.util.concurrent.ListenableFuture;
import com.lzy.okgo.cookie.SerializableCookie;
import io.grpc.Attributes;
import io.grpc.InternalChannelz;
import io.grpc.InternalLogId;
import io.grpc.InternalStatus;
import io.grpc.Metadata;
import io.grpc.ServerStreamTracer;
import io.grpc.Status;
import io.grpc.internal.GrpcUtil;
import io.grpc.internal.KeepAliveEnforcer;
import io.grpc.internal.KeepAliveManager;
import io.grpc.internal.LogExceptionRunnable;
import io.grpc.internal.MaxConnectionIdleManager;
import io.grpc.internal.ObjectPool;
import io.grpc.internal.SerializingExecutor;
import io.grpc.internal.ServerTransport;
import io.grpc.internal.ServerTransportListener;
import io.grpc.internal.StatsTraceContext;
import io.grpc.internal.TransportTracer;
import io.grpc.okhttp.ExceptionHandlingFrameWriter;
import io.grpc.okhttp.HandshakerSocketFactory;
import io.grpc.okhttp.OkHttpFrameLogger;
import io.grpc.okhttp.OkHttpServerStream;
import io.grpc.okhttp.OkHttpServerTransport;
import io.grpc.okhttp.OutboundFlowController;
import io.grpc.okhttp.internal.framed.ErrorCode;
import io.grpc.okhttp.internal.framed.FrameReader;
import io.grpc.okhttp.internal.framed.HeadersMode;
import io.grpc.okhttp.internal.framed.Http2;
import io.grpc.okhttp.internal.framed.Settings;
import io.grpc.okhttp.internal.framed.Variant;
import java.io.IOException;
import java.io.InputStream;
import java.net.Socket;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.TreeMap;
import java.util.concurrent.Executor;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import java.util.logging.Logger;
import okhttp3.internal.http2.Header;
import okio.Buffer;
import okio.BufferedSource;
import okio.ByteString;
import okio.Okio;
public final class OkHttpServerTransport implements ServerTransport, ExceptionHandlingFrameWriter.TransportExceptionHandler, OutboundFlowController.Transport {
private static final int GRACEFUL_SHUTDOWN_PING = 4369;
private static final int KEEPALIVE_PING = 57005;
private boolean abruptShutdown;
private Attributes attributes;
private final Socket bareSocket;
private final Config config;
private ScheduledFuture<?> forcefulCloseTimer;
private ExceptionHandlingFrameWriter frameWriter;
private Status goAwayStatus;
private boolean gracefulShutdown;
private boolean handshakeShutdown;
private final KeepAliveEnforcer keepAliveEnforcer;
private KeepAliveManager keepAliveManager;
private int lastStreamId;
private ServerTransportListener listener;
private final InternalLogId logId;
private ScheduledFuture<?> maxConnectionAgeMonitor;
private MaxConnectionIdleManager maxConnectionIdleManager;
private OutboundFlowController outboundFlow;
private ScheduledExecutorService scheduledExecutorService;
private ScheduledFuture<?> secondGoawayTimer;
private InternalChannelz.Security securityInfo;
private final TransportTracer tracer;
private Executor transportExecutor;
private static final Logger log = Logger.getLogger(OkHttpServerTransport.class.getName());
private static final ByteString HTTP_METHOD = ByteString.encodeUtf8(Header.TARGET_METHOD_UTF8);
private static final ByteString CONNECT_METHOD = ByteString.encodeUtf8("CONNECT");
private static final ByteString POST_METHOD = ByteString.encodeUtf8("POST");
private static final ByteString SCHEME = ByteString.encodeUtf8(Header.TARGET_SCHEME_UTF8);
private static final ByteString PATH = ByteString.encodeUtf8(Header.TARGET_PATH_UTF8);
private static final ByteString AUTHORITY = ByteString.encodeUtf8(Header.TARGET_AUTHORITY_UTF8);
private static final ByteString CONNECTION = ByteString.encodeUtf8("connection");
private static final ByteString HOST = ByteString.encodeUtf8(SerializableCookie.HOST);
private static final ByteString TE = ByteString.encodeUtf8("te");
private static final ByteString TE_TRAILERS = ByteString.encodeUtf8(GrpcUtil.TE_TRAILERS);
private static final ByteString CONTENT_TYPE = ByteString.encodeUtf8("content-type");
private static final ByteString CONTENT_LENGTH = ByteString.encodeUtf8("content-length");
private final Variant variant = new Http2();
private final Object lock = new Object();
private final Map<Integer, StreamState> streams = new TreeMap();
private int goAwayStreamId = Integer.MAX_VALUE;
public interface StreamState {
OutboundFlowController.StreamState getOutboundFlowState();
boolean hasReceivedEndOfStream();
void inboundDataReceived(Buffer buffer, int i, boolean z);
void inboundRstReceived(Status status);
int inboundWindowAvailable();
void transportReportStatus(Status status);
}
static Logger access$600() {
return log;
}
public OkHttpServerTransport(Config config, Socket socket) {
this.config = (Config) Preconditions.checkNotNull(config, "config");
this.bareSocket = (Socket) Preconditions.checkNotNull(socket, "bareSocket");
TransportTracer create = config.transportTracerFactory.create();
this.tracer = create;
create.setFlowControlWindowReader(new TransportTracer.FlowControlReader() {
@Override
public final TransportTracer.FlowControlWindows read() {
TransportTracer.FlowControlWindows readFlowControlWindow;
readFlowControlWindow = OkHttpServerTransport.this.readFlowControlWindow();
return readFlowControlWindow;
}
});
this.logId = InternalLogId.allocate(getClass(), socket.getRemoteSocketAddress().toString());
this.transportExecutor = config.transportExecutorPool.getObject();
this.scheduledExecutorService = config.scheduledExecutorServicePool.getObject();
this.keepAliveEnforcer = new KeepAliveEnforcer(config.permitKeepAliveWithoutCalls, config.permitKeepAliveTimeInNanos, TimeUnit.NANOSECONDS);
}
public void start(ServerTransportListener serverTransportListener) {
this.listener = (ServerTransportListener) Preconditions.checkNotNull(serverTransportListener, ServiceSpecificExtraArgs.CastExtraArgs.LISTENER);
final SerializingExecutor serializingExecutor = new SerializingExecutor(this.transportExecutor);
serializingExecutor.execute(new Runnable() {
@Override
public final void run() {
OkHttpServerTransport.this.m506lambda$start$0$iogrpcokhttpOkHttpServerTransport(serializingExecutor);
}
});
}
public void m506lambda$start$0$iogrpcokhttpOkHttpServerTransport(SerializingExecutor serializingExecutor) {
try {
this.bareSocket.setTcpNoDelay(true);
HandshakerSocketFactory.HandshakeResult handshake = this.config.handshakerSocketFactory.handshake(this.bareSocket, Attributes.EMPTY);
Socket socket = handshake.socket;
this.attributes = handshake.attributes;
AsyncSink sink = AsyncSink.sink(serializingExecutor, this, 10000);
sink.becomeConnected(Okio.sink(socket), socket);
ForwardingFrameWriter forwardingFrameWriter = new ForwardingFrameWriter(sink.limitControlFramesWriter(this.variant.newWriter(Okio.buffer(sink), false))) {
@Override
public void synReply(boolean z, int i, List<io.grpc.okhttp.internal.framed.Header> list) throws IOException {
OkHttpServerTransport.this.keepAliveEnforcer.resetCounters();
super.synReply(z, i, list);
}
@Override
public void headers(int i, List<io.grpc.okhttp.internal.framed.Header> list) throws IOException {
OkHttpServerTransport.this.keepAliveEnforcer.resetCounters();
super.headers(i, list);
}
@Override
public void data(boolean z, int i, Buffer buffer, int i2) throws IOException {
OkHttpServerTransport.this.keepAliveEnforcer.resetCounters();
super.data(z, i, buffer, i2);
}
};
synchronized (this.lock) {
this.securityInfo = handshake.securityInfo;
ExceptionHandlingFrameWriter exceptionHandlingFrameWriter = new ExceptionHandlingFrameWriter(this, forwardingFrameWriter);
this.frameWriter = exceptionHandlingFrameWriter;
this.outboundFlow = new OutboundFlowController(this, exceptionHandlingFrameWriter);
this.frameWriter.connectionPreface();
Settings settings = new Settings();
OkHttpSettingsUtil.set(settings, 7, this.config.flowControlWindow);
OkHttpSettingsUtil.set(settings, 6, this.config.maxInboundMetadataSize);
this.frameWriter.settings(settings);
if (this.config.flowControlWindow > 65535) {
this.frameWriter.windowUpdate(0, this.config.flowControlWindow - 65535);
}
this.frameWriter.flush();
}
if (this.config.keepAliveTimeNanos != Long.MAX_VALUE) {
KeepAliveManager keepAliveManager = new KeepAliveManager(new KeepAlivePinger(), this.scheduledExecutorService, this.config.keepAliveTimeNanos, this.config.keepAliveTimeoutNanos, true);
this.keepAliveManager = keepAliveManager;
keepAliveManager.onTransportStarted();
}
if (this.config.maxConnectionIdleNanos != Long.MAX_VALUE) {
MaxConnectionIdleManager maxConnectionIdleManager = new MaxConnectionIdleManager(this.config.maxConnectionIdleNanos);
this.maxConnectionIdleManager = maxConnectionIdleManager;
maxConnectionIdleManager.start(new Runnable() {
@Override
public final void run() {
OkHttpServerTransport.this.shutdown();
}
}, this.scheduledExecutorService);
}
if (this.config.maxConnectionAgeInNanos != Long.MAX_VALUE) {
this.maxConnectionAgeMonitor = this.scheduledExecutorService.schedule(new LogExceptionRunnable(new Runnable() {
@Override
public final void run() {
OkHttpServerTransport.this.m507lambda$startIo$1$iogrpcokhttpOkHttpServerTransport();
}
}), (long) (((Math.random() * 0.2d) + 0.9d) * this.config.maxConnectionAgeInNanos), TimeUnit.NANOSECONDS);
}
this.transportExecutor.execute(new FrameHandler(this.variant.newReader(Okio.buffer(Okio.source(socket)), false)));
} catch (IOException | Error | RuntimeException e) {
synchronized (this.lock) {
if (!this.handshakeShutdown) {
log.log(Level.INFO, "Socket failed to handshake", e);
}
}
GrpcUtil.closeQuietly(this.bareSocket);
terminated();
}
}
public void m507lambda$startIo$1$iogrpcokhttpOkHttpServerTransport() {
shutdown(Long.valueOf(this.config.maxConnectionAgeGraceInNanos));
}
@Override
public void shutdown() {
shutdown(Long.valueOf(TimeUnit.SECONDS.toNanos(1L)));
}
private void shutdown(Long l) {
synchronized (this.lock) {
if (!this.gracefulShutdown && !this.abruptShutdown) {
this.gracefulShutdown = true;
if (this.frameWriter == null) {
this.handshakeShutdown = true;
GrpcUtil.closeQuietly(this.bareSocket);
} else {
this.secondGoawayTimer = this.scheduledExecutorService.schedule(new Runnable() {
@Override
public final void run() {
OkHttpServerTransport.this.triggerGracefulSecondGoaway();
}
}, l.longValue(), TimeUnit.NANOSECONDS);
this.frameWriter.goAway(Integer.MAX_VALUE, ErrorCode.NO_ERROR, new byte[0]);
this.frameWriter.ping(false, 0, GRACEFUL_SHUTDOWN_PING);
this.frameWriter.flush();
}
}
}
}
public void triggerGracefulSecondGoaway() {
synchronized (this.lock) {
ScheduledFuture<?> scheduledFuture = this.secondGoawayTimer;
if (scheduledFuture == null) {
return;
}
scheduledFuture.cancel(false);
this.secondGoawayTimer = null;
this.frameWriter.goAway(this.lastStreamId, ErrorCode.NO_ERROR, new byte[0]);
this.goAwayStreamId = this.lastStreamId;
if (this.streams.isEmpty()) {
this.frameWriter.close();
} else {
this.frameWriter.flush();
}
}
}
@Override
public void shutdownNow(Status status) {
synchronized (this.lock) {
if (this.frameWriter == null) {
this.handshakeShutdown = true;
GrpcUtil.closeQuietly(this.bareSocket);
return;
}
abruptShutdown(ErrorCode.NO_ERROR, "", status, true);
}
}
@Override
public void onException(Throwable th) {
Preconditions.checkNotNull(th, "failureCause");
abruptShutdown(ErrorCode.INTERNAL_ERROR, "I/O failure", Status.UNAVAILABLE.withCause(th), false);
}
public void abruptShutdown(ErrorCode errorCode, String str, Status status, boolean z) {
synchronized (this.lock) {
if (this.abruptShutdown) {
return;
}
this.abruptShutdown = true;
this.goAwayStatus = status;
ScheduledFuture<?> scheduledFuture = this.secondGoawayTimer;
if (scheduledFuture != null) {
scheduledFuture.cancel(false);
this.secondGoawayTimer = null;
}
for (Map.Entry<Integer, StreamState> entry : this.streams.entrySet()) {
if (z) {
this.frameWriter.rstStream(entry.getKey().intValue(), ErrorCode.CANCEL);
}
entry.getValue().transportReportStatus(status);
}
this.streams.clear();
this.frameWriter.goAway(this.lastStreamId, errorCode, str.getBytes(GrpcUtil.US_ASCII));
this.goAwayStreamId = this.lastStreamId;
this.frameWriter.close();
this.forcefulCloseTimer = this.scheduledExecutorService.schedule(new Runnable() {
@Override
public final void run() {
OkHttpServerTransport.this.triggerForcefulClose();
}
}, 1L, TimeUnit.SECONDS);
}
}
public void triggerForcefulClose() {
GrpcUtil.closeQuietly(this.bareSocket);
}
public void terminated() {
synchronized (this.lock) {
ScheduledFuture<?> scheduledFuture = this.forcefulCloseTimer;
if (scheduledFuture != null) {
scheduledFuture.cancel(false);
this.forcefulCloseTimer = null;
}
}
KeepAliveManager keepAliveManager = this.keepAliveManager;
if (keepAliveManager != null) {
keepAliveManager.onTransportTermination();
}
MaxConnectionIdleManager maxConnectionIdleManager = this.maxConnectionIdleManager;
if (maxConnectionIdleManager != null) {
maxConnectionIdleManager.onTransportTermination();
}
ScheduledFuture<?> scheduledFuture2 = this.maxConnectionAgeMonitor;
if (scheduledFuture2 != null) {
scheduledFuture2.cancel(false);
}
this.transportExecutor = this.config.transportExecutorPool.returnObject(this.transportExecutor);
this.scheduledExecutorService = this.config.scheduledExecutorServicePool.returnObject(this.scheduledExecutorService);
this.listener.transportTerminated();
}
@Override
public ScheduledExecutorService getScheduledExecutorService() {
return this.scheduledExecutorService;
}
@Override
public ListenableFuture<InternalChannelz.SocketStats> getStats() {
ListenableFuture<InternalChannelz.SocketStats> immediateFuture;
synchronized (this.lock) {
immediateFuture = Futures.immediateFuture(new InternalChannelz.SocketStats(this.tracer.getStats(), this.bareSocket.getLocalSocketAddress(), this.bareSocket.getRemoteSocketAddress(), Utils.getSocketOptions(this.bareSocket), this.securityInfo));
}
return immediateFuture;
}
public TransportTracer.FlowControlWindows readFlowControlWindow() {
TransportTracer.FlowControlWindows flowControlWindows;
synchronized (this.lock) {
OutboundFlowController outboundFlowController = this.outboundFlow;
flowControlWindows = new TransportTracer.FlowControlWindows(outboundFlowController == null ? -1L : outboundFlowController.windowUpdate(null, 0), this.config.flowControlWindow * 0.5f);
}
return flowControlWindows;
}
@Override
public InternalLogId getLogId() {
return this.logId;
}
@Override
public OutboundFlowController.StreamState[] getActiveStreams() {
OutboundFlowController.StreamState[] streamStateArr;
synchronized (this.lock) {
streamStateArr = new OutboundFlowController.StreamState[this.streams.size()];
int i = 0;
for (StreamState streamState : this.streams.values()) {
streamStateArr[i] = streamState.getOutboundFlowState();
i++;
}
}
return streamStateArr;
}
public void streamClosed(int i, boolean z) {
synchronized (this.lock) {
this.streams.remove(Integer.valueOf(i));
if (this.streams.isEmpty()) {
this.keepAliveEnforcer.onTransportIdle();
MaxConnectionIdleManager maxConnectionIdleManager = this.maxConnectionIdleManager;
if (maxConnectionIdleManager != null) {
maxConnectionIdleManager.onTransportIdle();
}
}
if (this.gracefulShutdown && this.streams.isEmpty()) {
this.frameWriter.close();
} else if (z) {
this.frameWriter.flush();
}
}
}
public static String asciiString(ByteString byteString) {
for (int i = 0; i < byteString.size(); i++) {
if (byteString.getByte(i) >= 128) {
return byteString.string(GrpcUtil.US_ASCII);
}
}
return byteString.utf8();
}
public static int headerFind(List<io.grpc.okhttp.internal.framed.Header> list, ByteString byteString, int i) {
while (i < list.size()) {
if (list.get(i).name.equals(byteString)) {
return i;
}
i++;
}
return -1;
}
public static boolean headerContains(List<io.grpc.okhttp.internal.framed.Header> list, ByteString byteString) {
return headerFind(list, byteString, 0) != -1;
}
public static void headerRemove(List<io.grpc.okhttp.internal.framed.Header> list, ByteString byteString) {
int i = 0;
while (true) {
i = headerFind(list, byteString, i);
if (i == -1) {
return;
}
list.remove(i);
}
}
public static ByteString headerGetRequiredSingle(List<io.grpc.okhttp.internal.framed.Header> list, ByteString byteString) {
int headerFind = headerFind(list, byteString, 0);
if (headerFind != -1 && headerFind(list, byteString, headerFind + 1) == -1) {
return list.get(headerFind).value;
}
return null;
}
public static final class Config {
final int flowControlWindow;
final HandshakerSocketFactory handshakerSocketFactory;
final long keepAliveTimeNanos;
final long keepAliveTimeoutNanos;
final long maxConnectionAgeGraceInNanos;
final long maxConnectionAgeInNanos;
final long maxConnectionIdleNanos;
final int maxInboundMessageSize;
final int maxInboundMetadataSize;
final long permitKeepAliveTimeInNanos;
final boolean permitKeepAliveWithoutCalls;
final ObjectPool<ScheduledExecutorService> scheduledExecutorServicePool;
final List<? extends ServerStreamTracer.Factory> streamTracerFactories;
final ObjectPool<Executor> transportExecutorPool;
final TransportTracer.Factory transportTracerFactory;
public Config(OkHttpServerBuilder okHttpServerBuilder, List<? extends ServerStreamTracer.Factory> list) {
this.streamTracerFactories = (List) Preconditions.checkNotNull(list, "streamTracerFactories");
this.transportExecutorPool = (ObjectPool) Preconditions.checkNotNull(okHttpServerBuilder.transportExecutorPool, "transportExecutorPool");
this.scheduledExecutorServicePool = (ObjectPool) Preconditions.checkNotNull(okHttpServerBuilder.scheduledExecutorServicePool, "scheduledExecutorServicePool");
this.transportTracerFactory = (TransportTracer.Factory) Preconditions.checkNotNull(okHttpServerBuilder.transportTracerFactory, "transportTracerFactory");
this.handshakerSocketFactory = (HandshakerSocketFactory) Preconditions.checkNotNull(okHttpServerBuilder.handshakerSocketFactory, "handshakerSocketFactory");
this.keepAliveTimeNanos = okHttpServerBuilder.keepAliveTimeNanos;
this.keepAliveTimeoutNanos = okHttpServerBuilder.keepAliveTimeoutNanos;
this.flowControlWindow = okHttpServerBuilder.flowControlWindow;
this.maxInboundMessageSize = okHttpServerBuilder.maxInboundMessageSize;
this.maxInboundMetadataSize = okHttpServerBuilder.maxInboundMetadataSize;
this.maxConnectionIdleNanos = okHttpServerBuilder.maxConnectionIdleInNanos;
this.permitKeepAliveWithoutCalls = okHttpServerBuilder.permitKeepAliveWithoutCalls;
this.permitKeepAliveTimeInNanos = okHttpServerBuilder.permitKeepAliveTimeInNanos;
this.maxConnectionAgeInNanos = okHttpServerBuilder.maxConnectionAgeInNanos;
this.maxConnectionAgeGraceInNanos = okHttpServerBuilder.maxConnectionAgeGraceInNanos;
}
}
public class FrameHandler implements FrameReader.Handler, Runnable {
private int connectionUnacknowledgedBytesRead;
private final OkHttpFrameLogger frameLogger = new OkHttpFrameLogger(Level.FINE, OkHttpServerTransport.class);
private final FrameReader frameReader;
private boolean receivedSettings;
@Override
public void ackSettings() {
}
@Override
public void alternateService(int i, String str, ByteString byteString, String str2, int i2, long j) {
}
public FrameHandler(FrameReader frameReader) {
this.frameReader = frameReader;
}
@Override
public void run() {
Status status;
InputStream inputStream;
String name = Thread.currentThread().getName();
Thread.currentThread().setName("OkHttpServerTransport");
try {
this.frameReader.readConnectionPreface();
} catch (IOException unused) {
}
try {
if (!this.frameReader.nextFrame(this)) {
connectionError(ErrorCode.INTERNAL_ERROR, "Failed to read initial SETTINGS");
inputStream = OkHttpServerTransport.this.bareSocket.getInputStream();
} else if (this.receivedSettings) {
while (this.frameReader.nextFrame(this)) {
if (OkHttpServerTransport.this.keepAliveManager != null) {
OkHttpServerTransport.this.keepAliveManager.onDataReceived();
}
}
synchronized (OkHttpServerTransport.this.lock) {
status = OkHttpServerTransport.this.goAwayStatus;
}
if (status == null) {
status = Status.UNAVAILABLE.withDescription("TCP connection closed or IOException");
}
OkHttpServerTransport.this.abruptShutdown(ErrorCode.INTERNAL_ERROR, "I/O failure", status, false);
GrpcUtil.exhaust(OkHttpServerTransport.this.bareSocket.getInputStream());
GrpcUtil.closeQuietly(OkHttpServerTransport.this.bareSocket);
OkHttpServerTransport.this.terminated();
Thread.currentThread().setName(name);
return;
} else {
connectionError(ErrorCode.PROTOCOL_ERROR, "First HTTP/2 frame must be SETTINGS. RFC7540 section 3.5");
inputStream = OkHttpServerTransport.this.bareSocket.getInputStream();
}
GrpcUtil.exhaust(inputStream);
} catch (IOException unused2) {
}
GrpcUtil.closeQuietly(OkHttpServerTransport.this.bareSocket);
OkHttpServerTransport.this.terminated();
Thread.currentThread().setName(name);
}
@Override
public void headers(boolean z, boolean z2, int i, int i2, List<io.grpc.okhttp.internal.framed.Header> list, HeadersMode headersMode) {
int headerFind;
this.frameLogger.logHeaders(OkHttpFrameLogger.Direction.INBOUND, i, list, z2);
if ((i & 1) != 0) {
synchronized (OkHttpServerTransport.this.lock) {
if (i > OkHttpServerTransport.this.goAwayStreamId) {
return;
}
boolean z3 = i > OkHttpServerTransport.this.lastStreamId;
if (z3) {
OkHttpServerTransport.this.lastStreamId = i;
}
int headerBlockSize = headerBlockSize(list);
if (headerBlockSize <= OkHttpServerTransport.this.config.maxInboundMetadataSize) {
OkHttpServerTransport.headerRemove(list, ByteString.EMPTY);
String str = null;
ByteString byteString = null;
ByteString byteString2 = null;
ByteString byteString3 = null;
ByteString byteString4 = null;
while (list.size() > 0 && list.get(0).name.getByte(0) == 58) {
io.grpc.okhttp.internal.framed.Header remove = list.remove(0);
if (!OkHttpServerTransport.HTTP_METHOD.equals(remove.name) || byteString != null) {
if (!OkHttpServerTransport.SCHEME.equals(remove.name) || byteString2 != null) {
if (!OkHttpServerTransport.PATH.equals(remove.name) || byteString3 != null) {
if (OkHttpServerTransport.AUTHORITY.equals(remove.name) && byteString4 == null) {
byteString4 = remove.value;
} else {
streamError(i, ErrorCode.PROTOCOL_ERROR, "Unexpected pseudo header. RFC7540 section 8.1.2.1");
return;
}
} else {
byteString3 = remove.value;
}
} else {
byteString2 = remove.value;
}
} else {
byteString = remove.value;
}
}
for (int i3 = 0; i3 < list.size(); i3++) {
if (list.get(i3).name.getByte(0) == 58) {
streamError(i, ErrorCode.PROTOCOL_ERROR, "Pseudo header not before regular headers. RFC7540 section 8.1.2.1");
return;
}
}
if (OkHttpServerTransport.CONNECT_METHOD.equals(byteString) || !z3 || (byteString != null && byteString2 != null && byteString3 != null)) {
if (OkHttpServerTransport.headerContains(list, OkHttpServerTransport.CONNECTION)) {
streamError(i, ErrorCode.PROTOCOL_ERROR, "Connection-specific headers not permitted. RFC7540 section 8.1.2.2");
return;
} else if (z3) {
if (byteString4 == null && (headerFind = OkHttpServerTransport.headerFind(list, OkHttpServerTransport.HOST, 0)) != -1) {
if (OkHttpServerTransport.headerFind(list, OkHttpServerTransport.HOST, headerFind + 1) != -1) {
respondWithHttpError(i, z2, 400, Status.Code.INTERNAL, "Multiple host headers disallowed. RFC7230 section 5.4");
return;
}
byteString4 = list.get(headerFind).value;
}
ByteString byteString5 = byteString4;
OkHttpServerTransport.headerRemove(list, OkHttpServerTransport.HOST);
if (byteString3.size() != 0 && byteString3.getByte(0) == 47) {
String substring = OkHttpServerTransport.asciiString(byteString3).substring(1);
ByteString headerGetRequiredSingle = OkHttpServerTransport.headerGetRequiredSingle(list, OkHttpServerTransport.CONTENT_TYPE);
if (headerGetRequiredSingle != null) {
String asciiString = OkHttpServerTransport.asciiString(headerGetRequiredSingle);
if (GrpcUtil.isGrpcContentType(asciiString)) {
if (OkHttpServerTransport.POST_METHOD.equals(byteString)) {
ByteString headerGetRequiredSingle2 = OkHttpServerTransport.headerGetRequiredSingle(list, OkHttpServerTransport.TE);
if (OkHttpServerTransport.TE_TRAILERS.equals(headerGetRequiredSingle2)) {
OkHttpServerTransport.headerRemove(list, OkHttpServerTransport.CONTENT_LENGTH);
Metadata convertHeaders = Utils.convertHeaders(list);
StatsTraceContext newServerContext = StatsTraceContext.newServerContext(OkHttpServerTransport.this.config.streamTracerFactories, substring, convertHeaders);
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport okHttpServerTransport = OkHttpServerTransport.this;
OkHttpServerStream.TransportState transportState = new OkHttpServerStream.TransportState(okHttpServerTransport, i, okHttpServerTransport.config.maxInboundMessageSize, newServerContext, OkHttpServerTransport.this.lock, OkHttpServerTransport.this.frameWriter, OkHttpServerTransport.this.outboundFlow, OkHttpServerTransport.this.config.flowControlWindow, OkHttpServerTransport.this.tracer, substring);
Attributes attributes = OkHttpServerTransport.this.attributes;
if (byteString5 != null) {
str = OkHttpServerTransport.asciiString(byteString5);
}
OkHttpServerStream okHttpServerStream = new OkHttpServerStream(transportState, attributes, str, newServerContext, OkHttpServerTransport.this.tracer);
if (OkHttpServerTransport.this.streams.isEmpty()) {
OkHttpServerTransport.this.keepAliveEnforcer.onTransportActive();
if (OkHttpServerTransport.this.maxConnectionIdleManager != null) {
OkHttpServerTransport.this.maxConnectionIdleManager.onTransportActive();
}
}
OkHttpServerTransport.this.streams.put(Integer.valueOf(i), transportState);
OkHttpServerTransport.this.listener.streamCreated(okHttpServerStream, substring, convertHeaders);
transportState.onStreamAllocated();
if (z2) {
transportState.inboundDataReceived(new Buffer(), 0, z2);
}
}
return;
}
Status.Code code = Status.Code.INTERNAL;
Object[] objArr = new Object[2];
objArr[0] = OkHttpServerTransport.asciiString(OkHttpServerTransport.TE_TRAILERS);
objArr[1] = headerGetRequiredSingle2 == null ? "<missing>" : OkHttpServerTransport.asciiString(headerGetRequiredSingle2);
respondWithGrpcError(i, z2, code, String.format("Expected header TE: %s, but %s is received. Some intermediate proxy may not support trailers", objArr));
return;
}
respondWithHttpError(i, z2, 405, Status.Code.INTERNAL, "HTTP Method is not supported: " + OkHttpServerTransport.asciiString(byteString));
return;
}
respondWithHttpError(i, z2, 415, Status.Code.INTERNAL, "Content-Type is not supported: " + asciiString);
return;
}
respondWithHttpError(i, z2, 415, Status.Code.INTERNAL, "Content-Type is missing or duplicated");
return;
}
respondWithHttpError(i, z2, 404, Status.Code.UNIMPLEMENTED, "Expected path to start with /: " + OkHttpServerTransport.asciiString(byteString3));
return;
} else {
if (z2) {
synchronized (OkHttpServerTransport.this.lock) {
StreamState streamState = (StreamState) OkHttpServerTransport.this.streams.get(Integer.valueOf(i));
if (streamState == null) {
streamError(i, ErrorCode.STREAM_CLOSED, "Received headers for closed stream");
return;
} else if (streamState.hasReceivedEndOfStream()) {
streamError(i, ErrorCode.STREAM_CLOSED, "Received HEADERS for half-closed (remote) stream. RFC7540 section 5.1");
return;
} else {
streamState.inboundDataReceived(new Buffer(), 0, true);
return;
}
}
}
streamError(i, ErrorCode.PROTOCOL_ERROR, "Headers disallowed in the middle of the stream. RFC7540 section 8.1");
return;
}
}
streamError(i, ErrorCode.PROTOCOL_ERROR, "Missing required pseudo header. RFC7540 section 8.1.2.3");
return;
}
respondWithHttpError(i, z2, 431, Status.Code.RESOURCE_EXHAUSTED, String.format(Locale.US, "Request metadata larger than %d: %d", Integer.valueOf(OkHttpServerTransport.this.config.maxInboundMetadataSize), Integer.valueOf(headerBlockSize)));
return;
}
}
connectionError(ErrorCode.PROTOCOL_ERROR, "Clients cannot open even numbered streams. RFC7540 section 5.1.1");
}
private int headerBlockSize(List<io.grpc.okhttp.internal.framed.Header> list) {
long j = 0;
for (int i = 0; i < list.size(); i++) {
io.grpc.okhttp.internal.framed.Header header = list.get(i);
j += header.name.size() + 32 + header.value.size();
}
return (int) Math.min(j, 2147483647L);
}
@Override
public void data(boolean z, int i, BufferedSource bufferedSource, int i2) throws IOException {
this.frameLogger.logData(OkHttpFrameLogger.Direction.INBOUND, i, bufferedSource.getBuffer(), i2, z);
if (i == 0) {
connectionError(ErrorCode.PROTOCOL_ERROR, "Stream 0 is reserved for control messages. RFC7540 section 5.1.1");
} else if ((i & 1) == 0) {
connectionError(ErrorCode.PROTOCOL_ERROR, "Clients cannot open even numbered streams. RFC7540 section 5.1.1");
} else {
long j = i2;
bufferedSource.require(j);
synchronized (OkHttpServerTransport.this.lock) {
StreamState streamState = (StreamState) OkHttpServerTransport.this.streams.get(Integer.valueOf(i));
if (streamState == null) {
bufferedSource.skip(j);
streamError(i, ErrorCode.STREAM_CLOSED, "Received data for closed stream");
} else if (streamState.hasReceivedEndOfStream()) {
bufferedSource.skip(j);
streamError(i, ErrorCode.STREAM_CLOSED, "Received DATA for half-closed (remote) stream. RFC7540 section 5.1");
} else if (streamState.inboundWindowAvailable() < i2) {
bufferedSource.skip(j);
streamError(i, ErrorCode.FLOW_CONTROL_ERROR, "Received DATA size exceeded window size. RFC7540 section 6.9");
} else {
Buffer buffer = new Buffer();
buffer.write(bufferedSource.getBuffer(), j);
streamState.inboundDataReceived(buffer, i2, z);
int i3 = this.connectionUnacknowledgedBytesRead + i2;
this.connectionUnacknowledgedBytesRead = i3;
if (i3 >= OkHttpServerTransport.this.config.flowControlWindow * 0.5f) {
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport.this.frameWriter.windowUpdate(0, this.connectionUnacknowledgedBytesRead);
OkHttpServerTransport.this.frameWriter.flush();
}
this.connectionUnacknowledgedBytesRead = 0;
}
}
}
}
}
@Override
public void rstStream(int i, ErrorCode errorCode) {
this.frameLogger.logRstStream(OkHttpFrameLogger.Direction.INBOUND, i, errorCode);
if (!ErrorCode.NO_ERROR.equals(errorCode) && !ErrorCode.CANCEL.equals(errorCode) && !ErrorCode.STREAM_CLOSED.equals(errorCode)) {
Logger logger = OkHttpServerTransport.log;
Level level = Level.INFO;
logger.log(level, "Received RST_STREAM: " + errorCode);
}
Status withDescription = GrpcUtil.Http2Error.statusForCode(errorCode.httpCode).withDescription("RST_STREAM");
synchronized (OkHttpServerTransport.this.lock) {
StreamState streamState = (StreamState) OkHttpServerTransport.this.streams.get(Integer.valueOf(i));
if (streamState != null) {
streamState.inboundRstReceived(withDescription);
OkHttpServerTransport.this.streamClosed(i, false);
}
}
}
@Override
public void settings(boolean z, Settings settings) {
this.frameLogger.logSettings(OkHttpFrameLogger.Direction.INBOUND, settings);
synchronized (OkHttpServerTransport.this.lock) {
boolean z2 = false;
if (OkHttpSettingsUtil.isSet(settings, 7)) {
z2 = OkHttpServerTransport.this.outboundFlow.initialOutboundWindowSize(OkHttpSettingsUtil.get(settings, 7));
}
OkHttpServerTransport.this.frameWriter.ackSettings(settings);
OkHttpServerTransport.this.frameWriter.flush();
if (!this.receivedSettings) {
this.receivedSettings = true;
OkHttpServerTransport okHttpServerTransport = OkHttpServerTransport.this;
okHttpServerTransport.attributes = okHttpServerTransport.listener.transportReady(OkHttpServerTransport.this.attributes);
}
if (z2) {
OkHttpServerTransport.this.outboundFlow.writeStreams();
}
}
}
@Override
public void ping(boolean z, int i, int i2) {
if (!OkHttpServerTransport.this.keepAliveEnforcer.pingAcceptable()) {
OkHttpServerTransport.this.abruptShutdown(ErrorCode.ENHANCE_YOUR_CALM, "too_many_pings", Status.RESOURCE_EXHAUSTED.withDescription("Too many pings from client"), false);
return;
}
long j = (i << 32) | (i2 & 4294967295L);
if (!z) {
this.frameLogger.logPing(OkHttpFrameLogger.Direction.INBOUND, j);
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport.this.frameWriter.ping(true, i, i2);
OkHttpServerTransport.this.frameWriter.flush();
}
return;
}
this.frameLogger.logPingAck(OkHttpFrameLogger.Direction.INBOUND, j);
if (57005 == j) {
return;
}
if (4369 == j) {
OkHttpServerTransport.this.triggerGracefulSecondGoaway();
return;
}
Logger logger = OkHttpServerTransport.log;
Level level = Level.INFO;
logger.log(level, "Received unexpected ping ack: " + j);
}
@Override
public void goAway(int i, ErrorCode errorCode, ByteString byteString) {
this.frameLogger.logGoAway(OkHttpFrameLogger.Direction.INBOUND, i, errorCode, byteString);
Status withDescription = GrpcUtil.Http2Error.statusForCode(errorCode.httpCode).withDescription(String.format("Received GOAWAY: %s '%s'", errorCode, byteString.utf8()));
if (!ErrorCode.NO_ERROR.equals(errorCode)) {
OkHttpServerTransport.log.log(Level.WARNING, "Received GOAWAY: {0} {1}", new Object[]{errorCode, byteString.utf8()});
}
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport.this.goAwayStatus = withDescription;
}
}
@Override
public void pushPromise(int i, int i2, List<io.grpc.okhttp.internal.framed.Header> list) throws IOException {
this.frameLogger.logPushPromise(OkHttpFrameLogger.Direction.INBOUND, i, i2, list);
connectionError(ErrorCode.PROTOCOL_ERROR, "PUSH_PROMISE only allowed on peer-initiated streams. RFC7540 section 6.6");
}
@Override
public void windowUpdate(int i, long j) {
this.frameLogger.logWindowsUpdate(OkHttpFrameLogger.Direction.INBOUND, i, j);
synchronized (OkHttpServerTransport.this.lock) {
if (i == 0) {
OkHttpServerTransport.this.outboundFlow.windowUpdate(null, (int) j);
} else {
StreamState streamState = (StreamState) OkHttpServerTransport.this.streams.get(Integer.valueOf(i));
if (streamState != null) {
OkHttpServerTransport.this.outboundFlow.windowUpdate(streamState.getOutboundFlowState(), (int) j);
}
}
}
}
@Override
public void priority(int i, int i2, int i3, boolean z) {
this.frameLogger.logPriority(OkHttpFrameLogger.Direction.INBOUND, i, i2, i3, z);
}
private void connectionError(ErrorCode errorCode, String str) {
OkHttpServerTransport.this.abruptShutdown(errorCode, str, GrpcUtil.Http2Error.statusForCode(errorCode.httpCode).withDescription(String.format("HTTP2 connection error: %s '%s'", errorCode, str)), false);
}
private void streamError(int i, ErrorCode errorCode, String str) {
if (errorCode == ErrorCode.PROTOCOL_ERROR) {
OkHttpServerTransport.log.log(Level.FINE, "Responding with RST_STREAM {0}: {1}", new Object[]{errorCode, str});
}
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport.this.frameWriter.rstStream(i, errorCode);
OkHttpServerTransport.this.frameWriter.flush();
StreamState streamState = (StreamState) OkHttpServerTransport.this.streams.get(Integer.valueOf(i));
if (streamState != null) {
streamState.transportReportStatus(Status.INTERNAL.withDescription(String.format("Responded with RST_STREAM %s: %s", errorCode, str)));
OkHttpServerTransport.this.streamClosed(i, false);
}
}
}
private void respondWithHttpError(int i, boolean z, int i2, Status.Code code, String str) {
Metadata metadata = new Metadata();
metadata.put(InternalStatus.CODE_KEY, code.toStatus());
metadata.put(InternalStatus.MESSAGE_KEY, str);
List<io.grpc.okhttp.internal.framed.Header> createHttpResponseHeaders = Headers.createHttpResponseHeaders(i2, "text/plain; charset=utf-8", metadata);
Buffer writeUtf8 = new Buffer().writeUtf8(str);
synchronized (OkHttpServerTransport.this.lock) {
final Http2ErrorStreamState http2ErrorStreamState = new Http2ErrorStreamState(i, OkHttpServerTransport.this.lock, OkHttpServerTransport.this.outboundFlow, OkHttpServerTransport.this.config.flowControlWindow);
if (OkHttpServerTransport.this.streams.isEmpty()) {
OkHttpServerTransport.this.keepAliveEnforcer.onTransportActive();
if (OkHttpServerTransport.this.maxConnectionIdleManager != null) {
OkHttpServerTransport.this.maxConnectionIdleManager.onTransportActive();
}
}
OkHttpServerTransport.this.streams.put(Integer.valueOf(i), http2ErrorStreamState);
if (z) {
http2ErrorStreamState.inboundDataReceived(new Buffer(), 0, true);
}
OkHttpServerTransport.this.frameWriter.headers(i, createHttpResponseHeaders);
OkHttpServerTransport.this.outboundFlow.data(true, http2ErrorStreamState.getOutboundFlowState(), writeUtf8, true);
OkHttpServerTransport.this.outboundFlow.notifyWhenNoPendingData(http2ErrorStreamState.getOutboundFlowState(), new Runnable() {
@Override
public final void run() {
OkHttpServerTransport.FrameHandler.this.m508x6fc8eaa(http2ErrorStreamState);
}
});
}
}
public void m508x6fc8eaa(Http2ErrorStreamState http2ErrorStreamState) {
synchronized (OkHttpServerTransport.this.lock) {
if (!http2ErrorStreamState.hasReceivedEndOfStream()) {
OkHttpServerTransport.this.frameWriter.rstStream(http2ErrorStreamState.streamId, ErrorCode.NO_ERROR);
}
OkHttpServerTransport.this.streamClosed(http2ErrorStreamState.streamId, true);
}
}
private void respondWithGrpcError(int i, boolean z, Status.Code code, String str) {
Metadata metadata = new Metadata();
metadata.put(InternalStatus.CODE_KEY, code.toStatus());
metadata.put(InternalStatus.MESSAGE_KEY, str);
List<io.grpc.okhttp.internal.framed.Header> createResponseTrailers = Headers.createResponseTrailers(metadata, false);
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport.this.frameWriter.synReply(true, i, createResponseTrailers);
if (!z) {
OkHttpServerTransport.this.frameWriter.rstStream(i, ErrorCode.NO_ERROR);
}
OkHttpServerTransport.this.frameWriter.flush();
}
}
}
public final class KeepAlivePinger implements KeepAliveManager.KeepAlivePinger {
private KeepAlivePinger() {
}
@Override
public void ping() {
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport.this.frameWriter.ping(false, 0, OkHttpServerTransport.KEEPALIVE_PING);
OkHttpServerTransport.this.frameWriter.flush();
}
OkHttpServerTransport.this.tracer.reportKeepAliveSent();
}
@Override
public void onPingTimeout() {
synchronized (OkHttpServerTransport.this.lock) {
OkHttpServerTransport.this.goAwayStatus = Status.UNAVAILABLE.withDescription("Keepalive failed. Considering connection dead");
GrpcUtil.closeQuietly(OkHttpServerTransport.this.bareSocket);
}
}
}
public static class Http2ErrorStreamState implements StreamState, OutboundFlowController.Stream {
private final Object lock;
private final OutboundFlowController.StreamState outboundFlowState;
private boolean receivedEndOfStream;
private final int streamId;
private int window;
@Override
public void inboundRstReceived(Status status) {
}
@Override
public void onSentBytes(int i) {
}
@Override
public void transportReportStatus(Status status) {
}
Http2ErrorStreamState(int i, Object obj, OutboundFlowController outboundFlowController, int i2) {
this.streamId = i;
this.lock = obj;
this.outboundFlowState = outboundFlowController.createState(this, i);
this.window = i2;
}
@Override
public void inboundDataReceived(Buffer buffer, int i, boolean z) {
synchronized (this.lock) {
if (z) {
this.receivedEndOfStream = true;
}
this.window -= i;
try {
buffer.skip(buffer.size());
} catch (IOException e) {
throw new AssertionError(e);
}
}
}
@Override
public boolean hasReceivedEndOfStream() {
boolean z;
synchronized (this.lock) {
z = this.receivedEndOfStream;
}
return z;
}
@Override
public int inboundWindowAvailable() {
int i;
synchronized (this.lock) {
i = this.window;
}
return i;
}
@Override
public OutboundFlowController.StreamState getOutboundFlowState() {
OutboundFlowController.StreamState streamState;
synchronized (this.lock) {
streamState = this.outboundFlowState;
}
return streamState;
}
}
}