OLD | NEW |
1 # coding: utf-8 | 1 # coding: utf-8 |
2 | 2 |
3 # This Source Code Form is subject to the terms of the Mozilla Public | 3 # This Source Code Form is subject to the terms of the Mozilla Public |
4 # License, v. 2.0. If a copy of the MPL was not distributed with this | 4 # License, v. 2.0. If a copy of the MPL was not distributed with this |
5 # file, You can obtain one at http://mozilla.org/MPL/2.0/. | 5 # file, You can obtain one at http://mozilla.org/MPL/2.0/. |
6 | 6 |
7 import os | 7 import os |
8 import re | 8 import re |
9 import json | 9 import json |
10 import ConfigParser | 10 import ConfigParser |
(...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
82 author=get_optional('general', 'author'), | 82 author=get_optional('general', 'author'), |
83 homepage=get_optional('general', 'homepage'), | 83 homepage=get_optional('general', 'homepage'), |
84 updateURL=get_optional('general', 'updateURL'), | 84 updateURL=get_optional('general', 'updateURL'), |
85 allowedDomains=allowedDomains, | 85 allowedDomains=allowedDomains, |
86 allowAllDomains=allowAllDomains, | 86 allowAllDomains=allowAllDomains, |
87 allowSecurePages=allowSecurePages, | 87 allowSecurePages=allowSecurePages, |
88 startScripts=(get_optional('contentScripts', 'document_start') or '').split(
), | 88 startScripts=(get_optional('contentScripts', 'document_start') or '').split(
), |
89 endScripts=(get_optional('contentScripts', 'document_end') or '').split(), | 89 endScripts=(get_optional('contentScripts', 'document_end') or '').split(), |
90 menus=parse_section('menus', 2), | 90 menus=parse_section('menus', 2), |
91 toolbarItems=parse_section('toolbar_items'), | 91 toolbarItems=parse_section('toolbar_items'), |
92 popovers=parse_section('popovers') | 92 popovers=parse_section('popovers'), |
| 93 developerIdentifier=params.get('developerIdentifier') |
93 ).encode('utf-8') | 94 ).encode('utf-8') |
94 | 95 |
95 def createBackgroundPage(params): | 96 def createBackgroundPage(params): |
96 template = getTemplate('background.html.tmpl', autoEscape=True) | 97 template = getTemplate('background.html.tmpl', autoEscape=True) |
97 return template.render( | 98 return template.render( |
98 backgroundScripts=params['metadata'].get( | 99 backgroundScripts=params['metadata'].get( |
99 'general', 'backgroundScripts' | 100 'general', 'backgroundScripts' |
100 ).split() | 101 ).split() |
101 ).encode('utf-8') | 102 ).encode('utf-8') |
102 | 103 |
103 def createInfoModule(params): | 104 def createInfoModule(params): |
104 template = getTemplate('safariInfo.js.tmpl') | 105 template = getTemplate('safariInfo.js.tmpl') |
105 return template.render(params).encode('utf-8') | 106 return template.render(params).encode('utf-8') |
106 | 107 |
107 def fixAbsoluteUrls(files): | 108 def fixAbsoluteUrls(files): |
108 for filename, content in files.iteritems(): | 109 for filename, content in files.iteritems(): |
109 if os.path.splitext(filename)[1].lower() == '.html': | 110 if os.path.splitext(filename)[1].lower() == '.html': |
110 files[filename] = re.sub( | 111 files[filename] = re.sub( |
111 r'(<[^<>]*?\b(?:href|src)\s*=\s*["\']?)\/+', | 112 r'(<[^<>]*?\b(?:href|src)\s*=\s*["\']?)\/+', |
112 r'\1' + '/'.join(['..'] * filename.count('/') + ['']), | 113 r'\1' + '/'.join(['..'] * filename.count('/') + ['']), |
113 content, re.S | re.I | 114 content, re.S | re.I |
114 ) | 115 ) |
115 | 116 |
116 def createSignedXarArchive(outFile, files, keyFile): | 117 def get_certificates_and_key(keyfile): |
| 118 import M2Crypto |
| 119 |
| 120 certs = [] |
| 121 bio = M2Crypto.BIO.openfile(keyfile) |
| 122 |
| 123 try: |
| 124 key = M2Crypto.RSA.load_key_bio(bio) |
| 125 bio.reset() |
| 126 while True: |
| 127 try: |
| 128 certs.append(M2Crypto.X509.load_cert_bio(bio)) |
| 129 except M2Crypto.X509.X509Error: |
| 130 break |
| 131 finally: |
| 132 bio.close() |
| 133 |
| 134 return certs, key |
| 135 |
| 136 def get_developer_identifier(certs): |
| 137 for cert in certs: |
| 138 subject = cert.get_subject() |
| 139 for entry in subject.get_entries_by_nid(subject.nid['CN']): |
| 140 m = re.match(r'Safari Developer: \((.*?)\)', entry.get_data().as_text()) |
| 141 if m: |
| 142 return m.group(1) |
| 143 |
| 144 raise Exception('No Safari developer certificate found in chain') |
| 145 |
| 146 def createSignedXarArchive(outFile, files, certs, key): |
117 import subprocess | 147 import subprocess |
118 import tempfile | 148 import tempfile |
119 import shutil | 149 import shutil |
120 import M2Crypto | 150 import M2Crypto |
121 | 151 |
122 # write files to temporary directory and create a xar archive | 152 # write files to temporary directory and create a xar archive |
123 dirname = tempfile.mkdtemp() | 153 dirname = tempfile.mkdtemp() |
124 try: | 154 try: |
125 for filename, contents in files.iteritems(): | 155 for filename, contents in files.iteritems(): |
126 path = os.path.join(dirname, filename) | 156 path = os.path.join(dirname, filename) |
127 | 157 |
128 try: | 158 try: |
129 os.makedirs(os.path.dirname(path)) | 159 os.makedirs(os.path.dirname(path)) |
130 except OSError: | 160 except OSError: |
131 pass | 161 pass |
132 | 162 |
133 with open(path, 'wb') as file: | 163 with open(path, 'wb') as file: |
134 file.write(contents) | 164 file.write(contents) |
135 | 165 |
136 subprocess.check_output( | 166 subprocess.check_output( |
137 ['xar', '-czf', os.path.abspath(outFile), '--distribution'] + os.listdir(d
irname), | 167 ['xar', '-czf', os.path.abspath(outFile), '--distribution'] + os.listdir(d
irname), |
138 cwd=dirname | 168 cwd=dirname |
139 ) | 169 ) |
140 finally: | 170 finally: |
141 shutil.rmtree(dirname) | 171 shutil.rmtree(dirname) |
142 | 172 |
143 certificate_filenames = [] | 173 certificate_filenames = [] |
144 try: | 174 try: |
145 # load key and certificates from the all-in-one key file | 175 # write each certificate in DER format to a separate |
146 # and write each certificate in DER format to a seperate | |
147 # temporary file, that they can be passed to xar | 176 # temporary file, that they can be passed to xar |
148 bio = M2Crypto.BIO.openfile(keyFile) | 177 for cert in certs: |
149 try: | 178 fd, filename = tempfile.mkstemp() |
150 key = M2Crypto.RSA.load_key_bio(bio) | 179 try: |
151 | 180 certificate_filenames.append(filename) |
152 bio.reset() | 181 os.write(fd, cert.as_der()) |
153 while True: | 182 finally: |
154 try: | 183 os.close(fd) |
155 cert = M2Crypto.X509.load_cert_bio(bio) | |
156 except M2Crypto.X509.X509Error: | |
157 break | |
158 | |
159 fd, filename = tempfile.mkstemp() | |
160 try: | |
161 certificate_filenames.append(filename) | |
162 os.write(fd, cert.as_der()) | |
163 finally: | |
164 os.close(fd) | |
165 finally: | |
166 bio.close() | |
167 | 184 |
168 # add certificates and placeholder signature | 185 # add certificates and placeholder signature |
169 # to the xar archive, and get data to sign | 186 # to the xar archive, and get data to sign |
170 fd, digestinfo_filename = tempfile.mkstemp() | 187 fd, digestinfo_filename = tempfile.mkstemp() |
171 os.close(fd) | 188 os.close(fd) |
172 try: | 189 try: |
173 subprocess.check_call( | 190 subprocess.check_call( |
174 [ | 191 [ |
175 'xar', '--sign', '-f', outFile, | 192 'xar', '--sign', '-f', outFile, |
176 '--digestinfo-to-sign', digestinfo_filename, | 193 '--digestinfo-to-sign', digestinfo_filename, |
(...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
234 | 251 |
235 if metadata.has_section('preprocess'): | 252 if metadata.has_section('preprocess'): |
236 files.preprocess( | 253 files.preprocess( |
237 [f for f, _ in metadata.items('preprocess')], | 254 [f for f, _ in metadata.items('preprocess')], |
238 {'needsExt': True} | 255 {'needsExt': True} |
239 ) | 256 ) |
240 | 257 |
241 if metadata.has_section('import_locales'): | 258 if metadata.has_section('import_locales'): |
242 importGeckoLocales(params, files) | 259 importGeckoLocales(params, files) |
243 | 260 |
| 261 if keyFile: |
| 262 certs, key = get_certificates_and_key(keyFile) |
| 263 params['developerIdentifier'] = get_developer_identifier(certs) |
| 264 |
244 files['lib/info.js'] = createInfoModule(params) | 265 files['lib/info.js'] = createInfoModule(params) |
245 files['background.html'] = createBackgroundPage(params) | 266 files['background.html'] = createBackgroundPage(params) |
246 files['Info.plist'] = createManifest(params, files) | 267 files['Info.plist'] = createManifest(params, files) |
247 | 268 |
248 fixAbsoluteUrls(files) | 269 fixAbsoluteUrls(files) |
249 | 270 |
250 dirname = metadata.get('general', 'basename') + '.safariextension' | 271 dirname = metadata.get('general', 'basename') + '.safariextension' |
251 for filename in files.keys(): | 272 for filename in files.keys(): |
252 files[os.path.join(dirname, filename)] = files.pop(filename) | 273 files[os.path.join(dirname, filename)] = files.pop(filename) |
253 | 274 |
254 if not devenv and keyFile: | 275 if not devenv and keyFile: |
255 createSignedXarArchive(outFile, files, keyFile) | 276 createSignedXarArchive(outFile, files, certs, key) |
256 else: | 277 else: |
257 files.zip(outFile) | 278 files.zip(outFile) |
OLD | NEW |