mirror of https://github.com/poanetwork/quorum.git
824 lines
28 KiB
Go
824 lines
28 KiB
Go
package permission
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math/big"
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/ethclient"
|
|
|
|
"github.com/ethereum/go-ethereum/event"
|
|
|
|
"github.com/ethereum/go-ethereum/raft"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts/abi/bind"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/eth"
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/ethereum/go-ethereum/node"
|
|
"github.com/ethereum/go-ethereum/p2p"
|
|
"github.com/ethereum/go-ethereum/p2p/enode"
|
|
"github.com/ethereum/go-ethereum/params"
|
|
pbind "github.com/ethereum/go-ethereum/permission/bind"
|
|
)
|
|
|
|
type NodeOperation uint8
|
|
|
|
const (
|
|
NodeAdd NodeOperation = iota
|
|
NodeDelete
|
|
)
|
|
|
|
type PermissionCtrl struct {
|
|
node *node.Node
|
|
ethClnt bind.ContractBackend
|
|
eth *eth.Ethereum
|
|
key *ecdsa.PrivateKey
|
|
dataDir string
|
|
permUpgr *pbind.PermUpgr
|
|
permInterf *pbind.PermInterface
|
|
permNode *pbind.NodeManager
|
|
permAcct *pbind.AcctManager
|
|
permRole *pbind.RoleManager
|
|
permOrg *pbind.OrgManager
|
|
permConfig *types.PermissionConfig
|
|
|
|
startWaitGroup *sync.WaitGroup // waitgroup to make sure all dependenies are ready before we start the service
|
|
stopFeed event.Feed // broadcasting stopEvent when service is being stopped
|
|
errorChan chan error // channel to capture error when starting aysnc
|
|
|
|
mux sync.Mutex
|
|
}
|
|
|
|
// to signal all watches when service is stopped
|
|
type stopEvent struct {
|
|
}
|
|
|
|
// function reads the permissions config file passed and populates the
|
|
// config structure accordingly
|
|
func ParsePermissionConfig(dir string) (types.PermissionConfig, error) {
|
|
fullPath := filepath.Join(dir, params.PERMISSION_MODEL_CONFIG)
|
|
f, err := os.Open(fullPath)
|
|
if err != nil {
|
|
log.Error("can't open file", "file", fullPath, "error", err)
|
|
return types.PermissionConfig{}, err
|
|
}
|
|
defer func() {
|
|
_ = f.Close()
|
|
}()
|
|
|
|
var permConfig types.PermissionConfig
|
|
blob, err := ioutil.ReadFile(fullPath)
|
|
if err != nil {
|
|
log.Error("error reading file", "err", err, "file", fullPath)
|
|
}
|
|
|
|
err = json.Unmarshal(blob, &permConfig)
|
|
if err != nil {
|
|
log.Error("error unmarshalling the file", "err", err, "file", fullPath)
|
|
}
|
|
|
|
if len(permConfig.Accounts) == 0 {
|
|
return types.PermissionConfig{}, fmt.Errorf("no accounts given in %s. Network cannot boot up", params.PERMISSION_MODEL_CONFIG)
|
|
}
|
|
if permConfig.SubOrgDepth.Cmp(big.NewInt(0)) == 0 || permConfig.SubOrgBreadth.Cmp(big.NewInt(0)) == 0 {
|
|
return types.PermissionConfig{}, fmt.Errorf("sub org breadth depth not passed in %s. Network cannot boot up", params.PERMISSION_MODEL_CONFIG)
|
|
}
|
|
if permConfig.IsEmpty() {
|
|
return types.PermissionConfig{}, fmt.Errorf("missing contract addresses in %s", params.PERMISSION_MODEL_CONFIG)
|
|
}
|
|
|
|
return permConfig, nil
|
|
}
|
|
|
|
// Create a service instance for permissioning
|
|
//
|
|
// Permission Service depends on the following:
|
|
// 1. EthService to be ready
|
|
// 2. Downloader to sync up blocks
|
|
// 3. InProc RPC server to be ready
|
|
func NewQuorumPermissionCtrl(stack *node.Node, pconfig *types.PermissionConfig) (*PermissionCtrl, error) {
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
p := &PermissionCtrl{
|
|
node: stack,
|
|
key: stack.GetNodeKey(),
|
|
dataDir: stack.DataDir(),
|
|
permConfig: pconfig,
|
|
startWaitGroup: wg,
|
|
errorChan: make(chan error),
|
|
}
|
|
go func(_wg *sync.WaitGroup) {
|
|
log.Debug("permission service: waiting for InProcRPC Server")
|
|
stopChan, stopSubscription := p.subscribeStopEvent()
|
|
inProcRPCServerSub := stack.EventMux().Subscribe(rpc.InProcServerReadyEvent{})
|
|
defer func(start time.Time) {
|
|
log.Debug("permission service: InProcRPC server is ready", "took", time.Since(start))
|
|
stopSubscription.Unsubscribe()
|
|
inProcRPCServerSub.Unsubscribe()
|
|
_wg.Done()
|
|
}(time.Now())
|
|
select {
|
|
case <-inProcRPCServerSub.Chan():
|
|
case <-stopChan:
|
|
}
|
|
}(wg) // wait for inproc RPC to be ready
|
|
return p, nil
|
|
}
|
|
|
|
func (p *PermissionCtrl) bindContract(contractInstance interface{}, bindFunc func() (interface{}, error)) error {
|
|
element := reflect.ValueOf(contractInstance).Elem()
|
|
instance, err := bindFunc()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
element.Set(reflect.ValueOf(instance))
|
|
return nil
|
|
}
|
|
|
|
// This is to make sure all contract instances are ready and initialized
|
|
//
|
|
// Required to be call after standard service start lifecycle
|
|
func (p *PermissionCtrl) AfterStart() error {
|
|
log.Debug("permission service: binding contracts")
|
|
err := <-p.errorChan // capture any error happened during asyncStart. Also wait here if asyncStart is not yet finish
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := p.bindContract(&p.permUpgr, func() (interface{}, error) { return pbind.NewPermUpgr(p.permConfig.UpgrdAddress, p.ethClnt) }); err != nil {
|
|
return err
|
|
}
|
|
if err := p.bindContract(&p.permInterf, func() (interface{}, error) { return pbind.NewPermInterface(p.permConfig.InterfAddress, p.ethClnt) }); err != nil {
|
|
return err
|
|
}
|
|
if err := p.bindContract(&p.permAcct, func() (interface{}, error) { return pbind.NewAcctManager(p.permConfig.AccountAddress, p.ethClnt) }); err != nil {
|
|
return err
|
|
}
|
|
if err := p.bindContract(&p.permNode, func() (interface{}, error) { return pbind.NewNodeManager(p.permConfig.NodeAddress, p.ethClnt) }); err != nil {
|
|
return err
|
|
}
|
|
if err := p.bindContract(&p.permRole, func() (interface{}, error) { return pbind.NewRoleManager(p.permConfig.RoleAddress, p.ethClnt) }); err != nil {
|
|
return err
|
|
}
|
|
if err := p.bindContract(&p.permOrg, func() (interface{}, error) { return pbind.NewOrgManager(p.permConfig.OrgAddress, p.ethClnt) }); err != nil {
|
|
return err
|
|
}
|
|
|
|
// populate the initial list of permissioned nodes and account accesses
|
|
if err := p.populateInitPermissions(); err != nil {
|
|
return fmt.Errorf("populateInitPermissions failed: %v", err)
|
|
}
|
|
|
|
// set the default access to ReadOnly
|
|
types.SetDefaults(p.permConfig.NwAdminRole, p.permConfig.OrgAdminRole)
|
|
|
|
for _, f := range []func() error{
|
|
p.manageOrgPermissions, // monitor org management related events
|
|
p.manageNodePermissions, // monitor org level node management events
|
|
p.manageRolePermissions, // monitor org level role management events
|
|
p.manageAccountPermissions, // monitor org level account management events
|
|
} {
|
|
if err := f(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
log.Info("permission service: is now ready")
|
|
|
|
return nil
|
|
}
|
|
|
|
// start service asynchronously due to dependencies
|
|
func (p *PermissionCtrl) asyncStart() {
|
|
var ethereum *eth.Ethereum
|
|
// will be blocked here until node is up
|
|
if err := p.node.Service(ðereum); err != nil {
|
|
p.errorChan <- fmt.Errorf("dependent ethereum service not started")
|
|
return
|
|
}
|
|
defer func() {
|
|
p.errorChan <- nil
|
|
}()
|
|
// for cases where the node is joining an existing network, permission service
|
|
// can be brought up only after block syncing is complete. This function
|
|
// waits for block syncing before the starting permissions
|
|
p.startWaitGroup.Add(1)
|
|
go func(_wg *sync.WaitGroup) {
|
|
log.Debug("permission service: waiting for downloader")
|
|
stopChan, stopSubscription := p.subscribeStopEvent()
|
|
pollingTicker := time.NewTicker(10 * time.Millisecond)
|
|
defer func(start time.Time) {
|
|
log.Debug("permission service: downloader completed", "took", time.Since(start))
|
|
stopSubscription.Unsubscribe()
|
|
pollingTicker.Stop()
|
|
_wg.Done()
|
|
}(time.Now())
|
|
for {
|
|
select {
|
|
case <-pollingTicker.C:
|
|
if types.GetSyncStatus() && !ethereum.Downloader().Synchronising() {
|
|
return
|
|
}
|
|
case <-stopChan:
|
|
return
|
|
}
|
|
}
|
|
}(p.startWaitGroup) // wait for downloader to sync if any
|
|
|
|
log.Debug("permission service: waiting for all dependencies to be ready")
|
|
p.startWaitGroup.Wait()
|
|
client, err := p.node.Attach()
|
|
if err != nil {
|
|
p.errorChan <- fmt.Errorf("unable to create rpc client: %v", err)
|
|
return
|
|
}
|
|
p.ethClnt = ethclient.NewClient(client)
|
|
p.eth = ethereum
|
|
}
|
|
|
|
func (p *PermissionCtrl) Start(srvr *p2p.Server) error {
|
|
log.Debug("permission service: starting")
|
|
go func() {
|
|
log.Debug("permission service: starting async")
|
|
p.asyncStart()
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func (p *PermissionCtrl) APIs() []rpc.API {
|
|
return []rpc.API{
|
|
{
|
|
Namespace: "quorumPermission",
|
|
Version: "1.0",
|
|
Service: NewQuorumControlsAPI(p),
|
|
Public: true,
|
|
},
|
|
}
|
|
}
|
|
|
|
func (p *PermissionCtrl) Protocols() []p2p.Protocol {
|
|
return []p2p.Protocol{}
|
|
}
|
|
|
|
func (p *PermissionCtrl) Stop() error {
|
|
log.Info("permission service: stopping")
|
|
p.stopFeed.Send(stopEvent{})
|
|
log.Info("permission service: stopped")
|
|
return nil
|
|
}
|
|
|
|
// monitors org management related events happening via smart contracts
|
|
// and updates cache accordingly
|
|
func (p *PermissionCtrl) manageOrgPermissions() error {
|
|
chPendingApproval := make(chan *pbind.OrgManagerOrgPendingApproval, 1)
|
|
chOrgApproved := make(chan *pbind.OrgManagerOrgApproved, 1)
|
|
chOrgSuspended := make(chan *pbind.OrgManagerOrgSuspended, 1)
|
|
chOrgReactivated := make(chan *pbind.OrgManagerOrgSuspensionRevoked, 1)
|
|
|
|
opts := &bind.WatchOpts{}
|
|
var blockNumber uint64 = 1
|
|
opts.Start = &blockNumber
|
|
|
|
if _, err := p.permOrg.OrgManagerFilterer.WatchOrgPendingApproval(opts, chPendingApproval); err != nil {
|
|
return fmt.Errorf("failed WatchNodePendingApproval: %v", err)
|
|
}
|
|
|
|
if _, err := p.permOrg.OrgManagerFilterer.WatchOrgApproved(opts, chOrgApproved); err != nil {
|
|
return fmt.Errorf("failed WatchNodePendingApproval: %v", err)
|
|
}
|
|
|
|
if _, err := p.permOrg.OrgManagerFilterer.WatchOrgSuspended(opts, chOrgSuspended); err != nil {
|
|
return fmt.Errorf("failed WatchNodePendingApproval: %v", err)
|
|
}
|
|
|
|
if _, err := p.permOrg.OrgManagerFilterer.WatchOrgSuspensionRevoked(opts, chOrgReactivated); err != nil {
|
|
return fmt.Errorf("failed WatchNodePendingApproval: %v", err)
|
|
}
|
|
|
|
go func() {
|
|
stopChan, stopSubscription := p.subscribeStopEvent()
|
|
defer stopSubscription.Unsubscribe()
|
|
for {
|
|
select {
|
|
case evtPendingApproval := <-chPendingApproval:
|
|
types.OrgInfoMap.UpsertOrg(evtPendingApproval.OrgId, evtPendingApproval.PorgId, evtPendingApproval.UltParent, evtPendingApproval.Level, types.OrgStatus(evtPendingApproval.Status.Uint64()))
|
|
|
|
case evtOrgApproved := <-chOrgApproved:
|
|
types.OrgInfoMap.UpsertOrg(evtOrgApproved.OrgId, evtOrgApproved.PorgId, evtOrgApproved.UltParent, evtOrgApproved.Level, types.OrgApproved)
|
|
|
|
case evtOrgSuspended := <-chOrgSuspended:
|
|
types.OrgInfoMap.UpsertOrg(evtOrgSuspended.OrgId, evtOrgSuspended.PorgId, evtOrgSuspended.UltParent, evtOrgSuspended.Level, types.OrgSuspended)
|
|
|
|
case evtOrgReactivated := <-chOrgReactivated:
|
|
types.OrgInfoMap.UpsertOrg(evtOrgReactivated.OrgId, evtOrgReactivated.PorgId, evtOrgReactivated.UltParent, evtOrgReactivated.Level, types.OrgApproved)
|
|
case <-stopChan:
|
|
log.Info("quit org contract watch")
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func (p *PermissionCtrl) subscribeStopEvent() (chan stopEvent, event.Subscription) {
|
|
c := make(chan stopEvent)
|
|
s := p.stopFeed.Subscribe(c)
|
|
return c, s
|
|
}
|
|
|
|
// Monitors node management events and updates cache accordingly
|
|
func (p *PermissionCtrl) manageNodePermissions() error {
|
|
chNodeApproved := make(chan *pbind.NodeManagerNodeApproved, 1)
|
|
chNodeProposed := make(chan *pbind.NodeManagerNodeProposed, 1)
|
|
chNodeDeactivated := make(chan *pbind.NodeManagerNodeDeactivated, 1)
|
|
chNodeActivated := make(chan *pbind.NodeManagerNodeActivated, 1)
|
|
chNodeBlacklisted := make(chan *pbind.NodeManagerNodeBlacklisted)
|
|
chNodeRecoveryInit := make(chan *pbind.NodeManagerNodeRecoveryInitiated, 1)
|
|
chNodeRecoveryDone := make(chan *pbind.NodeManagerNodeRecoveryCompleted, 1)
|
|
|
|
opts := &bind.WatchOpts{}
|
|
var blockNumber uint64 = 1
|
|
opts.Start = &blockNumber
|
|
|
|
if _, err := p.permNode.NodeManagerFilterer.WatchNodeApproved(opts, chNodeApproved); err != nil {
|
|
return fmt.Errorf("failed WatchNodeApproved: %v", err)
|
|
}
|
|
|
|
if _, err := p.permNode.NodeManagerFilterer.WatchNodeProposed(opts, chNodeProposed); err != nil {
|
|
return fmt.Errorf("failed WatchNodeProposed: %v", err)
|
|
}
|
|
|
|
if _, err := p.permNode.NodeManagerFilterer.WatchNodeDeactivated(opts, chNodeDeactivated); err != nil {
|
|
return fmt.Errorf("failed NodeDeactivated: %v", err)
|
|
}
|
|
if _, err := p.permNode.NodeManagerFilterer.WatchNodeActivated(opts, chNodeActivated); err != nil {
|
|
return fmt.Errorf("failed WatchNodeActivated: %v", err)
|
|
}
|
|
|
|
if _, err := p.permNode.NodeManagerFilterer.WatchNodeBlacklisted(opts, chNodeBlacklisted); err != nil {
|
|
return fmt.Errorf("failed NodeBlacklisting: %v", err)
|
|
}
|
|
|
|
if _, err := p.permNode.NodeManagerFilterer.WatchNodeRecoveryInitiated(opts, chNodeRecoveryInit); err != nil {
|
|
return fmt.Errorf("failed NodeRecoveryInitiated: %v", err)
|
|
}
|
|
|
|
if _, err := p.permNode.NodeManagerFilterer.WatchNodeRecoveryCompleted(opts, chNodeRecoveryDone); err != nil {
|
|
return fmt.Errorf("failed NodeRecoveryCompleted: %v", err)
|
|
}
|
|
|
|
go func() {
|
|
stopChan, stopSubscription := p.subscribeStopEvent()
|
|
defer stopSubscription.Unsubscribe()
|
|
for {
|
|
select {
|
|
case evtNodeApproved := <-chNodeApproved:
|
|
p.updatePermissionedNodes(evtNodeApproved.EnodeId, NodeAdd)
|
|
types.NodeInfoMap.UpsertNode(evtNodeApproved.OrgId, evtNodeApproved.EnodeId, types.NodeApproved)
|
|
|
|
case evtNodeProposed := <-chNodeProposed:
|
|
types.NodeInfoMap.UpsertNode(evtNodeProposed.OrgId, evtNodeProposed.EnodeId, types.NodePendingApproval)
|
|
|
|
case evtNodeDeactivated := <-chNodeDeactivated:
|
|
p.updatePermissionedNodes(evtNodeDeactivated.EnodeId, NodeDelete)
|
|
types.NodeInfoMap.UpsertNode(evtNodeDeactivated.OrgId, evtNodeDeactivated.EnodeId, types.NodeDeactivated)
|
|
|
|
case evtNodeActivated := <-chNodeActivated:
|
|
p.updatePermissionedNodes(evtNodeActivated.EnodeId, NodeAdd)
|
|
types.NodeInfoMap.UpsertNode(evtNodeActivated.OrgId, evtNodeActivated.EnodeId, types.NodeApproved)
|
|
|
|
case evtNodeBlacklisted := <-chNodeBlacklisted:
|
|
types.NodeInfoMap.UpsertNode(evtNodeBlacklisted.OrgId, evtNodeBlacklisted.EnodeId, types.NodeBlackListed)
|
|
p.updateDisallowedNodes(evtNodeBlacklisted.EnodeId, NodeAdd)
|
|
p.updatePermissionedNodes(evtNodeBlacklisted.EnodeId, NodeDelete)
|
|
|
|
case evtNodeRecoveryInit := <-chNodeRecoveryInit:
|
|
types.NodeInfoMap.UpsertNode(evtNodeRecoveryInit.OrgId, evtNodeRecoveryInit.EnodeId, types.NodeRecoveryInitiated)
|
|
|
|
case evtNodeRecoveryDone := <-chNodeRecoveryDone:
|
|
types.NodeInfoMap.UpsertNode(evtNodeRecoveryDone.OrgId, evtNodeRecoveryDone.EnodeId, types.NodeApproved)
|
|
p.updateDisallowedNodes(evtNodeRecoveryDone.EnodeId, NodeDelete)
|
|
p.updatePermissionedNodes(evtNodeRecoveryDone.EnodeId, NodeAdd)
|
|
|
|
case <-stopChan:
|
|
log.Info("quit node contract watch")
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
// adds or deletes and entry from a given file
|
|
func (p *PermissionCtrl) updateFile(fileName, enodeId string, operation NodeOperation, createFile bool) {
|
|
// Load the nodes from the config file
|
|
var nodeList []string
|
|
index := 0
|
|
// if createFile is false means the file is already existing. read the file
|
|
if !createFile {
|
|
blob, err := ioutil.ReadFile(fileName)
|
|
if err != nil && !createFile {
|
|
log.Error("Failed to access the file", "fileName", fileName, "err", err)
|
|
return
|
|
}
|
|
|
|
if err := json.Unmarshal(blob, &nodeList); err != nil {
|
|
log.Error("Failed to load nodes list from file", "fileName", fileName, "err", err)
|
|
return
|
|
}
|
|
|
|
// logic to update the permissioned-nodes.json file based on action
|
|
|
|
recExists := false
|
|
for i, eid := range nodeList {
|
|
if eid == enodeId {
|
|
index = i
|
|
recExists = true
|
|
break
|
|
}
|
|
}
|
|
if (operation == NodeAdd && recExists) || (operation == NodeDelete && !recExists) {
|
|
return
|
|
}
|
|
}
|
|
if operation == NodeAdd {
|
|
nodeList = append(nodeList, enodeId)
|
|
} else {
|
|
nodeList = append(nodeList[:index], nodeList[index+1:]...)
|
|
}
|
|
blob, _ := json.Marshal(nodeList)
|
|
|
|
p.mux.Lock()
|
|
defer p.mux.Unlock()
|
|
|
|
if err := ioutil.WriteFile(fileName, blob, 0644); err != nil {
|
|
log.Error("Error writing new node info to file", "fileName", fileName, "err", err)
|
|
}
|
|
}
|
|
|
|
// updates node information in the permissioned-nodes.json file based on node
|
|
// management activities in smart contract
|
|
func (p *PermissionCtrl) updatePermissionedNodes(enodeId string, operation NodeOperation) {
|
|
log.Debug("updatePermissionedNodes", "DataDir", p.dataDir, "file", params.PERMISSIONED_CONFIG)
|
|
|
|
path := filepath.Join(p.dataDir, params.PERMISSIONED_CONFIG)
|
|
if _, err := os.Stat(path); err != nil {
|
|
log.Error("Read Error for permissioned-nodes.json file. This is because 'permissioned' flag is specified but no permissioned-nodes.json file is present", "err", err)
|
|
return
|
|
}
|
|
|
|
p.updateFile(path, enodeId, operation, false)
|
|
if operation == NodeDelete {
|
|
p.disconnectNode(enodeId)
|
|
}
|
|
}
|
|
|
|
//this function populates the black listed node information into the disallowed-nodes.json file
|
|
func (p *PermissionCtrl) updateDisallowedNodes(url string, operation NodeOperation) {
|
|
log.Debug("updateDisallowedNodes", "DataDir", p.dataDir, "file", params.BLACKLIST_CONFIG)
|
|
|
|
fileExists := true
|
|
path := filepath.Join(p.dataDir, params.BLACKLIST_CONFIG)
|
|
// Check if the file is existing. If the file is not existing create the file
|
|
if _, err := os.Stat(path); err != nil {
|
|
log.Error("Read Error for disallowed-nodes.json file", "err", err)
|
|
if _, err := os.OpenFile(path, os.O_CREATE|os.O_RDWR, 0644); err != nil {
|
|
log.Error("Failed to create disallowed-nodes.json file", "err", err)
|
|
return
|
|
}
|
|
fileExists = false
|
|
}
|
|
|
|
if fileExists {
|
|
p.updateFile(path, url, operation, false)
|
|
} else {
|
|
p.updateFile(path, url, operation, true)
|
|
}
|
|
}
|
|
|
|
// Monitors account access related events and updates the cache accordingly
|
|
func (p *PermissionCtrl) manageAccountPermissions() error {
|
|
chAccessModified := make(chan *pbind.AcctManagerAccountAccessModified)
|
|
chAccessRevoked := make(chan *pbind.AcctManagerAccountAccessRevoked)
|
|
chStatusChanged := make(chan *pbind.AcctManagerAccountStatusChanged)
|
|
|
|
opts := &bind.WatchOpts{}
|
|
var blockNumber uint64 = 1
|
|
opts.Start = &blockNumber
|
|
|
|
if _, err := p.permAcct.AcctManagerFilterer.WatchAccountAccessModified(opts, chAccessModified); err != nil {
|
|
return fmt.Errorf("failed AccountAccessModified: %v", err)
|
|
}
|
|
|
|
if _, err := p.permAcct.AcctManagerFilterer.WatchAccountAccessRevoked(opts, chAccessRevoked); err != nil {
|
|
return fmt.Errorf("failed AccountAccessRevoked: %v", err)
|
|
}
|
|
|
|
if _, err := p.permAcct.AcctManagerFilterer.WatchAccountStatusChanged(opts, chStatusChanged); err != nil {
|
|
return fmt.Errorf("failed AccountStatusChanged: %v", err)
|
|
}
|
|
|
|
go func() {
|
|
stopChan, stopSubscription := p.subscribeStopEvent()
|
|
defer stopSubscription.Unsubscribe()
|
|
for {
|
|
select {
|
|
case evtAccessModified := <-chAccessModified:
|
|
types.AcctInfoMap.UpsertAccount(evtAccessModified.OrgId, evtAccessModified.RoleId, evtAccessModified.Account, evtAccessModified.OrgAdmin, types.AcctStatus(int(evtAccessModified.Status.Uint64())))
|
|
|
|
case evtAccessRevoked := <-chAccessRevoked:
|
|
types.AcctInfoMap.UpsertAccount(evtAccessRevoked.OrgId, evtAccessRevoked.RoleId, evtAccessRevoked.Account, evtAccessRevoked.OrgAdmin, types.AcctActive)
|
|
|
|
case evtStatusChanged := <-chStatusChanged:
|
|
ac := types.AcctInfoMap.GetAccount(evtStatusChanged.Account)
|
|
types.AcctInfoMap.UpsertAccount(evtStatusChanged.OrgId, ac.RoleId, evtStatusChanged.Account, ac.IsOrgAdmin, types.AcctStatus(int(evtStatusChanged.Status.Uint64())))
|
|
case <-stopChan:
|
|
log.Info("quit account contract watch")
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
// Disconnect the node from the network
|
|
func (p *PermissionCtrl) disconnectNode(enodeId string) {
|
|
if p.eth.ChainConfig().Istanbul == nil && p.eth.ChainConfig().Clique == nil {
|
|
var raftService *raft.RaftService
|
|
if err := p.node.Service(&raftService); err == nil {
|
|
raftApi := raft.NewPublicRaftAPI(raftService)
|
|
|
|
//get the raftId for the given enodeId
|
|
raftId, err := raftApi.GetRaftId(enodeId)
|
|
if err == nil {
|
|
raftApi.RemovePeer(raftId)
|
|
} else {
|
|
log.Error("failed to get raft id", "err", err, "enodeId", enodeId)
|
|
}
|
|
}
|
|
} else {
|
|
// Istanbul or clique - disconnect the peer
|
|
server := p.node.Server()
|
|
if server != nil {
|
|
node, err := enode.ParseV4(enodeId)
|
|
if err == nil {
|
|
server.RemovePeer(node)
|
|
} else {
|
|
log.Error("failed parse node id", "err", err, "enodeId", enodeId)
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
// Thus function checks if the initial network boot up status and if no
|
|
// populates permissions model with details from permission-config.json
|
|
func (p *PermissionCtrl) populateInitPermissions() error {
|
|
auth := bind.NewKeyedTransactor(p.key)
|
|
permInterfSession := &pbind.PermInterfaceSession{
|
|
Contract: p.permInterf,
|
|
CallOpts: bind.CallOpts{
|
|
Pending: true,
|
|
},
|
|
TransactOpts: bind.TransactOpts{
|
|
From: auth.From,
|
|
Signer: auth.Signer,
|
|
GasLimit: 47000000,
|
|
GasPrice: big.NewInt(0),
|
|
},
|
|
}
|
|
|
|
networkInitialized, err := permInterfSession.GetNetworkBootStatus()
|
|
if err != nil {
|
|
// handle the scenario of no contract code.
|
|
log.Warn("Failed to retrieve network boot status ", "err", err)
|
|
return err
|
|
}
|
|
|
|
if !networkInitialized {
|
|
if err := p.bootupNetwork(permInterfSession); err != nil {
|
|
return err
|
|
}
|
|
} else {
|
|
//populate orgs, nodes, roles and accounts from contract
|
|
for _, f := range []func(auth *bind.TransactOpts) error{
|
|
p.populateOrgsFromContract,
|
|
p.populateNodesFromContract,
|
|
p.populateRolesFromContract,
|
|
p.populateAccountsFromContract,
|
|
} {
|
|
if err := f(auth); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// initialize the permissions model and populate initial values
|
|
func (p *PermissionCtrl) bootupNetwork(permInterfSession *pbind.PermInterfaceSession) error {
|
|
if _, err := permInterfSession.SetPolicy(p.permConfig.NwAdminOrg, p.permConfig.NwAdminRole, p.permConfig.OrgAdminRole); err != nil {
|
|
log.Error("bootupNetwork SetPolicy failed", "err", err)
|
|
return err
|
|
}
|
|
if _, err := permInterfSession.Init(p.permConfig.SubOrgBreadth, p.permConfig.SubOrgDepth); err != nil {
|
|
log.Error("bootupNetwork init failed", "err", err)
|
|
return err
|
|
}
|
|
|
|
types.OrgInfoMap.UpsertOrg(p.permConfig.NwAdminOrg, "", p.permConfig.NwAdminOrg, big.NewInt(1), types.OrgApproved)
|
|
types.RoleInfoMap.UpsertRole(p.permConfig.NwAdminOrg, p.permConfig.NwAdminRole, true, true, types.FullAccess, true)
|
|
// populate the initial node list from static-nodes.json
|
|
if err := p.populateStaticNodesToContract(permInterfSession); err != nil {
|
|
return err
|
|
}
|
|
// populate initial account access to full access
|
|
if err := p.populateInitAccountAccess(permInterfSession); err != nil {
|
|
return err
|
|
}
|
|
|
|
// update network status to boot completed
|
|
if err := p.updateNetworkStatus(permInterfSession); err != nil {
|
|
log.Error("failed to updated network boot status", "error", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// populates the account access details from contract into cache
|
|
func (p *PermissionCtrl) populateAccountsFromContract(auth *bind.TransactOpts) error {
|
|
//populate accounts
|
|
permAcctSession := &pbind.AcctManagerSession{
|
|
Contract: p.permAcct,
|
|
CallOpts: bind.CallOpts{
|
|
Pending: true,
|
|
},
|
|
}
|
|
|
|
if numberOfRoles, err := permAcctSession.GetNumberOfAccounts(); err == nil {
|
|
iOrgNum := numberOfRoles.Uint64()
|
|
for k := uint64(0); k < iOrgNum; k++ {
|
|
if addr, org, role, status, orgAdmin, err := permAcctSession.GetAccountDetailsFromIndex(big.NewInt(int64(k))); err == nil {
|
|
types.AcctInfoMap.UpsertAccount(org, role, addr, orgAdmin, types.AcctStatus(int(status.Int64())))
|
|
}
|
|
}
|
|
} else {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// populates the role details from contract into cache
|
|
func (p *PermissionCtrl) populateRolesFromContract(auth *bind.TransactOpts) error {
|
|
//populate roles
|
|
permRoleSession := &pbind.RoleManagerSession{
|
|
Contract: p.permRole,
|
|
CallOpts: bind.CallOpts{
|
|
Pending: true,
|
|
},
|
|
}
|
|
if numberOfRoles, err := permRoleSession.GetNumberOfRoles(); err == nil {
|
|
iOrgNum := numberOfRoles.Uint64()
|
|
for k := uint64(0); k < iOrgNum; k++ {
|
|
if roleStruct, err := permRoleSession.GetRoleDetailsFromIndex(big.NewInt(int64(k))); err == nil {
|
|
types.RoleInfoMap.UpsertRole(roleStruct.OrgId, roleStruct.RoleId, roleStruct.Voter, roleStruct.Admin, types.AccessType(int(roleStruct.AccessType.Int64())), roleStruct.Active)
|
|
}
|
|
}
|
|
|
|
} else {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// populates the node details from contract into cache
|
|
func (p *PermissionCtrl) populateNodesFromContract(auth *bind.TransactOpts) error {
|
|
//populate nodes
|
|
permNodeSession := &pbind.NodeManagerSession{
|
|
Contract: p.permNode,
|
|
CallOpts: bind.CallOpts{
|
|
Pending: true,
|
|
},
|
|
}
|
|
if numberOfNodes, err := permNodeSession.GetNumberOfNodes(); err == nil {
|
|
iOrgNum := numberOfNodes.Uint64()
|
|
for k := uint64(0); k < iOrgNum; k++ {
|
|
if nodeStruct, err := permNodeSession.GetNodeDetailsFromIndex(big.NewInt(int64(k))); err == nil {
|
|
types.NodeInfoMap.UpsertNode(nodeStruct.OrgId, nodeStruct.EnodeId, types.NodeStatus(int(nodeStruct.NodeStatus.Int64())))
|
|
}
|
|
}
|
|
} else {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// populates the org details from contract into cache
|
|
func (p *PermissionCtrl) populateOrgsFromContract(auth *bind.TransactOpts) error {
|
|
//populate orgs
|
|
permOrgSession := &pbind.OrgManagerSession{
|
|
Contract: p.permOrg,
|
|
CallOpts: bind.CallOpts{
|
|
Pending: true,
|
|
},
|
|
}
|
|
if numberOfOrgs, err := permOrgSession.GetNumberOfOrgs(); err == nil {
|
|
iOrgNum := numberOfOrgs.Uint64()
|
|
for k := uint64(0); k < iOrgNum; k++ {
|
|
if orgId, porgId, ultParent, level, status, err := permOrgSession.GetOrgInfo(big.NewInt(int64(k))); err == nil {
|
|
types.OrgInfoMap.UpsertOrg(orgId, porgId, ultParent, level, types.OrgStatus(int(status.Int64())))
|
|
}
|
|
}
|
|
} else {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Reads the node list from static-nodes.json and populates into the contract
|
|
func (p *PermissionCtrl) populateStaticNodesToContract(permissionsSession *pbind.PermInterfaceSession) error {
|
|
nodes := p2p.ParsePermissionedNodes(p.dataDir)
|
|
for _, node := range nodes {
|
|
_, err := permissionsSession.AddAdminNode(node.String())
|
|
if err != nil {
|
|
log.Warn("Failed to propose node", "err", err, "enode", node.EnodeID())
|
|
return err
|
|
}
|
|
types.NodeInfoMap.UpsertNode(p.permConfig.NwAdminOrg, node.String(), 2)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Invokes the initAccounts function of smart contract to set the initial
|
|
// set of accounts access to full access
|
|
func (p *PermissionCtrl) populateInitAccountAccess(permissionsSession *pbind.PermInterfaceSession) error {
|
|
for _, a := range p.permConfig.Accounts {
|
|
_, er := permissionsSession.AddAdminAccount(a)
|
|
if er != nil {
|
|
log.Warn("Error adding permission initial account list", "err", er, "account", a)
|
|
return er
|
|
}
|
|
types.AcctInfoMap.UpsertAccount(p.permConfig.NwAdminOrg, p.permConfig.NwAdminRole, a, true, 2)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// updates network boot status to true
|
|
func (p *PermissionCtrl) updateNetworkStatus(permissionsSession *pbind.PermInterfaceSession) error {
|
|
_, err := permissionsSession.UpdateNetworkBootStatus()
|
|
if err != nil {
|
|
log.Warn("Failed to udpate network boot status ", "err", err)
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// monitors role management related events and updated cache
|
|
func (p *PermissionCtrl) manageRolePermissions() error {
|
|
chRoleCreated := make(chan *pbind.RoleManagerRoleCreated, 1)
|
|
chRoleRevoked := make(chan *pbind.RoleManagerRoleRevoked, 1)
|
|
|
|
opts := &bind.WatchOpts{}
|
|
var blockNumber uint64 = 1
|
|
opts.Start = &blockNumber
|
|
|
|
if _, err := p.permRole.RoleManagerFilterer.WatchRoleCreated(opts, chRoleCreated); err != nil {
|
|
return fmt.Errorf("failed WatchRoleCreated: %v", err)
|
|
}
|
|
|
|
if _, err := p.permRole.RoleManagerFilterer.WatchRoleRevoked(opts, chRoleRevoked); err != nil {
|
|
return fmt.Errorf("failed WatchRoleRemoved: %v", err)
|
|
}
|
|
|
|
go func() {
|
|
stopChan, stopSubscription := p.subscribeStopEvent()
|
|
defer stopSubscription.Unsubscribe()
|
|
for {
|
|
select {
|
|
case evtRoleCreated := <-chRoleCreated:
|
|
types.RoleInfoMap.UpsertRole(evtRoleCreated.OrgId, evtRoleCreated.RoleId, evtRoleCreated.IsVoter, evtRoleCreated.IsAdmin, types.AccessType(int(evtRoleCreated.BaseAccess.Uint64())), true)
|
|
|
|
case evtRoleRevoked := <-chRoleRevoked:
|
|
if r := types.RoleInfoMap.GetRole(evtRoleRevoked.OrgId, evtRoleRevoked.RoleId); r != nil {
|
|
types.RoleInfoMap.UpsertRole(evtRoleRevoked.OrgId, evtRoleRevoked.RoleId, r.IsVoter, r.IsAdmin, r.Access, false)
|
|
} else {
|
|
log.Error("Revoke role - cache is missing role", "org", evtRoleRevoked.OrgId, "role", evtRoleRevoked.RoleId)
|
|
}
|
|
case <-stopChan:
|
|
log.Info("quit role contract watch")
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
return nil
|
|
}
|