diff options
Diffstat (limited to 'src/libcharon/plugins/load_tester/load_tester_control.c')
| -rw-r--r-- | src/libcharon/plugins/load_tester/load_tester_control.c | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/libcharon/plugins/load_tester/load_tester_control.c b/src/libcharon/plugins/load_tester/load_tester_control.c index d0e56176d..db2a31368 100644 --- a/src/libcharon/plugins/load_tester/load_tester_control.c +++ b/src/libcharon/plugins/load_tester/load_tester_control.c @@ -142,6 +142,31 @@ static bool equals(uintptr_t a, uintptr_t b) return a == b; } +METHOD(listener_t, alert, bool, + init_listener_t *this, ike_sa_t *ike_sa, alert_t alert, va_list args) +{ + if (alert == ALERT_RETRANSMIT_SEND) + { + uintptr_t id; + bool match = FALSE; + + id = ike_sa->get_unique_id(ike_sa); + this->mutex->lock(this->mutex); + if (this->initiated->get(this->initiated, (void*)id)) + { + match = TRUE; + } + this->mutex->unlock(this->mutex); + + if (match) + { + fprintf(this->stream, "*"); + fflush(this->stream); + } + } + return TRUE; +} + METHOD(listener_t, ike_state_change, bool, init_listener_t *this, ike_sa_t *ike_sa, ike_sa_state_t state) { @@ -214,6 +239,7 @@ static job_requeue_t initiate(FILE *stream) INIT(listener, .listener = { .ike_state_change = _ike_state_change, + .alert = _alert, }, .stream = stream, .initiated = hashtable_create((void*)hash, (void*)equals, count), |