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
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
|
#!/usr/bin/env python3
# bblock
# lock/unlock task to latest signature
#
# Copyright (c) 2023 BayLibre, SAS
# Author: Julien Stepahn <jstephan@baylibre.com>
#
# SPDX-License-Identifier: GPL-2.0-only
#
import os
import sys
import logging
scripts_path = os.path.dirname(os.path.realpath(__file__))
lib_path = scripts_path + "/lib"
sys.path = sys.path + [lib_path]
import scriptpath
scriptpath.add_bitbake_lib_path()
import bb.tinfoil
import bb.msg
import argparse_oe
myname = os.path.basename(sys.argv[0])
logger = bb.msg.logger_create(myname)
def getTaskSignatures(tinfoil, pn, tasks):
tinfoil.set_event_mask(
[
"bb.event.GetTaskSignatureResult",
"logging.LogRecord",
"bb.command.CommandCompleted",
"bb.command.CommandFailed",
]
)
ret = tinfoil.run_command("getTaskSignatures", pn, tasks)
if ret:
while True:
event = tinfoil.wait_event(1)
if event:
if isinstance(event, bb.command.CommandCompleted):
break
elif isinstance(event, bb.command.CommandFailed):
logger.error(str(event))
sys.exit(2)
elif isinstance(event, bb.event.GetTaskSignatureResult):
sig = event.sig
elif isinstance(event, logging.LogRecord):
logger.handle(event)
else:
logger.error("No result returned from getTaskSignatures command")
sys.exit(2)
return sig
def parseRecipe(tinfoil, recipe):
try:
tinfoil.parse_recipes()
d = tinfoil.parse_recipe(recipe)
except Exception:
logger.error("Failed to get recipe info for: %s" % recipe)
sys.exit(1)
return d
def bblockDump(lockfile):
try:
with open(lockfile, "r") as lockfile:
for line in lockfile:
print(line.strip())
except IOError:
return 1
return 0
def bblockReset(lockfile, pns, package_archs, tasks):
if not pns:
logger.info("Unlocking all recipes")
try:
os.remove(lockfile)
except FileNotFoundError:
pass
else:
logger.info("Unlocking {pns}".format(pns=pns))
tmp_lockfile = lockfile + ".tmp"
with open(lockfile, "r") as infile, open(tmp_lockfile, "w") as outfile:
for line in infile:
if not (
any(element in line for element in pns)
and any(element in line for element in package_archs.split())
):
outfile.write(line)
else:
if tasks and not any(element in line for element in tasks):
outfile.write(line)
os.remove(lockfile)
os.rename(tmp_lockfile, lockfile)
def main():
parser = argparse_oe.ArgumentParser(description="Lock and unlock a recipe")
parser.add_argument("pn", nargs="*", help="Space separated list of recipe to lock")
parser.add_argument(
"-t",
"--tasks",
help="Comma separated list of tasks",
type=lambda s: [
task if task.startswith("do_") else "do_" + task for task in s.split(",")
],
)
parser.add_argument(
"-r",
"--reset",
action="store_true",
help="Unlock pn recipes, or all recipes if pn is empty",
)
parser.add_argument(
"-d",
"--dump",
action="store_true",
help="Dump generated bblock.conf file",
)
global_args, unparsed_args = parser.parse_known_args()
with bb.tinfoil.Tinfoil() as tinfoil:
tinfoil.prepare(config_only=True)
package_archs = tinfoil.config_data.getVar("PACKAGE_ARCHS")
builddir = tinfoil.config_data.getVar("TOPDIR")
lockfile = "{builddir}/conf/bblock.conf".format(builddir=builddir)
if global_args.dump:
bblockDump(lockfile)
return 0
if global_args.reset:
bblockReset(lockfile, global_args.pn, package_archs, global_args.tasks)
return 0
with open(lockfile, "a") as lockfile:
s = ""
if lockfile.tell() == 0:
s = "# Generated by bblock\n"
s += 'SIGGEN_LOCKEDSIGS_TASKSIG_CHECK = "info"\n'
s += 'SIGGEN_LOCKEDSIGS_TYPES += "${PACKAGE_ARCHS}"\n'
s += "\n"
for pn in global_args.pn:
d = parseRecipe(tinfoil, pn)
package_arch = d.getVar("PACKAGE_ARCH")
siggen_locked_sigs_package_arch = d.getVar(
"SIGGEN_LOCKEDSIGS_{package_arch}".format(package_arch=package_arch)
)
sigs = getTaskSignatures(tinfoil, [pn], global_args.tasks)
for sig in sigs:
new_entry = "{pn}:{taskname}:{sig}".format(
pn=sig[0], taskname=sig[1], sig=sig[2]
)
if (
siggen_locked_sigs_package_arch
and not new_entry in siggen_locked_sigs_package_arch
) or not siggen_locked_sigs_package_arch:
s += 'SIGGEN_LOCKEDSIGS_{package_arch} += "{new_entry}"\n'.format(
package_arch=package_arch, new_entry=new_entry
)
lockfile.write(s)
return 0
if __name__ == "__main__":
try:
ret = main()
except Exception:
ret = 1
import traceback
traceback.print_exc()
sys.exit(ret)
|
