]> git.decadent.org.uk Git - dak.git/blob - dak/new_security_install.py
Merge commit 'mhy/securityqueue' into merge
[dak.git] / dak / new_security_install.py
1 #!/usr/bin/env python
2
3 # Wrapper for Debian Security team
4 # Copyright (C) 2006  Anthony Towns <ajt@debian.org>
5
6 # This program is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
10
11 # This program is distributed in the hope that it will be useful, but
12 # WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 # General Public License for more details.
15
16 # You should have received a copy of the GNU General Public License
17 # along with this program; if not, write to the Free Software
18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
19 # USA
20
21 ################################################################################
22
23 from daklib import queue
24 from daklib import logging
25 from daklib import utils
26 from daklib import database
27 import apt_pkg, os, sys, pwd, time, re, commands
28
29 re_taint_free = re.compile(r"^['/;\-\+\.~\s\w]+$");
30
31 Cnf = None
32 Options = None
33 Upload = None
34 Logger = None
35
36 advisory = None
37 changes = []
38 srcverarches = {}
39
40 def init():
41     global Cnf, Upload, Options, Logger
42
43     Cnf = utils.get_conf()
44     Cnf["Dinstall::Options::No-Mail"] = "y"
45     Arguments = [('h', "help", "Security-Install::Options::Help"),
46                  ('a', "automatic", "Security-Install::Options::Automatic"),
47                  ('n', "no-action", "Security-Install::Options::No-Action"),
48                  ('s', "sudo", "Security-Install::Options::Sudo"),
49                  (' ', "no-upload", "Security-Install::Options::No-Upload"),
50                  ('u', "fg-upload", "Security-Install::Options::Foreground-Upload"),
51                  (' ', "drop-advisory", "Security-Install::Options::Drop-Advisory"),
52                  ('A', "approve", "Security-Install::Options::Approve"),
53                  ('R', "reject", "Security-Install::Options::Reject"),
54                  ('D', "disembargo", "Security-Install::Options::Disembargo") ]
55
56     for i in Arguments:
57         Cnf[i[2]] = ""
58
59     arguments = apt_pkg.ParseCommandLine(Cnf, Arguments, sys.argv)
60
61     Options = Cnf.SubTree("Security-Install::Options")
62
63     whoami = os.getuid()
64     whoamifull = pwd.getpwuid(whoami)
65     username = whoamifull[0]
66     if username != "dak":
67         print "Non-dak user: %s" % username
68         Options["Sudo"] = "y"
69
70     if Options["Help"]:
71         print "help yourself"
72         sys.exit(0)
73
74     if len(arguments) == 0:
75         utils.fubar("Process what?")
76
77     Upload = queue.Upload(Cnf)
78     if Options["No-Action"]:
79         Options["Sudo"] = ""
80     if not Options["Sudo"] and not Options["No-Action"]:
81         Logger = Upload.Logger = logging.Logger(Cnf, "new-security-install")
82
83     return arguments
84
85 def quit():
86     if Logger:
87         Logger.close()
88     sys.exit(0)
89
90 def load_args(arguments):
91     global advisory, changes
92
93     adv_ids = {}
94     if not arguments[0].endswith(".changes"):
95         adv_ids [arguments[0]] = 1
96         arguments = arguments[1:]
97
98     null_adv_changes = []
99
100     changesfiles = {}
101     for a in arguments:
102         if "/" in a:
103             utils.fubar("can only deal with files in the current directory")
104         if not a.endswith(".changes"):
105             utils.fubar("not a .changes file: %s" % (a))
106         Upload.init_vars()
107         Upload.pkg.changes_file = a
108         Upload.update_vars()
109         if "adv id" in Upload.pkg.changes:
110             changesfiles[a] = 1
111             adv_ids[Upload.pkg.changes["adv id"]] = 1
112         else:
113             null_adv_changes.append(a)
114
115     adv_ids = adv_ids.keys()
116     if len(adv_ids) > 1:
117         utils.fubar("multiple advisories selected: %s" % (", ".join(adv_ids)))
118     if adv_ids == []:
119         advisory = None
120     else:
121         advisory = adv_ids[0]
122
123     changes = changesfiles.keys()
124     return null_adv_changes
125
126 def load_adv_changes():
127     global srcverarches, changes
128
129     for c in os.listdir("."):
130         if not c.endswith(".changes"): continue
131         Upload.init_vars()
132         Upload.pkg.changes_file = c
133         Upload.update_vars()
134         if "adv id" not in Upload.pkg.changes:
135             continue
136         if Upload.pkg.changes["adv id"] != advisory:
137             continue
138
139         if c not in changes: changes.append(c)
140         srcver = "%s %s" % (Upload.pkg.changes["source"],
141                             Upload.pkg.changes["version"])
142         srcverarches.setdefault(srcver, {})
143         for arch in Upload.pkg.changes["architecture"].keys():
144             srcverarches[srcver][arch] = 1
145
146 def advisory_info():
147     if advisory != None:
148         print "Advisory: %s" % (advisory)
149     print "Changes:"
150     for c in changes:
151         print " %s" % (c)
152
153     print "Packages:"
154     svs = srcverarches.keys()
155     svs.sort()
156     for sv in svs:
157         as = srcverarches[sv].keys()
158         as.sort()
159         print " %s (%s)" % (sv, ", ".join(as))
160
161 def prompt(opts, default):
162     p = ""
163     v = {}
164     for o in opts:
165         v[o[0].upper()] = o
166         if o[0] == default:
167             p += ", [%s]%s" % (o[0], o[1:])
168         else:
169             p += ", " + o
170     p = p[2:] + "? "
171     a = None
172
173     if Options["Automatic"]:
174         a = default
175
176     while a not in v:
177         a = utils.our_raw_input(p) + default
178         a = a[:1].upper()
179
180     return v[a]
181
182 def add_changes(extras):
183     for c in extras:
184         changes.append(c)
185         Upload.init_vars()
186         Upload.pkg.changes_file = c
187         Upload.update_vars()
188         srcver = "%s %s" % (Upload.pkg.changes["source"], Upload.pkg.changes["version"])
189         srcverarches.setdefault(srcver, {})
190         for arch in Upload.pkg.changes["architecture"].keys():
191             srcverarches[srcver][arch] = 1
192         Upload.pkg.changes["adv id"] = advisory
193         Upload.dump_vars(os.getcwd())
194
195 def yes_no(prompt):
196     if Options["Automatic"]: return True
197     while 1:
198         answer = utils.our_raw_input(prompt + " ").lower()
199         if answer in "yn":
200             return answer == "y"
201         print "Invalid answer; please try again."
202
203 def do_upload():
204     if Options["No-Upload"]:
205         print "Not uploading as requested"
206     elif Options["Foreground-Upload"]:
207         actually_upload(changes)
208     else:
209         child = os.fork()
210         if child == 0:
211             actually_upload(changes)
212             os._exit(0)
213         print "Uploading in the background"
214
215 def actually_upload(changes_files):
216     file_list = ""
217     suites = {}
218     component_mapping = {}
219     for component in Cnf.SubTree("Security-Install::ComponentMappings").List():
220         component_mapping[component] = Cnf["Security-Install::ComponentMappings::%s" % (component)]
221     uploads = {}; # uploads[uri] = file_list
222     changesfiles = {}; # changesfiles[uri] = file_list
223     package_list = {} # package_list[source_name][version]
224     changes_files.sort(utils.changes_compare)
225     for changes_file in changes_files:
226         changes_file = utils.validate_changes_file_arg(changes_file)
227         # Reset variables
228         components = {}
229         upload_uris = {}
230         file_list = []
231         Upload.init_vars()
232         # Parse the .dak file for the .changes file
233         Upload.pkg.changes_file = changes_file
234         Upload.update_vars()
235         files = Upload.pkg.files
236         changes = Upload.pkg.changes
237         dsc = Upload.pkg.dsc
238         # Build the file list for this .changes file
239         for file in files.keys():
240             poolname = os.path.join(Cnf["Dir::Root"], Cnf["Dir::PoolRoot"],
241                                     utils.poolify(changes["source"], files[file]["component"]),
242                                     file)
243             file_list.append(poolname)
244             orig_component = files[file].get("original component", files[file]["component"])
245             components[orig_component] = ""
246         # Determine the upload uri for this .changes file
247         for component in components.keys():
248             upload_uri = component_mapping.get(component)
249             if upload_uri:
250                 upload_uris[upload_uri] = ""
251         num_upload_uris = len(upload_uris.keys())
252         if num_upload_uris == 0:
253             utils.fubar("%s: No valid upload URI found from components (%s)."
254                         % (changes_file, ", ".join(components.keys())))
255         elif num_upload_uris > 1:
256             utils.fubar("%s: more than one upload URI (%s) from components (%s)."
257                         % (changes_file, ", ".join(upload_uris.keys()),
258                            ", ".join(components.keys())))
259         upload_uri = upload_uris.keys()[0]
260         # Update the file list for the upload uri
261         if not uploads.has_key(upload_uri):
262             uploads[upload_uri] = []
263         uploads[upload_uri].extend(file_list)
264         # Update the changes list for the upload uri
265         if not changesfiles.has_key(upload_uri):
266             changesfiles[upload_uri] = []
267         changesfiles[upload_uri].append(changes_file)
268         # Remember the suites and source name/version
269         for suite in changes["distribution"].keys():
270             suites[suite] = ""
271         # Remember the source name and version
272         if changes["architecture"].has_key("source") and \
273            changes["distribution"].has_key("testing"):
274             if not package_list.has_key(dsc["source"]):
275                 package_list[dsc["source"]] = {}
276             package_list[dsc["source"]][dsc["version"]] = ""
277
278     for uri in uploads.keys():
279         uploads[uri].extend(changesfiles[uri])
280         (host, path) = uri.split(":")
281         #        file_list = " ".join(uploads[uri])
282         print "Moving files to UploadQueue"
283         for filename in uploads[uri]:
284             utils.copy(filename, Cnf["Dir::Upload"])
285             remove_from_buildd(suites, filename)
286         #spawn("lftp -c 'open %s; cd %s; put %s'" % (host, path, file_list))
287
288     if not Options["No-Action"]:
289         filename = "%s/testing-processed" % (Cnf["Dir::Log"])
290         file = utils.open_file(filename, 'a')
291         for source in package_list.keys():
292             for version in package_list[source].keys():
293                 file.write(" ".join([source, version])+'\n')
294         file.close()
295
296 def remove_from_buildd(suites, filename):
297     """Check the buildd dir for each suite and remove the file if needed"""
298     builddbase = Cnf["Dir::QueueBuild"]
299     filebase = os.path.basename(filename)
300     for s in suites:
301         try:
302             os.unlink(os.path.join(builddbase, s, filebase))
303         except OSError, e:
304             utils.warn("Problem removing %s from buildd queue %s [%s]" % (filebase, s, str(e)))
305
306
307 def generate_advisory(template):
308     global changes, advisory
309
310     adv_packages = []
311     updated_pkgs = {};  # updated_pkgs[distro][arch][file] = {path,md5,size}
312
313     for arg in changes:
314         arg = utils.validate_changes_file_arg(arg)
315         Upload.pkg.changes_file = arg
316         Upload.init_vars()
317         Upload.update_vars()
318
319         src = Upload.pkg.changes["source"]
320         src_ver = "%s (%s)" % (src, Upload.pkg.changes["version"])
321         if src_ver not in adv_packages:
322             adv_packages.append(src_ver)
323
324         suites = Upload.pkg.changes["distribution"].keys()
325         for suite in suites:
326             if not updated_pkgs.has_key(suite):
327                 updated_pkgs[suite] = {}
328
329         files = Upload.pkg.files
330         for file in files.keys():
331             arch = files[file]["architecture"]
332             md5 = files[file]["md5sum"]
333             size = files[file]["size"]
334             poolname = Cnf["Dir::PoolRoot"] + \
335                 utils.poolify(src, files[file]["component"])
336             if arch == "source" and file.endswith(".dsc"):
337                 dscpoolname = poolname
338             for suite in suites:
339                 if not updated_pkgs[suite].has_key(arch):
340                     updated_pkgs[suite][arch] = {}
341                 updated_pkgs[suite][arch][file] = {
342                     "md5": md5, "size": size, "poolname": poolname }
343
344         dsc_files = Upload.pkg.dsc_files
345         for file in dsc_files.keys():
346             arch = "source"
347             if not dsc_files[file].has_key("files id"):
348                 continue
349
350             # otherwise, it's already in the pool and needs to be
351             # listed specially
352             md5 = dsc_files[file]["md5sum"]
353             size = dsc_files[file]["size"]
354             for suite in suites:
355                 if not updated_pkgs[suite].has_key(arch):
356                     updated_pkgs[suite][arch] = {}
357                 updated_pkgs[suite][arch][file] = {
358                     "md5": md5, "size": size, "poolname": dscpoolname }
359
360     if os.environ.has_key("SUDO_UID"):
361         whoami = long(os.environ["SUDO_UID"])
362     else:
363         whoami = os.getuid()
364     whoamifull = pwd.getpwuid(whoami)
365     username = whoamifull[4].split(",")[0]
366
367     Subst = {
368         "__ADVISORY__": advisory,
369         "__WHOAMI__": username,
370         "__DATE__": time.strftime("%B %d, %Y", time.gmtime(time.time())),
371         "__PACKAGE__": ", ".join(adv_packages),
372         "__DAK_ADDRESS__": Cnf["Dinstall::MyEmailAddress"]
373         }
374
375     if Cnf.has_key("Dinstall::Bcc"):
376         Subst["__BCC__"] = "Bcc: %s" % (Cnf["Dinstall::Bcc"])
377
378     adv = ""
379     archive = Cnf["Archive::%s::PrimaryMirror" % (utils.where_am_i())]
380     for suite in updated_pkgs.keys():
381         ver = Cnf["Suite::%s::Version" % suite]
382         if ver != "": ver += " "
383         suite_header = "%s %s(%s)" % (Cnf["Dinstall::MyDistribution"],
384                                        ver, suite)
385         adv += "%s\n%s\n\n" % (suite_header, "-"*len(suite_header))
386
387         arches = Cnf.ValueList("Suite::%s::Architectures" % suite)
388         if "source" in arches:
389             arches.remove("source")
390         if "all" in arches:
391             arches.remove("all")
392         arches.sort()
393
394         adv += "%s updates are available for %s.\n\n" % (
395                 suite.capitalize(), utils.join_with_commas_and(arches))
396
397         for a in ["source", "all"] + arches:
398             if not updated_pkgs[suite].has_key(a):
399                 continue
400
401             if a == "source":
402                 adv += "Source archives:\n\n"
403             elif a == "all":
404                 adv += "Architecture independent packages:\n\n"
405             else:
406                 adv += "%s architecture (%s)\n\n" % (a,
407                         Cnf["Architectures::%s" % a])
408
409             for file in updated_pkgs[suite][a].keys():
410                 adv += "  http://%s/%s%s\n" % (
411                                 archive, updated_pkgs[suite][a][file]["poolname"], file)
412                 adv += "    Size/MD5 checksum: %8s %s\n" % (
413                         updated_pkgs[suite][a][file]["size"],
414                         updated_pkgs[suite][a][file]["md5"])
415             adv += "\n"
416     adv = adv.rstrip()
417
418     Subst["__ADVISORY_TEXT__"] = adv
419
420     adv = utils.TemplateSubst(Subst, template)
421     return adv
422
423 def spawn(command):
424     if not re_taint_free.match(command):
425         utils.fubar("Invalid character in \"%s\"." % (command))
426
427     if Options["No-Action"]:
428         print "[%s]" % (command)
429     else:
430         (result, output) = commands.getstatusoutput(command)
431         if (result != 0):
432             utils.fubar("Invocation of '%s' failed:\n%s\n" % (command, output), result)
433
434
435 ##################### ! ! ! N O T E ! ! !  #####################
436 #
437 # These functions will be reinvoked by semi-priveleged users, be careful not
438 # to invoke external programs that will escalate privileges, etc.
439 #
440 ##################### ! ! ! N O T E ! ! !  #####################
441
442 def sudo(arg, fn, exit):
443     if Options["Sudo"]:
444         if advisory == None:
445             utils.fubar("Must set advisory name")
446         os.spawnl(os.P_WAIT, "/usr/bin/sudo", "/usr/bin/sudo", "-u", "dak", "-H",
447                   "/usr/local/bin/dak", "new-security-install", "-"+arg, "--", advisory)
448     else:
449         fn()
450     if exit:
451         quit()
452
453 def do_Approve(): sudo("A", _do_Approve, True)
454 def _do_Approve():
455     # 1. dump advisory in drafts
456     draft = "/org/security.debian.org/advisories/drafts/%s" % (advisory)
457     print "Advisory in %s" % (draft)
458     if not Options["No-Action"]:
459         adv_file = "./advisory.%s" % (advisory)
460         if not os.path.exists(adv_file):
461             adv_file = Cnf["Dir::Templates"]+"/security-install.advisory"
462         adv_fd = os.open(draft, os.O_RDWR|os.O_CREAT|os.O_EXCL, 0664)
463         os.write(adv_fd, generate_advisory(adv_file))
464         os.close(adv_fd)
465         adv_fd = None
466
467     # 2. run dak process-accepted on changes
468     print "Accepting packages..."
469     spawn("dak process-accepted -pa %s" % (" ".join(changes)))
470
471     # 3. run dak make-suite-file-list / apt-ftparchve / dak generate-releases
472     print "Updating file lists for apt-ftparchive..."
473     spawn("dak make-suite-file-list")
474     print "Updating Packages and Sources files..."
475     spawn("apt-ftparchive generate %s" % (utils.which_apt_conf_file()))
476     print "Updating Release files..."
477     spawn("dak generate-releases")
478     print "Triggering security mirrors..."
479     spawn("sudo -u archvsync -H /home/archvsync/signal_security")
480
481     # 4. chdir to done - do upload
482     if not Options["No-Action"]:
483         os.chdir(Cnf["Dir::Queue::Done"])
484     do_upload()
485
486 def do_Disembargo(): sudo("D", _do_Disembargo, True)
487 def _do_Disembargo():
488     if os.getcwd() != Cnf["Dir::Queue::Embargoed"].rstrip("/"):
489         utils.fubar("Can only disembargo from %s" % Cnf["Dir::Queue::Embargoed"])
490
491     dest = Cnf["Dir::Queue::Unembargoed"]
492     emb_q = database.get_or_set_queue_id("embargoed")
493     une_q = database.get_or_set_queue_id("unembargoed")
494
495     for c in changes:
496         print "Disembargoing %s" % (c)
497
498         Upload.init_vars()
499         Upload.pkg.changes_file = c
500         Upload.update_vars()
501
502         if "source" in Upload.pkg.changes["architecture"].keys():
503             print "Adding %s %s to disembargo table" % (Upload.pkg.changes["source"], Upload.pkg.changes["version"])
504             Upload.projectB.query("INSERT INTO disembargo (package, version) VALUES ('%s', '%s')" % (Upload.pkg.changes["source"], Upload.pkg.changes["version"]))
505
506         files = {}
507         for suite in Upload.pkg.changes["distribution"].keys():
508             if suite not in Cnf.ValueList("Dinstall::QueueBuildSuites"):
509                 continue
510             dest_dir = Cnf["Dir::QueueBuild"]
511             if Cnf.FindB("Dinstall::SecurityQueueBuild"):
512                 dest_dir = os.path.join(dest_dir, suite)
513             for file in Upload.pkg.files.keys():
514                 files[os.path.join(dest_dir, file)] = 1
515
516         files = files.keys()
517         Upload.projectB.query("BEGIN WORK")
518         for f in files:
519             Upload.projectB.query("UPDATE queue_build SET queue = %s WHERE filename = '%s' AND queue = %s" % (une_q, f, emb_q))
520         Upload.projectB.query("COMMIT WORK")
521
522         for file in Upload.pkg.files.keys():
523             utils.copy(file, os.path.join(dest, file))
524             os.unlink(file)
525
526     for c in changes:
527         utils.copy(c, os.path.join(dest, c))
528         os.unlink(c)
529         k = c[:-8] + ".dak"
530         utils.copy(k, os.path.join(dest, k))
531         os.unlink(k)
532
533 def do_Reject(): sudo("R", _do_Reject, True)
534 def _do_Reject():
535     global changes
536     for c in changes:
537         print "Rejecting %s..." % (c)
538         Upload.init_vars()
539         Upload.pkg.changes_file = c
540         Upload.update_vars()
541         files = {}
542         for suite in Upload.pkg.changes["distribution"].keys():
543             if suite not in Cnf.ValueList("Dinstall::QueueBuildSuites"):
544                 continue
545             dest_dir = Cnf["Dir::QueueBuild"]
546             if Cnf.FindB("Dinstall::SecurityQueueBuild"):
547                 dest_dir = os.path.join(dest_dir, suite)
548             for file in Upload.pkg.files.keys():
549                 files[os.path.join(dest_dir, file)] = 1
550
551         files = files.keys()
552
553         aborted = Upload.do_reject()
554         if not aborted:
555             os.unlink(c[:-8]+".dak")
556             for f in files:
557                 Upload.projectB.query(
558                     "DELETE FROM queue_build WHERE filename = '%s'" % (f))
559                 os.unlink(f)
560
561     print "Updating buildd information..."
562     spawn("/org/security.debian.org/dak/config/debian-security/cron.buildd")
563
564     adv_file = "./advisory.%s" % (advisory)
565     if os.path.exists(adv_file):
566         os.unlink(adv_file)
567
568 def do_DropAdvisory():
569     for c in changes:
570         Upload.init_vars()
571         Upload.pkg.changes_file = c
572         Upload.update_vars()
573         del Upload.pkg.changes["adv id"]
574         Upload.dump_vars(os.getcwd())
575     quit()
576
577 def do_Edit():
578     adv_file = "./advisory.%s" % (advisory)
579     if not os.path.exists(adv_file):
580         utils.copy(Cnf["Dir::Templates"]+"/security-install.advisory", adv_file)
581     editor = os.environ.get("EDITOR", "vi")
582     result = os.system("%s %s" % (editor, adv_file))
583     if result != 0:
584         utils.fubar("%s invocation failed for %s." % (editor, adv_file))
585
586 def do_Show():
587     adv_file = "./advisory.%s" % (advisory)
588     if not os.path.exists(adv_file):
589         adv_file = Cnf["Dir::Templates"]+"/security-install.advisory"
590     print "====\n%s\n====" % (generate_advisory(adv_file))
591
592 def do_Quit():
593     quit()
594
595 def main():
596     global changes
597
598     args = init()
599     extras = load_args(args)
600     if advisory:
601         load_adv_changes()
602     if extras:
603         if not advisory:
604             changes = extras
605         else:
606             if srcverarches == {}:
607                 if not yes_no("Create new advisory %s?" % (advisory)):
608                     print "Not doing anything, then"
609                     quit()
610             else:
611                 advisory_info()
612                 doextras = []
613                 for c in extras:
614                     if yes_no("Add %s to %s?" % (c, advisory)):
615                         doextras.append(c)
616                 extras = doextras
617             add_changes(extras)
618
619     if not advisory:
620         utils.fubar("Must specify an advisory id")
621
622     if not changes:
623         utils.fubar("No changes specified")
624
625     if Options["Approve"]:
626         advisory_info()
627         do_Approve()
628     elif Options["Reject"]:
629         advisory_info()
630         do_Reject()
631     elif Options["Disembargo"]:
632         advisory_info()
633         do_Disembargo()
634     elif Options["Drop-Advisory"]:
635         advisory_info()
636         do_DropAdvisory()
637     else:
638         while 1:
639             default = "Q"
640             opts = ["Approve", "Edit advisory"]
641             if os.path.exists("./advisory.%s" % advisory):
642                 default = "A"
643             else:
644                 default = "E"
645             if os.getcwd() == Cnf["Dir::Queue::Embargoed"].rstrip("/"):
646                 opts.append("Disembargo")
647             opts += ["Show advisory", "Reject", "Quit"]
648
649             advisory_info()
650             what = prompt(opts, default)
651
652             if what == "Quit":
653                 do_Quit()
654             elif what == "Approve":
655                 do_Approve()
656             elif what == "Edit advisory":
657                 do_Edit()
658             elif what == "Show advisory":
659                 do_Show()
660             elif what == "Disembargo":
661                 do_Disembargo()
662             elif what == "Reject":
663                 do_Reject()
664             else:
665                 utils.fubar("Impossible answer '%s', wtf?" % (what))
666
667 ################################################################################
668
669 if __name__ == '__main__':
670     main()
671
672 ################################################################################