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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
|
--[[
Copyright (c) 2012-2013 Kaarle Ritvanen
See LICENSE file for license details
--]]
local M = {}
local raise = require('acf2.error').raise
local Union = require('acf2.model.combination').Union
local fld = require('acf2.model.field')
local String = fld.String
local object = require('acf2.object')
local class = object.class
local super = object.super
local pth = require('acf2.path')
local update = require('acf2.util').update
local stringy = require('stringy')
local BaseIPAddress = class(String)
function BaseIPAddress:abs_mask_addr(context)
if self.mask_addr then
return pth.join(pth.parent(context.addr), self.mask_addr)
end
end
function BaseIPAddress:topology(context)
local res = super(self, BaseIPAddress):topology(context)
local maddr = self:abs_mask_addr(context)
if maddr then
table.insert(res, {path=context.path, addr=maddr, type=self.mask_type})
end
return res
end
function BaseIPAddress:invalid(context)
raise(context.path, 'Invalid IPv'..self.version..' address')
end
function BaseIPAddress:split(context, value)
local comps = stringy.split(value, '/')
if #comps == 1 then return value, self.length end
if #comps > 2 or not self.cidr then self:invalid(context) end
local mask = tonumber(comps[2])
if not mask or mask < 0 or mask > self.length then self:invalid(context) end
return comps[1], mask
end
function BaseIPAddress:_load(context)
local res = super(self, BaseIPAddress):_load(context)
local maddr = self:abs_mask_addr(context)
if res and maddr then
return res..'/'..(self:mask2cidr(context.txn:get(maddr)) or self.length)
end
return res
end
function BaseIPAddress:_save(context, value)
local maddr = self:abs_mask_addr(context)
if maddr then
local cidr
if value then value, cidr = self:split(context, value) end
context.txn:set(maddr, cidr and self:cidr2mask(cidr))
end
super(self, BaseIPAddress):_save(context, value)
end
M.IPv4Address = class(BaseIPAddress)
function M.IPv4Address:init(params)
super(self, M.IPv4Address):init(
update(params, {version=4, length=32, mask_type='string'})
)
end
function M.IPv4Address:validate(context, value)
super(self, M.IPv4Address):validate(context, value)
local address = self:split(context, value)
local function test(...)
if #{...} ~= 4 then return true end
for _, octet in ipairs{...} do
if tonumber(octet) > 255 then return true end
end
end
if test(address:match('^(%d+)%.(%d+)%.(%d+)%.(%d+)$')) then
self:invalid(context)
end
end
function M.IPv4Address:mask2cidr(mask)
local acc = 0
for i, comp in ipairs(stringy.split(mask, '.')) do
acc = acc + math.pow(256, 4 - i) * tonumber(comp)
end
local res = 32
while acc % 2 == 0 do
res = res - 1
assert(res > -1)
acc = acc / 2
end
return res
end
function M.IPv4Address:cidr2mask(cidr)
local acc = (math.pow(2, cidr) - 1) * math.pow(2, 32 - cidr)
local comps = {}
for i = 4,1,-1 do
comps[i] = acc % 256
acc = math.floor(acc / 256)
end
return table.concat(comps, '.')
end
M.IPv6Address = class(BaseIPAddress)
function M.IPv6Address:init(params)
super(self, M.IPv6Address):init(
update(params, {version=6, length=128, mask_type='number'})
)
end
function M.IPv6Address:validate(context, value)
super(self, M.IPv6Address):validate(context, value)
local address = self:split(context, value)
local function invalid() self:invalid(context) end
if address == '' then invalid() end
local comps = stringy.split(address, ':')
if #comps < 3 then invalid() end
local function collapse(i, ofs)
if comps[i] > '' then return end
if comps[i + ofs] > '' then invalid() end
table.remove(comps, i)
end
collapse(1, 1)
collapse(#comps, -1)
if #comps > 8 then invalid() end
local short = false
for _, comp in ipairs(comps) do
if comp == '' then
if short then invalid() end
short = true
elseif not comp:match('^%x%x?%x?%x?$') then invalid() end
end
if (
short and #comps == 3 and comps[2] == ''
) or (not short and #comps < 8) then
invalid()
end
end
function M.IPv6Address:mask2cidr(mask) return mask end
function M.IPv6Address:cidr2mask(cidr) return cidr end
M.IPAddress = class(Union)
function M.IPAddress:init(params)
super(self, M.IPAddress):init(
update(
params,
{types={M.IPv4Address, M.IPv6Address}, error='Invalid IP address'}
)
)
end
M.Port = class(fld.Integer)
function M.Port:validate(context, value)
super(self, M.Port):validate(context, value)
if value < 0 or value > 65535 then raise(context.path, 'Invalid port') end
end
return M
|