mirror of
https://github.com/gnh1201/caterpillar.git
synced 2025-05-15 06:01:06 +00:00
60 lines
1.5 KiB
Python
60 lines
1.5 KiB
Python
#!/usr/bin/python3
|
|
#
|
|
# container.py
|
|
# Linux Container (e.g. Docker) plugin for Caterpillar Proxy
|
|
#
|
|
# Caterpillar Proxy - The simple and parasitic web proxy with SPAM filter
|
|
# Namyheon Go (Catswords Research) <gnh1201@gmail.com>
|
|
# https://github.com/gnh1201/caterpillar
|
|
# Created at: 2024-03-04
|
|
# Updated at: 2024-07-06
|
|
#
|
|
|
|
import docker
|
|
|
|
from base import Extension, Logger
|
|
|
|
logger = Logger("Container")
|
|
|
|
|
|
class Container(Extension):
|
|
def __init__(self):
|
|
self.type = "rpcmethod"
|
|
self.method = "container_init"
|
|
self.exported_methods = ["container_run", "container_stop"]
|
|
|
|
# docker
|
|
self.client = docker.from_env()
|
|
|
|
def dispatch(self, type, id, params, conn):
|
|
logger.info("[*] Greeting! dispatch")
|
|
conn.send(b"Greeting! dispatch")
|
|
|
|
def container_run(self, type, id, params, conn):
|
|
devices = params["devices"]
|
|
image = params["image"]
|
|
devices = params["devices"]
|
|
name = params["name"]
|
|
environment = params["environment"]
|
|
volumes = params["volumes"]
|
|
|
|
container = self.client.containers.run(
|
|
image,
|
|
devices=devices,
|
|
name=name,
|
|
volumes=volumes,
|
|
environment=environment,
|
|
detach=True,
|
|
)
|
|
container.logs()
|
|
|
|
logger.info("[*] Running...")
|
|
|
|
def container_stop(self, type, id, params, conn):
|
|
name = params["name"]
|
|
|
|
container = self.client.containers.get(name)
|
|
container.stop()
|
|
|
|
logger.info("[*] Stopped")
|