/** * @file traffic_selector_substructure.c * * @brief Interface of traffic_selector_substructure_t. * */ /* * Copyright (C) 2005-2006 Martin Willi * Copyright (C) 2005 Jan Hutter * Hochschule fuer Technik Rapperswil * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the * Free Software Foundation; either version 2 of the License, or (at your * option) any later version. See . * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License * for more details. */ #include "traffic_selector_substructure.h" #include #include typedef struct private_traffic_selector_substructure_t private_traffic_selector_substructure_t; /** * Private data of an traffic_selector_substructure_t object. * */ struct private_traffic_selector_substructure_t { /** * Public traffic_selector_substructure_t interface. */ traffic_selector_substructure_t public; /** * Type of traffic selector. */ u_int8_t ts_type; /** * IP Protocol ID. */ u_int8_t ip_protocol_id; /** * Length of this payload. */ u_int16_t payload_length; /** * Start port number. */ u_int16_t start_port; /** * End port number. */ u_int16_t end_port; /** * Starting address. */ chunk_t starting_address; /** * Ending address. */ chunk_t ending_address; }; /** * Encoding rules to parse or generate a TS payload * * The defined offsets are the positions in a object of type * private_traffic_selector_substructure_t. * */ encoding_rule_t traffic_selector_substructure_encodings[] = { /* 1 Byte next ts type*/ { TS_TYPE, offsetof(private_traffic_selector_substructure_t, ts_type) }, /* 1 Byte IP protocol id*/ { U_INT_8, offsetof(private_traffic_selector_substructure_t, ip_protocol_id) }, /* Length of the whole payload*/ { PAYLOAD_LENGTH, offsetof(private_traffic_selector_substructure_t, payload_length) }, /* 2 Byte start port*/ { U_INT_16, offsetof(private_traffic_selector_substructure_t, start_port) }, /* 2 Byte end port*/ { U_INT_16, offsetof(private_traffic_selector_substructure_t, end_port) }, /* starting address is either 4 or 16 byte */ { ADDRESS, offsetof(private_traffic_selector_substructure_t, starting_address) }, /* ending address is either 4 or 16 byte */ { ADDRESS, offsetof(private_traffic_selector_substructure_t, ending_address) } }; /* 1 2 3 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ ! TS Type !IP Protocol ID*| Selector Length | +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ | Start Port* | End Port* | +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ ! ! ~ Starting Address* ~ ! ! +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ ! ! ~ Ending Address* ~ ! ! +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+ */ /** * Implementation of payload_t.verify. */ static status_t verify(private_traffic_selector_substructure_t *this) { if (this->start_port > this->end_port) { return FAILED; } switch (this->ts_type) { case TS_IPV4_ADDR_RANGE: { if ((this->starting_address.len != 4) || (this->ending_address.len != 4)) { /* ipv4 address must be 4 bytes long */ return FAILED; } break; } case TS_IPV6_ADDR_RANGE: { if ((this->starting_address.len != 16) || (this->ending_address.len != 16)) { /* ipv6 address must be 16 bytes long */ return FAILED; } break; } default: { /* not supported ts type */ return FAILED; } } return SUCCESS; } /** * Implementation of traffic_selector_substructure_t.get_encoding_rules. */ static void get_encoding_rules(private_traffic_selector_substructure_t *this, encoding_rule_t **rules, size_t *rule_count) { *rules = traffic_selector_substructure_encodings; *rule_count = sizeof(traffic_selector_substructure_encodings) / sizeof(encoding_rule_t); } /** * Implementation of payload_t.get_type. */ static payload_type_t get_payload_type(private_traffic_selector_substructure_t *this) { return TRAFFIC_SELECTOR_SUBSTRUCTURE; } /** * Implementation of payload_t.get_next_type. */ static payload_type_t get_next_type(private_traffic_selector_substructure_t *this) { return 0; } /** * Implementation of payload_t.set_next_type. */ static void set_next_type(private_traffic_selector_substructure_t *this,payload_type_t type) { } /** * Implementation of payload_t.get_length. */ static size_t get_length(private_traffic_selector_substructure_t *this) { return this->payload_length; } /** * Implementation of traffic_selector_substructure_t.get_traffic_selector. */ static traffic_selector_t *get_traffic_selector(private_traffic_selector_substructure_t *this) { traffic_selector_t *ts; ts = traffic_selector_create_from_bytes(this->ip_protocol_id, this->ts_type, this->starting_address, this->start_port, this->ending_address, this->end_port); return ts; } /** * recompute length field of the payload */ void compute_length(private_traffic_selector_substructure_t *this) { this->payload_length = TRAFFIC_SELECTOR_HEADER_LENGTH + this->ending_address.len + this->starting_address.len; } /** * Implementation of payload_t.destroy and traffic_selector_substructure_t.destroy. */ static void destroy(private_traffic_selector_substructure_t *this) { free(this->starting_address.ptr); free(this->ending_address.ptr); free(this); } /* * Described in header */ traffic_selector_substructure_t *traffic_selector_substructure_create() { private_traffic_selector_substructure_t *this = malloc_thing(private_traffic_selector_substructure_t); /* interface functions */ 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_payload_type; this->public.payload_interface.destroy = (void (*) (payload_t *))destroy; /* public functions */ this->public.get_traffic_selector = (traffic_selector_t* (*)(traffic_selector_substructure_t*))get_traffic_selector; this->public.destroy = (void (*) (traffic_selector_substructure_t *)) destroy; /* private variables */ this->payload_length = TRAFFIC_SELECTOR_HEADER_LENGTH; this->start_port = 0; this->end_port = 0; this->starting_address = chunk_empty; this->ending_address = chunk_empty; this->ip_protocol_id = 0; /* must be set to be valid */ this->ts_type = TS_IPV4_ADDR_RANGE; return (&(this->public)); } /* * Described in header */ traffic_selector_substructure_t *traffic_selector_substructure_create_from_traffic_selector(traffic_selector_t *traffic_selector) { private_traffic_selector_substructure_t *this = (private_traffic_selector_substructure_t*)traffic_selector_substructure_create(); this->ts_type = traffic_selector->get_type(traffic_selector); this->ip_protocol_id = traffic_selector->get_protocol(traffic_selector); this->start_port = traffic_selector->get_from_port(traffic_selector); this->end_port = traffic_selector->get_to_port(traffic_selector); this->starting_address = traffic_selector->get_from_address(traffic_selector); this->ending_address = traffic_selector->get_to_address(traffic_selector); compute_length(this); return &(this->public); }