Add the magic runes to the end, and reformat.

Move the two static functions to the top, so that forward prototyping
is not required.

--HG--
extra : convert_revision : svn%3Aeff31bef-be4a-0410-a8fe-e47997df2690/trunk%4092
issue20
tibs 2008-11-11 19:46:58 +00:00
rodzic 0b7868f9ca
commit ebf0a5da1b
1 zmienionych plików z 135 dodań i 135 usunięć

186
pcap.c
Wyświetl plik

@ -33,98 +33,6 @@
#include "pcap.h"
#include "misc_fns.h"
static int pcap_read_header(PCAP_reader_p ctx, pcap_hdr_t *out_hdr);
static int pcap_read_pktheader(PCAP_reader_p ctx, pcaprec_hdr_t *out_hdr);
int pcap_open(PCAP_reader_p *ctx_p, pcap_hdr_t *out_hdr,
const char *filename)
{
FILE *fptr = (filename ? fopen(filename, "rb") : stdin);
PCAP_reader_p ctx;
int rv;
(*ctx_p) = NULL;
if (!fptr)
{
// Couldn't open the file.
return -1;
}
ctx = (PCAP_reader_p)malloc(SIZEOF_PCAP_READER);
if (!ctx)
{
fclose(fptr);
// Out of memory.
return -2;
}
ctx->file = fptr;
rv = pcap_read_header(ctx, out_hdr);
if (rv != 1)
{
// Header read failed.
fclose(ctx->file);
free(ctx);
return -4;
}
(*ctx_p) = ctx;
return 0;
}
int pcap_read_next(PCAP_reader_p ctx, pcaprec_hdr_t *out_hdr,
uint8_t **out_data,
uint32_t *out_len)
{
int rv;
(*out_data) = NULL; (*out_len) = 0;
rv = pcap_read_pktheader(ctx, out_hdr);
if (rv != 1) { return rv; }
// Otherwise we now know how long our packet is ..
(*out_data) = (uint8_t *)malloc(out_hdr->incl_len);
if (!(*out_data))
{
// Out of memory.
return -3;
}
(*out_len) = out_hdr->incl_len;
rv = fread((*out_data), (*out_len), 1, ctx->file);
if (rv != 1)
{
free(*out_data); (*out_data) = NULL;
*out_len = 0;
if (feof(ctx->file))
{
// Ah. EOF.
return 0;
}
else
{
// Error. Curses.
return rv;
}
}
else if (rv == 1)
{
// Gotcha.
return 1;
}
return 0;
}
static int pcap_read_header(PCAP_reader_p ctx, pcap_hdr_t *hdr)
{
uint8_t hdr_val[SIZEOF_PCAP_HDR_ON_DISC];
@ -216,5 +124,97 @@ static int pcap_read_pktheader(PCAP_reader_p ctx, pcaprec_hdr_t *hdr)
return 1;
}
/* End file */
extern int pcap_open(PCAP_reader_p *ctx_p, pcap_hdr_t *out_hdr,
const char *filename)
{
FILE *fptr = (filename ? fopen(filename, "rb") : stdin);
PCAP_reader_p ctx;
int rv;
(*ctx_p) = NULL;
if (!fptr)
{
// Couldn't open the file.
return -1;
}
ctx = (PCAP_reader_p)malloc(SIZEOF_PCAP_READER);
if (!ctx)
{
fclose(fptr);
// Out of memory.
return -2;
}
ctx->file = fptr;
rv = pcap_read_header(ctx, out_hdr);
if (rv != 1)
{
// Header read failed.
fclose(ctx->file);
free(ctx);
return -4;
}
(*ctx_p) = ctx;
return 0;
}
extern int pcap_read_next(PCAP_reader_p ctx, pcaprec_hdr_t *out_hdr,
uint8_t **out_data,
uint32_t *out_len)
{
int rv;
(*out_data) = NULL; (*out_len) = 0;
rv = pcap_read_pktheader(ctx, out_hdr);
if (rv != 1) { return rv; }
// Otherwise we now know how long our packet is ..
(*out_data) = (uint8_t *)malloc(out_hdr->incl_len);
if (!(*out_data))
{
// Out of memory.
return -3;
}
(*out_len) = out_hdr->incl_len;
rv = fread((*out_data), (*out_len), 1, ctx->file);
if (rv != 1)
{
free(*out_data); (*out_data) = NULL;
*out_len = 0;
if (feof(ctx->file))
{
// Ah. EOF.
return 0;
}
else
{
// Error. Curses.
return rv;
}
}
else if (rv == 1)
{
// Gotcha.
return 1;
}
return 0;
}
// Local Variables:
// tab-width: 8
// indent-tabs-mode: nil
// c-basic-offset: 2
// End:
// vim: set tabstop=8 shiftwidth=2 expandtab: