1 // SPDX-License-Identifier: GPL-2.0
2 /* Copyright(c) 2009-2012  Realtek Corporation.*/
3 
4 #include "wifi.h"
5 #include "base.h"
6 #include "ps.h"
7 #include <linux/export.h>
8 #include "btcoexist/rtl_btc.h"
9 
10 bool rtl_ps_enable_nic(struct ieee80211_hw *hw)
11 {
12 	struct rtl_priv *rtlpriv = rtl_priv(hw);
13 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
14 	struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
15 	struct rtl_mac *rtlmac = rtl_mac(rtl_priv(hw));
16 
17 	/*<1> reset trx ring */
18 	if (rtlhal->interface == INTF_PCI)
19 		rtlpriv->intf_ops->reset_trx_ring(hw);
20 
21 	if (is_hal_stop(rtlhal))
22 		RT_TRACE(rtlpriv, COMP_ERR, DBG_WARNING,
23 			 "Driver is already down!\n");
24 
25 	/*<2> Enable Adapter */
26 	if (rtlpriv->cfg->ops->hw_init(hw))
27 		return false;
28 	rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_RETRY_LIMIT,
29 			&rtlmac->retry_long);
30 	RT_CLEAR_PS_LEVEL(ppsc, RT_RF_OFF_LEVL_HALT_NIC);
31 
32 	rtlpriv->cfg->ops->switch_channel(hw);
33 	rtlpriv->cfg->ops->set_channel_access(hw);
34 	rtlpriv->cfg->ops->set_bw_mode(hw,
35 			cfg80211_get_chandef_type(&hw->conf.chandef));
36 
37 	/*<3> Enable Interrupt */
38 	rtlpriv->cfg->ops->enable_interrupt(hw);
39 
40 	/*<enable timer> */
41 	rtl_watch_dog_timer_callback(&rtlpriv->works.watchdog_timer);
42 
43 	return true;
44 }
45 EXPORT_SYMBOL(rtl_ps_enable_nic);
46 
47 bool rtl_ps_disable_nic(struct ieee80211_hw *hw)
48 {
49 	struct rtl_priv *rtlpriv = rtl_priv(hw);
50 
51 	/*<1> Stop all timer */
52 	rtl_deinit_deferred_work(hw, true);
53 
54 	/*<2> Disable Interrupt */
55 	rtlpriv->cfg->ops->disable_interrupt(hw);
56 	tasklet_kill(&rtlpriv->works.irq_tasklet);
57 
58 	/*<3> Disable Adapter */
59 	rtlpriv->cfg->ops->hw_disable(hw);
60 
61 	return true;
62 }
63 EXPORT_SYMBOL(rtl_ps_disable_nic);
64 
65 static bool rtl_ps_set_rf_state(struct ieee80211_hw *hw,
66 				enum rf_pwrstate state_toset,
67 				u32 changesource)
68 {
69 	struct rtl_priv *rtlpriv = rtl_priv(hw);
70 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
71 	enum rf_pwrstate rtstate;
72 	bool actionallowed = false;
73 	u16 rfwait_cnt = 0;
74 
75 	/*Only one thread can change
76 	 *the RF state at one time, and others
77 	 *should wait to be executed.
78 	 */
79 	while (true) {
80 		spin_lock(&rtlpriv->locks.rf_ps_lock);
81 		if (ppsc->rfchange_inprogress) {
82 			spin_unlock(&rtlpriv->locks.rf_ps_lock);
83 
84 			RT_TRACE(rtlpriv, COMP_ERR, DBG_WARNING,
85 				 "RF Change in progress! Wait to set..state_toset(%d).\n",
86 				  state_toset);
87 
88 			/* Set RF after the previous action is done.  */
89 			while (ppsc->rfchange_inprogress) {
90 				rfwait_cnt++;
91 				mdelay(1);
92 				/*Wait too long, return false to avoid
93 				 *to be stuck here.
94 				 */
95 				if (rfwait_cnt > 100)
96 					return false;
97 			}
98 		} else {
99 			ppsc->rfchange_inprogress = true;
100 			spin_unlock(&rtlpriv->locks.rf_ps_lock);
101 			break;
102 		}
103 	}
104 
105 	rtstate = ppsc->rfpwr_state;
106 
107 	switch (state_toset) {
108 	case ERFON:
109 		ppsc->rfoff_reason &= (~changesource);
110 
111 		if ((changesource == RF_CHANGE_BY_HW) &&
112 		    (ppsc->hwradiooff)) {
113 			ppsc->hwradiooff = false;
114 		}
115 
116 		if (!ppsc->rfoff_reason) {
117 			ppsc->rfoff_reason = 0;
118 			actionallowed = true;
119 		}
120 
121 		break;
122 
123 	case ERFOFF:
124 
125 		if ((changesource == RF_CHANGE_BY_HW) && !ppsc->hwradiooff) {
126 			ppsc->hwradiooff = true;
127 		}
128 
129 		ppsc->rfoff_reason |= changesource;
130 		actionallowed = true;
131 		break;
132 
133 	case ERFSLEEP:
134 		ppsc->rfoff_reason |= changesource;
135 		actionallowed = true;
136 		break;
137 
138 	default:
139 		pr_err("switch case %#x not processed\n", state_toset);
140 		break;
141 	}
142 
143 	if (actionallowed)
144 		rtlpriv->cfg->ops->set_rf_power_state(hw, state_toset);
145 
146 	spin_lock(&rtlpriv->locks.rf_ps_lock);
147 	ppsc->rfchange_inprogress = false;
148 	spin_unlock(&rtlpriv->locks.rf_ps_lock);
149 
150 	return actionallowed;
151 }
152 
153 static void _rtl_ps_inactive_ps(struct ieee80211_hw *hw)
154 {
155 	struct rtl_priv *rtlpriv = rtl_priv(hw);
156 	struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
157 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
158 
159 	ppsc->swrf_processing = true;
160 
161 	if (ppsc->inactive_pwrstate == ERFON &&
162 	    rtlhal->interface == INTF_PCI) {
163 		if ((ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM) &&
164 		    RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM) &&
165 		    rtlhal->interface == INTF_PCI) {
166 			rtlpriv->intf_ops->disable_aspm(hw);
167 			RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
168 		}
169 	}
170 
171 	rtl_ps_set_rf_state(hw, ppsc->inactive_pwrstate,
172 			    RF_CHANGE_BY_IPS);
173 
174 	if (ppsc->inactive_pwrstate == ERFOFF &&
175 	    rtlhal->interface == INTF_PCI) {
176 		if (ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM &&
177 		    !RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
178 			rtlpriv->intf_ops->enable_aspm(hw);
179 			RT_SET_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
180 		}
181 	}
182 
183 	ppsc->swrf_processing = false;
184 }
185 
186 void rtl_ips_nic_off_wq_callback(void *data)
187 {
188 	struct rtl_works *rtlworks =
189 	    container_of_dwork_rtl(data, struct rtl_works, ips_nic_off_wq);
190 	struct ieee80211_hw *hw = rtlworks->hw;
191 	struct rtl_priv *rtlpriv = rtl_priv(hw);
192 	struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
193 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
194 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
195 	enum rf_pwrstate rtstate;
196 
197 	if (mac->opmode != NL80211_IFTYPE_STATION) {
198 		RT_TRACE(rtlpriv, COMP_ERR, DBG_WARNING,
199 			 "not station return\n");
200 		return;
201 	}
202 
203 	if (mac->p2p_in_use)
204 		return;
205 
206 	if (mac->link_state > MAC80211_NOLINK)
207 		return;
208 
209 	if (is_hal_stop(rtlhal))
210 		return;
211 
212 	if (rtlpriv->sec.being_setkey)
213 		return;
214 
215 	if (rtlpriv->cfg->ops->bt_coex_off_before_lps)
216 		rtlpriv->cfg->ops->bt_coex_off_before_lps(hw);
217 
218 	if (ppsc->inactiveps) {
219 		rtstate = ppsc->rfpwr_state;
220 
221 		/*
222 		 *Do not enter IPS in the following conditions:
223 		 *(1) RF is already OFF or Sleep
224 		 *(2) swrf_processing (indicates the IPS is still under going)
225 		 *(3) Connectted (only disconnected can trigger IPS)
226 		 *(4) IBSS (send Beacon)
227 		 *(5) AP mode (send Beacon)
228 		 *(6) monitor mode (rcv packet)
229 		 */
230 
231 		if (rtstate == ERFON &&
232 		    !ppsc->swrf_processing &&
233 		    (mac->link_state == MAC80211_NOLINK) &&
234 		    !mac->act_scanning) {
235 			RT_TRACE(rtlpriv, COMP_RF, DBG_TRACE,
236 				 "IPSEnter(): Turn off RF\n");
237 
238 			ppsc->inactive_pwrstate = ERFOFF;
239 			ppsc->in_powersavemode = true;
240 
241 			/* call before RF off */
242 			if (rtlpriv->cfg->ops->get_btc_status())
243 				rtlpriv->btcoexist.btc_ops->btc_ips_notify(rtlpriv,
244 									ppsc->inactive_pwrstate);
245 
246 			/*rtl_pci_reset_trx_ring(hw); */
247 			_rtl_ps_inactive_ps(hw);
248 		}
249 	}
250 }
251 
252 void rtl_ips_nic_off(struct ieee80211_hw *hw)
253 {
254 	struct rtl_priv *rtlpriv = rtl_priv(hw);
255 
256 	/* because when link with ap, mac80211 will ask us
257 	 * to disable nic quickly after scan before linking,
258 	 * this will cause link failed, so we delay 100ms here
259 	 */
260 	queue_delayed_work(rtlpriv->works.rtl_wq,
261 			   &rtlpriv->works.ips_nic_off_wq, MSECS(100));
262 }
263 
264 /* NOTICE: any opmode should exc nic_on, or disable without
265  * nic_on may something wrong, like adhoc TP
266  */
267 void rtl_ips_nic_on(struct ieee80211_hw *hw)
268 {
269 	struct rtl_priv *rtlpriv = rtl_priv(hw);
270 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
271 	enum rf_pwrstate rtstate;
272 
273 	cancel_delayed_work_sync(&rtlpriv->works.ips_nic_off_wq);
274 
275 	mutex_lock(&rtlpriv->locks.ips_mutex);
276 	if (ppsc->inactiveps) {
277 		rtstate = ppsc->rfpwr_state;
278 
279 		if (rtstate != ERFON &&
280 		    !ppsc->swrf_processing &&
281 		    ppsc->rfoff_reason <= RF_CHANGE_BY_IPS) {
282 
283 			ppsc->inactive_pwrstate = ERFON;
284 			ppsc->in_powersavemode = false;
285 			_rtl_ps_inactive_ps(hw);
286 			/* call after RF on */
287 			if (rtlpriv->cfg->ops->get_btc_status())
288 				rtlpriv->btcoexist.btc_ops->btc_ips_notify(rtlpriv,
289 									ppsc->inactive_pwrstate);
290 		}
291 	}
292 	mutex_unlock(&rtlpriv->locks.ips_mutex);
293 }
294 EXPORT_SYMBOL_GPL(rtl_ips_nic_on);
295 
296 /*for FW LPS*/
297 
298 /*
299  *Determine if we can set Fw into PS mode
300  *in current condition.Return TRUE if it
301  *can enter PS mode.
302  */
303 static bool rtl_get_fwlps_doze(struct ieee80211_hw *hw)
304 {
305 	struct rtl_priv *rtlpriv = rtl_priv(hw);
306 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
307 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
308 	u32 ps_timediff;
309 
310 	ps_timediff = jiffies_to_msecs(jiffies -
311 				       ppsc->last_delaylps_stamp_jiffies);
312 
313 	if (ps_timediff < 2000) {
314 		RT_TRACE(rtlpriv, COMP_POWER, DBG_LOUD,
315 			 "Delay enter Fw LPS for DHCP, ARP, or EAPOL exchanging state\n");
316 		return false;
317 	}
318 
319 	if (mac->link_state != MAC80211_LINKED)
320 		return false;
321 
322 	if (mac->opmode == NL80211_IFTYPE_ADHOC)
323 		return false;
324 
325 	return true;
326 }
327 
328 /* Change current and default preamble mode.*/
329 void rtl_lps_set_psmode(struct ieee80211_hw *hw, u8 rt_psmode)
330 {
331 	struct rtl_priv *rtlpriv = rtl_priv(hw);
332 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
333 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
334 	bool enter_fwlps;
335 
336 	if (mac->opmode == NL80211_IFTYPE_ADHOC)
337 		return;
338 
339 	if (mac->link_state != MAC80211_LINKED)
340 		return;
341 
342 	if (ppsc->dot11_psmode == rt_psmode && rt_psmode == EACTIVE)
343 		return;
344 
345 	/* Update power save mode configured. */
346 	ppsc->dot11_psmode = rt_psmode;
347 
348 	/*
349 	 *<FW control LPS>
350 	 *1. Enter PS mode
351 	 *   Set RPWM to Fw to turn RF off and send H2C fw_pwrmode
352 	 *   cmd to set Fw into PS mode.
353 	 *2. Leave PS mode
354 	 *   Send H2C fw_pwrmode cmd to Fw to set Fw into Active
355 	 *   mode and set RPWM to turn RF on.
356 	 */
357 
358 	if ((ppsc->fwctrl_lps) && ppsc->report_linked) {
359 		if (ppsc->dot11_psmode == EACTIVE) {
360 			RT_TRACE(rtlpriv, COMP_RF, DBG_DMESG,
361 				 "FW LPS leave ps_mode:%x\n",
362 				  FW_PS_ACTIVE_MODE);
363 			enter_fwlps = false;
364 			ppsc->pwr_mode = FW_PS_ACTIVE_MODE;
365 			ppsc->smart_ps = 0;
366 			rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_FW_LPS_ACTION,
367 						      (u8 *)(&enter_fwlps));
368 			if (ppsc->p2p_ps_info.opp_ps)
369 				rtl_p2p_ps_cmd(hw , P2P_PS_ENABLE);
370 
371 			if (rtlpriv->cfg->ops->get_btc_status())
372 				rtlpriv->btcoexist.btc_ops->btc_lps_notify(rtlpriv, rt_psmode);
373 		} else {
374 			if (rtl_get_fwlps_doze(hw)) {
375 				RT_TRACE(rtlpriv, COMP_RF, DBG_DMESG,
376 					 "FW LPS enter ps_mode:%x\n",
377 					 ppsc->fwctrl_psmode);
378 				if (rtlpriv->cfg->ops->get_btc_status())
379 					rtlpriv->btcoexist.btc_ops->btc_lps_notify(rtlpriv, rt_psmode);
380 				enter_fwlps = true;
381 				ppsc->pwr_mode = ppsc->fwctrl_psmode;
382 				ppsc->smart_ps = 2;
383 				rtlpriv->cfg->ops->set_hw_reg(hw,
384 							HW_VAR_FW_LPS_ACTION,
385 							(u8 *)(&enter_fwlps));
386 
387 			} else {
388 				/* Reset the power save related parameters. */
389 				ppsc->dot11_psmode = EACTIVE;
390 			}
391 		}
392 	}
393 }
394 
395 /* Interrupt safe routine to enter the leisure power save mode.*/
396 static void rtl_lps_enter_core(struct ieee80211_hw *hw)
397 {
398 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
399 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
400 	struct rtl_priv *rtlpriv = rtl_priv(hw);
401 
402 	if (!ppsc->fwctrl_lps)
403 		return;
404 
405 	if (rtlpriv->sec.being_setkey)
406 		return;
407 
408 	if (rtlpriv->link_info.busytraffic)
409 		return;
410 
411 	/*sleep after linked 10s, to let DHCP and 4-way handshake ok enough!! */
412 	if (mac->cnt_after_linked < 5)
413 		return;
414 
415 	if (mac->opmode == NL80211_IFTYPE_ADHOC)
416 		return;
417 
418 	if (mac->link_state != MAC80211_LINKED)
419 		return;
420 
421 	mutex_lock(&rtlpriv->locks.lps_mutex);
422 
423 	/* Don't need to check (ppsc->dot11_psmode == EACTIVE), because
424 	 * bt_ccoexist may ask to enter lps.
425 	 * In normal case, this constraint move to rtl_lps_set_psmode().
426 	 */
427 	RT_TRACE(rtlpriv, COMP_POWER, DBG_LOUD,
428 		 "Enter 802.11 power save mode...\n");
429 	rtl_lps_set_psmode(hw, EAUTOPS);
430 
431 	mutex_unlock(&rtlpriv->locks.lps_mutex);
432 }
433 
434 /* Interrupt safe routine to leave the leisure power save mode.*/
435 static void rtl_lps_leave_core(struct ieee80211_hw *hw)
436 {
437 	struct rtl_priv *rtlpriv = rtl_priv(hw);
438 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
439 	struct rtl_hal *rtlhal = rtl_hal(rtl_priv(hw));
440 
441 	mutex_lock(&rtlpriv->locks.lps_mutex);
442 
443 	if (ppsc->fwctrl_lps) {
444 		if (ppsc->dot11_psmode != EACTIVE) {
445 
446 			/*FIX ME */
447 			/*rtlpriv->cfg->ops->enable_interrupt(hw); */
448 
449 			if (ppsc->reg_rfps_level & RT_RF_LPS_LEVEL_ASPM &&
450 			    RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM) &&
451 			    rtlhal->interface == INTF_PCI) {
452 				rtlpriv->intf_ops->disable_aspm(hw);
453 				RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
454 			}
455 
456 			RT_TRACE(rtlpriv, COMP_POWER, DBG_LOUD,
457 				 "Busy Traffic,Leave 802.11 power save..\n");
458 
459 			rtl_lps_set_psmode(hw, EACTIVE);
460 		}
461 	}
462 	mutex_unlock(&rtlpriv->locks.lps_mutex);
463 }
464 
465 /* For sw LPS*/
466 void rtl_swlps_beacon(struct ieee80211_hw *hw, void *data, unsigned int len)
467 {
468 	struct rtl_priv *rtlpriv = rtl_priv(hw);
469 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
470 	struct ieee80211_hdr *hdr = data;
471 	struct ieee80211_tim_ie *tim_ie;
472 	u8 *tim;
473 	u8 tim_len;
474 	bool u_buffed;
475 	bool m_buffed;
476 
477 	if (mac->opmode != NL80211_IFTYPE_STATION)
478 		return;
479 
480 	if (!rtlpriv->psc.swctrl_lps)
481 		return;
482 
483 	if (rtlpriv->mac80211.link_state != MAC80211_LINKED)
484 		return;
485 
486 	if (!rtlpriv->psc.sw_ps_enabled)
487 		return;
488 
489 	if (rtlpriv->psc.fwctrl_lps)
490 		return;
491 
492 	if (likely(!(hw->conf.flags & IEEE80211_CONF_PS)))
493 		return;
494 
495 	/* check if this really is a beacon */
496 	if (!ieee80211_is_beacon(hdr->frame_control))
497 		return;
498 
499 	/* min. beacon length + FCS_LEN */
500 	if (len <= 40 + FCS_LEN)
501 		return;
502 
503 	/* and only beacons from the associated BSSID, please */
504 	if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
505 		return;
506 
507 	rtlpriv->psc.last_beacon = jiffies;
508 
509 	tim = rtl_find_ie(data, len - FCS_LEN, WLAN_EID_TIM);
510 	if (!tim)
511 		return;
512 
513 	if (tim[1] < sizeof(*tim_ie))
514 		return;
515 
516 	tim_len = tim[1];
517 	tim_ie = (struct ieee80211_tim_ie *) &tim[2];
518 
519 	if (!WARN_ON_ONCE(!hw->conf.ps_dtim_period))
520 		rtlpriv->psc.dtim_counter = tim_ie->dtim_count;
521 
522 	/* Check whenever the PHY can be turned off again. */
523 
524 	/* 1. What about buffered unicast traffic for our AID? */
525 	u_buffed = ieee80211_check_tim(tim_ie, tim_len,
526 				       rtlpriv->mac80211.assoc_id);
527 
528 	/* 2. Maybe the AP wants to send multicast/broadcast data? */
529 	m_buffed = tim_ie->bitmap_ctrl & 0x01;
530 	rtlpriv->psc.multi_buffered = m_buffed;
531 
532 	/* unicast will process by mac80211 through
533 	 * set ~IEEE80211_CONF_PS, So we just check
534 	 * multicast frames here */
535 	if (!m_buffed) {
536 		/* back to low-power land. and delay is
537 		 * prevent null power save frame tx fail */
538 		queue_delayed_work(rtlpriv->works.rtl_wq,
539 				   &rtlpriv->works.ps_work, MSECS(5));
540 	} else {
541 		RT_TRACE(rtlpriv, COMP_POWER, DBG_DMESG,
542 			 "u_bufferd: %x, m_buffered: %x\n", u_buffed, m_buffed);
543 	}
544 }
545 EXPORT_SYMBOL_GPL(rtl_swlps_beacon);
546 
547 void rtl_swlps_rf_awake(struct ieee80211_hw *hw)
548 {
549 	struct rtl_priv *rtlpriv = rtl_priv(hw);
550 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
551 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
552 
553 	if (!rtlpriv->psc.swctrl_lps)
554 		return;
555 	if (mac->link_state != MAC80211_LINKED)
556 		return;
557 
558 	if (ppsc->reg_rfps_level & RT_RF_LPS_LEVEL_ASPM &&
559 	    RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
560 		rtlpriv->intf_ops->disable_aspm(hw);
561 		RT_CLEAR_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
562 	}
563 
564 	mutex_lock(&rtlpriv->locks.lps_mutex);
565 	rtl_ps_set_rf_state(hw, ERFON, RF_CHANGE_BY_PS);
566 	mutex_unlock(&rtlpriv->locks.lps_mutex);
567 }
568 
569 void rtl_swlps_rfon_wq_callback(void *data)
570 {
571 	struct rtl_works *rtlworks =
572 	    container_of_dwork_rtl(data, struct rtl_works, ps_rfon_wq);
573 	struct ieee80211_hw *hw = rtlworks->hw;
574 
575 	rtl_swlps_rf_awake(hw);
576 }
577 
578 void rtl_swlps_rf_sleep(struct ieee80211_hw *hw)
579 {
580 	struct rtl_priv *rtlpriv = rtl_priv(hw);
581 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
582 	struct rtl_ps_ctl *ppsc = rtl_psc(rtl_priv(hw));
583 	u8 sleep_intv;
584 
585 	if (!rtlpriv->psc.sw_ps_enabled)
586 		return;
587 
588 	if ((rtlpriv->sec.being_setkey) ||
589 	    (mac->opmode == NL80211_IFTYPE_ADHOC))
590 		return;
591 
592 	/*sleep after linked 10s, to let DHCP and 4-way handshake ok enough!! */
593 	if ((mac->link_state != MAC80211_LINKED) || (mac->cnt_after_linked < 5))
594 		return;
595 
596 	if (rtlpriv->link_info.busytraffic)
597 		return;
598 
599 	spin_lock(&rtlpriv->locks.rf_ps_lock);
600 	if (rtlpriv->psc.rfchange_inprogress) {
601 		spin_unlock(&rtlpriv->locks.rf_ps_lock);
602 		return;
603 	}
604 	spin_unlock(&rtlpriv->locks.rf_ps_lock);
605 
606 	mutex_lock(&rtlpriv->locks.lps_mutex);
607 	rtl_ps_set_rf_state(hw, ERFSLEEP, RF_CHANGE_BY_PS);
608 	mutex_unlock(&rtlpriv->locks.lps_mutex);
609 
610 	if (ppsc->reg_rfps_level & RT_RF_OFF_LEVL_ASPM &&
611 	    !RT_IN_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM)) {
612 		rtlpriv->intf_ops->enable_aspm(hw);
613 		RT_SET_PS_LEVEL(ppsc, RT_PS_LEVEL_ASPM);
614 	}
615 
616 	/* here is power save alg, when this beacon is DTIM
617 	 * we will set sleep time to dtim_period * n;
618 	 * when this beacon is not DTIM, we will set sleep
619 	 * time to sleep_intv = rtlpriv->psc.dtim_counter or
620 	 * MAX_SW_LPS_SLEEP_INTV(default set to 5) */
621 
622 	if (rtlpriv->psc.dtim_counter == 0) {
623 		if (hw->conf.ps_dtim_period == 1)
624 			sleep_intv = hw->conf.ps_dtim_period * 2;
625 		else
626 			sleep_intv = hw->conf.ps_dtim_period;
627 	} else {
628 		sleep_intv = rtlpriv->psc.dtim_counter;
629 	}
630 
631 	if (sleep_intv > MAX_SW_LPS_SLEEP_INTV)
632 		sleep_intv = MAX_SW_LPS_SLEEP_INTV;
633 
634 	/* this print should always be dtim_conter = 0 &
635 	 * sleep  = dtim_period, that meaons, we should
636 	 * awake before every dtim */
637 	RT_TRACE(rtlpriv, COMP_POWER, DBG_DMESG,
638 		 "dtim_counter:%x will sleep :%d beacon_intv\n",
639 		  rtlpriv->psc.dtim_counter, sleep_intv);
640 
641 	/* we tested that 40ms is enough for sw & hw sw delay */
642 	queue_delayed_work(rtlpriv->works.rtl_wq, &rtlpriv->works.ps_rfon_wq,
643 			MSECS(sleep_intv * mac->vif->bss_conf.beacon_int - 40));
644 }
645 
646 void rtl_lps_change_work_callback(struct work_struct *work)
647 {
648 	struct rtl_works *rtlworks =
649 	    container_of(work, struct rtl_works, lps_change_work);
650 	struct ieee80211_hw *hw = rtlworks->hw;
651 	struct rtl_priv *rtlpriv = rtl_priv(hw);
652 
653 	if (rtlpriv->enter_ps)
654 		rtl_lps_enter_core(hw);
655 	else
656 		rtl_lps_leave_core(hw);
657 }
658 EXPORT_SYMBOL_GPL(rtl_lps_change_work_callback);
659 
660 void rtl_lps_enter(struct ieee80211_hw *hw)
661 {
662 	struct rtl_priv *rtlpriv = rtl_priv(hw);
663 
664 	if (!in_interrupt())
665 		return rtl_lps_enter_core(hw);
666 	rtlpriv->enter_ps = true;
667 	schedule_work(&rtlpriv->works.lps_change_work);
668 }
669 EXPORT_SYMBOL_GPL(rtl_lps_enter);
670 
671 void rtl_lps_leave(struct ieee80211_hw *hw)
672 {
673 	struct rtl_priv *rtlpriv = rtl_priv(hw);
674 
675 	if (!in_interrupt())
676 		return rtl_lps_leave_core(hw);
677 	rtlpriv->enter_ps = false;
678 	schedule_work(&rtlpriv->works.lps_change_work);
679 }
680 EXPORT_SYMBOL_GPL(rtl_lps_leave);
681 
682 void rtl_swlps_wq_callback(void *data)
683 {
684 	struct rtl_works *rtlworks = container_of_dwork_rtl(data,
685 				     struct rtl_works,
686 				     ps_work);
687 	struct ieee80211_hw *hw = rtlworks->hw;
688 	struct rtl_priv *rtlpriv = rtl_priv(hw);
689 	bool ps = false;
690 
691 	ps = (hw->conf.flags & IEEE80211_CONF_PS);
692 
693 	/* we can sleep after ps null send ok */
694 	if (rtlpriv->psc.state_inap) {
695 		rtl_swlps_rf_sleep(hw);
696 
697 		if (rtlpriv->psc.state && !ps) {
698 			rtlpriv->psc.sleep_ms = jiffies_to_msecs(jiffies -
699 						 rtlpriv->psc.last_action);
700 		}
701 
702 		if (ps)
703 			rtlpriv->psc.last_slept = jiffies;
704 
705 		rtlpriv->psc.last_action = jiffies;
706 		rtlpriv->psc.state = ps;
707 	}
708 }
709 
710 static void rtl_p2p_noa_ie(struct ieee80211_hw *hw, void *data,
711 			   unsigned int len)
712 {
713 	struct rtl_priv *rtlpriv = rtl_priv(hw);
714 	struct ieee80211_mgmt *mgmt = data;
715 	struct rtl_p2p_ps_info *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
716 	u8 *pos, *end, *ie;
717 	u16 noa_len;
718 	static u8 p2p_oui_ie_type[4] = {0x50, 0x6f, 0x9a, 0x09};
719 	u8 noa_num, index , i, noa_index = 0;
720 	bool find_p2p_ie = false , find_p2p_ps_ie = false;
721 
722 	pos = (u8 *)mgmt->u.beacon.variable;
723 	end = data + len;
724 	ie = NULL;
725 
726 	while (pos + 1 < end) {
727 		if (pos + 2 + pos[1] > end)
728 			return;
729 
730 		if (pos[0] == 221 && pos[1] > 4) {
731 			if (memcmp(&pos[2], p2p_oui_ie_type, 4) == 0) {
732 				ie = pos + 2+4;
733 				break;
734 			}
735 		}
736 		pos += 2 + pos[1];
737 	}
738 
739 	if (ie == NULL)
740 		return;
741 	find_p2p_ie = true;
742 	/*to find noa ie*/
743 	while (ie + 1 < end) {
744 		noa_len = READEF2BYTE((__le16 *)&ie[1]);
745 		if (ie + 3 + ie[1] > end)
746 			return;
747 
748 		if (ie[0] == 12) {
749 			find_p2p_ps_ie = true;
750 			if ((noa_len - 2) % 13 != 0) {
751 				RT_TRACE(rtlpriv, COMP_INIT, DBG_LOUD,
752 					 "P2P notice of absence: invalid length.%d\n",
753 					 noa_len);
754 				return;
755 			} else {
756 				noa_num = (noa_len - 2) / 13;
757 			}
758 			noa_index = ie[3];
759 			if (rtlpriv->psc.p2p_ps_info.p2p_ps_mode ==
760 			    P2P_PS_NONE || noa_index != p2pinfo->noa_index) {
761 				RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
762 					 "update NOA ie.\n");
763 				p2pinfo->noa_index = noa_index;
764 				p2pinfo->opp_ps = (ie[4] >> 7);
765 				p2pinfo->ctwindow = ie[4] & 0x7F;
766 				p2pinfo->noa_num = noa_num;
767 				index = 5;
768 				for (i = 0; i < noa_num; i++) {
769 					p2pinfo->noa_count_type[i] =
770 							READEF1BYTE(ie+index);
771 					index += 1;
772 					p2pinfo->noa_duration[i] =
773 						 READEF4BYTE((__le32 *)ie+index);
774 					index += 4;
775 					p2pinfo->noa_interval[i] =
776 						 READEF4BYTE((__le32 *)ie+index);
777 					index += 4;
778 					p2pinfo->noa_start_time[i] =
779 						 READEF4BYTE((__le32 *)ie+index);
780 					index += 4;
781 				}
782 
783 				if (p2pinfo->opp_ps == 1) {
784 					p2pinfo->p2p_ps_mode = P2P_PS_CTWINDOW;
785 					/* Driver should wait LPS entering
786 					 * CTWindow
787 					 */
788 					if (rtlpriv->psc.fw_current_inpsmode)
789 						rtl_p2p_ps_cmd(hw,
790 							       P2P_PS_ENABLE);
791 				} else if (p2pinfo->noa_num > 0) {
792 					p2pinfo->p2p_ps_mode = P2P_PS_NOA;
793 					rtl_p2p_ps_cmd(hw, P2P_PS_ENABLE);
794 				} else if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
795 					rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
796 				}
797 			}
798 			break;
799 		}
800 		ie += 3 + noa_len;
801 	}
802 
803 	if (find_p2p_ie == true) {
804 		if ((p2pinfo->p2p_ps_mode > P2P_PS_NONE) &&
805 		    (find_p2p_ps_ie == false))
806 			rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
807 	}
808 }
809 
810 static void rtl_p2p_action_ie(struct ieee80211_hw *hw, void *data,
811 			      unsigned int len)
812 {
813 	struct rtl_priv *rtlpriv = rtl_priv(hw);
814 	struct ieee80211_mgmt *mgmt = data;
815 	struct rtl_p2p_ps_info *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
816 	u8 noa_num, index , i , noa_index = 0;
817 	u8 *pos, *end, *ie;
818 	u16 noa_len;
819 	static u8 p2p_oui_ie_type[4] = {0x50, 0x6f, 0x9a, 0x09};
820 
821 	pos = (u8 *)&mgmt->u.action.category;
822 	end = data + len;
823 	ie = NULL;
824 
825 	if (pos[0] == 0x7f) {
826 		if (memcmp(&pos[1], p2p_oui_ie_type, 4) == 0)
827 			ie = pos + 3+4;
828 	}
829 
830 	if (ie == NULL)
831 		return;
832 
833 	RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, "action frame find P2P IE.\n");
834 	/*to find noa ie*/
835 	while (ie + 1 < end) {
836 		noa_len = READEF2BYTE((__le16 *)&ie[1]);
837 		if (ie + 3 + ie[1] > end)
838 			return;
839 
840 		if (ie[0] == 12) {
841 			RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, "find NOA IE.\n");
842 			RT_PRINT_DATA(rtlpriv, COMP_FW, DBG_LOUD, "noa ie ",
843 				      ie, noa_len);
844 			if ((noa_len - 2) % 13 != 0) {
845 				RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
846 					 "P2P notice of absence: invalid length.%d\n",
847 					 noa_len);
848 				return;
849 			} else {
850 				noa_num = (noa_len - 2) / 13;
851 			}
852 			noa_index = ie[3];
853 			if (rtlpriv->psc.p2p_ps_info.p2p_ps_mode ==
854 			    P2P_PS_NONE || noa_index != p2pinfo->noa_index) {
855 				p2pinfo->noa_index = noa_index;
856 				p2pinfo->opp_ps = (ie[4] >> 7);
857 				p2pinfo->ctwindow = ie[4] & 0x7F;
858 				p2pinfo->noa_num = noa_num;
859 				index = 5;
860 				for (i = 0; i < noa_num; i++) {
861 					p2pinfo->noa_count_type[i] =
862 							READEF1BYTE(ie+index);
863 					index += 1;
864 					p2pinfo->noa_duration[i] =
865 							 READEF4BYTE((__le32 *)ie+index);
866 					index += 4;
867 					p2pinfo->noa_interval[i] =
868 							 READEF4BYTE((__le32 *)ie+index);
869 					index += 4;
870 					p2pinfo->noa_start_time[i] =
871 							 READEF4BYTE((__le32 *)ie+index);
872 					index += 4;
873 				}
874 
875 				if (p2pinfo->opp_ps == 1) {
876 					p2pinfo->p2p_ps_mode = P2P_PS_CTWINDOW;
877 					/* Driver should wait LPS entering
878 					 * CTWindow
879 					 */
880 					if (rtlpriv->psc.fw_current_inpsmode)
881 						rtl_p2p_ps_cmd(hw,
882 							       P2P_PS_ENABLE);
883 				} else if (p2pinfo->noa_num > 0) {
884 					p2pinfo->p2p_ps_mode = P2P_PS_NOA;
885 					rtl_p2p_ps_cmd(hw, P2P_PS_ENABLE);
886 				} else if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
887 					rtl_p2p_ps_cmd(hw, P2P_PS_DISABLE);
888 				}
889 			}
890 			break;
891 		}
892 		ie += 3 + noa_len;
893 	}
894 }
895 
896 void rtl_p2p_ps_cmd(struct ieee80211_hw *hw , u8 p2p_ps_state)
897 {
898 	struct rtl_priv *rtlpriv = rtl_priv(hw);
899 	struct rtl_ps_ctl *rtlps = rtl_psc(rtl_priv(hw));
900 	struct rtl_p2p_ps_info  *p2pinfo = &(rtlpriv->psc.p2p_ps_info);
901 
902 	RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, " p2p state %x\n" , p2p_ps_state);
903 	switch (p2p_ps_state) {
904 	case P2P_PS_DISABLE:
905 		p2pinfo->p2p_ps_state = p2p_ps_state;
906 		rtlpriv->cfg->ops->set_hw_reg(hw, HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
907 					      &p2p_ps_state);
908 		p2pinfo->noa_index = 0;
909 		p2pinfo->ctwindow = 0;
910 		p2pinfo->opp_ps = 0;
911 		p2pinfo->noa_num = 0;
912 		p2pinfo->p2p_ps_mode = P2P_PS_NONE;
913 		if (rtlps->fw_current_inpsmode) {
914 			if (rtlps->smart_ps == 0) {
915 				rtlps->smart_ps = 2;
916 				rtlpriv->cfg->ops->set_hw_reg(hw,
917 					 HW_VAR_H2C_FW_PWRMODE,
918 					 &rtlps->pwr_mode);
919 			}
920 
921 		}
922 		break;
923 	case P2P_PS_ENABLE:
924 		if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
925 			p2pinfo->p2p_ps_state = p2p_ps_state;
926 
927 			if (p2pinfo->ctwindow > 0) {
928 				if (rtlps->smart_ps != 0) {
929 					rtlps->smart_ps = 0;
930 					rtlpriv->cfg->ops->set_hw_reg(hw,
931 						 HW_VAR_H2C_FW_PWRMODE,
932 						 &rtlps->pwr_mode);
933 				}
934 			}
935 			rtlpriv->cfg->ops->set_hw_reg(hw,
936 				 HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
937 				 &p2p_ps_state);
938 
939 		}
940 		break;
941 	case P2P_PS_SCAN:
942 	case P2P_PS_SCAN_DONE:
943 	case P2P_PS_ALLSTASLEEP:
944 		if (p2pinfo->p2p_ps_mode > P2P_PS_NONE) {
945 			p2pinfo->p2p_ps_state = p2p_ps_state;
946 			rtlpriv->cfg->ops->set_hw_reg(hw,
947 				 HW_VAR_H2C_FW_P2P_PS_OFFLOAD,
948 				 &p2p_ps_state);
949 		}
950 		break;
951 	default:
952 		break;
953 	}
954 	RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
955 		 "ctwindow %x oppps %x\n",
956 		 p2pinfo->ctwindow , p2pinfo->opp_ps);
957 	RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD,
958 		 "count %x duration %x index %x interval %x start time %x noa num %x\n",
959 		 p2pinfo->noa_count_type[0],
960 		 p2pinfo->noa_duration[0],
961 		 p2pinfo->noa_index,
962 		 p2pinfo->noa_interval[0],
963 		 p2pinfo->noa_start_time[0],
964 		 p2pinfo->noa_num);
965 	RT_TRACE(rtlpriv, COMP_FW, DBG_LOUD, "end\n");
966 }
967 
968 void rtl_p2p_info(struct ieee80211_hw *hw, void *data, unsigned int len)
969 {
970 	struct rtl_priv *rtlpriv = rtl_priv(hw);
971 	struct rtl_mac *mac = rtl_mac(rtl_priv(hw));
972 	struct ieee80211_hdr *hdr = data;
973 
974 	if (!mac->p2p)
975 		return;
976 	if (mac->link_state != MAC80211_LINKED)
977 		return;
978 	/* min. beacon length + FCS_LEN */
979 	if (len <= 40 + FCS_LEN)
980 		return;
981 
982 	/* and only beacons from the associated BSSID, please */
983 	if (!ether_addr_equal_64bits(hdr->addr3, rtlpriv->mac80211.bssid))
984 		return;
985 
986 	/* check if this really is a beacon */
987 	if (!(ieee80211_is_beacon(hdr->frame_control) ||
988 	      ieee80211_is_probe_resp(hdr->frame_control) ||
989 	      ieee80211_is_action(hdr->frame_control)))
990 		return;
991 
992 	if (ieee80211_is_action(hdr->frame_control))
993 		rtl_p2p_action_ie(hw , data , len - FCS_LEN);
994 	else
995 		rtl_p2p_noa_ie(hw , data , len - FCS_LEN);
996 }
997 EXPORT_SYMBOL_GPL(rtl_p2p_info);
998