diff options
Diffstat (limited to 'src/dns_decoder.cpp')
| -rw-r--r-- | src/dns_decoder.cpp | 32 |
1 files changed, 21 insertions, 11 deletions
diff --git a/src/dns_decoder.cpp b/src/dns_decoder.cpp index 784ad07..603f786 100644 --- a/src/dns_decoder.cpp +++ b/src/dns_decoder.cpp @@ -1894,22 +1894,32 @@ void dns_message_uuid_get0(struct dns_message *msg, char **value, size_t *value_ enum dns_message_type dns_message_type_get(struct dns_message *msg) { - if(NULL==msg) - { - return DNS_MESSAGE_MAX; - } - - return msg->type; + return ((NULL==msg) ? DNS_MESSAGE_MAX : msg->type); } int32_t dns_message_header_id_get(struct dns_message *msg) { - if(NULL==msg || msg->magic!=DNS_MESSAGE_MAGIC || msg->type==DNS_MESSAGE_TRANSACTION_BEGIN || msg->type==DNS_MESSAGE_TRANSACTION_END) - { - return -1; - } + return ((NULL==msg || msg->magic!=DNS_MESSAGE_MAGIC || msg->type==DNS_MESSAGE_TRANSACTION_BEGIN || msg->type==DNS_MESSAGE_TRANSACTION_END) ? -1 : msg->trans_identifier_id); +} + +int32_t dns_message_header_qdcount_get(struct dns_message *msg) +{ + return ((NULL==msg || msg->magic!=DNS_MESSAGE_MAGIC || msg->type==DNS_MESSAGE_TRANSACTION_BEGIN || msg->type==DNS_MESSAGE_TRANSACTION_END) ? -1 : msg->n_question); +} + +int32_t dns_message_header_ancount_get(struct dns_message *msg) +{ + return ((NULL==msg || msg->magic!=DNS_MESSAGE_MAGIC || msg->type==DNS_MESSAGE_TRANSACTION_BEGIN || msg->type==DNS_MESSAGE_TRANSACTION_END) ? -1 : msg->n_answer_rr); +} - return msg->trans_identifier_id; +int32_t dns_message_header_aucount_get(struct dns_message *msg) +{ + return ((NULL==msg || msg->magic!=DNS_MESSAGE_MAGIC || msg->type==DNS_MESSAGE_TRANSACTION_BEGIN || msg->type==DNS_MESSAGE_TRANSACTION_END) ? -1 : msg->n_authority_rr); +} + +int32_t dns_message_header_adcount_get(struct dns_message *msg) +{ + return ((NULL==msg || msg->magic!=DNS_MESSAGE_MAGIC || msg->type==DNS_MESSAGE_TRANSACTION_BEGIN || msg->type==DNS_MESSAGE_TRANSACTION_END) ? -1 : msg->n_additional_rr); } struct dns_flag *dns_message_header_flag_get0(struct dns_message *msg) |