mirror of
https://git.huckle.dev/Huckles-Minecraft-Archive/jpexs-decompiler.git
synced 2026-08-02 18:21:25 +00:00
307 lines
9.3 KiB
Java
307 lines
9.3 KiB
Java
/*
|
|
* Copyright (C) 2010-2015 JPEXS
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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 for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package com.jpexs.decompiler.flash.gui.debugger;
|
|
|
|
import com.jpexs.helpers.utf8.Utf8Helper;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.EOFException;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStream;
|
|
import java.net.InetAddress;
|
|
import java.net.ServerSocket;
|
|
import java.net.Socket;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.WeakHashMap;
|
|
import java.util.regex.Pattern;
|
|
|
|
/**
|
|
*
|
|
* @author JPEXS
|
|
*/
|
|
public class Debugger {
|
|
|
|
private static final Set<DebugListener> listeners = new HashSet<>();
|
|
|
|
public synchronized void addMessageListener(DebugListener l) {
|
|
listeners.add(l);
|
|
}
|
|
|
|
public synchronized void removeMessageListener(DebugListener l) {
|
|
listeners.remove(l);
|
|
}
|
|
|
|
private static class DebugHandler extends Thread {
|
|
|
|
private final Socket s;
|
|
|
|
private final int serverPort;
|
|
|
|
private static int maxid = 0;
|
|
|
|
private final int id;
|
|
|
|
public boolean finished = false;
|
|
private final Map<String, String> parameters = new HashMap<>();
|
|
|
|
public static final int MSG_STRING = 0;
|
|
public static final int MSG_LOADER_URL = 1;
|
|
public static final int MSG_LOADER_BYTES = 2;
|
|
|
|
public String getParameter(String name, String defValue) {
|
|
if (parameters.containsKey(name)) {
|
|
return parameters.get(name);
|
|
}
|
|
return defValue;
|
|
}
|
|
|
|
public int getVersionMajor() {
|
|
return Integer.parseInt(getParameter("debug.version.major", "1"));
|
|
}
|
|
|
|
public int getVersionMinor() {
|
|
return Integer.parseInt(getParameter("debug.version.major", "0"));
|
|
}
|
|
|
|
public boolean hasMsgType() {
|
|
return getVersionMajor() > 1 || getVersionMinor() > 0;
|
|
}
|
|
|
|
public DebugHandler(int serverPort, Socket s) {
|
|
this.s = s;
|
|
id = maxid++;
|
|
this.serverPort = serverPort;
|
|
}
|
|
|
|
public void cancel() {
|
|
try {
|
|
s.close();
|
|
} catch (IOException ex) {
|
|
//ignore
|
|
}
|
|
}
|
|
|
|
private int readType(InputStream is) throws IOException {
|
|
int type = is.read();
|
|
if (type == -1) {
|
|
throw new EOFException();
|
|
}
|
|
return type;
|
|
}
|
|
|
|
private byte[] readBytes(InputStream is) throws IOException {
|
|
int len = is.read();
|
|
if (len == -1) {
|
|
throw new EOFException();
|
|
}
|
|
int len2 = is.read();
|
|
if (len2 == -1) {
|
|
throw new EOFException();
|
|
}
|
|
int len3 = is.read();
|
|
if (len3 == -1) {
|
|
throw new EOFException();
|
|
}
|
|
int len4 = is.read();
|
|
if (len4 == -1) {
|
|
throw new EOFException();
|
|
}
|
|
len = (len << 24) + (len2 << 16) + (len3 << 8) + len4;
|
|
byte data[] = new byte[len];
|
|
int cnt;
|
|
int off = 0;
|
|
while (len > 0 && (cnt = is.read(data, off, len)) > 0) {
|
|
len -= cnt;
|
|
off += cnt;
|
|
}
|
|
return data;
|
|
}
|
|
|
|
private String readString(InputStream is) throws IOException {
|
|
int len = is.read();
|
|
if (len == -1) {
|
|
throw new EOFException();
|
|
}
|
|
int len2 = is.read();
|
|
if (len2 == -1) {
|
|
throw new EOFException();
|
|
}
|
|
len = (len << 8) + len2;
|
|
|
|
byte buf[] = new byte[len];
|
|
for (int i = 0; i < len; i++) {
|
|
int rd = is.read();
|
|
if (rd == -1) {
|
|
throw new EOFException();
|
|
}
|
|
buf[i] = (byte) rd;
|
|
}
|
|
return new String(buf, Utf8Helper.charset);
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
String clientName = Integer.toString(id);
|
|
try (InputStream is = s.getInputStream()) {
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
|
|
int c;
|
|
do {
|
|
c = is.read();
|
|
if (c != 0) {
|
|
baos.write(c);
|
|
}
|
|
|
|
} while (c > 0);
|
|
String ret = baos.toString("UTF-8");
|
|
|
|
if (ret.equals("<policy-file-request/>")) {
|
|
try (OutputStream os = s.getOutputStream()) {
|
|
os.write(("<cross-domain-policy><allow-access-from domain=\"*\" to-ports=\"" + serverPort + "\" /></cross-domain-policy>").getBytes("UTF-8"));
|
|
}
|
|
} else {
|
|
if (!ret.isEmpty()) {
|
|
String param[] = (ret.contains(";") ? ret.split(";") : new String[]{ret});
|
|
for (String p : param) {
|
|
if (p.contains("=")) {
|
|
String key = p.substring(0, p.indexOf("="));
|
|
String val = p.substring(p.indexOf("=") + 1);
|
|
parameters.put(key, val);
|
|
} else {
|
|
parameters.put(p, "true");
|
|
}
|
|
}
|
|
}
|
|
boolean hasType = hasMsgType();
|
|
String name = readString(is);
|
|
if (!name.isEmpty()) {
|
|
clientName = name;
|
|
}
|
|
while (true) {
|
|
int type = 0;
|
|
if (hasType) {
|
|
type = readType(is);
|
|
}
|
|
switch (type) {
|
|
case MSG_STRING:
|
|
ret = readString(is);
|
|
for (DebugListener l : listeners) {
|
|
l.onMessage(clientName, ret);
|
|
}
|
|
break;
|
|
case MSG_LOADER_URL:
|
|
ret = readString(is);
|
|
for (DebugListener l : listeners) {
|
|
l.onLoaderURL(clientName, ret);
|
|
}
|
|
break;
|
|
case MSG_LOADER_BYTES:
|
|
byte retB[] = readBytes(is);
|
|
for (DebugListener l : listeners) {
|
|
l.onLoaderBytes(clientName, retB);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
} catch (IOException ex) {
|
|
//ignore
|
|
}
|
|
try {
|
|
s.close();
|
|
} catch (IOException ex) {
|
|
//ignore
|
|
}
|
|
finished = true;
|
|
for (DebugListener l : listeners) {
|
|
l.onFinish(clientName);
|
|
}
|
|
}
|
|
}
|
|
|
|
private static class DebugServerThread extends Thread {
|
|
|
|
private final int port;
|
|
|
|
private ServerSocket ss;
|
|
|
|
private final Map<Integer, DebugHandler> handlers = new WeakHashMap<>();
|
|
|
|
public DebugServerThread(int port) {
|
|
this.port = port;
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
try {
|
|
ss = new ServerSocket(port, 50, InetAddress.getByName("localhost"));
|
|
ss.setReuseAddress(true);
|
|
while (true) {
|
|
Socket s = ss.accept();
|
|
DebugHandler h = new DebugHandler(port, s);
|
|
handlers.put(h.id, h);
|
|
h.start();
|
|
}
|
|
} catch (IOException ex) {
|
|
//ignore
|
|
}
|
|
}
|
|
}
|
|
|
|
private final int port;
|
|
|
|
public Debugger(int port) {
|
|
this.port = port;
|
|
}
|
|
|
|
private DebugServerThread server = null;
|
|
|
|
public synchronized void start() {
|
|
if (server == null) {
|
|
server = new DebugServerThread(port);
|
|
server.start();
|
|
}
|
|
}
|
|
|
|
public synchronized boolean isRunning() {
|
|
return server != null;
|
|
}
|
|
|
|
public int getPort() {
|
|
return port;
|
|
}
|
|
|
|
public synchronized void stop() {
|
|
if (server != null) {
|
|
try {
|
|
server.ss.close();
|
|
} catch (IOException ex) {
|
|
//ignore
|
|
}
|
|
for (DebugHandler h : server.handlers.values()) {
|
|
h.cancel();
|
|
}
|
|
server.handlers.clear();
|
|
server = null;
|
|
}
|
|
}
|
|
}
|