unix 中udp应用增强可靠性

#ifndef __unp_rtt_h
#define __unp_rtt_h
#include "unix.h"


struct rtt_info{
	float rtt_rtt;
	float rtt_srtt;
	float rtt_rttvar;
	float rtt_rto;
	float rtt_nrexmt;
	uint32_t rtt_base;
};
#define RTT_RXTMIN 2
#define RTT_RXTMAX 6
#define RTT_MAXNREXMT 3


void rtt_debug(struct rtt_info*);
void rtt_init(struct rtt_info*);
void rtt_newpack(struct rtt_info*);
int  rtt_start(struct rtt_info*);
void rtt_stop(struct rtt_info*,uint32_t);
int  rtt_timeout(struct rtt_info*);
uint32_t rtt_ts(struct rtt_info*);

extern int rtt_d_flag;
#endif 

------------------------------------------

#include "unp.h"
#include <time.h>


int rtt_d_flag = 0;

#define RTT_RTOCALC(ptr) ((ptr)->rtt_srtt +(4.0 * (ptr)->rtt_rttvar))


static float 
rtt_minmax(float rto)
{
	if(rto < RTT_RXTMIN)
		rto = RTT_RXTMIN;
	else if(rto > RTT_RXTMAX)
		rto = RTT_RXTMAX;
	return rto;
}
void rtt_init(struct rtt_info *ptr)
{
	struct timeval tv;
	::gettimeofday(&tv,NULL);
	ptr->rtt_base   = tv.tv_sec;
	ptr->rtt_srtt   = 0;
	ptr->rtt_rtt    = 0;
	ptr->rtt_rttvar = 0.75;
	ptr->rtt_rto    = rtt_minmax(RTT_RTOCALC(ptr));
}
uint32_t 
rtt_ts(struct rtt_info* ptr)
{
	uint32_t ts;
	struct timeval tv;
	gettimeofday(&tv,NULL);
	ts = ((tv.tv_sec - ptr->rtt_base) * 1000) + (tv.tv_usec/1000);
	return ts;
}
void 
rtt_newpack(struct rtt_info *ptr)
{
	ptr->rtt_nrexmt = 0;
}
int 
rtt_start(struct rtt_info *ptr)
{
	return (int)(ptr->rtt_rto + 0.5);
}
void 
rtt_stop(struct rtt_info *ptr,uint32_t ms)
{
	double delta;
	ptr->rtt_rtt = ms/1000.0;

	delta = ptr->rtt_rtt - ptr->rtt_srtt;
	ptr->rtt_srtt += delta/8;

	if(delta < 0.0)
		delta = -delta;

	ptr->rtt_rttvar += (delta - ptr->rtt_rttvar) / 4;
	ptr->rtt_rto= rtt_minmax(RTT_RTOCALC(ptr));
}
int 
rtt_timeout(struct rtt_info *ptr)
{
	ptr->rtt_rto *= 2;
	if(++ptr->rtt_nrexmt > RTT_MAXNREXMT)
		return -1;
	return 0;
}


---------------------------------------



#include "unp.h"
#include "unix.h"

#define RTT_DEBUG

static struct rtt_info rttinfo;
static int rttinit = 0;
static struct msghdr msgsend,msgrecv;
static struct hdr{
	uint32_t seq;
	uint32_t ts;
}sendhdr,recvhdr;


static void sig_alrm(int signo);
static sigjmp_buf jmpbuf;
ssize_t
dg_send_recv(int fd,void* outbuff,size_t outbytes,void *inbuff,size_t inbytes,
			 struct sockaddr* destaddr,socklen_t destlen)
{
	ssize_t n ;
	struct iovec iovsend[2],iovrecv[2];

	if(rttinit == 0){
		rtt_init(&rttinfo);
		rttinit = 1;
		rtt_d_flag = 1;
	}

	sendhdr.seq++;
	msgsend.msg_name = destaddr;
	msgsend.msg_namelen = destlen;
	msgsend.msg_iov = iovsend;
	msgsend.msg_iovlen = 2;
	iovsend[0].iov_base = &sendhdr;
	iovsend[0].iov_len  = sizeof(sendhdr);
	iovsend[1].iov_base = outbuff;
	iovsend[1].iov_len  = outbytes;


	msgrecv.msg_name = NULL;
	msgrecv.msg_namelen = 0;
	msgrecv.msg_iov = iovrecv;
	msgrecv.msg_iovlen = 2;
	iovrecv[0].iov_base = &recvhdr;
	iovrecv[0].iov_len  = sizeof(recvhdr);
	iovrecv[1].iov_base = inbuff;
	iovrecv[1].iov_len  = inbytes;


	signal(SIGALRM,sig_alrm);
	rtt_newpack(&rttinfo);

sendagain:
	sendhdr.ts = rtt_ts(&rttinfo);
	sendmsg(fd,&msgsend,0);

	alarm(rtt_start(&rttinfo));
	if(sigsetjmp(jmpbuf,1) != 0){
		if(rtt_timeout(&rttinfo) < 0){
			memset(&rttinfo,0,sizeof(rttinfo));
			errno = ETIMEDOUT;
			return -1;
		}
		goto sendagain;
	}
	do{
		n = recvmsg(fd,&msgrecv,0);
	}while(n < sizeof(recvhdr) || recvhdr.seq != sendhdr.seq);
	alarm(0);

	rtt_stop(&rttinfo,rtt_ts(&rttinfo) - recvhdr.ts);

	return n - sizeof(struct hdr);
}
static void
sig_alrm(int signo)
{
	siglongjmp(jmpbuf,1);
}
#include <iostream>
using namespace std;
int main()
{
	int fd = socket(AF_INET,SOCK_DGRAM,0);
	struct sockaddr_in addr;
	addr.sin_family = AF_INET;
	addr.sin_addr.s_addr = inet_addr("192.168.4.155");
	addr.sin_port = htons(5000);
	char outbuff[64] = {0};
	strcpy(outbuff,"Hello world,Hello China!");
	int len = strlen(outbuff);
	cout << "Test3 Client...." << endl;
	char inbuff[64] = {0};
	do{
		int ret = dg_send_recv(fd,outbuff,len,inbuff,63,(struct sockaddr*)&addr,sizeof(addr));
		if(ret < 0){
			cout << "dg_send_recv error[" << errno << "]" << endl;
			continue;
		}
		cout << "recv:" << inbuff << endl;
		sleep(4);
	}while(true);
	return 0;
};