+//

+// Created by bshp on 1/14/23.

+//

+

+#ifndef NEWEMV_ICCDATAEXTRACTOR_H

+#define NEWEMV_ICCDATAEXTRACTOR_H

+

+#include <iostream>

+#include <stdio.h>

+#include <stdlib.h>

+#include <string.h>

+#include <sstream>

+#include <vector>

+#include <iomanip>

+#include <memory>

+#include "Platform/PlatformBase.h"

+#include "TLVParser.h"

+

+#ifdef __linux__

+#include <unistd.h>

+#endif

+

+#if defined (TC_WINDOWS) && !defined (TC_PROTOTYPE)

+# include "Exception.h"

+#else

+# include "Platform/Exception.h"

+#endif

+

+#ifdef TC_WINDOWS

+#include <winscard.h>

+#include <windows.h>

+#endif

+#ifdef TC_UNIX

+#undef BOOL

+#include <PCSC/winscard.h>

+using VeraCrypt::byte;

+#define BOOL int

+//#include <unistd.h> //Works without on windows

+#endif

+

+#ifdef _WIN32

+#include <windows.h>

+#endif

+

+#ifdef _WIN64

+#define ssize_t __int64

+#else

+#define ssize_t long

+#endif

+

+#define SELECT_TYPE_SIZE 12 /* Size of the SELECT_TYPE APDU */

+

+/* Winscard function pointers definitions for windows import */

+#ifdef TC_WINDOWS

+typedef LONG (WINAPI *SCardEstablishContextPtr)(DWORD dwScope,LPCVOID pvReserved1, LPCVOID pvReserved2, LPSCARDCONTEXT phContext);

+typedef LONG (WINAPI *SCardReleaseContextPtr)(SCARDCONTEXT hContext);

+typedef LONG (WINAPI *SCardConnectAPtr)(SCARDCONTEXT hContext,LPCSTR szReader,DWORD dwShareMode,DWORD dwPreferredProtocols,LPSCARDHANDLE phCard, LPDWORD pdwActiveProtocol);

+typedef LONG (WINAPI *SCardDisconnectPtr)(SCARDHANDLE hCard, DWORD dwDisposition);

+typedef LONG (WINAPI *SCardTransmitPtr)(SCARDHANDLE hCard,LPCSCARD_IO_REQUEST pioSendPci,const BYTE* pbSendBuffer, DWORD cbSendLength,LPSCARD_IO_REQUEST pioRecvPci,BYTE* pbRecvBuffer, LPDWORD pcbRecvLength);

+typedef LONG (WINAPI *SCardListReadersAPtr)(SCARDCONTEXT hContext,LPCSTR mszGroups,LPSTR mszReaders, LPDWORD pcchReaders);

+typedef LONG (WINAPI *SCardFreeMemoryPtr)(SCARDCONTEXT hContext,LPCVOID pvMem);

+#endif

+

+namespace VeraCrypt

