+--
+-- lf_bulk_program.lua - A tool to clone a large number of tags at once.
+-- Updated 2017-04-18
+--
+-- The getopt-functionality is loaded from pm3/client/lualibs/getopt.lua
+-- Have a look there for further details
+getopt = require('getopt')
+bit32 = require('bit32')
+
+usage = [[ script run lf_bulk_program.lua -f facility -b base_id_num -c count
+
+ e.g:
+ script run lf_bulk_program.lua -f 1 -b 1000 -c 10
+]]
+author = "Brian Redbeard"
+desc =[[
+Perform bulk enrollment of 26 bit H10301 style RFID Tags
+For more info, check the comments in the code
+]]
+
+--[[Implement a function to simply visualize the bitstream in a text format
+--This is especially helpful for troubleshooting bitwise math issues]]--
+function toBits(num,bits)
+ -- returns a table of bits, most significant first.
+ bits = bits or math.max(1, select(2, math.frexp(num)))
+ local t = {} -- will contain the bits
+ for b = bits, 1, -1 do
+ t[b] = math.fmod(num, 2)
+ num = math.floor((num - t[b]) / 2)
+ end
+ return table.concat(t)
+end
+
+--[[Likely, I'm an idiot, but I couldn't find any parity functions in Lua
+ This can also be done with a combination of bitwise operations (in fact,
+ is the canonically "correct" way to do it, but my brain doesn't just
+ default to this and so counting some ones is good enough for me]]--
+local function evenparity(s)
+ local _, count = string.gsub(s, "1", "")
+
+ local p = count % 2
+ if (p == 0) then
+ return(false)
+ else
+ return(true)
+ end
+end
+
+
+local function isempty(s)
+ return s == nil or s == ''
+end
+
+--[[The Proxmark3 "clone" functions expect the data to be in hex format so
+ take the card id number and facility ID as arguments and construct the
+ hex. This should be easy enough to extend to non 26bit formats]]--
+local function cardHex(i,f)
+ fac = bit32.lshift(f,16)
+ id = bit32.bor(i, fac)
+ stream=toBits(id,26)
+
+ --As the function defaults to even parity and returns a boolean,
+ --perform a 'not' function to get odd parity
+ high = evenparity(string.sub(stream,0,12)) and 1 or 0
+ low = not evenparity(string.sub(stream,13)) and 1 or 0
+ bits = bit32.bor(bit32.lshift(id,1), low)
+ bits = bit32.bor(bits, bit32.lshift(high,25))
+
+ --Since the lua library bit32 is (obviously) 32 bits and we need to
+ --encode 36 bits to properly do a 26 bit tag with the preamble we need
+ --to create a higher order and lower order component which we will
+ --then assemble in the return. The math above defines the proper
+ --encoding as per HID/Weigand/etc. These bit flips are due to the
+ --format length check on bit 38 (cmdlfhid.c:64) and
+ --bit 31 (cmdlfhid.c:66).
+ preamble = bit32.bor(0, bit32.lshift(1,5))
+ bits = bit32.bor(bits, bit32.lshift(1,26))
+
+ return ("%04x%08x"):format(preamble,bits)
+
+end
+
+local function main(args)
+
+ --I really wish a better getopt function would be brought in supporting
+ --long arguments, but it seems this library was chosen for BSD style
+ --compatibility
+ for o, a in getopt.getopt(args, 'f:b:c:h') do
+ if o == 'f' then
+ if isempty(a) then
+ print("You did not supply a facility code, using 0")
+ facility = 0
+ else
+ facility = a
+ end
+ elseif o == 'b' then
+ if isempty(a) then
+ print("You must supply the flag -b (base id)")
+ return
+ else
+ baseid = a
+ end
+ elseif o == 'c' then
+ if isempty(a) then
+ print("You must supply the flag -c (count)")
+ return
+ else
+ count = a
+ end
+ elseif o == 'h' then
+ print(desc)
+ print(usage)
+ return
+ end
+ end
+
+ --Due to my earlier complaints about how this specific getopt library
+ --works, specifying ":" does not enforce supplying a value, thus we
+ --need to do these checks all over again.
+
+ if isempty(baseid) then
+ print("You must supply the flag -b (base id)")
+ print(usage)
+ return
+ end
+
+ if isempty(count) then
+ print("You must supply the flag -c (count)")
+ print(usage)
+ return
+ end
+
+ --If the facility ID is non specified, ensure we code it as zero
+ if isempty(facility) then
+ print("Using 0 for the facility code as -f was not supplied")
+ facility = 0
+ end
+
+ --The next baseid + count function presents a logic/UX conflict
+ --where users specifying -c 1 (count = 1) would try to program two
+ --tags. This makes it so that -c 0 & -c 1 both code one tag, and all
+ --other values encode the expected amount.
+ if tonumber(count) > 0 then count = count -1 end
+
+ endid = baseid + count
+
+ for cardnum = baseid,endid do
+ local card = cardHex(cardnum, facility)
+ print("Press enter to program card "..cardnum..":"..facility.." (hex: "..card..")")
+ --This would be better with "press any key", but we'll take
+ --what we can get.
+ io.read()
+ core.console( ('lf hid clone %s'):format(card) )
+ end
+end
+
+
+main(args)