py文件输入/输出、sys模块、os模块

举报
黑色地带(崛起) 发表于 2023/02/20 22:32:15 2023/02/20
【摘要】 py文件输入/输出、sys模块、os模块

 目录

一、文件输入/输出

简介:

示例:

二、sys模块

简介:

示例:

三、OS模块

简介:

示例:

四、脚本

示例:


(我所用为python3+pycharm)

一、文件输入/输出

简介:

假设有一个名为vuln_ b anners.txt 文本文件, 这个文件的每一行都列出了一个已知有漏洞的特定服务版本(的banner) 。

通过读取这个文本文件, 来判断banner 是否有漏洞的服务器,避免使用一大堆if语句去判断。


示例:

cat vuln_banners.txt
3Com 3CDaemon FTP Server Version 2.0
Ability Server 2.34
CCProxy Telnet Service Ready
ESMTP TABS Mail Server for Windows NT
FreeFloat Ftp Server (Version 1.00)
IMAP4revl MDaemon 9.6.4 ready
MailEnable Service, Version: 0-1.54
NetDecision-HTTP-Server 1.0
PSO Proxy 0.9
SAMBAR
Sami FTP Server 2.0.2
Spipe 1.0
TelSrv 1.5
WDaemon 6.8.5
WinGate 6.1.1
Xitami
YahooPOPs! Simple Mail Transfer Service Ready



在checkVulns函数中换上了新的代码。这里以只读模式('r')打开文本文件,用.readlinesO方法遍历文件中的每一行, 并将其与我们的banner做比较。

注: 必须用.strip('\r')方法去掉每一行的回车键。检测到匹配, 打印一个有漏洞的服务器的banner


def checkVulns(banner):
    f = open("vuln_banners.txt", 'r')
    for line in f.readlines():
        if line.strip('\n') in banner:
            print('Server is vulnerable: '+banner.strip('\n'))





二、sys模块

简介:

内置的sys模块使我们能访问到由Python 解释器使用或维护的对象, 其中包括标志、版本、整型数的最大尺寸、可用的模块、hook路径、标准错误/输入/输出的位置, 以及调用解释器的命令行参数


示例:

把一个文本文件的文件名作为命令行参数传递sys.argv 列表中含有所有的命令行参数

第一个sys.argv[0]元素中的是Python 脚本的名称

列表中的其余元素中则记录了之后所有的命令行参数

如果我们只是传递一个额外的参数, sys.argv 中应该包含两个元素(脚本名称、传入的参数)

import sys

if len(sys.argv) != 2:
    filename = sys.argv[1]
    print('Reading Vulnerabilities From: ' + filename)




三、OS模块

简介:

内置的OS 模块提供了丰富的适用于Mac、NT 或Posix 的操作系统的函数。这个模块允许程序独立地与操作系统环境、文件系统、用户数据库以及权限进行交互



示例:

用户把一个文本文件的文件名作为命令行参数传递进来, 先检查一下该文件是否存在、当前用户是否有权限读取该文件,其中任一条件不满足, 就向用户显示一条相应的错误信息

import sys
import os

if len(sys.argv) == 2:
    filename = sys.argv[1]
    if not os.path.isfile(filename):
        print(filename + ' does not exist.')
        exit(0)
    if not os.access(filename, os.R_OK):
        print(filename + ' access denied.')
        exit(0)
    print('Reading Vulnerabilities From: ' + filename)




四、脚本

示例:

重新整合上面学过的这些重新整合一下Python 扫描脚本的各个部分

import socket
import sys
import os

def retbanner(ip, port):
    try:
        socket.setdefaulttimeout(2)
        s = socket.socket()
        s.connect((ip, port))
        banner = s.recv (1024)
        return banner
    except Exception as e:
        return e


def checkVulns(banner,filename):
    f = open(filename,'r')
    for line in f.readlines():
        if line.strip ('\n') in banner:
            print('Server is vulnerable: ' + banner.strip ('\n'))


def main():
if len(sys.argv) == 2:
    filename = sys.argv[1]
    if not os.path.isfile(filename):
        print(filename + ' does not exist.')
        exit(0)
    if not os.access(filename, os.R_OK):
        print(filename + ' access denied.')
        exit(0)
    else:
        print('Usage: ' + str(sys.argv[0]) + ' <vuln filename>')
        exit(0)
        portList = [21, 22, 25, 80, 110, 443]
        for x in range(147, 150 ):
            ip = '192.168.190.' + str( x )
            for port in portList:
                banner = retbanner(ip,port )
                if banner:
                    print(ip + str(banner))
                    checkVulns(banner, filename)

if __name__ == '__main__':
    main()


【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。