#!/usr/bin/python

"""
Unfortunately, Mininet and OpenFlow don't always clean up
properly after themselves. Until they do (or until cleanup
functionality is integrated into the python code), this
script may be used to get rid of unwanted garbage. It may
also get rid of 'false positives', but hopefully nothing
irreplaceable!
"""

from subprocess import Popen, PIPE
import re

def sh( cmd ): 
   "Print a command and send it to the shell"
   print cmd
   return Popen( [ '/bin/sh', '-c', cmd ], 
      stdout=PIPE ).communicate()[ 0 ]

def cleanUpScreens():
   "Remove moldy old screen sessions."      
   r = r'(\d+.[hsc]\d+)'
   output = sh( 'screen -ls' ).split( '\n' )
   for line in output:
      m = re.search( r, line )
      if m is not None:
         quietRun( 'screen -S ' + m.group( 1 ) + ' -X kill' )
 
def cleanup():
   print "*** Removing all links of the pattern foo-ethX"
   links = sh( "ip link show | egrep -o '(\w+-eth\w+)'" ).split( '\n' )
   for link in links: 
      if link != '': sh( "ip link del " + link )

   print "*** Removing excess controllers/ofprotocols/ofdatapaths/pings/noxes"
   zombies = 'controller ofprotocol ofdatapath ping nox_core lt-nox_core'
   # Note: real zombie processes can't actually be killed, since they 
   # are already (un)dead. Then again,
   # you can't connect to them either, so they're mostly harmless.
   sh( 'killall -9 ' + zombies + ' 2> /dev/null' )

   print "*** Removing excess kernel datapaths"
   dps = sh( "ps ax | egrep -o 'dp[0-9]+' | sed 's/dp/nl:/'" ).split( '\n')
   for dp in dps: 
      if dp != '': sh( 'dpctl deldp ' + dp )

   print "*** Removing junk from /tmp"
   sh( 'rm -f /tmp/vconn* /tmp/vlogs* /tmp/*.out /tmp/*.log' )

   print "*** Removing old screen sessions"
   cleanUpScreens()

   print "*** Cleanup complete."

if __name__ == "__main__":
   cleanup()