m_lib/mcrypt.py

   1 #! /usr/bin/env python
   2
   3 #
   4 # useful function(s) for module crypt
   5 #
   6
   7
   8 from __future__ import print_function
   9 import random, crypt
  10
  11
  12 saltchars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcedfghijklmnopqrstuvwxyz0123456789./"
  13 len_salt = len(saltchars)
  14
  15 def gen_salt():
  16    """
  17       There are some difference among modern unicies. BSD/OS, for example,
  18       uses MD5 hash, and ignores salt completely. FreeBSD uses 3 different
  19       versions of crypt() - with standard salt, with extended 9-byte salt,
  20       and MD5 (again, ignoring salt at all).
  21       This function generates salt for standard "Broken DES"-based crypt().
  22    """
  23    r1 = random.randint(0, len_salt-1)
  24    r2 = random.randint(0, len_salt-1)
  25    return "%s%s" % (saltchars[r1], saltchars[r2])
  26
  27
  28 def test():
  29    passwd = raw_input("Enter password: ")
  30    salt = gen_salt()
  31    encrypted = crypt.crypt(passwd, salt)
  32
  33    pwd_file = open("test.pwd", 'w')
  34    pwd_file.write("%s:%s" % ("user", encrypted))
  35    pwd_file.close()
  36    print("Password file written")
  37
  38    pwd_file = open("test.pwd", 'r')
  39    username, encrypted = pwd_file.readline()[:-1].split(':')
  40    pwd_file.close()
  41
  42    if crypt.crypt(encrypted, encrypted):
  43       print("Password verified Ok")
  44    else:
  45       print("BAD password")
  46
  47 if __name__ == "__main__":
  48    test()