File: SCardBeginTransaction_SharedMode.py

package info (click to toggle)
pcsc-lite 2.4.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,700 kB
  • sloc: ansic: 13,894; python: 3,231; lex: 609; makefile: 246; sh: 54; xml: 22
file content (164 lines) | stat: -rwxr-xr-x 4,768 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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
#! /usr/bin/env python3

#   SCardBeginTransaction_SharedMode.py : Unitary test for
#   SCardReleaseContext()
#   Copyright (C) 2018  Ludovic Rousseau
#
#   This program is free software; you can redistribute it and/or modify
#   it under the terms of the GNU General Public License as published by
#   the Free Software Foundation; either version 3 of the License, or
#   (at your option) any later version.
#
#   This program 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 General Public License for more details.
#
#   You should have received a copy of the GNU General Public License along
#   with this program; if not, see <http://www.gnu.org/licenses/>.

"""
# SCardReleaseContext() should not release a PC/SC transaction not
# started by the released context
"""

import threading
import time

from smartcard.pcsc.PCSCExceptions import (
    BaseSCardException,
    EstablishContextException,
    ListReadersException,
    ReleaseContextException,
)
from smartcard.scard import SCARD_E_SHARING_VIOLATION  # SCardDisconnect,
from smartcard.scard import (
    SCARD_LEAVE_CARD,
    SCARD_PROTOCOL_ANY,
    SCARD_S_SUCCESS,
    SCARD_SCOPE_USER,
    SCARD_SHARE_SHARED,
    SCardBeginTransaction,
    SCardConnect,
    SCardEndTransaction,
    SCardEstablishContext,
    SCardListReaders,
    SCardReleaseContext,
    SCardTransmit,
)

RED = "\033[0;31m"
BLUE = "\033[0;34m"
NORMAL = "\033[00m"


def init_client():
    """init a client"""
    print("SCardEstablishContext")
    hresult, hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
    if hresult != SCARD_S_SUCCESS:
        raise EstablishContextException(hresult)

    print("SCardListReaders")
    hresult, readers = SCardListReaders(hcontext, [])
    if hresult != SCARD_S_SUCCESS:
        raise ListReadersException(hresult)
    reader = readers[0]
    print("Using reader:", reader)

    print("SCardConnect")
    hresult, hcard, dwActiveProtocol = SCardConnect(
        hcontext, reader, SCARD_SHARE_SHARED, SCARD_PROTOCOL_ANY
    )
    if hresult != SCARD_S_SUCCESS:
        raise BaseSCardException(hresult)

    return hcontext, hcard, reader, dwActiveProtocol


def thread_transaction():
    """Transcation on 2nd client"""
    # 2nd client
    hcontext2, hcard2, _reader, dwActiveProtocol = init_client()

    time.sleep(0.5)

    print("SCardBeginTransaction hcard2")
    hresult = SCardBeginTransaction(hcard2)
    if hresult != SCARD_S_SUCCESS:
        raise BaseSCardException(hresult)

    SELECT = [0x00, 0xA4, 0x00, 0x00, 0x02, 0x3F, 0x00]
    print("SCardTransmit hcard2")
    hresult, response = SCardTransmit(hcard2, dwActiveProtocol, SELECT)
    if hresult != SCARD_S_SUCCESS:
        raise BaseSCardException(hresult)
    print(response)

    time.sleep(5)

    print("SCardEndTransaction hcard2")
    hresult = SCardEndTransaction(hcard2, SCARD_LEAVE_CARD)
    if hresult == SCARD_E_SHARING_VIOLATION:
        print(RED + " ERROR!" + NORMAL)
    if hresult != SCARD_S_SUCCESS:
        raise BaseSCardException(hresult)

    print("SCardReleaseContext hcontext2")
    hresult = SCardReleaseContext(hcontext2)
    if hresult != SCARD_S_SUCCESS:
        raise ReleaseContextException(hresult)


def main():
    """main"""

    # 1st client
    hcontext1, _hcard1, _reader, dwActiveProtocol = init_client()

    # 3rd client
    hcontext3, hcard3, _reader, dwActiveProtocol = init_client()

    t = threading.Thread(target=thread_transaction)
    t.start()

    time.sleep(1)

    print("SCardReleaseContext hcontext1")
    hresult = SCardReleaseContext(hcontext1)
    if hresult not in (SCARD_S_SUCCESS, SCARD_E_SHARING_VIOLATION):
        raise ReleaseContextException(hresult)

    time.sleep(2)

    before = time.time()
    print("SCardBeginTransaction hcard3 (should block)")
    hresult = SCardBeginTransaction(hcard3)
    if hresult != SCARD_S_SUCCESS:
        raise BaseSCardException(hresult)
    after = time.time()
    delta = (after - before) * 1000
    print(f"Delta time: {delta} ms")
    if delta < 1000:
        print(RED + " ERROR! No blocking" + NORMAL)
    else:
        print(BLUE + "Blocking: OK" + NORMAL)

    SELECT = [0x00, 0xA4, 0x00, 0x00, 0x02, 0x3F, 0x00]
    print("SCardTransmit hcard3")
    hresult, response = SCardTransmit(hcard3, dwActiveProtocol, SELECT)
    if hresult != SCARD_S_SUCCESS:
        raise BaseSCardException(hresult)
    print(response)

    # wait for the thread
    t.join()

    print("SCardReleaseContext hcontext3")
    hresult = SCardReleaseContext(hcontext3)
    if hresult != SCARD_S_SUCCESS:
        raise ReleaseContextException(hresult)


if __name__ == "__main__":
    main()