sssp_name.py

#!/usr/bin/python

import os
from os.path import *
import re
import socket
import sys

from sssp_utils import *

def virus_check(file_names):
  # Define the server
  server = 'localhost'
  port = 4010

  # return code
  result = 0

  # and connect to it
  try:
      s = socket.socket(socket.AF_INET, socket.SOCK_STREAM);
      s.connect ((server, port))
  except:
      print "Can't connect"
      return(1)

  try:
      # Do the initial exchange
      if not exchangeGreetings(s):
          print "Greetings rejected!!"
          return(1)

      # Query the server to see what we can do
      s.send ('QUERY SERVER\n')
      if not accepted(s):
          print "QUERY Rejected!! \n"
          return(1)

      resp = receivemsg(s)

      scanfile = 0
      scandata = 0
      scandir = 0
      scandirr = 0

      for l in resp:
          parts = optionsyntax.findall(l)

          if (parts[0][0] == 'method' and parts[0][1] == 'SCANDATA'):
              #print "SCANDATA is allowed"
              scandata = 1
          elif(parts[0][0] == 'method' and parts[0][1] == 'SCANFILE'):
              #print "SCANFILE is allowed"
              scanfile = 1
          elif(parts[0][0] == 'method' and parts[0][1] == 'SCANDIR'):
              #print "SCANDIR is allowed"
              scandir = 1
          elif(parts[0][0] == 'method' and parts[0][1] == 'SCANDIRR'):
              #print "SCANDIRR is allowed"
              scandirr = 1

      if not scandata and not scanfile and not scandir:
          print "Nothing is allowed!"
          return(1)

      # For each file on the command line use the most permissive
      # request that we can. though using SCANDIRR isn't always the
      # best idea.

      for filename in file_names:
          filename = os.path.realpath(filename)

          #Send the scan request according to what we are allowed
          if scandirr:
              s.send ('SCANDIRR ' + filename + '\n')
              if not accepted(s):
                  print "SCANDIRR Rejected!!"
                  return(1)
          elif scandir:
              s.send ('SCANDIR ' + filename + '\n')
              if not accepted(s):
                  print "SCANDIR Rejected!!"
                  return(1)
          elif scanfile:
              s.send ('SCANFILE ' + filename + '\n')
              if not accepted(s):
                  print "SCANFILE Rejected!!"
                  return(1)
          elif scandata:
              if not exists (filename) or not isfile(filename):
                  print "No such file as " + filename
                  return(1)
              else:
                  filesize = os.stat (filename)[6]
                  s.send ('SCANDATA ' + str(filesize) + '\n')
                  if not accepted(s):
                      print "SCANDATA Rejected!!"
                      return(1)

                  thefile = open (filename)
                  while 1:
                      b = thefile.read(4096)
                      if len(b) == 0:
                          break;
                      s.send (b)
                  thefile.close()

          #Analyse the response to the scan request
          # looking for VIRUS and DONE statements.
          resp = receivemsg(s)
          for l in resp:
              if virussyntax.match(l):
                  parts = virussyntax.findall(l)
                  print "Virus: " + parts[0][0] + " in " + parts[0][1]
                  result += 2
              elif donesyntax.match(l):
                  parts = donesyntax.findall(l)
                  print parts[0][0] + " (" + parts[0][1] + ") " + parts[0][2]

      sayGoodbye(s)
      return (result)

  finally:
      s.close();

if __name__ == '__main__':
  virus_check(sys.argv[1:])