Summary
The current source tree still allows arbitrary code execution during supposedly safer allowlisted pickle loading. The allowlist trusts whole module namespaces instead of exact safe globals, so crafted pickles can invoke dangerous in-namespace callables through pickle REDUCE.
Details
- Vulnerability type: Remote code execution via unsafe deserialization
- Affected component:
nltk.picklesec.allowlisted_pickle_load, nltk.tokenize.punkt.punkt_pickle_load, nltk.parse.transitionparser.TransitionParser.parse
- Affected versions: Current source
v3.10.0-rc2; published 3.9.4 was not the claim target for this bypass.
- Patched versions: Not yet patched
- Root cause: Module-prefix allowlists include dangerous callables such as
nltk.tokenize.repp.ReppTokenizer._execute and numpy.f2py.crackfortran.myeval.
punkt_pickle_load() allowlists both nltk.tokenize.punkt and the whole nltk.tokenize namespace, which exposes ReppTokenizer._execute() and its subprocess.Popen(...) sink during unpickling. TransitionParser.parse() uses allowlisted_pickle_load(..., allowed_modules=("numpy", "scipy", "sklearn")), which permits numpy.f2py.crackfortran.myeval() and its attacker-controlled eval(...) path. I confirmed both gadgets create marker files before the caller returns or later aborts on type misuse.
PoC
Preconditions
- The application loads an attacker-controlled tokenizer or model artifact through these public loaders.
Steps
- Create a pickle whose REDUCE callable is
ReppTokenizer._execute and point its command to a harmless marker-file write.
- Pass that payload to
punkt_pickle_load(BytesIO(payload)) and observe the marker file is created during unpickling.
- Create a second pickle whose REDUCE callable is
numpy.f2py.crackfortran.myeval and load it through TransitionParser.parse().
- Observe the second marker file is created before
TransitionParser.parse() later fails on the returned object type.
Minimal reproducible excerpt
{'punkt_marker': 'PUNKT_RCE', 'transitionparser_marker': 'TP_RCE'}
Impact
Any caller that trusts these current allowlisted loaders can still execute attacker-controlled commands while loading model or tokenizer artifacts. This defeats the protection mechanism that replaced unrestricted pickle loading and creates a dangerous false sense of safety.
Remediation
Replace broad module-prefix allowlists with exact (module, qualname) pairs for the few safe classes or functions genuinely required. Do not allow entire namespaces such as nltk.tokenize or numpy, and keep post-load type validation only as a secondary defense.
Resources
Fix + attack demonstration (verified)
- tightened callers
find_class now, before the allowlists:
- Rejects any dotted
name → closes 4489 with zero legit impact.
- Denies dangerous modules (
os, subprocess, sys, builtins, numpy.f2py, nltk.tokenize.repp, …) even under a broad allowed_modules — a defense-in-depth backstop so a future too-broad allowlist can't silently reopen RCE.
builtins denied wholesale; safe primitives (int, str, …) must be named exactly via allowed_globals.
Callers tightened: punkt drops the broad nltk.tokenize (keeps nltk.tokenize.punkt + exact collections.defaultdict/builtins.int); transitionparser keeps numpy/scipy/sklearn (array unpickling needs their submodules) with the new guards blocking the gadgets.
Full pickle-sink audit
Every deserialization sink in the tree was reviewed: no raw pickle.load anywhere, and no joblib/numpy/torch/dill/yaml/marshal loaders. data.load + wordnet_app use RestrictedUnpickler (blocks all globals — safe); the remaining pickle_load sites (chartparser_app, tbl/demo) load user-selected or self-written files and keep their warning.
Attack demonstration (captured; fork clone)
=== EXPLOITS blocked ===
4489 sklearn.os.system (dotted) -> BLOCKED
x99w numpy.f2py.crackfortran.myeval -> BLOCKED
x99w nltk.tokenize.repp._execute -> BLOCKED
backstop os.system (os allowlisted) -> BLOCKED
backstop builtins.eval (exact global)-> BLOCKED
=== LEGIT loads still work ===
punkt round-trip via punkt_pickle_load -> OK
builtins.int (safe primitive) -> OK
Tests
test_pickle_allowlist_security.py — added 5 regressions (dotted traversal, both namespace gadgets, denied-module backstop, legit round-trip). Suite: 122 passed / 9 skipped (sklearn-dependent) across pickle/punkt/transition/tokenize. pre-commit (black/isort/ruff) clean.
References
Summary
The current source tree still allows arbitrary code execution during supposedly safer allowlisted pickle loading. The allowlist trusts whole module namespaces instead of exact safe globals, so crafted pickles can invoke dangerous in-namespace callables through pickle REDUCE.
Details
nltk.picklesec.allowlisted_pickle_load,nltk.tokenize.punkt.punkt_pickle_load,nltk.parse.transitionparser.TransitionParser.parsev3.10.0-rc2; published3.9.4was not the claim target for this bypass.nltk.tokenize.repp.ReppTokenizer._executeandnumpy.f2py.crackfortran.myeval.punkt_pickle_load()allowlists bothnltk.tokenize.punktand the wholenltk.tokenizenamespace, which exposesReppTokenizer._execute()and itssubprocess.Popen(...)sink during unpickling.TransitionParser.parse()usesallowlisted_pickle_load(..., allowed_modules=("numpy", "scipy", "sklearn")), which permitsnumpy.f2py.crackfortran.myeval()and its attacker-controlledeval(...)path. I confirmed both gadgets create marker files before the caller returns or later aborts on type misuse.PoC
Preconditions
Steps
ReppTokenizer._executeand point its command to a harmless marker-file write.punkt_pickle_load(BytesIO(payload))and observe the marker file is created during unpickling.numpy.f2py.crackfortran.myevaland load it throughTransitionParser.parse().TransitionParser.parse()later fails on the returned object type.Minimal reproducible excerpt
Impact
Any caller that trusts these current allowlisted loaders can still execute attacker-controlled commands while loading model or tokenizer artifacts. This defeats the protection mechanism that replaced unrestricted pickle loading and creates a dangerous false sense of safety.
Remediation
Replace broad module-prefix allowlists with exact
(module, qualname)pairs for the few safe classes or functions genuinely required. Do not allow entire namespaces such asnltk.tokenizeornumpy, and keep post-load type validation only as a secondary defense.Resources
Fix + attack demonstration (verified)
find_classnow, before the allowlists:name→ closes 4489 with zero legit impact.os,subprocess,sys,builtins,numpy.f2py,nltk.tokenize.repp, …) even under a broadallowed_modules— a defense-in-depth backstop so a future too-broad allowlist can't silently reopen RCE.builtinsdenied wholesale; safe primitives (int,str, …) must be named exactly viaallowed_globals.Callers tightened: punkt drops the broad
nltk.tokenize(keepsnltk.tokenize.punkt+ exactcollections.defaultdict/builtins.int); transitionparser keeps numpy/scipy/sklearn (array unpickling needs their submodules) with the new guards blocking the gadgets.Full pickle-sink audit
Every deserialization sink in the tree was reviewed: no raw
pickle.loadanywhere, and no joblib/numpy/torch/dill/yaml/marshal loaders.data.load+wordnet_appuseRestrictedUnpickler(blocks all globals — safe); the remainingpickle_loadsites (chartparser_app,tbl/demo) load user-selected or self-written files and keep their warning.Attack demonstration (captured; fork clone)
Tests
test_pickle_allowlist_security.py— added 5 regressions (dotted traversal, both namespace gadgets, denied-module backstop, legit round-trip). Suite: 122 passed / 9 skipped (sklearn-dependent) across pickle/punkt/transition/tokenize. pre-commit (black/isort/ruff) clean.References