From 7b0d1bcc01c0c6144de630fdebb913bc256e389d Mon Sep 17 00:00:00 2001
From: Mike Hiretsky <mh@calculate.ru>
Date: Fri, 19 Nov 2010 15:34:11 +0300
Subject: [PATCH] Add autokill processes.

---
 i18n/cl_builder_ru.mo | Bin 13374 -> 13842 bytes
 pym/cl_builder.py     |  46 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 46 insertions(+)

diff --git a/i18n/cl_builder_ru.mo b/i18n/cl_builder_ru.mo
index feb3325e2a3ee85fd1b4b8dc7d65404880ece911..9d6b44bfe785547d248fd6ab40b6575dd0472fb1 100644
GIT binary patch
delta 2808
zcmaLXdrZ}39LMqJcp7@}!d*lujtQJRf{J*H6hRV0A{6f>D(L|U1O>uq%%ho_i<*-;
zwMLqn|46F|2-9)0xiz<1`<?09%5-y5ZmnF9OaJJP-k)=RY`Im>&iD8FU7p`_`99C@
z1hyTi_1}(t;;5lC6UoFkPGdT-btDJM?=E8|;vhbaV<L@7#Tob{uEJT^fU5V2eSHO^
zxc&ndqjR(|8CZxDun`lC@tZbI+}t>Adll2T9za!$9AiuhPQe%~KsT;N8Zgb6jcqs;
zuh{-#KaU?9e$R_)X933I227xTQ_o2jH}>Nc?6oW0M9t_vYNoDn#w4K!)p0&f#<i&D
zEvO70Lk*-Gm8sjPdQQ@z=V`W!CH<RCoVc+SHNp<te$)VNp=SOF)p2CBF;AfzNs?KF
zdaoGgV-2dKv#0@oiIeaqDx(ik?L^b4KaP`hPV^uLb-fbRU=?Zx2XQ=hqGoy(mBKs7
z<je!q`%Z3Zrpc(il8YKp1;*nCn1P?62J~Yr`6s*P5jPl=iJ*7wyF~P23QogiI0u_?
z9(Ljk96)s#PwPtgTzm#MqXu>e`I+M!vhh>Q#5<@7MJ14b+A!(lLltvS9hIU|xF2WY
zX;jLuqh>l}JDOJ1VJfC$2`XbvsQM?+jeVGi-{Tw{#1-iAb2o>R7F5c+Q6s;B8rWTQ
z<3prRGlAa8wwaDviab;*H=>@uiElWJIgHvnE)I(@74vZ`=HNMe0sVJ4QA)hDvJ%&!
zzJgBNix$>k8R?jX0nEZd%)=>+;yK)kg?I|}z2CF1y`(`)vlG?fhp0>q;BtNcqnWlw
zwgG43N2mc@N7^tCaVaLa!)slRO0geH@GLICAyh{*CWlkI9yQbR_$>a68n~OCsrzNP
zM&JK?_Kg8t!h;lgUxhv_#!IMO8$qw?AltSaHIRcGreh}-;Z4*8()btCgo;q_??z?n
zJnFN&hg0d_ILL=mHx0RG8c@6YIBKNVke_jS!b{*qHC&D_;}O(AenVv<mK~|*rKlOd
zjmqRjq>8zLtg877{rb!%FfDadfU2+?)$lnaDaJvi^_YsU=sB_<%@0TwGlcqlllfMa
zu}pjsTd)Pc!VQ>58kS%OX5-xq@?XeFB0KPT+=%=Ym@}vuU&L*A3)Mg&d8RLO1~t%c
zkw@lFT#KosO%rLw1K5iTF_&*RU-wa&4PXJ@pGp4J@htMR7<Zvkc8Nm?{)%%khpm4Y
zcj8eT!SI!d<EYd>K&3c|yfYRv4_mPknY_7+nsF2#|7t8iE!7D>C)yO>p#zmp?N$}N
zI9yn&(2IOEW~Y5q1Fj+_5~J*Un`|{(m2$!vItzb(I@Q4I32oBMJPt~=)`dT%(Cp`9
z4WSurA(X1+L`D0pk&_biFAzgS5!(snXIcB$QHlOa&QuzS210veD^W|#Bbo?hVKbrq
z5lNI1D%xjS;xa-@sIrDoc8iE&BA!roRd$3iui36*_t>Z5y-;UgCE!cMI6_O^Or#Jh
z+SEQGS{D?xz#nI*upptmVqcos?G9H;{-WVaTkXy<L^YvxOeR(k+MO!vh(aQcSV=5w
zuW-d>YUw8tE<)QZf>=uA5-P6}`T+HB@OszBuEdC#;_BL3cUq%+M??J<Ut^=Mv8y*K
z%Q3#px5HP_<gR)AY5Pyn-HE=2?bUS^P1SW(kH2>;rrQx8dZDtuwzhuPaFaP*p17)*
zq?5rtR-e@$+-G%LeZjY^9_w<j&FZ%T?UOz65q(w<t#!4fjvtY6(h67?s1e`|p7X@|
z;{RSw(#xSLf#3nFC%8AbS1qW$Q|TEFuXW7&+`7PnJ;8VAwx41ASN*R&9%pv%KdLiQ
aH_ct<z3$+?;QPTt|MGV5-L7liX4l`wR+|U_

delta 2473
zcmYk+du)?c7{~Faov!Odwv|H1=xjmDuCPn(WEX`o$8DRN3}q;I*|>~tZaRs0rJ%+L
ziHXucOmN02#A(!kjfRl8f2K>OC@~@NAJGVkWQl_QVbm<f@2`C^p6q!)=RG@p&pFRI
z?|X1yyf^-x@0vqK876|ndl_aU*x=zpIqo$pz!SI#PvcE^84J;$ZB~udsD9g=`@NXM
zeG*sV1eRmwb!Pb(#kpp2Yo!vVq1*8}jBtMx)$wx-;RVdYj9KaabCCyF4MwpQOK`8_
zyUzd5qJB4xdd>{`FnGOwA2+L@Qbj{E-iW(V9gd(@^Z{z6Q#c=gMh)zmZMF~tsQ=fZ
zGPn^nkzJ@vy^iWP;r#z6M>pvSGQNeVgs}lN!!3@_pe8Va8t^3EfL|fWuwPL9|H5TB
zKQ}#r^{9ys;R4)`%H%QBbH;HFp2fH-m#FALGpHE{$=5t=M6Ij~m66Ag$=iO^ie5#X
ziBqTvoI^j(qjNcyp(fCUB+YgqL)f#Z<CXA{|57SPXeh%;jN$KCfyK;D19hQN`Xm<P
zQPf1fL=E^YM)5aXf)N_EfKH^z`cVD$aMAM<sI8jvlmAL87imyx1LQ}ks&Q;b4cLo|
z@fB2tPNMps!!XWZ01Fv4hB3So`>_r`L1p+))Wq}YtcexHsf4N2BhRqSNEU4vmHOvV
zDSXrU|2RJGV((CAqm7MTj=k83W2gbAaScZ3q>Kz=C%%R{<X3PX#{E3KpUPpZMmKq>
z!Wgc=9axWJxC(znz0YOLPWOYTt@#Kw;P<FZ6_cJ0Y)4J(4ZIo8qb3j_9dwObJ(blo
zY{G>&ic0YWw&5jgz?yJ+phr=u9l<)hjI~(Gk2UeFsP@BnH-7Eh7jsgZx$nUZ_zt${
z{r`uGPU||RrIqY<97Rp!3tWm<a4m*tq-r}+zk3Olp;JiK?GMy*a`}oWLv^SR+CDBi
z)Ni9E_zf;*eDjenrM?!Gq5<R?Hi}yLq|=^JlwRR7Bz4w@{Mn0K*gZRrdhltd-NVVf
zhx-a-jdlPx;rn<mdYGlopY>4DjGyA7_wyLG;K$gCZr;%+upOIm3Zt0AJG%;-@OC_a
zyt#G;nX{e8Ui6Zt%{YvD-Or#V{&zX~*XdluWg~W=RxpMy;03%D`^bNz+EFQ-#wIM_
zMg!l6EAcJV8MugTn9sXigWK_C9Kl1llupXPr&Z)%sScBWrL-Bf*FE?&9z+&lIV@Kz
zuIF+W4xzT@Gt}O@**6y|i~16(=yz8O8BZNr4!=F*G_JrN?SBE0MW|?2D$CQP_AA4w
zEA=YMR6n6ZcpITqD;s=UQmelOcMw|8{e&{7Y-}M^u1#ZZOjKrN>AREq=;Yu|B1WuE
zv}Fb2-PBZuh(SW9<pDyws>D7_kfPM?uElIZCq_j_u8U|V^n$9aCzNlUnl*%<P`*_L
z(%5##4))clroUZ%&YkAlLd+(#>%&BdP|@Z*NaX5<qVum0x{BT{4#s~anrQTf8g;(@
zU$$~Hi`Yi!ECz`?2pw3J4QW!JZ&e!-qux17bO;s@USb`QN$6x=En11bMiJtX<PY9J
uW->ds(v|4UJCQ8(edtPh{7reuc*O5c{#sP(O6(~KCA&*J8Oh1=R`0)gYwS4y

diff --git a/pym/cl_builder.py b/pym/cl_builder.py
index 9279f80..0e5051d 100644
--- a/pym/cl_builder.py
+++ b/pym/cl_builder.py
@@ -34,6 +34,7 @@ from cl_distr import IsoDistributive, DirectoryDistributive, \
 from cl_template import template,iniParser
 from cl_vars_share import varsShare
 from datetime import datetime
+from time import sleep
 
 from cl_kernel_utils import KernelConfig,InitRamFs
 
@@ -261,6 +262,22 @@ class cl_builder(color_print):
                 return data[5]
         return ""
 
+    def getProcessByRoot(self,rootPath):
+        def psax(x):
+            try:
+                rootPath = path.join('/proc',x,'root')
+                cmdLinePath = path.join('/proc',x,'cmdline')
+                if path.exists(rootPath) and path.exists(cmdLinePath):
+                    return {'pid':x,
+                            'root':os.readlink(rootPath),
+                            'cmdline':open(cmdLinePath,'r').read().strip()}
+            except:
+                pass
+            return {'pid':x,'root':'','cmdline':''}
+        procid = filter(lambda x:x.isdigit(),os.listdir('/proc'))
+        return filter(lambda x:x['root'].startswith(rootPath),
+                      map(psax,procid))
+
     def printInfo(self):
         self.printSUCCESS(_("Creating image of") + " Calculate Linux")
         self.defaultPrint("%s\n"%_("System information"))
@@ -295,6 +312,16 @@ class cl_builder(color_print):
                 else:
                     self.printWARNING(_("WARNING") +": " +
                      _("image will be created on disk which mounted to root"))
+            warningProc = \
+                self.getProcessByRoot(self.clVars.Get('cl_builder_path'))
+            if warningProc:
+                self.printWARNING(_("WARNING") +": " +
+                    _("continuation of the operation will kill follow process:"))
+                for procObj in warningProc:
+                    self.printWARNING(
+                        " %s"%" ".join(
+                        map(lambda x:x.replace("\n"," ").strip(),
+                        procObj['cmdline'].split('\x00')))[:75])
 
             self.defaultPrint("\n")
         else:
@@ -378,6 +405,25 @@ class cl_builder(color_print):
                 self.assembleIso = False
                 self.printERROR(_("Interrupting the image creating"))
                 return False
+
+        killMap = [(0,15,_("Terminating %s processes")),
+                   (4,9,_("Kill %s processes")),
+                   (2,0,None),
+                   (10,9,_("Repeat kill %s processes"))]
+        listing = self.getProcessByRoot(self.clVars.Get('cl_builder_path'))
+        for waitTime,killCode,msg in killMap:
+            sleep(waitTime)
+            listing = filter(lambda x:path.exists(path.join('/proc',x['pid'])),
+                             listing)
+            if not listing:
+                break
+            if msg:
+                self.printSUCCESS(msg%",".join(map(lambda x:x['pid'],listing))
+                                  +" ...")
+            if not killCode:
+                continue
+            for i in listing:
+                os.kill(int(i['pid']),killCode)
         self.printMessageForTest(_("Prepare data for live image"))
         self.prepareSourceDistributive(self.sourceDistr)
         self.printByResult(True)