Lua uvFifoListener ffi

src/lua/fileffi.lua

@@ -20,6 +20,7 @@ ffi.cdef [[
 typedef struct { char opaque[?]; } LuaFile;
 typedef struct { uint32_t size; uint8_t data[?]; } LuaBuffer;
 typedef struct { char opaque[?]; } LuaSlice;
+typedef struct { char opaque[?]; } LuaServer;
 
 enum FileOps {
     READ,
@@ -49,6 +50,11 @@ LuaFile* subFile(LuaFile*, uint64_t start, int64_t size);
 
 LuaFile* uvFifo(const char* address, int port);
 
+LuaServer* uvFifoListener();
+
+void stopListener(LuaServer* server);
+void startListener(LuaServer* server, unsigned port, void (*cb)(LuaFile* fifo));
+
 void closeFile(LuaFile* wrapper);
 
 uint64_t readFileRawPtr(LuaFile* wrapper, void* dst, uint64_t size);
@@ -339,6 +345,40 @@ local function createFileWrapper(wrapper)
     return file
 end
 
+local function createUvListener(port, cb)
+    if type(port) ~= 'number' then error 'port argument must be an unsigned number' end
+    if (port) == nil then error 'must provide a port' end
+    if cb == nil then error 'must provide a callback function' end
+
+    _callback = function(fifo)  end
+
+    if cb ~= nil then
+        if type(cb) ~= 'function' then error 'callback must be a function' end
+        _callback = function(fifo)
+            file = createFileWrapper(fifo)
+            cb(file)
+        end
+    end
+
+    _callback = ffi.cast('void (*)(LuaFile* fifo)', _callback)
+
+    local wrapper = C.uvFifoListener()
+
+    local listener = {
+        _type = "LuaServer",
+        _wrapper = wrapper,
+        _proxy = newproxy(),
+        _cb = _callback,
+        _port = port,
+        startListener = function(listener) C.startListener(listener._wrapper, listener._port, listener._cb)  end,
+        stopListener = function(listener) C.stopListener(listener._wrapper) end,
+    }
+    -- Use a proxy instead of doing this on the wrapper directly using ffi.gc, because of a bug in LuaJIT,
+    -- where circular references on finalizers using ffi.gc won't actually collect anything.
+    debug.setmetatable(listener._proxy, { __gc = function()  C.stopListener(listener._wrapper) end })
+    return listener
+end
+
 local function open(filename, t)
     if (t == nil) then t = 'READ' end
     if (t == 'DOWNLOAD_URL_AND_WAIT') then
@@ -395,6 +435,10 @@ local function uvFifo(address, port)
     return createFileWrapper(C.uvFifo(address, port))
 end
 
+local function uvFifoListener(port, callback)
+    return createUvListener(port, callback)
+end
+
 if (type(Support) ~= 'table') then Support = {} end
 
 Support.NewLuaBuffer = function(size)
@@ -408,6 +452,8 @@ Support.File = {
     buffer = buffer,
     zReader = zReader,
     uvFifo = uvFifo,
+    uvFifoListener = uvFifoListener,
+    _createUvListener = createUvListener,
     _createFileWrapper = createFileWrapper,
     _createSliceWrapper = createSliceWrapper,
 }

src/lua/luafile.cc

@@ -27,6 +27,7 @@
 namespace {
 
 using LuaFile = PCSX::LuaFFI::LuaFile;
+using LuaServer = PCSX::LuaFFI::LuaServer;
 
 enum FileOps {
     READ,
@@ -73,6 +74,22 @@ LuaFile* subFile(LuaFile* wrapper, uint64_t start, int64_t size) {
 }
 LuaFile* uvFifo(const char* address, int port) { return new LuaFile(new PCSX::UvFifo(address, port)); }
 
+LuaServer* uvFifoListener() {return new LuaServer(new PCSX::UvFifoListener());}
+
+void startListener(LuaServer* server, unsigned port, void (*cb)(LuaFile* fifo)) { server->m_listener->start(port, PCSX::g_system->getLoop(), &server->m_async, [cb, server](PCSX::UvFifo* fifo) {
+        if (fifo) {
+            cb(new LuaFile(fifo));
+        } else {
+            server->m_async.data = server;
+            uv_close(reinterpret_cast<uv_handle_t*>(&server->m_async), [](uv_handle_t* handle) {});
+        }
+    });
+}
+
+void stopListener(LuaServer* server) {
+   server->m_listener->stop();
+}
+
 void closeFile(LuaFile* wrapper) { wrapper->file->close(); }
 
 uint64_t readFileRawPtr(LuaFile* wrapper, void* dst, uint64_t size) { return wrapper->file->read(dst, size); }
@@ -194,6 +211,9 @@ static void registerAllSymbols(PCSX::Lua L) {
     REGISTER(L, bufferFileEmpty);
     REGISTER(L, subFile);
     REGISTER(L, uvFifo);
+    REGISTER(L, uvFifoListener);
+    REGISTER(L, stopListener);
+    REGISTER(L, startListener);
 
     REGISTER(L, closeFile);
 

src/lua/luafile.h

@@ -23,6 +23,7 @@
 
 #include "lua/luawrapper.h"
 #include "support/file.h"
+#include "support/uvfile.h"
 
 namespace PCSX {
 
@@ -52,6 +53,12 @@ struct LuaFile {
     IO<File> file;
 };
 
+struct LuaServer {
+    LuaServer(UvFifoListener* listener) : m_listener(listener) {}
+    UvFifoListener* m_listener = nullptr;
+    uv_async_t m_async;
+};
+
 void open_file(Lua);
 }  // namespace LuaFFI