new stuff

4 files changed, 29 insertions, 13 deletions

diff --git a/kernel/fs/fifo.c b/kernel/fs/fifo.c
index 7c286db..f2ad933 100644
--- a/kernel/fs/fifo.c
+++ b/kernel/fs/fifo.c
@@ -9,11 +9,17 @@ void fifo_close(vfs_fd_t *fd) {
   return;
 }
 
+void fifo_realloc(FIFO_FILE *file) {
+  file->buffer_len += 4096;
+  file->buffer = krealloc(file->buffer, file->buffer_len);
+}
+
 int fifo_object_write(u8 *buffer, u64 offset, u64 len, FIFO_FILE *file) {
   (void)offset;
   file->has_data = 1;
   if (file->write_len + len >= file->buffer_len) {
     file->can_write = 0;
+    fifo_realloc(file);
     return -EAGAIN;
   }
   memcpy(file->buffer + file->write_len, buffer, len);
diff --git a/kernel/fs/vfs.c b/kernel/fs/vfs.c
index f2b763c..505d6ae 100644
--- a/kernel/fs/vfs.c
+++ b/kernel/fs/vfs.c
@@ -375,7 +375,23 @@ int vfs_pwrite(int fd, void *buf, u64 count, u64 offset) {
   if (!(vfs_fd->flags & O_WRITE)) {
     return -EBADF;
   }
-  return raw_vfs_pwrite(vfs_fd, buf, count, offset);
+  int rc = raw_vfs_pwrite(vfs_fd, buf, count, offset);
+  if (-EAGAIN == rc) {
+    if (!(vfs_fd->flags & O_NONBLOCK)) {
+      struct pollfd fds;
+      do {
+        fds.fd = fd;
+        fds.events = POLLOUT;
+        fds.revents = 0;
+        int rc = poll(&fds, 1, 0);
+        if (-EINTR == rc) {
+          return -EINTR;
+        }
+      } while (!(fds.revents & POLLOUT));
+      return vfs_pwrite(fd, buf, count, offset);
+    }
+  }
+  return rc;
 }
 
 vfs_vm_object_t *vfs_get_vm_object(int fd, u64 length, u64 offset) {
diff --git a/kernel/network/tcp.c b/kernel/network/tcp.c
index 1a3641b..997b699 100644
--- a/kernel/network/tcp.c
+++ b/kernel/network/tcp.c
@@ -218,6 +218,7 @@ void handle_tcp(ipv4_t src_ip, const u8 *payload, u32 payload_length) {
           (u8 *)(payload + header->data_offset * sizeof(u32)), 0,
           tcp_payload_length, incoming_connection->data_file);
       assert(len >= 0);
+      assert(len == tcp_payload_length);
       incoming_connection->ack += len;
       tcp_send_ack(incoming_connection);
     }
diff --git a/kernel/socket.c b/kernel/socket.c
index 36d0fcb..39775ce 100644
--- a/kernel/socket.c
+++ b/kernel/socket.c
@@ -9,12 +9,8 @@
 #include <sched/scheduler.h>
 #include <socket.h>
 
-// FIXME: Make these more dynamic
 OPEN_UNIX_SOCKET *un_sockets[100] = {0};
 
-// struct TcpConnection *tcp_sockets[100];
-// struct TcpListen *tcp_listen[100];
-
 void gen_ipv4(ipv4_t *ip, u8 i1, u8 i2, u8 i3, u8 i4) {
   ip->a[0] = i1;
   ip->a[1] = i2;
@@ -195,14 +191,11 @@ int tcp_read(u32 socket, u8 *buffer, u64 buffer_size, u64 *out) {
     return 0;
   }
 
-  int rc = 0;
-  for (; rc <= 0;) {
-    rc = fifo_object_read(buffer, 0, buffer_size, con->data_file);
-    if (rc <= 0) {
-      enable_interrupts();
-      rc = 0;
-      return 0;
-    }
+  int rc = fifo_object_read(buffer, 0, buffer_size, con->data_file);
+  if (rc <= 0) {
+    enable_interrupts();
+    rc = 0;
+    return 0;
   }
   if (out) {
     *out = rc;