Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,7 @@ import asyncio
|
|
3 |
import asyncssh
|
4 |
import os
|
5 |
import sys
|
|
|
6 |
|
7 |
# Класс сессии: обрабатывает подключение и ввод данных
|
8 |
class MySSHServerSession(asyncssh.SSHServerSession):
|
@@ -31,27 +32,30 @@ class MySSHServer(asyncssh.SSHServer):
|
|
31 |
return MySSHServerSession()
|
32 |
|
33 |
async def begin_auth(self, username):
|
34 |
-
# Простой способ
|
35 |
return True
|
36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
# Функция для запуска сервера
|
38 |
async def start_ssh_server():
|
39 |
# Читаем порт из переменной окружения PORT (по умолчанию 2222)
|
40 |
port = int(os.environ.get('PORT', 2222))
|
41 |
|
42 |
-
#
|
43 |
-
|
44 |
-
host_key_file = os.environ.get('HOST_KEY_FILE')
|
45 |
-
if host_key_file and os.path.exists(host_key_file):
|
46 |
-
host_keys = [host_key_file]
|
47 |
-
else:
|
48 |
-
print("Генерация временного ключа сервера...", file=sys.stderr)
|
49 |
-
host_keys = [asyncssh.generate_private_key('ssh-rsa')]
|
50 |
|
51 |
try:
|
|
|
|
|
|
|
52 |
server = await asyncssh.create_server(
|
53 |
MySSHServer, '', port,
|
54 |
-
server_host_keys=
|
55 |
)
|
56 |
except (OSError, asyncssh.Error) as exc:
|
57 |
sys.exit('Ошибка при запуске сервера: ' + str(exc))
|
|
|
3 |
import asyncssh
|
4 |
import os
|
5 |
import sys
|
6 |
+
from io import BytesIO
|
7 |
|
8 |
# Класс сессии: обрабатывает подключение и ввод данных
|
9 |
class MySSHServerSession(asyncssh.SSHServerSession):
|
|
|
32 |
return MySSHServerSession()
|
33 |
|
34 |
async def begin_auth(self, username):
|
35 |
+
# Простой способ проверки: разрешаем любые логины
|
36 |
return True
|
37 |
|
38 |
+
# Генерация временного SSH-ключа в памяти
|
39 |
+
def generate_temp_ssh_key():
|
40 |
+
private_key = asyncssh.generate_private_key('ssh-rsa')
|
41 |
+
private_key_str = private_key.private_bytes().decode('utf-8')
|
42 |
+
return private_key_str
|
43 |
+
|
44 |
# Функция для запуска сервера
|
45 |
async def start_ssh_server():
|
46 |
# Читаем порт из переменной окружения PORT (по умолчанию 2222)
|
47 |
port = int(os.environ.get('PORT', 2222))
|
48 |
|
49 |
+
# Генерация временного ключа в памяти
|
50 |
+
private_key_str = generate_temp_ssh_key()
|
|
|
|
|
|
|
|
|
|
|
|
|
51 |
|
52 |
try:
|
53 |
+
# Используем приватный ключ из памяти
|
54 |
+
private_key = asyncssh.import_private_key(private_key_str)
|
55 |
+
|
56 |
server = await asyncssh.create_server(
|
57 |
MySSHServer, '', port,
|
58 |
+
server_host_keys=[private_key]
|
59 |
)
|
60 |
except (OSError, asyncssh.Error) as exc:
|
61 |
sys.exit('Ошибка при запуске сервера: ' + str(exc))
|