3rd-party/libtins
1
0
Fork 0
mirror of https://github.com/mfontanini/libtins synced 2026-01-23 02:35:57 +01:00
Code Issues Releases Wiki Activity

Add missing member functions to ICMP extension classes

Browse Source
This commit is contained in:
Matias Fontanini
2015-12-14 20:22:50 -08:00
parent 187e7b1ca3
commit 6e026fcb66
3 changed files with 88 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/icmp_extension.cpp
View File

@@ -12,6 +12,11 @@ const uint32_t ICMPExtension::BASE_HEADER_SIZE = sizeof(uint16_t) + sizeof(uint8
// ICMPExtension class
ICMPExtension::ICMPExtension()
: extension_class_(0), extension_type_(0) {
}
ICMPExtension::ICMPExtension(const uint8_t* buffer, uint32_t total_sz) {
// Check for the base header (u16 length + u8 clss + u8 type)
if (total_sz < BASE_HEADER_SIZE) {
@@ -31,6 +36,18 @@ ICMPExtension::ICMPExtension(const uint8_t* buffer, uint32_t total_sz) {
payload_.assign(buffer, buffer + length);
}
void ICMPExtension::extension_class(uint8_t value) {
extension_class_ = value;
}
void ICMPExtension::extension_type(uint8_t value) {
extension_type_ = value;
}
void ICMPExtension::payload(const payload_type& value) {
payload_ = value;
}
uint32_t ICMPExtension::size() const {
return BASE_HEADER_SIZE + payload_.size();
}
@@ -87,6 +104,13 @@ void ICMPExtensionsStructure::reserved(small_uint<12> value) {
version_and_reserved_ = Endian::host_to_be(current_value);
}
void ICMPExtensionsStructure::version(small_uint<4> value) {
uint16_t current_value = Endian::be_to_host(version_and_reserved_);
current_value &= 0xfff;
current_value |= value << 12;
version_and_reserved_ = Endian::host_to_be(current_value);
}
bool ICMPExtensionsStructure::validate_extensions(const uint8_t* buffer, uint32_t total_sz) {
if (total_sz < BASE_HEADER_SIZE) {
return false;