go.ligato.io/vpp-agent/v3@v3.5.0/plugins/vpp/wireguardplugin/wireguardplugin.go (about)

     1  //  Copyright (c) 2020 Doc.ai and/or its affiliates.
     2  //
     3  //  Licensed under the Apache License, Version 2.0 (the "License");
     4  //  you may not use this file except in compliance with the License.
     5  //  You may obtain a copy of the License at:
     6  //
     7  //      http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  //  Unless required by applicable law or agreed to in writing, software
    10  //  distributed under the License is distributed on an "AS IS" BASIS,
    11  //  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  //  See the License for the specific language governing permissions and
    13  //  limitations under the License.
    14  
    15  //go:generate descriptor-adapter --descriptor-name Peer --value-type *vpp_wg.Peer --meta-type *wgidx.WgMetadata --import "go.ligato.io/vpp-agent/v3/plugins/vpp/wireguardplugin/wgidx" --import "go.ligato.io/vpp-agent/v3/proto/ligato/vpp/wireguard" --output-dir "descriptor"
    16  
    17  package wireguardplugin
    18  
    19  import (
    20  	"github.com/pkg/errors"
    21  	"go.ligato.io/cn-infra/v2/health/statuscheck"
    22  	"go.ligato.io/cn-infra/v2/infra"
    23  
    24  	"go.ligato.io/vpp-agent/v3/plugins/govppmux"
    25  	kvs "go.ligato.io/vpp-agent/v3/plugins/kvscheduler/api"
    26  	"go.ligato.io/vpp-agent/v3/plugins/vpp/ifplugin"
    27  	"go.ligato.io/vpp-agent/v3/plugins/vpp/wireguardplugin/descriptor"
    28  	"go.ligato.io/vpp-agent/v3/plugins/vpp/wireguardplugin/descriptor/adapter"
    29  	"go.ligato.io/vpp-agent/v3/plugins/vpp/wireguardplugin/vppcalls"
    30  
    31  	_ "go.ligato.io/vpp-agent/v3/plugins/vpp/wireguardplugin/vppcalls/vpp2106"
    32  	_ "go.ligato.io/vpp-agent/v3/plugins/vpp/wireguardplugin/vppcalls/vpp2202"
    33  	_ "go.ligato.io/vpp-agent/v3/plugins/vpp/wireguardplugin/vppcalls/vpp2210"
    34  )
    35  
    36  type WgPlugin struct {
    37  	Deps
    38  	// handler
    39  	WgHandler vppcalls.WgVppAPI
    40  
    41  	peerDescriptor *descriptor.WgPeerDescriptor
    42  }
    43  
    44  type Deps struct {
    45  	infra.PluginDeps
    46  	KVScheduler kvs.KVScheduler
    47  	VPP         govppmux.API
    48  	IfPlugin    ifplugin.API
    49  	StatusCheck statuscheck.PluginStatusWriter // optional
    50  }
    51  
    52  func (p *WgPlugin) Init() (err error) {
    53  	if !p.VPP.IsPluginLoaded("wireguard") {
    54  		p.Log.Warnf("VPP plugin wireguard was disabled by VPP")
    55  		return nil
    56  	}
    57  
    58  	// init Wg handler
    59  	p.WgHandler = vppcalls.CompatibleWgVppHandler(p.VPP, p.IfPlugin.GetInterfaceIndex(), p.Log)
    60  	if p.WgHandler == nil {
    61  		return errors.New("Wireguard handler is not available")
    62  	}
    63  
    64  	p.peerDescriptor = descriptor.NewWgPeerDescriptor(p.WgHandler, p.Log)
    65  	peerDescriptor := adapter.NewPeerDescriptor(p.peerDescriptor.GetDescriptor())
    66  	err = p.KVScheduler.RegisterKVDescriptor(peerDescriptor)
    67  	if err != nil {
    68  		return err
    69  	}
    70  
    71  	return nil
    72  }
    73  
    74  // AfterInit registers plugin with StatusCheck.
    75  func (p *WgPlugin) AfterInit() error {
    76  	if p.StatusCheck != nil {
    77  		p.StatusCheck.Register(p.PluginName, nil)
    78  	}
    79  	return nil
    80  }