Grcov report - lib.rs

1

// Copyright (C) Moondance Labs Ltd.

2

// This file is part of Tanssi.

3

4

// Tanssi is free software: you can redistribute it and/or modify

5

// it under the terms of the GNU General Public License as published by

6

// the Free Software Foundation, either version 3 of the License, or

7

// (at your option) any later version.

8

9

// Tanssi is distributed in the hope that it will be useful,

10

// but WITHOUT ANY WARRANTY; without even the implied warranty of

11

// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

12

// GNU General Public License for more details.

13

14

// You should have received a copy of the GNU General Public License

15

// along with Tanssi.  If not, see <http://www.gnu.org/licenses/>

16

17

//! # Author Noting Pallet

18

//!

19

//! This pallet notes the author of the different containerChains that have registered:

20

//!

21

//! The set of container chains is retrieved thanks to the GetContainerChains trait

22

//! For each containerChain, we inspect the Header stored in the relayChain as

23

//! a generic header. This is the first requirement for containerChains.

24

//!

25

//! The second requirement is that an Aura digest with the slot number for the containerChains

26

//! needs to exist

27

//!

28

//! Using those two requirements we can select who the author was based on the collators assigned

29

//! to that containerChain, by simply assigning the slot position.

30

31

#![cfg_attr(not(feature = "std"), no_std)]

32

33

use {

34

    cumulus_pallet_parachain_system::RelaychainStateProvider,

35

    cumulus_primitives_core::{

36

        relay_chain::{BlakeTwo256, BlockNumber, HeadData},

37

        ParaId,

38

},

39

    dp_core::well_known_keys::PARAS_HEADS_INDEX,

40

    frame_support::{dispatch::PostDispatchInfo, pallet_prelude::*, Hashable},

41

    frame_system::pallet_prelude::*,

42

    nimbus_primitives::SlotBeacon,

43

    parity_scale_codec::{Decode, Encode},

44

    sp_consensus_aura::{inherents::InherentType, Slot, AURA_ENGINE_ID},

45

    sp_inherents::{InherentIdentifier, IsFatalError},

46

    sp_runtime::{traits::Header, DigestItem, DispatchResult, RuntimeString},

47

    tp_author_noting_inherent::INHERENT_IDENTIFIER,

48

    tp_traits::{

49

        AuthorNotingHook, ContainerChainBlockInfo, GenericStateProof, GenericStorageReader,

50

        GetContainerChainAuthor, GetCurrentContainerChains, LatestAuthorInfoFetcher,

51

        NativeStorageReader, ReadEntryErr,

52

},

53

};

54

55

#[cfg(test)]

56

mod mock;

57

58

#[cfg(test)]

59

mod tests;

60

pub mod weights;

61

pub use weights::WeightInfo;

62

63

#[cfg(any(test, feature = "runtime-benchmarks"))]

64

mod benchmarks;

65

#[cfg(feature = "runtime-benchmarks")]

66

mod mock_proof;

67

68

pub use pallet::*;

69

70

17207

#[frame_support::pallet]

71

pub mod pallet {

72

    use super::*;

73

74

    #[pallet::config]

75

    pub trait Config: frame_system::Config {

76

        /// The overarching event type.

77

        type RuntimeEvent: From<Event<Self>> + IsType<<Self as frame_system::Config>::RuntimeEvent>;

78

79

        type ContainerChains: GetCurrentContainerChains;

80

81

        type SlotBeacon: SlotBeacon;

82

83

        type ContainerChainAuthor: GetContainerChainAuthor<Self::AccountId>;

84

85

        /// An entry-point for higher-level logic to react to containers chains authoring.

86

///

87

        /// Typically, this can be a hook to reward block authors.

88

        type AuthorNotingHook: AuthorNotingHook<Self::AccountId>;

89

90

        type RelayOrPara: RelayOrPara;

91

92

        /// Weight information for extrinsics in this pallet.

93

        type WeightInfo: WeightInfo;

94

95

96

14132

    #[pallet::error]

97

    pub enum Error<T> {

98

        /// The new value for a configuration parameter is invalid.

99

        FailedReading,

100

        FailedDecodingHeader,

101

        AuraDigestFirstItem,

102

        AsPreRuntimeError,

103

