import urllib, os, sys, socket, _winreg, string, psyco

import urllib, os, sys, socket, _winreg, string, psyco

psyco.full()

# Change the user agent to SpySweeper to avoid detection
class AppURLopener(urllib.FancyURLopener):
def __init__(self, *args):
self.version = "SpySweeper"
urllib.FancyURLopener.__init__(self, *args)

urllib._urlopener = AppURLopener()

# Set a few defaults
default_path = "C:\\Program Files\\Webroot\\Spy Sweeper"
host = "www.webroot.com"
force = False

# Process command line arguments
if len(sys.argv) > 0:
for x in sys.argv:
x = string.lower(x)
if x == "-f" or x == "-force":
force = True
elif x == string.lower(sys.argv[0]):
pass
else:
print "\nInvalid command line argument,", x
os._exit(1)

# Set up screen
os.system('cls')
print "Updating Webroot SpySweeper\n"

# Get SpySweeper location
try:
hkey = _winreg.OpenKey(_winreg.HKEY_CURRENT_USER, 'Software\Webroot\SpySweeper')
ssID = _winreg.QueryValueEx(hkey, 'id')
ssPath = ssID[0]
except WindowsError, (errno, strerror):
if errno == 2:
if not os.path.exists(default_path + "\\SpySweeper.exe"):
print "Error: SpySweeper not found in registry"
sys.exit(1)
else:
ssPath = default_path
else:
print "Error: ", (errno, strerror)
sys.exit(1)

# Verify that masters.mst is where it's supposed to be.
mstPath = ssPath + '\\masters\\'
if not os.path.exists(mstPath + 'masters.mst') and not force:
print "Error: SpySweeper definitions not found!"
os._exit(1)

# Get installed mst version number
try:
if not os.path.exists(ssPath + '\\Version\\version.ver'):
print "Unable to determine verion! Please reinstall SpySweeper."
sys.exit(1)
else:
f = file(ssPath + '\\Version\\version.ver', 'r')
current_version = string.strip(f.read())
print "Installed mst version ......", current_version
f.close()
except WindowsError, (errno, strerror):
if errno == 2:
print "Error: SpySweeper not found in registry"
sys.exit(1)
else:
print "Error: ", (errno, strerror)
exit(1)

try:
# Connect to webroot server
print "Server's mst version .......",
response = urllib.urlopen("http://www.webroot.com/php/disp0201.php?oc=49&pc=64000")
data = response.read()
server_version = data
print data

# Download if newer or if forced
if data > current_version or force:
print "Downloading update ......... ",
response = urllib.urlopen("http://www.webroot.com/php/disp0201.php?pc=64000&oc=50")
length = int(response.info()["Content-Length"])
amt = (length/10) + 10
data = ""
while True:
print '%s' % (chr(8) + '#'),
dl = response.read(amt)
if dl == "":
break
data += dl
print chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+chr(8)+"OK "
f = file(ssPath + '\\version\\version.ver', 'w')
f.write(server_version)
f.close()
else:
print "\nLatest update already installed"
os._exit(0)

if os.path.exists(mstPath + 'masters.mst'):
print "Backing up old file ........",
if os.path.exists(mstPath + 'masters.bak'):
os.unlink(mstPath + 'masters.bak')
os.rename(mstPath + 'masters.mst', mstPath + 'masters.bak')
print "OK"
print "Updating masters file ......",
f = file(mstPath + 'masters.mst', 'wb')
f.write(data)
f.close()
print "OK"
print "\nSpySweeper successfuly updated"

except KeyboardInterrupt:
print "\n\nCanceled by user"
sys.exit(1)
except socket.gaierror:
print "Unable to resolve host"
sys.exit(1)
except socket.error:
print "Unable to connect:", sys.exc_info()[1]
sys.exit(1)
except:
print "Unexpected error:", sys.exc_info()
raise