func TestGetNeighbor(test *testing.T) {
assert := assert.New(test)
s := server.NewBgpServer()
go s.Serve()
g := api.NewGrpcServer(s, ":50051")
go g.Serve()
cli, err := NewGoBGPClient("")
err = cli.StartServer(&config.Global{
Config: config.GlobalConfig{
As: 1,
RouterId: "1.1.1.1",
Port: 1790,
},
})
assert.Nil(err)
err = cli.AddNeighbor(&config.Neighbor{
Config: config.NeighborConfig{
NeighborAddress: "10.0.0.1",
PeerAs: 2,
},
})
assert.Nil(err)
_, err = cli.GetNeighbor("10.0.0.1")
assert.Nil(err)
_, err = cli.GetNeighbor("10.0.0.2")
assert.Equal(err.Error(), "not found neighbor 10.0.0.2")
}
//createBgpServer creates and starts a bgp server and correspoinding grpc server
func createBgpServer() (bgpServer *bgpserver.BgpServer, grpcServer *bgpserver.Server) {
bgpServer = bgpserver.NewBgpServer(bgp.BGP_PORT)
if bgpServer == nil {
log.Errorf("Error creating bgp server")
return
} else {
go bgpServer.Serve()
}
// start grpc Server
grpcServer = bgpserver.NewGrpcServer(bgpserver.GRPC_PORT, bgpServer.GrpcReqCh)
if grpcServer == nil {
log.Errorf("Error creating bgp server")
return
} else {
go grpcServer.Serve()
}
return
}
//createBgpServer creates and starts a bgp server and correspoinding grpc server
func createBgpServer() (bgpServer *gobgp.BgpServer, grpcServer *api.Server) {
bgpServer = gobgp.NewBgpServer()
if bgpServer == nil {
log.Errorf("Error creating bgp server")
return
} else {
go bgpServer.Serve()
}
// start grpc Server
grpcServer = api.NewGrpcServer(bgpServer, ":50051")
if grpcServer == nil {
log.Errorf("Error creating bgp server")
return
} else {
go grpcServer.Serve()
}
return
}
func Crud(crud int, in *nlan.Router, con *context.Context) {
loopback := in.Loopback
embedded := in.EmbeddedBgp // quagga-bgpd(false) or gobgp(true)
ospf := in.GetOspf()
bgp := in.GetBgp()
cmd, _ := con.GetCmd()
log.Print("Router called...")
var s *gobgp.BgpServer
var g *gobgp.Server
if embedded {
cmd("/etc/init.d/quagga", "stop")
s = gobgp.NewBgpServer()
go s.Serve()
g = gobgp.NewGrpcServer(50051, s.GrpcReqCh)
go g.Serve()
}
var crudRouter func(string, bool, *gobgp.BgpServer, []*nlan.Ospf, map[string]*nlan.Attrs, *context.Context)
switch crud {
case env.ADD:
crudRouter = addRouter
case env.UPDATE:
crudRouter = updateRouter
case env.DELETE:
crudRouter = deleteRouter
default:
log.Fatal("CRUD unidentified")
}
log.Printf("Loopback: %s", loopback)
crudRouter(loopback, embedded, s, ospf, bgp, con)
log.Print("crudRouter() completed")
}
func main() {
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGTERM, syscall.SIGUSR1)
var opts struct {
ConfigFile string `short:"f" long:"config-file" description:"specifying a config file"`
ConfigType string `short:"t" long:"config-type" description:"specifying config type (toml, yaml, json)" default:"toml"`
LogLevel string `short:"l" long:"log-level" description:"specifying log level"`
LogPlain bool `short:"p" long:"log-plain" description:"use plain format for logging (json by default)"`
UseSyslog string `short:"s" long:"syslog" description:"use syslogd"`
Facility string `long:"syslog-facility" description:"specify syslog facility"`
DisableStdlog bool `long:"disable-stdlog" description:"disable standard logging"`
CPUs int `long:"cpus" description:"specify the number of CPUs to be used"`
Ops bool `long:"openswitch" description:"openswitch mode"`
GrpcHosts string `long:"api-hosts" description:"specify the hosts that gobgpd listens on" default:":50051"`
GracefulRestart bool `short:"r" long:"graceful-restart" description:"flag restart-state in graceful-restart capability"`
Dry bool `short:"d" long:"dry-run" description:"check configuration"`
PProfHost string `long:"pprof-host" description:"specify the host that gobgpd listens on for pprof" default:"localhost:6060"`
PProfDisable bool `long:"pprof-disable" description:"disable pprof profiling"`
}
_, err := flags.Parse(&opts)
if err != nil {
os.Exit(1)
}
if opts.CPUs == 0 {
runtime.GOMAXPROCS(runtime.NumCPU())
} else {
if runtime.NumCPU() < opts.CPUs {
log.Errorf("Only %d CPUs are available but %d is specified", runtime.NumCPU(), opts.CPUs)
os.Exit(1)
}
runtime.GOMAXPROCS(opts.CPUs)
}
if !opts.PProfDisable {
go func() {
log.Println(http.ListenAndServe(opts.PProfHost, nil))
}()
}
switch opts.LogLevel {
case "debug":
log.SetLevel(log.DebugLevel)
case "info":
log.SetLevel(log.InfoLevel)
default:
log.SetLevel(log.InfoLevel)
}
if opts.DisableStdlog == true {
log.SetOutput(ioutil.Discard)
} else {
log.SetOutput(os.Stdout)
}
if opts.UseSyslog != "" {
dst := strings.SplitN(opts.UseSyslog, ":", 2)
network := ""
addr := ""
if len(dst) == 2 {
network = dst[0]
addr = dst[1]
}
facility := syslog.Priority(0)
switch opts.Facility {
case "kern":
facility = syslog.LOG_KERN
case "user":
facility = syslog.LOG_USER
case "mail":
facility = syslog.LOG_MAIL
case "daemon":
facility = syslog.LOG_DAEMON
case "auth":
facility = syslog.LOG_AUTH
case "syslog":
facility = syslog.LOG_SYSLOG
case "lpr":
facility = syslog.LOG_LPR
case "news":
facility = syslog.LOG_NEWS
case "uucp":
facility = syslog.LOG_UUCP
case "cron":
facility = syslog.LOG_CRON
case "authpriv":
facility = syslog.LOG_AUTHPRIV
case "ftp":
facility = syslog.LOG_FTP
case "local0":
facility = syslog.LOG_LOCAL0
case "local1":
facility = syslog.LOG_LOCAL1
case "local2":
facility = syslog.LOG_LOCAL2
case "local3":
facility = syslog.LOG_LOCAL3
case "local4":
facility = syslog.LOG_LOCAL4
case "local5":
facility = syslog.LOG_LOCAL5
case "local6":
facility = syslog.LOG_LOCAL6
case "local7":
facility = syslog.LOG_LOCAL7
}
hook, err := logrus_syslog.NewSyslogHook(network, addr, syslog.LOG_INFO|facility, "bgpd")
if err != nil {
log.Error("Unable to connect to syslog daemon, ", opts.UseSyslog)
os.Exit(1)
} else {
log.AddHook(hook)
}
}
if opts.LogPlain {
if opts.DisableStdlog {
log.SetFormatter(&log.TextFormatter{
DisableColors: true,
})
}
} else {
log.SetFormatter(&log.JSONFormatter{})
}
configCh := make(chan *config.BgpConfigSet)
if opts.Dry {
go config.ReadConfigfileServe(opts.ConfigFile, opts.ConfigType, configCh)
c := <-configCh
if opts.LogLevel == "debug" {
p.Println(c)
}
os.Exit(0)
}
log.Info("gobgpd started")
bgpServer := server.NewBgpServer()
go bgpServer.Serve()
// start grpc Server
grpcServer := api.NewGrpcServer(bgpServer, opts.GrpcHosts)
go func() {
if err := grpcServer.Serve(); err != nil {
log.Fatalf("failed to listen grpc port: %s", err)
}
}()
if opts.Ops {
m, err := ops.NewOpsManager(opts.GrpcHosts)
if err != nil {
log.Errorf("Failed to start ops config manager: %s", err)
os.Exit(1)
}
log.Info("Coordination with OpenSwitch")
m.Serve()
} else if opts.ConfigFile != "" {
go config.ReadConfigfileServe(opts.ConfigFile, opts.ConfigType, configCh)
}
var c *config.BgpConfigSet = nil
for {
select {
case newConfig := <-configCh:
var added, deleted, updated []config.Neighbor
var updatePolicy bool
if c == nil {
c = newConfig
if err := bgpServer.Start(&newConfig.Global); err != nil {
log.Fatalf("failed to set global config: %s", err)
}
if newConfig.Zebra.Config.Enabled {
if err := bgpServer.StartZebraClient(&newConfig.Zebra); err != nil {
log.Fatalf("failed to set zebra config: %s", err)
}
}
if len(newConfig.Collector.Config.Url) > 0 {
if err := bgpServer.StartCollector(&newConfig.Collector.Config); err != nil {
log.Fatalf("failed to set collector config: %s", err)
}
}
for _, c := range newConfig.RpkiServers {
if err := bgpServer.AddRpki(&c.Config); err != nil {
log.Fatalf("failed to set rpki config: %s", err)
}
}
for _, c := range newConfig.BmpServers {
if err := bgpServer.AddBmp(&c.Config); err != nil {
log.Fatalf("failed to set bmp config: %s", err)
}
}
for _, c := range newConfig.MrtDump {
if len(c.Config.FileName) == 0 {
continue
}
if err := bgpServer.EnableMrt(&c.Config); err != nil {
log.Fatalf("failed to set mrt config: %s", err)
}
}
p := config.ConfigSetToRoutingPolicy(newConfig)
if err := bgpServer.UpdatePolicy(*p); err != nil {
log.Fatalf("failed to set routing policy: %s", err)
}
added = newConfig.Neighbors
if opts.GracefulRestart {
for i, n := range added {
if n.GracefulRestart.Config.Enabled {
added[i].GracefulRestart.State.LocalRestarting = true
}
}
}
} else {
added, deleted, updated, updatePolicy = config.UpdateConfig(c, newConfig)
if updatePolicy {
log.Info("Policy config is updated")
p := config.ConfigSetToRoutingPolicy(newConfig)
bgpServer.UpdatePolicy(*p)
}
c = newConfig
}
for i, p := range added {
log.Infof("Peer %v is added", p.Config.NeighborAddress)
bgpServer.AddNeighbor(&added[i])
}
for i, p := range deleted {
log.Infof("Peer %v is deleted", p.Config.NeighborAddress)
bgpServer.DeleteNeighbor(&deleted[i])
}
for i, p := range updated {
log.Infof("Peer %v is updated", p.Config.NeighborAddress)
u, _ := bgpServer.UpdateNeighbor(&updated[i])
updatePolicy = updatePolicy || u
}
if updatePolicy {
bgpServer.SoftResetIn("", bgp.RouteFamily(0))
}
case sig := <-sigCh:
switch sig {
case syscall.SIGKILL, syscall.SIGTERM:
bgpServer.Shutdown()
case syscall.SIGUSR1:
runtime.GC()
debug.FreeOSMemory()
}
}
}
}
func main() {
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGTERM)
var opts struct {
ConfigFile string `short:"f" long:"config-file" description:"specifying a config file"`
LogLevel string `short:"l" long:"log-level" description:"specifying log level"`
LogPlain bool `short:"p" long:"log-plain" description:"use plain format for logging (json by default)"`
UseSyslog string `short:"s" long:"syslog" description:"use syslogd"`
Facility string `long:"syslog-facility" description:"specify syslog facility"`
DisableStdlog bool `long:"disable-stdlog" description:"disable standard logging"`
CPUs int `long:"cpus" description:"specify the number of CPUs to be used"`
}
_, err := flags.Parse(&opts)
if err != nil {
os.Exit(1)
}
if opts.CPUs == 0 {
runtime.GOMAXPROCS(runtime.NumCPU())
} else {
if runtime.NumCPU() < opts.CPUs {
log.Errorf("Only %d CPUs are available but %d is specified", runtime.NumCPU(), opts.CPUs)
os.Exit(1)
}
runtime.GOMAXPROCS(opts.CPUs)
}
switch opts.LogLevel {
case "debug":
log.SetLevel(log.DebugLevel)
case "info":
log.SetLevel(log.InfoLevel)
default:
log.SetLevel(log.InfoLevel)
}
if opts.DisableStdlog == true {
log.SetOutput(ioutil.Discard)
} else {
log.SetOutput(os.Stdout)
}
if opts.UseSyslog != "" {
dst := strings.SplitN(opts.UseSyslog, ":", 2)
network := ""
addr := ""
if len(dst) == 2 {
network = dst[0]
addr = dst[1]
}
facility := syslog.Priority(0)
switch opts.Facility {
case "kern":
facility = syslog.LOG_KERN
case "user":
facility = syslog.LOG_USER
case "mail":
facility = syslog.LOG_MAIL
case "daemon":
facility = syslog.LOG_DAEMON
case "auth":
facility = syslog.LOG_AUTH
case "syslog":
facility = syslog.LOG_SYSLOG
case "lpr":
facility = syslog.LOG_LPR
case "news":
facility = syslog.LOG_NEWS
case "uucp":
facility = syslog.LOG_UUCP
case "cron":
facility = syslog.LOG_CRON
case "authpriv":
facility = syslog.LOG_AUTHPRIV
case "ftp":
facility = syslog.LOG_FTP
case "local0":
facility = syslog.LOG_LOCAL0
case "local1":
facility = syslog.LOG_LOCAL1
case "local2":
facility = syslog.LOG_LOCAL2
case "local3":
facility = syslog.LOG_LOCAL3
case "local4":
facility = syslog.LOG_LOCAL4
case "local5":
facility = syslog.LOG_LOCAL5
case "local6":
facility = syslog.LOG_LOCAL6
case "local7":
facility = syslog.LOG_LOCAL7
}
hook, err := logrus_syslog.NewSyslogHook(network, addr, syslog.LOG_INFO|facility, "bgpd")
if err != nil {
log.Error("Unable to connect to syslog daemon, ", opts.UseSyslog)
os.Exit(1)
} else {
log.AddHook(hook)
}
}
if opts.LogPlain == false {
log.SetFormatter(&log.JSONFormatter{})
}
if opts.ConfigFile == "" {
opts.ConfigFile = "gobgpd.conf"
}
configCh := make(chan config.BgpConfigSet)
reloadCh := make(chan bool)
go config.ReadConfigfileServe(opts.ConfigFile, configCh, reloadCh)
reloadCh <- true
bgpServer := server.NewBgpServer(bgp.BGP_PORT)
go bgpServer.Serve()
// start grpc Server
grpcServer := server.NewGrpcServer(server.GRPC_PORT, bgpServer.GrpcReqCh)
go grpcServer.Serve()
var bgpConfig *config.Bgp = nil
var policyConfig *config.RoutingPolicy = nil
for {
select {
case newConfig := <-configCh:
var added, deleted, updated []config.Neighbor
if bgpConfig == nil {
bgpServer.SetGlobalType(newConfig.Bgp.Global)
bgpConfig = &newConfig.Bgp
bgpServer.SetRpkiConfig(newConfig.Bgp.RpkiServers)
bgpServer.SetBmpConfig(newConfig.Bgp.BmpServers)
added = newConfig.Bgp.Neighbors.NeighborList
deleted = []config.Neighbor{}
updated = []config.Neighbor{}
} else {
bgpConfig, added, deleted, updated = config.UpdateConfig(bgpConfig, &newConfig.Bgp)
}
if policyConfig == nil {
policyConfig = &newConfig.Policy
bgpServer.SetPolicy(newConfig.Policy)
} else {
if config.CheckPolicyDifference(policyConfig, &newConfig.Policy) {
log.Info("Policy config is updated")
bgpServer.UpdatePolicy(newConfig.Policy)
}
}
for _, p := range added {
log.Infof("Peer %v is added", p.NeighborConfig.NeighborAddress)
bgpServer.PeerAdd(p)
}
for _, p := range deleted {
log.Infof("Peer %v is deleted", p.NeighborConfig.NeighborAddress)
bgpServer.PeerDelete(p)
}
for _, p := range updated {
log.Infof("Peer %v is updated", p.NeighborConfig.NeighborAddress)
bgpServer.PeerUpdate(p)
}
case sig := <-sigCh:
switch sig {
case syscall.SIGHUP:
log.Info("reload the config file")
reloadCh <- true
case syscall.SIGKILL, syscall.SIGTERM:
bgpServer.Shutdown()
}
}
}
}
func main() {
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGTERM, syscall.SIGUSR1)
var opts struct {
ConfigFile string `short:"f" long:"config-file" description:"specifying a config file"`
ConfigType string `short:"t" long:"config-type" description:"specifying config type (toml, yaml, json)" default:"toml"`
LogLevel string `short:"l" long:"log-level" description:"specifying log level"`
LogPlain bool `short:"p" long:"log-plain" description:"use plain format for logging (json by default)"`
UseSyslog string `short:"s" long:"syslog" description:"use syslogd"`
Facility string `long:"syslog-facility" description:"specify syslog facility"`
DisableStdlog bool `long:"disable-stdlog" description:"disable standard logging"`
CPUs int `long:"cpus" description:"specify the number of CPUs to be used"`
Ops bool `long:"openswitch" description:"openswitch mode"`
GrpcHosts string `long:"api-hosts" description:"specify the hosts that gobgpd listens on" default:":50051"`
GracefulRestart bool `short:"r" long:"graceful-restart" description:"flag restart-state in graceful-restart capability"`
Dry bool `short:"d" long:"dry-run" description:"check configuration"`
}
_, err := flags.Parse(&opts)
if err != nil {
os.Exit(1)
}
if opts.CPUs == 0 {
runtime.GOMAXPROCS(runtime.NumCPU())
} else {
if runtime.NumCPU() < opts.CPUs {
log.Errorf("Only %d CPUs are available but %d is specified", runtime.NumCPU(), opts.CPUs)
os.Exit(1)
}
runtime.GOMAXPROCS(opts.CPUs)
}
go func() {
log.Println(http.ListenAndServe("localhost:6060", nil))
}()
switch opts.LogLevel {
case "debug":
log.SetLevel(log.DebugLevel)
case "info":
log.SetLevel(log.InfoLevel)
default:
log.SetLevel(log.InfoLevel)
}
if opts.DisableStdlog == true {
log.SetOutput(ioutil.Discard)
} else {
log.SetOutput(os.Stdout)
}
if opts.UseSyslog != "" {
dst := strings.SplitN(opts.UseSyslog, ":", 2)
network := ""
addr := ""
if len(dst) == 2 {
network = dst[0]
addr = dst[1]
}
facility := syslog.Priority(0)
switch opts.Facility {
case "kern":
facility = syslog.LOG_KERN
case "user":
facility = syslog.LOG_USER
case "mail":
facility = syslog.LOG_MAIL
case "daemon":
facility = syslog.LOG_DAEMON
case "auth":
facility = syslog.LOG_AUTH
case "syslog":
facility = syslog.LOG_SYSLOG
case "lpr":
facility = syslog.LOG_LPR
case "news":
facility = syslog.LOG_NEWS
case "uucp":
facility = syslog.LOG_UUCP
case "cron":
facility = syslog.LOG_CRON
case "authpriv":
facility = syslog.LOG_AUTHPRIV
case "ftp":
facility = syslog.LOG_FTP
case "local0":
facility = syslog.LOG_LOCAL0
case "local1":
facility = syslog.LOG_LOCAL1
case "local2":
facility = syslog.LOG_LOCAL2
case "local3":
facility = syslog.LOG_LOCAL3
case "local4":
facility = syslog.LOG_LOCAL4
case "local5":
facility = syslog.LOG_LOCAL5
case "local6":
facility = syslog.LOG_LOCAL6
case "local7":
facility = syslog.LOG_LOCAL7
}
hook, err := logrus_syslog.NewSyslogHook(network, addr, syslog.LOG_INFO|facility, "bgpd")
if err != nil {
log.Error("Unable to connect to syslog daemon, ", opts.UseSyslog)
os.Exit(1)
} else {
log.AddHook(hook)
}
}
if opts.LogPlain {
if opts.DisableStdlog {
log.SetFormatter(&log.TextFormatter{
DisableColors: true,
})
}
} else {
log.SetFormatter(&log.JSONFormatter{})
}
configCh := make(chan *config.BgpConfigSet)
if opts.Dry {
go config.ReadConfigfileServe(opts.ConfigFile, opts.ConfigType, configCh)
c := <-configCh
if opts.LogLevel == "debug" {
p.Println(c)
}
os.Exit(0)
}
log.Info("gobgpd started")
bgpServer := server.NewBgpServer()
go bgpServer.Serve()
// start grpc Server
grpcServer := server.NewGrpcServer(opts.GrpcHosts, bgpServer.GrpcReqCh)
go func() {
if err := grpcServer.Serve(); err != nil {
log.Fatalf("failed to listen grpc port: %s", err)
}
}()
if opts.Ops {
m, err := ops.NewOpsManager(bgpServer.GrpcReqCh)
if err != nil {
log.Errorf("Failed to start ops config manager: %s", err)
os.Exit(1)
}
log.Info("Coordination with OpenSwitch")
m.Serve()
} else if opts.ConfigFile != "" {
go config.ReadConfigfileServe(opts.ConfigFile, opts.ConfigType, configCh)
}
var c *config.BgpConfigSet = nil
for {
select {
case newConfig := <-configCh:
var added, deleted, updated []config.Neighbor
var updatePolicy bool
if c == nil {
c = newConfig
if err := bgpServer.SetGlobalType(newConfig.Global); err != nil {
log.Fatalf("failed to set global config: %s", err)
}
if err := bgpServer.SetRpkiConfig(newConfig.RpkiServers); err != nil {
log.Fatalf("failed to set rpki config: %s", err)
}
if err := bgpServer.SetBmpConfig(newConfig.BmpServers); err != nil {
log.Fatalf("failed to set bmp config: %s", err)
}
if err := bgpServer.SetMrtConfig(newConfig.MrtDump); err != nil {
log.Fatalf("failed to set mrt config: %s", err)
}
p := config.ConfigSetToRoutingPolicy(newConfig)
if err := bgpServer.SetRoutingPolicy(*p); err != nil {
log.Fatalf("failed to set routing policy: %s", err)
}
added = newConfig.Neighbors
if opts.GracefulRestart {
for i, n := range added {
if n.GracefulRestart.Config.Enabled {
added[i].GracefulRestart.State.LocalRestarting = true
}
}
}
} else {
added, deleted, updated, updatePolicy = config.UpdateConfig(c, newConfig)
if updatePolicy {
log.Info("Policy config is updated")
p := config.ConfigSetToRoutingPolicy(newConfig)
bgpServer.UpdatePolicy(*p)
}
c = newConfig
}
for _, p := range added {
log.Infof("Peer %v is added", p.Config.NeighborAddress)
bgpServer.PeerAdd(p)
}
for _, p := range deleted {
log.Infof("Peer %v is deleted", p.Config.NeighborAddress)
bgpServer.PeerDelete(p)
}
for _, p := range updated {
log.Infof("Peer %v is updated", p.Config.NeighborAddress)
u, _ := bgpServer.PeerUpdate(p)
updatePolicy = updatePolicy || u
}
if updatePolicy {
// TODO: we want to apply the new policies to the existing
// routes here. Sending SOFT_RESET_IN to all the peers works
// for the change of in and import policies. SOFT_RESET_OUT is
// necessary for the export policy but we can't blindly
// execute SOFT_RESET_OUT because we unnecessarily advertize
// the existing routes. Needs to investigate the changes of
// policies and handle only affected peers.
ch := make(chan *server.GrpcResponse)
bgpServer.GrpcReqCh <- &server.GrpcRequest{
RequestType: server.REQ_NEIGHBOR_SOFT_RESET_IN,
Name: "all",
ResponseCh: ch,
}
<-ch
}
case sig := <-sigCh:
switch sig {
case syscall.SIGKILL, syscall.SIGTERM:
bgpServer.Shutdown()
case syscall.SIGUSR1:
runtime.GC()
debug.FreeOSMemory()
}
}
}
}
func main() {
sigCh := make(chan os.Signal, 1)
signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGTERM)
var opts struct {
ConfigFile string `short:"f" long:"config-file" description:"specifying a config file"`
ConfigType string `short:"t" long:"config-type" description:"specifying config type (toml, yaml, json)" default:"toml"`
LogLevel string `short:"l" long:"log-level" description:"specifying log level"`
LogPlain bool `short:"p" long:"log-plain" description:"use plain format for logging (json by default)"`
UseSyslog string `short:"s" long:"syslog" description:"use syslogd"`
Facility string `long:"syslog-facility" description:"specify syslog facility"`
DisableStdlog bool `long:"disable-stdlog" description:"disable standard logging"`
CPUs int `long:"cpus" description:"specify the number of CPUs to be used"`
Ops bool `long:"openswitch" description:"openswitch mode"`
GrpcPort int `long:"grpc-port" description:"grpc port" default:"50051"`
}
_, err := flags.Parse(&opts)
if err != nil {
os.Exit(1)
}
if opts.CPUs == 0 {
runtime.GOMAXPROCS(runtime.NumCPU())
} else {
if runtime.NumCPU() < opts.CPUs {
log.Errorf("Only %d CPUs are available but %d is specified", runtime.NumCPU(), opts.CPUs)
os.Exit(1)
}
runtime.GOMAXPROCS(opts.CPUs)
}
go func() {
log.Println(http.ListenAndServe("localhost:6060", nil))
}()
switch opts.LogLevel {
case "debug":
log.SetLevel(log.DebugLevel)
case "info":
log.SetLevel(log.InfoLevel)
default:
log.SetLevel(log.InfoLevel)
}
if opts.DisableStdlog == true {
log.SetOutput(ioutil.Discard)
} else {
log.SetOutput(os.Stdout)
}
if opts.UseSyslog != "" {
dst := strings.SplitN(opts.UseSyslog, ":", 2)
network := ""
addr := ""
if len(dst) == 2 {
network = dst[0]
addr = dst[1]
}
facility := syslog.Priority(0)
switch opts.Facility {
case "kern":
facility = syslog.LOG_KERN
case "user":
facility = syslog.LOG_USER
case "mail":
facility = syslog.LOG_MAIL
case "daemon":
facility = syslog.LOG_DAEMON
case "auth":
facility = syslog.LOG_AUTH
case "syslog":
facility = syslog.LOG_SYSLOG
case "lpr":
facility = syslog.LOG_LPR
case "news":
facility = syslog.LOG_NEWS
case "uucp":
facility = syslog.LOG_UUCP
case "cron":
facility = syslog.LOG_CRON
case "authpriv":
facility = syslog.LOG_AUTHPRIV
case "ftp":
facility = syslog.LOG_FTP
case "local0":
facility = syslog.LOG_LOCAL0
case "local1":
facility = syslog.LOG_LOCAL1
case "local2":
facility = syslog.LOG_LOCAL2
case "local3":
facility = syslog.LOG_LOCAL3
case "local4":
facility = syslog.LOG_LOCAL4
case "local5":
facility = syslog.LOG_LOCAL5
case "local6":
facility = syslog.LOG_LOCAL6
case "local7":
facility = syslog.LOG_LOCAL7
}
hook, err := logrus_syslog.NewSyslogHook(network, addr, syslog.LOG_INFO|facility, "bgpd")
if err != nil {
log.Error("Unable to connect to syslog daemon, ", opts.UseSyslog)
os.Exit(1)
} else {
log.AddHook(hook)
}
}
if opts.LogPlain == false {
log.SetFormatter(&log.JSONFormatter{})
}
log.Info("gobgpd started")
configCh := make(chan config.BgpConfigSet)
reloadCh := make(chan bool)
bgpServer := server.NewBgpServer()
if opts.Ops {
m, err := ops.NewOpsConfigManager(bgpServer.GrpcReqCh)
if err != nil {
log.Errorf("Failed to start ops config manager: %s", err)
os.Exit(1)
}
go m.Serve()
} else if opts.ConfigFile != "" {
go config.ReadConfigfileServe(opts.ConfigFile, opts.ConfigType, configCh, reloadCh)
reloadCh <- true
}
go bgpServer.Serve()
// start grpc Server
grpcServer := server.NewGrpcServer(opts.GrpcPort, bgpServer.GrpcReqCh)
go func() {
if err := grpcServer.Serve(); err != nil {
log.Fatalf("failed to listen grpc port: %s", err)
}
}()
var bgpConfig *config.Bgp = nil
var policyConfig *config.RoutingPolicy = nil
for {
select {
case newConfig := <-configCh:
var added, deleted, updated []config.Neighbor
if bgpConfig == nil {
bgpServer.SetGlobalType(newConfig.Bgp.Global)
bgpConfig = &newConfig.Bgp
bgpServer.SetRpkiConfig(newConfig.Bgp.RpkiServers)
added = newConfig.Bgp.Neighbors
deleted = []config.Neighbor{}
updated = []config.Neighbor{}
} else {
bgpConfig, added, deleted, updated = config.UpdateConfig(bgpConfig, &newConfig.Bgp)
}
if policyConfig == nil {
policyConfig = &newConfig.Policy
// FIXME: Currently the following code
// is safe because the above
// SetRpkiConfig will be blocked
// because the length of rpkiConfigCh
// is zero. So server.GlobalRib is
// allocated before the above
// SetPolicy. But this should be
// handled more cleanly.
if err := bgpServer.SetRoutingPolicy(newConfig.Policy); err != nil {
log.Fatal(err)
}
} else {
if config.CheckPolicyDifference(policyConfig, &newConfig.Policy) {
log.Info("Policy config is updated")
bgpServer.UpdatePolicy(newConfig.Policy)
}
}
for _, p := range added {
log.Infof("Peer %v is added", p.Config.NeighborAddress)
bgpServer.PeerAdd(p)
}
for _, p := range deleted {
log.Infof("Peer %v is deleted", p.Config.NeighborAddress)
bgpServer.PeerDelete(p)
}
for _, p := range updated {
log.Infof("Peer %v is updated", p.Config.NeighborAddress)
bgpServer.PeerUpdate(p)
}
case sig := <-sigCh:
switch sig {
case syscall.SIGHUP:
log.Info("reload the config file")
reloadCh <- true
case syscall.SIGKILL, syscall.SIGTERM:
bgpServer.Shutdown()
}
}
}
}
func NewBgpService(conf *config.BalancerConfig) (*BgpService, error) {
return &BgpService{
bgp: gobgp.NewBgpServer(),
config: conf,
}, nil
}