#!/usr/bin/env python3
#
# Copyright (C) 2021 Amlogic, Inc. All rights reserved.
#
# This source code is subject to the terms and conditions defined in the
# file 'LICENSE' which is part of this source code package.

TARGET_NAME = 'dgpk2'
MAX_IDX = 100000000
DGPK2_SIZE = 16

def get_args():
	from argparse import ArgumentParser

	parser = ArgumentParser()
	parser.add_argument('--soc', type = str, required = True, help = 'soc')
	parser.add_argument('--count', type = int, required = True, help = 'device count')
	parser.add_argument('--out_dir', type = str, required = True, help = 'output directory')
	return parser.parse_args()

# dgpk: device global protect key
def main():
	import os

	args = get_args()

	if not os.path.exists(args.out_dir):
		os.mkdir(args.out_dir)

	if args.count > MAX_IDX:
		args.count = MAX_IDX
		print('the max number of ' + TARGET_NAME + ' derived at one time is ' + MAX_IDX)

	derived_num = 0
	for i in range(1, args.count + 1):
		dgpk2 = os.urandom(DGPK2_SIZE)

		fpath = args.out_dir + '/' + args.soc + '_' + TARGET_NAME + '_' + str(i).zfill(9) + '.bin'
		f = open(fpath, 'wb')
		f.write(dgpk2)
		f.close()

		derived_num = derived_num + 1
		print(os.path.abspath(fpath) + ' derived successfully')

	print(str(derived_num) + ' ' + TARGET_NAME + ' files derived successfully')

if __name__ == '__main__':
	main()