src/third_party/tlslite/scripts/tlsdb.py

   1 #! python
   2
   3 import sys
   4 import os
   5 import socket
   6 import thread
   7 import math
   8
   9 try:
  10     import cryptoIDlib
  11     cryptoIDlibLoaded = True
  12 except:
  13     cryptoIDlibLoaded = False
  14
  15
  16 if __name__ != "__main__":
  17     raise "This must be run as a command, not used as a module!"
  18
  19
  20 from tlslite.api import *
  21
  22 if len(sys.argv) == 1 or (len(sys.argv)==2 and sys.argv[1].lower().endswith("help")):
  23     print ""
  24     print "Version: 0.3.8"
  25     print ""
  26     print "RNG: %s" % prngName
  27     print ""
  28     print "Modules:"
  29     if cryptlibpyLoaded:
  30         print "  cryptlib_py : Loaded"
  31     else:
  32         print "  cryptlib_py : Not Loaded"
  33     if m2cryptoLoaded:
  34         print "  M2Crypto    : Loaded"
  35     else:
  36         print "  M2Crypto    : Not Loaded"
  37     if pycryptoLoaded:
  38         print "  pycrypto    : Loaded"
  39     else:
  40         print "  pycrypto    : Not Loaded"
  41     if gmpyLoaded:
  42         print "  GMPY        : Loaded"
  43     else:
  44         print "  GMPY        : Not Loaded"
  45     if cryptoIDlibLoaded:
  46         print "  cryptoIDlib : Loaded"
  47     else:
  48         print "  cryptoIDlib : Not Loaded"
  49     print ""
  50     print "Commands:"
  51     print ""
  52     print "  createsrp       <db>"
  53     print "  createsharedkey <db>"
  54     print ""
  55     print "  add    <db> <user> <pass> [<bits>]"
  56     print "  del    <db> <user>"
  57     print "  check  <db> <user> [<pass>]"
  58     print "  list   <db>"
  59     sys.exit()
  60
  61 cmd = sys.argv[1].lower()
  62
  63 class Args:
  64     def __init__(self, argv):
  65         self.argv = argv
  66     def get(self, index):
  67         if len(self.argv)<=index:
  68             raise SyntaxError("Not enough arguments")
  69         return self.argv[index]
  70     def getLast(self, index):
  71         if len(self.argv)>index+1:
  72             raise SyntaxError("Too many arguments")
  73         return self.get(index)
  74
  75 args = Args(sys.argv)
  76
  77 def reformatDocString(s):
  78     lines = s.splitlines()
  79     newLines = []
  80     for line in lines:
  81         newLines.append("  " + line.strip())
  82     return "\n".join(newLines)
  83
  84 try:
  85     if cmd == "help":
  86         command = args.getLast(2).lower()
  87         if command == "valid":
  88             print ""
  89         else:
  90             print "Bad command: '%s'" % command
  91
  92     elif cmd == "createsrp":
  93         dbName = args.get(2)
  94
  95         db = VerifierDB(dbName)
  96         db.create()
  97
  98     elif cmd == "createsharedkey":
  99         dbName = args.getLast(2)
 100
 101         db = SharedKeyDB(dbName)
 102         db.create()
 103
 104     elif cmd == "add":
 105         dbName = args.get(2)
 106         username = args.get(3)
 107         password = args.get(4)
 108
 109         try:
 110             db = VerifierDB(dbName)
 111             db.open()
 112             if username in db:
 113                 print "User already in database!"
 114                 sys.exit()
 115             bits = int(args.getLast(5))
 116             N, g, salt, verifier = VerifierDB.makeVerifier(username, password, bits)
 117             db[username] = N, g, salt, verifier
 118         except ValueError:
 119             db = SharedKeyDB(dbName)
 120             db.open()
 121             if username in db:
 122                 print "User already in database!"
 123                 sys.exit()
 124             args.getLast(4)
 125             db[username] = password
 126
 127     elif cmd == "del":
 128         dbName = args.get(2)
 129         username = args.getLast(3)
 130
 131         try:
 132             db = VerifierDB(dbName)
 133             db.open()
 134         except ValueError:
 135             db = SharedKeyDB(dbName)
 136             db.open()
 137
 138         del(db[username])
 139
 140     elif cmd == "check":
 141         dbName = args.get(2)
 142         username = args.get(3)
 143         if len(sys.argv)>=5:
 144             password = args.getLast(4)
 145         else:
 146             password = None
 147
 148         try:
 149             db = VerifierDB(dbName)
 150             db.open()
 151         except ValueError:
 152             db = SharedKeyDB(dbName)
 153             db.open()
 154
 155         try:
 156             db[username]
 157             print "Username exists"
 158
 159             if password:
 160                 if db.check(username, password):
 161                     print "Password is correct"
 162                 else:
 163                     print "Password is wrong"
 164         except KeyError:
 165             print "Username does not exist"
 166             sys.exit()
 167
 168     elif cmd == "list":
 169         dbName = args.get(2)
 170
 171         try:
 172             db = VerifierDB(dbName)
 173             db.open()
 174         except ValueError:
 175             db = SharedKeyDB(dbName)
 176             db.open()
 177
 178         if isinstance(db, VerifierDB):
 179             print "Verifier Database"
 180             def numBits(n):
 181                 if n==0:
 182                     return 0
 183                 return int(math.floor(math.log(n, 2))+1)
 184             for username in db.keys():
 185                 N, g, s, v = db[username]
 186                 print numBits(N), username
 187         else:
 188             print "Shared Key Database"
 189             for username in db.keys():
 190                 print username
 191     else:
 192         print "Bad command: '%s'" % cmd
 193 except:
 194     raise