src/virtualization/buildtools/romfsck.py

#!/usr/bin/env python
# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import sys

def main():
  with open(sys.argv[1], 'rb') as image:
    with open(sys.argv[2], 'w') as output:
      image.seek(0, 2)
      if (image.tell() % 1024):
        print('File not block aligned')
        exit(-1)
      image.seek(0)
      magic = image.read(8)
      if (magic != b'-rom1fs-'):
        print('Bad header magic {}'.format(magic))
        exit(-1)
      image.seek(0)
      csum = 0
      i = 3
      for b in image.read(512):
        csum += (b if isinstance(b, int) else ord(b)) << (8 * i)
        if (i > 0):
          i -= 1
        else:
          i = 3
      csum &= 0xFFFFFFFF
      if (csum != 0):
        print('Bad checksum {}'.format(csum))
        exit(-1)
      output.write('ok')

if __name__ == '__main__':
  sys.exit(main())