| /* |
| * Copyright 2018-present Open Networking Foundation |
| |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package common |
| |
| import ( |
| "flag" |
| "fmt" |
| "io/ioutil" |
| "net" |
| "strings" |
| |
| "github.com/imdario/mergo" |
| log "github.com/sirupsen/logrus" |
| "gopkg.in/yaml.v2" |
| ) |
| |
| var tagAllocationValues = []string{ |
| "unknown", |
| "shared", |
| "unique", |
| } |
| |
| type TagAllocation int |
| |
| func (t TagAllocation) String() string { |
| return tagAllocationValues[t] |
| } |
| |
| func tagAllocationFromString(s string) (TagAllocation, error) { |
| for i, v := range tagAllocationValues { |
| if v == strings.TrimSpace(s) { |
| return TagAllocation(i), nil |
| } |
| } |
| log.WithFields(log.Fields{ |
| "ValidValues": strings.Join(tagAllocationValues[1:], ", "), |
| }).Errorf("%s-is-not-a-valid-tag-allocation", s) |
| return TagAllocation(0), fmt.Errorf("%s-is-not-a-valid-tag-allocation", s) |
| } |
| |
| const ( |
| _ TagAllocation = iota |
| TagAllocationShared |
| TagAllocationUnique |
| ) |
| |
| type BBRCliOptions struct { |
| *GlobalConfig |
| BBSimIp string |
| BBSimPort string |
| BBSimApiPort string |
| LogFile string |
| } |
| |
| type GlobalConfig struct { |
| BBSim BBSimConfig |
| Olt OltConfig |
| BBR BBRConfig |
| } |
| |
| type OltConfig struct { |
| Model string `yaml:"model"` |
| Vendor string `yaml:"vendor"` |
| HardwareVersion string `yaml:"hardware_version"` |
| FirmwareVersion string `yaml:"firmware_version"` |
| DeviceId string `yaml:"device_id"` |
| DeviceSerialNumber string `yaml:"device_serial_number"` |
| PonPorts uint32 `yaml:"pon_ports"` |
| NniPorts uint32 `yaml:"nni_ports"` |
| OnusPonPort uint32 `yaml:"onus_per_port"` |
| Technology string `yaml:"technology"` |
| ID int `yaml:"id"` |
| OltRebootDelay int `yaml:"reboot_delay"` |
| PortStatsInterval int `yaml:"port_stats_interval"` |
| } |
| |
| type BBSimConfig struct { |
| ConfigFile string |
| ServiceConfigFile string |
| DhcpRetry bool `yaml:"dhcp_retry"` |
| AuthRetry bool `yaml:"auth_retry"` |
| LogLevel string `yaml:"log_level"` |
| LogCaller bool `yaml:"log_caller"` |
| Delay int `yaml:"delay"` |
| CpuProfile *string `yaml:"cpu_profile"` |
| OpenOltAddress string `yaml:"openolt_address"` |
| ApiAddress string `yaml:"api_address"` |
| RestApiAddress string `yaml:"rest_api_address"` |
| LegacyApiAddress string `yaml:"legacy_api_address"` |
| LegacyRestApiAddress string `yaml:"legacy_rest_api_address"` |
| SadisRestAddress string `yaml:"sadis_rest_address"` |
| SadisServer bool `yaml:"sadis_server"` |
| KafkaAddress string `yaml:"kafka_address"` |
| Events bool `yaml:"enable_events"` |
| ControlledActivation string `yaml:"controlled_activation"` |
| EnablePerf bool `yaml:"enable_perf"` |
| KafkaEventTopic string `yaml:"kafka_event_topic"` |
| } |
| |
| type BBRConfig struct { |
| Log string `yaml:"log"` |
| LogLevel string `yaml:"log_level"` |
| LogCaller bool `yaml:"log_caller"` |
| } |
| |
| type ServiceYaml struct { |
| Name string |
| CTag int `yaml:"c_tag"` |
| STag int `yaml:"s_tag"` |
| NeedsEapol bool `yaml:"needs_eapol"` |
| NeedsDchp bool `yaml:"needs_dhcp"` |
| NeedsIgmp bool `yaml:"needs_igmp"` |
| CTagAllocation string `yaml:"c_tag_allocation"` |
| STagAllocation string `yaml:"s_tag_allocation"` |
| TechnologyProfileID int `yaml:"tp_id"` |
| UniTagMatch int `yaml:"uni_tag_match"` |
| ConfigureMacAddress bool `yaml:"configure_mac_address"` |
| UsPonCTagPriority uint8 `yaml:"us_pon_c_tag_priority"` |
| UsPonSTagPriority uint8 `yaml:"us_pon_s_tag_priority"` |
| DsPonCTagPriority uint8 `yaml:"ds_pon_c_tag_priority"` |
| DsPonSTagPriority uint8 `yaml:"ds_pon_s_tag_priority"` |
| } |
| type YamlServiceConfig struct { |
| Workflow string |
| Services []ServiceYaml `yaml:"services,flow"` |
| } |
| |
| func (cfg *YamlServiceConfig) String() string { |
| str := fmt.Sprintf("[workflow: %s, Services: ", cfg.Workflow) |
| |
| for _, s := range cfg.Services { |
| str = fmt.Sprintf("%s[", str) |
| str = fmt.Sprintf("%sname=%s, c_tag=%d, s_tag=%d, ", |
| str, s.Name, s.CTag, s.STag) |
| str = fmt.Sprintf("%sc_tag_allocation=%s, s_tag_allocation=%s, ", |
| str, s.CTagAllocation, s.STagAllocation) |
| str = fmt.Sprintf("%sneeds_eapol=%t, needs_dhcp=%t, needs_igmp=%t", |
| str, s.NeedsEapol, s.NeedsDchp, s.NeedsIgmp) |
| str = fmt.Sprintf("%stp_id=%d, uni_tag_match=%d", |
| str, s.TechnologyProfileID, s.UniTagMatch) |
| str = fmt.Sprintf("%s]", str) |
| } |
| str = fmt.Sprintf("%s]", str) |
| return str |
| } |
| |
| var ( |
| Config *GlobalConfig |
| Services []ServiceYaml |
| ) |
| |
| // Load the BBSim configuration. This is a combination of CLI parameters and YAML files |
| // We proceed in this order: |
| // - Read CLI parameters |
| // - Using those we read the yaml files (config and services) |
| // - we merge the configuration (CLI has priority over yaml files) |
| func LoadConfig() { |
| |
| Config = getDefaultOps() |
| |
| cliConf := readCliParams() |
| |
| yamlConf, err := loadBBSimConf(cliConf.BBSim.ConfigFile) |
| |
| if err != nil { |
| log.WithFields(log.Fields{ |
| "file": cliConf.BBSim.ConfigFile, |
| "err": err, |
| }).Fatal("Can't read config file") |
| } |
| |
| // merging Yaml and Default Values |
| if err := mergo.Merge(Config, yamlConf, mergo.WithOverride); err != nil { |
| log.WithFields(log.Fields{ |
| "err": err, |
| }).Fatal("Can't merge YAML and Config") |
| } |
| |
| // merging CLI values on top of the yaml ones |
| if err := mergo.Merge(Config, cliConf, mergo.WithOverride); err != nil { |
| log.WithFields(log.Fields{ |
| "err": err, |
| }).Fatal("Can't merge CLI and Config") |
| } |
| |
| services, err := loadBBSimServices(Config.BBSim.ServiceConfigFile) |
| |
| if err != nil { |
| log.WithFields(log.Fields{ |
| "file": Config.BBSim.ServiceConfigFile, |
| "err": err, |
| }).Fatal("Can't read services file") |
| } |
| |
| Services = services |
| |
| } |
| |
| func readCliParams() *GlobalConfig { |
| |
| conf := getDefaultOps() |
| |
| configFile := flag.String("config", conf.BBSim.ConfigFile, "Configuration file path") |
| servicesFile := flag.String("services", conf.BBSim.ServiceConfigFile, "Service Configuration file path") |
| |
| olt_id := flag.Int("olt_id", conf.Olt.ID, "OLT device ID") |
| nni := flag.Int("nni", int(conf.Olt.NniPorts), "Number of NNI ports per OLT device to be emulated") |
| pon := flag.Int("pon", int(conf.Olt.PonPorts), "Number of PON ports per OLT device to be emulated") |
| onu := flag.Int("onu", int(conf.Olt.OnusPonPort), "Number of ONU devices per PON port to be emulated") |
| |
| openolt_address := flag.String("openolt_address", conf.BBSim.OpenOltAddress, "IP address:port") |
| api_address := flag.String("api_address", conf.BBSim.ApiAddress, "IP address:port") |
| rest_api_address := flag.String("rest_api_address", conf.BBSim.RestApiAddress, "IP address:port") |
| |
| profileCpu := flag.String("cpuprofile", "", "write cpu profile to file") |
| |
| logLevel := flag.String("logLevel", conf.BBSim.LogLevel, "Set the log level (trace, debug, info, warn, error)") |
| logCaller := flag.Bool("logCaller", conf.BBSim.LogCaller, "Whether to print the caller filename or not") |
| |
| delay := flag.Int("delay", conf.BBSim.Delay, "The delay between ONU DISCOVERY batches in milliseconds (1 ONU per each PON PORT at a time") |
| |
| controlledActivation := flag.String("ca", conf.BBSim.ControlledActivation, "Set the mode for controlled activation of PON ports and ONUs") |
| enablePerf := flag.Bool("enableperf", conf.BBSim.EnablePerf, "Setting this flag will cause BBSim to not store data like traffic schedulers, flows of ONUs etc..") |
| enableEvents := flag.Bool("enableEvents", conf.BBSim.Events, "Enable sending BBSim events on configured kafka server") |
| kafkaAddress := flag.String("kafkaAddress", conf.BBSim.KafkaAddress, "IP:Port for kafka") |
| kafkaEventTopic := flag.String("kafkaEventTopic", conf.BBSim.KafkaEventTopic, "Ability to configure the topic on which BBSim publishes events on Kafka") |
| dhcpRetry := flag.Bool("dhcpRetry", conf.BBSim.DhcpRetry, "Set this flag if BBSim should retry DHCP upon failure until success") |
| authRetry := flag.Bool("authRetry", conf.BBSim.AuthRetry, "Set this flag if BBSim should retry EAPOL (Authentication) upon failure until success") |
| flag.Parse() |
| |
| conf.Olt.ID = int(*olt_id) |
| conf.Olt.NniPorts = uint32(*nni) |
| conf.Olt.PonPorts = uint32(*pon) |
| conf.Olt.OnusPonPort = uint32(*onu) |
| conf.BBSim.ConfigFile = *configFile |
| conf.BBSim.ServiceConfigFile = *servicesFile |
| conf.BBSim.CpuProfile = profileCpu |
| conf.BBSim.LogLevel = *logLevel |
| conf.BBSim.LogCaller = *logCaller |
| conf.BBSim.Delay = *delay |
| conf.BBSim.ControlledActivation = *controlledActivation |
| conf.BBSim.EnablePerf = *enablePerf |
| conf.BBSim.Events = *enableEvents |
| conf.BBSim.KafkaAddress = *kafkaAddress |
| conf.BBSim.OpenOltAddress = *openolt_address |
| conf.BBSim.ApiAddress = *api_address |
| conf.BBSim.RestApiAddress = *rest_api_address |
| conf.BBSim.KafkaEventTopic = *kafkaEventTopic |
| conf.BBSim.AuthRetry = *authRetry |
| conf.BBSim.DhcpRetry = *dhcpRetry |
| |
| // update device id if not set |
| if conf.Olt.DeviceId == "" { |
| conf.Olt.DeviceId = net.HardwareAddr{0xA, 0xA, 0xA, 0xA, 0xA, byte(conf.Olt.ID)}.String() |
| } |
| |
| return conf |
| } |
| |
| func getDefaultOps() *GlobalConfig { |
| |
| c := &GlobalConfig{ |
| BBSimConfig{ |
| ConfigFile: "configs/bbsim.yaml", |
| ServiceConfigFile: "configs/att-services.yaml", |
| LogLevel: "debug", |
| LogCaller: false, |
| Delay: 200, |
| OpenOltAddress: ":50060", |
| ApiAddress: ":50070", |
| RestApiAddress: ":50071", |
| LegacyApiAddress: ":50072", |
| LegacyRestApiAddress: ":50073", |
| SadisRestAddress: ":50074", |
| SadisServer: true, |
| KafkaAddress: ":9092", |
| Events: false, |
| ControlledActivation: "default", |
| EnablePerf: false, |
| KafkaEventTopic: "", |
| DhcpRetry: false, |
| AuthRetry: false, |
| }, |
| OltConfig{ |
| Vendor: "BBSim", |
| Model: "asfvolt16", |
| HardwareVersion: "emulated", |
| FirmwareVersion: "", |
| DeviceSerialNumber: "BBSM00000001", |
| PonPorts: 1, |
| NniPorts: 1, |
| OnusPonPort: 1, |
| Technology: "XGS-PON", |
| ID: 0, |
| OltRebootDelay: 10, |
| PortStatsInterval: 20, |
| }, |
| BBRConfig{ |
| LogLevel: "debug", |
| LogCaller: false, |
| }, |
| } |
| return c |
| } |
| |
| // LoadBBSimConf loads the BBSim configuration from a YAML file |
| func loadBBSimConf(filename string) (*GlobalConfig, error) { |
| yamlConfig := getDefaultOps() |
| |
| yamlFile, err := ioutil.ReadFile(filename) |
| if err != nil { |
| log.WithFields(log.Fields{ |
| "err": err, |
| "filename": filename, |
| }).Error("Cannot load BBSim configuration file. Using defaults.") |
| return yamlConfig, nil |
| } |
| |
| err = yaml.Unmarshal(yamlFile, yamlConfig) |
| if err != nil { |
| return nil, err |
| } |
| |
| return yamlConfig, nil |
| } |
| |
| // LoadBBSimServices parses a file describing the services that need to be created for each UNI |
| func loadBBSimServices(filename string) ([]ServiceYaml, error) { |
| |
| yamlServiceCfg := YamlServiceConfig{} |
| |
| yamlFile, err := ioutil.ReadFile(filename) |
| if err != nil { |
| return nil, err |
| } |
| |
| err = yaml.Unmarshal([]byte(yamlFile), &yamlServiceCfg) |
| if err != nil { |
| return nil, err |
| } |
| |
| for _, service := range yamlServiceCfg.Services { |
| |
| if service.CTagAllocation == "" || service.STagAllocation == "" { |
| log.Fatal("c_tag_allocation and s_tag_allocation are mandatory fields") |
| } |
| |
| if _, err := tagAllocationFromString(string(service.CTagAllocation)); err != nil { |
| log.WithFields(log.Fields{ |
| "err": err, |
| }).Fatal("c_tag_allocation is not valid") |
| } |
| } |
| |
| log.WithFields(log.Fields{ |
| "services": yamlServiceCfg.String(), |
| }).Debug("BBSim services description correctly loaded") |
| return yamlServiceCfg.Services, nil |
| } |
| |
| // This is only used by BBR |
| func GetBBROpts() BBRCliOptions { |
| |
| bbsimIp := flag.String("bbsimIp", "127.0.0.1", "BBSim IP") |
| bbsimPort := flag.String("bbsimPort", "50060", "BBSim Port") |
| bbsimApiPort := flag.String("bbsimApiPort", "50070", "BBSim API Port") |
| logFile := flag.String("logfile", "", "Log to a file") |
| |
| LoadConfig() |
| |
| bbrOptions := BBRCliOptions{ |
| Config, |
| *bbsimIp, |
| *bbsimPort, |
| *bbsimApiPort, |
| *logFile, |
| } |
| |
| return bbrOptions |
| } |