From 00de4511009a7fbbb88142b2b2a0faeaa0e3e7d9 Mon Sep 17 00:00:00 2001
From: zhaomingwork <zhaomingwork@qq.com>
Date: 星期一, 15 五月 2023 22:05:26 +0800
Subject: [PATCH] add ws_server_online support wss

---
 funasr/runtime/python/websocket/ws_server_online.py |   13 +++++++++++--
 1 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/funasr/runtime/python/websocket/ws_server_online.py b/funasr/runtime/python/websocket/ws_server_online.py
index a35b127..ba50f62 100644
--- a/funasr/runtime/python/websocket/ws_server_online.py
+++ b/funasr/runtime/python/websocket/ws_server_online.py
@@ -7,7 +7,7 @@
 import logging
 import tracemalloc
 import numpy as np
-
+import ssl
 from parse_args import args
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
@@ -94,7 +94,16 @@
 				await websocket.send(message)
 
 
+if len(args.certfile)>0:
+  ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
 
-start_server = websockets.serve(ws_serve, args.host, args.port, subprotocols=["binary"], ping_interval=None)
+  # Generate with Lets Encrypt, copied to this location, chown to current user and 400 permissions
+  ssl_cert = args.certfile
+  ssl_key = args.keyfile
+
+  ssl_context.load_cert_chain(ssl_cert, keyfile=ssl_key)
+  start_server = websockets.serve(ws_serve, args.host, args.port, subprotocols=["binary"], ping_interval=None,ssl=ssl_context)
+else:
+  start_server = websockets.serve(ws_serve, args.host, args.port, subprotocols=["binary"], ping_interval=None)
 asyncio.get_event_loop().run_until_complete(start_server)
 asyncio.get_event_loop().run_forever()
\ No newline at end of file

--
Gitblit v1.9.1