Support ":" in cookie key

可以在 Cookie Key 中使用冒号

> Cookie 标准禁止在Key中使用冒号，部分网站中会存在该情况

Author: Medicean

hackhttp/__init__.py

@@ -76,7 +76,7 @@
 """
 
 __title__ = 'hackhttp'
-__version__ = '1.0.3'
+__version__ = '1.0.4'
 __build__ = 0x020700
 __author__ = 'BugScanTeam'
 __author_email__ = '[email protected]'

hackhttp/hackhttp.py

@@ -14,6 +14,7 @@
 import cookielib
 import copy
 import time
+import string
 
 
 class httpheader(mimetools.Message):
@@ -38,6 +39,27 @@ def get(self, key, d=None):
         return self.dict.get(key, d)
 
 
+class MorselHook(Cookie.Morsel):
+    """
+    Support ":" in Cookie key.
+
+    >>> import inspect
+    >>> (inspect.getargspec(MorselHook.set)[3])[0]
+    "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!#$%&'*+-.^_`|~:"
+    >>> cookie = Cookie.SimpleCookie()
+    >>> cookie.load("key:key=abc; key=val")
+    >>> print cookie
+    Set-Cookie: key=val;
+    Set-Cookie: key:key=abc;
+    """
+    def set(
+        self, key, val, coded_val,
+            LegalChars=Cookie._LegalChars + ':',
+            idmap=string._idmap, translate=string.translate):
+        return super(MorselHook, self).set(
+            key, val, coded_val, LegalChars, idmap, translate)
+
+
 class httpconpool():
     # 创建的连接总数, key 为 conhash
     connected = {}
@@ -141,6 +163,7 @@ def __init__(self, conpool=None, cookie_str=None, throw_exception=True):
             self.conpool = httpconpool(10)
         else:
             self.conpool = conpool
+        Cookie.Morsel = MorselHook
         self.initcookie = Cookie.SimpleCookie()
         if cookie_str:
             if not cookie_str.endswith(';'):