tests/appctl.py

   1 # Copyright (c) 2012 Nicira, Inc.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at:
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 import argparse
  16 import sys
  17
  18 import ovs.daemon
  19 import ovs.unixctl
  20 import ovs.unixctl.client
  21 import ovs.util
  22 import ovs.vlog
  23 from ovs.fatal_signal import signal_alarm
  24
  25
  26 def connect_to_target(target):
  27     error, str_result = ovs.unixctl.socket_name_from_target(target)
  28     if error:
  29         ovs.util.ovs_fatal(error, str_result)
  30     else:
  31         socket_name = str_result
  32
  33     error, client = ovs.unixctl.client.UnixctlClient.create(socket_name)
  34     if error:
  35         ovs.util.ovs_fatal(error, "cannot connect to \"%s\"" % socket_name)
  36
  37     return client
  38
  39
  40 def main():
  41     parser = argparse.ArgumentParser(description="Python Implementation of"
  42                                      " ovs-appctl.")
  43     parser.add_argument("-t", "--target", default="ovs-vswitchd",
  44                         help="pidfile or socket to contact")
  45
  46     parser.add_argument("command", metavar="COMMAND",
  47                         help="Command to run.")
  48     parser.add_argument("argv", metavar="ARG", nargs="*",
  49                         help="Arguments to the command.")
  50     parser.add_argument("-T", "--timeout", metavar="SECS",
  51                         help="wait at most SECS seconds for a response")
  52     args = parser.parse_args()
  53
  54     if args.timeout:
  55         signal_alarm(int(args.timeout))
  56
  57     ovs.vlog.Vlog.init()
  58     target = args.target
  59     client = connect_to_target(target)
  60     err_no, error, result = client.transact(args.command, args.argv)
  61     client.close()
  62
  63     if err_no:
  64         ovs.util.ovs_fatal(err_no, "%s: transaction error" % target)
  65     elif error is not None:
  66         sys.stderr.write(error)
  67         ovs.util.ovs_error(0, "%s: server returned an error" % target)
  68         sys.exit(2)
  69     else:
  70         assert result is not None
  71         sys.stdout.write(result)
  72
  73
  74 if __name__ == '__main__':
  75     main()