simcard communication python class

This commit is contained in:
Christina Quast 2015-04-06 00:33:52 +02:00
parent 267afe37ec
commit 5149cd6e78
1 changed files with 66 additions and 86 deletions

View File

@ -3,101 +3,81 @@
from smartcard.scard import *
import smartcard.util
CMD_SEL_ROOT = [0xA0, 0xA4, 0x00, 0x00, 0x02, 0x3F, 0x00]
CMD_SEL_FILE = [0xA0, 0xA4, 0x00, 0x00, 0x02, 0x7F, 0x20]
CMD_GET_DATA = [0xA0, 0xC0, 0x00, 0x00, 0x16]
# SuperSIM ATR
atr_supersim= [0x3B, 0x9A, 0x94, 0x00, 0x92, 0x02, 0x75, 0x93, 0x11, 0x00, 0x01, 0x02, 0x02, 0x19]
import array
# Faster sysmocom SIM
ATR_SYSMOCOM1 = [0x3B, 0x99, 0x18, 0x00, 0x11, 0x88, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x60]
ATR_SYSMOCOM2 = [0x3B, 0x99, 0x11, 0x00, 0x11, 0x88, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x60]
NEW_ATR = ATR_SYSMOCOM2
class SmartcardException(Exception):
pass
def pattern_match(inpt):
if (inpt == ATR_SYSMOCOM1):
return NEW_ATR
elif (inpt == CMD_SEL_FILE):
return CMD_SEL_ROOT
else:
return inpt
class SmartcardConnection:
# hcard, dwActiveProtocol, hcontext, reader
def connect_to_card(hcontext, reader):
hresult, hcard, dwActiveProtocol = SCardConnect(hcontext, reader,
SCARD_SHARE_SHARED, SCARD_PROTOCOL_T0 | SCARD_PROTOCOL_T1)
if hresult != SCARD_S_SUCCESS:
raise Exception('Unable to connect: ' +
SCardGetErrorMessage(hresult))
print 'Connected with active protocol', dwActiveProtocol
return hresult, hcard, dwActiveProtocol
def establish_context():
hresult, hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
if hresult != SCARD_S_SUCCESS:
raise Exception('Failed to establish context : ' +
SCardGetErrorMessage(hresult))
print 'Context established!'
def __init__(self):
self.establish_context()
self.connect_card()
hresult, readers = SCardListReaders(hcontext, [])
if hresult != SCARD_S_SUCCESS:
raise Exception('Failed to list readers: ' +
SCardGetErrorMessage(hresult))
print 'PCSC Readers:', readers
if len(readers) < 1:
raise Exception('No smart card readers')
reader = readers[0]
print "Using reader:", reader
return (hcontext, reader)
def release_context(hcontext):
hresult = SCardReleaseContext(hcontext)
if hresult != SCARD_S_SUCCESS:
raise Exception('Failed to release context: ' +
def connect_card(self):
hresult, self.hcard, self.dwActiveProtocol = SCardConnect(self.hcontext, self.reader,
SCARD_SHARE_SHARED, SCARD_PROTOCOL_T0 | SCARD_PROTOCOL_T1)
if hresult != SCARD_S_SUCCESS:
raise SmartcardException('Unable to connect: ' +
SCardGetErrorMessage(hresult))
print 'Released context.'
print 'Connected with active protocol', self.dwActiveProtocol
def send_receive_cmd(cmd, hcard, dwActiveProtocol):
print("Response: ")
new_cmd = pattern_match(cmd)
print(' '.join([hex(i) for i in cmd]))
hresult, response = SCardTransmit(hcard, dwActiveProtocol,
new_cmd)
if hresult != SCARD_S_SUCCESS:
raise Exception('Failed to transmit: ' +
SCardGetErrorMessage(hresult))
resp = pattern_match(response)
print 'Ans: ' + smartcard.util.toHexString(resp,
smartcard.util.HEX)
return response
def establish_context(self):
hresult, self.hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
if hresult != SCARD_S_SUCCESS:
raise SmartcardException('Failed to establish context : ' +
SCardGetErrorMessage(hresult))
print 'Context established!'
def disconnect_card(hcard):
hresult = SCardDisconnect(hcard, SCARD_UNPOWER_CARD)
if hresult != SCARD_S_SUCCESS:
raise Exception('Failed to disconnect: ' +
SCardGetErrorMessage(hresult))
print 'Disconnected'
hresult, readers = SCardListReaders(self.hcontext, [])
if hresult != SCARD_S_SUCCESS:
raise SmartcardException('Failed to list readers: ' +
SCardGetErrorMessage(hresult))
print 'PCSC Readers:', readers
if len(readers) < 1:
raise SmartcardException('No smart card readers')
self.reader = readers[0]
print "Using reader:", self.reader
def release_context(self):
hresult = SCardReleaseContext(self.hcontext)
if hresult != SCARD_S_SUCCESS:
raise SmartcardException('Failed to release context: ' +
SCardGetErrorMessage(hresult))
print 'Released context.'
def send_receive_cmd(self, cmd):
print("Cmd: ")
hresult, resp = SCardTransmit(self.hcard, self.dwActiveProtocol,
cmd.tolist())
if hresult != SCARD_S_SUCCESS:
raise SmartcardException('Failed to transmit: ' +
SCardGetErrorMessage(hresult))
print 'Ans: ' + smartcard.util.toHexString(resp,
smartcard.util.HEX)
return array.array('B', resp)
def disconnect_card(self):
hresult = SCardDisconnect(self.hcard, SCARD_UNPOWER_CARD)
if hresult != SCARD_S_SUCCESS:
raise SmartcardException('Failed to disconnect: ' +
SCardGetErrorMessage(hresult))
print 'Disconnected'
def do_intercept(cmd, dwActiveProtocol):
return send_receive_cmd(cmd, hcard, dwActiveProtocol)
def ccid_raw_init():
(hcontext, reader) = establish_context()
hresult, hcard, dwActiveProtocol = connect_to_card(hcontext, reader)
return hcard, hcontext, dwActiveProtocol
def ccid_raw_exit(hcard, hcontext):
disconnect_card(hcard)
release_context(hcontext)
def close(self):
self.disconnect_card()
self.release_context()
if __name__ == '__main__':
hcard, hcontext, dwActiveProtocol = ccid_raw_init()
do_intercept(CMD_SEL_ROOT, dwActiveProtocol)
do_intercept(CMD_SEL_FILE, dwActiveProtocol)
do_intercept(CMD_GET_DATA, dwActiveProtocol)
ccid_raw_exit()
import constants
sm_con = SmartcardConnection()
print(sm_con.send_receive_cmd(constants.CMD_SEL_ROOT))
print(sm_con.send_receive_cmd(constants.CMD_SEL_FILE))
print(sm_con.send_receive_cmd(constants.CMD_GET_DATA))
sm_con.close()