VeraCrypt
aboutsummaryrefslogtreecommitdiff
path: root/src/Common/zlib/inffast.h
blob: e5c1aa4ca8cd5244423680865609c71ab68f9ab6 (plain) 
1
2
3
4
5
6
7
8
9
10
11

/* inffast.h -- header to use inffast.c
 * Copyright (C) 1995-2003, 2010 Mark Adler
 * For conditions of distribution and use, see copyright notice in zlib.h
 */

/* WARNING: this file should *not* be used by applications. It is
   part of the implementation of the compression library and is
   subject to change. Applications should only use zlib.h.
 */

void ZLIB_INTERNAL inflate_fast OF((z_streamp strm, unsigned start));
>188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 
#include "SCardLoader.h"
#include "PCSCException.h"

#ifndef TC_WINDOWS
#include <dlfcn.h>
#define LoadLibrary(x) dlopen(x, RTLD_NOW | RTLD_LOCAL)
#define FreeLibrary(x) dlclose(x)
#define GetProcAddress(x, y) dlsym(x, y)
typedef void* HMODULE;
#ifdef TC_MACOSX
#if !defined(USE_SCARD_CONTROL_112)
#define SCardControlName "SCardControl132"
#else
#define SCardControlName "SCardControl"
#endif
#else
#define SCardControlName "SCardControl"
#endif
#define SCardConnectName "SCardConnect"
#define SCardStatusName "SCardStatus"
#define SCardGetStatusChangeName "SCardGetStatusChange"
#define SCardListReaderGroupsName "SCardListReaderGroups"
#define SCardListReadersName "SCardListReaders"
#else
#define SCardControlName "SCardControl"
#define SCardConnectName "SCardConnectW"
#define SCardStatusName "SCardStatusW"
#define SCardGetStatusChangeName "SCardGetStatusChangeW"
#define SCardListReaderGroupsName "SCardListReaderGroupsW"
#define SCardListReadersName "SCardListReadersW"
#endif

using namespace std;

namespace VeraCrypt
{
	HMODULE SCardLoader::hScardModule = NULL;
	SCARDCONTEXT SCardLoader::hScardContext = 0;
	SCardEstablishContextPtr SCardLoader::scardEstablishContext = NULL;
	SCardReleaseContextPtr SCardLoader::scardReleaseContext = NULL;
	SCardIsValidContextPtr SCardLoader::scardIsValidContext = NULL;
#ifndef TC_MACOSX
	SCardFreeMemoryPtr SCardLoader::scardFreeMemory = NULL;
#endif
	SCardConnectPtr SCardLoader::scardConnect = NULL;
	SCardReconnectPtr SCardLoader::scardReconnect = NULL;
	SCardDisconnectPtr SCardLoader::scardDisconnect = NULL;     
	SCardBeginTransactionPtr SCardLoader::scardBeginTransaction = NULL;
	SCardEndTransactionPtr SCardLoader::scardEndTransaction = NULL;
	SCardStatusPtr SCardLoader::scardStatus = NULL;
	SCardGetStatusChangePtr SCardLoader::scardGetStatusChange = NULL;
	SCardControlPtr SCardLoader::scardControl = NULL;
	SCardTransmitPtr SCardLoader::scardTransmit = NULL;
	SCardListReaderGroupsPtr SCardLoader::scardListReaderGroups = NULL;
	SCardListReadersPtr SCardLoader::scardListReaders = NULL;
	SCardCancelPtr SCardLoader::scardCancel = NULL;
	SCardGetAttribPtr SCardLoader::scardGetAttrib = NULL;
	SCardSetAttribPtr SCardLoader::scardSetAttrib = NULL;
	SCARD_IO_REQUEST* SCardLoader::scardT0Pci = NULL;
	SCARD_IO_REQUEST* SCardLoader::scardT1Pci = NULL;
	SCARD_IO_REQUEST* SCardLoader::scardRawPci = NULL;
	bool SCardLoader::bInitialized = false;

#ifdef TC_WINDOWS
	wstring SCardLoader::GetSCardPath()
#else
	string SCardLoader::GetSCardPath()
#endif
	{
#ifdef TC_WINDOWS
		wchar_t winscardPath[TC_MAX_PATH];
		if (GetSystemDirectory(winscardPath, TC_MAX_PATH))
		{
			StringCbCat(winscardPath, sizeof(winscardPath), L"\\Winscard.dll");
		}
		else
			StringCbCopy(winscardPath, sizeof(winscardPath), L"C:\\Windows\\System32\\Winscard.dll");
		return winscardPath;
#elif TC_MACOSX
		return "/System/Library/Frameworks/PCSC.framework/PCSC";
#else
		string pcscPath = "";
		FILE* pipe = 
#ifdef TC_LINUX
			popen("ldconfig -p 2>&1", "r");
#else
			popen("ldconfig -r 2>&1", "r"); // FreeBSD
#endif
		if (pipe)
		{
			char buffer[128];
			while (!feof(pipe))
			{
				if (fgets(buffer, 128, pipe) != NULL)
				{
					string line(buffer);
					if (line.find("libpcsclite.so") != string::npos)
					{
						size_t pos = line.find("=>");
						if (pos != string::npos)
						{
							pcscPath = line.substr(pos + 3);
							pos = pcscPath.find_first_of(" \t\r\n");
							if (pos != string::npos)
								pcscPath = pcscPath.substr(0, pos);
							break;
						}
					}
				}
			}
			pclose(pipe);
		}

		if (pcscPath == "")
		{
			pcscPath = "libpcsclite.so";
		}
		
		return pcscPath;
#endif
	}

