-
Notifications
You must be signed in to change notification settings - Fork 0
/
NFCReader.py
executable file
·130 lines (111 loc) · 4.53 KB
/
NFCReader.py
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
#! /usr/bin/env python
import re, argparse
from smartcard.System import readers
import datetime, sys
#ACS ACR122U NFC Reader
#Suprisingly, to get data from the tag, it is a handshake protocol
#You send it a command to get data back
#This command below is based on the "API Driver Manual of ACR122U NFC Contactless Smart Card Reader"
COMMAND = [0xFF, 0xCA, 0x00, 0x00, 0x00] #handshake cmd needed to initiate data transfer
# Disable the standard buzzer when a tag is detected, Source : https://gist.github.com/nixme/2717287
DISABLEBEEP = [0xFF, 0x00, 0x52, 0x00, 0x00]
# get all the available readers
r = readers()
print("Available readers:", r )
def stringParser(dataCurr):
#--------------String Parser--------------#
#([85, 203, 230, 191], 144, 0) -> [85, 203, 230, 191]
if isinstance(dataCurr, tuple):
temp = dataCurr[0]
code = dataCurr[1]
#[85, 203, 230, 191] -> [85, 203, 230, 191]
else:
temp = dataCurr
code = 0
dataCurr = ''
#[85, 203, 230, 191] -> bfe6cb55 (int to hex reversed)
for val in temp:
# dataCurr += (hex(int(val))).lstrip('0x') # += bf
dataCurr += format(val, '#04x')[2:] # += bf
#bfe6cb55 -> BFE6CB55
dataCurr = dataCurr.upper()
#if return is successful
if (code == 144):
return dataCurr
def readTag(page):
try:
connection = reader.createConnection()
status_connection = connection.connect()
connection.transmit(DISABLEBEEP)
connection.transmit(COMMAND)
#Read command [FF, B0, 00, page, #bytes]
resp = connection.transmit([0xFF, 0xB0, 0x00, int(page), 0x04])
dataCurr = stringParser(resp)
#only allows new tags to be worked so no duplicates
if(dataCurr is not None):
print(dataCurr)
else:
print("Something went wrong. Page " + str(page))
except:
print("Exception")
def writeTag(page, value):
if type(value) != str:
print("Value input should be a string" )
exit()
while(1):
if len(value) == 8:
try:
connection = reader.createConnection()
status_connection = connection.connect()
connection.transmit(DISABLEBEEP)
connection.transmit(COMMAND)
WRITE_COMMAND = [0xFF, 0xD6, 0x00, int(page), 0x04, int(value[0:2], 16), int(value[2:4], 16), int(value[4:6], 16), int(value[6:8], 16)]
# Let's write a page Page 9 is usually 00000000
resp = connection.transmit(WRITE_COMMAND)
if resp[1] == 144:
print("Wrote " + value + " to page " + str(page))
break
except:
continue
else:
print("Must have a full 4 byte write value" )
break
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Read / write NFC tags')
usingreader_group = parser.add_argument_group('usingreader')
usingreader_group.add_argument('--usingreader', nargs=1, metavar='READER_ID', help='Reader to use [0-X], default is 0')
read_group = parser.add_argument_group('read')
read_group.add_argument('--read', nargs='+', help='Pages to read. Can be a x-x range, or list of pages')
write_group = parser.add_argument_group('write')
write_group.add_argument('--write', nargs=2, metavar=('PAGE', 'DATA'), help='Page number and hex value to write.')
args = parser.parse_args()
#Choosing which reader to use
if args.usingreader:
usingreader = args.usingreader[0]
if (int(usingreader) >= 0 and int(usingreader) <= len(r)-1):
reader = r[int(usingreader)]
print("Using: [",int(usingreader),"]", reader )
else:
reader = r[0]
print("Using: [ 0 ]", reader )
else:
reader = r[0]
print("Using: [ 0 ]", reader )
#Only going to write one page at a time
if args.write:
page = args.write[0]
data = args.write[1]
if len(data) == 8:
writeTag(int(page), data)
else:
raise argparse.ArgumentTypeError("Must have a full 4 byte write value")
#Page numbers are sent as ints, not hex, to the reader
if args.read:
for page in args.read:
if "-" in page:
start = int(page.split("-")[0])
end = int(page.split("-")[1])
for new_page in xrange(start, end+1):
readTag(new_page)
else:
readTag(page)