批量增加Linux系统账号、重置账号密码、FTP账号批量测试
批量增加Linux系统账号、重置账号密码是用Linux Shell脚本来做的;批量FTP账号测试是用Python脚本来做的。这些脚本都是读取一个用户名和密码文件,然后基于该用户名密码文件进行自动批量测试。
这些脚本并没有多少技术含量,只是初次写起来需要调试耗费一点时间,希望能给需要的人节省时间,避免不必要的加班 ^_^
用户名密码文件格式如下:
每一行都是用户名和密码,用户名和密码之间用空格或者TAB制表符分割。
批量增加系统账号
#!/bin/bash cat ./ftp_account.txt | while read line do if [ "$line" = "" ] then exit fi account=`echo $line|awk -F ' ' '{print $1}'` password=`echo $line|awk -F ' ' '{print $2}'` loginShell=`which nologin` useradd $account -s $loginShell -b /home/log/cachelogbak/newlog echo "$account:$password"| chpasswd done
代码工作流程:
首先循环读取当前目录下一个叫做 ftp_account.txt 的用户名密码文件。每次一行,使用每一行之前先判断是不是空行(或到了最后一行了),然后切分读入的一行,切分为2个元素,第一个元素是用户名,第二个为密码,由于ubuntu和centos的nologin shell路径不同,因此用which找出具体路径,然后用useradd来创建相应用户并设置该账号的home目录和shell,最后调用chpasswd来更改该用户的密码。
批量重置账号密码
#!/bin/bash cat ./ftp_account.txt | while read line do if [ "$line" = "" ] then exit fi account=`echo $line|awk -F ' ' '{print $1}'` password=`echo $line|awk -F ' ' '{print $2}'` echo "$account:$password"| chpasswd done
代码工作流程:
其实就是前面第一个脚本的简略版,只不过省略了创建用户账号的那一步,仅仅调用chpasswd来更改密码。
批量测试FTP账号
#!/usr/bin/python #-*-coding:utf-8 -*- import os import sys import time import signal import datetime from ftplib import FTP """ ===ftp_account.txt content=== user1 passwd1 user2 passwd2 user3 passwd3 """ #====================================== def sigHandle(sig, frame): print('\nPython script terminated') os._exit(0) signal.signal(signal.SIGINT, sigHandle) #====================================== if os.path.exists('./ftp_account.txt'): fd = open('./ftp_account.txt', 'r') else: print "no such file" os._exit(-1) ftp = FTP() i = 0 j = 0 MAX = 0 fail_dict = {} while 1: line = fd.readline() if not line: break line = line.strip('\n').split() if len(line) != 2: print "wrong file format, break..." continue try: print "正在测试 %s" % line[0] ftp.connect(host="192.168.1.49", timeout=30) ftp.login(line[0], line[1]) ftp.close() i += 1 except Exception as e: if len(line[0]) > MAX: MAX = len(line[0]) fail_dict[line[0]] = str(e) j += 1 print "成功%d个," % i + "失败%d个" % j for each in fail_dict: print("失败账号: {:<%d}," % MAX).format(each) + \ (" 失败原因: {:<%d}" % MAX).format(fail_dict[each]) # 如果在Python 2.6上无法使用上面的print代码,请使用下面的代码 # for each in fail_dict: # print"失败账号: %s," % each + " 失败原因: %s" % fail_dict[each]
代码工作流程:
首先判断是否存在ftp_account.txt这个用户名密码文件,不存在则退出。
然后用Python自带的ftp库ftplib来创建一个ftp对象,并且声明3个计数器,i用于记录成功的账户数,j用来记录失败的账户数,MAX则用于最后的字符串格式化对齐。fail_dict字典则用来详细记录错误的ftp账户和错误原因。
接下来循环读取 ftp_account.txt 的用户名密码文件。每次一行,使用每一行之前先判断是不是空行(或到了最后一行了),然后切分读入的一行,切分为含2个元素的list列表,第一个元素是用户名,第二个为密码。如果list长度不为2,则说明用户名密码文件格式错误,跳过这错误的一行。然后开始连接指定的ftp服务器192.168.1.49,并设置超时为30秒。如果连接成功,紧接着关闭连接,并增加成功计数;如果失败,则抛出异常,增加失败计数,并将失败的账号和原因记录到fail_dict,同时找出失败账号字符串长度最长的那个值用于后面的格式化对齐。最后打印出成功失败数,以及哪些失败和失败原因。
sigHandle函数是用来在中途退出上面的脚本时阻止一大堆烦人的输出。