Use GDataInputStream to read data in the control and command services.

1 files changed, 16 insertions, 26 deletions

diff --git a/command_service.c b/command_service.c
index 5ef7fed..1174687 100644
--- a/command_service.c
+++ b/command_service.c
@@ -11,38 +11,28 @@ static gboolean service_incoming(GSocketService *service,
 		gpointer user_data) {
 	GError *error = NULL;
 	GSocket *socket = g_socket_connection_get_socket(connection);
-	gchar buffer[0x400];
+	GDataInputStream *input = g_data_input_stream_new(g_io_stream_get_input_stream((GIOStream*)connection));
+	g_data_input_stream_set_newline_type(input, G_DATA_STREAM_NEWLINE_TYPE_LF);
+	gchar *line;
+	gsize size;
 
-	while(g_socket_is_connected(socket) == TRUE) {
-		gssize size = g_socket_receive(socket, buffer, 0x400, NULL, &error);
+	while(g_socket_is_connected(socket) == TRUE &&
+			(line = g_data_input_stream_read_line(input, &size, NULL, &error)) != NULL) {
+		commands_handle(connection, line, &error);
 
-		if(size < 0) {
+		if(error != NULL) {
 			g_warning(error->message);
+			gchar *error_buffer = g_strdup_printf("%s\n", error->message);
+			g_socket_send(socket, error_buffer, strlen(error_buffer), NULL, NULL);
+			g_free(error_buffer);
 			g_error_free(error);
-			return FALSE;
-		} else if(size == 0) {
-			break;
+			error = NULL;
 		}
+	}
 
-		gchar *pos, *eol;
-		pos = buffer;
-		while((eol = g_strstr_len(pos, size, "\n")) != NULL || (eol = g_strstr_len(pos, size, "\r")) != NULL) {
-			*eol = '\0';
-
-			commands_handle(connection, pos, &error);
-
-			if(error != NULL) {
-				g_warning(error->message);
-				gchar *error_buffer = g_strdup_printf("%s\n", error->message);
-				g_socket_send(socket, error_buffer, strlen(error_buffer), NULL, NULL);
-				g_free(error_buffer);
-				g_error_free(error);
-				error = NULL;
-			}
-
-			size -= eol - pos + 1;
-			pos = eol + 1;
-		}
+	if(error != NULL) {
+		g_warning(error->message);
+		g_error_free(error);
 	}
 
 	return FALSE;