blob: 080370ca9adee2dd953456d45d29e1fa7398e12e [file] [log] [blame]
apatrick@chromium.org1218a882010-07-29 08:02:22 +09001#!/usr/bin/env python
scottmg@chromium.orgd5e26de2012-05-23 16:08:56 +09002# Copyright (c) 2012 The Chromium Authors. All rights reserved.
apatrick@chromium.org1218a882010-07-29 08:02:22 +09003# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
5
maruel@chromium.org1f144a32011-11-24 04:13:44 +09006"""Extracts a single file from a CAB archive."""
apatrick@chromium.org1218a882010-07-29 08:02:22 +09007
8import os
grt@chromium.org509c7542011-12-14 07:20:46 +09009import shutil
apatrick@chromium.org1218a882010-07-29 08:02:22 +090010import subprocess
11import sys
apatrick@chromium.org38ff3d22011-12-13 09:34:15 +090012import tempfile
apatrick@chromium.org1218a882010-07-29 08:02:22 +090013
scottmg@chromium.orgd5e26de2012-05-23 16:08:56 +090014def run_quiet(*args):
mseaborn@chromium.org20697d02013-11-02 11:05:53 +090015 """Run 'expand' suppressing noisy output. Returns returncode from process."""
scottmg@chromium.orgd5e26de2012-05-23 16:08:56 +090016 popen = subprocess.Popen(args, stdout=subprocess.PIPE)
17 out, _ = popen.communicate()
18 if popen.returncode:
19 # expand emits errors to stdout, so if we fail, then print that out.
20 print out
21 return popen.returncode
apatrick@chromium.org3faf3c82011-12-13 12:32:12 +090022
maruel@chromium.org1f144a32011-11-24 04:13:44 +090023def main():
24 if len(sys.argv) != 4:
25 print 'Usage: extract_from_cab.py cab_path archived_file output_dir'
26 return 1
apatrick@chromium.org1218a882010-07-29 08:02:22 +090027
maruel@chromium.org1f144a32011-11-24 04:13:44 +090028 [cab_path, archived_file, output_dir] = sys.argv[1:]
apatrick@chromium.org1218a882010-07-29 08:02:22 +090029
grt@chromium.org509c7542011-12-14 07:20:46 +090030 # Expand.exe does its work in a fixed-named temporary directory created within
31 # the given output directory. This is a problem for concurrent extractions, so
32 # create a unique temp dir within the desired output directory to work around
33 # this limitation.
34 temp_dir = tempfile.mkdtemp(dir=output_dir)
35
apatrick@chromium.org38ff3d22011-12-13 09:34:15 +090036 try:
37 # Invoke the Windows expand utility to extract the file.
scottmg@chromium.orgd5e26de2012-05-23 16:08:56 +090038 level = run_quiet('expand', cab_path, '-F:' + archived_file, temp_dir)
grt@chromium.org509c7542011-12-14 07:20:46 +090039 if level == 0:
40 # Move the output file into place, preserving expand.exe's behavior of
41 # paving over any preexisting file.
42 output_file = os.path.join(output_dir, archived_file)
43 try:
44 os.remove(output_file)
45 except OSError:
46 pass
47 os.rename(os.path.join(temp_dir, archived_file), output_file)
apatrick@chromium.org38ff3d22011-12-13 09:34:15 +090048 finally:
grt@chromium.org509c7542011-12-14 07:20:46 +090049 shutil.rmtree(temp_dir, True)
50
51 if level != 0:
52 return level
maruel@chromium.org1f144a32011-11-24 04:13:44 +090053
54 # The expand utility preserves the modification date and time of the archived
55 # file. Touch the extracted file. This helps build systems that compare the
56 # modification times of input and output files to determine whether to do an
57 # action.
58 os.utime(os.path.join(output_dir, archived_file), None)
59 return 0
60
61
62if __name__ == '__main__':
63 sys.exit(main())