]>
jfr.im git - yt-dlp.git/blob - youtube_dl/update.py
4 from zipimport
import zipimporter
7 from .version
import __version__
9 def rsa_verify(message
, signature
, key
):
10 from struct
import pack
11 from hashlib
import sha256
12 from sys
import version_info
14 if version_info
[0] == 2: return x
15 else: return x
.encode('latin1')
16 assert(type(message
) == type(b('')))
22 signature
= pow(int(signature
, 16), key
[1], key
[0])
25 raw_bytes
.insert(0, pack("B", signature
& 0xFF))
27 signature
= (block_size
- len(raw_bytes
)) * b('\x00') + b('').join(raw_bytes
)
28 if signature
[0:2] != b('\x00\x01'): return False
29 signature
= signature
[2:]
30 if not b('\x00') in signature
: return False
31 signature
= signature
[signature
.index(b('\x00'))+1:]
32 if not signature
.startswith(b('\x30\x31\x30\x0D\x06\x09\x60\x86\x48\x01\x65\x03\x04\x02\x01\x05\x00\x04\x20')): return False
33 signature
= signature
[19:]
34 if signature
!= sha256(message
).digest(): return False
37 def update_self(to_screen
, verbose
, filename
):
38 """Update the program file with the latest version from the repository"""
40 UPDATE_URL
= "http://rg3.github.com/youtube-dl/update/"
41 VERSION_URL
= UPDATE_URL
+ 'LATEST_VERSION'
42 JSON_URL
= UPDATE_URL
+ 'versions.json'
43 UPDATES_RSA_KEY
= (0x9d60ee4d8f805312fdb15a62f87b95bd66177b91df176765d13514a0f1754bcd2057295c5b6f1d35daa6742c3ffc9a82d3e118861c207995a8031e151d863c9927e304576bc80692bc8e094896fcf11b66f3e29e04e3a71e9a11558558acea1840aec37fc396fb6b65dc81a1c4144e03bd1c011de62e3f1357b327d08426fe93, 65537)
46 if not isinstance(globals().get('__loader__'), zipimporter
) and not hasattr(sys
, "frozen"):
47 to_screen(u
'It looks like you installed youtube-dl with pip, setup.py or a tarball. Please use that to update.')
50 # Check if there is a new version
52 newversion
= compat_urllib_request
.urlopen(VERSION_URL
).read().decode('utf-8').strip()
54 if verbose
: to_screen(compat_str(traceback
.format_exc()))
55 to_screen(u
'ERROR: can\'t find the current version. Please try again later.')
57 if newversion
== __version__
:
58 to_screen(u
'youtube-dl is up-to-date (' + __version__
+ ')')
61 # Download and check versions info
63 versions_info
= compat_urllib_request
.urlopen(JSON_URL
).read().decode('utf-8')
64 versions_info
= json
.loads(versions_info
)
66 if verbose
: to_screen(compat_str(traceback
.format_exc()))
67 to_screen(u
'ERROR: can\'t obtain versions info. Please try again later.')
69 if not 'signature' in versions_info
:
70 to_screen(u
'ERROR: the versions file is not signed or corrupted. Aborting.')
72 signature
= versions_info
['signature']
73 del versions_info
['signature']
74 if not rsa_verify(json
.dumps(versions_info
, sort_keys
=True).encode('utf-8'), signature
, UPDATES_RSA_KEY
):
75 to_screen(u
'ERROR: the versions file signature is invalid. Aborting.')
78 to_screen(u
'Updating to version ' + versions_info
['latest'] + '...')
79 version
= versions_info
['versions'][versions_info
['latest']]
80 if version
.get('notes'):
81 to_screen(u
'PLEASE NOTE:')
82 for note
in version
['notes']:
85 if not os
.access(filename
, os
.W_OK
):
86 to_screen(u
'ERROR: no write permissions on %s' % filename
)
90 if hasattr(sys
, "frozen"):
91 exe
= os
.path
.abspath(filename
)
92 directory
= os
.path
.dirname(exe
)
93 if not os
.access(directory
, os
.W_OK
):
94 to_screen(u
'ERROR: no write permissions on %s' % directory
)
98 urlh
= compat_urllib_request
.urlopen(version
['exe'][0])
99 newcontent
= urlh
.read()
101 except (IOError, OSError) as err
:
102 if verbose
: to_screen(compat_str(traceback
.format_exc()))
103 to_screen(u
'ERROR: unable to download latest version')
106 newcontent_hash
= hashlib
.sha256(newcontent
).hexdigest()
107 if newcontent_hash
!= version
['exe'][1]:
108 to_screen(u
'ERROR: the downloaded file hash does not match. Aborting.')
112 with open(exe
+ '.new', 'wb') as outf
:
113 outf
.write(newcontent
)
114 except (IOError, OSError) as err
:
115 if verbose
: to_screen(compat_str(traceback
.format_exc()))
116 to_screen(u
'ERROR: unable to write the new version')
120 bat
= os
.path
.join(directory
, 'youtube-dl-updater.bat')
123 echo Updating youtube-dl...
124 ping 127.0.0.1 -n 5 -w 1000 > NUL
125 move /Y "%s.new" "%s"
127 \n""" %(exe
, exe
, bat
))
131 except (IOError, OSError) as err
:
132 if verbose
: to_screen(compat_str(traceback
.format_exc()))
133 to_screen(u
'ERROR: unable to overwrite current version')
137 elif isinstance(globals().get('__loader__'), zipimporter
):
139 urlh
= compat_urllib_request
.urlopen(version
['bin'][0])
140 newcontent
= urlh
.read()
142 except (IOError, OSError) as err
:
143 if verbose
: to_screen(compat_str(traceback
.format_exc()))
144 to_screen(u
'ERROR: unable to download latest version')
147 newcontent_hash
= hashlib
.sha256(newcontent
).hexdigest()
148 if newcontent_hash
!= version
['bin'][1]:
149 to_screen(u
'ERROR: the downloaded file hash does not match. Aborting.')
153 with open(filename
, 'wb') as outf
:
154 outf
.write(newcontent
)
155 except (IOError, OSError) as err
:
156 if verbose
: to_screen(compat_str(traceback
.format_exc()))
157 to_screen(u
'ERROR: unable to overwrite current version')
160 to_screen(u
'Updated youtube-dl. Restart youtube-dl to use the new version.')