PLD_Builder/request.py

   1 # vi: encoding=utf-8 ts=8 sts=4 sw=4 et
   2
   3 from xml.dom.minidom import *
   4 import string
   5 import time
   6 import xml.sax.saxutils
   7 import fnmatch
   8 import os
   9 import urllib
  10 import cgi
  11
  12 import util
  13 import log
  14 from acl import acl
  15 from config import config
  16
  17 __all__ = ['parse_request', 'parse_requests']
  18
  19 def text(e):
  20     res = ""
  21     for n in e.childNodes:
  22         if n.nodeType != Element.TEXT_NODE:
  23             log.panic("xml: text expected in <%s>, got %d" % (e.nodeName, n.nodeType))
  24         res += n.nodeValue
  25     return res
  26
  27 def attr(e, a, default = None):
  28     try:
  29         return e.attributes[a].value
  30     except:
  31         if default != None:
  32             return default
  33         raise
  34
  35 def escape(s):
  36     return xml.sax.saxutils.escape(s)
  37
  38 # return timestamp with timezone information
  39 # so we could parse it in javascript
  40 def tzdate(t):
  41     # as strftime %z is unofficial, and does not work, need to make it numeric ourselves
  42 #    date = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(t))
  43     date = time.strftime("%a %b %d %Y %H:%M:%S", time.localtime(t))
  44     # NOTE: the altzone is showing CURRENT timezone, not what the "t" reflects
  45     # NOTE: when DST is off timezone gets it right, altzone not
  46     if time.daylight:
  47         tzoffset = time.altzone
  48     else:
  49         tzoffset = time.timezone
  50     tz = '%+05d' % (-tzoffset / 3600 * 100)
  51     return date + ' ' + tz
  52
  53 def is_blank(e):
  54     return e.nodeType == Element.TEXT_NODE and string.strip(e.nodeValue) == ""
  55
  56 class Group:
  57     def __init__(self, e):
  58         self.batches = []
  59         self.kind = 'group'
  60         self.id = attr(e, "id")
  61         self.no = int(attr(e, "no"))
  62         self.priority = 2
  63         self.time = time.time()
  64         self.requester = ""
  65         self.max_jobs = 0
  66         self.requester_email = ""
  67         self.flags = string.split(attr(e, "flags", ""))
  68         for c in e.childNodes:
  69             if is_blank(c): continue
  70
  71             if c.nodeType != Element.ELEMENT_NODE:
  72                 log.panic("xml: evil group child %d" % c.nodeType)
  73             if c.nodeName == "batch":
  74                 self.batches.append(Batch(c))
  75             elif c.nodeName == "requester":
  76                 self.requester = text(c)
  77                 self.requester_email = attr(c, "email", "")
  78             elif c.nodeName == "priority":
  79                 self.priority = int(text(c))
  80             elif c.nodeName == "time":
  81                 self.time = int(text(c))
  82             elif c.nodeName == "maxjobs":
  83                 self.max_jobs = int(text(c))
  84             else:
  85                 log.panic("xml: evil group child (%s)" % c.nodeName)
  86         # note that we also check that group is sorted WRT deps
  87         m = {}
  88         for b in self.batches:
  89             deps = []
  90             m[b.b_id] = b
  91             for dep in b.depends_on:
  92                 if m.has_key(dep):
  93                     # avoid self-deps
  94                     if id(m[dep]) != id(b):
  95                         deps.append(m[dep])
  96                 else:
  97                     log.panic("xml: dependency not found in group")
  98             b.depends_on = deps
  99         if self.requester_email == "" and self.requester != "":
 100             self.requester_email = acl.user(self.requester).mail_to()
 101
 102     def dump(self, f):
 103         f.write("group: %d (id=%s pri=%d)\n" % (self.no, self.id, self.priority))
 104         f.write("  from: %s\n" % self.requester)
 105         f.write("  flags: %s\n" % string.join(self.flags))
 106         f.write("  time: %s\n" % time.asctime(time.localtime(self.time)))
 107         for b in self.batches:
 108             b.dump(f)
 109         f.write("\n")
 110
 111     def dump_html(self, f):
 112         f.write(
 113             "<div id=\"%(no)d\" class=\"%(flags)s\">\n"
 114             "<a href=\"#%(no)d\">%(no)d</a>. <span id=\"tz\">%(time)s</span> from <b>%(requester)s</b> "
 115             "<small>%(id)s, prio=%(priority)d, jobs=%(max_jobs)d, %(flags)s</small>\n"
 116         % {
 117             'no': self.no,
 118             'id': '<a href="srpms/%(id)s">%(id)s</a>' % {'id': self.id},
 119             'time': escape(tzdate(self.time)),
 120             'requester': escape(self.requester),
 121             'priority': self.priority,
 122             'max_jobs': self.max_jobs,
 123             'flags': string.join(self.flags)
 124         })
 125         f.write("<ul>\n")
 126         for b in self.batches:
 127             b.dump_html(f, self.id)
 128         f.write("</ul>\n")
 129         f.write("</div>\n")
 130
 131     def write_to(self, f):
 132         f.write("""
 133        <group id="%s" no="%d" flags="%s">
 134          <requester email='%s'>%s</requester>
 135          <time>%d</time>
 136          <priority>%d</priority>
 137          <maxjobs>%d</maxjobs>\n""" % (self.id, self.no, string.join(self.flags),
 138                     escape(self.requester_email), escape(self.requester),
 139                     self.time, self.priority, self.max_jobs))
 140         for b in self.batches:
 141             b.write_to(f)
 142         f.write("       </group>\n\n")
 143
 144     def is_done(self):
 145         ok = 1
 146         for b in self.batches:
 147             if not b.is_done():
 148                 ok = 0
 149         return ok
 150
 151 class Batch:
 152     def __init__(self, e):
 153         self.bconds_with = []
 154         self.bconds_without = []
 155         self.builders = []
 156         self.builders_status = {}
 157         self.builders_status_time = {}
 158         self.builders_status_buildtime = {}
 159         self.kernel = ""
 160         self.target = []
 161         self.branch = ""
 162         self.src_rpm = ""
 163         self.info = ""
 164         self.spec = ""
 165         self.command = ""
 166         self.command_flags = []
 167         self.skip = []
 168         self.gb_id = ""
 169         self.b_id = attr(e, "id")
 170         self.depends_on = string.split(attr(e, "depends-on"))
 171         self.upgraded = True
 172         for c in e.childNodes:
 173             if is_blank(c): continue
 174
 175             if c.nodeType != Element.ELEMENT_NODE:
 176                 log.panic("xml: evil batch child %d" % c.nodeType)
 177             if c.nodeName == "src-rpm":
 178                 self.src_rpm = text(c)
 179             elif c.nodeName == "spec":
 180                 # normalize specname, specname is used as buildlog and we don't
 181                 # want to be exposed to directory traversal attacks
 182                 self.spec = text(c).split('/')[-1]
 183             elif c.nodeName == "command":
 184                 self.spec = "COMMAND"
 185                 self.command = text(c).strip()
 186                 self.command_flags = string.split(attr(c, "flags", ""))
 187             elif c.nodeName == "info":
 188                 self.info = text(c)
 189             elif c.nodeName == "kernel":
 190                 self.kernel = text(c)
 191             elif c.nodeName == "target":
 192                 self.target.append(text(c))
 193             elif c.nodeName == "skip":
 194                 self.skip.append(text(c))
 195             elif c.nodeName == "branch":
 196                 self.branch = text(c)
 197             elif c.nodeName == "builder":
 198                 key = text(c)
 199                 self.builders.append(key)
 200                 self.builders_status[key] = attr(c, "status", "?")
 201                 self.builders_status_time[key] = attr(c, "time", "0")
 202                 self.builders_status_buildtime[key] = "0" #attr(c, "buildtime", "0")
 203             elif c.nodeName == "with":
 204                 self.bconds_with.append(text(c))
 205             elif c.nodeName == "without":
 206                 self.bconds_without.append(text(c))
 207             else:
 208                 log.panic("xml: evil batch child (%s)" % c.nodeName)
 209
 210     def is_done(self):
 211         ok = 1
 212         for b in self.builders:
 213             s = self.builders_status[b]
 214             if not s.startswith("OK") and not s.startswith("SKIP") and not s.startswith("UNSUPP") and not s.startswith("FAIL"):
 215                 ok = 0
 216         return ok
 217
 218     def dump(self, f):
 219         f.write("  batch: %s/%s\n" % (self.src_rpm, self.spec))
 220         f.write("    info: %s\n" % self.info)
 221         f.write("    kernel: %s\n" % self.kernel)
 222         f.write("    target: %s\n" % self.target_string())
 223         f.write("    branch: %s\n" % self.branch)
 224         f.write("    bconds: %s\n" % self.bconds_string())
 225         builders = []
 226         for b in self.builders:
 227             builders.append("%s:%s" % (b, self.builders_status[b]))
 228         f.write("    builders: %s\n" % string.join(builders))
 229
 230     def is_command(self):
 231         return self.command != ""
 232
 233     def dump_html(self, f, rid):
 234         f.write("<li>\n")
 235         if self.is_command():
 236             desc = "SH: <pre>%s</pre> flags: [%s]" % (self.command, ' '.join(self.command_flags))
 237         else:
 238             package_url = "http://git.pld-linux.org/gitweb.cgi?p=packages/%(package)s.git;f=%(spec)s;h=%(branch)s;a=shortlog" % {
 239                 'spec': self.spec,
 240                 'branch': self.branch,
 241                 'package': self.spec[:-5],
 242             }
 243             desc = "%(src_rpm)s (<a href=\"%(package_url)s\">%(spec)s -r %(branch)s</a>%(bconds)s)" % {
 244                 'src_rpm': self.src_rpm,
 245                 'spec': self.spec,
 246                 'branch': self.branch,
 247                 'bconds': self.bconds_string() + self.kernel_string() + self.target_string(),
 248                 'package_url': package_url,
 249             }
 250         f.write("%s <small>[" % desc)
 251         builders = []
 252         for b in self.builders:
 253             s = self.builders_status[b]
 254             if s.startswith("OK"):
 255                 c = "green"
 256             elif s.startswith("FAIL"):
 257                 c = "red"
 258             elif s.startswith("SKIP"):
 259                 c = "blue"
 260             elif s.startswith("UNSUPP"):
 261                 c = "fuchsia"
 262             else:
 263                 c = "black"
 264             link_pre = ""
 265             link_post = ""
 266             if (s.startswith("OK") or s.startswith("SKIP") or s.startswith("UNSUPP") or s.startswith("FAIL")) and len(self.spec) > 5:
 267                 if self.is_command():
 268                     bl_name = "command"
 269                 else:
 270                     bl_name = self.spec[:len(self.spec)-5]
 271                 lin_ar = b.replace('noauto-','')
 272                 path = "/%s/%s/%s,%s.bz2" % (lin_ar.replace('-','/'), s, bl_name, rid)
 273                 is_ok = 0
 274                 if s.startswith("OK"):
 275                     is_ok = 1
 276                 bld = lin_ar.split('-')
 277                 tree_name = '-'.join(bld[:-1])
 278                 tree_arch = '-'.join(bld[-1:])
 279                 link_pre = "<a href=\"http://buildlogs.pld-linux.org/index.php?dist=%s&arch=%s&ok=%d&name=%s&id=%s&action=tail\">" \
 280                         % (urllib.quote(tree_name), urllib.quote(tree_arch), is_ok, urllib.quote(bl_name), urllib.quote(rid))
 281                 link_post = "</a>"
 282
 283             def ftime(s):
 284                 t = float(s)
 285                 if t > 0:
 286                     return time.asctime(time.localtime(t))
 287                 else:
 288                     return 'N/A'
 289
 290             tooltip = "last update: %(time)s\nbuild time: %(buildtime)s" % {
 291                 'time' : ftime(self.builders_status_time[b]),
 292                 'buildtime' : ftime(self.builders_status_buildtime[b]),
 293             }
 294             builders.append(link_pre +
 295                 "<font color='%(color)s'><b title=\"%(tooltip)s\">%(builder)s:%(status)s</b></font>" % {
 296                     'color' : c,
 297                     'builder' : b,
 298                     'status' : s,
 299                     'tooltip' : cgi.escape(tooltip, True),
 300             }
 301             + link_post)
 302         f.write("%s]</small></li>\n" % string.join(builders))
 303
 304     def rpmbuild_opts(self):
 305         """
 306             return all rpmbuild options related to this build
 307         """
 308         bconds = self.bconds_string() + self.kernel_string() + self.target_string()
 309         rpmdefs = \
 310             "--define '_topdir %(echo $HOME/rpm)' " \
 311             "--define '_specdir %{_topdir}/packages/%{name}' "  \
 312             "--define '_sourcedir %{_specdir}' " \
 313             "--define '_builddir %{_topdir}/BUILD/%{name}' "
 314         return rpmdefs + bconds
 315
 316     def kernel_string(self):
 317         r = ""
 318         if self.kernel != "":
 319             r = " --define 'alt_kernel " + self.kernel + "'"
 320         return r
 321
 322     def target_string(self):
 323         if len(self.target) > 0:
 324             return " --target " + ",".join(self.target)
 325         else:
 326             return ""
 327
 328     def bconds_string(self):
 329         r = ""
 330         for b in self.bconds_with:
 331             r = r + " --with " + b
 332         for b in self.bconds_without:
 333             r = r + " --without " + b
 334         return r
 335
 336     def default_target(self, arch):
 337         self.target.append("%s-pld-linux" % arch)
 338
 339     def write_to(self, f):
 340         f.write("""
 341          <batch id='%s' depends-on='%s'>
 342            <src-rpm>%s</src-rpm>
 343            <command flags="%s">%s</command>
 344            <spec>%s</spec>
 345            <branch>%s</branch>
 346            <info>%s</info>\n""" % (self.b_id,
 347                  string.join(map(lambda (b): b.b_id, self.depends_on)),
 348                  escape(self.src_rpm),
 349                  escape(' '.join(self.command_flags)), escape(self.command),
 350                  escape(self.spec), escape(self.branch), escape(self.info)))
 351         if self.kernel != "":
 352             f.write("           <kernel>%s</kernel>\n" % escape(self.kernel))
 353         for b in self.bconds_with:
 354             f.write("           <with>%s</with>\n" % escape(b))
 355         for b in self.target:
 356             f.write("           <target>%s</target>\n" % escape(b))
 357         for b in self.bconds_without:
 358             f.write("           <without>%s</without>\n" % escape(b))
 359         for b in self.builders:
 360             if self.builders_status_buildtime.has_key(b):
 361                 t = self.builders_status_buildtime[b]
 362             else:
 363                 t = "0"
 364             f.write("           <builder status='%s' time='%s' buildtime='%s'>%s</builder>\n" % \
 365                     (escape(self.builders_status[b]), self.builders_status_time[b], t, escape(b)))
 366         f.write("         </batch>\n")
 367
 368     def log_line(self, l):
 369         log.notice(l)
 370         if self.logfile != None:
 371             util.append_to(self.logfile, l)
 372
 373     def expand_builders(batch, all_builders):
 374         all = []
 375         for bld in batch.builders:
 376             res = []
 377             for my_bld in all_builders:
 378                 if fnmatch.fnmatch(my_bld, bld):
 379                     res.append(my_bld)
 380             if res != []:
 381                 all.extend(res)
 382             else:
 383                 all.append(bld)
 384         batch.builders = all
 385
 386 class Notification:
 387     def __init__(self, e):
 388         self.batches = []
 389         self.kind = 'notification'
 390         self.group_id = attr(e, "group-id")
 391         self.builder = attr(e, "builder")
 392         self.batches = {}
 393         self.batches_buildtime = {}
 394         for c in e.childNodes:
 395             if is_blank(c): continue
 396             if c.nodeType != Element.ELEMENT_NODE:
 397                 log.panic("xml: evil notification child %d" % c.nodeType)
 398             if c.nodeName == "batch":
 399                 id = attr(c, "id")
 400                 status = attr(c, "status")
 401                 buildtime = attr(c, "buildtime", "0")
 402                 if not status.startswith("OK") and not status.startswith("SKIP") and not status.startswith("UNSUPP") and not status.startswith("FAIL"):
 403                     log.panic("xml notification: bad status: %s" % status)
 404                 self.batches[id] = status
 405                 self.batches_buildtime[id] = buildtime
 406             else:
 407                 log.panic("xml: evil notification child (%s)" % c.nodeName)
 408
 409     def apply_to(self, q):
 410         for r in q.requests:
 411             if r.kind == "group":
 412                 for b in r.batches:
 413                     if self.batches.has_key(b.b_id):
 414                         b.builders_status[self.builder] = self.batches[b.b_id]
 415                         b.builders_status_time[self.builder] = time.time()
 416                         b.builders_status_buildtime[self.builder] = "0" #self.batches_buildtime[b.b_id]
 417
 418 def build_request(e):
 419     if e.nodeType != Element.ELEMENT_NODE:
 420         log.panic("xml: evil request element")
 421     if e.nodeName == "group":
 422         return Group(e)
 423     elif e.nodeName == "notification":
 424         return Notification(e)
 425     elif e.nodeName == "command":
 426         # FIXME
 427         return Command(e)
 428     else:
 429         log.panic("xml: evil request [%s]" % e.nodeName)
 430
 431 def parse_request(f):
 432     d = parseString(f)
 433     return build_request(d.documentElement)
 434
 435 def parse_requests(f):
 436     d = parseString(f)
 437     res = []
 438     for r in d.documentElement.childNodes:
 439         if is_blank(r): continue
 440         res.append(build_request(r))
 441     return res