Compare commits

...

2 Commits

3 changed files with 14 additions and 7 deletions

1
.gitignore vendored
View File

@@ -1,2 +1,3 @@
main main
user_file.txt user_file.txt
*.o

8
main.c
View File

@@ -66,6 +66,14 @@ int main() {
} else { } else {
if (dest_address[i] == NULL) { if (dest_address[i] == NULL) {
dest_address[i] = fetch_address(buffer); dest_address[i] = fetch_address(buffer);
if (dest_address[i] == NULL) {
printf("Invalid message format\n");
exit(241);
}
if (check_ip_ver(dest_address[i]) == -1) {
printf("Invalid address: %s\n",dest_address[i]);
exit(242);
}
} }

View File

@@ -4,14 +4,12 @@
char* fetch_address(char* message) { char* fetch_address(char* message) {
char* token = malloc (sizeof(char) * strlen(message)); char* token = malloc (sizeof(char) * strlen(message));
token = strtok(message," "); token = strtok(message," \n");
if (strcmp(token,"TO:") == 0) { if (strcmp(token,"TO:") == 0) {
token = strtok(NULL,""); token = strtok(NULL," \n");
} else {
return NULL;
} }
if (check_ip_ver(token) == -1) {
return NULL;
} else {
return token; return token;
} }
}