131 lines
3.6 KiB
Python
131 lines
3.6 KiB
Python
"""
|
|
Scyther : An automatic verifier for security protocols.
|
|
Copyright (C) 2007-2013 Cas Cremers
|
|
|
|
This program is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU General Public License
|
|
as published by the Free Software Foundation; either version 2
|
|
of the License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
"""
|
|
|
|
#
|
|
# FindDot.py
|
|
#
|
|
# Since version 2.31, GraphViz no longer sets the PATH environment variable on Windows.
|
|
# The sole reason of existence for this file is to solve this issue automatically if possible.
|
|
|
|
#---------------------------------------------------------------------------
|
|
""" Import externals """
|
|
import sys
|
|
import os
|
|
#---------------------------------------------------------------------------
|
|
""" Import internals """
|
|
from . import Misc
|
|
#---------------------------------------------------------------------------
|
|
|
|
DOTLOCATION = None
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def testDot(fpath):
|
|
|
|
try:
|
|
cmd = "%s -V" % (fpath)
|
|
(sts,sout,serr) = Misc.safeCommandOutput(cmd)
|
|
|
|
if sts != -1:
|
|
if "version" in str(sout) + str(serr):
|
|
return True
|
|
except:
|
|
pass
|
|
|
|
return False
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def scanPrefix(pf,name):
|
|
|
|
if pf.endswith("*"):
|
|
import glob
|
|
|
|
gl = glob.glob(pf)
|
|
for pf in gl:
|
|
for root,dirs,files in os.walk(pf):
|
|
for d in dirs:
|
|
npf = os.path.join(root,d)
|
|
res = scanPrefix(npf,name)
|
|
if res != None:
|
|
return res
|
|
|
|
return None
|
|
|
|
fpath = os.path.join(pf,name)
|
|
if len(pf) > 0:
|
|
fpath = "\"%s\"" % (fpath)
|
|
if testDot(fpath) == True:
|
|
return fpath
|
|
|
|
return None
|
|
|
|
|
|
def scanLocations():
|
|
if sys.platform.startswith("win"):
|
|
prefixes = ["", \
|
|
"C:\Program Files\Graphviz*", \
|
|
"C:\Program Files (x86)\Graphviz*" ]
|
|
name = "dot.exe"
|
|
else:
|
|
prefixes = [""]
|
|
name = "dot"
|
|
|
|
for pf in prefixes:
|
|
path = scanPrefix(pf,name)
|
|
if path != None:
|
|
return path
|
|
|
|
return None
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
def findDot():
|
|
global DOTLOCATION
|
|
|
|
# Cache the results
|
|
if DOTLOCATION != None:
|
|
return DOTLOCATION
|
|
|
|
DOTLOCATION = scanLocations()
|
|
if DOTLOCATION == None:
|
|
Misc.panic("""
|
|
Could not find the required 'dot' program, which is part of the Graphviz suite.
|
|
Please install it from http://www.graphviz.org/
|
|
|
|
Ubuntu users: install the 'graphviz' package.
|
|
|
|
Windows users: make sure that Graphviz is installed and
|
|
that the location of the 'dot' program is in
|
|
the PATH environment variable.
|
|
|
|
Restarting your system may be needed for Scyther to locate any newly installed
|
|
programs.
|
|
""")
|
|
return DOTLOCATION
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
if __name__ == '__main__':
|
|
Misc.panic(findDot())
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
# vim: set ts=4 sw=4 et list lcs=tab\:>-:
|