#define _GNU_SOURCE

#include <ctype.h>
#include <libgen.h>
#include <pwd.h>
#include <stdbool.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>

#include <curl/curl.h>

#include "include/strings.c"

#define LINEBUFFER_MAX 256

#define OUIBUFFER_MAX 8192

const char *ouiurl = "https://standards-oui.ieee.org/oui/oui.txt";

/*===========================================================================*/
static bool downloadoui(char *ouiname)
{
static size_t bread;
static CURLcode ret;
static CURL *hnd;
static FILE *fhoui;
static FILE *fhouitmp;
static char ouibuff[OUIBUFFER_MAX];

fprintf(stdout, "start downloading oui from https://standards-oui.ieee.org to: %s\

n", ouiname);
if((fhouitmp = tmpfile()) == NULL)
{
fprintf(stderr, "failed to create temporary download file\n");
return false;
}
hnd = curl_easy_init ();
curl_easy_setopt(hnd, CURLOPT_URL, "https://standards-oui.ieee.org/oui/oui.txt");
curl_easy_setopt(hnd, CURLOPT_NOPROGRESS, 1L);
curl_easy_setopt(hnd, CURLOPT_MAXREDIRS, 5L);
curl_easy_setopt(hnd, CURLOPT_WRITEDATA, fhouitmp);
curl_easy_setopt(hnd, CURLOPT_NOPROGRESS, 0L);
ret = curl_easy_perform(hnd);
curl_easy_cleanup(hnd);
if(ret != 0)
{
fprintf(stderr, "\ndownload not successful\n");
exit(EXIT_FAILURE);
}
rewind(fhouitmp);
if((fhoui = fopen(ouiname, "w")) == NULL)
{
fprintf(stderr, "\nerror creating file %s\n", ouiname);
exit(EXIT_FAILURE);
}
while (!feof(fhouitmp))
{
 bread = fread(ouibuff, 1, sizeof(ouibuff), fhouitmp);
if(bread > 0) fwrite(ouibuff, 1, bread, fhoui);
}
fclose(fhoui);
fprintf(stdout, "\ndownload finished\n");
return true;
}
/*===========================================================================*/
static size_t chop(char *buffer, size_t len)
{
static char *ptr;

ptr = buffer +len -1;

while (len) {
if (*ptr != '\n') break;
*ptr-- = 0;
len--;
}

while (len) {
if (*ptr != '\r') break;
*ptr-- = 0;
len--;
}
return len;
}
/*---------------------------------------------------------------------------*/
static int fgetline(FILE *inputstream, size_t size, char *buffer)
{
if (feof(inputstream)) return -1;
char *buffptr = fgets (buffer, size, inputstream);

if (buffptr == NULL) return -1;

size_t len = strlen(buffptr);

len = chop(buffptr, len);

return len;
}
/*===========================================================================*/
static void getessidinfo(char *essidname)
{
static int l, p;
static uint8_t essidbuffer[66];

l = strlen(essidname);
if((l < 2) || (l > 64))
{
fprintf(stderr, "not a valid ESSID hex string\n");
return;
}
if((l %2) != 0)
{
fprintf(stderr, "not a valid hex string\n");
return;
}
for(p = 0; p < l; p++)
{
if(!isxdigit((unsigned char)essidname[p]))
 {
fprintf(stderr, "not a valid hex string\n");
return;
}
}

memset(&essidbuffer, 0, 66);
if(hex2bin(essidname, essidbuffer, l /2) == false)
{
fprintf(stderr, "not a valid ESSID hex string\n");
return;
}
fprintf(stdout, "%s\n", essidbuffer);
return;
}
/*===========================================================================*/
static void gethexessidinfo(char *hexessidname)
{
static int l, p;
l = strlen(hexessidname);
for(p = 0; p < l; p++)
{
fprintf(stdout, "%02x", hexessidname[p]);
}
fprintf(stdout, "\n");
return;
}
/*===========================================================================*/
static void get16800info(char *ouiname, char *hash16800line)
{
static int len;
static int l, l1;
static FILE* fhoui;
static char *vendorptr;
static char *essidptr;
static char *passwdptr;

static unsigned long long int macap;

static unsigned long long int macsta;
static unsigned long long int ouiap;
static unsigned long long int ouista;
static unsigned long long int vendoroui;

static char linein[LINEBUFFER_MAX];

static uint8_t essidbuffer[66];
static char vendorapname[256];
static char vendorstaname[256];

sscanf(&hash16800line[33], "%12llx", &macap);

ouiap = macap >> 24;
sscanf(&hash16800line[46], "%12llx", &macsta);
ouista = macsta >> 24;

essidptr = hash16800line +59;

l = strlen(essidptr);

passwdptr = strrchr(essidptr, ':');

if(passwdptr != NULL)
{
 l1 = strlen(passwdptr);
if(l1 > 1)
{
l -= l1;
}
}
if((l%2 != 0) || (l > 64))
{
fprintf(stderr, "xxx wrong ESSID length %s\n", essidptr);
return;
}
memset(&essidbuffer, 0, 66);
if(hex2bin(essidptr, essidbuffer, l /2) == false)
{
fprintf(stderr, "wrong ESSID %s\n", essidptr);
return;
}

if ((fhoui = fopen(ouiname, "r")) == NULL)

{
fprintf(stderr, "unable to open database %s\n", ouiname);
exit (EXIT_FAILURE);
}

strncpy(vendorapname, "unknown", 8);

strncpy(vendorstaname, "unknown", 8);

while((len = fgetline(fhoui, LINEBUFFER_MAX, linein)) != -1)

{
if (len < 10)
continue;
if(strstr(linein, "(base 16)") != NULL)
{
sscanf(linein, "%06llx", &vendoroui);
if(ouiap == vendoroui)
{
vendorptr = strrchr(linein, '\t');
if(vendorptr != NULL)
{
strncpy(vendorapname, vendorptr +1,255);
}
}
if(ouista == vendoroui)
{
vendorptr = strrchr(linein, '\t');
if(vendorptr != NULL)
{
strncpy(vendorstaname, vendorptr +1,255);
}
}
}
}
if(isasciistring(l /2, essidbuffer) == true)
{
fprintf(stdout, "\nESSID..: %s\n", essidbuffer);
}
else
{
fprintf(stdout, "\nESSID..: $HEX[%s]\n", essidbuffer);
 }

fprintf(stdout, "MAC_AP.: %012llx\n"

"VENDOR.: %s\n"
"MAC_STA: %012llx\n"
"VENDOR.: %s\n\n"
, macap, vendorapname, macsta, vendorstaname);

fclose(fhoui);
return;
}
/*===========================================================================*/
static void get2500info(char *ouiname, char *hash2500line)
{
static int len;
static int l, l1;
static FILE* fhoui;
static char *vendorptr;
static char *essidptr;
static char *passwdptr;

static unsigned long long int macap;

static unsigned long long int macsta;
static unsigned long long int ouiap;
static unsigned long long int ouista;
static unsigned long long int vendoroui;

static char linein[LINEBUFFER_MAX];

static uint8_t essidbuffer[72];
static char vendorapname[256];
static char vendorstaname[256];

sscanf(&hash2500line[33], "%12llx", &macap);

ouiap = macap >> 24;
sscanf(&hash2500line[46], "%12llx", &macsta);
ouista = macsta >> 24;

essidptr = hash2500line +59;

l = strlen(essidptr);
passwdptr = strrchr(hash2500line, ':');
if((passwdptr -hash2500line) > 59)
{
if(passwdptr != NULL)
{
l1 = strlen(passwdptr);
if(l1 > 1)
{
l -= l1;
}
}
}
if(l > 70)
{
fprintf(stderr, "wrong ESSID length %s %d\n", essidptr, l);
return;
}
memset(&essidbuffer, 0, 72);
memcpy(&essidbuffer, essidptr, l);
if ((fhoui = fopen(ouiname, "r")) == NULL)
{
fprintf(stderr, "unable to open database %s\n", ouiname);
exit (EXIT_FAILURE);
}

strncpy(vendorapname, "unknown", 8);

strncpy(vendorstaname, "unknown", 8);

while((len = fgetline(fhoui, LINEBUFFER_MAX, linein)) != -1)

{
if (len < 10)
continue;
if(strstr(linein, "(base 16)") != NULL)
{
sscanf(linein, "%06llx", &vendoroui);
if(ouiap == vendoroui)
{
vendorptr = strrchr(linein, '\t');
if(vendorptr != NULL)
{
strncpy(vendorapname, vendorptr +1,255);
}
}
if(ouista == vendoroui)
{
vendorptr = strrchr(linein, '\t');
if(vendorptr != NULL)
{
strncpy(vendorstaname, vendorptr +1,255);
}
}
}
}
fprintf(stdout, "\nESSID..: %s\n", essidbuffer);
fprintf(stdout, "MAC_AP.: %012llx\n"
"VENDOR.: %s\n"
"MAC_STA: %012llx\n"
"VENDOR.: %s\n\n"
, macap, vendorapname, macsta, vendorstaname);

fclose(fhoui);
return;
}
/*===========================================================================*/
static void getoui(char *ouiname, unsigned long long int oui)
{
static int len;
static FILE* fhoui;
static char *vendorptr;
static unsigned long long int vendoroui;
static char linein[LINEBUFFER_MAX];
static char vendorapname[256];
#ifdef BIG_ENDIAN_HOST
int lsb = oui & 0xf;
#else
int lsb = (oui >> 16) & 0xf;
#endif
if ((fhoui = fopen(ouiname, "r")) == NULL)
{
fprintf(stderr, "unable to open database %s\n", ouiname);
exit (EXIT_FAILURE);
}

strncpy(vendorapname, "unknown", 8);

while((len = fgetline(fhoui, LINEBUFFER_MAX, linein)) != -1)
{
if (len < 10)
continue;

if(strstr(linein, "(base 16)") != NULL)

{
sscanf(linein, "%06llx", &vendoroui);
if(oui == vendoroui)
{
vendorptr = strrchr(linein, '\t');
if(vendorptr != NULL)
{
strncpy(vendorapname, vendorptr +1,255);
break;
}
}
}
}

fprintf(stdout, "\nVENDOR: %s (%s)%s\n\n", vendorapname,

oui == 0xffffff ? "broadcast" : lsb & 2 ? "LAA - likely randomized!" :
"UAA",
oui == 0xffffff ? "" : lsb & 1 ? ", multicast" : ", unicast");

fclose(fhoui);
return;
}
/*===========================================================================*/
static void getvendor(char *ouiname, char *vendorstring)
{
static int len;

static FILE* fhoui;

static char *vendorptr;
static unsigned long long int vendoroui;

char linein[LINEBUFFER_MAX];

if ((fhoui = fopen(ouiname, "r")) == NULL)

{
fprintf(stderr, "unable to open database %s\n", ouiname);
exit (EXIT_FAILURE);
}

while((len = fgetline(fhoui, LINEBUFFER_MAX, linein)) != -1)

{
if (len < 10)
continue;

if(strstr(linein, "(base 16)") != NULL)

 {
if(strstr(linein, vendorstring) != NULL)
{
sscanf(linein, "%06llx", &vendoroui);
vendorptr = strrchr(linein, '\t');
fprintf(stdout, "%06llx%s\n", vendoroui, vendorptr);
}
}
}
fclose(fhoui);
return;
}
/*===========================================================================*/
__attribute__ ((noreturn))
static void usage(char *eigenname)
{
fprintf(stdout, "%s %s (C) %s ZeroBeat\n"
"usage: %s <options>\n"
"\n"
"options:\n"
"-d : download https://standards-oui.ieee.org/oui/oui.txt\n"
" : and save to ~/.hcxtools/oui.txt\n"
" : internet connection required\n"
"-m <mac> : mac (six bytes of mac addr) or \n"
" : oui (fist three bytes of mac addr)\n"
"-p <hashline> : input PMKID and/or EAPOL hashline (hashmode 22000 or 16800)\
n"
"-P <hashline> : input EAPOL hashline from potfile (hashcat <= 5.1.0)\n"
"-e <ESSID> : input ESSID\n"
"-x <xdigit> : input ESSID in hex\n"
"-v <vendor> : vendor name\n"
"-h : this help screen\n"
"\n", eigenname, VERSION_TAG, VERSION_YEAR, eigenname);
exit(EXIT_SUCCESS);
}
/*===========================================================================*/
int main(int argc, char *argv[])
{
static int auswahl;
static int mode = 0;
static int ret;
static int l;
static int p1, p2;
static unsigned long long int oui = 0;
static char *hashlineend;

static uid_t uid;

static struct passwd *pwd;
static struct stat statinfo;
static char *vendorname = NULL;
static char *hexessidname = NULL;
static char *essidname = NULL;
static char *hash16800line = NULL;
static char *hash2500line = NULL;
static char *ouiname = NULL;
static char confdirname[] = ".hcxtools";
static char ouinameuser[] = ".hcxtools/oui.txt";
static char ouinamesystemwide[] = "/usr/share/ieee-data/oui.txt";
static char pmkidtype[] = {"WPA*01*" };
static char eapoltype[] = {"WPA*02*" };
static char pmkidtypeend[] = {"***" };

while ((auswahl = getopt(argc, argv, "m:v:p:P:e:x:dh")) != -1)

{
switch (auswahl)
{
case 'd':
mode = 'd';
break;

case 'm':
l= strlen(optarg);
if((l > 17) || (l < 6))
{
fprintf(stderr, "error wrong oui size %s (need eg.
11:22:33:44:55:aa or 112233)\n", optarg);
exit(EXIT_FAILURE);
}
p2 = 0;
for(p1 = 0; p1 < l; p1++)
{
if(isxdigit((unsigned char)optarg[p1]))
{
optarg[p2] = optarg[p1];
p2++;
}
}
optarg[6] = 0;
for(p1 = 0; p1 < 6; p1++)
{
if(!isxdigit((unsigned char)optarg[p1]))
{
fprintf(stderr, "error wrong oui %s (need eg.
11:22:33:44:55:aa or 112233)\n", optarg);
exit(EXIT_FAILURE);
}
}
oui = strtoull(optarg, NULL, 16);
mode = 'm';
break;

case 'p':
hash16800line = optarg;
l = strlen(hash16800line);
if(l < 61)
{
fprintf(stderr, "error hashline too short %s\n", optarg);
exit(EXIT_FAILURE);
}
if(memcmp(&pmkidtype, hash16800line, 7) == 0)
{
hash16800line += 7;
l -=7;
if(memcmp(&pmkidtypeend, &hash16800line[l -3], 3) == 0)
hash16800line[l -3] = 0;
}
if(memcmp(&eapoltype, hash16800line, 7) == 0)
 {
hash16800line += 7;
hashlineend = strchr(&hash16800line[59], '*');
if(hashlineend != NULL) hashlineend[0] = 0;
}
if(((hash16800line[32] != ':') && (hash16800line[45] != ':') &&
(hash16800line[58] != ':')) && ((hash16800line[32] != '*') && (hash16800line[45] !=
'*') && (hash16800line[58] != '*')))
{
fprintf(stderr, "error hashline wrong format %s\n", optarg);
exit(EXIT_FAILURE);
}
mode = 'p';
break;

case 'P':
hash2500line = optarg;
l = strlen(hash2500line);
if(l < 61)
{
fprintf(stderr, "error hashline too short %s\n", optarg);
exit(EXIT_FAILURE);
}
if((hash2500line[32] != ':') && (hash2500line[45] != ':') &&
(hash2500line[58] != ':'))
{
fprintf(stderr, "error hashline wrong format %s\n", optarg);
exit(EXIT_FAILURE);
}
mode = 'P';
break;

case 'v':
vendorname = optarg;
mode = 'v';
break;

case 'e':
hexessidname = optarg;
mode = 'e';
break;

case 'x':
essidname = optarg;
mode = 'x';
break;

default:
usage(basename(argv[0]));
}
}

if(argc > 3)
{
fprintf(stderr, "only one argument allowed\n");
exit(EXIT_FAILURE);
}
uid = getuid();
pwd = getpwuid(uid);
if(pwd == NULL)
{
fprintf(stderr, "failed to get home dir\n");
exit(EXIT_FAILURE);
}
ret = chdir(pwd->pw_dir);
if(ret == -1)
{
fprintf(stderr, "failed to change dir\n");
}
if(stat(confdirname, &statinfo) == -1)
{
if(mkdir(confdirname,0755) == -1)
{
fprintf(stderr, "failed to create conf dir\n");
exit(EXIT_FAILURE);
}
}
if(mode == 'd')
{
downloadoui(ouinameuser);
return EXIT_SUCCESS;
}

if(stat(ouinamesystemwide, &statinfo) == 0)
{
ouiname = ouinamesystemwide;
}
if(stat(ouinameuser, &statinfo) == 0)
{
ouiname = ouinameuser;
}
if(ouiname == NULL)
{
fprintf(stderr, "failed read oui.txt\n"
"use download option -d to download it\n"
"or download file https://standards-oui.ieee.org/oui/oui.txt\n"
"and save it to ~/.hcxtools/oui.txt\n");
exit(EXIT_FAILURE);
}
if(stat(ouiname, &statinfo) < 0)
{
fprintf(stderr, "failed read oui.txt\n"
"use download option -d to download it\n"
"or download file https://standards-oui.ieee.org/oui/oui.txt\n"
"and save it to ~/.hcxtools/oui.txt\n");
exit(EXIT_FAILURE);
}
if(mode == 'm')
{
getoui(ouiname, oui);
}
else if(mode == 'p')
{
get16800info(ouiname, hash16800line);
}
else if(mode == 'P')
{
get2500info(ouiname, hash2500line);
 }

else if(mode == 'v')

{
getvendor(ouiname, vendorname);
}

else if(mode == 'e')

{
gethexessidinfo(hexessidname);
}

else if(mode == 'x')

{
getessidinfo(essidname);
}

else
{
usage(basename(argv[0]));
return EXIT_FAILURE;
}

return EXIT_SUCCESS;
}