#include <netinet/in.h>
#include "smb.hpp"
smb::smb(const uint8_t *data, size_t length)
{
if (length < 36) {
memset(magic, 0, 4);
return;
}
memcpy(&size, data + 2, 2);
memcpy(magic, data + 4, 4);
command = data[8];
memcpy(&nt_status, data + 9, 4);
flags = data[13];
memcpy(&flags2, data + 14, 2);
memcpy(&pid_hi, data + 16, 2);
memcpy(signature, data + 18, 8);
memcpy(&reserved, data + 26, 2);
memcpy(&tid, data + 28, 2);
memcpy(&pid, data + 30, 2);
memcpy(&uid, data + 32, 2);
memcpy(&mid, data + 34, 2);
}
std::ostream &operator<<(std::ostream &lhs, const smb &rhs)
{
lhs << "Magic: ";
for (int i = 1; i < 4; i++) {
lhs << rhs.magic[i];
}
lhs << std::endl;
lhs << "Command: " << (uint16_t)rhs.command << std::endl
<< "NT Status: " << rhs.nt_status << std::endl
<< "Flags: " << (uint16_t)rhs.flags << std::endl
<< "Flags2: " << rhs.flags2 << std::endl
<< "Pid Hi: " << rhs.pid_hi << std::endl
<< "Tid: " << rhs.tid << std::endl
<< "Pid: " << rhs.pid << std::endl
<< "Uid: " << rhs.uid << std::endl
<< "Mid: " << rhs.mid << std::endl;
return lhs;
}