Implement queue system to display peers broadcasting

log.c

@@ -1,5 +1,6 @@
 #include "log.h"
 #include <stdio.h>
+#include <string.h>
 #include <stdlib.h>
 #include <stdarg.h>
 #include <time.h>
@@ -16,6 +17,10 @@ const char* get_current_time()
 void log_message(LogLevel level, const char* format, ...)
 {
 	const char* level_strings[] = { "INFO", "WARNING", "ERROR" };
+	static char* last_message = NULL;
+	static unsigned int repeat_count = 0;
+	static unsigned int exp_count = 1;
+	char* message = NULL;
 
 	// Open the log file in append mode
 	FILE* log_file = fopen("logfile.txt", "a");
@@ -26,19 +31,48 @@ void log_message(LogLevel level, const char* format, ...)
 
 	va_list args;
 
-	// Write the log message to the file
 	va_start(args, format);
-	fprintf(log_file, "[%s] [%s] ", get_current_time(), level_strings[level]);
+	int message_size =  vsnprintf(NULL, 0, format, args) + 1;
-	vfprintf(log_file, format, args);
-	fprintf(log_file, "\n");
-	va_end(args);
-	// Write to stdout
-	va_start(args, format);
-	printf("[%s] ", level_strings[level]);
-	vprintf(format, args);
-	printf("\n");
 	va_end(args);
 
-	// Close the log file
+	va_start(args, format);
+	message = malloc(message_size);
+	vsnprintf(message, message_size, format, args);
+	va_end(args);
+
+	if (!last_message) goto output;
+	// If current and last messages different
+	if (strcmp(message, last_message)) {
+		repeat_count = 0;
+		exp_count = 1;
+		goto output;
+	}
+	repeat_count++;
+	if (repeat_count < LOG_MAX_REPEAT) goto output;
+
+	unsigned int bit_width = sizeof(exp_count) * 8; // Calculate the number of bits in an int
+	unsigned int leftmost_bit_mask = 1 << (bit_width - 1); // Create a mask with the
+	while (exp_count < repeat_count && exp_count < leftmost_bit_mask) exp_count = exp_count << 1;
+	if (repeat_count != exp_count && repeat_count != LOG_MAX_REPEAT) goto clean;
+
+	fprintf(log_file, "[%s] [%s] %s !! x%u !!\n", get_current_time(), level_strings[level], last_message, repeat_count);
+	printf("[%s] %s !! x%u !!\n", level_strings[level], last_message, repeat_count);
+
+	goto clean;
+output:
+	// Write the log message to the file
+	fprintf(log_file, "[%s] [%s] ", get_current_time(), level_strings[level]);
+	fprintf(log_file, "%s", message);
+	fprintf(log_file, "\n");
+	// Write to stdout
+	printf("[%s] ", level_strings[level]);
+	printf("%s", message);
+	printf("\n");
+	// Record last message
+	if (last_message) free(last_message);
+	last_message = malloc(message_size);
+	strcpy(last_message, message);
+clean:
+	free(message);
 	fclose(log_file);
 }

log.h

@@ -2,6 +2,8 @@
 #ifndef LOG_H
 #define LOG_H
 
