Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 1 | # Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
Ben Kwa | 966db08 | 2017-06-05 14:17:23 -0700 | [diff] [blame] | 5 | """This tool can be used to set up a base container for test. For example, |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 6 | python lxc.py -s -p /tmp/container |
| 7 | This command will download and setup base container in directory /tmp/container. |
| 8 | After that command finishes, you can run lxc command to work with the base |
| 9 | container, e.g., |
| 10 | lxc-start -P /tmp/container -n base -d |
| 11 | lxc-attach -P /tmp/container -n base |
| 12 | """ |
| 13 | |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 14 | import argparse |
| 15 | import logging |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 16 | |
| 17 | import common |
| 18 | from autotest_lib.client.bin import utils |
Ben Kwa | 966db08 | 2017-06-05 14:17:23 -0700 | [diff] [blame] | 19 | from autotest_lib.site_utils import lxc |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 20 | |
| 21 | |
| 22 | def parse_options(): |
| 23 | """Parse command line inputs. |
| 24 | |
| 25 | @raise argparse.ArgumentError: If command line arguments are invalid. |
| 26 | """ |
| 27 | parser = argparse.ArgumentParser() |
| 28 | parser.add_argument('-s', '--setup', action='store_true', |
| 29 | default=False, |
| 30 | help='Set up base container.') |
| 31 | parser.add_argument('-p', '--path', type=str, |
| 32 | help='Directory to store the container.', |
Ben Kwa | 966db08 | 2017-06-05 14:17:23 -0700 | [diff] [blame] | 33 | default=lxc.DEFAULT_CONTAINER_PATH) |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 34 | parser.add_argument('-f', '--force_delete', action='store_true', |
| 35 | default=False, |
| 36 | help=('Force to delete existing containers and rebuild ' |
| 37 | 'base containers.')) |
Dan Shi | 2a0f61e | 2016-06-21 14:47:21 -0700 | [diff] [blame] | 38 | parser.add_argument('-n', '--name', type=str, |
| 39 | help='Name of the base container.', |
Ben Kwa | 966db08 | 2017-06-05 14:17:23 -0700 | [diff] [blame] | 40 | default=lxc.BASE) |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 41 | options = parser.parse_args() |
| 42 | if not options.setup and not options.force_delete: |
| 43 | raise argparse.ArgumentError( |
| 44 | 'Use --setup to setup a base container, or --force_delete to ' |
| 45 | 'delete all containers in given path.') |
| 46 | return options |
| 47 | |
| 48 | |
| 49 | def main(): |
| 50 | """main script.""" |
Dan Shi | ca3be48 | 2015-05-05 23:23:53 -0700 | [diff] [blame] | 51 | # Force to run the setup as superuser. |
| 52 | # TODO(dshi): crbug.com/459344 Set remove this enforcement when test |
| 53 | # container can be unprivileged container. |
| 54 | if utils.sudo_require_password(): |
| 55 | logging.warn('SSP requires root privilege to run commands, please ' |
| 56 | 'grant root access to this process.') |
| 57 | utils.run('sudo true') |
| 58 | |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 59 | options = parse_options() |
Ben Kwa | d48cbcb | 2017-08-25 08:59:21 -0700 | [diff] [blame] | 60 | image = lxc.BaseImage(container_path=options.path) |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 61 | if options.setup: |
Ben Kwa | d48cbcb | 2017-08-25 08:59:21 -0700 | [diff] [blame] | 62 | image.setup(name=options.name, force_delete=options.force_delete) |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 63 | elif options.force_delete: |
Ben Kwa | d48cbcb | 2017-08-25 08:59:21 -0700 | [diff] [blame] | 64 | image.cleanup() |
Dan Shi | 767dced | 2015-02-01 00:21:07 -0800 | [diff] [blame] | 65 | |
| 66 | |
| 67 | if __name__ == '__main__': |
| 68 | main() |