#ifndef RX_CHECK_H
#define RX_CHECK_H
/*
Hanoh Haim
Cisco Systems, Inc.
*/
/*
Copyright (c) 2015-2015 Cisco Systems, Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
#include "timer_wheel_pq.h"
#include "rx_check_header.h"
#include "time_histogram.h"
#include "utl_jitter.h"
typedef enum {
CLIENT_SIDE = 0,
SERVER_SIDE = 1,
CS_NUM = 2,
CS_INVALID = 255
} pkt_dir_enum_t;
typedef uint8_t pkt_dir_t ;
void flow_aging_callback(CFlowTimerHandle * t);
class CRxCheckFlowPerDir {
public:
CRxCheckFlowPerDir(){
m_flags=0;
m_pkts=0;
m_seq=0;
m_flow_size=0;
}
uint16_t m_flow_size; // how many packets in this direction
uint16_t m_pkts;
uint16_t m_seq;
private:
uint16_t m_flags;
public:
void set_fif_seen(uint16_t flow_per_dir){
m_pkts=1;
m_seq=1;
m_flags |=2;
m_flow_size=flow_per_dir;
}
bool is_fif_seen(){
return ( (m_flags & 2) ==2 ?true:false);
}
void set_init_not_from_first_packet(){
m_flags |=1;
}
bool is_init_not_from_first_pkt(){
return ( (m_flags & 2) ==2 ?true:false);
}
};
class CRxCheckFlow {
public:
CRxCheckFlow(){
m_aging_timer_handle.m_callback =flow_aging_callback;
m_aging_timer_handle.m_object = (void *)this;
m_aging_timer_handle.m_id= 0x1234;
m_oo_err=0;
m_flags=0;
}
public:
/* timestamp of FIF */
uint64_t m_flow_id; /* key*/
CRxCheckFlowPerDir m_dir[CS_NUM];
CFlowTimerHandle m_aging_timer_handle;
uint16_t m_oo_err; /* out of order issue */
uint16_t m_flags;
public:
uint16_t get_total_pkt_seen(void){
return (m_dir[0].m_pkts+
m_dir[1].m_pkts);
}
uint16_t get_total_pkt_expected(void){
return (m_dir[0].m_flow_size+
m_dir[1].m_flow_size);
}
bool is_all_pkts_seen(void){
int i;
int c=0;
for (i=0; i<2; i++) {
if ( (m_dir[i].m_pkts!=m_dir[i].m_flow_size) ){
return (false);
}
if (m_dir[i].m_flow_size>0) {
c++;
}
}
int expc=is_both_dir()?2:1;
if ( expc == c ) {
return ( (m_oo_err==0)?true:false );
}
return (false);
}
void set_both_dir(){
m_flags |=2;
}
bool is_both_dir(){
return ( (m_flags & 2) == 2 ?true:false);
}
void set_aged_correctly(){
m_flags |=4;
}
bool is_aged_correctly(){
return ( (m_flags & 4) ==4 ?true:false);
}
};
class CRxCheckFlowTableStats {
public:
uint64_t m_total_rx;
uint64_t m_total_rx_bytes;
uint64_t m_lookup;
uint64_t m_found;
uint64_t m_fif;
uint64_t m_add;
uint64_t m_remove;
uint64_t m_active;
uint64_t m_err_drop;
uint64_t m_err_aged;
uint64_t m_err_no_magic;
uint64_t m_err_wrong_pkt_id;
uint64_t m_err_fif_seen_twice;
uint64_t m_err_open_with_no_fif_pkt;
uint64_t m_err_oo_dup; /* got same packets id twice expect 1 got 0 */
uint64_t m_err_oo_early; /* miss packet ,expect 1 got 2 */
uint64_t m_err_oo_late; /* early packet ,expect 7 got 6 */
uint64_t m_err_flow_length_changed; /* early packet ,expect 7 got 6 */
uint64_t get_total_err(void){
return (m_err_drop+m_err_aged+
m_err_no_magic+
m_err_wrong_pkt_id+
m_err_fif_seen_twice+
m_err_open_with_no_fif_pkt+
m_err_oo_dup+
m_err_oo_early+
m_err_oo_late+m_err_flow_length_changed);
}
public:
void Clear();
void Dump(FILE *fd);
void dump_json(std::string & json);
};
typedef CRxCheckFlow * rx_check_flow_ptr;
typedef std::map<uint64_t, rx_check_flow_ptr, std::less<uint64_t> > rx_check_flow_map_t;
typedef rx_check_flow_map_t::iterator rx_check_flow_map_iter_t;
class CRxCheckFlowTableMap {
public:
virtual bool Create(int max_size);
virtual void Delete();
virtual bool remove(uint64_t fid );
virtual CRxCheckFlow * lookup(uint64_t fid );
virtual CRxCheckFlow * add(uint64_t fid );
virtual void remove_all(void);
void dump_all(FILE *fd);
uint64_t count(void);
public:
rx_check_flow_map_t m_map;
};
class uint64_tHashEnv
{
public:
static uint32_t Hash(uint64_t x)
{
return ( (x >>40) ^ (x & 0xffffffff));
}
};
template<class T>
class CMyFSA {
public:
bool Create(uint32_t size, bool supportGetNext, bool ctorRequired){
return(true);
}
void Delete(){
}
void Reset(){
}
T * GetNewItem(){
return (new T());
}
void ReturnItem(T *obj){
delete obj;
}
};
#if 0
typedef CHashEntry<uint64_t,CRxCheckFlow> rx_c_hash_ent_t;
typedef CCloseHash<uint64_t,CRxCheckFlow,uint64_tHashEnv,CMyFSA<rx_c_hash_ent_t> > rx_c_hash_t;
class CRxCheckFlowTableHash {
public:
bool Create(int max_size){
return ( m_hash.Create(max_size,0,false,false,true) );
}
void Delete(){
m_hash.Delete();
}
bool remove(uint64_t fid ) {
return(m_hash.Remove(fid)==hsOK?true:false);
}
CRxCheckFlow * lookup(uint64_t fid ){
rx_c_hash_ent_t *lp=m_hash.Find(fid);
if (lp) {
return (&lp->value);
}else{
return ((CRxCheckFlow *)NULL);
}
}
CRxCheckFlow * add(uint64_t fid ){
rx_c_hash_ent_t *lp;
assert(m_hash.Insert(fid,lp)==hsOK);
return (&lp->value);
}
void remove_all(void){
}
void dump_all(FILE *fd){
m_hash.Dump(fd);
}
uint64_t count(void){
return ( m_hash.GetSize());
}
public:
rx_c_hash_t m_hash;
};
#endif
// must be 2^
#define MAX_TEMPLATES_STATS 32
#define MAX_TEMPLATES_STATS_MASK (MAX_TEMPLATES_STATS-1)
class CPerTxthreadTemplateInfo {
public:
CPerTxthreadTemplateInfo(){
Clear();
}
void Clear(){
memset(m_template_info,0,sizeof(m_template_info));
}
void Dump(FILE *fd);
void Add(CPerTxthreadTemplateInfo * obj);
void inc_template(uint8_t index){
if (index<MAX_TEMPLATES_STATS) {
m_template_info[index]++;
}else{
m_template_info[MAX_TEMPLATES_STATS-1]++;
}
}
std::string dump_as_json(std::string name);
uint64_t m_template_info[MAX_TEMPLATES_STATS];
};
class CPerTemplateInfo {
public:
CPerTemplateInfo() {
reset();
}
void reset(){
m_errors=0;
m_rx_pkts=0;
m_jitter.reset();
}
void calc(double dtime){
m_jitter.calc(dtime);
}
uint32_t get_jitter_usec(){
return ((uint32_t)(m_jitter.get_jitter()*1000000.0));
}
void inc_error_counter(void){
m_errors++;
}
uint64_t get_error_counter(){
return (m_errors);
}
void inc_rx_counter(void){
m_rx_pkts++;
}
uint64_t get_rx_counter(){
return (m_rx_pkts);
}
private:
uint64_t m_rx_pkts;
CJitter m_jitter;
uint64_t m_errors;
};
class RxCheckManager {
public:
bool Create();
void Delete();
void handle_packet(CRx_check_header * rxh);
void Dump(FILE *fd);
void DumpShort(FILE *fd);
void DumpTemplate(FILE *fd,bool verbose);
void DumpTemplateFull(FILE *fd);
uint32_t getTemplateMaxJitter();
void template_dump_json(std::string & json);
uint64_t getTotalRx(void){
return ( m_stats.m_total_rx );
}
void tw_drain();
void tw_handle();
void dump_json(std::string & json );
protected:
void update_template_err(uint8_t template_id);
CPerTemplateInfo * get_template(uint8_t index){
uint8_t _id;
if ( index < MAX_TEMPLATES_STATS_MASK ){
_id=index;
}else{
_id=MAX_TEMPLATES_STATS_MASK;
}
return (&m_template_info[_id]);
}
bool on_flow_end(CRxCheckFlow * lp);
friend void flow_aging_callback(CFlowTimerHandle * t);
public:
CTimerWheel m_tw;
CRxCheckFlowTableMap m_ft;
CRxCheckFlowTableStats m_stats;
CTimeHistogram m_hist;
CPerTemplateInfo m_template_info[MAX_TEMPLATES_STATS];
bool m_on_drain;
public:
dsec_t m_cur_time;
};
#endif