+#define LOG_MAX_REPEAT 5
+
 typedef enum {
 	LOG_INFO,
 	LOG_WARNING,

main.c

@@ -39,6 +39,7 @@ void destroy();
 int select_device(SDL_KeyboardEvent* key_event, int max_index);
 void application_state_to_string(application_state state, char* string);
 
+void* broadcast_t(void* args);
 void* peer_discovery_t(void* args);
 
 SDL_Window* window = NULL;
@@ -59,8 +60,8 @@ int main(int argc, char* argv[])
 	char device_selected_fmt[] = "Device selected: %i";
 
 	state state;
-	//state.application_state = SOCKET_LISTEN;
+	state.application_state = SOCKET_LISTEN;
-	state.application_state = SELECTING_DEVICE;
+	//state.application_state = SELECTING_DEVICE;
 	if (!audio_init(&state.audio)) return 1;
 	state.device_index = -1;
 	state.update_ui = 0;
@@ -90,6 +91,9 @@ int main(int argc, char* argv[])
 	pthread_t peer_discovery_thread;
 	pthread_create(&peer_discovery_thread, NULL, peer_discovery_t, &state);
 
+	pthread_t broadcast_thread;
+	pthread_create(&broadcast_thread, NULL, broadcast_t, &state);
+
 	SDL_Event event;
 	while (1) {
 		sem_wait(&state.sem);
@@ -116,6 +120,19 @@ int main(int argc, char* argv[])
 			application_state_to_string(state.application_state, char_buffer);
 			text_texture_load(&state_text_texture, char_buffer);
 			log_message(LOG_INFO, "Update UI State: %s", char_buffer);
+
+			int i = 0;
+			struct sockaddr_in* peer_addr;
+			while (net_get_peers(&peer_addr)) {
+				if (i >= socket_frame.len) break;;
+				char* ip = inet_ntoa(peer_addr->sin_addr);
+				int port = ntohs(peer_addr->sin_port);
+				sprintf(char_buffer, "%s:%i", ip, port);
+				text_texture_load(&socket_frame.textures[i], char_buffer);
+				i++;
+			}
+			for (; i < socket_frame.len; i++) text_texture_load(&socket_frame.textures[i], "");
+
 			state.update_ui = 0;
 		}
 
@@ -168,6 +185,7 @@ int handle_input(SDL_Event* event, state* state)
 			break;
 		case SDLK_2:
 			log_message(LOG_INFO, "Socket button 2 pressed");
+			net_print_peers();
 			break;
 		}
 		break;
@@ -357,13 +375,22 @@ void application_state_to_string(application_state state, char* string)
 	}
 }
 
+void* broadcast_t(void* args)
+{
+	while (1) {
+		net_broadcast();
+		sleep(1);
+	}
+}
+
 void* peer_discovery_t(void* args)
 {
 	state* state = args;
 	while (1) {
+		int added_client = net_listen_peers();
+		int pruned_client = net_prune_peers();
 		sem_wait(&state->sem);
-		net_listen_clients();
+		state->update_ui = added_client || pruned_client;
 		sem_post(&state->sem);
-		sleep(1);
 	}
 }

net.c

@@ -1,6 +1,8 @@
 #include "net.h"
 #include "log.h"
 #include <stdio.h>
