BadReader.py

#!/usr/bin/python3

# Bad Reader
# A sandbox program to experiment reading and writing on a SLE4442 "smart" card
# Author : Heartbroken-Dev
# License : Apache License 2.0
# Dependencies : pcscd pcsc-tools libpcsclite1 libpcsclite-dev pyscard
# Version : PoC 2

from smartcard.CardType import AnyCardType
from smartcard.CardRequest import CardRequest
from smartcard.util import toHexString
from smartcard.Exceptions import CardRequestTimeoutException
import BRutils as utils
import BRcommands as cmds

def enterPrompt(cardService):
	requestedExit = False
	requestedDisconnect = False
	cardReader = cardService.connection.getReader() # TODO : to be protected by try..except clause if cardService not connected

	while not requestedExit:

		cmdIn = input(cardReader + " >>> ") # or any other prompt-like thing
		cmdList = cmdIn.split()

		if cmdList[0] == "disconnect":
			requestedDisconnect = True
			requestedExit = True # Required to exit the prompt for reconnection as is
			cmds.Disconnect.execute(cardService, cmdList)
		elif cmdList[0] == "exit":
			if not requestedDisconnect:
				print("Disconnecting card before exiting") # TODO : Yellow ?
				cmds.Disconnect.execute(cardService, cmdList)
			print("Exiting")
			requestedExit = True
		elif cmdList[0] == "getATR":
			cmds.GetATR.execute(cardService, cmdList)
		else: # default to help()
			cmds.Help.execute(cardService, cmdList) # TODO : add specific message when there's a typo ?

	if requestedDisconnect:
		return utils.const.STATUS_PROMPT_DISCONNECTED
	else:
		return utils.const.STATUS_PROMPT_EXITING

def attemptConnection(cardRequest):

	stillWaiting = True

	print("Waiting for card insertion", flush=True, end='')

	while stillWaiting:
		try:
			stillWaiting = False
			print(".", flush=True, end='')
			cardService = cardRequest.waitforcard()
		except CardRequestTimeoutException:
			stillWaiting = True

	print("!")
	cardService.connection.connect()
	reader = cardService.connection.getReader()
	cardATR = toHexString(cardService.connection.getATR())

	print("Successfully connected on reader", reader)
	print("Card ATR showing", cardATR)

	return cardService

def main():

	mainCardType = AnyCardType()
	mainCardRequest = CardRequest(timeout=1, cardType=mainCardType)

	# Introduction
	print("Bad Reader, PoC")
	print("By Heartbroken-Dev, licensed under Apache-2.0")

	while True: # Pseudo do: .. while()

		# Attempt to connect to a card
		mainCardService = attemptConnection(mainCardRequest)

		promptStatus = enterPrompt(mainCardService)

		if promptStatus == utils.const.STATUS_PROMPT_EXITING:
			break

if __name__ == '__main__':
	main()