        NonDecodableSlot,

104

        AuthorNotFound,

105

        NonAuraDigest,

106

107

108

396

    #[pallet::pallet]

109

    pub struct Pallet<T>(PhantomData<T>);

110

111

27998

    #[pallet::hooks]

112

    impl<T: Config> Hooks<BlockNumberFor<T>> for Pallet<T> {

113

13999

        fn on_initialize(_n: BlockNumberFor<T>) -> Weight {

114

13999

            let mut weight = Weight::zero();

115

13999

116

13999

            // We clear this storage item to make sure its always included

117

13999

            DidSetContainerAuthorData::<T>::kill();

118

13999

119

13999

            weight += T::DbWeight::get().writes(1);

120

13999

121

13999

            // The read onfinalizes

122

13999

            weight += T::DbWeight::get().reads(1);

123

13999

124

13999

            weight

125

13999

126

127

13995

        fn on_finalize(_: BlockNumberFor<T>) {

128

13995

            assert!(

129

13995

                <DidSetContainerAuthorData<T>>::exists(),

130

1

                "Container chain author data needs to be present in every block!"

131

);

132

13994

133

134

135

72809

    #[pallet::call]

136

    impl<T: Config> Pallet<T> {

137

        #[pallet::call_index(0)]

138

        #[pallet::weight((T::WeightInfo::set_latest_author_data(<T::ContainerChains as GetCurrentContainerChains>::MaxContainerChains::get()), DispatchClass::Mandatory))]

139

        pub fn set_latest_author_data(

140

            origin: OriginFor<T>,

141

            data: InherentDataOf<T>,

142

14024

        ) -> DispatchResultWithPostInfo {

143

14024

            ensure_none(origin)?;

144

145

14024

            assert!(

146

14024

                !<DidSetContainerAuthorData<T>>::exists(),

147

1

                "DidSetContainerAuthorData must be updated only once in a block",

148

);

149

150

14023

            let registered_para_ids = T::ContainerChains::current_container_chains();

151

14023

            let mut total_weight =

152

14023

                T::WeightInfo::set_latest_author_data(registered_para_ids.len() as u32);

153

14023

154

14023

            // We do this first to make sure we don't do 2 reads (parachains and relay state)

155

14023

            // when we have no containers registered

156

14023

            // Essentially one can pass an empty proof if no container-chains are registered

157

14023

            if !registered_para_ids.is_empty() {

158

14013

                let storage_reader = T::RelayOrPara::create_storage_reader(data);

159

14013

160

14013

                let parent_tanssi_slot = u64::from(T::SlotBeacon::slot()).into();

161

162

                // TODO: we should probably fetch all authors-containers first

163

                // then pass the vector to the hook, this would allow for a better estimation

164

41654

                for para_id in registered_para_ids {

165

27645

                    match Self::fetch_block_info_from_proof(

166

27645

                        &storage_reader,

167

27645

                        para_id,

168

27645

                        parent_tanssi_slot,

169

27645

) {

170

13498

                        Ok(block_info) => {

171

13498

                            LatestAuthor::<T>::mutate(

172

13498

                                para_id,

173

13498

                                |maybe_old_block_info: &mut Option<

174

                                    ContainerChainBlockInfo<T::AccountId>,

175

13498

                                >| {

176

13498

                                    if let Some(ref mut old_block_info) = maybe_old_block_info {

177

13133

                                        if block_info.block_number > old_block_info.block_number {

178

13129

                                            // We only reward author if the block increases

179

13129

                                            total_weight = total_weight.saturating_add(

180

13129

                                                T::AuthorNotingHook::on_container_author_noted(

181

13129

                                                    &block_info.author,

182

13129

                                                    block_info.block_number,

183

13129

                                                    para_id,

184

13129

),

185

13129

);

186

13129

                                            let _ = core::mem::replace(old_block_info, block_info);

187

13129

188

365

                                    } else {

189

365

                                        // If there is no previous block, we should reward the author of the first block

190

365

                                        total_weight = total_weight.saturating_add(

191

365

                                            T::AuthorNotingHook::on_container_author_noted(

192

365

                                                &block_info.author,

193

365

                                                block_info.block_number,

194

365

                                                para_id,

195

365

),

196

365

);

197

365

                                        let _ = core::mem::replace(

198

365

                                            maybe_old_block_info,

199

365

                                            Some(block_info),

200

365

);

201

365

202

13498

},

203

13498

);

204

13498

205

14147

                        Err(e) => log::warn!(

206

14132

                            "Author-noting error {:?} found in para {:?}",

207

14132

e,

208

14132

                            u32::from(para_id)

209

),

210

211

212

10

213

214

            // We correctly set the data

215

14019

            DidSetContainerAuthorData::<T>::put(true);

216

14019

217

14019

            Ok(PostDispatchInfo {

218

14019

                actual_weight: Some(total_weight),

219

14019

                pays_fee: Pays::No,

220

14019

})

221

222

223

        #[pallet::call_index(1)]

224

        #[pallet::weight(T::WeightInfo::set_author())]

225

        pub fn set_author(

226

            origin: OriginFor<T>,

227

            para_id: ParaId,

228

            block_number: BlockNumber,

229

            author: T::AccountId,

230

            latest_slot_number: Slot,

231

19

        ) -> DispatchResult {

232

19

            ensure_root(origin)?;

233

16

            LatestAuthor::<T>::insert(

234

16

                para_id,

235

16

                ContainerChainBlockInfo {

236

16

                    block_number,

237

16

                    author: author.clone(),

238

16

                    latest_slot_number,

239

16

},

240

16

);

241

16

            Self::deposit_event(Event::LatestAuthorChanged {

242

16

                para_id,

243

16

                block_number,

244

16

                new_author: author,

245

16

                latest_slot_number,

246

16

});

247

16

            Ok(())

248

249

250

        #[pallet::call_index(2)]

251

        #[pallet::weight(T::WeightInfo::kill_author_data())]

252

59

        pub fn kill_author_data(origin: OriginFor<T>, para_id: ParaId) -> DispatchResult {

253

59

            ensure_root(origin)?;

254

56

            LatestAuthor::<T>::remove(para_id);

255

56

            Self::deposit_event(Event::RemovedAuthorData { para_id });

256

56

            Ok(())

257

258

259

260

    #[pallet::event]

261

72

    #[pallet::generate_deposit(pub(super) fn deposit_event)]

262

    pub enum Event<T: Config> {

263

1

        /// Latest author changed

264

        LatestAuthorChanged {

265

            para_id: ParaId,

266

            block_number: BlockNumber,

267

            new_author: T::AccountId,

268

            latest_slot_number: Slot,

269

},

270

1

        /// Removed author data

271

        RemovedAuthorData { para_id: ParaId },

272

273

274

13626

    #[pallet::storage]

275

    pub(super) type LatestAuthor<T: Config> =

276

        StorageMap<_, Blake2_128Concat, ParaId, ContainerChainBlockInfo<T::AccountId>, OptionQuery>;

277

278

    /// Was the containerAuthorData set?

279

112076

    #[pallet::storage]

280

    pub(super) type DidSetContainerAuthorData<T: Config> = StorageValue<_, bool, ValueQuery>;

281

282

    #[pallet::inherent]

283

    impl<T: Config> ProvideInherent for Pallet<T> {

284

        type Call = Call<T>;

285

        type Error = InherentError;

286

        // TODO, what should we put here

287

        const INHERENT_IDENTIFIER: InherentIdentifier =

288

            tp_author_noting_inherent::INHERENT_IDENTIFIER;

289

290

        fn is_inherent_required(_: &InherentData) -> Result<Option<Self::Error>, Self::Error> {

291

            // Return Ok(Some(_)) unconditionally because this inherent is required in every block

292

            Ok(Some(InherentError::Other(

293

                sp_runtime::RuntimeString::Borrowed("Pallet Author Noting Inherent required"),

294

)))

295

296

297

14000

        fn create_inherent(data: &InherentData) -> Option<Self::Call> {

298

14000

            let data = T::RelayOrPara::create_inherent_arg(data);

299

14000

300

14000

            Some(Call::set_latest_author_data { data })

301

14000

302

303

13984

        fn is_inherent(call: &Self::Call) -> bool {

304

13984

            matches!(call, Call::set_latest_author_data { .. })

305

13984

306

307

308

309

impl<T: Config> Pallet<T> {

310

    /// Fetch author and block number from a proof of header

311

27644

    fn fetch_block_info_from_proof<S: GenericStorageReader>(

312

27644

        relay_state_proof: &S,

313

27644

        para_id: ParaId,

314

27644

        tanssi_slot: Slot,

315

27644

    ) -> Result<ContainerChainBlockInfo<T::AccountId>, Error<T>> {

316

27644

        let bytes = para_id.twox_64_concat();

317

27644

        // CONCAT

318

27644

        let key = [PARAS_HEADS_INDEX, bytes.as_slice()].concat();

319

        // We might encounter empty vecs

320

        // We only note if we can decode

321

        // In this process several errors can occur, but we will only log if such errors happen

322

        // We first take the HeadData

323

        // If the readError was that the key was not provided (identified by the Proof error),

324

        // then panic

325

27644

        let head_data = relay_state_proof

326

27644

            .read_entry::<HeadData>(key.as_slice(), None)

327

27644

            .map_err(|e| match e {

328

3

                ReadEntryErr::Proof => panic!("Invalid proof provided for para head key"),

329

5

                _ => Error::<T>::FailedReading,

330

27644

            })?;

331

332

        // We later take the Header decoded

333

27639

        let author_header = sp_runtime::generic::Header::<BlockNumber, BlakeTwo256>::decode(

334

27639

            &mut head_data.0.as_slice(),

335

27639

336

27639

        .map_err(|_| Error::<T>::FailedDecodingHeader)?;

337

338

        // Return author from first aura log.

339

        // If there are no aura logs, it iterates over all the logs, then returns the error from the first element.

340

        // This is because it is hard to return a `Vec<Error<T>>`.

341

27635

        let mut first_error = None;

342

27635

        for aura_digest in author_header.digest().logs() {

343

27633

            match Self::author_from_log(aura_digest, para_id, &author_header, tanssi_slot) {

344

13498

                Ok(x) => return Ok(x),

345

14135

                Err(e) => {

346

14135

                    if first_error.is_none() {

347

14135

                        first_error = Some(e);

348

14135

349

350

351

352

353

14134

        Err(first_error.unwrap_or(Error::<T>::AuraDigestFirstItem))

354

27641

355

356

    /// Get block author from aura digest

357

27633

    fn author_from_log(

358

27633

        aura_digest: &DigestItem,

359

27633

        para_id: ParaId,

360

27633

        author_header: &sp_runtime::generic::Header<BlockNumber, BlakeTwo256>,

361

27633

        tanssi_slot: Slot,

362

27633

    ) -> Result<ContainerChainBlockInfo<T::AccountId>, Error<T>> {

363

        // We decode the digest as pre-runtime digest

364

27633

        let (id, mut data) = aura_digest

365

27633

            .as_pre_runtime()

366

27633

            .ok_or(Error::<T>::AsPreRuntimeError)?;

367

368

        // Match against the Aura digest

369

27633

        if id == AURA_ENGINE_ID {

370

            // DecodeSlot

371

27631

            let slot = InherentType::decode(&mut data).map_err(|_| Error::<T>::NonDecodableSlot)?;

372

373

            // Fetch Author

374

27630

            let author = T::ContainerChainAuthor::author_for_slot(slot, para_id)

375

27630

                .ok_or(Error::<T>::AuthorNotFound)?;

376

377

13498

            Ok(ContainerChainBlockInfo {

378

13498

                block_number: author_header.number,

379

13498

                author,

380

13498

                // We store the slot number of the current tanssi block to have a time-based notion

381

13498

                // of when the last block of a container chain was included.

382

13498

                // Note that this is not the slot of the container chain block, and it does not

383

13498

                // indicate when that block was created, but when it was included in tanssi.

384

13498

                latest_slot_number: tanssi_slot,

385

13498

})

386

        } else {

387

2

            Err(Error::<T>::NonAuraDigest)

388

389

27633

390

391

36

    pub fn latest_author(para_id: ParaId) -> Option<ContainerChainBlockInfo<T::AccountId>> {

392

36

        LatestAuthor::<T>::get(para_id)

393

36

394

395

396

#[derive(Encode)]

397

#[cfg_attr(feature = "std", derive(Debug, Decode))]

398

pub enum InherentError {

399

    Other(RuntimeString),

400

401

402

impl IsFatalError for InherentError {

403

    fn is_fatal_error(&self) -> bool {

404

        match *self {

405

            InherentError::Other(_) => true,

406

407

408

409

410

impl InherentError {

411

    /// Try to create an instance ouf of the given identifier and data.

412

    #[cfg(feature = "std")]

413

    pub fn try_from(id: &InherentIdentifier, data: &[u8]) -> Option<Self> {

414

        if id == &INHERENT_IDENTIFIER {

415

            <InherentError as parity_scale_codec::Decode>::decode(&mut &data[..]).ok()

416

        } else {

417

            None

418

419

420

421

422

impl<T: Config> LatestAuthorInfoFetcher<T::AccountId> for Pallet<T> {

423

20

    fn get_latest_author_info(para_id: ParaId) -> Option<ContainerChainBlockInfo<T::AccountId>> {

424

20

        LatestAuthor::<T>::get(para_id)

425

20

426

427

428

/// This pallet has slightly different behavior when used in a parachain vs when used in a relay chain

429

/// (solochain). The main difference is:

430

/// In relay mode, we don't need a storage proof, so the inherent doesn't need any input argument,

431

/// and instead of reading from a storage proof we read from storage directly.

432

pub trait RelayOrPara {

433

    type InherentArg: TypeInfo + Clone + PartialEq + Encode + Decode + core::fmt::Debug;

434

    type GenericStorageReader: GenericStorageReader;

435

436

    fn create_inherent_arg(data: &InherentData) -> Self::InherentArg;

437

    fn create_storage_reader(data: Self::InherentArg) -> Self::GenericStorageReader;

438

439

    #[cfg(feature = "runtime-benchmarks")]

440

    fn set_current_relay_chain_state(state: cumulus_pallet_parachain_system::RelayChainState);

441

442

443

pub type InherentDataOf<T> = <<T as Config>::RelayOrPara as RelayOrPara>::InherentArg;

444

445

pub struct RelayMode;

446

pub struct ParaMode<RCSP: RelaychainStateProvider>(PhantomData<RCSP>);

447

448

impl RelayOrPara for RelayMode {

449

    type InherentArg = ();

450

    type GenericStorageReader = NativeStorageReader;

451

452

    fn create_inherent_arg(_data: &InherentData) -> Self::InherentArg {

453

        // This ignores the inherent data entirely, so it is compatible with clients that don't add our inherent

454

455

456

56

    fn create_storage_reader(_data: Self::InherentArg) -> Self::GenericStorageReader {

457

56

        NativeStorageReader

458

56

459

460

    #[cfg(feature = "runtime-benchmarks")]

461

    fn set_current_relay_chain_state(_state: cumulus_pallet_parachain_system::RelayChainState) {}

462

463

464

impl<RCSP: RelaychainStateProvider> RelayOrPara for ParaMode<RCSP> {

465

    type InherentArg = tp_author_noting_inherent::OwnParachainInherentData;

466

    type GenericStorageReader = GenericStateProof<cumulus_primitives_core::relay_chain::Block>;

467

468

14000

    fn create_inherent_arg(data: &InherentData) -> Self::InherentArg {

469

14000

        data.get_data(&INHERENT_IDENTIFIER)

470

14000

            .ok()

471

14000

            .flatten()

472

14000

            .expect("there is not data to be posted; qed")

473

14000

474

475

14005

    fn create_storage_reader(data: Self::InherentArg) -> Self::GenericStorageReader {

476

14005

        let tp_author_noting_inherent::OwnParachainInherentData {

477

14005

            relay_storage_proof,

478

14005

        } = data;

479

14005

480

14005

        let relay_chain_state = RCSP::current_relay_chain_state();

481

14005

        let relay_storage_root = relay_chain_state.state_root;

482

14005

483

14005

        GenericStateProof::new(relay_storage_root, relay_storage_proof)

484

14005

            .expect("Invalid relay chain state proof")

485

14005

486

487

    #[cfg(feature = "runtime-benchmarks")]

488

    fn set_current_relay_chain_state(state: cumulus_pallet_parachain_system::RelayChainState) {

489

        RCSP::set_current_relay_chain_state(state)

490

491