File: SCardExclusiveBehaviour.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 (128 lines) | stat: -rwxr-xr-x 4,303 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
#!/usr/bin/env python3

"""
# Check the return value of SCardConnect() on a reader already used in
# SCARD_SHARE_EXCLUSIVE mode
"""

# Copyright (c) 2010 Jean-Luc Giraud (jlgiraud@mac.com)
# All rights reserved.

# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
#
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
# 3. The name of the author may not be used to endorse or promote products
#    derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
# IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
# OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
# IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
# NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
# THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.


import sys

from smartcard.Exceptions import NoReadersException
from smartcard.pcsc.PCSCExceptions import (
    EstablishContextException,
    ListReadersException,
)
from smartcard.scard import (
    SCARD_E_SHARING_VIOLATION,
    SCARD_PROTOCOL_ANY,
    SCARD_S_SUCCESS,
    SCARD_SCOPE_USER,
    SCARD_SHARE_EXCLUSIVE,
    SCARD_SHARE_SHARED,
    SCardConnect,
    SCardDisconnect,
    SCardEstablishContext,
    SCardGetErrorMessage,
    SCardListReaders,
)


def Connect(mode):
    """Connect"""
    hresult, hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
    if hresult != SCARD_S_SUCCESS:
        raise EstablishContextException(hresult)

    hresult, readers = SCardListReaders(hcontext, [])
    if hresult != SCARD_S_SUCCESS:
        raise ListReadersException(hresult)
    print("PC/SC Readers:", readers)
    if len(readers) <= 0:
        raise NoReadersException()
    reader = readers[0]
    print("Using reader:", reader)

    hresult, hcard, _dwActiveProtocol = SCardConnect(
        hcontext, reader, mode, SCARD_PROTOCOL_ANY
    )
    return hresult, hcontext, hcard, reader


def ConnectWithReader(readerName, mode):
    """ConnectWithReader"""
    hresult, hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
    if hresult != SCARD_S_SUCCESS:
        raise EstablishContextException(hresult)

    hresult, hcard, _dwActiveProtocol = SCardConnect(
        hcontext, readerName, mode, SCARD_PROTOCOL_ANY
    )
    return hresult, hcontext, hcard


def main():
    """docstring for main"""

    hresult, _hcontext1, hcard1, readerName = Connect(SCARD_SHARE_EXCLUSIVE)
    if hresult != SCARD_S_SUCCESS:
        print(f"Test1: Error creating first exclusive connection 0x{hresult:X}")
        return

    hresult, _hcontext2, _hcardTest2 = ConnectWithReader(readerName, SCARD_SHARE_SHARED)
    if hresult != SCARD_E_SHARING_VIOLATION:
        print(
            f"Test1: Expected \
{SCardGetErrorMessage(SCARD_E_SHARING_VIOLATION)} (SCARD_E_SHARING_VIOLATION) \
but got {SCardGetErrorMessage(hresult)}"
        )
        return

    SCardDisconnect(hcard1, 0)

    hresult, _hcontext1, hcard1 = ConnectWithReader(readerName, SCARD_SHARE_SHARED)
    if hresult != SCARD_S_SUCCESS:
        print(f"Test2: Error creating first shared connection {hresult:X}")
        return

    hresult, _hcontext2, _hcardTest2 = ConnectWithReader(
        readerName, SCARD_SHARE_EXCLUSIVE
    )
    if hresult != SCARD_E_SHARING_VIOLATION:
        print(
            "Test2: Expected \
{SCardGetErrorMessage(SCARD_E_SHARING_VIOLATION)} (SCARD_E_SHARING_VIOLATION) \
but got {SCardGetErrorMessage(hresult)}"
        )
        sys.exit(1)
    print("Test successful")


if __name__ == "__main__":
    main()