From 14528099a40f4e4801756b8c82572abdc9778e44 Mon Sep 17 00:00:00 2001
From: coolsnowwolf <coolsnowwolf@gmail.com>
Date: Wed, 13 Sep 2017 22:20:23 +0800
Subject: [PATCH] set cpu affinity for wifi and app running core load balance

---
 .../base-files/etc/init.d/set_cpu_affinity    | 79 +++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100755 target/linux/ipq806x/base-files/etc/init.d/set_cpu_affinity

diff --git a/target/linux/ipq806x/base-files/etc/init.d/set_cpu_affinity b/target/linux/ipq806x/base-files/etc/init.d/set_cpu_affinity
new file mode 100755
index 000000000..01290eee9
--- /dev/null
+++ b/target/linux/ipq806x/base-files/etc/init.d/set_cpu_affinity
@@ -0,0 +1,79 @@
+#!/bin/sh /etc/rc.common
+
+START=99
+
+set_irq_affinity() {
+	local name="$1"
+	local val="$2"
+  
+case "$name" in
+wifi0)
+  	local irq_wifi0=`grep -E -m1 'ath10k_ahb|qcom-pcie-msi' /proc/interrupts | cut -d: -f1 | tail -n1 | tr -d ' '`
+	[ -n "$irq_wifi0" ] || echo "$name irq not found."
+	echo "$val" > "/proc/irq/$irq_wifi0/smp_affinity"
+	;;
+wifi1)
+  	local irq_wifi1=`grep -E -m2 'ath10k_ahb|qcom-pcie-msi' /proc/interrupts | cut -d: -f1 | tail -n1 | tr -d ' '`
+	[ -n "$irq_wifi1" ] || echo "$name irq not found."
+	echo "$val" > "/proc/irq/$irq_wifi1/smp_affinity"
+	;;
+wifi2)
+  	local irq_wifi1=`grep -E -m3 'ath10k_ahb|qcom-pcie-msi' /proc/interrupts | cut -d: -f1 | tail -n1 | tr -d ' '`
+	[ -n "$irq_wifi1" ] || echo "$name irq not found."
+	echo "$val" > "/proc/irq/$irq_wifi1/smp_affinity"
+	;;
+*)
+  	local irq=`grep -m 1 "$name" /proc/interrupts | cut -d: -f1 | sed 's, *,,'`
+	[ -n "$irq" ] || echo "$name irq not found."
+	echo "$val" > "/proc/irq/$irq/smp_affinity"
+	;;
+esac
+}
+
+start() {
+
+. /lib/functions.sh
+
+local board=$(board_name)
+
+case "$board" in
+rt-ac58u |\
+fritz4040)
+  set_irq_affinity edma_eth_tx4 2
+	set_irq_affinity edma_eth_tx5 2
+	set_irq_affinity edma_eth_tx6 2
+	set_irq_affinity edma_eth_tx7 2
+	set_irq_affinity edma_eth_rx2 2
+	set_irq_affinity edma_eth_tx8 4
+	set_irq_affinity edma_eth_tx9 4
+	set_irq_affinity edma_eth_tx10 4
+	set_irq_affinity edma_eth_tx11 4
+	set_irq_affinity edma_eth_rx4 4
+	set_irq_affinity edma_eth_tx12 8
+	set_irq_affinity edma_eth_tx13 8
+	set_irq_affinity edma_eth_tx14 8
+	set_irq_affinity edma_eth_tx15 8
+	set_irq_affinity edma_eth_rx6 8
+	set_irq_affinity wifi0 2
+	set_irq_affinity wifi1 4
+	set_irq_affinity wifi2 8
+	;;
+ap148 |\
+c2600 |\
+db149 |\
+d7800 |\
+ea8500 |\
+nbg6817 |\
+r7500 |\
+r7500v2 |\
+r7800 |\
+vr2600v)
+  set_irq_affinity eth0 2
+	set_irq_affinity eth1 2
+	set_irq_affinity wifi0 2
+	;;
+*)
+	echo "Unsupported hardware. CPU affinity is not adjusted."
+	;;
+esac
+}