ISMAIL MASSERAN
129 lines
4.3 KiB
Python
129 lines
4.3 KiB
Python
"""PCSCReaderGroups organizes smartcard readers as groups.
|
|
|
|
__author__ = "gemalto 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
|
|
"""
|
|
|
|
from smartcard.pcsc.PCSCExceptions import (
|
|
EstablishContextException,
|
|
ListReadersException,
|
|
ReleaseContextException,
|
|
)
|
|
from smartcard.reader.ReaderGroups import innerreadergroups, readergroups
|
|
from smartcard.scard import (
|
|
SCARD_S_SUCCESS,
|
|
SCARD_SCOPE_USER,
|
|
SCardEstablishContext,
|
|
SCardForgetReaderGroup,
|
|
SCardGetErrorMessage,
|
|
SCardIntroduceReaderGroup,
|
|
SCardListReaderGroups,
|
|
SCardReleaseContext,
|
|
error,
|
|
)
|
|
|
|
# pylint: disable=too-few-public-methods
|
|
|
|
|
|
class pcscinnerreadergroups(innerreadergroups):
|
|
"""Smartcard PCSC readers groups inner class.
|
|
|
|
The PCSCReaderGroups singleton manages the creation of the unique
|
|
instance of this class.
|
|
"""
|
|
|
|
def __init__(self, initlist=None):
|
|
"""Constructor."""
|
|
innerreadergroups.__init__(self, initlist)
|
|
self.unremovablegroups = ["SCard$DefaultReaders"]
|
|
|
|
def getreadergroups(self):
|
|
"""Returns the list of smartcard reader groups."""
|
|
innerreadergroups.getreadergroups(self)
|
|
|
|
hresult, hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
|
|
if hresult != SCARD_S_SUCCESS:
|
|
raise EstablishContextException(hresult)
|
|
hresult, readers = SCardListReaderGroups(hcontext)
|
|
if hresult != SCARD_S_SUCCESS:
|
|
raise ListReadersException(hresult)
|
|
hresult = SCardReleaseContext(hcontext)
|
|
if hresult != SCARD_S_SUCCESS:
|
|
raise ReleaseContextException(hresult)
|
|
return readers
|
|
|
|
def addreadergroup(self, newgroup):
|
|
"""Add a reader group"""
|
|
|
|
hresult, hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
|
|
if SCARD_S_SUCCESS != hresult:
|
|
raise error("Failed to establish context: " + SCardGetErrorMessage(hresult))
|
|
try:
|
|
hresult = SCardIntroduceReaderGroup(hcontext, newgroup)
|
|
if SCARD_S_SUCCESS != hresult:
|
|
raise error(
|
|
"Unable to introduce reader group: " + SCardGetErrorMessage(hresult)
|
|
)
|
|
|
|
innerreadergroups.addreadergroup(self, newgroup)
|
|
|
|
finally:
|
|
hresult = SCardReleaseContext(hcontext)
|
|
if SCARD_S_SUCCESS != hresult:
|
|
raise error(
|
|
"Failed to release context: " + SCardGetErrorMessage(hresult)
|
|
)
|
|
|
|
def removereadergroup(self, group):
|
|
"""Remove a reader group"""
|
|
|
|
hresult, hcontext = SCardEstablishContext(SCARD_SCOPE_USER)
|
|
if SCARD_S_SUCCESS != hresult:
|
|
raise error("Failed to establish context: " + SCardGetErrorMessage(hresult))
|
|
try:
|
|
hresult = SCardForgetReaderGroup(hcontext, group)
|
|
if hresult != SCARD_S_SUCCESS:
|
|
raise error(
|
|
"Unable to forget reader group: " + SCardGetErrorMessage(hresult)
|
|
)
|
|
|
|
innerreadergroups.removereadergroup(self, group)
|
|
|
|
finally:
|
|
hresult = SCardReleaseContext(hcontext)
|
|
if SCARD_S_SUCCESS != hresult:
|
|
raise error(
|
|
"Failed to release context: " + SCardGetErrorMessage(hresult)
|
|
)
|
|
|
|
|
|
class PCSCReaderGroups(readergroups):
|
|
"""PCSC readers groups."""
|
|
|
|
def __init__(self, initlist=None):
|
|
"""Create a single instance of pcscinnerreadergroups on first call"""
|
|
self.innerclazz = pcscinnerreadergroups
|
|
readergroups.__init__(self, initlist)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print(PCSCReaderGroups().getreadergroups())
|