dolphin/Source/Core/Common/PcapFile.cpp

67 lines
1.7 KiB
C++

// Copyright 2014 Dolphin Emulator Project
// Licensed under GPLv2
// Refer to the license.txt file included.
#include <chrono>
#include "Common/CommonTypes.h"
#include "Common/FileUtil.h"
#include "Common/PcapFile.h"
namespace {
const u32 PCAP_MAGIC = 0xa1b2c3d4;
const u16 PCAP_VERSION_MAJOR = 2;
const u16 PCAP_VERSION_MINOR = 4;
const u32 PCAP_CAPTURE_LENGTH = 65535;
// TODO(delroth): Make this configurable at PCAP creation time?
const u32 PCAP_DATA_LINK_TYPE = 147; // Reserved for internal use.
// Designed to be directly written into the PCAP file. The PCAP format is
// endian independent, so this works just fine.
#pragma pack(push, 1)
struct PCAPHeader
{
u32 magic_number;
u16 version_major;
u16 version_minor;
s32 tz_offset; // Offset in seconds from the GMT timezone.
u32 ts_accuracy; // In practice, 0.
u32 capture_length; // Size at which we truncate packets.
u32 data_link_type;
};
struct PCAPRecordHeader
{
u32 ts_sec;
u32 ts_usec;
u32 size_in_file; // Size after eventual truncation.
u32 real_size; // Size before eventual truncation.
};
#pragma pack(pop)
} // namespace
void PCAP::AddHeader()
{
PCAPHeader hdr = {
PCAP_MAGIC, PCAP_VERSION_MAJOR, PCAP_VERSION_MINOR,
0, 0, PCAP_CAPTURE_LENGTH, PCAP_DATA_LINK_TYPE
};
m_fp->WriteBytes(&hdr, sizeof (hdr));
}
void PCAP::AddPacket(const u8* bytes, size_t size)
{
std::chrono::system_clock::time_point now(std::chrono::system_clock::now());
auto ts = now.time_since_epoch();
PCAPRecordHeader rec_hdr = {
(u32)std::chrono::duration_cast<std::chrono::seconds>(ts).count(),
(u32)(std::chrono::duration_cast<std::chrono::microseconds>(ts).count() % 1000000),
(u32)size, (u32)size
};
m_fp->WriteBytes(&rec_hdr, sizeof (rec_hdr));
m_fp->WriteBytes(bytes, size);
}