/* IEEE 802.11 SoftMAC layer
 * Copyright (c) 2005 Andrea Merello <andreamrl@tiscali.it>
 *
 * Mostly extracted from the rtl8180-sa2400 driver for the 
 * in-kernel generic ieee802.11 stack.
 *
 * Some pieces of code might be stolen from ipw2100 driver
 * copyright of who own it's copyright ;-)
 *
 * PS wx handler mostly stolen from hostap, copyright who
 * own it's copyright ;-)
 *
 * released under the GPL
 */


#include "ieee80211.h"

/* FIXME: add A freqs */

const long ieee80211_wlan_frequencies[] = {  
	2412, 2417, 2422, 2427, 
	2432, 2437, 2442, 2447, 
	2452, 2457, 2462, 2467, 
	2472, 2484  
};


int ieee80211_wx_set_freq(struct ieee80211_device *ieee, struct iw_request_info *a,
			     union iwreq_data *wrqu, char *b)
{
	int ret;
	struct iw_freq *fwrq = & wrqu->freq;

	down(&ieee->wx_sem);
	
	if(ieee->iw_mode == IW_MODE_INFRA){ 
		ret = -EOPNOTSUPP;
		goto out;
	}

	/* if setting by freq convert to channel */
	if (fwrq->e == 1) {
		if ((fwrq->m >= (int) 2.412e8 &&
		     fwrq->m <= (int) 2.487e8)) {
			int f = fwrq->m / 100000;
			int c = 0;
			
			while ((c < 14) && (f != ieee80211_wlan_frequencies[c]))
				c++;
			
			/* hack to fall through */
			fwrq->e = 0;
			fwrq->m = c + 1;
		}
	}
	
	if (fwrq->e > 0 || fwrq->m > 14 || fwrq->m < 1 ){ 
		ret = -EOPNOTSUPP;
		goto out;
	
	}else { /* Set the channel */
		
		
		ieee->current_network.channel = fwrq->m;
		ieee->set_chan(ieee->dev, ieee->current_network.channel);
		
		if(ieee->iw_mode == IW_MODE_ADHOC || ieee->iw_mode == IW_MODE_MASTER)
			if(ieee->state == IEEE80211_LINKED){
			
			ieee80211_stop_send_beacons(ieee);
			ieee80211_start_send_beacons(ieee);
			}
	}

	ret = 0;
out:
	up(&ieee->wx_sem);
	return ret;
}


int ieee80211_wx_get_freq(struct ieee80211_device *ieee,
			     struct iw_request_info *a,
			     union iwreq_data *wrqu, char *b)
{
	struct iw_freq *fwrq = & wrqu->freq;

	if (ieee->current_network.channel == 0)
		return -1;
	
	fwrq->m = ieee->current_network.channel;
	fwrq->e = 0;
	
	return 0;
}

int ieee80211_wx_get_wap(struct ieee80211_device *ieee, 
			    struct iw_request_info *info, 
			    union iwreq_data *wrqu, char *extra)
{
	unsigned long flags;	
	
	wrqu->ap_addr.sa_family = ARPHRD_ETHER;
	
	if (ieee->iw_mode == IW_MODE_MONITOR)
		return -1;
	
	/* We want avoid to give to the user inconsistent infos*/
	spin_lock_irqsave(&ieee->lock, flags);
	
	if (ieee->state != IEEE80211_LINKED && 
		ieee->state != IEEE80211_LINKED_SCANNING &&
		ieee->wap_set == 0)
		
		memset(wrqu->ap_addr.sa_data, 0, ETH_ALEN);
	else
		memcpy(wrqu->ap_addr.sa_data, 
		       ieee->current_network.bssid, ETH_ALEN);
	
	spin_unlock_irqrestore(&ieee->lock, flags);
	
	return 0;
}


