Include stdint.h for UINTxx_MAX defines
[strongswan.git] / src / libcharon / encoding / payloads / transform_attribute.c
index 8bf2dde..d20f77c 100644 (file)
@@ -1,5 +1,6 @@
 /*
- * Copyright (C) 2005-2006 Martin Willi
+ * Copyright (C) 2005-2010 Martin Willi
+ * Copyright (C) 2010 revosec AG
  * Copyright (C) 2005 Jan Hutter
  * Hochschule fuer Technik Rapperswil
  *
 
 #include <string.h>
 #include <stddef.h>
+#include <stdint.h>
 
 #include "transform_attribute.h"
 
 #include <encoding/payloads/encodings.h>
 #include <library.h>
 
+ENUM(tattr_ph1_names, TATTR_PH1_ENCRYPTION_ALGORITHM, TATTR_PH1_GROUP_ORDER,
+       "ENCRYPTION_ALGORITHM",
+       "HASH_ALGORITHM",
+       "AUTH_METHOD",
+       "GROUP",
+       "GROUP_TYPE",
+       "GROUP_PRIME",
+       "GROUP_GENONE",
+       "GROUP_GENTWO",
+       "GROUP_CURVE_A",
+       "GROUP_CURVE_B",
+       "LIFE_TYPE",
+       "LIFE_DURATION",
+       "PRF",
+       "KEY_LENGTH",
+       "FIELD_SIZE",
+       "GROUP_ORDER",
+);
+
+ENUM(tattr_ph2_names, TATTR_PH2_SA_LIFE_TYPE, TATTR_PH2_EXT_SEQ_NUMBER,
+       "SA_LIFE_TYPE",
+       "SA_LIFE_DURATION",
+       "GROUP",
+       "ENCAP_MODE",
+       "AUTH_ALGORITHM",
+       "KEY_LENGTH",
+       "KEY_ROUNDS",
+       "COMP_DICT_SIZE",
+       "COMP_PRIV_ALGORITHM",
+       "ECN_TUNNEL",
+       "EXT_SEQ_NUMBER",
+);
+
+ENUM(tattr_ikev2_names, TATTR_IKEV2_KEY_LENGTH, TATTR_IKEV2_KEY_LENGTH,
+       "KEY_LENGTH",
+);
+
+
 typedef struct private_transform_attribute_t private_transform_attribute_t;
 
 /**
  * Private data of an transform_attribute_t object.
- *
  */
 struct private_transform_attribute_t {
+
        /**
         * Public transform_attribute_t interface.
         */
@@ -56,31 +96,25 @@ struct private_transform_attribute_t {
         * Attribute value as chunk if attribute_format is 0 (FALSE).
         */
        chunk_t attribute_value;
-};
-
 
-ENUM_BEGIN(transform_attribute_type_name, ATTRIBUTE_UNDEFINED, ATTRIBUTE_UNDEFINED,
-       "ATTRIBUTE_UNDEFINED");
-ENUM_NEXT(transform_attribute_type_name, KEY_LENGTH, KEY_LENGTH, ATTRIBUTE_UNDEFINED,
-       "KEY_LENGTH");
-ENUM_END(transform_attribute_type_name, KEY_LENGTH);
+       /**
+        * Payload type, TRANSFORM_ATTRIBUTE or TRANSFORM_ATTRIBUTE_V1
+        */
+       payload_type_t type;
+};
 
 /**
- * Encoding rules to parse or generate a Transform attribute.
- *
- * The defined offsets are the positions in a object of type
- * private_transform_attribute_t.
- *
+ * Encoding rules for IKEv1/IKEv2 transform attributes
  */
