mirror of
https://github.com/wangyu-/UDPspeeder.git
synced 2025-01-18 22:09:35 +08:00
153 lines
3.9 KiB
C
153 lines
3.9 KiB
C
/*
|
|
* connection.h
|
|
*
|
|
* Created on: Sep 23, 2017
|
|
* Author: root
|
|
*/
|
|
|
|
#ifndef CONNECTION_H_
|
|
#define CONNECTION_H_
|
|
|
|
extern int disable_anti_replay;
|
|
|
|
#include "connection.h"
|
|
#include "common.h"
|
|
#include "log.h"
|
|
#include "delay_manager.h"
|
|
#include "fd_manager.h"
|
|
#include "fec_manager.h"
|
|
|
|
extern int report_interval;
|
|
|
|
struct conv_manager_t // manage the udp connections
|
|
{
|
|
//typedef hash_map map;
|
|
unordered_map<u64_t,u32_t> u64_to_conv; //conv and u64 are both supposed to be uniq
|
|
unordered_map<u32_t,u64_t> conv_to_u64;
|
|
unordered_map<u32_t,u64_t> conv_last_active_time;
|
|
|
|
unordered_map<u32_t,u64_t>::iterator clear_it;
|
|
|
|
//void (*clear_function)(uint64_t u64) ;
|
|
|
|
long long last_clear_time;
|
|
|
|
conv_manager_t();
|
|
conv_manager_t(const conv_manager_t &b)
|
|
{
|
|
assert(0==1);
|
|
}
|
|
~conv_manager_t();
|
|
int get_size();
|
|
void reserve();
|
|
void clear();
|
|
u32_t get_new_conv();
|
|
int is_conv_used(u32_t conv);
|
|
int is_u64_used(u64_t u64);
|
|
u32_t find_conv_by_u64(u64_t u64);
|
|
u64_t find_u64_by_conv(u32_t conv);
|
|
int update_active_time(u32_t conv);
|
|
int insert_conv(u32_t conv,u64_t u64);
|
|
int erase_conv(u32_t conv);
|
|
int clear_inactive(char * ip_port=0);
|
|
int clear_inactive0(char * ip_port);
|
|
};
|
|
|
|
|
|
struct inner_stat_t
|
|
{
|
|
u64_t input_packet_num;
|
|
u64_t input_packet_size;
|
|
u64_t output_packet_num;
|
|
u64_t output_packet_size;
|
|
};
|
|
struct stat_t
|
|
{
|
|
u64_t last_report_time;
|
|
inner_stat_t normal_to_fec;
|
|
inner_stat_t fec_to_normal;
|
|
stat_t()
|
|
{
|
|
memset(this,0,sizeof(stat_t));
|
|
}
|
|
void report_as_client()
|
|
{
|
|
if(report_interval!=0 &&get_current_time()-last_report_time>u64_t(report_interval)*1000)
|
|
{
|
|
last_report_time=get_current_time();
|
|
inner_stat_t &a=normal_to_fec;
|
|
inner_stat_t &b=fec_to_normal;
|
|
mylog(log_info,"[report]client-->server:(original:%llu pkt;%llu byte) (fec:%llu pkt,%llu byte) server-->client:(original:%llu pkt;%llu byte) (fec:%llu pkt;%llu byte)\n",
|
|
a.input_packet_num,a.input_packet_size,a.output_packet_num,a.output_packet_size,
|
|
b.output_packet_num,b.output_packet_size,b.input_packet_num,b.input_packet_size
|
|
);
|
|
}
|
|
}
|
|
void report_as_server(ip_port_t &ip_port)
|
|
{
|
|
if(report_interval!=0 &&get_current_time()-last_report_time>u64_t(report_interval)*1000)
|
|
{
|
|
last_report_time=get_current_time();
|
|
inner_stat_t &a=fec_to_normal;
|
|
inner_stat_t &b=normal_to_fec;
|
|
mylog(log_info,"[report][%s]client-->server:(original:%llu pkt;%llu byte) (fec:%llu pkt;%llu byte) server-->client:(original:%llu pkt;%llu byte) (fec:%llu pkt;%llu byte)\n",
|
|
ip_port.to_s(),
|
|
a.output_packet_num,a.output_packet_size,a.input_packet_num,a.input_packet_size,
|
|
b.input_packet_num,b.input_packet_size,b.output_packet_num,b.output_packet_size
|
|
);
|
|
}
|
|
}
|
|
};
|
|
|
|
|
|
struct conn_info_t //stores info for a raw connection.for client ,there is only one connection,for server there can be thousand of connection since server can
|
|
//handle multiple clients
|
|
{
|
|
conv_manager_t conv_manager;
|
|
fec_encode_manager_t fec_encode_manager;
|
|
fec_decode_manager_t fec_decode_manager;
|
|
my_timer_t timer;
|
|
//ip_port_t ip_port;
|
|
u64_t last_active_time;
|
|
stat_t stat;
|
|
conn_info_t()
|
|
{
|
|
}
|
|
void update_active_time()
|
|
{
|
|
last_active_time=get_current_time();
|
|
}
|
|
conn_info_t(const conn_info_t &b)
|
|
{
|
|
assert(0==1);
|
|
}
|
|
};
|
|
|
|
struct conn_manager_t //manager for connections. for client,we dont need conn_manager since there is only one connection.for server we use one conn_manager for all connections
|
|
{
|
|
|
|
unordered_map<u64_t,conn_info_t*> mp;//<ip,port> to conn_info_t;
|
|
unordered_map<u64_t,conn_info_t*>::iterator clear_it;
|
|
long long last_clear_time;
|
|
|
|
conn_manager_t();
|
|
conn_manager_t(const conn_info_t &b)
|
|
{
|
|
assert(0==1);
|
|
}
|
|
int exist(ip_port_t);
|
|
conn_info_t *& find_p(ip_port_t); //be aware,the adress may change after rehash
|
|
conn_info_t & find(ip_port_t) ; //be aware,the adress may change after rehash
|
|
int insert(ip_port_t);
|
|
|
|
int erase(unordered_map<u64_t,conn_info_t*>::iterator erase_it);
|
|
int clear_inactive();
|
|
int clear_inactive0();
|
|
|
|
};
|
|
|
|
extern conn_manager_t conn_manager;
|
|
|
|
|
|
#endif /* CONNECTION_H_ */
|