mirror of
https://github.com/adulau/crl-monitor.git
synced 2024-12-22 16:56:01 +00:00
-i option added - Indexing subject field
This commit is contained in:
parent
5bfca0f3d9
commit
685b0b5ae9
1 changed files with 23 additions and 3 deletions
|
@ -32,8 +32,22 @@ argParser.add_argument('-v', action='store_true', help='Verbose output')
|
|||
argParser.add_argument('-k', default=False, action='store_true', help='Add certificate to keystore')
|
||||
argParser.add_argument('-d', default=None, help='Certificate directory')
|
||||
argParser.add_argument('-r', default='-', help='Read from a file, default is stdin')
|
||||
argParser.add_argument('-i', default=False, action='store_true', help='Enable full-text indexing (default is disabled)')
|
||||
args = argParser.parse_args()
|
||||
|
||||
if args.i:
|
||||
from whoosh.index import create_in, exists_in, open_dir
|
||||
from whoosh.fields import *
|
||||
schema = Schema(path=ID(stored=True,unique=True), content=TEXT)
|
||||
indexpath = '/tmp/findex'
|
||||
if not os.path.exists(indexpath):
|
||||
os.mkdir(indexpath)
|
||||
if not exists_in(indexpath):
|
||||
ix = create_in(indexpath, schema)
|
||||
else:
|
||||
ix = open_dir(indexpath)
|
||||
writer = ix.writer()
|
||||
|
||||
if args.s:
|
||||
try:
|
||||
#Redis structure Set of (Subject) per FP
|
||||
|
@ -60,13 +74,13 @@ for cert in fileinput.input(args.r):
|
|||
p = args.d + "/" + bpath(ha=fp)
|
||||
if not os.path.exists(p):
|
||||
os.makedirs(p)
|
||||
fn = os.path.join(p,fp)
|
||||
fn = os.path.join(p, fp)
|
||||
if not os.path.exists(fn):
|
||||
f = open(fn, 'w+')
|
||||
f.write(certb)
|
||||
f.close()
|
||||
if args.v:
|
||||
print "Certificate saved in " +fn
|
||||
print "Certificate saved in "+fn
|
||||
|
||||
try:
|
||||
x509 = X509.load_cert_string(certb, X509.FORMAT_DER)
|
||||
|
@ -74,10 +88,16 @@ for cert in fileinput.input(args.r):
|
|||
print "At line number "+str(fileinput.lineno())+" parsing error"
|
||||
pass
|
||||
subject = x509.get_subject().as_text()
|
||||
issuer = x509.get_issuer().as_text()
|
||||
if subject is not None:
|
||||
if not args.s:
|
||||
print fp +","+ subject
|
||||
print fp+","+subject
|
||||
if args.i:
|
||||
writer.update_document(path=unicode(fp), content=unicode(subject)+" "+unicode(issuer))
|
||||
elif args.s:
|
||||
r.sadd(fp, subject)
|
||||
else:
|
||||
sys.exit(1)
|
||||
|
||||
if args.i:
|
||||
writer.commit()
|
||||
|
|
Loading…
Reference in a new issue