关于bottle WEB框架中签名cookie的一点理解

首先要理解一个概念

MAC (message authenticate code)

消息认证码(带密钥的Hash函数):密码学中,通信实体双方使用的一种验证机制,保证消息数据完整性的一种工具。

构造方法由M.Bellare提出,安全性依赖于Hash函数,故也称带密钥的Hash函数。消息认证码是基于密钥和消息摘要所获得的一个值,

可用于数据源发认证和完整性校验。

 

消息认证码(带密钥的Hash函数):密码学中,通信实体双方使用的一种验证机制,保证消息数据完整性的一种工具。 构造方法由M.Bellare提出,安全性依赖于Hash函数,故也称带密钥的Hash函数。消息认证码是基于密钥和消息摘要所获得的一个值,可用于数据源发认证和完整性校验。
在发送数据之前,发送方首先使用通信双方协商好的散列函数计算其摘要值。在双方共享的会话密钥作用下,由摘要值获得消息验证码。之后,它和数据一起被发送。接收方收到报文后,首先利用会话密钥还原摘要值,同时利用散列函数在本地计算所收到数据的摘要值,并将这两个数据进行比对。若两者相等,则报文通过认证。

 

 签名cookie就是基于这种原理。

一般发送的数据都会做base64编码,关于base64编码可以看这个链接

 blog.xiayf.cn/2016/01/24/base64-encoding/

总的来说:base64并不是用来加密的,base64是一种数据编码方式,目的让数据符合传输协议的要求,将二进制的数据转化为一种文本数据

在Python中提供了两个模块,去实现。以下是base64的基本用法

>>> import base64
>>> s='b'asdafsdafa'
SyntaxError: invalid syntax
>>> s=b'asdasdas'
>>> s_64 = base64.bencode(s)

    s_64 = base64.bencode(s)
AttributeError: module 'base64' has no attribute 'bencode'
>>> s_64 = base64.b64encode(s)
>>> s_64
b'YXNkYXNkYXM='
>>> base64.b64decode(s_64)
b'asdasdas'
>>> 

 

对于mac,Python有个模块hmac基本用法如下

>>> import hmac
>>> hmac.new(b'slat')
<hmac.HMAC object at 0x0000000004110B70>
>>> hmac = hmac.new(b'slat')
>>> hmac.update(b'asdas')
>>> hmac.digest()
b'\xe8\\\xb6\x11\x9dj\rY\x06I\x1f[\x06\xeb\xeb\xf3'
>>> 

 

 文档说明

class HMAC(builtins.object)
 |  RFC 2104 HMAC class.  Also complies with RFC 4231.
 |  
 |  This supports the API for Cryptographic Hash Functions (PEP 247).
 |  
 |  Methods defined here:
 |  
 |  __init__(self, key, msg=None, digestmod=None)
 |      Create a new HMAC object.
 |      
 |      key:       key for the keyed hash object.
 |      msg:       Initial input for the hash, if provided.
 |      digestmod: A module supporting PEP 247.  *OR*
 |                 A hashlib constructor returning a new hash object. *OR*
 |                 A hash name suitable for hashlib.new().
 |                 Defaults to hashlib.md5.
 |                 Implicit default to hashlib.md5 is deprecated and will be
 |                 removed in Python 3.6.
 |      
 |      Note: key and msg must be a bytes or bytearray objects.
 |  
 |  copy(self)
 |      Return a separate copy of this hashing object.
 |      
 |      An update to this copy won't affect the original object.
 |  
 |  digest(self)
 |      Return the hash value of this hashing object.
 |      
 |      This returns a string containing 8-bit data.  The object is
 |      not altered in any way by this function; you can continue
 |      updating the object after calling this function.
 |  
 |  hexdigest(self)
 |      Like digest(), but returns a string of hexadecimal digits instead.
 |  
 |  update(self, msg)
 |      Update this hashing object with the string msg.
View Code

 

立即了基础,来看bottle框架设置签名cookie和获取签名cookie的值的源码.

设置签名cookie