+#include <stdlib.h>
+#include <time.h>
 
 typedef struct {
 	struct sockaddr_in addr;
@@ -8,6 +10,18 @@ typedef struct {
 	socklen_t addr_len;
 } net_info;
 
+typedef struct peer_addr_node {
+	struct sockaddr_in addr;
+	time_t time_last;
+	time_t time_elapsed;
+	struct peer_addr_node* next;
+} peer_addr_node;
+static peer_addr_node* peer_list = NULL;
+
+static int peer_list_add(struct sockaddr_in* addr);
+static int peer_list_exists(struct sockaddr_in* addr);
+static void peer_list_destroy();
+
 static net_info broadcast_info;
 static net_info listen_info;
 static net_info peer_info;
@@ -18,7 +32,7 @@ static int init = 0;
 int net_init(unsigned int port)
 {
 	PORT = port;
-	// broadcast socket init
+	// broadcast socket
 	broadcast_info.socket = socket(AF_INET, SOCK_DGRAM, 0);
 	if (broadcast_info.socket < 0) {
 		log_message(LOG_ERROR, "Error creating broadcast socket");
@@ -36,7 +50,7 @@ int net_init(unsigned int port)
 
 	broadcast_info.addr_len = sizeof(broadcast_info.addr);
 
-	// listen socket init
+	// listen socket
 	listen_info.socket = socket(AF_INET, SOCK_DGRAM, 0);
 	if (listen_info.socket < 0) {
 		log_message(LOG_ERROR, "Error creating listen socket", strerror(errno));
@@ -71,6 +85,14 @@ int net_init(unsigned int port)
 	return 1;
 }
 
+void net_destroy()
+{
+	close(broadcast_info.socket);
+	close(listen_info.socket);
+	close(peer_info.socket);
+	peer_list_destroy();
+}
+
 int net_broadcast()
 {
 	if (!init) return 0;;
@@ -88,59 +110,92 @@ int net_broadcast()
 	return 1;
 }
 
-int net_listen_clients()
+int net_listen_peers()
 {
 	if (!init) return 0;
 
-	if (peer_info.socket >= 0) {
+	net_info broadcast_listen;
-		log_message(LOG_WARNING, "Peer socket already created");
-		log_message(LOG_INFO,
-		            "Peer socket address:\n%s:%d",
-		            inet_ntoa(peer_info.addr.sin_addr),
-		            ntohs(peer_info.addr.sin_port));
-		return 1;
-	}
 
-	peer_info.socket = socket(AF_INET, SOCK_DGRAM, 0);
+	broadcast_listen.socket = socket(AF_INET, SOCK_DGRAM, 0);
-	if (peer_info.socket < 0) {
+	if (broadcast_listen.socket < 0) {
 		log_message(LOG_ERROR, "Error creating listen socket", strerror(errno));
-		close(peer_info.socket);
+		close(broadcast_listen.socket);
 		return 0;
 	}
 
 	char buffer[64];
 
-	memset(&peer_info.addr, 0, sizeof(peer_info.addr));
+	memset(&broadcast_listen.addr, 0, sizeof(broadcast_listen.addr));
-	peer_info.addr.sin_family = AF_INET;
+	broadcast_listen.addr.sin_family = AF_INET;
-	peer_info.addr.sin_addr.s_addr = INADDR_ANY;
+	broadcast_listen.addr.sin_addr.s_addr = INADDR_ANY;
-	peer_info.addr.sin_port = htons(PORT);
+	broadcast_listen.addr.sin_port = htons(PORT);
 
-	peer_info.addr_len = sizeof(peer_info.addr);
+	broadcast_listen.addr_len = sizeof(broadcast_listen.addr);
 
 	int retv = recvfrom(listen_info.socket, buffer, sizeof(buffer), 0,
-	                    (struct sockaddr*)&peer_info.addr,
+	                    (struct sockaddr*)&broadcast_listen.addr,
-	                    &peer_info.addr_len);
+	                    &broadcast_listen.addr_len);
+
+	close(broadcast_listen.socket);
 
 	if (retv < 0) {
+		if (errno != EAGAIN)
 			log_message(LOG_ERROR, "Failed to receive broadcast message: %s", strerror(errno));
-		close(peer_info.socket);
+		close(broadcast_listen.socket);
-		peer_info.socket = -1;
+		broadcast_listen.socket = -1;
 		return 0;
 	}
 
 	buffer[retv] = '\0';
+
 	log_message(LOG_INFO,
-	            "Broadcast message received from:\n%s:%d",
+	            "Broadcast message received from:\n%s:%d\n%s",
-	            inet_ntoa(peer_info.addr.sin_addr),
+	            inet_ntoa(broadcast_listen.addr.sin_addr),
-	            ntohs(peer_info.addr.sin_port));
+	            ntohs(broadcast_listen.addr.sin_port),
-	log_message(LOG_INFO, "%s", buffer);
+	            buffer);
+
+	retv = peer_list_exists(&broadcast_listen.addr);
+	if (retv) return 0;
+
+	peer_list_add(&broadcast_listen.addr);
 
 	return 1;
 }
 
-int net_send()
+int net_prune_peers()
 {
 	if (!init) return 0;
+	if (!peer_list) return 0;
+	int pruned = 0;
+	peer_addr_node* node = peer_list;
+	peer_addr_node* last_node = NULL;
+	while (node) {
+		peer_addr_node* next_node = node->next;
+
+		node->time_elapsed += time(NULL) - node->time_last;
+		node->time_last = time(NULL);
+
+		if (node->time_elapsed < CLIENT_MAX_TIME) {
+			last_node = node;
+			goto next;
+		}
+
+		pruned = 1;
+		free(node);
+		if (node == peer_list)
+			peer_list = next_node;
+		if (last_node != NULL)
+			last_node->next = next_node;
+next:
+		node = next_node;
+	}
+	return pruned;
+}
+
+int net_send(void* buffer, size_t size)
+{
+	if (!init) return 0;
+	if (peer_info.socket < 0) return 0;
 	return 1;
 }
 
@@ -149,3 +204,74 @@ int net_receive()
 	if (!init) return 0;
 	return 1;
 }
+
+void net_print_peers()
+{
+	if (!peer_list) return;
+	peer_addr_node* node = peer_list;
+	while (node) {
+		log_message(LOG_INFO,
+		            "Peers :\n%s:%d",
+		            inet_ntoa(node->addr.sin_addr),
+		            ntohs(node->addr.sin_port));
+		node = node->next;
+	}
+}
+
+int net_get_peers(struct sockaddr_in** peer_addr)
+{
+	if (!peer_list) return 0;
+	static peer_addr_node* node = NULL;
+	static int done = 0;
+
+	if (done) {
+		done = 0;
+		return 0;
+	}
+
+	if (!node) node = peer_list;
+	*peer_addr = &node->addr;
+
+	if (!node->next) done = 1;
+	node = node->next;
+	return 1;
+}
+
+static int peer_list_add(struct sockaddr_in* addr)
+{
+	peer_addr_node* new_node = malloc(sizeof(peer_addr_node));
+	size_t size = sizeof(struct sockaddr_in);
+	memset(&new_node->addr, 0, sizeof(size));
+	memcpy(&new_node->addr, addr, sizeof(size));
+	new_node->time_last = time(NULL);
+	new_node->time_elapsed = 0;
+	new_node->next = peer_list ? peer_list : NULL;
+	peer_list = new_node;
+	return 1;
+}
+
+static int peer_list_exists(struct sockaddr_in* addr)
+{
+	if (!peer_list) return 0;
+	peer_addr_node* node = peer_list;
+	while (node) {
+		int address_match = addr->sin_addr.s_addr == node->addr.sin_addr.s_addr;
+		int port_match = addr->sin_port == node->addr.sin_port;
+		if (address_match && port_match) {
+			node->time_elapsed = 0;
+			return 1;
+		}
+		node = node->next;
+	}
+	return 0;
+}
+
+static void peer_list_destroy()
+{
+	peer_addr_node* node = peer_list;
+	while (node) {
+		peer_addr_node* next_node = node->next;
+		free(node);
+		node = next_node;
+	}
+}

net.h

@@ -7,10 +7,16 @@
 #include <string.h>
 #include <ifaddrs.h>
 
+#define CLIENT_MAX_TIME 4
+
 int net_init(unsigned int port);
+void net_destroy();
 int net_broadcast();
-int net_listen_clients();
+int net_listen_peers();
-int net_send();
+int net_prune_peers();
+int net_send(void* buffer, size_t size);
 int net_receive();
+void net_print_peers();
+int net_get_peers(struct sockaddr_in** peer_addr);
 
 #endif

texture.c

@@ -63,14 +63,13 @@ int text_texture_render(text_texture* text_texture, int x, int y)
 	return 1;
 }
 
