Merge pull request #46 from moteus/master

Support `socket_action` interface.

Author: moteus

doc/lcurl.ldoc

@@ -24,6 +24,13 @@ function easy() end
 --
 -- @tparam[opt] table options
 -- @treturn[1] multi new curl multi object
+--
+-- @usage
+-- m = curl.multi{
+--   socketfunction = handle_socket;
+--   timerfunction  = start_timeout;
+-- }
+--
 function multi() end
 
 --- Create Share object
@@ -433,6 +440,18 @@ function info_read() end
 -- c:setopt{maxconnects = 10}
 function setopt() end
 
+--- Perform socket action.
+--
+-- @tparam[opt=curl.SOCKET_TIMEOUT] number socket
+-- @tparam[opt=0] number mask
+-- @treturn multi self
+--
+-- @usage
+-- c:socket_action()
+-- c:socket_action(sock_fd, curl.CSELECT_IN)
+-- c:socket_action(sock_fd, curl.CSELECT_OUT)
+function socket_action() end
+
 --- Set timer callback.
 --
 -- @tparam function timer timer callback

examples/cURLv3/multi-uv.lua

@@ -0,0 +1,162 @@
+local curl = require "cURL"
+local uv   = require "lluv"
+
+local fprintf = function(f, ...) f:write((string.format(...))) end
+local printf  = function(...) fprintf(io.stdout, ...) end
+
+local stderr = io.stderr
+
+local timeout, curl_handle
+
+local ACTION_NAMES = {
+  [curl.POLL_IN     ] = "POLL_IN";
+  [curl.POLL_INOUT  ] = "POLL_INOUT";
+  [curl.POLL_OUT    ] = "POLL_OUT";
+  [curl.POLL_NONE   ] = "POLL_NONE";
+  [curl.POLL_REMOVE ] = "POLL_REMOVE";
+}
+local EVENT_NAMES = {
+  [ uv.READABLE ] = "READABLE";
+  [ uv.WRITABLE ] = "WRITABLE";
+  [ uv.READABLE + uv.WRITABLE ] = "READABLE + WRITABLE";
+}
+local FLAGS = {
+  [ uv.READABLE ]               = curl.CSELECT_IN;
+  [ uv.WRITABLE ]               = curl.CSELECT_OUT;
+  [ uv.READABLE + uv.WRITABLE ] = curl.CSELECT_IN + curl.CSELECT_OUT;
+
+}
+
+local trace = function() end or print
+
+local FILES, CONTEXT = {}, {}
+
+function create_curl_context(sockfd)
+  local context = {
+    sockfd      = sockfd;
+    poll_handle = uv.poll_socket(sockfd);
+  }
+  context.poll_handle.data = context
+  
+  return context
+end
+
+function destroy_curl_context(context)
+  context.poll_handle:close()
+end 
+
+function add_download(url, num)
+  local filename = tostring(num) ..  ".download"
+  local file = io.open(filename, "w")
+  if not file then
+    fprintf(stderr, "Error opening %s\n", filename)
+    return
+  end
+
+  local handle = curl.easy{
+    url           = url;
+    writefunction = file;
+  }
+
+  FILES[handle] = file
+
+  curl_handle:add_handle(handle)
+  fprintf(stderr, "Added download %s -> %s\n", url, filename);
+end
+
+function check_multi_info()
+  while true do
+    local easy, ok, err = curl_handle:info_read()
+    if not easy then curl_handle:close() error(err) end
+    if easy == 0 then break end
+
+    local context = CONTEXT[e]
+    if context then destroy_curl_context(context) end
+    local file = FILES[easy]
+    if file then FILES[easy] = nil, file:close() end
+    local done_url = easy:getinfo_effective_url()
+    easy:close()
+    if ok then
+      printf("%s DONE\n", done_url);
+    elseif data == "error" then
+      printf("%s ERROR - %s\n", done_url, tostring(err));
+    end
+  end
+end
+
+function curl_perform(handle, err, events)
+  -- calls by libuv --
+  trace("UV::POLL", handle, err, EVENT_NAMES[events] or events)
+
+  local flags = assert(FLAGS[events], ("unknown event:" .. events))
+
+  context = handle.data
+
+  curl_handle:socket_action(context.sockfd, flags)
+
+  check_multi_info()
+end
+
+function on_timeout(timer)
+  -- calls by libuv --
+  trace("UV::TIMEOUT", timer)
+
+  local running_handles, err = curl_handle:socket_action()
+
+  check_multi_info()
+end
+
+function start_timeout(timeout_ms)
+  -- calls by curl --
+  trace("CURL::TIMEOUT", timeout_ms)
+
+  -- 0 means directly call socket_action, but we'll do it in a bit
+  if timeout_ms <= 0 then timeout_ms = 1 end
+
+  timeout:stop():start(timeout_ms, 0, on_timeout)
+end
+
+local handle_socket = function(...)
+  local ok, err = pcall(handle_socket_impl, ...)
+  if not ok then uv.defer(function() error(err) end) end
+end
+
+function handle_socket_impl(easy, s, action)
+  -- calls by curl --
+
+  trace("CURL::SOCKET", easy, s, ACTION_NAMES[action] or action)
+
+  local curl_context = CONTEXT[easy] or create_curl_context(s)
+  CONTEXT[easy] = curl_context
+
+  assert(curl_context.sockfd == s)
+
+  if action == curl.POLL_IN then
+    curl_context.poll_handle:start(uv.READABLE, curl_perform)
+  elseif action == curl.POLL_OUT then
+    curl_context.poll_handle:start(uv.WRITABLE, curl_perform)
+  elseif action == curl.POLL_REMOVE then
+    CONTEXT[easy] = nil
+    destroy_curl_context(curl_context)
+  end
+end
+
+timeout = uv.timer()
+
+curl_handle = curl.multi{
+  socketfunction = handle_socket;
+  timerfunction  = start_timeout;
+}
+
+curl_handle = curl.multi{
+  socketfunction = handle_socket;
+  timerfunction  = start_timeout;
+}
+
+for i = 1, math.huge do
+  local url = arg[i]
+  if not url then break end
+  add_download(url, i)
+end
+
+uv.run(loop, UV_RUN_DEFAULT)

