refind network and monitor wifi connection

This commit is contained in:
gcw_4spBpAfv
2026-04-02 18:02:34 +08:00
parent fffca13941
commit ec80107128
4 changed files with 877 additions and 129 deletions

View File

@@ -21,4 +21,5 @@ files:
- shot_id_generator.py - shot_id_generator.py
- time_sync.py - time_sync.py
- version.py - version.py
- vision.py - vision.cpython-311-riscv64-linux-gnu.so
- wifi.py

View File

@@ -80,4 +80,26 @@
自动关机的时机: 超过配置的idle时长 自动关机的时机: 超过配置的idle时长
禁止自动关机的情况1.校准中2.OTA中 禁止自动关机的情况1.校准中2.OTA中
重启计时的时机1.校准完成2.命令触发射箭3.真实触发射箭4.初始化完成 重启计时的时机1.校准完成2.命令触发射箭3.真实触发射箭4.初始化完成
9. 9. Wifi网络监控
有两次发现wifi网络下有些消息发送很慢但具体是什么缘故还不清楚现在增加了wifi网络下的检测并一旦发现wifi的网络质量差就会切换到4G。
WiFi 连接成功
启动后台监测线程
每 5 秒循环:
测量 RTT (1 样本600ms timeout)
获取 RSSI
更新缓存
判断是否差:
- RTT >= 600ms → 差
- RTT >= 350ms 且 RSSI <= -80dBm → 差
如果质量差:
快速重试2次如果其中任意一次网络恢复了继续使用wifi。否则
调用 _switch_to_4g_due_to_poor_wifi()
关闭 WiFi socket
重置连接状态
尝试切换到 4G
上层检测到连接断开:
重新 connect_server() → 自动选择 4G

View File

