X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=lightning%2Fsrc%2Fln%2Fscript.rs;fp=lightning%2Fsrc%2Fln%2Fscript.rs;h=589c3f311936c7885e1363c96b42055465f09991;hb=a21abe2b5ab67228a820cde7ce9b1c59970ca75d;hp=c2987f7393ff5d90e10de8229de1869af456d325;hpb=633f7329fc7df3015a00e87cb08a06b2d4973707;p=rust-lightning

diff --git a/lightning/src/ln/script.rs b/lightning/src/ln/script.rs
index c2987f73..589c3f31 100644
--- a/lightning/src/ln/script.rs
+++ b/lightning/src/ln/script.rs
@@ -10,6 +10,7 @@ use bitcoin::secp256k1::key::PublicKey;
 use ln::features::InitFeatures;
 
 use std::convert::TryFrom;
+use core::num::NonZeroU8;
 
 /// A script pubkey for shutting down a channel as defined by [BOLT #2].
 ///
@@ -60,7 +61,8 @@ impl ShutdownScript {
 	/// # Panics
 	///
 	/// This function may panic if given a segwit program with an invalid length.
-	pub fn new_witness_program(version: u5, program: &[u8]) -> Self {
+	pub fn new_witness_program(version: NonZeroU8, program: &[u8]) -> Self {
+		let version = u5::try_from_u8(version.get()).expect("Invalid segwit version");
 		let script = Script::new_witness_program(version, program);
 		Self::try_from(script).expect("Invalid segwit program")
 	}