Initial release

Author: ned

.gitignore

@@ -0,0 +1,7 @@
+bin
+build
+scripts
+sweet.conf
+sweet-workspace
+workspace
+test-env

LICENSE

@@ -0,0 +1,29 @@
+This software is distributed under the New BSD license.
+
+Copyright (c) 2014, AppliedTrust, Inc.
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of Sweet, the name of AppliedTrust, nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

Makefile

@@ -0,0 +1,72 @@
+VERSION=$(shell bin/sweet64 --version)
+
+####
+linux32: bindata
+	GOOS=linux GOARCH=386 go build -o bin/sweet32 cmd/main.go
+
+all: bindata binaries packages
+
+binaries: bindata linux32 linux64 darwin64
+
+bindata:
+	GOOS=linux GOARCH=amd64 $(GOPATH)/bin/go-bindata -pkg="sweet" tmpl/ static/
+
+linux64: bindata
+	GOOS=linux GOARCH=amd64 go build -o bin/sweet64 cmd/main.go
+
+darwin64: bindata
+	GOOS=darwin GOARCH=amd64 go build -o bin/sweet-osx cmd/main.go
+
+####
+packages: rpm32 deb32 rpm64 deb64
+
+rpm32:
+	rm -rf build/rpm
+	mkdir -p build/rpm/sweet/usr/local/bin
+	cp bin/sweet32 build/rpm/sweet/usr/local/bin/sweet
+	fpm -s dir -t rpm -n sweet -a i386 --epoch 0 -v $(VERSION) -C build/rpm/sweet .
+	mv sweet-$(VERSION)-1.i386.rpm bin/
+
+rpm64:
+	rm -rf build/rpm
+	mkdir -p build/rpm/sweet/usr/local/bin
+	cp bin/sweet64 build/rpm/sweet/usr/local/bin/sweet
+	fpm -s dir -t rpm -n sweet -a x86_64 --epoch 0 -v $(VERSION) -C build/rpm/sweet .
+	mv sweet-$(VERSION)-1.x86_64.rpm bin/
+
+deb32:
+	rm -rf build/deb
+	mkdir -p build/deb/sweet/usr/local/bin
+	cp bin/sweet32 build/deb/sweet/usr/local/bin/sweet
+	fpm -s dir -t deb -n sweet -a i386 -v $(VERSION) -C build/deb/sweet .
+	mv sweet_$(VERSION)_i386.deb bin/
+
+deb64:
+	rm -rf build/deb
+	mkdir -p build/deb/sweet/usr/local/bin
+	cp bin/sweet64 build/deb/sweet/usr/local/bin/sweet
+	fpm -s dir -t deb -n sweet -a amd64 -v $(VERSION) -C build/deb/sweet .
+	mv sweet_$(VERSION)_amd64.deb bin/
+
+#### 
+release:
+	$(GOPATH)/bin/github-release release --user appliedtrust --repo sweet --tag $(VERSION) \
+		--name "Sweet $(VERSION)" \
+		--description "Network device configuration backups and change alerts for the 21st century." \
+		--pre-release
+	$(GOPATH)/bin/github-release upload --user appliedtrust --repo sweet --tag $(VERSION) \
+		--name "sweet-linux-32" \
+		--file bin/sweet32
+	$(GOPATH)/bin/github-release upload --user appliedtrust --repo sweet --tag $(VERSION) \
+		--name "sweet-linux-64" \
+		--file bin/sweet64
+	$(GOPATH)/bin/github-release upload --user appliedtrust --repo sweet --tag $(VERSION) \
+		--name "sweet-osx" \
+		--file bin/sweet-osx
+	$(GOPATH)/bin/github-release upload --user appliedtrust --repo sweet --tag $(VERSION) \
+		--name bin/sweet_$(VERSION)_i386.deb \
+		--file bin/sweet_$(VERSION)_i386.deb
+	$(GOPATH)/bin/github-release upload --user appliedtrust --repo sweet --tag $(VERSION) \
+		--name bin/sweet-$(VERSION)-1.i386.rpm \
+		--file bin/sweet-$(VERSION)-1.i386.rpm
+

bindata.go