-int text_texture_destroy(text_texture* text_texture)
+void text_texture_destroy(text_texture* text_texture)
 {
 	if (!text_texture->texture) {
-		return 0;
+		return;
 	}
 	SDL_DestroyTexture(text_texture->texture);
 	TTF_CloseFont(text_texture->font);
-	return 1;
 }
 
 int text_texture_frame_init(text_texture_frame* text_texture_frame, int len, SDL_Renderer* renderer, int font_size)
@@ -78,6 +77,7 @@ int text_texture_frame_init(text_texture_frame* text_texture_frame, int len, SDL
 	text_texture_frame->textures = malloc(len*sizeof(text_texture));
 	for (int i = 0; i < len; i++) {
 		text_texture_init(&text_texture_frame->textures[i], renderer, font_size);
+		text_texture_load(&text_texture_frame->textures[i], "");
 	}
 	text_texture_frame->len = len;
 	return 1;
@@ -94,12 +94,11 @@ int text_texture_frame_render(text_texture_frame* text_texture_frame, int x, int
 	return 1;
 }
 
-int text_texture_frame_destroy(text_texture_frame* text_texture_frame)
+void text_texture_frame_destroy(text_texture_frame* text_texture_frame)
 {
 	for (int i = 0; i < text_texture_frame->len; i++) {
 		text_texture_destroy(&text_texture_frame->textures[i]);
 	}
 	free(text_texture_frame->textures);
 	text_texture_frame->textures = NULL;
-	return 1;
 }

texture.h

@@ -18,12 +18,12 @@ typedef struct {
 } text_texture_frame;
 
 int text_texture_init(text_texture* text_texture, SDL_Renderer* renderer, int font_size);
+void text_texture_destroy(text_texture* text_texture);
 int text_texture_load(text_texture* text_texture, char* string);
 int text_texture_render(text_texture* text_texture, int x, int y);
-int text_texture_destroy(text_texture* text_texture);
 
 int text_texture_frame_init(text_texture_frame* text_texture_frame, int len, SDL_Renderer* renderer, int font_size);
+void text_texture_frame_destroy(text_texture_frame* text_texture_frame);
 int text_texture_frame_render(text_texture_frame* text_texture_frame, int x, int y);
-int text_texture_frame_destroy(text_texture_frame* text_texture_frame);
 
 #endif