mirror of
https://github.com/revanced/jadb.git
synced 2024-06-10 07:07:49 +02:00
76 lines
2.3 KiB
Java
76 lines
2.3 KiB
Java
package se.vidstige.jadb;
|
|
|
|
import java.io.*;
|
|
import java.net.Socket;
|
|
import java.nio.charset.StandardCharsets;
|
|
|
|
class Transport implements Closeable {
|
|
|
|
private final OutputStream outputStream;
|
|
private final InputStream inputStream;
|
|
private final DataInputStream dataInput;
|
|
private final DataOutputStream dataOutput;
|
|
|
|
private Transport(OutputStream outputStream, InputStream inputStream) {
|
|
this.outputStream = outputStream;
|
|
this.inputStream = inputStream;
|
|
this.dataInput = new DataInputStream(inputStream);
|
|
this.dataOutput = new DataOutputStream(outputStream);
|
|
}
|
|
|
|
public Transport(Socket socket) throws IOException {
|
|
this(socket.getOutputStream(), socket.getInputStream());
|
|
}
|
|
|
|
public String readString() throws IOException {
|
|
String encodedLength = readString(4);
|
|
int length = Integer.parseInt(encodedLength, 16);
|
|
return readString(length);
|
|
}
|
|
|
|
public void readResponseTo(OutputStream output) throws IOException {
|
|
Stream.copy(inputStream, output);
|
|
}
|
|
|
|
public InputStream getInputStream() {
|
|
return inputStream;
|
|
}
|
|
|
|
public void verifyResponse() throws IOException, JadbException {
|
|
String response = readString(4);
|
|
if (!"OKAY".equals(response)) {
|
|
String error = readString();
|
|
throw new JadbException("command failed: " + error);
|
|
}
|
|
}
|
|
|
|
public String readString(int length) throws IOException {
|
|
byte[] responseBuffer = new byte[length];
|
|
dataInput.readFully(responseBuffer);
|
|
return new String(responseBuffer, StandardCharsets.UTF_8);
|
|
}
|
|
|
|
private String getCommandLength(String command) {
|
|
return String.format("%04x", command.getBytes().length);
|
|
}
|
|
|
|
public void send(String command) throws IOException {
|
|
OutputStreamWriter writer = new OutputStreamWriter(outputStream, StandardCharsets.UTF_8);
|
|
writer.write(getCommandLength(command));
|
|
writer.write(command);
|
|
writer.flush();
|
|
}
|
|
|
|
public SyncTransport startSync() throws IOException, JadbException {
|
|
send("sync:");
|
|
verifyResponse();
|
|
return new SyncTransport(dataOutput, dataInput);
|
|
}
|
|
|
|
@Override
|
|
public void close() throws IOException {
|
|
dataInput.close();
|
|
dataOutput.close();
|
|
}
|
|
}
|