mirror of
https://github.com/AutoMaker-Org/automaker.git
synced 2026-01-30 06:12:03 +00:00
feat: enhance terminal input validation and update keyboard shortcuts
- Added validation for terminal input to ensure it is a string and limited to 1MB to prevent memory issues. - Implemented checks for terminal resize dimensions to ensure they are positive integers within specified bounds. - Updated keyboard shortcuts for terminal actions to use Alt key combinations instead of Ctrl+Shift for better accessibility.
This commit is contained in:
@@ -297,11 +297,34 @@ terminalWss.on(
|
||||
|
||||
switch (msg.type) {
|
||||
case "input":
|
||||
// Validate input data type and length
|
||||
if (typeof msg.data !== "string") {
|
||||
ws.send(JSON.stringify({ type: "error", message: "Invalid input type" }));
|
||||
break;
|
||||
}
|
||||
// Limit input size to 1MB to prevent memory issues
|
||||
if (msg.data.length > 1024 * 1024) {
|
||||
ws.send(JSON.stringify({ type: "error", message: "Input too large" }));
|
||||
break;
|
||||
}
|
||||
// Write user input to terminal
|
||||
terminalService.write(sessionId, msg.data);
|
||||
break;
|
||||
|
||||
case "resize":
|
||||
// Validate resize dimensions are positive integers within reasonable bounds
|
||||
if (
|
||||
typeof msg.cols !== "number" ||
|
||||
typeof msg.rows !== "number" ||
|
||||
!Number.isInteger(msg.cols) ||
|
||||
!Number.isInteger(msg.rows) ||
|
||||
msg.cols < 1 ||
|
||||
msg.cols > 1000 ||
|
||||
msg.rows < 1 ||
|
||||
msg.rows > 500
|
||||
) {
|
||||
break; // Silently ignore invalid resize requests
|
||||
}
|
||||
// Resize terminal with deduplication and rate limiting
|
||||
if (msg.cols && msg.rows) {
|
||||
const now = Date.now();
|
||||
|
||||
@@ -2,6 +2,7 @@
|
||||
* Common utilities and state for terminal routes
|
||||
*/
|
||||
|
||||
import { randomBytes } from "crypto";
|
||||
import { createLogger } from "../../lib/logger.js";
|
||||
import type { Request, Response, NextFunction } from "express";
|
||||
import { getTerminalService } from "../../services/terminal-service.js";
|
||||
@@ -49,12 +50,10 @@ export function getTokenData(
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate a secure random token
|
||||
* Generate a cryptographically secure random token
|
||||
*/
|
||||
export function generateToken(): string {
|
||||
return `term-${Date.now()}-${Math.random()
|
||||
.toString(36)
|
||||
.slice(2, 17)}${Math.random().toString(36).slice(2, 17)}`;
|
||||
return `term-${randomBytes(32).toString("base64url")}`;
|
||||
}
|
||||
|
||||
/**
|
||||
|
||||
@@ -668,8 +668,8 @@ export function TerminalPanel({
|
||||
// Use event.code for keyboard-layout-independent key detection
|
||||
const code = event.code;
|
||||
|
||||
// Ctrl+Shift+D - Split right (uses Ctrl+Shift to avoid Alt+D readline conflict)
|
||||
if (event.ctrlKey && event.shiftKey && !event.altKey && !event.metaKey && code === 'KeyD') {
|
||||
// Alt+D - Split right
|
||||
if (event.altKey && !event.ctrlKey && !event.shiftKey && !event.metaKey && code === 'KeyD') {
|
||||
event.preventDefault();
|
||||
if (canTrigger) {
|
||||
lastShortcutTimeRef.current = now;
|
||||
@@ -678,8 +678,8 @@ export function TerminalPanel({
|
||||
return false;
|
||||
}
|
||||
|
||||
// Ctrl+Shift+S - Split down (uses Ctrl+Shift to avoid readline conflicts)
|
||||
if (event.ctrlKey && event.shiftKey && !event.altKey && !event.metaKey && code === 'KeyS') {
|
||||
// Alt+S - Split down
|
||||
if (event.altKey && !event.ctrlKey && !event.shiftKey && !event.metaKey && code === 'KeyS') {
|
||||
event.preventDefault();
|
||||
if (canTrigger) {
|
||||
lastShortcutTimeRef.current = now;
|
||||
@@ -688,8 +688,8 @@ export function TerminalPanel({
|
||||
return false;
|
||||
}
|
||||
|
||||
// Ctrl+Shift+W - Close terminal (uses Ctrl+Shift to avoid readline conflicts)
|
||||
if (event.ctrlKey && event.shiftKey && !event.altKey && !event.metaKey && code === 'KeyW') {
|
||||
// Alt+W - Close terminal
|
||||
if (event.altKey && !event.ctrlKey && !event.shiftKey && !event.metaKey && code === 'KeyW') {
|
||||
event.preventDefault();
|
||||
if (canTrigger) {
|
||||
lastShortcutTimeRef.current = now;
|
||||
@@ -698,8 +698,8 @@ export function TerminalPanel({
|
||||
return false;
|
||||
}
|
||||
|
||||
// Ctrl+Shift+T - New terminal tab (uses Ctrl+Shift for consistency)
|
||||
if (event.ctrlKey && event.shiftKey && !event.altKey && !event.metaKey && code === 'KeyT') {
|
||||
// Alt+T - New terminal tab
|
||||
if (event.altKey && !event.ctrlKey && !event.shiftKey && !event.metaKey && code === 'KeyT') {
|
||||
event.preventDefault();
|
||||
if (canTrigger && onNewTabRef.current) {
|
||||
lastShortcutTimeRef.current = now;
|
||||
@@ -1757,7 +1757,7 @@ export function TerminalPanel({
|
||||
e.stopPropagation();
|
||||
onSplitHorizontal();
|
||||
}}
|
||||
title="Split Right (Cmd+D)"
|
||||
title="Split Right (Alt+D)"
|
||||
>
|
||||
<SplitSquareHorizontal className="h-3 w-3" />
|
||||
</Button>
|
||||
@@ -1769,7 +1769,7 @@ export function TerminalPanel({
|
||||
e.stopPropagation();
|
||||
onSplitVertical();
|
||||
}}
|
||||
title="Split Down (Cmd+Shift+D)"
|
||||
title="Split Down (Alt+S)"
|
||||
>
|
||||
<SplitSquareVertical className="h-3 w-3" />
|
||||
</Button>
|
||||
@@ -1799,7 +1799,7 @@ export function TerminalPanel({
|
||||
e.stopPropagation();
|
||||
onClose();
|
||||
}}
|
||||
title="Close Terminal (Cmd+W)"
|
||||
title="Close Terminal (Alt+W)"
|
||||
>
|
||||
<X className="h-3 w-3" />
|
||||
</Button>
|
||||
|
||||
Reference in New Issue
Block a user