@@ -5,6 +5,7 @@
提供TCP通信、数据包打包/解析、队列管理等功能 提供TCP通信、数据包打包/解析、队列管理等功能
""" """
import json import json
import re
from math import e from math import e
import struct import struct
from maix import time from maix import time
@@ -21,6 +22,7 @@ from power import get_bus_voltage, voltage_to_percent
# from laser import laser_manager # from laser import laser_manager
# from ota import ota_manager # from ota import ota_manager
from logger_manager import logger_manager from logger_manager import logger_manager
from wifi import wifi_manager
class NetworkManager: class NetworkManager:
@@ -49,14 +51,11 @@ class NetworkManager:
self._raw_line_data = [] self._raw_line_data = []
self._manual_trigger_flag = False self._manual_trigger_flag = False
# WiFi 相关状态 # 网络类型状态
self._network_type = None # "wifi" 或 "4g" 或 None self._network_type = None # "wifi" 或 "4G" 或 None
self._wifi_connected = False # 本次上电曾因 WiFi 质量差切换到 4G 后,直至关机不再改回 WiFi
self._wifi_ip = None self._session_force_4g = False
self._wifi_socket = None
self._wifi_socket_lock = threading.Lock()
self._prefer_wifi = True # 是否优先使用WiFi
self._wifi_recv_buffer = b"" # WiFi接收缓冲区
self._initialized = True self._initialized = True
# 导入 archery_netcore 模块,并检查是否存在 parse_packet 和 make_packet 函数 # 导入 archery_netcore 模块,并检查是否存在 parse_packet 和 make_packet 函数
@@ -117,12 +116,12 @@ class NetworkManager:
@property @property
def wifi_connected(self): def wifi_connected(self):
"""WiFi是否已连接""" """WiFi是否已连接"""
return self._wifi_connected return wifi_manager.wifi_connected
@property @property
def wifi_ip(self): def wifi_ip(self):
"""WiFi IP地址""" """WiFi IP地址"""
return self._wifi_ip return wifi_manager.wifi_ip
# ==================== 内部状态管理方法 ==================== # ==================== 内部状态管理方法 ====================
@@ -200,39 +199,15 @@ class NetworkManager:
self._password = default_id + "." self._password = default_id + "."
return default_id return default_id
# ==================== WiFi 管理方法 ==================== # ==================== WiFi 管理方法(委托给 wifi_manager====================
def is_wifi_connected(self): def is_wifi_connected(self):
"""检查WiFi是否已连接""" """检查WiFi是否已连接"""
# 优先用 MaixPy network如果可用 return wifi_manager.is_wifi_connected()
try:
from maix import network
wlan = network.WLAN(network.TYPE_WIFI)
if wlan.isconnected():
self._wifi_connected = True
return True
except:
pass
# 兜底:看系统 wlan0 有没有 IP
try:
ip = os.popen("ifconfig wlan0 2>/dev/null | grep 'inet ' | awk '{print $2}'").read().strip()
if ip:
self._wifi_connected = True
self._wifi_ip = ip
return True
except:
pass
self._wifi_connected = False
return False
def connect_wifi(self, ssid, password, verify_host=None, verify_port=None, persist=True, timeout_s=20): def connect_wifi(self, ssid, password, verify_host=None, verify_port=None, persist=True, timeout_s=20):
""" """
连接 Wi-Fi先用新凭证尝试连接并验证可用性;失败自动回滚;成功后再决定是否落盘 连接 Wi-Fi委托给 wifi_manager
重要:系统的 /etc/init.d/S30wifi 通常会读取 /boot/wifi.ssid 与 /boot/wifi.pass 来连接 WiFi。
因此要“真正尝试连接新 WiFi”必须临时写入 /boot/ 触发重启;若失败则把旧值写回去(回滚)。
Returns: Returns:
(ip, error): IP地址和错误信息成功时error为None (ip, error): IP地址和错误信息成功时error为None
@@ -369,6 +344,77 @@ class NetworkManager:
# ==================== 网络选择策略 ==================== # ==================== 网络选择策略 ====================
def _get_wifi_rssi_dbm(self):
"""获取 WiFi 信号强度(委托给 wifi_manager"""
return wifi_manager._get_wifi_rssi_dbm()
def _measure_wifi_tcp_rtt_ms(self, host, port, samples=3, per_sample_timeout_ms=900):
"""测量 WiFi TCP RTT委托给 wifi_manager"""
return wifi_manager._measure_wifi_tcp_rtt_ms(host, port, samples, per_sample_timeout_ms)
def _is_wifi_quality_bad(self, wifi_rtt_ms, wifi_rssi_dbm):
"""判断 WiFi 质量是否差(委托给 wifi_manager"""
return wifi_manager._is_wifi_quality_bad(wifi_rtt_ms, wifi_rssi_dbm)
def is_4g_available(self):
"""
快速判断 4G 是否可用(不做 DNS仅验证 SIM/附着/有 IP
兼容性兜底:若未拿到 IP再补充设置 APN 并激活 PDP context
"""
try:
atc = hardware_manager.at_client
if atc is None:
return False
with self.get_uart_lock():
# 1) SIM 就绪
r = atc.send("AT+CPIN?", "READY", 3000)
if "READY" not in r:
return False
# 2) 尝试附着到网络
# 不同模块返回可能略有差异,所以做宽松解析
r2 = atc.send("AT+CGATT?", "OK", 3000)
m = re.search(r"\+CGATT:\s*(\d+)", r2)
attached = int(m.group(1)) == 1 if m else None
if attached is False or attached is None:
atc.send("AT+CGATT=1", "OK", 5000)
# 3) 查询 PDP 地址(有 IP 表示网络可用)
def _extract_ip(resp: str):
m3 = re.search(r"(\d{1,3}\.){3}\d{1,3}", resp or "")
if not m3:
return None
ip = m3.group(0)
if ip.startswith("0.") or ip.startswith("127."):
return None
return ip
r3 = atc.send("AT+CGPADDR=1", "OK", 3000)
ip = _extract_ip(r3)
if ip:
return True
# 4) 若没 IP补充设置 APN + 激活 PDP某些网络/模组需要这一步)
atc.send('AT+CGDCONT=1,"IP","CMNET"', "OK", 3000)
qact_resp = atc.send("AT+CGACT?", "OK", 3000)
if "+CGACT:" not in (qact_resp or "") or "1,1" not in (qact_resp or ""):
atc.send("AT+CGACT=1,1", "OK", 10000)
r4 = atc.send("AT+CGPADDR=1", "OK", 3000)
ip2 = _extract_ip(r4)
if ip2:
return True
return False
except Exception:
return False
def _apply_session_force_4g(self):
"""锁定本次会话为 4G直到关机期间不再回切 WiFi"""
self._session_force_4g = True
self._network_type = "4g"
def select_network(self, prefer_wifi=None): def select_network(self, prefer_wifi=None):
""" """
自动选择网络WiFi优先 自动选择网络WiFi优先
@@ -380,33 +426,109 @@ class NetworkManager:
"wifi""4g" 或 None无可用网络 "wifi""4g" 或 None无可用网络
""" """
if prefer_wifi is None: if prefer_wifi is None:
prefer_wifi = self._prefer_wifi prefer_wifi = wifi_manager.prefer_wifi
# 策略1如果指定优先WiFi且WiFi可用使用WiFi # 本次会话锁定:只要 4G 可用就一直用 4G
if prefer_wifi and self.is_wifi_connected(): if self._session_force_4g:
# 检查WiFi是否能连接到服务器 self.logger.info("[NET] 会话锁定 4G继续使用 4G跳过 WiFi 质量评估)")
if self.is_server_reachable(self._server_ip, self._server_port, timeout=3):
self._network_type = "wifi"
self.logger.info(f"[NET] 选择WiFi网络IP: {self._wifi_ip}")
import os
os.environ["TZ"] = "Asia/Shanghai"
os.system("ntpdate pool.ntp.org")
return "wifi"
else:
self.logger.warning("[NET] WiFi已连接但无法访问服务器尝试4G")
# 策略2如果WiFi可用使用WiFi
if self.is_wifi_connected():
if self.is_server_reachable(self._server_ip, self._server_port, timeout=3):
self._network_type = "wifi"
self.logger.info(f"[NET] 选择WiFi网络IP: {self._wifi_ip}")
return "wifi"
# 策略3回退到4G
self.logger.info("[NET] WiFi不可用或无法连接服务器使用4G网络")
self._network_type = "4g" self._network_type = "4g"
return "4g" return "4g"
host = self._server_ip
port = self._server_port
# 1) 开机先尝试 WiFi并评估质量
if prefer_wifi and self.is_wifi_connected():
wifi_rssi_dbm = self._get_wifi_rssi_dbm()
wifi_rtt_ms, wifi_reachable = self._measure_wifi_tcp_rtt_ms(
host, port,
samples=getattr(config, "WIFI_QUALITY_RTT_SAMPLES", 3),
per_sample_timeout_ms=900,
)
wifi_bad = self._is_wifi_quality_bad(wifi_rtt_ms, wifi_rssi_dbm)
try:
self.logger.info(
f"[NET] WiFi质量评估rtt_ms(median)={wifi_rtt_ms:.1f}, rssi_dbm={wifi_rssi_dbm}, "
f"reachable={wifi_reachable}, bad={wifi_bad}"
)
except Exception:
pass
# 如果质量差且 4G 可用 -> 切换 4G 并锁定本次会话
if wifi_bad and self.is_4g_available():
self._apply_session_force_4g()
self.logger.warning("[NET] WiFi质量差且 4G 可用 -> 切换到 4G并锁定本次会话")
return "4g"
# 否则:仍以 WiFi 为主,但如果服务器不可达则回退到 4G
if wifi_reachable or self.is_server_reachable(host, port, timeout=3):
self._network_type = "wifi"
self.logger.info(f"[NET] 选择WiFi网络IP: {wifi_manager.wifi_ip}")
try:
os.environ["TZ"] = "Asia/Shanghai"
os.system("ntpdate pool.ntp.org")
except Exception:
pass
return "wifi"
# WiFi可用但服务器不可达
self.logger.warning("[NET] WiFi可用但服务器不可达尝试4G")
# 2) 如果 WiFi 没法用/不想用,回退到 4G
if self.is_4g_available():
self._network_type = "4g"
self.logger.info("[NET] 选择4G网络")
return "4g"
# 3) 两者都不可用
self.logger.error("[NET] WiFi 与 4G 均不可用")
return None
def _start_wifi_quality_monitor(self):
"""
启动 WiFi 质量后台监测线程(委托给 wifi_manager
"""
wifi_manager.start_quality_monitor(
network_type_callback=lambda: self._network_type,
on_poor_quality_callback=self._switch_to_4g_due_to_poor_wifi
)
def _stop_wifi_quality_monitor(self):
"""停止 WiFi 质量监测线程(委托给 wifi_manager"""
wifi_manager.stop_quality_monitor()
def get_wifi_quality_status(self):
"""获取当前 WiFi 质量状态(委托给 wifi_manager"""
return wifi_manager.get_wifi_quality_status()
def _switch_to_4g_due_to_poor_wifi(self):
"""
由于 WiFi 质量差,切换到 4G 网络
"""
self.logger.info("[WiFi->4G] 开始切换到 4G 网络")
# 1. 标记本次上电强制使用 4G
self._session_force_4g = True
# 2. 关闭 WiFi socket
wifi_manager.disconnect_wifi()
# 3. 重置连接状态
self._tcp_connected = False
self._network_type = None # 清空,让 select_network 重新选择
# 4. 检查 4G 是否可用
if self.is_4g_available():
self._network_type = "4g"
self.logger.info("[WiFi->4G] 切换成功,将使用 4G 网络")
return True
else:
self.logger.error("[WiFi->4G] 4G 不可用,无法切换")
# 回退:继续使用 WiFi虽然质量差
self._session_force_4g = False
return False
def safe_enqueue(self, data_dict, msg_type=2, high=False): def safe_enqueue(self, data_dict, msg_type=2, high=False):
"""线程安全地将消息加入队列(公共方法)""" """线程安全地将消息加入队列(公共方法)"""
@@ -485,32 +607,36 @@ class NetworkManager:
"""通过WiFi建立TCP连接""" """通过WiFi建立TCP连接"""
try: try:
# 创建TCP socket # 创建TCP socket
self._wifi_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) wifi_manager.wifi_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self._wifi_socket.settimeout(5.0) # 5秒超时 wifi_manager.wifi_socket.settimeout(5.0) # 5秒超时
# 连接到服务器 # 连接到服务器
addr_info = socket.getaddrinfo(config.SERVER_IP, config.SERVER_PORT, addr_info = socket.getaddrinfo(config.SERVER_IP, config.SERVER_PORT,
socket.AF_INET, socket.SOCK_STREAM)[0] socket.AF_INET, socket.SOCK_STREAM)[0]
self._wifi_socket.connect(addr_info[-1]) wifi_manager.wifi_socket.connect(addr_info[-1])
# 设置非阻塞模式(用于接收数据) # 设置非阻塞模式(用于接收数据)
self._wifi_socket.setblocking(False) wifi_manager.wifi_socket.setblocking(False)
# 加快消息发送 # 加快消息发送
self._wifi_socket.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1) wifi_manager.wifi_socket.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
self._tcp_connected = True self._tcp_connected = True
self.logger.info("[WIFI-TCP] TCP连接已建立") self.logger.info("[WIFI-TCP] TCP 连接已建立")
# 启动 WiFi 质量后台监测
self._start_wifi_quality_monitor()
return True return True
except Exception as e: except Exception as e:
self.logger.error(f"[WIFI-TCP] 连接失败: {e}") self.logger.error(f"[WIFI-TCP] 连接失败: {e}")
if self._wifi_socket: if wifi_manager.wifi_socket:
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
return False return False
@@ -543,12 +669,12 @@ class NetworkManager:
def _check_wifi_connection(self): def _check_wifi_connection(self):
"""检查WiFi TCP连接是否仍然有效""" """检查WiFi TCP连接是否仍然有效"""
if not self._wifi_socket: if not wifi_manager.wifi_socket:
return False return False
try: try:
# send(b"") 在很多实现里是 no-op无法可靠探测断线。 # send(b"") 在很多实现里是 no-op无法可靠探测断线。
# 用非阻塞 peek 来判断若对端已关闭recv 会返回 b""。 # 用非阻塞 peek 来判断若对端已关闭recv 会返回 b""。
data = self._wifi_socket.recv(1, socket.MSG_PEEK | socket.MSG_DONTWAIT) data = wifi_manager.wifi_socket.recv(1, socket.MSG_PEEK | socket.MSG_DONTWAIT)
if data == b"": if data == b"":
raise OSError("wifi socket closed") raise OSError("wifi socket closed")
return True return True
@@ -561,7 +687,7 @@ class NetworkManager:
if err in (11, 35, 10035): # EAGAIN/EWOULDBLOCK on linux/mac/win if err in (11, 35, 10035): # EAGAIN/EWOULDBLOCK on linux/mac/win
return True return True
# 某些平台会把无数据可读/超时抛成 socket.timeout / TimeoutErrorerrno 可能为 None # 某些平台会把"无数据可读/超时"抛成 socket.timeout / TimeoutErrorerrno 可能为 None
# 这不代表断线:视为 benign交给真正的 send/recv 去判定断线。 # 这不代表断线:视为 benign交给真正的 send/recv 去判定断线。
if (err is None) and (("timed out" in str(e).lower()) or isinstance(e, (TimeoutError, socket.timeout))): if (err is None) and (("timed out" in str(e).lower()) or isinstance(e, (TimeoutError, socket.timeout))):
try: try:
@@ -571,7 +697,7 @@ class NetworkManager:
return True return True
# 某些嵌入式 socket 实现可能不支持 MSG_PEEK/MSG_DONTWAIT或返回 EINVAL/ENOTSUP。 # 某些嵌入式 socket 实现可能不支持 MSG_PEEK/MSG_DONTWAIT或返回 EINVAL/ENOTSUP。
# 这种情况不代表断线:选择无法检测但不判死,交给真正的 send/recv 去触发断线处理。 # 这种情况不代表断线:选择"无法检测但不判死",交给真正的 send/recv 去触发断线处理。
# 常见EINVAL(22), ENOTSUP(95), EOPNOTSUPP(95), EINTR(4) # 常见EINVAL(22), ENOTSUP(95), EOPNOTSUPP(95), EINTR(4)
if err in (4, 22, 95): if err in (4, 22, 95):
try: try:
@@ -586,31 +712,31 @@ class NetworkManager:
except: except:
pass pass
# 明确的连接不可用错误才判定断线并清理 # 明确的"连接不可用"错误才判定断线并清理
# 常见ENOTCONN(107), ECONNRESET(104), EPIPE(32), EBADF(9) # 常见ENOTCONN(107), ECONNRESET(104), EPIPE(32), EBADF(9)
if err in (9, 32, 104, 107, 10054, 10057): if err in (9, 32, 104, 107, 10054, 10057):
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
self._tcp_connected = False self._tcp_connected = False
return False return False
# socket已断开或不可用清理 # socket已断开或不可用清理
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
self._tcp_connected = False self._tcp_connected = False
return False return False
except Exception: except Exception:
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
self._tcp_connected = False self._tcp_connected = False
return False return False
@@ -628,14 +754,18 @@ class NetworkManager:
self._network_type = None self._network_type = None
def _disconnect_tcp_via_wifi(self): def _disconnect_tcp_via_wifi(self):
"""断开WiFi TCP连接""" """断开 WiFi TCP 连接并停止监测"""
with self._wifi_socket_lock: # 先停止监测线程
if self._wifi_socket: self._stop_wifi_quality_monitor()
# 再关闭 socket
with wifi_manager.wifi_socket_lock:
if wifi_manager.wifi_socket:
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
def _disconnect_tcp_via_4g(self): def _disconnect_tcp_via_4g(self):
link_id = getattr(config, "TCP_LINK_ID", 0) link_id = getattr(config, "TCP_LINK_ID", 0)
@@ -672,16 +802,16 @@ class NetworkManager:
def _tcp_send_raw_via_wifi(self, data: bytes, max_retries=2) -> bool: def _tcp_send_raw_via_wifi(self, data: bytes, max_retries=2) -> bool:
"""通过WiFi socket发送TCP数据""" """通过WiFi socket发送TCP数据"""
if not self._wifi_socket: if not wifi_manager.wifi_socket:
return False return False
with self._wifi_socket_lock: with wifi_manager.wifi_socket_lock:
for attempt in range(max_retries): for attempt in range(max_retries):
try: try:
# 标准socket发送 # 标准 socket 发送
total_sent = 0 total_sent = 0
while total_sent < len(data): while total_sent < len(data):
sent = self._wifi_socket.send(data[total_sent:]) sent = wifi_manager.wifi_socket.send(data[total_sent:])
if sent == 0: if sent == 0:
# socket连接已断开 # socket连接已断开
self.logger.warning(f"[WIFI-TCP] 发送失败socket已断开尝试 {attempt+1}/{max_retries}") self.logger.warning(f"[WIFI-TCP] 发送失败socket已断开尝试 {attempt+1}/{max_retries}")
@@ -698,19 +828,19 @@ class NetworkManager:
self.logger.error(f"[WIFI-TCP] 发送异常: {e}(尝试 {attempt+1}/{max_retries}") self.logger.error(f"[WIFI-TCP] 发送异常: {e}(尝试 {attempt+1}/{max_retries}")
# 发送异常通常意味着连接已不可用,主动关闭以触发重连 # 发送异常通常意味着连接已不可用,主动关闭以触发重连
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
self._tcp_connected = False self._tcp_connected = False
return False return False
except Exception as e: except Exception as e:
self.logger.error(f"[WIFI-TCP] 未知错误: {e}(尝试 {attempt+1}/{max_retries}") self.logger.error(f"[WIFI-TCP] 未知错误: {e}(尝试 {attempt+1}/{max_retries}")
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
self._tcp_connected = False self._tcp_connected = False
return False return False
@@ -747,15 +877,14 @@ class NetworkManager:
verify_mode = getattr(config, "SSL_VERIFY_MODE", 0) verify_mode = getattr(config, "SSL_VERIFY_MODE", 0)
# 1) 配置认证方式 # 1) 配置认证方式
# r = hardware_manager.at_client.send(f'AT+MSSLCFG="auth",{ssl_id},{auth_mode}', "OK", 3000) r = hardware_manager.at_client.send(f'AT+MSSLCFG="auth",{ssl_id},{auth_mode}', "OK", 3000)
r = hardware_manager.at_client.send(f'AT+MSSLCFG="auth",{ssl_id},0', "OK", 3000) # r = hardware_manager.at_client.send(f'AT+MSSLCFG="auth",{ssl_id},0', "OK", 3000)
self.logger.info(f"[4G-TCP] AT+MSSLCFG=\"auth\",{ssl_id},0 response: {r}") self.logger.info(f"[4G-TCP] AT+MSSLCFG=\"auth\",{ssl_id},0 response: {r}")
if "OK" not in r: if "OK" not in r:
return False return False
# 2) 写入根证书(只有 verify_mode=1 才需要) # 2) 写入根证书(只有 verify_mode=1 才需要)
# if verify_mode == 1: if verify_mode == 1:
if False:
cert_filename = getattr(config, "SSL_CERT_FILENAME", None) cert_filename = getattr(config, "SSL_CERT_FILENAME", None)
cert_path = getattr(config, "SSL_CERT_PATH", None) cert_path = getattr(config, "SSL_CERT_PATH", None)
if not cert_filename or not cert_path: if not cert_filename or not cert_path:
@@ -767,6 +896,7 @@ class NetworkManager:
# 按手册AT+MSSLCERTWR="file",0,size -> 等待 ">" -> 写入证书内容 -> 等 OK # 按手册AT+MSSLCERTWR="file",0,size -> 等待 ">" -> 写入证书内容 -> 等 OK
r = hardware_manager.at_client.send(f'AT+MSSLCERTWR="{cert_filename}",0,{len(cert_data)}', ">", 5000) r = hardware_manager.at_client.send(f'AT+MSSLCERTWR="{cert_filename}",0,{len(cert_data)}', ">", 5000)
self.logger.info(f"[4G-TCP] AT+MSSLCERTWR=\"{cert_filename}\",0,{len(cert_data)} response: {r}")
if ">" not in r: if ">" not in r:
return False return False
@@ -774,18 +904,36 @@ class NetworkManager:
hardware_manager.uart4g.write(cert_data) hardware_manager.uart4g.write(cert_data)
r = hardware_manager.at_client.send("", "OK", 8000) r = hardware_manager.at_client.send("", "OK", 8000)
self.logger.info(f"[4G-TCP] AT+MSSLCERTWR=\"{cert_filename}\",0,{len(cert_data)} response: {r}")
if "OK" not in r: if "OK" not in r:
return False return False
r = hardware_manager.at_client.send(f'AT+MSSLCHECK="{cert_filename}"', "OK", 8000)
if "OK" not in r:
self.logger.error(f"[4G-TCP] MSSLCHECK failed, response: {r}")
return False
else:
self.logger.info(f"[4G-TCP] MSSLCHECK response: {r}")
r = hardware_manager.at_client.send(f'AT+MSSLLIST=1', "OK", 3000)
self.logger.info(f"[4G-TCP] AT+MSSLLIST=1 response: {r}")
r = hardware_manager.at_client.send(f'AT+MSSLCERTRD="{cert_filename}"', "OK", 3000)
self.logger.info(f"[4G-TCP] AT+MSSLCERTRD=\"{cert_filename}\" response: {r}")
# 3) 引用根证书 # 3) 引用根证书
r = hardware_manager.at_client.send(f'AT+MSSLCFG="cert",{ssl_id},"{cert_filename}"', "OK", 3000) r = hardware_manager.at_client.send(f'AT+MSSLCFG="cert",{ssl_id},"{cert_filename}"', "OK", 3000)
if "OK" not in r: if "OK" not in r:
return False return False
else:
self.logger.info(f"[4G-TCP] MSSLCFG(cert) response: {r}")
# 4) 绑定 TCP 通道到 ssl_id并启用 # 4) 绑定 TCP 通道到 ssl_id并启用
r = hardware_manager.at_client.send(f'AT+MIPCFG="ssl",{link_id},{ssl_id},1', "OK", 3000) r = hardware_manager.at_client.send(f'AT+MIPCFG="ssl",{link_id},{ssl_id},1', "OK", 3000)
self.logger.info(f"[4G-TCP] AT+MIPCFG=\"ssl\",{link_id},{ssl_id},1 response: {r}") self.logger.info(f"[4G-TCP] AT+MIPCFG=\"ssl\",{link_id},{ssl_id},1 response: {r}")
if "OK" not in r: if "OK" not in r:
self.logger.error(f"[4G-TCP] MIPCFG(ssl) failed, response: {r}")
return False return False
return True return True
@@ -800,13 +948,13 @@ class NetworkManager:
Returns: Returns:
bytes: 接收到的数据,如果没有数据则返回 b"" bytes: 接收到的数据,如果没有数据则返回 b""
""" """
if not self._wifi_socket: if not wifi_manager.wifi_socket:
return b"" return b""
try: try:
# 这里保持 socket 为非阻塞模式(连接时已 setblocking(False))。 # 这里保持 socket 为非阻塞模式(连接时已 setblocking(False))。
# 不要反复 settimeout(),否则会把 socket 切回阻塞+超时,并导致 conncheck 误报 timed out。 # 不要反复 settimeout(),否则会把 socket 切回"阻塞+超时",并导致 conncheck 误报 timed out。
data = self._wifi_socket.recv(4096) # 每次最多接收4KB无数据会抛 BlockingIOError data = wifi_manager.wifi_socket.recv(4096) # 每次最多接收4KB无数据会抛 BlockingIOError
return data return data
except BlockingIOError: except BlockingIOError:
@@ -818,10 +966,10 @@ class NetworkManager:
# 关闭socket # 关闭socket
try: try:
self._wifi_socket.close() wifi_manager.wifi_socket.close()
except: except:
pass pass
self._wifi_socket = None wifi_manager.wifi_socket = None
self._tcp_connected = False self._tcp_connected = False
return b"" return b""
@@ -1150,19 +1298,19 @@ class NetworkManager:
data = self.receive_tcp_data_via_wifi(timeout_ms=50) data = self.receive_tcp_data_via_wifi(timeout_ms=50)
if data: if data:
# 将数据添加到缓冲区 # 将数据添加到缓冲区
self._wifi_recv_buffer += data wifi_manager.recv_buffer += data
# 尝试从缓冲区解析完整的数据包 # 尝试从缓冲区解析完整的数据包
while len(self._wifi_recv_buffer) >= 12: # 至少需要12字节的头部 while len(wifi_manager.recv_buffer) >= 12: # 至少需要12字节的头部
# 解析头部 # 解析头部
try: try:
body_len, msg_type, checksum = struct.unpack(">III", self._wifi_recv_buffer[:12]) body_len, msg_type, checksum = struct.unpack(">III", wifi_manager.recv_buffer[:12])
total_len = 12 + body_len total_len = 12 + body_len
if len(self._wifi_recv_buffer) >= total_len: if len(wifi_manager.recv_buffer) >= total_len:
# 有完整的数据包 # 有完整的数据包
payload = self._wifi_recv_buffer[:total_len] payload = wifi_manager.recv_buffer[:total_len]
self._wifi_recv_buffer = self._wifi_recv_buffer[total_len:] wifi_manager.recv_buffer = wifi_manager.recv_buffer[total_len:]
item = (0, payload) # link_id=0 for WiFi item = (0, payload) # link_id=0 for WiFi
break break
else: else:
@@ -1170,7 +1318,7 @@ class NetworkManager:
break break
except: except:
# 解析失败,清空缓冲区 # 解析失败,清空缓冲区
self._wifi_recv_buffer = b"" wifi_manager.recv_buffer = b""
break break
elif self._network_type == "4g": elif self._network_type == "4g":
# 4G接收数据 # 4G接收数据
@@ -1296,6 +1444,11 @@ class NetworkManager:
# 自动判断模式如果没有明确指定根据WiFi连接状态和凭证决定 # 自动判断模式如果没有明确指定根据WiFi连接状态和凭证决定
if mode not in ("4g", "wifi"): if mode not in ("4g", "wifi"):
self.logger.info("ota missing mode, auto-detecting...") self.logger.info("ota missing mode, auto-detecting...")
# 若本次会话已锁定 4G则 OTA 自动也走 4G避免后续回切导致体验不一致
if self._session_force_4g:
mode = "4g"
self.logger.info("ota auto-selected: 4g (session locked on 4g)")
else:
# 只有同时满足WiFi已连接 且 提供了WiFi凭证才使用WiFi # 只有同时满足WiFi已连接 且 提供了WiFi凭证才使用WiFi
if self.is_wifi_connected() and ssid and password: if self.is_wifi_connected() and ssid and password:
mode = "wifi" mode = "wifi"

