added support for config generation for ceos

clab/config.go

@@ -19,13 +19,16 @@ const (
 	dockerNetName     = "clab"
 	dockerNetIPv4Addr = "172.20.20.0/24"
 	dockerNetIPv6Addr = "2001:172:20:20::/80"
-
-	defaultConfigTemplate = "/etc/containerlab/templates/srl/srlconfig.tpl"
 )
 
 // supported kinds
 var kinds = []string{"srl", "ceos", "linux", "alpine", "bridge"}
 
+var defaultConfigTemplates = map[string]string{
+	"srl":  "/etc/containerlab/templates/srl/srlconfig.tpl",
+	"ceos": "/etc/containerlab/templates/arista/ceos.cfg.tpl",
+}
+
 var srlTypes = map[string]string{
 	"ixr6":  "topology-7250IXR6.yml",
 	"ixr10": "topology-7250IXR10.yml",
@@ -229,7 +232,7 @@ func (c *cLab) configInitialization(nodeCfg *NodeConfig, kind string) string {
 	if c.Config.Topology.Defaults.Config != "" {
 		return c.Config.Topology.Defaults.Config
 	}
-	return defaultConfigTemplate
+	return defaultConfigTemplates[kind]
 }
 
 func (c *cLab) imageInitialization(nodeCfg *NodeConfig, kind string) string {
@@ -312,9 +315,7 @@ func (c *cLab) NewNode(nodeName string, nodeCfg NodeConfig, idx int) error {
 	case "ceos":
 		// initialize the global parameters with defaults, can be overwritten later
 		node.Config = c.configInitialization(&nodeCfg, node.Kind)
-		//node.License = t.SRLLicense
 		node.Image = c.imageInitialization(&nodeCfg, node.Kind)
-		//node.NodeType = "ixr6"
 		node.Position = c.positionInitialization(&nodeCfg, node.Kind)
 
 		// initialize specifc container information
@@ -330,7 +331,6 @@ func (c *cLab) NewNode(nodeName string, nodeCfg NodeConfig, idx int) error {
 		node.User = "root"
 		node.Group = c.groupInitialization(&nodeCfg, node.Kind)
 		node.NodeType = nodeCfg.Type
-		node.Config = nodeCfg.Config
 
 		node.Sysctls = make(map[string]string)
 		node.Sysctls["net.ipv4.ip_forward"] = "0"
@@ -340,6 +340,10 @@ func (c *cLab) NewNode(nodeName string, nodeCfg NodeConfig, idx int) error {
 		node.Sysctls["net.ipv6.conf.all.autoconf"] = "0"
 		node.Sysctls["net.ipv6.conf.default.autoconf"] = "0"
 
+		// mount config file
+		cfgPath := filepath.Join(node.LabDir, "config", "startup-config")
+		node.Binds = append(node.Binds, fmt.Sprint(cfgPath, ":/mnt/flash/startup-config"))
+
 	case "srl":
 		// initialize the global parameters with defaults, can be overwritten later
 		node.Config = c.configInitialization(&nodeCfg, node.Kind)

clab/file.go

@@ -132,19 +132,22 @@ func CreateDirectory(path string, perm os.FileMode) {
 	}
 }
 
-// CreateNodeDirStructure create the directory structure and files for the clab
+// CreateNodeDirStructure create the directory structure and files for the lab nodes
 func (c *cLab) CreateNodeDirStructure(node *Node) (err error) {
 	c.m.RLock()
 	defer c.m.RUnlock()
+
+	// create node directory in the lab directory
+	if node.Kind != "linux" && node.Kind != "bridge" {
+		CreateDirectory(node.LabDir, 0777)
+	}
+
 	switch node.Kind {
 	case "srl":
 		log.Infof("Create directory structure for SRL container: %s", node.ShortName)
 		var src string
 		var dst string
 
-		// create node directory in lab
-		CreateDirectory(node.LabDir, 0777)
-
 		// copy license file to node specific directory in lab
 		src = node.License
 		dst = path.Join(node.LabDir, "license.key")
@@ -180,9 +183,19 @@ func (c *cLab) CreateNodeDirStructure(node *Node) (err error) {
 		}
 		log.Debugf("CopyFile src %s -> dst %s succeeded\n", src, dst)
 
-	case "alpine":
 	case "linux":
 	case "ceos":
+		// generate config directory
+		CreateDirectory(path.Join(node.LabDir, "config"), 0777)
+		cfg := path.Join(node.LabDir, "config", "startup-config")
+		if !fileExists(cfg) {
+			err = node.generateConfig(cfg)
+			if err != nil {
+				log.Errorf("node=%s, failed to generate config: %v", node.ShortName, err)
+			}
+		} else {
+			log.Debugf("Config file exists for node %s", node.ShortName)
+		}
 	case "bridge":
 	default:
 	}
@@ -192,6 +205,7 @@ func (c *cLab) CreateNodeDirStructure(node *Node) (err error) {
 
 // GenerateConfig generates configuration for the nodes
 func (node *Node) generateConfig(dst string) error {
+	log.Debugf("generating config for node %s from file %s", node.ShortName, node.Config)
 	tpl, err := template.New(filepath.Base(node.Config)).ParseFiles(node.Config)
 	if err != nil {
 		return err

templates/arista/ceos.cfg.tpl

@@ -0,0 +1 @@
+hostname {{ .ShortName }}