#!/usr/bin/python
# coding=UTF-8
import sys
import MySQLdb
""""""
def matchpasswd(login,passwd):
    #用户名和密码，用于验证输入的用户名和密码。
    #你可以自定义验证通过的要求，验证通过，返回True，验证不通过返回False
    if (login == '修改为你的用户名' and passwd == '修改为你的验证密码'):
        return True
    else:
        return False
def write_client_ip_to_db(username,client_ip):
    #数据库地址，数据库名，数据库密码，数据库用户名，数据库编码格式
    db=MySQLdb.connect("localhost","squid","修改为你的密码","squid",charset='utf8')
    # 使用cursor()方法获取操作游标 
    cursor = db.cursor()
    # 向sq_client_ip表中插入用户名和用户ip
    sql = "INSERT INTO sq_client_ip(username, client_ip) VALUES ('%s', '%s')" % (username,client_ip)
    try:
       # 执行sql语句
       cursor.execute(sql)
       #Ok 后面的print语句可以向cache日志中输出提示。/var/log/squid/cache.log查看
       print("执行sql语句")
       # 提交到数据库执行
       db.commit()
    except:
       #打印错误内容
       info = sys.exc_info()  
       print( info[0], ":", info[1]) 
	   #发生错误时回滚
       db.rollback()

    # 关闭数据库连接
    db.close()
    return
while True:
    # read a line from stdin
    line = sys.stdin.readline()
    line = line.strip()
    #提取输入参数。
    username,password,client_ip = line.split()
    #判断密码是否匹配
    if matchpasswd(username, password):
        #输出ok表示验证通过
        sys.stdout.write('OK\n 密码验证通过')
      	#将客户端ip添加至数据库ip白名单
      	write_client_ip_to_db(username,client_ip)
      	
    else:
        sys.stdout.write('ERR\n 密码验证失败')
    # Flush the output to stdout.
    sys.stdout.flush()

#!/usr/bin/python
# coding=UTF-8
import sys
import MySQLdb
"""当用户连接squid，传递用户客户端ip地址到外部辅助器，辅助器通过标准输入读取。辅助器查询数据库符合客户端ip的所有项，如果查询结果，显示客户ip在数据库中，则输出ok，否则输出ERR"""
def matchclientip(client_ip):
    #数据库地址，数据库名，数据库密码，数据库用户名，数据库编码格式
    db=MySQLdb.connect("localhost","squid","你的数据库密码","squid",charset='utf8')
    # 使用cursor()方法获取操作游标 
    cursor = db.cursor()
    # SQL 插入语句
    sql = "SELECT * FROM sq_client_ip  WHERE client_ip = '%s'" % (client_ip)
    try:
       # 执行sql语句
       cursor.execute(sql)
       result = cursor.fetchone()
       if result:
          exist_ip = True
       else:
          exist_ip = False
    except:
       # 发生错误时回滚
       db.rollback()
       exist_ip = False

    # 关闭数据库连接
    db.close()
    return exist_ip
while True:
    # 从标准输入读取一行，“client_ip”
    line = sys.stdin.readline()
    #提取输入参数。squid，输入给辅助器的字符串是"0 xx.xx.xx.xx"格式
    #利用split分割字符串，得到0和client_ip的list
    client_ip = line.strip().split()[1]
    #判断密码是否匹配
    if matchclientip(client_ip):
        #输出ok表示验证通过
        sys.stdout.write('OK\n')
        #print会将信息输出到/var/log/squid/cache.log中。
        print("Acl辅助器验证成功")
    else:
        sys.stdout.write('ERR\n')
        print("ACL辅助器验证失败")
    # Flush the output to stdout.
    sys.stdout.flush()