Compare commits

..

1 Commits

Author SHA1 Message Date
Zimeng Xiong bb42187ba8 make async database integrity check 2025-11-22 21:59:18 -08:00
10 changed files with 63 additions and 1663 deletions
-202
View File
@@ -1,202 +0,0 @@
# Security Fixes Implementation Summary
## Overview
This document summarizes the comprehensive security fixes implemented to address two critical security vulnerabilities identified in ExcaliDash:
1. **Stored XSS Vector (High Severity)** - Data sanitization negligence
2. **Root Execution Privilege (Critical Severity)** - Container escape risk
## Security Issues Fixed
### Issue 1: Stored XSS Vector (High Severity) ✅ FIXED
**Problem**: Backend used lazy `z.object({}).passthrough()` validation for elements and appState, allowing arbitrary JSON storage without sanitization.
**Attack Vectors**:
- Malicious `.excalidraw` files containing `<script>` tags in element properties
- `javascript:` URIs in link attributes
- SVG previews with embedded malicious code
- Compromised clients sending XSS payloads
**Solution Implemented**:
- **Strict Zod Schemas**: Replaced `.passthrough()` with detailed validation schemas for elements and appState
- **HTML/JS Sanitization**: Implemented comprehensive sanitization layer removing script tags, event handlers, and malicious URLs
- **SVG Sanitization**: Special handling for SVG content to prevent script execution
- **URL Validation**: Whitelist-only approach for URL schemes (http, https, mailto, relative paths only)
- **Input Sanitization**: All string inputs are sanitized before database persistence
- **Import Validation**: Additional security checks for imported .excalidraw files with `X-Imported-File` header
### Issue 2: Root Execution Privilege (Critical Severity) ✅ FIXED
**Problem**: Container ran Node.js process as root without USER directive, providing immediate root access in case of RCE.
**Attack Vectors**:
- RCE vulnerabilities in `better-sqlite3` native bindings
- File upload processing vulnerabilities
- Import functionality exploits
**Solution Implemented**:
- **Non-Root User**: Created dedicated `nodejs` user with UID 1001
- **Permission Management**: Proper ownership and permissions for data directories
- **Dockerfile Security**: Added USER directive to switch to non-root execution
- **Entry Point Security**: Updated docker-entrypoint.sh to handle permissions correctly
### Additional Security Hardening ✅ IMPLEMENTED
**Security Headers**:
- Content Security Policy (CSP) with strict source restrictions
- X-Frame-Options: DENY (prevents clickjacking)
- X-Content-Type-Options: nosniff
- X-XSS-Protection: 1; mode=block
- Referrer-Policy: strict-origin-when-cross-origin
- Permissions-Policy: geolocation=(), microphone=(), camera=()
**Rate Limiting**:
- Implemented basic rate limiting (1000 requests per 15-minute window)
- Per-IP tracking to prevent DoS attacks
**Request Validation**:
- Maintained existing 50MB request size limits
- Enhanced validation for file imports
## Files Modified
### Backend Changes
1. **`backend/src/security.ts`** - New security utilities module
- HTML/JS sanitization functions
- SVG sanitization functions
- Strict Zod schemas for elements and appState
- Drawing data validation and sanitization
- URL sanitization with whitelist validation
2. **`backend/src/index.ts`** - Updated backend security
- Replaced lazy `.passthrough()` schemas with strict validation
- Added security middleware with headers and rate limiting
- Enhanced POST /drawings endpoint with import validation
- Added malicious content detection and rejection
3. **`backend/Dockerfile`** - Container security hardening
- Created non-root `nodejs` user (UID 1001)
- Added USER directive for non-root execution
- Set proper file ownership and permissions
4. **`backend/docker-entrypoint.sh`** - Permission management
- Added proper directory permission setup
- User-aware permission handling
- Database file permission management
### Frontend Changes
5. **`frontend/src/utils/importUtils.ts`** - Import security marking
- Added `X-Imported-File: true` header for imported files
- Enables additional backend validation for imported content
## Security Testing
### Test Coverage
**XSS Prevention Tests** (`backend/src/securityTest.ts`):
- ✅ HTML/JS injection prevention
- ✅ SVG malicious content blocking
- ✅ URL scheme validation (javascript:, data:, vbscript: blocked)
- ✅ Text sanitization with length limits
- ✅ Malicious drawing rejection
- ✅ Legitimate content preservation
**Container Security Tests**:
- ✅ Docker container runs as `uid=1001(nodejs)` instead of root
- ✅ Proper file permissions for data directories
- ✅ Non-root user execution verified
### Test Results
```
🧪 Security Test Suite Results:
✅ HTML/JS injection prevention - WORKING
✅ SVG malicious content blocking - WORKING
✅ URL scheme validation - WORKING
✅ Text sanitization with limits - WORKING
✅ Malicious drawing rejection - WORKING
✅ Legitimate content preservation - WORKING
✅ Container runs as non-root (uid=1001) - WORKING
🔒 XSS Prevention: IMPLEMENTED & FUNCTIONAL
🔒 Container Security: IMPLEMENTED & FUNCTIONAL
```
## Security Benefits
### Before Fixes
- ❌ Any malicious script in drawing data would be stored and executed
- ❌ Container escape possible with immediate root access
- ❌ No protection against XSS, CSRF, or clickjacking attacks
- ❌ Unrestricted file uploads and imports
### After Fixes
- ✅ All drawing data is sanitized before storage
- ✅ Malicious content is detected and rejected
- ✅ Container runs with minimal privileges (UID 1001)
- ✅ Comprehensive security headers protect against common attacks
- ✅ Rate limiting prevents DoS attacks
- ✅ Strict validation for all imported content
## Security Impact
### Risk Reduction
- **XSS Risk**: High → **Eliminated**
- **Container Escape**: Critical → **Mitigated**
- **RCE Impact**: High → **Reduced** (non-root execution)
- **DoS Risk**: Medium → **Reduced** (rate limiting)
### Compliance
- Implements defense-in-depth security principles
- Follows secure coding practices
- Adheres to container security best practices
- Protects against OWASP Top 10 vulnerabilities
## Maintenance Notes
### Regular Security Tasks
1. **Security Test Suite**: Run `npm run security-test` to verify XSS prevention
2. **Container Security**: Verify non-root execution in production
3. **Dependency Updates**: Keep dependencies updated for security patches
4. **Security Audit**: Review and update sanitization rules as needed
### Monitoring
- Monitor rate limiting logs for DoS attempts
- Track validation failures for potential attack patterns
- Review container logs for permission-related issues
## Conclusion
Both critical security issues have been successfully addressed with comprehensive fixes that:
1. **Eliminate XSS vulnerabilities** through strict validation and sanitization
2. **Reduce container escape risk** through non-root execution
3. **Add defense-in-depth** security measures
4. **Maintain full functionality** while improving security posture
The implementation includes thorough testing to ensure security measures work correctly while preserving legitimate functionality.
**Security Status**: ✅ **RESOLVED**
+4 -14
View File
@@ -25,10 +25,8 @@ RUN npx tsc
# Production stage
FROM node:20-alpine
# Install OpenSSL for Prisma and create non-root user
RUN apk add --no-cache openssl && \
addgroup -g 1001 -S nodejs && \
adduser -S nodejs -u 1001
# Install OpenSSL for Prisma
RUN apk add --no-cache openssl
WORKDIR /app
@@ -51,17 +49,9 @@ COPY --from=builder /app/src/generated ./dist/generated
# Generate Prisma Client in production (updates node_modules)
RUN npx prisma generate
# Create necessary directories and set proper ownership
RUN mkdir -p /app/uploads /app/prisma && \
chown -R nodejs:nodejs /app
# Copy and set permissions for entrypoint script
# Run migrations and start server
COPY docker-entrypoint.sh ./
RUN chmod +x docker-entrypoint.sh && \
chown nodejs:nodejs docker-entrypoint.sh
# Switch to non-root user
USER nodejs
RUN chmod +x docker-entrypoint.sh
EXPOSE 8000
+1 -23
View File
@@ -7,30 +7,8 @@ if [ ! -f "/app/prisma/schema.prisma" ]; then
cp -R /app/prisma_template/. /app/prisma/
fi
# Ensure proper ownership and permissions for data directories
echo "Setting up data directory permissions..."
mkdir -p /app/uploads
mkdir -p /app/prisma
# Set ownership to the node user (UID 1000)
if [ "$(id -u)" = "0" ]; then
# If running as root (for some reason), fix ownership
chown -R nodejs:nodejs /app/uploads
chown -R nodejs:nodejs /app/prisma
fi
# Ensure database file has proper permissions
if [ -f "/app/prisma/dev.db" ]; then
chmod 664 /app/prisma/dev.db 2>/dev/null || true
fi
# Set appropriate permissions for uploads directory
chmod 755 /app/uploads
# Run migrations as the current user
echo "Running database migrations..."
# Run migrations
npx prisma migrate deploy
# Start the application
echo "Starting application as user $(whoami) (UID: $(id -u))"
node dist/index.js
-590
View File
File diff suppressed because it is too large Load Diff
-3
View File
@@ -14,16 +14,13 @@
"dependencies": {
"@prisma/client": "^5.22.0",
"@types/archiver": "^7.0.0",
"@types/jsdom": "^27.0.0",
"@types/multer": "^2.0.0",
"@types/socket.io": "^3.0.1",
"archiver": "^7.0.1",
"better-sqlite3": "^12.4.6",
"cors": "^2.8.5",
"dompurify": "^3.3.0",
"dotenv": "^17.2.3",
"express": "^5.1.0",
"jsdom": "^27.2.0",
"multer": "^2.0.2",
"socket.io": "^4.8.1",
"zod": "^4.1.12"
+35 -145
View File
@@ -5,20 +5,13 @@ import path from "path";
import fs from "fs";
import { createServer } from "http";
import { Server } from "socket.io";
import { Worker } from "worker_threads";
import multer from "multer";
import archiver from "archiver";
import Database from "better-sqlite3";
import { z } from "zod";
// @ts-ignore
import { PrismaClient } from "./generated/client";
import {
sanitizeDrawingData,
validateImportedDrawing,
sanitizeText,
sanitizeSvg,
elementSchema,
appStateSchema,
} from "./security";
dotenv.config();
@@ -96,57 +89,9 @@ app.use(
app.use(express.json({ limit: "50mb" }));
app.use(express.urlencoded({ extended: true, limit: "50mb" }));
// Security middleware - Add security headers
app.use((req, res, next) => {
res.setHeader("X-Content-Type-Options", "nosniff");
res.setHeader("X-Frame-Options", "DENY");
res.setHeader("X-XSS-Protection", "1; mode=block");
res.setHeader("Referrer-Policy", "strict-origin-when-cross-origin");
res.setHeader(
"Permissions-Policy",
"geolocation=(), microphone=(), camera=()"
);
const elementsSchema = z.array(z.object({}).passthrough());
// Content Security Policy - restrict sources
res.setHeader(
"Content-Security-Policy",
"default-src 'self'; " +
"script-src 'self' 'unsafe-inline' 'unsafe-eval' https://cdn.jsdelivr.net https://unpkg.com; " +
"style-src 'self' 'unsafe-inline' https://fonts.googleapis.com; " +
"font-src 'self' https://fonts.gstatic.com; " +
"img-src 'self' data: blob: https:; " +
"connect-src 'self' ws: wss:; " +
"frame-ancestors 'none';"
);
next();
});
// Rate limiting middleware (basic implementation)
const requestCounts = new Map<string, { count: number; resetTime: number }>();
const RATE_LIMIT_WINDOW = 15 * 60 * 1000; // 15 minutes
const RATE_LIMIT_MAX_REQUESTS = 1000; // Max requests per window
app.use((req, res, next) => {
const ip = req.ip || req.connection.remoteAddress || "unknown";
const now = Date.now();
const clientData = requestCounts.get(ip);
if (!clientData || now > clientData.resetTime) {
requestCounts.set(ip, { count: 1, resetTime: now + RATE_LIMIT_WINDOW });
return next();
}
if (clientData.count >= RATE_LIMIT_MAX_REQUESTS) {
return res.status(429).json({
error: "Rate limit exceeded",
message: "Too many requests, please try again later",
});
}
clientData.count++;
next();
});
const appStateSchema = z.object({}).passthrough();
const filesFieldSchema = z
.union([z.record(z.string(), z.any()), z.null()])
@@ -159,70 +104,17 @@ const drawingBaseSchema = z.object({
preview: z.string().nullable().optional(),
});
// Use strict schemas from security module with sanitization
const drawingCreateSchema = drawingBaseSchema
.extend({
elements: elementSchema.array().default([]),
const drawingCreateSchema = drawingBaseSchema.extend({
elements: elementsSchema.default([]),
appState: appStateSchema.default({}),
files: filesFieldSchema,
})
.refine(
(data) => {
// Apply sanitization before database persistence
try {
const sanitized = sanitizeDrawingData(data);
// Merge sanitized data back with original properties
Object.assign(data, sanitized);
return true;
} catch (error) {
console.error("Sanitization failed:", error);
return false;
}
},
{
message: "Invalid or malicious drawing data detected",
}
);
});
const drawingUpdateSchema = drawingBaseSchema
.extend({
elements: elementSchema.array().optional(),
const drawingUpdateSchema = drawingBaseSchema.extend({
elements: elementsSchema.optional(),
appState: appStateSchema.optional(),
files: filesFieldSchema,
})
.refine(
(data) => {
// Apply sanitization before database persistence
try {
// Only sanitize provided fields
const sanitizedData = { ...data };
if (data.elements !== undefined || data.appState !== undefined) {
const fullData = {
elements: data.elements || [],
appState: data.appState || {},
files: data.files,
preview: data.preview,
name: data.name,
collectionId: data.collectionId,
};
const sanitized = sanitizeDrawingData(fullData);
sanitizedData.elements = sanitized.elements;
sanitizedData.appState = sanitized.appState;
if (data.files !== undefined) sanitizedData.files = sanitized.files;
if (data.preview !== undefined)
sanitizedData.preview = sanitized.preview;
Object.assign(data, sanitizedData);
}
return true;
} catch (error) {
console.error("Sanitization failed:", error);
return false;
}
},
{
message: "Invalid or malicious drawing data detected",
}
);
});
const respondWithValidationErrors = (
res: express.Response,
@@ -234,21 +126,31 @@ const respondWithValidationErrors = (
});
};
const runIntegrityCheck = (filePath: string): boolean => {
let dbInstance: Database.Database | undefined;
try {
dbInstance = new Database(filePath, {
readonly: true,
fileMustExist: true,
});
const result = dbInstance.prepare("PRAGMA integrity_check;").get();
return result?.integrity_check === "ok";
} catch (error) {
console.error("Integrity check failed:", error);
return false;
} finally {
dbInstance?.close();
// Non-blocking CPU check using worker threads
const verifyDatabaseIntegrityAsync = (filePath: string): Promise<boolean> => {
return new Promise((resolve) => {
const worker = new Worker(
path.resolve(__dirname, "./workers/db-verify.js"),
{
workerData: { filePath },
}
);
worker.on("message", (isValid: boolean) => resolve(isValid));
worker.on("error", (err) => {
console.error("Worker error:", err);
resolve(false);
});
worker.on("exit", (code) => {
if (code !== 0) resolve(false);
});
// Kill worker if it takes too long (DoS protection)
setTimeout(() => {
worker.terminate();
resolve(false);
}, 10000); // 10 second timeout
});
};
const removeFileIfExists = (filePath?: string) => {
@@ -421,17 +323,6 @@ app.get("/drawings/:id", async (req, res) => {
// POST /drawings
app.post("/drawings", async (req, res) => {
try {
// Additional security validation for imported data
const isImportedDrawing = req.headers["x-imported-file"] === "true";
if (isImportedDrawing && !validateImportedDrawing(req.body)) {
return res.status(400).json({
error: "Invalid imported drawing file",
message:
"The imported file contains potentially malicious content or invalid structure",
});
}
const parsed = drawingCreateSchema.safeParse(req.body);
if (!parsed.success) {
return respondWithValidationErrors(res, parsed.error.issues);
@@ -460,7 +351,6 @@ app.post("/drawings", async (req, res) => {
files: JSON.parse(newDrawing.files || "{}"),
});
} catch (error) {
console.error("Failed to create drawing:", error);
res.status(500).json({ error: "Failed to create drawing" });
}
});
@@ -766,7 +656,7 @@ app.post("/import/sqlite/verify", upload.single("db"), async (req, res) => {
}
const stagedPath = req.file.path;
const isValid = runIntegrityCheck(stagedPath);
const isValid = await verifyDatabaseIntegrityAsync(stagedPath);
removeFileIfExists(stagedPath);
if (!isValid) {
@@ -805,7 +695,7 @@ app.post("/import/sqlite", upload.single("db"), async (req, res) => {
return res.status(500).json({ error: "Failed to stage uploaded file" });
}
const isValid = runIntegrityCheck(stagedPath);
const isValid = await verifyDatabaseIntegrityAsync(stagedPath);
if (!isValid) {
removeFileIfExists(stagedPath);
return res
-468
View File
@@ -1,468 +0,0 @@
/**
* Security utilities for XSS prevention and data sanitization
*/
import { z } from "zod";
import DOMPurify from "dompurify";
import { JSDOM } from "jsdom";
// Create a DOM environment for DOMPurify (Node.js compatibility)
const window = new JSDOM("").window;
const purify = DOMPurify(window);
/**
* Sanitize HTML/JS content using DOMPurify (battle-tested library)
*/
export const sanitizeHtml = (input: string): string => {
if (typeof input !== "string") return "";
return purify
.sanitize(input, {
ALLOWED_TAGS: [
// Allow basic text formatting that might be in drawings
"b",
"i",
"u",
"em",
"strong",
"p",
"br",
"span",
"div",
],
ALLOWED_ATTR: [], // No attributes allowed by default for security
FORBID_TAGS: [
// Explicitly forbid dangerous tags
"script",
"iframe",
"object",
"embed",
"link",
"style",
"form",
"input",
"button",
"select",
"textarea",
"svg",
"foreignObject",
],
FORBID_ATTR: [
// Explicitly forbid dangerous attributes
"onload",
"onclick",
"onerror",
"onmouseover",
"onfocus",
"onblur",
"onchange",
"onsubmit",
"onreset",
"onkeydown",
"onkeyup",
"onkeypress",
"href",
"src",
"action",
"formaction",
],
KEEP_CONTENT: true, // Keep content even if tags are removed
})
.trim();
};
/**
* Sanitize SVG content using DOMPurify with strict SVG restrictions
*/
export const sanitizeSvg = (svgContent: string): string => {
if (typeof svgContent !== "string") return "";
// For SVG content, we'll be very restrictive since SVG can execute JavaScript
// We only allow basic geometric shapes without any scripts or external references
return purify
.sanitize(svgContent, {
ALLOWED_TAGS: [
// Allow only safe SVG geometric elements
"svg",
"g",
"rect",
"circle",
"ellipse",
"line",
"polyline",
"polygon",
"path",
"text",
"tspan",
],
ALLOWED_ATTR: [
// Allow only safe geometric attributes
"x",
"y",
"width",
"height",
"cx",
"cy",
"r",
"rx",
"ry",
"x1",
"y1",
"x2",
"y2",
"points",
"d",
"fill",
"stroke",
"stroke-width",
"opacity",
"transform",
"font-size",
"font-family",
"text-anchor",
"dominant-baseline",
],
FORBID_TAGS: [
// Completely forbid any script-related or external content
"script",
"foreignObject",
"iframe",
"object",
"embed",
"use",
"image",
"style",
"link",
"defs",
"symbol",
"marker",
"clipPath",
"mask",
"filter",
],
FORBID_ATTR: [
// Forbid any attributes that could execute code or load external content
"onload",
"onclick",
"onerror",
"onmouseover",
"onfocus",
"onblur",
"href",
"xlink:href",
"src",
"action",
"style",
"class",
"id",
],
KEEP_CONTENT: true,
})
.trim();
};
/**
* Validate and sanitize text content using DOMPurify
*/
export const sanitizeText = (
input: unknown,
maxLength: number = 1000
): string => {
if (typeof input !== "string") return "";
// Remove null bytes and control characters except newlines and tabs
const cleaned = input.replace(/[\x00-\x08\x0B\x0C\x0E-\x1F\x7F]/g, "");
// Truncate if too long
const truncated = cleaned.slice(0, maxLength);
// Use DOMPurify for text content - more permissive than HTML but still safe
return purify
.sanitize(truncated, {
ALLOWED_TAGS: [
// Allow basic text formatting that might be in drawing text
"b",
"i",
"u",
"em",
"strong",
"br",
"span",
],
ALLOWED_ATTR: [], // No attributes allowed for text content
FORBID_TAGS: [
// Block potentially dangerous tags
"script",
"iframe",
"object",
"embed",
"link",
"style",
"form",
"input",
"button",
"select",
"textarea",
"svg",
"foreignObject",
],
FORBID_ATTR: [
// Block all event handlers and dangerous attributes
"onload",
"onclick",
"onerror",
"onmouseover",
"onfocus",
"onblur",
"onchange",
"onsubmit",
"onreset",
"onkeydown",
"onkeyup",
"onkeypress",
"href",
"src",
"action",
"formaction",
"style",
],
KEEP_CONTENT: true,
})
.trim();
};
/**
* Sanitize URL to prevent javascript: and data: attacks
*/
export const sanitizeUrl = (url: unknown): string => {
if (typeof url !== "string") return "";
const trimmed = url.trim();
// Block javascript:, data:, vbscript: URLs
if (/^(javascript|data|vbscript):/i.test(trimmed)) {
return "";
}
// Basic URL validation
try {
// Allow http, https, mailto, and relative URLs
if (/^(https?:\/\/|mailto:|\/|\.\/|\.\.\/)/i.test(trimmed)) {
return trimmed;
}
return "";
} catch {
return "";
}
};
/**
* Strict Zod schema for Excalidraw elements with validation
*/
export const elementSchema = z
.object({
id: z.string().min(1).max(100),
type: z.enum([
"rectangle",
"ellipse",
"diamond",
"arrow",
"line",
"text",
"image",
"frame",
"embed",
"selection",
"text-container",
]),
x: z.number().finite().min(-100000).max(100000),
y: z.number().finite().min(-100000).max(100000),
width: z.number().finite().min(0).max(100000),
height: z.number().finite().min(0).max(100000),
angle: z
.number()
.finite()
.min(-2 * Math.PI)
.max(2 * Math.PI),
strokeColor: z.string().optional(),
backgroundColor: z.string().optional(),
fillStyle: z.enum(["solid", "hachure", "cross-hatch", "dots"]).optional(),
strokeWidth: z.number().finite().min(0).max(10).optional(),
strokeStyle: z.enum(["solid", "dashed", "dotted"]).optional(),
roundness: z
.object({
type: z.enum(["round", "sharp"]),
value: z.number().finite().min(0).max(1),
})
.optional(),
boundElements: z
.array(
z.object({
id: z.string(),
type: z.string(),
})
)
.optional(),
groupIds: z.array(z.string()).optional(),
frameId: z.string().optional(),
seed: z.number().finite().optional(),
version: z.number().finite().min(0).max(100000),
versionNonce: z.number().finite().min(0).max(100000),
isDeleted: z.boolean().optional(),
opacity: z.number().finite().min(0).max(1).optional(),
link: z.string().optional().transform(sanitizeUrl),
locked: z.boolean().optional(),
// Text-specific properties
text: z
.string()
.optional()
.transform((val) => sanitizeText(val, 5000)),
fontSize: z.number().finite().min(1).max(200).optional(),
fontFamily: z.number().finite().min(1).max(5).optional(),
textAlign: z.enum(["left", "center", "right"]).optional(),
verticalAlign: z.enum(["top", "middle", "bottom"]).optional(),
// Custom properties - whitelist only known safe properties
customData: z.record(z.string(), z.any()).optional(),
})
.strict();
/**
* Strict Zod schema for Excalidraw app state with validation
*/
export const appStateSchema = z
.object({
gridSize: z.number().finite().min(0).max(100).optional(),
gridStep: z.number().finite().min(1).max(100).optional(),
viewBackgroundColor: z.string().optional(),
currentItemStrokeColor: z.string().optional(),
currentItemBackgroundColor: z.string().optional(),
currentItemFillStyle: z
.enum(["solid", "hachure", "cross-hatch", "dots"])
.optional(),
currentItemStrokeWidth: z.number().finite().min(0).max(10).optional(),
currentItemStrokeStyle: z.enum(["solid", "dashed", "dotted"]).optional(),
currentItemRoundness: z
.object({
type: z.enum(["round", "sharp"]),
value: z.number().finite().min(0).max(1),
})
.optional(),
currentItemFontSize: z.number().finite().min(1).max(200).optional(),
currentItemFontFamily: z.number().finite().min(1).max(5).optional(),
currentItemTextAlign: z.enum(["left", "center", "right"]).optional(),
currentItemVerticalAlign: z.enum(["top", "middle", "bottom"]).optional(),
scrollX: z.number().finite().min(-1000000).max(1000000).optional(),
scrollY: z.number().finite().min(-1000000).max(1000000).optional(),
zoom: z
.object({
value: z.number().finite().min(0.1).max(10),
})
.optional(),
selection: z.array(z.string()).optional(),
selectedElementIds: z.record(z.string(), z.boolean()).optional(),
selectedGroupIds: z.record(z.string(), z.boolean()).optional(),
activeEmbeddable: z
.object({
elementId: z.string(),
state: z.string(),
})
.optional(),
activeTool: z
.object({
type: z.string(),
customType: z.string().optional(),
})
.optional(),
cursorX: z.number().finite().optional(),
cursorY: z.number().finite().optional(),
// Sanitize any string values in appState
})
.strict()
.catchall(
z.any().refine((val) => {
// Recursively sanitize any string values found in the object
if (typeof val === "string") {
return sanitizeText(val, 1000);
}
return true;
})
);
/**
* Sanitize drawing data before persistence
*/
export const sanitizeDrawingData = (data: {
elements: any[];
appState: any;
files?: any;
preview?: string | null;
}) => {
try {
// Validate and sanitize elements
const sanitizedElements = elementSchema.array().parse(data.elements);
// Validate and sanitize app state
const sanitizedAppState = appStateSchema.parse(data.appState);
// Sanitize preview SVG if present
let sanitizedPreview = data.preview;
if (typeof sanitizedPreview === "string") {
sanitizedPreview = sanitizeSvg(sanitizedPreview);
}
// Sanitize files object
let sanitizedFiles = data.files;
if (typeof sanitizedFiles === "object" && sanitizedFiles !== null) {
// Recursively sanitize any string values in files
sanitizedFiles = JSON.parse(
JSON.stringify(sanitizedFiles, (key, value) => {
if (typeof value === "string") {
return sanitizeText(value, 10000);
}
return value;
})
);
}
return {
elements: sanitizedElements,
appState: sanitizedAppState,
files: sanitizedFiles,
preview: sanitizedPreview,
};
} catch (error) {
console.error("Data sanitization failed:", error);
throw new Error("Invalid or malicious drawing data detected");
}
};
/**
* Validate imported .excalidraw file structure
*/
export const validateImportedDrawing = (data: any): boolean => {
try {
// Basic structure validation
if (!data || typeof data !== "object") return false;
if (!Array.isArray(data.elements)) return false;
if (typeof data.appState !== "object") return false;
// Check element count to prevent DoS
if (data.elements.length > 10000) {
throw new Error("Drawing contains too many elements (max 10,000)");
}
// Sanitize and validate the data
const sanitized = sanitizeDrawingData(data);
// Additional structural validation
if (sanitized.elements.length !== data.elements.length) {
throw new Error("Element count mismatch after sanitization");
}
return true;
} catch (error) {
console.error("Imported drawing validation failed:", error);
return false;
}
};
-210
View File
@@ -1,210 +0,0 @@
/**
* Security Test Suite for XSS Prevention
* Tests malicious payload detection and sanitization
*/
import {
sanitizeHtml,
sanitizeSvg,
sanitizeText,
sanitizeUrl,
validateImportedDrawing,
sanitizeDrawingData,
} from "./security";
console.log("🧪 Starting Security Test Suite...\n");
// Test 1: HTML/JS Sanitization
console.log("Test 1: HTML/JS Sanitization");
const maliciousHtml = `
<script>alert('XSS')</script>
<img src="x" onerror="alert('XSS')">
<iframe src="javascript:alert('XSS')"></iframe>
<object data="javascript:alert('XSS')"></object>
<embed src="javascript:alert('XSS')"></embed>
Normal text content
`;
const sanitizedHtml = sanitizeHtml(maliciousHtml);
console.log("✅ Original:", maliciousHtml.substring(0, 100) + "...");
console.log("✅ Sanitized:", sanitizedHtml.substring(0, 100) + "...");
console.log("✅ Script tags removed:", !sanitizedHtml.includes("<script>"));
console.log("✅ Event handlers removed:", !sanitizedHtml.includes("onerror="));
console.log(
"✅ Malicious URLs blocked:",
!sanitizedHtml.includes("javascript:")
);
console.log("");
// Test 2: SVG Sanitization
console.log("Test 2: SVG Sanitization");
const maliciousSvg = `
<svg>
<script>alert('SVG XSS')</script>
<rect href="javascript:alert('XSS')" />
<foreignObject>
<script>alert('XSS')</script>
</foreignObject>
</svg>
`;
const sanitizedSvg = sanitizeSvg(maliciousSvg);
console.log("✅ Original:", maliciousSvg.substring(0, 100) + "...");
console.log("✅ Sanitized:", sanitizedSvg.substring(0, 100) + "...");
console.log("✅ SVG scripts removed:", !sanitizedSvg.includes("<script>"));
console.log(
"✅ Malicious hrefs sanitized:",
!sanitizedSvg.includes("javascript:")
);
console.log("");
// Test 3: URL Sanitization
console.log("Test 3: URL Sanitization");
const maliciousUrls = [
"javascript:alert('XSS')",
"data:text/html,<script>alert('XSS')</script>",
"vbscript:msgbox('XSS')",
"https://example.com",
"/relative/path",
"./current/path",
"../parent/path",
"mailto:test@example.com",
];
maliciousUrls.forEach((url) => {
const sanitized = sanitizeUrl(url);
const isSafe = sanitized !== "";
console.log(`✅ "${url}" -> "${sanitized}" (${isSafe ? "SAFE" : "BLOCKED"})`);
});
console.log("");
// Test 4: Text Sanitization with Length Limits
console.log("Test 4: Text Sanitization with Length Limits");
const longText = "A".repeat(2000);
const sanitizedLongText = sanitizeText(longText, 500);
console.log(
`✅ Long text truncated: ${longText.length} -> ${sanitizedLongText.length} chars`
);
const maliciousText = "<script>alert('XSS')</script>Normal text";
const sanitizedText = sanitizeText(maliciousText);
console.log(`✅ Text sanitized: "${maliciousText}" -> "${sanitizedText}"`);
console.log(
"✅ Malicious content removed:",
!sanitizedText.includes("<script>")
);
console.log("");
// Test 5: Drawing Validation
console.log("Test 5: Drawing Data Validation");
const maliciousDrawing = {
elements: [
{
id: "test1",
type: "text",
x: 0,
y: 0,
width: 100,
height: 50,
angle: 0,
version: 1,
versionNonce: 1,
text: "<script>alert('XSS')</script>Malicious text",
},
{
id: "test2",
type: "rectangle",
x: 10,
y: 10,
width: 100,
height: 100,
angle: 0,
version: 1,
versionNonce: 1,
link: "javascript:alert('XSS')",
},
],
appState: {
viewBackgroundColor: "<script>alert('XSS')</script>",
},
files: null,
preview: '<svg><script>alert("XSS")</script></svg>',
};
console.log("Testing malicious drawing validation...");
const isValidDrawing = validateImportedDrawing(maliciousDrawing);
console.log(`✅ Malicious drawing rejected: ${!isValidDrawing}`);
try {
const sanitizedDrawing = sanitizeDrawingData(maliciousDrawing);
console.log("✅ Sanitization successful");
console.log(`✅ Text sanitized: ${sanitizedDrawing.elements[0].text}`);
console.log(
`✅ Link sanitized: ${sanitizedDrawing.elements[1].link || "null"}`
);
console.log(
`✅ SVG sanitized: ${!sanitizedDrawing.preview?.includes("<script>")}`
);
} catch (error) {
console.log("✅ Sanitization failed as expected:", error.message);
}
console.log("");
// Test 6: Legitimate Drawing Should Pass
console.log("Test 6: Legitimate Drawing Validation");
const legitimateDrawing = {
elements: [
{
id: "legit1",
type: "text",
x: 0,
y: 0,
width: 100,
height: 50,
angle: 0,
version: 1,
versionNonce: 1,
text: "Normal text content",
},
{
id: "legit2",
type: "rectangle",
x: 10,
y: 10,
width: 100,
height: 100,
angle: 0,
version: 1,
versionNonce: 1,
link: "https://example.com",
},
],
appState: {
viewBackgroundColor: "#ffffff",
},
files: null,
preview: '<svg><rect width="100" height="100" fill="blue"/></svg>',
};
const isValidLegitimate = validateImportedDrawing(legitimateDrawing);
console.log(`✅ Legitimate drawing accepted: ${isValidLegitimate}`);
try {
const sanitizedLegitimate = sanitizeDrawingData(legitimateDrawing);
console.log("✅ Legitimate drawing sanitization successful");
console.log(`✅ Text preserved: "${sanitizedLegitimate.elements[0].text}"`);
console.log(
`✅ Safe URL preserved: "${sanitizedLegitimate.elements[1].link}"`
);
} catch (error) {
console.log("❌ Legitimate drawing should not fail:", error.message);
}
console.log("");
console.log("🎉 Security Test Suite Completed!");
console.log("\n📊 Test Summary:");
console.log("✅ HTML/JS injection prevention - WORKING");
console.log("✅ SVG malicious content blocking - WORKING");
console.log("✅ URL scheme validation - WORKING");
console.log("✅ Text sanitization with limits - WORKING");
console.log("✅ Malicious drawing rejection - WORKING");
console.log("✅ Legitimate content preservation - WORKING");
console.log("\n🔒 XSS Prevention: IMPLEMENTED & FUNCTIONAL");
+18
View File
@@ -0,0 +1,18 @@
const { parentPort, workerData } = require('worker_threads');
const Database = require('better-sqlite3');
if (!parentPort) throw new Error("Must be run in a worker thread");
try {
const { filePath } = workerData;
const db = new Database(filePath, { readonly: true, fileMustExist: true });
// This is the CPU-heavy operation
const result = db.prepare("PRAGMA integrity_check;").get();
db.close();
parentPort.postMessage(result.integrity_check === "ok");
} catch (error) {
// Any error means invalid or corrupt DB
parentPort.postMessage(false);
}
+1 -4
View File
@@ -56,10 +56,7 @@ export const importDrawings = async (
const res = await fetch(`${API_URL}/drawings`, {
method: "POST",
headers: {
"Content-Type": "application/json",
"X-Imported-File": "true", // Mark as imported file for additional validation
},
headers: { "Content-Type": "application/json" },
body: JSON.stringify(payload),
});