File: ExclusiveTransmitCardConnection.py

package info (click to toggle)
pyscard 1.6.10-2
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 1,360 kB
  • ctags: 1,516
  • sloc: python: 6,387; ansic: 2,065; makefile: 2
file content (78 lines) | stat: -rw-r--r-- 3,331 bytes parent folder | download
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
"""Sample CardConnectionDecorator that provides exclusive transmit()

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

Copyright 2001-2010 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.CardConnection import CardConnection
from smartcard.CardConnectionDecorator import CardConnectionDecorator
from smartcard.Exceptions import CardConnectionException
from smartcard.scard import SCardBeginTransaction, SCardEndTransaction, SCARD_LEAVE_CARD
from smartcard.scard import SCardGetErrorMessage
from smartcard.pcsc import PCSCCardConnection
import smartcard.pcsc

class ExclusiveTransmitCardConnection( CardConnectionDecorator ):
    '''This decorator uses SCardBeginTransaction/SCardEndTransaction
    to preserve other processes of threads to access the card during transmit().'''
    def __init__( self, cardconnection ):
        CardConnectionDecorator.__init__( self, cardconnection )

    def lock( self ):
        '''Lock card with SCardBeginTransaction.'''

        component=self.component
        while True:
            if isinstance( component, smartcard.pcsc.PCSCCardConnection.PCSCCardConnection ):
                hresult = SCardBeginTransaction( component.hcard )
                if 0!=hresult:
                    raise CardConnectionException( 'Failed to lock with SCardBeginTransaction' + SCardGetErrorMessage(hresult) )
                else:
                    #print 'locked'
                    pass
                break
            if hasattr( component, 'component' ):
                component=component.component
            else:
                break

    def unlock( self ):
        '''Unlock card with SCardEndTransaction.'''
        component=self.component
        while True:
            if isinstance( component, smartcard.pcsc.PCSCCardConnection.PCSCCardConnection ):
                hresult = SCardEndTransaction( component.hcard, SCARD_LEAVE_CARD )
                if 0!=hresult:
                    raise CardConnectionException( 'Failed to unlock with SCardEndTransaction' + SCardGetErrorMessage(hresult) )
                else:
                    #print 'unlocked'
                    pass
                break
            if hasattr( component, 'component' ):
                component=component.component
            else:
                break


    def transmit( self, bytes, protocol=None ):
        '''Gain exclusive access to card during APDU transmission for if this
        decorator decorates a PCSCCardConnection.'''
        data, sw1, sw2 = CardConnectionDecorator.transmit( self, bytes, protocol )
        return data, sw1, sw2