Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* The MIT License (MIT)
*
* Copyright (c) 2019 Code Technology Studio
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package org.dromara.jpom.ssh;
import org.apache.hc.core5.http.*;
import org.apache.hc.core5.http.impl.io.HttpRequestExecutor;
import org.apache.hc.core5.http.io.HttpClientConnection;
import org.apache.hc.core5.http.io.HttpResponseInformationCallback;
import org.apache.hc.core5.http.io.entity.AbstractHttpEntity;
import org.apache.hc.core5.http.message.BasicClassicHttpRequest;
import org.apache.hc.core5.http.message.StatusLine;
import org.apache.hc.core5.http.protocol.HttpContext;
import org.apache.hc.core5.http.protocol.HttpCoreContext;
import org.apache.hc.core5.io.Closer;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Objects;
/**
* https://github.com/docker-java/docker-java
*
* @author docker-java-transport-httpclient5
*/
class HijackingHttpRequestExecutor extends HttpRequestExecutor {
static final String HIJACKED_INPUT_ATTRIBUTE = "com.github.docker-java.hijackedInput";
HijackingHttpRequestExecutor(ConnectionReuseStrategy connectionReuseStrategy) {
super(connectionReuseStrategy);
}
@Override
public ClassicHttpResponse execute(
ClassicHttpRequest request,
HttpClientConnection conn,
HttpResponseInformationCallback informationCallback,
HttpContext context
) throws IOException, HttpException {
Objects.requireNonNull(request, "HTTP request");
Objects.requireNonNull(conn, "Client connection");
Objects.requireNonNull(context, "HTTP context");
InputStream hijackedInput = (InputStream) context.getAttribute(HIJACKED_INPUT_ATTRIBUTE);
if (hijackedInput != null) {
return executeHijacked(request, conn, context, hijackedInput);
}
return super.execute(request, conn, informationCallback, context);
}
private ClassicHttpResponse executeHijacked(
ClassicHttpRequest request,
HttpClientConnection conn,
HttpContext context,
InputStream hijackedInput
) throws HttpException, IOException {
try {
context.setAttribute(HttpCoreContext.SSL_SESSION, conn.getSSLSession());
context.setAttribute(HttpCoreContext.CONNECTION_ENDPOINT, conn.getEndpointDetails());
final ProtocolVersion transportVersion = request.getVersion();
if (transportVersion != null) {
context.setProtocolVersion(transportVersion);
}
conn.sendRequestHeader(request);
conn.sendRequestEntity(request);
conn.flush();
ClassicHttpResponse response = conn.receiveResponseHeader();
if (response.getCode() != HttpStatus.SC_SWITCHING_PROTOCOLS) {
conn.terminateRequest(request);
throw new ProtocolException("Expected 101 Switching Protocols, got: " + new StatusLine(response));
}
Thread thread = new Thread(() -> {
try {
BasicClassicHttpRequest fakeRequest = new BasicClassicHttpRequest("POST", "/");
fakeRequest.setHeader(HttpHeaders.CONTENT_LENGTH, Long.MAX_VALUE);
fakeRequest.setEntity(new HijackedEntity(hijackedInput));
conn.sendRequestEntity(fakeRequest);
} catch (Exception e) {
throw new RuntimeException(e);
}
});
thread.setName("docker-java-ssh-httpclient5-hijacking-stream-" + System.identityHashCode(request));
thread.setDaemon(true);
thread.start();
// 101 -> 200
response.setCode(200);
conn.receiveResponseEntity(response);
return response;
} catch (final HttpException | IOException | RuntimeException ex) {
Closer.closeQuietly(conn);
throw ex;
}
}
private static class HijackedEntity extends AbstractHttpEntity {
private final InputStream inStream;
HijackedEntity(InputStream inStream) {
super((String) null, null, false);
this.inStream = inStream;
}
@Override
public void writeTo(OutputStream outStream) throws IOException {
byte[] buffer = new byte[1024];
int read;
while ((read = inStream.read(buffer)) != -1) {
outStream.write(buffer, 0, read);
outStream.flush();
}
}
@Override
public InputStream getContent() {
return inStream;
}
@Override
public boolean isStreaming() {
return true;
}
@Override
public boolean isRepeatable() {
return false;
}
@Override
public void close() throws IOException {
inStream.close();
}
@Override
public long getContentLength() {
return -1;
}
}
}
