#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
from __future__ import print_function
import base64

"""
    cbytes.py

    Output a c file with the DER certificate.
    Read der file as input
"""
import sys, fileinput, binascii

if len(sys.argv) not in [2, 3]:
    print('usage: %s <certificate.der|hex-input> [-s]' % sys.argv[0])
    print('    -s: just output c string (for general use)')
    sys.exit(1)

buf = None
try:
    buf = bytearray(open(sys.argv[1], 'rb').read())
except:
    n = sys.argv[1].replace('\n', '')
    n = sys.argv[1].replace('\r', '')
    buf = bytearray(binascii.unhexlify(n))

c_str = ''
size = len(buf)

a = ''.join(map(lambda c: '\\x%02x' % c, buf))

for i in range(0, len(a), 80):
    c_str += "\"" + a[i : i + 80] + "\"\n"

if '-s' in sys.argv:
    print(c_str)
    sys.exit(0)

print('// generated')
print('#include <stdint.h>')
print()
print('code uint8_t __attest[] = \n%s;' % c_str)
print('const uint16_t __attest_size = sizeof(__attest)-1;')

b = base64.b64encode(buf)
print('b64: ')
print(b)