Initial staking pallet

Cargo.toml

@@ -42,6 +42,9 @@ members = [
   "substrate/in-instructions/primitives",
   "substrate/in-instructions/pallet",
 
+  "substrate/staking/primitives",
+  "substrate/staking/pallet",
+
   "substrate/validator-sets/primitives",
   "substrate/validator-sets/pallet",
 

common/std-shims/Cargo.toml

@@ -13,7 +13,7 @@ all-features = true
 rustdoc-args = ["--cfg", "docsrs"]
 
 [dependencies]
-spin = "0.9"
+spin = { version = "0.9", features = ["mutex", "once"] }
 hashbrown = "0.14"
 
 [features]

common/std-shims/src/sync.rs

@@ -29,42 +29,22 @@ pub use mutex_shim::{ShimMutex as Mutex, MutexGuard};
 pub use std::sync::OnceLock;
 #[cfg(not(feature = "std"))]
 mod oncelock_shim {
-  use super::Mutex;
+  use spin::Once;
 
-  pub struct OnceLock<T>(Mutex<bool>, Option<T>);
+  pub struct OnceLock<T>(Once<T>);
   impl<T> OnceLock<T> {
     pub const fn new() -> OnceLock<T> {
-      OnceLock(Mutex::new(false), None)
+      OnceLock(Once::new())
     }
-
-    // These return a distinct Option in case of None so another caller using get_or_init doesn't
-    // transform it from None to Some
     pub fn get(&self) -> Option<&T> {
-      if !*self.0.lock() {
-        None
-      } else {
-        self.1.as_ref()
-      }
+      self.0.poll()
     }
     pub fn get_mut(&mut self) -> Option<&mut T> {
-      if !*self.0.lock() {
-        None
-      } else {
-        self.1.as_mut()
-      }
+      self.0.get_mut()
     }
 
     pub fn get_or_init<F: FnOnce() -> T>(&self, f: F) -> &T {
-      let mut lock = self.0.lock();
-      if !*lock {
-        unsafe {
-          core::ptr::addr_of!(self.1).cast_mut().write_unaligned(Some(f()));
-        }
-      }
-      *lock = true;
-      drop(lock);
-
-      self.get().unwrap()
+      self.0.call_once(f)
     }
   }
 }

common/zalloc/Cargo.toml

@@ -13,7 +13,7 @@ all-features = true
 rustdoc-args = ["--cfg", "docsrs"]
 
 [dependencies]
-zeroize = "^1.5"
+zeroize = { version = "^1.5", default-features = false }
 
 [features]
 # Commented for now as it requires nightly and we don't use nightly

coordinator/tributary/Cargo.toml

@@ -32,7 +32,7 @@ scale = { package = "parity-scale-codec", version = "3", features = ["derive"] }
 futures = "0.3"
 tendermint = { package = "tendermint-machine", path = "./tendermint" }
 
-tokio = { version = "1", features = ["macros", "sync", "time", "rt"] }
+tokio = { version = "1", features = ["sync", "time", "rt"] }
 
 [features]
 tests = []

coordinator/tributary/tendermint/Cargo.toml

@@ -16,4 +16,7 @@ log = "0.4"
 parity-scale-codec = { version = "3", features = ["derive"] }
 
 futures = "0.3"
-tokio = { version = "1", features = ["macros", "sync", "time", "rt"] }
+tokio = { version = "1", features = ["sync", "time"] }
+
+[dev-dependencies]
+tokio = { version = "1", features = ["rt-multi-thread", "macros"] }

deny.toml

@@ -61,6 +61,9 @@ exceptions = [
 
   { allow = ["AGPL-3.0"], name = "serai-validator-sets-pallet" },
 
+  { allow = ["AGPL-3.0"], name = "serai-staking-primitives" },
+  { allow = ["AGPL-3.0"], name = "serai-staking-pallet" },
+
   { allow = ["AGPL-3.0"], name = "serai-runtime" },
   { allow = ["AGPL-3.0"], name = "serai-node" },
 

deploy/coins/bitcoin/Dockerfile

@@ -26,12 +26,15 @@ RUN grep bitcoin-${BITCOIN_VERSION}-x86_64-linux-gnu.tar.gz SHA256SUMS | sha256s
 # Prepare Image
 RUN tar xzvf bitcoin-${BITCOIN_VERSION}-x86_64-linux-gnu.tar.gz
 
-FROM debian:bullseye-slim as image
+FROM debian:bookworm-slim as image
 
 WORKDIR /home/bitcoin
 COPY --from=builder /home/bitcoin/* .
 RUN mv bin/* /bin && mv lib/* /lib
 COPY ./scripts /scripts
 
+# Upgrade packages
+RUN apt update && apt upgrade -y
+
 EXPOSE 8332 8333 18332 18333 18443 18444
 VOLUME ["/home/bitcoin/.bitcoin"]

deploy/coins/monero/Dockerfile

@@ -27,12 +27,15 @@ RUN gpg --keyserver hkp://keyserver.ubuntu.com:80 --keyserver-options no-self-si
 RUN tar -xvjf monero-linux-x64-v${MONERO_VERSION}.tar.bz2 --strip-components=1
 
 # Prepare Image
-FROM debian:bullseye-slim as image
+FROM debian:bookworm-slim as image
 
 WORKDIR /home/monero
 COPY --from=builder /home/monero/* .
 RUN mv * /bin/
 COPY ./scripts /scripts
 
+# Upgrade packages
+RUN apt update && apt upgrade -y
+
 EXPOSE 18080 18081
 VOLUME /home/monero/.bitmonero

deploy/message-queue/Dockerfile

@@ -1,4 +1,4 @@
-FROM rust:1.71 as builder
+FROM rust:1.71-slim-bookworm as builder
 LABEL description="STAGE 1: Build"
 
 # Add files for build
@@ -29,7 +29,7 @@ RUN --mount=type=cache,target=/root/.cargo \
   mv /serai/target/release/serai-message-queue /serai/bin
 
 # Prepare Image
-FROM debian:bullseye-slim as image
+FROM debian:bookworm-slim as image
 LABEL description="STAGE 2: Copy and Run"
 
 WORKDIR /home/serai
@@ -38,6 +38,9 @@ WORKDIR /home/serai
 COPY --from=builder /serai/bin/* /bin/
 COPY --from=builder /serai/AGPL-3.0 .
 
+# Upgrade packages
+RUN apt update && apt upgrade -y
+
 # Run message-queue
 EXPOSE 2287
 CMD ["serai-message-queue"]

deploy/processor/Dockerfile

@@ -1,4 +1,4 @@
-FROM docker.io/paritytech/ci-linux:production as builder
+FROM rust:1.71-slim-bookworm as builder
 LABEL description="STAGE 1: Build"
 
 # Add files for build
@@ -16,6 +16,11 @@ ADD AGPL-3.0 /serai
 
 WORKDIR /serai
 
+RUN apt update && apt upgrade -y && apt install -y clang libssl-dev
+
+# Add the wasm toolchain
+RUN rustup target add wasm32-unknown-unknown
+
 # Mount the caches and build
 RUN --mount=type=cache,target=/root/.cargo \
   --mount=type=cache,target=/usr/local/cargo/registry \
@@ -27,7 +32,7 @@ RUN --mount=type=cache,target=/root/.cargo \
   mv /serai/target/release/serai-processor /serai/bin
 
 # Prepare Image
-FROM debian:bullseye-slim as image
+FROM debian:bookworm-slim as image
 LABEL description="STAGE 2: Copy and Run"
 
 WORKDIR /home/serai
@@ -36,8 +41,8 @@ WORKDIR /home/serai
 COPY --from=builder /serai/bin/* /bin/
 COPY --from=builder /serai/AGPL-3.0 .
 
-# Install openssl
-RUN apt update && apt upgrade -y && apt install -y libssl
+# Upgrade packages and install openssl
+RUN apt update && apt upgrade -y && apt install -y libssl-dev
 
 # Run processor
 CMD ["serai-processor"]

deploy/serai/Dockerfile

@@ -1,4 +1,4 @@
-FROM docker.io/paritytech/ci-linux:production as builder
+FROM rust:1.71-slim-bookworm as builder
 LABEL description="STAGE 1: Build"
 
 # Add files for build
@@ -16,34 +16,20 @@ ADD AGPL-3.0 /serai
 
 WORKDIR /serai
 
-# Update Rust
-RUN rustup update
-
-# Install Solc @ 0.8.16
-RUN --mount=type=cache,target=/root/.cache \
-  --mount=type=cache,target=/root/.local \
-  --mount=type=cache,target=/root/.solc-select \
-  pip3 install solc-select==0.2.1
-RUN --mount=type=cache,target=/root/.cache \
-  --mount=type=cache,target=/root/.local \
-  --mount=type=cache,target=/root/.solc-select \
-  solc-select install 0.8.16
-RUN --mount=type=cache,target=/root/.cache \
-  --mount=type=cache,target=/root/.local \
-  --mount=type=cache,target=/root/.solc-select \
-  solc-select use 0.8.16
-
-# Mount cargo and the Serai cache
-RUN --mount=type=cache,target=/root/.local \
-  --mount=type=cache,target=/root/.solc-select \
-  --mount=type=cache,target=/root/.cache \
-  --mount=type=cache,target=/usr/local/cargo/git \
+RUN apt update && apt upgrade -y && apt install -y make clang libssl-dev protobuf-compiler
+
+# Add the wasm toolchain
+RUN rustup target add wasm32-unknown-unknown
+
+# Mount the caches and build
+RUN --mount=type=cache,target=/root/.cargo \
   --mount=type=cache,target=/usr/local/cargo/registry \
+  --mount=type=cache,target=/usr/local/cargo/git \
   --mount=type=cache,target=/serai/target \
   cd substrate/node && cargo build --release
 
 # Prepare Image
-FROM ubuntu:latest as image
+FROM debian:bookworm-slim as image
 LABEL description="STAGE 2: Copy and Run"
 
 WORKDIR /home/serai
@@ -52,6 +38,9 @@ WORKDIR /home/serai
 COPY --from=builder /serai/target/release/ /bin/
 COPY --from=builder /serai/AGPL-3.0 .
 
+# Upgrade packages
+RUN apt update && apt upgrade -y
+
 # Run node
 EXPOSE 30333 9615 9933 9944
 CMD ["serai-node"]

substrate/runtime/Cargo.toml

@@ -57,6 +57,9 @@ pallet-grandpa = { git = "https://github.com/serai-dex/substrate", default-featu
 
 pallet-authority-discovery = { git = "https://github.com/serai-dex/substrate", default-features = false }
 
+pallet-session = { git = "https://github.com/serai-dex/substrate", default-features = false }
+staking-pallet = { package = "serai-staking-pallet", path = "../staking/pallet", default-features = false }
+
 frame-system-rpc-runtime-api = { git = "https://github.com/serai-dex/substrate", default-features = false }
 pallet-transaction-payment-rpc-runtime-api = { git = "https://github.com/serai-dex/substrate", default-features = false }
 
@@ -114,6 +117,8 @@ std = [
 ]
 
 runtime-benchmarks = [
+  "hex-literal",
+
   "sp-runtime/runtime-benchmarks",
 
   "frame-system/runtime-benchmarks",
@@ -125,8 +130,8 @@ runtime-benchmarks = [
   "pallet-balances/runtime-benchmarks",
   "pallet-assets/runtime-benchmarks",
 
-	"pallet-babe/runtime-benchmarks",
-	"pallet-grandpa/runtime-benchmarks",
+  "pallet-babe/runtime-benchmarks",
+  "pallet-grandpa/runtime-benchmarks",
 ]
 
 default = ["std"]

substrate/runtime/src/lib.rs

@@ -61,6 +61,8 @@ use babe::AuthorityId as BabeId;
 use grandpa::AuthorityId as GrandpaId;
 use sp_authority_discovery::AuthorityId as AuthorityDiscoveryId;
 
+use pallet_session::PeriodicSessions;
+
 /// An index to a block.
 pub type BlockNumber = u64;
 
@@ -354,6 +356,32 @@ impl authority_discovery::Config for Runtime {
   type MaxAuthorities = MaxAuthorities;
 }
 
+const SESSION_LENGTH: BlockNumber = 5 * DAYS;
+type Sessions = PeriodicSessions<ConstU32<{ SESSION_LENGTH }>, ConstU32<{ SESSION_LENGTH }>>;
+
+pub struct IdentityValidatorIdOf;
+impl Convert<Public, Option<Public>> for IdentityValidatorIdOf {
+  fn convert(key: Public) -> Option<Public> {
+    Some(key)
+  }
+}
+
+impl pallet_session::Config for Runtime {
+  type RuntimeEvent = RuntimeEvent;
+  type ValidatorId = AccountId;
+  type ValidatorIdOf = IdentityValidatorIdOf;
+  type ShouldEndSession = Sessions;
+  type NextSessionRotation = Sessions;
+  type SessionManager = ();
+  type SessionHandler = <SessionKeys as OpaqueKeys>::KeyTypeIdProviders;
+  type Keys = SessionKeys;
+  type WeightInfo = pallet_session::weights::SubstrateWeight<Runtime>;
+}
+
+impl staking_pallet::Config for Runtime {
+  type Currency = Balances;
+}
+
 pub type Header = generic::Header<BlockNumber, BlakeTwo256>;
 pub type Block = generic::Block<Header, UncheckedExtrinsic>;
 pub type SignedExtra = (
@@ -393,6 +421,7 @@ construct_runtime!(
     ValidatorSets: validator_sets,
 
     Session: session,
+    Staking: staking,
     Babe: babe,
     Grandpa: grandpa,
 

substrate/staking/pallet/Cargo.toml

@@ -0,0 +1,36 @@
+[package]
+name = "serai-staking-pallet"
+version = "0.1.0"
+description = "Staking pallet for Serai"
+license = "AGPL-3.0-only"
+repository = "https://github.com/serai-dex/serai/tree/develop/substrate/staking/pallet"
+authors = ["Luke Parker <[email protected]>"]
+edition = "2021"
+
+[package.metadata.docs.rs]
+all-features = true
+rustdoc-args = ["--cfg", "docsrs"]
+
+[dependencies]
+parity-scale-codec = { version = "3", default-features = false, features = ["derive"] }
+scale-info = { version = "2", default-features = false, features = ["derive"] }
+
+sp-runtime = { git = "https://github.com/serai-dex/substrate", default-features = false }
+
+frame-system = { git = "https://github.com/serai-dex/substrate", default-features = false }
+frame-support = { git = "https://github.com/serai-dex/substrate", default-features = false }
+
+staking-primitives = { package = "serai-staking-primitives", path = "../primitives", default-features = false }
+
+[features]
+std = [
+  "frame-system/std",
+  "frame-support/std",
+]
+
+runtime-benchmarks = [
+  "frame-system/runtime-benchmarks",
+  "frame-support/runtime-benchmarks",
+]
+
+default = ["std"]

substrate/staking/pallet/LICENSE

@@ -0,0 +1,15 @@
+AGPL-3.0-only license
+
+Copyright (c) 2022 Luke Parker
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License Version 3 as
+published by the Free Software Foundation.
+
+This program 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 Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program. If not, see <http://www.gnu.org/licenses/>.