Annotation of embedaddon/strongswan/src/libstrongswan/credentials/credential_factory.c, revision 1.1.1.1
1.1 misho 1: /*
2: * Copyright (C) 2008 Martin Willi
3: * HSR Hochschule fuer Technik Rapperswil
4: *
5: * This program is free software; you can redistribute it and/or modify it
6: * under the terms of the GNU General Public License as published by the
7: * Free Software Foundation; either version 2 of the License, or (at your
8: * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
9: *
10: * This program is distributed in the hope that it will be useful, but
11: * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
12: * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
13: * for more details.
14: */
15:
16: #include <stdint.h>
17:
18: #include "credential_factory.h"
19:
20: #include <utils/debug.h>
21: #include <collections/linked_list.h>
22: #include <threading/thread_value.h>
23: #include <threading/rwlock.h>
24: #include <credentials/certificates/x509.h>
25: #include <credentials/containers/container.h>
26:
27: ENUM(credential_type_names, CRED_PRIVATE_KEY, CRED_CONTAINER,
28: "CRED_PRIVATE_KEY",
29: "CRED_PUBLIC_KEY",
30: "CRED_CERTIFICATE",
31: "CRED_CONTAINER",
32: );
33:
34: typedef struct private_credential_factory_t private_credential_factory_t;
35:
36: /**
37: * private data of credential_factory
38: */
39: struct private_credential_factory_t {
40:
41: /**
42: * public functions
43: */
44: credential_factory_t public;
45:
46: /**
47: * list with entry_t
48: */
49: linked_list_t *constructors;
50:
51: /**
52: * Thread specific recursiveness counter
53: */
54: thread_value_t *recursive;
55:
56: /**
57: * lock access to builders
58: */
59: rwlock_t *lock;
60: };
61:
62: typedef struct entry_t entry_t;
63: struct entry_t {
64: /** kind of credential builder */
65: credential_type_t type;
66: /** subtype of credential, e.g. certificate_type_t */
67: int subtype;
68: /** registered with final flag? */
69: bool final;
70: /** builder function */
71: builder_function_t constructor;
72: };
73:
74: METHOD(credential_factory_t, add_builder, void,
75: private_credential_factory_t *this, credential_type_t type, int subtype,
76: bool final, builder_function_t constructor)
77: {
78: entry_t *entry = malloc_thing(entry_t);
79:
80: entry->type = type;
81: entry->subtype = subtype;
82: entry->final = final;
83: entry->constructor = constructor;
84: this->lock->write_lock(this->lock);
85: this->constructors->insert_last(this->constructors, entry);
86: this->lock->unlock(this->lock);
87: }
88:
89: METHOD(credential_factory_t, remove_builder, void,
90: private_credential_factory_t *this, builder_function_t constructor)
91: {
92: enumerator_t *enumerator;
93: entry_t *entry;
94:
95: this->lock->write_lock(this->lock);
96: enumerator = this->constructors->create_enumerator(this->constructors);
97: while (enumerator->enumerate(enumerator, &entry))
98: {
99: if (entry->constructor == constructor)
100: {
101: this->constructors->remove_at(this->constructors, enumerator);
102: free(entry);
103: }
104: }
105: enumerator->destroy(enumerator);
106: this->lock->unlock(this->lock);
107: }
108:
109: METHOD(credential_factory_t, create, void*,
110: private_credential_factory_t *this, credential_type_t type, int subtype, ...)
111: {
112: enumerator_t *enumerator;
113: entry_t *entry;
114: va_list args;
115: void *construct = NULL;
116: int failures = 0;
117: uintptr_t level;
118:
119: level = (uintptr_t)this->recursive->get(this->recursive);
120: this->recursive->set(this->recursive, (void*)level + 1);
121:
122: this->lock->read_lock(this->lock);
123: enumerator = this->constructors->create_enumerator(this->constructors);
124: while (enumerator->enumerate(enumerator, &entry))
125: {
126: if (entry->type == type && entry->subtype == subtype)
127: {
128: va_start(args, subtype);
129: construct = entry->constructor(subtype, args);
130: va_end(args);
131: if (construct)
132: {
133: break;
134: }
135: failures++;
136: }
137: }
138: enumerator->destroy(enumerator);
139: this->lock->unlock(this->lock);
140:
141: if (!construct && !level)
142: {
143: enum_name_t *names;
144:
145: switch (type)
146: {
147: case CRED_CERTIFICATE:
148: names = certificate_type_names;
149: break;
150: case CRED_CONTAINER:
151: names = container_type_names;
152: break;
153: case CRED_PRIVATE_KEY:
154: case CRED_PUBLIC_KEY:
155: default:
156: names = key_type_names;
157: break;
158: }
159: DBG1(DBG_LIB, "building %N - %N failed, tried %d builders",
160: credential_type_names, type, names, subtype, failures);
161: }
162: this->recursive->set(this->recursive, (void*)level);
163: return construct;
164: }
165:
166: CALLBACK(builder_filter, bool,
167: void *null, enumerator_t *orig, va_list args)
168: {
169: entry_t *entry;
170: credential_type_t *type;
171: int *subtype;
172:
173: VA_ARGS_VGET(args, type, subtype);
174:
175: while (orig->enumerate(orig, &entry))
176: {
177: if (entry->final)
178: {
179: *type = entry->type;
180: *subtype = entry->subtype;
181: return TRUE;
182: }
183: }
184: return FALSE;
185: }
186:
187: METHOD(credential_factory_t, create_builder_enumerator, enumerator_t*,
188: private_credential_factory_t *this)
189: {
190: this->lock->read_lock(this->lock);
191: return enumerator_create_filter(
192: this->constructors->create_enumerator(this->constructors),
193: builder_filter, this->lock, (void*)this->lock->unlock);
194: }
195:
196: METHOD(credential_factory_t, destroy, void,
197: private_credential_factory_t *this)
198: {
199: this->constructors->destroy_function(this->constructors, free);
200: this->recursive->destroy(this->recursive);
201: this->lock->destroy(this->lock);
202: free(this);
203: }
204:
205: /*
206: * see header file
207: */
208: credential_factory_t *credential_factory_create()
209: {
210: private_credential_factory_t *this;
211:
212: INIT(this,
213: .public = {
214: .create = _create,
215: .create_builder_enumerator = _create_builder_enumerator,
216: .add_builder = _add_builder,
217: .remove_builder = _remove_builder,
218: .destroy = _destroy,
219: },
220: .constructors = linked_list_create(),
221: .recursive = thread_value_create(NULL),
222: .lock = rwlock_create(RWLOCK_TYPE_DEFAULT),
223: );
224:
225: return &this->public;
226: }
227:
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>