modules/livecd_stage2_target.py

   1 """
   2 LiveCD stage2 target, builds upon previous LiveCD stage1 tarball
   3 """
   4 # NOTE: That^^ docstring has influence catalyst-spec(5) man page generation.
   5
   6 import os,string,types,stat,shutil
   7 from catalyst_support import *
   8 from generic_stage_target import *
   9
  10 class livecd_stage2_target(generic_stage_target):
  11         """
  12         Builder class for a LiveCD stage2 build.
  13         """
  14         def __init__(self,spec,addlargs):
  15                 self.required_values=["boot/kernel"]
  16
  17                 self.valid_values=[]
  18
  19                 self.valid_values.extend(self.required_values)
  20                 self.valid_values.extend(["livecd/cdtar","livecd/empty","livecd/rm",\
  21                         "livecd/unmerge","livecd/iso","livecd/gk_mainargs","livecd/type",\
  22                         "livecd/readme","livecd/motd","livecd/overlay",\
  23                         "livecd/modblacklist","livecd/splash_theme","livecd/rcadd",\
  24                         "livecd/rcdel","livecd/fsscript","livecd/xinitrc",\
  25                         "livecd/root_overlay","livecd/users","portage_overlay",\
  26                         "livecd/fstype","livecd/fsops","livecd/linuxrc","livecd/bootargs",\
  27                         "gamecd/conf","livecd/xdm","livecd/xsession","livecd/volid"])
  28
  29                 generic_stage_target.__init__(self,spec,addlargs)
  30                 if "livecd/type" not in self.settings:
  31                         self.settings["livecd/type"] = "generic-livecd"
  32
  33                 file_locate(self.settings, ["cdtar","controller_file"])
  34
  35         def set_source_path(self):
  36                 self.settings["source_path"]=normpath(self.settings["storedir"]+"/builds/"+self.settings["source_subpath"]+".tar.bz2")
  37                 if os.path.isfile(self.settings["source_path"]):
  38                         self.settings["source_path_hash"]=generate_hash(self.settings["source_path"])
  39                 else:
  40                         self.settings["source_path"]=normpath(self.settings["storedir"]+"/tmp/"+self.settings["source_subpath"]+"/")
  41                 if not os.path.exists(self.settings["source_path"]):
  42                         raise CatalystError,"Source Path: "+self.settings["source_path"]+" does not exist."
  43
  44         def set_spec_prefix(self):
  45             self.settings["spec_prefix"]="livecd"
  46
  47         def set_target_path(self):
  48                 self.settings["target_path"]=normpath(self.settings["storedir"]+"/builds/"+self.settings["target_subpath"]+"/")
  49                 if "AUTORESUME" in self.settings \
  50                         and os.path.exists(self.settings["autoresume_path"]+"setup_target_path"):
  51                                 print "Resume point detected, skipping target path setup operation..."
  52                 else:
  53                         # first clean up any existing target stuff
  54                         if os.path.isdir(self.settings["target_path"]):
  55                                 cmd("rm -rf "+self.settings["target_path"],
  56                                 "Could not remove existing directory: "+self.settings["target_path"],env=self.env)
  57                                 touch(self.settings["autoresume_path"]+"setup_target_path")
  58                         if not os.path.exists(self.settings["target_path"]):
  59                                 os.makedirs(self.settings["target_path"])
  60
  61         def run_local(self):
  62                 # what modules do we want to blacklist?
  63                 if "livecd/modblacklist" in self.settings:
  64                         try:
  65                                 myf=open(self.settings["chroot_path"]+"/etc/modprobe.d/blacklist.conf","a")
  66                         except:
  67                                 self.unbind()
  68                                 raise CatalystError,"Couldn't open "+self.settings["chroot_path"]+"/etc/modprobe.d/blacklist.conf."
  69
  70                         myf.write("\n#Added by Catalyst:")
  71                         for x in self.settings["livecd/modblacklist"]:
  72                                 myf.write("\nblacklist "+x)
  73                         myf.close()
  74
  75         def unpack(self):
  76                 unpack=True
  77                 display_msg=None
  78
  79                 clst_unpack_hash=read_from_clst(self.settings["autoresume_path"]+"unpack")
  80
  81                 if os.path.isdir(self.settings["source_path"]):
  82                         unpack_cmd="rsync -a --delete "+self.settings["source_path"]+" "+self.settings["chroot_path"]
  83                         display_msg="\nStarting rsync from "+self.settings["source_path"]+"\nto "+\
  84                                 self.settings["chroot_path"]+" (This may take some time) ...\n"
  85                         error_msg="Rsync of "+self.settings["source_path"]+" to "+self.settings["chroot_path"]+" failed."
  86                         invalid_snapshot=False
  87
  88                 if "AUTORESUME" in self.settings:
  89                         if os.path.isdir(self.settings["source_path"]) and \
  90                                 os.path.exists(self.settings["autoresume_path"]+"unpack"):
  91                                 print "Resume point detected, skipping unpack operation..."
  92                                 unpack=False
  93                         elif "source_path_hash" in self.settings:
  94                                 if self.settings["source_path_hash"] != clst_unpack_hash:
  95                                         invalid_snapshot=True
  96
  97                 if unpack:
  98                         self.mount_safety_check()
  99                         if invalid_snapshot:
 100                                 print "No Valid Resume point detected, cleaning up  ..."
 101                                 #os.remove(self.settings["autoresume_path"]+"dir_setup")
 102                                 self.clear_autoresume()
 103                                 self.clear_chroot()
 104                                 #self.dir_setup()
 105
 106                         if not os.path.exists(self.settings["chroot_path"]):
 107                                 os.makedirs(self.settings["chroot_path"])
 108
 109                         if not os.path.exists(self.settings["chroot_path"]+"/tmp"):
 110                                 os.makedirs(self.settings["chroot_path"]+"/tmp",1777)
 111
 112                         if "PKGCACHE" in self.settings:
 113                                 if not os.path.exists(self.settings["pkgcache_path"]):
 114                                         os.makedirs(self.settings["pkgcache_path"],0755)
 115
 116                         if not display_msg:
 117                                 raise CatalystError,"Could not find appropriate source. Please check the 'source_subpath' setting in the spec file."
 118
 119                         print display_msg
 120                         cmd(unpack_cmd,error_msg,env=self.env)
 121
 122                         if "source_path_hash" in self.settings:
 123                                 myf=open(self.settings["autoresume_path"]+"unpack","w")
 124                                 myf.write(self.settings["source_path_hash"])
 125                                 myf.close()
 126                         else:
 127                                 touch(self.settings["autoresume_path"]+"unpack")
 128
 129         def set_action_sequence(self):
 130                 self.settings["action_sequence"]=["unpack","unpack_snapshot",\
 131                                 "config_profile_link","setup_confdir","portage_overlay",\
 132                                 "bind","chroot_setup","setup_environment","run_local",\
 133                                 "build_kernel"]
 134                 if "FETCH" not in self.settings:
 135                         self.settings["action_sequence"] += ["bootloader","preclean",\
 136                                 "livecd_update","root_overlay","fsscript","rcupdate","unmerge",\
 137                                 "unbind","remove","empty","target_setup",\
 138                                 "setup_overlay","create_iso"]
 139                 self.settings["action_sequence"].append("clear_autoresume")
 140
 141 def register(foo):
 142         foo.update({"livecd-stage2":livecd_stage2_target})
 143         return foo