first init

2026-09-01 10:12:31 +08:00
commit ca6e55f853
718 changed files with 204746 additions and 0 deletions
@@ -0,0 +1,93 @@
+"""Sample CardConnectionDecorator that provides exclusive transmit()
+
+__author__ = "https://www.gemalto.com/"
+
+Copyright 2001-2012 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
+"""
+
+import smartcard.pcsc
+from smartcard.CardConnectionDecorator import CardConnectionDecorator
+from smartcard.Exceptions import CardConnectionException
+from smartcard.scard import (
+    SCARD_LEAVE_CARD,
+    SCARD_S_SUCCESS,
+    SCardBeginTransaction,
+    SCardEndTransaction,
+    SCardGetErrorMessage,
+)
+
+
+class ExclusiveTransmitCardConnection(CardConnectionDecorator):
+    """This decorator uses
+    L{SCardBeginTransaction}/L{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 L{SCardBeginTransaction}."""
+
+        # pylint: disable=duplicate-code
+
+        component = self.component
+        while True:
+            if isinstance(
+                component, smartcard.pcsc.PCSCCardConnection.PCSCCardConnection
+            ):
+                hresult = SCardBeginTransaction(component.hcard)
+                if SCARD_S_SUCCESS != hresult:
+                    raise CardConnectionException(
+                        "Failed to lock with SCardBeginTransaction: "
+                        + SCardGetErrorMessage(hresult)
+                    )
+                break
+            if hasattr(component, "component"):
+                component = component.component
+            else:
+                break
+
+    def unlock(self):
+        """Unlock card with L{SCardEndTransaction}."""
+
+        # pylint: disable=duplicate-code
+
+        component = self.component
+        while True:
+            if isinstance(
+                component, smartcard.pcsc.PCSCCardConnection.PCSCCardConnection
+            ):
+                hresult = SCardEndTransaction(component.hcard, SCARD_LEAVE_CARD)
+                if SCARD_S_SUCCESS != hresult:
+                    raise CardConnectionException(
+                        "Failed to unlock with SCardEndTransaction: "
+                        + SCardGetErrorMessage(hresult)
+                    )
+                break
+            if hasattr(component, "component"):
+                component = component.component
+            else:
+                break
+
+    def transmit(self, command, protocol=None):
+        """Gain exclusive access to card during APDU transmission for if this
+        decorator decorates a PCSCCardConnection."""
+        data, sw1, sw2 = CardConnectionDecorator.transmit(self, command, protocol)
+        return data, sw1, sw2