-encoding_rule_t transform_attribute_encodings[] = {
+static encoding_rule_t encodings[] = {
        /* Flag defining the format of this payload */
-       { ATTRIBUTE_FORMAT,                     offsetof(private_transform_attribute_t, attribute_format)                       },
+       { ATTRIBUTE_FORMAT,                     offsetof(private_transform_attribute_t, attribute_format)                       },
        /* type of the attribute as 15 bit unsigned integer */
        { ATTRIBUTE_TYPE,                       offsetof(private_transform_attribute_t, attribute_type)                         },
        /* Length or value, depending on the attribute format flag */
-       { ATTRIBUTE_LENGTH_OR_VALUE,    offsetof(private_transform_attribute_t, attribute_length_or_value)      },
+       { ATTRIBUTE_LENGTH_OR_VALUE,offsetof(private_transform_attribute_t, attribute_length_or_value)  },
        /* Value of attribute if attribute format flag is zero */
-       { ATTRIBUTE_VALUE,                      offsetof(private_transform_attribute_t, attribute_value)                        }
+       { ATTRIBUTE_VALUE,                      offsetof(private_transform_attribute_t, attribute_value)                        }
 };
 
 /*
@@ -95,231 +129,153 @@ encoding_rule_t transform_attribute_encodings[] = {
       +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
 */
 
-/**
- * Implementation of payload_t.verify.
- */
-static status_t verify(private_transform_attribute_t *this)
+METHOD(payload_t, verify, status_t,
+       private_transform_attribute_t *this)
 {
-       if (this->attribute_type != KEY_LENGTH)
-       {
-               return FAILED;
-       }
-
        return SUCCESS;
 }
 
-/**
- * Implementation of payload_t.get_encoding_rules.
- */
-static void get_encoding_rules(private_transform_attribute_t *this, encoding_rule_t **rules, size_t *rule_count)
+METHOD(payload_t, get_encoding_rules, int,
+       private_transform_attribute_t *this, encoding_rule_t **rules)
 {
-       *rules = transform_attribute_encodings;
-       *rule_count = sizeof(transform_attribute_encodings) / sizeof(encoding_rule_t);
+       *rules = encodings;
+       return countof(encodings);
 }
 
-/**
- * Implementation of payload_t.get_type.
- */
-static payload_type_t get_type(private_transform_attribute_t *this)
+METHOD(payload_t, get_header_length, int,
+       private_transform_attribute_t *this)
 {
-       return TRANSFORM_ATTRIBUTE;
+       return 0;
 }
 
-/**
- * Implementation of payload_t.get_next_type.
- */
-static payload_type_t get_next_type(private_transform_attribute_t *this)
+METHOD(payload_t, get_type, payload_type_t,
+       private_transform_attribute_t *this)
 {
-       return (NO_PAYLOAD);
+       return this->type;
 }
 
-/**
- * Implementation of payload_t.set_next_type.
- */
-static void set_next_type(private_transform_attribute_t *this,payload_type_t type)
+METHOD(payload_t, get_next_type, payload_type_t,
+       private_transform_attribute_t *this)
 {
+       return NO_PAYLOAD;
 }
 
-/**
- * Implementation of transform_attribute_t.get_length.
- */
-static size_t get_length(private_transform_attribute_t *this)
+METHOD(payload_t, set_next_type, void,
+       private_transform_attribute_t *this, payload_type_t type)
 {
-       if (this->attribute_format == TRUE)
-       {
-               /*Attribute size is only 4 byte */
-               return 4;
-       }
-       return (this->attribute_length_or_value + 4);
 }
 
-/**
- * Implementation of transform_attribute_t.set_value_chunk.
- */
-static void set_value_chunk(private_transform_attribute_t *this, chunk_t value)
+METHOD(payload_t, get_length, size_t,
+       private_transform_attribute_t *this)
 {
-       if (this->attribute_value.ptr != NULL)
+       if (this->attribute_format)
        {
-               /* free existing value */
-               free(this->attribute_value.ptr);
-               this->attribute_value.ptr = NULL;
-               this->attribute_value.len = 0;
-
-       }
-
-       if (value.len > 2)
-       {
-               this->attribute_value.ptr = clalloc(value.ptr,value.len);
-               this->attribute_value.len = value.len;
-               this->attribute_length_or_value = value.len;
-               /* attribute has not a fixed length */
-               this->attribute_format = FALSE;
-       }
-       else
-       {
-               memcpy(&(this->attribute_length_or_value),value.ptr,value.len);
+               return 4;
        }
+       return this->attribute_length_or_value + 4;
 }
 
-/**
- * Implementation of transform_attribute_t.set_value.
- */
-static void set_value(private_transform_attribute_t *this, u_int16_t value)
+METHOD(transform_attribute_t, get_value_chunk, chunk_t,
+       private_transform_attribute_t *this)
 {
-       if (this->attribute_value.ptr != NULL)
+       if (this->attribute_format)
        {
-               /* free existing value */
-               free(this->attribute_value.ptr);
-               this->attribute_value.ptr = NULL;
-               this->attribute_value.len = 0;
-
+               return chunk_from_thing(this->attribute_length_or_value);
        }
-       this->attribute_length_or_value = value;
+       return this->attribute_value;
 }
 
-/**
- * Implementation of transform_attribute_t.get_value_chunk.
- */
-static chunk_t get_value_chunk (private_transform_attribute_t *this)
+METHOD(transform_attribute_t, get_value, u_int64_t,
+       private_transform_attribute_t *this)
 {
-       chunk_t value;
+       u_int64_t value = 0;
 
-       if (this->attribute_format == FALSE)
+       if (this->attribute_format)
        {
-               value.ptr = this->attribute_value.ptr;
-               value.len = this->attribute_value.len;
+               return this->attribute_length_or_value;
        }
-       else
+       if (this->attribute_value.len > sizeof(value))
        {
-               value.ptr = (void *) &(this->attribute_length_or_value);
-               value.len = 2;
+               return UINT64_MAX;
        }
-
-       return value;
+       memcpy(((char*)&value) + sizeof(value) - this->attribute_value.len,
+                  this->attribute_value.ptr, this->attribute_value.len);
+       return untoh64((char*)&value);
 }
 
-/**
- * Implementation of transform_attribute_t.get_value.
- */
-static u_int16_t get_value (private_transform_attribute_t *this)
+METHOD(transform_attribute_t, get_attribute_type, u_int16_t,
+       private_transform_attribute_t *this)
 {
-       return this->attribute_length_or_value;
+       return this->attribute_type;
 }
 
-
-/**
- * Implementation of transform_attribute_t.set_attribute_type.
- */
-static void set_attribute_type (private_transform_attribute_t *this, u_int16_t type)
+METHOD2(payload_t, transform_attribute_t, destroy, void,
+       private_transform_attribute_t *this)
 {
-       this->attribute_type = type & 0x7FFF;
+       free(this->attribute_value.ptr);
+       free(this);
 }
 
-/**
- * Implementation of transform_attribute_t.get_attribute_type.
+/*
+ * Described in header.
  */
-static u_int16_t get_attribute_type (private_transform_attribute_t *this)
+transform_attribute_t *transform_attribute_create(payload_type_t type)
 {
-       return this->attribute_type;
+       private_transform_attribute_t *this;
+
+       INIT(this,
+               .public = {
+                       .payload_interface = {
+                               .verify = _verify,
+                               .get_encoding_rules = _get_encoding_rules,
+                               .get_header_length = _get_header_length,
+                               .get_length = _get_length,
+                               .get_next_type = _get_next_type,
+                               .set_next_type = _set_next_type,
+                               .get_type = _get_type,
+                               .destroy = _destroy,
+                       },
+                       .get_value_chunk = _get_value_chunk,
+                       .get_value = _get_value,
+                       .get_attribute_type = _get_attribute_type,
+                       .destroy = _destroy,
+               },
+               .attribute_format = FALSE,
+               .type = type,
+       );
+       return &this->public;
 }
 
-/**
- * Implementation of transform_attribute_t.clone.
+/*
+ * Described in header.
  */
-static transform_attribute_t * _clone(private_transform_attribute_t *this)
+transform_attribute_t *transform_attribute_create_value(payload_type_t type,
+                                                       transform_attribute_type_t kind, u_int64_t value)
 {
-       private_transform_attribute_t *new_clone;
+       private_transform_attribute_t *this;
 
-       new_clone = (private_transform_attribute_t *) transform_attribute_create();
+       this = (private_transform_attribute_t*)transform_attribute_create(type);
 
-       new_clone->attribute_format = this->attribute_format;
-       new_clone->attribute_type = this->attribute_type;
-       new_clone->attribute_length_or_value = this->attribute_length_or_value;
+       this->attribute_type = kind & 0x7FFF;
 
-       if (!new_clone->attribute_format)
+       if (value <= UINT16_MAX)
        {
-               new_clone->attribute_value.ptr = clalloc(this->attribute_value.ptr,this->attribute_value.len);
-               new_clone->attribute_value.len = this->attribute_value.len;
+               this->attribute_length_or_value = value;
+               this->attribute_format = TRUE;
        }
+       else if (value <= UINT32_MAX)
+       {
+               u_int32_t val32;
 
-       return (transform_attribute_t *) new_clone;
-}
-
-/**
- * Implementation of transform_attribute_t.destroy and payload_t.destroy.
- */
-static void destroy(private_transform_attribute_t *this)
-{
-       if (this->attribute_value.ptr != NULL)
+               val32 = htonl(value);
+               this->attribute_value = chunk_clone(chunk_from_thing(val32));
+               this->attribute_length_or_value = sizeof(val32);
+       }
+       else
        {
-               free(this->attribute_value.ptr);
+               htoun64(&value, value);
+               this->attribute_value = chunk_clone(chunk_from_thing(value));
+               this->attribute_length_or_value = sizeof(value);
        }
-       free(this);
-}
-
-/*
- * Described in header.
- */
-transform_attribute_t *transform_attribute_create()
-{
-       private_transform_attribute_t *this = malloc_thing(private_transform_attribute_t);
-
-       /* payload interface */
-       this->public.payload_interface.verify = (status_t (*) (payload_t *))verify;
-       this->public.payload_interface.get_encoding_rules = (void (*) (payload_t *, encoding_rule_t **, size_t *) ) get_encoding_rules;
-       this->public.payload_interface.get_length = (size_t (*) (payload_t *)) get_length;
-       this->public.payload_interface.get_next_type = (payload_type_t (*) (payload_t *)) get_next_type;
-       this->public.payload_interface.set_next_type = (void (*) (payload_t *,payload_type_t)) set_next_type;
-       this->public.payload_interface.get_type = (payload_type_t (*) (payload_t *)) get_type;
-       this->public.payload_interface.destroy = (void (*) (payload_t *))destroy;
-
-       /* public functions */
-       this->public.set_value_chunk = (void (*) (transform_attribute_t *,chunk_t)) set_value_chunk;
-       this->public.set_value = (void (*) (transform_attribute_t *,u_int16_t)) set_value;
-       this->public.get_value_chunk = (chunk_t (*) (transform_attribute_t *)) get_value_chunk;
-       this->public.get_value = (u_int16_t (*) (transform_attribute_t *)) get_value;
-       this->public.set_attribute_type = (void (*) (transform_attribute_t *,u_int16_t type)) set_attribute_type;
-       this->public.get_attribute_type = (u_int16_t (*) (transform_attribute_t *)) get_attribute_type;
-       this->public.clone = (transform_attribute_t * (*) (transform_attribute_t *)) _clone;
-       this->public.destroy = (void (*) (transform_attribute_t *)) destroy;
-
-       /* set default values of the fields */
-       this->attribute_format = TRUE;
-       this->attribute_type = 0;
-       this->attribute_length_or_value = 0;
-       this->attribute_value.ptr = NULL;
-       this->attribute_value.len = 0;
-
-       return (&(this->public));
-}
-
-/*
- * Described in header.
- */
-transform_attribute_t *transform_attribute_create_key_length(u_int16_t key_length)
-{
-       transform_attribute_t *attribute = transform_attribute_create();
-       attribute->set_attribute_type(attribute,KEY_LENGTH);
-       attribute->set_value(attribute,key_length);
-       return attribute;
+       return &this->public;
 }