// Copyright (c) 2015 D1SM.net
package net.fs.rudp;
import java.io.IOException;
import java.net.DatagramPacket;
import java.net.InetAddress;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Random;
import net.fs.rudp.message.PingMessage;
import net.fs.rudp.message.PingMessage2;
import net.fs.utils.ByteIntConvert;
import net.fs.utils.MLog;
import net.fs.utils.MessageCheck;
public class ClientControl {
int clientId;
Thread sendThread;
Object synlock=new Object();
private HashMap<Integer, SendRecord> sendRecordTable=new HashMap<Integer, SendRecord>();
HashMap<Integer, SendRecord> sendRecordTable_remote=new HashMap<Integer, SendRecord>();
long startSendTime=0;
int maxSpeed=(int) (1024*1024);
int initSpeed=(int) maxSpeed;
int currentSpeed=initSpeed;
int lastTime=-1;
Object syn_timeid=new Object();
long sended=0;
long markTime=0;
long lastSendPingTime,lastReceivePingTime=System.currentTimeMillis();
Random ran=new Random();
HashMap<Integer, Long> pingTable=new HashMap<Integer, Long>();
public int pingDelay=250;
int clientId_real=-1;
long needSleep_All,trueSleep_All;
int maxAcked=0;
long lastLockTime;
Route route;
InetAddress dstIp;
int dstPort;
public HashMap<Integer, ConnectionUDP> connTable=new HashMap<Integer, ConnectionUDP>();
Object syn_connTable=new Object();
Object syn_tunTable=new Object();
String password;
public ResendManage resendMange;
boolean closed=false;
{
resendMange = new ResendManage();
}
ClientControl(Route route,int clientId,InetAddress dstIp,int dstPort){
this.clientId=clientId;
this.route=route;
this.dstIp=dstIp;
this.dstPort=dstPort;
}
public void onReceivePacket(DatagramPacket dp){
byte[] dpData=dp.getData();
int sType=0;
sType=MessageCheck.checkSType(dp);
int remote_clientId=ByteIntConvert.toInt(dpData, 8);
if(sType==net.fs.rudp.message.MessageType.sType_PingMessage){
PingMessage pm=new PingMessage(dp);
sendPingMessage2(pm.getPingId(),dp.getAddress(),dp.getPort());
currentSpeed=pm.getDownloadSpeed()*1024;
}else if(sType==net.fs.rudp.message.MessageType.sType_PingMessage2){
PingMessage2 pm=new PingMessage2(dp);
lastReceivePingTime=System.currentTimeMillis();
Long t=pingTable.get(pm.getPingId());
if(t!=null){
pingDelay=(int) (System.currentTimeMillis()-t);
String protocal="";
if(route.isUseTcpTun()){
protocal="tcp";
}else {
protocal="udp";
}
//MLog.println(" receive_ping222: "+pm.getPingId()+" "+new Date());
MLog.println("delay_"+protocal+" "+pingDelay+"ms "+dp.getAddress().getHostAddress()+":"+dp.getPort());
}
}
}
public void sendPacket(DatagramPacket dp) throws IOException{
//加密
route.sendPacket(dp);
}
void addConnection(ConnectionUDP conn){
synchronized (syn_connTable) {
connTable.put(conn.connectId, conn);
}
}
void removeConnection(ConnectionUDP conn){
synchronized (syn_connTable) {
connTable.remove(conn.connectId);
}
}
public void close(){
closed=true;
route.clientManager.removeClient(clientId);
synchronized (syn_connTable) {
Iterator<Integer> it=getConnTableIterator();
while(it.hasNext()){
final ConnectionUDP conn=connTable.get(it.next());
if(conn!=null){
Route.es.execute(new Runnable() {
@Override
public void run() {
conn.stopnow=true;
conn.destroy(true);
}
});
}
}
}
}
Iterator<Integer> getConnTableIterator(){
Iterator<Integer> it=null;
synchronized (syn_connTable) {
it=new CopiedIterator(connTable.keySet().iterator());
}
return it;
}
public void updateClientId(int newClientId){
clientId_real=newClientId;
sendRecordTable.clear();
sendRecordTable_remote.clear();
}
public void onSendDataPacket(ConnectionUDP conn){
}
public void sendPingMessage(){
int pingid=Math.abs(ran.nextInt());
long pingTime=System.currentTimeMillis();
pingTable.put(pingid, pingTime);
lastSendPingTime=System.currentTimeMillis();
PingMessage lm=new PingMessage(0,route.localclientId,pingid,Route.localDownloadSpeed,Route.localUploadSpeed);
lm.setDstAddress(dstIp);
lm.setDstPort(dstPort);
try {
sendPacket(lm.getDatagramPacket());
} catch (IOException e) {
//e.printStackTrace();
}
}
public void sendPingMessage2(int pingId,InetAddress dstIp,int dstPort){
PingMessage2 lm=new PingMessage2(0,route.localclientId,pingId);
lm.setDstAddress(dstIp);
lm.setDstPort(dstPort);
try {
sendPacket(lm.getDatagramPacket());
} catch (IOException e) {
e.printStackTrace();
}
}
public void onReceivePing(PingMessage pm){
if(route.mode==2){
currentSpeed=pm.getDownloadSpeed()*1024;
//#MLog.println("更新对方速度: "+currentSpeed);
}
}
SendRecord getSendRecord(int timeId){
SendRecord record=null;
synchronized (syn_timeid) {
record=sendRecordTable.get(timeId);
if(record==null){
record=new SendRecord();
record.setTimeId(timeId);
sendRecordTable.put(timeId, record);
}
}
return record;
}
public int getCurrentTimeId(){
long current=System.currentTimeMillis();
if(startSendTime==0){
startSendTime=current;
}
int timeId=(int) ((current-startSendTime)/1000);
return timeId;
}
public int getTimeId(long time){
int timeId=(int) ((time-startSendTime)/1000);
return timeId;
}
//纳秒
public synchronized void sendSleep(long startTime,int length){
if(route.mode==1){
currentSpeed=Route.localUploadSpeed;
}
if(sended==0){
markTime=startTime;
}
sended+=length;
//10K sleep
if(sended>10*1024){
long needTime=(long) (1000*1000*1000f*sended/currentSpeed);
long usedTime=System.nanoTime()-markTime;
if(usedTime<needTime){
long sleepTime=needTime-usedTime;
needSleep_All+=sleepTime;
long moreTime=trueSleep_All-needSleep_All;
if(moreTime>0){
if(sleepTime<=moreTime){
sleepTime=0;
trueSleep_All-=sleepTime;
}
}
long s=needTime/(1000*1000);
int n=(int) (needTime%(1000*1000));
long t1=System.nanoTime();
if(sleepTime>0){
try {
Thread.sleep(s, n);
} catch (InterruptedException e) {
e.printStackTrace();
}
trueSleep_All+=(System.nanoTime()-t1);
//#MLog.println("sssssssssss "+(trueSleep_All-needSleep_All)/(1000*1000));
}
////#MLog.println("sleepb "+sleepTime+" l "+sended+" s "+s+" n "+n+" tt "+(moreTime));
}
sended=0;
}
}
public Object getSynlock() {
return synlock;
}
public void setSynlock(Object synlock) {
this.synlock = synlock;
}
public void setClientId(int clientId) {
this.clientId = clientId;
}
public int getClientId_real() {
return clientId_real;
}
public void setClientId_real(int clientId_real) {
this.clientId_real = clientId_real;
lastReceivePingTime=System.currentTimeMillis();
}
public long getLastSendPingTime() {
return lastSendPingTime;
}
public void setLastSendPingTime(long lastSendPingTime) {
this.lastSendPingTime = lastSendPingTime;
}
public long getLastReceivePingTime() {
return lastReceivePingTime;
}
public void setLastReceivePingTime(long lastReceivePingTime) {
this.lastReceivePingTime = lastReceivePingTime;
}
public String getPassword() {
return password;
}
public void setPassword(String password) {
this.password = password;
}
}
