feat: radio1 5GHz/6GHz rogue AP via UCI with hop pause

2026-08-17 23:28:54 -05:00
parent b2098bae7d
commit 4effc08a25
2 changed files with 182 additions and 2 deletions
@@ -159,6 +159,97 @@ class GetApRadio1AbsentTest(unittest.TestCase):
        self.assertEqual(payload['wpa']['channel'], 1)


+class SetApRadio1Test(unittest.TestCase):
+    def setUp(self):
+        self.uci = {}
+        self.runs = []
+
+        def fake_uci_get(section):
+            return self.uci.get(section)
+
+        def fake_run(args, timeout=20, input_data=None):
+            self.runs.append((list(args), input_data))
+            a = list(args)
+            if a[:2] == ['uci', 'set']:
+                k, _, v = a[2].partition('=')
+                self.uci[k] = v
+            if a[:2] == ['uci', 'get']:
+                return (0, self.uci.get(a[2], '') + '\n', '')
+            return (0, '', '')
+
+        server._uci_wifi_iface = fake_uci_get
+        server._uci_section = fake_uci_get
+        server.device_run = fake_run
+        server.daemon_sock_call = lambda method, path, body=None, timeout=10: (
+            200, {'success': True})
+
+    def test_5g_open_writes_radio1_sections(self):
+        server.h_pineap_wifi_set_ap(ctx({'open': {
+            'ssid': 'CorpGuest', 'hidden': False, 'enabled': True,
+            'channel': 36, 'country': 'US'}}))
+        self.assertEqual(self.uci['wireless.radio1.band'], '5g')
+        self.assertEqual(self.uci['wireless.radio1.channel'], '36')
+        self.assertEqual(self.uci['wireless.radio1.htmode'], 'VHT80')
+        self.assertEqual(self.uci['wireless.wlan1open'], 'wifi-iface')
+        self.assertEqual(self.uci['wireless.wlan1open.device'], 'radio1')
+        self.assertEqual(self.uci['wireless.wlan1open.ssid'], 'CorpGuest')
+        self.assertEqual(self.uci['wireless.wlan1open.encryption'], 'none')
+        self.assertEqual(self.uci['pineapd.wlan1mon.hop'], '0')
+        self.assertIn(['wifi', 'reload'], [r[0] for r in self.runs])
+        self.assertIn(['/etc/init.d/pineapd', 'reload'], [r[0] for r in self.runs])
+
+    def test_6g_wpa_sae(self):
+        server.h_pineap_wifi_set_ap(ctx({'wpa': {
+            'ssid': 'Corp', 'passphrase': 'secret123', 'enctype': 'sae',
+            'hidden': False, 'enabled': True, 'channel': 181}}))
+        self.assertEqual(self.uci['wireless.radio1.band'], '6g')
+        self.assertEqual(self.uci['wireless.radio1.htmode'], 'HE80')
+        self.assertEqual(self.uci['wireless.wlan1wpa.encryption'], 'sae')
+
+    def test_6g_rejects_psk2(self):
+        status, payload = server.h_pineap_wifi_set_ap(ctx({'wpa': {
+            'ssid': 'Corp', 'passphrase': 'secret123', 'enctype': 'psk2',
+            'hidden': False, 'enabled': True, 'channel': 181}}))
+        self.assertEqual(status, 400)
+
+    def test_6g_disable_removes_radio1(self):
+        self.uci['pineapd.wlan1mon.hop'] = '0'
+        self.uci['wlan1wpa'] = {'device': 'radio1'}
+        status, payload = server.h_pineap_wifi_set_ap(ctx({'wpa': {
+            'ssid': 'Corp', 'passphrase': 'secret123', 'enctype': 'sae',
+            'hidden': False, 'enabled': False, 'channel': 181}}))
+        self.assertEqual(status, 200)
+        self.assertEqual(self.uci['wireless.radio1.channel'], 'auto')
+        self.assertEqual(self.uci['pineapd.wlan1mon.hop'], '1')
+
+    def test_2g_still_uses_daemon_path(self):
+        calls = []
+
+        def fake_sock(method, path, body=None, timeout=10):
+            if method == 'GET':
+                return 200, {'loghandshake': True}
+            calls.append((method, path, body))
+            return 200, {'success': True}
+
+        server.daemon_sock_call = fake_sock
+        server.h_pineap_wifi_set_ap(ctx({'open': {
+            'ssid': 'pager-open', 'hidden': False, 'enabled': True,
+            'channel': 11, 'country': 'US'}}))
+        put = [c for c in calls if c[0] == 'PUT']
+        self.assertEqual(len(put), 1)
+        self.assertEqual(put[0][1], '/api/settings/wifi/set_ap')
+        self.assertEqual(put[0][2]['configs'][0]['interface'], 'wlan0open')
+
+    def test_2g_removes_existing_radio1(self):
+        self.uci['pineapd.wlan1mon.hop'] = '0'
+        self.uci['wlan1open'] = {'device': 'radio1'}
+        server.h_pineap_wifi_set_ap(ctx({'open': {
+            'ssid': 'pager-open', 'hidden': False, 'enabled': True,
+            'channel': 11, 'country': 'US'}}))
+        self.assertEqual(self.uci['pineapd.wlan1mon.hop'], '1')
+        self.assertEqual(self.uci.get('wireless.radio1.channel'), 'auto')
+
+
 class GetApRadioChannelFallbackTest(unittest.TestCase):
    """Regression: iface without a channel option inherits the radio channel."""