mirror of
https://github.com/wakatime/sublime-wakatime.git
synced 2023-08-10 21:13:02 +03:00
255 lines
7.0 KiB
Python
255 lines
7.0 KiB
Python
# This file is part of 'NTLM Authorization Proxy Server' http://sourceforge.net/projects/ntlmaps/
|
|
# Copyright 2001 Dmitry A. Rozmanov <dima@xenon.spb.ru>
|
|
#
|
|
# This library is free software: you can redistribute it and/or
|
|
# modify it under the terms of the GNU Lesser General Public
|
|
# License as published by the Free Software Foundation, either
|
|
# version 3 of the License, or (at your option) any later version.
|
|
|
|
# This library is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# Lesser General Public License for more details.
|
|
|
|
# You should have received a copy of the GNU Lesser General Public
|
|
# License along with this library. If not, see <http://www.gnu.org/licenses/> or <http://www.gnu.org/licenses/lgpl.txt>.
|
|
import six
|
|
|
|
from ntlm_auth.U32 import U32
|
|
from ntlm_auth.des_data import des_SPtrans, des_skb
|
|
|
|
def c2l(c):
|
|
"char[4] to unsigned long"
|
|
l = U32(c[0])
|
|
l = l | (U32(c[1]) << 8)
|
|
l = l | (U32(c[2]) << 16)
|
|
l = l | (U32(c[3]) << 24)
|
|
return l
|
|
|
|
|
|
def l2c(l):
|
|
"unsigned long to char[4]"
|
|
c = []
|
|
c.append(int(l & U32(0xFF)))
|
|
c.append(int((l >> 8) & U32(0xFF)))
|
|
c.append(int((l >> 16) & U32(0xFF)))
|
|
c.append(int((l >> 24) & U32(0xFF)))
|
|
return c
|
|
|
|
|
|
def D_ENCRYPT(tup, u, t, s):
|
|
L, R, S = tup
|
|
# print 'LRS1', L, R, S, u, t, '-->',
|
|
u = (R ^ s[S])
|
|
t = R ^ s[S + 1]
|
|
t = ((t >> 4) + (t << 28))
|
|
L = L ^ (des_SPtrans[1][int((t) & U32(0x3f))] |
|
|
des_SPtrans[3][int((t >> 8) & U32(0x3f))] |
|
|
des_SPtrans[5][int((t >> 16) & U32(0x3f))] |
|
|
des_SPtrans[7][int((t >> 24) & U32(0x3f))] |
|
|
des_SPtrans[0][int((u) & U32(0x3f))] |
|
|
des_SPtrans[2][int((u >> 8) & U32(0x3f))] |
|
|
des_SPtrans[4][int((u >> 16) & U32(0x3f))] |
|
|
des_SPtrans[6][int((u >> 24) & U32(0x3f))])
|
|
# print 'LRS:', L, R, S, u, t
|
|
return (L, R, S), u, t, s
|
|
|
|
|
|
def PERM_OP(tup, n, m):
|
|
"tup - (a, b, t)"
|
|
a, b, t = tup
|
|
t = ((a >> n) ^ b) & m
|
|
b = b ^ t
|
|
a = a ^ (t << n)
|
|
return (a, b, t)
|
|
|
|
|
|
def HPERM_OP(tup, n, m):
|
|
"tup - (a, t)"
|
|
a, t = tup
|
|
t = ((a << (16 - n)) ^ a) & m
|
|
a = a ^ t ^ (t >> (16 - n))
|
|
return a, t
|
|
|
|
|
|
shifts2 = [0, 0, 1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 0]
|
|
|
|
|
|
class DES:
|
|
KeySched = None # des_key_schedule
|
|
|
|
def __init__(self, key_str):
|
|
self.KeySched = des_set_key(key_str)
|
|
|
|
def decrypt(self, str):
|
|
# block - UChar[]
|
|
block = []
|
|
|
|
for i in six.iterbytes(str):
|
|
block.append(i)
|
|
|
|
# print block
|
|
block = des_ecb_encrypt(block, self.KeySched, 0)
|
|
|
|
res = b''
|
|
for i in block:
|
|
res = res + six.int2byte(i)
|
|
|
|
return res
|
|
|
|
def encrypt(self, plaintext):
|
|
# block - UChar[]
|
|
|
|
block = []
|
|
for i in plaintext:
|
|
block.append(i)
|
|
|
|
block = des_ecb_encrypt(block, self.KeySched, 1)
|
|
|
|
res = b''
|
|
|
|
for i in block:
|
|
res += six.int2byte(i)
|
|
|
|
return res
|
|
|
|
|
|
def des_encript(input, ks, encrypt):
|
|
# input - U32[]
|
|
# output - U32[]
|
|
# ks - des_key_shedule - U32[2][16]
|
|
# encrypt - int
|
|
# l, r, t, u - U32
|
|
# i - int
|
|
# s - U32[]
|
|
|
|
l = input[0]
|
|
r = input[1]
|
|
t = U32(0)
|
|
u = U32(0)
|
|
|
|
r, l, t = PERM_OP((r, l, t), 4, U32(0x0f0f0f0f))
|
|
l, r, t = PERM_OP((l, r, t), 16, U32(0x0000ffff))
|
|
r, l, t = PERM_OP((r, l, t), 2, U32(0x33333333))
|
|
l, r, t = PERM_OP((l, r, t), 8, U32(0x00ff00ff))
|
|
r, l, t = PERM_OP((r, l, t), 1, U32(0x55555555))
|
|
|
|
t = (r << 1) | (r >> 31)
|
|
r = (l << 1) | (l >> 31)
|
|
l = t
|
|
|
|
s = ks # ???????????????
|
|
# print l, r
|
|
if encrypt:
|
|
for i in range(0, 32, 4):
|
|
rtup, u, t, s = D_ENCRYPT((l, r, i + 0), u, t, s)
|
|
l = rtup[0]
|
|
r = rtup[1]
|
|
rtup, u, t, s = D_ENCRYPT((r, l, i + 2), u, t, s)
|
|
r = rtup[0]
|
|
l = rtup[1]
|
|
else:
|
|
for i in range(30, 0, -4):
|
|
rtup, u, t, s = D_ENCRYPT((l, r, i - 0), u, t, s)
|
|
l = rtup[0]
|
|
r = rtup[1]
|
|
rtup, u, t, s = D_ENCRYPT((r, l, i - 2), u, t, s)
|
|
r = rtup[0]
|
|
l = rtup[1]
|
|
# print l, r
|
|
l = (l >> 1) | (l << 31)
|
|
r = (r >> 1) | (r << 31)
|
|
|
|
r, l, t = PERM_OP((r, l, t), 1, U32(0x55555555))
|
|
l, r, t = PERM_OP((l, r, t), 8, U32(0x00ff00ff))
|
|
r, l, t = PERM_OP((r, l, t), 2, U32(0x33333333))
|
|
l, r, t = PERM_OP((l, r, t), 16, U32(0x0000ffff))
|
|
r, l, t = PERM_OP((r, l, t), 4, U32(0x0f0f0f0f))
|
|
|
|
output = [l]
|
|
output.append(r)
|
|
l, r, t, u = U32(0), U32(0), U32(0), U32(0)
|
|
return output
|
|
|
|
|
|
def des_ecb_encrypt(input, ks, encrypt):
|
|
# input - des_cblock - UChar[8]
|
|
# output - des_cblock - UChar[8]
|
|
# ks - des_key_shedule - U32[2][16]
|
|
# encrypt - int
|
|
|
|
# print input
|
|
l0 = c2l(input[0:4])
|
|
l1 = c2l(input[4:8])
|
|
ll = [l0]
|
|
ll.append(l1)
|
|
# print ll
|
|
ll = des_encript(ll, ks, encrypt)
|
|
# print ll
|
|
l0 = ll[0]
|
|
l1 = ll[1]
|
|
output = l2c(l0)
|
|
output = output + l2c(l1)
|
|
# print output
|
|
l0, l1, ll[0], ll[1] = U32(0), U32(0), U32(0), U32(0)
|
|
return output
|
|
|
|
|
|
def des_set_key(key):
|
|
# key - des_cblock - UChar[8]
|
|
# schedule - des_key_schedule
|
|
|
|
# register unsigned long c,d,t,s;
|
|
# register unsigned char *in;
|
|
# register unsigned long *k;
|
|
# register int i;
|
|
|
|
# k = schedule
|
|
# in = key
|
|
|
|
k = []
|
|
c = c2l(key[0:4])
|
|
d = c2l(key[4:8])
|
|
t = U32(0)
|
|
|
|
d, c, t = PERM_OP((d, c, t), 4, U32(0x0f0f0f0f))
|
|
c, t = HPERM_OP((c, t), -2, U32(0xcccc0000))
|
|
d, t = HPERM_OP((d, t), -2, U32(0xcccc0000))
|
|
d, c, t = PERM_OP((d, c, t), 1, U32(0x55555555))
|
|
c, d, t = PERM_OP((c, d, t), 8, U32(0x00ff00ff))
|
|
d, c, t = PERM_OP((d, c, t), 1, U32(0x55555555))
|
|
|
|
d = (((d & U32(0x000000ff)) << 16) | (d & U32(0x0000ff00)) | ((d & U32(0x00ff0000)) >> 16) | (
|
|
(c & U32(0xf0000000)) >> 4))
|
|
c = c & U32(0x0fffffff)
|
|
|
|
for i in range(16):
|
|
if (shifts2[i]):
|
|
c = ((c >> 2) | (c << 26))
|
|
d = ((d >> 2) | (d << 26))
|
|
else:
|
|
c = ((c >> 1) | (c << 27))
|
|
d = ((d >> 1) | (d << 27))
|
|
c = c & U32(0x0fffffff)
|
|
d = d & U32(0x0fffffff)
|
|
|
|
s = des_skb[0][int((c) & U32(0x3f))] | \
|
|
des_skb[1][int(((c >> 6) & U32(0x03)) | ((c >> 7) & U32(0x3c)))] | \
|
|
des_skb[2][int(((c >> 13) & U32(0x0f)) | ((c >> 14) & U32(0x30)))] | \
|
|
des_skb[3][int(((c >> 20) & U32(0x01)) | ((c >> 21) & U32(0x06)) | ((c >> 22) & U32(0x38)))]
|
|
|
|
t = des_skb[4][int((d) & U32(0x3f))] | \
|
|
des_skb[5][int(((d >> 7) & U32(0x03)) | ((d >> 8) & U32(0x3c)))] | \
|
|
des_skb[6][int((d >> 15) & U32(0x3f))] | \
|
|
des_skb[7][int(((d >> 21) & U32(0x0f)) | ((d >> 22) & U32(0x30)))]
|
|
# print s, t
|
|
|
|
k.append(((t << 16) | (s & U32(0x0000ffff))) & U32(0xffffffff))
|
|
s = ((s >> 16) | (t & U32(0xffff0000)))
|
|
s = (s << 4) | (s >> 28)
|
|
k.append(s & U32(0xffffffff))
|
|
|
|
schedule = k
|
|
|
|
return schedule
|