rsyslog/rsyslog
1
0
Fork 0
mirror of https://github.com/rsyslog/rsyslog.git synced 2025-12-21 17:10:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

experiemental milestone: tcpflood support buffering in TLS mode

Browse Source 
This needs to be checked, it is not yet 100% correct
This commit is contained in:
Rainer Gerhards 2011-02-04 09:22:58 +01:00
parent 0c880a4a30
commit 233157e520
1 changed files with 30 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/tcpflood.c
View File

@ -100,6 +100,7 @@
#define NETTEST_INPUT_CONF_FILE "nettest.input.conf" /* name of input file, must match $IncludeConfig in .conf files */
#define MAX_EXTRADATA_LEN 100*1024
#define MAX_SENDBUF 2 * MAX_EXTRADATA_LEN
static char *targetIP = "127.0.0.1";
static char *msgPRI = "167";
@ -397,6 +398,8 @@ int sendMessages(struct instdata *inst)
int lenSend = 0;
char *statusText = "";
char buf[MAX_EXTRADATA_LEN + 1024];
char sendBuf[MAX_SENDBUF];
int offsSendBuf = 0;
if(!bSilent) {
if(dataFile == NULL) {
@ -438,7 +441,16 @@ int sendMessages(struct instdata *inst)
} else if(transport == TP_UDP) {
lenSend = sendto(udpsock, buf, lenBuf, 0, &udpRcvr, sizeof(udpRcvr));
} else if(transport == TP_TLS) {
lenSend = sendTLS(socknum, buf, lenBuf);
if(offsSendBuf + lenBuf < MAX_SENDBUF) {
memcpy(sendBuf+offsSendBuf, buf, lenBuf);
offsSendBuf += lenBuf;
lenSend = lenBuf; /* simulate "good" call */
} else {
lenSend = sendTLS(socknum, sendBuf, offsSendBuf);
lenSend = (lenSend == offsSendBuf) ? lenBuf : -1;
memcpy(sendBuf, buf, lenBuf);
offsSendBuf = lenBuf;
}
}
if(lenSend != lenBuf) {
printf("\r%5.5d\n", i);
@ -469,6 +481,11 @@ int sendMessages(struct instdata *inst)
++msgNum;
++i;
}
if(transport == TP_TLS && offsSendBuf != 0) {
/* send remaining buffer */
lenSend = sendTLS(socknum, sendBuf, offsSendBuf);
printf("TLS send buffer of %d messages remaining, sent %d\n", offsSendBuf, lenSend);
}
if(!bSilent)
printf("\r%8.8d %s sent\n", i, statusText);
@ -731,7 +748,18 @@ initTLSSess(int i)
static int
sendTLS(int i, char *buf, int lenBuf)
{
return gnutls_record_send(sessArray[i], buf, lenBuf);
int lenSent;
int r;
lenSent = 0;
while(lenSent != lenBuf) {
r = gnutls_record_send(sessArray[i], buf + lenSent, lenBuf - lenSent);
if(r < 0)
break;
lenSent += r;
}
return lenSent;
}
static void