Compare commits
	
		
			126 Commits
		
	
	
		
			v0.22a
			...
			812c77790a
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 812c77790a | |||
| 7575da794e | |||
| 6f9bb504be | |||
| 7c3e14f0f1 | |||
| a4ff9c1d51 | |||
| 00fd283987 | |||
| 7cf4594f34 | |||
| 37fea55e42 | |||
| c15ee68d2e | |||
| 54fd1f8064 | |||
| 89db166040 | |||
| 9c18a83ee8 | |||
| 561ae4386a | |||
| 667c73216e | |||
| 9e506145a8 | |||
| d636d963ff | |||
| 56b4d8ea84 | |||
| abaa4a11f9 | |||
| 74602dc0df | |||
| be756edea7 | |||
| 546382ded7 | |||
| 07eccffbd1 | |||
| 7809107ea4 | |||
| 5ab0a59db4 | |||
| 1b27753718 | |||
| c97f5317dd | |||
| 38266af66e | |||
| fabe9cc330 | |||
| 0637686f91 | |||
| 99f7bed6f3 | |||
| 9b86d786fe | |||
| efd28c03ad | |||
| e1786a4d08 | |||
| bf1037c1d3 | |||
| cbdcb039cb | |||
| 09ac7464f5 | |||
| de1a92dec8 | |||
| d16a2d6789 | |||
| 35b75ee7fb | |||
| 4d92e880c9 | |||
| 52916f9ae9 | |||
| 31961b8e3a | |||
| 128e2aa0c4 | |||
| 0c7293ae66 | |||
| e1f6b4f6f9 | |||
| f29aeb2e23 | |||
| 1cb7c5fec7 | |||
| 3dae685fc4 | |||
| 9b90f9c812 | |||
| e13437b79e | |||
| 484e05e8d1 | |||
| e7a6bdd376 | |||
| 170dce31b3 | |||
| 1295fb86f6 | |||
| d3410c281a | |||
| 51dc7d1588 | |||
| 92d8beb58f | |||
| 745811c39b | |||
| 276d01ed4c | |||
| 7266496cac | |||
| 5b0de24508 | |||
| 2a8836721c | |||
| 0f6b7b8b80 | |||
| 1462c383d3 | |||
| 29ce0d9b58 | |||
| 42e1085ad4 | |||
| 0f4f76a9a2 | |||
| e87699e2dc | |||
| 4d8bf6e0d5 | |||
| d9e1e20afc | |||
| fcf7d68d06 | |||
| f919ff2ec3 | |||
| 1bc248fdcc | |||
| a7aaa11de6 | |||
| 71f345dff4 | |||
| e1410bf209 | |||
| c585678be9 | |||
| 9218ffafe1 | |||
| 1c04f62ed8 | |||
| 2151034a02 | |||
| e84c43c759 | |||
| d4f6b9ddc7 | |||
| 86e08ec0f7 | |||
| b4fd7caca7 | |||
| 29e8736fbc | |||
| 203c4bff3b | |||
| 7356c0d3d0 | |||
| a446a19a08 | |||
| 0bf825ee5a | |||
| 542d2f96f6 | |||
| 6c6cb7edc8 | |||
| 31fa6904db | |||
| fb3ee07585 | |||
| 57c8bba09b | |||
| 249ab19173 | |||
| bc92f29900 | |||
| e4fc9c3a6c | |||
| 43f26d099f | |||
| dbd9153513 | |||
| 7b5ae7ce6e | |||
| 77a2e9dabf | |||
| 6821b14407 | |||
| f9ce3601df | |||
| 4f85f2e6ac | |||
| e0f371693a | |||
| 0f97270a6a | |||
| 7dbbf8a757 | |||
| 0053fd6c8b | |||
| 4aa1c81fea | |||
| 966a3d57c1 | |||
| a12c268be2 | |||
| 7b34495cf6 | |||
| 1c40e29eff | |||
| ef78245902 | |||
| 910be4ea31 | |||
| 285229009f | |||
| 19dd2dfb33 | |||
| 139ea18422 | |||
| 3bedf019dc | |||
| da74456d6a | |||
| f40db69b9d | |||
| 239bcd4b95 | |||
| 5a3d26a52c | |||
| bf20c815ce | |||
| f1c4fd960d | |||
| 8d18bfe55d | 
							
								
								
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							@ -1,3 +1,2 @@
 | 
				
			|||||||
gocage
 | 
					gocage
 | 
				
			||||||
go.sum
 | 
					go.sum
 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										3
									
								
								CHANGELOG
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										3
									
								
								CHANGELOG
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,3 @@
 | 
				
			|||||||
 | 
					v.0.33b : Support jaling datasets on differents pools : jail_zfs_dataset now have to include the pool name  
 | 
				
			||||||
 | 
					v.0.33c : Parallelize start/stop of jails with same priority  
 | 
				
			||||||
 | 
					
 | 
				
			||||||
							
								
								
									
										96
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										96
									
								
								README.md
									
									
									
									
									
								
							@ -4,8 +4,15 @@ GoCage
 | 
				
			|||||||
Jail management tool for FreeBSD, written in Go.  
 | 
					Jail management tool for FreeBSD, written in Go.  
 | 
				
			||||||
Support iocage jails, so they can coexist.  
 | 
					Support iocage jails, so they can coexist.  
 | 
				
			||||||
Gocage is meant to be a complete jail management tool with network, snapshots, jail cloning support and a web interface. This is the hypothetic future.  
 | 
					Gocage is meant to be a complete jail management tool with network, snapshots, jail cloning support and a web interface. This is the hypothetic future.  
 | 
				
			||||||
 | 
					Gocage can handle multiple datastores, so you can have jails on HDD storage and jails on SSD storage.  
 | 
				
			||||||
 | 
					
 | 
				
			||||||
At present time, it list and stops jails.  
 | 
					From v0.33b, due to multi ZFS pool support, gocage is no longer 100% compatible with iocage.  
 | 
				
			||||||
 | 
					Zfs datasets now should be specified with the ZFS pool. e.g. :  
 | 
				
			||||||
 | 
					<code>
 | 
				
			||||||
 | 
					Config.Jail_zfs = 1
 | 
				
			||||||
 | 
					Config.Jail_zfs_dataset = myzfspool/poudriere
 | 
				
			||||||
 | 
					Config.Jail_zfs_mountpoint = none
 | 
				
			||||||
 | 
					</code>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
List jails
 | 
					List jails
 | 
				
			||||||
----------
 | 
					----------
 | 
				
			||||||
@ -118,3 +125,90 @@ Stop jails
 | 
				
			|||||||
----------
 | 
					----------
 | 
				
			||||||
