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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
#!/usr/bin/env python3
# Copyright 2020 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Chroot helper functions."""
import os
from pathlib import Path
import subprocess
from typing import Iterable, List, Union
def InChroot() -> bool:
"""Returns True if currently in the chroot."""
return "CROS_WORKON_SRCROOT" in os.environ
def VerifyInsideChroot() -> None:
"""Checks whether the script invoked was executed in the chroot.
Raises:
AssertionError: The script was run outside the chroot.
"""
assert InChroot(), "Script should be run inside the chroot."
def VerifyOutsideChroot() -> None:
"""Checks whether the script invoked was executed in the chroot.
Raises:
AssertionError: The script was run inside the chroot.
"""
assert not InChroot(), "Script should be run outside the chroot."
def VerifyChromeOSRoot(chromeos_root: Union[Path, str]) -> None:
"""Checks whether the path actually points to ChromiumOS checkout root.
Raises:
AssertionError: The path is not ChromiumOS checkout root.
"""
subdir = "src/third_party/chromiumos-overlay"
path = Path(chromeos_root).expanduser() / subdir
msg = f"Wrong ChromeOS path. No {subdir} directory in {chromeos_root} ."
assert path.is_dir(), msg
def FindChromeOSRootAbove(chromeos_tree_path: Path) -> Path:
"""Returns the root of a ChromeOS tree, given a path in said tree.
May return `chromeos_tree_path`, if that's already the root of the tree.
Raises:
ValueError if the given path is not in a ChromeOS tree.
"""
if (chromeos_tree_path / ".repo").exists():
return chromeos_tree_path
for parent in chromeos_tree_path.parents:
if (parent / ".repo").exists():
return parent
raise ValueError(f"{chromeos_tree_path} is not in a repo checkout")
def GetChrootEbuildPaths(
chromeos_root: Union[Path, str],
packages: Iterable[str],
chroot_name: str = "chroot",
out_dir: str = "out",
) -> List[str]:
"""Gets the chroot path(s) of the package(s).
Args:
chromeos_root: The absolute path to the chromeos tree to use.
packages: A list of a package/packages to
be used to find their chroot path.
chroot_name: name of the chroot to enter.
out_dir: name of the out directory for the chroot.
Returns:
A list of chroot paths of the packages' ebuild files.
Raises:
ValueError: Failed to get the chroot path of a package.
"""
chroot_paths = []
cros_sdk = [
"cros_sdk",
f"--chroot={chroot_name}",
f"--out-dir={out_dir}",
]
# Find the chroot path for each package's ebuild.
for package in packages:
chroot_path = subprocess.check_output(
cros_sdk + ["--", "equery", "w", package],
cwd=chromeos_root,
encoding="utf-8",
)
chroot_paths.append(chroot_path.strip())
return chroot_paths
def ConvertChrootPathsToAbsolutePaths(
chromeos_root: str,
chroot_paths: List[str],
) -> List[str]:
"""Converts the chroot path(s) to absolute symlink path(s).
Args:
chromeos_root: The absolute path to the chroot.
chroot_paths: A list of chroot paths to convert to absolute paths.
Returns:
A list of absolute path(s).
Raises:
ValueError: Invalid prefix for the chroot path or
invalid chroot paths were provided.
"""
abs_paths = []
chroot_prefix = "/mnt/host/source/"
# Iterate through the chroot paths.
# For each chroot file path, remove '/mnt/host/source/' prefix
# and combine the chroot path with the result and add it to the list.
for chroot_path in chroot_paths:
if not chroot_path.startswith(chroot_prefix):
raise ValueError(
"Invalid prefix for the chroot path: %s" % chroot_path
)
rel_path = chroot_path[len(chroot_prefix) :]
# combine the chromeos root path + '/src/...'
abs_path = os.path.join(chromeos_root, rel_path)
abs_paths.append(abs_path)
return abs_paths
|
