pintos-os.org Git - openvswitch/blob - tests/appctl.py

   1 # Copyright (c) 2012 Nicira Networks.
   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.util
  21 import ovs.vlog
  22
  23
  24 def connect_to_target(target):
  25     error, str_result = ovs.unixctl.socket_name_from_target(target)
  26     if error:
  27         ovs.util.ovs_fatal(error, str_result)
  28     else:
  29         socket_name = str_result
  30
  31     error, client = ovs.unixctl.UnixctlClient.create(socket_name)
  32     if error:
  33         ovs.util.ovs_fatal(error, "cannot connect to \"%s\"" % socket_name)
  34
  35     return client
  36
  37
  38 def main():
  39     parser = argparse.ArgumentParser(description="Python Implementation of"
  40                                      " ovs-appctl.")
  41     parser.add_argument("-t", "--target", default="ovs-vswitchd",
  42                         help="pidfile or socket to contact")
  43
  44     parser.add_argument("command", metavar="COMMAND",
  45                         help="Command to run.")
  46     parser.add_argument("argv", metavar="ARG", nargs="*",
  47                         help="Arguments to the command.")
  48     args = parser.parse_args()
  49
  50     ovs.vlog.Vlog.init()
  51     target = args.target
  52     client = connect_to_target(target)
  53     err_no, error, result = client.transact(args.command, args.argv)
  54     client.close()
  55
  56     if err_no:
  57         ovs.util.ovs_fatal(err_no, "%s: transaction error" % target)
  58     elif error is not None:
  59         sys.stderr.write(error)
  60         ovs.util.ovs_error(0, "%s: server returned an error" % target)
  61         sys.exit(2)
  62     else:
  63         assert result is not None
  64         sys.stdout.write(result)
  65
  66
  67 if __name__ == '__main__':
  68     main()