1 |
vletoux |
1 |
/* OpenPGP Smart Card Mini Driver |
2 |
|
|
Copyright (C) 2009 Vincent Le Toux |
3 |
|
|
|
4 |
|
|
This library is Free software; you can redistribute it and/or |
5 |
|
|
modify it under the terms of the GNU Lesser General Public |
6 |
|
|
License version 2.1 as published by the Free Software Foundation. |
7 |
|
|
|
8 |
|
|
This library is distributed in the hope that it will be useful, |
9 |
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
10 |
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
11 |
|
|
Lesser General Public License for more details. |
12 |
|
|
|
13 |
|
|
You should have received a copy of the GNU Lesser General Public |
14 |
|
|
License along with this library; if not, write to the Free Software |
15 |
|
|
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
16 |
|
|
*/ |
17 |
|
|
|
18 |
|
|
#include <windows.h> |
19 |
|
|
#include <tchar.h> |
20 |
|
|
#include <cardmod.h> |
21 |
|
|
|
22 |
|
|
#pragma comment(lib,"Scarddlg") |
23 |
|
|
#pragma comment(lib,"Winscard") |
24 |
|
|
|
25 |
|
|
HMODULE hModule = NULL; |
26 |
|
|
CARD_DATA CardData; |
27 |
|
|
PCARD_DATA pCardData = NULL; |
28 |
|
|
SCARD_ATRMASK atr; |
29 |
|
|
TCHAR szCard[256]; |
30 |
|
|
|
31 |
|
|
extern "C" { |
32 |
|
|
|
33 |
|
|
// |
34 |
|
|
// Heap helpers |
35 |
|
|
// |
36 |
|
|
|
37 |
|
|
LPVOID WINAPI _Alloc(__in SIZE_T cBytes) |
38 |
|
|
{ |
39 |
|
|
return HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, cBytes); |
40 |
|
|
} |
41 |
|
|
|
42 |
|
|
LPVOID WINAPI _ReAlloc( |
43 |
|
|
__in LPVOID pvMem, |
44 |
|
|
__in SIZE_T cBytes) |
45 |
|
|
{ |
46 |
|
|
return HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, pvMem, cBytes); |
47 |
|
|
} |
48 |
|
|
|
49 |
|
|
void WINAPI _Free( |
50 |
|
|
__in LPVOID pvMem) |
51 |
|
|
{ |
52 |
|
|
HeapFree(GetProcessHeap(), 0, pvMem); |
53 |
|
|
} |
54 |
|
|
|
55 |
|
|
// |
56 |
|
|
// Dummy data caching stubs to satisfy the card module callback requirements |
57 |
|
|
// |
58 |
|
|
|
59 |
|
|
DWORD WINAPI _CacheAddFileStub( |
60 |
|
|
IN PVOID pvCacheContext, |
61 |
|
|
IN LPWSTR wszTag, |
62 |
|
|
IN DWORD dwFlags, |
63 |
|
|
IN PBYTE pbData, |
64 |
|
|
IN DWORD cbData) |
65 |
|
|
{ |
66 |
|
|
UNREFERENCED_PARAMETER(pvCacheContext); |
67 |
|
|
UNREFERENCED_PARAMETER(wszTag); |
68 |
|
|
UNREFERENCED_PARAMETER(dwFlags); |
69 |
|
|
UNREFERENCED_PARAMETER(pbData); |
70 |
|
|
UNREFERENCED_PARAMETER(cbData); |
71 |
|
|
return ERROR_SUCCESS; |
72 |
|
|
} |
73 |
|
|
|
74 |
|
|
DWORD WINAPI _CacheLookupFileStub( |
75 |
|
|
IN PVOID pvCacheContext, |
76 |
|
|
IN LPWSTR wszTag, |
77 |
|
|
IN DWORD dwFlags, |
78 |
|
|
IN PBYTE *ppbData, |
79 |
|
|
IN PDWORD pcbData) |
80 |
|
|
{ |
81 |
|
|
UNREFERENCED_PARAMETER(pvCacheContext); |
82 |
|
|
UNREFERENCED_PARAMETER(wszTag); |
83 |
|
|
UNREFERENCED_PARAMETER(dwFlags); |
84 |
|
|
UNREFERENCED_PARAMETER(ppbData); |
85 |
|
|
UNREFERENCED_PARAMETER(pcbData); |
86 |
|
|
return ERROR_NOT_FOUND; |
87 |
|
|
} |
88 |
|
|
|
89 |
|
|
DWORD WINAPI _CacheDeleteFileStub( |
90 |
|
|
IN PVOID pvCacheContext, |
91 |
|
|
IN LPWSTR wszTag, |
92 |
|
|
IN DWORD dwFlags) |
93 |
|
|
{ |
94 |
|
|
UNREFERENCED_PARAMETER(pvCacheContext); |
95 |
|
|
UNREFERENCED_PARAMETER(wszTag); |
96 |
|
|
UNREFERENCED_PARAMETER(dwFlags); |
97 |
|
|
return ERROR_SUCCESS; |
98 |
|
|
} |
99 |
|
|
} |
100 |
|
|
|
101 |
|
|
DWORD Connect(BOOL fSystemDll) |
102 |
|
|
{ |
103 |
|
|
DWORD dwReturn = 0; |
104 |
|
|
SCARDCONTEXT hSCardContext = NULL; |
105 |
|
|
SCARDHANDLE hSCardHandle = NULL; |
106 |
|
|
TCHAR szCardModule[256]; |
107 |
|
|
TCHAR szReader[256]; |
108 |
|
|
DWORD dwCardModuleSize = ARRAYSIZE(szCardModule); |
109 |
|
|
DWORD dwReaderSize = ARRAYSIZE(szReader); |
110 |
|
|
OPENCARDNAME_EX dlgStruct; |
111 |
|
|
PFN_CARD_ACQUIRE_CONTEXT pfnCardAcquireContext; |
112 |
|
|
|
113 |
|
|
__try |
114 |
|
|
{ |
115 |
|
|
// find a smart card |
116 |
|
|
///////////////////// |
117 |
|
|
|
118 |
|
|
dwReturn = SCardEstablishContext(SCARD_SCOPE_USER, |
119 |
|
|
NULL, |
120 |
|
|
NULL, |
121 |
|
|
&hSCardContext ); |
122 |
|
|
if ( SCARD_S_SUCCESS != dwReturn ) |
123 |
|
|
{ |
124 |
|
|
__leave; |
125 |
|
|
} |
126 |
|
|
|
127 |
|
|
// Initialize the structure. |
128 |
|
|
memset(&dlgStruct, 0, sizeof(dlgStruct)); |
129 |
|
|
dlgStruct.dwStructSize = sizeof(dlgStruct); |
130 |
|
|
dlgStruct.hSCardContext = hSCardContext; |
131 |
|
|
dlgStruct.dwFlags = SC_DLG_MINIMAL_UI; |
132 |
|
|
dlgStruct.lpstrRdr = szReader; |
133 |
|
|
dlgStruct.nMaxRdr = dwReaderSize; |
134 |
|
|
dlgStruct.lpstrCard = szCard; |
135 |
|
|
dlgStruct.nMaxCard = ARRAYSIZE(szCard); |
136 |
|
|
dlgStruct.lpstrTitle = L"Select Card"; |
137 |
|
|
dlgStruct.dwShareMode = 0; |
138 |
|
|
// Display the select card dialog box. |
139 |
|
|
dwReturn = SCardUIDlgSelectCard(&dlgStruct); |
140 |
|
|
if ( SCARD_S_SUCCESS != dwReturn ) |
141 |
|
|
{ |
142 |
|
|
__leave; |
143 |
|
|
} |
144 |
|
|
|
145 |
|
|
// find the dll path / name |
146 |
|
|
//////////////////////////// |
147 |
|
|
if (fSystemDll) |
148 |
|
|
{ |
149 |
|
|
|
150 |
|
|
dwReturn = SCardGetCardTypeProviderName( |
151 |
|
|
hSCardContext, |
152 |
|
|
szCard, |
153 |
|
|
SCARD_PROVIDER_CARD_MODULE, |
154 |
|
|
(PTSTR) &szCardModule, |
155 |
|
|
&dwCardModuleSize); |
156 |
|
|
if (0 == dwCardModuleSize) |
157 |
|
|
{ |
158 |
|
|
dwReturn = (DWORD) SCARD_E_UNKNOWN_CARD; |
159 |
|
|
__leave; |
160 |
|
|
} |
161 |
|
|
} |
162 |
|
|
else |
163 |
|
|
{ |
164 |
vletoux |
8 |
#ifdef _M_X64 |
165 |
vletoux |
1 |
_tcscpy_s(szCardModule, dwCardModuleSize, TEXT("openpgpmdrv64.dll")); |
166 |
|
|
#else |
167 |
|
|
_tcscpy_s(szCardModule, dwCardModuleSize, TEXT("openpgpmdrv32.dll")); |
168 |
|
|
#endif |
169 |
|
|
} |
170 |
|
|
// connect to the smart card |
171 |
|
|
//////////////////////////// |
172 |
|
|
DWORD dwProtocol, dwState; |
173 |
|
|
dwReturn = SCardConnect(hSCardContext,szReader,SCARD_SHARE_SHARED,SCARD_PROTOCOL_T1|SCARD_PROTOCOL_T0, &hSCardHandle, &dwProtocol); |
174 |
|
|
if ( SCARD_S_SUCCESS != dwReturn ) |
175 |
|
|
{ |
176 |
|
|
__leave; |
177 |
|
|
} |
178 |
|
|
atr.cbAtr = 32; |
179 |
|
|
dwReturn = SCardStatus(hSCardHandle, szReader, &dwReaderSize, &dwState, &dwProtocol, atr.rgbAtr,&atr.cbAtr); |
180 |
|
|
if ( SCARD_S_SUCCESS != dwReturn ) |
181 |
|
|
{ |
182 |
|
|
__leave; |
183 |
|
|
} |
184 |
|
|
// load |
185 |
|
|
//////// |
186 |
|
|
if (NULL == (hModule = LoadLibrary(szCardModule))) |
187 |
|
|
{ |
188 |
|
|
dwReturn = GetLastError(); |
189 |
|
|
__leave; |
190 |
|
|
} |
191 |
|
|
|
192 |
|
|
if (NULL == (pfnCardAcquireContext = |
193 |
|
|
(PFN_CARD_ACQUIRE_CONTEXT) GetProcAddress( |
194 |
|
|
hModule, "CardAcquireContext"))) |
195 |
|
|
{ |
196 |
|
|
dwReturn = GetLastError(); |
197 |
|
|
__leave; |
198 |
|
|
} |
199 |
|
|
// initialize context |
200 |
|
|
////////////////////// |
201 |
|
|
pCardData = &CardData; |
202 |
|
|
pCardData->dwVersion = CARD_DATA_CURRENT_VERSION; |
203 |
|
|
pCardData->pfnCspAlloc = _Alloc; |
204 |
|
|
pCardData->pfnCspFree = _Free; |
205 |
|
|
pCardData->pfnCspReAlloc = _ReAlloc; |
206 |
|
|
pCardData->pfnCspCacheAddFile = _CacheAddFileStub; |
207 |
|
|
pCardData->pfnCspCacheLookupFile = _CacheLookupFileStub; |
208 |
|
|
pCardData->pfnCspCacheDeleteFile = _CacheDeleteFileStub; |
209 |
|
|
pCardData->hScard = hSCardHandle; |
210 |
|
|
pCardData->hSCardCtx = hSCardContext; |
211 |
|
|
pCardData->cbAtr = atr.cbAtr; |
212 |
|
|
pCardData->pbAtr = atr.rgbAtr; |
213 |
|
|
pCardData->pwszCardName = szCard; |
214 |
vletoux |
8 |
//dwReturn = SCardBeginTransaction(hSCardHandle); |
215 |
vletoux |
1 |
if ( SCARD_S_SUCCESS != dwReturn ) |
216 |
|
|
{ |
217 |
|
|
__leave; |
218 |
|
|
} |
219 |
|
|
dwReturn = pfnCardAcquireContext(pCardData, 0); |
220 |
|
|
} |
221 |
|
|
__finally |
222 |
|
|
{ |
223 |
|
|
if (dwReturn != 0) |
224 |
|
|
{ |
225 |
|
|
if (hSCardHandle) |
226 |
|
|
{ |
227 |
|
|
SCardEndTransaction(hSCardHandle,SCARD_LEAVE_CARD); |
228 |
|
|
SCardDisconnect(hSCardHandle,0); |
229 |
|
|
} |
230 |
|
|
if (hSCardContext) |
231 |
|
|
SCardReleaseContext(hSCardContext); |
232 |
|
|
} |
233 |
|
|
} |
234 |
|
|
return dwReturn; |
235 |
|
|
} |
236 |
|
|
|
237 |
|
|
DWORD Disconnect() |
238 |
|
|
{ |
239 |
|
|
DWORD dwReturn = 0; |
240 |
|
|
if (pCardData) |
241 |
|
|
{ |
242 |
|
|
if (pCardData->hScard) |
243 |
|
|
{ |
244 |
|
|
SCardEndTransaction(pCardData->hScard,SCARD_LEAVE_CARD); |
245 |
|
|
SCardDisconnect(pCardData->hScard,0); |
246 |
|
|
} |
247 |
|
|
if (pCardData->hSCardCtx) |
248 |
|
|
SCardReleaseContext(pCardData->hSCardCtx); |
249 |
|
|
pCardData = NULL; |
250 |
|
|
} |
251 |
|
|
else |
252 |
|
|
{ |
253 |
|
|
dwReturn = SCARD_E_COMM_DATA_LOST; |
254 |
|
|
} |
255 |
|
|
return dwReturn; |
256 |
|
|
} |