summaryrefslogtreecommitdiff
path: root/src/tools/cbr/cbr_client.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/tools/cbr/cbr_client.c')
-rw-r--r--src/tools/cbr/cbr_client.c68
1 files changed, 49 insertions, 19 deletions
diff --git a/src/tools/cbr/cbr_client.c b/src/tools/cbr/cbr_client.c
index ff7d4057..b0c04f39 100644
--- a/src/tools/cbr/cbr_client.c
+++ b/src/tools/cbr/cbr_client.c
@@ -24,18 +24,29 @@
#include <ouroboros/dev.h>
#include <ouroboros/time_utils.h>
-int client_main(int duration, int size, long rate)
+static void busy_wait_until(const struct timespec * deadline)
+{
+ struct timespec now;
+ clock_gettime(CLOCK_REALTIME, &now);
+ while (now.tv_sec < deadline->tv_sec)
+ clock_gettime(CLOCK_REALTIME, &now);
+ while (now.tv_sec == deadline->tv_sec
+ && now.tv_nsec < deadline->tv_nsec)
+ clock_gettime(CLOCK_REALTIME, &now);
+}
+
+int client_main(int duration, int size, long rate, bool flood, bool sleep)
{
int fd = 0;
int result = 0;
bool stop = false;
char buf[size];
long seqnr = 0;
- unsigned long gap = size * 8 * (BILLION / rate); /* ns */
+ unsigned long gap = size * 8.0 * (BILLION / (double) rate);
struct timespec start;
struct timespec end;
- struct timespec interval = {(gap / BILLION), gap % BILLION};
+ struct timespec intv = {(gap / BILLION), gap % BILLION};
int ms;
if (ap_init(CLIENT_AP_NAME)) {
@@ -62,24 +73,43 @@ int client_main(int duration, int size, long rate)
}
clock_gettime(CLOCK_REALTIME, &start);
- while (!stop) {
- memcpy(buf, &seqnr, sizeof(seqnr));
-
- if (flow_write(fd, buf, size) == -1) {
- printf("Failed to write SDU.\n");
- stop = true;
- continue;
+ if (!flood) {
+ while (!stop) {
+ clock_gettime(CLOCK_REALTIME, &end);
+ ts_add(&end, &intv, &end);
+ memcpy(buf, &seqnr, sizeof(seqnr));
+
+ if (flow_write(fd, buf, size) == -1) {
+ stop = true;
+ continue;
+ }
+
+ if (sleep)
+ nanosleep(&intv, NULL);
+ else
+ busy_wait_until(&end);
+
+ ++seqnr;
+
+ if (ts_diff_us(&start, &end) / MILLION
+ >= (long) duration)
+ stop = true;
+ }
+ } else { /* flood */
+ while (!stop) {
+ clock_gettime(CLOCK_REALTIME, &end);
+ if (flow_write(fd, buf, size) == -1) {
+ stop = true;
+ continue;
+ }
+
+ ++seqnr;
+
+ if (ts_diff_us(&start, &end) / MILLION
+ >= (long) duration)
+ stop = true;
}
- nanosleep(&interval, NULL);
-
- seqnr++;
-
- clock_gettime(CLOCK_REALTIME, &end);
-
- if (duration != 0
- && ts_diff_us(&start, &end) / MILLION >= (long) duration)
- stop = true;
}
clock_gettime(CLOCK_REALTIME, &end);