/*
* wfaTGConfig: store the traffic profile setting that will be used to
* instruct traffic generation.
* input: cmd -- not used
* response: send success back to controller
* return: success or fail
* Note: the profile storage is a global space.
*/
int wfaTGConfig(int len, BYTE *caCmdBuf, int *respLen, BYTE *respBuf)
{
int ret = FALSE;
tgStream_t *myStream = NULL;
dutCmdResponse_t *confResp = &gGenericResp;
/* if the stream table over maximum, reset it */
if(slotCnt == WFA_MAX_TRAFFIC_STREAMS)
slotCnt = 0;
if(slotCnt == 0)
{
printf("resetting stream table\n");
wMEMSET(gStreams, 0, WFA_MAX_TRAFFIC_STREAMS*sizeof(tgStream_t));
}
DPRINT_INFO(WFA_OUT, "entering tcConfig ...\n");
myStream = &gStreams[slotCnt++];
wMEMSET(myStream, 0, sizeof(tgStream_t));
wMEMCPY(&myStream->profile, caCmdBuf, len);
myStream->id = ++streamId; /* the id start from 1 */
myStream->tblidx = slotCnt-1;
#if 0
DPRINT_INFO(WFA_OUT, "profile %i direction %i dest ip %s dport %i source %s sport %i rate %i duration %i size %i class %i delay %i\n", myStream->profile.profile, myStream->profile.direction, myStream->profile.dipaddr, myStream->profile.dport, myStream->profile.sipaddr, myStream->profile.sport, myStream->profile.rate, myStream->profile.duration, myStream->profile.pksize, myStream->profile.trafficClass, myStream->profile.startdelay);
#endif
confResp->status = STATUS_COMPLETE;
confResp->streamId = myStream->id;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_CONFIG_RESP_TLV, sizeof(dutCmdResponse_t), (BYTE *)confResp, respBuf);
*respLen = WFA_TLV_HDR_LEN + sizeof(dutCmdResponse_t);
return ret;
}
int wfaConnectUDPPeer(int mysock, char *daddr, int dport)
{
struct sockaddr_in peerAddr;
wMEMSET(&peerAddr, 0, sizeof(peerAddr));
peerAddr.sin_family = AF_INET;
inet_aton(daddr, &peerAddr.sin_addr);
peerAddr.sin_port = htons(dport);
wCONNECT(mysock, (struct sockaddr *)&peerAddr, sizeof(peerAddr));
return mysock;
}
/*
* wfaSetProcPriority():
* With the linux 2.6 kernel, it allows an application process dynamically
* adjust its running priority level. In order to achieve higher control of
* packet sending/receiving and timer response, it is helpful to raise the
* process priority level over others and lower it back once it finishes.
*
* This is purely used for performance tuning purpose and not required to
* port if it is not needed.
*/
int wfaSetProcPriority(int set)
{
int maxprio, currprio;
struct sched_param schp;
wMEMSET(&schp, 0, sizeof(schp));
sched_getparam(0, &schp);
currprio = schp.sched_priority;
if(set != 0)
{
if(geteuid() == 0)
{
maxprio = sched_get_priority_max(SCHED_FIFO);
if(maxprio == -1)
{
return WFA_FAILURE;
}
schp.sched_priority = maxprio;
if(sched_setscheduler(0, SCHED_FIFO, &schp) != 0)
{
}
}
if(mlockall(MCL_CURRENT | MCL_FUTURE) != 0)
{
}
}
else
{
if(geteuid() == 0)
{
schp.sched_priority = 0;
if(sched_setscheduler(0, SCHED_OTHER, &schp) != 0)
{
}
}
munlockall();
}
return currprio;
}
/*
* wfaCreateTCPServSock(): initially create a TCP socket
* intput: port -- TCP socket port to listen
* return: socket id;
*/
int wfaCreateTCPServSock(unsigned short port)
{
int sock; /* socket to create */
struct sockaddr_in servAddr; /* Local address */
const int on = 1;
/* Create socket for incoming connections */
if ((sock = wSOCKET(PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0)
{
DPRINT_ERR(WFA_ERR, "createTCPServSock socket() failed");
return WFA_FAILURE;
}
/* Construct local address structure */
wMEMSET(&servAddr, 0, sizeof(servAddr));
wfaGetifAddr(gnetIf, &servAddr);
servAddr.sin_family = AF_INET; /* Internet address family */
//servAddr.sin_addr.s_addr = htonl(INADDR_ANY); /* Any incoming interface */
servAddr.sin_port = htons(port); /* Local port */
wSETSOCKOPT(sock, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
/* Bind to the local address */
if (wBIND(sock, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0)
{
DPRINT_ERR(WFA_ERR, "bind() failed");
return WFA_FAILURE;
}
/* Mark the socket so it will listen for incoming connections */
if (wLISTEN(sock, MAXPENDING) < 0)
{
DPRINT_ERR(WFA_ERR, "listen() failed");
return WFA_FAILURE;
}
return sock;
}
void * wfa_wmm_thread(void *thr_param)
{
int myId = ((tgThrData_t *)thr_param)->tid;
tgWMM_t *my_wmm = &wmm_thr[myId];
tgStream_t *myStream = NULL;
int myStreamId, i=0,rttime=0,difftime=0, rcvCount=0,sendCount=0;
int mySock = -1, status, respLen = 0, nbytes = 0, ret=0, j=0;
tgProfile_t *myProfile;
pthread_attr_t tattr;
#ifdef WFA_WMM_PS_EXT
tgThrData_t *tdata =(tgThrData_t *) thr_param;
StationProcStatetbl_t curr_state;
#endif
//#ifdef WFA_VOICE_EXT
struct timeval lstime, lrtime;
int asn = 1; /* everytime it starts from 1, and to ++ */
//#endif
wPT_ATTR_INIT(&tattr);
wPT_ATTR_SETSCH(&tattr, SCHED_RR);
while(1)
{
int sleepTotal=0,sendFailCount=0;
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread::begin while loop for each send/rcv before mutex lock\n");
pthread_mutex_lock(&my_wmm->thr_flag_mutex);
/* it needs to reset the thr_flag to wait again */
while(my_wmm->thr_flag == 0)
{
printf("Mutex wait\n");
/*
* in normal situation, the signal indicates the thr_flag changed to
* a valid number (stream id), then it will go out the loop and do
* work.
*/
wPT_COND_WAIT(&my_wmm->thr_flag_cond, &my_wmm->thr_flag_mutex);
}
wPT_MUTEX_UNLOCK(&my_wmm->thr_flag_mutex);
myStreamId = my_wmm->thr_flag;
my_wmm->thr_flag=0;
/* use the flag as a stream id to file the profile */
myStream = findStreamProfile(myStreamId);
myProfile = &myStream->profile;
if(myProfile == NULL)
{
status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_SEND_RESP_TLV, 4, (BYTE *)&status, respBuf);
respLen = WFA_TLV_HDR_LEN+4;
/*
* send it back to control agent.
*/
continue;
}
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread::Mutex unlocked\n");
switch(myProfile->direction)
{
case DIRECT_SEND:
mySock = wfaCreateUDPSock(myProfile->sipaddr, myProfile->sport);
if (mySock < 0)
{
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread SEND ERROR failed create UDP socket! \n");
break;
}
mySock = wfaConnectUDPPeer(mySock, myProfile->dipaddr, myProfile->dport);
sendThrId = myId;
/*
* Set packet/socket priority TOS field
*/
wfaTGSetPrio(mySock, myProfile->trafficClass);
/*
* set a proper priority
*/
wfaSetThreadPrio(myId, myProfile->trafficClass);
/* if delay is too long, it must be something wrong */
if(myProfile->startdelay > 0 && myProfile->startdelay<100)
{
wSLEEP(myProfile->startdelay);
}
/*
* set timer fire up
*/
if(myProfile->maxcnt == 0)
{
wSIGNAL(SIGALRM, tmout_stop_send);
wALARM(myProfile->duration );
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread SEND set stop alarm for %d sec \n", myProfile->duration);
}
if(myProfile->profile == PROF_MCAST)
{
wfaSetSockMcastSendOpt(mySock);
}
if (myProfile->profile == PROF_IPTV || myProfile->profile == PROF_FILE_TX || myProfile->profile == PROF_MCAST)
{
int iOptVal, iOptLen;
getsockopt(mySock, SOL_SOCKET, SO_SNDBUF, (char *)&iOptVal, (socklen_t *)&iOptLen);
iOptVal = iOptVal * 16;
setsockopt(mySock, SOL_SOCKET, SO_SNDBUF, (char *)&iOptVal, (socklen_t )iOptLen);
if ( (myProfile->rate != 0 ) /* WFA_SEND_FIX_BITRATE_MAX_FRAME_RATE)*/ &&
(myProfile->pksize * myProfile->rate * 8 < WFA_SEND_FIX_BITRATE_MAX) )
wfaSendBitrateData(mySock, myStreamId, respBuf, &respLen);
else
{
wfaSendLongFile(mySock, myStreamId, respBuf, &respLen);
}
/* wfaSendLongFile(mySock, myStreamId, respBuf, &respLen); */
if(mySock != -1)
{
wCLOSE(mySock);
mySock = -1;
}
}
else if(myProfile->profile == PROF_TRANSC || myProfile->profile == PROF_START_SYNC || myProfile->profile == PROF_CALI_RTD)
{
#if 0
struct timeval nxtime, curtime;
int ioflags = wFCNTL(mySock, F_GETFL, 0);
#endif
struct timeval tmout;
gtgTransac = myStreamId;
sentTranPkts = 0;
#if 0
gettimeofday(&nxtime, NULL);
nxtime.tv_usec += 20000; /* fixed 20 min sec timeout */
if(nxtime.tv_usec >= 1000000)
{
nxtime.tv_sec +=1;
nxtime.tv_usec -= 1000000;
}
wFCNTL(mySock, F_SETFL, ioflags | O_NONBLOCK);
#endif
gettimeofday(&lstime,0);
DPRINT_INFO(WFA_OUT, "Start sending traffic,at sec %d usec %d\n", (int )lstime.tv_sec, (int)lstime.tv_usec);
tmout.tv_sec = 0;
tmout.tv_usec = 15000; // set for 15000 microsec timeout for rcv
ret = setsockopt(mySock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tmout, (socklen_t) sizeof(tmout));
rcvCount=0; sendFailCount=0;
j=0; sendCount=0;
sleepTotal = 0;
while(gtgTransac != 0)
{
gettimeofday(&lstime, NULL);
#ifdef WFA_VOICE_EXT
/*
* If your device is BIG ENDIAN, you need to
* modify the the function calls
*/
int2BuffBigEndian(asn++, &((tgHeader_t *)trafficBuf)->hdr[8]);
int2BuffBigEndian(lstime.tv_sec, &((tgHeader_t *)trafficBuf)->hdr[12]);
int2BuffBigEndian(lstime.tv_usec, &((tgHeader_t *)trafficBuf)->hdr[16]);
#else
j++;
i=0;
do
{
#endif /* WFA_VOICE_EXT */
if(gtgTransac != 0/* && nbytes <= 0 */)
{
if(respBuf == NULL)
{
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread SEND,PROF_TRANSC::a Null respBuf\n");
}
memset(respBuf, 0, WFA_RESP_BUF_SZ);
respLen = 0;
memset(trafficBuf ,0, MAX_UDP_LEN + 1);
if(wfaSendShortFile(mySock, myStreamId,
trafficBuf, 0, respBuf, &respLen) == DONE)
{
if(wfaCtrlSend(gxcSockfd, respBuf, respLen) != respLen)
{
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread SEND,PROF_TRANSC::wfaCtrlSend Error for wfaSendShortFile\n");
}
sendFailCount++;
i--;
usleep(1000);
}
else
{
i++;
sendCount++;
}
//sentTranPkts++; /* send routine already incresed this counter */
if((myProfile->maxcnt>0) &&(sendCount == myProfile->maxcnt))
{
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread SEND,PROF_TRANSC::meet maxcnt=%d; end loop\n",myProfile->maxcnt);
gtgTransac = 0; /* break snd/rcv wile loop */
break;
}
nbytes = wfaRecvFile(mySock, myStreamId, (char *)trafficBuf);
if(nbytes <= 0)
{/* Do not print any msg it will slow down process on snd/rcv */
//setsockopt(mySock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tmout, (socklen_t) sizeof(tmout));
//printf("PROF_TRANSC::time out event, wfaRecvFile failed,resend a new packet ...\n");
//tmout.tv_sec = 0;
//tmout.tv_usec = 3000; // set for 20 minlsec timeout
//setsockopt(mySock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tmout, (socklen_t) sizeof(tmout));
#if 0 /* if your socket APIs does not support "recvfrom" timeout, this is the way to loop the descriptor */
gettimeofday(&curtime, NULL);
if(((curtime.tv_sec - nxtime.tv_sec) * 1000000 + (curtime.tv_usec - nxtime.tv_usec)) < 20000)
{
continue;
}
nxtime.tv_usec = curtime.tv_usec += 20000;
if(nxtime.tv_usec >= 1000000)
{
nxtime.tv_sec = curtime.tv_sec +=1;
nxtime.tv_usec -= 1000000;
}
#endif
//continue;
}
else
{
rcvCount++;
nbytes = 0;
}
} /* if gtgTransac != 0 */
#ifdef WFA_VOICE_EXT
/*
* Roundtrip time delay:
* 1. retrieve the timestamp
* 2. calculate the Trt(1) roundtrip delay
* 3. store the minimum Trt(1)
* 4. store Cdut(t1) and Ctm(2)
*/
gettimeofday(&lrtime, NULL);
/* get a round trip time */
rttime = wfa_ftime_diff(&lstime, &lrtime);
if(min_rttime > rttime)
{
min_rttime = rttime;
if(gtgCaliRTD != 0)
{
gtgPktRTDelay = min_rttime;
}
}
if(gtgCaliRTD != 0 )
{
usleep(20000); /* wait a few min seconds before retrying the next calibration */
}
#else
/* not voice case */
/* for do-while loop for frame rate per sec */
}while ((i <= myProfile->rate + myProfile->rate/3) && (myProfile->rate !=0) && (gtgTransac != 0 ));
if(myProfile->maxcnt == 0)
{
gettimeofday(&lrtime, NULL);
rttime = wfa_itime_diff(&lstime, &lrtime);
/* we cover frame rate = 0 case without any sleep to continue push data */
if (((difftime = 1000000 - rttime) > 0) && (myProfile->rate != 0))
{
if ( j < myProfile->duration)
{
usleep (difftime);
sleepTotal = sleepTotal + difftime/1000;
}
}
if (j > myProfile->duration + 2)
{ /* avoid infinite loop */
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread SEND over time %d sec, stop sending\n",myProfile->duration);
break;
}
}
#endif /* WFA_VOICE_EXT */
} /* while */
if(mySock != -1)
{
wCLOSE(mySock);
mySock = -1;
}
DPRINT_INFO(WFA_OUT, "wfa_wmm_thread SEND::Sending stats back, sendCount=%d rcvCount=%d sleepTotal in mil-sec=%d sendFailCount=%d frmRate=%d do count=%d\n", sendCount,rcvCount,sleepTotal,sendFailCount, myProfile->rate, j);
}/* else if(myProfile->profile == PROF_TRANSC || myProfile->profile == PROF_START_SYNC || myProfile->profile == PROF_CALI_RTD) */
wMEMSET(respBuf, 0, WFA_RESP_BUF_SZ);
wSLEEP(1);
/*
* uses thread that is saved at last to pack the items and ships
* it to CA.
*/
if(myId == sendThrId)
{
wfaSentStatsResp(gxcSockfd, respBuf);
printf("done stats\n");
sendThrId = 0;
}
break;
case DIRECT_RECV:
/*
* Test WMM-PS
*/
if(myProfile->profile == PROF_UAPSD)
{
#ifdef WFA_WMM_PS_EXT /* legacy code not used now */
wmmps_info.sta_test = B_D;
wmmps_info.ps_thread = myId;
wmmps_info.rcv_state = 0;
wmmps_info.tdata = tdata;
wmmps_info.dscp = wfaTGSetPrio(psSockfd, TG_WMM_AC_BE);
tdata->state_num=0;
/*
* default timer value
*/
while(gtgWmmPS>0)
{
if(resetsnd)
{
tdata->state_num = 0;
resetsnd = 0;
}
if (wmmps_info.sta_test > LAST_TEST)
break;
tdata->state = stationProcStatetbl[wmmps_info.sta_test];
curr_state = tdata->state[tdata->state_num];
curr_state.statefunc(curr_state.pw_offon,curr_state.sleep_period,&(tdata->state_num));
}
#endif /* WFA_WMM_PS_EXT */
}
else if (myProfile->profile == PROF_IPTV || myProfile->profile == PROF_FILE_TX || myProfile->profile == PROF_MCAST)
{
char recvBuf[MAX_RCV_BUF_LEN+1];
int iOptVal, iOptLen;
struct timeval tmout;
#ifdef WFA_VOICE_EXT
struct timeval currtime;
FILE *e2eoutp = NULL;
char e2eResults[124];
int le2eCnt = 0;
#endif
mySock = wfaCreateUDPSock(myProfile->dipaddr, myProfile->dport);
if(mySock == -1)
{
printf("Error open socket\n");
continue;
}
if (myProfile->profile == PROF_MCAST)
{
int so = wfaSetSockMcastRecvOpt(mySock, myProfile->dipaddr);
if(so < 0)
{
DPRINT_ERR(WFA_ERR, "Join the multicast group failed\n");
wCLOSE(mySock);
continue;
}
}
tgSockfds[myStream->tblidx] = mySock;
#ifdef WFA_VOICE_EXT
/* only for receive stream needs to create a stats storage */
tgE2EStats_t *e2esp = NULL;
int totalE2Cnt = 220 * WFA_G_CODEC_RATE;
printf("init E2Cnt %i\n", totalE2Cnt);
if(myProfile->profile == PROF_IPTV)
{
e2esp = malloc(totalE2Cnt * sizeof(tgE2EStats_t));
if(e2esp == NULL)
{
}
}
#endif
/* increase the rec queue size */
getsockopt(mySock, SOL_SOCKET, SO_RCVBUF, (char *)&iOptVal, (socklen_t *)&iOptLen);
iOptVal = iOptVal * 10;
setsockopt(mySock, SOL_SOCKET, SO_RCVBUF, (char *)&iOptVal, (socklen_t )iOptLen);
/* set timeout for blocking receive */
tmout.tv_sec = 0;
tmout.tv_usec = 200000; /* set the receive time out to 200 ms */
setsockopt(mySock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tmout, (socklen_t) sizeof(tmout));
wfaSetThreadPrio(myId, TG_WMM_AC_VO); /* try to raise the receiver higher priority than sender */
for(;;)
{
nbytes = wfaRecvFile(mySock, myStreamId, (char *)recvBuf);
if(nbytes <= 0)
{
/* due to timeout */
if(tgSockfds[myStream->tblidx] >=0 )
continue;
break;
}
#ifdef WFA_VOICE_EXT
if(myProfile->profile == PROF_IPTV)
{
struct timeval ttval, currTimeVal;
int sn = bigEndianBuff2Int(&((tgHeader_t *)recvBuf)->hdr[8]);
ttval.tv_sec = bigEndianBuff2Int(&((tgHeader_t *)recvBuf)->hdr[12]);
ttval.tv_usec = bigEndianBuff2Int(&((tgHeader_t *)recvBuf)->hdr[16]);
gettimeofday(&currTimeVal, NULL);
/*
* take the end2end stats, limit to the max voice pkt number
*/
if(le2eCnt < totalE2Cnt)
{
tgE2EStats_t *ep = e2esp + le2eCnt++;
ep->seqnum = sn;
ep->rsec = ttval.tv_sec;
ep->rusec = ttval.tv_usec;
ep->lsec = currTimeVal.tv_sec;
ep->lusec = currTimeVal.tv_usec;
if(ep->lusec < 0)
{
ep->lsec -=1;
ep->lusec += 1000000;
}
else if(ep->lusec >= 1000000)
{
ep->lsec += 1;
ep->lusec -= 1000000;
}
}
}
#endif /* WFA_VOICE_EXT */
wfaSetThreadPrio(myId, TG_WMM_AC_BE); /* put it back down */
} /* while */
my_wmm->thr_flag = 0;
#ifdef WFA_VOICE_EXT
if(myProfile->profile == PROF_IPTV)
{
int j;
wGETTIMEOFDAY(&currtime, NULL);
sprintf(e2eResults, "/tmp/e2e%u-%i.txt", (unsigned int) currtime.tv_sec, myStreamId);
printf("file %s cnt %i\n", e2eResults, le2eCnt);
e2eoutp = fopen(e2eResults, "w+");
if(e2eoutp != NULL)
{
fprintf(e2eoutp, "roundtrip delay: %i\n", (int) (1000000*gtgPktRTDelay));
for(j = 0; j< le2eCnt; j++)
{
tgE2EStats_t *ep = e2esp+j;
fprintf(e2eoutp, "%i:%i:%i:%i:%i\n", ep->seqnum, ep->lsec, ep->lusec, ep->rsec, ep->rusec);
}
fclose(e2eoutp);
}
if(e2esp != NULL)
free(e2esp);
}
#endif
}
else if(myProfile->profile == PROF_TRANSC || myProfile->profile == PROF_START_SYNC || myProfile->profile == PROF_CALI_RTD)
{
struct timeval tmout;
mySock = wfaCreateUDPSock(myProfile->sipaddr, myProfile->sport);
if(mySock < 0)
{
/* return error */
my_wmm->thr_flag = 0;
continue;
}
tgSockfds[myStream->tblidx] = mySock;
totalTranPkts = 0xFFFFFFF0;
gtgTransac = myStreamId;
/* set timeout for blocking receive */
tmout.tv_sec = 0;
tmout.tv_usec = 400000; /* set the receive time out to 400 ms, 200ms is too short */
setsockopt(mySock, SOL_SOCKET, SO_RCVTIMEO, (char *)&tmout, (socklen_t) sizeof(tmout));
while(gtgTransac != 0)
{
memset(trafficBuf, 0, sizeof((char*)trafficBuf));
if(mySock != -1)
{
int i = gtgTransac;
nbytes = 0;
/* check for data as long as we are in a transaction */
while ((gtgTransac != 0) && (nbytes <= 0))
{
nbytes = wfaRecvFile(mySock, i, (char *)trafficBuf);
}
/* It is the end of a transaction, go out of the loop */
if (gtgTransac == 0) break;
}
else
{
break;
}
#ifdef WFA_VOICE_EXT
/* for a transaction receiver, it just needs to send the local time back */
gettimeofday(&lstime, NULL);
int2BuffBigEndian(lstime.tv_sec, &((tgHeader_t *)trafficBuf)->hdr[12]);
int2BuffBigEndian(lstime.tv_usec, &((tgHeader_t *)trafficBuf)->hdr[16]);
#endif
memset(respBuf, 0, WFA_RESP_BUF_SZ);
respLen = 0;
if(wfaSendShortFile(mySock, gtgTransac, trafficBuf, nbytes, respBuf, &respLen) == DONE)
{
if(wfaCtrlSend(gxcSockfd, (BYTE *)respBuf, respLen)!=respLen)
{
DPRINT_WARNING(WFA_WNG, "wfaCtrlSend Error\n");
}
}
}
my_wmm->thr_flag = 0;
//////////////////// Wifi Alliance Added
if(mySock != -1)
{
wCLOSE(mySock);
mySock = -1;
}
//////////////////// Wifi Alliance Added
}
break;
default:
DPRINT_ERR(WFA_ERR, "Unknown covered case\n");
}
}
}
/* This is going to be a blocking SEND till it finishes */
int wfaSendLongFile(int mySockfd, int streamid, BYTE *aRespBuf, int *aRespLen)
{
tgProfile_t *theProf = NULL;
tgStream_t *myStream = NULL;
struct sockaddr_in toAddr;
char *packBuf;
int packLen;
int bytesSent;
dutCmdResponse_t sendResp;
int sleepTime = 0;
int throttledRate = 0;
struct timeval before, after,af;
int difftime = 0, counter = 0;
struct timeval stime;
int throttled_est_cost;
int act_sleep_time;
gettimeofday(&af,0);
DPRINT_INFO(WFA_OUT, "Entering sendLongFile %i\n", streamid);
/* find the profile */
myStream = findStreamProfile(streamid);
if(myStream == NULL)
{
return FALSE;
}
theProf = &myStream->profile;
if(theProf == NULL)
{
return FALSE;
}
packLen = theProf->pksize;
/* allocate a buf */
packBuf = (char *)malloc(packLen);
wMEMSET(packBuf, 1, packLen);
/* fill in the header */
wSTRNCPY(packBuf, "1345678", sizeof(tgHeader_t));
/* initialize the destination address */
wMEMSET(&toAddr, 0, sizeof(toAddr));
toAddr.sin_family = AF_INET;
toAddr.sin_addr.s_addr = inet_addr(theProf->dipaddr);
toAddr.sin_port = htons(theProf->dport);
/* if a frame rate and duration are defined, then we know
* interval for each packet and how many packets it needs to
* send.
*/
if(theProf->duration != 0)
{
printf("duration %i\n", theProf->duration);
/*
* use this to decide periodical interval sleep time and frames to send
* int the each interval.
* Each device should adopt a own algorithm for better performance
*/
wfaTxSleepTime(theProf->profile, theProf->rate, &sleepTime, &throttledRate);
/*
* alright, we need to raise the priority level of the process
* to improve the real-time performance of packet sending.
* Since this is for tuning purpose, it is optional implementation.
*/
//wfaSetProcPriority(60);
//interval = 1*1000000/theProf->rate ; // in usec;
// Here assumes it takes 20 usec to send a packet
throttled_est_cost = throttledRate * 20; // MUST estimate the cost per ppk
act_sleep_time = sleepTime - adj_latency;
if (act_sleep_time <= 0)
act_sleep_time = sleepTime;
printf("sleep time %i act_sleep_time %i\n", sleepTime, act_sleep_time);
runLoop=1;
while(runLoop)
{
counter++;
/* fill in the counter */
int2BuffBigEndian(counter, &((tgHeader_t *)packBuf)->hdr[8]);
/*
* the following code is only used to slow down
* over fast traffic flooding the buffer and cause
* packet drop or the other end not able to receive due to
* some limitations, purely for experiment purpose.
* each implementation needs some fine tune to it.
*/
if(counter ==1)
{
wGETTIMEOFDAY(&before, NULL);
before.tv_usec += sleepTime;
if(before.tv_usec > 1000000)
{
before.tv_usec -= 1000000;
before.tv_sec +=1;
}
}
if(throttledRate != 0)
{
if(counter%throttledRate == 0)
{
wGETTIMEOFDAY(&after, NULL);
difftime = wfa_itime_diff(&after, &before);
if(difftime > adj_latency)
{
// too much time left, go sleep
wUSLEEP(difftime-adj_latency);
wGETTIMEOFDAY(&after, NULL);
difftime = wfa_itime_diff(&after, &before);
}
// burn the rest to absort latency
if(difftime >0)
buzz_time(difftime);
before.tv_usec += sleepTime;
if(before.tv_usec > 1000000)
{
before.tv_usec -= 1000000;
before.tv_sec +=1;
}
}
} // otherwise, it floods
/*
* Fill the timestamp to the header.
*/
wGETTIMEOFDAY(&stime, NULL);
int2BuffBigEndian(stime.tv_sec, &((tgHeader_t *)packBuf)->hdr[12]);
int2BuffBigEndian(stime.tv_usec, &((tgHeader_t *)packBuf)->hdr[16]);
bytesSent = wfaTrafficSendTo(mySockfd, packBuf, packLen,
(struct sockaddr *)&toAddr);
if(bytesSent != -1)
{
myStream->stats.txPayloadBytes += bytesSent;
myStream->stats.txFrames++ ;
}
else
{
int errsv = errno;
switch(errsv)
{
case EAGAIN:
case ENOBUFS:
DPRINT_ERR(WFA_ERR, "send error\n");
wUSLEEP(1000); /* hold for 1 ms */
counter-- ;
myStream->stats.txFrames--;
break;
case ECONNRESET:
runLoop = 0;
break;
case EPIPE:
runLoop = 0;
break;
default:
perror("sendto: ");
DPRINT_ERR(WFA_ERR, "Packet sent error\n");
}
}
}
/*
* lower back to an original level if the process is raised previously
* It is optional.
*/
//wfaSetProcPriority(30);
}
else /* invalid parameters */
{
/* encode a TLV for response for "invalid ..." */
sendResp.status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_SEND_RESP_TLV, 4,
(BYTE *)&sendResp, (BYTE *)aRespBuf);
/* done here */
*aRespLen = WFA_TLV_HDR_LEN + 4;
return DONE;
}
gtgSend = 0;
/* free the buffer */
wFREE(packBuf);
//printf("done sending long\n");
/* return statistics */
sendResp.status = STATUS_COMPLETE;
sendResp.streamId = myStream->id;
wMEMCPY(&sendResp.cmdru.stats, &myStream->stats, sizeof(tgStats_t));
#if 0
DPRINT_INFO(WFA_OUT, "stream Id %u tx %u total %llu\n", myStream->id, myStream->stats.txFrames, myStream->stats.txPayloadBytes);
#endif
wfaEncodeTLV(WFA_TRAFFIC_AGENT_SEND_RESP_TLV, sizeof(dutCmdResponse_t),
(BYTE *)&sendResp, (BYTE *)aRespBuf);
*aRespLen = WFA_TLV_HDR_LEN + sizeof(dutCmdResponse_t);
return DONE;
}
int wfaTGReset(int len, BYTE *parms, int *respLen, BYTE *respBuf)
{
dutCmdResponse_t *resetResp = &gGenericResp;
int i;
/* need to reset all traffic socket fds */
if(btSockfd != -1)
{
wCLOSE(btSockfd);
btSockfd = -1;
}
for(i = 0; i<WFA_MAX_TRAFFIC_STREAMS; i++)
{
if(tgSockfds[i] != -1)
{
wCLOSE(tgSockfds[i]);
tgSockfds[i] = -1;
}
}
/* reset the timer alarm if it was armed */
wALARM(0);
/* just reset the flags for the command */
gtgRecv = 0;
gtgSend = 0;
gtgTransac = 0;
#ifdef WFA_VOICE_EXT
gtgCaliRTD = 0;
min_rttime = 0xFFFFFFFF;
gtgPktRTDelay = 0xFFFFFFFF;
#endif
totalTranPkts = 0;
//gtimeOut = 0;
runLoop = 0;
usedThread = 0;
#ifdef WFA_WMM_PS_EXT
gtgWmmPS = 0;
gtgPsPktRecvd = 0;
if(psSockfd != -1)
{
wCLOSE(psSockfd);
psSockfd = -1;
}
wMEMSET(&wmmps_info, 0, sizeof(wfaWmmPS_t));
// wfaSetDUTPwrMgmt(PS_OFF);
#endif
e2eResults[0] = '\0';
/* Also need to clean up WMM streams NOT DONE YET!*/
slotCnt = 0; /* reset stream profile container */
wMEMSET(gStreams, 0, WFA_MAX_TRAFFIC_STREAMS);
/*
* After be asked to reset, it should lower itself back to
* normal level. It is optional implementation if it is not called
* while it starts sending/receiving for raising priority level.
*/
//wfaSetProcPriority(20);
/* encode a TLV for response for "complete ..." */
resetResp->status = STATUS_COMPLETE;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_RESET_RESP_TLV, 4,
(BYTE *)resetResp, respBuf);
*respLen = WFA_TLV_HDR_LEN + 4;
return TRUE;
}
/*
* wfaTGSendStart: instruct traffic generator to start sending based on a profile
* input: cmd -- not used
* response: inform controller for "running"
* return: success or failed
*/
int wfaTGSendStart(int len, BYTE *parms, int *respLen, BYTE *respBuf)
{
int i=0, streamid=0;
int numStreams = len/4;
tgProfile_t *theProfile;
tgStream_t *myStream = NULL;
dutCmdResponse_t staSendResp;
DPRINT_INFO(WFA_OUT, "Entering tgSendStart for %i streams ...\n", numStreams);
for(i=0; i<numStreams; i++)
{
wMEMCPY(&streamid, parms+(4*i), 4);
myStream = findStreamProfile(streamid);
if(myStream == NULL)
{
staSendResp.status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_SEND_RESP_TLV, 4, (BYTE *)&staSendResp, respBuf);
*respLen = WFA_TLV_HDR_LEN + 4;
return TRUE;
}
theProfile = &myStream->profile;
if(theProfile == NULL)
{
staSendResp.status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_SEND_RESP_TLV, 4, (BYTE *)&staSendResp, respBuf);
*respLen = WFA_TLV_HDR_LEN + 4;
return TRUE;
}
if(theProfile->direction != DIRECT_SEND)
{
staSendResp.status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_SEND_RESP_TLV, 4, (BYTE *)&staSendResp, respBuf);
*respLen = WFA_TLV_HDR_LEN + 4;
return TRUE;
}
/*
* need to reset the stats
*/
wMEMSET(&myStream->stats, 0, sizeof(tgStats_t));
// mark the stream active;
myStream->state = WFA_STREAM_ACTIVE;
switch(theProfile->profile)
{
case PROF_FILE_TX:
case PROF_MCAST:
case PROF_TRANSC:
gtgTransac = streamid;
gtgSend = streamid;
case PROF_CALI_RTD:
gtgCaliRTD = streamid;
case PROF_IPTV:
gtgSend = streamid;
/*
* singal the thread to Sending WMM traffic
*/
//if(usedThread <
wmm_thr[usedThread].thr_flag = streamid;
wPT_MUTEX_LOCK(&wmm_thr[usedThread].thr_flag_mutex);
wPT_COND_SIGNAL(&wmm_thr[usedThread].thr_flag_cond);
wPT_MUTEX_UNLOCK(&wmm_thr[usedThread].thr_flag_mutex);
usedThread++;
//wfaSetProcPriority(90);
break;
}
}
*respLen = 0;
return TRUE;
}
/*
* tgRecvStop: instruct traffic generator to stop receiving based on a profile
* input: cmd -- not used
* response: inform controller for "complete"
* return: success or failed
*/
int wfaTGRecvStop(int len, BYTE *parms, int *respLen, BYTE *respBuf)
{
int status = STATUS_COMPLETE, i;
int numStreams = len/4;
unsigned int streamid;
tgProfile_t *theProfile;
tgStream_t *myStream=NULL;
dutCmdResponse_t statResp;
BYTE dutRspBuf[WFA_RESP_BUF_SZ];
int id_cnt = 0;
DPRINT_INFO(WFA_OUT, "entering tgRecvStop with length %d\n",len);
/* in case that send-stream not done yet, an optional delay */
while(sendThrId != 0)
sleep(1);
/*
* After finishing the receiving command, it should lower itself back to
* normal level. It is optional implementation if it is not called
* while it starts receiving for raising priority level.
*/
//wfaSetProcPriority(30);
wMEMSET(dutRspBuf, 0, WFA_RESP_BUF_SZ);
for(i=0; i<numStreams; i++)
{
wMEMCPY(&streamid, parms+(4*i), 4);
printf(" stop stream id %i\n", streamid);
myStream = findStreamProfile(streamid);
if(myStream == NULL)
{
status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_RECV_STOP_RESP_TLV, 4, (BYTE *)&status, respBuf);
*respLen = WFA_TLV_HDR_LEN + 4;
printf("stream table empty\n");
continue;
}
theProfile = &myStream->profile;
if(theProfile == NULL)
{
status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_RECV_STOP_RESP_TLV, 4, (BYTE *)&status, respBuf);
*respLen = WFA_TLV_HDR_LEN + 4;
return TRUE;
}
if(theProfile->direction != DIRECT_RECV)
{
status = STATUS_INVALID;
wfaEncodeTLV(WFA_TRAFFIC_AGENT_RECV_STOP_RESP_TLV, 4, (BYTE *)&status, respBuf);
*respLen = WFA_TLV_HDR_LEN + 4;
return TRUE;
}
/* reset its flags , close sockets */
switch(theProfile->profile)
{
case PROF_TRANSC:
case PROF_CALI_RTD:
gtgTransac = 0;
case PROF_MCAST:
case PROF_FILE_TX:
case PROF_IPTV:
gtgRecv = 0;
if(tgSockfds[myStream->tblidx] != -1)
{
wCLOSE(tgSockfds[myStream->tblidx]);
tgSockfds[myStream->tblidx] = -1;
}
break;
case PROF_UAPSD:
#ifdef WFA_WMM_PS_EXT
gtgWmmPS = 0;
gtgPsPktRecvd = 0;
if(psSockfd != -1)
{
wCLOSE(psSockfd);
psSockfd = -1;
}
wMEMSET(&wmmps_info, 0, sizeof(wfaWmmPS_t));
wfaSetDUTPwrMgmt(PS_OFF);
#endif /* WFA_WMM_PS_EXT */
break;
}
/* encode a TLV for response for "complete/error ..." */
statResp.status = STATUS_COMPLETE;
statResp.streamId = streamid;
#if 1
DPRINT_INFO(WFA_OUT, "stream Id %u rx %u total %llu\n", streamid, myStream->stats.rxFrames, myStream->stats.rxPayloadBytes);
#endif
wMEMCPY(&statResp.cmdru.stats, &myStream->stats, sizeof(tgStats_t));
wMEMCPY((dutRspBuf + i * sizeof(dutCmdResponse_t)), (BYTE *)&statResp, sizeof(dutCmdResponse_t));
id_cnt++;
// Not empty it but require to reset the entire table before test starts.
//wMEMSET(myStream, 0, sizeof(tgStream_t));
}
// mark the stream inactive
myStream->state = WFA_STREAM_INACTIVE;
printf("Sending back the statistics at recvstop\n");
wfaEncodeTLV(WFA_TRAFFIC_AGENT_RECV_STOP_RESP_TLV, id_cnt * sizeof(dutCmdResponse_t), dutRspBuf, respBuf);
/* done here */
*respLen = WFA_TLV_HDR_LEN + numStreams * sizeof(dutCmdResponse_t);
return TRUE;
}
/* RecvStart: instruct traffic generator to start receiving
* based on a profile
* input: cmd -- not used
* response: inform controller for "running"
* return: success or failed
*/
int wfaTGRecvStart(int len, BYTE *parms, int *respLen, BYTE *respBuf)
{
int status = STATUS_COMPLETE, i;
int numStreams = len/4;
int streamid;
tgProfile_t *theProfile;
tgStream_t *myStream;
DPRINT_INFO(WFA_OUT, "entering tgRecvStart\n");
/*
* The function wfaSetProcPriority called here is to enhance the real-time
* performance for packet receiving. It is only for tuning and optional
* to implement
*/
//wfaSetProcPriority(60);
for(i=0; i<numStreams; i++)
{
wMEMCPY(&streamid, parms+(4*i), 4); /* changed from 2 to 4, bug reported by n.ojanen */
myStream = findStreamProfile(streamid);
if(myStream == NULL)
{
status = STATUS_INVALID;
return status;
}
theProfile = &myStream->profile;
if(theProfile == NULL)
{
status = STATUS_INVALID;
return status;
}
/* calculate the frame interval which is used to derive its jitter */
if(theProfile->rate != 0 && theProfile->rate < 5000)
myStream->fmInterval = 1000000/theProfile->rate; /* in ms */
else
myStream->fmInterval = 0;
if(theProfile->direction != DIRECT_RECV)
{
status = STATUS_INVALID;
return status;
}
wMEMSET(&myStream->stats, 0, sizeof(tgStats_t));
// mark the stream active
myStream->state = WFA_STREAM_ACTIVE;
switch(theProfile->profile)
{
#ifdef WFA_WPA2_SINGLE_THREAD
case PROF_MCAST:
case PROF_FILE_TX:
// case PROF_IPTV:
btSockfd = wfaCreateUDPSock(theProfile->dipaddr, theProfile->dport);
gtgRecv = streamid;
if(btSockfd < 0)
status = STATUS_ERROR;
else
{
/* get current flags setting */
int ioflags = wFCNTL(btSockfd, F_GETFL, 0);
/* set only BLOCKING flag to non-blocking */
wFCNTL(btSockfd, F_SETFL, ioflags | O_NONBLOCK);
}
break;
#else
case PROF_TRANSC:
case PROF_CALI_RTD: /* Calibrate roundtrip delay */
gtgTransac = streamid;
case PROF_MCAST:
case PROF_FILE_TX:
case PROF_IPTV:
gtgRecv = streamid;
wmm_thr[usedThread].thr_flag = streamid;
wPT_MUTEX_LOCK(&wmm_thr[usedThread].thr_flag_mutex);
wPT_COND_SIGNAL(&wmm_thr[usedThread].thr_flag_cond);
wPT_MUTEX_UNLOCK(&wmm_thr[usedThread].thr_flag_mutex);
printf("Recv Start in thread %i for streamid %i\n", usedThread, streamid);
usedThread++;
break;
#endif
case PROF_UAPSD:
#ifdef WFA_WMM_PS_EXT
status = STATUS_COMPLETE;
psSockfd = wfaCreateUDPSock(theProfile->dipaddr, WFA_WMMPS_UDP_PORT);
wmmps_info.sta_state = 0;
wmmps_info.wait_state = WFA_WAIT_STAUT_00;
wMEMSET(&wmmps_info.psToAddr, 0, sizeof(wmmps_info.psToAddr));
wmmps_info.psToAddr.sin_family = AF_INET;
wmmps_info.psToAddr.sin_addr.s_addr = inet_addr(theProfile->sipaddr);
wmmps_info.psToAddr.sin_port = htons(theProfile->sport);
wmmps_info.reset = 0;
wmm_thr[usedThread].thr_flag = streamid;
wmmps_info.streamid = streamid;
wPT_MUTEX_LOCK(&wmm_thr[usedThread].thr_flag_mutex);
wPT_COND_SIGNAL(&wmm_thr[usedThread].thr_flag_cond);
gtgWmmPS = streamid;;
wPT_MUTEX_UNLOCK(&wmm_thr[usedThread].thr_flag_mutex);
usedThread++;
#endif /* WFA_WMM_PS_EXT */
break;
}
}
/* encode a TLV for response for "complete/error ..." */
wfaEncodeTLV(WFA_TRAFFIC_AGENT_RECV_START_RESP_TLV, sizeof(int),
(BYTE *)&status, respBuf);
*respLen = WFA_TLV_HDR_LEN + sizeof(int);
return TRUE;
}
/* always receive from a specified IP address and Port */
int wfaRecvFile(int mySockfd, int streamid, char *recvBuf)
{
/* how many packets are received */
char *packBuf = recvBuf;
struct sockaddr_in fromAddr;
tgProfile_t *theProf;
tgStream_t *myStream;
unsigned int bytesRecvd;
int lostPkts;
/* find the profile */
myStream = findStreamProfile(streamid);
if(myStream == NULL)
{
return FALSE;
}
theProf = &myStream->profile;
if(theProf == NULL)
{
return FALSE;
}
wMEMSET(packBuf, 0, MAX_UDP_LEN);
wMEMSET(&fromAddr, 0, sizeof(fromAddr));
fromAddr.sin_family = AF_INET;
fromAddr.sin_addr.s_addr = inet_addr(theProf->dipaddr);
fromAddr.sin_port = htons(theProf->dport);
if(gtgRecv && gtgTransac)
{
fromAddr.sin_addr.s_addr = inet_addr(theProf->sipaddr);
fromAddr.sin_port = htons(theProf->sport);
}
else if(gtgSend && gtgTransac)
{
fromAddr.sin_addr.s_addr = inet_addr(theProf->dipaddr);
fromAddr.sin_port = htons(theProf->dport);
}
/* it is always to receive at least one packet, in case more in the
queue, just pick them up.
*/
bytesRecvd = wfaTrafficRecv(mySockfd, packBuf, (struct sockaddr *)&fromAddr);
if(bytesRecvd != -1)
{
myStream->stats.rxFrames++;
myStream->stats.rxPayloadBytes +=bytesRecvd;
/*
* Get the lost packet count
*/
lostPkts = bigEndianBuff2Int(&((tgHeader_t *)packBuf)->hdr[8]) - 1 - myStream->lastPktSN;
myStream->stats.lostPkts += lostPkts;
myStream->lastPktSN = bigEndianBuff2Int(&((tgHeader_t *)packBuf)->hdr[8]);
}
else
{
#if 0
DPRINT_ERR(WFA_ERR, "Packet received error\n");
perror("receive error");
#endif
}
return (bytesRecvd);
}
/* this only sends one packet a time */
int wfaSendShortFile(int mySockfd, int streamid, BYTE *sendBuf, int pksize, BYTE *aRespBuf, int *aRespLen)
{
BYTE *packBuf = sendBuf;
struct sockaddr_in toAddr;
tgProfile_t *theProf;
tgStream_t *myStream;
int packLen, bytesSent=-1;
dutCmdResponse_t sendResp;
if(mySockfd == -1)
{
/* stop */
gtgTransac = 0;
//gtimeOut = 0;
gtgRecv = 0;
gtgSend = 0;
printf("stop short traffic\n");
myStream = findStreamProfile(streamid);
if(myStream != NULL)
{
sendResp.status = STATUS_COMPLETE;
sendResp.streamId = streamid;
wMEMCPY(&sendResp.cmdru.stats, &myStream->stats, sizeof(tgStats_t));
wfaEncodeTLV(WFA_TRAFFIC_AGENT_SEND_RESP_TLV, sizeof(dutCmdResponse_t), (BYTE *)&sendResp, aRespBuf);
*aRespLen = WFA_TLV_HDR_LEN + sizeof(dutCmdResponse_t);
}
return DONE;
}
/* find the profile */
myStream = findStreamProfile(streamid);
if(myStream == NULL)
{
return FALSE;
}
theProf = &myStream->profile;
if(theProf == NULL)
{
return FALSE;
}
if(pksize == 0)
packLen = theProf->pksize;
else
packLen = pksize;
wMEMSET(&toAddr, 0, sizeof(toAddr));
toAddr.sin_family = AF_INET;
toAddr.sin_addr.s_addr = inet_addr(theProf->sipaddr);
toAddr.sin_port = htons(theProf->sport);
if(gtgRecv && gtgTransac)
{
// printf("mySock %i sipaddr %s sport %i\n", mySockfd, theProf->sipaddr, theProf->sport);
toAddr.sin_addr.s_addr = inet_addr(theProf->sipaddr);
toAddr.sin_port = htons(theProf->sport);
}
else if(gtgSend && gtgTransac)
{
// printf("mySock %i dipaddr %s dport %i\n", mySockfd, theProf->dipaddr, theProf->dport);
toAddr.sin_addr.s_addr = inet_addr(theProf->dipaddr);
toAddr.sin_port = htons(theProf->dport);
}
int2BuffBigEndian(myStream->stats.txFrames, &((tgHeader_t *)packBuf)->hdr[8]);
if(mySockfd != -1)
bytesSent = wfaTrafficSendTo(mySockfd, (char *)packBuf, packLen, (struct sockaddr *)&toAddr);
if(bytesSent != -1)
{
myStream->stats.txFrames++;
myStream->stats.txPayloadBytes += bytesSent;
}
else
{
int errsv = errno;
switch(errsv)
{
case EAGAIN:
case ENOBUFS:
DPRINT_ERR(WFA_ERR, "send error\n");
wUSLEEP(1000); /* hold for 1 ms */
myStream->stats.txFrames--;
break;
default:
;;
//perror("sendto: ");
}
}
sentTranPkts++;
return TRUE;
}
