paritytech · s0me0ne-unkn0wn · Aug 1, 2023 · Aug 2, 2023 · Aug 2, 2023 · Aug 2, 2023
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -2,14 +2,6 @@
 name = "polkadot"
 path = "src/main.rs"
 
-[[bin]]
-name = "polkadot-execute-worker"
-path = "src/bin/execute-worker.rs"
-
-[[bin]]
-name = "polkadot-prepare-worker"
-path = "src/bin/prepare-worker.rs"
-
 [package]
 name = "polkadot"
 description = "Implementation of a `https://polkadot.network` node in Rust based on the Substrate framework."
@@ -38,24 +30,31 @@ polkadot-node-core-pvf = { path = "node/core/pvf" }
 polkadot-node-core-pvf-prepare-worker = { path = "node/core/pvf/prepare-worker" }
 polkadot-overseer = { path = "node/overseer" }
 
-# Needed for worker binaries.
-polkadot-node-core-pvf-common = { path = "node/core/pvf/common" }
-polkadot-node-core-pvf-execute-worker = { path = "node/core/pvf/execute-worker" }
-
 [dev-dependencies]
 assert_cmd = "2.0.4"
 nix = { version = "0.26.1", features = ["signal"] }
 tempfile = "3.2.0"
 tokio = "1.24.2"
 substrate-rpc-client = { git = "https://github.com/paritytech/substrate", branch = "master" }
 polkadot-core-primitives = { path = "core-primitives" }
+polkadot-execute-worker = { path = "bin/execute-worker" }
+polkadot-prepare-worker = { path = "bin/prepare-worker" }
 
 [build-dependencies]
+ansi_term = "0.12.1"
+regex = "1.9.3"
+rustc_version = "0.4.0"
 substrate-build-script-utils = { git = "https://github.com/paritytech/substrate", branch = "master" }
 
+# To get the binary names.
+polkadot-execute-worker = { path = "bin/execute-worker" }
+polkadot-prepare-worker = { path = "bin/prepare-worker" }
+
 [workspace]
 members = [
 	"cli",
+	"bin/execute-worker",
+	"bin/prepare-worker",
 	"core-primitives",
 	"erasure-coding",
 	"erasure-coding/fuzzer",
@@ -142,6 +141,12 @@ members = [
 	"utils/generate-bags",
 ]
 
+default-members = [
+	".",
+	"bin/execute-worker",
+	"bin/prepare-worker",
+]
+
 [badges]
 maintenance = { status = "actively-developed" }
 

diff --git a/bin/execute-worker/Cargo.toml b/bin/execute-worker/Cargo.toml
@@ -0,0 +1,16 @@
+[package]
+name = "polkadot-execute-worker"
+version.workspace = true
+authors.workspace = true
+edition.workspace = true
+
+[[bin]]
+name = "polkadot-execute-worker"
+path = "src/main.rs"
+
+[dependencies]
+polkadot-node-core-pvf-common = { path = "../../node/core/pvf/common" }
+polkadot-node-core-pvf-execute-worker = { path = "../../node/core/pvf/execute-worker" }
+
+[build-dependencies]
+substrate-build-script-utils = { git = "https://github.com/paritytech/substrate", branch = "master" }
diff --git a/bin/execute-worker/build.rs b/bin/execute-worker/build.rs
@@ -0,0 +1,22 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is part of Polkadot.
+
+// Polkadot is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// Polkadot is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with Polkadot.  If not, see <http://www.gnu.org/licenses/>.
+
+fn main() {
+	substrate_build_script_utils::generate_cargo_keys();
+	// For the node/worker version check, make sure we always rebuild the node and binary workers
+	// when the version changes.
+	substrate_build_script_utils::rerun_if_git_head_changed();
+}
diff --git a/bin/execute-worker/src/lib.rs b/bin/execute-worker/src/lib.rs
@@ -0,0 +1,18 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is part of Polkadot.
+
+// Polkadot is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// Polkadot is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with Polkadot.  If not, see <http://www.gnu.org/licenses/>.
+
+/// Execute worker binary name
+pub const BINARY_NAME: &str = "polkadot-execute-worker";
diff --git a/bin/execute-worker/src/main.rs b/bin/execute-worker/src/main.rs
@@ -0,0 +1,33 @@
+// Copyright (C) Parity Technologies (UK) Ltd.
+// This file is part of Polkadot.
+
+// Polkadot is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// Polkadot is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+
+// You should have received a copy of the GNU General Public License
+// along with Polkadot.  If not, see <http://www.gnu.org/licenses/>.
+
+//! Execute worker.
+
+#[cfg(test)]
+use polkadot_execute_worker::BINARY_NAME;
+
+polkadot_node_core_pvf_common::decl_worker_main!(
+	"execute-worker",
+	polkadot_node_core_pvf_execute_worker::worker_entrypoint,
+	env!("SUBSTRATE_CLI_IMPL_VERSION")
+);
+
+#[test]
+fn name_test() {
+	// If binary name ever needs to change, it must be changed in the package manifest
+	// as well as in the constant value
+	assert_eq!(env!("CARGO_BIN_NAME"), BINARY_NAME);
+}
diff --git a/src/bin/execute-worker.rs → bin/execute-worker/tests/integration.rs b/src/bin/execute-worker.rs → bin/execute-worker/tests/integration.rs
@@ -14,10 +14,10 @@
 // You should have received a copy of the GNU General Public License
 // along with Polkadot.  If not, see <http://www.gnu.org/licenses/>.
 
-//! Execute worker.
+//! Dummy integration test needed for the binary to get build when other intergation
+//! tests are run
 
-polkadot_node_core_pvf_common::decl_worker_main!(
-	"execute-worker",
-	polkadot_node_core_pvf_execute_worker::worker_entrypoint,
-	env!("SUBSTRATE_CLI_IMPL_VERSION")
-);
+#[test]
+fn dummy_test() {
+	assert!(true);
+}
diff --git a/bin/prepare-worker/Cargo.toml b/bin/prepare-worker/Cargo.toml
@@ -0,0 +1,16 @@
+[package]
+name = "polkadot-prepare-worker"
+version.workspace = true
+authors.workspace = true
+edition.workspace = true
+
+[[bin]]
+name = "polkadot-prepare-worker"
+path = "src/main.rs"
+
+[dependencies]
+polkadot-node-core-pvf-common = { path = "../../node/core/pvf/common" }
+polkadot-node-core-pvf-prepare-worker = { path = "../../node/core/pvf/prepare-worker" }
+
+[build-dependencies]
+substrate-build-script-utils = { git = "https://github.com/paritytech/substrate", branch = "master" }