src/nepi/resources/planetlab/scripts/pl-vif-down.py

   1 #
   2 #    NEPI, a framework to manage network experiments
   3 #    Copyright (C) 2013 INRIA
   4 #
   5 #    This program is free software: you can redistribute it and/or modify
   6 #    it under the terms of the GNU General Public License as published by
   7 #    the Free Software Foundation, either version 3 of the License, or
   8 #    (at your option) any later version.
   9 #
  10 #    This program is distributed in the hope that it will be useful,
  11 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 #    GNU General Public License for more details.
  14 #
  15 #    You should have received a copy of the GNU General Public License
  16 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 #
  18 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
  19
  20 import base64
  21 import socket
  22 import vsys
  23
  24 from optparse import OptionParser
  25
  26 STOP_MSG = "STOP"
  27
  28 def get_options():
  29     usage = ("usage: %prog -N <vif-name> -D <delete> -S <socket-name>")
  30
  31     parser = OptionParser(usage = usage)
  32
  33     parser.add_option("-N", "--vif-name", dest="vif_name",
  34         help = "The name of the virtual interface, or a "
  35                 "unique numeric identifier to name the interface "
  36                 "if GRE mode is used.",
  37         type="str")
  38
  39     parser.add_option("-D", "--delete", dest="delete",
  40             action="store_true",
  41             default = False,
  42             help="Removes virtual interface if GRE mode was used")
  43
  44     parser.add_option("-S", "--socket-name", dest="socket_name",
  45         help = "Name for the unix socket used to interact with this process",
  46         default = "tap.sock", type="str")
  47
  48     (options, args) = parser.parse_args()
  49
  50     return (options.vif_name, options.delete, options.socket_name)
  51
  52 if __name__ == '__main__':
  53
  54     (vif_name, delete, socket_name) = get_options()
  55
  56     # If a socket name is sent, send the STOP message and wait for a reply
  57     if socket_name:
  58         sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
  59         sock.connect(socket_name)
  60         encoded = base64.b64encode(STOP_MSG)
  61         sock.send("%s\n" % encoded)
  62         reply = sock.recv(1024)
  63         reply = base64.b64decode(reply)
  64         print reply
  65
  66     # Else, use the vsys interface to set the virtual interface down
  67     elif vif_name:
  68         vsys.vif_down(vif_name, delete = delete)
  69
  70