forked from dolphin-emu/dolphin
		
	
		
			
				
	
	
		
			67 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			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);
 | 
						|
}
 |