

#include <cstring>
#include <tins/bootp.h>
#include <tins/exceptions.h>
#include <tins/memory_helpers.h>

using std::copy;

using Tins::Memory::InputMemoryStream;
using Tins::Memory::OutputMemoryStream;

namespace Tins{

BootP::BootP() 
: bootp_(), vend_(64) {

}

BootP::BootP(const uint8_t* buffer, uint32_t total_sz, uint32_t vend_field_size) 
: vend_(vend_field_size) {
    InputMemoryStream stream(buffer, total_sz);
    stream.read(bootp_);
    if (!stream.can_read(vend_field_size)) {
        throw malformed_packet();
    }
    stream.read(vend_, vend_field_size);
}

uint32_t BootP::header_size() const {
    return static_cast<uint32_t>(sizeof(bootp_) + vend_.size());
}

void BootP::opcode(uint8_t code) {
    bootp_.opcode = code;
}

void BootP::htype(uint8_t type) {
    bootp_.htype = type;
}

void BootP::hlen(uint8_t length) {
    bootp_.hlen = length;
}

void BootP::hops(uint8_t count) {
    bootp_.hops = count;
}

void BootP::xid(uint32_t identifier) {
    bootp_.xid = Endian::host_to_be(identifier);
}

void BootP::secs(uint16_t value) {
    bootp_.secs = Endian::host_to_be(value);
}

void BootP::padding(uint16_t value) {
    bootp_.padding = Endian::host_to_be(value);
}

void BootP::ciaddr(ipaddress_type address) {
    bootp_.ciaddr = address;
}

void BootP::yiaddr(ipaddress_type address) {
    bootp_.yiaddr = address;
}

void BootP::siaddr(ipaddress_type address) {
    bootp_.siaddr = address;
}

void BootP::giaddr(ipaddress_type address) {
    bootp_.giaddr = address;
}

void BootP::sname(const uint8_t* new_sname) {
    copy(new_sname, new_sname + sizeof(bootp_.sname), bootp_.sname);
}

void BootP::file(const uint8_t* new_file) {
    copy(new_file, new_file + sizeof(bootp_.file), bootp_.file);
}

void BootP::vend(const vend_type& newvend_) {
    vend_ = newvend_;
}

void BootP::write_serialization(uint8_t* buffer, uint32_t total_sz) {
    OutputMemoryStream stream(buffer, total_sz);
    stream.write(bootp_);
    stream.write(vend_.begin(), vend_.end());
}

bool BootP::matches_response(const uint8_t* ptr, uint32_t total_sz) const {
    if (total_sz < sizeof(bootp_)) {
        return false;
    }
    const bootp_header* bootp_ptr = (const bootp_header *)ptr;
    return bootp_ptr->xid == bootp_.xid;
}

} // Tins
