File: ProviderContextPool.cpp

package info (click to toggle)
beid 3.5.2.dfsg-10
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 147,240 kB
  • ctags: 34,507
  • sloc: cpp: 149,944; ansic: 41,577; java: 8,927; cs: 6,528; sh: 2,426; perl: 1,866; xml: 805; python: 463; makefile: 263; lex: 92
file content (103 lines) | stat: -rw-r--r-- 2,571 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
/* ****************************************************************************

 * eID Middleware Project.
 * Copyright (C) 2008-2009 FedICT.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version
 * 3.0 as published by the Free Software Foundation.
 *
 * This software 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 this software; if not, see
 * http://www.gnu.org/licenses/.

**************************************************************************** */
#include "ProviderContextPool.h"
#include "../common/eidErrors.h"
#include "../common/MWException.h"

using namespace eIDMW;

CProviderContextPool::CProviderContextPool()
{
	m_ProvCounter = 0;
}

CProviderContextPool::~CProviderContextPool()
{
	CAutoMutex oAutMutex(&m_oMutex);

	while (m_pool.size() > 0)
	{
		tProvCtxPool::iterator it = m_pool.begin();
		delete(it->second);
		m_pool.erase(it);
	}
}

HCRYPTPROV CProviderContextPool::AddProviderContext(LPCSTR szContainer,
	DWORD dwFlags, PVTableProvStruc pVTable)
{
	CAutoMutex oAutMutex(&m_oMutex);

	// Provider handle = 1 for the 1st context, 2 for the 2nd context, ...
	HCRYPTPROV hProv = ++m_ProvCounter;

	CProviderContext *poProvCtx = new CProviderContext(hProv,
		szContainer, dwFlags, pVTable);
	if (poProvCtx == NULL)
		throw CMWEXCEPTION(EIDMW_ERR_MEMORY);

	m_pool[hProv] = poProvCtx;

	return hProv;
}

CProviderContext * CProviderContextPool::GetProviderContext(
	HCRYPTPROV hProv, bool bThrowException)
{
	CProviderContext *poProvCtx = NULL;

	CAutoMutex oAutMutex(&m_oMutex);

	tProvCtxPool::iterator it = m_pool.begin();
	for ( ; it != m_pool.end(); it++)
	{
		if (it->first == hProv)
		{
			poProvCtx = it->second;
			break;
		}
	}

	if (poProvCtx == NULL && bThrowException)
		throw CMWEXCEPTION(EIDMW_ERR_PARAM_BAD);

	return poProvCtx;
}

BOOL CProviderContextPool::DeleteProviderContext(HCRYPTPROV hProv)
{
	BOOL bRet = FALSE;

	CAutoMutex oAutMutex(&m_oMutex);

	tProvCtxPool::iterator it = m_pool.begin();
	for ( ; it != m_pool.end(); it++)
	{
		if (it->first == hProv)
		{
			delete it->second;
			m_pool.erase(it);
			bRet = TRUE;
			break;
		}
	}

	return bRet;
}