htmlencoder.py 文件源码

def main():
    parser = _create_argument_parser()
    args = parser.parse_args()

    # Alphabet for the experiments
    I = [ord(x) for x in string.punctuation + string.lowercase]

    enc_learn = EncoderLearner(I)
    print '[+] Learning HTML Encoder: ',
    sanitizer = enc_learn.learn_mealy_machine()
    print 'OK'

    print '[+] Saving transducer model in file {}.txt: '.format(args.outfile),
    sanitizer.save(args.outfile + '.txt')
    print 'OK'

    if args.save_bek:
        print '[+] Saving BEK program in file {}.bek: '.format(args.outfile),
        bek = BekProgram()
        bek.create_from_transducer(sanitizer)
        bek.save(args.outfile + '.bek')
        print 'OK'