idevicesyslog: fix reading from syslog_relay (simple socket connection)

1 files changed, 5 insertions, 25 deletions

diff --git a/tools/idevicesyslog.c b/tools/idevicesyslog.c
index 393c88e..227bf3d 100644
--- a/tools/idevicesyslog.c
+++ b/tools/idevicesyslog.c
@@ -116,36 +116,16 @@ int main(int argc, char *argv[])
                         printf("ERROR: Could not open usbmux connection.\n");
                 } else {
                         while (!quit_flag) {
-                                char *receive = NULL;
-                                uint32_t datalen = 0, bytes = 0, recv_bytes = 0;
-
-                                ret = idevice_connection_receive(conn, (char *) &datalen, sizeof(datalen), &bytes);
-                                if (ret < 0) {
+                                char c;
+                                uint32_t bytes = 0;
+                                if ((idevice_connection_receive(conn, &c, 1, &bytes) < 0) || (bytes != 1)) {
                                         fprintf(stderr, "Error receiving data. Exiting...\n");
                                         break;
                                 }
-
-                                datalen = be32toh(datalen);
-
-                                if (datalen == 0)
-                                        continue;
-
-                                recv_bytes += bytes;
-                                receive = (char *) malloc(sizeof(char) * datalen);
-
-                                while (!quit_flag && (recv_bytes <= datalen)) {
-                                        ret = idevice_connection_receive(conn, receive, datalen, &bytes);
-
-                                        if (bytes == 0)
-                                                break;
-
-                                        recv_bytes += bytes;
-
-                                        fwrite(receive, sizeof(char), bytes, stdout);
+                                if (c != 0) {
+                                        putchar(c);
                                         fflush(stdout);
                                 }
-
-                                free(receive);
                         }
                 }
                 idevice_disconnect(conn);