ast.py 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426
  1. """
  2. AbstractSyntaxTree classes for the Bitbake language
  3. """
  4. # Copyright (C) 2003, 2004 Chris Larson
  5. # Copyright (C) 2003, 2004 Phil Blundell
  6. # Copyright (C) 2009 Holger Hans Peter Freyther
  7. #
  8. # SPDX-License-Identifier: GPL-2.0-only
  9. #
  10. import bb
  11. from bb import methodpool
  12. from bb.parse import logger
  13. class StatementGroup(list):
  14. def eval(self, data):
  15. for statement in self:
  16. statement.eval(data)
  17. class AstNode(object):
  18. def __init__(self, filename, lineno):
  19. self.filename = filename
  20. self.lineno = lineno
  21. class IncludeNode(AstNode):
  22. def __init__(self, filename, lineno, what_file, force):
  23. AstNode.__init__(self, filename, lineno)
  24. self.what_file = what_file
  25. self.force = force
  26. def eval(self, data):
  27. """
  28. Include the file and evaluate the statements
  29. """
  30. s = data.expand(self.what_file)
  31. logger.debug(2, "CONF %s:%s: including %s", self.filename, self.lineno, s)
  32. # TODO: Cache those includes... maybe not here though
  33. if self.force:
  34. bb.parse.ConfHandler.include(self.filename, s, self.lineno, data, "include required")
  35. else:
  36. bb.parse.ConfHandler.include(self.filename, s, self.lineno, data, False)
  37. class ExportNode(AstNode):
  38. def __init__(self, filename, lineno, var):
  39. AstNode.__init__(self, filename, lineno)
  40. self.var = var
  41. def eval(self, data):
  42. data.setVarFlag(self.var, "export", 1, op = 'exported')
  43. class UnsetNode(AstNode):
  44. def __init__(self, filename, lineno, var):
  45. AstNode.__init__(self, filename, lineno)
  46. self.var = var
  47. def eval(self, data):
  48. loginfo = {
  49. 'variable': self.var,
  50. 'file': self.filename,
  51. 'line': self.lineno,
  52. }
  53. data.delVar(self.var,**loginfo)
  54. class UnsetFlagNode(AstNode):
  55. def __init__(self, filename, lineno, var, flag):
  56. AstNode.__init__(self, filename, lineno)
  57. self.var = var
  58. self.flag = flag
  59. def eval(self, data):
  60. loginfo = {
  61. 'variable': self.var,
  62. 'file': self.filename,
  63. 'line': self.lineno,
  64. }
  65. data.delVarFlag(self.var, self.flag, **loginfo)
  66. class DataNode(AstNode):
  67. """
  68. Various data related updates. For the sake of sanity
  69. we have one class doing all this. This means that all
  70. this need to be re-evaluated... we might be able to do
  71. that faster with multiple classes.
  72. """
  73. def __init__(self, filename, lineno, groupd):
  74. AstNode.__init__(self, filename, lineno)
  75. self.groupd = groupd
  76. def getFunc(self, key, data):
  77. if 'flag' in self.groupd and self.groupd['flag'] is not None:
  78. return data.getVarFlag(key, self.groupd['flag'], expand=False, noweakdefault=True)
  79. else:
  80. return data.getVar(key, False, noweakdefault=True, parsing=True)
  81. def eval(self, data):
  82. groupd = self.groupd
  83. key = groupd["var"]
  84. loginfo = {
  85. 'variable': key,
  86. 'file': self.filename,
  87. 'line': self.lineno,
  88. }
  89. if "exp" in groupd and groupd["exp"] is not None:
  90. data.setVarFlag(key, "export", 1, op = 'exported', **loginfo)
  91. op = "set"
  92. if "ques" in groupd and groupd["ques"] is not None:
  93. val = self.getFunc(key, data)
  94. op = "set?"
  95. if val is None:
  96. val = groupd["value"]
  97. elif "colon" in groupd and groupd["colon"] is not None:
  98. e = data.createCopy()
  99. op = "immediate"
  100. val = e.expand(groupd["value"], key + "[:=]")
  101. elif "append" in groupd and groupd["append"] is not None:
  102. op = "append"
  103. val = "%s %s" % ((self.getFunc(key, data) or ""), groupd["value"])
  104. elif "prepend" in groupd and groupd["prepend"] is not None:
  105. op = "prepend"
  106. val = "%s %s" % (groupd["value"], (self.getFunc(key, data) or ""))
  107. elif "postdot" in groupd and groupd["postdot"] is not None:
  108. op = "postdot"
  109. val = "%s%s" % ((self.getFunc(key, data) or ""), groupd["value"])
  110. elif "predot" in groupd and groupd["predot"] is not None:
  111. op = "predot"
  112. val = "%s%s" % (groupd["value"], (self.getFunc(key, data) or ""))
  113. else:
  114. val = groupd["value"]
  115. flag = None
  116. if 'flag' in groupd and groupd['flag'] is not None:
  117. flag = groupd['flag']
  118. elif groupd["lazyques"]:
  119. flag = "_defaultval"
  120. loginfo['op'] = op
  121. loginfo['detail'] = groupd["value"]
  122. if flag:
  123. data.setVarFlag(key, flag, val, **loginfo)
  124. else:
  125. data.setVar(key, val, parsing=True, **loginfo)
  126. class MethodNode(AstNode):
  127. tr_tbl = str.maketrans('/.+-@%&', '_______')
  128. def __init__(self, filename, lineno, func_name, body, python, fakeroot):
  129. AstNode.__init__(self, filename, lineno)
  130. self.func_name = func_name
  131. self.body = body
  132. self.python = python
  133. self.fakeroot = fakeroot
  134. def eval(self, data):
  135. text = '\n'.join(self.body)
  136. funcname = self.func_name
  137. if self.func_name == "__anonymous":
  138. funcname = ("__anon_%s_%s" % (self.lineno, self.filename.translate(MethodNode.tr_tbl)))
  139. self.python = True
  140. text = "def %s(d):\n" % (funcname) + text
  141. bb.methodpool.insert_method(funcname, text, self.filename, self.lineno - len(self.body) - 1)
  142. anonfuncs = data.getVar('__BBANONFUNCS', False) or []
  143. anonfuncs.append(funcname)
  144. data.setVar('__BBANONFUNCS', anonfuncs)
  145. if data.getVar(funcname, False):
  146. # clean up old version of this piece of metadata, as its
  147. # flags could cause problems
  148. data.delVarFlag(funcname, 'python')
  149. data.delVarFlag(funcname, 'fakeroot')
  150. if self.python:
  151. data.setVarFlag(funcname, "python", "1")
  152. if self.fakeroot:
  153. data.setVarFlag(funcname, "fakeroot", "1")
  154. data.setVarFlag(funcname, "func", 1)
  155. data.setVar(funcname, text, parsing=True)
  156. data.setVarFlag(funcname, 'filename', self.filename)
  157. data.setVarFlag(funcname, 'lineno', str(self.lineno - len(self.body)))
  158. class PythonMethodNode(AstNode):
  159. def __init__(self, filename, lineno, function, modulename, body):
  160. AstNode.__init__(self, filename, lineno)
  161. self.function = function
  162. self.modulename = modulename
  163. self.body = body
  164. def eval(self, data):
  165. # Note we will add root to parsedmethods after having parse
  166. # 'this' file. This means we will not parse methods from
  167. # bb classes twice
  168. text = '\n'.join(self.body)
  169. bb.methodpool.insert_method(self.modulename, text, self.filename, self.lineno - len(self.body) - 1)
  170. data.setVarFlag(self.function, "func", 1)
  171. data.setVarFlag(self.function, "python", 1)
  172. data.setVar(self.function, text, parsing=True)
  173. data.setVarFlag(self.function, 'filename', self.filename)
  174. data.setVarFlag(self.function, 'lineno', str(self.lineno - len(self.body) - 1))
  175. class ExportFuncsNode(AstNode):
  176. def __init__(self, filename, lineno, fns, classname):
  177. AstNode.__init__(self, filename, lineno)
  178. self.n = fns.split()
  179. self.classname = classname
  180. def eval(self, data):
  181. for func in self.n:
  182. calledfunc = self.classname + "_" + func
  183. if data.getVar(func, False) and not data.getVarFlag(func, 'export_func', False):
  184. continue
  185. if data.getVar(func, False):
  186. data.setVarFlag(func, 'python', None)
  187. data.setVarFlag(func, 'func', None)
  188. for flag in [ "func", "python" ]:
  189. if data.getVarFlag(calledfunc, flag, False):
  190. data.setVarFlag(func, flag, data.getVarFlag(calledfunc, flag, False))
  191. for flag in [ "dirs" ]:
  192. if data.getVarFlag(func, flag, False):
  193. data.setVarFlag(calledfunc, flag, data.getVarFlag(func, flag, False))
  194. data.setVarFlag(func, "filename", "autogenerated")
  195. data.setVarFlag(func, "lineno", 1)
  196. if data.getVarFlag(calledfunc, "python", False):
  197. data.setVar(func, " bb.build.exec_func('" + calledfunc + "', d)\n", parsing=True)
  198. else:
  199. if "-" in self.classname:
  200. bb.fatal("The classname %s contains a dash character and is calling an sh function %s using EXPORT_FUNCTIONS. Since a dash is illegal in sh function names, this cannot work, please rename the class or don't use EXPORT_FUNCTIONS." % (self.classname, calledfunc))
  201. data.setVar(func, " " + calledfunc + "\n", parsing=True)
  202. data.setVarFlag(func, 'export_func', '1')
  203. class AddTaskNode(AstNode):
  204. def __init__(self, filename, lineno, func, before, after):
  205. AstNode.__init__(self, filename, lineno)
  206. self.func = func
  207. self.before = before
  208. self.after = after
  209. def eval(self, data):
  210. bb.build.addtask(self.func, self.before, self.after, data)
  211. class DelTaskNode(AstNode):
  212. def __init__(self, filename, lineno, func):
  213. AstNode.__init__(self, filename, lineno)
  214. self.func = func
  215. def eval(self, data):
  216. bb.build.deltask(self.func, data)
  217. class BBHandlerNode(AstNode):
  218. def __init__(self, filename, lineno, fns):
  219. AstNode.__init__(self, filename, lineno)
  220. self.hs = fns.split()
  221. def eval(self, data):
  222. bbhands = data.getVar('__BBHANDLERS', False) or []
  223. for h in self.hs:
  224. bbhands.append(h)
  225. data.setVarFlag(h, "handler", 1)
  226. data.setVar('__BBHANDLERS', bbhands)
  227. class InheritNode(AstNode):
  228. def __init__(self, filename, lineno, classes):
  229. AstNode.__init__(self, filename, lineno)
  230. self.classes = classes
  231. def eval(self, data):
  232. bb.parse.BBHandler.inherit(self.classes, self.filename, self.lineno, data)
  233. def handleInclude(statements, filename, lineno, m, force):
  234. statements.append(IncludeNode(filename, lineno, m.group(1), force))
  235. def handleExport(statements, filename, lineno, m):
  236. statements.append(ExportNode(filename, lineno, m.group(1)))
  237. def handleUnset(statements, filename, lineno, m):
  238. statements.append(UnsetNode(filename, lineno, m.group(1)))
  239. def handleUnsetFlag(statements, filename, lineno, m):
  240. statements.append(UnsetFlagNode(filename, lineno, m.group(1), m.group(2)))
  241. def handleData(statements, filename, lineno, groupd):
  242. statements.append(DataNode(filename, lineno, groupd))
  243. def handleMethod(statements, filename, lineno, func_name, body, python, fakeroot):
  244. statements.append(MethodNode(filename, lineno, func_name, body, python, fakeroot))
  245. def handlePythonMethod(statements, filename, lineno, funcname, modulename, body):
  246. statements.append(PythonMethodNode(filename, lineno, funcname, modulename, body))
  247. def handleExportFuncs(statements, filename, lineno, m, classname):
  248. statements.append(ExportFuncsNode(filename, lineno, m.group(1), classname))
  249. def handleAddTask(statements, filename, lineno, m):
  250. func = m.group("func")
  251. before = m.group("before")
  252. after = m.group("after")
  253. if func is None:
  254. return
  255. statements.append(AddTaskNode(filename, lineno, func, before, after))
  256. def handleDelTask(statements, filename, lineno, m):
  257. func = m.group("func")
  258. if func is None:
  259. return
  260. statements.append(DelTaskNode(filename, lineno, func))
  261. def handleBBHandlers(statements, filename, lineno, m):
  262. statements.append(BBHandlerNode(filename, lineno, m.group(1)))
  263. def handleInherit(statements, filename, lineno, m):
  264. classes = m.group(1)
  265. statements.append(InheritNode(filename, lineno, classes))
  266. def runAnonFuncs(d):
  267. code = []
  268. for funcname in d.getVar("__BBANONFUNCS", False) or []:
  269. code.append("%s(d)" % funcname)
  270. bb.utils.better_exec("\n".join(code), {"d": d})
  271. def finalize(fn, d, variant = None):
  272. saved_handlers = bb.event.get_handlers().copy()
  273. try:
  274. for var in d.getVar('__BBHANDLERS', False) or []:
  275. # try to add the handler
  276. handlerfn = d.getVarFlag(var, "filename", False)
  277. if not handlerfn:
  278. bb.fatal("Undefined event handler function '%s'" % var)
  279. handlerln = int(d.getVarFlag(var, "lineno", False))
  280. bb.event.register(var, d.getVar(var, False), (d.getVarFlag(var, "eventmask") or "").split(), handlerfn, handlerln)
  281. bb.event.fire(bb.event.RecipePreFinalise(fn), d)
  282. bb.data.expandKeys(d)
  283. runAnonFuncs(d)
  284. tasklist = d.getVar('__BBTASKS', False) or []
  285. bb.event.fire(bb.event.RecipeTaskPreProcess(fn, list(tasklist)), d)
  286. bb.build.add_tasks(tasklist, d)
  287. bb.parse.siggen.finalise(fn, d, variant)
  288. d.setVar('BBINCLUDED', bb.parse.get_file_depends(d))
  289. bb.event.fire(bb.event.RecipeParsed(fn), d)
  290. finally:
  291. bb.event.set_handlers(saved_handlers)
  292. def _create_variants(datastores, names, function, onlyfinalise):
  293. def create_variant(name, orig_d, arg = None):
  294. if onlyfinalise and name not in onlyfinalise:
  295. return
  296. new_d = bb.data.createCopy(orig_d)
  297. function(arg or name, new_d)
  298. datastores[name] = new_d
  299. for variant in list(datastores.keys()):
  300. for name in names:
  301. if not variant:
  302. # Based on main recipe
  303. create_variant(name, datastores[""])
  304. else:
  305. create_variant("%s-%s" % (variant, name), datastores[variant], name)
  306. def multi_finalize(fn, d):
  307. appends = (d.getVar("__BBAPPEND") or "").split()
  308. for append in appends:
  309. logger.debug(1, "Appending .bbappend file %s to %s", append, fn)
  310. bb.parse.BBHandler.handle(append, d, True)
  311. onlyfinalise = d.getVar("__ONLYFINALISE", False)
  312. safe_d = d
  313. d = bb.data.createCopy(safe_d)
  314. try:
  315. finalize(fn, d)
  316. except bb.parse.SkipRecipe as e:
  317. d.setVar("__SKIPPED", e.args[0])
  318. datastores = {"": safe_d}
  319. extended = d.getVar("BBCLASSEXTEND") or ""
  320. if extended:
  321. # the following is to support bbextends with arguments, for e.g. multilib
  322. # an example is as follows:
  323. # BBCLASSEXTEND = "multilib:lib32"
  324. # it will create foo-lib32, inheriting multilib.bbclass and set
  325. # BBEXTENDCURR to "multilib" and BBEXTENDVARIANT to "lib32"
  326. extendedmap = {}
  327. variantmap = {}
  328. for ext in extended.split():
  329. eext = ext.split(':', 2)
  330. if len(eext) > 1:
  331. extendedmap[ext] = eext[0]
  332. variantmap[ext] = eext[1]
  333. else:
  334. extendedmap[ext] = ext
  335. pn = d.getVar("PN")
  336. def extendfunc(name, d):
  337. if name != extendedmap[name]:
  338. d.setVar("BBEXTENDCURR", extendedmap[name])
  339. d.setVar("BBEXTENDVARIANT", variantmap[name])
  340. else:
  341. d.setVar("PN", "%s-%s" % (pn, name))
  342. bb.parse.BBHandler.inherit(extendedmap[name], fn, 0, d)
  343. safe_d.setVar("BBCLASSEXTEND", extended)
  344. _create_variants(datastores, extendedmap.keys(), extendfunc, onlyfinalise)
  345. for variant in datastores.keys():
  346. if variant:
  347. try:
  348. if not onlyfinalise or variant in onlyfinalise:
  349. finalize(fn, datastores[variant], variant)
  350. except bb.parse.SkipRecipe as e:
  351. datastores[variant].setVar("__SKIPPED", e.args[0])
  352. datastores[""] = d
  353. return datastores