Smail/backend/config/ssl_export_test.go
smail 1790e1f4f6
Some checks are pending
Security Checks / security (push) Waiting to run
修复只读证书同步并拒绝硬链接私钥
2026-09-20 04:53:18 +08:00

122 lines
4.0 KiB
Go

package config
import (
"bytes"
"crypto/tls"
"os"
"path/filepath"
"runtime"
"testing"
"time"
)
func TestExportProjectTLSKeepsOldBundleOnInvalidInput(t *testing.T) {
if runtime.GOOS != "linux" {
if err := ExportProjectTLS(t.TempDir(), filepath.Join(t.TempDir(), "tls.pem")); err == nil {
t.Fatal("unsupported platform allowed exporting a private key")
}
t.Skip("Linux Nginx export requires POSIX permission checks")
}
root := t.TempDir()
outputDirectory := t.TempDir()
if err := os.Chmod(outputDirectory, 0700); err != nil {
t.Fatal(err)
}
destination := filepath.Join(outputDirectory, "tls.pem")
runtime := validRuntimeConfig()
runtime.Root.Domains = DomainConfig{Primary: "example.com", Web: "web.example.com", Accepted: []string{"example.com"}}
key, certificate := testManagedCertificate(t, "example.com", time.Now().Add(-time.Hour), time.Now().Add(time.Hour), 2048, "*.example.com")
ref, err := stageFlatSSLAtRoot(root, ManagedSSLModeDefault, key, certificate, nil, nil)
if err != nil {
t.Fatal(err)
}
defer releaseManagedSSL(root, SSLConfig{CertificateRef: ref.CertificateRef})
runtime.SSL.TLS = SSLConfig{Management: SSLManagementProject, Challenge: SSLChallengeDNS, Mode: ManagedSSLModeDefault, DefaultDirectory: "ssl", CertificateRef: ref.CertificateRef, PrivateKeyRef: ref.PrivateKeyRef}
writeRuntimeFixture(t, root, runtime)
if err := ExportProjectTLS(root, destination); err != nil {
t.Fatal(err)
}
bundle, err := os.ReadFile(destination)
if err != nil {
t.Fatal(err)
}
if _, err := tls.X509KeyPair(bundle, bundle); err != nil {
t.Fatal(err)
}
before, err := os.Stat(destination)
if err != nil || before.Mode().Perm() != 0600 {
t.Fatal("export must be private")
}
if err := ExportProjectTLS(root, destination); err != nil {
t.Fatal(err)
}
after, _ := os.Stat(destination)
if !os.SameFile(before, after) {
t.Fatal("unchanged certificate must not be rewritten")
}
for _, target := range []string{ref.PrivateKeyPath, ref.CertificatePath, destination} {
if err := os.Chmod(target, 0644); err != nil {
t.Fatal(err)
}
if err := ExportProjectTLS(root, destination); err == nil {
t.Fatal("unsafe file permissions accepted")
}
if err := os.Chmod(target, 0600); err != nil {
t.Fatal(err)
}
}
if err := os.Chmod(outputDirectory, 0777); err != nil {
t.Fatal(err)
}
if err := ExportProjectTLS(root, destination); err == nil {
t.Fatal("writable output directory accepted")
}
if err := os.Chmod(outputDirectory, 0700); err != nil {
t.Fatal(err)
}
link := filepath.Join(t.TempDir(), "linked")
if err := os.Symlink(outputDirectory, link); err != nil {
t.Fatal(err)
}
if err := ExportProjectTLS(root, filepath.Join(link, "tls.pem")); err == nil {
t.Fatal("linked output directory accepted")
}
hardLink := filepath.Join(t.TempDir(), "linked.key")
if err := os.Link(ref.PrivateKeyPath, hardLink); err != nil {
t.Fatal(err)
}
if err := ExportProjectTLS(root, destination); err == nil {
t.Fatal("hard-linked private key accepted")
}
if err := os.Remove(hardLink); err != nil {
t.Fatal(err)
}
if err := os.WriteFile(ref.PrivateKeyPath, []byte("invalid"), 0600); err != nil {
t.Fatal(err)
}
if err := ExportProjectTLS(root, destination); err == nil {
t.Fatal("invalid key was exported")
}
preserved, _ := os.ReadFile(destination)
if !bytes.Equal(bundle, preserved) {
t.Fatal("failed export replaced valid bundle")
}
runtime.SSL.TLS.CertificateRef = "ssl/unmanaged.crt"
runtime.SSL.TLS.PrivateKeyRef = "ssl/unmanaged.key"
writeRuntimeFixture(t, root, runtime)
if err := ExportProjectTLS(root, destination); err == nil {
t.Fatal("unmanaged input accepted")
}
for _, mode := range []string{SSLManagementCustom, SSLManagementWebServer} {
runtime.SSL.TLS.Management = mode
writeRuntimeFixture(t, root, runtime)
if err := ExportProjectTLS(root, destination); err != nil {
t.Fatal(err)
}
preserved, _ := os.ReadFile(destination)
if !bytes.Equal(bundle, preserved) {
t.Fatal("external mode changed exported bundle")
}
}
}