void removeMapRule(String name)
{
MapRule mapRule = getMapRule(name);
if (mapRule != null)
{
mapList.Remove(mapRule);
mapRuleTable.Remove(mapRule.listen_port);
if (mapRule.serverSocket != null)
{
try
{
mapRule.serverSocket.Close();
}
catch (Exception e)
{
MLog.info(e.Message);
}
}
try
{
saveMapRule();
}
catch (Exception e)
{
MLog.info(e.Message);
}
}
}
void addMapRule(MapRule mapRule)
{
if (getMapRule(mapRule.name) != null)
{
throw new Exception("映射 " + mapRule.name + " 已存在,请修改名称!");
}
Socket serverSocket = null;
try
{
serverSocket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);//serverSocket = new Socket(mapRule.getListen_port());c#不这样创建socket
serverSocket.Connect(IPAddress.Parse("127.0.0.1"), mapRule.getListen_port());
listen(serverSocket);
mapList.Add(mapRule);
mapRuleTable.Add(mapRule.listen_port, mapRule);
saveMapRule();
}
catch (Exception e2)
{
//e2.printStackTrace();
throw new Exception("端口 " + mapRule.listen_port + " 已经被占用!");
}
finally
{
// if(serverSocket!=null){
// serverSocket.close();
// }
}
}
MapRule getMapRule(string name)
{
MapRule rule = null;
foreach (MapRule r in mapList)
{
if (r.getName().Equals(name))
{
rule = r;
break;
}
}
return(rule);
}
void updateMapRule(MapRule mapRule_origin, MapRule mapRule_new)
{
if (getMapRule(mapRule_new.name) != null && !mapRule_origin.name.Equals(mapRule_new.name))
{
throw new Exception("映射 " + mapRule_new.name + " 已存在,请修改名称!");
}
Socket serverSocket = null;
if (mapRule_origin.listen_port != mapRule_new.listen_port)
{
try
{
//serverSocket = new ServerSocket(mapRule_new.getListen_port());
serverSocket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
serverSocket.Connect(IPAddress.Parse("127.0.0.1"), mapRule_new.getListen_port());
listen(serverSocket);
mapRule_origin.isUsing = false;
if (mapRule_origin.serverSocket != null)
{
mapRule_origin.serverSocket.Close();
}
mapRule_origin.serverSocket = serverSocket;
mapRuleTable.Remove(mapRule_origin.listen_port);
mapRuleTable.Add(mapRule_new.listen_port, mapRule_new);
}
catch (Exception e2)
{
//e2.printStackTrace();
throw new Exception("端口 " + mapRule_new.getListen_port() + " 已经被占用!");
}
finally
{
// if(serverSocket!=null){
// serverSocket.close();
// }
}
}
mapRule_origin.name = mapRule_new.name;
mapRule_origin.listen_port = mapRule_new.listen_port;
mapRule_origin.dst_port = mapRule_new.dst_port;
saveMapRule();
}
void listen(Socket serverSocket)
{
Route.es.execute(() =>
{
while (true)
{
try
{
Socket socket = serverSocket.Accept();
Route.es.execute(() =>
{
int listenPort = (serverSocket.LocalEndPoint as IPEndPoint).Port;
MapRule mapRule = mapRuleTable[listenPort];
if (mapRule != null)
{
Route route = null;
if (mapClient.isUseTcp())
{
route = mapClient.route_tcp;
}
else
{
route = mapClient.route_udp;
}
PortMapProcess process = new PortMapProcess(mapClient, route, socket, mapClient.serverAddress, mapClient.serverPort, null,
null, mapRule.dst_port);
}
});
}
catch (Exception e)
{
MLog.info(e.Message);
break;
}
}
});
}
void loadMapRule()
{
//string content;
//JSONObject json=null;
//try {
// content = readFileUtf8(configFilePath);
// json=JSONObject.parseObject(content);
//} catch (Exception e) {
// //e.printStackTrace();
//}
//if(json!=null&&json.containsKey("map_list")){
// JSONArray json_map_list=json.getJSONArray("map_list");
// for(int i=0;i<json_map_list.size();i++){
// JSONObject json_rule=(JSONObject) json_map_list.get(i);
// MapRule mapRule=new MapRule();
// mapRule.name=json_rule.getString("name");
// mapRule.listen_port=json_rule.getIntValue("listen_port");
// mapRule.dst_port=json_rule.getIntValue("dst_port");
// mapList.add(mapRule);
// ServerSocket serverSocket;
// try {
// serverSocket = new ServerSocket(mapRule.getListen_port());
// listen(serverSocket);
// mapRule.serverSocket=serverSocket;
// } catch (IOException e) {
// mapRule.using=true;
// e.printStackTrace();
// }
// mapRuleTable.Add(mapRule.listen_port, mapRule);
// }
//}
///用JSON类改写
string content;
MapList maplistObject = null;// 用来辅助转换json的临时类
try
{
content = readFileUtf8(configFilePath);
maplistObject = JSON.parse <MapList>(content);
}
catch (Exception e)
{
MLog.info(e.Message);
}
if (maplistObject != null && maplistObject.map_list != null)
{
foreach (var mapPair in maplistObject.map_list)//这里显得繁琐了，以后要是能改造MapRule就能简单好多
{
MapRule mapRule = new MapRule();
mapRule.dst_port = mapPair.dst_port;
mapRule.listen_port = mapPair.listen_port;
mapRule.name = mapPair.name;
mapList.Add(mapRule);
Socket serverSocket;
try
{
serverSocket = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
serverSocket.Connect(IPAddress.Parse("127.0.0.1"), mapRule.getListen_port());
listen(serverSocket);
mapRule.serverSocket = serverSocket;
}
catch (Exception e)
{
mapRule.isUsing = true;
MLog.info(e.Message);
}
mapRuleTable.Add(mapRule.listen_port, mapRule);
}
}
}