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);
 | |
| }
 |