plugeth/eth/plugin_hooks.go

package eth

import (
  "github.com/ethereum/go-ethereum/plugins"
  "github.com/ethereum/go-ethereum/node"
  "github.com/ethereum/go-ethereum/params"
  "github.com/ethereum/go-ethereum/core/vm"
  "github.com/ethereum/go-ethereum/consensus"
  "github.com/ethereum/go-ethereum/consensus/ethash"
  "github.com/ethereum/go-ethereum/eth/ethconfig"
  "github.com/ethereum/go-ethereum/ethdb"
  "github.com/ethereum/go-ethereum/log"
)

func PluginCreateConsensusEngine(pl *plugins.PluginLoader, stack *node.Node, chainConfig *params.ChainConfig, config *ethash.Config, notify []string, noverify bool, db ethdb.Database) consensus.Engine {
  fnList := pl.Lookup("CreateConsensusEngine", func(item interface{}) bool {
    _, ok := item.(func(*node.Node, *params.ChainConfig, *ethash.Config, []string, bool, ethdb.Database) consensus.Engine)
    return ok
  })
  for _, fni := range fnList {
    if fn, ok := fni.(func(*node.Node, *params.ChainConfig, *ethash.Config, []string, bool, ethdb.Database) consensus.Engine); ok {
      return fn(stack, chainConfig, config, notify, noverify, db)
    }
  }
  return ethconfig.CreateConsensusEngine(stack, chainConfig, config, notify, noverify, db)
}

func pluginCreateConsensusEngine(stack *node.Node, chainConfig *params.ChainConfig, config *ethash.Config, notify []string, noverify bool, db ethdb.Database) consensus.Engine {
  if plugins.DefaultPluginLoader == nil {
		log.Warn("Attempting CreateConsensusEngine, but default PluginLoader has not been initialized")
		return ethconfig.CreateConsensusEngine(stack, chainConfig, config, notify, noverify, db)
	}
  return PluginCreateConsensusEngine(plugins.DefaultPluginLoader, stack, chainConfig, config, notify, noverify, db)
}

func PluginUpdateBlockchainVMConfig(pl *plugins.PluginLoader, cfg *vm.Config) {
  fnList := plugins.Lookup("UpdateBlockchainVMConfig", func(item interface{}) bool {
    _, ok := item.(func(*vm.Config))
    return ok
  })
  for _, fni := range fnList {
    if fn, ok := fni.(func(*vm.Config)); ok {
      fn(cfg)
      return
    }
  }
}


func pluginUpdateBlockchainVMConfig(cfg *vm.Config) {
  if plugins.DefaultPluginLoader == nil {
		log.Warn("Attempting CreateConsensusEngine, but default PluginLoader has not been initialized")
    return
  }
  PluginUpdateBlockchainVMConfig(plugins.DefaultPluginLoader, cfg)
}