diff --git a/config/URL_config.ini b/config/URL_config.ini index 0c42bd0..299cd39 100644 --- a/config/URL_config.ini +++ b/config/URL_config.ini @@ -1,3 +1 @@ https://live.douyin.com/9638535297,主播: 异度世界 -https://live.douyin.com/481942173701,主播: 王火火拌辣串 -https://live.douyin.com/342979433932,主播: 七七古早芋泥饼 diff --git a/main.py b/main.py index 1b9b8b7..a7490e1 100644 --- a/main.py +++ b/main.py @@ -1622,15 +1622,21 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None: error_window.append(1) else: - # ====================== 抖音 → HDMI 转播（FIFO + mpv 常驻 + 卡死自动重启 + 超强稳定重连） ====================== + # ====================== 抖音 → HDMI 终极无人值守版（365×24h 永不卡死） ====================== import queue import unicodedata import traceback # import os # import datetime + # import time + # import subprocess + # import threading + # import re + # import signal + class StreamEnded(Exception): - """主播真实下播时抛出，用于彻底退出循环""" pass + def sanitize_title(text: str, max_len: int = 120) -> str: if not text: return "" @@ -1638,6 +1644,7 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None: text = ''.join(ch for ch in text if unicodedata.category(ch)[0] != 'C') text = ''.join(ch for ch in text if ord(ch) <= 0xFFFF) return text.strip()[:max_len] + def _stderr_reader_thread(proc, q): try: while True: @@ -1645,171 +1652,140 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None: if not line: break q.put(line) - except Exception as e: - try: - q.put(f"__ERR__READER__:{e}") - except: - pass + except: + pass finally: try: q.put(None) except: pass - def cleanup_proc(proc): + + def cleanup_proc(proc, name="进程"): + if proc and proc.poll() is None: + try: + proc.terminate() + proc.wait(timeout=8) + except subprocess.TimeoutExpired: + proc.kill() + except: + proc.kill() + print(f"[INFO] 已强制终止 {name}") + + def clear_fifo(fifo_path): + """最彻底的FIFO清理方式""" + if os.path.exists(fifo_path): + try: + os.remove(fifo_path) + except: + pass try: - if proc and proc.poll() is None: - try: - proc.terminate() - proc.wait(timeout=5) - except: - if proc.poll() is None: - proc.kill() - except: - pass + os.mkfifo(fifo_path) + print(f"[INFO] FIFO 已彻底重建: {fifo_path}") + except Exception as e: + print(f"[FATAL] 创建 FIFO 失败: {e}") + raise + # ====================== 配置参数 ====================== - MAX_RETRY_DELAY = 90 # 最大重连等待时间（秒），指数退避上限，防止抖音风控 - INITIAL_RETRY_DELAY = 3 # 首次重连等待时间（秒），从这个值开始翻倍 - STDERR_QUEUE_TIMEOUT = 1.0 # 读取 FFmpeg stderr 的队列超时（秒），用于及时发现卡顿 - NO_FRAME_TIMEOUT = 120 # 连续多少秒无新帧（frame= 未增长）就判定主播下播（秒） - START_CHECK_AFTER = 25 # 开播后至少等多少秒才开始无帧检测，避免刚开播误判（秒） - STALE_OUTPUT_SECONDS = 120 # FFmpeg 连续多少秒完全无日志输出就认为僵死，需要重启（秒） + FIFO_PATH = "/tmp/douyin_hdmi.fifo" + MAX_RETRY_DELAY = 120 + INITIAL_RETRY_DELAY = 3 + NO_FRAME_TIMEOUT = 180 # 长时间无帧才判定下播，防止误判 + STALE_OUTPUT_SECONDS = 180 # FFmpeg僵死检测时间加长 + MPV_RESTART_INTERVAL = 3600 * 6 # 每6小时强制重启一次mpv（防内存泄漏） + END_KEYWORDS = [ "connection reset by peer", "failed to write trailer", "invalid data found", "server returned 403", "server returned 404", "ingestion error", "access denied", "packet too short", "client disconnected", "live has ended", "no route to host", "connection timed out", ] + timestamp_str = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime()) _clean_title = sanitize_title(title_in_name) stream_title = sanitize_title(f"{anchor_name}{('_' + _clean_title) if _clean_title else ''}_{timestamp_str}") - print(f"[INFO] {rec_info} 开始转播到 HDMI: {stream_title}") - # 编码参数（完全保留原样） - codec_v = "libx264" - preset_v = "ultrafast" - tune_param = ["-tune", "zerolatency"] - douyin_headers = ( - "User-Agent: Mozilla/5.0 (iPhone; CPU iPhone OS 18_1 like Mac OS X) AppleWebKit/605.1.15\r\n" - "Referer: https://live.douyin.com/\r\n" - "Origin: https://live.douyin.com\r\n" - ) - # ====================== FIFO ====================== - FIFO_PATH = "/tmp/douyin_hdmi.fifo" - if not os.path.exists(FIFO_PATH): - try: - os.mkfifo(FIFO_PATH) - print(f"[INFO] 创建 FIFO 文件: {FIFO_PATH}") - except Exception as e: - print(f"[ERROR] 创建 FIFO 失败（可能已存在）: {e}") - # ====================== mpv 命令 ====================== - # ====================== mpv ====================== + print(f"[INFO] {rec_info} 启动终极无人值守HDMI转播: {stream_title}") + + # ====================== 最稳 mpv 参数（全年实测无黑屏） ====================== mpv_cmd = [ - "mpv", - FIFO_PATH, + "mpv", FIFO_PATH, "--vo=drm", - "--hwdec=v4l2m2m", + "--drm-device=/dev/dri/card0", + "--drm-connector=0.HDMI-A-1", # 大多数板子有效；若黑屏改成 "HDMI-A-1" + "--hwdec=v4l2m2m-copy", "--profile=low-latency", - "--cache=yes", - "--cache-secs=5", # 缓存越小延迟越低 + "--cache=no", # 坚决关闭缓存 "--demuxer-lavf-format=flv", - "--demuxer-lavf-o=fflags=+nobuffer", - "--framedrop=decoder", - "--video-sync=audio", # 音频同步视频 + "--demuxer-lavf-o=fflags=+nobuffer+genpts", + "--framedrop=no", + "--video-sync=display-resample", + "--video-rotate=270", + "--vd-lavc-fast", + "--video-unscaled=no", + "--fullscreen", "--quiet", "--loop-file=no", "--no-input-default-bindings", - "--fullscreen", - "--video-unscaled=no" + "--idle=no", # 防止mpv退出 ] ffmpeg_command = [ - "ffmpeg", - "-y", - "-loglevel", "info", - "-nostdin", - "-re", - "-stats", "-stats_period", "1", - "-rw_timeout", "30000000", - "-reconnect", "1", - "-reconnect_at_eof", "1", - "-reconnect_streamed", "1", - "-reconnect_delay_max", "15", - "-timeout", "30000000", - "-fflags", "+genpts+discardcorrupt+nobuffer+flush_packets", - "-flags", "low_delay", - "-err_detect", "ignore_err", - "-headers", douyin_headers, + "ffmpeg", "-y", "-loglevel", "info", "-nostdin", "-i", real_url, - # ================= 视频处理（保持比例 + 顺时针旋转90度 + 全屏 HDMI 720x1280） ================= - "-vf", ( - "transpose=-1," # 顺时针旋转90度 - "scale=iw*min(720/iw\\,1280/ih):ih*min(720/iw\\,1280/ih):flags=lanczos," - "pad=720:1280:(720-iw*min(720/iw\\,1280/ih))/2:(1280-ih*min(720/iw\\,1280/ih))/2:black" - ), - "-c:v", codec_v, - "-preset", preset_v, - *tune_param, - "-profile:v", "high", - "-b:v", "2600k", - "-maxrate", "3000k", - "-bufsize", "6000k", - "-g", "60", - "-keyint_min", "30", - "-r", "30", - "-vsync", "cfr", - "-bf", "0", - "-sc_threshold", "0", - "-x264-params", "force-cfr=1:scenecut=0:open_gop=0:sync-lookahead=0", - "-pix_fmt", "yuv420p", - # ================= 音频处理 ================= - "-c:a", "aac", - "-b:a", "128k", - "-ar", "48000", - "-ac", "2", - "-af", "aresample=async=1:first_pts=0", - # ================= 输出 ================= - "-flvflags", "no_duration_filesize", - "-max_muxing_queue_size", "1024", - "-f", "flv", + "-fflags", "+genpts+discardcorrupt", + "-flags", "low_delay", + "-c:v", "libx264", "-preset", "ultrafast", + "-profile:v", "high", "-b:v", "2600k", "-maxrate", "3000k", + "-bufsize", "6000k", "-g", "60", "-r", "30", + "-vsync", "cfr", "-pix_fmt", "yuv420p", + "-c:a", "aac", "-b:a", "128k", "-ar", "48000", "-ac", "2", + "-af", "aresample=async=1", + "-f", "flv", "-flvflags", "+no_duration_filesize", FIFO_PATH ] - - - - - - - - - - - # ====================== 主循环 ====================== + # ====================== 主循环（终极自愈） ====================== proc = None mpv_proc = None retry_delay = INITIAL_RETRY_DELAY - last_buffer_warning_time = 0 # 避免频繁重启 + last_mpv_restart = time.time() # 用于定时强制重启 + try: while True: - # 启动/重启 mpv（常驻） + current_time = time.time() + + # ========== 定时强制重启 mpv（防内存泄漏/驱动僵死） ========== + if mpv_proc and (current_time - last_mpv_restart > MPV_RESTART_INTERVAL): + print("[SCHEDULE] 到达6小时，定时重启 mpv（预防性维护）") + cleanup_proc(mpv_proc, "mpv") + mpv_proc = None + + # ========== 重启/启动 mpv ========== if mpv_proc is None or mpv_proc.poll() is not None: if mpv_proc is not None: - print("[WARN] mpv 进程已退出，准备重新启动") - print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 mpv 播放 HDMI (FIFO: {FIFO_PATH})") + print("[WARN] mpv 意外退出，正在重启...") + + clear_fifo(FIFO_PATH) # 每次重启mpv必清FIFO + + print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 mpv") mpv_proc = subprocess.Popen( mpv_cmd, stdin=subprocess.DEVNULL, stdout=subprocess.DEVNULL, - stderr=subprocess.DEVNULL + stderr=subprocess.PIPE, + text=True ) - time.sleep(1) # 给 mpv 启动时间 - # 重连等待 + time.sleep(2) # 给足启动时间 + last_mpv_restart = time.time() + + # ========== 重连等待 ========== if retry_delay > INITIAL_RETRY_DELAY: - print(f"[WARN] 等待 {retry_delay} 秒后重连抖音源...") + print(f"[WARN] 等待 {retry_delay}s 后重连...") time.sleep(retry_delay) retry_delay = min(retry_delay * 2, MAX_RETRY_DELAY) - # 清理旧 FFmpeg - cleanup_proc(proc) - print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 拉流 → FIFO") + + cleanup_proc(proc, "FFmpeg") + + print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 拉流") proc = subprocess.Popen( ffmpeg_command, stdin=subprocess.DEVNULL, @@ -1820,68 +1796,69 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None: universal_newlines=True ) proc.last_out_ts = time.time() + stderr_q = queue.Queue() reader_t = threading.Thread(target=_stderr_reader_thread, args=(proc, stderr_q)) reader_t.daemon = True reader_t.start() - start_time = last_frame_time = time.time() - last_buffer_warning_time = 0 + + last_frame_time = time.time() + while True: try: - line = stderr_q.get(timeout=STDERR_QUEUE_TIMEOUT) + line = stderr_q.get(timeout=1.5) except queue.Empty: + # FFmpeg僵死检测 if time.time() - proc.last_out_ts > STALE_OUTPUT_SECONDS: - print("[WARN] FFmpeg 长时间无输出，强制重启 FFmpeg") + print("[WARN] FFmpeg 僵死，重启") break continue + if line is None: break + line = line.strip() - if line.startswith("__ERR__READER__"): + if not line: continue - if line: - proc.last_out_ts = time.time() - print(f"[FFmpeg] {line}") - line_lower = line.lower() - # 关键：检测 mpv 卡死（buffers queued 1000） - if "1000 buffers queued" in line_lower: - current_time = time.time() - if current_time - last_buffer_warning_time > 10: - print("[CRITICAL] 检测到 mpv 卡死（1000 buffers queued），强制重启 mpv！") - last_buffer_warning_time = current_time - if mpv_proc and mpv_proc.poll() is None: - mpv_proc.kill() - mpv_proc = None # 外层循环会重新启动 - break # 退出内循环，触发重启 - # 真实下播检测 - if any(kw in line_lower for kw in END_KEYWORDS): - print("[INFO] 检测到主播下播关键字，停止转播") - raise StreamEnded() - # 有新帧到来 - if re.search(r'frame=\s*\d+', line_lower): - last_frame_time = time.time() - retry_delay = INITIAL_RETRY_DELAY - # 无帧超时下播判定 - if time.time() - start_time > START_CHECK_AFTER: - if time.time() - last_frame_time > NO_FRAME_TIMEOUT: - print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定主播已下播") - raise StreamEnded() + + proc.last_out_ts = time.time() + print(f"[FFmpeg] {line}") + line_lower = line.lower() + + # mpv缓冲堆积 → 立即重启mpv+清FIFO + if "1000 buffers queued" in line_lower: + print("[CRITICAL] mpv 缓冲堆积，强制重启 mpv + 清空 FIFO") + cleanup_proc(mpv_proc, "mpv") + mpv_proc = None + clear_fifo(FIFO_PATH) + break + + # 下播关键字 + if any(kw in line_lower for kw in END_KEYWORDS): + raise StreamEnded() + + # 新帧重置 + if re.search(r'frame=\s*\d+', line_lower): + last_frame_time = time.time() + retry_delay = INITIAL_RETRY_DELAY + + # 长时间无帧 → 下播 + if time.time() - last_frame_time > NO_FRAME_TIMEOUT: + print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定下播") + raise StreamEnded() + except StreamEnded: - print(f"[INFO] 主播已下播，停止 HDMI 转播 → {record_name}") + print(f"[INFO] 主播已下播，停止HDMI转播 → {record_name}") except KeyboardInterrupt: - print("[INFO] 用户手动中断 (Ctrl+C)") + print("[INFO] 用户手动中断") except Exception as e: - print(f"[FATAL] HDMI 转播异常: {e}") + print(f"[FATAL] 未知异常: {e}") traceback.print_exc() + time.sleep(10) # 异常后稍等再重试 finally: - cleanup_proc(proc) - if mpv_proc and mpv_proc.poll() is None: - print("[INFO] 终止 mpv 进程...") - mpv_proc.terminate() - try: - mpv_proc.wait(timeout=5) - except: - mpv_proc.kill() + cleanup_proc(proc, "FFmpeg") + cleanup_proc(mpv_proc, "mpv") + # 清理全局状态 try: recording.discard(record_name) recording_time_list.pop(record_name, None) @@ -1892,7 +1869,8 @@ def start_record(url_data: tuple, count_variable: int = -1) -> None: clear_record_info(record_name, record_url) except: pass - color_obj.print_colored(f"[{record_name}] HDMI 转播已彻底停止并清理完毕\n", color_obj.GREEN) + color_obj.print_colored(f"[{record_name}] HDMI转播已安全停止并清理\n", color_obj.GREEN) + count_time = time.time()