From 9e56ad94589054428e5ce4de86f75840cffc88ad Mon Sep 17 00:00:00 2001
From: Vanya Sergeev <vsergeev@gmail.com>
Date: Tue, 5 Jul 2016 03:13:46 -0700
Subject: [PATCH] blocks/sources/rtlsdr: add sigterm handler to allow device
 close

the sigterm handlers cancel's the async read, allowing device close
after rtlsdr_read_async() returns.

also rename read_async_callback to read_callback for less ambiguity
between it and the async.callback() helper function.

resolves #8.
---
 radio/blocks/sources/rtlsdr.lua | 35 +++++++++++++++++++++++++++++----
 1 file changed, 31 insertions(+), 4 deletions(-)

diff --git a/radio/blocks/sources/rtlsdr.lua b/radio/blocks/sources/rtlsdr.lua
index bfb4cd110..d1e951565 100644
--- a/radio/blocks/sources/rtlsdr.lua
+++ b/radio/blocks/sources/rtlsdr.lua
@@ -29,6 +29,7 @@ local ffi = require('ffi')
 local block = require('radio.core.block')
 local platform = require('radio.core.platform')
 local types = require('radio.types')
+local async = require('radio.core.async')
 
 local RtlSdrSource = block.factory("RtlSdrSource")
 
@@ -172,10 +173,26 @@ function RtlSdrSource:initialize_rtlsdr()
     end
 end
 
-local function read_async_callback_factory(pipes)
+local function sigterm_handler_factory(dev)
+    local ffi = require('ffi')
+
+    ffi.cdef[[
+        typedef struct rtlsdr_dev rtlsdr_dev_t;
+        int rtlsdr_cancel_async(rtlsdr_dev_t *dev);
+    ]]
+    local librtlsdr = ffi.load('rtlsdr')
+
+    local function sigterm_handler(sig)
+        librtlsdr.rtlsdr_cancel_async(ffi.cast('rtlsdr_dev_t *', dev))
+    end
+
+    return ffi.cast('void (*)(int)', sigterm_handler)
+end
+
+local function read_callback_factory(pipes)
     local out = types.ComplexFloat32.vector()
 
-    local function read_async_callback(buf, len, ctx)
+    local function read_callback(buf, len, ctx)
         -- Size output vector
         out:resize(len/2)
 
@@ -191,18 +208,28 @@ local function read_async_callback_factory(pipes)
         end
     end
 
-    return read_async_callback
+    return read_callback
 end
 
 function RtlSdrSource:run()
     -- Initialize the rtlsdr in our own running process
     self:initialize_rtlsdr()
 
+    -- Register SIGTERM signal handler
+    local handler, handler_state = async.callback(sigterm_handler_factory, tonumber(ffi.cast("intptr_t", self.dev[0])))
+    ffi.C.signal(ffi.C.SIGTERM, handler)
+
     -- Start asynchronous read
-    local ret = librtlsdr.rtlsdr_read_async(self.dev[0], read_async_callback_factory(self.outputs[1].pipes), nil, 0, 32768)
+    local ret = librtlsdr.rtlsdr_read_async(self.dev[0], read_callback_factory(self.outputs[1].pipes), nil, 0, 32768)
     if ret ~= 0 then
         error("rtlsdr_read_async(): " .. tostring(ret))
     end
+
+    -- Close rtlsdr
+    ret = librtlsdr.rtlsdr_close(self.dev[0])
+    if ret ~= 0 then
+        error("rtlsdr_close(): " .. tostring(ret))
+    end
 end
 
 return RtlSdrSource