#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <thread>

using namespace std;

void send_message(int sockfd, struct sockaddr_in addr) {
	while (1) {
		char buffer[1024];
		bzero(buffer, 1024);
		cin.getline(buffer, 1024);
		sendto(sockfd, buffer, 1024, 0, (struct sockaddr*)&addr, sizeof(addr));
	}
}

int main(int argc, char** argv) {

	if (argc != 3) {
		printf("Usage: %s ip port\n", argv[0]);
		exit(0);
	}
	if (strlen(*argv) < 3) {
		printf("Usage: %s ip port\n", argv[0]);
		exit(0);
	}

	char* ip = argv[1];
	int port = atoi(argv[2]);

	int sockfd;
	struct sockaddr_in addr;
	char buffer[1024];
	socklen_t addr_size;

	sockfd = socket(AF_INET, SOCK_DGRAM, 0);
	if (sockfd < 0) {
		perror("[-] socket error");
		exit(1);
	}

	memset(&addr, '\0', sizeof(addr));
	addr.sin_family = AF_INET;
	addr.sin_port = htons(port);
	addr.sin_addr.s_addr = inet_addr(ip);

	char username[50];
	cout << "Enter tour username:" << endl;
	bzero(username, 50);
	cin.getline(username, 50);
	sendto(sockfd, username, 50, 0, (struct sockaddr*)&addr, sizeof(addr));
	cout << "Your username: " << username << endl;

	thread t1(send_message, sockfd, addr);

	while (1) {
		bzero(buffer, 1024);
		addr_size = sizeof(addr);
		recvfrom(sockfd, buffer, 1024, 0, (struct sockaddr*)&addr, &addr_size);
		cout << buffer << endl;
	}
    

	t1.join();
    return 0;

}