#include <netlink/genl/genl.h>

#include "wifi_command.h"

static int ackHandler(struct nl_msg *msg, void *arg)
{
	*((int *)arg) = 0;

	return NL_STOP;
}

static int finishHandler(struct nl_msg *msg, void *arg)
{
	*((int *)arg) = 0;

	return NL_SKIP;
}

static int errorHandler(struct sockaddr_nl *nla, struct nlmsgerr *err, void *arg)
{
	*((int *)arg) = 0;

	return NL_SKIP;
}

static int msgHandler(struct nl_msg *msg, void *arg)
{
	struct nlattr *attr[NL80211_ATTR_MAX + 1];
	WifiCommand *cmd = (WifiCommand *)arg;
	struct genlmsghdr *gnlh = (struct genlmsghdr *)nlmsg_data(nlmsg_hdr(msg));

	nla_parse(attr, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0),
			genlmsg_attrlen(gnlh, 0), NULL);

	cmd->parser(attr);

	return NL_SKIP;
}

wifi_error sendMsg(struct nl_sock *sk, struct nl_msg *msg, void *arg)
{
	int err;
	struct nl_cb *cb;

       	cb = nl_cb_alloc(NL_CB_DEFAULT);
	if (cb == NULL) {
		ALOGE("%s: alloc cb failed");
		return WIFI_ERROR_OUT_OF_MEMORY;
	}

	nl_cb_err(cb, NL_CB_CUSTOM, errorHandler, &err);
	nl_cb_set(cb, NL_CB_FINISH, NL_CB_CUSTOM, finishHandler, &err);
	nl_cb_set(cb, NL_CB_ACK, NL_CB_CUSTOM, ackHandler, &err);
	nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, msgHandler, arg);

	err = nl_send_auto_complete(sk, msg);
	while (err > 0) {
		int res = nl_recvmsgs(sk, cb);
		if (res < 0)
			ALOGE("%s: recv msg failed: %d", res);
	}

	nl_cb_put(cb);

	return err ? WIFI_ERROR_UNKNOWN : WIFI_SUCCESS;
}

wifi_error WifiCommand::send()
{
	return sendMsg(sock, nlmsg(), (void *)this);
}

WifiCommand::~WifiCommand()
{
	nlmsg_free(msg);
}

WifiCommand::WifiCommand(struct nl_sock *sk, int family_id, int flags, int nl80211_cmd)
{
	sock = sk;

	msg = nlmsg_alloc();
	if (msg)
		genlmsg_put(msg, 0, 0, family_id, 0, flags, nl80211_cmd, 0);
	else
		ALOGE("nlmsg alloc failed, nl80211 cmd: %d", nl80211_cmd);
}