自己动手写工具包——目录扫描器

写下这个帖子是对我python学习过程的一个记录,也是对自己的一个督促。
希望各位大神多多帮助,指出不足之处!


#学python大概有了一个星期,自从前天准备做一个《自己动手写工具包》的python项目,下午研究了一下urllib,就把它写出来了。

#还是有很多的小漏洞的,我会慢慢的修补。


#coding:utf-8
import os
import sys
import urllib

def open_Disc(path):
	try:
		temp_Disc = [];
		result_Disc = [];
		Disc = file(path,'r');
		for Disc_line in Disc.readlines():
			temp_Disc.append(Disc_line);
		Disc.close();
		#去重复
		for i in temp_Disc:
			if i not in result_Disc:
				result_Disc.append(i);
		print("File open success!");
		return result_Disc;
	except IOError,e:
		print("No such file!");
		sys.exit();
		
def check_url(Disc,Url):
	index = 0;
	while index < len(Disc):
		complete_url = Url+Disc[index];
		complete_url = complete_url[:-1];
		sock = urllib.urlopen(complete_url);
		index+=1;
		return_code = sock.getcode();
		if return_code == 200 or return_code == 301 or return_code == 302 or return_code == 403:
			print(complete_url+"\t\t\t"+str(return_code));

def help():
	print("Use webScan like this:");
	print("wenScan.py <host> <disc absolute path>");

if __name__ == "__main__":
	if len(sys.argv) == 2:
		if str(sys.argv[1]) == "--help" or str(sys.argv[1]) == '-h':
			help();
		else:
			print("Use option like:--help -h");
	elif len(sys.argv) == 3:
		url = sys.argv[1];
		path = sys.argv[2];

		Disc = open_Disc(path);
		check_url(Disc,url);
	else:
		print("");
		print("Too many options or no option!");
		print("Use option like:--help -h");
		sys.exit();

本程序在windows下试验成功,没有使用多线程。
使用方法:python ***.py http://xx.xx/ D:\disc.txt
参数1是你要扫描的网站,必须要加http://
参数2是你的目录字典,一定要注意,在字典的最后一行加一个回车。


(反正我知道也没有人会去用它)


我的实力非常差劲,希望各位大神包容,不要喷!

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值