blob: b0b4a8b960cc33bbc84d76eb4a45e01565b066b3 [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 sh import docker, git, nproc, uname # type: ignore
29from typing import Any, Callable, Dict, Iterable, Optional
Patrick Williams02871c92021-02-01 20:57:19 -060030
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060031try:
32 # Python before 3.8 doesn't have TypedDict, so reroute to standard 'dict'.
33 from typing import TypedDict
34except:
35
36 class TypedDict(dict): # type: ignore
37 # We need to do this to eat the 'total' argument.
38 def __init_subclass__(cls, **kwargs):
39 super().__init_subclass__()
40
41
42# Declare some variables used in package definitions.
Patrick Williamsaae36d12021-02-04 16:30:04 -060043prefix = "/usr/local"
Patrick Williams02871c92021-02-01 20:57:19 -060044proc_count = nproc().strip()
Patrick Williams02871c92021-02-01 20:57:19 -060045
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060046
47class PackageDef(TypedDict, total=False):
48 """ Package Definition for packages dictionary. """
49
50 # rev [optional]: Revision of package to use.
51 rev: str
52 # url [optional]: lambda function to create URL: (package, rev) -> url.
53 url: Callable[[str, str], str]
54 # depends [optional]: List of package dependencies.
55 depends: Iterable[str]
56 # build_type [required]: Build type used for package.
57 # Currently supported: autoconf, cmake, custom, make, meson
58 build_type: str
59 # build_steps [optional]: Steps to run for 'custom' build_type.
60 build_steps: Iterable[str]
61 # config_flags [optional]: List of options to pass configuration tool.
62 config_flags: Iterable[str]
63 # config_env [optional]: List of environment variables to set for config.
64 config_env: Iterable[str]
65 # custom_post_dl [optional]: List of steps to run after download, but
66 # before config / build / install.
67 custom_post_dl: Iterable[str]
Patrick Williams6bce2ca2021-02-12 21:13:37 -060068 # custom_post_install [optional]: List of steps to run after install.
69 custom_post_install: Iterable[str]
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060070
71 # __tag [private]: Generated Docker tag name for package stage.
72 __tag: str
73 # __package [private]: Package object associated with this package.
74 __package: Any # Type is Package, but not defined yet.
75
Patrick Williams02871c92021-02-01 20:57:19 -060076
Patrick Williams72043242021-02-02 10:31:45 -060077# Packages to include in image.
78packages = {
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060079 "boost": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -050080 rev="1.80.0",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060081 url=(
Patrick Williams5f2549e2021-04-12 12:05:10 -050082 lambda pkg, rev: f"https://downloads.yoctoproject.org/mirror/sources/{pkg}_{rev.replace('.', '_')}.tar.bz2"
Patrick Williams2abc4a42021-02-03 06:11:40 -060083 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060084 build_type="custom",
85 build_steps=[
Patrick Williamsaae36d12021-02-04 16:30:04 -060086 f"./bootstrap.sh --prefix={prefix} --with-libraries=context,coroutine",
87 "./b2",
88 f"./b2 install --prefix={prefix}",
89 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060090 ),
91 "USCiLab/cereal": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -050092 rev="v1.3.2",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -060093 build_type="custom",
94 build_steps=[f"cp -a include/cereal/ {prefix}/include/"],
95 ),
Ed Tanousc7198552022-07-01 08:15:50 -070096 "danmar/cppcheck": PackageDef(
Patrick Williamsbe4bd082022-10-03 08:59:12 -050097 rev="2.9",
Ed Tanousc7198552022-07-01 08:15:50 -070098 build_type="cmake",
99 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600100 "CLIUtils/CLI11": PackageDef(
101 rev="v1.9.1",
102 build_type="cmake",
103 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600104 "-DBUILD_TESTING=OFF",
105 "-DCLI11_BUILD_DOCS=OFF",
106 "-DCLI11_BUILD_EXAMPLES=OFF",
107 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600108 ),
109 "fmtlib/fmt": PackageDef(
William A. Kennington III652d8ae2022-10-02 17:01:16 -0700110 rev="9.1.0",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600111 build_type="cmake",
112 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600113 "-DFMT_DOC=OFF",
114 "-DFMT_TEST=OFF",
115 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600116 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600117 "Naios/function2": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500118 rev="4.2.1",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600119 build_type="custom",
120 build_steps=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600121 f"mkdir {prefix}/include/function2",
122 f"cp include/function2/function2.hpp {prefix}/include/function2/",
123 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600124 ),
Patrick Williamsed9414e2022-09-08 11:23:01 -0500125 # release-1.12.1
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600126 "google/googletest": PackageDef(
Patrick Williamsed9414e2022-09-08 11:23:01 -0500127 rev="58d77fa8070e8cec2dc1ed015d66b454c8d78850",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600128 build_type="cmake",
William A. Kennington III4dd32c02021-05-28 01:58:13 -0700129 config_env=["CXXFLAGS=-std=c++20"],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600130 config_flags=["-DTHREADS_PREFER_PTHREAD_FLAG=ON"],
131 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600132 "nlohmann/json": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500133 rev="v3.11.2",
Patrick Williams6bce2ca2021-02-12 21:13:37 -0600134 build_type="cmake",
135 config_flags=["-DJSON_BuildTests=OFF"],
136 custom_post_install=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600137 f"ln -s {prefix}/include/nlohmann/json.hpp {prefix}/include/json.hpp",
138 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600139 ),
Patrick Williams02871c92021-02-01 20:57:19 -0600140 # Snapshot from 2019-05-24
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600141 "linux-test-project/lcov": PackageDef(
142 rev="v1.15",
143 build_type="make",
144 ),
Patrick Williamsc1977832022-09-27 16:54:34 -0500145 # dev-5.15 2022-09-27
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600146 "openbmc/linux": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500147 rev="c9fb275212dac5b300311f6f6b1dcc5ed18a3493",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600148 build_type="custom",
149 build_steps=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600150 f"make -j{proc_count} defconfig",
151 f"make INSTALL_HDR_PATH={prefix} headers_install",
152 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600153 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600154 "LibVNC/libvncserver": PackageDef(
155 rev="LibVNCServer-0.9.13",
156 build_type="cmake",
157 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600158 "leethomason/tinyxml2": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500159 rev="9.0.0",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600160 build_type="cmake",
161 ),
Patrick Williams02871c92021-02-01 20:57:19 -0600162 # version from /meta-openembedded/meta-oe/recipes-devtools/boost-url/boost-url_git.bb
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600163 "CPPAlliance/url": PackageDef(
Ed Tanousab640cd2022-02-03 16:33:27 -0800164 rev="d740a92d38e3a8f4d5b2153f53b82f1c98e312ab",
Ed Tanouseed466e2021-09-20 10:04:55 -0700165 build_type="custom",
166 build_steps=[f"cp -a include/** {prefix}/include/"],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600167 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600168 "tristanpenman/valijson": PackageDef(
Patrick Williamsc1977832022-09-27 16:54:34 -0500169 rev="v0.7",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600170 build_type="cmake",
171 config_flags=[
Patrick Williams0eedeed2021-02-06 19:06:09 -0600172 "-Dvalijson_BUILD_TESTS=0",
173 "-Dvalijson_INSTALL_HEADERS=1",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600174 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600175 ),
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600176 "open-power/pdbg": PackageDef(build_type="autoconf"),
177 "openbmc/gpioplus": PackageDef(
178 depends=["openbmc/stdplus"],
179 build_type="meson",
180 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600181 "-Dexamples=false",
182 "-Dtests=disabled",
183 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600184 ),
185 "openbmc/phosphor-dbus-interfaces": PackageDef(
186 depends=["openbmc/sdbusplus"],
187 build_type="meson",
William A. Kennington III4fe87772022-02-11 15:44:29 -0800188 config_flags=["-Dgenerate_md=false"],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600189 ),
190 "openbmc/phosphor-logging": PackageDef(
191 depends=[
Patrick Williams83394612021-02-03 07:12:50 -0600192 "USCiLab/cereal",
Patrick Williams83394612021-02-03 07:12:50 -0600193 "openbmc/phosphor-dbus-interfaces",
194 "openbmc/sdbusplus",
195 "openbmc/sdeventplus",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600196 ],
Patrick Williamsf79ce4c2021-04-30 16:00:49 -0500197 build_type="meson",
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600198 config_flags=[
Patrick Williams5eabdae2022-04-14 14:34:34 -0500199 f"-Dyamldir={prefix}/share/phosphor-dbus-yaml/yaml",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600200 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600201 ),
202 "openbmc/phosphor-objmgr": PackageDef(
203 depends=[
Brad Bishop11e57622022-09-14 16:10:25 -0400204 "CLIUtils/CLI11",
Patrick Williams70af95c2022-09-27 16:55:41 -0500205 "boost",
Patrick Williams83394612021-02-03 07:12:50 -0600206 "leethomason/tinyxml2",
Patrick Williams70af95c2022-09-27 16:55:41 -0500207 "openbmc/phosphor-dbus-interfaces",
Patrick Williams83394612021-02-03 07:12:50 -0600208 "openbmc/phosphor-logging",
209 "openbmc/sdbusplus",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600210 ],
Brad Bishop1197e352021-08-03 19:25:46 -0400211 build_type="meson",
212 config_flags=[
213 "-Dtests=disabled",
214 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600215 ),
216 "openbmc/pldm": PackageDef(
217 depends=[
Patrick Williams83394612021-02-03 07:12:50 -0600218 "CLIUtils/CLI11",
219 "boost",
220 "nlohmann/json",
221 "openbmc/phosphor-dbus-interfaces",
222 "openbmc/phosphor-logging",
223 "openbmc/sdbusplus",
224 "openbmc/sdeventplus",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600225 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600226 build_type="meson",
227 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600228 "-Dlibpldm-only=enabled",
229 "-Doem-ibm=enabled",
230 "-Dtests=disabled",
231 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600232 ),
233 "openbmc/sdbusplus": PackageDef(
234 build_type="meson",
235 custom_post_dl=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600236 "cd tools",
237 f"./setup.py install --root=/ --prefix={prefix}",
238 "cd ..",
239 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600240 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600241 "-Dexamples=disabled",
242 "-Dtests=disabled",
243 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600244 ),
245 "openbmc/sdeventplus": PackageDef(
Patrick Williams70af95c2022-09-27 16:55:41 -0500246 depends=[
247 "Naios/function2",
248 "openbmc/stdplus",
249 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600250 build_type="meson",
251 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600252 "-Dexamples=false",
253 "-Dtests=disabled",
254 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600255 ),
256 "openbmc/stdplus": PackageDef(
Patrick Williams70af95c2022-09-27 16:55:41 -0500257 depends=[
Patrick Williams70af95c2022-09-27 16:55:41 -0500258 "fmtlib/fmt",
William A. Kennington IIIca1bf0c2022-10-05 02:23:30 -0700259 "google/googletest",
260 "Naios/function2",
Patrick Williams70af95c2022-09-27 16:55:41 -0500261 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600262 build_type="meson",
263 config_flags=[
Patrick Williamsaae36d12021-02-04 16:30:04 -0600264 "-Dexamples=false",
265 "-Dtests=disabled",
William A. Kennington IIIca1bf0c2022-10-05 02:23:30 -0700266 "-Dgtest=enabled",
Patrick Williamsaae36d12021-02-04 16:30:04 -0600267 ],
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600268 ),
269} # type: Dict[str, PackageDef]
Patrick Williams02871c92021-02-01 20:57:19 -0600270
271# Define common flags used for builds
Patrick Williams02871c92021-02-01 20:57:19 -0600272configure_flags = " ".join(
273 [
274 f"--prefix={prefix}",
275 ]
276)
277cmake_flags = " ".join(
278 [
Patrick Williams02871c92021-02-01 20:57:19 -0600279 "-DBUILD_SHARED_LIBS=ON",
Patrick Williams0f2086b2021-02-05 06:49:49 -0600280 "-DCMAKE_BUILD_TYPE=RelWithDebInfo",
Patrick Williams02871c92021-02-01 20:57:19 -0600281 f"-DCMAKE_INSTALL_PREFIX:PATH={prefix}",
Patrick Williams0f2086b2021-02-05 06:49:49 -0600282 "-GNinja",
283 "-DCMAKE_MAKE_PROGRAM=ninja",
Patrick Williams02871c92021-02-01 20:57:19 -0600284 ]
285)
286meson_flags = " ".join(
287 [
288 "--wrap-mode=nodownload",
289 f"-Dprefix={prefix}",
290 ]
291)
292
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600293
294class Package(threading.Thread):
295 """Class used to build the Docker stages for each package.
296
297 Generally, this class should not be instantiated directly but through
298 Package.generate_all().
299 """
300
301 # Copy the packages dictionary.
302 packages = packages.copy()
303
304 # Lock used for thread-safety.
305 lock = threading.Lock()
306
307 def __init__(self, pkg: str):
308 """ pkg - The name of this package (ex. foo/bar ) """
309 super(Package, self).__init__()
310
311 self.package = pkg
312 self.exception = None # type: Optional[Exception]
313
314 # Reference to this package's
315 self.pkg_def = Package.packages[pkg]
316 self.pkg_def["__package"] = self
317
318 def run(self) -> None:
319 """ Thread 'run' function. Builds the Docker stage. """
320
321 # In case this package has no rev, fetch it from Github.
322 self._update_rev()
323
324 # Find all the Package objects that this package depends on.
325 # This section is locked because we are looking into another
326 # package's PackageDef dict, which could be being modified.
327 Package.lock.acquire()
328 deps: Iterable[Package] = [
329 Package.packages[deppkg]["__package"]
330 for deppkg in self.pkg_def.get("depends", [])
331 ]
332 Package.lock.release()
333
334 # Wait until all the depends finish building. We need them complete
335 # for the "COPY" commands.
336 for deppkg in deps:
337 deppkg.join()
338
339 # Generate this package's Dockerfile.
340 dockerfile = f"""
341FROM {docker_base_img_name}
342{self._df_copycmds()}
343{self._df_build()}
344"""
345
346 # Generate the resulting tag name and save it to the PackageDef.
347 # This section is locked because we are modifying the PackageDef,
348 # which can be accessed by other threads.
349 Package.lock.acquire()
350 tag = Docker.tagname(self._stagename(), dockerfile)
351 self.pkg_def["__tag"] = tag
352 Package.lock.release()
353
354 # Do the build / save any exceptions.
355 try:
356 Docker.build(self.package, tag, dockerfile)
357 except Exception as e:
358 self.exception = e
359
360 @classmethod
361 def generate_all(cls) -> None:
362 """Ensure a Docker stage is created for all defined packages.
363
364 These are done in parallel but with appropriate blocking per
365 package 'depends' specifications.
366 """
367
368 # Create a Package for each defined package.
369 pkg_threads = [Package(p) for p in cls.packages.keys()]
370
371 # Start building them all.
Patrick Williams6dbd7802021-02-20 08:34:10 -0600372 # This section is locked because threads depend on each other,
373 # based on the packages, and they cannot 'join' on a thread
374 # which is not yet started. Adding a lock here allows all the
375 # threads to start before they 'join' their dependencies.
376 Package.lock.acquire()
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600377 for t in pkg_threads:
378 t.start()
Patrick Williams6dbd7802021-02-20 08:34:10 -0600379 Package.lock.release()
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600380
381 # Wait for completion.
382 for t in pkg_threads:
383 t.join()
384 # Check if the thread saved off its own exception.
385 if t.exception:
386 print(f"Package {t.package} failed!", file=sys.stderr)
387 raise t.exception
388
389 @staticmethod
390 def df_all_copycmds() -> str:
391 """Formulate the Dockerfile snippet necessary to copy all packages
392 into the final image.
393 """
394 return Package.df_copycmds_set(Package.packages.keys())
395
396 @classmethod
397 def depcache(cls) -> str:
398 """Create the contents of the '/tmp/depcache'.
399 This file is a comma-separated list of "<pkg>:<rev>".
400 """
401
402 # This needs to be sorted for consistency.
403 depcache = ""
404 for pkg in sorted(cls.packages.keys()):
405 depcache += "%s:%s," % (pkg, cls.packages[pkg]["rev"])
406 return depcache
407
408 def _update_rev(self) -> None:
409 """ Look up the HEAD for missing a static rev. """
410
411 if "rev" in self.pkg_def:
412 return
413
Patrick Williams65b21fb2021-02-12 21:21:14 -0600414 # Check if Jenkins/Gerrit gave us a revision and use it.
415 if gerrit_project == self.package and gerrit_rev:
416 print(
417 f"Found Gerrit revision for {self.package}: {gerrit_rev}",
418 file=sys.stderr,
419 )
420 self.pkg_def["rev"] = gerrit_rev
421 return
422
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600423 # Ask Github for all the branches.
424 lookup = git("ls-remote", "--heads", f"https://github.com/{self.package}")
425
426 # Find the branch matching {branch} (or fallback to master).
427 # This section is locked because we are modifying the PackageDef.
428 Package.lock.acquire()
429 for line in lookup.split("\n"):
430 if f"refs/heads/{branch}" in line:
431 self.pkg_def["rev"] = line.split()[0]
432 elif f"refs/heads/master" in line and "rev" not in self.pkg_def:
433 self.pkg_def["rev"] = line.split()[0]
434 Package.lock.release()
435
436 def _stagename(self) -> str:
437 """ Create a name for the Docker stage associated with this pkg. """
438 return self.package.replace("/", "-").lower()
439
440 def _url(self) -> str:
441 """ Get the URL for this package. """
442 rev = self.pkg_def["rev"]
443
444 # If the lambda exists, call it.
445 if "url" in self.pkg_def:
446 return self.pkg_def["url"](self.package, rev)
447
448 # Default to the github archive URL.
449 return f"https://github.com/{self.package}/archive/{rev}.tar.gz"
450
451 def _cmd_download(self) -> str:
452 """Formulate the command necessary to download and unpack to source."""
453
454 url = self._url()
455 if ".tar." not in url:
456 raise NotImplementedError(
457 f"Unhandled download type for {self.package}: {url}"
458 )
459
460 cmd = f"curl -L {url} | tar -x"
461
462 if url.endswith(".bz2"):
463 cmd += "j"
464 elif url.endswith(".gz"):
465 cmd += "z"
466 else:
467 raise NotImplementedError(
468 f"Unknown tar flags needed for {self.package}: {url}"
469 )
470
471 return cmd
472
473 def _cmd_cd_srcdir(self) -> str:
474 """ Formulate the command necessary to 'cd' into the source dir. """
475 return f"cd {self.package.split('/')[-1]}*"
476
477 def _df_copycmds(self) -> str:
478 """ Formulate the dockerfile snippet necessary to COPY all depends. """
479
480 if "depends" not in self.pkg_def:
481 return ""
482 return Package.df_copycmds_set(self.pkg_def["depends"])
483
484 @staticmethod
485 def df_copycmds_set(pkgs: Iterable[str]) -> str:
486 """Formulate the Dockerfile snippet necessary to COPY a set of
487 packages into a Docker stage.
488 """
489
490 copy_cmds = ""
491
492 # Sort the packages for consistency.
493 for p in sorted(pkgs):
494 tag = Package.packages[p]["__tag"]
495 copy_cmds += f"COPY --from={tag} {prefix} {prefix}\n"
496 # Workaround for upstream docker bug and multiple COPY cmds
497 # https://github.com/moby/moby/issues/37965
498 copy_cmds += "RUN true\n"
499
500 return copy_cmds
501
502 def _df_build(self) -> str:
503 """Formulate the Dockerfile snippet necessary to download, build, and
504 install a package into a Docker stage.
505 """
506
507 # Download and extract source.
508 result = f"RUN {self._cmd_download()} && {self._cmd_cd_srcdir()} && "
509
510 # Handle 'custom_post_dl' commands.
511 custom_post_dl = self.pkg_def.get("custom_post_dl")
512 if custom_post_dl:
513 result += " && ".join(custom_post_dl) + " && "
514
515 # Build and install package based on 'build_type'.
516 build_type = self.pkg_def["build_type"]
517 if build_type == "autoconf":
518 result += self._cmd_build_autoconf()
519 elif build_type == "cmake":
520 result += self._cmd_build_cmake()
521 elif build_type == "custom":
522 result += self._cmd_build_custom()
523 elif build_type == "make":
524 result += self._cmd_build_make()
525 elif build_type == "meson":
526 result += self._cmd_build_meson()
527 else:
528 raise NotImplementedError(
529 f"Unhandled build type for {self.package}: {build_type}"
530 )
531
Patrick Williams6bce2ca2021-02-12 21:13:37 -0600532 # Handle 'custom_post_install' commands.
533 custom_post_install = self.pkg_def.get("custom_post_install")
534 if custom_post_install:
535 result += " && " + " && ".join(custom_post_install)
536
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600537 return result
538
539 def _cmd_build_autoconf(self) -> str:
540 options = " ".join(self.pkg_def.get("config_flags", []))
541 env = " ".join(self.pkg_def.get("config_env", []))
542 result = "./bootstrap.sh && "
543 result += f"{env} ./configure {configure_flags} {options} && "
544 result += f"make -j{proc_count} && make install"
545 return result
546
547 def _cmd_build_cmake(self) -> str:
548 options = " ".join(self.pkg_def.get("config_flags", []))
549 env = " ".join(self.pkg_def.get("config_env", []))
550 result = "mkdir builddir && cd builddir && "
551 result += f"{env} cmake {cmake_flags} {options} .. && "
552 result += "cmake --build . --target all && "
553 result += "cmake --build . --target install && "
554 result += "cd .."
555 return result
556
557 def _cmd_build_custom(self) -> str:
558 return " && ".join(self.pkg_def.get("build_steps", []))
559
560 def _cmd_build_make(self) -> str:
561 return f"make -j{proc_count} && make install"
562
563 def _cmd_build_meson(self) -> str:
564 options = " ".join(self.pkg_def.get("config_flags", []))
565 env = " ".join(self.pkg_def.get("config_env", []))
566 result = f"{env} meson builddir {meson_flags} {options} && "
567 result += "ninja -C builddir && ninja -C builddir install"
568 return result
569
570
571class Docker:
572 """Class to assist with Docker interactions. All methods are static."""
573
574 @staticmethod
575 def timestamp() -> str:
576 """ Generate a timestamp for today using the ISO week. """
577 today = date.today().isocalendar()
578 return f"{today[0]}-W{today[1]:02}"
579
580 @staticmethod
581 def tagname(pkgname: str, dockerfile: str) -> str:
582 """ Generate a tag name for a package using a hash of the Dockerfile. """
583 result = docker_image_name
584 if pkgname:
585 result += "-" + pkgname
586
587 result += ":" + Docker.timestamp()
588 result += "-" + sha256(dockerfile.encode()).hexdigest()[0:16]
589
590 return result
591
592 @staticmethod
593 def build(pkg: str, tag: str, dockerfile: str) -> None:
594 """Build a docker image using the Dockerfile and tagging it with 'tag'."""
595
596 # If we're not forcing builds, check if it already exists and skip.
597 if not force_build:
598 if docker.image.ls(tag, "--format", '"{{.Repository}}:{{.Tag}}"'):
599 print(f"Image {tag} already exists. Skipping.", file=sys.stderr)
600 return
601
602 # Build it.
603 # Capture the output of the 'docker build' command and send it to
604 # stderr (prefixed with the package name). This allows us to see
605 # progress but not polute stdout. Later on we output the final
606 # docker tag to stdout and we want to keep that pristine.
607 #
608 # Other unusual flags:
609 # --no-cache: Bypass the Docker cache if 'force_build'.
610 # --force-rm: Clean up Docker processes if they fail.
611 docker.build(
612 proxy_args,
613 "--network=host",
614 "--force-rm",
615 "--no-cache=true" if force_build else "--no-cache=false",
616 "-t",
617 tag,
618 "-",
619 _in=dockerfile,
620 _out=(
621 lambda line: print(
622 pkg + ":", line, end="", file=sys.stderr, flush=True
623 )
624 ),
625 )
626
627
628# Read a bunch of environment variables.
629docker_image_name = os.environ.get("DOCKER_IMAGE_NAME", "openbmc/ubuntu-unit-test")
630force_build = os.environ.get("FORCE_DOCKER_BUILD")
631is_automated_ci_build = os.environ.get("BUILD_URL", False)
Patrick Williams5b08dc62022-09-27 16:36:57 -0500632distro = os.environ.get("DISTRO", "ubuntu:kinetic")
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600633branch = os.environ.get("BRANCH", "master")
634ubuntu_mirror = os.environ.get("UBUNTU_MIRROR")
635http_proxy = os.environ.get("http_proxy")
636
Patrick Williams65b21fb2021-02-12 21:21:14 -0600637gerrit_project = os.environ.get("GERRIT_PROJECT")
638gerrit_rev = os.environ.get("GERRIT_PATCHSET_REVISION")
639
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600640# Set up some common variables.
641username = os.environ.get("USER", "root")
642homedir = os.environ.get("HOME", "/root")
643gid = os.getgid()
644uid = os.getuid()
645
Josh Lehan6825a012022-03-17 18:31:39 -0700646# Use well-known constants if user is root
647if username == "root":
648 homedir = "/root"
649 gid = 0
650 uid = 0
651
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600652# Determine the architecture for Docker.
653arch = uname("-m").strip()
654if arch == "ppc64le":
655 docker_base = "ppc64le/"
656elif arch == "x86_64":
657 docker_base = ""
Thang Q. Nguyen051b05b2021-12-10 08:30:35 +0000658elif arch == "aarch64":
Thang Q. Nguyenf98f1a82021-12-22 01:59:19 +0000659 docker_base = "arm64v8/"
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600660else:
661 print(
662 f"Unsupported system architecture({arch}) found for docker image",
663 file=sys.stderr,
664 )
665 sys.exit(1)
666
Patrick Williams02871c92021-02-01 20:57:19 -0600667# Special flags if setting up a deb mirror.
668mirror = ""
669if "ubuntu" in distro and ubuntu_mirror:
670 mirror = f"""
671RUN echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME) main restricted universe multiverse" > /etc/apt/sources.list && \\
672 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-updates main restricted universe multiverse" >> /etc/apt/sources.list && \\
673 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-security main restricted universe multiverse" >> /etc/apt/sources.list && \\
674 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-proposed main restricted universe multiverse" >> /etc/apt/sources.list && \\
675 echo "deb {ubuntu_mirror} $(. /etc/os-release && echo $VERSION_CODENAME)-backports main restricted universe multiverse" >> /etc/apt/sources.list
676"""
677
678# Special flags for proxying.
679proxy_cmd = ""
Adrian Ambrożewicz34ec77e2021-06-02 10:23:38 +0200680proxy_keyserver = ""
Patrick Williams02871c92021-02-01 20:57:19 -0600681proxy_args = []
682if http_proxy:
683 proxy_cmd = f"""
684RUN echo "[http]" >> {homedir}/.gitconfig && \
685 echo "proxy = {http_proxy}" >> {homedir}/.gitconfig
686"""
Adrian Ambrożewicz34ec77e2021-06-02 10:23:38 +0200687 proxy_keyserver = f"--keyserver-options http-proxy={http_proxy}"
688
Patrick Williams02871c92021-02-01 20:57:19 -0600689 proxy_args.extend(
690 [
691 "--build-arg",
692 f"http_proxy={http_proxy}",
693 "--build-arg",
Lei YUd461cd62021-02-18 14:25:49 +0800694 f"https_proxy={http_proxy}",
Patrick Williams02871c92021-02-01 20:57:19 -0600695 ]
696 )
697
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600698# Create base Dockerfile.
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600699dockerfile_base = f"""
700FROM {docker_base}{distro}
Patrick Williams02871c92021-02-01 20:57:19 -0600701
702{mirror}
703
704ENV DEBIAN_FRONTEND noninteractive
705
Patrick Williams8949d3c2022-04-27 16:41:27 -0500706ENV PYTHONPATH "/usr/local/lib/python3.10/site-packages/"
Patrick Williams02871c92021-02-01 20:57:19 -0600707
Patrick Williamsbb16ac12021-04-12 12:23:51 -0500708# Sometimes the ubuntu key expires and we need a way to force an execution
709# of the apt-get commands for the dbgsym-keyring. When this happens we see
710# an error like: "Release: The following signatures were invalid:"
711# Insert a bogus echo that we can change here when we get this error to force
712# the update.
713RUN echo "ubuntu keyserver rev as of 2021-04-21"
714
Patrick Williams02871c92021-02-01 20:57:19 -0600715# We need the keys to be imported for dbgsym repos
716# New releases have a package, older ones fall back to manual fetching
717# https://wiki.ubuntu.com/Debug%20Symbol%20Packages
Patrick Williams50837432021-02-06 12:24:05 -0600718RUN apt-get update && apt-get dist-upgrade -yy && \
Patrick Williamsf79ce4c2021-04-30 16:00:49 -0500719 ( apt-get install gpgv ubuntu-dbgsym-keyring || \
Patrick Williams50837432021-02-06 12:24:05 -0600720 ( apt-get install -yy dirmngr && \
721 apt-key adv --keyserver keyserver.ubuntu.com \
Adrian Ambrożewicz34ec77e2021-06-02 10:23:38 +0200722 {proxy_keyserver} \
Patrick Williams50837432021-02-06 12:24:05 -0600723 --recv-keys F2EDC64DC5AEE1F6B9C621F0C8CAB6595FDFF622 ) )
Patrick Williams02871c92021-02-01 20:57:19 -0600724
725# Parse the current repo list into a debug repo list
726RUN sed -n '/^deb /s,^deb [^ ]* ,deb http://ddebs.ubuntu.com ,p' /etc/apt/sources.list >/etc/apt/sources.list.d/debug.list
727
728# Remove non-existent debug repos
729RUN sed -i '/-\(backports\|security\) /d' /etc/apt/sources.list.d/debug.list
730
731RUN cat /etc/apt/sources.list.d/debug.list
732
733RUN apt-get update && apt-get dist-upgrade -yy && apt-get install -yy \
Patrick Williams274e3a92022-09-08 07:08:31 -0500734 gcc-12 \
735 g++-12 \
Patrick Williams02871c92021-02-01 20:57:19 -0600736 libc6-dbg \
737 libc6-dev \
738 libtool \
739 bison \
740 libdbus-1-dev \
741 flex \
742 cmake \
743 python3 \
744 python3-dev\
745 python3-yaml \
746 python3-mako \
747 python3-pip \
748 python3-setuptools \
749 python3-git \
750 python3-socks \
751 pkg-config \
752 autoconf \
753 autoconf-archive \
754 libsystemd-dev \
755 systemd \
756 libssl-dev \
757 libevdev-dev \
Patrick Williams02871c92021-02-01 20:57:19 -0600758 libjpeg-dev \
759 libpng-dev \
760 ninja-build \
761 sudo \
762 curl \
763 git \
764 dbus \
765 iputils-ping \
Patrick Williams27a646b2022-09-27 16:57:44 -0500766 clang-15 \
767 clang-format-15 \
768 clang-tidy-15 \
769 clang-tools-15 \
Patrick Williams02871c92021-02-01 20:57:19 -0600770 shellcheck \
771 npm \
772 iproute2 \
773 libnl-3-dev \
774 libnl-genl-3-dev \
775 libconfig++-dev \
776 libsnmp-dev \
777 valgrind \
778 valgrind-dbg \
779 libpam0g-dev \
780 xxd \
781 libi2c-dev \
782 wget \
783 libldap2-dev \
784 libprotobuf-dev \
William A. Kennington IIIdafe7522021-05-28 01:51:06 -0700785 liburing-dev \
Patrick Williams8949d3c2022-04-27 16:41:27 -0500786 liburing2-dbgsym \
Patrick Williams02871c92021-02-01 20:57:19 -0600787 libperlio-gzip-perl \
788 libjson-perl \
789 protobuf-compiler \
790 libgpiod-dev \
791 device-tree-compiler \
Patrick Williams02871c92021-02-01 20:57:19 -0600792 libpciaccess-dev \
793 libmimetic-dev \
794 libxml2-utils \
Patrick Williams0eedeed2021-02-06 19:06:09 -0600795 libxml-simple-perl \
John Wedig9adf68d2021-11-16 14:00:39 -0800796 rsync \
797 libcryptsetup-dev
Patrick Williams02871c92021-02-01 20:57:19 -0600798
Manojkiran Eda87111bb2021-08-14 11:26:16 +0530799RUN npm install -g eslint@latest eslint-plugin-json@latest
800
Patrick Williams5b08dc62022-09-27 16:36:57 -0500801# Kinetic comes with GCC-12, so skip this.
802#RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-12 12 \
803# --slave /usr/bin/g++ g++ /usr/bin/g++-12 \
804# --slave /usr/bin/gcov gcov /usr/bin/gcov-12 \
805# --slave /usr/bin/gcov-dump gcov-dump /usr/bin/gcov-dump-12 \
806# --slave /usr/bin/gcov-tool gcov-tool /usr/bin/gcov-tool-12
807#RUN update-alternatives --install /usr/bin/cpp cpp /usr/bin/cpp-12 12
Patrick Williams02871c92021-02-01 20:57:19 -0600808
Patrick Williams27a646b2022-09-27 16:57:44 -0500809RUN update-alternatives --install /usr/bin/clang clang /usr/bin/clang-15 1000 \
810 --slave /usr/bin/clang++ clang++ /usr/bin/clang++-15 \
811 --slave /usr/bin/clang-tidy clang-tidy /usr/bin/clang-tidy-15 \
812 --slave /usr/bin/clang-format clang-format /usr/bin/clang-format-15 \
813 --slave /usr/bin/run-clang-tidy run-clang-tidy.py /usr/bin/run-clang-tidy-15 \
814 --slave /usr/bin/scan-build scan-build /usr/bin/scan-build-15
Patrick Williams02871c92021-02-01 20:57:19 -0600815
Patrick Williams50837432021-02-06 12:24:05 -0600816"""
817
818if is_automated_ci_build:
819 dockerfile_base += f"""
820# Run an arbitrary command to polute the docker cache regularly force us
821# to re-run `apt-get update` daily.
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600822RUN echo {Docker.timestamp()}
Patrick Williams50837432021-02-06 12:24:05 -0600823RUN apt-get update && apt-get dist-upgrade -yy
824
825"""
826
827dockerfile_base += f"""
Patrick Williams02871c92021-02-01 20:57:19 -0600828RUN pip3 install inflection
829RUN pip3 install pycodestyle
830RUN pip3 install jsonschema
Michael Shenfb612a52022-08-05 00:58:31 +0000831RUN pip3 install meson==0.63.0
Ed Tanous5278efb2022-09-26 13:59:29 -0700832RUN pip3 install packaging
Patrick Williams02871c92021-02-01 20:57:19 -0600833RUN pip3 install protobuf
Manojkiran Edae6f120a2021-03-20 11:13:43 +0530834RUN pip3 install codespell
Ed Tanousca8c4a82022-02-08 13:58:22 -0800835RUN pip3 install requests
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600836"""
Patrick Williams02871c92021-02-01 20:57:19 -0600837
Ed Tanousfb9948a2022-06-21 09:10:24 -0700838# Note, we use sha1s here because the newest gitlint release doesn't include
839# some features we need. Next time they release, we can rely on a direct
840# release tag
841dockerfile_base += f"""
842RUN pip3 install git+https://github.com/jorisroovers/gitlint.git@8ede310d62d5794efa7518b235f899f8a8ad6a68\#subdirectory=gitlint-core
843RUN pip3 install git+https://github.com/jorisroovers/gitlint.git@8ede310d62d5794efa7518b235f899f8a8ad6a68
844"""
845
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600846# Build the base and stage docker images.
847docker_base_img_name = Docker.tagname("base", dockerfile_base)
848Docker.build("base", docker_base_img_name, dockerfile_base)
849Package.generate_all()
Patrick Williams02871c92021-02-01 20:57:19 -0600850
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600851# Create the final Dockerfile.
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600852dockerfile = f"""
Patrick Williams02871c92021-02-01 20:57:19 -0600853# Build the final output image
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600854FROM {docker_base_img_name}
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600855{Package.df_all_copycmds()}
Patrick Williams02871c92021-02-01 20:57:19 -0600856
857# Some of our infrastructure still relies on the presence of this file
858# even though it is no longer needed to rebuild the docker environment
859# NOTE: The file is sorted to ensure the ordering is stable.
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600860RUN echo '{Package.depcache()}' > /tmp/depcache
Patrick Williams02871c92021-02-01 20:57:19 -0600861
862# Final configuration for the workspace
Josh Lehan6825a012022-03-17 18:31:39 -0700863RUN grep -q {gid} /etc/group || groupadd -f -g {gid} {username}
Patrick Williams02871c92021-02-01 20:57:19 -0600864RUN mkdir -p "{os.path.dirname(homedir)}"
865RUN grep -q {uid} /etc/passwd || useradd -d {homedir} -m -u {uid} -g {gid} {username}
866RUN sed -i '1iDefaults umask=000' /etc/sudoers
867RUN echo "{username} ALL=(ALL) NOPASSWD: ALL" >>/etc/sudoers
868
Andrew Geissler305a9a52021-04-07 11:08:40 -0500869# Ensure user has ability to write to /usr/local for different tool
870# and data installs
Andrew Geissler7bb00b12021-05-10 15:12:08 -0500871RUN chown -R {username}:{username} /usr/local/share
Andrew Geissler305a9a52021-04-07 11:08:40 -0500872
Patrick Williams02871c92021-02-01 20:57:19 -0600873{proxy_cmd}
874
875RUN /bin/bash
876"""
877
Patrick Williamsa18d9c52021-02-05 09:52:26 -0600878# Do the final docker build
Patrick Williamsee3c9ee2021-02-12 20:56:01 -0600879docker_final_img_name = Docker.tagname(None, dockerfile)
880Docker.build("final", docker_final_img_name, dockerfile)
881
Patrick Williams00536fb2021-02-11 14:28:49 -0600882# Print the tag of the final image.
883print(docker_final_img_name)