[File_Filter] 파일 분류하는 방법

python2 기반으로 작성했지만

print문에 ()만 붙이면 python3에서도 동작하네요!!

 

악성코드와 상용프로그램을 한 곳에 

저장되어 있어서...

 

이것을 분류하려고 합니다.

 

악성코드는 자신의 해시값으로

파일명이 되어있습니다.

 

import os
import shutil
import sys

folder_name = sys.argv[1]
listfile_name = sys.argv[2]


if len(sys.argv) != 3:
	print " [-] python file_filter.py [악성코드가 있는 디렉터리] [악성코드 파일명을 저장한.txt]\n"
	print " [*] Using_Example"
	print " [*] python file_filter.py /home/maias/Desktop/data/ /home/maias/Desktop/data/1st_answer.csv"

	exit(1)

listfile_name = listfile_name.split("\n")[0]

with open(listfile_name, 'r') as fd:
	file_list = fd.read().split('\n')
	
dir_file_list = os.listdir(folder_name)
# 디렉터리에 있는 파일명을 리스트로 저장

for dir_file_name in dir_file_list:
	if dir_file_name in file_list:
		shutil.move(folder_name + "/" + dir_file_name, folder_name + "/malware/" + dir_file_name)
# 파일이름이 txt에 있다면 malware디렉터리로 이동

 

현재는 리눅스기반의 디렉터리로 작성했지만

/를 \\로 변경하면 되는 부분이라

사용하는데 문제는 없을 것으로 보이네요.