blob: 664726cc9a2f7c13715c19fbf66f34a5b7958b09 [file] [log] [blame]
Patrick Williams02871c92021-02-01 20:57:19 -06001#!/usr/bin/env python3
2#
3# Build the required docker image to run package unit tests
4#
5# Script Variables:
6# DOCKER_IMG_NAME: <optional, the name of the docker image to generate>
7# default is openbmc/ubuntu-unit-test
8# DISTRO: <optional, the distro to build a docker image against>
Patrick Williams50837432021-02-06 12:24:05 -06009# FORCE_DOCKER_BUILD: <optional, a non-zero value with force all Docker
10# images to be rebuilt rather than reusing caches.>
11# BUILD_URL: <optional, used to detect running under CI context
12# (ex. Jenkins)>
Patrick Williams02871c92021-02-01 20:57:19 -060013# BRANCH: <optional, branch to build from each of the openbmc/
14# repositories>
15# default is master, which will be used if input branch not
16# provided or not found
17# UBUNTU_MIRROR: <optional, the URL of a mirror of Ubuntu to override the
18# default ones in /etc/apt/sources.list>
19# default is empty, and no mirror is used.
20# http_proxy The HTTP address of the proxy server to connect to.
21# Default: "", proxy is not setup if this is not set
22
23import os
24import sys
Patrick Williamsb16f3e22021-02-06 08:16:47 -060025import threading
Patrick Williamsa18d9c52021-02-05 09:52:26 -060026from datetime import date
27from hashlib import sha256
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060028from typing import Any, Callable, Dict, Iterable, Optional
Patrick Williams02871c92021-02-01 20:57:19 -060029
Patrick Williams41d86212022-11-25 18:28:43 -060030from sh import docker, git, nproc, uname # type: ignore
31
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060032try:
33 # Python before 3.8 doesn't have TypedDict, so reroute to standard 'dict'.
34 from typing import TypedDict
Patrick Williams41d86212022-11-25 18:28:43 -060035except Exception:
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060036
37 class TypedDict(dict): # type: ignore
38 # We need to do this to eat the 'total' argument.
Patrick Williams41d86212022-11-25 18:28:43 -060039 def __init_subclass__(cls, **kwargs: Any) -> None:
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060040 super().__init_subclass__()
41
42
43# Declare some variables used in package definitions.
Patrick Williamsaae36d12021-02-04 16:30:04 -060044prefix = "/usr/local"
Patrick Williams02871c92021-02-01 20:57:19 -060045proc_count = nproc().strip()
Patrick Williams02871c92021-02-01 20:57:19 -060046
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060047
48class PackageDef(TypedDict, total=False):
Patrick Williams05fb2a02022-10-11 17:22:33 -050049 """Package Definition for packages dictionary."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060050
51 # rev [optional]: Revision of package to use.
52 rev: str
53 # url [optional]: lambda function to create URL: (package, rev) -> url.
54 url: Callable[[str, str], str]
55 # depends [optional]: List of package dependencies.
56 depends: Iterable[str]
57 # build_type [required]: Build type used for package.
58 # Currently supported: autoconf, cmake, custom, make, meson
59 build_type: str
60 # build_steps [optional]: Steps to run for 'custom' build_type.
61 build_steps: Iterable[str]
62 # config_flags [optional]: List of options to pass configuration tool.
63 config_flags: Iterable[str]
64 # config_env [optional]: List of environment variables to set for config.
65 config_env: Iterable[str]
66 # custom_post_dl [optional]: List of steps to run after download, but
67 # before config / build / install.
68 custom_post_dl: Iterable[str]
Patrick Williams6bce2ca2021-02-12 21:13:37 -060069 # custom_post_install [optional]: List of steps to run after install.
70 custom_post_install: Iterable[str]
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060071
72 # __tag [private]: Generated Docker tag name for package stage.
73 __tag: str
74 # __package [private]: Package object associated with this package.
75 __package: Any # Type is Package, but not defined yet.
76
Patrick Williams02871c92021-02-01 20:57:19 -060077
Patrick Williams72043242021-02-02 10:31:45 -060078# Packages to include in image.
79packages = {
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060080 "boost": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -050081 rev="1.80.0",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060082 url=(
Patrick Williams5f2549e2021-04-12 12:05:10 -050083 lambda pkg, rev: f"https://downloads.yoctoproject.org/mirror/sources/{pkg}_{rev.replace('.', '_')}.tar.bz2"
Patrick Williams2abc4a42021-02-03 06:11:40 -060084 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060085 build_type="custom",
86 build_steps=[
Patrick Williamsaae36d12021-02-04 16:30:04 -060087 f"./bootstrap.sh --prefix={prefix} --with-libraries=context,coroutine",
88 "./b2",
89 f"./b2 install --prefix={prefix}",
90 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060091 ),
92 "USCiLab/cereal": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -050093 rev="v1.3.2",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060094 build_type="custom",
95 build_steps=[f"cp -a include/cereal/ {prefix}/include/"],
96 ),
Ed Tanousc7198552022-07-01 08:15:50 -070097 "danmar/cppcheck": PackageDef(
Patrick Williamsbe4bd082022-10-03 08:59:12 -050098 rev="2.9",
Ed Tanousc7198552022-07-01 08:15:50 -070099 build_type="cmake",
100 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600101 "CLIUtils/CLI11": PackageDef(
102 rev="v1.9.1",
103 build_type="cmake",
104 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600105 "-DBUILD_TESTING=OFF",
106 "-DCLI11_BUILD_DOCS=OFF",
107 "-DCLI11_BUILD_EXAMPLES=OFF",
108 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600109 ),
110 "fmtlib/fmt": PackageDef(
William A. Kennington III652d8ae2022-10-02 17:01:16 -0700111 rev="9.1.0",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600112 build_type="cmake",
113 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600114 "-DFMT_DOC=OFF",
115 "-DFMT_TEST=OFF",
116 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600117 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600118 "Naios/function2": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500119 rev="4.2.1",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600120 build_type="custom",
121 build_steps=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600122 f"mkdir {prefix}/include/function2",
123 f"cp include/function2/function2.hpp {prefix}/include/function2/",
124 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600125 ),
Patrick Williamsed9414e2022-09-08 11:23:01 -0500126 # release-1.12.1
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600127 "google/googletest": PackageDef(
Patrick Williamsed9414e2022-09-08 11:23:01 -0500128 rev="58d77fa8070e8cec2dc1ed015d66b454c8d78850",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600129 build_type="cmake",
William A. Kennington III4dd32c02021-05-28 01:58:13 -0700130 config_env=["CXXFLAGS=-std=c++20"],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600131 config_flags=["-DTHREADS_PREFER_PTHREAD_FLAG=ON"],
132 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600133 "nlohmann/json": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500134 rev="v3.11.2",
Patrick Williams6bce2ca2021-02-12 21:13:37 -0600135 build_type="cmake",
136 config_flags=["-DJSON_BuildTests=OFF"],
137 custom_post_install=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600138 f"ln -s {prefix}/include/nlohmann/json.hpp {prefix}/include/json.hpp",
139 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600140 ),
Patrick Williams02871c92021-02-01 20:57:19 -0600141 # Snapshot from 2019-05-24
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600142 "linux-test-project/lcov": PackageDef(
143 rev="v1.15",
144 build_type="make",
145 ),
Patrick Williams001055b2022-11-28 07:58:27 -0600146 # dev-6.0 2022-11-28
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600147 "openbmc/linux": PackageDef(
Patrick Williams001055b2022-11-28 07:58:27 -0600148 rev="1b16243b004ce4d977a9f3b9d9e715cf5028f867",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600149 build_type="custom",
150 build_steps=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600151 f"make -j{proc_count} defconfig",
152 f"make INSTALL_HDR_PATH={prefix} headers_install",
153 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600154 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600155 "LibVNC/libvncserver": PackageDef(
156 rev="LibVNCServer-0.9.13",
157 build_type="cmake",
158 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600159 "leethomason/tinyxml2": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500160 rev="9.0.0",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600161 build_type="cmake",
162 ),
Patrick Williams02871c92021-02-01 20:57:19 -0600163 # version from /meta-openembedded/meta-oe/recipes-devtools/boost-url/boost-url_git.bb
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600164 "CPPAlliance/url": PackageDef(
Ed Tanousab640cd2022-02-03 16:33:27 -0800165 rev="d740a92d38e3a8f4d5b2153f53b82f1c98e312ab",
Ed Tanouseed466e2021-09-20 10:04:55 -0700166 build_type="custom",
167 build_steps=[f"cp -a include/** {prefix}/include/"],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600168 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600169 "tristanpenman/valijson": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500170 rev="v0.7",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600171 build_type="cmake",
172 config_flags=[
Patrick Williams0eedeed2021-02-06 19:06:09 -0600173 "-Dvalijson_BUILD_TESTS=0",
174 "-Dvalijson_INSTALL_HEADERS=1",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600175 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600176 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600177 "open-power/pdbg": PackageDef(build_type="autoconf"),
178 "openbmc/gpioplus": PackageDef(
179 depends=["openbmc/stdplus"],
180 build_type="meson",
181 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600182 "-Dexamples=false",
183 "-Dtests=disabled",
184 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600185 ),
186 "openbmc/phosphor-dbus-interfaces": PackageDef(
187 depends=["openbmc/sdbusplus"],
188 build_type="meson",
William A. Kennington III4fe87772022-02-11 15:44:29 -0800189 config_flags=["-Dgenerate_md=false"],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600190 ),
191 "openbmc/phosphor-logging": PackageDef(
192 depends=[
Patrick Williams83394612021-02-03 07:12:50 -0600193 "USCiLab/cereal",
Patrick Williams83394612021-02-03 07:12:50 -0600194 "openbmc/phosphor-dbus-interfaces",
195 "openbmc/sdbusplus",
196 "openbmc/sdeventplus",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600197 ],
Patrick Williamsf79ce4c2021-04-30 16:00:49 -0500198 build_type="meson",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600199 config_flags=[
William A. Kennington III6c98f282022-10-05 13:37:04 -0700200 "-Dlibonly=true",
201 "-Dtests=disabled",
Patrick Williams5eabdae2022-04-14 14:34:34 -0500202 f"-Dyamldir={prefix}/share/phosphor-dbus-yaml/yaml",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600203 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600204 ),
205 "openbmc/phosphor-objmgr": PackageDef(
206 depends=[
Brad Bishop11e57622022-09-14 16:10:25 -0400207 "CLIUtils/CLI11",
Patrick Williams70af95c2022-09-27 16:55:41 -0500208 "boost",
Patrick Williams83394612021-02-03 07:12:50 -0600209 "leethomason/tinyxml2",
Patrick Williams70af95c2022-09-27 16:55:41 -0500210 "openbmc/phosphor-dbus-interfaces",
Patrick Williams83394612021-02-03 07:12:50 -0600211 "openbmc/phosphor-logging",
212 "openbmc/sdbusplus",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600213 ],
Brad Bishop1197e352021-08-03 19:25:46 -0400214 build_type="meson",
215 config_flags=[
216 "-Dtests=disabled",
217 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600218 ),
Manojkiran Eda1c19e452022-10-03 11:01:59 +0530219 "openbmc/libpldm": PackageDef(
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600220 build_type="meson",
221 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600222 "-Doem-ibm=enabled",
223 "-Dtests=disabled",
224 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600225 ),
226 "openbmc/sdbusplus": PackageDef(
227 build_type="meson",
228 custom_post_dl=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600229 "cd tools",
230 f"./setup.py install --root=/ --prefix={prefix}",
231 "cd ..",
232 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600233 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600234 "-Dexamples=disabled",
235 "-Dtests=disabled",
236 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600237 ),
238 "openbmc/sdeventplus": PackageDef(
Patrick Williams70af95c2022-09-27 16:55:41 -0500239 depends=[
240 "Naios/function2",
241 "openbmc/stdplus",
242 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600243 build_type="meson",
244 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600245 "-Dexamples=false",
246 "-Dtests=disabled",
247 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600248 ),
249 "openbmc/stdplus": PackageDef(
Patrick Williams70af95c2022-09-27 16:55:41 -0500250 depends=[
Patrick Williams70af95c2022-09-27 16:55:41 -0500251 "fmtlib/fmt",
William A. Kennington IIIca1bf0c2022-10-05 02:23:30 -0700252 "google/googletest",
253 "Naios/function2",
Patrick Williams70af95c2022-09-27 16:55:41 -0500254 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600255 build_type="meson",
256 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600257 "-Dexamples=false",
258 "-Dtests=disabled",
William A. Kennington IIIca1bf0c2022-10-05 02:23:30 -0700259 "-Dgtest=enabled",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600260 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600261 ),
262} # type: Dict[str, PackageDef]
Patrick Williams02871c92021-02-01 20:57:19 -0600263
264# Define common flags used for builds
Patrick Williams02871c92021-02-01 20:57:19 -0600265configure_flags = " ".join(
266 [
267 f"--prefix={prefix}",
268 ]
269)
270cmake_flags = " ".join(
271 [
Patrick Williams02871c92021-02-01 20:57:19 -0600272 "-DBUILD_SHARED_LIBS=ON",
Patrick Williams0f2086b2021-02-05 06:49:49 -0600273 "-DCMAKE_BUILD_TYPE=RelWithDebInfo",
Patrick Williams02871c92021-02-01 20:57:19 -0600274 f"-DCMAKE_INSTALL_PREFIX:PATH={prefix}",
Patrick Williams0f2086b2021-02-05 06:49:49 -0600275 "-GNinja",
276 "-DCMAKE_MAKE_PROGRAM=ninja",
Patrick Williams02871c92021-02-01 20:57:19 -0600277 ]
278)
279meson_flags = " ".join(
280 [
281 "--wrap-mode=nodownload",
282 f"-Dprefix={prefix}",
283 ]
284)
285
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600286
287class Package(threading.Thread):
288 """Class used to build the Docker stages for each package.
289
290 Generally, this class should not be instantiated directly but through
291 Package.generate_all().
292 """
293
294 # Copy the packages dictionary.
295 packages = packages.copy()
296
297 # Lock used for thread-safety.
298 lock = threading.Lock()
299
300 def __init__(self, pkg: str):
Patrick Williams05fb2a02022-10-11 17:22:33 -0500301 """pkg - The name of this package (ex. foo/bar )"""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600302 super(Package, self).__init__()
303
304 self.package = pkg
305 self.exception = None # type: Optional[Exception]
306
307 # Reference to this package's
308 self.pkg_def = Package.packages[pkg]
309 self.pkg_def["__package"] = self
310
311 def run(self) -> None:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500312 """Thread 'run' function. Builds the Docker stage."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600313
314 # In case this package has no rev, fetch it from Github.
315 self._update_rev()
316
317 # Find all the Package objects that this package depends on.
318 # This section is locked because we are looking into another
319 # package's PackageDef dict, which could be being modified.
320 Package.lock.acquire()
321 deps: Iterable[Package] = [
322 Package.packages[deppkg]["__package"]
323 for deppkg in self.pkg_def.get("depends", [])
324 ]
325 Package.lock.release()
326
327 # Wait until all the depends finish building. We need them complete
328 # for the "COPY" commands.
329 for deppkg in deps:
330 deppkg.join()
331
332 # Generate this package's Dockerfile.
333 dockerfile = f"""
334FROM {docker_base_img_name}
335{self._df_copycmds()}
336{self._df_build()}
337"""
338
339 # Generate the resulting tag name and save it to the PackageDef.
340 # This section is locked because we are modifying the PackageDef,
341 # which can be accessed by other threads.
342 Package.lock.acquire()
343 tag = Docker.tagname(self._stagename(), dockerfile)
344 self.pkg_def["__tag"] = tag
345 Package.lock.release()
346
347 # Do the build / save any exceptions.
348 try:
349 Docker.build(self.package, tag, dockerfile)
350 except Exception as e:
351 self.exception = e
352
353 @classmethod
354 def generate_all(cls) -> None:
355 """Ensure a Docker stage is created for all defined packages.
356
357 These are done in parallel but with appropriate blocking per
358 package 'depends' specifications.
359 """
360
361 # Create a Package for each defined package.
362 pkg_threads = [Package(p) for p in cls.packages.keys()]
363
364 # Start building them all.
Patrick Williams6dbd7802021-02-20 08:34:10 -0600365 # This section is locked because threads depend on each other,
366 # based on the packages, and they cannot 'join' on a thread
367 # which is not yet started. Adding a lock here allows all the
368 # threads to start before they 'join' their dependencies.
369 Package.lock.acquire()
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600370 for t in pkg_threads:
371 t.start()
Patrick Williams6dbd7802021-02-20 08:34:10 -0600372 Package.lock.release()
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600373
374 # Wait for completion.
375 for t in pkg_threads:
376 t.join()
377 # Check if the thread saved off its own exception.
378 if t.exception:
379 print(f"Package {t.package} failed!", file=sys.stderr)
380 raise t.exception
381
382 @staticmethod
383 def df_all_copycmds() -> str:
384 """Formulate the Dockerfile snippet necessary to copy all packages
385 into the final image.
386 """
387 return Package.df_copycmds_set(Package.packages.keys())
388
389 @classmethod
390 def depcache(cls) -> str:
391 """Create the contents of the '/tmp/depcache'.
392 This file is a comma-separated list of "<pkg>:<rev>".
393 """
394
395 # This needs to be sorted for consistency.
396 depcache = ""
397 for pkg in sorted(cls.packages.keys()):
398 depcache += "%s:%s," % (pkg, cls.packages[pkg]["rev"])
399 return depcache
400
401 def _update_rev(self) -> None:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500402 """Look up the HEAD for missing a static rev."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600403
404 if "rev" in self.pkg_def:
405 return
406
Patrick Williams65b21fb2021-02-12 21:21:14 -0600407 # Check if Jenkins/Gerrit gave us a revision and use it.
408 if gerrit_project == self.package and gerrit_rev:
409 print(
410 f"Found Gerrit revision for {self.package}: {gerrit_rev}",
411 file=sys.stderr,
412 )
413 self.pkg_def["rev"] = gerrit_rev
414 return
415
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600416 # Ask Github for all the branches.
Patrick Williams05fb2a02022-10-11 17:22:33 -0500417 lookup = git(
418 "ls-remote", "--heads", f"https://github.com/{self.package}"
419 )
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600420
421 # Find the branch matching {branch} (or fallback to master).
422 # This section is locked because we are modifying the PackageDef.
423 Package.lock.acquire()
424 for line in lookup.split("\n"):
425 if f"refs/heads/{branch}" in line:
426 self.pkg_def["rev"] = line.split()[0]
Patrick Williamsc7d73642022-10-11 17:22:06 -0500427 elif (
428 "refs/heads/master" in line or "refs/heads/main" in line
429 ) and "rev" not in self.pkg_def:
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600430 self.pkg_def["rev"] = line.split()[0]
431 Package.lock.release()
432
433 def _stagename(self) -> str:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500434 """Create a name for the Docker stage associated with this pkg."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600435 return self.package.replace("/", "-").lower()
436
437 def _url(self) -> str:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500438 """Get the URL for this package."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600439 rev = self.pkg_def["rev"]
440
441 # If the lambda exists, call it.
442 if "url" in self.pkg_def:
443 return self.pkg_def["url"](self.package, rev)
444
445 # Default to the github archive URL.
446 return f"https://github.com/{self.package}/archive/{rev}.tar.gz"
447
448 def _cmd_download(self) -> str:
449 """Formulate the command necessary to download and unpack to source."""
450
451 url = self._url()
452 if ".tar." not in url:
453 raise NotImplementedError(
454 f"Unhandled download type for {self.package}: {url}"
455 )
456
457 cmd = f"curl -L {url} | tar -x"
458
459 if url.endswith(".bz2"):
460 cmd += "j"
461 elif url.endswith(".gz"):
462 cmd += "z"
463 else:
464 raise NotImplementedError(
465 f"Unknown tar flags needed for {self.package}: {url}"
466 )
467
468 return cmd
469
470 def _cmd_cd_srcdir(self) -> str:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500471 """Formulate the command necessary to 'cd' into the source dir."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600472 return f"cd {self.package.split('/')[-1]}*"
473
474 def _df_copycmds(self) -> str:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500475 """Formulate the dockerfile snippet necessary to COPY all depends."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600476
477 if "depends" not in self.pkg_def:
478 return ""
479 return Package.df_copycmds_set(self.pkg_def["depends"])
480
481 @staticmethod
482 def df_copycmds_set(pkgs: Iterable[str]) -> str:
483 """Formulate the Dockerfile snippet necessary to COPY a set of
484 packages into a Docker stage.
485 """
486
487 copy_cmds = ""
488
489 # Sort the packages for consistency.
490 for p in sorted(pkgs):
491 tag = Package.packages[p]["__tag"]
492 copy_cmds += f"COPY --from={tag} {prefix} {prefix}\n"
493 # Workaround for upstream docker bug and multiple COPY cmds
494 # https://github.com/moby/moby/issues/37965
495 copy_cmds += "RUN true\n"
496
497 return copy_cmds
498
499 def _df_build(self) -> str:
500 """Formulate the Dockerfile snippet necessary to download, build, and
501 install a package into a Docker stage.
502 """
503
504 # Download and extract source.
505 result = f"RUN {self._cmd_download()} && {self._cmd_cd_srcdir()} && "
506
507 # Handle 'custom_post_dl' commands.
508 custom_post_dl = self.pkg_def.get("custom_post_dl")
509 if custom_post_dl:
510 result += " && ".join(custom_post_dl) + " && "
511
512 # Build and install package based on 'build_type'.
513 build_type = self.pkg_def["build_type"]
514 if build_type == "autoconf":
515 result += self._cmd_build_autoconf()
516 elif build_type == "cmake":
517 result += self._cmd_build_cmake()
518 elif build_type == "custom":
519 result += self._cmd_build_custom()
520 elif build_type == "make":
521 result += self._cmd_build_make()
522 elif build_type == "meson":
523 result += self._cmd_build_meson()
524 else:
525 raise NotImplementedError(
526 f"Unhandled build type for {self.package}: {build_type}"
527 )
528
Patrick Williams6bce2ca2021-02-12 21:13:37 -0600529 # Handle 'custom_post_install' commands.
530 custom_post_install = self.pkg_def.get("custom_post_install")
531 if custom_post_install:
532 result += " && " + " && ".join(custom_post_install)
533
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600534 return result
535
536 def _cmd_build_autoconf(self) -> str:
537 options = " ".join(self.pkg_def.get("config_flags", []))
538 env = " ".join(self.pkg_def.get("config_env", []))
539 result = "./bootstrap.sh && "
540 result += f"{env} ./configure {configure_flags} {options} && "
541 result += f"make -j{proc_count} && make install"
542 return result
543
544 def _cmd_build_cmake(self) -> str:
545 options = " ".join(self.pkg_def.get("config_flags", []))
546 env = " ".join(self.pkg_def.get("config_env", []))
547 result = "mkdir builddir && cd builddir && "
548 result += f"{env} cmake {cmake_flags} {options} .. && "
549 result += "cmake --build . --target all && "
550 result += "cmake --build . --target install && "
551 result += "cd .."
552 return result
553
554 def _cmd_build_custom(self) -> str:
555 return " && ".join(self.pkg_def.get("build_steps", []))
556
557 def _cmd_build_make(self) -> str:
558 return f"make -j{proc_count} && make install"
559
560 def _cmd_build_meson(self) -> str:
561 options = " ".join(self.pkg_def.get("config_flags", []))
562 env = " ".join(self.pkg_def.get("config_env", []))
563 result = f"{env} meson builddir {meson_flags} {options} && "
564 result += "ninja -C builddir && ninja -C builddir install"
565 return result
566
567
568class Docker:
569 """Class to assist with Docker interactions. All methods are static."""
570
571 @staticmethod
572 def timestamp() -> str:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500573 """Generate a timestamp for today using the ISO week."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600574 today = date.today().isocalendar()
575 return f"{today[0]}-W{today[1]:02}"
576
577 @staticmethod
Patrick Williams41d86212022-11-25 18:28:43 -0600578 def tagname(pkgname: Optional[str], dockerfile: str) -> str:
Patrick Williams05fb2a02022-10-11 17:22:33 -0500579 """Generate a tag name for a package using a hash of the Dockerfile."""
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600580 result = docker_image_name
581 if pkgname:
582 result += "-" + pkgname
583
584 result += ":" + Docker.timestamp()
585 result += "-" + sha256(dockerfile.encode()).hexdigest()[0:16]
586
587 return result
588
589 @staticmethod
590 def build(pkg: str, tag: str, dockerfile: str) -> None:
591 """Build a docker image using the Dockerfile and tagging it with 'tag'."""
592
593 # If we're not forcing builds, check if it already exists and skip.
594 if not force_build:
595 if docker.image.ls(tag, "--format", '"{{.Repository}}:{{.Tag}}"'):
Patrick Williams05fb2a02022-10-11 17:22:33 -0500596 print(
597 f"Image {tag} already exists. Skipping.", file=sys.stderr
598 )
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600599 return
600
601 # Build it.
602 # Capture the output of the 'docker build' command and send it to
603 # stderr (prefixed with the package name). This allows us to see
604 # progress but not polute stdout. Later on we output the final
605 # docker tag to stdout and we want to keep that pristine.
606 #
607 # Other unusual flags:
608 # --no-cache: Bypass the Docker cache if 'force_build'.
609 # --force-rm: Clean up Docker processes if they fail.
610 docker.build(
611 proxy_args,
612 "--network=host",
613 "--force-rm",
614 "--no-cache=true" if force_build else "--no-cache=false",
615 "-t",
616 tag,
617 "-",
618 _in=dockerfile,
619 _out=(
620 lambda line: print(
621 pkg + ":", line, end="", file=sys.stderr, flush=True
622 )
623 ),
624 )
625
626
627# Read a bunch of environment variables.
Patrick Williams05fb2a02022-10-11 17:22:33 -0500628docker_image_name = os.environ.get(
629 "DOCKER_IMAGE_NAME", "openbmc/ubuntu-unit-test"
630)
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600631force_build = os.environ.get("FORCE_DOCKER_BUILD")
632is_automated_ci_build = os.environ.get("BUILD_URL", False)
Patrick Williams5b08dc62022-09-27 16:36:57 -0500633distro = os.environ.get("DISTRO", "ubuntu:kinetic")
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600634branch = os.environ.get("BRANCH", "master")
635ubuntu_mirror = os.environ.get("UBUNTU_MIRROR")
636http_proxy = os.environ.get("http_proxy")
637
Patrick Williams65b21fb2021-02-12 21:21:14 -0600638gerrit_project = os.environ.get("GERRIT_PROJECT")
639gerrit_rev = os.environ.get("GERRIT_PATCHSET_REVISION")
640
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600641# Set up some common variables.
642username = os.environ.get("USER", "root")
643homedir = os.environ.get("HOME", "/root")
644gid = os.getgid()
645uid = os.getuid()
646
Josh Lehan6825a012022-03-17 18:31:39 -0700647# Use well-known constants if user is root
648if username == "root":
649 homedir = "/root"
650 gid = 0
651 uid = 0
652
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600653# Determine the architecture for Docker.
654arch = uname("-m").strip()
655if arch == "ppc64le":
656 docker_base = "ppc64le/"
657elif arch == "x86_64":
658 docker_base = ""
Thang Q. Nguyen051b05b2021-12-10 08:30:35 +0000659elif arch == "aarch64":
Thang Q. Nguyenf98f1a82021-12-22 01:59:19 +0000660 docker_base = "arm64v8/"
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600661else:
662 print(
663 f"Unsupported system architecture({arch}) found for docker image",
664 file=sys.stderr,
665 )
666 sys.exit(1)
667
Patrick Williams02871c92021-02-01 20:57:19 -0600668# Special flags if setting up a deb mirror.
669mirror = ""
670if "ubuntu" in distro and ubuntu_mirror:
671 mirror = f"""
672RUN echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME) main restricted universe multiverse" > /etc/apt/sources.list && \\
673 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-updates main restricted universe multiverse" >> /etc/apt/sources.list && \\
674 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-security main restricted universe multiverse" >> /etc/apt/sources.list && \\
675 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-proposed main restricted universe multiverse" >> /etc/apt/sources.list && \\
676 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-backports main restricted universe multiverse" >> /etc/apt/sources.list
677"""
678
679# Special flags for proxying.
680proxy_cmd = ""
Adrian Ambrożewicz34ec77e2021-06-02 10:23:38 +0200681proxy_keyserver = ""
Patrick Williams02871c92021-02-01 20:57:19 -0600682proxy_args = []
683if http_proxy:
684 proxy_cmd = f"""
685RUN echo "[http]" >> {homedir}/.gitconfig && \
686 echo "proxy = {http_proxy}" >> {homedir}/.gitconfig
687"""
Adrian Ambrożewicz34ec77e2021-06-02 10:23:38 +0200688 proxy_keyserver = f"--keyserver-options http-proxy={http_proxy}"
689
Patrick Williams02871c92021-02-01 20:57:19 -0600690 proxy_args.extend(
691 [
692 "--build-arg",
693 f"http_proxy={http_proxy}",
694 "--build-arg",
Lei YUd461cd62021-02-18 14:25:49 +0800695 f"https_proxy={http_proxy}",
Patrick Williams02871c92021-02-01 20:57:19 -0600696 ]
697 )
698
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600699# Create base Dockerfile.
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600700dockerfile_base = f"""
701FROM {docker_base}{distro}
Patrick Williams02871c92021-02-01 20:57:19 -0600702
703{mirror}
704
705ENV DEBIAN_FRONTEND noninteractive
706
Patrick Williams8949d3c2022-04-27 16:41:27 -0500707ENV PYTHONPATH "/usr/local/lib/python3.10/site-packages/"
Patrick Williams02871c92021-02-01 20:57:19 -0600708
Patrick Williamsbb16ac12021-04-12 12:23:51 -0500709# Sometimes the ubuntu key expires and we need a way to force an execution
710# of the apt-get commands for the dbgsym-keyring. When this happens we see
711# an error like: "Release: The following signatures were invalid:"
712# Insert a bogus echo that we can change here when we get this error to force
713# the update.
714RUN echo "ubuntu keyserver rev as of 2021-04-21"
715
Patrick Williams02871c92021-02-01 20:57:19 -0600716# We need the keys to be imported for dbgsym repos
717# New releases have a package, older ones fall back to manual fetching
718# https://wiki.ubuntu.com/Debug%20Symbol%20Packages
Patrick Williams50837432021-02-06 12:24:05 -0600719RUN apt-get update && apt-get dist-upgrade -yy && \
Patrick Williamsf79ce4c2021-04-30 16:00:49 -0500720 ( apt-get install gpgv ubuntu-dbgsym-keyring || \
Patrick Williams50837432021-02-06 12:24:05 -0600721 ( apt-get install -yy dirmngr && \
722 apt-key adv --keyserver keyserver.ubuntu.com \
Adrian Ambrożewicz34ec77e2021-06-02 10:23:38 +0200723 {proxy_keyserver} \
Patrick Williams50837432021-02-06 12:24:05 -0600724 --recv-keys F2EDC64DC5AEE1F6B9C621F0C8CAB6595FDFF622 ) )
Patrick Williams02871c92021-02-01 20:57:19 -0600725
726# Parse the current repo list into a debug repo list
727RUN sed -n '/^deb /s,^deb [^ ]* ,deb http://ddebs.ubuntu.com ,p' /etc/apt/sources.list >/etc/apt/sources.list.d/debug.list
728
729# Remove non-existent debug repos
Patrick Williams41d86212022-11-25 18:28:43 -0600730RUN sed -i '/-\\(backports\\|security\\) /d' /etc/apt/sources.list.d/debug.list
Patrick Williams02871c92021-02-01 20:57:19 -0600731
732RUN cat /etc/apt/sources.list.d/debug.list
733
734RUN apt-get update && apt-get dist-upgrade -yy && apt-get install -yy \
Patrick Williams274e3a92022-09-08 07:08:31 -0500735 gcc-12 \
736 g++-12 \
Patrick Williams02871c92021-02-01 20:57:19 -0600737 libc6-dbg \
738 libc6-dev \
739 libtool \
740 bison \
741 libdbus-1-dev \
742 flex \
743 cmake \
744 python3 \
745 python3-dev\
746 python3-yaml \
747 python3-mako \
748 python3-pip \
749 python3-setuptools \
750 python3-git \
751 python3-socks \
752 pkg-config \
753 autoconf \
754 autoconf-archive \
755 libsystemd-dev \
756 systemd \
757 libssl-dev \
758 libevdev-dev \
Patrick Williams02871c92021-02-01 20:57:19 -0600759 libjpeg-dev \
760 libpng-dev \
761 ninja-build \
762 sudo \
763 curl \
764 git \
765 dbus \
766 iputils-ping \
Patrick Williams27a646b2022-09-27 16:57:44 -0500767 clang-15 \
768 clang-format-15 \
769 clang-tidy-15 \
770 clang-tools-15 \
Patrick Williams02871c92021-02-01 20:57:19 -0600771 shellcheck \
772 npm \
773 iproute2 \
774 libnl-3-dev \
775 libnl-genl-3-dev \
776 libconfig++-dev \
777 libsnmp-dev \
778 valgrind \
779 valgrind-dbg \
780 libpam0g-dev \
781 xxd \
782 libi2c-dev \
783 wget \
784 libldap2-dev \
785 libprotobuf-dev \
William A. Kennington IIIdafe7522021-05-28 01:51:06 -0700786 liburing-dev \
Patrick Williams8949d3c2022-04-27 16:41:27 -0500787 liburing2-dbgsym \
Patrick Williams02871c92021-02-01 20:57:19 -0600788 libperlio-gzip-perl \
789 libjson-perl \
790 protobuf-compiler \
791 libgpiod-dev \
792 device-tree-compiler \
Patrick Williams02871c92021-02-01 20:57:19 -0600793 libpciaccess-dev \
794 libmimetic-dev \
795 libxml2-utils \
Patrick Williams0eedeed2021-02-06 19:06:09 -0600796 libxml-simple-perl \
John Wedig9adf68d2021-11-16 14:00:39 -0800797 rsync \
798 libcryptsetup-dev
Patrick Williams02871c92021-02-01 20:57:19 -0600799
Manojkiran Eda87111bb2021-08-14 11:26:16 +0530800RUN npm install -g eslint@latest eslint-plugin-json@latest
801
Patrick Williams5b08dc62022-09-27 16:36:57 -0500802# Kinetic comes with GCC-12, so skip this.
803#RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-12 12 \
804# --slave /usr/bin/g++ g++ /usr/bin/g++-12 \
805# --slave /usr/bin/gcov gcov /usr/bin/gcov-12 \
806# --slave /usr/bin/gcov-dump gcov-dump /usr/bin/gcov-dump-12 \
807# --slave /usr/bin/gcov-tool gcov-tool /usr/bin/gcov-tool-12
808#RUN update-alternatives --install /usr/bin/cpp cpp /usr/bin/cpp-12 12
Patrick Williams02871c92021-02-01 20:57:19 -0600809
Patrick Williams27a646b2022-09-27 16:57:44 -0500810RUN update-alternatives --install /usr/bin/clang clang /usr/bin/clang-15 1000 \
811 --slave /usr/bin/clang++ clang++ /usr/bin/clang++-15 \
812 --slave /usr/bin/clang-tidy clang-tidy /usr/bin/clang-tidy-15 \
813 --slave /usr/bin/clang-format clang-format /usr/bin/clang-format-15 \
814 --slave /usr/bin/run-clang-tidy run-clang-tidy.py /usr/bin/run-clang-tidy-15 \
815 --slave /usr/bin/scan-build scan-build /usr/bin/scan-build-15
Patrick Williams02871c92021-02-01 20:57:19 -0600816
Patrick Williams50837432021-02-06 12:24:05 -0600817"""
818
819if is_automated_ci_build:
820 dockerfile_base += f"""
821# Run an arbitrary command to polute the docker cache regularly force us
822# to re-run `apt-get update` daily.
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600823RUN echo {Docker.timestamp()}
Patrick Williams50837432021-02-06 12:24:05 -0600824RUN apt-get update && apt-get dist-upgrade -yy
825
826"""
827
Patrick Williams41d86212022-11-25 18:28:43 -0600828dockerfile_base += """
Patrick Williams02871c92021-02-01 20:57:19 -0600829RUN pip3 install inflection
830RUN pip3 install pycodestyle
831RUN pip3 install jsonschema
Michael Shenfb612a52022-08-05 00:58:31 +0000832RUN pip3 install meson==0.63.0
Ed Tanous5278efb2022-09-26 13:59:29 -0700833RUN pip3 install packaging
Patrick Williams02871c92021-02-01 20:57:19 -0600834RUN pip3 install protobuf
Manojkiran Edae6f120a2021-03-20 11:13:43 +0530835RUN pip3 install codespell
Ed Tanousca8c4a82022-02-08 13:58:22 -0800836RUN pip3 install requests
Patrick Williamsc2e7c932022-11-27 14:13:21 -0600837RUN pip3 install gitlint
Ed Tanousfb9948a2022-06-21 09:10:24 -0700838"""
839
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600840# Build the base and stage docker images.
841docker_base_img_name = Docker.tagname("base", dockerfile_base)
842Docker.build("base", docker_base_img_name, dockerfile_base)
843Package.generate_all()
Patrick Williams02871c92021-02-01 20:57:19 -0600844
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600845# Create the final Dockerfile.
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600846dockerfile = f"""
Patrick Williams02871c92021-02-01 20:57:19 -0600847# Build the final output image
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600848FROM {docker_base_img_name}
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600849{Package.df_all_copycmds()}
Patrick Williams02871c92021-02-01 20:57:19 -0600850
851# Some of our infrastructure still relies on the presence of this file
852# even though it is no longer needed to rebuild the docker environment
853# NOTE: The file is sorted to ensure the ordering is stable.
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600854RUN echo '{Package.depcache()}' > /tmp/depcache
Patrick Williams02871c92021-02-01 20:57:19 -0600855
856# Final configuration for the workspace
Josh Lehan6825a012022-03-17 18:31:39 -0700857RUN grep -q {gid} /etc/group || groupadd -f -g {gid} {username}
Patrick Williams02871c92021-02-01 20:57:19 -0600858RUN mkdir -p "{os.path.dirname(homedir)}"
859RUN grep -q {uid} /etc/passwd || useradd -d {homedir} -m -u {uid} -g {gid} {username}
860RUN sed -i '1iDefaults umask=000' /etc/sudoers
861RUN echo "{username} ALL=(ALL) NOPASSWD: ALL" >>/etc/sudoers
862
Andrew Geissler305a9a52021-04-07 11:08:40 -0500863# Ensure user has ability to write to /usr/local for different tool
864# and data installs
Andrew Geissler7bb00b12021-05-10 15:12:08 -0500865RUN chown -R {username}:{username} /usr/local/share
Andrew Geissler305a9a52021-04-07 11:08:40 -0500866
Patrick Williams02871c92021-02-01 20:57:19 -0600867{proxy_cmd}
868
869RUN /bin/bash
870"""
871
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600872# Do the final docker build
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600873docker_final_img_name = Docker.tagname(None, dockerfile)
874Docker.build("final", docker_final_img_name, dockerfile)
875
Patrick Williams00536fb2021-02-11 14:28:49 -0600876# Print the tag of the final image.
877print(docker_final_img_name)