	void SCardLoader::Initialize()
	{
		if (bInitialized)
			return;

		hScardModule = LoadLibrary(GetSCardPath().c_str());
		if (hScardModule)
		{
			scardEstablishContext = (SCardEstablishContextPtr)GetProcAddress(hScardModule, "SCardEstablishContext");
			scardReleaseContext = (SCardReleaseContextPtr)GetProcAddress(hScardModule, "SCardReleaseContext");
			scardIsValidContext = (SCardIsValidContextPtr)GetProcAddress(hScardModule, "SCardIsValidContext");
#ifndef TC_MACOSX
			scardFreeMemory = (SCardFreeMemoryPtr)GetProcAddress(hScardModule, "SCardFreeMemory");
#endif
			scardConnect = (SCardConnectPtr)GetProcAddress(hScardModule, SCardConnectName);
			scardReconnect = (SCardReconnectPtr)GetProcAddress(hScardModule, "SCardReconnect");
			scardDisconnect = (SCardDisconnectPtr)GetProcAddress(hScardModule, "SCardDisconnect");
			scardBeginTransaction = (SCardBeginTransactionPtr)GetProcAddress(hScardModule, "SCardBeginTransaction");
			scardEndTransaction = (SCardEndTransactionPtr)GetProcAddress(hScardModule, "SCardEndTransaction");
			scardStatus = (SCardStatusPtr)GetProcAddress(hScardModule, SCardStatusName);
			scardGetStatusChange = (SCardGetStatusChangePtr)GetProcAddress(hScardModule, SCardGetStatusChangeName);
			scardControl = (SCardControlPtr)GetProcAddress(hScardModule, SCardControlName);
			scardTransmit = (SCardTransmitPtr)GetProcAddress(hScardModule, "SCardTransmit");
			scardListReaderGroups = (SCardListReaderGroupsPtr)GetProcAddress(hScardModule, SCardListReaderGroupsName);
			scardListReaders = (SCardListReadersPtr)GetProcAddress(hScardModule, SCardListReadersName);
			scardCancel = (SCardCancelPtr)GetProcAddress(hScardModule, "SCardCancel");
			scardGetAttrib = (SCardGetAttribPtr)GetProcAddress(hScardModule, "SCardGetAttrib");
			scardSetAttrib = (SCardSetAttribPtr)GetProcAddress(hScardModule, "SCardSetAttrib");
			scardT0Pci = (SCARD_IO_REQUEST*)GetProcAddress(hScardModule, "g_rgSCardT0Pci");
			scardT1Pci = (SCARD_IO_REQUEST*)GetProcAddress(hScardModule, "g_rgSCardT1Pci");
			scardRawPci = (SCARD_IO_REQUEST*)GetProcAddress(hScardModule, "g_rgSCardRawPci");
			if (
#ifndef TC_MACOSX
				scardFreeMemory &&
#endif
				scardEstablishContext && scardReleaseContext && scardIsValidContext && scardConnect && scardReconnect && scardDisconnect &&
				scardBeginTransaction && scardEndTransaction && scardStatus && scardGetStatusChange && scardControl && scardTransmit &&
				scardListReaderGroups && scardListReaders && scardCancel && scardGetAttrib && scardSetAttrib && scardT0Pci && scardT1Pci && scardRawPci)
			{
				if (SCARD_S_SUCCESS == scardEstablishContext(SCARD_SCOPE_SYSTEM, NULL, NULL, &hScardContext))
				{
					bInitialized = true;
				}
			}
		}

		if (!bInitialized)
		{
			Finalize();
		}
	}

	void SCardLoader::Finalize()
	{
		if (hScardContext)
		{
			scardReleaseContext(hScardContext);
			hScardContext = 0;
		}

		if (hScardModule)
		{
			FreeLibrary(hScardModule);
			hScardModule = NULL;
		}

		scardEstablishContext = NULL;
		scardReleaseContext = NULL;
		scardIsValidContext = NULL;
#ifndef TC_MACOSX
		scardFreeMemory = NULL;
#endif
		scardConnect = NULL;
		scardReconnect = NULL;
		scardDisconnect = NULL;
		scardBeginTransaction = NULL;
		scardEndTransaction = NULL;
		scardStatus = NULL;
		scardGetStatusChange = NULL;
		scardControl = NULL;
		scardTransmit = NULL;
		scardListReaderGroups = NULL;
		scardListReaders = NULL;
		scardCancel = NULL;
		scardGetAttrib = NULL;
		scardSetAttrib = NULL;
		scardT0Pci = NULL;
		scardT1Pci = NULL;
		scardRawPci = NULL;

		bInitialized = false;
	}

