#!/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 != '-rom1fs-'):
        print 'Bad header magic ' + magic
        exit(-1)
      image.seek(0)
      csum = 0
      i = 3
      for b in image.read(512):
        csum += ord(b) << (8 * i)
        if (i > 0):
          i -= 1
        else:
          i = 3
      csum &= 0xFFFFFFFF
      if (csum != 0):
        print 'Bad checksum ' + csum
        exit(-1)
      output.write('ok')

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