- added build_message to ike_sa
[strongswan.git] / Source / charon / message.c
1 /**
2 * @file message.c
3 *
4 * @brief Class message_t. Object of this type represents an IKEv2-Message.
5 *
6 */
7
8 /*
9 * Copyright (C) 2005 Jan Hutter, Martin Willi
10 * Hochschule fuer Technik Rapperswil
11 *
12 * This program is free software; you can redistribute it and/or modify it
13 * under the terms of the GNU General Public License as published by the
14 * Free Software Foundation; either version 2 of the License, or (at your
15 * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
16 *
17 * This program is distributed in the hope that it will be useful, but
18 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
19 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
20 * for more details.
21 */
22
23 #include <stdlib.h>
24
25 #include "message.h"
26
27 #include "types.h"
28 #include "globals.h"
29 #include "ike_sa_id.h"
30 #include "generator.h"
31 #include "utils/linked_list.h"
32 #include "utils/allocator.h"
33 #include "utils/logger_manager.h"
34 #include "payloads/encodings.h"
35 #include "payloads/payload.h"
36 #include "parser.h"
37
38
39 /**
40 * Supported payload entry used in message_rule_t
41 *
42 */
43 typedef struct supported_payload_entry_s supported_payload_entry_t;
44
45 struct supported_payload_entry_s {
46 /**
47 * Payload type
48 */
49 payload_type_t payload_type;
50
51 /**
52 * Minimal occurence of this payload
53 */
54 size_t min_occurence;
55
56 /**
57 * Max occurence of this payload
58 */
59 size_t max_occurence;
60 };
61
62 /**
63 * Message Rule used to find out which payloads are supported by each message type
64 *
65 */
66 typedef struct message_rule_s message_rule_t;
67
68 struct message_rule_s {
69 /**
70 * Type of message
71 */
72 exchange_type_t exchange_type;
73
74 /**
75 * Is message a request or response
76 */
77 bool is_request;
78 /**
79 * Number of supported payloads
80 */
81 size_t supported_payloads_count;
82 /**
83 * Pointer to first supported payload entry
84 */
85 supported_payload_entry_t *supported_payloads;
86 };
87
88
89 static supported_payload_entry_t supported_ike_sa_init_i_payloads[] =
90 {
91 {SECURITY_ASSOCIATION,1,1},
92 {KEY_EXCHANGE,1,1},
93 {NONCE,1,1},
94 };
95
96 static supported_payload_entry_t supported_ike_sa_init_r_payloads[] =
97 {
98 {SECURITY_ASSOCIATION,1,1},
99 {KEY_EXCHANGE,1,1},
100 {NONCE,1,1},
101 };
102
103 static message_rule_t message_rules[] = {
104 {IKE_SA_INIT,TRUE,(sizeof(supported_ike_sa_init_i_payloads)/sizeof(supported_payload_entry_t)),supported_ike_sa_init_i_payloads},
105 {IKE_SA_INIT,FALSE,(sizeof(supported_ike_sa_init_r_payloads)/sizeof(supported_payload_entry_t)),supported_ike_sa_init_r_payloads}
106 };
107
108 /**
109 * Entry for a payload in the internal used linked list
110 *
111 */
112 typedef struct payload_entry_s payload_entry_t;
113
114 struct payload_entry_s {
115 /**
116 * Type of payload
117 */
118 payload_type_t payload_type;
119 /**
120 * Data struct holding the data of given payload
121 */
122 void *data_struct;
123 };
124
125
126 /**
127 * Private data of an message_t object
128 */
129 typedef struct private_message_s private_message_t;
130
131 struct private_message_s {
132
133 /**
134 * Public part of a message_t object
135 */
136 message_t public;
137
138
139 /**
140 * Minor version of message
141 */
142 u_int8_t major_version;
143
144 /**
145 * Major version of message
146 */
147 u_int8_t minor_version;
148
149 /**
150 * First Payload in message
151 */
152 payload_type_t first_payload;
153
154 /**
155 * Assigned exchange type
156 */
157 exchange_type_t exchange_type;
158
159
160 /**
161 * TRUE if message is request.
162 * FALSE if message is reply.
163 */
164 bool is_request;
165
166 /**
167 * Message ID of this message
168 */
169 u_int32_t message_id;
170
171 /**
172 * ID of assigned IKE_SA
173 */
174 ike_sa_id_t *ike_sa_id;
175
176 /**
177 * Assigned UDP packet.
178 *
179 * Stores incoming packet or last generated one.
180 */
181 packet_t *packet;
182
183 /**
184 * Linked List where payload data are stored in
185 */
186 linked_list_t *payloads;
187
188 /**
189 * Assigned parser to parse Header and Body of this message
190 */
191 parser_t *parser;
192
193 /**
194 * logger for this message
195 */
196 logger_t *logger;
197
198 /**
199 * Gets a list of supported payloads of this message type
200 *
201 * @param this calling object
202 * @param[out] supported_payloads first entry of supported payloads
203 * @param[out] supported_payloads_count number of supported payload entries
204 *
205 * @return SUCCESS
206 * NOT_FOUND if no supported payload definition could be found
207 */
208 status_t (*get_supported_payloads) (private_message_t *this, supported_payload_entry_t **supported_payloads,size_t *supported_payloads_count);
209
210 };
211
212 /**
213 * Implements private_message_t's get_supported_payloads function.
214 * See #private_message_t.get_supported_payloads.
215 */
216 status_t get_supported_payloads (private_message_t *this, supported_payload_entry_t **supported_payloads,size_t *supported_payloads_count)
217 {
218 int i;
219 exchange_type_t exchange_type = this->public.get_exchange_type(&(this->public));
220 bool is_request = this->public.get_request(&(this->public));
221
222
223 for (i = 0; i < (sizeof(message_rules) / sizeof(message_rule_t)); i++)
224 {
225 if ((exchange_type == message_rules[i].exchange_type) &&
226 (is_request == message_rules[i].is_request))
227 {
228 /* found rule for given exchange_type*/
229 *supported_payloads = message_rules[i].supported_payloads;
230 *supported_payloads_count = message_rules[i].supported_payloads_count;
231
232 return SUCCESS;
233 }
234
235
236 }
237 *supported_payloads = NULL;
238 *supported_payloads_count = 0;
239 return NOT_FOUND;
240 }
241
242 /**
243 * Implements message_t's set_ike_sa_id function.
244 * See #message_s.set_ike_sa_id.
245 */
246 static status_t set_ike_sa_id (private_message_t *this,ike_sa_id_t *ike_sa_id)
247 {
248 status_t status;
249 status = ike_sa_id->clone(ike_sa_id,&(this->ike_sa_id));
250 return status;
251 }
252
253 /**
254 * Implements message_t's get_ike_sa_id function.
255 * See #message_s.get_ike_sa_id.
256 */
257 static status_t get_ike_sa_id (private_message_t *this,ike_sa_id_t **ike_sa_id)
258 {
259 status_t status;
260 if (this->ike_sa_id == NULL)
261 {
262 return FAILED;
263 }
264 status = this->ike_sa_id->clone(this->ike_sa_id,ike_sa_id);
265 return status;
266 }
267
268
269 /**
270 * Implements message_t's set_message_id function.
271 * See #message_s.set_message_id.
272 */
273 static status_t set_message_id (private_message_t *this,u_int32_t message_id)
274 {
275 this->message_id = message_id;
276 return SUCCESS;
277 }
278
279
280 /**
281 * Implements message_t's set_message_id function.
282 * See #message_s.set_message_id.
283 */
284 static u_int32_t get_message_id (private_message_t *this)
285 {
286 return this->message_id;
287 }
288
289 /**
290 * Implements message_t's set_major_version function.
291 * See #message_s.set_major_version.
292 */
293 static status_t set_major_version (private_message_t *this,u_int8_t major_version)
294 {
295 this->major_version = major_version;
296 return SUCCESS;
297 }
298
299
300 /**
301 * Implements message_t's get_major_version function.
302 * See #message_s.get_major_version.
303 */
304 static u_int8_t get_major_version (private_message_t *this)
305 {
306 return this->major_version;
307 }
308
309 /**
310 * Implements message_t's set_minor_version function.
311 * See #message_s.set_minor_version.
312 */
313 static status_t set_minor_version (private_message_t *this,u_int8_t minor_version)
314 {
315 this->minor_version = minor_version;
316 return SUCCESS;
317 }
318
319
320 /**
321 * Implements message_t's get_minor_version function.
322 * See #message_s.get_minor_version.
323 */
324 static u_int8_t get_minor_version (private_message_t *this)
325 {
326 return this->minor_version;
327 }
328
329 /**
330 * Implements message_t's set_exchange_type function.
331 * See #message_s.set_exchange_type.
332 */
333 static status_t set_exchange_type (private_message_t *this,exchange_type_t exchange_type)
334 {
335 this->exchange_type = exchange_type;
336 return SUCCESS;
337 }
338
339
340 /**
341 * Implements message_t's get_exchange_type function.
342 * See #message_s.get_exchange_type.
343 */
344 static exchange_type_t get_exchange_type (private_message_t *this)
345 {
346 return this->exchange_type;
347 }
348
349
350 /**
351 * Implements message_t's set_request function.
352 * See #message_s.set_request.
353 */
354 static status_t set_request (private_message_t *this,bool request)
355 {
356 this->is_request = request;
357 return SUCCESS;
358 }
359
360 /**
361 * Implements message_t's get_request function.
362 * See #message_s.get_request.
363 */
364 static exchange_type_t get_request (private_message_t *this)
365 {
366 return this->is_request;
367 }
368
369 static status_t add_payload(private_message_t *this, payload_t *payload)
370 {
371 if (this->payloads->insert_last(this->payloads, payload) != SUCCESS)
372 {
373 return OUT_OF_RES;
374 }
375 return SUCCESS;
376 }
377
378 static status_t set_source(private_message_t *this, host_t *host)
379 {
380 if (this->packet->source != NULL)
381 {
382 this->packet->source->destroy(this->packet->source);
383 }
384 this->packet->source = host;
385 return SUCCESS;
386 }
387
388 static status_t set_destination(private_message_t *this, host_t *host)
389 {
390 if (this->packet->destination != NULL)
391 {
392 this->packet->destination->destroy(this->packet->destination);
393 }
394 this->packet->destination = host;
395 return SUCCESS;
396 }
397
398 static status_t get_source(private_message_t *this, host_t **host)
399 {
400 *host = this->packet->source;
401 return SUCCESS;
402 }
403
404 static status_t get_destination(private_message_t *this, host_t **host)
405 {
406 *host = this->packet->destination;
407 return SUCCESS;
408 }
409
410
411 static status_t get_payload_iterator(private_message_t *this, linked_list_iterator_t **iterator)
412 {
413 return this->payloads->create_iterator(this->payloads, iterator, TRUE);
414 }
415
416
417 /**
418 * Implements message_t's generate function.
419 * See #message_s.generate.
420 */
421 static status_t generate(private_message_t *this, packet_t **packet)
422 {
423 generator_t *generator;
424 ike_header_t *ike_header;
425 payload_t *payload, *next_payload;
426 linked_list_iterator_t *iterator;
427 u_int64_t initiator_spi, responder_spi;
428 bool is_initiator;
429 status_t status;
430
431 if (this->exchange_type == EXCHANGE_TYPE_UNDEFINED)
432 {
433 return INVALID_STATE;
434 }
435
436 if (this->packet->source == NULL ||
437 this->packet->destination == NULL)
438 {
439 return INVALID_STATE;
440 }
441
442 ike_header = ike_header_create();
443 if (ike_header == NULL)
444 {
445 return OUT_OF_RES;
446 }
447
448 this->ike_sa_id->get_values(this->ike_sa_id, &initiator_spi, &responder_spi, &is_initiator);
449
450 ike_header->set_exchange_type(ike_header, this->exchange_type);
451 ike_header->set_initiator_flag(ike_header, is_initiator);
452 ike_header->set_message_id(ike_header, this->message_id);
453 ike_header->set_response_flag(ike_header, !this->is_request);
454 ike_header->set_initiator_flag(ike_header, is_initiator);
455 ike_header->set_initiator_spi(ike_header, initiator_spi);
456 ike_header->set_responder_spi(ike_header, responder_spi);
457
458 generator = generator_create();
459 if (generator == NULL)
460 {
461 return OUT_OF_RES;
462 }
463
464 payload = (payload_t*)ike_header;
465
466 if (this->payloads->create_iterator(this->payloads, &iterator, TRUE) != SUCCESS)
467 {
468 generator->destroy(generator);
469 ike_header->destroy(ike_header);
470 return OUT_OF_RES;
471 }
472 while(iterator->has_next(iterator))
473 {
474 iterator->current(iterator, (void**)&next_payload);
475 payload->set_next_type(payload, next_payload->get_type(next_payload));
476 status = generator->generate_payload(generator, payload);
477 if (status != SUCCESS)
478 {
479 generator->destroy(generator);
480 ike_header->destroy(ike_header);
481 return status;
482 }
483 payload = next_payload;
484 }
485 iterator->destroy(iterator);
486
487 payload->set_next_type(payload, NO_PAYLOAD);
488 status = generator->generate_payload(generator, payload);
489 if (status != SUCCESS)
490 {
491 generator->destroy(generator);
492 ike_header->destroy(ike_header);
493 return status;
494 }
495
496 ike_header->destroy(ike_header);
497
498
499
500 if (this->packet->data.ptr != NULL)
501 {
502 allocator_free(this->packet->data.ptr);
503 }
504
505 status = generator->write_to_chunk(generator, &(this->packet->data));
506 if (status != SUCCESS)
507 {
508 generator->destroy(generator);
509 return status;
510 }
511
512 this->packet->clone(this->packet, packet);
513
514 generator->destroy(generator);
515 return SUCCESS;
516 }
517
518 /**
519 * Implements message_t's parse_header function.
520 * See #message_s.parse_header.
521 */
522 static status_t parse_header (private_message_t *this)
523 {
524 ike_header_t *ike_header;
525 status_t status;
526
527 this->parser->reset_context(this->parser);
528 status = this->parser->parse_payload(this->parser,HEADER,(payload_t **) &ike_header);
529 if (status != SUCCESS)
530 {
531 this->logger->log(this->logger, ERROR, "Header could not be parsed");
532 return status;
533
534 }
535
536 /* verify payload */
537 status = ike_header->payload_interface.verify(&(ike_header->payload_interface));
538 if (status != SUCCESS)
539 {
540 this->logger->log(this->logger, ERROR, "Header could not be verified");
541 return status;
542 }
543
544 if (this->ike_sa_id != NULL)
545 {
546 this->ike_sa_id->destroy(this->ike_sa_id);
547 }
548
549 this->ike_sa_id = ike_sa_id_create(ike_header->get_initiator_spi(ike_header),
550 ike_header->get_responder_spi(ike_header),
551 !ike_header->get_initiator_flag(ike_header));
552 if (this->ike_sa_id == NULL)
553 {
554 this->logger->log(this->logger, ERROR, "Could not creaee ike_sa_id object");
555 ike_header->destroy(ike_header);
556 return OUT_OF_RES;
557 }
558 this->exchange_type = ike_header->get_exchange_type(ike_header);
559 this->message_id = ike_header->get_message_id(ike_header);
560 this->is_request = (!(ike_header->get_response_flag(ike_header)));
561 this->major_version = ike_header->get_maj_version(ike_header);
562 this->minor_version = ike_header->get_min_version(ike_header);
563 this->first_payload = ike_header->payload_interface.get_next_type(&(ike_header->payload_interface));
564
565 ike_header->destroy(ike_header);
566 return SUCCESS;
567 }
568
569 /**
570 * Implements message_t's parse_body function.
571 * See #message_s.parse_body.
572 */
573 static status_t parse_body (private_message_t *this)
574 {
575 status_t status;
576 int i;
577 payload_type_t current_payload_type = this->first_payload;
578 supported_payload_entry_t *supported_payloads;
579 size_t supported_payloads_count;
580
581 if (this->get_supported_payloads (this,&supported_payloads,&supported_payloads_count) != SUCCESS)
582 {
583 /* message type is not supported */
584 return FAILED;
585 }
586
587 while (current_payload_type != NO_PAYLOAD)
588 {
589 payload_t *current_payload;
590
591 bool supported = FALSE;
592 for (i = 0; i < supported_payloads_count;i++)
593 {
594 if (supported_payloads[i].payload_type == current_payload_type)
595 {
596 supported = TRUE;
597 break;
598 }
599 }
600 if (!supported && (current_payload_type != NO_PAYLOAD))
601 {
602 /* type not supported */
603 status = NOT_SUPPORTED;
604 this->logger->log(this->logger, ERROR, "Payload type %s not supported",mapping_find(payload_type_m,current_payload_type));
605 break;
606 }
607
608 status = this->parser->parse_payload(this->parser,current_payload_type,(payload_t **) &current_payload);
609 if (status != SUCCESS)
610 {
611 this->logger->log(this->logger, ERROR, "Payload type %s could not be parsed",mapping_find(payload_type_m,current_payload_type));
612 break;
613 }
614
615 current_payload_type = current_payload->get_next_type(current_payload);
616
617 status = current_payload->verify(current_payload);
618 if (status != SUCCESS)
619 {
620 this->logger->log(this->logger, ERROR, "Payload type %s could not be verified",mapping_find(payload_type_m,current_payload_type));
621 status = VERIFY_ERROR;
622 break;
623 }
624
625 status = this->payloads->insert_last(this->payloads,current_payload);
626 if (status != SUCCESS)
627 {
628 this->logger->log(this->logger, ERROR, "Could not insert current payload to internal list cause of ressource exhausting");
629 break;
630 }
631
632 }
633 if (status != SUCCESS)
634 {
635 /* already parsed payload is destroyed later in destroy call from outside this object */
636 }
637 else
638 {
639 linked_list_iterator_t *iterator;
640
641 status = this->payloads->create_iterator(this->payloads,&iterator,TRUE);
642 if (status != SUCCESS)
643 {
644 this->logger->log(this->logger, ERROR, "Could not create iterator to check supported payloads");
645 return status;
646 }
647
648
649 /* check for payloads with wrong count*/
650 for (i = 0; i < supported_payloads_count;i++)
651 {
652 size_t min_occurence = supported_payloads[i].min_occurence;
653 size_t max_occurence = supported_payloads[i].max_occurence;
654 payload_type_t payload_type = supported_payloads[i].payload_type;
655 size_t found_payloads = 0;
656
657 iterator->reset(iterator);
658
659 while(iterator->has_next(iterator))
660 {
661 payload_t *current_payload;
662 status = iterator->current(iterator,(void **)&current_payload);
663 if (status != SUCCESS)
664 {
665 this->logger->log(this->logger, CONTROL|MORE, "Could not get payload from internal list");
666 iterator->destroy(iterator);
667 return status;
668 }
669 if (current_payload->get_type(current_payload) == payload_type)
670 {
671 found_payloads++;
672 if (found_payloads > max_occurence)
673 {
674 this->logger->log(this->logger, CONTROL|MORE, "Payload of type %s more than %d times (%d) occured in current message",
675 mapping_find(payload_type_m,current_payload->get_type(current_payload)),max_occurence,found_payloads);
676 iterator->destroy(iterator);
677 return NOT_SUPPORTED;
678 }
679 }
680
681 }
682 if (found_payloads < min_occurence)
683 {
684 this->logger->log(this->logger, CONTROL|MORE, "Payload of type %s not occured %d times",
685 mapping_find(payload_type_m,payload_type),min_occurence);
686 iterator->destroy(iterator);
687 return NOT_SUPPORTED;
688 }
689
690 }
691 iterator->destroy(iterator);
692 }
693 return status;
694 }
695
696
697
698 /**
699 * Implements message_t's destroy function.
700 * See #message_s.destroy.
701 */
702 static status_t destroy (private_message_t *this)
703 {
704 linked_list_iterator_t *iterator;
705
706 if (this->packet != NULL)
707 {
708 this->packet->destroy(this->packet);
709 }
710 if (this->ike_sa_id != NULL)
711 {
712 this->ike_sa_id->destroy(this->ike_sa_id);
713 }
714
715 this->payloads->create_iterator(this->payloads, &iterator, TRUE);
716 while (iterator->has_next(iterator))
717 {
718 payload_t *payload;
719 iterator->current(iterator, (void**)&payload);
720 this->logger->log(this->logger, CONTROL|MORE, "Destroying payload of type %s",
721 mapping_find(payload_type_m, payload->get_type(payload)));
722 payload->destroy(payload);
723 }
724 iterator->destroy(iterator);
725 this->payloads->destroy(this->payloads);
726 this->parser->destroy(this->parser);
727
728 allocator_free(this);
729 return SUCCESS;
730 }
731
732 /*
733 * Described in Header-File
734 */
735 message_t *message_create_from_packet(packet_t *packet)
736 {
737 private_message_t *this = allocator_alloc_thing(private_message_t);
738 if (this == NULL)
739 {
740 return NULL;
741 }
742
743 /* public functions */
744 this->public.set_major_version = (status_t(*)(message_t*, u_int8_t))set_major_version;
745 this->public.get_major_version = (u_int8_t(*)(message_t*))get_major_version;
746 this->public.set_minor_version = (status_t(*)(message_t*, u_int8_t))set_minor_version;
747 this->public.get_minor_version = (u_int8_t(*)(message_t*))get_minor_version;
748 this->public.set_message_id = (status_t(*)(message_t*, u_int32_t))set_message_id;
749 this->public.get_message_id = (u_int32_t(*)(message_t*))get_message_id;
750 this->public.set_ike_sa_id = (status_t(*)(message_t*, ike_sa_id_t *))set_ike_sa_id;
751 this->public.get_ike_sa_id = (status_t(*)(message_t*, ike_sa_id_t **))get_ike_sa_id;
752 this->public.set_exchange_type = (status_t(*)(message_t*, exchange_type_t))set_exchange_type;
753 this->public.get_exchange_type = (exchange_type_t(*)(message_t*))get_exchange_type;
754 this->public.set_request = (status_t(*)(message_t*, bool))set_request;
755 this->public.get_request = (bool(*)(message_t*))get_request;
756 this->public.add_payload = (status_t(*)(message_t*,payload_t*))add_payload;
757 this->public.generate = (status_t (*) (message_t *, packet_t**)) generate;
758 this->public.set_source = (status_t (*) (message_t*,host_t*)) set_source;
759 this->public.get_source = (status_t (*) (message_t*,host_t**)) get_source;
760 this->public.set_destination = (status_t (*) (message_t*,host_t*)) set_destination;
761 this->public.get_destination = (status_t (*) (message_t*,host_t**)) get_destination;
762 this->public.get_payload_iterator = (status_t (*) (message_t *, linked_list_iterator_t **)) get_payload_iterator;
763 this->public.parse_header = (status_t (*) (message_t *)) parse_header;
764 this->public.parse_body = (status_t (*) (message_t *)) parse_body;
765 this->public.destroy = (status_t(*)(message_t*))destroy;
766
767 /* public values */
768 this->exchange_type = EXCHANGE_TYPE_UNDEFINED;
769 this->is_request = TRUE;
770 this->ike_sa_id = NULL;
771 this->first_payload = NO_PAYLOAD;
772 this->message_id = 0;
773
774 /* private functions */
775 this->get_supported_payloads = get_supported_payloads;
776
777 /* private values */
778 if (packet == NULL)
779 {
780 packet = packet_create();
781 }
782 if (packet == NULL)
783 {
784 allocator_free(this);
785 return NULL;
786 }
787 this->packet = packet;
788 this->payloads = linked_list_create();
789 if (this->payloads == NULL)
790 {
791 allocator_free(this);
792 return NULL;
793 }
794
795 /* parser is created from data of packet */
796 this->parser = parser_create(this->packet->data);
797 if (this->parser == NULL)
798 {
799 this->payloads->destroy(this->payloads);
800 allocator_free(this);
801 return NULL;
802 }
803
804 this->logger = global_logger_manager->create_logger(global_logger_manager, MESSAGE, NULL);
805 if (this->logger == NULL)
806 {
807 this->parser->destroy(this->parser);
808 this->payloads->destroy(this->payloads);
809 allocator_free(this);
810 return NULL;
811 }
812
813 return (&this->public);
814 }
815
816 /*
817 * Described in Header-File
818 */
819 message_t *message_create()
820 {
821 return message_create_from_packet(NULL);
822 }