1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
|
# DistUpgradeView.py
#
# Copyright (c) 2004,2005 Canonical
#
# Author: Michael Vogt <michael.vogt@ubuntu.com>
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 2 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
# USA
import apt
import logging
import time
import sys
from DistUpgradeView import DistUpgradeView, InstallProgress
from DistUpgradeConfigParser import DistUpgradeConfig
import os
import pty
import apt_pkg
import select
import fcntl
import string
import re
import subprocess
import copy
class NonInteractiveFetchProgress(apt.progress.FetchProgress):
def updateStatus(self, uri, descr, shortDescr, status):
#logging.debug("Fetch: updateStatus %s %s" % (uri, status))
pass
class NonInteractiveInstallProgress(InstallProgress):
def __init__(self):
InstallProgress.__init__(self)
logging.debug("seting up environ for non-interactive use")
os.environ["DEBIAN_FRONTEND"] = "noninteractive"
os.environ["APT_LISTCHANGES_FRONTEND"] = "none"
self.config = DistUpgradeConfig(".")
if self.config.getboolean("NonInteractive","ForceOverwrite"):
apt_pkg.Config.Set("DPkg::Options::","--force-overwrite")
# default to 1200 sec timeout
self.timeout = 1200
try:
self.timeout = self.config.getint("NonInteractive","TerminalTimeout")
except Exception, e:
pass
def error(self, pkg, errormsg):
logging.error("got a error from dpkg for pkg: '%s': '%s'" % (pkg, errormsg))
# re-run maintainer script with sh -x to get a better idea
# what went wrong
#
# FIXME: this is just a approximation for now, we also need
# to pass:
# - a version after configure
# - a version after remove (if upgrade to new version)
#
# not everything is a shell script
#
# if the new preinst fails, its not yet in /var/lib/dpkg/info
# so this is inaccurate as well
prefix = "/var/lib/dpkg/info/"
error_map = { "post-installation" : ("postinst","configure"),
"pre-installation" : ("preinst", "configure"),
"pre-removal" : ("prerm","remvove"),
"post-removal": ("postrm","remove"),
}
for msg in error_map:
if msg in errormsg:
environ = copy.copy(os.environ)
maintainer_script = "%s/%s.%s" % (prefix, pkg, error_map[msg][0])
interp = open(maintainer_script).readline()[2:].strip().split()[0]
if ("bash" in interp) or ("/bin/sh" in interp):
debug_opts = "-x"
elif ("perl" in interp):
debug_opts = "-d"
environ["PERLDB_OPTS"] = "AutoTrace NonStop"
else:
logging.warning("unknown interpreter: '%s'" % interp)
logging.debug("re-runing %s with %s %s (%s)" % (error_map[msg][0], interp, debug_opts, environ))
cmd = [interp, debug_opts,
maintainer_script,
error_map[msg][1],
]
print cmd
ret = subprocess.call(cmd, env=environ)
logging.debug("%s script returned: %s" % (error_map[msg][0],ret))
def conffile(self, current, new):
logging.warning("got a conffile-prompt from dpkg for file: '%s'" % current)
try:
# don't overwrite
os.write(self.master_fd,"n\n")
except Exception, e:
logging.error("error '%s' when trying to write to the conffile"%e)
def startUpdate(self):
self.last_activity = time.time()
def updateInterface(self):
if self.statusfd == None:
return
if (self.last_activity + self.timeout) < time.time():
logging.warning("no activity %s seconds (%s) - sending ctrl-c" % (self.timeout, self.status))
os.write(self.master_fd,chr(3))
# read status fd from dpkg
# from python-apt/apt/progress.py (but modified a bit)
# -------------------------------------------------------------
res = select.select([self.statusfd],[],[],0.1)
while len(res[0]) > 0:
self.last_activity = time.time()
while not self.read.endswith("\n"):
self.read += os.read(self.statusfd.fileno(),1)
if self.read.endswith("\n"):
s = self.read
#print s
(status, pkg, percent, status_str) = string.split(s, ":")
if status == "pmerror":
self.error(pkg,status_str)
elif status == "pmconffile":
# we get a string like this:
# 'current-conffile' 'new-conffile' useredited distedited
match = re.compile("\s*\'(.*)\'\s*\'(.*)\'.*").match(status_str)
if match:
self.conffile(match.group(1), match.group(2))
elif status == "pmstatus":
if (float(percent) != self.percent or
status_str != self.status):
self.statusChange(pkg, float(percent), status_str.strip())
self.percent = float(percent)
self.status = string.strip(status_str)
sys.stdout.write("[%s] %s: %s\n" % (float(percent), pkg, status_str.strip()))
sys.stdout.flush()
self.read = ""
res = select.select([self.statusfd],[],[],0.1)
# -------------------------------------------------------------
#fcntl.fcntl(self.master_fd, fcntl.F_SETFL, os.O_NDELAY)
# read master fd (terminal output)
res = select.select([self.master_fd],[],[],0.1)
while len(res[0]) > 0:
self.last_activity = time.time()
try:
s = os.read(self.master_fd, 1)
sys.stdout.write("%s" % s)
except OSError,e:
# happens after we are finished because the fd is closed
return
res = select.select([self.master_fd],[],[],0.1)
sys.stdout.flush()
def fork(self):
logging.debug("doing a pty.fork()")
(self.pid, self.master_fd) = pty.fork()
if self.pid != 0:
logging.debug("pid is: %s" % self.pid)
return self.pid
class DistUpgradeViewNonInteractive(DistUpgradeView):
" non-interactive version of the upgrade view "
def __init__(self):
self.config = DistUpgradeConfig(".")
def getOpCacheProgress(self):
" return a OpProgress() subclass for the given graphic"
return apt.progress.OpProgress()
def getFetchProgress(self):
" return a fetch progress object "
return NonInteractiveFetchProgress()
def getInstallProgress(self, cache=None):
" return a install progress object "
return NonInteractiveInstallProgress()
def updateStatus(self, msg):
""" update the current status of the distUpgrade based
on the current view
"""
pass
def setStep(self, step):
""" we have 5 steps current for a upgrade:
1. Analyzing the system
2. Updating repository information
3. Performing the upgrade
4. Post upgrade stuff
5. Complete
"""
pass
def confirmChanges(self, summary, changes, downloadSize, actions=None):
DistUpgradeView.confirmChanges(self, summary, changes, downloadSize, actions)
logging.debug("toinstall: '%s'" % self.toInstall)
logging.debug("toupgrade: '%s'" % self.toUpgrade)
logging.debug("toremove: '%s'" % self.toRemove)
return True
def askYesNoQuestion(self, summary, msg):
" ask a Yes/No question and return True on 'Yes' "
return True
def confirmRestart(self):
" generic ask about the restart, can be overriden "
logging.debug("confirmRestart() called")
# ignore if we don't have this option
try:
# rebooting here makes sense if we run e.g. in qemu
return self.config.getboolean("NonInteractive","RealReboot")
except Exception, e:
logging.debug("no RealReboot found, returning false (%s) " % e)
return False
def error(self, summary, msg, extended_msg=None):
" display a error "
logging.error("%s %s (%s)" % (summary, msg, extended_msg))
def abort(self):
logging.error("view.abort called")
if __name__ == "__main__":
view = DistUpgradeViewNonInteractive()
fp = NonInteractiveFetchProgress()
ip = NonInteractiveInstallProgress()
ip.error("linux-image-2.6.17-10-generic","post-installation script failed")
cache = apt.Cache()
for pkg in sys.argv[1:]:
#if cache[pkg].isInstalled:
# cache[pkg].markDelete()
#else:
cache[pkg].markInstall()
cache.commit(fp,ip)
time.sleep(2)
sys.exit(0)
|
