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.
この脆弱性の修正方法
7 件の Shoulder 検出ルールに基づく Deserialization of Untrusted Data の予防策。
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
コードの脆弱性を見つける
Shoulderを使用してコードのDeserialization of Untrusted Dataパターンをスキャンしましょう。 7 ルール.
# Scan with Shoulder CLI npx @shoulderdev/cli trust --cwe=502 # Or scan entire project npx @shoulderdev/cli trust .
検出ルール (7)
コードレビューで注目すべき点
これらのパターンはDeserialization of Untrusted Dataの潜在的な脆弱性を示しています。コードレビューとセキュリティ監査中に探してください。
コードベースをスキャン: Deserialization of Untrusted Data
Shoulder CLI はコードベース全体から脆弱なパターンを見つけます。