diff --git a/cw/cw340_aes_sca.py b/cw/cw340_aes_sca.py new file mode 100755 index 00000000..b33f4b16 --- /dev/null +++ b/cw/cw340_aes_sca.py @@ -0,0 +1,154 @@ +#!/usr/bin/env python3 +# Copyright lowRISC contributors. +# Licensed under the Apache License, Version 2.0, see LICENSE for details. +# SPDX-License-Identifier: Apache-2.0 +import binascii +import random +import signal +import sys +import time +from datetime import datetime +from functools import partial +from pathlib import Path + +import chipwhisperer as cw +import numpy as np +import yaml +from Crypto.Cipher import AES +from tqdm import tqdm + +from util import device, plot, trace_util + +def setup_target(scope): + target = cw.target(scope) + target.baud = int(115200) + target.flush() + version = None + ping_cnt = 0 + while not version: + if ping_cnt == 3: + print("no response from target") + sys.exit(0) + target.write('v' + '\n') + ping_cnt += 1 + time.sleep(0.5) + version = target.read().strip() + print(version) + return target + +def setup_scope(cfg): + scope = cw.scope() + scope.gain.db = cfg["cwfpgahusky"]["scope_gain"] + scope.adc.basic_mode = "rising_edge" + scope.clock.clkgen_src = 'extclk' + scope.clock.clkgen_freq = cfg["cwfpgahusky"]["pll_frequency"] + scope.clock.adc_mul = cfg["cwfpgahusky"]["adc_mul"] + scope.clock.extclk_monitor_enabled = False + scope.adc.samples = cfg["cwfpgahusky"]["num_samples"] + if cfg["cwfpgahusky"]["offset"] >= 0: + scope.adc.offset = cfg["cwfpgahusky"]["offset"] + else: + scope.adc.offset = 0 + scope.adc.presamples = -cfg["cwfpgahusky"]["offset"] + scope.trigger.triggers = "tio4" + scope.io.tio1 = "serial_tx" + scope.io.tio2 = "serial_rx" + scope.io.hs2 = "disabled" + scope.clock.clkgen_src = 'extclk' + if not scope.clock.clkgen_locked: + print("scope.clock.clkgen is not locked") + sys.exit(0) + if not scope.clock.adc_locked: + print("scope.clock.adc is not locked") + sys.exit(0) + return scope + +def prepare_aes(cfg): + # Preparation of Key and plaintext generation + # Generate key at random based on test_random_seed (not used atm) + random.seed(cfg["test"]["test_random_seed"]) + key = bytearray(cfg["test"]["key_len_bytes"]) + for i in range(0, cfg["test"]["key_len_bytes"]): + key[i] = random.randint(0, 255) + # Load initial key and text values from cfg + key = bytearray(cfg["test"]["key"]) + print(f'Using key: {binascii.b2a_hex(bytes(key))}') + text = bytearray(cfg["test"]["text"]) + # Prepare generation of new texts/keys by encryption using key_for_generation + key_for_gen = bytearray(cfg["test"]["key_for_gen"]) + cipher_gen = AES.new(bytes(key_for_gen), AES.MODE_ECB) + return key, text, key_for_gen, cipher_gen + + +if __name__ == '__main__': + # Load configuration from file + with open('simple_capture_aes_sca_cw340.yaml') as f: + cfg = yaml.load(f, Loader=yaml.FullLoader) + scope = setup_scope(cfg) + target = setup_target(scope) + + # Create ChipWhisperer project for storage of traces and metadata + project = cw.create_project(cfg["capture"]["project_name"], overwrite=True) + + key, text, key_for_gen, cipher_gen = prepare_aes(cfg) + + # Set key + target.simpleserial_write("k", key) + + # Cipher to compute expected responses + cipher = AES.new(bytes(key), AES.MODE_ECB) + + # Main loop for measurements with progress bar + for _ in tqdm(range(cfg["capture"]["num_traces"]), desc='Capturing', ncols=80): + scope.arm() + + # Generate new text for this iteration + text = bytearray(cipher_gen.encrypt(text)) + # Load text and trigger execution + target.simpleserial_write('p', text) + + # Capture trace + ret = scope.capture(poll_done=False) + i = 0 + while not target.is_done(): + i += 1 + time.sleep(0.05) + if i > 100: + print("Warning: Target did not finish operation") + if ret: + print("Warning: Timeout happened during capture") + + # Get trace + wave = scope.get_last_trace(as_int=True) + + + # Get response from device and verify + response = target.simpleserial_read('r', target.output_len, ack=False) + if binascii.b2a_hex(response) != binascii.b2a_hex(cipher.encrypt(bytes(text))): + raise RuntimeError(f'Bad ciphertext: {response} != {cipher.encrypt(bytes(text))}.') + + # Sanity check retrieved data (wave) and create CW Trace + if len(wave) >= 1: + trace = cw.Trace(wave, text, response, key) + else: + raise RuntimeError('Capture failed.') + + # Check if ADC range has been exceeded for Husky. + trace_util.check_range(trace.wave, scope.adc.bits_per_sample) + + # Append CW trace to CW project storage + project.traces.append(trace, dtype=np.uint16) + + # Save metadata and entire configuration cfg to project file + project.settingsDict['datetime'] = datetime.now().strftime("%m/%d/%Y, %H:%M:%S") + project.settingsDict['cfg'] = cfg + sample_rate = int(round(scope.clock.adc_freq, -6)) + project.settingsDict['sample_rate'] = sample_rate + project.save() + + # Create and show test plot + if cfg["capture"]["show_plot"]: + plot.save_plot_to_file(project.waves, None, cfg["capture"]["plot_traces"], + cfg["capture"]["trace_image_filename"], add_mean_stddev=True) + print(f'Created plot with {cfg["capture"]["plot_traces"]} traces: ' + f'{Path(cfg["capture"]["trace_image_filename"]).resolve()}') diff --git a/cw/simple_capture_aes_sca_cw340.yaml b/cw/simple_capture_aes_sca_cw340.yaml new file mode 100644 index 00000000..6e3cb7e9 --- /dev/null +++ b/cw/simple_capture_aes_sca_cw340.yaml @@ -0,0 +1,22 @@ +cwfpgahusky: + baudrate: 115200 + output_len_bytes: 16 + pll_frequency: 100000000 + adc_mul: 2 + num_segments: 1 + num_samples: 1200 + offset: -40 + scope_gain: 31.5 +capture: + scope_select: husky + num_traces: 1000 + project_name: "projects/simple_capture_aes_sca" + show_plot: True + plot_traces: 100 + trace_image_filename: "projects/simple_capture_aes_sca_sample_traces.html" +test: + key_len_bytes: 16 + test_random_seed: 0 + key: [0x81, 0x1E, 0x37, 0x31, 0xB0, 0x12, 0x0A, 0x78, 0x42, 0x78, 0x1E, 0x22, 0xB2, 0x5C, 0xDD, 0xF9] + text: [0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA] + key_for_gen: [0x12, 0x34, 0x56, 0x78, 0x9A, 0xBC, 0xDE, 0xF1, 0x23, 0x45, 0x67, 0x89, 0xAB, 0xCD, 0xE0, 0xF0]