148 lines
		
	
	
		
			5.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			148 lines
		
	
	
		
			5.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright 2016 The go-ethereum Authors
 | 
						|
// This file is part of the go-ethereum library.
 | 
						|
//
 | 
						|
// The go-ethereum library is free software: you can redistribute it and/or modify
 | 
						|
// it under the terms of the GNU Lesser General Public License as published by
 | 
						|
// the Free Software Foundation, either version 3 of the License, or
 | 
						|
// (at your option) any later version.
 | 
						|
//
 | 
						|
// The go-ethereum library 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 Lesser General Public License for more details.
 | 
						|
//
 | 
						|
// You should have received a copy of the GNU Lesser General Public License
 | 
						|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 | 
						|
// Package release contains the node service that tracks client releases.
 | 
						|
package release
 | 
						|
 | 
						|
import (
 | 
						|
	"fmt"
 | 
						|
	"strings"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"github.com/ethereum/go-ethereum/accounts/abi/bind"
 | 
						|
	"github.com/ethereum/go-ethereum/common"
 | 
						|
	"github.com/ethereum/go-ethereum/eth"
 | 
						|
	"github.com/ethereum/go-ethereum/logger"
 | 
						|
	"github.com/ethereum/go-ethereum/logger/glog"
 | 
						|
	"github.com/ethereum/go-ethereum/node"
 | 
						|
	"github.com/ethereum/go-ethereum/p2p"
 | 
						|
	"github.com/ethereum/go-ethereum/rpc"
 | 
						|
)
 | 
						|
 | 
						|
// Interval to check for new releases
 | 
						|
const releaseRecheckInterval = time.Hour
 | 
						|
 | 
						|
// Config contains the configurations of the release service.
 | 
						|
type Config struct {
 | 
						|
	Oracle common.Address // Ethereum address of the release oracle
 | 
						|
	Major  uint32         // Major version component of the release
 | 
						|
	Minor  uint32         // Minor version component of the release
 | 
						|
	Patch  uint32         // Patch version component of the release
 | 
						|
	Commit [20]byte       // Git SHA1 commit hash of the release
 | 
						|
}
 | 
						|
 | 
						|
// ReleaseService is a node service that periodically checks the blockchain for
 | 
						|
// newly released versions of the client being run and issues a warning to the
 | 
						|
// user about it.
 | 
						|
type ReleaseService struct {
 | 
						|
	config Config          // Current version to check releases against
 | 
						|
	oracle *ReleaseOracle  // Native binding to the release oracle contract
 | 
						|
	quit   chan chan error // Quit channel to terminate the version checker
 | 
						|
}
 | 
						|
 | 
						|
// NewReleaseService creates a new service to periodically check for new client
 | 
						|
// releases and notify the user of such.
 | 
						|
func NewReleaseService(ctx *node.ServiceContext, config Config) (node.Service, error) {
 | 
						|
	// Retrieve the Ethereum service dependency to access the blockchain
 | 
						|
	var ethereum *eth.Ethereum
 | 
						|
	if err := ctx.Service(ðereum); err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	// Construct the release service
 | 
						|
	contract, err := NewReleaseOracle(config.Oracle, eth.NewContractBackend(ethereum))
 | 
						|
	if err != nil {
 | 
						|
		return nil, err
 | 
						|
	}
 | 
						|
	return &ReleaseService{
 | 
						|
		config: config,
 | 
						|
		oracle: contract,
 | 
						|
		quit:   make(chan chan error),
 | 
						|
	}, nil
 | 
						|
}
 | 
						|
 | 
						|
// Protocols returns an empty list of P2P protocols as the release service does
 | 
						|
// not have a networking component.
 | 
						|
func (r *ReleaseService) Protocols() []p2p.Protocol { return nil }
 | 
						|
 | 
						|
// APIs returns an empty list of RPC descriptors as the release service does not
 | 
						|
// expose any functioanlity to the outside world.
 | 
						|
func (r *ReleaseService) APIs() []rpc.API { return nil }
 | 
						|
 | 
						|
// Start spawns the periodic version checker goroutine
 | 
						|
func (r *ReleaseService) Start(server *p2p.Server) error {
 | 
						|
	go r.checker()
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
// Stop terminates all goroutines belonging to the service, blocking until they
 | 
						|
// are all terminated.
 | 
						|
func (r *ReleaseService) Stop() error {
 | 
						|
	errc := make(chan error)
 | 
						|
	r.quit <- errc
 | 
						|
	return <-errc
 | 
						|
}
 | 
						|
 | 
						|
// checker runs indefinitely in the background, periodically checking for new
 | 
						|
// client releases.
 | 
						|
func (r *ReleaseService) checker() {
 | 
						|
	// Set up the timers to periodically check for releases
 | 
						|
	timer := time.NewTimer(0) // Immediately fire a version check
 | 
						|
	defer timer.Stop()
 | 
						|
 | 
						|
	for {
 | 
						|
		select {
 | 
						|
		// If the time arrived, check for a new release
 | 
						|
		case <-timer.C:
 | 
						|
			// Rechedule the timer before continuing
 | 
						|
			timer.Reset(releaseRecheckInterval)
 | 
						|
 | 
						|
			// Retrieve the current version, and handle missing contracts gracefully
 | 
						|
			version, err := r.oracle.CurrentVersion(nil)
 | 
						|
			if err != nil {
 | 
						|
				if err == bind.ErrNoCode {
 | 
						|
					glog.V(logger.Debug).Infof("Release oracle not found at %x", r.config.Oracle)
 | 
						|
					continue
 | 
						|
				}
 | 
						|
				glog.V(logger.Error).Infof("Failed to retrieve current release: %v", err)
 | 
						|
				continue
 | 
						|
			}
 | 
						|
			// Version was successfully retrieved, notify if newer than ours
 | 
						|
			if version.Major > r.config.Major ||
 | 
						|
				(version.Major == r.config.Major && version.Minor > r.config.Minor) ||
 | 
						|
				(version.Major == r.config.Major && version.Minor == r.config.Minor && version.Patch > r.config.Patch) {
 | 
						|
 | 
						|
				warning := fmt.Sprintf("Client v%d.%d.%d-%x seems older than the latest upstream release v%d.%d.%d-%x",
 | 
						|
					r.config.Major, r.config.Minor, r.config.Patch, r.config.Commit[:4], version.Major, version.Minor, version.Patch, version.Commit[:4])
 | 
						|
				howtofix := fmt.Sprintf("Please check https://github.com/ethereum/go-ethereum/releases for new releases")
 | 
						|
				separator := strings.Repeat("-", len(warning))
 | 
						|
 | 
						|
				glog.V(logger.Warn).Info(separator)
 | 
						|
				glog.V(logger.Warn).Info(warning)
 | 
						|
				glog.V(logger.Warn).Info(howtofix)
 | 
						|
				glog.V(logger.Warn).Info(separator)
 | 
						|
			} else {
 | 
						|
				glog.V(logger.Debug).Infof("Client v%d.%d.%d-%x seems up to date with upstream v%d.%d.%d-%x",
 | 
						|
					r.config.Major, r.config.Minor, r.config.Patch, r.config.Commit[:4], version.Major, version.Minor, version.Patch, version.Commit[:4])
 | 
						|
			}
 | 
						|
 | 
						|
		// If termination was requested, return
 | 
						|
		case errc := <-r.quit:
 | 
						|
			errc <- nil
 | 
						|
			return
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 |