first_bottle.py
response.set_cookie('account',username,secret='salt')

 

 

    def set_cookie(self, name, value, secret=None, **options):
 
        if not self._cookies:
            self._cookies = SimpleCookie()

        if secret:
            value = touni(cookie_encode((name, value), secret))
        elif not isinstance(value, basestring):
            raise TypeError('Secret key missing for non-string Cookie.')

        if len(value) > 4096: raise ValueError('Cookie value to long.')
        self._cookies[name] = value

        for key, value in options.items():
            if key == 'max_age':
                if isinstance(value, timedelta):
                    value = value.seconds + value.days * 24 * 3600
            if key == 'expires':
                if isinstance(value, (datedate, datetime)):
                    value = value.timetuple()
                elif isinstance(value, (int, float)):
                    value = time.gmtime(value)
                value = time.strftime("%a, %d %b %Y %H:%M:%S GMT", value)
            self._cookies[name][key.replace('_', '-')] = value

 self._cookie默认是None,SimpleCookie继承BaseCookie,BaseCookie继承一个字典,所以暂且认为self.cookie是一个字典,

if secret  是判断,如果设置了密钥,就执行这一步,

 

def tob(s, enc='utf8'):
    return s.encode(enc) if isinstance(s, unicode) else bytes(s)
def touni(s, enc='utf8', err='strict'):
    return s.decode(enc, err) if isinstance(s, bytes) else unicode(s)
tonat = touni if py3k else tob

 

touni 函数是返回str类型的字符串,这里unicode=str,unicode(s) 相当于str(s)

 

def cookie_encode(data, key):
    ''' Encode and sign a pickle-able object. Return a (byte) string '''
    msg = base64.b64encode(pickle.dumps(data, -1))
    sig = base64.b64encode(hmac.new(tob(key), msg).digest())
    return tob('!') + sig + tob('?') + msg

 

pickle.dumps将数据序列化,返回的是bytes类型的字符串,然后编码为base64    sig 是先用hmac加密,

最后将msg(消息) 和sig(签名)拼接,这样一个签名cookie就设置好了,注意这里的msg是一个(name,value)包含cookie的key和value

这样一个签名cookie就设置好了

 

理解了签名cookie的设置,再看获得签名cookie的值就比较简单了。。

大致原理是拿到cookie的值,通过?分割出message 和sig  ,再拿message和secret 进行hmac 拿到新的sig,这个新的sig与分割出来的sig比较,如果一致,表示没有被篡改,这样吧message 用base64decode

然后pickle.loads 就拿到原来的数组了。数组的[1]就是那个value,

def cookie_decode(data, key):
    ''' Verify and decode an encoded string. Return an object or None.'''
    data = tob(data)
    if cookie_is_encoded(data):
        sig, msg = data.split(tob('?'), 1)if _lscmp(sig[1:], base64.b64encode(hmac.new(tob(key), msg).digest())):
            return pickle.loads(base64.b64decode(msg))

 

 

因为之前setcookie时在自古穿前面加了一个感叹 号!  ,所以切片sig[1:]     

def _lscmp(a, b):
    ''' Compares two strings in a cryptographically safe way:
        Runtime is not affected by length of common prefix. '''
    return not sum(0 if x==y else 1 for x, y in zip(a, b)) and len(a) == len(b)

 

上面这个函数是逐个字符比较,如果比较的字符都相等那么就返回0,否则返回1,这样如果是两个字符串完全匹配,就都是0,调用sum() 相加肯定返回0 ,否则肯定不是1,但是必须在长度相等的条件下才可以

测试代码
>>> a='asdas' >>> unicode(a) Traceback (most recent call last): File "<pyshell#1>", line 1, in <module> unicode(a) NameError: name 'unicode' is not defined >>> b='asd' >>> (0 if x==y else 1 for x,y in zip(a,b)) <generator object <genexpr> at 0x0000000003170200> >>> sum((0 if x==y else 1 for x,y in zip(a,b))) 0 >>> s=zip(a,b) >>> s <zip object at 0x0000000003147948> >>> for i in s: print(i) ('a', 'a') ('s', 's') ('d', 'd')

 

 

为什么比较字符串相等不直接用 a==b?

 

于洋 回到顶部

posted @ 2017-11-23 20:34  dragonbird  阅读(669)  评论(0编辑  收藏  举报