U ٪zc-@sddlmZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZmZmZGdddeZed kreddZed ed d dS) )TemplateN)gettext) SourcesList)get_disturl_downloadablecountry_mirrorc@seZdZdZdZdZdejkZddZ ddZ d d Z d d Z d dZ d"ddZddZddZddZddZddZddZddZd d!ZdS)#DistUpgradeFetcherCorez2 base class (without GUI) for the upgrade fetcher http://archive.ubuntu.com/ubuntumainZDEBUG_UPDATE_MANAGERcCs||_t|_||_g|_dS)N)new_distrcurrent_dist_name _progress run_options)selfr ZprogressrD/usr/lib/python3/dist-packages/DistUpgrade/DistUpgradeFetcherCore.py__init__-szDistUpgradeFetcherCore.__init__cCs|jrtj|ddS)z" helper to show debug information  N)DEBUGsysstderrwrite)rmsgrrr_debug4szDistUpgradeFetcherCore._debugcCsdS)NTrrrrrshowReleaseNotes9sz'DistUpgradeFetcherCore.showReleaseNotescCst|t|dS)z dummy implementation for error display, should be overwriten by subclasses that want to more fancy method F)print)rZsummarymessagerrrerror<szDistUpgradeFetcherCore.errorcCsv|jjrr|jdtj|jj}|jdtj|jj}ttdtj|tj|d| ||rrdSdS)N/z0authenticate '%(file)s' against '%(signature)s' )file signatureTF) r upgradeToolSigtmpdirospathbasename upgradeToolr_gpgauthenticate)rfZsigrrr authenticateDs   z#DistUpgradeFetcherCore.authenticateNcCs@dg}|sd}|d|g7}|d||g7}tj|tjd}|dkS)zx authenticated a file against a given signature, if no keyring is given use the apt default keyring zapt-keyz./usr/share/keyrings/ubuntu-archive-keyring.gpgz --keyringZverify)rr) subprocesscallPIPE)rr!r"ZkeyringZgpgZretrrrr*Ps z&DistUpgradeFetcherCore.gpgauthenticatec Cstj|jtj|j}ttdtj|tj|sBdSz@t |jdtj|jd}|D]}| |qh| Wn8t j k r}ztd|WYdSd}~XYnXdS)Nzextracting '%s'Fr rzfailed to open tarfile (%s)T)r%r&joinr$r'urirr)existstarfileopenextractcloseZ ReadErrorloggingr)rfnameZtarZtarinfoerrrextractDistUpgrader`s"     z*DistUpgradeFetcherCore.extractDistUpgradercCsHd|j|jjf|_}tj|sD|tdtddtdSdS)Nz%s/%szCould not run the upgrade toolz. zThis is most likely a bug in the upgrade tool. Please report it as a bug using the command 'ubuntu-bug ubuntu-release-upgrader-core'.T) r$r namescriptr%r&r3rr))rr=rrrverifyDistUpraderqs   z(DistUpgradeFetcherCore.verifyDistUpradercCs|d|jtdd}t}|jD]}|js&|js&|jdksBq&|j|krNq&|j |rz|j |jkrz|j |j krz|S|j |jkr&d|j kr&|j|t |d}t||jr|S||jq&|ddS) a try to figure what the mirror is from current sources.list do this by looing for matching DEFAULT_COMPONENT, current dist in sources.list and then doing a http HEAD/ftp size request to see if the uri is available on this server zmirror_from_sources_list: %sF)Z withMatcherZdebr Nzno mirror found)rr rsetlistZdisabledZinvalidtyper2 startswithZdistDEFAULT_COMPONENTcompslenradd)rr2Z default_uriZsourcesseenr:Z mirror_urirrrmirror_from_sources_lists*        z/DistUpgradeFetcherCore.mirror_from_sources_listc Cs||jr,|d|||j}|r,|St|}t}|j|d}zt||js`td|Wn<tk r}z|d||jdd}W5d}~XYnX|S)z expand the uri so that it uses a mirror if the url starts with a well known string (like archive.ubuntu.com) ztrying to find suitable mirror)Z countrymirrorzfailed to download %sz url '%s' could not be downloadedr?N) rCDEFAULT_MIRRORrrIrrZsafe_substituter Exception)rr2Znew_uriZ uri_templatemr:rrr _expandUris    z!DistUpgradeFetcherCore._expandUricCsNtjdd}||_t|td||jdkrPtj ddtj ddt |j }|j jdk r||j j}tj||td d }||j jdk rJ||j j|_tj||jtd d }||}||jkrtd |d Stj|j jtj|j jfD]:}tj|r.tj|dks td|d Sq dSd S)z. download the tarball with the upgrade script zubuntu-release-upgrader-)prefixzusing tmpdir: '%s'rzDebug::Acquire::http1zDebug::Acquire::ftpNzUpgrade tool signature)Zdescrz Upgrade toolzfetch result != continue (%s)Fzfile '%s' missingT)tempfileZmkdtempr$r%chdirr8debugrapt_pkgZconfigr@ZAcquirerr r#rMZ AcquireFiler)r(r2runZRESULT_CONTINUEZwarningr&r'r3getsize)rr$Zfetcherr2Zaf1Zaf2resultr+rrrfetchDistUpgradersD        z(DistUpgradeFetcherCore.fetchDistUpgradercCsB|jg|j}tdkr0tdddg|nt|j|dS)Nrz /usr/bin/sudoZsudoz-E)r=rr%getuidexecv)rargsrrrrunDistUpgraders z&DistUpgradeFetcherCore.runDistUpgradercCstdt|jdS)Nz..)r%rQshutilZrmtreer$rrrrcleanups zDistUpgradeFetcherCore.cleanupc Cs>|s dS|s,|tdtddS|sT|tdtd|dS|st|tdtddS|s|tdtd|dSz4t |j tj st d |j }d |_ ||Wnht k r8}zH|j d kr|td td WY"d S|td td|jW5d}~XYnXdS)NzFailed to fetchz=Fetching the upgrade failed. There may be a network problem. zAuthentication failedz_Authenticating the upgrade failed. There may be a problem with the network or with the server. zFailed to extractz[Extracting the upgrade failed. There may be a problem with the network or with the server. zVerification failedz[Verifying the upgrade failed. There may be a problem with the network or with the server. zCan not execute '%s' zCan not run the upgradezyThis usually is caused by a system where /tmp is mounted noexec. Please remount without noexec and run the upgrade again.FzThe error message is '%s'.T)rrWrr)r,r]r;r>r%accessr=X_OKOSErrorerrnor[strerror)rexr:rrrrTsP          zDistUpgradeFetcherCore.run)N)__name__ __module__ __qualname____doc__rJrDr%environrrrrrr,r*r;r>rIrMrWr[r]rTrrrrr &s&  'r __main__zgot mirror: '%s'zghttp://archive.ubuntu.com/ubuntu/dists/intrepid-proposed/main/dist-upgrader-all/0.93.34/intrepid.tar.gzr )stringrr%rSr8r4rPr\rr-rr)Zaptsources.sourceslistrZutilsrrrobjectr redrrIrrrrs*   v