64 lines
1.8 KiB
C++
64 lines
1.8 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);
|
|
}
|