diff options
| author | Vadim Kochan <vadim4j@gmail.com> | 2016-08-13 02:11:12 +0300 |
|---|---|---|
| committer | Tobias Klauser <tklauser@distanz.ch> | 2016-09-21 09:54:11 +0200 |
| commit | d5bcf8d0f5a2a549c3cbe71e7704bb557c7eb6a3 (patch) | |
| tree | 24391949bd00fc3b7bb285db28c5249a8cef6e37 | |
| parent | 208db0e4a3ceb21a65527d87198cf6f49c4b06dc (diff) | |
trafgen: proto: Randomize proto field at runtime
Add dynamic proto field function which can generate random value in
specified range (default 0 - MAX_UINT32).
Signed-off-by: Vadim Kochan <vadim4j@gmail.com>
Signed-off-by: Tobias Klauser <tklauser@distanz.ch>
| -rw-r--r-- | trafgen_proto.c | 27 | ||||
| -rw-r--r-- | trafgen_proto.h | 1 |
2 files changed, 28 insertions, 0 deletions
diff --git a/trafgen_proto.c b/trafgen_proto.c index 6c37b21..8c316b1 100644 --- a/trafgen_proto.c +++ b/trafgen_proto.c @@ -469,6 +469,31 @@ static void field_inc_func(struct proto_field *field) } } +static inline uint32_t field_rand(struct proto_field *field) +{ + return field->func.min + (rand() % ((field->func.max - field->func.min) + 1)); +} + +static void field_rnd_func(struct proto_field *field) +{ + if (field->len == 1) { + proto_field_set_u8(field->hdr, field->id, + (uint8_t) field_rand(field)); + } else if (field->len == 2) { + proto_field_set_be16(field->hdr, field->id, + (uint16_t) field_rand(field)); + } else if (field->len == 4) { + proto_field_set_be32(field->hdr, field->id, + (uint32_t) field_rand(field)); + } else if (field->len > 4) { + uint8_t *bytes = __proto_field_get_bytes(field); + uint32_t i; + + for (i = 0; i < field->len; i++) + bytes[i] = (uint8_t) field_rand(field); + } +} + void proto_field_func_add(struct proto_hdr *hdr, uint32_t fid, struct proto_field_func *func) { @@ -499,6 +524,8 @@ void proto_field_func_add(struct proto_hdr *hdr, uint32_t fid, } field->func.update_field = field_inc_func; + } else if (func->type & PROTO_FIELD_FUNC_RND) { + field->func.update_field = field_rnd_func; } } diff --git a/trafgen_proto.h b/trafgen_proto.h index 0a1c3de..41e9383 100644 --- a/trafgen_proto.h +++ b/trafgen_proto.h @@ -54,6 +54,7 @@ struct proto_hdr { enum proto_field_func_t { PROTO_FIELD_FUNC_INC = 1 << 0, PROTO_FIELD_FUNC_MIN = 1 << 1, + PROTO_FIELD_FUNC_RND = 1 << 2, }; struct proto_field_func { |