import os

import hashlib

import sys

import socket

import string

for root, dir, files in os.walk(str(sys.argv[1])):

for fp in files:

try:

fn = root+fp

infile = open(fn, "rb")

content = infile.read()

infile.close()

m = hashlib.md5()

m.update(content)

hash = m.hexdigest()

mhr = socket.socket(socket.AF_INET,\

socket.SOCK_STREAM)

mhr.connect(("server", 43))

mhr.send(str(hash + "\r\n"))

response = ''

while True:

d = mhr.recv(4096)

response += d

if d == '':

break

if "NO_DATA" not in response:

print "<INFECTED>:"+str(fn)

except:

pass