Many common libraries that are often used for reading configuration files and deserializing objects are very dangerous because they can allow execution of arbitrary code. By default, libraries such as PyYAML and pickle do not provide strong separation of data and code, and thus allow code to be embedded inside the input.
Often the input to these libraries is untrusted or only partially trusted. These unsafe inputs can come from configuration files or be provided via REST APIs. For example, we often use YAML for configuration files but YAML files can also contain embedded Python code. This may provide an attacker with a method to execute code.
Many, but not all, of these libraries, offer safe interfaces that disable features that enable code execution. You always want to use the safe functions to load input. Often the obvious function to use is not the safe one and we should check the documentation for libraries not covered here.
We often use YAML, pickle, or eval to load data into our Python programs, but this is dangerous. PyYAML has a safe way to load code, but pickle and eval do not.
Module | Problem | Use | Avoid |
---|---|---|---|
PyYAML | Allows creating arbitrary Python objects. | yaml.safe_load | yaml.load |
pickle | Allows creating arbitrary Python objects. | Do not use | pickle.load, pickle.loads |
eval | Runs all input as Python code | Do not use | eval |
yaml.load is the obvious function to use but is dangerous:
import yaml
import pickle
conf_str = '''
!!python/object:__main__.AttackerObj
key: 'value'
'''
conf = yaml.load(conf_str)
Neither pickle nor eval should be used with input that is untrusted.
import pickle
f = open('myfile', 'r')
s = f.read()
pickle.load(s) # can execute code in myfile
eval(s) # executes myfile contents as python code
Here we use PyYAMLs safe YAML loading function:
import yaml
conf_str = '''
- key: 'value'
- key: 'value'
'''
conf = yaml.safe_load(conf_str)
There is no safe alternative for pickle.load.
- Anyone that can control the input passed to dangerous libraries can gain arbitrary code execution on the system running the dangerous library.