`gocage stop test`
 | 
					`gocage stop test`
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Multi datastore  
 | 
				
			||||||
 | 
					----------
 | 
				
			||||||
 | 
					A datastore is a ZFS dataset mounted. It should be declared in gocage.conf.yml, specifying its ZFS mountpoint :
 | 
				
			||||||
 | 
					<pre><code>
 | 
				
			||||||
 | 
					datastore:
 | 
				
			||||||
 | 
					  - /iocage
 | 
				
			||||||
 | 
					  - /fastiocage
 | 
				
			||||||
 | 
					</pre></code>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					In gocage commands, datastore name is the mountpoint without its "/" prefix.  
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					### List datastores
 | 
				
			||||||
 | 
					<pre><code>
 | 
				
			||||||
 | 
					gocage datastore list 
 | 
				
			||||||
 | 
					+============+=============+============+===========+==========+============+
 | 
				
			||||||
 | 
					| Name       | Mountpoint  | ZFSDataset | Available | Used     | Referenced |
 | 
				
			||||||
 | 
					+============+=============+============+===========+==========+============+
 | 
				
			||||||
 | 
					| iocage     | /iocage     | hdd/iocage | 1.6 TB    | 414.9 GB | 27.5 KB    |
 | 
				
			||||||
 | 
					+------------+-------------+------------+-----------+----------+------------+
 | 
				
			||||||
 | 
					| fastiocage | /fastiocage | ssd/iocage | 1.5 TB    | 65.3 KB  | 34.6 KB    |
 | 
				
			||||||
 | 
					+------------+-------------+------------+-----------+----------+------------+
 | 
				
			||||||
 | 
					</pre></code>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					### Filter datastores  
 | 
				
			||||||
 | 
					As with jails and snapshots, you can filter by name:  
 | 
				
			||||||
 | 
					<pre><code>
 | 
				
			||||||
 | 
					gocage datastore list iocage
 | 
				
			||||||
 | 
					+============+=============+============+===========+==========+============+
 | 
				
			||||||
 | 
					| Name       | Mountpoint  | ZFSDataset | Available | Used     | Referenced |
 | 
				
			||||||
 | 
					+============+=============+============+===========+==========+============+
 | 
				
			||||||
 | 
					| iocage     | /iocage     | hdd/iocage | 1.6 TB    | 414.9 GB | 27.5 KB    |
 | 
				
			||||||
 | 
					+------------+-------------+------------+-----------+----------+------------+
 | 
				
			||||||
 | 
					</pre></code>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					### Sort datastores  
 | 
				
			||||||
 | 
					You can sort datastores:  
 | 
				
			||||||
 | 
					<pre><code>
 | 
				
			||||||
 | 
					gocage datastore list -s -Available
 | 
				
			||||||
 | 
					+============+=============+============+===========+==========+============+
 | 
				
			||||||
 | 
					| Name       | Mountpoint  | ZFSDataset | Available | Used     | Referenced |
 | 
				
			||||||
 | 
					+============+=============+============+===========+==========+============+
 | 
				
			||||||
 | 
					| iocage     | /iocage     | hdd/iocage | 1.6 TB    | 415.0 GB | 27.5 KB    |
 | 
				
			||||||
 | 
					+------------+-------------+------------+-----------+----------+------------+
 | 
				
			||||||
 | 
					| fastiocage | /fastiocage | ssd/iocage | 1.5 TB    | 65.3 KB  | 34.6 KB    |
 | 
				
			||||||
 | 
					+------------+-------------+------------+-----------+----------+------------+
 | 
				
			||||||
 | 
					</pre></code>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					See [cmd/struct.go](https://git.nosd.in/yo/gocage/src/branch/master/cmd/struct.go) for field names.
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Migrating jails  
 | 
				
			||||||
 | 
					----------
 | 
				
			||||||
 | 
					With multi datastore comes the need to migrate a jail between datastores.  
 | 
				
			||||||
 | 
					Migration can be done with a minimal downtime, using zfs differential send/receive.  
 | 
				
			||||||
 | 
					Source jail datasets are sent to the destination datastore, jail is stopped and a last differential sync is done before starting jail on new datastore.  
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					### Warning 
 | 
				
			||||||
 | 
					Be aware the moment you migrate a jail to another datastore than /iocage default, you lose compatibility with iocage.  
 | 
				
			||||||
 | 
					Then you need to disable iocage service, and enable gocage so the jails will start automatically at boot.  
 | 
				
			||||||
 | 
					Also make sure, if you don't destroy source jail, that it won't have the "boot" property set or you will have the 2 jails up at boot.  
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					<pre><code>
 | 
				
			||||||
 | 
					gocage migrate -d fastiocage srv-random
 | 
				
			||||||
 | 
					Snapshot data/iocage/jails/srv-random: Done
 | 
				
			||||||
 | 
					Snapshot data/iocage/jails/srv-random/root: Done
 | 
				
			||||||
 | 
					Migrate jail config dataset to fastdata/iocage/jails/srv-random: Done
 | 
				
			||||||
 | 
					Migrate jail filesystem dataset to fastdata/iocage/jails/srv-random/root: Done
 | 
				
			||||||
 | 
					</pre></code>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					Fetch  
 | 
				
			||||||
 | 
					----------
 | 
				
			||||||
 | 
					Files can be fetched from custom repository, or from local directory with "from" option.  
 | 
				
			||||||
 | 
					For example if you destroyed releases/12.3-RELEASE and still have the downloaded files in /iocage/download/12.3-RELEASE:  
 | 
				
			||||||
 | 
					<pre><code>
 | 
				
			||||||
 | 
					gocage fetch -r 12.3 -o iocage --from file:/iocage/download
 | 
				
			||||||
 | 
					</pre></code>
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					TODO
 | 
				
			||||||
 | 
					----------
 | 
				
			||||||
 | 
					gocage update  
 | 
				
			||||||
 | 
					gocage upgrade  
 | 
				
			||||||
 | 
					gocage create  
 | 
				
			||||||
 | 
					gocage destroy  
 | 
				
			||||||
 | 
					gocage init  
 | 
				
			||||||
 | 
					  create default pool with defaults.json
 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										6
									
								
								TODO.md
									
									
									
									
									
								
							
							
						
						
									
										6
									
								
								TODO.md
									
									
									
									
									
								
							@ -1,4 +1,8 @@
 | 
				
			|||||||
Replicating jails between two servers (use zrepl)
 | 
					Replicating jails between two servers (use zrepl)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					DEBUG: 
 | 
				
			||||||
 | 
					- cmd/list.go:275:
 | 
				
			||||||
 | 
					    // FIXME ??? Shouldn't be ioc-$Name ?
 | 
				
			||||||
 | 
					    j.InternalName = rj.Name
 | 
				
			||||||
 | 
					- WriteConfigToDisk don't write neither "release" in cmd stop neither "last_started" in cmd start
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										46
									
								
								cmd/console.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										46
									
								
								cmd/console.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,46 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"os"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"log"
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
 | 
						"syscall"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func ShellJail(args []string) error {
 | 
				
			||||||
 | 
						// We cant shell more than one jail bc we replace gocage execution with jexec, so there wont be no return to gocage
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							cj, err := getJailFromArray(args[0], gJails)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error getting jail %s: %v\n", args[0], err)
 | 
				
			||||||
 | 
								return err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							if err := shellJail(cj); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("%v\n", err)
 | 
				
			||||||
 | 
								return err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func shellJail(jail *Jail) error {
 | 
				
			||||||
 | 
						if false == jail.Running {
 | 
				
			||||||
 | 
							return errors.New("Jail is not running")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						jid := strconv.Itoa(jail.JID)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						//err := syscall.Exec("/usr/sbin/jexec", []string{"jexec", jid, "/bin/csh"}, os.Environ())
 | 
				
			||||||
 | 
						err := syscall.Exec("/usr/sbin/jexec", []string{"jexec", jid, "login", "-f", "root"}, os.Environ())
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// We should never get here, as syscall.Exec replace the gocage binary execution with jexec
 | 
				
			||||||
 | 
						// This means the moment syscall.Exec fires, gocage execution halt.
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							log.Printf("Exec returned %v\n", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										108
									
								
								cmd/datastore.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										108
									
								
								cmd/datastore.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,108 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"reflect"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					) 
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func ListDatastores(args []string, display bool) error {
 | 
				
			||||||
 | 
						/***************************************************************
 | 
				
			||||||
 | 
						/   Filter datastores by names given on command line
 | 
				
			||||||
 | 
						/**************************************************************/
 | 
				
			||||||
 | 
						for _, d := range args {
 | 
				
			||||||
 | 
							cmd := fmt.Sprintf("zfs list -p -H -o name,used,available,referenced %s", d)
 | 
				
			||||||
 | 
							out, err := executeCommand(cmd)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								if strings.HasSuffix(err.Error(), "No such file or directory") {
 | 
				
			||||||
 | 
									return errors.New(fmt.Sprintf("Datastore does not exist: %s", d))
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									return errors.New(fmt.Sprintf("Error executing command %s: %v; command returned: %s\n", cmd, err, out))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							fields := strings.Fields(out)
 | 
				
			||||||
 | 
							if len(fields) < 4 {
 | 
				
			||||||
 | 
								return errors.New(fmt.Sprintf("Error parsing output of \"%s\": Not enough fields", cmd))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							u, _ := strconv.ParseUint(fields[1], 10, 64)
 | 
				
			||||||
 | 
							a, _ := strconv.ParseUint(fields[2], 10, 64)
 | 
				
			||||||
 | 
							r, _ := strconv.ParseUint(fields[3], 10, 64)
 | 
				
			||||||
 | 
							ds := Datastore{Name: d[1:], Mountpoint: d, ZFSDataset: fields[0], Used: u, Referenced: r, Available: a}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							err = loadDefaultsForDatastore(&ds)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							gDatastores = append(gDatastores, ds)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						fields := strings.Split(gDisplayDColumns, ",")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if true == display {
 | 
				
			||||||
 | 
							/***************************************************************
 | 
				
			||||||
 | 
							/   Sort datastores
 | 
				
			||||||
 | 
							/     We support 3 sort criteria max
 | 
				
			||||||
 | 
							/**************************************************************/
 | 
				
			||||||
 | 
							if len(gSortDSFields) > 0 && gSortDSFields != "none" {		
 | 
				
			||||||
 | 
								ds := initDatastoreSortStruct()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								// The way we manage criteria quantity is not very elegant...
 | 
				
			||||||
 | 
								var fct1, fct2, fct3 *reflect.Value
 | 
				
			||||||
 | 
								for i, c := range strings.Split(gSortDSFields, ",") {
 | 
				
			||||||
 | 
									var fctName string
 | 
				
			||||||
 | 
									if strings.HasPrefix(c, "-") {
 | 
				
			||||||
 | 
										fctName = fmt.Sprintf("%sDec", strings.Replace(c, "-", "", 1))
 | 
				
			||||||
 | 
									} else { // Par defaut (pas de prefix +/-) on considere un tri incremental
 | 
				
			||||||
 | 
										fctName = fmt.Sprintf("%sInc", strings.Replace(c, "+", "", 1))
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									// Get function by its name
 | 
				
			||||||
 | 
									fct, _, err := getStructFieldValue(ds, fctName)
 | 
				
			||||||
 | 
									
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										fieldName := strings.Replace(strings.Replace(c, "-", "", 1), "+", "", 1)
 | 
				
			||||||
 | 
										return errors.New(fmt.Sprintf("ERROR getting DatastoreSort struct field %s. Please check the field name: %s\n", fctName, fieldName))
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									switch i + 1 {
 | 
				
			||||||
 | 
									case 1:
 | 
				
			||||||
 | 
										fct1 = fct
 | 
				
			||||||
 | 
									case 2:
 | 
				
			||||||
 | 
										fct2 = fct
 | 
				
			||||||
 | 
									case 3:
 | 
				
			||||||
 | 
										fct3 = fct
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								switch len(strings.Split(gSortDSFields, ",")) {
 | 
				
			||||||
 | 
								case 1:
 | 
				
			||||||
 | 
									DatastoresOrderedBy(fct1.Interface().(datastoreLessFunc)).Sort(gDatastores)
 | 
				
			||||||
 | 
								case 2:
 | 
				
			||||||
 | 
									DatastoresOrderedBy(fct1.Interface().(datastoreLessFunc), fct2.Interface().(datastoreLessFunc)).Sort(gDatastores)
 | 
				
			||||||
 | 
								case 3:
 | 
				
			||||||
 | 
									DatastoresOrderedBy(fct1.Interface().(datastoreLessFunc), fct2.Interface().(datastoreLessFunc), fct3.Interface().(datastoreLessFunc)).Sort(gDatastores)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							displayDatastoresFields(gDatastores, fields)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * Load jails default config from $datastore/defaults.json
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func loadDefaultsForDatastore(ds *Datastore) error {
 | 
				
			||||||
 | 
						jc, err := getJailConfig(fmt.Sprintf("%s/defaults.json", ds.Mountpoint))
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						ds.DefaultJailConfig = jc
 | 
				
			||||||
 | 
						//gDefaultConfig = append(gDefaultConfig, jc)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										374
									
								
								cmd/fetch.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										374
									
								
								cmd/fetch.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,374 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"io"
 | 
				
			||||||
 | 
						"os"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"bufio"
 | 
				
			||||||
 | 
						"bytes"
 | 
				
			||||||
 | 
						//"errors"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
						"net/http"
 | 
				
			||||||
 | 
						//"archive/tar"
 | 
				
			||||||
 | 
						"encoding/hex"
 | 
				
			||||||
 | 
						"crypto/sha256"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						//"github.com/ulikunitz/xz"
 | 
				
			||||||
 | 
						log "github.com/sirupsen/logrus"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const (
 | 
				
			||||||
 | 
						ReleaseServer = "download.freebsd.org"
 | 
				
			||||||
 | 
						ReleaseRootDir = "ftp/releases"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var (
 | 
				
			||||||
 | 
						FetchFiles = []string{"base.txz", "lib32.txz", "src.txz"}
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// TODO: Check if files already exist
 | 
				
			||||||
 | 
					// Fetch release files, verify, put in datastore under ${datastore}/download
 | 
				
			||||||
 | 
					// Only support http and file protocols
 | 
				
			||||||
 | 
					func fetchRelease(release string, proto string, arch string, datastore string, fetchFrom string) error {
 | 
				
			||||||
 | 
						var ds Datastore
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						log.SetReportCaller(true)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if len(fetchFrom) > 0 {
 | 
				
			||||||
 | 
							proto = strings.Split(fetchFrom, ":")[0]
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if false == strings.EqualFold(proto, "http") &&
 | 
				
			||||||
 | 
							false == strings.EqualFold(proto, "file") {
 | 
				
			||||||
 | 
							return fmt.Errorf("Unsupported protocol: %s\n", proto)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						for _, ds = range gDatastores {
 | 
				
			||||||
 | 
							if strings.EqualFold(datastore, ds.Name) {
 | 
				
			||||||
 | 
								break
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if false == strings.EqualFold(datastore, ds.Name) {
 | 
				
			||||||
 | 
							return fmt.Errorf("Datastore not found: %s\n", datastore)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Check datastore have a download dataset, and it is mounted
 | 
				
			||||||
 | 
						downloadDsName := fmt.Sprintf("%s/download", ds.ZFSDataset)
 | 
				
			||||||
 | 
						downloadDsMountPoint := fmt.Sprintf("%s/download", ds.Mountpoint)
 | 
				
			||||||
 | 
						exist, err := doZfsDatasetExist(downloadDsName)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return fmt.Errorf("Error accessing dataset %s: %v\n", downloadDsName, err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if false == exist {
 | 
				
			||||||
 | 
							// Then create dataset
 | 
				
			||||||
 | 
							if err := createZfsDataset(downloadDsName, downloadDsMountPoint, "lz4"); err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("Error creating dataset %s: %v\n", downloadDsName, err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Create download/XX.X-RELEASE dataset if necessary
 | 
				
			||||||
 | 
						thisDownloadDsName := fmt.Sprintf("%s/%s-RELEASE", downloadDsName, release)
 | 
				
			||||||
 | 
						thisDownloadDsMountPoint := fmt.Sprintf("%s/%s-RELEASE", downloadDsMountPoint, release)
 | 
				
			||||||
 | 
						exist, err = doZfsDatasetExist(thisDownloadDsName)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return fmt.Errorf("Error accessing dataset %s: %v\n", thisDownloadDsName, err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if false == exist {
 | 
				
			||||||
 | 
							// Then create dataset
 | 
				
			||||||
 | 
							if err := createZfsDataset(thisDownloadDsName, thisDownloadDsMountPoint, "lz4"); err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("Error creating dataset %s: %v\n", thisDownloadDsName, err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var fetchUrl string
 | 
				
			||||||
 | 
						if len(fetchFrom) > 0 {
 | 
				
			||||||
 | 
							fetchUrl = fmt.Sprintf("%s/%s-RELEASE", fetchFrom, release)
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							fetchUrl = fmt.Sprintf("%s://%s/%s/%s/%s-RELEASE", proto, ReleaseServer, ReleaseRootDir, arch, release)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						log.Debugf("FetchURL = %s", fetchUrl)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// check if proto/server/arch/release is available
 | 
				
			||||||
 | 
						if strings.EqualFold(proto, "http") {
 | 
				
			||||||
 | 
							resp, err := http.Get(fetchUrl)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("Can not get %s: %v\n", fetchUrl, err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							defer resp.Body.Close()
 | 
				
			||||||
 | 
							if resp.StatusCode != 200 {
 | 
				
			||||||
 | 
								return fmt.Errorf("Get %s returned %d, check release name\n", fetchUrl, resp.StatusCode)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Fetch files
 | 
				
			||||||
 | 
						// Get MANIFEST so we get sha256 sums
 | 
				
			||||||
 | 
						if err := fetchFile(proto, fetchUrl, "MANIFEST", thisDownloadDsMountPoint, []byte{}); err != nil {
 | 
				
			||||||
 | 
							return fmt.Errorf("%v\n", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						// Build an array of "file;checksum"
 | 
				
			||||||
 | 
						checksumMap, err := buildFileChecksumFromManifest(fmt.Sprintf("%s/MANIFEST", thisDownloadDsMountPoint), FetchFiles)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return fmt.Errorf("%v\n", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Fetch remaining files, verify integrity and write to disk
 | 
				
			||||||
 | 
						for f, c := range checksumMap {
 | 
				
			||||||
 | 
							if err := fetchFile(proto, fetchUrl, f, thisDownloadDsMountPoint, c); err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("%v\n", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Extract release files stored in iocage/download/$RELEASE/ to iocage/releases/$RELEASE/root/
 | 
				
			||||||
 | 
					func extractRelease(release string, datastore string) {
 | 
				
			||||||
 | 
						log.SetReportCaller(true)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						var ds Datastore
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						for _, ds = range gDatastores {
 | 
				
			||||||
 | 
							if strings.EqualFold(datastore, ds.Name) {
 | 
				
			||||||
 | 
								break
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if false == strings.EqualFold(datastore, ds.Name) {
 | 
				
			||||||
 | 
							fmt.Printf("Datastore not found: %s\n", datastore)
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Check datastore have a releases dataset, and it is mounted
 | 
				
			||||||
 | 
						releaseDsName := fmt.Sprintf("%s/releases", ds.ZFSDataset)
 | 
				
			||||||
 | 
						releaseDsMountPoint := fmt.Sprintf("%s/releases", ds.Mountpoint)
 | 
				
			||||||
 | 
						exist, err := doZfsDatasetExist(releaseDsName)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error accessing dataset %s: %v\n", releaseDsName, err)
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if false == exist {
 | 
				
			||||||
 | 
							// Then create dataset
 | 
				
			||||||
 | 
							if err := createZfsDataset(releaseDsName, releaseDsMountPoint, "lz4"); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error creating dataset %s: %v\n", releaseDsName, err)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Create releases/XX.X-RELEASE dataset if necessary
 | 
				
			||||||
 | 
						thisReleaseDsName := fmt.Sprintf("%s/%s-RELEASE", releaseDsName, release)
 | 
				
			||||||
 | 
						thisReleaseDsMountPoint := fmt.Sprintf("%s/%s-RELEASE", releaseDsMountPoint, release)
 | 
				
			||||||
 | 
						exist, err = doZfsDatasetExist(thisReleaseDsName)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error accessing dataset %s: %v\n", thisReleaseDsName, err)
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if false == exist {
 | 
				
			||||||
 | 
							// Then create dataset
 | 
				
			||||||
 | 
							if err := createZfsDataset(thisReleaseDsName, thisReleaseDsMountPoint, "lz4"); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error creating dataset %s: %v\n", thisReleaseDsName, err)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Create releases/XX.X-RELEASE/root dataset if necessary
 | 
				
			||||||
 | 
						thisReleaseRootDsName := fmt.Sprintf("%s/root", thisReleaseDsName)
 | 
				
			||||||
 | 
						thisReleaseRootDsMountPoint := fmt.Sprintf("%s/root", thisReleaseDsMountPoint)
 | 
				
			||||||
 | 
						exist, err = doZfsDatasetExist(thisReleaseRootDsName)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error accessing dataset %s: %v\n", thisReleaseRootDsName, err)
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if false == exist {
 | 
				
			||||||
 | 
							// Then create dataset
 | 
				
			||||||
 | 
							if err := createZfsDataset(thisReleaseRootDsName, thisReleaseRootDsMountPoint, "lz4"); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error creating dataset %s: %v\n", thisReleaseRootDsName, err)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Now extract download/$RELEASE/*.txz to releases/XX.X-RELEASE/root
 | 
				
			||||||
 | 
						downloadDsMountPoint := fmt.Sprintf("%s/download", ds.Mountpoint)
 | 
				
			||||||
 | 
						downloadDir := fmt.Sprintf("%s/%s-RELEASE", downloadDsMountPoint, release)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						d, err := os.Open(downloadDir)
 | 
				
			||||||
 | 
						defer d.Close()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Can not read %s directory: %v\n", downloadDir, err)
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						files, err := d.Readdir(0)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Can not browse %s directory: %v\n", downloadDir, err)
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Extract every .txz files
 | 
				
			||||||
 | 
						for _, fi := range files {
 | 
				
			||||||
 | 
							if false == fi.IsDir() {
 | 
				
			||||||
 | 
								if strings.HasSuffix(fi.Name(), ".txz") {
 | 
				
			||||||
 | 
									ar := fmt.Sprintf("%s/%s", downloadDir, fi.Name())
 | 
				
			||||||
 | 
									fmt.Printf("Extracting file %s... ", ar)
 | 
				
			||||||
 | 
									// pure Go method, sorry this is so slow. Also I did not handle permissions in this
 | 
				
			||||||
 | 
					/*				f, err := os.Open(ar)
 | 
				
			||||||
 | 
									defer f.Close()
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("Can not open %s: %v\n", ar, err)
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									// xz reader
 | 
				
			||||||
 | 
									r, err := xz.NewReader(f)
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("Can not read %s: %v\n", ar, err)
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									// tar reader
 | 
				
			||||||
 | 
									tr := tar.NewReader(r)
 | 
				
			||||||
 | 
									// Iterate through the files in the archive.
 | 
				
			||||||
 | 
									for {
 | 
				
			||||||
 | 
										hdr, err := tr.Next()
 | 
				
			||||||
 | 
										if err == io.EOF {
 | 
				
			||||||
 | 
											// end of tar archive
 | 
				
			||||||
 | 
											break
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											log.Fatal(err)
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										switch hdr.Typeflag {
 | 
				
			||||||
 | 
											case tar.TypeDir:
 | 
				
			||||||
 | 
												// create a directory
 | 
				
			||||||
 | 
												dest := fmt.Sprintf("%s/%s", thisReleaseRootDsMountPoint, hdr.Name)
 | 
				
			||||||
 | 
												// FIXME: Access rights?
 | 
				
			||||||
 | 
												err = os.MkdirAll(dest, 0777)
 | 
				
			||||||
 | 
												if err != nil {
 | 
				
			||||||
 | 
													log.Fatal(err)
 | 
				
			||||||
 | 
												}
 | 
				
			||||||
 | 
											case tar.TypeReg, tar.TypeRegA:
 | 
				
			||||||
 | 
												// write a file
 | 
				
			||||||
 | 
												dest := fmt.Sprintf("%s/%s", thisReleaseRootDsMountPoint, hdr.Name)
 | 
				
			||||||
 | 
												w, err := os.Create(dest)
 | 
				
			||||||
 | 
												defer w.Close()
 | 
				
			||||||
 | 
												if err != nil {
 | 
				
			||||||
 | 
													log.Fatal(err)
 | 
				
			||||||
 | 
												}
 | 
				
			||||||
 | 
												_, err = io.Copy(w, tr)
 | 
				
			||||||
 | 
												if err != nil {
 | 
				
			||||||
 | 
													log.Fatal(err)
 | 
				
			||||||
 | 
												}
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
					*/
 | 
				
			||||||
 | 
									cmd := fmt.Sprintf("/usr/bin/tar xpf %s -C %s", ar, thisReleaseRootDsMountPoint)
 | 
				
			||||||
 | 
									out, err := executeCommand(cmd)
 | 
				
			||||||
 | 
									if err != nil && len(out) > 0 {
 | 
				
			||||||
 | 
										fmt.Printf("Error: %v: %s\n", err, out)
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										fmt.Printf("Done\n")
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func fetchFile(proto, baseUrl, fileName, storeDir string, checksum []byte) error {
 | 
				
			||||||
 | 
						// Check storeDir exist
 | 
				
			||||||
 | 
						_, err := os.Stat(storeDir)
 | 
				
			||||||
 | 
						if os.IsNotExist(err) {
 | 
				
			||||||
 | 
							return fmt.Errorf("Directory does not exist: %s\n", storeDir)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						url := fmt.Sprintf("%s/%s", baseUrl, fileName)
 | 
				
			||||||
 | 
						fmt.Printf("Fetching %s...", url)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						var body []byte
 | 
				
			||||||
 | 
						if strings.EqualFold(proto, "http") {
 | 
				
			||||||
 | 
							resp, err := http.Get(url)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf(" Error\n")
 | 
				
			||||||
 | 
								return fmt.Errorf("Can not get %s: %v\n", url, err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							defer resp.Body.Close()
 | 
				
			||||||
 | 
							body, err = io.ReadAll(resp.Body)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("Can not read %s response body: %v\n", url, err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						} else if strings.EqualFold(proto, "file") {
 | 
				
			||||||
 | 
							url = strings.Replace(url, "file:", "", 1)
 | 
				
			||||||
 | 
							f, err := os.Open(url)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("Error accessing file %s", url)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							defer f.Close()
 | 
				
			||||||
 | 
							body, err = io.ReadAll(f)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("Can not read file %s: %v\n", url, err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Check integrity
 | 
				
			||||||
 | 
						if len(checksum) > 0 {
 | 
				
			||||||
 | 
							err = checkIntegrity(body, checksum)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return fmt.Errorf("Error checking integrity")
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						dest := fmt.Sprintf("%s/%s", storeDir, fileName)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						f, err := os.Create(dest)       // creates a file at current directory
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return fmt.Errorf("Can not create file %s: %v\n", dest, err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						defer f.Close()
 | 
				
			||||||
 | 
						f.Write(body)
 | 
				
			||||||
 | 
						fmt.Printf(" Done\n")
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func checkIntegrity(data []byte, checksum []byte) error {
 | 
				
			||||||
 | 
						sum := sha256.Sum256(data)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if false == bytes.Equal(checksum[:],sum[:]) {
 | 
				
			||||||
 | 
							return fmt.Errorf("Invalid checksum: %x != %x", sum, checksum)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Get checksum from manifest, for each file in fileList
 | 
				
			||||||
 | 
					/* MANIFEST format:
 | 
				
			||||||
 | 
					 * base-dbg.txz    a5b51f3d54686509e91ca9c30e9f1cd93dc757f25c643609b3c35e7119c0531d        1654    base_dbg        "Base system (Debugging)"       off
 | 
				
			||||||
 | 
					 * base.txz        e85b256930a2fbc04b80334106afecba0f11e52e32ffa197a88d7319cf059840        26492   base    "Base system (MANDATORY)"       on
 | 
				
			||||||
 | 
					 * kernel-dbg.txz  6b47a6cb83637af1f489aa8cdb802d9db936ea864887188cfc69d8075762214e        912     kernel_dbg      "Kernel (Debugging)"    on
 | 
				
			||||||
 | 
					 */
 | 
				
			||||||
 | 
					func buildFileChecksumFromManifest(manifest string, fileList []string) (map[string][]byte, error) {
 | 
				
			||||||
 | 
						var ckarr = make(map[string][]byte)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						rm, err := os.Open(manifest)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return ckarr, fmt.Errorf("Unable to open MANIFEST: %v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						fscan := bufio.NewScanner(rm)
 | 
				
			||||||
 | 
						fscan.Split(bufio.ScanLines)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// For each MANIFEST line...
 | 
				
			||||||
 | 
						for fscan.Scan() {
 | 
				
			||||||
 | 
							fields := strings.Fields(fscan.Text())
 | 
				
			||||||
 | 
							fn := fields[0]
 | 
				
			||||||
 | 
							fck := fields[1]
 | 
				
			||||||
 | 
							hexSum, err := hex.DecodeString(fck)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return ckarr, fmt.Errorf("Invalid value for checksum %s", fck)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							// ... Find the corresponding file in fileList, then add to checksum array ckarr
 | 
				
			||||||
 | 
							for _, f := range fileList {
 | 
				
			||||||
 | 
								if strings.EqualFold(f, fn) {
 | 
				
			||||||
 | 
									ckarr[fn] = hexSum
 | 
				
			||||||
 | 
									break
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if len(ckarr) < len(fileList) {
 | 
				
			||||||
 | 
							return ckarr, fmt.Errorf("Missing file in MANIFEST")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return ckarr, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										241
									
								
								cmd/jailhost.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										241
									
								
								cmd/jailhost.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,241 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"net"
 | 
				
			||||||
 | 
						"regexp"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
						"io/ioutil"
 | 
				
			||||||
 | 
						"golang.org/x/net/route"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					var defaultRoute4 = [4]byte{0, 0, 0, 0}
 | 
				
			||||||
 | 
					var defaultRoute6 = [16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}
 | 
				
			||||||
 | 
					var local6 = [16]byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func Inet4AddrToString(ip4 route.Inet4Addr) string {
 | 
				
			||||||
 | 
						return fmt.Sprintf("%v.%v.%v.%v", ip4.IP[0], ip4.IP[1], ip4.IP[2], ip4.IP[3])
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func Inet6AddrToString(ip6 route.Inet6Addr) string {
 | 
				
			||||||
 | 
						return fmt.Sprintf("%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x",
 | 
				
			||||||
 | 
							ip6.IP[0], ip6.IP[1], ip6.IP[2], ip6.IP[3], ip6.IP[4], ip6.IP[5], ip6.IP[6], ip6.IP[7],
 | 
				
			||||||
 | 
							ip6.IP[8], ip6.IP[9], ip6.IP[10], ip6.IP[11], ip6.IP[12], ip6.IP[13], ip6.IP[14], ip6.IP[15])
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/*****************************************************************************
 | 
				
			||||||
 | 
					* Initialize default_interface, default_gatway4, default_gateway6
 | 
				
			||||||
 | 
					*****************************************************************************/
 | 
				
			||||||
 | 
					func (jh *JailHost) InitNetworkProperties() {
 | 
				
			||||||
 | 
						rib, _ := route.FetchRIB(0, route.RIBTypeRoute, 0)
 | 
				
			||||||
 | 
						messages, err := route.ParseRIB(route.RIBTypeRoute, rib)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							panic(err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, message := range messages {
 | 
				
			||||||
 | 
							route_message := message.(*route.RouteMessage)
 | 
				
			||||||
 | 
							addresses := route_message.Addrs
 | 
				
			||||||
 | 
							card_index := route_message.Index
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if addresses[0].Family() == 2 {
 | 
				
			||||||
 | 
								var destination4, gateway4 *route.Inet4Addr
 | 
				
			||||||
 | 
								ok := false
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if destination4, ok = addresses[0].(*route.Inet4Addr); !ok {
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if gateway4, ok = addresses[1].(*route.Inet4Addr); !ok {
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if destination4 == nil || gateway4 == nil {
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if destination4.IP == defaultRoute4 {
 | 
				
			||||||
 | 
									card, _ := net.InterfaceByIndex(card_index)
 | 
				
			||||||
 | 
									//fmt.Printf("Default IPv4 gateway is %v on card %s\n", Inet4AddrToString(*gateway4), card.Name)
 | 
				
			||||||
 | 
									jh.default_interface = card.Name
 | 
				
			||||||
 | 
									jh.default_gateway4 = Inet4AddrToString(*gateway4)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							} else if addresses[0].Family() == 28 {
 | 
				
			||||||
 | 
								var destination6, gateway6 *route.Inet6Addr
 | 
				
			||||||
 | 
								ok := false
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if destination6, ok = addresses[0].(*route.Inet6Addr); !ok {
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if gateway6, ok = addresses[1].(*route.Inet6Addr); !ok {
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if destination6 == nil || gateway6 == nil {
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								if destination6.IP == defaultRoute6 && gateway6.IP != local6 {
 | 
				
			||||||
 | 
									card, _ := net.InterfaceByIndex(card_index)
 | 
				
			||||||
 | 
									//fmt.Printf("Default IPv6 gateway is %v on card %s\n", Inet6AddrToString(*gateway6), card.Name)
 | 
				
			||||||
 | 
									jh.default_interface = card.Name
 | 
				
			||||||
 | 
									jh.default_gateway6 = Inet6AddrToString(*gateway6)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (jh *JailHost) GetDefaultInterface() string {
 | 
				
			||||||
 | 
						if len(jh.default_interface) == 0 {
 | 
				
			||||||
 | 
							jh.InitNetworkProperties()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return jh.default_interface
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (jh *JailHost) GetDefaultGateway4() string {
 | 
				
			||||||
 | 
						if len(jh.default_gateway4) == 0 {
 | 
				
			||||||
 | 
							jh.InitNetworkProperties()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return jh.default_gateway4
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (jh *JailHost) GetDefaultGateway6() string {
 | 
				
			||||||
 | 
						if len(jh.default_gateway6) == 0 {
 | 
				
			||||||
 | 
							jh.InitNetworkProperties()
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return jh.default_gateway6
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/*****************************************************************************
 | 
				
			||||||
 | 
					 * Get all network interfaces
 | 
				
			||||||
 | 
					 ****************************************************************************/
 | 
				
			||||||
 | 
					func (jh *JailHost) GetInterfaces() ([]string, error) {
 | 
				
			||||||
 | 
						var names []string
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						interfaces, err := net.Interfaces()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return names, fmt.Errorf("Error listing network interfaces: %v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						for _, n := range interfaces {
 | 
				
			||||||
 | 
							names = append(names, n.Name)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return names, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (jh *JailHost) GetBridgeMTU(bridgeName string) (int, error) {
 | 
				
			||||||
 | 
						bridge, err := net.InterfaceByName(bridgeName)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return 0, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return bridge.MTU, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/*****************************************************************************
 | 
				
			||||||
 | 
					 * Get all IPv4 currently in use on host
 | 
				
			||||||
 | 
					 ****************************************************************************/
 | 
				
			||||||
 | 
					func getHostInUseIPv4() ([]string, error) {
 | 
				
			||||||
 | 
						var ips []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						re := regexp.MustCompile(ifconfigipv4re)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						out, err := executeCommand("/sbin/ifconfig")
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return ips, fmt.Errorf("Error executing \"/sbin/ifconfig\": %s", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, line := range strings.Split(out, "\n") {
 | 
				
			||||||
 | 
							if re.MatchString(line) {
 | 
				
			||||||
 | 
								ips = append(ips, re.FindStringSubmatch(line)[1])
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return ips, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func getHostname() (string, error) {
 | 
				
			||||||
 | 
						out, err := executeCommand("/bin/hostname")
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return "", fmt.Errorf("Error executing \"/bin/hostname\": %v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return strings.Split(out, "\n")[0], nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func getHostId() (string, error) {
 | 
				
			||||||
 | 
						var content []byte
 | 
				
			||||||
 | 
						var err     error
 | 
				
			||||||
 | 
						// return empty string if file does not exist
 | 
				
			||||||
 | 
						if content, err = ioutil.ReadFile("/etc/hostid"); err != nil {
 | 
				
			||||||
 | 
							if strings.HasSuffix(err.Error(), "no such file or directory") {
 | 
				
			||||||
 | 
								return "", nil
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							return "", err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return strings.Split(string(content), "\n")[0], nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func getArch() (string, error) {
 | 
				
			||||||
 | 
						out, err := executeCommand("/usr/bin/uname -p")
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return "", fmt.Errorf("Error executing \"/usr/bin/uname -p\": %v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return strings.Split(out, "\n")[0], nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func getFreeBSDVersion() (FreeBSDVersion, error) {
 | 
				
			||||||
 | 
						var version FreeBSDVersion
 | 
				
			||||||
 | 
						regex := `([0-9]{1,2})(\.)?([0-9]{1,2})?\-([^\-]*)(\-)?(p[0-9]{1,2})?`
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						re := regexp.MustCompile(regex)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						out, err := executeCommand("/bin/freebsd-version")
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return version, fmt.Errorf("Error executing \"/bin/freebsd-version\": %v", err)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if re.MatchString(out) {
 | 
				
			||||||
 | 
							version.major, err = strconv.Atoi(re.FindStringSubmatch(out)[1])
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return version, err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							version.minor, err = strconv.Atoi(re.FindStringSubmatch(out)[3])
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return version, err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							version.flavor = strings.Trim(re.FindStringSubmatch(out)[4], "\n")
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// Skip the 'p' starting patch level
 | 
				
			||||||
 | 
							if len(re.FindStringSubmatch(out)[6]) > 0 {
 | 
				
			||||||
 | 
								version.patchLevel, err = strconv.Atoi(re.FindStringSubmatch(out)[6][1:])
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									return version, err
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return version, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func NewJailHost() (JailHost, error) {
 | 
				
			||||||
 | 
						var jh  JailHost
 | 
				
			||||||
 | 
						var err error
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if jh.hostname, err = getHostname(); err != nil {
 | 
				
			||||||
 | 
							return jh, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if jh.arch, err = getArch(); err != nil {
 | 
				
			||||||
 | 
							return jh, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if jh.hostid, err = getHostId(); err != nil {
 | 
				
			||||||
 | 
							return jh, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if jh.version, err = getFreeBSDVersion(); err != nil {
 | 
				
			||||||
 | 
							return jh, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						return jh, nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										528
									
								
								cmd/list.go
									
									
									
									
									
								
							
							
						
						
									
										528
									
								
								cmd/list.go
									
									
									
									
									
								
							@ -1,316 +1,121 @@
 | 
				
			|||||||
package cmd
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"os"
 | 
						"encoding/json"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
 | 
						"gocage/jail"
 | 
				
			||||||
 | 
						"io/ioutil"
 | 
				
			||||||
	"log"
 | 
						"log"
 | 
				
			||||||
	"errors"
 | 
						"os"
 | 
				
			||||||
	"reflect"
 | 
						"reflect"
 | 
				
			||||||
	"strings"
 | 
						"strings"
 | 
				
			||||||
	"io/ioutil"
 | 
					 | 
				
			||||||
	"gocage/jail"
 | 
					 | 
				
			||||||
	"encoding/json"
 | 
					 | 
				
			||||||
	"github.com/spf13/viper"
 | 
					 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
// Recurse into structure, returning reflect.Value of wanted field.
 | 
					 * List all properties a jail have, with their internal name
 | 
				
			||||||
//  Nested fields are named with a dot (ex "MyStruct.MyField")
 | 
					 * Only print properties name. To get name & values, use GetJailProperties()
 | 
				
			||||||
func getStructFieldValue(parentStruct interface{}, fieldName string) (*reflect.Value, string, error) {
 | 
					 *******************************************************************************/
 | 
				
			||||||
	v := reflect.ValueOf(parentStruct)
 | 
					func ListJailsProps(args []string) {
 | 
				
			||||||
 | 
						var conf Jail
 | 
				
			||||||
/*	if v.Kind() == reflect.Ptr {
 | 
						var result []string
 | 
				
			||||||
		v = v.Elem()
 | 
						// Mandatory constructor to init default values
 | 
				
			||||||
	}
 | 
						jailconf, err := NewJailConfig()
 | 
				
			||||||
*/
 | 
						if err != nil {
 | 
				
			||||||
	if false {
 | 
							fmt.Printf("Error allocating JailConfig: %s\n", err.Error())
 | 
				
			||||||
		for i := 0 ; i < v.NumField(); i++ {
 | 
					 | 
				
			||||||
			f := v.Field(i)
 | 
					 | 
				
			||||||
			if f.Kind() == reflect.String {
 | 
					 | 
				
			||||||
				fmt.Printf("%v=%v\n", v.Type().Field(i).Name, f.Interface())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if strings.Contains(fieldName, ".") {
 | 
					 | 
				
			||||||
		fs := strings.Split(fieldName, ".")
 | 
					 | 
				
			||||||
		//f := v.FieldByName(fs[0])
 | 
					 | 
				
			||||||
		f := v.Elem().FieldByName(fs[0])
 | 
					 | 
				
			||||||
		if f.Kind() == reflect.Struct {
 | 
					 | 
				
			||||||
			return getStructFieldValue(f.Interface(), strings.Join(fs[1:], "."))
 | 
					 | 
				
			||||||
		} else {
 | 
					 | 
				
			||||||
			log.Fatalln(fmt.Sprintf("%s is not a struct: %s\n", fs[0], f.Kind().String()))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	} else {
 | 
					 | 
				
			||||||
		f := v.FieldByName(fieldName)
 | 
					 | 
				
			||||||
		if f.IsValid() {
 | 
					 | 
				
			||||||
			return &f, fieldName, nil
 | 
					 | 
				
			||||||
		} else {
 | 
					 | 
				
			||||||
			return &v, fieldName, errors.New(fmt.Sprintf("Field not found: %s", fieldName))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return &v, fieldName, nil
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
// TODO : Replace by getStructFieldValue
 | 
					 | 
				
			||||||
// Recurse into structure, returning reflect.Value of wanted field.
 | 
					 | 
				
			||||||
//  Nested fields are named with a dot (ex "MyStruct.MyField")
 | 
					 | 
				
			||||||
func getStructField(parentStruct interface{}, fieldName string) (reflect.Value, string) {
 | 
					 | 
				
			||||||
	v := reflect.ValueOf(parentStruct)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if v.Kind() == reflect.Ptr {
 | 
					 | 
				
			||||||
		v = v.Elem()
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if false {
 | 
					 | 
				
			||||||
		for i := 0 ; i < v.NumField(); i++ {
 | 
					 | 
				
			||||||
			f := v.Field(i)
 | 
					 | 
				
			||||||
			if f.Kind() == reflect.String {
 | 
					 | 
				
			||||||
				fmt.Printf("%v=%v\n", v.Type().Field(i).Name, f.Interface())
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if strings.Contains(fieldName, ".") {
 | 
					 | 
				
			||||||
		fs := strings.Split(fieldName, ".")
 | 
					 | 
				
			||||||
		f := v.FieldByName(fs[0])
 | 
					 | 
				
			||||||
		if f.Kind() == reflect.Struct {
 | 
					 | 
				
			||||||
			return getStructField(f.Interface(), strings.Join(fs[1:], "."))
 | 
					 | 
				
			||||||
		} else {
 | 
					 | 
				
			||||||
			log.Fatalln(fmt.Sprintf("%s is not a struct: %s\n", fs[0], f.Kind().String()))
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	return v, fieldName
 | 
					 | 
				
			||||||
}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
/* Pretty display of jails field
 | 
					 | 
				
			||||||
   Fields to show are given in a string array parameter
 | 
					 | 
				
			||||||
   Ex. : displayJails(["Name", "JID", "RootPath"])
 | 
					 | 
				
			||||||
*/
 | 
					 | 
				
			||||||
func displayStructFields(jails []Jail, valsToDisplay []string) {
 | 
					 | 
				
			||||||
	/* A line is defined by :
 | 
					 | 
				
			||||||
		Nr of fields
 | 
					 | 
				
			||||||
		For each field :
 | 
					 | 
				
			||||||
			Its name
 | 
					 | 
				
			||||||
			Its max length
 | 
					 | 
				
			||||||
			Its value
 | 
					 | 
				
			||||||
	*/
 | 
					 | 
				
			||||||
	type Field struct {
 | 
					 | 
				
			||||||
		Name	string
 | 
					 | 
				
			||||||
		MaxLen	int
 | 
					 | 
				
			||||||
		Value	string
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	type Line []Field
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	type Output []Line
 | 
					 | 
				
			||||||
	var out Output
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	//fmt.Printf("%d fields in a %d items structure\n", len(valsToDisplay), len(jails))
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// Browse structure to get max length and values
 | 
					 | 
				
			||||||
	for _, j := range jails {
 | 
					 | 
				
			||||||
		// Have to use a pointer, else reflect.Value.Elem() will panic : https://pkg.go.dev/reflect#Value.Elem
 | 
					 | 
				
			||||||
		tj := &j
 | 
					 | 
				
			||||||
		
 | 
					 | 
				
			||||||
		line := make([]Field, len(valsToDisplay))
 | 
					 | 
				
			||||||
		for i, f := range valsToDisplay {
 | 
					 | 
				
			||||||
			a, f := getStructField(tj, f)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
			field := Field {
 | 
					 | 
				
			||||||
				Name: f,
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			if a.FieldByName(f).IsValid() {
 | 
					 | 
				
			||||||
				// For now this just contains this item length, will adjust later
 | 
					 | 
				
			||||||
				//  We need to divide length by number of items in string fields (because we can have more than 1 ip4_addr)
 | 
					 | 
				
			||||||
				if reflect.TypeOf(a.FieldByName(f).Interface()).Kind() == reflect.String {
 | 
					 | 
				
			||||||
					itnr := len(strings.Split(string(a.FieldByName(f).Interface().(string)), ","))
 | 
					 | 
				
			||||||
					field.MaxLen = len(fmt.Sprintf("%v", a.FieldByName(f).Interface())) / itnr
 | 
					 | 
				
			||||||
				} else {
 | 
					 | 
				
			||||||
					field.MaxLen = len(fmt.Sprintf("%v", a.FieldByName(f).Interface()))
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				field.Value = fmt.Sprintf("%v", a.FieldByName(f).Interface())
 | 
					 | 
				
			||||||
			} else {
 | 
					 | 
				
			||||||
				fmt.Printf("Invalid field name: %s\n", f)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			line[i] = field
 | 
					 | 
				
			||||||
			
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		out = append(out, line)
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	if len(out) == 0 {
 | 
					 | 
				
			||||||
		fmt.Printf("Nothing to see here!\n")
 | 
					 | 
				
			||||||
		return
 | 
							return
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Get real maximum length
 | 
						conf.Config = jailconf
 | 
				
			||||||
	maxlen := make([]int, len(valsToDisplay))
 | 
					 | 
				
			||||||
	for i := 0; i< len(valsToDisplay); i++ {
 | 
					 | 
				
			||||||
		maxlen[i] = len(valsToDisplay[i])
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	for _, l := range out {
 | 
					 | 
				
			||||||
		for i, f := range l {
 | 
					 | 
				
			||||||
			if f.MaxLen > maxlen[i] {
 | 
					 | 
				
			||||||
				maxlen[i] = f.MaxLen
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Align maxlen to the real maximum length
 | 
						result = getStructFieldNames(conf, result, "")
 | 
				
			||||||
	for io, l := range out {
 | 
					 | 
				
			||||||
		for ii, _ := range l {
 | 
					 | 
				
			||||||
			// We need to access slice by index if we want to modify content
 | 
					 | 
				
			||||||
			out[io][ii].MaxLen = maxlen[ii]
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	totalLen := 0
 | 
						for _, f := range result {
 | 
				
			||||||
	// For each field, add separator and 2 blank spaces 
 | 
							fmt.Printf("%s\n", f)
 | 
				
			||||||
	for _, f := range out[0] {
 | 
					 | 
				
			||||||
		totalLen += f.MaxLen + 3
 | 
					 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	// Then add starting delimiter 
 | 
					 | 
				
			||||||
	totalLen += 1
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	Debug := false
 | 
					 | 
				
			||||||
	if Debug == true {
 | 
					 | 
				
			||||||
		for _, f := range out[0] {
 | 
					 | 
				
			||||||
			fmt.Printf("%s max length : %d\n", f.Name, f.MaxLen)
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// Lets draw things on the screen!
 | 
					 | 
				
			||||||
	// First, headers: 1st separator line
 | 
					 | 
				
			||||||
	for i, f := range out[0] {
 | 
					 | 
				
			||||||
		if i == 0 {	fmt.Printf("+")	}
 | 
					 | 
				
			||||||
		for i := 0 ; i < f.MaxLen+2 ; i++ {	fmt.Printf("=")	}
 | 
					 | 
				
			||||||
		fmt.Printf("+")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	fmt.Printf("\n")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	//  Column names
 | 
					 | 
				
			||||||
	for i, f := range out[0] {
 | 
					 | 
				
			||||||
		if i == 0 {
 | 
					 | 
				
			||||||
			fmt.Printf("|")
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		/* Use vlsToDisplay to get real name (with "Config.")
 | 
					 | 
				
			||||||
		fmt.Printf(" %s", f.Name)
 | 
					 | 
				
			||||||
		for i := len(f.Name)+1 ; i < f.MaxLen+1 ; i++ {  */
 | 
					 | 
				
			||||||
		fmt.Printf(" %s", valsToDisplay[i])
 | 
					 | 
				
			||||||
		for i := len(valsToDisplay[i])+1 ; i < f.MaxLen+1 ; i++ {
 | 
					 | 
				
			||||||
			fmt.Printf(" ")
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		fmt.Printf(" |")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	//  Finally separator line
 | 
					 | 
				
			||||||
	fmt.Printf("\n")
 | 
					 | 
				
			||||||
	for i, f := range out[0] {
 | 
					 | 
				
			||||||
		if i == 0 {	fmt.Printf("+")	}
 | 
					 | 
				
			||||||
		for i := 0 ; i < f.MaxLen+2 ; i++ {	fmt.Printf("=")	}
 | 
					 | 
				
			||||||
		fmt.Printf("+")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	fmt.Printf("\n")
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	// Then display data
 | 
					 | 
				
			||||||
	//  Loop through lines
 | 
					 | 
				
			||||||
	for _, l := range out {
 | 
					 | 
				
			||||||
		
 | 
					 | 
				
			||||||
		// Loop through fields
 | 
					 | 
				
			||||||
		//  In case we need to add a line for a 2nd IP, or whatever object
 | 
					 | 
				
			||||||
		var supplines = make(map[string]string)
 | 
					 | 
				
			||||||
		for i, f := range l {
 | 
					 | 
				
			||||||
			if i == 0 {
 | 
					 | 
				
			||||||
				fmt.Printf("|")
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			// Special cases of value displaying
 | 
					 | 
				
			||||||
			if f.Name == "JID" && f.Value == "0" {
 | 
					 | 
				
			||||||
				fmt.Printf("  ")
 | 
					 | 
				
			||||||
			} else if f.Name == "Ip4_addr" {
 | 
					 | 
				
			||||||
				ia := strings.Split(f.Value, ",")
 | 
					 | 
				
			||||||
				// If we have more than 1 value we need to finish this line, and store value for writing at the end of line loop
 | 
					 | 
				
			||||||
				for i, inter := range ia {
 | 
					 | 
				
			||||||
					if i > 0 {
 | 
					 | 
				
			||||||
						supplines[f.Name] = inter
 | 
					 | 
				
			||||||
					} else {
 | 
					 | 
				
			||||||
						fmt.Printf(" %s", inter)
 | 
					 | 
				
			||||||
					}
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				//fmt.Printf(" %s", strings.Split(strings.Split(f.Value, "|")[1], "/")[0])
 | 
					 | 
				
			||||||
			} else {
 | 
					 | 
				
			||||||
				fmt.Printf(" %s", f.Value)
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			// Complete with spaces to the max length
 | 
					 | 
				
			||||||
			for i := len(f.Value)+1 ; i < f.MaxLen+1 ; i++ {
 | 
					 | 
				
			||||||
				fmt.Printf(" ")
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
			fmt.Printf(" |")
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		// Draw supplementary lines
 | 
					 | 
				
			||||||
		if len(supplines) > 0 {
 | 
					 | 
				
			||||||
			for i, f := range l {
 | 
					 | 
				
			||||||
				if i == 0 {
 | 
					 | 
				
			||||||
					fmt.Printf("\n|")
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				// Overwrite value in this scope
 | 
					 | 
				
			||||||
				v, exists := supplines[f.Name]
 | 
					 | 
				
			||||||
				if exists {
 | 
					 | 
				
			||||||
					fmt.Printf(" %s", v)
 | 
					 | 
				
			||||||
				} else {
 | 
					 | 
				
			||||||
					// 1st option : Do not redisplay precedent line values
 | 
					 | 
				
			||||||
					fmt.Printf(" ")
 | 
					 | 
				
			||||||
					// 2nd option : Redisplay precedenet line values
 | 
					 | 
				
			||||||
					//fmt.Printf(" %s", f.Value)
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				// Complete with spaces to the max length
 | 
					 | 
				
			||||||
				for i := len(v)+1 ; i < f.MaxLen+1 ; i++ {
 | 
					 | 
				
			||||||
					fmt.Printf(" ")
 | 
					 | 
				
			||||||
				}
 | 
					 | 
				
			||||||
				fmt.Printf(" |")
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		// Draw line separator between jails
 | 
					 | 
				
			||||||
		if !gNoLineSep {
 | 
					 | 
				
			||||||
			fmt.Printf("\n")
 | 
					 | 
				
			||||||
			for i, f := range out[0] {
 | 
					 | 
				
			||||||
				if i == 0 {	fmt.Printf("+")	}
 | 
					 | 
				
			||||||
				for i := 0 ; i < f.MaxLen+2 ; i++ {	fmt.Printf("-")	}
 | 
					 | 
				
			||||||
				fmt.Printf("+")
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		fmt.Printf("\n")
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	if gNoLineSep {
 | 
					 | 
				
			||||||
		for i, f := range out[0] {
 | 
					 | 
				
			||||||
			if i == 0 {	fmt.Printf("+")	}
 | 
					 | 
				
			||||||
			for i := 0 ; i < f.MaxLen+2 ; i++ {	fmt.Printf("-")	}
 | 
					 | 
				
			||||||
			fmt.Printf("+")
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
	
 | 
					 | 
				
			||||||
	fmt.Printf("\n")
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
/* Get Jails from datastores. Store config and running metadata into gJails global var */
 | 
					 * Get Jails from datastores. Store config and running metadata
 | 
				
			||||||
 | 
					 *   into gJails global var
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
func ListJails(args []string, display bool) {
 | 
					func ListJails(args []string, display bool) {
 | 
				
			||||||
	fields := strings.Split(gDisplayColumns, ",")
 | 
						type uniqueJailName struct {
 | 
				
			||||||
 | 
							jail       string
 | 
				
			||||||
	for _, d := range viper.GetStringSlice("datastore") {
 | 
							unique     bool
 | 
				
			||||||
		listJailsFromDatastore(d)
 | 
							uniqueName bool
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						var nameChecked []*uniqueJailName
 | 
				
			||||||
 | 
						var curCheck *uniqueJailName
 | 
				
			||||||
 | 
						var found bool
 | 
				
			||||||
 | 
						var skip bool
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						for _, ds := range gDatastores {
 | 
				
			||||||
 | 
							listJailsFromDatastore(ds, args, display)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Only when displaying jails, we accept to process multiple same name jails
 | 
				
			||||||
 | 
						if false == display {
 | 
				
			||||||
 | 
							for _, j := range gJails {
 | 
				
			||||||
 | 
								// If already checked and was using distinctive name, skip this occurence
 | 
				
			||||||
 | 
								for i, n := range nameChecked {
 | 
				
			||||||
 | 
									if strings.EqualFold(n.jail, j.Name) {
 | 
				
			||||||
 | 
										found = true
 | 
				
			||||||
 | 
										if false == n.unique && true == n.uniqueName {
 | 
				
			||||||
 | 
											skip = true
 | 
				
			||||||
 | 
										} else {
 | 
				
			||||||
 | 
											curCheck = nameChecked[i]
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										break
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								if true == skip {
 | 
				
			||||||
 | 
									continue
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								// Initialize if not found in nameChecked
 | 
				
			||||||
 | 
								if false == found {
 | 
				
			||||||
 | 
									curCheck = &uniqueJailName{jail: j.Name,
 | 
				
			||||||
 | 
														unique: true,
 | 
				
			||||||
 | 
														uniqueName: false}
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									found = false
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								if countOfJailsWithThisName(j.Name) > 1 {
 | 
				
			||||||
 | 
									//fmt.Printf("DEBUG: Jail %s exist multiple times, now checking if specified with full name\n", j.Name)
 | 
				
			||||||
 | 
									curCheck.unique = false
 | 
				
			||||||
 | 
									
 | 
				
			||||||
 | 
									for _, a := range args {
 | 
				
			||||||
 | 
										//fmt.Printf("DEBUG: comparing %s/%s with %s\n", j.Datastore, j.Name, a)
 | 
				
			||||||
 | 
										if strings.EqualFold(a, fmt.Sprintf("%s/%s", j.Datastore, j.Name)) {
 | 
				
			||||||
 | 
											//fmt.Printf("DEBUG: Was specified with full name, GG!\n")
 | 
				
			||||||
 | 
											curCheck.uniqueName = true
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								nameChecked = append(nameChecked, curCheck)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// Now check
 | 
				
			||||||
 | 
							for _, a := range args {
 | 
				
			||||||
 | 
								for _, n := range nameChecked {
 | 
				
			||||||
 | 
									if strings.EqualFold(n.jail, a) && false == n.unique && false == n.uniqueName {
 | 
				
			||||||
 | 
										fmt.Printf("There is more than one jail named \"%s\", please use datastore/jail format\n", n.jail)
 | 
				
			||||||
 | 
										os.Exit(1)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						fields := strings.Split(gDisplayJColumns, ",")
 | 
				
			||||||
	
 | 
						
 | 
				
			||||||
	// This is the structure we will filter, then display
 | 
						// This is the structure we will filter, then display
 | 
				
			||||||
	var jails []Jail
 | 
						var jails []Jail
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	/************************************************************************************
 | 
						/***************************************************************
 | 
				
			||||||
	/   Filter jails with "filter" options
 | 
						/   Filter jails with "filter" options
 | 
				
			||||||
	/***********************************************************************************/
 | 
						/**************************************************************/
 | 
				
			||||||
	if len(gFilterJails) > 0 && gFilterJails != "none" {
 | 
						if len(gFilterJails) > 0 && gFilterJails != "none" {
 | 
				
			||||||
		flts := make(map[string]string)
 | 
							flts := make(map[string]string)
 | 
				
			||||||
		for _, flt := range strings.Split(gFilterJails, ",") {
 | 
							for _, flt := range strings.Split(gFilterJails, ",") {
 | 
				
			||||||
@ -341,37 +146,35 @@ func ListJails(args []string, display bool) {
 | 
				
			|||||||
		jails = gJails
 | 
							jails = gJails
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/***************************************************************
 | 
				
			||||||
	/************************************************************************************
 | 
					 | 
				
			||||||
	/   Filter jails by names given on command line
 | 
						/   Filter jails by names given on command line
 | 
				
			||||||
	/***********************************************************************************/
 | 
						/**************************************************************/
 | 
				
			||||||
	if len(args) > 0 {
 | 
						if len(args) > 0 {
 | 
				
			||||||
		var js []Jail
 | 
							var js []Jail
 | 
				
			||||||
		for _, a := range args {
 | 
							for _, a := range args {
 | 
				
			||||||
			for _, j := range jails {
 | 
								for _, j := range jails {
 | 
				
			||||||
				if j.Name == a {
 | 
									if j.Name == a {
 | 
				
			||||||
					js = append(js, j)
 | 
										js = append(js, j)
 | 
				
			||||||
					break
 | 
					 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		jails = js
 | 
							jails = js
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	/************************************************************************************
 | 
						/***************************************************************
 | 
				
			||||||
	/   Sort jails
 | 
						/   Sort jails
 | 
				
			||||||
	/     We support 3 sort criteria max
 | 
						/     We support 3 sort criteria max
 | 
				
			||||||
	/***********************************************************************************/
 | 
						/**************************************************************/
 | 
				
			||||||
	if len(gSortFields) > 0 && gSortFields != "none" {
 | 
						if len(gSortJailFields) > 0 && gSortJailFields != "none" {
 | 
				
			||||||
		js := initSortStruct()
 | 
							js := initJailSortStruct()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		// The way we manage criteria quantity is not very elegant...
 | 
							// The way we manage criteria quantity is not very elegant...
 | 
				
			||||||
		var fct1, fct2, fct3 *reflect.Value
 | 
							var fct1, fct2, fct3 *reflect.Value
 | 
				
			||||||
		for i, c := range strings.Split(gSortFields, ",") {
 | 
							for i, c := range strings.Split(gSortJailFields, ",") {
 | 
				
			||||||
			var fctName string
 | 
								var fctName string
 | 
				
			||||||
			if strings.HasPrefix(c, "-") {
 | 
								if strings.HasPrefix(c, "-") {
 | 
				
			||||||
				fctName = fmt.Sprintf("%sDec", strings.Replace(c, "-", "", 1))
 | 
									fctName = fmt.Sprintf("%sDec", strings.Replace(c, "-", "", 1))
 | 
				
			||||||
			} else {	// Par defaut (pas de prefix +/-) on considere un tri incremental	
 | 
								} else { // Par defaut (pas de prefix +/-) on considere un tri incremental
 | 
				
			||||||
				fctName = fmt.Sprintf("%sInc", strings.Replace(c, "+", "", 1))
 | 
									fctName = fmt.Sprintf("%sInc", strings.Replace(c, "+", "", 1))
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@ -382,82 +185,102 @@ func ListJails(args []string, display bool) {
 | 
				
			|||||||
				fmt.Printf("ERROR getting JailSort struct field %s. Please check the field name: %s\n", fctName, fieldName)
 | 
									fmt.Printf("ERROR getting JailSort struct field %s. Please check the field name: %s\n", fctName, fieldName)
 | 
				
			||||||
				return
 | 
									return
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
			switch i+1 {
 | 
								switch i + 1 {
 | 
				
			||||||
				case 1:	fct1 = fct
 | 
								case 1:
 | 
				
			||||||
				case 2: fct2 = fct
 | 
									fct1 = fct
 | 
				
			||||||
				case 3: fct3 = fct
 | 
								case 2:
 | 
				
			||||||
 | 
									fct2 = fct
 | 
				
			||||||
 | 
								case 3:
 | 
				
			||||||
 | 
									fct3 = fct
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		switch len(strings.Split(gSortFields, ",")) {
 | 
							switch len(strings.Split(gSortJailFields, ",")) {
 | 
				
			||||||
			case 1:
 | 
							case 1:
 | 
				
			||||||
				OrderedBy(fct1.Interface().(lessFunc)).Sort(jails)
 | 
								JailsOrderedBy(fct1.Interface().(jailLessFunc)).Sort(jails)
 | 
				
			||||||
			case 2:
 | 
							case 2:
 | 
				
			||||||
				OrderedBy(fct1.Interface().(lessFunc), fct2.Interface().(lessFunc)).Sort(jails)
 | 
								JailsOrderedBy(fct1.Interface().(jailLessFunc), fct2.Interface().(jailLessFunc)).Sort(jails)
 | 
				
			||||||
			case 3:
 | 
							case 3:
 | 
				
			||||||
				OrderedBy(fct1.Interface().(lessFunc), fct2.Interface().(lessFunc), fct3.Interface().(lessFunc)).Sort(jails)
 | 
								JailsOrderedBy(fct1.Interface().(jailLessFunc), fct2.Interface().(jailLessFunc), fct3.Interface().(jailLessFunc)).Sort(jails)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	/************************************************************************************
 | 
						/***************************************************************
 | 
				
			||||||
	/   Finally, display jails
 | 
						/   Finally, display jails
 | 
				
			||||||
	/***********************************************************************************/
 | 
						/**************************************************************/
 | 
				
			||||||
	if display {
 | 
						if display {
 | 
				
			||||||
		displayStructFields(jails, fields)
 | 
							displayJailsFields(jails, fields)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func listJailsFromDatastore(ds Datastore, args []string, accept_multiple bool) {
 | 
				
			||||||
	
 | 
						
 | 
				
			||||||
func listJailsFromDatastore(datastore string) {
 | 
						fileInfo, err := os.Stat(ds.Mountpoint)
 | 
				
			||||||
	fileInfo, err := os.Stat(datastore)
 | 
						if err != nil {
 | 
				
			||||||
	if err != nil { log.Fatalln(fmt.Sprintf("Unable to access %s, check path and/or rights", datastore)) }
 | 
							log.Fatalln(fmt.Sprintf("Unable to access %s, check path and/or rights", ds.Mountpoint))
 | 
				
			||||||
	if fileInfo.IsDir() == false { log.Fatalln(fmt.Sprintf("%s is not a directory", datastore)) }
 | 
						}
 | 
				
			||||||
 | 
						if fileInfo.IsDir() == false {
 | 
				
			||||||
 | 
							log.Fatalln(fmt.Sprintf("%s is not a directory", ds.Mountpoint))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// A datastore have to contain a "jails" directory
 | 
						// A datastore have to contain a "jails" directory
 | 
				
			||||||
	jailsDir := fmt.Sprintf("%s/jails", datastore)
 | 
						jailsDir := fmt.Sprintf("%s/jails", ds.Mountpoint)
 | 
				
			||||||
	fileInfo, err = os.Stat(jailsDir)
 | 
						fileInfo, err = os.Stat(jailsDir)
 | 
				
			||||||
	if err != nil { log.Fatalln(fmt.Sprintf("Unable to access %s, check path and/or rights", jailsDir)) }
 | 
						if err != nil {
 | 
				
			||||||
	if fileInfo.IsDir() == false { log.Fatalln(fmt.Sprintf("%s is not a directory", jailsDir)) }
 | 
							log.Fatalln(fmt.Sprintf("Unable to access %s, check path and/or rights", jailsDir))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						if fileInfo.IsDir() == false {
 | 
				
			||||||
 | 
							log.Fatalln(fmt.Sprintf("%s is not a directory", jailsDir))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	listJailsFromDirectory(jailsDir)
 | 
						listJailsFromDirectory(jailsDir, ds.Name)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func listJailsFromDirectory(dir string, dsname string) ([]Jail, error) {
 | 
				
			||||||
func listJailsFromDirectory(dir string) ([]Jail) {
 | 
					 | 
				
			||||||
	files, err := ioutil.ReadDir(dir)
 | 
						files, err := ioutil.ReadDir(dir)
 | 
				
			||||||
	if err != nil { log.Fatalln(fmt.Sprintf("Unable to browse %s, check path and/or rights", dir)) }
 | 
						if err != nil {
 | 
				
			||||||
 | 
							log.Fatalln(fmt.Sprintf("Unable to browse %s, check path and/or rights", dir))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	for _, fi := range files {
 | 
						for _, fi := range files {
 | 
				
			||||||
		if fi.IsDir() == true {
 | 
							if fi.IsDir() == true {
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		// 1. Get conf from config.json
 | 
								// 1. Get conf from config.json
 | 
				
			||||||
			jailConfPath := fmt.Sprintf("%s/%s/%s", dir, fi.Name(), "config.json")
 | 
								jailConfPath := fmt.Sprintf("%s/%s/%s", dir, fi.Name(), "config.json")
 | 
				
			||||||
			jailConf, err := getJailConfig(jailConfPath)
 | 
								jailConf, err := getJailConfig(jailConfPath)
 | 
				
			||||||
			if err != nil { log.Println("ERROR reading jail config for %s", jailConfPath) }
 | 
								if err != nil {
 | 
				
			||||||
	
 | 
									log.Println("ERROR reading jail config for %s", jailConfPath)
 | 
				
			||||||
		// 2. Build jail object from config
 | 
					 | 
				
			||||||
			jailRootPath := fmt.Sprintf("%s/%s/%s", dir, fi.Name(), "root")
 | 
					 | 
				
			||||||
			j := Jail{
 | 
					 | 
				
			||||||
				Name: jailConf.Host_hostname,
 | 
					 | 
				
			||||||
				Config: jailConf,
 | 
					 | 
				
			||||||
				ConfigPath: jailConfPath,
 | 
					 | 
				
			||||||
				RootPath: jailRootPath,
 | 
					 | 
				
			||||||
				Running: false,
 | 
					 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		// 3. Add current running informations
 | 
								// 2. Build jail object from config
 | 
				
			||||||
 | 
								jailRootPath := fmt.Sprintf("%s/%s/%s", dir, fi.Name(), "root")
 | 
				
			||||||
 | 
								j := Jail{
 | 
				
			||||||
 | 
									Name:       jailConf.Host_hostuuid,
 | 
				
			||||||
 | 
									Config:     jailConf,
 | 
				
			||||||
 | 
									ConfigPath: jailConfPath,
 | 
				
			||||||
 | 
									Datastore:  dsname,
 | 
				
			||||||
 | 
									RootPath:   jailRootPath,
 | 
				
			||||||
 | 
									Running:    false,
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								// 3. Add current running informations
 | 
				
			||||||
			rjails, err := jail.GetJails()
 | 
								rjails, err := jail.GetJails()
 | 
				
			||||||
			if err != nil { log.Fatalln("Unable to list running jails") }
 | 
								if err != nil {
 | 
				
			||||||
 | 
									log.Fatalln("Unable to list running jails")
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
			for _, rj := range rjails {
 | 
								for _, rj := range rjails {
 | 
				
			||||||
				if rj.Path == j.RootPath {
 | 
									if rj.Path == j.RootPath {
 | 
				
			||||||
					j.JID = rj.Jid
 | 
										j.JID = rj.Jid
 | 
				
			||||||
					j.Running = true
 | 
										j.Running = true
 | 
				
			||||||
 | 
										// FIXME ??? Shouldn't be ioc-$Name ?
 | 
				
			||||||
					j.InternalName = rj.Name
 | 
										j.InternalName = rj.Name
 | 
				
			||||||
 | 
										j.Devfs_ruleset = rj.Devfs_ruleset
 | 
				
			||||||
					break
 | 
										break
 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
/* This op take some 600ms for ~40 jails :^( */
 | 
								// 4. Get Zpool
 | 
				
			||||||
 | 
								/* This op take some 600ms for ~40 jails :^( */
 | 
				
			||||||
			out, err := executeCommand(fmt.Sprintf("zfs list -H -o name %s", j.RootPath))
 | 
								out, err := executeCommand(fmt.Sprintf("zfs list -H -o name %s", j.RootPath))
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
				fmt.Printf("ERROR getting dataset from %s: %s\n", j.RootPath, err.Error())
 | 
									fmt.Printf("ERROR getting dataset from %s: %s\n", j.RootPath, err.Error())
 | 
				
			||||||
@ -465,21 +288,40 @@ func listJailsFromDirectory(dir string) ([]Jail) {
 | 
				
			|||||||
				j.Zpool = strings.Split(strings.TrimSuffix(out, "\n"), "/")[0]
 | 
									j.Zpool = strings.Split(strings.TrimSuffix(out, "\n"), "/")[0]
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								// Check if jail with the same name already exist on another DS
 | 
				
			||||||
 | 
								for _, jj := range gJails {
 | 
				
			||||||
 | 
									if strings.EqualFold(jj.Name, j.Name) {
 | 
				
			||||||
 | 
										/*fmt.Printf("Warning: A jail with name %s exist on datastore %s, use full name to interact!\n", j.Name, jj.Datastore)
 | 
				
			||||||
 | 
										fmt.Printf("(Ex: gocage start %s/%s)\n", jj.Datastore, j.Name)*/
 | 
				
			||||||
 | 
										// Add Datastore  to avoid confusion
 | 
				
			||||||
 | 
										if false == strings.Contains(gDisplayJColumns, "Datastore") {
 | 
				
			||||||
 | 
											gDisplayJColumns += ",Datastore"
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								
 | 
				
			||||||
			gJails = append(gJails, j)
 | 
								gJails = append(gJails, j)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return gJails
 | 
						return gJails, nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					 | 
				
			||||||
func getJailConfig(jailConfigPath string) (JailConfig, error) {
 | 
					func getJailConfig(jailConfigPath string) (JailConfig, error) {
 | 
				
			||||||
	content, err := ioutil.ReadFile(jailConfigPath)
 | 
						content, err := ioutil.ReadFile(jailConfigPath)
 | 
				
			||||||
	if err != nil { log.Fatalln(fmt.Sprintf("Unable to read %s, check path and/or rights", jailConfigPath)) }
 | 
						if err != nil {
 | 
				
			||||||
 | 
							log.Fatalln(fmt.Sprintf("Unable to read %s, check path and/or rights", jailConfigPath))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	var jc JailConfig
 | 
						// Mandatory constructor to init default values
 | 
				
			||||||
 | 
						jc, err := NewJailConfig()
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return jc, err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
	err = json.Unmarshal([]byte(content), &jc)
 | 
						err = json.Unmarshal([]byte(content), &jc)
 | 
				
			||||||
	if err != nil { log.Fatalln(fmt.Sprintf("Error occured during unmarshaling %s: %s", jailConfigPath, err.Error())) }
 | 
						if err != nil {
 | 
				
			||||||
 | 
							log.Fatalln(fmt.Sprintf("Error occured during unmarshaling %s: %s", jailConfigPath, err.Error()))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return jc, err
 | 
						return jc, err
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										224
									
								
								cmd/migrate.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										224
									
								
								cmd/migrate.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,224 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"os"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"bufio"
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const (
 | 
				
			||||||
 | 
						// 1MB
 | 
				
			||||||
 | 
						BUFFER_SIZE = 512
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * Migrate jail to another zfs pool
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func MigrateJail(args []string) {
 | 
				
			||||||
 | 
						var jailNames []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for _, a := range args {
 | 
				
			||||||
 | 
								jailNames = append(jailNames, a)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, jn := range jailNames {
 | 
				
			||||||
 | 
							cj, err := getJailFromArray(jn, gJails)
 | 
				
			||||||
 | 
							if cj == nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error getting jail %s: Not found\n", jn)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// Check if destination datastore exist & get current DS
 | 
				
			||||||
 | 
							destDS, err := getDatastoreFromArray(gMigrateDestDatastore, gDatastores)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error getting datastore \"%s\": %v\n", gMigrateDestDatastore, err)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							curDS, err := getDatastoreFromArray(cj.Datastore, gDatastores)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error getting datastore \"%s\": %v\n", gMigrateDestDatastore, err)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							if cj.Running == true && gYesToAll == false {
 | 
				
			||||||
 | 
								fmt.Printf("WARNING: Jail %s is running\n", cj.Name)
 | 
				
			||||||
 | 
								fmt.Printf("Migration will stop it for data sync before starting on the new pool. You will be prompted for shutdown.\n")
 | 
				
			||||||
 | 
								fmt.Printf("Continue? (y/n) ")
 | 
				
			||||||
 | 
								scanr := bufio.NewScanner(os.Stdin)
 | 
				
			||||||
 | 
								scanr.Scan()
 | 
				
			||||||
 | 
								if false == strings.EqualFold(scanr.Text(), "y") {
 | 
				
			||||||
 | 
									fmt.Printf("Migration aborted\n")
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							/* TODO : Check dest pool (gMigrateDestDatastore) existence
 | 
				
			||||||
 | 
									zfs snapshot /iocage/jails/$jail@gocage_mig_first_snap
 | 
				
			||||||
 | 
									zfs snapshot /iocage/jails/$jail/root@gocage_mig_first_snap
 | 
				
			||||||
 | 
									zfs send jail@gocage_mig_first_snap | zfs receive destpool/jails/jail_name
 | 
				
			||||||
 | 
									zfs send jail/root@gocage_mig_first_snap | zfs receive destpool/jails/jail_name/root
 | 
				
			||||||
 | 
									shutdown jail if needed
 | 
				
			||||||
 | 
									if jail was shutdown
 | 
				
			||||||
 | 
										zfs snapshot /iocage/jails/$jail@gocage_mig_last_snap to get last data
 | 
				
			||||||
 | 
										zfs send -i jail@gocage_mig_first_snap jail/root@gocage_mig_last_snap | zfs receive -F destpool/jails/jail_name
 | 
				
			||||||
 | 
										start jail on new dest
 | 
				
			||||||
 | 
										zfs destroy destpool/jails/jail_name@gocage_mig_first_snap
 | 
				
			||||||
 | 
										zfs destroy destpool/jails/jail_name/root@gocage_mig_first_snap
 | 
				
			||||||
 | 
							*/
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Snapshot config
 | 
				
			||||||
 | 
							dsconf := strings.Join([]string{curDS.ZFSDataset, "jails", jn}, "/")
 | 
				
			||||||
 | 
							fmt.Printf("Snapshot %s: ", dsconf)
 | 
				
			||||||
 | 
							if err = zfsSnapshot(dsconf, "gocage_mig_init"); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							fmt.Printf("Done\n")
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// Snapshot filesystem
 | 
				
			||||||
 | 
							dsdata := strings.Join([]string{curDS.ZFSDataset, "jails", jn, "root"}, "/")
 | 
				
			||||||
 | 
							fmt.Printf("Snapshot %s: ", dsdata)
 | 
				
			||||||
 | 
							if err := zfsSnapshot(dsdata, "gocage_mig_init"); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							fmt.Printf("Done\n")
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// TODO: Browse jailed ZFS dataset to migrate them (see poudriere for example)
 | 
				
			||||||
 | 
							// zfs send -R src/poudriere@init | zfs receive -Fu dest/poudriere
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							dsconfdest := strings.Join([]string{destDS.ZFSDataset, "jails", jn}, "/")
 | 
				
			||||||
 | 
							fmt.Printf("Migrate jail config dataset to %s: ", dsconfdest)
 | 
				
			||||||
 | 
							if err := zfsCopy(fmt.Sprintf("%s@gocage_mig_init", dsconf), dsconfdest); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							fmt.Printf("Done\n")
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							dsdatadest := strings.Join([]string{destDS.ZFSDataset, "jails", jn, "root"}, "/")
 | 
				
			||||||
 | 
							fmt.Printf("Migrate jail filesystem dataset to %s: ", dsdatadest)
 | 
				
			||||||
 | 
							if err := zfsCopy(fmt.Sprintf("%s@gocage_mig_init", dsdata), dsdatadest); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							fmt.Printf("Done\n")
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// Running jail needs a last snapshot for an incremental send/recv after shutting down.
 | 
				
			||||||
 | 
							if cj.Running == true {
 | 
				
			||||||
 | 
								fmt.Printf("Shutdown jail %s for last data sync, this could take some time.\n", cj.Name)
 | 
				
			||||||
 | 
								if gYesToAll == false {
 | 
				
			||||||
 | 
									fmt.Printf("Continue? (y/n) ")
 | 
				
			||||||
 | 
									scanr := bufio.NewScanner(os.Stdin)
 | 
				
			||||||
 | 
									scanr.Scan()
 | 
				
			||||||
 | 
									if false == strings.EqualFold(scanr.Text(), "y") {
 | 
				
			||||||
 | 
										fmt.Printf("Migration aborted. Now cleaning destination pool.\n")
 | 
				
			||||||
 | 
										if err := CleanMigrateMess([]string{cj.Name}); err != nil {
 | 
				
			||||||
 | 
											fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										// TODO : Remove destination datasets, or handle the cas at the beginning of current function
 | 
				
			||||||
 | 
										// (when snapshot already exist on source and dest)
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								StopJail([]string{cj.Name})
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								fmt.Printf("Snapshot %s: ", dsconf)
 | 
				
			||||||
 | 
								if err = zfsSnapshot(dsconf, "gocage_mig_last_sync"); err != nil {
 | 
				
			||||||
 | 
									fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								fmt.Printf("Done\n")
 | 
				
			||||||
 | 
								fmt.Printf("Snapshot %s: ", dsdata)
 | 
				
			||||||
 | 
								if err := zfsSnapshot(dsdata, "gocage_mig_last_sync"); err != nil {
 | 
				
			||||||
 | 
									fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								fmt.Printf("Done\n")
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								fmt.Printf("Synchronize jail config to %s: ", dsconfdest)
 | 
				
			||||||
 | 
								if err := zfsCopyIncremental(fmt.Sprintf("%s@gocage_mig_init", dsconf), 
 | 
				
			||||||
 | 
															fmt.Sprintf("%s@gocage_mig_last_sync", dsconf),  
 | 
				
			||||||
 | 
															 dsconfdest); err != nil {
 | 
				
			||||||
 | 
									fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								fmt.Printf("Done\n")
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								fmt.Printf("Synchronize jail filesystem dataset to %s: ", dsdatadest)
 | 
				
			||||||
 | 
								if err := zfsCopyIncremental(fmt.Sprintf("%s@gocage_mig_init", dsdata), 
 | 
				
			||||||
 | 
															fmt.Sprintf("%s@gocage_mig_last_sync", dsdata),  
 | 
				
			||||||
 | 
															dsdatadest); err != nil {
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
									fmt.Printf("Error: %v\n", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								fmt.Printf("Done\n")
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								// TODO : Start jail on new datastore (! Currently ListJails won't support 2 jails with same name !)
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								// TODO : zfs destroy destpool/jails/jail_name@gocage_mig_first_snap
 | 
				
			||||||
 | 
								// TODO : zfs destroy destpool/jails/jail_name/root@gocage_mig_first_snap
 | 
				
			||||||
 | 
								// TODO : zfs destroy -r srcpool/iocage/jails/$jail
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Clean snapshots from an aborted migration
 | 
				
			||||||
 | 
					func CleanMigrateMess(args []string) error {
 | 
				
			||||||
 | 
						var jailNames []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for _, a := range args {
 | 
				
			||||||
 | 
								jailNames = append(jailNames, a)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, jn := range jailNames {
 | 
				
			||||||
 | 
							cj, err := getJailFromArray(jn, gJails)
 | 
				
			||||||
 | 
							if cj == nil {
 | 
				
			||||||
 | 
								return errors.New(fmt.Sprintf("Error getting jail %s: Not found\n", jn))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							curDS, err := getDatastoreFromArray(cj.Datastore, gDatastores)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								return errors.New(fmt.Sprintf("Error getting datastore \"%s\": %v\n", cj.Datastore, err))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							cmd := fmt.Sprintf("zfs destroy %s@gocage_mig_init", strings.Join([]string{curDS.ZFSDataset, "jails", jn}, "/"))
 | 
				
			||||||
 | 
							out, err := executeCommand(cmd)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								if false == strings.HasSuffix(out, "could not find any snapshots to destroy; check snapshot names.\n") {
 | 
				
			||||||
 | 
									fmt.Printf("Error executing command %s: %v; command returned: %s\n", cmd, err, out)
 | 
				
			||||||
 | 
									return errors.New(fmt.Sprintf("Error executing command %s: %v; command returned: %s\n", cmd, err, out))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							cmd = fmt.Sprintf("zfs destroy %s@gocage_mig_init", strings.Join([]string{curDS.ZFSDataset, "jails", jn, "root"}, "/"))
 | 
				
			||||||
 | 
							out, err = executeCommand(cmd)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								if false == strings.HasSuffix(out, "could not find any snapshots to destroy; check snapshot names.\n") {
 | 
				
			||||||
 | 
									fmt.Printf("Error executing command %s: %v; command returned: %s\n", cmd, err, out)
 | 
				
			||||||
 | 
									return errors.New(fmt.Sprintf("Error executing command %s: %v; command returned: %s\n", cmd, err, out))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							cmd = fmt.Sprintf("zfs destroy %s@gocage_mig_last_sync", strings.Join([]string{curDS.ZFSDataset, "jails", jn}, "/"))
 | 
				
			||||||
 | 
							out, err = executeCommand(cmd)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								if false == strings.HasSuffix(out, "could not find any snapshots to destroy; check snapshot names.\n") {
 | 
				
			||||||
 | 
									fmt.Printf("Error executing command %s: %v; command returned: %s\n", cmd, err, out)
 | 
				
			||||||
 | 
									return errors.New(fmt.Sprintf("Error executing command %s: %v; command returned: %s\n", cmd, err, out))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							cmd = fmt.Sprintf("zfs destroy %s@gocage_mig_last_sync", strings.Join([]string{curDS.ZFSDataset, "jails", jn, "root"}, "/"))
 | 
				
			||||||
 | 
							out, err = executeCommand(cmd)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								if false == strings.HasSuffix(out, "could not find any snapshots to destroy; check snapshot names.\n") {
 | 
				
			||||||
 | 
									fmt.Printf("Error executing command %s: %v; command returned: %s\n", cmd, err, out)
 | 
				
			||||||
 | 
									return errors.New(fmt.Sprintf("Error executing command %s: %v; command returned: %s\n", cmd, err, out))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										123
									
								
								cmd/properties.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										123
									
								
								cmd/properties.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,123 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"reflect"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func GetJailProperties(args []string) {
 | 
				
			||||||
 | 
						var props []string
 | 
				
			||||||
 | 
						var jail *Jail
 | 
				
			||||||
 | 
						var err error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for i, a := range args {
 | 
				
			||||||
 | 
								// Last arg is the jail name
 | 
				
			||||||
 | 
								if i == len(args)-1 {
 | 
				
			||||||
 | 
									jail, err = getJailFromArray(a, gJails)
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("Error: %s\n", err.Error())
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									props = append(props, a)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(jail.Name) == 0 || len(args) == 0 {
 | 
				
			||||||
 | 
							// TODO : Show help
 | 
				
			||||||
 | 
							fmt.Printf("Error\n")
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if isStringInArray(props, "all") {
 | 
				
			||||||
 | 
							var result []string
 | 
				
			||||||
 | 
							result = getStructFieldNames(*jail, result, "")
 | 
				
			||||||
 | 
							props = result
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, p := range props {
 | 
				
			||||||
 | 
							v, err := getJailProperty(jail, p)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error: %s\n", err.Error())
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							} else {
 | 
				
			||||||
 | 
								fmt.Printf("%s = %s\n", p, v)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func getJailProperty(jail *Jail, propName string) (string, error) {
 | 
				
			||||||
 | 
						val, _, err := getStructFieldValue(jail, propName)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return "", err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						switch val.Kind() {
 | 
				
			||||||
 | 
						case reflect.String:
 | 
				
			||||||
 | 
							return val.String(), nil
 | 
				
			||||||
 | 
						case reflect.Int:
 | 
				
			||||||
 | 
							return strconv.FormatInt(val.Int(), 10), nil
 | 
				
			||||||
 | 
						case reflect.Bool:
 | 
				
			||||||
 | 
							return strconv.FormatBool(val.Bool()), nil
 | 
				
			||||||
 | 
						default:
 | 
				
			||||||
 | 
							return "", errors.New(fmt.Sprintf("Error: Unknown type for property %s: %s\n", propName, val.Kind()))
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return "", errors.New("Unknown error in getJailProperty")
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func SetJailProperties(args []string) {
 | 
				
			||||||
 | 
						type properties struct {
 | 
				
			||||||
 | 
							name  string
 | 
				
			||||||
 | 
							value string
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						var jail Jail
 | 
				
			||||||
 | 
						var props []properties
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for i, a := range args {
 | 
				
			||||||
 | 
								// This is the jail name
 | 
				
			||||||
 | 
								if i == len(args)-1 {
 | 
				
			||||||
 | 
									jail.Name = a
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									kv := strings.Split(a, "=")
 | 
				
			||||||
 | 
									if len(kv) != 2 {
 | 
				
			||||||
 | 
										// TODO : Show help
 | 
				
			||||||
 | 
										fmt.Printf("Error parsing args: %s\n", a)
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										p := properties{name: kv[0], value: kv[1]}
 | 
				
			||||||
 | 
										props = append(props, p)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(jail.Name) == 0 || len(args) == 0 {
 | 
				
			||||||
 | 
							// TODO : Show help
 | 
				
			||||||
 | 
							fmt.Printf("Error\n")
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Get jail by index to modify it
 | 
				
			||||||
 | 
						for i, _ := range gJails {
 | 
				
			||||||
 | 
							if gJails[i].Name == jail.Name {
 | 
				
			||||||
 | 
								for _, p := range props {
 | 
				
			||||||
 | 
									err := setStructFieldValue(&gJails[i], p.name, p.value)
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("Error: %s\n", err.Error())
 | 
				
			||||||
 | 
										return
 | 
				
			||||||
 | 
									} else {
 | 
				
			||||||
 | 
										fmt.Printf("%s: %s set to %s\n", gJails[i].Name, p.name, p.value)
 | 
				
			||||||
 | 
										gJails[i].ConfigUpdated = true
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										521
									
								
								cmd/root.go
									
									
									
									
									
								
							
							
						
						
									
										521
									
								
								cmd/root.go
									
									
									
									
									
								
							@ -1,77 +1,125 @@
 | 
				
			|||||||
package cmd
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"os"
 | 
					 | 
				
			||||||
	"fmt"
 | 
					 | 
				
			||||||
	"strconv"
 | 
					 | 
				
			||||||
	"strings"
 | 
					 | 
				
			||||||
	"encoding/json"
 | 
						"encoding/json"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"io/ioutil"
 | 
				
			||||||
 | 
						"os"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	"github.com/spf13/cobra"
 | 
						"github.com/spf13/cobra"
 | 
				
			||||||
	"github.com/spf13/viper"
 | 
						"github.com/spf13/viper"
 | 
				
			||||||
 | 
						// TODO : Use log
 | 
				
			||||||
 | 
						log "github.com/sirupsen/logrus"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
const (
 | 
					const (
 | 
				
			||||||
	gVersion = "0.022a"
 | 
						gVersion = "0.33c"
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// TODO : Get from $jail_zpool/defaults.json
 | 
				
			||||||
 | 
						MIN_DYN_DEVFS_RULESET = 1000
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
var (
 | 
					var (
 | 
				
			||||||
	gJails			[]Jail
 | 
						gJailHost   JailHost
 | 
				
			||||||
 | 
						gJails      []Jail
 | 
				
			||||||
 | 
						gDatastores []Datastore
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	gUseSudo		bool
 | 
						gUseSudo bool
 | 
				
			||||||
 | 
						gForce   bool
 | 
				
			||||||
 | 
						gDebug   bool
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	gConfigFile		string
 | 
						gConfigFile      string
 | 
				
			||||||
	gDisplayColumns	string
 | 
						gDisplayJColumns string
 | 
				
			||||||
	gFilterJails	string
 | 
						gDisplaySColumns string
 | 
				
			||||||
	gSortFields		string
 | 
						gDisplayDColumns string
 | 
				
			||||||
	gNoLineSep		bool
 | 
						gFilterJails     string
 | 
				
			||||||
 | 
						gFilterSnaps     string
 | 
				
			||||||
 | 
						gFilterDS        string
 | 
				
			||||||
 | 
						gSortJailFields  string
 | 
				
			||||||
 | 
						gSortSnapFields  string
 | 
				
			||||||
 | 
						gSortDSFields    string
 | 
				
			||||||
 | 
						gNoJailLineSep   bool
 | 
				
			||||||
 | 
						gNoSnapLineSep   bool
 | 
				
			||||||
 | 
						gNoDSLineSep     bool
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	gHostVersion	float64
 | 
						gHostVersion float64
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						gTimeZone     string
 | 
				
			||||||
 | 
						gSnapshotName string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						gMigrateDestDatastore string
 | 
				
			||||||
 | 
						gYesToAll             bool
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						gFetchRelease    string
 | 
				
			||||||
 | 
						gFetchIntoDS     string
 | 
				
			||||||
 | 
						gFetchFrom       string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	rootCmd = & cobra.Command{
 | 
						rootCmd = &cobra.Command{
 | 
				
			||||||
    	Use:    "gocage",
 | 
							Use:   "gocage",
 | 
				
			||||||
	    Short:  "GoCage is a FreeBSD Jail management tool",
 | 
							Short: "GoCage is a FreeBSD Jail management tool",
 | 
				
			||||||
	    Long:   `GoCage is a jail management tool. It support VNET, host-only, NAT networks. Provides snapshots and cloning.
 | 
							Long: `GoCage is a jail management tool. It support VNET, host-only, NAT networks. Provides snapshots and cloning.
 | 
				
			||||||
It support iocage jails and can coexist with iocage.`,
 | 
					It support iocage jails and can coexist with iocage.`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
	    Run: func(cmd *cobra.Command, args []string) {
 | 
								fv, _ := getFreeBSDVersion()
 | 
				
			||||||
	        fmt.Println("Here we are in the Run")
 | 
								fmt.Printf("GoCage v.%s on FreeBSD %d.%d-%s\n", gVersion, fv.major, fv.minor, fv.flavor)
 | 
				
			||||||
			cleanAfterRun()
 | 
								fmt.Printf("Use -h flag to display help\n")
 | 
				
			||||||
	    },
 | 
							},
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	versionCmd = &cobra.Command{
 | 
						versionCmd = &cobra.Command{
 | 
				
			||||||
	    Use:   "version",
 | 
							Use:   "version",
 | 
				
			||||||
	    Short: "Print the version number of GoCage",
 | 
							Short: "Print the version number of GoCage",
 | 
				
			||||||
	    Long:  `Let this show you how much fail I had to get this *cough* perfect`,
 | 
							Long:  `Let this show you how much fail I had to get this *cough* perfect`,
 | 
				
			||||||
	    Run: func(cmd *cobra.Command, args []string) {
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
	        fmt.Printf("GoCage v.%s on FreeBSD %.1f\n", gVersion, gHostVersion)
 | 
								fv, _ := getFreeBSDVersion()
 | 
				
			||||||
			cleanAfterRun()
 | 
								fmt.Printf("GoCage v.%s on FreeBSD %d.%d-%s\n", gVersion, fv.major, fv.minor, fv.flavor)
 | 
				
			||||||
	    },
 | 
							},
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	listCmd = &cobra.Command{
 | 
						listCmd = &cobra.Command{
 | 
				
			||||||
		Use:   "list",
 | 
							Use:   "list",
 | 
				
			||||||
		Short: "Print jails",
 | 
							Short: "Print jails",
 | 
				
			||||||
		Long:  `Display jails, their IP and OS.
 | 
							Long: `Display jails, their IP and OS.
 | 
				
			||||||
Jail list can be restricted by adding name on command line
 | 
					Jail list can be restricted by adding name on command line
 | 
				
			||||||
ex: gocage list srv-db srv-web`,
 | 
					ex: gocage list srv-db srv-web`,
 | 
				
			||||||
		Run: func(cmd *cobra.Command, args []string) {
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
			ListJails(args, true)
 | 
								ListJails(args, true)
 | 
				
			||||||
			cleanAfterRun()
 | 
					 | 
				
			||||||
		},
 | 
							},
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						listPropsCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "properties",
 | 
				
			||||||
 | 
							Short: "Print jails properties",
 | 
				
			||||||
 | 
							Long:  "Display jails properties. You can use properties to filter, get or set them.",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								ListJailsProps(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
					/*	destroyCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "destroy",
 | 
				
			||||||
 | 
							Short: "destroy jails",
 | 
				
			||||||
 | 
							Long: `Destroy jail filesystem, snapshots and configuration file.`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								DestroyJails(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					*/
 | 
				
			||||||
	stopCmd = &cobra.Command{
 | 
						stopCmd = &cobra.Command{
 | 
				
			||||||
		Use:   "stop",
 | 
							Use:   "stop",
 | 
				
			||||||
		Short: "stop jail",
 | 
							Short: "stop jail",
 | 
				
			||||||
		Long:  `shutdown jail`,
 | 
							Long:  "shutdown jail",
 | 
				
			||||||
		Run: func(cmd *cobra.Command, args []string) {
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
			// Get the inventory
 | 
								// Load inventory
 | 
				
			||||||
			ListJails(args, false)
 | 
								ListJails(args, false)
 | 
				
			||||||
			StopJail(args)
 | 
								if len(args) == 0 {
 | 
				
			||||||
			cleanAfterRun()
 | 
									StopAllRunningJails()
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									StopJail(args)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
		},
 | 
							},
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@ -79,43 +127,283 @@ ex: gocage list srv-db srv-web`,
 | 
				
			|||||||
		Use:   "start",
 | 
							Use:   "start",
 | 
				
			||||||
		Short: "start jail",
 | 
							Short: "start jail",
 | 
				
			||||||
		Run: func(cmd *cobra.Command, args []string) {
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
			// Get the inventory
 | 
								// Load inventory
 | 
				
			||||||
			ListJails(args, false)
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								if len(args) == 0 {
 | 
				
			||||||
 | 
									StartJailsAtBoot()
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									StartJail(args)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								WriteConfigToDisk("", false, false)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						restartCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "restart",
 | 
				
			||||||
 | 
							Short: "restart jail",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								StopJail(args)
 | 
				
			||||||
			StartJail(args)
 | 
								StartJail(args)
 | 
				
			||||||
			cleanAfterRun()
 | 
								WriteConfigToDisk("", false, false)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						shellCmd = &cobra.Command {
 | 
				
			||||||
 | 
							Use:   "console",
 | 
				
			||||||
 | 
							Short: "Execute shell on jail",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								ShellJail(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						setCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "set",
 | 
				
			||||||
 | 
							Short: "Set a jail property",
 | 
				
			||||||
 | 
							Long: `Set jail property value. Specify property=value, end command with jail name.
 | 
				
			||||||
 | 
					Multiples properties can be specified, separated with space (Ex: gocage set allow_mlock=1 boot=1 myjail)`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								SetJailProperties(args)
 | 
				
			||||||
 | 
								WriteConfigToDisk("", true, false)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						getCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "get",
 | 
				
			||||||
 | 
							Short: "Get a jail property",
 | 
				
			||||||
 | 
							Long: `Get jail property value. Specify property, end command with jail name.
 | 
				
			||||||
 | 
					Multiples properties can be specified, separated with space (Ex: gocage get allow_mlock boot myjail)
 | 
				
			||||||
 | 
					For all properties specify "all" (Ex: gocage get all myjail)`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								GetJailProperties(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						snapshotCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "snapshot",
 | 
				
			||||||
 | 
							Short: "snapshot jail",
 | 
				
			||||||
 | 
							Long:  "Commands to manage jail snapshots. If no arguments given, ",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						snapshotListCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "list",
 | 
				
			||||||
 | 
							Short: "list snapshots",
 | 
				
			||||||
 | 
							Long: `List snapshots of a jail by specifying its name. 
 | 
				
			||||||
 | 
					List all snapshots if no jail name specified.
 | 
				
			||||||
 | 
					You can specify multiple jails.`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								ListJailsSnapshots(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						snapshotCreateCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "create",
 | 
				
			||||||
 | 
							Short: "create snapshots",
 | 
				
			||||||
 | 
							Long:  `Create snapshot of a jail by specifying snapshot name and jail name.`,
 | 
				
			||||||
 | 
							//		You can specify multiple jails.`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								CreateJailSnapshot(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						snapshotRollbackCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "rollback",
 | 
				
			||||||
 | 
							Short: "Rollback snapshots",
 | 
				
			||||||
 | 
							Long:  `Rollback jail to specifyed snapshot.`,
 | 
				
			||||||
 | 
							//		You can specify multiple jails.`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								RollbackJailSnapshot(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						snapshotDeleteCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "destroy",
 | 
				
			||||||
 | 
							Short: "destroy snapshots",
 | 
				
			||||||
 | 
							Long:  `Destroy snapshot of a jail by specifying snapshot name and jail name.
 | 
				
			||||||
 | 
									You can specify multiple snapshots separated by comma.`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								DeleteJailSnapshot(args)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						migrateCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "migrate",
 | 
				
			||||||
 | 
							Short: "Migrate jail to another datastore",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								MigrateJail(args)
 | 
				
			||||||
 | 
								WriteConfigToDisk("", false, false)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						migrateCleanCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "clean",
 | 
				
			||||||
 | 
							Short: "Clean previous aborted/in error jail migration",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								// Load inventory
 | 
				
			||||||
 | 
								ListJails(args, false)
 | 
				
			||||||
 | 
								err := CleanMigrateMess(args)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									fmt.Printf("%v", err)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						datastoreCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "datastore",
 | 
				
			||||||
 | 
							Short: "list datastores",
 | 
				
			||||||
 | 
							Long:  "Commands to manage datastores. If no arguments given, list them.",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								ListDatastores(args, true)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						datastoreListCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "list",
 | 
				
			||||||
 | 
							Short: "list datastores",
 | 
				
			||||||
 | 
							Long: `List datastore by specifying its name. 
 | 
				
			||||||
 | 
					List all datastores if no name specified.
 | 
				
			||||||
 | 
					You can specify multiple datastores.`,
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								ListDatastores(args, true)
 | 
				
			||||||
 | 
							},
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						fetchCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "fetch",
 | 
				
			||||||
 | 
							Short: "Fetch FreeBSD release to local datastore",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								err := fetchRelease(gFetchRelease, "http", gJailHost.arch, gFetchIntoDS, gFetchFrom)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									fmt.Printf("%v\n", err)
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									extractRelease(gFetchRelease, gFetchIntoDS)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
						 },
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						UpdateCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "update",
 | 
				
			||||||
 | 
						 Short: "Update FreeBSD release",
 | 
				
			||||||
 | 
						 Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
							 ListJails(args, false)
 | 
				
			||||||
 | 
							 UpdateJail(args)
 | 
				
			||||||
 | 
						 },
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						testCmd = &cobra.Command{
 | 
				
			||||||
 | 
							Use:   "test",
 | 
				
			||||||
 | 
							Short: "temporary command to test some code snippet",
 | 
				
			||||||
 | 
							Run: func(cmd *cobra.Command, args []string) {
 | 
				
			||||||
 | 
								fmt.Printf("Nope\n")
 | 
				
			||||||
		},
 | 
							},
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// TODO : Init log level and log output
 | 
				
			||||||
 | 
					 | 
				
			||||||
func init() {
 | 
					func init() {
 | 
				
			||||||
    cobra.OnInitialize(initConfig)
 | 
						var err error
 | 
				
			||||||
	
 | 
						
 | 
				
			||||||
// Global switches
 | 
						cobra.OnInitialize(initConfig)
 | 
				
			||||||
    rootCmd.PersistentFlags().StringVarP(&gConfigFile, "config", "c", "/usr/local/etc/gocage.conf.yml", "GoCage configuration file")
 | 
					 | 
				
			||||||
    rootCmd.PersistentFlags().BoolVarP(&gUseSudo, "sudo", "u", false, "Use sudo to run commands")
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Command dependant switches
 | 
						// Global switches
 | 
				
			||||||
    listCmd.PersistentFlags().StringVarP(&gDisplayColumns, "outcol", "o", "JID,Name,Config.Release,Config.Ip4_addr,Running", "Show these columns in output")
 | 
						rootCmd.PersistentFlags().StringVarP(&gConfigFile, "config", "c", "/usr/local/etc/gocage.conf.yml", "GoCage configuration file")
 | 
				
			||||||
    listCmd.PersistentFlags().BoolVarP(&gNoLineSep, "nolinesep", "l", false, "Do not display line separator between jails")
 | 
						rootCmd.PersistentFlags().BoolVarP(&gUseSudo, "sudo", "u", false, "Use sudo to run commands")
 | 
				
			||||||
    listCmd.PersistentFlags().StringVarP(&gFilterJails, "filter", "f", "none", "Only display jails with these values. Ex: \"gocage list -f Config.Boot=1\" will only list started on boot jails")
 | 
						rootCmd.PersistentFlags().StringVarP(&gTimeZone, "timezone", "t", "", "Specify timezone. Will get from /var/db/zoneinfo if not set.")
 | 
				
			||||||
    listCmd.PersistentFlags().StringVarP(&gSortFields, "sort", "s", "none", "Display jails sorted by field values. Ex: \"gocage list -s +Name,-Config.Priority\" will sort jails by their decreasing name, then increasing start priority. 3 critera max supported.")
 | 
						rootCmd.PersistentFlags().BoolVar(&gDebug, "debug", false, "Debug mode")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Now declare commands
 | 
						// Command dependant switches
 | 
				
			||||||
    rootCmd.AddCommand(versionCmd)
 | 
					 | 
				
			||||||
    rootCmd.AddCommand(listCmd)
 | 
					 | 
				
			||||||
    rootCmd.AddCommand(stopCmd)
 | 
					 | 
				
			||||||
    rootCmd.AddCommand(startCmd)
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Get FreeBSD version
 | 
						// We reuse these flags in "gocage snapshot list myjail" and 'gocage datastore list" commands
 | 
				
			||||||
	out, err := executeCommand("freebsd-version")
 | 
						listCmd.Flags().StringVarP(&gDisplayJColumns, "outcol", "o", "JID,Name,Config.Release,Config.Ip4_addr,Running", "Show these columns in output")
 | 
				
			||||||
 | 
						listCmd.Flags().BoolVarP(&gNoJailLineSep, "nolinesep", "l", false, "Do not display line separator between jails")
 | 
				
			||||||
 | 
						listCmd.Flags().StringVarP(&gFilterJails, "filter", "f", "none", "Only display jails with these values. Ex: \"gocage list -f Config.Boot=1\" will only list started on boot jails")
 | 
				
			||||||
 | 
						listCmd.Flags().StringVarP(&gSortJailFields, "sort", "s", "none", "Display jails sorted by field values. Ex: \"gocage list -s +Name,-Config.Priority\" will sort jails by their decreasing name, then increasing start priority. 3 critera max supported.")
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
					//	destroyCmd.Flags().BoolVarP(&gForce, "force", "f", false, "Force stop jail if running")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						snapshotListCmd.Flags().StringVarP(&gDisplaySColumns, "outcol", "o", "Jailname,Name,Creation,Referenced,Used", "Show these columns in output")
 | 
				
			||||||
 | 
						snapshotListCmd.Flags().BoolVarP(&gNoSnapLineSep, "nolinesep", "l", false, "Do not display line separator between snapshots")
 | 
				
			||||||
 | 
						snapshotListCmd.Flags().StringVarP(&gFilterSnaps, "filter", "f", "none", "Only display snapshots with these values. Ex: \"gocage snapshot list -f Config.Boot=1\" will only list started on boot jails")
 | 
				
			||||||
 | 
						snapshotListCmd.Flags().StringVarP(&gSortSnapFields, "sort", "s", "none", "Display snapshots sorted by field values. Ex: \"gocage snapshot list -s +Jailname,-Used\" will sort snapshots by jail decreasing name, then increasing used space. 3 critera max supported.")
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						datastoreListCmd.Flags().StringVarP(&gDisplayDColumns, "outcol", "o", "Name,Mountpoint,ZFSDataset,Available,Used,Referenced", "Show these columns in output")
 | 
				
			||||||
 | 
						datastoreListCmd.Flags().BoolVarP(&gNoDSLineSep, "nolinesep", "l", false, "Do not display line separator between datastores")
 | 
				
			||||||
 | 
						datastoreListCmd.Flags().StringVarP(&gFilterDS, "filter", "f", "none", "Only display datastores with these values. Ex: \"gocage datastore list -f Config.Boot=1\" will only list started on boot jails")
 | 
				
			||||||
 | 
						datastoreListCmd.Flags().StringVarP(&gSortDSFields, "sort", "s", "none", "Display datastores sorted by field values. Ex: \"gocage datastore list -s +Jailname,-Used\" will sort snapshots by jail decreasing name, then increasing used space. 3 critera max supported.")
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// This is local flag : Only available to gocage snapshot create command
 | 
				
			||||||
 | 
						snapshotCreateCmd.Flags().StringVarP(&gSnapshotName, "snapname", "n", "", "Name of the snapshot to create")
 | 
				
			||||||
 | 
						snapshotCreateCmd.MarkFlagRequired("snapname")
 | 
				
			||||||
 | 
						snapshotDeleteCmd.Flags().StringVarP(&gSnapshotName, "snapname", "n", "", "Name of the snapshot to destroy")
 | 
				
			||||||
 | 
						snapshotDeleteCmd.MarkFlagRequired("snapname")
 | 
				
			||||||
 | 
						snapshotRollbackCmd.Flags().StringVarP(&gSnapshotName, "snapname", "n", "", "Name of the snapshot to rollback to")
 | 
				
			||||||
 | 
						snapshotRollbackCmd.MarkFlagRequired("snapname")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						migrateCmd.Flags().StringVarP(&gMigrateDestDatastore, "datastore", "d", "", "Path of destination datastore for jail (Ex: \"/iocage\")")
 | 
				
			||||||
 | 
						migrateCmd.Flags().BoolVarP(&gYesToAll, "yes", "y", false, "Answer yes to all questions")
 | 
				
			||||||
 | 
						migrateCmd.MarkFlagRequired("datastore")
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						fetchCmd.Flags().StringVarP(&gFetchRelease, "release", "r", "", "Release to fetch (e.g.: \"13.1\"")
 | 
				
			||||||
 | 
						fetchCmd.Flags().StringVarP(&gFetchIntoDS, "datastore", "o", "", "Datastore release will be saved to")
 | 
				
			||||||
 | 
						fetchCmd.Flags().StringVarP(&gFetchFrom, "from", "d", "", "Repository to download from. Should contain XY.Z-RELEASE. File protocol supported")
 | 
				
			||||||
 | 
						fetchCmd.MarkFlagRequired("release")
 | 
				
			||||||
 | 
						fetchCmd.MarkFlagRequired("datastore")
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Now declare commands
 | 
				
			||||||
 | 
						rootCmd.AddCommand(versionCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(listCmd)
 | 
				
			||||||
 | 
						listCmd.AddCommand(listPropsCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(stopCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(startCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(restartCmd)
 | 
				
			||||||
 | 
					//	rootCmd.AddCommand(destroyCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(shellCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(getCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(setCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(snapshotCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(migrateCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(datastoreCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(fetchCmd)
 | 
				
			||||||
 | 
						rootCmd.AddCommand(UpdateCmd)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						rootCmd.AddCommand(testCmd)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						snapshotCmd.AddCommand(snapshotListCmd)
 | 
				
			||||||
 | 
						snapshotCmd.AddCommand(snapshotCreateCmd)
 | 
				
			||||||
 | 
						snapshotCmd.AddCommand(snapshotDeleteCmd)
 | 
				
			||||||
 | 
						snapshotCmd.AddCommand(snapshotRollbackCmd)
 | 
				
			||||||
 | 
						migrateCmd.AddCommand(migrateCleanCmd)
 | 
				
			||||||
 | 
						datastoreCmd.AddCommand(datastoreListCmd)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Get FreeBSD version, hostname, hostid
 | 
				
			||||||
 | 
						gJailHost, err = NewJailHost()
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		fmt.Printf("Error running \"freebsd-version\": %s", err.Error())
 | 
							fmt.Printf("Error initializing JailHost properties: %v\n", err)
 | 
				
			||||||
		os.Exit(1)
 | 
							os.Exit(1)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	gHostVersion, _ = strconv.ParseFloat(strings.Split(out, "-")[0], 32)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func initConfig() {
 | 
					func initConfig() {
 | 
				
			||||||
@ -124,8 +412,6 @@ func initConfig() {
 | 
				
			|||||||
		os.Exit(1)
 | 
							os.Exit(1)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
//	fmt.Printf("We are in initConfig(), with config file %s\n", gConfigFile)
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
	viper.SetConfigFile(gConfigFile)
 | 
						viper.SetConfigFile(gConfigFile)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	if err := viper.ReadInConfig(); err != nil {
 | 
						if err := viper.ReadInConfig(); err != nil {
 | 
				
			||||||
@ -133,52 +419,121 @@ func initConfig() {
 | 
				
			|||||||
		os.Exit(1)
 | 
							os.Exit(1)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	
 | 
						
 | 
				
			||||||
//	fmt.Println("Using config file:", viper.ConfigFileUsed())
 | 
						// Load default configs from datastores
 | 
				
			||||||
//	fmt.Printf("datastore in config : %s\n", viper.GetStringSlice("datastore"))
 | 
						err := ListDatastores(viper.GetStringSlice("datastore"), false)
 | 
				
			||||||
//	fmt.Printf("datastore.0 in config : %s\n", viper.GetStringSlice("datastore.0"))
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("ERROR: error checking datastores: %v\n", err)
 | 
				
			||||||
 | 
							os.Exit(1)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						//	fmt.Println("Using config file:", viper.ConfigFileUsed())
 | 
				
			||||||
 | 
						//	fmt.Printf("datastore in config : %s\n", viper.GetStringSlice("datastore"))
 | 
				
			||||||
 | 
						//	fmt.Printf("datastore.0 in config : %s\n", viper.GetStringSlice("datastore.0"))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Command line flags have priority on config file
 | 
						// Command line flags have priority on config file
 | 
				
			||||||
	if rootCmd.Flags().Lookup("sudo") != nil && false == rootCmd.Flags().Lookup("sudo").Changed {
 | 
						if rootCmd.Flags().Lookup("sudo") != nil && false == rootCmd.Flags().Lookup("sudo").Changed {
 | 
				
			||||||
		gUseSudo = viper.GetBool("sudo")
 | 
							gUseSudo = viper.GetBool("sudo")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						if rootCmd.Flags().Lookup("timezone") != nil && false == rootCmd.Flags().Lookup("timezone").Changed {
 | 
				
			||||||
 | 
							gTimeZone = viper.GetString("timezone")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						// If neither on cmdline nor config file, get from /var/db/zoneinfo
 | 
				
			||||||
 | 
						if len(gTimeZone) == 0 {
 | 
				
			||||||
 | 
							tz, err := ioutil.ReadFile("/var/db/zoneinfo")
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Println("Error reading /var/db/zoneinfo: %s\n", err.Error())
 | 
				
			||||||
 | 
								os.Exit(1)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							gTimeZone = strings.Trim(string(tz), "\n")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
	if listCmd.Flags().Lookup("outcol") != nil && false == listCmd.Flags().Lookup("outcol").Changed {
 | 
						if listCmd.Flags().Lookup("outcol") != nil && false == listCmd.Flags().Lookup("outcol").Changed {
 | 
				
			||||||
		gDisplayColumns = viper.GetString("outcol")
 | 
							gDisplayJColumns = viper.GetString("outcol")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if listCmd.Flags().Lookup("nolinesep") != nil && false == listCmd.Flags().Lookup("nolinesep").Changed {
 | 
						if listCmd.Flags().Lookup("nolinesep") != nil && false == listCmd.Flags().Lookup("nolinesep").Changed {
 | 
				
			||||||
		gNoLineSep = viper.GetBool("nolinesep")
 | 
							gNoJailLineSep = viper.GetBool("nolinesep")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if listCmd.Flags().Lookup("filter") != nil && false == listCmd.Flags().Lookup("filter").Changed {
 | 
						if listCmd.Flags().Lookup("filter") != nil && false == listCmd.Flags().Lookup("filter").Changed {
 | 
				
			||||||
		gFilterJails = viper.GetString("filter")
 | 
							gFilterJails = viper.GetString("filter")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if listCmd.Flags().Lookup("sort") != nil && false == listCmd.Flags().Lookup("sort").Changed {
 | 
						if listCmd.Flags().Lookup("sort") != nil && false == listCmd.Flags().Lookup("sort").Changed {
 | 
				
			||||||
		gSortFields = viper.GetString("sort")
 | 
							gSortJailFields = viper.GetString("sort")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	if len(strings.Split(gSortFields, ",")) > 3 {
 | 
						if len(strings.Split(gSortJailFields, ",")) > 3 {
 | 
				
			||||||
		fmt.Printf("More than 3 sort criteria, this is not supported!\n")
 | 
							fmt.Printf("More than 3 sort criteria is not supported!\n")
 | 
				
			||||||
		os.Exit(1)
 | 
							os.Exit(1)
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						if gDebug {
 | 
				
			||||||
 | 
							log.SetLevel(log.DebugLevel)
 | 
				
			||||||
 | 
							log.Debugf("Debug mode enabled\n")
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// Called after execution
 | 
					/********************************************************************************
 | 
				
			||||||
func cleanAfterRun() {
 | 
					 * Write jail(s) config which been updated to disk.
 | 
				
			||||||
 | 
					 * If name is specified, work on the jail. If name is empty string, work on all.
 | 
				
			||||||
 | 
					 * If changeauto not set, values which are in "auto" mode on disk
 | 
				
			||||||
 | 
					 *  won't be overwritten (p.ex defaultrouter wont be overwritten with current
 | 
				
			||||||
 | 
					 *  default route, so if route change on jailhost this will reflect on jail next
 | 
				
			||||||
 | 
					 *  start)
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func WriteConfigToDisk(jailName string, changeauto bool, forceWrite bool) {
 | 
				
			||||||
	for _, j := range gJails {
 | 
						for _, j := range gJails {
 | 
				
			||||||
		if j.ConfigUpdated {
 | 
							if len(jailName) > 0 && j.Name == jailName || len(jailName) == 0 {
 | 
				
			||||||
			// TODO : Marshall to disk
 | 
								if j.ConfigUpdated || forceWrite {
 | 
				
			||||||
			fmt.Printf("Config for jail %s will be updated\n", j.Name)
 | 
									log.Debug("%s config has changed, write changes to disk\n", j.Name)
 | 
				
			||||||
			marshaled, err := json.MarshalIndent(j.Config, "", "    ")
 | 
					
 | 
				
			||||||
			if err != nil {
 | 
									// we will manipulate properties so get a copy
 | 
				
			||||||
				fmt.Printf("ERROR marshaling config: %s\n", err.Error())
 | 
									jc := j.Config
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									if changeauto == false {
 | 
				
			||||||
 | 
										// Overwrite "auto" properties
 | 
				
			||||||
 | 
										ondiskjc, err := getJailConfig(j.ConfigPath)
 | 
				
			||||||
 | 
										if err != nil {
 | 
				
			||||||
 | 
											panic(err)
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
										// TODO : List all fields, then call getStructFieldValue to compare value with "auto"
 | 
				
			||||||
 | 
										// If "auto" then keep it that way before writing ondiskjc to disk
 | 
				
			||||||
 | 
										var properties []string
 | 
				
			||||||
 | 
										properties = getStructFieldNames(ondiskjc, properties, "")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
										for _, p := range properties {
 | 
				
			||||||
 | 
											v, _, err := getStructFieldValue(ondiskjc, p)
 | 
				
			||||||
 | 
											if err != nil {
 | 
				
			||||||
 | 
												panic(err)
 | 
				
			||||||
 | 
											}
 | 
				
			||||||
 | 
											if v.String() == "auto" {
 | 
				
			||||||
 | 
												err = setStructFieldValue(&jc, p, "auto")
 | 
				
			||||||
 | 
												if err != nil {
 | 
				
			||||||
 | 
													fmt.Printf("ERROR sanitizing config: %s\n", err.Error())
 | 
				
			||||||
 | 
													os.Exit(1)
 | 
				
			||||||
 | 
												}
 | 
				
			||||||
 | 
											}
 | 
				
			||||||
 | 
										}
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
									marshaled, err := json.MarshalIndent(jc, "", "    ")
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("ERROR marshaling config: %s\n", err.Error())
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									
 | 
				
			||||||
 | 
									//fmt.Printf("DEBUG: Will write config to disk, with content:\n")
 | 
				
			||||||
 | 
									//fmt.Printf(string(marshaled))
 | 
				
			||||||
 | 
									
 | 
				
			||||||
 | 
									if os.WriteFile(j.ConfigPath, []byte(marshaled), 0644); err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("Error writing config file %s: %v\n", j.ConfigPath, err)
 | 
				
			||||||
 | 
										os.Exit(1)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
			fmt.Printf(string(marshaled))
 | 
					 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func Execute() {
 | 
					func Execute() {
 | 
				
			||||||
    if err := rootCmd.Execute(); err != nil {
 | 
						if err := rootCmd.Execute(); err != nil {
 | 
				
			||||||
        fmt.Fprintln(os.Stderr, err)
 | 
							fmt.Fprintln(os.Stderr, err)
 | 
				
			||||||
        os.Exit(1)
 | 
							os.Exit(1)
 | 
				
			||||||
    }
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										330
									
								
								cmd/snapshots.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										330
									
								
								cmd/snapshots.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,330 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"bufio"
 | 
				
			||||||
 | 
						"errors"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						"os"
 | 
				
			||||||
 | 
						"regexp"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
 | 
						"strings"
 | 
				
			||||||
 | 
						"reflect"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * List all snapshots jails have
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func ListJailsSnapshots(args []string) {
 | 
				
			||||||
 | 
						var jailNames []string
 | 
				
			||||||
 | 
						var snapshots []Snapshot
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						/***************************************************************
 | 
				
			||||||
 | 
						/   Filter snapshots by jailname
 | 
				
			||||||
 | 
						/**************************************************************/
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for _, a := range args {
 | 
				
			||||||
 | 
								/*if countOfJailsWithThisName(a) > 1 {
 | 
				
			||||||
 | 
									fmt.Printf("Nope")
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}*/
 | 
				
			||||||
 | 
								jailNames = append(jailNames, a)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(jailNames) == 0 || len(args) == 0 {
 | 
				
			||||||
 | 
							for _, j := range gJails {
 | 
				
			||||||
 | 
								snapshots = append(snapshots, listJailSnapshots(j)...)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						} else {
 | 
				
			||||||
 | 
							for _, cj := range gJails {
 | 
				
			||||||
 | 
								for _, jn := range jailNames {
 | 
				
			||||||
 | 
									if strings.EqualFold(cj.Name, jn) {
 | 
				
			||||||
 | 
										snapshots = append(snapshots, listJailSnapshots(cj)...)
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						fields := strings.Split(gDisplaySColumns, ",")
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						/***************************************************************
 | 
				
			||||||
 | 
						/   Sort snapshots
 | 
				
			||||||
 | 
						/     We support 3 sort criteria max
 | 
				
			||||||
 | 
						/**************************************************************/
 | 
				
			||||||
 | 
						if len(gSortSnapFields) > 0 && gSortSnapFields != "none" {		
 | 
				
			||||||
 | 
							ss := initSnapshotSortStruct()
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// The way we manage criteria quantity is not very elegant...
 | 
				
			||||||
 | 
							var fct1, fct2, fct3 *reflect.Value
 | 
				
			||||||
 | 
							for i, c := range strings.Split(gSortSnapFields, ",") {
 | 
				
			||||||
 | 
								var fctName string
 | 
				
			||||||
 | 
								if strings.HasPrefix(c, "-") {
 | 
				
			||||||
 | 
									fctName = fmt.Sprintf("%sDec", strings.Replace(c, "-", "", 1))
 | 
				
			||||||
 | 
								} else { // Par defaut (pas de prefix +/-) on considere un tri incremental
 | 
				
			||||||
 | 
									fctName = fmt.Sprintf("%sInc", strings.Replace(c, "+", "", 1))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								// Get function by its name
 | 
				
			||||||
 | 
								fct, _, err := getStructFieldValue(ss, fctName)
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									fieldName := strings.Replace(strings.Replace(c, "-", "", 1), "+", "", 1)
 | 
				
			||||||
 | 
									fmt.Printf("ERROR getting SnapshotSort struct field %s. Please check the field name: %s\n", fctName, fieldName)
 | 
				
			||||||
 | 
									return
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								switch i + 1 {
 | 
				
			||||||
 | 
								case 1:
 | 
				
			||||||
 | 
									fct1 = fct
 | 
				
			||||||
 | 
								case 2:
 | 
				
			||||||
 | 
									fct2 = fct
 | 
				
			||||||
 | 
								case 3:
 | 
				
			||||||
 | 
									fct3 = fct
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							switch len(strings.Split(gSortSnapFields, ",")) {
 | 
				
			||||||
 | 
							case 1:
 | 
				
			||||||
 | 
								SnapshotsOrderedBy(fct1.Interface().(snapshotLessFunc)).Sort(snapshots)
 | 
				
			||||||
 | 
							case 2:
 | 
				
			||||||
 | 
								SnapshotsOrderedBy(fct1.Interface().(snapshotLessFunc), fct2.Interface().(snapshotLessFunc)).Sort(snapshots)
 | 
				
			||||||
 | 
							case 3:
 | 
				
			||||||
 | 
								SnapshotsOrderedBy(fct1.Interface().(snapshotLessFunc), fct2.Interface().(snapshotLessFunc), fct3.Interface().(snapshotLessFunc)).Sort(snapshots)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						displaySnapshotsFields(snapshots, fields)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * List all snapshots a jail have
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func listJailSnapshots(jail Jail) []Snapshot {
 | 
				
			||||||
 | 
						var snapshots []Snapshot
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// 1. List all datasets
 | 
				
			||||||
 | 
						// TODO : Include mounted filesystems?
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						curDS, err := getDatastoreFromArray(jail.Datastore, gDatastores)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error getting datastore \"%s\": %v\n", jail.Datastore, err)
 | 
				
			||||||
 | 
							return snapshots
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						rootDataset := fmt.Sprintf("%s/%s/%s", curDS.ZFSDataset, "jails", jail.Name)
 | 
				
			||||||
 | 
						cmd := fmt.Sprintf("zfs list -p -r -H -o name,mountpoint,used,referenced,creation -t snapshot %s", rootDataset)
 | 
				
			||||||
 | 
						out, err := executeCommand(cmd)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error: %s\n", err.Error())
 | 
				
			||||||
 | 
							return snapshots
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						for _, line := range strings.Split(out, "\n") {
 | 
				
			||||||
 | 
							if len(line) > 0 {
 | 
				
			||||||
 | 
								ls := strings.Split(line, "\t")
 | 
				
			||||||
 | 
								// Parse creation date so we can use it to sort snapshots
 | 
				
			||||||
 | 
								//creationts, err := time.ParseInLocation(dateLayout, ls[4], loc)
 | 
				
			||||||
 | 
								creationts, err := strconv.ParseInt(ls[4], 10, 64)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									fmt.Println("Error while parsing date %s:", ls[4], err)
 | 
				
			||||||
 | 
									return snapshots
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
								// Get subdir to append to snapshot name
 | 
				
			||||||
 | 
								subdir := strings.Replace(strings.Split(ls[0], "@")[0], rootDataset, "", 1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								u, _ := strconv.ParseUint(ls[2], 10, 64)
 | 
				
			||||||
 | 
								r, _ := strconv.ParseUint(ls[3], 10, 64)
 | 
				
			||||||
 | 
								snapshots = append(snapshots, Snapshot{Datastore: curDS.Name,
 | 
				
			||||||
 | 
									Name:       fmt.Sprintf("%s%s", strings.Split(ls[0], "@")[1], subdir),
 | 
				
			||||||
 | 
									Jailname:   jail.Name,
 | 
				
			||||||
 | 
									Mountpoint: ls[1],
 | 
				
			||||||
 | 
									Used:       u,
 | 
				
			||||||
 | 
									Referenced: r,
 | 
				
			||||||
 | 
									Creation:   time.Unix(creationts, 0)})
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Sort snapshots by creation date
 | 
				
			||||||
 | 
						ss := initSnapshotSortStruct()
 | 
				
			||||||
 | 
						SnapshotsOrderedBy(ss.CreationInc).Sort(snapshots)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return snapshots
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * Create snapshot for jail(s)
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func CreateJailSnapshot(args []string) {
 | 
				
			||||||
 | 
						var jailNames []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for _, a := range args {
 | 
				
			||||||
 | 
								jailNames = append(jailNames, a)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, cj := range gJails {
 | 
				
			||||||
 | 
							for _, jn := range jailNames {
 | 
				
			||||||
 | 
								if strings.EqualFold(cj.Name, jn) {
 | 
				
			||||||
 | 
									createJailSnapshot(cj)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * Create snapshot for a jail
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func createJailSnapshot(jail Jail) error {
 | 
				
			||||||
 | 
						curDS, _ := getDatastoreFromArray(jail.Datastore, gDatastores)
 | 
				
			||||||
 | 
						rootDataset := fmt.Sprintf("%s/%s/%s", curDS.ZFSDataset, "jails", jail.Name)
 | 
				
			||||||
 | 
						cmd := fmt.Sprintf("zfs snapshot -r %s@%s", rootDataset, gSnapshotName)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						_, err := executeCommand(cmd)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error creating snapshot %s@%s: %s\n", rootDataset, gSnapshotName, err.Error())
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						fmt.Printf("Snapshot %s@%s created\n", rootDataset, gSnapshotName)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * Delete snapshot for jail(s)
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func DeleteJailSnapshot(args []string) {
 | 
				
			||||||
 | 
						var jailNames []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for _, a := range args {
 | 
				
			||||||
 | 
								jailNames = append(jailNames, a)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, cj := range gJails {
 | 
				
			||||||
 | 
							for _, jn := range jailNames {
 | 
				
			||||||
 | 
								if strings.EqualFold(cj.Name, jn) {
 | 
				
			||||||
 | 
									deleteJailSnapshot(cj)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * Delete snapshot for a jail
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func deleteJailSnapshot(jail Jail) error {
 | 
				
			||||||
 | 
						var snaptodel []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// Get all recursive snapshots
 | 
				
			||||||
 | 
						curDS, _ := getDatastoreFromArray(jail.Datastore, gDatastores)
 | 
				
			||||||
 | 
						rootDataset := fmt.Sprintf("%s/%s/%s", curDS.ZFSDataset, "jails", jail.Name)
 | 
				
			||||||
 | 
						cmd := fmt.Sprintf("zfs list -r -H -o name -t snapshot %s", rootDataset)
 | 
				
			||||||
 | 
						out, err := executeCommand(cmd)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error: listing snapshots: %s\n", err.Error())
 | 
				
			||||||
 | 
							return nil
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, line := range strings.Split(out, "\n") {
 | 
				
			||||||
 | 
							if len(line) > 0 {
 | 
				
			||||||
 | 
								ls := strings.Split(line, "@")
 | 
				
			||||||
 | 
								for _, sn := range strings.Split(gSnapshotName, ",") {
 | 
				
			||||||
 | 
									matched, _ := regexp.Match(fmt.Sprintf("^%s(\\/.*)?$", sn), []byte(ls[1]))
 | 
				
			||||||
 | 
									if matched {
 | 
				
			||||||
 | 
										snaptodel = append(snaptodel, strings.Join(ls, "@"))
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, s := range snaptodel {
 | 
				
			||||||
 | 
							cmd := fmt.Sprintf("zfs destroy %s", s)
 | 
				
			||||||
 | 
							_, err := executeCommand(cmd)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error deleting snapshot %s: %s\n", s, err.Error())
 | 
				
			||||||
 | 
								return nil
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							fmt.Printf("Snapshot %s deleted\n", s)
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func RollbackJailSnapshot(args []string) error {
 | 
				
			||||||
 | 
						var jailNames []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if len(args) > 0 {
 | 
				
			||||||
 | 
							for _, a := range args {
 | 
				
			||||||
 | 
								jailNames = append(jailNames, a)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, cj := range gJails {
 | 
				
			||||||
 | 
							for _, jn := range jailNames {
 | 
				
			||||||
 | 
								if strings.EqualFold(cj.Name, jn) {
 | 
				
			||||||
 | 
									rollbackJailSnapshot(cj)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					/********************************************************************************
 | 
				
			||||||
 | 
					 * rollback jail to snapshot gSnapshotName, destroy this snapshots and more
 | 
				
			||||||
 | 
					 *   recents snapshots and bookmarks
 | 
				
			||||||
 | 
					 *******************************************************************************/
 | 
				
			||||||
 | 
					func rollbackJailSnapshot(jail Jail) error {
 | 
				
			||||||
 | 
						var snaptorb []string
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						if jail.Running {
 | 
				
			||||||
 | 
							fmt.Printf("Jail should be stoped to rollback, should we stop and rollback? (y/n)\n")
 | 
				
			||||||
 | 
							scanr := bufio.NewScanner(os.Stdin)
 | 
				
			||||||
 | 
							if scanr.Scan() {
 | 
				
			||||||
 | 
								if !strings.EqualFold(scanr.Text(), "y") {
 | 
				
			||||||
 | 
									return errors.New("Jail is running")
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									err := stopJail(&jail)
 | 
				
			||||||
 | 
									if err != nil {
 | 
				
			||||||
 | 
										return err
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						// We need to rollback parent and childs
 | 
				
			||||||
 | 
						// Get all recursive snapshots
 | 
				
			||||||
 | 
						curDS, _ := getDatastoreFromArray(jail.Datastore, gDatastores)
 | 
				
			||||||
 | 
						rootDataset := fmt.Sprintf("%s/%s/%s", curDS.ZFSDataset, "jails", jail.Name)
 | 
				
			||||||
 | 
						cmd := fmt.Sprintf("zfs list -r -H -o name -t snapshot %s", rootDataset)
 | 
				
			||||||
 | 
						out, err := executeCommand(cmd)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							fmt.Printf("Error: listing snapshots: %s\n", err.Error())
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, line := range strings.Split(out, "\n") {
 | 
				
			||||||
 | 
							if len(line) > 0 {
 | 
				
			||||||
 | 
								ls := strings.Split(line, "@")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
								matched, _ := regexp.Match(fmt.Sprintf("^%s(\\/.*)?$", gSnapshotName), []byte(ls[1]))
 | 
				
			||||||
 | 
								if matched {
 | 
				
			||||||
 | 
									snaptorb = append(snaptorb, strings.Join(ls, "@"))
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, s := range snaptorb {
 | 
				
			||||||
 | 
							cmd := fmt.Sprintf("zfs rollback -r %s", s)
 | 
				
			||||||
 | 
							_, err := executeCommand(cmd)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error rolling back snapshot %s: %s\n", s, err.Error())
 | 
				
			||||||
 | 
								return err
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						fmt.Printf("Jail is back to %s\n", gSnapshotName)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										1553
									
								
								cmd/start.go
									
									
									
									
									
								
							
							
						
						
									
										1553
									
								
								cmd/start.go
									
									
									
									
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										205
									
								
								cmd/stop.go
									
									
									
									
									
								
							
							
						
						
									
										205
									
								
								cmd/stop.go
									
									
									
									
									
								
							@ -3,12 +3,16 @@ package cmd
 | 
				
			|||||||
import (
 | 
					import (
 | 
				
			||||||
	"os"
 | 
						"os"
 | 
				
			||||||
	"fmt"
 | 
						"fmt"
 | 
				
			||||||
//	"log"
 | 
						//"log"
 | 
				
			||||||
 | 
						"sync"
 | 
				
			||||||
	"errors"
 | 
						"errors"
 | 
				
			||||||
	"regexp"
 | 
						"regexp"
 | 
				
			||||||
	"os/exec"
 | 
						"os/exec"
 | 
				
			||||||
//	"reflect"
 | 
						//"reflect"
 | 
				
			||||||
 | 
						"strconv"
 | 
				
			||||||
	"strings"
 | 
						"strings"
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						log "github.com/sirupsen/logrus"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// TODO : Use SYS_RCTL_GET_RACCT syscall
 | 
					// TODO : Use SYS_RCTL_GET_RACCT syscall
 | 
				
			||||||
@ -38,7 +42,6 @@ func removeRctlRules(jail string, rules []string) error {
 | 
				
			|||||||
	return nil
 | 
						return nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					 | 
				
			||||||
// TODO: Validate with  >1 dataset
 | 
					// TODO: Validate with  >1 dataset
 | 
				
			||||||
func umountAndUnjailZFS(jail *Jail) error {
 | 
					func umountAndUnjailZFS(jail *Jail) error {
 | 
				
			||||||
	var ds []string
 | 
						var ds []string
 | 
				
			||||||
@ -48,10 +51,10 @@ func umountAndUnjailZFS(jail *Jail) error {
 | 
				
			|||||||
 | 
					
 | 
				
			||||||
	for _, zd := range ds {
 | 
						for _, zd := range ds {
 | 
				
			||||||
		// 1. Get dataset and childs
 | 
							// 1. Get dataset and childs
 | 
				
			||||||
		cmd := fmt.Sprintf("zfs list -H -r -o name -S name %s/%s", jail.Zpool, zd)
 | 
							cmd := fmt.Sprintf("zfs list -H -r -o name -S name %s", zd)
 | 
				
			||||||
		out, err := executeCommand(cmd)
 | 
							out, err := executeCommand(cmd)
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
			fmt.Printf(fmt.Sprintf("ERROR listing dataset %s/%s\n", jail.Zpool, zd))
 | 
								fmt.Printf(fmt.Sprintf("ERROR listing dataset %s\n", zd))
 | 
				
			||||||
			os.Exit(1)
 | 
								os.Exit(1)
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
		for _, c := range strings.Split(out, "\n") {
 | 
							for _, c := range strings.Split(out, "\n") {
 | 
				
			||||||
@ -69,22 +72,22 @@ func umountAndUnjailZFS(jail *Jail) error {
 | 
				
			|||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// 2. Unjail dataset from the host
 | 
						// 2. Unjail dataset from the host
 | 
				
			||||||
	cmd := fmt.Sprintf("zfs unjail %s %s/%s", jail.InternalName, jail.Zpool, ds[len(ds)-1])
 | 
						cmd := fmt.Sprintf("zfs unjail %s %s", jail.InternalName, ds[len(ds)-1])
 | 
				
			||||||
	_, err := executeCommand(cmd)
 | 
						_, err := executeCommand(cmd)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		fmt.Printf("ERROR unjailing %s/%s: %s\n", jail.Zpool, ds[len(ds)-1], err.Error())
 | 
							fmt.Printf("ERROR unjailing %s: %s\n", ds[len(ds)-1], err.Error())
 | 
				
			||||||
		return err
 | 
							return err
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return nil
 | 
						return nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					 | 
				
			||||||
func destroyVNetInterfaces(jail *Jail) error {
 | 
					func destroyVNetInterfaces(jail *Jail) error {
 | 
				
			||||||
	for _, i := range strings.Split(jail.Config.Ip4_addr, ",") {
 | 
						for _, i := range strings.Split(jail.Config.Ip4_addr, ",") {
 | 
				
			||||||
		iname := fmt.Sprintf("%s.%d", strings.Split(i, "|")[0], jail.JID)
 | 
							iname := fmt.Sprintf("%s.%d", strings.Split(i, "|")[0], jail.JID)
 | 
				
			||||||
		fmt.Printf("%s: ", iname)
 | 
							fmt.Printf("%s: ", iname)
 | 
				
			||||||
		_, err := executeCommand(fmt.Sprintf("ifconfig %s destroy", iname))
 | 
							_, err := executeCommand(fmt.Sprintf("ifconfig %s destroy", iname))
 | 
				
			||||||
 | 
							//_, err := executeScript(fmt.Sprintf("ifconfig %s destroy >/dev/null 2>&1", iname))
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
			return err
 | 
								return err
 | 
				
			||||||
		} else {
 | 
							} else {
 | 
				
			||||||
@ -101,16 +104,17 @@ func destroyVNetInterfaces(jail *Jail) error {
 | 
				
			|||||||
//  or else it will require a restart of "devfs" service.
 | 
					//  or else it will require a restart of "devfs" service.
 | 
				
			||||||
// But, stoppign the jail already removes this >1000 ID.
 | 
					// But, stoppign the jail already removes this >1000 ID.
 | 
				
			||||||
// So no need to call this function.
 | 
					// So no need to call this function.
 | 
				
			||||||
func deleteDevfsRuleset(jail *Jail) error {
 | 
					func deleteDevfsRuleset(ruleset int) error {
 | 
				
			||||||
	cmd := "devfs rule showsets"
 | 
						cmd := "devfs rule showsets"
 | 
				
			||||||
	out, err := executeCommand(cmd)
 | 
						out, err := executeCommand(cmd)
 | 
				
			||||||
	if err != nil {
 | 
						if err != nil {
 | 
				
			||||||
		return errors.New(fmt.Sprintf("ERROR listing rulesets: %s", err.Error()))
 | 
							return errors.New(fmt.Sprintf("ERROR listing rulesets: %s", err.Error()))
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
	
 | 
						
 | 
				
			||||||
 | 
						rs := strconv.Itoa(ruleset)
 | 
				
			||||||
	for _, r := range strings.Split(out, "\n") {
 | 
						for _, r := range strings.Split(out, "\n") {
 | 
				
			||||||
		if r == jail.Config.Devfs_ruleset {
 | 
							if r == rs {
 | 
				
			||||||
			cmd := fmt.Sprintf("devfs rule -s %s delset", jail.Config.Devfs_ruleset)
 | 
								cmd := fmt.Sprintf("devfs rule -s %d delset", ruleset)
 | 
				
			||||||
			_, err := executeCommand(cmd)
 | 
								_, err := executeCommand(cmd)
 | 
				
			||||||
			return err
 | 
								return err
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
@ -119,7 +123,6 @@ func deleteDevfsRuleset(jail *Jail) error {
 | 
				
			|||||||
	return nil
 | 
						return nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					 | 
				
			||||||
func umountJailFsFromHost(jail *Jail, mountpoint string) error {
 | 
					func umountJailFsFromHost(jail *Jail, mountpoint string) error {
 | 
				
			||||||
	cmd := "mount -p"
 | 
						cmd := "mount -p"
 | 
				
			||||||
	out, err := executeCommand(cmd)
 | 
						out, err := executeCommand(cmd)
 | 
				
			||||||
@ -135,7 +138,7 @@ func umountJailFsFromHost(jail *Jail, mountpoint string) error {
 | 
				
			|||||||
				cmd = fmt.Sprintf("umount %s%s", jail.RootPath, mountpoint)
 | 
									cmd = fmt.Sprintf("umount %s%s", jail.RootPath, mountpoint)
 | 
				
			||||||
				_, err := executeCommand(cmd)
 | 
									_, err := executeCommand(cmd)
 | 
				
			||||||
				if err != nil {
 | 
									if err != nil {
 | 
				
			||||||
					return errors.New(fmt.Sprintf("Error umounting %s/%s: %s", jail.RootPath, mountpoint, err.Error()))
 | 
										return errors.New(fmt.Sprintf("Error umounting %s%s: %s", jail.RootPath, mountpoint, err.Error()))
 | 
				
			||||||
				}
 | 
									}
 | 
				
			||||||
				return nil
 | 
									return nil
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
@ -145,7 +148,6 @@ func umountJailFsFromHost(jail *Jail, mountpoint string) error {
 | 
				
			|||||||
	return nil
 | 
						return nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					 | 
				
			||||||
// Internal usage only
 | 
					// Internal usage only
 | 
				
			||||||
func stopJail(jail *Jail) error {
 | 
					func stopJail(jail *Jail) error {
 | 
				
			||||||
	cmd := "jail -q"
 | 
						cmd := "jail -q"
 | 
				
			||||||
@ -167,6 +169,79 @@ func stopJail(jail *Jail) error {
 | 
				
			|||||||
	return nil
 | 
						return nil
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Stop all running jails by reverse priority
 | 
				
			||||||
 | 
					// Parallelize up to gMaxThreads
 | 
				
			||||||
 | 
					// Only parallelize same priority level jails
 | 
				
			||||||
 | 
					func StopAllRunningJails() {
 | 
				
			||||||
 | 
						var stopList []Jail
 | 
				
			||||||
 | 
						var wg *sync.WaitGroup
 | 
				
			||||||
 | 
						var curThNb int
 | 
				
			||||||
 | 
						var curPri int
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Get boot enabled jails
 | 
				
			||||||
 | 
						for _, j := range gJails {
 | 
				
			||||||
 | 
							if j.Running == true {
 | 
				
			||||||
 | 
								stopList = append(stopList, j)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Order by priority
 | 
				
			||||||
 | 
						js := initJailSortStruct()
 | 
				
			||||||
 | 
						fct, _, err := getStructFieldValue(js, "Config.PriorityDec")
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							log.Errorf("ERROR getting JailSort struct field \"Config.PriorityDec\"\n")
 | 
				
			||||||
 | 
							return
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						JailsOrderedBy(fct.Interface().(jailLessFunc)).Sort(stopList)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						wg = new(sync.WaitGroup)
 | 
				
			||||||
 | 
						curThNb = 0
 | 
				
			||||||
 | 
						for i, j := range stopList {
 | 
				
			||||||
 | 
							jFullName := fmt.Sprintf("%s/%s", j.Datastore, j.Name)
 | 
				
			||||||
 | 
							log.Debugf("Stopping %s with priority %s\n", jFullName, j.Config.Priority)
 | 
				
			||||||
 | 
							jailPri, err := strconv.Atoi(j.Config.Priority)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								panic(fmt.Sprintf("Invalid format for Priority (Jail %s)\n", jFullName))
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							if (curThNb >= gMaxThreads || i == 0) {
 | 
				
			||||||
 | 
								// FIXME : Use a pool instead of waiting for all threads to run a new one
 | 
				
			||||||
 | 
								wg.Wait()
 | 
				
			||||||
 | 
								curThNb = 0
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
								wg.Add(1)
 | 
				
			||||||
 | 
								curThNb++
 | 
				
			||||||
 | 
								curPri = jailPri
 | 
				
			||||||
 | 
								go func(jailFullName string) {
 | 
				
			||||||
 | 
									defer wg.Done()
 | 
				
			||||||
 | 
									StopJail([]string{jailFullName})
 | 
				
			||||||
 | 
								}(jFullName)
 | 
				
			||||||
 | 
							} else {
 | 
				
			||||||
 | 
								if (curPri == jailPri) {
 | 
				
			||||||
 | 
									wg.Add(1)
 | 
				
			||||||
 | 
									curThNb++
 | 
				
			||||||
 | 
									go func(jailFullName string) {
 | 
				
			||||||
 | 
										defer wg.Done()
 | 
				
			||||||
 | 
										StopJail([]string{jailFullName})
 | 
				
			||||||
 | 
									}(jFullName)
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									wg.Wait()
 | 
				
			||||||
 | 
									curThNb = 0
 | 
				
			||||||
 | 
								
 | 
				
			||||||
 | 
									wg.Add(1)
 | 
				
			||||||
 | 
									curThNb++
 | 
				
			||||||
 | 
									curPri = jailPri
 | 
				
			||||||
 | 
									go func(jailFullName string) {
 | 
				
			||||||
 | 
										defer wg.Done()
 | 
				
			||||||
 | 
										StopJail([]string{jailFullName})
 | 
				
			||||||
 | 
									}(jFullName)
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						wg.Wait()
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
/*
 | 
					/*
 | 
				
			||||||
	Stop jail:
 | 
						Stop jail:
 | 
				
			||||||
		Remove rctl rules
 | 
							Remove rctl rules
 | 
				
			||||||
@ -191,25 +266,41 @@ func stopJail(jail *Jail) error {
 | 
				
			|||||||
func StopJail(args []string) {
 | 
					func StopJail(args []string) {
 | 
				
			||||||
	// Current jail were stopping
 | 
						// Current jail were stopping
 | 
				
			||||||
	var cj *Jail
 | 
						var cj *Jail
 | 
				
			||||||
 | 
						var err error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	for _, j := range args {
 | 
						for _, a := range args {
 | 
				
			||||||
		fmt.Printf("> Stopping jail %s\n", j)
 | 
							// Check if jail exist and is distinctly named
 | 
				
			||||||
 | 
							cj, err = getJailFromArray(a, gJails)
 | 
				
			||||||
		for _, rj := range gJails {
 | 
							if err != nil {
 | 
				
			||||||
			if rj.Name == j {
 | 
								fmt.Printf("Error getting jail: %s\n", err)
 | 
				
			||||||
				cj = &rj
 | 
					 | 
				
			||||||
				break
 | 
					 | 
				
			||||||
			}
 | 
					 | 
				
			||||||
		}
 | 
					 | 
				
			||||||
		if cj == nil {
 | 
					 | 
				
			||||||
			fmt.Printf("Jail not found: %s\n", j)
 | 
					 | 
				
			||||||
			continue
 | 
								continue
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if cj.Running == false {
 | 
							if cj.Running == false {
 | 
				
			||||||
			fmt.Printf("Jail %s is not running!\n", cj.Name)
 | 
								fmt.Printf("Jail %s is not running!\n", cj.Name)
 | 
				
			||||||
			continue
 | 
								continue
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							fmt.Printf("> Stopping jail %s\n", a)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							// Get current version to update config.json
 | 
				
			||||||
 | 
							cvers, err := executeCommandInJail(cj, "/bin/freebsd-version")
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("ERROR: %s\n", err.Error())
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							cvers = strings.TrimRight(cvers, "\n")
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							//fmt.Sprintf(cj.Config.Release, cvers)
 | 
				
			||||||
 | 
							//cj.Config.Release = cvers
 | 
				
			||||||
 | 
							//cj.ConfigUpdated = true
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// This is working in this context, but value is not available in WriteConfigToDisk context :/
 | 
				
			||||||
 | 
							setStructFieldValue(cj, "Config.Release", cvers)
 | 
				
			||||||
 | 
							fmt.Printf("DEBUG: release was set, now is : %s\n", cj.Config.Release)
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// We need to get the real Config object, not a copy of it		
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		out, err := executeCommand(fmt.Sprintf("rctl jail:%s", cj.InternalName))
 | 
							out, err := executeCommand(fmt.Sprintf("rctl jail:%s", cj.InternalName))
 | 
				
			||||||
		if err == nil && len(out) > 0 {
 | 
							if err == nil && len(out) > 0 {
 | 
				
			||||||
			fmt.Printf("  > Remove RCTL rules:\n")
 | 
								fmt.Printf("  > Remove RCTL rules:\n")
 | 
				
			||||||
@ -221,17 +312,17 @@ func StopJail(args []string) {
 | 
				
			|||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if len (cj.Config.Exec_prestop) > 0 {
 | 
							if len(cj.Config.Exec_prestop) > 0 {
 | 
				
			||||||
			fmt.Printf("  > Execute prestop:\n")
 | 
								fmt.Printf("  > Execute pre-stop:\n")
 | 
				
			||||||
			_, err := executeCommand(cj.Config.Exec_prestop)
 | 
								_, err := executeCommand(cj.Config.Exec_prestop)
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
				fmt.Printf("ERROR: %s\n", err.Error())
 | 
									fmt.Printf("ERROR: %s\n", err.Error())
 | 
				
			||||||
			} else {
 | 
								} else {
 | 
				
			||||||
				fmt.Printf("  > Execute prestop: OK\n")
 | 
									fmt.Printf("  > Execute pre-stop: OK\n")
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if len (cj.Config.Exec_stop) > 0 {
 | 
							if len(cj.Config.Exec_stop) > 0 {
 | 
				
			||||||
			fmt.Printf("  > Execute stop:\n")
 | 
								fmt.Printf("  > Execute stop:\n")
 | 
				
			||||||
			_, err := executeCommandInJail(cj, cj.Config.Exec_stop)
 | 
								_, err := executeCommandInJail(cj, cj.Config.Exec_stop)
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
@ -261,13 +352,13 @@ func StopJail(args []string) {
 | 
				
			|||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		/*fmt.Printf("  > Remove devfsruleset %s:\n", cj.Config.Devfs_ruleset)
 | 
							fmt.Printf("  > Remove devfs ruleset %d: \n", cj.Devfs_ruleset)
 | 
				
			||||||
		err = deleteDevfsRuleset(cj)
 | 
							err = deleteDevfsRuleset(cj.Devfs_ruleset)
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
			fmt.Printf("ERROR: %s\n", err.Error())
 | 
								fmt.Printf("ERROR: %s\n", err.Error())
 | 
				
			||||||
		} else {
 | 
							} else {
 | 
				
			||||||
			fmt.Printf("  > Remove devfsruleset %s: OK\n", cj.Config.Devfs_ruleset)
 | 
								fmt.Printf("  > Remove devfsruleset %d: OK\n", cj.Devfs_ruleset)
 | 
				
			||||||
		}*/
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		fmt.Printf("  > Stop jail %s:\n", cj.Name)
 | 
							fmt.Printf("  > Stop jail %s:\n", cj.Name)
 | 
				
			||||||
		err = stopJail(cj)
 | 
							err = stopJail(cj)
 | 
				
			||||||
@ -297,15 +388,16 @@ func StopJail(args []string) {
 | 
				
			|||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if cj.Config.Mount_fdescfs > 0 {
 | 
							// FIXME: /dev/fd is mounted even with Mount_fdescfs = 0 ?!
 | 
				
			||||||
 | 
							//if cj.Config.Mount_fdescfs > 0 {
 | 
				
			||||||
			fmt.Printf("  > Umount fdescfs:\n")
 | 
								fmt.Printf("  > Umount fdescfs:\n")
 | 
				
			||||||
			err := umountJailFsFromHost(cj, "/dev/fd")
 | 
								err = umountJailFsFromHost(cj, "/dev/fd")
 | 
				
			||||||
			if err != nil {
 | 
								if err != nil {
 | 
				
			||||||
				fmt.Printf("ERROR: %s\n", err.Error())
 | 
									fmt.Printf("ERROR: %s\n", err.Error())
 | 
				
			||||||
			} else {
 | 
								} else {
 | 
				
			||||||
				fmt.Printf("  > Umount fdescfs: OK\n")
 | 
									fmt.Printf("  > Umount fdescfs: OK\n")
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							//}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		if cj.Config.Mount_devfs > 0 {
 | 
							if cj.Config.Mount_devfs > 0 {
 | 
				
			||||||
			fmt.Printf("  > Umount devfs:\n")
 | 
								fmt.Printf("  > Umount devfs:\n")
 | 
				
			||||||
@ -318,7 +410,6 @@ func StopJail(args []string) {
 | 
				
			|||||||
		}
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
		// Remove local mounts from $JAIL/fstab
 | 
							// Remove local mounts from $JAIL/fstab
 | 
				
			||||||
		// The way we get fstab is presumptuous
 | 
					 | 
				
			||||||
		fstab := strings.Replace(cj.ConfigPath, "config.json", "fstab", 1)
 | 
							fstab := strings.Replace(cj.ConfigPath, "config.json", "fstab", 1)
 | 
				
			||||||
		mounts, err := getFstab(fstab)
 | 
							mounts, err := getFstab(fstab)
 | 
				
			||||||
		if err != nil {
 | 
							if err != nil {
 | 
				
			||||||
@ -338,6 +429,44 @@ func StopJail(args []string) {
 | 
				
			|||||||
				fmt.Printf("  > Umount mountpoints from %s: OK\n", fstab)
 | 
									fmt.Printf("  > Umount mountpoints from %s: OK\n", fstab)
 | 
				
			||||||
			}
 | 
								}
 | 
				
			||||||
		}
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// TODO: Execute poststop
 | 
				
			||||||
 | 
							if len(cj.Config.Exec_poststop) > 0 {
 | 
				
			||||||
 | 
								fmt.Printf("  > Execute post-stop:\n")
 | 
				
			||||||
 | 
								_, err := executeCommand(cj.Config.Exec_poststop)
 | 
				
			||||||
 | 
								if err != nil {
 | 
				
			||||||
 | 
									fmt.Printf("ERROR: %s\n", err.Error())
 | 
				
			||||||
 | 
								} else {
 | 
				
			||||||
 | 
									fmt.Printf("  > Execute post-stop: OK\n")
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// Remove parameter file
 | 
				
			||||||
 | 
							pfile := fmt.Sprintf("/var/run/jail.%s.conf", cj.InternalName)
 | 
				
			||||||
 | 
							if err = os.Remove(pfile); err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error deleting parameter file %s\n", pfile)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							// We need this to get a reference to cj.Running (bc cj.Running is just a copy of value in the scope of StopJail())
 | 
				
			||||||
 | 
							for i, j := range gJails {
 | 
				
			||||||
 | 
								if strings.EqualFold(j.Name, cj.Name) && strings.EqualFold(j.Datastore, cj.Datastore) {
 | 
				
			||||||
 | 
									if err = setStructFieldValue(&gJails[i], "Running", "false"); err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("ERROR: setting Running property to false: %s\n", err.Error())
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									if err = setStructFieldValue(&gJails[i], "JID", "0"); err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("ERROR: setting JID property to 0: %s\n", err.Error())
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									if err = setStructFieldValue(&gJails[i], "InternalName", ""); err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("ERROR: clearing InternalName property: %s\n", err.Error())
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
									if err = setStructFieldValue(&gJails[i], "Devfs_ruleset", "0"); err != nil {
 | 
				
			||||||
 | 
										fmt.Printf("ERROR: setting Devfs_ruleset property to 0: %s\n", err.Error())
 | 
				
			||||||
 | 
									}
 | 
				
			||||||
 | 
								}
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
							fmt.Printf("DEBUG: release = %s\n", cj.Config.Release)
 | 
				
			||||||
 | 
							WriteConfigToDisk(cj.Name, false, true)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										971
									
								
								cmd/struct.go
									
									
									
									
									
								
							
							
						
						
									
										971
									
								
								cmd/struct.go
									
									
									
									
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										157
									
								
								cmd/update.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										157
									
								
								cmd/update.go
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,157 @@
 | 
				
			|||||||
 | 
					package cmd
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					import (
 | 
				
			||||||
 | 
						"os"
 | 
				
			||||||
 | 
						"fmt"
 | 
				
			||||||
 | 
						//"log"
 | 
				
			||||||
 | 
						"time"
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					const (
 | 
				
			||||||
 | 
						fbsdUpdateConfig = `
 | 
				
			||||||
 | 
						# $FreeBSD: releng/12.2/usr.sbin/freebsd-update/freebsd-update.conf 337338 2018-08-04 22:25:41Z brd $
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Trusted keyprint.  Changing this is a Bad Idea unless you've received
 | 
				
			||||||
 | 
						# a PGP-signed email from <security-officer@FreeBSD.org> telling you to
 | 
				
			||||||
 | 
						# change it and explaining why.
 | 
				
			||||||
 | 
						KeyPrint 800651ef4b4c71c27e60786d7b487188970f4b4169cc055784e21eb71d410cc5
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Server or server pool from which to fetch updates.  You can change
 | 
				
			||||||
 | 
						# this to point at a specific server if you want, but in most cases
 | 
				
			||||||
 | 
						# using a "nearby" server won't provide a measurable improvement in
 | 
				
			||||||
 | 
						# performance.
 | 
				
			||||||
 | 
						ServerName update.FreeBSD.org
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Components of the base system which should be kept updated.
 | 
				
			||||||
 | 
						Components world
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Example for updating the userland and the kernel source code only:
 | 
				
			||||||
 | 
						# Components src/base src/sys world
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Paths which start with anything matching an entry in an IgnorePaths
 | 
				
			||||||
 | 
						# statement will be ignored.
 | 
				
			||||||
 | 
						IgnorePaths
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Paths which start with anything matching an entry in an IDSIgnorePaths
 | 
				
			||||||
 | 
						# statement will be ignored by "freebsd-update IDS".
 | 
				
			||||||
 | 
						IDSIgnorePaths /usr/share/man/cat
 | 
				
			||||||
 | 
						IDSIgnorePaths /usr/share/man/whatis
 | 
				
			||||||
 | 
						IDSIgnorePaths /var/db/locate.database
 | 
				
			||||||
 | 
						IDSIgnorePaths /var/log
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Paths which start with anything matching an entry in an UpdateIfUnmodified
 | 
				
			||||||
 | 
						# statement will only be updated if the contents of the file have not been
 | 
				
			||||||
 | 
						# modified by the user (unless changes are merged; see below).
 | 
				
			||||||
 | 
						UpdateIfUnmodified /etc/ /var/ /root/ /.cshrc /.profile
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# When upgrading to a new FreeBSD release, files which match MergeChanges
 | 
				
			||||||
 | 
						# will have any local changes merged into the version from the new release.
 | 
				
			||||||
 | 
						MergeChanges /etc/
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						### Default configuration options:
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Directory in which to store downloaded updates and temporary
 | 
				
			||||||
 | 
						# files used by FreeBSD Update.
 | 
				
			||||||
 | 
						# WorkDir /var/db/freebsd-update
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Destination to send output of "freebsd-update cron" if an error
 | 
				
			||||||
 | 
						# occurs or updates have been downloaded.
 | 
				
			||||||
 | 
						# MailTo root
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Is FreeBSD Update allowed to create new files?
 | 
				
			||||||
 | 
						# AllowAdd yes
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Is FreeBSD Update allowed to delete files?
 | 
				
			||||||
 | 
						# AllowDelete yes
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# If the user has modified file ownership, permissions, or flags, should
 | 
				
			||||||
 | 
						# FreeBSD Update retain this modified metadata when installing a new version
 | 
				
			||||||
 | 
						# of that file?
 | 
				
			||||||
 | 
						# KeepModifiedMetadata yes
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# When upgrading between releases, should the list of Components be
 | 
				
			||||||
 | 
						# read strictly (StrictComponents yes) or merely as a list of components
 | 
				
			||||||
 | 
						# which *might* be installed of which FreeBSD Update should figure out
 | 
				
			||||||
 | 
						# which actually are installed and upgrade those (StrictComponents no)?
 | 
				
			||||||
 | 
						# StrictComponents no
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# When installing a new kernel perform a backup of the old one first
 | 
				
			||||||
 | 
						# so it is possible to boot the old kernel in case of problems.
 | 
				
			||||||
 | 
						# BackupKernel yes
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# If BackupKernel is enabled, the backup kernel is saved to this
 | 
				
			||||||
 | 
						# directory.
 | 
				
			||||||
 | 
						# BackupKernelDir /boot/kernel.old
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# When backing up a kernel also back up debug symbol files?
 | 
				
			||||||
 | 
						# BackupKernelSymbolFiles no
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						# Create a new boot environment when installing patches
 | 
				
			||||||
 | 
						# CreateBootEnv yes
 | 
				
			||||||
 | 
						`
 | 
				
			||||||
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// Internal usage only
 | 
				
			||||||
 | 
					func updateJail(jail *Jail) error {
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Create default config as temporary file
 | 
				
			||||||
 | 
						cfgFile, err := os.CreateTemp("", "gocage-jail-update-")
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						cfgFile.Write([]byte(fbsdUpdateConfig))
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						defer cfgFile.Close()
 | 
				
			||||||
 | 
						//defer os.Remove(cfgFile.Name())
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						cmd := fmt.Sprintf("/usr/sbin/freebsd-update --not-running-from-cron -f %s -b %s --currently-running %s fetch install", 
 | 
				
			||||||
 | 
										   cfgFile.Name(), jail.RootPath, jail.Config.Release)
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						fmt.Printf("DEBUG: Prepare to execute \"%s\"\n", cmd)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						err = executeCommandWithOutputToStdout(cmd)
 | 
				
			||||||
 | 
						if err != nil {
 | 
				
			||||||
 | 
							return err
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						// Get and write new release into config.json
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						
 | 
				
			||||||
 | 
						return nil
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func UpdateJail(args []string) {
 | 
				
			||||||
 | 
						// Current jail were stopping
 | 
				
			||||||
 | 
						var cj *Jail
 | 
				
			||||||
 | 
						var err error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
						for _, a := range args {
 | 
				
			||||||
 | 
							// Check if jail exist and is distinctly named
 | 
				
			||||||
 | 
							cj, err = getJailFromArray(a, gJails)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("Error getting jail: %s\n", err)
 | 
				
			||||||
 | 
								continue
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							fmt.Printf("  > Snapshot jail %s\n", cj.Name)
 | 
				
			||||||
 | 
							// Set snapshot name
 | 
				
			||||||
 | 
							dt := time.Now()
 | 
				
			||||||
 | 
							curDate := fmt.Sprintf("%s", dt.Format("2006-01-02_15-04-05"))
 | 
				
			||||||
 | 
							gSnapshotName = fmt.Sprintf("goc_update_%s_%s", cj.Config.Release, curDate)
 | 
				
			||||||
 | 
							err := createJailSnapshot(*cj)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("  > Snapshot jail %s: ERROR: %s\n", cj.Name, err.Error())
 | 
				
			||||||
 | 
								return
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
							fmt.Printf("  > Snapshot jail %s: OK\n", cj.Name)
 | 
				
			||||||
 | 
							
 | 
				
			||||||
 | 
							fmt.Printf("  > Update jail %s\n", cj.Name)
 | 
				
			||||||
 | 
							err = updateJail(cj)
 | 
				
			||||||
 | 
							if err != nil {
 | 
				
			||||||
 | 
								fmt.Printf("ERROR: %s\n", err.Error())
 | 
				
			||||||
 | 
							} else {
 | 
				
			||||||
 | 
								fmt.Printf("  > Update jail %s: OK\n", cj.Name)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
							
								
								
									
										2003
									
								
								cmd/utils.go
									
									
									
									
									
								
							
							
						
						
									
										2003
									
								
								cmd/utils.go
									
									
									
									
									
								
							
										
											
												File diff suppressed because it is too large
												Load Diff
											
										
									
								
							
							
								
								
									
										5
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										5
									
								
								go.mod
									
									
									
									
									
								
							@ -3,8 +3,13 @@ module gocage
 | 
				
			|||||||
go 1.17
 | 
					go 1.17
 | 
				
			||||||
 | 
					
 | 
				
			||||||
require (
 | 
					require (
 | 
				
			||||||
 | 
						github.com/c-robinson/iplib v1.0.3
 | 
				
			||||||
 | 
						github.com/c2h5oh/datasize v0.0.0-20220606134207-859f65c6625b
 | 
				
			||||||
 | 
						github.com/google/shlex v0.0.0-20191202100458-e7afc7fbc510
 | 
				
			||||||
 | 
						github.com/sirupsen/logrus v1.8.1
 | 
				
			||||||
	github.com/spf13/cobra v1.2.1
 | 
						github.com/spf13/cobra v1.2.1
 | 
				
			||||||
	github.com/spf13/viper v1.9.0
 | 
						github.com/spf13/viper v1.9.0
 | 
				
			||||||
 | 
						golang.org/x/net v0.0.0-20210503060351-7fd8e65b6420
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
require (
 | 
					require (
 | 
				
			||||||
 | 
				
			|||||||
@ -1,11 +1,12 @@
 | 
				
			|||||||
# List datastore root directory (the one containing jails/ and defaults.json)
 | 
					# List datastore(s) root directory (the one containing jails/ and defaults.json)
 | 
				
			||||||
datastore:
 | 
					datastore:
 | 
				
			||||||
    - /iocage
 | 
					    - /iocage
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Prefix all commands with sudo
 | 
				
			||||||
sudo: false
 | 
					sudo: false
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# Columns to display when "gocage list". Column names are struct fields, see cmd/struct.go
 | 
					# Columns to display when "gocage list". Column names are struct fields, see cmd/struct.go
 | 
				
			||||||
outcol: 'JID,Name,Config.Release,Config.Ip4_addr,Running'
 | 
					outcol: 'JID,Name,Config.Release,Config.Ip4_addr,Running'
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# Do not add line separator between jails
 | 
					# Do not add line separator between jails
 | 
				
			||||||
nolinesep: false
 | 
					nolinesep: false
 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										36
									
								
								jail/main.go
									
									
									
									
									
								
							
							
						
						
									
										36
									
								
								jail/main.go
									
									
									
									
									
								
							@ -14,16 +14,17 @@ package jail
 | 
				
			|||||||
*/
 | 
					*/
 | 
				
			||||||
import "C"
 | 
					import "C"
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
	"strconv"
 | 
					    "strconv"
 | 
				
			||||||
//    "syscall"
 | 
					//    "syscall"
 | 
				
			||||||
    "unsafe"
 | 
					    "unsafe"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
type Jail struct {
 | 
					type Jail struct {
 | 
				
			||||||
    Name	string
 | 
					    Name          string
 | 
				
			||||||
    Jid		int
 | 
					    Jid           int
 | 
				
			||||||
	Path	string
 | 
					    Path          string
 | 
				
			||||||
 | 
					    Devfs_ruleset int
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@ -33,8 +34,8 @@ func GetJails() ([]Jail, error) {
 | 
				
			|||||||
    var jl Jail
 | 
					    var jl Jail
 | 
				
			||||||
    var err error
 | 
					    var err error
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Make "params" a list of 4 jails parameters
 | 
						// Make "params" a list of 5 jails parameters
 | 
				
			||||||
    params := make([]C.struct_jailparam, 4)
 | 
					    params := make([]C.struct_jailparam, 5)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    // initialize parameter names
 | 
					    // initialize parameter names
 | 
				
			||||||
    csname := C.CString("name")
 | 
					    csname := C.CString("name")
 | 
				
			||||||
@ -43,27 +44,31 @@ func GetJails() ([]Jail, error) {
 | 
				
			|||||||
    defer C.free(unsafe.Pointer(csjid))
 | 
					    defer C.free(unsafe.Pointer(csjid))
 | 
				
			||||||
    cspath := C.CString("path")
 | 
					    cspath := C.CString("path")
 | 
				
			||||||
    defer C.free(unsafe.Pointer(cspath))
 | 
					    defer C.free(unsafe.Pointer(cspath))
 | 
				
			||||||
 | 
						csdevfsrs := C.CString("devfs_ruleset")
 | 
				
			||||||
 | 
						defer C.free(unsafe.Pointer(csdevfsrs))
 | 
				
			||||||
    cslastjid := C.CString("lastjid")
 | 
					    cslastjid := C.CString("lastjid")
 | 
				
			||||||
    defer C.free(unsafe.Pointer(cslastjid))
 | 
					    defer C.free(unsafe.Pointer(cslastjid))
 | 
				
			||||||
	
 | 
						
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    // initialize params struct with parameter names
 | 
					    // initialize params struct with parameter names
 | 
				
			||||||
    C.jailparam_init(¶ms[0], csname)
 | 
					    C.jailparam_init(¶ms[0], csname)
 | 
				
			||||||
    C.jailparam_init(¶ms[1], csjid)
 | 
					    C.jailparam_init(¶ms[1], csjid)
 | 
				
			||||||
    C.jailparam_init(¶ms[2], cspath)
 | 
					    C.jailparam_init(¶ms[2], cspath)
 | 
				
			||||||
 | 
						C.jailparam_init(¶ms[3], csdevfsrs)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    // The key to retrieve jail. lastjid = 0 returns first jail and its jid as jailparam_get return value
 | 
					    // The key to retrieve jail. lastjid = 0 returns first jail and its jid as jailparam_get return value
 | 
				
			||||||
    C.jailparam_init(¶ms[3], cslastjid)
 | 
					    C.jailparam_init(¶ms[4], cslastjid)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    lastjailid := 0
 | 
					    lastjailid := 0
 | 
				
			||||||
    cslastjidval := C.CString(strconv.Itoa(lastjailid))
 | 
					    cslastjidval := C.CString(strconv.Itoa(lastjailid))
 | 
				
			||||||
    defer C.free(unsafe.Pointer(cslastjidval))
 | 
					    defer C.free(unsafe.Pointer(cslastjidval))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    C.jailparam_import(¶ms[3], cslastjidval)
 | 
					    C.jailparam_import(¶ms[4], cslastjidval)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    // loop on existing jails
 | 
					    // loop on existing jails
 | 
				
			||||||
    for lastjailid >= 0 {
 | 
					    for lastjailid >= 0 {
 | 
				
			||||||
        // get parameter values
 | 
					        // get parameter values
 | 
				
			||||||
        lastjailid = int(C.jailparam_get(¶ms[0], 4, 0))
 | 
					        lastjailid = int(C.jailparam_get(¶ms[0], 5, 0))
 | 
				
			||||||
        if lastjailid > 0 {
 | 
					        if lastjailid > 0 {
 | 
				
			||||||
            nametmp := C.jailparam_export(¶ms[0])
 | 
					            nametmp := C.jailparam_export(¶ms[0])
 | 
				
			||||||
            jl.Name = C.GoString(nametmp)
 | 
					            jl.Name = C.GoString(nametmp)
 | 
				
			||||||
@ -75,23 +80,28 @@ func GetJails() ([]Jail, error) {
 | 
				
			|||||||
            // Memory mgmt : Non gere par Go
 | 
					            // Memory mgmt : Non gere par Go
 | 
				
			||||||
            C.free(unsafe.Pointer(jidtmp))
 | 
					            C.free(unsafe.Pointer(jidtmp))
 | 
				
			||||||
            
 | 
					            
 | 
				
			||||||
			pathtmp := C.jailparam_export(¶ms[2])
 | 
					            pathtmp := C.jailparam_export(¶ms[2])
 | 
				
			||||||
            jl.Path = C.GoString(pathtmp)
 | 
					            jl.Path = C.GoString(pathtmp)
 | 
				
			||||||
            // Memory mgmt : Non gere par Go
 | 
					            // Memory mgmt : Non gere par Go
 | 
				
			||||||
            C.free(unsafe.Pointer(pathtmp))
 | 
					            C.free(unsafe.Pointer(pathtmp))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					            drstmp := C.jailparam_export(¶ms[3])
 | 
				
			||||||
 | 
					            jl.Devfs_ruleset, _ = strconv.Atoi(C.GoString(drstmp))
 | 
				
			||||||
 | 
					            // Memory mgmt : Non gere par Go
 | 
				
			||||||
 | 
					            C.free(unsafe.Pointer(drstmp))
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            jls = append(jls, jl)
 | 
					            jls = append(jls, jl)
 | 
				
			||||||
            //log.Debug("Got jid " + strconv.Itoa(jl.jid) + " with name " + jl.name)
 | 
					            //log.Debug("Got jid " + strconv.Itoa(jl.jid) + " with name " + jl.name)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
            // Prepare next loop iteration
 | 
					            // Prepare next loop iteration
 | 
				
			||||||
            cslastjidval := C.CString(strconv.Itoa(lastjailid))
 | 
					            cslastjidval := C.CString(strconv.Itoa(lastjailid))
 | 
				
			||||||
            defer C.free(unsafe.Pointer(cslastjidval))
 | 
					            defer C.free(unsafe.Pointer(cslastjidval))
 | 
				
			||||||
            C.jailparam_import(¶ms[3], cslastjidval)
 | 
					            C.jailparam_import(¶ms[4], cslastjidval)
 | 
				
			||||||
        }
 | 
					        }
 | 
				
			||||||
    }
 | 
					    }
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	// Free 4 items of params list
 | 
						// Free 5 items of params list
 | 
				
			||||||
    C.jailparam_free(¶ms[0], 4)
 | 
					    C.jailparam_free(¶ms[0], 5)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
    return jls, err
 | 
					    return jls, err
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										7
									
								
								main.go
									
									
									
									
									
								
							
							
						
						
									
										7
									
								
								main.go
									
									
									
									
									
								
							@ -1,10 +1,9 @@
 | 
				
			|||||||
package main
 | 
					package main
 | 
				
			||||||
 | 
					
 | 
				
			||||||
import (
 | 
					import (
 | 
				
			||||||
    "gocage/cmd"
 | 
						"gocage/cmd"
 | 
				
			||||||
)
 | 
					)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
func main () {
 | 
					func main() {
 | 
				
			||||||
    cmd.Execute()
 | 
						cmd.Execute()
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					 | 
				
			||||||
 | 
				
			|||||||
							
								
								
									
										43
									
								
								service/gocage
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										43
									
								
								service/gocage
									
									
									
									
									
										Executable file
									
								
							@ -0,0 +1,43 @@
 | 
				
			|||||||
 | 
					#!/bin/sh
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# $FreeBSD$
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# PROVIDE: gocage
 | 
				
			||||||
 | 
					# REQUIRE: LOGIN cleanvar
 | 
				
			||||||
 | 
					# KEYWORD: shutdown
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# Add the following lines to /etc/rc.conf to enable :
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# gocage_enable="YES"
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					# gocage_conf="/usr/local/etc/gocage.conf.yml"
 | 
				
			||||||
 | 
					#
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					. /etc/rc.subr
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					name="gocage"
 | 
				
			||||||
 | 
					rcvar=gocage_enable
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					# read configuration and set defaults
 | 
				
			||||||
 | 
					load_rc_config "$name"
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					: ${gocage_enable:="NO"}
 | 
				
			||||||
 | 
					: ${gocage_conf="/usr/local/etc/gocage.conf.yml"}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					start_cmd=${name}_start
 | 
				
			||||||
 | 
					stop_cmd=${name}_stop
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					gocage_start()
 | 
				
			||||||
 | 
					{
 | 
				
			||||||
 | 
					    echo "Gocage starting jails... "
 | 
				
			||||||
 | 
					    /usr/local/bin/gocage -c ${gocage_conf} start
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					gocage_stop()
 | 
				
			||||||
 | 
					{
 | 
				
			||||||
 | 
					    echo "Gocage stopping jails... "
 | 
				
			||||||
 | 
					    /usr/local/bin/gocage  -c ${gocage_conf} stop
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					run_rc_command "$1"
 | 
				
			||||||
		Reference in New Issue
	
	Block a user