feat: add ethereumjs (#107)

package_io/constants.star

@@ -3,7 +3,8 @@ EL_CLIENT_TYPE = struct(
 	erigon="erigon",
 	nethermind="nethermind",
 	besu="besu",
-	reth="reth"
+	reth="reth",
+	ethereumjs="ethereumjs"
 )
 
 CL_CLIENT_TYPE = struct(

package_io/input_parser.star

@@ -3,7 +3,8 @@ DEFAULT_EL_IMAGES = {
 	"erigon":		"thorax/erigon:devel",
 	"nethermind":	"nethermind/nethermind:latest",
 	"besu":			"hyperledger/besu:develop",
-	"reth": 		"ghcr.io/paradigmxyz/reth"
+	"reth": 		"ghcr.io/paradigmxyz/reth",
+	"ethereumjs":	"ethpandaops/ethereumjs:master",
 }
 
 DEFAULT_CL_IMAGES = {

src/el/ethereumjs/ethereumjs_launcher.star

@@ -0,0 +1,179 @@
+shared_utils = import_module("github.com/kurtosis-tech/eth-network-package/shared_utils/shared_utils.star")
+input_parser = import_module("github.com/kurtosis-tech/eth-network-package/package_io/input_parser.star")
+el_client_context = import_module("github.com/kurtosis-tech/eth-network-package/src/el/el_client_context.star")
+el_admin_node_info = import_module("github.com/kurtosis-tech/eth-network-package/src/el/el_admin_node_info.star")
+
+package_io = import_module("github.com/kurtosis-tech/eth-network-package/package_io/constants.star")
+
+
+RPC_PORT_NUM		= 8545
+WS_PORT_NUM			= 8546
+WS_PORT_ENGINE_NUM 	= 8547
+DISCOVERY_PORT_NUM	= 30303
+ENGINE_RPC_PORT_NUM = 8551
+#METRICS_PORT_NUM = 9001
+
+# The min/max CPU/memory that the execution node can use
+EXECUTION_MIN_CPU = 100
+EXECUTION_MAX_CPU = 1000
+EXECUTION_MIN_MEMORY = 256
+EXECUTION_MAX_MEMORY = 1024
+
+# Port IDs
+RPC_PORT_ID			= "rpc"
+WS_PORT_ID			= "ws"
+TCP_DISCOVERY_PORT_ID = "tcp-discovery"
+UDP_DISCOVERY_PORT_ID = "udp-discovery"
+ENGINE_RPC_PORT_ID	= "engine-rpc"
+WS_PORT_ENGINE_ID	= "ws-engine"
+#METRICS_PORT_ID = "metrics"
+
+GENESIS_DATA_MOUNT_DIRPATH = "/genesis"
+
+PREFUNDED_KEYS_MOUNT_DIRPATH = "/prefunded-keys"
+
+# The dirpath of the execution data directory on the client container
+EXECUTION_DATA_DIRPATH_ON_CLIENT_CONTAINER = "/execution-data"
+
+PRIVATE_IP_ADDRESS_PLACEHOLDER = "KURTOSIS_IP_ADDR_PLACEHOLDER"
+
+USED_PORTS = {
+	RPC_PORT_ID: shared_utils.new_port_spec(RPC_PORT_NUM, shared_utils.TCP_PROTOCOL),
+	WS_PORT_ID: shared_utils.new_port_spec(WS_PORT_NUM, shared_utils.TCP_PROTOCOL),
+	WS_PORT_ENGINE_ID: shared_utils.new_port_spec(WS_PORT_ENGINE_NUM, shared_utils.TCP_PROTOCOL),
+	TCP_DISCOVERY_PORT_ID: shared_utils.new_port_spec(DISCOVERY_PORT_NUM, shared_utils.TCP_PROTOCOL),
+	UDP_DISCOVERY_PORT_ID: shared_utils.new_port_spec(DISCOVERY_PORT_NUM, shared_utils.UDP_PROTOCOL),
+	ENGINE_RPC_PORT_ID: shared_utils.new_port_spec(ENGINE_RPC_PORT_NUM, shared_utils.TCP_PROTOCOL),
+#    METRICS_PORT_ID: shared_utils.new_port_spec(METRICS_PORT_NUM, shared_utils.TCP_PROTOCOL)
+}
+
+ENTRYPOINT_ARGS = []
+
+VERBOSITY_LEVELS = {
+	package_io.GLOBAL_CLIENT_LOG_LEVEL.error: "error",
+	package_io.GLOBAL_CLIENT_LOG_LEVEL.warn:  "warn",
+	package_io.GLOBAL_CLIENT_LOG_LEVEL.info:  "info",
+	package_io.GLOBAL_CLIENT_LOG_LEVEL.debug: "debug",
+	package_io.GLOBAL_CLIENT_LOG_LEVEL.trace: "trace",
+}
+
+
+def launch(
+	plan,
+	launcher,
+	service_name,
+	image,
+	participant_log_level,
+	global_log_level,
+	# If empty then the node will be launched as a bootnode
+	existing_el_clients,
+	el_min_cpu,
+	el_max_cpu,
+	el_min_mem,
+	el_max_mem,
+	extra_params):
+
+
+	log_level = input_parser.get_client_log_level_or_default(participant_log_level, global_log_level, VERBOSITY_LEVELS)
+
+	el_min_cpu = el_min_cpu if int(el_min_cpu) > 0 else EXECUTION_MIN_CPU
+	el_max_cpu = el_max_cpu if int(el_max_cpu) > 0 else EXECUTION_MAX_CPU
+	el_min_mem = el_min_mem if int(el_min_mem) > 0 else EXECUTION_MIN_MEMORY
+	el_max_mem = el_max_mem if int(el_max_mem) > 0 else EXECUTION_MAX_MEMORY
+
+
+	config, jwt_secret_json_filepath_on_client = get_config(
+		launcher.el_genesis_data,
+		image,
+		existing_el_clients,
+		log_level,
+		el_min_cpu,
+		el_max_cpu,
+		el_min_mem,
+		el_max_mem,
+		extra_params
+	)
+
+	service = plan.add_service(service_name, config)
+
+	enode = el_admin_node_info.get_enode_for_node(plan, service_name, RPC_PORT_ID)
+
+	jwt_secret = shared_utils.read_file_from_service(plan, service_name, jwt_secret_json_filepath_on_client)
+
+	return el_client_context.new_el_client_context(
+		"ethereumjs",
+		"", # ethereumjs has no enr
+		enode,
+		service.ip_address,
+		RPC_PORT_NUM,
+		WS_PORT_NUM,
+		ENGINE_RPC_PORT_NUM,
+		jwt_secret,
+		service_name,
+	)
+
+def get_config(
+	genesis_data,
+	image,
+	existing_el_clients,
+	verbosity_level,
+	el_min_cpu,
+	el_max_cpu,
+	el_min_mem,
+	el_max_mem,
+	extra_params):
+
+	genesis_json_filepath_on_client = shared_utils.path_join(GENESIS_DATA_MOUNT_DIRPATH, genesis_data.geth_genesis_json_relative_filepath)
+	jwt_secret_json_filepath_on_client = shared_utils.path_join(GENESIS_DATA_MOUNT_DIRPATH, genesis_data.jwt_secret_relative_filepath)
+
+	cmd = [
+		"--gethGenesis=" + genesis_json_filepath_on_client,
+		"--dataDir=" + EXECUTION_DATA_DIRPATH_ON_CLIENT_CONTAINER,
+		"--port={0}".format(DISCOVERY_PORT_NUM),
+		"--rpc",
+		"--rpcAddr=0.0.0.0",
+		"--rpcPort={0}".format(RPC_PORT_NUM),
+		"--rpcCors=*",
+		"--rpcEngine",
+		"--rpcEngineAddr=0.0.0.0",
+		"--rpcEnginePort={0}".format(ENGINE_RPC_PORT_NUM),
+		"--ws",
+		"--wsAddr=0.0.0.0",
+		"--wsPort={0}".format(WS_PORT_NUM),
+		"--wsEnginePort={0}".format(WS_PORT_ENGINE_NUM),
+		"--wsEngineAddr=0.0.0.0",
+		"--jwt-secret={0}".format(jwt_secret_json_filepath_on_client),
+		"--extIP={0}".format(PRIVATE_IP_ADDRESS_PLACEHOLDER),
+		"--sync=full",
+		"--isSingleNode=true",
+		"--logLevel={0}".format(verbosity_level),
+	]
+
+	if len(existing_el_clients) > 0:
+		cmd.append("--bootnodes=" + ",".join([ctx.enode for ctx in existing_el_clients[:package_io.MAX_ENODE_ENTRIES]]))
+
+	if len(extra_params) > 0:
+		# this is a repeated<proto type>, we convert it into Starlark
+		cmd.extend([param for param in extra_params])
+
+
+	return ServiceConfig(
+		image = image,
+		ports = USED_PORTS,
+		cmd = cmd,
+		files = {
+			GENESIS_DATA_MOUNT_DIRPATH: genesis_data.files_artifact_uuid,
+		},
+		entrypoint = ENTRYPOINT_ARGS,
+		private_ip_address_placeholder = PRIVATE_IP_ADDRESS_PLACEHOLDER,
+		min_cpu = el_min_cpu,
+		max_cpu = el_max_cpu,
+		min_memory = el_min_mem,
+		max_memory = el_max_mem
+	), jwt_secret_json_filepath_on_client
+
+
+def new_ethereumjs_launcher(el_genesis_data):
+	return struct(
+		el_genesis_data = el_genesis_data,
+	)

src/participant_network.star

@@ -9,7 +9,7 @@ besu = import_module("github.com/kurtosis-tech/eth-network-package/src/el/besu/b
 erigon = import_module("github.com/kurtosis-tech/eth-network-package/src/el/erigon/erigon_launcher.star")
 nethermind = import_module("github.com/kurtosis-tech/eth-network-package/src/el/nethermind/nethermind_launcher.star")
 reth = import_module("github.com/kurtosis-tech/eth-network-package/src/el/reth/reth_launcher.star")
-
+ethereumjs = import_module("github.com/kurtosis-tech/eth-network-package/src/el/ethereumjs/ethereumjs_launcher.star")
 
 lighthouse = import_module("github.com/kurtosis-tech/eth-network-package/src/cl/lighthouse/lighthouse_launcher.star")
 lodestar = import_module("github.com/kurtosis-tech/eth-network-package/src/cl/lodestar/lodestar_launcher.star")
@@ -119,6 +119,7 @@ def launch_participant_network(plan, participants, network_params, global_log_le
 		package_io.EL_CLIENT_TYPE.erigon : {"launcher": erigon.new_erigon_launcher(network_params.network_id, el_genesis_data), "launch_method": erigon.launch},
 		package_io.EL_CLIENT_TYPE.nethermind : {"launcher": nethermind.new_nethermind_launcher(el_genesis_data), "launch_method": nethermind.launch},
 		package_io.EL_CLIENT_TYPE.reth : {"launcher": reth.new_reth_launcher(el_genesis_data), "launch_method": reth.launch},
+		package_io.EL_CLIENT_TYPE.ethereumjs : {"launcher": ethereumjs.new_ethereumjs_launcher(el_genesis_data), "launch_method": ethereumjs.launch},
 	}
 
 	all_el_client_contexts = []