@@ -0,0 +1,49 @@
+package sweet
+
+import (
+	"io/ioutil"
+	"os"
+	"strings"
+	"testing"
+)
+
+func TestBindata(t *testing.T) {
+	directories := []string{"tmpl", "static"}
+	files := map[string]bool{}
+
+	// finding all bindata source files
+	for _, dir := range directories {
+		dirFiles, err := ioutil.ReadDir(dir)
+		if err != nil {
+			t.Errorf("Can't find bindata source directory: %s", dir)
+		}
+		for _, f := range dirFiles {
+			files[dir+string(os.PathSeparator)+f.Name()] = true
+		}
+	}
+
+	// checking dashboard template file
+	filename := "tmpl/index.html"
+	if _, exists := files[filename]; !exists {
+		t.Errorf("Critical bindata source file %s is missing", filename)
+	}
+	asset, err := Asset(filename)
+	if err != nil {
+		t.Errorf("Bindata is missing a critical file: %s", filename)
+	}
+	if !strings.Contains(string(asset), "Sweet status dashboard") {
+		t.Errorf("%s is missing expected HTML meta tag", filename)
+	}
+
+	// checking each source file is present in bindata
+	for filename = range files {
+		asset, err := Asset(filename)
+		if err != nil {
+			t.Errorf("Bindata is missing a file: %s", filename)
+		}
+		if len(asset) < 1 {
+			t.Errorf("Zero-length bindata file: %s", filename)
+		}
+	}
+
+}

bindata_test.go

@@ -0,0 +1,63 @@
+package sweet
+
+import (
+	"fmt"
+	"time"
+)
+
+func CollectCisco(device DeviceAccess) map[string]string {
+	result := make(map[string]string)
+	result["err"] = ""
+
+	c, err := newSSHCollector(device)
+	if err != nil {
+		result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+		return result
+	}
+
+	if err := expect("assword:", c.Receive); err != nil {
+		result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+		return result
+	}
+	c.Send <- device.Config["pass"] + "\n"
+	multi := []string{"#", ">", "assword:"}
+	m, err := expectMulti(multi, c.Receive)
+	if err != nil {
+		result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+		return result
+	}
+	if m == "assword:" { // bad pw
+		result["err"] = fmt.Sprintf("%s: Bad login password.", device.Hostname)
+		return result
+	} else if m == ">" { // attempt enable
+		c.Send <- "enable\n"
+		if err := expect("assword:", c.Receive); err != nil {
+			result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+			return result
+		}
+		c.Send <- device.Config["enable"] + "\n"
+		if err := expect("#", c.Receive); err != nil {
+			result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+			return result
+		}
+	}
+	c.Send <- "terminal length 0\n"
+	if err := expect("#", c.Receive); err != nil {
+		result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+		return result
+	}
+	c.Send <- "terminal pager 0\n"
+	if err := expect("#", c.Receive); err != nil {
+		result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+		return result
+	}
+	c.Send <- "show conf\n"
+	result["config"], err = timeoutSave(c.Receive, 500*time.Millisecond)
+	if err != nil {
+		result["err"] = fmt.Sprintf("%s: %s", device.Hostname, err.Error())
+		return result
+	}
+	c.Send <- "exit\n"
+
+	return result
+}

cisco.go

@@ -0,0 +1,43 @@
+package sweet
+
+import (
+	"os"
+	"strings"
+	"testing"
+	"time"
+)
+
+func TestCiscoGood(t *testing.T) {
+	d := new(DeviceAccess)
+	d.Config = make(map[string]string)
+	d.Method = "cisco"
+	d.Timeout = 10 * time.Second
+
+	if os.Getenv("SWEET_TEST_HOST") == "" {
+		t.Error("Test requries SWEET_TEST_HOST environment variable")
+		return
+	}
+	if os.Getenv("SWEET_TEST_USER") == "" {
+		t.Error("Test requries SWEET_TEST_USER environment variable")
+		return
+	}
+	if os.Getenv("SWEET_TEST_PASS") == "" {
+		t.Error("Test requries SWEET_TEST_PASS environment variable")
+		return
+	}
+
+	d.Hostname = os.Getenv("SWEET_TEST_HOST")
+	d.Config["user"] = os.Getenv("SWEET_TEST_USER")
+	d.Config["pass"] = os.Getenv("SWEET_TEST_PASS")
+
+	d.Target = d.Hostname
+
+	s := CollectCisco(*d)
+	if !strings.Contains(s["config"], "aaa authorization commands") {
+		t.Errorf("Config missing aaa line")
+	}
+	if !strings.Contains(s["config"], "ntp access-group peer") {
+		t.Errorf("Config missing ntp line close to end")
+	}
+
+}