diff options
Diffstat (limited to 'src/buffer.cpp')
-rw-r--r-- | src/buffer.cpp | 257 |
1 files changed, 257 insertions, 0 deletions
diff --git a/src/buffer.cpp b/src/buffer.cpp new file mode 100644 index 0000000..250a806 --- /dev/null +++ b/src/buffer.cpp @@ -0,0 +1,257 @@ +/* + * anytun + * + * The secure anycast tunneling protocol (satp) defines a protocol used + * for communication between any combination of unicast and anycast + * tunnel endpoints. It has less protocol overhead than IPSec in Tunnel + * mode and allows tunneling of every ETHER TYPE protocol (e.g. + * ethernet, ip, arp ...). satp directly includes cryptography and + * message authentication based on the methodes used by SRTP. It is + * intended to deliver a generic, scaleable and secure solution for + * tunneling and relaying of packets of any protocol. + * + * + * Copyright (C) 2007 anytun.org <satp@wirdorange.org> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 2 + * as published by the Free Software Foundation. + * + * 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. + * + * You should have received a copy of the GNU General Public License + * along with this program (see the file COPYING included with this + * distribution); if not, write to the Free Software Foundation, Inc., + * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#include <stdexcept> +#include <string> +#include <sstream> +#include <iostream> +#include <boost/archive/text_oarchive.hpp> +#include <boost/archive/text_iarchive.hpp> +#include "datatypes.h" +#include "buffer.h" + +Buffer::Buffer(bool allow_realloc) : buf_(0), length_(0), real_length_(0), allow_realloc_(allow_realloc) +{ +} + +Buffer::Buffer(u_int32_t length, bool allow_realloc) : length_(length), real_length_(length_ + Buffer::OVER_SIZE_), + allow_realloc_(allow_realloc) +{ + buf_ = new u_int8_t[real_length_]; + if(!buf_) { + length_ = 0; + real_length_ = 0; + throw std::bad_alloc(); + } + std::memset(buf_, 0, real_length_); +} + +Buffer::Buffer(u_int8_t* data, u_int32_t length, bool allow_realloc) : length_(length), real_length_(length + Buffer::OVER_SIZE_), + allow_realloc_(allow_realloc) +{ + if(!data) { + length_ = 0; + real_length_ = 0; + return; + } + + buf_ = new u_int8_t[real_length_]; + if(!buf_) { + length_ = 0; + real_length_ = 0; + throw std::bad_alloc(); + } + std::memcpy(buf_, data, length_); +} + +Buffer::Buffer(std::string hex_data, bool allow_realloc) : length_(hex_data.size()/2), + real_length_(length_ + Buffer::OVER_SIZE_), + allow_realloc_(allow_realloc) +{ + buf_ = new u_int8_t[real_length_]; + if(!buf_) { + length_ = 0; + real_length_ = 0; + throw std::bad_alloc(); + } + + for(u_int32_t i=0; i<length_; ++i) + { + u_int32_t tmp; + std::istringstream ss(std::string(hex_data.c_str(), i*2, 2)); + if(!(ss >> std::hex >> tmp)) tmp = 0; + buf_[i] = tmp; + } +} + +Buffer::~Buffer() +{ + if(buf_) + delete[] buf_; +} + +Buffer::Buffer(const Buffer &src) : length_(src.length_), real_length_(src.real_length_), allow_realloc_(src.allow_realloc_) +{ + buf_ = new u_int8_t[real_length_]; + if(!buf_) { + length_ = 0; + real_length_ = 0; + throw std::bad_alloc(); + } + std::memcpy(buf_, src.buf_, length_); +} + +void Buffer::operator=(const Buffer &src) +{ + if(buf_) + delete[] buf_; + + length_ = src.length_; + real_length_ = src.real_length_; + allow_realloc_ = src.allow_realloc_; + + buf_ = new u_int8_t[real_length_]; + if(!buf_) { + length_ = 0; + real_length_ = 0; + throw std::bad_alloc(); + } + std::memcpy(buf_, src.buf_, length_); +} + + + +bool Buffer::operator==(const Buffer &cmp) const +{ + if(length_ != cmp.length_) + return false; + + if(!std::memcmp(buf_, cmp.buf_, length_)) + return true; + + return false; +} + +Buffer Buffer::operator^(const Buffer &xor_by) const +{ + u_int32_t res_length = (xor_by.length_ > length_) ? xor_by.length_ : length_; + u_int32_t min_length = (xor_by.length_ < length_) ? xor_by.length_ : length_; + Buffer res(res_length); + + for( u_int32_t index = 0; index < min_length; index++ ) + res[index] = buf_[index] ^ xor_by[index]; + + return res; +} + +u_int32_t Buffer::getLength() const +{ + return length_; +} + +void Buffer::setLength(u_int32_t new_length) +{ + if(new_length == length_) + return; + + if(new_length > real_length_) + { + if(!allow_realloc_) + throw std::out_of_range("buffer::setLength() - reallocation not allowed for this Buffer"); + + u_int8_t* old_buf = buf_; + u_int32_t old_length = length_; + + length_ = new_length; + real_length_ = length_ + Buffer::OVER_SIZE_; + + buf_ = new u_int8_t[real_length_]; + if(!buf_) { + length_ = 0; + real_length_ = 0; + if(old_buf) + delete[] old_buf; + + throw std::bad_alloc(); + } + std::memcpy(buf_, old_buf, old_length); + + if(old_buf) + delete[] old_buf; + + old_buf = &buf_[old_length]; + std::memset(old_buf, 0, real_length_ - old_length); + } + else + length_ = new_length; + + reinit(); +} + + +u_int8_t* Buffer::getBuf() +{ + return buf_; +} + +u_int8_t& Buffer::operator[](u_int32_t index) +{ + if(index >= length_) + throw std::out_of_range("buffer::operator[]"); + + return buf_[index]; +} + +u_int8_t Buffer::operator[](u_int32_t index) const +{ + if(index >= length_) + throw std::out_of_range("buffer::operator[] const"); + + return buf_[index]; +} + +Buffer::operator u_int8_t*() +{ + return buf_; +} + +std::string Buffer::getHexDump() const +{ + std::stringstream ss; + ss << "Length=" << length_ << std::endl << std::hex << std::uppercase; + for( u_int32_t index = 0; index < length_; index++ ) + { + ss << std::setw(2) << std::setfill('0') << u_int32_t(buf_[index]) << " "; + if(!((index+1) % 16)) { + ss << std::endl; + continue; + } + if(!((index+1) % 8)) + ss << " "; + } + return ss.str(); +} + +std::string Buffer::getHexDumpOneLine() const +{ + std::stringstream ss; + ss << length_ << " Bytes,'" << std::hex << std::uppercase; + for( u_int32_t index = 0; index < length_; index++ ) + { + ss << std::setw(2) << std::setfill('0') << u_int32_t(buf_[index]); + } + ss << "'"; + return ss.str(); +} + +bool Buffer::isReallocAllowed() const +{ + return allow_realloc_; +} |