cloudflared-mirror/cmd/cloudflared/updater/workers_service_test.go

300 lines
7.9 KiB
Go
Raw Normal View History

2020-08-12 16:16:14 +00:00
package updater
import (
"archive/tar"
"bytes"
"compress/gzip"
"crypto/sha256"
"encoding/json"
"fmt"
"io/ioutil"
"log"
"net/http"
"net/http/httptest"
2020-08-12 16:16:14 +00:00
"os"
"runtime"
"testing"
"github.com/stretchr/testify/require"
2020-08-12 16:16:14 +00:00
)
func respondWithJSON(w http.ResponseWriter, v interface{}, status int) {
data, _ := json.Marshal(v)
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(status)
w.Write(data)
}
func respondWithData(w http.ResponseWriter, b []byte, status int) {
w.Header().Set("Content-Type", "application/octet-stream")
w.WriteHeader(status)
w.Write(b)
}
const mostRecentVersion = "2021.2.5"
const mostRecentBetaVersion = "2021.3.0"
const knownBuggyVersion = "2020.12.0"
const expectedUserMsg = "This message is expected when running a known buggy version"
2020-08-12 16:16:14 +00:00
func updateHandler(w http.ResponseWriter, r *http.Request) {
version := mostRecentVersion
host := fmt.Sprintf("http://%s", r.Host)
2020-08-12 16:16:14 +00:00
url := host + "/download"
query := r.URL.Query()
if query.Get(BetaKeyName) == "true" {
version = mostRecentBetaVersion
2020-08-12 16:16:14 +00:00
url = host + "/beta"
}
requestedVersion := query.Get(VersionKeyName)
if requestedVersion != "" {
version = requestedVersion
url = fmt.Sprintf("%s?version=%s", url, requestedVersion)
}
if query.Get(ArchitectureKeyName) != runtime.GOARCH || query.Get(OSKeyName) != runtime.GOOS {
respondWithJSON(w, VersionResponse{Error: "unsupported os and architecture"}, http.StatusBadRequest)
return
}
h := sha256.New()
fmt.Fprint(h, version)
checksum := fmt.Sprintf("%x", h.Sum(nil))
var userMessage = ""
if query.Get(ClientVersionName) == knownBuggyVersion {
userMessage = expectedUserMsg
}
v := VersionResponse{URL: url, Version: version, Checksum: checksum, UserMessage: userMessage}
2020-08-12 16:16:14 +00:00
respondWithJSON(w, v, http.StatusOK)
}
func gzipUpdateHandler(w http.ResponseWriter, r *http.Request) {
log.Println("got a request!")
version := "2020.09.02"
h := sha256.New()
fmt.Fprint(h, version)
checksum := fmt.Sprintf("%x", h.Sum(nil))
url := fmt.Sprintf("http://%s/gzip-download.tgz", r.Host)
v := VersionResponse{URL: url, Version: version, Checksum: checksum}
2020-08-12 16:16:14 +00:00
respondWithJSON(w, v, http.StatusOK)
}
func compressedDownloadHandler(w http.ResponseWriter, r *http.Request) {
version := "2020.09.02"
buf := new(bytes.Buffer)
gw := gzip.NewWriter(buf)
tw := tar.NewWriter(gw)
header := &tar.Header{
Size: int64(len(version)),
Name: "download",
}
tw.WriteHeader(header)
tw.Write([]byte(version))
tw.Close()
gw.Close()
respondWithData(w, buf.Bytes(), http.StatusOK)
}
func downloadHandler(w http.ResponseWriter, r *http.Request) {
version := mostRecentVersion
2020-08-12 16:16:14 +00:00
requestedVersion := r.URL.Query().Get(VersionKeyName)
if requestedVersion != "" {
version = requestedVersion
}
respondWithData(w, []byte(version), http.StatusOK)
}
func betaHandler(w http.ResponseWriter, r *http.Request) {
respondWithData(w, []byte(mostRecentBetaVersion), http.StatusOK)
2020-08-12 16:16:14 +00:00
}
func failureHandler(w http.ResponseWriter, r *http.Request) {
respondWithJSON(w, VersionResponse{Error: "unsupported os and architecture"}, http.StatusBadRequest)
}
func createServer() *httptest.Server {
2020-08-12 16:16:14 +00:00
mux := http.NewServeMux()
mux.HandleFunc("/updater", updateHandler)
mux.HandleFunc("/download", downloadHandler)
mux.HandleFunc("/beta", betaHandler)
mux.HandleFunc("/fail", failureHandler)
mux.HandleFunc("/compressed", gzipUpdateHandler)
mux.HandleFunc("/gzip-download.tgz", compressedDownloadHandler)
return httptest.NewServer(mux)
2020-08-12 16:16:14 +00:00
}
func createTestFile(t *testing.T, path string) {
f, err := os.Create(path)
require.NoError(t, err)
2020-08-12 16:16:14 +00:00
fmt.Fprint(f, "2020.08.04")
f.Close()
}
func TestUpdateService(t *testing.T) {
ts := createServer()
defer ts.Close()
2020-08-12 16:16:14 +00:00
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
log.Println("server url: ", ts.URL)
2020-08-12 16:16:14 +00:00
s := NewWorkersService("2020.8.2", fmt.Sprintf("%s/updater", ts.URL), testFilePath, Options{})
2020-08-12 16:16:14 +00:00
v, err := s.Check()
require.NoError(t, err)
require.Equal(t, v.Version(), mostRecentVersion)
2020-08-12 16:16:14 +00:00
require.NoError(t, v.Apply())
2020-08-12 16:16:14 +00:00
dat, err := ioutil.ReadFile(testFilePath)
require.NoError(t, err)
2020-08-12 16:16:14 +00:00
require.Equal(t, string(dat), mostRecentVersion)
2020-08-12 16:16:14 +00:00
}
func TestBetaUpdateService(t *testing.T) {
ts := createServer()
defer ts.Close()
2020-08-12 16:16:14 +00:00
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
s := NewWorkersService("2020.8.2", fmt.Sprintf("%s/updater", ts.URL), testFilePath, Options{IsBeta: true})
2020-08-12 16:16:14 +00:00
v, err := s.Check()
require.NoError(t, err)
require.Equal(t, v.Version(), mostRecentBetaVersion)
2020-08-12 16:16:14 +00:00
require.NoError(t, v.Apply())
2020-08-12 16:16:14 +00:00
dat, err := ioutil.ReadFile(testFilePath)
require.NoError(t, err)
2020-08-12 16:16:14 +00:00
require.Equal(t, string(dat), mostRecentBetaVersion)
2020-08-12 16:16:14 +00:00
}
func TestFailUpdateService(t *testing.T) {
ts := createServer()
defer ts.Close()
2020-08-12 16:16:14 +00:00
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
s := NewWorkersService("2020.8.2", fmt.Sprintf("%s/fail", ts.URL), testFilePath, Options{})
2020-08-12 16:16:14 +00:00
v, err := s.Check()
require.Error(t, err)
require.Nil(t, v)
2020-08-12 16:16:14 +00:00
}
func TestNoUpdateService(t *testing.T) {
ts := createServer()
defer ts.Close()
2020-08-12 16:16:14 +00:00
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
s := NewWorkersService(mostRecentVersion, fmt.Sprintf("%s/updater", ts.URL), testFilePath, Options{})
2020-08-12 16:16:14 +00:00
v, err := s.Check()
require.NoError(t, err)
require.NotNil(t, v)
require.Empty(t, v.Version())
2020-08-12 16:16:14 +00:00
}
func TestForcedUpdateService(t *testing.T) {
ts := createServer()
defer ts.Close()
2020-08-12 16:16:14 +00:00
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
s := NewWorkersService("2020.8.5", fmt.Sprintf("%s/updater", ts.URL), testFilePath, Options{IsForced: true})
2020-08-12 16:16:14 +00:00
v, err := s.Check()
require.NoError(t, err)
require.Equal(t, v.Version(), mostRecentVersion)
2020-08-12 16:16:14 +00:00
require.NoError(t, v.Apply())
2020-08-12 16:16:14 +00:00
dat, err := ioutil.ReadFile(testFilePath)
require.NoError(t, err)
2020-08-12 16:16:14 +00:00
require.Equal(t, string(dat), mostRecentVersion)
2020-08-12 16:16:14 +00:00
}
func TestUpdateSpecificVersionService(t *testing.T) {
ts := createServer()
defer ts.Close()
2020-08-12 16:16:14 +00:00
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
reqVersion := "2020.9.1"
s := NewWorkersService("2020.8.2", fmt.Sprintf("%s/updater", ts.URL), testFilePath, Options{RequestedVersion: reqVersion})
2020-08-12 16:16:14 +00:00
v, err := s.Check()
require.NoError(t, err)
require.Equal(t, reqVersion, v.Version())
2020-08-12 16:16:14 +00:00
require.NoError(t, v.Apply())
2020-08-12 16:16:14 +00:00
dat, err := ioutil.ReadFile(testFilePath)
require.NoError(t, err)
2020-08-12 16:16:14 +00:00
require.Equal(t, reqVersion, string(dat))
2020-08-12 16:16:14 +00:00
}
func TestCompressedUpdateService(t *testing.T) {
ts := createServer()
defer ts.Close()
2020-08-12 16:16:14 +00:00
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
s := NewWorkersService("2020.8.2", fmt.Sprintf("%s/compressed", ts.URL), testFilePath, Options{})
2020-08-12 16:16:14 +00:00
v, err := s.Check()
require.NoError(t, err)
require.Equal(t, "2020.09.02", v.Version())
2020-08-12 16:16:14 +00:00
require.NoError(t, v.Apply())
2020-08-12 16:16:14 +00:00
dat, err := ioutil.ReadFile(testFilePath)
require.NoError(t, err)
2020-08-12 16:16:14 +00:00
require.Equal(t, "2020.09.02", string(dat))
2020-08-12 16:16:14 +00:00
}
func TestUpdateWhenRunningKnownBuggyVersion(t *testing.T) {
ts := createServer()
defer ts.Close()
testFilePath := "tmpfile"
createTestFile(t, testFilePath)
defer os.Remove(testFilePath)
s := NewWorkersService(knownBuggyVersion, fmt.Sprintf("%s/updater", ts.URL), testFilePath, Options{})
v, err := s.Check()
require.NoError(t, err)
require.Equal(t, v.Version(), mostRecentVersion)
require.Equal(t, v.UserMessage(), expectedUserMsg)
}
2020-08-12 16:16:14 +00:00
func TestVersionParsing(t *testing.T) {
require.False(t, IsNewerVersion("2020.8.2", "2020.8.2"))
require.True(t, IsNewerVersion("2020.8.2", "2020.8.3"))
require.True(t, IsNewerVersion("2020.8.2", "2021.1.2"))
require.True(t, IsNewerVersion("2020.8.2", "2020.9.1"))
require.True(t, IsNewerVersion("2020.8.2", "2020.12.45"))
require.False(t, IsNewerVersion("2020.8.2", "2020.6.3"))
require.False(t, IsNewerVersion("DEV", "2020.8.5"))
require.False(t, IsNewerVersion("2020.8.2", "asdlkfjasdf"))
require.True(t, IsNewerVersion("3.0.1", "4.2.1"))
2020-08-12 16:16:14 +00:00
}