1 /*
2  * EAP-TNC - TNCS (IF-IMV, IF-TNCCS, and IF-TNCCS-SOH)
3  * Copyright (c) 2007-2008, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14 
15 #include "includes.h"
16 #include <dlfcn.h>
17 
18 #include "common.h"
19 #include "base64.h"
20 #include "tncs.h"
21 #include "eap_common/eap_tlv_common.h"
22 #include "eap_common/eap_defs.h"
23 
24 
25 /* TODO: TNCS must be thread-safe; review the code and add locking etc. if
26  * needed.. */
27 
28 #define TNC_CONFIG_FILE "/etc/tnc_config"
29 #define IF_TNCCS_START \
30 "<?xml version=\"1.0\"?>\n" \
31 "<TNCCS-Batch BatchId=\"%d\" Recipient=\"TNCS\" " \
32 "xmlns=\"http://www.trustedcomputinggroup.org/IWG/TNC/1_0/IF_TNCCS#\" " \
33 "xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" " \
34 "xsi:schemaLocation=\"http://www.trustedcomputinggroup.org/IWG/TNC/1_0/" \
35 "IF_TNCCS# https://www.trustedcomputinggroup.org/XML/SCHEMA/TNCCS_1.0.xsd\">\n"
36 #define IF_TNCCS_END "\n</TNCCS-Batch>"
37 
38 /* TNC IF-IMV */
39 
40 typedef unsigned long TNC_UInt32;
41 typedef unsigned char *TNC_BufferReference;
42 
43 typedef TNC_UInt32 TNC_IMVID;
44 typedef TNC_UInt32 TNC_ConnectionID;
45 typedef TNC_UInt32 TNC_ConnectionState;
46 typedef TNC_UInt32 TNC_RetryReason;
47 typedef TNC_UInt32 TNC_IMV_Action_Recommendation;
48 typedef TNC_UInt32 TNC_IMV_Evaluation_Result;
49 typedef TNC_UInt32 TNC_MessageType;
50 typedef TNC_MessageType *TNC_MessageTypeList;
51 typedef TNC_UInt32 TNC_VendorID;
52 typedef TNC_UInt32 TNC_Subtype;
53 typedef TNC_UInt32 TNC_Version;
54 typedef TNC_UInt32 TNC_Result;
55 typedef TNC_UInt32 TNC_AttributeID;
56 
57 typedef TNC_Result (*TNC_TNCS_BindFunctionPointer)(
58 	TNC_IMVID imvID,
59 	char *functionName,
60 	void **pOutfunctionPointer);
61 
62 #define TNC_RESULT_SUCCESS 0
63 #define TNC_RESULT_NOT_INITIALIZED 1
64 #define TNC_RESULT_ALREADY_INITIALIZED 2
65 #define TNC_RESULT_NO_COMMON_VERSION 3
66 #define TNC_RESULT_CANT_RETRY 4
67 #define TNC_RESULT_WONT_RETRY 5
68 #define TNC_RESULT_INVALID_PARAMETER 6
69 #define TNC_RESULT_CANT_RESPOND 7
70 #define TNC_RESULT_ILLEGAL_OPERATION 8
71 #define TNC_RESULT_OTHER 9
72 #define TNC_RESULT_FATAL 10
73 
74 #define TNC_CONNECTION_STATE_CREATE 0
75 #define TNC_CONNECTION_STATE_HANDSHAKE 1
76 #define TNC_CONNECTION_STATE_ACCESS_ALLOWED 2
77 #define TNC_CONNECTION_STATE_ACCESS_ISOLATED 3
78 #define TNC_CONNECTION_STATE_ACCESS_NONE 4
79 #define TNC_CONNECTION_STATE_DELETE 5
80 
81 #define TNC_IFIMV_VERSION_1 1
82 
83 #define TNC_VENDORID_ANY ((TNC_VendorID) 0xffffff)
84 #define TNC_SUBTYPE_ANY ((TNC_Subtype) 0xff)
85 
86 /* TNCC-TNCS Message Types */
87 #define TNC_TNCCS_RECOMMENDATION		0x00000001
88 #define TNC_TNCCS_ERROR				0x00000002
89 #define TNC_TNCCS_PREFERREDLANGUAGE		0x00000003
90 #define TNC_TNCCS_REASONSTRINGS			0x00000004
91 
92 /* Possible TNC_IMV_Action_Recommendation values: */
93 enum IMV_Action_Recommendation {
94 	TNC_IMV_ACTION_RECOMMENDATION_ALLOW,
95 	TNC_IMV_ACTION_RECOMMENDATION_NO_ACCESS,
96 	TNC_IMV_ACTION_RECOMMENDATION_ISOLATE,
97 	TNC_IMV_ACTION_RECOMMENDATION_NO_RECOMMENDATION
98 };
99 
100 /* Possible TNC_IMV_Evaluation_Result values: */
101 enum IMV_Evaluation_Result {
102 	TNC_IMV_EVALUATION_RESULT_COMPLIANT,
103 	TNC_IMV_EVALUATION_RESULT_NONCOMPLIANT_MINOR,
104 	TNC_IMV_EVALUATION_RESULT_NONCOMPLIANT_MAJOR,
105 	TNC_IMV_EVALUATION_RESULT_ERROR,
106 	TNC_IMV_EVALUATION_RESULT_DONT_KNOW
107 };
108 
109 struct tnc_if_imv {
110 	struct tnc_if_imv *next;
111 	char *name;
112 	char *path;
113 	void *dlhandle; /* from dlopen() */
114 	TNC_IMVID imvID;
115 	TNC_MessageTypeList supported_types;
116 	size_t num_supported_types;
117 
118 	/* Functions implemented by IMVs (with TNC_IMV_ prefix) */
119 	TNC_Result (*Initialize)(
120 		TNC_IMVID imvID,
121 		TNC_Version minVersion,
122 		TNC_Version maxVersion,
123 		TNC_Version *pOutActualVersion);
124 	TNC_Result (*NotifyConnectionChange)(
125 		TNC_IMVID imvID,
126 		TNC_ConnectionID connectionID,
127 		TNC_ConnectionState newState);
128 	TNC_Result (*ReceiveMessage)(
129 		TNC_IMVID imvID,
130 		TNC_ConnectionID connectionID,
131 		TNC_BufferReference message,
132 		TNC_UInt32 messageLength,
133 		TNC_MessageType messageType);
134 	TNC_Result (*SolicitRecommendation)(
135 		TNC_IMVID imvID,
136 		TNC_ConnectionID connectionID);
137 	TNC_Result (*BatchEnding)(
138 		TNC_IMVID imvID,
139 		TNC_ConnectionID connectionID);
140 	TNC_Result (*Terminate)(TNC_IMVID imvID);
141 	TNC_Result (*ProvideBindFunction)(
142 		TNC_IMVID imvID,
143 		TNC_TNCS_BindFunctionPointer bindFunction);
144 };
145 
146 
147 #define TNC_MAX_IMV_ID 10
148 
149 struct tncs_data {
150 	struct tncs_data *next;
151 	struct tnc_if_imv *imv; /* local copy of tncs_global_data->imv */
152 	TNC_ConnectionID connectionID;
153 	unsigned int last_batchid;
154 	enum IMV_Action_Recommendation recommendation;
155 	int done;
156 
157 	struct conn_imv {
158 		u8 *imv_send;
159 		size_t imv_send_len;
160 		enum IMV_Action_Recommendation recommendation;
161 		int recommendation_set;
162 	} imv_data[TNC_MAX_IMV_ID];
163 
164 	char *tncs_message;
165 };
166 
167 
168 struct tncs_global {
169 	struct tnc_if_imv *imv;
170 	TNC_ConnectionID next_conn_id;
171 	struct tncs_data *connections;
172 };
173 
174 static struct tncs_global *tncs_global_data = NULL;
175 
176 
177 static struct tnc_if_imv * tncs_get_imv(TNC_IMVID imvID)
178 {
179 	struct tnc_if_imv *imv;
180 
181 	if (imvID >= TNC_MAX_IMV_ID || tncs_global_data == NULL)
182 		return NULL;
183 	imv = tncs_global_data->imv;
184 	while (imv) {
185 		if (imv->imvID == imvID)
186 			return imv;
187 		imv = imv->next;
188 	}
189 	return NULL;
190 }
191 
192 
193 static struct tncs_data * tncs_get_conn(TNC_ConnectionID connectionID)
194 {
195 	struct tncs_data *tncs;
196 
197 	if (tncs_global_data == NULL)
198 		return NULL;
199 
200 	tncs = tncs_global_data->connections;
201 	while (tncs) {
202 		if (tncs->connectionID == connectionID)
203 			return tncs;
204 		tncs = tncs->next;
205 	}
206 
207 	wpa_printf(MSG_DEBUG, "TNC: Connection ID %lu not found",
208 		   (unsigned long) connectionID);
209 
210 	return NULL;
211 }
212 
213 
214 /* TNCS functions that IMVs can call */
215 TNC_Result TNC_TNCS_ReportMessageTypes(
216 	TNC_IMVID imvID,
217 	TNC_MessageTypeList supportedTypes,
218 	TNC_UInt32 typeCount)
219 {
220 	TNC_UInt32 i;
221 	struct tnc_if_imv *imv;
222 
223 	wpa_printf(MSG_DEBUG, "TNC: TNC_TNCS_ReportMessageTypes(imvID=%lu "
224 		   "typeCount=%lu)",
225 		   (unsigned long) imvID, (unsigned long) typeCount);
226 
227 	for (i = 0; i < typeCount; i++) {
228 		wpa_printf(MSG_DEBUG, "TNC: supportedTypes[%lu] = %lu",
229 			   i, supportedTypes[i]);
230 	}
231 
232 	imv = tncs_get_imv(imvID);
233 	if (imv == NULL)
234 		return TNC_RESULT_INVALID_PARAMETER;
235 	os_free(imv->supported_types);
236 	imv->supported_types =
237 		os_malloc(typeCount * sizeof(TNC_MessageTypeList));
238 	if (imv->supported_types == NULL)
239 		return TNC_RESULT_FATAL;
240 	os_memcpy(imv->supported_types, supportedTypes,
241 		  typeCount * sizeof(TNC_MessageTypeList));
242 	imv->num_supported_types = typeCount;
243 
244 	return TNC_RESULT_SUCCESS;
245 }
246 
247 
248 TNC_Result TNC_TNCS_SendMessage(
249 	TNC_IMVID imvID,
250 	TNC_ConnectionID connectionID,
251 	TNC_BufferReference message,
252 	TNC_UInt32 messageLength,
253 	TNC_MessageType messageType)
254 {
255 	struct tncs_data *tncs;
256 	unsigned char *b64;
257 	size_t b64len;
258 
259 	wpa_printf(MSG_DEBUG, "TNC: TNC_TNCS_SendMessage(imvID=%lu "
260 		   "connectionID=%lu messageType=%lu)",
261 		   imvID, connectionID, messageType);
262 	wpa_hexdump_ascii(MSG_DEBUG, "TNC: TNC_TNCS_SendMessage",
263 			  message, messageLength);
264 
265 	if (tncs_get_imv(imvID) == NULL)
266 		return TNC_RESULT_INVALID_PARAMETER;
267 
268 	tncs = tncs_get_conn(connectionID);
269 	if (tncs == NULL)
270 		return TNC_RESULT_INVALID_PARAMETER;
271 
272 	b64 = base64_encode(message, messageLength, &b64len);
273 	if (b64 == NULL)
274 		return TNC_RESULT_FATAL;
275 
276 	os_free(tncs->imv_data[imvID].imv_send);
277 	tncs->imv_data[imvID].imv_send_len = 0;
278 	tncs->imv_data[imvID].imv_send = os_zalloc(b64len + 100);
279 	if (tncs->imv_data[imvID].imv_send == NULL) {
280 		os_free(b64);
281 		return TNC_RESULT_OTHER;
282 	}
283 
284 	tncs->imv_data[imvID].imv_send_len =
285 		os_snprintf((char *) tncs->imv_data[imvID].imv_send,
286 			    b64len + 100,
287 			    "<IMC-IMV-Message><Type>%08X</Type>"
288 			    "<Base64>%s</Base64></IMC-IMV-Message>",
289 			    (unsigned int) messageType, b64);
290 
291 	os_free(b64);
292 
293 	return TNC_RESULT_SUCCESS;
294 }
295 
296 
297 TNC_Result TNC_TNCS_RequestHandshakeRetry(
298 	TNC_IMVID imvID,
299 	TNC_ConnectionID connectionID,
300 	TNC_RetryReason reason)
301 {
302 	wpa_printf(MSG_DEBUG, "TNC: TNC_TNCS_RequestHandshakeRetry");
303 	/* TODO */
304 	return TNC_RESULT_SUCCESS;
305 }
306 
307 
308 TNC_Result TNC_TNCS_ProvideRecommendation(
309 	TNC_IMVID imvID,
310 	TNC_ConnectionID connectionID,
311 	TNC_IMV_Action_Recommendation recommendation,
312 	TNC_IMV_Evaluation_Result evaluation)
313 {
314 	struct tncs_data *tncs;
315 
316 	wpa_printf(MSG_DEBUG, "TNC: TNC_TNCS_ProvideRecommendation(imvID=%lu "
317 		   "connectionID=%lu recommendation=%lu evaluation=%lu)",
318 		   (unsigned long) imvID, (unsigned long) connectionID,
319 		   (unsigned long) recommendation, (unsigned long) evaluation);
320 
321 	if (tncs_get_imv(imvID) == NULL)
322 		return TNC_RESULT_INVALID_PARAMETER;
323 
324 	tncs = tncs_get_conn(connectionID);
325 	if (tncs == NULL)
326 		return TNC_RESULT_INVALID_PARAMETER;
327 
328 	tncs->imv_data[imvID].recommendation = recommendation;
329 	tncs->imv_data[imvID].recommendation_set = 1;
330 
331 	return TNC_RESULT_SUCCESS;
332 }
333 
334 
335 TNC_Result TNC_TNCS_GetAttribute(
336 	TNC_IMVID imvID,
337 	TNC_ConnectionID connectionID,
338 	TNC_AttributeID attribureID,
339 	TNC_UInt32 bufferLength,
340 	TNC_BufferReference buffer,
341 	TNC_UInt32 *pOutValueLength)
342 {
343 	wpa_printf(MSG_DEBUG, "TNC: TNC_TNCS_GetAttribute");
344 	/* TODO */
345 	return TNC_RESULT_SUCCESS;
346 }
347 
348 
349 TNC_Result TNC_TNCS_SetAttribute(
350 	TNC_IMVID imvID,
351 	TNC_ConnectionID connectionID,
352 	TNC_AttributeID attribureID,
353 	TNC_UInt32 bufferLength,
354 	TNC_BufferReference buffer)
355 {
356 	wpa_printf(MSG_DEBUG, "TNC: TNC_TNCS_SetAttribute");
357 	/* TODO */
358 	return TNC_RESULT_SUCCESS;
359 }
360 
361 
362 TNC_Result TNC_TNCS_BindFunction(
363 	TNC_IMVID imvID,
364 	char *functionName,
365 	void **pOutFunctionPointer)
366 {
367 	wpa_printf(MSG_DEBUG, "TNC: TNC_TNCS_BindFunction(imcID=%lu, "
368 		   "functionName='%s')", (unsigned long) imvID, functionName);
369 
370 	if (tncs_get_imv(imvID) == NULL)
371 		return TNC_RESULT_INVALID_PARAMETER;
372 
373 	if (pOutFunctionPointer == NULL)
374 		return TNC_RESULT_INVALID_PARAMETER;
375 
376 	if (os_strcmp(functionName, "TNC_TNCS_ReportMessageTypes") == 0)
377 		*pOutFunctionPointer = TNC_TNCS_ReportMessageTypes;
378 	else if (os_strcmp(functionName, "TNC_TNCS_SendMessage") == 0)
379 		*pOutFunctionPointer = TNC_TNCS_SendMessage;
380 	else if (os_strcmp(functionName, "TNC_TNCS_RequestHandshakeRetry") ==
381 		 0)
382 		*pOutFunctionPointer = TNC_TNCS_RequestHandshakeRetry;
383 	else if (os_strcmp(functionName, "TNC_TNCS_ProvideRecommendation") ==
384 		 0)
385 		*pOutFunctionPointer = TNC_TNCS_ProvideRecommendation;
386 	else if (os_strcmp(functionName, "TNC_TNCS_GetAttribute") == 0)
387 		*pOutFunctionPointer = TNC_TNCS_GetAttribute;
388 	else if (os_strcmp(functionName, "TNC_TNCS_SetAttribute") == 0)
389 		*pOutFunctionPointer = TNC_TNCS_SetAttribute;
390 	else
391 		*pOutFunctionPointer = NULL;
392 
393 	return TNC_RESULT_SUCCESS;
394 }
395 
396 
397 static void * tncs_get_sym(void *handle, char *func)
398 {
399 	void *fptr;
400 
401 	fptr = dlsym(handle, func);
402 
403 	return fptr;
404 }
405 
406 
407 static int tncs_imv_resolve_funcs(struct tnc_if_imv *imv)
408 {
409 	void *handle = imv->dlhandle;
410 
411 	/* Mandatory IMV functions */
412 	imv->Initialize = tncs_get_sym(handle, "TNC_IMV_Initialize");
413 	if (imv->Initialize == NULL) {
414 		wpa_printf(MSG_ERROR, "TNC: IMV does not export "
415 			   "TNC_IMV_Initialize");
416 		return -1;
417 	}
418 
419 	imv->SolicitRecommendation = tncs_get_sym(
420 		handle, "TNC_IMV_SolicitRecommendation");
421 	if (imv->SolicitRecommendation == NULL) {
422 		wpa_printf(MSG_ERROR, "TNC: IMV does not export "
423 			   "TNC_IMV_SolicitRecommendation");
424 		return -1;
425 	}
426 
427 	imv->ProvideBindFunction =
428 		tncs_get_sym(handle, "TNC_IMV_ProvideBindFunction");
429 	if (imv->ProvideBindFunction == NULL) {
430 		wpa_printf(MSG_ERROR, "TNC: IMV does not export "
431 			   "TNC_IMV_ProvideBindFunction");
432 		return -1;
433 	}
434 
435 	/* Optional IMV functions */
436 	imv->NotifyConnectionChange =
437 		tncs_get_sym(handle, "TNC_IMV_NotifyConnectionChange");
438 	imv->ReceiveMessage = tncs_get_sym(handle, "TNC_IMV_ReceiveMessage");
439 	imv->BatchEnding = tncs_get_sym(handle, "TNC_IMV_BatchEnding");
440 	imv->Terminate = tncs_get_sym(handle, "TNC_IMV_Terminate");
441 
442 	return 0;
443 }
444 
445 
446 static int tncs_imv_initialize(struct tnc_if_imv *imv)
447 {
448 	TNC_Result res;
449 	TNC_Version imv_ver;
450 
451 	wpa_printf(MSG_DEBUG, "TNC: Calling TNC_IMV_Initialize for IMV '%s'",
452 		   imv->name);
453 	res = imv->Initialize(imv->imvID, TNC_IFIMV_VERSION_1,
454 			      TNC_IFIMV_VERSION_1, &imv_ver);
455 	wpa_printf(MSG_DEBUG, "TNC: TNC_IMV_Initialize: res=%lu imv_ver=%lu",
456 		   (unsigned long) res, (unsigned long) imv_ver);
457 
458 	return res == TNC_RESULT_SUCCESS ? 0 : -1;
459 }
460 
461 
462 static int tncs_imv_terminate(struct tnc_if_imv *imv)
463 {
464 	TNC_Result res;
465 
466 	if (imv->Terminate == NULL)
467 		return 0;
468 
469 	wpa_printf(MSG_DEBUG, "TNC: Calling TNC_IMV_Terminate for IMV '%s'",
470 		   imv->name);
471 	res = imv->Terminate(imv->imvID);
472 	wpa_printf(MSG_DEBUG, "TNC: TNC_IMV_Terminate: %lu",
473 		   (unsigned long) res);
474 
475 	return res == TNC_RESULT_SUCCESS ? 0 : -1;
476 }
477 
478 
479 static int tncs_imv_provide_bind_function(struct tnc_if_imv *imv)
480 {
481 	TNC_Result res;
482 
483 	wpa_printf(MSG_DEBUG, "TNC: Calling TNC_IMV_ProvideBindFunction for "
484 		   "IMV '%s'", imv->name);
485 	res = imv->ProvideBindFunction(imv->imvID, TNC_TNCS_BindFunction);
486 	wpa_printf(MSG_DEBUG, "TNC: TNC_IMV_ProvideBindFunction: res=%lu",
487 		   (unsigned long) res);
488 
489 	return res == TNC_RESULT_SUCCESS ? 0 : -1;
490 }
491 
492 
493 static int tncs_imv_notify_connection_change(struct tnc_if_imv *imv,
494 					     TNC_ConnectionID conn,
495 					     TNC_ConnectionState state)
496 {
497 	TNC_Result res;
498 
499 	if (imv->NotifyConnectionChange == NULL)
500 		return 0;
501 
502 	wpa_printf(MSG_DEBUG, "TNC: Calling TNC_IMV_NotifyConnectionChange(%d)"
503 		   " for IMV '%s'", (int) state, imv->name);
504 	res = imv->NotifyConnectionChange(imv->imvID, conn, state);
505 	wpa_printf(MSG_DEBUG, "TNC: TNC_IMC_NotifyConnectionChange: %lu",
506 		   (unsigned long) res);
507 
508 	return res == TNC_RESULT_SUCCESS ? 0 : -1;
509 }
510 
511 
512 static int tncs_load_imv(struct tnc_if_imv *imv)
513 {
514 	if (imv->path == NULL) {
515 		wpa_printf(MSG_DEBUG, "TNC: No IMV configured");
516 		return -1;
517 	}
518 
519 	wpa_printf(MSG_DEBUG, "TNC: Opening IMV: %s (%s)",
520 		   imv->name, imv->path);
521 	imv->dlhandle = dlopen(imv->path, RTLD_LAZY);
522 	if (imv->dlhandle == NULL) {
523 		wpa_printf(MSG_ERROR, "TNC: Failed to open IMV '%s' (%s): %s",
524 			   imv->name, imv->path, dlerror());
525 		return -1;
526 	}
527 
528 	if (tncs_imv_resolve_funcs(imv) < 0) {
529 		wpa_printf(MSG_ERROR, "TNC: Failed to resolve IMV functions");
530 		return -1;
531 	}
532 
533 	if (tncs_imv_initialize(imv) < 0 ||
534 	    tncs_imv_provide_bind_function(imv) < 0) {
535 		wpa_printf(MSG_ERROR, "TNC: Failed to initialize IMV");
536 		return -1;
537 	}
538 
539 	return 0;
540 }
541 
542 
543 static void tncs_free_imv(struct tnc_if_imv *imv)
544 {
545 	os_free(imv->name);
546 	os_free(imv->path);
547 	os_free(imv->supported_types);
548 }
549 
550 static void tncs_unload_imv(struct tnc_if_imv *imv)
551 {
552 	tncs_imv_terminate(imv);
553 
554 	if (imv->dlhandle)
555 		dlclose(imv->dlhandle);
556 
557 	tncs_free_imv(imv);
558 }
559 
560 
561 static int tncs_supported_type(struct tnc_if_imv *imv, unsigned int type)
562 {
563 	size_t i;
564 	unsigned int vendor, subtype;
565 
566 	if (imv == NULL || imv->supported_types == NULL)
567 		return 0;
568 
569 	vendor = type >> 8;
570 	subtype = type & 0xff;
571 
572 	for (i = 0; i < imv->num_supported_types; i++) {
573 		unsigned int svendor, ssubtype;
574 		svendor = imv->supported_types[i] >> 8;
575 		ssubtype = imv->supported_types[i] & 0xff;
576 		if ((vendor == svendor || svendor == TNC_VENDORID_ANY) &&
577 		    (subtype == ssubtype || ssubtype == TNC_SUBTYPE_ANY))
578 			return 1;
579 	}
580 
581 	return 0;
582 }
583 
584 
585 static void tncs_send_to_imvs(struct tncs_data *tncs, unsigned int type,
586 			      const u8 *msg, size_t len)
587 {
588 	struct tnc_if_imv *imv;
589 	TNC_Result res;
590 
591 	wpa_hexdump_ascii(MSG_MSGDUMP, "TNC: Message to IMV(s)", msg, len);
592 
593 	for (imv = tncs->imv; imv; imv = imv->next) {
594 		if (imv->ReceiveMessage == NULL ||
595 		    !tncs_supported_type(imv, type))
596 			continue;
597 
598 		wpa_printf(MSG_DEBUG, "TNC: Call ReceiveMessage for IMV '%s'",
599 			   imv->name);
600 		res = imv->ReceiveMessage(imv->imvID, tncs->connectionID,
601 					  (TNC_BufferReference) msg, len,
602 					  type);
603 		wpa_printf(MSG_DEBUG, "TNC: ReceiveMessage: %lu",
604 			   (unsigned long) res);
605 	}
606 }
607 
608 
609 static void tncs_batch_ending(struct tncs_data *tncs)
610 {
611 	struct tnc_if_imv *imv;
612 	TNC_Result res;
613 
614 	for (imv = tncs->imv; imv; imv = imv->next) {
615 		if (imv->BatchEnding == NULL)
616 			continue;
617 
618 		wpa_printf(MSG_DEBUG, "TNC: Call BatchEnding for IMV '%s'",
619 			   imv->name);
620 		res = imv->BatchEnding(imv->imvID, tncs->connectionID);
621 		wpa_printf(MSG_DEBUG, "TNC: BatchEnding: %lu",
622 			   (unsigned long) res);
623 	}
624 }
625 
626 
627 static void tncs_solicit_recommendation(struct tncs_data *tncs)
628 {
629 	struct tnc_if_imv *imv;
630 	TNC_Result res;
631 
632 	for (imv = tncs->imv; imv; imv = imv->next) {
633 		if (tncs->imv_data[imv->imvID].recommendation_set)
634 			continue;
635 
636 		wpa_printf(MSG_DEBUG, "TNC: Call SolicitRecommendation for "
637 			   "IMV '%s'", imv->name);
638 		res = imv->SolicitRecommendation(imv->imvID,
639 						 tncs->connectionID);
640 		wpa_printf(MSG_DEBUG, "TNC: SolicitRecommendation: %lu",
641 			   (unsigned long) res);
642 	}
643 }
644 
645 
646 void tncs_init_connection(struct tncs_data *tncs)
647 {
648 	struct tnc_if_imv *imv;
649 	int i;
650 
651 	for (imv = tncs->imv; imv; imv = imv->next) {
652 		tncs_imv_notify_connection_change(
653 			imv, tncs->connectionID, TNC_CONNECTION_STATE_CREATE);
654 		tncs_imv_notify_connection_change(
655 			imv, tncs->connectionID,
656 			TNC_CONNECTION_STATE_HANDSHAKE);
657 	}
658 
659 	for (i = 0; i < TNC_MAX_IMV_ID; i++) {
660 		os_free(tncs->imv_data[i].imv_send);
661 		tncs->imv_data[i].imv_send = NULL;
662 		tncs->imv_data[i].imv_send_len = 0;
663 	}
664 }
665 
666 
667 size_t tncs_total_send_len(struct tncs_data *tncs)
668 {
669 	int i;
670 	size_t len = 0;
671 
672 	for (i = 0; i < TNC_MAX_IMV_ID; i++)
673 		len += tncs->imv_data[i].imv_send_len;
674 	if (tncs->tncs_message)
675 		len += os_strlen(tncs->tncs_message);
676 	return len;
677 }
678 
679 
680 u8 * tncs_copy_send_buf(struct tncs_data *tncs, u8 *pos)
681 {
682 	int i;
683 
684 	for (i = 0; i < TNC_MAX_IMV_ID; i++) {
685 		if (tncs->imv_data[i].imv_send == NULL)
686 			continue;
687 
688 		os_memcpy(pos, tncs->imv_data[i].imv_send,
689 			  tncs->imv_data[i].imv_send_len);
690 		pos += tncs->imv_data[i].imv_send_len;
691 		os_free(tncs->imv_data[i].imv_send);
692 		tncs->imv_data[i].imv_send = NULL;
693 		tncs->imv_data[i].imv_send_len = 0;
694 	}
695 
696 	if (tncs->tncs_message) {
697 		size_t len = os_strlen(tncs->tncs_message);
698 		os_memcpy(pos, tncs->tncs_message, len);
699 		pos += len;
700 		os_free(tncs->tncs_message);
701 		tncs->tncs_message = NULL;
702 	}
703 
704 	return pos;
705 }
706 
707 
708 char * tncs_if_tnccs_start(struct tncs_data *tncs)
709 {
710 	char *buf = os_malloc(1000);
711 	if (buf == NULL)
712 		return NULL;
713 	tncs->last_batchid++;
714 	os_snprintf(buf, 1000, IF_TNCCS_START, tncs->last_batchid);
715 	return buf;
716 }
717 
718 
719 char * tncs_if_tnccs_end(void)
720 {
721 	char *buf = os_malloc(100);
722 	if (buf == NULL)
723 		return NULL;
724 	os_snprintf(buf, 100, IF_TNCCS_END);
725 	return buf;
726 }
727 
728 
729 static int tncs_get_type(char *start, unsigned int *type)
730 {
731 	char *pos = os_strstr(start, "<Type>");
732 	if (pos == NULL)
733 		return -1;
734 	pos += 6;
735 	*type = strtoul(pos, NULL, 16);
736 	return 0;
737 }
738 
739 
740 static unsigned char * tncs_get_base64(char *start, size_t *decoded_len)
741 {
742 	char *pos, *pos2;
743 	unsigned char *decoded;
744 
745 	pos = os_strstr(start, "<Base64>");
746 	if (pos == NULL)
747 		return NULL;
748 
749 	pos += 8;
750 	pos2 = os_strstr(pos, "</Base64>");
751 	if (pos2 == NULL)
752 		return NULL;
753 	*pos2 = '\0';
754 
755 	decoded = base64_decode((unsigned char *) pos, os_strlen(pos),
756 				decoded_len);
757 	*pos2 = '<';
758 	if (decoded == NULL) {
759 		wpa_printf(MSG_DEBUG, "TNC: Failed to decode Base64 data");
760 	}
761 
762 	return decoded;
763 }
764 
765 
766 static enum tncs_process_res tncs_derive_recommendation(struct tncs_data *tncs)
767 {
768 	enum IMV_Action_Recommendation rec;
769 	struct tnc_if_imv *imv;
770 	TNC_ConnectionState state;
771 	char *txt;
772 
773 	wpa_printf(MSG_DEBUG, "TNC: No more messages from IMVs");
774 
775 	if (tncs->done)
776 		return TNCCS_PROCESS_OK_NO_RECOMMENDATION;
777 
778 	tncs_solicit_recommendation(tncs);
779 
780 	/* Select the most restrictive recommendation */
781 	rec = TNC_IMV_ACTION_RECOMMENDATION_NO_RECOMMENDATION;
782 	for (imv = tncs->imv; imv; imv = imv->next) {
783 		TNC_IMV_Action_Recommendation irec;
784 		irec = tncs->imv_data[imv->imvID].recommendation;
785 		if (irec == TNC_IMV_ACTION_RECOMMENDATION_NO_ACCESS)
786 			rec = TNC_IMV_ACTION_RECOMMENDATION_NO_ACCESS;
787 		if (irec == TNC_IMV_ACTION_RECOMMENDATION_ISOLATE &&
788 		    rec != TNC_IMV_ACTION_RECOMMENDATION_NO_ACCESS)
789 			rec = TNC_IMV_ACTION_RECOMMENDATION_ISOLATE;
790 		if (irec == TNC_IMV_ACTION_RECOMMENDATION_ALLOW &&
791 		    rec == TNC_IMV_ACTION_RECOMMENDATION_NO_RECOMMENDATION)
792 			rec = TNC_IMV_ACTION_RECOMMENDATION_ALLOW;
793 	}
794 
795 	wpa_printf(MSG_DEBUG, "TNC: Recommendation: %d", rec);
796 	tncs->recommendation = rec;
797 	tncs->done = 1;
798 
799 	txt = NULL;
800 	switch (rec) {
801 	case TNC_IMV_ACTION_RECOMMENDATION_ALLOW:
802 	case TNC_IMV_ACTION_RECOMMENDATION_NO_RECOMMENDATION:
803 		txt = "allow";
804 		state = TNC_CONNECTION_STATE_ACCESS_ALLOWED;
805 		break;
806 	case TNC_IMV_ACTION_RECOMMENDATION_ISOLATE:
807 		txt = "isolate";
808 		state = TNC_CONNECTION_STATE_ACCESS_ISOLATED;
809 		break;
810 	case TNC_IMV_ACTION_RECOMMENDATION_NO_ACCESS:
811 		txt = "none";
812 		state = TNC_CONNECTION_STATE_ACCESS_NONE;
813 		break;
814 	default:
815 		state = TNC_CONNECTION_STATE_ACCESS_ALLOWED;
816 		break;
817 	}
818 
819 	if (txt) {
820 		os_free(tncs->tncs_message);
821 		tncs->tncs_message = os_zalloc(200);
822 		if (tncs->tncs_message) {
823 			os_snprintf(tncs->tncs_message, 199,
824 				    "<TNCC-TNCS-Message><Type>%08X</Type>"
825 				    "<XML><TNCCS-Recommendation type=\"%s\">"
826 				    "</TNCCS-Recommendation></XML>"
827 				    "</TNCC-TNCS-Message>",
828 				    TNC_TNCCS_RECOMMENDATION, txt);
829 		}
830 	}
831 
832 	for (imv = tncs->imv; imv; imv = imv->next) {
833 		tncs_imv_notify_connection_change(imv, tncs->connectionID,
834 						  state);
835 	}
836 
837 	switch (rec) {
838 	case TNC_IMV_ACTION_RECOMMENDATION_ALLOW:
839 		return TNCCS_RECOMMENDATION_ALLOW;
840 	case TNC_IMV_ACTION_RECOMMENDATION_NO_ACCESS:
841 		return TNCCS_RECOMMENDATION_NO_ACCESS;
842 	case TNC_IMV_ACTION_RECOMMENDATION_ISOLATE:
843 		return TNCCS_RECOMMENDATION_ISOLATE;
844 	case TNC_IMV_ACTION_RECOMMENDATION_NO_RECOMMENDATION:
845 		return TNCCS_RECOMMENDATION_NO_RECOMMENDATION;
846 	default:
847 		return TNCCS_PROCESS_ERROR;
848 	}
849 }
850 
851 
852 enum tncs_process_res tncs_process_if_tnccs(struct tncs_data *tncs,
853 					    const u8 *msg, size_t len)
854 {
855 	char *buf, *start, *end, *pos, *pos2, *payload;
856 	unsigned int batch_id;
857 	unsigned char *decoded;
858 	size_t decoded_len;
859 
860 	buf = os_malloc(len + 1);
861 	if (buf == NULL)
862 		return TNCCS_PROCESS_ERROR;
863 
864 	os_memcpy(buf, msg, len);
865 	buf[len] = '\0';
866 	start = os_strstr(buf, "<TNCCS-Batch ");
867 	end = os_strstr(buf, "</TNCCS-Batch>");
868 	if (start == NULL || end == NULL || start > end) {
869 		os_free(buf);
870 		return TNCCS_PROCESS_ERROR;
871 	}
872 
873 	start += 13;
874 	while (*start == ' ')
875 		start++;
876 	*end = '\0';
877 
878 	pos = os_strstr(start, "BatchId=");
879 	if (pos == NULL) {
880 		os_free(buf);
881 		return TNCCS_PROCESS_ERROR;
882 	}
883 
884 	pos += 8;
885 	if (*pos == '"')
886 		pos++;
887 	batch_id = atoi(pos);
888 	wpa_printf(MSG_DEBUG, "TNC: Received IF-TNCCS BatchId=%u",
889 		   batch_id);
890 	if (batch_id != tncs->last_batchid + 1) {
891 		wpa_printf(MSG_DEBUG, "TNC: Unexpected IF-TNCCS BatchId "
892 			   "%u (expected %u)",
893 			   batch_id, tncs->last_batchid + 1);
894 		os_free(buf);
895 		return TNCCS_PROCESS_ERROR;
896 	}
897 	tncs->last_batchid = batch_id;
898 
899 	while (*pos != '\0' && *pos != '>')
900 		pos++;
901 	if (*pos == '\0') {
902 		os_free(buf);
903 		return TNCCS_PROCESS_ERROR;
904 	}
905 	pos++;
906 	payload = start;
907 
908 	/*
909 	 * <IMC-IMV-Message>
910 	 * <Type>01234567</Type>
911 	 * <Base64>foo==</Base64>
912 	 * </IMC-IMV-Message>
913 	 */
914 
915 	while (*start) {
916 		char *endpos;
917 		unsigned int type;
918 
919 		pos = os_strstr(start, "<IMC-IMV-Message>");
920 		if (pos == NULL)
921 			break;
922 		start = pos + 17;
923 		end = os_strstr(start, "</IMC-IMV-Message>");
924 		if (end == NULL)
925 			break;
926 		*end = '\0';
927 		endpos = end;
928 		end += 18;
929 
930 		if (tncs_get_type(start, &type) < 0) {
931 			*endpos = '<';
932 			start = end;
933 			continue;
934 		}
935 		wpa_printf(MSG_DEBUG, "TNC: IMC-IMV-Message Type 0x%x", type);
936 
937 		decoded = tncs_get_base64(start, &decoded_len);
938 		if (decoded == NULL) {
939 			*endpos = '<';
940 			start = end;
941 			continue;
942 		}
943 
944 		tncs_send_to_imvs(tncs, type, decoded, decoded_len);
945 
946 		os_free(decoded);
947 
948 		start = end;
949 	}
950 
951 	/*
952 	 * <TNCC-TNCS-Message>
953 	 * <Type>01234567</Type>
954 	 * <XML><TNCCS-Foo type="foo"></TNCCS-Foo></XML>
955 	 * <Base64>foo==</Base64>
956 	 * </TNCC-TNCS-Message>
957 	 */
958 
959 	start = payload;
960 	while (*start) {
961 		unsigned int type;
962 		char *xml, *xmlend, *endpos;
963 
964 		pos = os_strstr(start, "<TNCC-TNCS-Message>");
965 		if (pos == NULL)
966 			break;
967 		start = pos + 19;
968 		end = os_strstr(start, "</TNCC-TNCS-Message>");
969 		if (end == NULL)
970 			break;
971 		*end = '\0';
972 		endpos = end;
973 		end += 20;
974 
975 		if (tncs_get_type(start, &type) < 0) {
976 			*endpos = '<';
977 			start = end;
978 			continue;
979 		}
980 		wpa_printf(MSG_DEBUG, "TNC: TNCC-TNCS-Message Type 0x%x",
981 			   type);
982 
983 		/* Base64 OR XML */
984 		decoded = NULL;
985 		xml = NULL;
986 		xmlend = NULL;
987 		pos = os_strstr(start, "<XML>");
988 		if (pos) {
989 			pos += 5;
990 			pos2 = os_strstr(pos, "</XML>");
991 			if (pos2 == NULL) {
992 				*endpos = '<';
993 				start = end;
994 				continue;
995 			}
996 			xmlend = pos2;
997 			xml = pos;
998 		} else {
999 			decoded = tncs_get_base64(start, &decoded_len);
1000 			if (decoded == NULL) {
1001 				*endpos = '<';
1002 				start = end;
1003 				continue;
1004 			}
1005 		}
1006 
1007 		if (decoded) {
1008 			wpa_hexdump_ascii(MSG_MSGDUMP,
1009 					  "TNC: TNCC-TNCS-Message Base64",
1010 					  decoded, decoded_len);
1011 			os_free(decoded);
1012 		}
1013 
1014 		if (xml) {
1015 			wpa_hexdump_ascii(MSG_MSGDUMP,
1016 					  "TNC: TNCC-TNCS-Message XML",
1017 					  (unsigned char *) xml,
1018 					  xmlend - xml);
1019 		}
1020 
1021 		start = end;
1022 	}
1023 
1024 	os_free(buf);
1025 
1026 	tncs_batch_ending(tncs);
1027 
1028 	if (tncs_total_send_len(tncs) == 0)
1029 		return tncs_derive_recommendation(tncs);
1030 
1031 	return TNCCS_PROCESS_OK_NO_RECOMMENDATION;
1032 }
1033 
1034 
1035 static struct tnc_if_imv * tncs_parse_imv(int id, char *start, char *end,
1036 					  int *error)
1037 {
1038 	struct tnc_if_imv *imv;
1039 	char *pos, *pos2;
1040 
1041 	if (id >= TNC_MAX_IMV_ID) {
1042 		wpa_printf(MSG_DEBUG, "TNC: Too many IMVs");
1043 		return NULL;
1044 	}
1045 
1046 	imv = os_zalloc(sizeof(*imv));
1047 	if (imv == NULL) {
1048 		*error = 1;
1049 		return NULL;
1050 	}
1051 
1052 	imv->imvID = id;
1053 
1054 	pos = start;
1055 	wpa_printf(MSG_DEBUG, "TNC: Configured IMV: %s", pos);
1056 	if (pos + 1 >= end || *pos != '"') {
1057 		wpa_printf(MSG_ERROR, "TNC: Ignoring invalid IMV line '%s' "
1058 			   "(no starting quotation mark)", start);
1059 		os_free(imv);
1060 		return NULL;
1061 	}
1062 
1063 	pos++;
1064 	pos2 = pos;
1065 	while (pos2 < end && *pos2 != '"')
1066 		pos2++;
1067 	if (pos2 >= end) {
1068 		wpa_printf(MSG_ERROR, "TNC: Ignoring invalid IMV line '%s' "
1069 			   "(no ending quotation mark)", start);
1070 		os_free(imv);
1071 		return NULL;
1072 	}
1073 	*pos2 = '\0';
1074 	wpa_printf(MSG_DEBUG, "TNC: Name: '%s'", pos);
1075 	imv->name = os_strdup(pos);
1076 
1077 	pos = pos2 + 1;
1078 	if (pos >= end || *pos != ' ') {
1079 		wpa_printf(MSG_ERROR, "TNC: Ignoring invalid IMV line '%s' "
1080 			   "(no space after name)", start);
1081 		os_free(imv);
1082 		return NULL;
1083 	}
1084 
1085 	pos++;
1086 	wpa_printf(MSG_DEBUG, "TNC: IMV file: '%s'", pos);
1087 	imv->path = os_strdup(pos);
1088 
1089 	return imv;
1090 }
1091 
1092 
1093 static int tncs_read_config(struct tncs_global *global)
1094 {
1095 	char *config, *end, *pos, *line_end;
1096 	size_t config_len;
1097 	struct tnc_if_imv *imv, *last;
1098 	int id = 0;
1099 
1100 	last = NULL;
1101 
1102 	config = os_readfile(TNC_CONFIG_FILE, &config_len);
1103 	if (config == NULL) {
1104 		wpa_printf(MSG_ERROR, "TNC: Could not open TNC configuration "
1105 			   "file '%s'", TNC_CONFIG_FILE);
1106 		return -1;
1107 	}
1108 
1109 	end = config + config_len;
1110 	for (pos = config; pos < end; pos = line_end + 1) {
1111 		line_end = pos;
1112 		while (*line_end != '\n' && *line_end != '\r' &&
1113 		       line_end < end)
1114 			line_end++;
1115 		*line_end = '\0';
1116 
1117 		if (os_strncmp(pos, "IMV ", 4) == 0) {
1118 			int error = 0;
1119 
1120 			imv = tncs_parse_imv(id++, pos + 4, line_end, &error);
1121 			if (error)
1122 				return -1;
1123 			if (imv) {
1124 				if (last == NULL)
1125 					global->imv = imv;
1126 				else
1127 					last->next = imv;
1128 				last = imv;
1129 			}
1130 		}
1131 	}
1132 
1133 	os_free(config);
1134 
1135 	return 0;
1136 }
1137 
1138 
1139 struct tncs_data * tncs_init(void)
1140 {
1141 	struct tncs_data *tncs;
1142 
1143 	if (tncs_global_data == NULL)
1144 		return NULL;
1145 
1146 	tncs = os_zalloc(sizeof(*tncs));
1147 	if (tncs == NULL)
1148 		return NULL;
1149 	tncs->imv = tncs_global_data->imv;
1150 	tncs->connectionID = tncs_global_data->next_conn_id++;
1151 	tncs->next = tncs_global_data->connections;
1152 	tncs_global_data->connections = tncs;
1153 
1154 	return tncs;
1155 }
1156 
1157 
1158 void tncs_deinit(struct tncs_data *tncs)
1159 {
1160 	int i;
1161 	struct tncs_data *prev, *conn;
1162 
1163 	if (tncs == NULL)
1164 		return;
1165 
1166 	for (i = 0; i < TNC_MAX_IMV_ID; i++)
1167 		os_free(tncs->imv_data[i].imv_send);
1168 
1169 	prev = NULL;
1170 	conn = tncs_global_data->connections;
1171 	while (conn) {
1172 		if (conn == tncs) {
1173 			if (prev)
1174 				prev->next = tncs->next;
1175 			else
1176 				tncs_global_data->connections = tncs->next;
1177 			break;
1178 		}
1179 		prev = conn;
1180 		conn = conn->next;
1181 	}
1182 
1183 	os_free(tncs->tncs_message);
1184 	os_free(tncs);
1185 }
1186 
1187 
1188 int tncs_global_init(void)
1189 {
1190 	struct tnc_if_imv *imv;
1191 
1192 	tncs_global_data = os_zalloc(sizeof(*tncs_global_data));
1193 	if (tncs_global_data == NULL)
1194 		return -1;
1195 
1196 	if (tncs_read_config(tncs_global_data) < 0) {
1197 		wpa_printf(MSG_ERROR, "TNC: Failed to read TNC configuration");
1198 		goto failed;
1199 	}
1200 
1201 	for (imv = tncs_global_data->imv; imv; imv = imv->next) {
1202 		if (tncs_load_imv(imv)) {
1203 			wpa_printf(MSG_ERROR, "TNC: Failed to load IMV '%s'",
1204 				   imv->name);
1205 			goto failed;
1206 		}
1207 	}
1208 
1209 	return 0;
1210 
1211 failed:
1212 	tncs_global_deinit();
1213 	return -1;
1214 }
1215 
1216 
1217 void tncs_global_deinit(void)
1218 {
1219 	struct tnc_if_imv *imv, *prev;
1220 
1221 	if (tncs_global_data == NULL)
1222 		return;
1223 
1224 	imv = tncs_global_data->imv;
1225 	while (imv) {
1226 		tncs_unload_imv(imv);
1227 
1228 		prev = imv;
1229 		imv = imv->next;
1230 		os_free(prev);
1231 	}
1232 
1233 	os_free(tncs_global_data);
1234 }
1235 
1236 
1237 struct wpabuf * tncs_build_soh_request(void)
1238 {
1239 	struct wpabuf *buf;
1240 
1241 	/*
1242 	 * Build a SoH Request TLV (to be used inside SoH EAP Extensions
1243 	 * Method)
1244 	 */
1245 
1246 	buf = wpabuf_alloc(8 + 4);
1247 	if (buf == NULL)
1248 		return NULL;
1249 
1250 	/* Vendor-Specific TLV (Microsoft) - SoH Request */
1251 	wpabuf_put_be16(buf, EAP_TLV_VENDOR_SPECIFIC_TLV); /* TLV Type */
1252 	wpabuf_put_be16(buf, 8); /* Length */
1253 
1254 	wpabuf_put_be32(buf, EAP_VENDOR_MICROSOFT); /* Vendor_Id */
1255 
1256 	wpabuf_put_be16(buf, 0x02); /* TLV Type - SoH Request TLV */
1257 	wpabuf_put_be16(buf, 0); /* Length */
1258 
1259 	return buf;
1260 }
1261 
1262 
1263 struct wpabuf * tncs_process_soh(const u8 *soh_tlv, size_t soh_tlv_len,
1264 				 int *failure)
1265 {
1266 	wpa_hexdump(MSG_DEBUG, "TNC: SoH TLV", soh_tlv, soh_tlv_len);
1267 	*failure = 0;
1268 
1269 	/* TODO: return MS-SoH Response TLV */
1270 
1271 	return NULL;
1272 }
1273