2013-10-19 78 views
-1

我正在构建一个允许用户上传PDF文件以供其他用户下载的Web应用程序(python和Django)。如何防止用户上传嵌入在pdf中的病毒?PDF和Django中的病毒

更新: 我在使用clamcv的django片段中发现了此代码。这会做这份工作吗?

def clean_file(self): 
    file = self.cleaned_data.get('file', '') 
    #check a file in form for viruses 
    if file: 
     from tempfile import mkstemp 
     import pyclamav 
     import os 
     tmpfile = mkstemp()[1] 
     f = open(tmpfile, 'wb') 
     f.write(file.read()) 
     f.close() 
     isvirus, name = pyclamav.scanfile(tmpfile) 
     os.unlink(tmpfile) 
     if isvirus: 
      raise forms.ValidationError(\ 
      "WARNING! Virus \"%s\" was detected in this file. \ 
      Check your system." % name) 

    return file 
+2

安装了病毒扫描和上传后,以及在存储之前扫描的文件... –

回答

1

好,一般可以使用任何病毒扫描软件来完成这个任务:只是

  • 产生这就要求你的文件
  • 使用病毒扫描程序的命令行字符串蟒蛇子进程运行命令行字符串像这样:

    try: 
        command_string = 'my_virusscanner -parameters ' + uploaded_file 
        result = subprocess.check_output(command_string,stderr=subprocess.STDOUT,shell=True) 
        #if needed, do something with "result"    
    except subprocess.CalledProcessError as e: 
        #if your scanner gives an error code when detecting a virus, you'll end up here 
        pass 
    except: 
        #something else went wrong 
        #check sys.exc_info() for info 
        pass 
    

如果不检查源代码,我认为pyclamav.scanfile或多或少都是一样的 - 所以如果你信任clamav,你应该会很好。如果您不信任ist,请使用您选择的病毒扫描程序的上述方法。

0

您可以使用django-safe-filefield包来验证上传的文件扩展名与MIME类型匹配。例如:

settings.py

CLAMAV_SOCKET = 'unix://tmp/clamav.sock' # or tcp://127.0.0.1:3310 

CLAMAV_TIMEOUT = 30 # 30 seconds timeout, None by default which means infinite 

forms.py

from safe_filefield.forms import SafeFileField 

class MyForm(forms.Form): 
    attachment = SafeFileField(
     scan_viruses=True, 
    )