Spaces:

hijnu
/

exec

Running

aigems commited on Sep 22, 2024

Commit

2317df1

1 Parent(s): 565dfa2

ok

Files changed (4) hide show

Dockerfile CHANGED Viewed

@@ -4,9 +4,21 @@ FROM node:20
 # 设置工作目录
 WORKDIR /app
-# 安装必要的工具和 Dropbear
 RUN apt-get update && apt-get install -y \
-    dropbear \
     && rm -rf /var/lib/apt/lists/*
 # 复制 package.json 和 package-lock.json（如果存在）
@@ -16,23 +28,28 @@ COPY package*.json ./
 # 安装依赖，包括 ssh2
 RUN npm install
-# 复制其余的源代码
 COPY . .
 # 创建一个新用户，UID 为 1000
 RUN if id 1000 >/dev/null 2>&1; then \
-        userdel -f $(id -nu 1000); \
     fi \
     && useradd -u 1000 -m -s /bin/bash user \
     && echo 'user:password' | chpasswd
 # 设置 Dropbear
 RUN mkdir -p /etc/dropbear
-RUN dropbearkey -t rsa -f /etc/dropbear/dropbear_rsa_host_key || true
 # 更改目录所有权
 RUN chown -R user:user /app /etc/dropbear
 # 启动脚本
 COPY start.sh /start.sh
 RUN chmod +x /start.sh

 # 设置工作目录
 WORKDIR /app
+# 安装必要的工具和最新版本的 Dropbear
 RUN apt-get update && apt-get install -y \
+    build-essential \
+    zlib1g-dev \
+    libssl-dev \
+    wget \
+    && wget https://matt.ucc.asn.au/dropbear/releases/dropbear-latest.tar.bz2 \
+    && tar xjf dropbear-latest.tar.bz2 \
+    && cd dropbear-* \
+    && ./configure \
+    && make \
+    && make install \
+    && cd .. \
+    && rm -rf dropbear-* \
+    && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 # 复制 package.json 和 package-lock.json（如果存在）
 # 安装依赖，包括 ssh2
 RUN npm install
+# 复制其余的源代码和公共文件
 COPY . .
 # 创建一个新用户，UID 为 1000
 RUN if id 1000 >/dev/null 2>&1; then \
+    userdel -f $(id -nu 1000); \
     fi \
     && useradd -u 1000 -m -s /bin/bash user \
     && echo 'user:password' | chpasswd
 # 设置 Dropbear
 RUN mkdir -p /etc/dropbear
 # 更改目录所有权
 RUN chown -R user:user /app /etc/dropbear
+# 切换到新用户
+USER user
+# 暴露 Web 应用端口和 Dropbear SSH 端口
+EXPOSE 7860 2202
 # 启动脚本
 COPY start.sh /start.sh
 RUN chmod +x /start.sh

app.js CHANGED Viewed

@@ -11,32 +11,45 @@ app.get('/', (req, res) => {
     res.sendFile(__dirname + '/public/index.html');
 });
-app.post('/ssh', (req, res) => {
     const conn = new Client();
-    conn.on('ready', () => {
-        conn.exec(req.body.command, (err, stream) => {
-            if (err) {
-                res.status(500).json({ error: err.message });
-                return;
-            }
-            let data = '';
-            stream.on('close', (code, signal) => {
-                conn.end();
-                res.json({ output: data });
-            }).on('data', (chunk) => {
-                data += chunk;
-            }).stderr.on('data', (chunk) => {
-                data += chunk;
             });
         });
-    }).on('error', (err) => {
-        res.status(500).json({ error: err.message });
-    }).connect({
-        host: 'localhost',
-        port: 2202,
-        username: 'user',
-        password: 'password'
-    });
 });
 app.listen(port, () => {

     res.sendFile(__dirname + '/public/index.html');
 });
+app.post('/ssh', async (req, res) => {
     const conn = new Client();
+    try {
+        await new Promise((resolve, reject) => {
+            conn.on('ready', resolve)
+                .on('error', reject)
+                .connect({
+                    host: 'localhost',
+                    port: 2202,
+                    username: 'user',
+                    password: 'password',
+                    readyTimeout: 5000,
+                    debug: console.log
+                });
+        });
+        console.log('SSH Connection established');
+        const { stdout, stderr } = await new Promise((resolve, reject) => {
+            conn.exec(req.body.command, (err, stream) => {
+                if (err) reject(err);
+                let stdout = '', stderr = '';
+                stream.on('close', (code, signal) => {
+                    resolve({ stdout, stderr, code, signal });
+                }).on('data', (data) => {
+                    stdout += data;
+                }).stderr.on('data', (data) => {
+                    stderr += data;
+                });
             });
         });
+        conn.end();
+        res.json({ output: stdout, error: stderr });
+    } catch (error) {
+        console.error('SSH error:', error);
+        res.status(500).json({ error: error.message });
+    }
 });
 app.listen(port, () => {

public/index.html CHANGED Viewed

@@ -6,7 +6,9 @@
     <title>Web SSH</title>
     <style>
         body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
-        #output { white-space: pre-wrap; background-color: #f0f0f0; padding: 10px; border-radius: 5px; }
     </style>
 </head>
 <body>

     <title>Web SSH</title>
     <style>
         body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
+        #output { white-space: pre-wrap; background-color: #f0f0f0; padding: 10px; border-radius: 5px; max-height: 400px; overflow-y: auto; }
+        #command { width: 70%; padding: 5px; }
+        button { padding: 5px 10px; }
     </style>
 </head>
 <body>

start.sh CHANGED Viewed

@@ -1,12 +1,35 @@
 #!/bin/bash
 # 检查并生成 SSH 主机密钥（如果不存在）
 if [ ! -f /etc/dropbear/dropbear_rsa_host_key ]; then
     dropbearkey -t rsa -f /etc/dropbear/dropbear_rsa_host_key
 fi
-# 启动 Dropbear，使用 2202 端口
-dropbear -R -p 2202
 # 启动 Node.js 应用
 npm start

 #!/bin/bash
+echo "===== System Information ====="
+uname -a
+echo "CPU: $(lscpu | grep 'Model name' | cut -f 2 -d ":")"
+echo "Memory: $(free -h | awk '/^Mem:/ {print $2}')"
+echo "Disk: $(df -h / | awk 'NR==2 {print $2}')"
+echo "==============================="
 # 检查并生成 SSH 主机密钥（如果不存在）
 if [ ! -f /etc/dropbear/dropbear_rsa_host_key ]; then
     dropbearkey -t rsa -f /etc/dropbear/dropbear_rsa_host_key
 fi
+# 启动 Dropbear，使用 2202 端口，允许密码认证
+dropbear -R -p 2202 -w -F -E
+# 检查 Dropbear 是否成功启动
+for i in {1..5}; do
+    if netstat -tuln | grep :2202 > /dev/null; then
+        echo "Dropbear started successfully on port 2202"
+        break
+    fi
+    if [ $i -eq 5 ]; then
+        echo "Failed to start Dropbear after 5 attempts"
+        exit 1
+    fi
+    echo "Waiting for Dropbear to start... (attempt $i)"
+    sleep 1
+done
+echo "Dropbear version: $(dropbear -V 2>&1)"
 # 启动 Node.js 应用
 npm start