From 9bb27fbea67c5c8b6c4bf4d01b4e197ebe19fe5e Mon Sep 17 00:00:00 2001 From: seshagiriprabhu Date: Tue, 25 Feb 2014 22:18:27 +0530 Subject: [PATCH] bound checking in messenger testing --- testing/Messenger_test.c | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/testing/Messenger_test.c b/testing/Messenger_test.c index 895a23d9..85ab3cd9 100644 --- a/testing/Messenger_test.c +++ b/testing/Messenger_test.c @@ -164,9 +164,12 @@ int main(int argc, char *argv[]) char temp_hex_id[128]; printf("\nEnter the address of the friend you wish to add (38 bytes HEX format):\n"); - if (scanf("%s", temp_hex_id) != 1) { - return 1; - } + if (!fgets(temp_hex_id, sizeof(temp_hex_id), stdin)) + exit(0); + + if ((strlen(temp_hex_id) > 0) && (temp_hex_id[strlen(temp_hex_id) - 1] == '\n')) + temp_hex_id[strlen(temp_hex_id) - 1] = '\0'; + uint8_t *bin_id = hex_string_to_bin(temp_hex_id); int num = m_addfriend(m, bin_id, (uint8_t *)"Install Gentoo", sizeof("Install Gentoo"));