int ieee80211_wx_set_wap(struct ieee80211_device *ieee,
			 struct iw_request_info *info,
			 union iwreq_data *awrq,
			 char *extra)
{
	
	int ret = 0;
	u8 zero[] = {0,0,0,0,0,0};
	unsigned long flags;
	
	short ifup = ieee->proto_started;//dev->flags & IFF_UP;
	struct sockaddr *temp = (struct sockaddr *)awrq;

	ieee->sync_scan_hurryup = 1;

	down(&ieee->wx_sem);
	/* use ifconfig hw ether */
	if (ieee->iw_mode == IW_MODE_MASTER){
		ret = -1;
		goto out;
	}
	
	if (temp->sa_family != ARPHRD_ETHER){
		ret = -EINVAL;
		goto out;
	}
	
	if (ifup)
		ieee80211_stop_protocol(ieee);
	
	/* just to avoid to give inconsistent infos in the
	 * get wx method. not really needed otherwise 
	 */
	spin_lock_irqsave(&ieee->lock, flags);
	
	memcpy(ieee->current_network.bssid, temp->sa_data, ETH_ALEN); 
	ieee->wap_set = memcmp(temp->sa_data, zero,ETH_ALEN)!=0;
	
	spin_unlock_irqrestore(&ieee->lock, flags);
	
	if (ifup)
		ieee80211_start_protocol(ieee);
	
out:
	up(&ieee->wx_sem);
	return ret;
}
	
 int ieee80211_wx_get_essid(struct ieee80211_device *ieee, struct iw_request_info *a,union iwreq_data *wrqu,char *b)
{
	int len,ret = 0;
	unsigned long flags;
	
	if (ieee->iw_mode == IW_MODE_MONITOR)
		return -1;
	
	/* We want avoid to give to the user inconsistent infos*/	
	spin_lock_irqsave(&ieee->lock, flags);
	
	if (ieee->current_network.ssid[0] == '\0' ||
		ieee->current_network.ssid_len == 0){ 
		ret = -1;
		goto out;
	}
	
	if (ieee->state != IEEE80211_LINKED && 
		ieee->state != IEEE80211_LINKED_SCANNING &&
		ieee->ssid_set == 0){
		ret = -1;
		goto out;
	}
	len = ieee->current_network.ssid_len;
	wrqu->essid.length = len;
	strncpy(b,ieee->current_network.ssid,len);
	wrqu->essid.flags = 1;

out:
	spin_unlock_irqrestore(&ieee->lock, flags);
	
	return ret;
	
}

int ieee80211_wx_set_rate(struct ieee80211_device *ieee, 
			     struct iw_request_info *info,
			     union iwreq_data *wrqu, char *extra)
{

	u32 target_rate = wrqu->bitrate.value;
	
	ieee->rate = target_rate/100000;
	//FIXME: we might want to limit rate also in management protocols.
	return 0; 
}



int ieee80211_wx_get_rate(struct ieee80211_device *ieee, 
			     struct iw_request_info *info, 
			     union iwreq_data *wrqu, char *extra)
{
	
	wrqu->bitrate.value = ieee->rate * 100000;
	
	return 0;
}

int ieee80211_wx_set_mode(struct ieee80211_device *ieee, struct iw_request_info *a,
			     union iwreq_data *wrqu, char *b)
{
	
	ieee->sync_scan_hurryup = 1;
	
	down(&ieee->wx_sem);
	
	if (wrqu->mode == ieee->iw_mode)
		goto out;
	
	if (wrqu->mode == IW_MODE_MONITOR){
	
		ieee->dev->type = ARPHRD_IEEE80211;
	}else{
		ieee->dev->type = ARPHRD_ETHER;
	}
	
	if (!ieee->proto_started){
		ieee->iw_mode = wrqu->mode;
	}else{
		ieee80211_stop_protocol(ieee);
		ieee->iw_mode = wrqu->mode;
		ieee80211_start_protocol(ieee);
	}

out:
	up(&ieee->wx_sem);
	return 0;
}

