added low-level ioLib.available for io_stream's

res/modules/internal/stream_providers/bytearray.lua

@@ -102,6 +102,15 @@ function lib.flush(descriptor)
     requireDescriptor(descriptor)
 end
 
+function lib.available(descriptor)
+    requireDescriptor(descriptor)
+
+    local buf = buffers[descriptor]
+    local pos = positions[descriptor]
+
+    return math.max(#buf - pos + 1, 0)
+end
+
 function lib.is_alive(descriptor)
     return buffers[descriptor] ~= nil
 end

res/modules/internal/stream_providers/file.lua

@@ -5,6 +5,7 @@ local lib = {
     write = file.__write_descriptor,
     seek = file.__seek_descriptor,
     flush = file.__flush_descriptor,
+    available = file.__available_descriptor,
     is_alive = file.__has_descriptor,
     close = file.__close_descriptor
 }

res/modules/internal/stream_providers/named_pipe_unix.lua

@@ -9,6 +9,7 @@ int close(int fd);
 ssize_t read(int fd, void *buf, size_t count);
 ssize_t write(int fd, const void *buf, size_t count);
 int fcntl(int fd, int cmd, ...);
+int ioctl(int fd, unsigned long request, ...);
 
 const char *strerror(int errnum);
 ]]
@@ -20,6 +21,7 @@ local O_WRONLY   = 0x1
 local O_RDWR     = 0x2
 local O_NONBLOCK = 0x800
 local F_GETFL    = 3
+local FIONREAD   = 0x541B
 
 local function getError()
     local err = FFI.errno()
@@ -66,6 +68,18 @@ function lib.flush(fd)
     -- no flush on unix
 end
 
+function lib.available(fd)
+    if fd == nil or fd < 0 then return 0 end
+
+    local bytes_ready = FFI.new("int[1]", 0)
+
+    if C.ioctl(fd, FIONREAD, bytes_ready) == -1 then
+        return 0
+    end
+
+    return tonumber(bytes_ready[0])
+end
+
 function lib.is_alive(fd)
     if fd == nil or fd < 0 then return false end
 

res/modules/internal/stream_providers/named_pipe_windows.lua

@@ -105,6 +105,22 @@ function lib.flush(handle)
     C.FlushFileBuffers(handle)
 end
 
+function lib.available(handle)
+    if handle == nil or handle == INVALID_HANDLE_VALUE then
+        return 0
+    end
+
+    local bytes_available = FFI.new("DWORD[1]")
+
+    local success = C.PeekNamedPipe(handle, nil, 0, nil, bytes_available, nil)
+
+    if success == 0 then
+        return 0
+    end
+
+    return tonumber(bytes_available[0])
+end
+
 function lib.is_alive(handle)
     if handle == nil or handle == INVALID_HANDLE_VALUE then
         return false

res/modules/io_stream.lua

@@ -42,14 +42,17 @@ local function readFully(result, readFunc)
 end
 
 --[[
-
 descriptor - descriptor of stream for provided I/O library
 binaryMode - if enabled, most methods will expect bytes instead of strings
 ioLib - I/O library. Should include the following functions:
     read(descriptor: int, length: int) -> Bytearray
         May return bytearray with a smaller size if bytes have not arrived yet or have run out
     write(descriptor: int, data: Bytearray)
+    seek(descriptor: int, mode: string, offset: int)
+        Mode may be 'b' (relative begin), 'c' (relative current), 'e' (relative end + 1)
     flush(descriptor: int)
+    available(descriptor: int) -> int
+        May return 0 if environment is not support available method
     is_alive(descriptor: int) -> bool
     close(descriptor: int)
 --]]
@@ -130,7 +133,7 @@ function io_stream:available(length)
         else
             return #self.readBuffer >= length
         end
-    end
+    else return self.ioLib.available(self.descriptor) end
 end
 
 function io_stream:__update_read_buffer()