Include stdint.h for UINTxx_MAX defines
[strongswan.git] / src / libcharon / encoding / payloads / transform_attribute.c
index fa34401..d20f77c 100644 (file)
@@ -17,6 +17,7 @@
 
 #include <string.h>
 #include <stddef.h>
+#include <stdint.h>
 
 #include "transform_attribute.h"
 
@@ -107,13 +108,13 @@ struct private_transform_attribute_t {
  */
 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)  },
        /* 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)                        }
 };
 
 /*
@@ -141,6 +142,12 @@ METHOD(payload_t, get_encoding_rules, int,
        return countof(encodings);
 }
 
+METHOD(payload_t, get_header_length, int,
+       private_transform_attribute_t *this)
+{
+       return 0;
+}
+
 METHOD(payload_t, get_type, payload_type_t,
        private_transform_attribute_t *this)
 {
@@ -168,31 +175,6 @@ METHOD(payload_t, get_length, size_t,
        return this->attribute_length_or_value + 4;
 }
 
-METHOD(transform_attribute_t, set_value_chunk, void,
-       private_transform_attribute_t *this, chunk_t value)
-{
-       chunk_free(&this->attribute_value);
-
-       if (value.len != 2)
-       {
-               this->attribute_value = chunk_clone(value);
-               this->attribute_length_or_value = value.len;
-               this->attribute_format = FALSE;
-       }
-       else
-       {
-               memcpy(&this->attribute_length_or_value, value.ptr, value.len);
-       }
-}
-
-METHOD(transform_attribute_t, set_value, void,
-       private_transform_attribute_t *this, u_int16_t value)
-{
-       chunk_free(&this->attribute_value);
-       this->attribute_length_or_value = value;
-       this->attribute_format = TRUE;
-}
-
 METHOD(transform_attribute_t, get_value_chunk, chunk_t,
        private_transform_attribute_t *this)
 {
@@ -203,16 +185,22 @@ METHOD(transform_attribute_t, get_value_chunk, chunk_t,
        return this->attribute_value;
 }
 
-METHOD(transform_attribute_t, get_value, u_int16_t,
+METHOD(transform_attribute_t, get_value, u_int64_t,
        private_transform_attribute_t *this)
 {
-       return this->attribute_length_or_value;
-}
+       u_int64_t value = 0;
 
-METHOD(transform_attribute_t, set_attribute_type, void,
-       private_transform_attribute_t *this, u_int16_t type)
-{
-       this->attribute_type = type & 0x7FFF;
+       if (this->attribute_format)
+       {
+               return this->attribute_length_or_value;
+       }
+       if (this->attribute_value.len > sizeof(value))
+       {
+               return UINT64_MAX;
+       }
+       memcpy(((char*)&value) + sizeof(value) - this->attribute_value.len,
+                  this->attribute_value.ptr, this->attribute_value.len);
+       return untoh64((char*)&value);
 }
 
 METHOD(transform_attribute_t, get_attribute_type, u_int16_t,
@@ -221,24 +209,6 @@ METHOD(transform_attribute_t, get_attribute_type, u_int16_t,
        return this->attribute_type;
 }
 
-METHOD(transform_attribute_t, clone_, transform_attribute_t*,
-       private_transform_attribute_t *this)
-{
-       private_transform_attribute_t *new;
-
-       new = (private_transform_attribute_t*)transform_attribute_create(this->type);
-
-       new->attribute_format = this->attribute_format;
-       new->attribute_type = this->attribute_type;
-       new->attribute_length_or_value = this->attribute_length_or_value;
-
-       if (!new->attribute_format)
-       {
-               new->attribute_value = chunk_clone(this->attribute_value);
-       }
-       return &new->public;
-}
-
 METHOD2(payload_t, transform_attribute_t, destroy, void,
        private_transform_attribute_t *this)
 {
@@ -258,22 +228,19 @@ transform_attribute_t *transform_attribute_create(payload_type_t type)
                        .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,
                        },
-                       .set_value_chunk = _set_value_chunk,
-                       .set_value = _set_value,
                        .get_value_chunk = _get_value_chunk,
                        .get_value = _get_value,
-                       .set_attribute_type = _set_attribute_type,
                        .get_attribute_type = _get_attribute_type,
-                       .clone = _clone_,
                        .destroy = _destroy,
                },
-               .attribute_format = TRUE,
+               .attribute_format = FALSE,
                .type = type,
        );
        return &this->public;
@@ -283,13 +250,32 @@ transform_attribute_t *transform_attribute_create(payload_type_t type)
  * Described in header.
  */
 transform_attribute_t *transform_attribute_create_value(payload_type_t type,
-                                                       transform_attribute_type_t kind, u_int16_t value)
+                                                       transform_attribute_type_t kind, u_int64_t value)
 {
-       transform_attribute_t *attribute;
+       private_transform_attribute_t *this;
 
-       attribute = transform_attribute_create(type);
-       attribute->set_attribute_type(attribute, kind);
-       attribute->set_value(attribute, value);
+       this = (private_transform_attribute_t*)transform_attribute_create(type);
 
-       return attribute;
+       this->attribute_type = kind & 0x7FFF;
+
+       if (value <= UINT16_MAX)
+       {
+               this->attribute_length_or_value = value;
+               this->attribute_format = TRUE;
+       }
+       else if (value <= UINT32_MAX)
+       {
+               u_int32_t val32;
+
+               val32 = htonl(value);
+               this->attribute_value = chunk_clone(chunk_from_thing(val32));
+               this->attribute_length_or_value = sizeof(val32);
+       }
+       else
+       {
+               htoun64(&value, value);
+               this->attribute_value = chunk_clone(chunk_from_thing(value));
+               this->attribute_length_or_value = sizeof(value);
+       }
+       return &this->public;
 }