2
* FreeRDP: A Remote Desktop Protocol Client
3
* pcap File Format Utils
5
* Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com>
7
* Licensed under the Apache License, Version 2.0 (the "License");
8
* you may not use this file except in compliance with the License.
9
* You may obtain a copy of the License at
11
* http://www.apache.org/licenses/LICENSE-2.0
13
* Unless required by applicable law or agreed to in writing, software
14
* distributed under the License is distributed on an "AS IS" BASIS,
15
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16
* See the License for the specific language governing permissions and
17
* limitations under the License.
27
#include <sys/timeb.h>
35
int gettimeofday(struct timeval* tp, void* tz)
37
struct _timeb timebuffer;
39
tp->tv_sec = (long) timebuffer.time;
40
tp->tv_usec = timebuffer.millitm * 1000;
45
#include <freerdp/types.h>
46
#include <freerdp/utils/memory.h>
48
#include <freerdp/utils/pcap.h>
50
#define PCAP_MAGIC 0xA1B2C3D4
52
void pcap_read_header(rdpPcap* pcap, pcap_header* header)
54
fread((void*) header, sizeof(pcap_header), 1, pcap->fp);
57
void pcap_write_header(rdpPcap* pcap, pcap_header* header)
59
fwrite((void*) header, sizeof(pcap_header), 1, pcap->fp);
62
void pcap_read_record_header(rdpPcap* pcap, pcap_record_header* record)
64
fread((void*) record, sizeof(pcap_record_header), 1, pcap->fp);
67
void pcap_write_record_header(rdpPcap* pcap, pcap_record_header* record)
69
fwrite((void*) record, sizeof(pcap_record_header), 1, pcap->fp);
72
void pcap_read_record(rdpPcap* pcap, pcap_record* record)
74
pcap_read_record_header(pcap, &record->header);
75
record->length = record->header.incl_len;
76
record->data = xmalloc(record->length);
77
fread(record->data, record->length, 1, pcap->fp);
80
void pcap_write_record(rdpPcap* pcap, pcap_record* record)
82
pcap_write_record_header(pcap, &record->header);
83
fwrite(record->data, record->length, 1, pcap->fp);
86
void pcap_add_record(rdpPcap* pcap, void* data, uint32 length)
91
if (pcap->tail == NULL)
93
pcap->tail = (pcap_record*) xzalloc(sizeof(pcap_record));
94
pcap->head = pcap->tail;
95
pcap->record = pcap->head;
100
record = (pcap_record*) xzalloc(sizeof(pcap_record));
101
pcap->tail->next = record;
105
if (pcap->record == NULL)
106
pcap->record = record;
109
record->length = length;
110
record->header.incl_len = length;
111
record->header.orig_len = length;
113
gettimeofday(&tp, 0);
114
record->header.ts_sec = tp.tv_sec;
115
record->header.ts_usec = tp.tv_usec;
118
boolean pcap_has_next_record(rdpPcap* pcap)
120
if (pcap->file_size - (ftell(pcap->fp)) <= 16)
126
boolean pcap_get_next_record_header(rdpPcap* pcap, pcap_record* record)
128
if (pcap_has_next_record(pcap) != true)
131
pcap_read_record_header(pcap, &record->header);
132
record->length = record->header.incl_len;
133
record->data = xmalloc(record->length);
138
boolean pcap_get_next_record_content(rdpPcap* pcap, pcap_record* record)
140
fread(record->data, record->length, 1, pcap->fp);
144
boolean pcap_get_next_record(rdpPcap* pcap, pcap_record* record)
146
if (pcap_has_next_record(pcap) != true)
149
pcap_read_record(pcap, record);
154
rdpPcap* pcap_open(char* name, boolean write)
158
FILE *pcap_fp = fopen(name, write ? "w+" : "r");
161
perror("opening pcap dump");
165
pcap = (rdpPcap*) xzalloc(sizeof(rdpPcap));
171
pcap->record_count = 0;
176
pcap->header.magic_number = 0xA1B2C3D4;
177
pcap->header.version_major = 2;
178
pcap->header.version_minor = 4;
179
pcap->header.thiszone = 0;
180
pcap->header.sigfigs = 0;
181
pcap->header.snaplen = 0xFFFFFFFF;
182
pcap->header.network = 0;
183
pcap_write_header(pcap, &pcap->header);
187
fseek(pcap->fp, 0, SEEK_END);
188
pcap->file_size = (int) ftell(pcap->fp);
189
fseek(pcap->fp, 0, SEEK_SET);
190
pcap_read_header(pcap, &pcap->header);
197
void pcap_flush(rdpPcap* pcap)
199
while (pcap->record != NULL)
201
pcap_write_record(pcap, pcap->record);
202
pcap->record = pcap->record->next;
205
if (pcap->fp != NULL)
209
void pcap_close(rdpPcap* pcap)
213
if (pcap->fp != NULL)