Update server.js

server.js

@@ -1,5 +1,5 @@
 const WebSocket = require('ws');
-const { exec } = require('child_process');
+const { spawn } = require('child_process');
 
 // Create a WebSocket server listening on port 7860
 const wss = new WebSocket.Server({ port: 7860 });
@@ -18,24 +18,31 @@ wss.on('connection', (ws) => {
         // Convert message (Buffer) to string
         const command = message.toString();
 
-        // Execute the command with superuser privileges
-        exec(command, (error, stdout, stderr) => {
-            if (error) {
-                ws.send(`Error: ${error.message}`);
-            }
-            if (stderr) {
-                ws.send(`stderr: ${stderr}`);
-            }
-            if (stdout) {
-                ws.send(`stdout: ${stdout}`);
-            }
-        }).stdout.on('data', (data) => {
-            // Send the stdout data to the WebSocket client
+        // Split command into command and arguments
+        const [cmd, ...args] = command.split(' ');
+
+        // Spawn a child process to execute the command
+        const process = spawn(cmd, args, { shell: true });
+
+        // Send stdout data to the WebSocket client
+        process.stdout.on('data', (data) => {
             ws.send(`stdout: ${data}`);
-        }).stderr.on('data', (data) => {
-            // Send the stderr data to the WebSocket client
+        });
+
+        // Send stderr data to the WebSocket client
+        process.stderr.on('data', (data) => {
             ws.send(`stderr: ${data}`);
         });
+
+        // Handle process completion
+        process.on('close', (code) => {
+            ws.send(`Process exited with code ${code}`);
+        });
+
+        // Handle errors
+        process.on('error', (err) => {
+            ws.send(`Error: ${err.message}`);
+        });
     });
 
     // Event listener for client disconnects