snapshot of https://github.com/livekit/livekit.git @ 08c5433449c106943231037d0b892895f6e92d3b (2026-09-15)

pkg/service/utils_test.go

@@ -0,0 +1,129 @@
+// Copyright 2023 LiveKit, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package service_test
+
+import (
+	"bytes"
+	"compress/gzip"
+	"context"
+	"net/http"
+	"testing"
+	"time"
+
+	"github.com/redis/go-redis/v9"
+	"github.com/stretchr/testify/require"
+
+	"github.com/livekit/livekit-server/pkg/service"
+)
+
+func redisClientDocker(t testing.TB) *redis.Client {
+	addr := runRedis(t)
+	cli := redis.NewClient(&redis.Options{
+		Addr: addr,
+	})
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	if err := cli.Ping(ctx).Err(); err != nil {
+		_ = cli.Close()
+		t.Fatal(err)
+	}
+	t.Cleanup(func() {
+		_ = cli.Close()
+	})
+	return cli
+}
+
+func redisClient(t testing.TB) *redis.Client {
+	cli := redis.NewClient(&redis.Options{
+		Addr: "localhost:6379",
+	})
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	err := cli.Ping(ctx).Err()
+	if err == nil {
+		t.Cleanup(func() {
+			_ = cli.Close()
+		})
+		return cli
+	}
+	_ = cli.Close()
+	t.Logf("local redis not available: %v", err)
+
+	t.Logf("starting redis in docker")
+	return redisClientDocker(t)
+}
+
+func TestIsValidDomain(t *testing.T) {
+	list := map[string]bool{
+		"turn.myhost.com":  true,
+		"turn.google.com":  true,
+		"https://host.com": false,
+		"turn://host.com":  false,
+	}
+	for key, result := range list {
+		service.IsValidDomain(key)
+		require.Equal(t, service.IsValidDomain(key), result)
+	}
+}
+
+func compress(t *testing.T, payload []byte) []byte {
+	t.Helper()
+	var buf bytes.Buffer
+	gw, err := gzip.NewWriterLevel(&buf, gzip.BestCompression)
+	require.NoError(t, err)
+	_, err = gw.Write(payload)
+	require.NoError(t, err)
+	require.NoError(t, gw.Close())
+	return buf.Bytes()
+}
+
+func TestDecompressGzip(t *testing.T) {
+	t.Run("small payload", func(t *testing.T) {
+		out, err := service.DecompressGzip(compress(t, []byte("hello world")))
+		require.NoError(t, err)
+		require.Equal(t, []byte("hello world"), out)
+	})
+
+	t.Run("payload exactly at cap", func(t *testing.T) {
+		raw := make([]byte, http.DefaultMaxHeaderBytes)
+		out, err := service.DecompressGzip(compress(t, raw))
+		require.NoError(t, err)
+		require.Len(t, out, http.DefaultMaxHeaderBytes)
+	})
+
+	t.Run("payload one byte over capd", func(t *testing.T) {
+		raw := make([]byte, http.DefaultMaxHeaderBytes+1)
+		_, err := service.DecompressGzip(compress(t, raw))
+		require.ErrorIs(t, err, service.ErrGzipTooLarge)
+	})
+
+	t.Run("gzip decompression bomb", func(t *testing.T) {
+		// 100 MB of zeros
+		raw := make([]byte, 100<<20)
+		compressed := compress(t, raw)
+		require.Less(t, len(compressed), 1<<20,
+			"sanity: bomb input should compress dramatically")
+
+		_, err := service.DecompressGzip(compressed)
+		require.ErrorIs(t, err, service.ErrGzipTooLarge)
+	})
+
+	t.Run("malformed gzip compression", func(t *testing.T) {
+		_, err := service.DecompressGzip([]byte("not gzip data"))
+		require.Error(t, err)
+		require.Contains(t, err.Error(), "cannot read decompressed")
+	})
+}