added update & uboot functions

update_test.go

@@ -0,0 +1,179 @@
+package alpine_builder
+
+import (
+	"bytes"
+	"compress/gzip"
+	"crypto/rand"
+	"encoding/binary"
+	"io"
+	"io/ioutil"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func init() {
+	ubootFile = "test_uboot"
+	ubootRemountRW = "true"
+	ubootRemountRO = "true"
+	rootPartitionA = "test_rootA"
+	rootPartitionB = "test_rootB"
+}
+
+func TestLoadUbootDat(t *testing.T) {
+	ass := assert.New(t)
+
+	// file missing
+	data, err := loadUbootDat()
+	ass.EqualError(err, "failed to open file: open test_uboot: no such file or directory")
+	ass.Equal([]byte{1, 0, 1}, data[:3])
+
+	// file with invalid data
+	ass.NoError(ioutil.WriteFile(ubootFile, []byte{}, os.ModePerm))
+
+	data, err = loadUbootDat()
+	ass.EqualError(err, "invalid dat file -> fallback to defaults")
+	ass.Equal([]byte{1, 0, 1}, data[:3])
+
+	// file with invalid CRC
+	testData := make([]byte, 1024)
+	testData[0] = 1
+	testData[1] = 2
+	testData[2] = 2
+
+	ass.NoError(ioutil.WriteFile(ubootFile, testData, os.ModePerm))
+	data, err = loadUbootDat()
+	ass.EqualError(err, "invalid crc -> fallback to defaults")
+	ass.Equal([]byte{1, 0, 1}, data[:3])
+
+	// file with valid CRC
+	binary.LittleEndian.PutUint32(testData[1020:], 0x982E8B7A)
+
+	ass.NoError(ioutil.WriteFile(ubootFile, testData, os.ModePerm))
+	data, err = loadUbootDat()
+	ass.NoError(err)
+	ass.Equal(testData, data)
+
+	_ = os.Remove(ubootFile)
+}
+
+func TestSaveUbootDat(t *testing.T) {
+	ass := assert.New(t)
+
+	testData := make([]byte, 1024)
+	testData[0] = 1
+	testData[1] = 2
+	testData[2] = 2
+	ass.NoError(saveUbootDat(testData))
+
+	binary.LittleEndian.PutUint32(testData[1020:], 0x982E8B7A)
+
+	data, err := ioutil.ReadFile(ubootFile)
+	ass.NoError(err)
+	ass.Equal(testData, data)
+
+	_ = os.Remove(ubootFile)
+}
+
+func TestUBootResetCounter(t *testing.T) {
+	ass := assert.New(t)
+
+	// write test file
+	testData := make([]byte, 1024)
+	testData[0] = 1
+	testData[1] = 2
+	testData[2] = 2
+	binary.LittleEndian.PutUint32(testData[1020:], 0x982E8B7A)
+	ass.NoError(ioutil.WriteFile(ubootFile, testData, os.ModePerm))
+
+	ass.NoError(UBootResetCounter())
+
+	data, err := ioutil.ReadFile(ubootFile)
+	ass.NoError(err)
+	ass.Zero(data[1])
+
+	_ = os.Remove(ubootFile)
+}
+
+func TestUBootActive(t *testing.T) {
+	ass := assert.New(t)
+
+	// write test file
+	testData := make([]byte, 1024)
+	testData[0] = 1
+	testData[1] = 2
+	testData[2] = 2
+	binary.LittleEndian.PutUint32(testData[1020:], 0x982E8B7A)
+	ass.NoError(ioutil.WriteFile(ubootFile, testData, os.ModePerm))
+
+	ass.Equal(uint8(2), UBootActive())
+
+	_ = os.Remove(ubootFile)
+}
+
+func TestUBootSetActive(t *testing.T) {
+	ass := assert.New(t)
+
+	// write test file
+	testData := make([]byte, 1024)
+	testData[0] = 1
+	testData[1] = 2
+	testData[2] = 2
+	binary.LittleEndian.PutUint32(testData[1020:], 0x982E8B7A)
+	ass.NoError(ioutil.WriteFile(ubootFile, testData, os.ModePerm))
+
+	ass.NoError(UBootSetActive(1))
+
+	data, err := ioutil.ReadFile(ubootFile)
+	ass.NoError(err)
+	ass.Equal(uint8(1), data[2])
+
+	_ = os.Remove(ubootFile)
+}
+
+func TestUpdateSystem(t *testing.T) {
+	ass := assert.New(t)
+
+	// test uboot file
+	testData := make([]byte, 1024)
+	testData[0] = 1
+	testData[1] = 2
+	testData[2] = 2
+	binary.LittleEndian.PutUint32(testData[1020:], 0x982E8B7A)
+	ass.NoError(ioutil.WriteFile(ubootFile, testData, os.ModePerm))
+
+	// generate test image content
+	size := int64(1024 * 1024 * 5)
+	testImgData := make([]byte, size)
+	buffer := bytes.NewBuffer(testImgData)
+	_, err := io.CopyN(buffer, rand.Reader, size)
+	ass.NoError(err)
+
+	// write compressed image file
+	file, err := os.Create("test_image.gz")
+	ass.NoError(err)
+	gzipWriter := gzip.NewWriter(file)
+	_, err = gzipWriter.Write(testImgData)
+	ass.NoError(err)
+	ass.NoError(gzipWriter.Close())
+	ass.NoError(file.Close())
+
+	ass.NoError(ioutil.WriteFile(rootPartitionA, nil, os.ModePerm))
+	ass.NoError(UpdateSystem("test_image.gz"))
+
+	// check if image was written
+	data, err := ioutil.ReadFile(rootPartitionA)
+	ass.NoError(err)
+	ass.Equal(testImgData, data)
+
+	// check if uboot dat was updated
+	data, err = ioutil.ReadFile(ubootFile)
+	ass.NoError(err)
+	ass.Equal(uint8(1), data[2])
+
+	_ = os.Remove("test_image.gz")
+	_ = os.Remove(ubootFile)
+	_ = os.Remove(rootPartitionA)
+	_ = os.Remove(rootPartitionB)
+}