1 files changed, 149 insertions, 0 deletions
diff --git a/dynamic_twt_manager.h b/dynamic_twt_manager.h
new file mode 100644
index 0000000..495329b
--- /dev/null
+++ b/dynamic_twt_manager.h
@@ -0,0 +1,149 @@
+// SPDX-License-Identifier: GPL-2.0-only
+/*
+ * Driver for Wifi performance tracker
+ *
+ * Copyright 2022 Google LLC.
+ *
+ * Author: Star Chang <starchang@google.com>
+ */
+#ifndef __TP_TRACKER_DYNAMIC_TWT_SETUP_H
+#define __TP_TRACKER_DYNAMIC_TWT_SETUP_H
+
+#include "debugfs.h"
+
+struct wlan_ptracker_client;
+struct wlan_ptracker_core;
+
+struct dytwt_setup_param {
+	u8 config_id;
+	u8 nego_type;
+	u8 trigger_type;
+	u32 wake_duration;
+	u32 wake_interval;
+};
+
+struct dytwt_cap {
+	u16 device_cap;
+	u16 peer_cap;
+	u32 rssi;
+	int link_speed;
+};
+
+struct dytwt_pwr_state {
+	u64 awake;
+	u64 asleep;
+	u64 count;
+};
+
+struct dytwt_status {
+	u32 config_id;
+	u32 flow_id;
+	u32 flow_flags;
+	u32 setup_cmd;
+	u32 channel;
+	u32 nego_type;
+	u32 wake_dur;
+	u32 wake_int;
+};
+
+struct dytwt_stats {
+	u32 config_id;
+	u32 sp_seq;		/* sequence number of the service period */
+	u32 tx_ucast_pkts;	/* Number of unicast Tx packets in TWT SPs */
+	u32 tx_pkts_min;	/* Minimum number of Tx packets in a TWT SP */
+	u32 tx_pkts_max;	/* Maximum number of Tx packets in a TWT SP */
+	u32 tx_pkts_avg;	/* Average number of Tx packets in each TWT SP */
+	u32 tx_failures;	/* Tx packets failure count */
+	u32 rx_ucast_pkts;	/* Number of unicast Rx packets in TWT SPs */
+	u32 rx_pkts_min;	/* Minimum number of Rx packets in a TWT SP */
+	u32 rx_pkts_max;	/* Maximum number of Rx packets in a TWT SP */
+	u32 rx_pkts_avg;	/* Average number of Rx packets in each TWT SP */
+	u32 rx_pkts_retried;	/* retried Rx packets count */
+	u32 tx_pkt_sz_avg;	/* Average Tx packet size in TWT SPs */
+	u32 rx_pkt_sz_avg;	/* Average Rx Packet size in TWT SPs */
+	u32 eosp_dur_avg;	/* Average Wake duration in SPs ended due to EOSP */
+	u32 eosp_count;		/* Count of TWT SPs ended due to EOSP */
+};
+
+struct dytwt_client_ops {
+	int (*setup)(void *priv, struct dytwt_setup_param *param);
+	int (*teardown)(void *priv, struct dytwt_setup_param *param);
+	int (*get_cap)(void *priv, struct dytwt_cap *cap);
+	int (*get_pwrstates)(void *priv, struct dytwt_pwr_state *state);
+	int (*get_stats)(void *priv, struct dytwt_stats *stats);
+	int (*get_status)(void *priv, struct dytwt_status *status);
+};
+
+enum {
+	TWT_ACTION_SETUP,
+	TWT_ACTION_TEARDOWN,
+	TWT_ACTION_MAX,
+};
+
+enum {
+	TWT_TEST_FORCE_STATE = 1,
+	TWT_TEST_CAP,
+	TWT_TEST_PWRSTATS,
+	TWT_TEST_ONOFF,
+	TWT_TEST_SET_PARAM,
+	TWT_TEST_DUMP_STATS,
+	TWT_TEST_DUMP_STATUS,
+	TWT_TEST_MAX,
+};
+
+struct dytwt_scene_action {
+	u32 action;
+	struct dytwt_setup_param param;
+};
+
+struct dytwt_entry {
+	/* base should put as first membor */
+	struct history_entry base;
+	bool apply;
+	u32 rate;
+	u32 reason;
+	struct dytwt_pwr_state pwr;
+} __align(void *);
+
+struct dytwt_statistic {
+	u64 awake;
+	u64 asleep;
+};
+
+#define DYTWT_COUNTER_MAX 6
+#define DYTWT_COUNTER_TOTAL 5
+struct dytwt_counters {
+	u64 total_awake;
+	u64 total_sleep;
+	u64 total_sleep_cnt;
+	u64 prev_awake;
+	u64 prev_asleep;
+	u64 prev_asleep_cnt;
+	struct dytwt_statistic scene[DYTWT_COUNTER_MAX];
+};
+
+struct dytwt_manager {
+	u32 prev;
+	u32 feature_flag;
+	u32 state;
+	struct history_manager *hm;
+	u32 rssi_threshold;
+	u32 link_threshold;
+	bool twt_cap;
+	struct delayed_work wq;
+	struct delayed_work setup_wq;
+	struct wlan_ptracker_core *core;
+	struct dytwt_counters counters;
+	struct kobject kobj;
+	struct dentry *dir;
+};
+
+struct dytwt_kobj_attr {
+	struct attribute attr;
+	ssize_t (*show)(struct dytwt_manager *, char *);
+	ssize_t (*store)(struct dytwt_manager *, const char *, size_t count);
+};
+
+extern int dytwt_init(struct wlan_ptracker_core *core);
+extern void dytwt_exit(struct wlan_ptracker_core *core);
+#endif /* __TP_TRACKER_DYNAMIC_TWT_SETUP_H */