#! /usr/bin/env python
"""Sample script for APDU error checking with a custom error checker.

__author__ = "http://www.gemalto.com"

Copyright 2001-2009 gemalto
Author: Jean-Daniel Aussel, mailto:jean-daniel.aussel@gemalto.com

This file is part of pyscard.

pyscard is free software; you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 2.1 of the License, or
(at your option) any later version.

pyscard is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with pyscard; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
"""

from smartcard.CardType import AnyCardType
from smartcard.CardRequest import CardRequest
from smartcard.CardConnectionObserver import ConsoleCardConnectionObserver

from smartcard.sw.ErrorCheckingChain import ErrorCheckingChain
from smartcard.sw.ErrorChecker import ErrorChecker
from smartcard.sw.SWExceptions import SWException



class MyErrorChecker( ErrorChecker ):
    """Our custom error checker that will except if 0x61<sw1<0x70."""
    def __call__( self, data, sw1, sw2 ):
        print sw1, sw2
        if 0x61<sw1 and 0x70>sw1:
            raise SWException( data, sw1, sw2 )

# define the apdus used in this script
GET_RESPONSE = [0XA0, 0XC0, 00, 00 ]
SELECT = [0xA0, 0xA4, 0x00, 0x00, 0x02]
DF_TELECOM = [0x7F, 0x10]

if __name__ == '__main__':

    print 'Insert a card within 10 seconds'
    print 'Cards without a DF_TELECOM will except'

    # request any card
    cardtype = AnyCardType()
    cardrequest = CardRequest( timeout=10, cardType=cardtype )
    cardservice = cardrequest.waitforcard()

    # our error checking chain
    errorchain=[]
    errorchain=[ ErrorCheckingChain( [], MyErrorChecker() ) ]
    cardservice.connection.setErrorCheckingChain( errorchain )

    # attach the console tracer
    observer=ConsoleCardConnectionObserver()
    cardservice.connection.addObserver( observer )

    # send a few apdus; exceptions will occur upon errors
    cardservice.connection.connect()

    try:
        SELECT = [0xA0, 0xA4, 0x00, 0x00, 0x02]
        DF_TELECOM = [0x7F, 0x10]
        apdu = SELECT+DF_TELECOM
        response, sw1, sw2 = cardservice.connection.transmit( apdu )
        if sw1 == 0x9F:
            GET_RESPONSE = [0XA0, 0XC0, 00, 00 ]
            apdu = GET_RESPONSE + [sw2]
            response, sw1, sw2 = cardservice.connection.transmit( apdu )
    except SWException, e:
        print e, "%x %x" % (e.sw1, e.sw2 )

        cardservice.connection.disconnect()

    import sys
    if 'win32'==sys.platform:
        print 'press Enter to continue'
        sys.stdin.read(1)