Deserialization of Untrusted Data
The product deserializes untrusted data without sufficiently verifying that the resulting data will be valid.
Many programming languages allow the serialization of objects for storage or transmission. When untrusted data is deserialized, it can lead to code execution, denial of service, or other unintended consequences.
Como corrigir esta vulnerabilidade
Estratégias de prevenção para Deserialization of Untrusted Data baseadas em 7 regras de detecção do Shoulder.
Use strict typed structs instead of interface{} and avoid gob with untrusted data
package main import ( - "encoding/gob" - "net/http" - ) - - func handler(w http.ResponseWriter, r *http.Request) { - // Vulnerable: gob decoding untrusted HTTP body - dec := gob.NewDecoder(r.Body) - var data interface{} - if err := dec.Decode(&data); err != nil { - http.Error(w, err.Error(), 400) + "encoding/json" + "net/http" + ) + + type UserRequest struct { + Name string `json:"name"` + Email string `json:"email"` + } + + func handler(w http.ResponseWriter, r *http.Request) { + // Safe: typed struct with JSON (data-only, no code execution) + var req UserRequest + dec := json.NewDecoder(r.Body) + dec.DisallowUnknownFields() + if err := dec.Decode(&req); err != nil { + http.Error(w, "Invalid request", 400) return } }
Validate all training data against strict schemas and apply content moderation before ingestion
func indexHandler(w http.ResponseWriter, r *http.Request) { var docs []Document json.NewDecoder(r.Body).Decode(&docs) + + validate := validator.New() + for _, doc := range docs { + if err := validate.Struct(doc); err != nil { + http.Error(w, "validation failed", http.StatusBadRequest) + return + } + if flagged, _ := moderationCheck(doc.Content); flagged { + http.Error(w, "content policy violation", http.StatusBadRequest) + return + } + } vectorDB.Upsert(docs) }
Validate training data against schemas and use content moderation before fine-tuning
app.post('/finetune', async (req, res) => { - await openai.files.create({ - file: req.body.trainingData, + const validated = trainingSchema.parse(req.body.trainingData); + const moderated = await moderateContent(validated); + await openai.files.create({ + file: moderated, purpose: 'fine-tune' }); });
Use JSON.parse() instead of node-serialize, and yaml.SAFE_SCHEMA for YAML parsing
const express = require('express'); - const serialize = require('node-serialize'); - const app = express(); - - app.post('/restore', (req, res) => { - const sessionData = req.body.session; - const session = serialize.deserialize(sessionData); - req.session = session; - res.json({ restored: true }); + const app = express(); + + app.post('/restore', (req, res) => { + try { + const session = JSON.parse(req.body.session); + req.session = session; + res.json({ restored: true }); + } catch (e) { + res.status(400).json({ error: 'Invalid session data' }); + } });
Validate training data with Pydantic schemas and apply content moderation before ingestion
- @app.route('/finetune', methods=['POST']) - def finetune(): - training_data = request.json['data'] - client.files.create(file=training_data, purpose='fine-tune') + from pydantic import BaseModel, validator + + class TrainingExample(BaseModel): + prompt: str + completion: str + + @validator('prompt', 'completion') + def validate_length(cls, v): + if len(v) > 4000: + raise ValueError('Content too long') + return v + + @app.route('/finetune', methods=['POST']) + async def finetune(): + examples = [TrainingExample(**ex) for ex in request.json['data']] + moderation = await openai.moderations.create( + input=[ex.completion for ex in examples] + ) + if any(r.flagged for r in moderation.results): + return {'error': 'Content policy violation'}, 400 + client.files.create(file=json.dumps([ex.dict() for ex in examples]), purpose='fine-tune') return {'status': 'queued'}
Replace pickle/marshal with JSON or other safe serialization formats
- import pickle - from flask import request - - @app.route('/load', methods=['POST']) - def load(): - data = request.get_data() - obj = pickle.loads(data) + import json + from flask import request + + @app.route('/load', methods=['POST']) + def load(): + data = request.get_data() + obj = json.loads(data) return str(obj)
Use yaml.safe_load() instead of yaml.load() to prevent code execution
import yaml def parse_config(yaml_string): - config = yaml.load(yaml_string) + config = yaml.safe_load(yaml_string) return config
Encontre vulnerabilidades no seu código
Use o Shoulder para escanear seu código em busca de padrões Deserialization of Untrusted Data. 7 regras.
# Scan with Shoulder CLI npx @shoulderdev/cli trust --cwe=502 # Or scan entire project npx @shoulderdev/cli trust .
Regras de Detecção (7)
O que observar nas revisões de código
Estes padrões indicam vulnerabilidades potenciais de Deserialization of Untrusted Data. Procure-os durante revisões de código e auditorias de segurança.
Escaneie seu código para Deserialization of Untrusted Data
O Shoulder CLI encontra padrões vulneráveis em todo o seu código.