-
Notifications
You must be signed in to change notification settings - Fork 2
/
server.py
175 lines (153 loc) · 7.16 KB
/
server.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
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
from flask import Flask, render_template, request, json
import subprocess
import os
import time
import logging
logging.basicConfig(filename='error.log', level=logging.DEBUG)
DUMPS_DIR = "DumpDir"
KEYFILE = "key-file.txt"
HOST = "localhost"
PORT = "5000"
def runCommand(command):
process = subprocess.Popen(
command,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
shell=True)
stdout, stderr = process.communicate()
return stdout, stderr, process.returncode
### Partie Flask (Serveur Web)
app = Flask('__name__', static_folder="res")
app.config['DEBUG'] = False
# Page principal
@app.route('/')
def root():
if request.args.get("error"): # On affiche l'erreur si il y en a une.
error = request.args.get("error")
return render_template('index.html', error=error)
else:
return render_template('index.html')
# Page d'attente du badge du client
@app.route('/waitingBadge')
def waitingBadge():
nbBadge = request.args.get('nbBadge', default=1, type=int)
return render_template('readBadge.html', nbBadge=nbBadge)
# Une fois que le badge est détecté
@app.route('/detectBadge')
def detectBadge():
while True: # Boucle infinie tant que le badge n'est pas detecté.
stdout, stderr, return_code = runCommand('nfc-list')
time.sleep(2)
stdout = stdout.decode("utf-8")
lines = stdout.split('\n')
if "Interface opened" in lines[1]:
if len(lines) > 6:
badge_type = lines[4].replace(' ', ' ')
if '00 04' in badge_type:
result = {'status': 'OK'}
break
else: # Si le badge n'est pas compatble avec le badge cloner.
result = {'status': 'ERROR', 'message': 'Badge non compatible !'}
break
else:
logging.warning(stdout)
result = {'status': 'ERROR', 'message': 'Lecteur NFC non connecté !'}
break
return json.dumps(result)
# Permet de lire et copier le badge du client (Route appellée avec du JS depuis la page readBadge.html)
@app.route('/readBadge')
def readBadge():
while True: # Boucle infinie tant que le badge n'est pas detecté.
stdout, stderr, return_code = runCommand('nfc-list')
time.sleep(2)
stdout = stdout.decode("utf-8")
lines = stdout.split('\n')
if "Interface opened" in lines[1]:
if len(lines) > 6:
badge_type = lines[4].replace(' ', ' ')
if '00 04' in badge_type:
badge_UID = lines[5].replace('UID (NFCID1): ', '').replace(' ', '')
# Si un dump existe déjà pour cet UID et qu'il est complet
if ((os.path.isfile('%s/%s.dmp' % (DUMPS_DIR, badge_UID))) and (os.path.getsize('%s/%s.dmp' % (DUMPS_DIR, badge_UID)))) == 1024:
result = {'status': 'OK', 'UID': badge_UID}
break
else:
stdout, stderr, status_code = runCommand(
'mfoc -f %s -P 500 -O %s/%s.dmp' % (KEYFILE, DUMPS_DIR, badge_UID))
time.sleep(2)
if status_code == 0: # Si on a réussi à avoir le dump
if os.path.getsize('%s/%s.dmp' % (DUMPS_DIR, badge_UID)) == 1024: # Mifare classic 1k
with open('%s/%s.dmp' % (DUMPS_DIR, badge_UID), 'rb') as f:
hexdata = f.read().hex()
sector11 = hexdata[1408:1536] # On récupère le secteur 11
if sector11[0:96] != "0" * 96: # Si le secteur est pas vide
result = {'status': 'ERROR', 'message': 'Badge protégé ! Ne pas copier !'}
break
else:
result = {'status': 'OK', 'UID': badge_UID}
break
else:
result = {'status': 'OK', 'UID': badge_UID}
break
else:
result = {'status': 'ERROR', 'message': 'Impossible de cracker le badge !'}
break
else: # Si le badge n'est pas compatble avec le badge cloner.
logging.warning(stdout)
result = {'status': 'ERROR', 'message': 'Badge non compatible !'}
break
else:
logging.warning(stdout)
result = {'status': 'ERROR', 'message': 'Lecteur NFC non connecté !'}
break
return json.dumps(result)
# Page d'attente des badges à cloner
@app.route('/writeBadge/<nbBadge>/<uid>')
def writeBadge(nbBadge, uid):
if request.args.get("error"): # On affiche l'erreur si il y en a une.
error = request.args.get("error")
return render_template('writeBadge.html', nbBadge=int(nbBadge), uid=uid, error=error)
return render_template('writeBadge.html', nbBadge=int(nbBadge), uid=uid)
# Permet de cloner les badges
@app.route('/copyBadge/<uid>')
def copyBadge(uid):
while True: # Boucle infinie tant que le badge n'est pas detecté.
stdout, stderr, return_code = runCommand('nfc-list')
time.sleep(2)
stdout = stdout.decode("utf-8")
lines = stdout.split('\n')
if "Interface opened" in lines[1]:
if len(lines) > 6:
badge_type = lines[4].replace(' ', ' ')
if '00 04' in badge_type:
stdout, stderr, status_code = runCommand('mfoc -P 500 -O %s/new.dmp' % DUMPS_DIR)
time.sleep(2)
if status_code == 0: # Si on a réussi à avoir le dump
stdout, stderr, status_code = runCommand(
'nfc-mfclassic W a %s/%s.dmp %s/new.dmp' % (DUMPS_DIR, uid, DUMPS_DIR))
time.sleep(2)
if status_code == 0: # Si on a réussi à copier
result = {'status': 'OK', 'message': 'Badge copié !'}
break
else:
result = {'status': 'ERROR', 'message': 'Avez-vous mis un badge réinscriptible ?'}
break
else:
result = {'status': 'ERROR', 'message': 'Impossible d\'écrire sur ce badge'}
break
else: # Si le badge n'est pas compatble avec le badge cloner.
logging.warning(stdout)
result = {'status': 'ERROR', 'message': 'Badge non compatible !'}
break
else:
logging.warning(stdout)
result = {'status': 'ERROR', 'message': 'Lecteur NFC non connecté !'}
break
return json.dumps(result)
@app.route('/wait')
def wait():
raison = request.args.get("raison")
redirect = request.args.get("redirect")
return render_template('wait.html', raison=raison, redirect=redirect)
app.run(host=HOST, port=PORT)
### Fin partie Flask ###