#if LINUX_VERSION_CODE >= KERNEL_VERSION(2,6,20)
void ieee80211_wx_sync_scan_wq(struct work_struct *work)
{
	struct ieee80211_device *ieee = container_of(work, struct ieee80211_device, wx_sync_scan_wq);
#else
void ieee80211_wx_sync_scan_wq(struct ieee80211_device *ieee)
{
#endif
	short chan;

	chan = ieee->current_network.channel;
	
	netif_carrier_off(ieee->dev);
	
	if (ieee->data_hard_stop)
		ieee->data_hard_stop(ieee->dev);
	
	ieee80211_stop_send_beacons(ieee);
		
	ieee->state = IEEE80211_LINKED_SCANNING;
	ieee->link_change(ieee->dev);
	
	ieee80211_start_scan_syncro(ieee);
	
	ieee->set_chan(ieee->dev, chan);
	
	ieee->state = IEEE80211_LINKED;
	ieee->link_change(ieee->dev);
	
	if (ieee->data_hard_resume)
		ieee->data_hard_resume(ieee->dev);
	
	if(ieee->iw_mode == IW_MODE_ADHOC || ieee->iw_mode == IW_MODE_MASTER)
		ieee80211_start_send_beacons(ieee);
	
	netif_carrier_on(ieee->dev);
	
	up(&ieee->wx_sem);
	
}

int ieee80211_wx_set_scan(struct ieee80211_device *ieee, struct iw_request_info *a,
			     union iwreq_data *wrqu, char *b)
{
	int ret = 0;
	
	down(&ieee->wx_sem);
	
	if (ieee->iw_mode == IW_MODE_MONITOR || !(ieee->proto_started)){ 
		ret = -1;
		goto out;
	}
	
	if ( ieee->state == IEEE80211_LINKED){
#if LINUX_VERSION_CODE > KERNEL_VERSION(2,5,0)	
		queue_work(ieee->wq, &ieee->wx_sync_scan_wq);
#else
		schedule_task(&ieee->wx_sync_scan_wq);
#endif
		/* intentionally forget to up sem */
		return 0;
	}
		
out:
	up(&ieee->wx_sem);
	return ret;
}

int ieee80211_wx_set_essid(struct ieee80211_device *ieee, 
			      struct iw_request_info *a,
			      union iwreq_data *wrqu, char *extra)
{
	
	int ret=0,len;
	short proto_started;
	unsigned long flags;
	
	ieee->sync_scan_hurryup = 1;
	
	down(&ieee->wx_sem);
	
	proto_started = ieee->proto_started;
	
	if (wrqu->essid.length > IW_ESSID_MAX_SIZE){
		ret= -E2BIG;
		goto out;
	}
	
	if (ieee->iw_mode == IW_MODE_MONITOR){
		ret= -1;
		goto out;
	}
	
	if(proto_started)
		ieee80211_stop_protocol(ieee);
	
	/* this is just to be sure that the GET wx callback
	 * has consisten infos. not needed otherwise
	 */
	spin_lock_irqsave(&ieee->lock, flags);
	
	if (wrqu->essid.flags && wrqu->essid.length) {
		len = ((wrqu->essid.length-1) < IW_ESSID_MAX_SIZE) ? (wrqu->essid.length-1) : IW_ESSID_MAX_SIZE;
#if LINUX_VERSION_CODE <  KERNEL_VERSION(2,6,20)		
		strncpy(ieee->current_network.ssid, extra, len);
		ieee->current_network.ssid_len = len;
#else
		strncpy(ieee->current_network.ssid, extra, len+1);
		ieee->current_network.ssid_len = len+1;
#endif
		ieee->ssid_set = 1;
	}
	else{ 
		ieee->ssid_set = 0;
		ieee->current_network.ssid[0] = '\0';
		ieee->current_network.ssid_len = 0;
	}
	
	spin_unlock_irqrestore(&ieee->lock, flags);
	
	if (proto_started)
		ieee80211_start_protocol(ieee);
out:
	up(&ieee->wx_sem);
	return ret;
}

 int ieee80211_wx_get_mode(struct ieee80211_device *ieee, struct iw_request_info *a,
			     union iwreq_data *wrqu, char *b)
{

	wrqu->mode = ieee->iw_mode;
	return 0;
}

 int ieee80211_wx_set_rawtx(struct ieee80211_device *ieee, 
			       struct iw_request_info *info, 
			       union iwreq_data *wrqu, char *extra)
{
	
	int *parms = (int *)extra;
	int enable = (parms[0] > 0);
	short prev = ieee->raw_tx;

	down(&ieee->wx_sem);
	
	if(enable) 
		ieee->raw_tx = 1;
	else 
		ieee->raw_tx = 0;

	printk(KERN_INFO"raw TX is %s\n", 
	      ieee->raw_tx ? "enabled" : "disabled");

	if(ieee->iw_mode == IW_MODE_MONITOR)
	{
		if(prev == 0 && ieee->raw_tx){
			if (ieee->data_hard_resume)
				ieee->data_hard_resume(ieee->dev);
	
			netif_carrier_on(ieee->dev);	
		}
		
		if(prev && ieee->raw_tx == 1)
			netif_carrier_off(ieee->dev); 
	}
	
	up(&ieee->wx_sem);
	
	return 0;
}
 
int ieee80211_wx_get_name(struct ieee80211_device *ieee, 
			     struct iw_request_info *info, 
			     union iwreq_data *wrqu, char *extra)
{
	strcpy(wrqu->name, "802.11");
	if(ieee->modulation & IEEE80211_CCK_MODULATION){
		strcat(wrqu->name, "b");
		if(ieee->modulation & IEEE80211_OFDM_MODULATION)
			strcat(wrqu->name, "/g");
	}else if(ieee->modulation & IEEE80211_OFDM_MODULATION)
		strcat(wrqu->name, "g");
	
	if((ieee->state == IEEE80211_LINKED) || 
		(ieee->state == IEEE80211_LINKED_SCANNING))
		strcat(wrqu->name," linked");
	else if(ieee->state != IEEE80211_NOLINK)
		strcat(wrqu->name," link..");
	
	
	return 0;
}


/* this is mostly stolen from hostap */
int ieee80211_wx_set_power(struct ieee80211_device *ieee,
				 struct iw_request_info *info,
				 union iwreq_data *wrqu, char *extra)
{
	int ret = 0;

	if(
		(!ieee->sta_wake_up) ||
		(!ieee->ps_request_tx_ack) ||
		(!ieee->enter_sleep_state) ||
		(!ieee->ps_is_queue_empty)){
		
		printk("ERROR. PS mode is tryied to be use but\
driver missed a callback\n\n");	
	
		return -1;
	}
	
	down(&ieee->wx_sem);
	
	if (wrqu->power.disabled){
		ieee->ps = IEEE80211_PS_DISABLED;
		
		goto exit;
	}
	switch (wrqu->power.flags & IW_POWER_MODE) {
	case IW_POWER_UNICAST_R:
		ieee->ps = IEEE80211_PS_UNICAST;
		
		break;
	case IW_POWER_ALL_R:
		ieee->ps = IEEE80211_PS_UNICAST | IEEE80211_PS_MBCAST;	
		break;
		
	case IW_POWER_ON:
		ieee->ps = IEEE80211_PS_DISABLED;
		break;
		
	default:
		ret = -EINVAL;
		goto exit;
	}

	if (wrqu->power.flags & IW_POWER_TIMEOUT) {
				      
		ieee->ps_timeout = wrqu->power.value / 1000;
		printk("Timeout %d\n",ieee->ps_timeout);
	}
	
	if (wrqu->power.flags & IW_POWER_PERIOD) {
		
		ret = -EOPNOTSUPP;
		goto exit;
		//wrq->value / 1024;
		
	}
exit:
	up(&ieee->wx_sem);
	return ret;

}

/* this is stolen from hostap */
int ieee80211_wx_get_power(struct ieee80211_device *ieee,
				 struct iw_request_info *info,
				 union iwreq_data *wrqu, char *extra)
{
	int ret =0;
	
	down(&ieee->wx_sem);
	
	if(ieee->ps == IEEE80211_PS_DISABLED){	
		wrqu->power.disabled = 1;
		goto exit;
	}

	wrqu->power.disabled = 0;

//	if ((wrqu->power.flags & IW_POWER_TYPE) == IW_POWER_TIMEOUT) {
		wrqu->power.flags = IW_POWER_TIMEOUT;
		wrqu->power.value = ieee->ps_timeout * 1000;
//	} else {
//		ret = -EOPNOTSUPP;
//		goto exit;
		//wrqu->power.flags = IW_POWER_PERIOD;
		//wrqu->power.value = ieee->current_network.dtim_period *
		//	ieee->current_network.beacon_interval * 1024;
//	}


	if (ieee->ps & IEEE80211_PS_MBCAST)
		wrqu->power.flags |= IW_POWER_ALL_R;
	else
		wrqu->power.flags |= IW_POWER_UNICAST_R;

exit:
	up(&ieee->wx_sem);
	return ret;

}
#if (LINUX_VERSION_CODE > KERNEL_VERSION(2,5,0))
EXPORT_SYMBOL(ieee80211_wx_get_essid);
EXPORT_SYMBOL(ieee80211_wx_set_essid);
EXPORT_SYMBOL(ieee80211_wx_set_rate);
EXPORT_SYMBOL(ieee80211_wx_get_rate);
EXPORT_SYMBOL(ieee80211_wx_set_wap);
EXPORT_SYMBOL(ieee80211_wx_get_wap);
EXPORT_SYMBOL(ieee80211_wx_set_mode);
EXPORT_SYMBOL(ieee80211_wx_get_mode);
EXPORT_SYMBOL(ieee80211_wx_set_scan);
EXPORT_SYMBOL(ieee80211_wx_get_freq);
EXPORT_SYMBOL(ieee80211_wx_set_freq);
EXPORT_SYMBOL(ieee80211_wx_set_rawtx);
EXPORT_SYMBOL(ieee80211_wx_get_name);
EXPORT_SYMBOL(ieee80211_wx_set_power);
EXPORT_SYMBOL(ieee80211_wx_get_power);
EXPORT_SYMBOL(ieee80211_wlan_frequencies);
#else
EXPORT_SYMBOL_NOVERS(ieee80211_wx_get_essid);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_essid);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_rate);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_get_rate);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_wap);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_get_wap);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_mode);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_get_mode);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_scan);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_get_freq);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_freq);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_rawtx);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_get_name);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_set_power);
EXPORT_SYMBOL_NOVERS(ieee80211_wx_get_power);
EXPORT_SYMBOL_NOVERS(ieee80211_wlan_frequencies);
#endif
