forked from KNEMC/KLALB
106 lines
2.9 KiB
Java
106 lines
2.9 KiB
Java
package org.kne.cloud.network.klalb;
|
|
|
|
import java.io.DataInputStream;
|
|
import java.io.IOException;
|
|
import java.net.ConnectException;
|
|
import java.net.Socket;
|
|
import java.util.Collection;
|
|
import java.util.Collections;
|
|
import java.util.HashMap;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.UUID;
|
|
import java.util.WeakHashMap;
|
|
|
|
public class KLALBServer {
|
|
WeakHashMap<UUID, IOThreadManager> whm=new WeakHashMap<>();
|
|
public KLALBServer(int port, Map<String,ServiceElement> services,List<Tunnel> tunnels) throws IOException {
|
|
TCPListener tcpl=new TCPListener(port);
|
|
tcpl.setCon((s)->{
|
|
try {
|
|
//s.setSoTimeout(10000);
|
|
TCPConnection tcc=new TCPConnection(null,s);
|
|
DataInputStream din=tcc.getDin();
|
|
int val=din.readShort()&0xffff;
|
|
if(val!=59649) {
|
|
return;
|
|
}
|
|
int x=din.read();
|
|
if(x==0) {
|
|
synchronized (services) {
|
|
tcc.getDout().writeInt(services.size());
|
|
Collection<ServiceElement> cll=services.values();
|
|
for (Iterator<ServiceElement> iterator = cll.iterator(); iterator.hasNext();) {
|
|
ServiceElement tunnel = iterator.next();
|
|
tcc.getDout().writeUTF(tunnel.toString());
|
|
}
|
|
}
|
|
synchronized (tunnels) {
|
|
int counts=tunnels.size();
|
|
tcc.getDout().writeInt(counts);
|
|
for (int i = 0; i < counts; i++) {
|
|
tcc.getDout().writeUTF(tunnels.get(i).toString());
|
|
}
|
|
}
|
|
tcc.getDout().flush();
|
|
return;
|
|
}
|
|
|
|
String lip=din.readUTF();
|
|
int lport=din.readInt();
|
|
String lname=din.readUTF();
|
|
ServiceElement eas=services.get(lname);
|
|
System.out.println(eas);
|
|
|
|
if(eas==null||(!eas.ipport.equals(new IPPort(lip, lport)))){
|
|
return;
|
|
}
|
|
|
|
String name = din.readUTF();
|
|
String ip = din.readUTF();
|
|
int portx=din.readInt();
|
|
Tunnel tll=Tunnel.newTunnel(name, ip, portx);
|
|
tcc.setTunnel(tll);
|
|
|
|
UUID uid=new UUID(din.readLong(),din.readLong());
|
|
|
|
System.out.println("隧道连接:"+tll);
|
|
IOThreadManager nx = null;
|
|
if(whm.containsKey(uid)) {
|
|
nx=whm.get(uid);
|
|
}else {
|
|
nx=new IOThreadManager();
|
|
Socket soc=new Socket("192.168.1.233",eas.ipport.getPort());
|
|
nx.setLocal(new TCPConnection(null, soc));
|
|
nx.startLocal();
|
|
whm.put(uid, nx);
|
|
|
|
}
|
|
nx.handleSocket(tcc);
|
|
int n=nx.getTcps().size();
|
|
if(n<=0) {
|
|
nx.closeLocal();
|
|
System.out.println("连接已关闭");
|
|
}
|
|
}catch(ConnectException e) {
|
|
System.out.println("连接本地服务失败，请检查你的服务程序");
|
|
}catch(IOException e) {
|
|
e.printStackTrace();
|
|
}finally {
|
|
try {
|
|
s.close();
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
});
|
|
tcpl.open();
|
|
|
|
}
|
|
|
|
/*public static void main(String[] args) throws IOException {
|
|
KLALBServer kc=new KLALBServer(4569);
|
|
}*/
|
|
}
|