+{

+ class IccDataExtractor {

+ private:

+

+ /* Used for loading winscard on windows */

+ #ifdef TC_WINDOWS

+ /* Winscard Library Handle */

+ HMODULE WinscardLibraryHandle;

+

+ /* Winscard function pointers */

+ SCardEstablishContextPtr WSCardEstablishContext;

+ SCardReleaseContextPtr WSCardReleaseContext;

+ SCardConnectAPtr WSCardConnectA;

+ SCardDisconnectPtr WSCardDisconnect;

+ SCardFreeMemoryPtr WSCardFreeMemory;

+ SCardListReadersAPtr WSCardListReadersA;

+ SCardTransmitPtr WSCardTransmit;

+

+ /* Is the winscard library loaded */

+ static bool Initialized;

+ #endif

+

+ /* SELECT_TYPES FOR DIFFERENT AIDs*/

+ const static BYTE SELECT_MASTERCARD[SELECT_TYPE_SIZE];

+ const static BYTE SELECT_VISA[SELECT_TYPE_SIZE];

+ const static BYTE SELECT_AMEX[SELECT_TYPE_SIZE];

+ const static BYTE * SELECT_TYPES[3];

+

+

+ SCARDCONTEXT hContext; /* Handle that identifies the resource manager context.*/

+

+ SCARDHANDLE hCard; /* A handle that identifies the connection to the smart card in the designated reader*/

+

+ std::vector<char*> readers; /* Card reader list */

+

+ unsigned long int nbReaders; /* Number of connected (available) readers */

+

+ LPSTR mszReaders; /* Names of the reader groups defined to the system, as a multi-string. Use a NULL value to

+ * list all readers in the system */

+

+ DWORD dwActiveProtocol; /* A flag that indicates the established active protocol.

+ * SCARD_PROTOCOL_T0: An asynchronous, character-oriented half-duplex transmission protocol.

+ * SCARD_PROTOCOL_T1: An asynchronous, block-oriented half-duplex transmission protocol.*/

+

+

+ /* Establishing the resource manager context (the scope) within which database operations are performed.

+ * The module of the smart card subsystem that manages access to multiple readers and smart cards. The

+ * resource manager identifies and tracks resources, allocates readers and resources across multiple

+ * applications,and supports transaction primitives for accessing services available on a given card.*/

+ int EstablishRSContext();

+

+ /* Connecting to the card in the given reader*/

+ int ConnectCard(unsigned long int reader_nb);

+

+ /* Disconnect the card currently connected*/

+ int DisconnectCard();

+

+ /* Testing if the card contains the application of the given EMV type */

+ bool TestingCardType(const int SELECT_TYPE_NUMBER);

+

+ /* Getting the ICC Public Key Certificates and the Issuer Public Key Certificates by parsing the application

+ * (!NEED TO TEST CARD TYPE TO SELECT APPLICATION FIRST!)*/

+ void GetCerts(vector<byte> &CERTS);

+

+ /* Getting CPCL data from the card and put it into a reference*/

+ void GetCPCL(vector<byte> &v);

+

+ /* Getting the PAN by parsing the application

+ * (!NEED TO TEST CARD TYPE TO SELECT APPLICATION FIRST!)*/

+ void GetPAN(vector<byte> &v);

+

+ /* Helper function to make a string from plain arrays and various standard containers of bytes */

+ template<typename TInputIter>

+ void make_hex_string(TInputIter first, TInputIter last, std::string& panString, bool use_uppercase = true, bool insert_spaces = false);

+

+ public:

+ IccDataExtractor();

+

+ ~IccDataExtractor();

+

+ /* Used to initialize the winscard library on windows to make sure the dll is in System32 */

+ #ifdef TC_WINDOWS

+ void IccDataExtractor::InitLibrary();

+ #endif

+

+ /* Detecting available readers and filling the reader table. Returns

+ * the number of available readers */

+ unsigned long GetReaders();

+

+

+ /* Getting an ICC Public Key Certificates, an Issuer Public Key Certificates and the CPCL data

+ * from the card designated by the reader number. Appending them into a byte vector */

+ void GettingAllCerts(int readerNumber, vector<byte> &v);

+

+ /* Getting the PAN from the card designated by the reader number */

+ void GettingPAN(int readerNumber, string& panString);

+ };

+

+ struct PCSCException: public Exception

+ {

+ PCSCException(LONG errorCode = (LONG) -1): ErrorCode(errorCode), SubjectErrorCodeValid(false), SubjectErrorCode((uint64)-1){}

+ PCSCException(LONG errorCode, uint64 subjectErrorCode): ErrorCode(errorCode), SubjectErrorCodeValid(true), SubjectErrorCode(subjectErrorCode){}

+

+ #ifdef TC_HEADER_Platform_Exception

+ virtual ~PCSCException() throw () { }

+ TC_SERIALIZABLE_EXCEPTION(PCSCException);

+ #else

+

+ void Show(HWND parent) const;

+ #endif

+

+ operator string () const;

+ LONG GetErrorCode() const { return ErrorCode; }

+

+ protected:

+ LONG ErrorCode;

+ bool SubjectErrorCodeValid;

+ uint64 SubjectErrorCode;

+ };

+

+ #ifdef TC_HEADER_Platform_Exception

+

+ #define TC_EXCEPTION(NAME) TC_EXCEPTION_DECL(NAME,Exception)

+

+ #undef TC_EXCEPTION_SET

+ #define TC_EXCEPTION_SET \

+ TC_EXCEPTION_NODECL (PCSCException); \

+ TC_EXCEPTION (WinscardLibraryNotInitialized); \

+ TC_EXCEPTION (InvalidEMVPath); \

+ TC_EXCEPTION (EMVKeyfileDataNotFound); \

+ TC_EXCEPTION (EMVPANNotFound); \

+ TC_EXCEPTION (EMVUnknownCardType);

+ TC_EXCEPTION_SET;

+

+ #undef TC_EXCEPTION

+

+ #else // !TC_HEADER_Platform_Exception

+

+ struct WinscardLibraryNotInitialized: public Exception

+ {

+ void Show(HWND parent) const { Error("WINSCARD_MODULE_INIT_FAILED", parent); }

+ };

+

+ struct InvalidEMVPath: public Exception

+ {

+ void Show(HWND parent) const { Error("INVALID_EMV_PATH", parent); }

+ };

+

+ struct EMVKeyfileDataNotFound: public Exception

+ {

+ void Show(HWND parent) const { Error("EMV_KEYFILE_DATA_NOT_FOUND", parent); }

+ };

+

+ struct EMVPANNotFound: public Exception

+ {

+ void Show(HWND parent) const { Error("EMV_PAN_NOT_FOUND", parent); }

+ };

+

+ struct EMVUnknownCardType: public Exception

+ {

+ void Show(HWND parent) const { Error("EMV_UNKNOWN_CARD_TYPE", parent); }

+ };

+

+ #endif // !TC_HEADER_Platform_Exception

+}

+

+#endif //NEWEMV_ICCDATAEXTRACTOR_H