From 91468b4a79c56730a11bb0f7a5dc88a17bc29546 Mon Sep 17 00:00:00 2001
From: nakul kachhwaha <quic_nkachhwa@quicinc.com>
Date: Tue, 5 Apr 2022 12:15:05 +0530
Subject: [PATCH] qcacld-3.0: Add ucfg API to set 6ghz relaxed conn policy flag

Add ucfg API to set 6ghz relaxed connection policy flag.

change-Id: I659bf86c4308691d5adc9f2a4d2ed3c38bbc63f0
CRs-Fixed: 3166847
---
 .../mlme/dispatcher/inc/wlan_mlme_api.h       | 19 +++++++++++++++++++
 .../mlme/dispatcher/inc/wlan_mlme_ucfg_api.h  | 19 ++++++++++++++++++-
 .../mlme/dispatcher/src/wlan_mlme_api.c       | 15 +++++++++++++++
 3 files changed, 52 insertions(+), 1 deletion(-)

diff --git a/components/mlme/dispatcher/inc/wlan_mlme_api.h b/components/mlme/dispatcher/inc/wlan_mlme_api.h
index 343a573235..487febb278 100644
--- a/components/mlme/dispatcher/inc/wlan_mlme_api.h
+++ b/components/mlme/dispatcher/inc/wlan_mlme_api.h
@@ -2242,6 +2242,18 @@ wlan_mlme_set_rf_test_mode_enabled(struct wlan_objmgr_psoc *psoc, bool value);
 QDF_STATUS
 wlan_mlme_is_relaxed_6ghz_conn_policy_enabled(struct wlan_objmgr_psoc *psoc,
 					      bool *value);
+
+/**
+ * wlan_mlme_set_relaxed_6ghz_conn_policy_enabled() - Set the 6ghz relaxed
+ *                                                    connection policy flag
+ * @psoc: psoc context
+ * @value: True/False
+ *
+ * Return: QDF_STATUS
+ */
+QDF_STATUS
+wlan_mlme_set_relaxed_6ghz_conn_policy(struct wlan_objmgr_psoc *psoc,
+				       bool value);
 #else
 static inline QDF_STATUS
 wlan_mlme_is_relaxed_6ghz_conn_policy_enabled(struct wlan_objmgr_psoc *psoc,
@@ -2250,6 +2262,13 @@ wlan_mlme_is_relaxed_6ghz_conn_policy_enabled(struct wlan_objmgr_psoc *psoc,
 	*value = false;
 	return QDF_STATUS_SUCCESS;
 }
+
+static inline QDF_STATUS
+wlan_mlme_set_relaxed_6ghz_conn_policy(struct wlan_objmgr_psoc *psoc,
+				       bool value)
+{
+	return QDF_STATUS_SUCCESS;
+}
 #endif
 
 /**
diff --git a/components/mlme/dispatcher/inc/wlan_mlme_ucfg_api.h b/components/mlme/dispatcher/inc/wlan_mlme_ucfg_api.h
index a2a308ae7a..09d156ebea 100644
--- a/components/mlme/dispatcher/inc/wlan_mlme_ucfg_api.h
+++ b/components/mlme/dispatcher/inc/wlan_mlme_ucfg_api.h
@@ -2764,7 +2764,7 @@ ucfg_mlme_set_rf_test_mode_enabled(struct wlan_objmgr_psoc *psoc, bool value)
  * ucfg_mlme_is_relaxed_6ghz_conn_policy_enabled() - Get 6ghz relaxed
  *                                                   connection policy flag
  * @psoc: pointer to psoc object
- * @value: Value that needs to be set from the caller
+ * @value: pointer to hold the value of flag
  *
  * Inline UCFG API to be used by HDD/OSIF callers
  *
@@ -2777,6 +2777,23 @@ ucfg_mlme_is_relaxed_6ghz_conn_policy_enabled(struct wlan_objmgr_psoc *psoc,
 	return wlan_mlme_is_relaxed_6ghz_conn_policy_enabled(psoc, value);
 }
 
+/**
+ * ucfg_mlme_set_relaxed_6ghz_conn_policy() - Set 6ghz relaxed
+ *                                            connection policy flag
+ * @psoc: pointer to psoc object
+ * @value: Value that needs to be set
+ *
+ * Inline UCFG API to be used by HDD/OSIF callers
+ *
+ * Return: QDF Status
+ */
+static inline QDF_STATUS
+ucfg_mlme_set_relaxed_6ghz_conn_policy(struct wlan_objmgr_psoc *psoc,
+				       bool value)
+{
+	return wlan_mlme_set_relaxed_6ghz_conn_policy(psoc, value);
+}
+
 /**
  * ucfg_mlme_get_opr_rate() - Get operational rate set
  * @psoc: pointer to vdev object
diff --git a/components/mlme/dispatcher/src/wlan_mlme_api.c b/components/mlme/dispatcher/src/wlan_mlme_api.c
index 702cfcbfec..44223ec4a8 100644
--- a/components/mlme/dispatcher/src/wlan_mlme_api.c
+++ b/components/mlme/dispatcher/src/wlan_mlme_api.c
@@ -3047,6 +3047,21 @@ wlan_mlme_is_relaxed_6ghz_conn_policy_enabled(struct wlan_objmgr_psoc *psoc,
 
 	return QDF_STATUS_SUCCESS;
 }
+
+QDF_STATUS
+wlan_mlme_set_relaxed_6ghz_conn_policy(struct wlan_objmgr_psoc *psoc,
+				       bool value)
+{
+	struct wlan_mlme_psoc_ext_obj *mlme_obj;
+
+	mlme_obj = mlme_get_psoc_ext_obj(psoc);
+	if (!mlme_obj)
+		return QDF_STATUS_E_FAILURE;
+
+	mlme_obj->cfg.gen.relaxed_6ghz_conn_policy = value;
+
+	return QDF_STATUS_SUCCESS;
+}
 #endif
 
 QDF_STATUS