flm01/mote/v2/openwrt/package/flukso/luasrc/heartbeat.lua

162 lines
5.0 KiB
Lua
Executable File

#!/usr/bin/env lua
--[[
heartbeat.lua - send a heartbeat to the flukso server
Copyright (C) 2008-2009 jokamajo.org
2011 Bart Van Der Meerssche <bart.vandermeerssche@flukso.net>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
]]--
if not arg[1] then
print ('Please pass the reset argument as a boolean to the script.')
os.exit(1)
end
local dbg = require 'dbg'
local nixio = require 'nixio'
nixio.fs = require 'nixio.fs'
local uci = require 'luci.model.uci'.cursor()
local luci = require 'luci'
luci.sys = require 'luci.sys'
luci.json = require 'luci.json'
local httpclient = require 'luci.httpclient'
-- parse and load /etc/config/flukso
local FLUKSO = uci:get_all('flukso')
-- WAN settings
local WAN_ENABLED = true
if tonumber(FLUKSO.daemon.enable_wan_branch) == 0 then WAN_ENABLED = false end
local WAN_BASE_URL = FLUKSO.daemon.wan_base_url .. 'device/'
local WAN_KEY = '0123456789abcdef0123456789abcdef'
uci:foreach('system', 'system', function(x) WAN_KEY = x.key end) -- quirky but it works
local DEVICE = '0123456789abcdef0123456789abcdef'
uci:foreach('system', 'system', function(x) DEVICE = x.device end)
local UPGRADE_URL = FLUKSO.daemon.upgrade_url
-- https header helpers
local FLUKSO_VERSION = '000'
uci:foreach('system', 'system', function(x) FLUKSO_VERSION = x.version end)
local USER_AGENT = 'Fluksometer v' .. FLUKSO_VERSION
local CACERT = FLUKSO.daemon.cacert
-- collect relevant monitoring points
function collect_mp()
local monitor = {}
monitor.reset = tonumber(arg[1])
monitor.version = tonumber(FLUKSO_VERSION)
monitor.uptime = math.floor(luci.sys.uptime())
system, model, monitor.memtotal, monitor.memcached, monitor.membuffers, monitor.memfree = luci.sys.sysinfo()
return monitor
end
-- terminate when WAN reporting is not set
if not WAN_ENABLED then
os.exit(2)
end
-- open the connection to the syslog deamon, specifying our identity
nixio.openlog('heartbeat', 'pid')
local monitor = collect_mp()
local monitor_json = luci.json.encode(monitor)
-- phone home
local headers = {}
headers['Content-Type'] = 'application/json'
headers['X-Version'] = '1.0'
headers['User-Agent'] = USER_AGENT
headers['Connection'] = 'close'
local options = {}
options.sndtimeo = 5
options.rcvtimeo = 5
-- We don't enable peer cert verification so we can still update/upgrade
-- the Fluksometer via the heartbeat call even when the cacert has expired.
-- Disabling validation does mean that the server has to include an hmac
-- digest in the reply that the Fluksometer needs to verify, this to prevent
-- man-in-the-middle attacks.
options.tls_context_set_verify = 'none'
-- options.cacert = CACERT
options.method = 'POST'
options.headers = headers
options.body = luci.json.encode(monitor)
local hash = nixio.crypto.hmac('sha1', WAN_KEY)
hash:update(options.body)
options.headers['X-Digest'] = hash:final()
local http_persist = httpclient.create_persistent()
local url = WAN_BASE_URL .. DEVICE
local response_json, code, call_info = http_persist(url, options)
if code == 200 then
nixio.syslog('info', string.format('%s %s: %s', options.method, url, code))
else
nixio.syslog('err', string.format('%s %s: %s', options.method, url, code))
-- if available, send additional error info to the syslog
if type(call_info) == 'string' then
nixio.syslog('err', call_info)
elseif type(call_info) == 'table' then
local auth_error = call_info.headers['WWW-Authenticate']
if auth_error then
nixio.syslog('err', string.format('WWW-Authenticate: %s', auth_error))
end
end
os.exit(3)
end
-- verify the reply's digest
hash = nixio.crypto.hmac('sha1', WAN_KEY)
hash:update(response_json)
if call_info.headers['X-Digest'] ~= hash:final() then
nixio.syslog('err', 'Incorrect digest in the heartbeat reply. Discard response.')
os.exit(4)
end
local response = luci.json.decode(response_json)
-- use heartbeat return timestamp to sync time in case ntpclient doesn't get the job done
local TIMESTAMP_MIN = 1234567890
local TIMESTAMP = tonumber(response.timestamp)
if os.time() < TIMESTAMP_MIN then
nixio.settimeofday(TIMESTAMP)
end
-- check whether we have to reset or upgrade
if response.upgrade == monitor.version then
os.execute('reboot')
elseif response.upgrade > monitor.version then
os.execute('wget -P /tmp ' .. UPGRADE_URL .. 'upgrade.' .. response.upgrade)
os.execute('chmod a+x /tmp/upgrade.' .. response.upgrade)
os.execute('/tmp/upgrade.' .. response.upgrade)
os.execute('rm /tmp/upgrade.' .. response.upgrade)
end