diff options
Diffstat (limited to 'libfreerdp/utils/pcap.c')
| -rw-r--r-- | libfreerdp/utils/pcap.c | 286 |
1 files changed, 286 insertions, 0 deletions
diff --git a/libfreerdp/utils/pcap.c b/libfreerdp/utils/pcap.c new file mode 100644 index 0000000..db50909 --- /dev/null +++ b/libfreerdp/utils/pcap.c @@ -0,0 +1,286 @@ +/** + * FreeRDP: A Remote Desktop Protocol Implementation + * pcap File Format Utils + * + * Copyright 2011 Marc-Andre Moreau <marcandre.moreau@gmail.com> + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include <freerdp/config.h> + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> + +#include <winpr/wtypes.h> +#include <winpr/assert.h> +#include <winpr/file.h> +#include <winpr/crt.h> +#include <freerdp/log.h> + +#define TAG FREERDP_TAG("utils") + +#ifndef _WIN32 +#include <sys/time.h> +#else +#include <time.h> +#include <sys/timeb.h> +#include <winpr/windows.h> + +int gettimeofday(struct timeval* tp, void* tz) +{ + struct _timeb timebuffer; + _ftime(&timebuffer); + tp->tv_sec = (long)timebuffer.time; + tp->tv_usec = timebuffer.millitm * 1000; + return 0; +} +#endif + +#include <freerdp/types.h> +#include <freerdp/utils/pcap.h> + +#define PCAP_MAGIC 0xA1B2C3D4 + +struct rdp_pcap +{ + FILE* fp; + char* name; + BOOL write; + INT64 file_size; + size_t record_count; + pcap_header header; + pcap_record* head; + pcap_record* tail; + pcap_record* record; +}; + +static BOOL pcap_read_header(rdpPcap* pcap, pcap_header* header) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(header); + + return fread(header, sizeof(pcap_header), 1, pcap->fp) == 1; +} + +static BOOL pcap_write_header(rdpPcap* pcap, const pcap_header* header) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(header); + + return fwrite(header, sizeof(pcap_header), 1, pcap->fp) == 1; +} + +static BOOL pcap_read_record_header(rdpPcap* pcap, pcap_record_header* record) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(record); + + return fread(record, sizeof(pcap_record_header), 1, pcap->fp) == 1; +} + +static BOOL pcap_write_record_header(rdpPcap* pcap, const pcap_record_header* record) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(record); + + return fwrite(record, sizeof(pcap_record_header), 1, pcap->fp) == 1; +} + +static BOOL pcap_read_record(rdpPcap* pcap, pcap_record* record) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(record); + + if (!pcap_read_record_header(pcap, &record->header)) + return FALSE; + + record->length = record->header.incl_len; + record->data = malloc(record->length); + if (!record->data) + return FALSE; + + if (fread(record->data, record->length, 1, pcap->fp) != 1) + { + free(record->data); + record->data = NULL; + return FALSE; + } + return TRUE; +} + +static BOOL pcap_write_record(rdpPcap* pcap, const pcap_record* record) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(record); + + return pcap_write_record_header(pcap, &record->header) && + (fwrite(record->cdata, record->length, 1, pcap->fp) == 1); +} + +BOOL pcap_add_record(rdpPcap* pcap, const void* data, size_t length) +{ + pcap_record* record = NULL; + struct timeval tp; + + WINPR_ASSERT(pcap); + WINPR_ASSERT(data || (length == 0)); + WINPR_ASSERT(length <= UINT32_MAX); + + record = (pcap_record*)calloc(1, sizeof(pcap_record)); + if (!record) + return FALSE; + + record->cdata = data; + record->length = length; + record->header.incl_len = (UINT32)length; + record->header.orig_len = (UINT32)length; + + gettimeofday(&tp, 0); + record->header.ts_sec = (UINT32)tp.tv_sec; + record->header.ts_usec = (UINT32)tp.tv_usec; + + if (pcap->tail == NULL) + { + pcap->tail = record; + if (!pcap->tail) + return FALSE; + + pcap->head = pcap->tail; + } + else + { + record->next = pcap->tail; + pcap->tail = record; + } + + if (pcap->record == NULL) + pcap->record = record; + + return TRUE; +} + +BOOL pcap_has_next_record(const rdpPcap* pcap) +{ + WINPR_ASSERT(pcap); + + if (pcap->file_size - (_ftelli64(pcap->fp)) <= 16) + return FALSE; + + return TRUE; +} + +BOOL pcap_get_next_record_header(rdpPcap* pcap, pcap_record* record) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(record); + + if (pcap_has_next_record(pcap) != TRUE) + return FALSE; + + pcap_read_record_header(pcap, &record->header); + record->length = record->header.incl_len; + + return TRUE; +} + +BOOL pcap_get_next_record_content(rdpPcap* pcap, pcap_record* record) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(record); + + return fread(record->data, record->length, 1, pcap->fp) == 1; +} + +BOOL pcap_get_next_record(rdpPcap* pcap, pcap_record* record) +{ + WINPR_ASSERT(pcap); + WINPR_ASSERT(record); + + return pcap_has_next_record(pcap) && pcap_read_record(pcap, record); +} + +rdpPcap* pcap_open(const char* name, BOOL write) +{ + rdpPcap* pcap = NULL; + + WINPR_ASSERT(name); + + pcap = (rdpPcap*)calloc(1, sizeof(rdpPcap)); + if (!pcap) + goto fail; + + pcap->name = _strdup(name); + pcap->write = write; + pcap->record_count = 0; + pcap->fp = winpr_fopen(name, write ? "w+b" : "rb"); + + if (pcap->fp == NULL) + goto fail; + + if (write) + { + pcap->header.magic_number = PCAP_MAGIC; + pcap->header.version_major = 2; + pcap->header.version_minor = 4; + pcap->header.thiszone = 0; + pcap->header.sigfigs = 0; + pcap->header.snaplen = UINT32_MAX; + pcap->header.network = 0; + if (!pcap_write_header(pcap, &pcap->header)) + goto fail; + } + else + { + _fseeki64(pcap->fp, 0, SEEK_END); + pcap->file_size = _ftelli64(pcap->fp); + _fseeki64(pcap->fp, 0, SEEK_SET); + if (!pcap_read_header(pcap, &pcap->header)) + goto fail; + } + + return pcap; + +fail: + pcap_close(pcap); + return NULL; +} + +void pcap_flush(rdpPcap* pcap) +{ + WINPR_ASSERT(pcap); + + while (pcap->record != NULL) + { + pcap_write_record(pcap, pcap->record); + pcap->record = pcap->record->next; + } + + if (pcap->fp != NULL) + fflush(pcap->fp); +} + +void pcap_close(rdpPcap* pcap) +{ + if (!pcap) + return; + + pcap_flush(pcap); + + if (pcap->fp != NULL) + fclose(pcap->fp); + + free(pcap->name); + free(pcap); +} |