-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcheck_disk.lua
114 lines (100 loc) · 3.46 KB
/
check_disk.lua
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
--
-- Copyright (c) 2009, Cloudkick, Inc.
-- All right reserved.
--
module(..., package.seeall);
local alien = require 'alien'
local util = require 'util'
local Check = util.Check
local log = util.log
local structs = require 'structs'
local function win_getvalue(args)
local f = alien.Kernel32.GetDiskFreeSpaceExA
f:types{ ret = "int", abi = "stdcall", "string", "pointer", "pointer", "pointer" }
ulargeint = alien.defstruct{
{ "low", "ulong" },
{ "high", "ulong" }
}
local disk_free = ulargeint:new()
local disk_total = ulargeint:new()
local disk_total_free = ulargeint:new()
log.dbg("calling alien.Kernel32.GetDiskFreeSpaceExA('%q')", args.path)
assert( f(args.path, disk_free(), disk_total(), disk_total_free()) ~= 0 )
local r = {}
r.bsize = 1024
r.blocks = (disk_total.high * 2^32 + disk_total.low) / r.bsize
r.bfree = (disk_total_free.high * 2^32 + disk_free.low) / r.bsize
r.capacity = 100 - ((r.bfree/r.blocks) * 100)
return r
end
local function getvalue(args)
local libc = alien.default
local buf = alien.buffer(structs.statfs.__size)
local rv = 0
if equus.p_is_darwin() == 1 then
libc.statfs64:types("int", "string", "pointer")
log.dbg("calling libc.statfs64('%q')", args.path)
rv = libc.statfs64(args.path, buf:topointer())
log.dbg("libc.statfs64('%q') = %d", args.path, rv)
else
libc.statfs:types("int", "string", "pointer")
log.dbg("calling libc.statfs('%q')", args.path)
rv = libc.statfs(args.path, buf:topointer())
log.dbg("libc.statfs('%q') = %d", args.path, rv)
end
if rv < 0 then
error("libc.statfs returned ".. rv)
end
local r = {}
r.bsize = buf:get(structs.statfs.f_bsize+1, "uint")
r.blocks = buf:get(structs.statfs.f_blocks+1, "long")
r.bavail = buf:get(structs.statfs.f_bavail+1, "long") -- blocks available to non-root users. this is what we actually care about
r.bfree = buf:get(structs.statfs.f_bfree+1, "long") -- blocks available to root
local blocks_total_nonroot = r.blocks - r.bfree + r.bavail
r.blocks_used = blocks_total_nonroot - r.bavail
r.capacity = (r.blocks_used / blocks_total_nonroot) * 100
if r.capacity > 100 then
log.msg("capacity calculation returned %s, capping at 100%%", r.capacity)
r.capacity = 100
end
return r
end
function run(rcheck, args)
if args.path == nil then
if equus.p_is_windows() == 1 then
args.path = 'c:\\'
else
args.path = '/'
end
else
args.path = args.path[1]
end
local rv = nil
local r = nil
if equus.p_is_windows() == 1 then
rv, r = pcall(win_getvalue, args)
else
rv, r = pcall(getvalue, args)
end
if rv then
rcheck:add_metric('capacity', r.capacity, Check.enum.double)
rcheck:add_metric('bsize', r.bsize, Check.enum.uint64)
rcheck:add_metric('blocks', r.blocks, Check.enum.uint64)
rcheck:add_metric('bfree', r.bfree, Check.enum.uint64)
local used = (r.blocks - r.bfree)
if equus.p_is_windows() == 0 then
rcheck:add_metric('bavail', r.bavail, Check.enum.uint64)
used = r.blocks_used
end
local free = (r.blocks - used)
local total_gb = (r.blocks * r.bsize) / 1024 / 1024 / 1024
local used_gb = (used * r.bsize) / 1024 / 1024 / 1024
local free_gb = ((r.blocks - used) * r.bsize) / 1024 / 1024 / 1024
rcheck:set_status('capacity: %.2f%% total: %.2f GB, used: %.2f GB, free: %.2f GB',
r.capacity, total_gb, used_gb, free_gb)
else
log.err("disk failed err: %s", r)
rcheck:set_error("%s", r)
end
return rcheck
end