572
wifi.py Normal file
View File

@@ -0,0 +1,572 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
WiFi管理模块
提供WiFi连接、网络检测、质量监测等功能
"""
import os
import re
import socket
import threading
import time as std_time
from maix import time
import config
from logger_manager import logger_manager
class WiFiManager:
"""WiFi管理器单例"""
_instance = None
def __new__(cls):
if cls._instance is None:
cls._instance = super(WiFiManager, cls).__new__(cls)
cls._instance._initialized = False
return cls._instance
def __init__(self):
if self._initialized:
return
# WiFi 相关状态
self._wifi_connected = False
self._wifi_ip = None
self._wifi_socket = None
self._wifi_socket_lock = threading.Lock()
self._prefer_wifi = True # 是否优先使用 WiFi
self._recv_buffer = b"" # TCP 接收缓冲区
# WiFi 质量监测(后台线程)
self._wifi_quality_monitor_thread = None
self._wifi_quality_stop_event = threading.Event()
self._last_wifi_rtt_ms = None # 最近一次测量的 RTT
self._last_wifi_rssi_dbm = None # 最近一次测量的 RSSI
# 服务器相关(用于网络检测)
try:
import archery_netcore as _netcore
self._server_ip = _netcore.get_config().get("SERVER_IP")
self._server_port = _netcore.get_config().get("SERVER_PORT")
except Exception:
self._server_ip = getattr(config, "SERVER_IP", None)
self._server_port = getattr(config, "SERVER_PORT", None)
self._initialized = True
@property
def logger(self):
"""获取 logger 对象"""
return logger_manager.logger
@property
def wifi_connected(self):
"""WiFi是否已连接"""
return self._wifi_connected
@property
def wifi_ip(self):
"""WiFi IP地址"""
return self._wifi_ip
@property
def wifi_socket(self):
"""WiFi socket对象"""
return self._wifi_socket
@wifi_socket.setter
def wifi_socket(self, value):
"""设置WiFi socket对象"""
self._wifi_socket = value
@property
def wifi_socket_lock(self):
"""获取WiFi socket锁"""
return self._wifi_socket_lock
@property
def prefer_wifi(self):
"""是否优先使用WiFi"""
return self._prefer_wifi
@prefer_wifi.setter
def prefer_wifi(self, value):
"""设置是否优先使用WiFi"""
self._prefer_wifi = value
@property
def last_wifi_rtt_ms(self):
"""最近一次测量的RTT"""
return self._last_wifi_rtt_ms
@property
def last_wifi_rssi_dbm(self):
"""最近一次测量的RSSI"""
return self._last_wifi_rssi_dbm
@property
def recv_buffer(self):
"""TCP接收缓冲区"""
return self._recv_buffer
@recv_buffer.setter
def recv_buffer(self, value):
"""设置TCP接收缓冲区"""
self._recv_buffer = value
# ==================== WiFi 连接方法 ====================
def is_wifi_connected(self):
"""检查WiFi是否已连接"""
# 优先用 MaixPy network如果可用
try:
from maix import network
wlan = network.WLAN(network.TYPE_WIFI)
if wlan.isconnected():
self._wifi_connected = True
return True
except:
self.logger.warning("Failed to check WiFi connection using MaixPy network", exc_info=True)
# 兜底:看系统 wlan0 有没有 IP
try:
ip = os.popen("ifconfig wlan0 2>/dev/null | grep 'inet ' | awk '{print $2}'").read().strip()
if ip:
self._wifi_connected = True
self._wifi_ip = ip
return True
except:
self.logger.warning("Failed to check WiFi connection using system command", exc_info=True)
self._wifi_connected = False
return False
def connect_wifi(self, ssid, password, verify_callback=None, persist=True, timeout_s=20):
"""
连接 Wi-Fi先用新凭证尝试连接并验证可用性失败自动回滚成功后再决定是否落盘
重要:系统的 /etc/init.d/S30wifi 通常会读取 /boot/wifi.ssid 与 /boot/wifi.pass 来连接 WiFi。
因此要"真正尝试连接新 WiFi",必须临时写入 /boot/ 触发重启;若失败则把旧值写回去(回滚)。
Args:
ssid: WiFi SSID
password: WiFi密码
verify_callback: 验证回调函数,接收 (ip) 参数,返回 (success: bool, error: str)
persist: 是否持久化保存凭证
timeout_s: 连接超时时间(秒)
Returns:
(ip, error): IP地址和错误信息成功时error为None
"""
# 配置文件路径定义
conf_path = "/etc/wpa_supplicant.conf"
ssid_file = "/boot/wifi.ssid"
pass_file = "/boot/wifi.pass"
def _read_text(path: str):
try:
if os.path.exists(path):
with open(path, "r", encoding="utf-8") as f:
return f.read()
except Exception:
return None
return None
def _write_text(path: str, content: str):
with open(path, "w", encoding="utf-8") as f:
f.write(content)
def _restore_boot(old_ssid: str | None, old_pass: str | None):
# 还原 /boot 凭证:原来没有就删除,原来有就写回
try:
if old_ssid is None:
if os.path.exists(ssid_file):
os.remove(ssid_file)
else:
_write_text(ssid_file, old_ssid)
except Exception:
pass
try:
if old_pass is None:
if os.path.exists(pass_file):
os.remove(pass_file)
else:
_write_text(pass_file, old_pass)
except Exception:
pass
old_conf = _read_text(conf_path)
old_boot_ssid = _read_text(ssid_file)
old_boot_pass = _read_text(pass_file)
try:
# 生成 wpa_supplicant 配置(写 /etc 作为辅助,具体是否生效取决于 S30wifi 脚本)
net_conf = os.popen(f'wpa_passphrase "{ssid}" "{password}"').read()
if "network={" not in net_conf:
raise RuntimeError("Failed to generate wpa config")
try:
_write_text(
conf_path,
"ctrl_interface=/var/run/wpa_supplicant\n"
"update_config=1\n\n"
+ net_conf,
)
except Exception:
# 不强制要求写 /etc 成功(某些系统只用 /boot
pass
# ====== 临时写入 /boot 凭证,触发 WiFi 服务真正尝试连接新 SSID ======
_write_text(ssid_file, ssid.strip())
_write_text(pass_file, password.strip())
# 重启 Wi-Fi 服务
os.system("/etc/init.d/S30wifi restart")
# 等待获取 IP
wait_s = int(timeout_s) if timeout_s and timeout_s > 0 else 20
wait_s = min(max(wait_s, 5), 60)
for _ in range(wait_s):
ip = os.popen("ifconfig wlan0 2>/dev/null | grep 'inet ' | awk '{print $2}'").read().strip()
if ip:
# 拿到 IP 不代表可上网/可访问目标;继续做可达性验证
self._wifi_connected = True
self._wifi_ip = ip
self.logger.info(f"[WIFI] 已连接IP: {ip},开始验证网络可用性...")
# 验证能访问指定目标(通过回调函数)
if verify_callback:
success, error = verify_callback(ip)
if not success:
raise RuntimeError(error or "Verification failed")
# ====== 验证通过 ======
if not persist:
# 不持久化:把 /boot 恢复成旧值(不重启,当前连接保持不变)
_restore_boot(old_boot_ssid, old_boot_pass)
self.logger.info("[WIFI] 网络验证通过,但按 persist=False 回滚 /boot 凭证(不重启)")
else:
self.logger.info("[WIFI] 网络验证通过,/boot 凭证已保留(持久化)")
return ip, None
std_time.sleep(1)
raise RuntimeError("Timeout: No IP obtained")
except Exception as e:
# 失败:回滚 /boot 和 /etc重启 WiFi 恢复旧网络
_restore_boot(old_boot_ssid, old_boot_pass)
try:
if old_conf is not None:
_write_text(conf_path, old_conf)
except Exception:
pass
try:
os.system("/etc/init.d/S30wifi restart")
except Exception:
pass
self._wifi_connected = False
self._wifi_ip = None
self.logger.error(f"[WIFI] 连接/验证失败,已回滚: {e}")
return None, str(e)
def disconnect_wifi(self):
"""断开WiFi连接并清理资源"""
if self._wifi_socket:
try:
self._wifi_socket.close()
except Exception:
pass
finally:
self._wifi_socket = None
self._wifi_connected = False
self._wifi_ip = None
# ==================== WiFi 质量监测 ====================
def _get_wifi_rssi_dbm(self):
"""
获取 WiFi 信号强度dBm越大越好比如 -40 比 -80 好)
由于不同固件实现差异,这里做多策略兜底,失败返回 None
"""
# 1) 优先使用iw dev wlan0 link
# 你提供的输出示例包含signal: -58 dBm
try:
out = os.popen("iw dev wlan0 link 2>/dev/null").read()
if out:
m = re.search(r"signal:\s*(-?\d+(?:\.\d+)?)\s*dBm", out, re.IGNORECASE)
if m:
v = float(m.group(1))
# 合理范围兜底
if -120.0 <= v <= 0.0:
return v
m2 = re.search(r"signal:\s*(-?\d+(?:\.\d+)?)", out, re.IGNORECASE)
if m2:
v = float(m2.group(1))
if -120.0 <= v <= 0.0:
return v
except Exception:
pass
# 2) 兜底iwconfig
try:
out = os.popen("iwconfig wlan0 2>/dev/null").read()
m = re.search(r"Signal level[=:]\s*(-?\d+(?:\.\d+)?)\s*dBm", out, re.IGNORECASE)
if m:
v = float(m.group(1))
if -120.0 <= v <= 0.0:
return v
m2 = re.search(r"Signal level[=:]\s*(-?\d+(?:\.\d+)?)", out, re.IGNORECASE)
if m2:
v = float(m2.group(1))
if -120.0 <= v <= 0.0:
return v
except Exception:
pass
return None
def _measure_wifi_tcp_rtt_ms(self, host, port, samples=3, per_sample_timeout_ms=900):
"""
测量:在当前 WiFi 下TCP 建连耗时RTT 的近似)
Args:
host: 目标主机
port: 目标端口
samples: 采样次数
per_sample_timeout_ms: 每次采样超时时间(毫秒)
Returns:
(median_rtt_ms, reachable_bool)
"""
rtts = []
reachable = False
addr = None
# 先解析一次地址,避免每次样本都做 DNS
try:
addr_info = socket.getaddrinfo(host, port)[0]
addr = (addr_info[0], addr_info[1], addr_info[2], addr_info[-1])
except Exception:
return float("inf"), False
for _ in range(max(1, int(samples or 1))):
s = None
try:
s = socket.socket(addr[0], addr[1], addr[2])
s.settimeout(max(0.1, float(per_sample_timeout_ms) / 1000.0))
t0 = time.ticks_ms()
s.connect(addr[-1])
elapsed_ms = abs(time.ticks_diff(time.ticks_ms(), t0))
rtts.append(float(elapsed_ms))
reachable = True
except Exception:
# 单个样本失败不影响整体,只要有成功样本就继续
pass
finally:
try:
if s:
s.close()
except Exception:
pass
# 小间隔,避免过度占用
try:
time.sleep_ms(100)
except Exception:
pass
if not rtts:
return float("inf"), False
rtts_sorted = sorted(rtts)
mid = len(rtts_sorted) // 2
if len(rtts_sorted) % 2 == 1:
median = rtts_sorted[mid]
else:
median = (rtts_sorted[mid - 1] + rtts_sorted[mid]) / 2.0
return median, reachable
def _is_wifi_quality_bad(self, wifi_rtt_ms, wifi_rssi_dbm):
"""
综合判断 WiFi 质量是否差:
- RTT中位数超过阈值 -> bad
- 若启用 RSSI信号弱(RSSI更差于阈值) 且 RTT 也偏高 -> bad
"""
if wifi_rtt_ms >= config.WIFI_QUALITY_RTT_BAD_MS:
return True
if not getattr(config, "WIFI_QUALITY_USE_RSSI", False):
return False
if wifi_rssi_dbm is None:
return False
# "rtt_warn + rssi_bad" 联合条件
if wifi_rtt_ms >= config.WIFI_QUALITY_RTT_WARN_MS and wifi_rssi_dbm <= config.WIFI_QUALITY_RSSI_BAD_DBM:
return True
return False
def get_wifi_quality_status(self):
"""
获取当前 WiFi 质量状态(用于调试或显示)
Returns:
dict: {"rtt_ms": float, "rssi_dbm": float, "is_bad": bool}
"""
rtt = self._last_wifi_rtt_ms
rssi = self._last_wifi_rssi_dbm
is_bad = False
if rtt is not None and rtt != float("inf"):
is_bad = self._is_wifi_quality_bad(rtt, rssi)
return {
"rtt_ms": rtt if rtt is not None and rtt != float("inf") else None,
"rssi_dbm": rssi,
"is_bad": is_bad
}
# ==================== 后台质量监测线程 ====================
def start_quality_monitor(self, network_type_callback, on_poor_quality_callback):
"""
启动 WiFi 质量后台监测线程(每 5 秒测量一次 RTT 和 RSSI
只在 WiFi 连接时运行,不影响业务发送性能
Args:
network_type_callback: 获取当前网络类型的回调函数
on_poor_quality_callback: WiFi质量差时的回调函数
"""
if self._wifi_quality_monitor_thread is not None:
self.logger.warning("[WiFi Monitor] 监测线程已在运行")
return
self._network_type_callback = network_type_callback
self._on_poor_quality_callback = on_poor_quality_callback
self._wifi_quality_stop_event.clear()
self._wifi_quality_monitor_thread = threading.Thread(
target=self._quality_monitor_loop,
daemon=True,
name="wifi_quality_monitor"
)
self._wifi_quality_monitor_thread.start()
self.logger.info("[WiFi Monitor] 已启动后台监测线程")
def stop_quality_monitor(self):
"""停止 WiFi 质量监测线程"""
if self._wifi_quality_monitor_thread is None:
return
self._wifi_quality_stop_event.set()
try:
self._wifi_quality_monitor_thread.join(timeout=2.0)
except Exception as e:
self.logger.error(f"[WiFi Monitor] 停止线程失败:{e}")
finally:
self._wifi_quality_monitor_thread = None
self.logger.info("[WiFi Monitor] 已停止后台监测线程")
def _quality_monitor_loop(self):
"""
WiFi 质量监测循环(后台线程)
每 5 秒测量一次 RTT 和 RSSI发现质量差则触发切换
"""
while not self._wifi_quality_stop_event.is_set():
try:
# 只在 WiFi 连接时才测量
network_type = self._network_type_callback()
if network_type == "wifi" and self._wifi_socket:
# 测量 RTT1 个样本,快速测量)
rtt_ms, reachable = self._measure_wifi_tcp_rtt_ms(
self._server_ip, self._server_port,
samples=1, per_sample_timeout_ms=600
)
# 获取 RSSI
rssi_dbm = self._get_wifi_rssi_dbm()
# 更新缓存
self._last_wifi_rtt_ms = rtt_ms if reachable else None
self._last_wifi_rssi_dbm = rssi_dbm
self.logger.debug(f"[WiFi Monitor] - RTT={rtt_ms:.0f}ms, RSSI={rssi_dbm:.0f}dBm")
# 判断质量是否差(切换前做 2 次快速复测,防止瞬时抖动)
def _is_bad_now(_reachable, _rtt, _rssi):
if (not _reachable) or (_rtt is None) or (_rtt == float("inf")):
return True
return self._is_wifi_quality_bad(_rtt, _rssi)
bad = _is_bad_now(reachable, rtt_ms, rssi_dbm)
if bad:
self.logger.warning("[WiFi Monitor] 质量差,切换前快速重试 2 次每次间隔1秒")
for retry_idx in range(2):
time.sleep_ms(1000)
rtt2, reachable2 = self._measure_wifi_tcp_rtt_ms(
self._server_ip, self._server_port,
samples=1, per_sample_timeout_ms=600
)
rssi2 = self._get_wifi_rssi_dbm()
# 更新缓存,便于外部查看最新状态
self._last_wifi_rtt_ms = rtt2 if reachable2 else None
self._last_wifi_rssi_dbm = rssi2
bad2 = _is_bad_now(reachable2, rtt2, rssi2)
try:
self.logger.info(
f"[WiFi Monitor] 复测{retry_idx+1}/2: reachable={reachable2}, "
f"rtt={rtt2 if rtt2 != float('inf') else -1:.0f}ms, rssi={rssi2}, bad={bad2}"
)
except Exception:
pass
if not bad2:
self.logger.info("[WiFi Monitor] 复测恢复正常,继续保留 WiFi不切换")
bad = False
break
if bad:
self.logger.warning("[WiFi Monitor] 复测仍差/不通,尝试切换到 4G")
self._on_poor_quality_callback()
# 休眠 5 秒
time.sleep(5)
except Exception as e:
self.logger.error(f"[WiFi Monitor] 监测异常:{e}")
# 异常后继续循环,避免线程退出
continue
# 全局 WiFi 管理器实例
wifi_manager = WiFiManager()
# ==================== 兼容旧接口的函数 ====================
def is_wifi_connected():
"""尽量判断当前是否有 Wi-Fi有则走 Wi-Fi OTA否则走 4G OTA"""
return wifi_manager.is_wifi_connected()
def connect_wifi(ssid, password, verify_callback=None, persist=True, timeout_s=20):
"""
连接 Wi-Fi 并将凭证持久化保存到 /boot/ 目录,
以便设备重启后自动连接。
Args:
ssid: WiFi SSID
password: WiFi密码
verify_callback: 验证回调函数,接收 (ip) 参数,返回 (success: bool, error: str)
persist: 是否持久化保存
timeout_s: 超时时间(秒)
Returns:
(ip, error): IP地址和错误信息成功时error为None
"""
return wifi_manager.connect_wifi(ssid, password, verify_callback, persist, timeout_s)