wlanhcx2essid.c

#define _GNU_SOURCE
#include <stdarg.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <limits.h>
#include <time.h>
#include <pcap.h>
#include <sys/stat.h>
#include <stdio_ext.h>
#include "common.h"

/*===========================================================================*/
/* globale Variablen */

hcx_t *hcxdata = NULL;
hcx_t *hcxdata2 = NULL;

/*===========================================================================*/
int sort_by_essid(const void *a, const void *b) 
{ 
hcx_t *ia = (hcx_t *)a;
hcx_t *ib = (hcx_t *)b;

return memcmp(ia->essid, ib->essid, 32);
}
/*===========================================================================*/
int writehccapx(char *hcxoutname, int hcxrecords, int hcxrecords2)
{
hcx_t *zeigerhcx;
hcx_t *zeigerhcx1;
hcx_t *zeigerhcx2;

FILE *fhhcx;
int c, c2;

if(hcxoutname == NULL)
	return FALSE;

if((fhhcx = fopen(hcxoutname, "ab")) == NULL)
	{
	fprintf(stderr, "error opening file %s", hcxoutname);
	return FALSE;
	}

qsort(hcxdata, hcxrecords, HCX_SIZE, sort_by_essid);
qsort(hcxdata2, hcxrecords2, HCX_SIZE, sort_by_essid);

c = 0;
while(c < hcxrecords)
	{
	zeigerhcx = hcxdata +c;
	zeigerhcx1 = hcxdata +c +1;
	fwrite(zeigerhcx, HCX_SIZE, 1, fhhcx);
	if((zeigerhcx->essid_len == zeigerhcx1->essid_len) && (memcmp(zeigerhcx->essid, zeigerhcx1->essid, zeigerhcx->essid_len) == 0))
		{
		c++;
		continue;
		}
	
	zeigerhcx2 = hcxdata2;
	c2 = 0;
	while(c2 < hcxrecords2)
		{
		if((zeigerhcx->essid_len == zeigerhcx2->essid_len) && (memcmp(zeigerhcx->essid, zeigerhcx2->essid, zeigerhcx->essid_len) == 0))
			fwrite(zeigerhcx2, HCX_SIZE, 1, fhhcx);
		zeigerhcx2++;
		c2++;
		}

	c++;
	}
fclose(fhhcx);
return TRUE;
}
/*===========================================================================*/
int readhccapx2(char *hcxinname)
{
struct stat statinfo;
FILE *fhhcx;
int hcxsize = 0;

if(hcxinname == NULL)
	return FALSE;

if(stat(hcxinname, &statinfo) != 0)
	{
	fprintf(stderr, "can't stat %s\n", hcxinname);
	return FALSE;
	}

if((statinfo.st_size % HCX_SIZE) != 0)
	{
	fprintf(stderr, "file corrupt\n");
	return FALSE;
	}

if((fhhcx = fopen(hcxinname, "rb")) == NULL)
	{
	fprintf(stderr, "error opening file %s", hcxinname);
	return FALSE;
	}

hcxdata2 = malloc(statinfo.st_size);
if(hcxdata2 == NULL)	
		{
		fprintf(stderr, "out of memory to store second hccapx data\n");
		return FALSE;
		}

hcxsize = fread(hcxdata2, 1, statinfo.st_size +HCX_SIZE, fhhcx);
if(hcxsize != statinfo.st_size)	
	{
	fprintf(stderr, "error reading hccapx file %s", hcxinname);
	return FALSE;
	}
fclose(fhhcx);
printf("%ld records readed from %s\n", hcxsize / HCX_SIZE, hcxinname);
return hcxsize / HCX_SIZE;
}
/*===========================================================================*/
int readhccapx(char *hcxinname)
{
struct stat statinfo;
FILE *fhhcx;
int hcxsize = 0;

if(hcxinname == NULL)
	return FALSE;

if(stat(hcxinname, &statinfo) != 0)
	{
	fprintf(stderr, "can't stat %s\n", hcxinname);
	return FALSE;
	}

if((statinfo.st_size % HCX_SIZE) != 0)
	{
	fprintf(stderr, "file corrupt\n");
	return FALSE;
	}

if((fhhcx = fopen(hcxinname, "rb")) == NULL)
	{
	fprintf(stderr, "error opening file %s", hcxinname);
	return FALSE;
	}

hcxdata = malloc(statinfo.st_size);
if(hcxdata == NULL)	
		{
		fprintf(stderr, "out of memory to store hccapx data\n");
		return FALSE;
		}

hcxsize = fread(hcxdata, 1, statinfo.st_size +HCX_SIZE, fhhcx);
if(hcxsize != statinfo.st_size)	
	{
	fprintf(stderr, "error reading hccapx file %s", hcxinname);
	return FALSE;
	}
fclose(fhhcx);

printf("%ld records readed from %s\n", hcxsize / HCX_SIZE, hcxinname);
return hcxsize / HCX_SIZE;
}
/*===========================================================================*/
static void usage(char *eigenname)
{
printf("%s %s (C) %s ZeroBeat\n"
	"usage: %s <options>\n"
	"\n"
	"options:\n"
	"-i <file> : input hccapx file\n"
	"-s <file> : input second hccapx file\n"
	"-o <file> : output hccapx file (merged by essid)\n"
	"-I        : show info about hccapx file\n"
	"\n", eigenname, VERSION, VERSION_JAHR, eigenname);
exit(EXIT_FAILURE);
}
/*===========================================================================*/
int main(int argc, char *argv[])
{
int auswahl;
int hcxorgrecords = 0;
int hcxorgrecords2 = 0;

char *eigenname = NULL;
char *eigenpfadname = NULL;
char *hcxinname = NULL;
char *hcxinname2 = NULL;
char *hcxoutname = NULL;


eigenpfadname = strdupa(argv[0]);
eigenname = basename(eigenpfadname);

setbuf(stdout, NULL);
while ((auswahl = getopt(argc, argv, "i:s:o:Ihv")) != -1)
	{
	switch (auswahl)
		{
		case 'i':
		hcxinname = optarg;
		break;

		case 'o':
		hcxoutname = optarg;
		break;

		case 's':
		hcxinname2 = optarg;
		break;

		default:
		usage(eigenname);
		break;
		}
	}

hcxorgrecords = readhccapx(hcxinname);
if(hcxorgrecords == 0)
	return EXIT_SUCCESS;

hcxorgrecords2 = readhccapx2(hcxinname2);
if(hcxorgrecords2 == 0)
	return EXIT_SUCCESS;

writehccapx(hcxoutname, hcxorgrecords, hcxorgrecords2);


if(hcxdata2 != NULL)
	free(hcxdata2);


if(hcxdata != NULL)
	free(hcxdata);

return EXIT_SUCCESS;
}