Merge "Assume input from /dev/stdin if no argument provided to btsnooz.py." into cw-e-dev

import base64

import fileinput

import struct

import sys

import zlib

def main():

  if len(sys.argv) != 2:

    sys.stderr.write('Usage: %s <bugreport>\n' % sys.argv[0])

  if len(sys.argv) > 2:

    sys.stderr.write('Usage: %s [bugreport]\n' % sys.argv[0])

    exit(1)

  with open(sys.argv[1]) as f:

    while True:

      line = f.readline()

      if not line:

        break

  iterator = fileinput.input()

  for line in iterator:

    if line.startswith('--- BEGIN:BTSNOOP_LOG_SUMMARY'):

        decode_snooz(base64.standard_b64decode(f.readline()))

      decode_snooz(base64.standard_b64decode(iterator.next()))

      sys.exit(0)

  sys.stderr.write('No btsnooz section found in bugreport.\n');