File: manage_pool_members.lua

package info (click to toggle)
ntopng 5.2.1%2Bdfsg1-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 121,832 kB
  • sloc: javascript: 143,431; cpp: 71,175; ansic: 11,108; sh: 4,687; makefile: 911; python: 587; sql: 512; pascal: 234; perl: 118; ruby: 52; exp: 4
file content (66 lines) | stat: -rw-r--r-- 1,861 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
--
-- (C) 2013-22 - ntop.org
--

local dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
package.path = dirs.installdir .. "/scripts/lua/modules/pools/?.lua;" .. package.path

require "lua_utils"
local host_pools = require "host_pools"
local json = require "dkjson"

sendHTTPContentTypeHeader('application/json')

--[[ EXAMPLE PAYLOAD:

   local ret = {associations = {
   ["DE:AD:BE:EE:FF:FF"] = {group = "maina", connectivity = "pass"},
   ["AB:AB:AB:AB:AB:AB"] = {group = "simon", connectivity = "reject"}}
   }
   return ret
--]]

-- Instantiate host pools
local s = host_pools:create()

local r = {}

local pools_list = {}

-- Table with pool names as keys
for _, pool in pairs(s:get_all_pools()) do
   pools_list[pool["name"]] = pool
end

local res = {associations = _POST["associations"]}

for member, info in pairs(_POST["associations"] or {}) do
   local pool = info["group"]
   local connectivity = info["connectivity"]

   if pools_list[pool] == nil then
      res["associations"][member]["status"] = "ERROR"
      res["associations"][member]["status_msg"] = "Unable to find a group with the specified name"
   else
      local pool_id = pools_list[pool]["pool_id"]
      if connectivity == "pass" then
	 if s:bind_member(member, pool_id) == true then
	    res["associations"][member]["status"] = "OK"
	 end
      elseif info["connectivity"] == "reject" then
	 s:bind_member(member, host_pools.DEFAULT_POOL_ID)
	 res["associations"][member]["status"] = "OK"
      else
	 res["associations"][member]["status"] = "ERROR"
	 res["associations"][member]["status_msg"] = "Unknown association: allowed associations are 'pass' and 'reject'"
      end
   end

end

-- Formerly an array with interfaces as keys. Now that pools are global, placeholder "_all_" is used
r["_all_"] = res


print(json.encode(r))