File: namespace_utils_unittest.cc

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (72 lines) | stat: -rw-r--r-- 1,995 bytes parent folder | download | duplicates (11)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "sandbox/linux/services/namespace_utils.h"

#include <errno.h>
#include <sched.h>
#include <sys/types.h>
#include <sys/wait.h>

#include "base/check.h"
#include "base/posix/eintr_wrapper.h"
#include "base/process/launch.h"
#include "sandbox/linux/services/credentials.h"
#include "sandbox/linux/tests/unit_tests.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace sandbox {

namespace {

SANDBOX_TEST(NamespaceUtils, KernelSupportsUnprivilegedNamespace) {
  const bool can_create_user_ns = Credentials::CanCreateProcessInNewUserNS();
  const bool supports_user_ns =
      NamespaceUtils::KernelSupportsUnprivilegedNamespace(CLONE_NEWUSER);
  // can_create_user_ns implies supports_user_ns, but the converse is not
  // necessarily true, as creating a user namespace can fail for various
  // reasons.
  if (can_create_user_ns) {
    SANDBOX_ASSERT(supports_user_ns);
  }
}

SANDBOX_TEST(NamespaceUtils, WriteToIdMapFile) {
  if (!Credentials::CanCreateProcessInNewUserNS()) {
    return;
  }

  const uid_t uid = getuid();
  const gid_t gid = getgid();

  const bool supports_deny_setgroups =
      NamespaceUtils::KernelSupportsDenySetgroups();

  const pid_t pid =
      base::ForkWithFlags(CLONE_NEWUSER | SIGCHLD, nullptr, nullptr);
  ASSERT_NE(-1, pid);
  if (pid == 0) {
    if (supports_deny_setgroups) {
      RAW_CHECK(NamespaceUtils::DenySetgroups());
    }

    RAW_CHECK(getuid() != uid);
    RAW_CHECK(NamespaceUtils::WriteToIdMapFile("/proc/self/uid_map", uid));
    RAW_CHECK(getuid() == uid);

    RAW_CHECK(getgid() != gid);
    RAW_CHECK(NamespaceUtils::WriteToIdMapFile("/proc/self/gid_map", gid));
    RAW_CHECK(getgid() == gid);

    _exit(0);
  }

  int status = 42;
  SANDBOX_ASSERT_EQ(pid, HANDLE_EINTR(waitpid(pid, &status, 0)));
  SANDBOX_ASSERT_EQ(0, status);
}

}  // namespace.

}  // namespace sandbox.