made tools/findbits.py importable as a Python module

This commit is contained in:
dn337t@gmail.com 2012-06-20 15:07:30 +00:00
parent 9f69393035
commit 79c4de2a0f

View file

@ -75,7 +75,8 @@ def domatch(number,binary):
print ' Reverse: (%s)' % reversed,
search(inverted,reversed)
if(len(sys.argv) < 3):
def main():
if(len(sys.argv) < 3):
print
print '\t'+sys.argv[0] + ' - Search bitstream for a known number'
print
@ -92,14 +93,14 @@ if(len(sys.argv) < 3):
print
os._exit(True)
bases= {
bases= {
2:'BINARY',
8:'OCTAL',
10:'DECIMAL',
16:'HEX',
}
for base in 2,8,10,16:
for base in 2,8,10,16:
try:
number= int(sys.argv[1],base)
print
@ -111,3 +112,6 @@ for base in 2,8,10,16:
domatch(binstring(number),sys.argv[2])
except:
continue
if __name__ == '__main__':
main()