Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 24 additions & 12 deletions cmd/containerd/server/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -92,22 +92,34 @@ func CreateTopLevelDirectories(config *srvconfig.Config) error {
if err := os.Chmod(config.TempDir, 0o700); err != nil && !errors.Is(err, os.ErrPermission) {
return err
}
if runtime.GOOS == "windows" {
// On Windows, the Host Compute Service (vmcompute) will read the
// TEMP/TMP setting from the calling process when creating the
// tempdir to extract an image layer to. This allows the
// administrator to align the tempdir location with the same volume
// as the snapshot dir to avoid a copy operation when moving the
// extracted layer to the snapshot dir location.
os.Setenv("TEMP", config.TempDir)
os.Setenv("TMP", config.TempDir)
} else {
os.Setenv("TMPDIR", config.TempDir)
}
setTempDirEnv(config.TempDir)
}
return nil
}

// setTempDirEnv points the process' temp-directory environment variables at
// tempDir so that components (and the OS) honor the configured temp location.
func setTempDirEnv(tempDir string) {
if runtime.GOOS == "windows" {
// On Windows, the Host Compute Service (vmcompute) will read the
// TEMP/TMP setting from the calling process when creating the
// tempdir to extract an image layer to. This allows the
// administrator to align the tempdir location with the same volume
// as the snapshot dir to avoid a copy operation when moving the
// extracted layer to the snapshot dir location.
os.Setenv("TEMP", tempDir)
os.Setenv("TMP", tempDir)
// Since Go 1.21, os.MkdirTemp/os.TempDir resolve the temp dir via
// Windows' GetTempPath2W. For processes running as SYSTEM (as
// containerd does under the SCM), that API reads SystemTemp rather
// than TEMP/TMP, so set it too to keep the override effective.
// https://cs.opensource.google/go/go/+/refs/tags/go1.21.0:src/os/file_windows.go
os.Setenv("SystemTemp", tempDir)
} else {
os.Setenv("TMPDIR", tempDir)
}
}

// New creates and initializes a new containerd server
func New(ctx context.Context, config *srvconfig.Config) (*Server, error) {
if err := apply(ctx, config); err != nil {
Expand Down
23 changes: 23 additions & 0 deletions cmd/containerd/server/server_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ import (
"context"
"iter"
"os"
"runtime"
"slices"
"testing"

Expand Down Expand Up @@ -172,3 +173,25 @@ func TestMigration(t *testing.T) {
t.Fatal(err)
}
}

func TestSetTempDirEnv(t *testing.T) {
const tempDir = "/tmp/path/for/testing/temp"

var keys []string
if runtime.GOOS == "windows" {
keys = []string{"TEMP", "TMP", "SystemTemp"}
} else {
keys = []string{"TMPDIR"}
}
for _, k := range keys {
t.Setenv(k, "")
}

setTempDirEnv(tempDir)

for _, k := range keys {
if got := os.Getenv(k); got != tempDir {
t.Errorf("expected %s=%q, got %q", k, tempDir, got)
}
}
}
3 changes: 2 additions & 1 deletion internal/cri/nri/nri_api_linux.go
Original file line number Diff line number Diff line change
Expand Up @@ -682,7 +682,8 @@ func (p *criPodSandbox) GetPid() uint32 {
}

func (p *criPodSandbox) GetIPs() []string {
if p.IP == "" {
// IP and AdditionalIPs are promoted fields from the embedded Sandbox struct.
if p.Sandbox == nil || p.IP == "" {
return nil
}
ips := append([]string{p.IP}, p.AdditionalIPs...)
Expand Down
81 changes: 81 additions & 0 deletions internal/cri/nri/nri_api_linux_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
/*
Copyright The containerd Authors.

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 nri

import (
"testing"

sstore "github.com/containerd/containerd/v2/internal/cri/store/sandbox"
"github.com/stretchr/testify/assert"
)

func TestCRIPodSandboxGetIPs(t *testing.T) {
testCases := []struct {
name string
pod *criPodSandbox
expected []string
}{
{
name: "nil Sandbox",
pod: &criPodSandbox{
Sandbox: nil,
},
expected: nil,
},
{
name: "empty primary IP",
pod: &criPodSandbox{
Sandbox: &sstore.Sandbox{
Metadata: sstore.Metadata{
IP: "",
},
},
},
expected: nil,
},
{
name: "single primary IP",
pod: &criPodSandbox{
Sandbox: &sstore.Sandbox{
Metadata: sstore.Metadata{
IP: "10.0.0.1",
},
},
},
expected: []string{"10.0.0.1"},
},
{
name: "primary and additional IPs",
pod: &criPodSandbox{
Sandbox: &sstore.Sandbox{
Metadata: sstore.Metadata{
IP: "10.0.0.1",
AdditionalIPs: []string{"fd00::1"},
},
},
},
expected: []string{"10.0.0.1", "fd00::1"},
},
}

for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
ips := tc.pod.GetIPs()
assert.Equal(t, tc.expected, ips)
})
}
}
Loading