bugfix: windows does not support ngx_*_lua_ffi_worker_pids.

Signed-off-by: lijunlong <lijunlong@openresty.com>

Author: zhuizhuhaomeng

.travis.yml

@@ -1,6 +1,6 @@
 ---
 sudo: required
-dist: bionic
+dist: focal
 
 branches:
  only:

lib/resty/core/worker.lua

@@ -2,20 +2,25 @@
 
 
 local ffi = require "ffi"
+local jit = require "jit"
 local base = require "resty.core.base"
+local ffi_cast = ffi.cast
 
 
 local C = ffi.C
-local ffi_new = ffi.new
 local new_tab = base.new_tab
 local subsystem = ngx.config.subsystem
+local get_string_buf = base.get_string_buf
+local get_size_ptr = base.get_size_ptr
 
 
 local ngx_lua_ffi_worker_id
 local ngx_lua_ffi_worker_pid
 local ngx_lua_ffi_worker_pids
 local ngx_lua_ffi_worker_count
 local ngx_lua_ffi_worker_exiting
+local ffi_intp_type = ffi.typeof("int *")
+local ffi_int_size = ffi.sizeof("int")
 
 
 ngx.worker = new_tab(0, 4)
@@ -25,31 +30,42 @@ if subsystem == "http" then
  ffi.cdef[[
  int ngx_http_lua_ffi_worker_id(void);
  int ngx_http_lua_ffi_worker_pid(void);
- int ngx_http_lua_ffi_worker_pids(int *pids, size_t *pids_len);
  int ngx_http_lua_ffi_worker_count(void);
  int ngx_http_lua_ffi_worker_exiting(void);
  ]]
 
  ngx_lua_ffi_worker_id = C.ngx_http_lua_ffi_worker_id
  ngx_lua_ffi_worker_pid = C.ngx_http_lua_ffi_worker_pid
- ngx_lua_ffi_worker_pids = C.ngx_http_lua_ffi_worker_pids
  ngx_lua_ffi_worker_count = C.ngx_http_lua_ffi_worker_count
  ngx_lua_ffi_worker_exiting = C.ngx_http_lua_ffi_worker_exiting
+ if jit.os ~= "Windows" then
+ ffi.cdef[[
+ int ngx_http_lua_ffi_worker_pids(int *pids, size_t *pids_len);
+ ]]
+
+ ngx_lua_ffi_worker_pids = C.ngx_http_lua_ffi_worker_pids
+ end
 
 elseif subsystem == "stream" then
  ffi.cdef[[
  int ngx_stream_lua_ffi_worker_id(void);
  int ngx_stream_lua_ffi_worker_pid(void);
- int ngx_stream_lua_ffi_worker_pids(int *pids, size_t *pids_len);
  int ngx_stream_lua_ffi_worker_count(void);
  int ngx_stream_lua_ffi_worker_exiting(void);
  ]]
 
  ngx_lua_ffi_worker_id = C.ngx_stream_lua_ffi_worker_id
  ngx_lua_ffi_worker_pid = C.ngx_stream_lua_ffi_worker_pid
- ngx_lua_ffi_worker_pids = C.ngx_stream_lua_ffi_worker_pids
  ngx_lua_ffi_worker_count = C.ngx_stream_lua_ffi_worker_count
  ngx_lua_ffi_worker_exiting = C.ngx_stream_lua_ffi_worker_exiting
+
+ if jit.os ~= "Windows" then
+ ffi.cdef[[
+ int ngx_stream_lua_ffi_worker_pids(int *pids, size_t *pids_len);
+ ]]
+
+ ngx_lua_ffi_worker_pids = C.ngx_stream_lua_ffi_worker_pids
+ end
 end
 
 
@@ -62,23 +78,32 @@ function ngx.worker.pid()
  return ngx_lua_ffi_worker_pid()
 end
 
-local size_ptr = ffi_new("size_t[1]")
-local pids_ptr = ffi_new("int[1024]") -- using NGX_MAX_PROCESSES
 
-function ngx.worker.pids()
- if ngx.get_phase() == "init" or ngx.get_phase() == "init_worker" then
- return nil, "API disabled in the current context"
- end
+if jit.os ~= "Windows" then
+ function ngx.worker.pids()
+ if ngx.get_phase() == "init" or ngx.get_phase() == "init_worker" then
+ return nil, "API disabled in the current context"
+ end
+
+ local pids = {}
+ local size_ptr = get_size_ptr()
+ local worker_cnt = ngx.worker.count()
+ if worker_cnt == 0 then
+ return pids
+ end
 
- local res = ngx_lua_ffi_worker_pids(pids_ptr, size_ptr)
+ size_ptr[0] = worker_cnt
+ local pids_ptr = get_string_buf(worker_cnt * ffi_int_size)
+ local intp_buf = ffi_cast(ffi_intp_type, pids_ptr)
+ local res = ngx_lua_ffi_worker_pids(intp_buf, size_ptr)
 
- local pids = {}
- if res == 0 then
- for i = 1, tonumber(size_ptr[0]) do
- pids[i] = pids_ptr[i-1]
+  if res == 0 then
+  for i = 1, tonumber(size_ptr[0]) do
+  pids[i] = intp_buf[i-1]
+ end
  end
+ return pids
  end
- return pids
 end
 
 function ngx.worker.id()