ó
¡ôXc @ s¤ d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z e d „ Z e d „ Z
d d „ Z e d „ Z
e d „ Z e d „ Z d S( iÿÿÿÿN( t
close_all_fdsc
C s) g } t j j j ƒ \ } } t j d t d | | | f d i | d 6ƒ } | j ƒ | j | j ƒ | j
ƒ | j
ƒ t j j j ƒ \ }
} t j d t ƒ \ } }
t j d t
d i | d 6d | | |
f ƒ } | j ƒ | j | j ƒ | j
ƒ | j
ƒ |
j
ƒ y4 t j d d g d |
d t j d
t d d ƒ} WnA t k
r‡t j d
d g d |
d t j d
t d d ƒ} n X|
j
ƒ | j | j ƒ t j d d d d d | d | g d | j d | d
t d d ƒ} | j j
ƒ | j | j ƒ x! | D] } t j j j | ƒ qW| S( Nt targett argst kwargst debugt duplext pigzs -ct stdint stdoutt close_fdst bufsizeiÿÿÿÿt gzipt opensslt encs -es -aes-128-cbcs -Ks -iv( t euca2oolst bundlet utilt open_pipe_fileobjst multiprocessingt Processt _create_tarball_from_streamt startt appendt pidt closet Pipet Falset _calc_sha1_for_pipet
subprocesst Popent PIPEt Truet OSErrorR t waitpid_in_thread( t infilet outfilet enc_keyt enc_ivt tarinfoR t pidst tar_out_rt tar_out_wt tar_pt digest_out_rt digest_out_wt digest_result_rt digest_result_wt digest_pR R R ( ( s? /usr/lib/python2.7/site-packages/euca2ools/bundle/pipes/core.pyt create_bundle_pipeline$ sL
c
C s g } t j d d d d d | d | g d | d t j d t d
d ƒ} | j | j ƒ | j ƒ y: t j d d
d g d | j d t j d t d
d ƒ} WnG t k
rå t j d d
d g d | j d t j d t d
d ƒ} n X| j | j ƒ | j j ƒ t j
j j ƒ \ } } t
j d t ƒ \ }
} t
j d t d i | d 6d | j | | f ƒ } | j ƒ | j | j ƒ | j j ƒ | j ƒ | j ƒ t
j d t d | | f d i | d 6ƒ }
|
j ƒ | j ƒ | j |
j ƒ x! | D] } t j
j j | ƒ qõW|
S( s˜
Create a pipeline to perform the unbundle operation on infile input.
The resulting unbundled image will be written to 'outfile'.
:param outfile: file obj to write unbundled image to
:param enc_key: the encryption key used to bundle the image
:param enc_iv: the encyrption initialization vector used in the bundle
:returns multiprocess pipe to read sha1 digest of written image
R R
s -ds -aes-128-cbcs -Ks -ivR R R R
iÿÿÿÿR s -cR R R R R R ( R R R R R R R R R R R R R R R R R R R t _extract_from_tarball_streamR! ( R" R# R$ R% R R' R R R+ R, R- R. R/ R* R ( ( s? /usr/lib/python2.7/site-packages/euca2ools/bundle/pipes/core.pyt create_unbundle_pipelineZ sH
c C s« d } | r | j ƒ n zm xf | j s„ | j t j ƒ } | r€ | j | ƒ | j ƒ | t | ƒ 7} | r | j | ƒ q q Pq WWd | rœ | j ƒ n | j
ƒ X| S( sÞ
Synchronously copy data from infile to outfile, updating a progress bar
with the total number of bytes copied along the way if one was provided,
and return the number of bytes copied.
This method must be run on the main thread.
:param infile: file obj to read input from
:param outfile: file obj to write output to
:param progressbar: progressbar object to update with i/o information
:param maxbytes: Int maximum number of bytes to write
i N( R t closedt readR t BUFSIZEt writet flusht lent updatet finishR ( R" R# t progressbart
bytes_writtent chunk( ( s? /usr/lib/python2.7/site-packages/euca2ools/bundle/pipes/core.pyt copy_with_progressbar– s"
c C sÎ t | | | g ƒ t j ƒ } z‰ yd xJ t rq | j t j ƒ } | rm | j | ƒ | j | ƒ | j ƒ q( Pq( W| j
| j ƒ ƒ Wn t k
r¦ | s d S‚ n XWd | j
ƒ | j
ƒ | j
ƒ Xd S( s¨
Read data from infile and write it to outfile, calculating a running SHA1
digest along the way. When infile hits end-of-file, send the digest in
hex form to result_mpconn and exit.
:param infile: file obj providing input for digest
:param outfile: file obj destination for writing output
:param digest_out_pipe_w: fileobj to write digest to
:param debug: boolean used in exception handling
N( R t hashlibt sha1R R4 R R5 R9 R6 R7 t sendt hexdigestt IOErrorR ( R" R# t digest_out_pipe_wR t digestR= ( ( s? /usr/lib/python2.7/site-packages/euca2ools/bundle/pipes/core.pyR ¹ s&
c C s– t d | | g ƒ t j d d d | d t j ƒ } z< y | j | d | ƒWn t k
rn | sh d S‚ n XWd | j ƒ | j ƒ | j ƒ Xd S( Nt
except_fdst modes w|t fileobjR
( R t tarfilet openR R5 t addfileRC R ( R" R# R& R t tarball( ( s? /usr/lib/python2.7/site-packages/euca2ools/bundle/pipes/core.pyR Ú s
c C sœ t | | g ƒ t j d d d | ƒ } zN y) t | ƒ } t j | j | ƒ | ƒ Wn t k
rt | sn d S‚ n XWd | j ƒ | j ƒ | j ƒ Xd S( sÝ
Perform tar extract on infile and write to outfile
:param infile: file obj providing input for tar
:param outfile: file obj destination for tar output
:param debug: boolean used in exception handling
RG s r|RH N( R RI RJ t nextt shutilt copyfileobjt extractfileRC R ( R" R# R RL R& ( ( s? /usr/lib/python2.7/site-packages/euca2ools/bundle/pipes/core.pyR1 ë s
( R? R RN R RI t euca2ools.bundle.utilR R R R0 R2 t NoneR> R R R1 ( ( ( s? /usr/lib/python2.7/site-packages/euca2ools/bundle/pipes/core.pyt <module> s 5 < # !