qcacld-3.0: Store TWT SI and SP in twt peer private object

Store TWT SI and SP in twt peer private object. This info will be
use in LL_LT_SAP to decide CSA via GATT message or ECSA action
frame.

Change-Id: I163b38a46267a2f1453cdb91d90dc97d0e6265e4
CRs-Fixed: 3517009

components/umac/twt/core/src/wlan_twt_main.c

@@ -1759,6 +1759,131 @@ wlan_twt_process_add_initial_nego(struct wlan_objmgr_psoc *psoc,
 				   WLAN_TWT_SETUP_STATE_ACTIVE);
 }
 
+/*
+ * wlan_twt_clear_wake_dur_and_interval() - Clear cached TWT wake duration and
+ * wake interval of peer.
+ * @psoc: Pointer to psoc object
+ * @vdev_id: Vdev Id
+ * @peer_mac: Peer mac address
+ * @dialog_id: Dialog Id
+ *
+ * Return: QDF_STATUS
+ */
+static void
+wlan_twt_clear_wake_dur_and_interval(struct wlan_objmgr_psoc *psoc,
+				     uint8_t vdev_id,
+				     struct qdf_mac_addr *peer_mac,
+				     uint8_t dialog_id)
+{
+	struct twt_peer_priv_obj *peer_priv;
+	struct wlan_objmgr_peer *peer;
+	uint8_t i;
+
+	peer = wlan_objmgr_get_peer_by_mac(psoc, peer_mac->bytes,
+					   WLAN_TWT_ID);
+	if (!peer) {
+		twt_err("Peer object not found "QDF_MAC_ADDR_FMT,
+			QDF_MAC_ADDR_REF(peer_mac->bytes));
+		return;
+	}
+	peer_priv = wlan_objmgr_peer_get_comp_private_obj(peer,
+							  WLAN_UMAC_COMP_TWT);
+	if (!peer_priv) {
+		wlan_objmgr_peer_release_ref(peer, WLAN_TWT_ID);
+		twt_err(" peer twt component object is NULL");
+		return;
+	}
+
+	qdf_mutex_acquire(&peer_priv->twt_peer_lock);
+
+	for (i = 0; i < peer_priv->num_twt_sessions; i++) {
+		if (peer_priv->session_info[i].dialog_id == dialog_id) {
+			peer_priv->session_info[i].dialog_id = TWT_ALL_SESSIONS_DIALOG_ID;
+			peer_priv->session_info[i].wake_dur = 0;
+			peer_priv->session_info[i].wake_interval = 0;
+			break;
+		}
+	}
+
+	twt_debug("vdev:%d peer:" QDF_MAC_ADDR_FMT " dialog_id:%d wake_dur:%d wake_interval:%d",
+		  vdev_id,
+		  QDF_MAC_ADDR_REF(peer_mac->bytes),
+		  peer_priv->session_info[i].dialog_id,
+		  peer_priv->session_info[i].wake_dur,
+		  peer_priv->session_info[i].wake_interval);
+
+	qdf_mutex_release(&peer_priv->twt_peer_lock);
+	wlan_objmgr_peer_release_ref(peer, WLAN_TWT_ID);
+}
+
+/*
+ * wlan_twt_set_wake_dur_and_interval() - Set TWT wake duration and wake
+ * interval of peer.
+ * @psoc: Pointer to psoc object
+ * @vdev_id: Vdev Id
+ * @peer_mac: Peer mac address
+ * @dialog_id: Dialog Id
+ * @wake_dur: TWT wake duration
+ * @wake_interval: TWT wake interval
+ *
+ * Return: QDF_STATUS
+ */
+static void
+wlan_twt_set_wake_dur_and_interval(struct wlan_objmgr_psoc *psoc,
+				   uint8_t vdev_id,
+				   struct qdf_mac_addr *peer_mac,
+				   uint8_t dialog_id,
+				   uint32_t wake_dur,
+				   uint32_t wake_interval)
+{
+	struct twt_peer_priv_obj *peer_priv;
+	struct wlan_objmgr_peer *peer;
+	uint8_t i;
+
+	peer = wlan_objmgr_get_peer_by_mac(psoc, peer_mac->bytes,
+					   WLAN_TWT_ID);
+	if (!peer) {
+		twt_err("Peer object not found "QDF_MAC_ADDR_FMT,
+			QDF_MAC_ADDR_REF(peer_mac->bytes));
+		return;
+	}
+
+	peer_priv = wlan_objmgr_peer_get_comp_private_obj(peer,
+							  WLAN_UMAC_COMP_TWT);
+	if (!peer_priv) {
+		wlan_objmgr_peer_release_ref(peer, WLAN_TWT_ID);
+		twt_err(" peer twt component object is NULL");
+		return;
+	}
+
+	qdf_mutex_acquire(&peer_priv->twt_peer_lock);
+
+	for (i = 0; i < peer_priv->num_twt_sessions; i++) {
+		if (peer_priv->session_info[i].dialog_id == dialog_id) {
+			peer_priv->session_info[i].wake_dur = wake_dur;
+			peer_priv->session_info[i].wake_interval = wake_interval;
+				break;
+		} else {
+			if (peer_priv->session_info[i].dialog_id == TWT_ALL_SESSIONS_DIALOG_ID) {
+				peer_priv->session_info[i].dialog_id = dialog_id;
+				peer_priv->session_info[i].wake_dur = wake_dur;
+				peer_priv->session_info[i].wake_interval = wake_interval;
+				break;
+			}
+		}
+	}
+
+	twt_debug("vdev:%d peer:" QDF_MAC_ADDR_FMT " dialog_id:%d wake_dur:%d wake_interval:%d",
+		  vdev_id,
+		  QDF_MAC_ADDR_REF(peer_mac->bytes),
+		  peer_priv->session_info[i].dialog_id,
+		  peer_priv->session_info[i].wake_dur,
+		  peer_priv->session_info[i].wake_interval);
+
+	qdf_mutex_release(&peer_priv->twt_peer_lock);
+	wlan_objmgr_peer_release_ref(peer, WLAN_TWT_ID);
+}
+
 QDF_STATUS
 wlan_twt_setup_complete_event_handler(struct wlan_objmgr_psoc *psoc,
 				    struct twt_add_dialog_complete_event *event)
@@ -1789,6 +1914,12 @@ wlan_twt_setup_complete_event_handler(struct wlan_objmgr_psoc *psoc,
 
 	switch (opmode) {
 	case QDF_SAP_MODE:
+		wlan_twt_set_wake_dur_and_interval(
+					psoc, vdev_id,
+					&event->params.peer_macaddr,
+					event->params.dialog_id,
+					event->additional_params.wake_dur_us,
+					event->additional_params.wake_intvl_us);
 		qdf_status = mlme_twt_osif_setup_complete_ind(psoc, event,
 							      false);
 		break;
@@ -1923,6 +2054,10 @@ wlan_twt_teardown_complete_event_handler(struct wlan_objmgr_psoc *psoc,
 	case QDF_SAP_MODE:
 		mlme_twt_osif_teardown_complete_ind(psoc, event);
 
+		wlan_twt_clear_wake_dur_and_interval(psoc, vdev_id,
+						     &event->peer_macaddr,
+						     event->dialog_id);
+
 		/*
 		 * If this is an unsolicited TWT del event initiated from the
 		 * peer, then no need to clear the active command in progress