src/l52util.c

@@ -155,3 +155,27 @@ void *lutil_newudatap_impl(lua_State *L, size_t size, const void *p){
   lutil_setmetatablep(L, p);
   return obj;
 }
+
+void lutil_pushint64(lua_State *L, int64_t v){
+  if(sizeof(lua_Integer) >= sizeof(int64_t)){
+    lua_pushinteger(L, (lua_Integer)v);
+    return;
+  }
+  lua_pushnumber(L, (lua_Number)v);
+}
+
+int64_t lutil_checkint64(lua_State *L, int idx){
+  if(sizeof(lua_Integer) >= sizeof(int64_t))
+    return luaL_checkinteger(L, idx);
+  return (int64_t)luaL_checknumber(L, idx);
+}
+
+int64_t lutil_optint64(lua_State *L, int idx, int64_t v){
+  if(sizeof(lua_Integer) >= sizeof(int64_t))
+    return luaL_optinteger(L, idx, v);
+  return (int64_t)luaL_optnumber(L, idx, v);
+}
+
+void lutil_pushnvalues(lua_State *L, int n){
+  for(;n;--n) lua_pushvalue(L, -n);
+}

src/l52util.h

@@ -13,47 +13,52 @@
 
 #include "lua.h"
 #include "lauxlib.h"
+#include <stdint.h>
 
 #if LUA_VERSION_NUM >= 503 /* Lua 5.3 */
 
-#ifndef luaL_optint
-# define luaL_optint luaL_optinteger
-#endif
-
 #ifndef luaL_checkint
-# define luaL_checkint luaL_checkinteger
+#define luaL_checkint  luaL_checkinteger
 #endif
 
 #ifndef luaL_checklong
-# define luaL_checklong luaL_checkinteger
+#define luaL_checklong  luaL_checkinteger
 #endif
 
+#ifndef luaL_optint
+#define luaL_optint luaL_optinteger
 #endif
 
+#ifndef luaL_optlong
+#define luaL_optlong luaL_optinteger
+#endif
 
-#if LUA_VERSION_NUM >= 502 // lua 5.2
+#endif
 
-// lua_rawgetp
-// lua_rawsetp
-// luaL_setfuncs
-// lua_absindex
-#ifndef lua_objlen
+#if LUA_VERSION_NUM >= 502 /* Lua 5.2 */
 
-#define lua_objlen      lua_rawlen
+/* lua_rawgetp */
+/* lua_rawsetp */
+/* luaL_setfuncs */
+/* lua_absindex */
 
+#ifndef lua_objlen
+#define lua_objlen      lua_rawlen
 #endif
 
 int   luaL_typerror (lua_State *L, int narg, const char *tname);
 
 #ifndef luaL_register
-
 void luaL_register (lua_State *L, const char *libname, const luaL_Reg *l);
+#endif
 
+#ifndef lua_equal
+#define lua_equal(L,idx1,idx2) lua_compare(L,(idx1),(idx2),LUA_OPEQ)
 #endif
 
-#else                      // lua 5.1
+#else                      /* Lua 5.1 */
 
-// functions form lua 5.2
+/* functions form lua 5.2 */
 
 # define lua_absindex(L, i) (((i)>0)?(i):((i)<=LUA_REGISTRYINDEX?(i):(lua_gettop(L)+(i)+1)))
 # define lua_rawlen  lua_objlen
@@ -78,5 +83,12 @@ int   lutil_createmetap   (lua_State *L, const void *p, const luaL_Reg *methods,
 
 void *lutil_newudatap_impl     (lua_State *L, size_t size, const void *p);
 
-#endif
+void lutil_pushint64(lua_State *L, int64_t v);
 
+int64_t lutil_checkint64(lua_State *L, int idx);
+
+int64_t lutil_optint64(lua_State *L, int idx, int64_t v);
+
+void lutil_pushnvalues(lua_State *L, int n);
+
+#endif