	SCARDCONTEXT SCardLoader::GetSCardContext()
	{
		return hScardContext;
	}

	LONG SCardLoader::SCardEstablishContext(DWORD dwScope, LPCVOID pvReserved1, LPCVOID pvReserved2, LPSCARDCONTEXT phContext)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardEstablishContext(dwScope, pvReserved1, pvReserved2, phContext);
	}

	LONG SCardLoader::SCardReleaseContext(SCARDCONTEXT hContext)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardReleaseContext(hContext);
	}

	LONG SCardLoader::SCardIsValidContext(SCARDCONTEXT hContext)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardIsValidContext(hContext);
	}

#ifndef TC_MACOSX
	LONG SCardLoader::SCardFreeMemory(SCARDCONTEXT hContext, LPCVOID pvMem)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardFreeMemory(hContext, pvMem);
	}
#endif

	LONG SCardLoader::SCardConnect(SCARDCONTEXT hContext, LPCTSTR szReader, DWORD dwShareMode, DWORD dwPreferredProtocols, LPSCARDHANDLE phCard, LPDWORD pdwActiveProtocol)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardConnect(hContext, szReader, dwShareMode, dwPreferredProtocols, phCard, pdwActiveProtocol);
	}
	
	LONG SCardLoader::SCardReconnect(SCARDHANDLE hCard, DWORD dwShareMode, DWORD dwPreferredProtocols, DWORD dwInitialization, LPDWORD pdwActiveProtocol)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardReconnect(hCard, dwShareMode, dwPreferredProtocols, dwInitialization, pdwActiveProtocol);
	}
	
	LONG SCardLoader::SCardDisconnect(SCARDHANDLE hCard, DWORD dwDisposition)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardDisconnect(hCard, dwDisposition);
	}
	
	LONG SCardLoader::SCardBeginTransaction(SCARDHANDLE hCard)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardBeginTransaction(hCard);
	}
	
	LONG SCardLoader::SCardEndTransaction(SCARDHANDLE hCard, DWORD dwDisposition)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardEndTransaction(hCard, dwDisposition);
	}
	
	LONG SCardLoader::SCardStatus(SCARDHANDLE hCard, LPTSTR mszReaderNames, LPDWORD pcchReaderLen, LPDWORD pdwState, LPDWORD pdwProtocol, BYTE* pbAtr, LPDWORD pcbAtrLen)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardStatus(hCard, mszReaderNames, pcchReaderLen, pdwState, pdwProtocol, pbAtr, pcbAtrLen);
	}
	
	LONG SCardLoader::SCardGetStatusChange(SCARDCONTEXT hContext, DWORD dwTimeout, LPSCARD_READERSTATE rgReaderStates, DWORD cReaders)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardGetStatusChange(hContext, dwTimeout, rgReaderStates, cReaders);
	}
	
	LONG SCardLoader::SCardControl(SCARDHANDLE hCard, DWORD dwControlCode, LPCVOID pbSendBuffer, DWORD cbSendLength, LPVOID pbRecvBuffer, DWORD cbRecvLength, LPDWORD lpBytesReturned)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardControl(hCard, dwControlCode, pbSendBuffer, cbSendLength, pbRecvBuffer, cbRecvLength, lpBytesReturned);
	}
	
	LONG SCardLoader::SCardTransmit(SCARDHANDLE hCard, LPCSCARD_IO_REQUEST pioSendPci, const BYTE* pbSendBuffer, DWORD cbSendLength, LPSCARD_IO_REQUEST pioRecvPci, BYTE* pbRecvBuffer, LPDWORD pcbRecvLength)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardTransmit(hCard, pioSendPci, pbSendBuffer, cbSendLength, pioRecvPci, pbRecvBuffer, pcbRecvLength);
	}
	
	LONG SCardLoader::SCardListReaderGroups(SCARDCONTEXT hContext, LPTSTR mszGroups, LPDWORD pcchGroups)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardListReaderGroups(hContext, mszGroups, pcchGroups);
	}
	
	LONG SCardLoader::SCardListReaders(SCARDCONTEXT hContext, LPCTSTR mszGroups, LPTSTR mszReaders, LPDWORD pcchReaders)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardListReaders(hContext, mszGroups, mszReaders, pcchReaders);
	}
	
	LONG SCardLoader::SCardCancel(SCARDCONTEXT hContext)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardCancel(hContext);
	}
	
	LONG SCardLoader::SCardGetAttrib(SCARDHANDLE hCard, DWORD dwAttrId, BYTE* pbAttr, LPDWORD pcbAttrLen)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardGetAttrib(hCard, dwAttrId, pbAttr, pcbAttrLen);
	}
	
	LONG SCardLoader::SCardSetAttrib(SCARDHANDLE hCard, DWORD dwAttrId, const BYTE* pbAttr, DWORD cbAttrLen)
	{
		Initialize();

		if (!bInitialized)
			throw ScardLibraryInitializationFailed();

		return scardSetAttrib(hCard, dwAttrId, pbAttr, cbAttrLen);
	}
}