/* * Copyright (c) 2010 Remko Tronçon * Licensed under the GNU General Public License v3. * See Documentation/Licenses/GPLv3.txt for more information. */ #pragma once #include <cstring> #include <vector> #include <iostream> #include "Swiften/Base/String.h" namespace Swift { class ByteArray { public: typedef std::vector<char>::const_iterator const_iterator; ByteArray() : data_() {} ByteArray(const String& s) : data_(s.getUTF8String().begin(), s.getUTF8String().end()) {} ByteArray(const char* c) { while (*c) { data_.push_back(*c); ++c; } } ByteArray(const char* c, size_t n) { if (n > 0) { data_.resize(n); memcpy(&data_[0], c, n); } } const char* getData() const { return data_.empty() ? NULL : &data_[0]; } char* getData() { return data_.empty() ? NULL : &data_[0]; } size_t getSize() const { return data_.size(); } bool isEmpty() const { return data_.empty(); } void resize(size_t size) { return data_.resize(size); } friend ByteArray operator+(const ByteArray& a, const ByteArray&b) { ByteArray result(a); result.data_.insert(result.data_.end(), b.data_.begin(), b.data_.end()); return result; } friend ByteArray operator+(const ByteArray& a, char b) { ByteArray x; x.resize(1); x[0] = b; return a + x; } ByteArray& operator+=(const ByteArray& b) { data_.insert(data_.end(), b.data_.begin(), b.data_.end()); return *this; } ByteArray& operator+=(char c) { data_.push_back(c); return *this; } friend bool operator==(const ByteArray& a, const ByteArray& b) { return a.data_ == b.data_; } const char& operator[](size_t i) const { return data_[i]; } char& operator[](size_t i) { return data_[i]; } const_iterator begin() const { return data_.begin(); } const_iterator end() const { return data_.end(); } String toString() const { return String(getData(), getSize()); } void readFromFile(const String& file); private: std::vector<char> data_; }; } std::ostream& operator<<(std::ostream& os, const Swift::ByteArray& s);