1. def verify_binary_integrity(filepath, expected_checksum, checksum_function='simple_sum'):
  2. """
  3. Verifies the integrity of a binary file using a specified checksum function.
  5. Args:
  6. filepath (str): The path to the binary file.
  7. expected_checksum (int): The expected checksum value.
  8. checksum_function (str): The name of the checksum function to use.
  9. Options: 'simple_sum', 'sum_of_bytes'
  11. Returns:
  12. bool: True if the checksum matches, False otherwise. Returns False if file not found.
  13. """
  14. try:
  15. with open(filepath, 'rb') as f:
  16. data = f.read()
  18. if checksum_function == 'simple_sum':
  19. # Simple sum of bytes (summing all byte values)
  20. calculated_checksum = sum(data)
  21. elif checksum_function == 'sum_of_bytes':
  22. #Sum of all bytes in the file
  23. calculated_checksum = sum(data)
  24. else:
  25. print("Invalid checksum function specified.")
  26. return False
  28. return calculated_checksum == expected_checksum
  29. except FileNotFoundError:
  30. print(f"File not found: {filepath}")
  31. return False
  32. except Exception as e:
  33. print(f"An error occurred: {e}")
  34. return False
  38. if __name__ == '__main__':
  39. # Example usage:
  40. file_path = "test_file.bin" # Replace with your binary file path
  42. # Create a dummy binary file for testing
  43. with open(file_path, 'wb') as f:
  44. f.write(b'\x01\x02\x03\x04\x05')
  46. expected_checksum = 15 # Simple sum of bytes (1+2+3+4+5)
  48. if verify_binary_integrity(file_path, expected_checksum, 'simple_sum'):
  49. print("File integrity verified.")
  50. else:
  51. print("File integrity check failed.")
  53. # Test with an incorrect checksum
  54. expected_checksum = 20
  55. if verify_binary_integrity(file_path, expected_checksum, 'simple_sum'):
  56. print("File integrity verified.")
  57. else:
  58. print("File integrity check failed.")
  60. # Test with a non-existent file
  61. if verify_binary_integrity("nonexistent_file.bin", 10, 'simple_sum'):
  62. print("File integrity verified.")
  63. else:
  64. print("File integrity check failed.")

Add your comment