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 version 2 as
   7 #    published by the Free Software Foundation;
   8 #
   9 #    This program is distributed in the hope that it will be useful,
  10 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 #    GNU General Public License for more details.
  13 #
  14 #    You should have received a copy of the GNU General Public License
  15 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 #
  17 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
  18
  19 import base64
  20 import socket
  21 import vsys
  22
  23 from optparse import OptionParser
  24
  25 STOP_MSG = "STOP"
  26
  27 def get_options():
  28     usage = ("usage: %prog -u <slicename> -N <vif-name> -t <vif-type> "
  29             "-D <delete> -S <socket-name>")
  30
  31     parser = OptionParser(usage = usage)
  32
  33     parser.add_option("-u", "--slicename", dest="slicename",
  34         help = "The name of the PlanetLab slice ",
  35         type="str")
  36
  37     parser.add_option("-N", "--vif-name", dest="vif_name",
  38         help = "The name of the virtual interface, or a "
  39                 "unique numeric identifier to name the interface "
  40                 "if GRE mode is used.",
  41         type="str")
  42
  43     parser.add_option("-t", "--vif-type", dest="vif_type",
  44             help = "Virtual interface type. Either IFF_TAP or IFF_TUN. "
  45             "Defaults to IFF_TAP. ", type="str")
  46
  47     parser.add_option("-D", "--delete", dest="delete",
  48             action="store_true",
  49             default = False,
  50             help="Removes virtual interface if GRE mode was used")
  51
  52     parser.add_option("-S", "--socket-name", dest="socket_name",
  53         help = "Name for the unix socket used to interact with this process",
  54         type="str")
  55
  56     (options, args) = parser.parse_args()
  57
  58     vif_type = vsys.IFF_TAP
  59     if options.vif_type and options.vif_type == "IFF_TUN":
  60         vif_type = vsys.IFF_TUN
  61
  62     return (options.socket_name, options.vif_name, options.slicename,
  63             vif_type, options.delete)
  64
  65 if __name__ == '__main__':
  66
  67     (socket_name, vif_name, slicename, vif_type, delete) = get_options()
  68
  69     # If a socket name is sent, send the STOP message and wait for a reply
  70     if socket_name:
  71         sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
  72         try:
  73             sock.connect(socket_name)
  74             encoded = base64.b64encode(STOP_MSG)
  75             sock.send("%s\n" % encoded)
  76             reply = sock.recv(1024)
  77             reply = base64.b64decode(reply)
  78             print reply
  79         except:
  80             print "Did not properly shutdown device"
  81     # If a slicename is provided, use it to remove a GRE device
  82     elif slicename:
  83         import pwd
  84         import getpass
  85
  86         sliceid = pwd.getpwnam(slicename).pw_uid
  87
  88         if vif_type == vsys.IFF_TAP:
  89             vif_prefix = "tap"
  90         else:
  91             vif_prefix = "tun"
  92
  93         # if_name should be a unique numeric vif id
  94         vif_name = "%s%s-%s" % (vif_prefix, sliceid, vif_name)
  95
  96         vsys.vif_down(vif_name, delete = True)
  97
  98     # Else, use the vsys interface to set the virtual interface down
  99     else:
 100         vsys.vif_down(vif_name)
 101
 102