-
Notifications
You must be signed in to change notification settings - Fork 37
/
NFCReader.py
145 lines (126 loc) · 4.85 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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
#! /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
# 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):
readingLoop = 1
while(readingLoop):
try:
connection = reader.createConnection()
status_connection = connection.connect()
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
break
else:
print "Something went wrong. Page " + str(page)
break
except Exception,e:
if(waiting_for_beacon ==1):
continue
else:
readingLoop=0
print str(e)
break
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(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 Exception, e:
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')
wait_group = parser.add_argument_group('wait')
wait_group.add_argument('--wait', nargs=1, metavar='0|1', help='Wait for beacon before returns [0|1], default is 1')
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)]
else:
reader = r[0]
else:
reader = r[0]
print "Using:", reader
#Disabling wait for answer if wait == 0
if args.wait:
wait = args.wait[0]
if (int(wait) == 0 ):
waiting_for_beacon = 0
else:
waiting_for_beacon = 1
else:
waiting_for_beacon = 1
print "Using:", 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)