diff --git a/douyin_srs_ffplay.py b/douyin_srs_ffplay.py new file mode 100644 index 0000000..214c187 --- /dev/null +++ b/douyin_srs_ffplay.py @@ -0,0 +1,230 @@ +# douyin_srs_ffplay.py + +import queue +import unicodedata +import traceback +import subprocess +import threading +import re +import time +import datetime + +class StreamEnded(Exception): + pass + +def sanitize_title(text: str, max_len: int = 120) -> str: + if not text: + return "" + text = unicodedata.normalize("NFKC", text) + 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: + line = proc.stderr.readline() + if not line: + break + q.put(line) + except: + pass + finally: + try: + q.put(None) + except: + pass + +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 start_douyin_srs_ffplay( + title, + anchor_name, + real_url, + rec_info, + record_name, + recording, + recording_time_list, + running_list, + monitoring, + clear_record_info, + color_obj +): + # ====================== 配置参数 ====================== + SRS_RTMP_URL = "rtmp://127.0.0.1/live/douyin" + SRS_PLAY_URL = "http://127.0.0.1:8080/live/douyin.flv" + + MAX_RETRY_DELAY = 120 + INITIAL_RETRY_DELAY = 3 + NO_FRAME_TIMEOUT = 180 + STALE_OUTPUT_SECONDS = 180 + FFPLAY_RESTART_INTERVAL = 3600 * 6 + + 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) + stream_title = sanitize_title(f"{anchor_name}{('_' + _clean_title) if _clean_title else ''}_{timestamp_str}") + print(f"[INFO] {rec_info} 启动 SRS → ffplay HDMI 转播（低CPU原画版）: {stream_title}") + + 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" + ) + + # ====================== FFmpeg 推流到 SRS（直抄原画，低CPU） ====================== + ffmpeg_command = [ + "ffmpeg", "-y", "-loglevel", "info", "-nostdin", + "-headers", douyin_headers, + "-rw_timeout", "30000000", + "-reconnect", "1", "-reconnect_at_eof", "1", "-reconnect_streamed", "1", + "-reconnect_delay_max", "7", + "-fflags", "+genpts+discardcorrupt+igndts", + "-err_detect", "ignore_err", + "-i", real_url, + "-c:v", "copy", + "-c:a", "copy", + "-f", "flv", SRS_RTMP_URL + ] + + # ====================== ffplay 最佳参数（低CPU + 零漂移 + 全屏） ====================== + ffplay_cmd = [ + "ffplay", + "-fflags", "nobuffer+genpts", + "-flags", "low_delay", + "-framedrop", + "-sync", "ext", + "-vf", "transpose=2", + "-autoexit", + "-fs", + "-loglevel", "quiet", + "-i", SRS_PLAY_URL + ] + + ffmpeg_proc = None + ffplay_proc = None + stderr_q = queue.Queue() + reader_t = None + retry_delay = INITIAL_RETRY_DELAY + last_ffplay_restart = time.time() + + try: + while True: + current_time = time.time() + + if ffplay_proc and (current_time - last_ffplay_restart > FFPLAY_RESTART_INTERVAL): + print("[SCHEDULE] 6小时预防性重启 ffplay") + cleanup_proc(ffplay_proc, "ffplay") + ffplay_proc = None + + if ffplay_proc is None or ffplay_proc.poll() is not None: + if ffplay_proc is not None: + print("[WARN] ffplay 退出，重启中...") + + print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 ffplay 播放 SRS 原画流") + ffplay_proc = subprocess.Popen( + ffplay_cmd, + stdin=subprocess.DEVNULL, + stdout=subprocess.DEVNULL, + stderr=subprocess.PIPE, + text=True + ) + time.sleep(2) + last_ffplay_restart = time.time() + + if retry_delay > INITIAL_RETRY_DELAY: + print(f"[WARN] 等待 {retry_delay}s 后重连抖音源...") + time.sleep(retry_delay) + retry_delay = min(retry_delay * 2, MAX_RETRY_DELAY) + + cleanup_proc(ffmpeg_proc, "FFmpeg") + + print(f"[INFO][{datetime.datetime.now():%H:%M:%S}] 启动 FFmpeg 直抄推流到 SRS") + ffmpeg_proc = subprocess.Popen( + ffmpeg_command, + stdin=subprocess.DEVNULL, + stdout=subprocess.DEVNULL, + stderr=subprocess.PIPE, + text=True, + bufsize=1, + universal_newlines=True + ) + ffmpeg_proc.last_out_ts = time.time() + + stderr_q = queue.Queue() + reader_t = threading.Thread(target=_stderr_reader_thread, args=(ffmpeg_proc, stderr_q)) + reader_t.daemon = True + reader_t.start() + + last_frame_time = time.time() + + while True: + try: + line = stderr_q.get(timeout=1.5) + except queue.Empty: + if time.time() - ffmpeg_proc.last_out_ts > STALE_OUTPUT_SECONDS: + print("[WARN] FFmpeg 僵死，重启") + break + continue + + if line is None: + break + + line = line.strip() + if not line: + continue + + ffmpeg_proc.last_out_ts = time.time() + print(f"[FFmpeg] {line}") + line_lower = line.lower() + + 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() - last_frame_time > NO_FRAME_TIMEOUT: + print(f"[INFO] {NO_FRAME_TIMEOUT}s 无新帧，判定下播") + raise StreamEnded() + + except StreamEnded: + print(f"[INFO] 主播已下播，停止 SRS → ffplay HDMI 转播 → {record_name}") + except KeyboardInterrupt: + print("[INFO] 用户手动中断") + except Exception as e: + print(f"[FATAL] 未知异常: {e}") + traceback.print_exc() + time.sleep(10) + finally: + cleanup_proc(ffmpeg_proc, "FFmpeg") + cleanup_proc(ffplay_proc, "ffplay") + try: + recording.discard(record_name) + recording_time_list.pop(record_name, None) + if record_name in running_list: + running_list.remove(record_name) + # global monitoring + # monitoring = max(0, monitoring - 1) + monitoring[0] = max(0, monitoring[0] - 1) # 修改为列表元素 + clear_record_info(record_name, real_url) + except: + pass + color_obj.print_colored(f"[{record_name}] SRS → ffplay HDMI 转播已彻底停止并清理完毕\n", color_obj.GREEN) \ No newline at end of file