Skip to content

Commit

Permalink
cryptolib/botan: Add CMCE scripts
Browse files Browse the repository at this point in the history
  • Loading branch information
wagner committed Mar 27, 2024
1 parent 8a42d94 commit 364ba88
Show file tree
Hide file tree
Showing 5 changed files with 287 additions and 0 deletions.
2 changes: 2 additions & 0 deletions cryptolib/botan/algo_mceliece/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
bin/
results/
8 changes: 8 additions & 0 deletions cryptolib/botan/algo_mceliece/Makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
all: cmce

cmce: cmce.cpp
mkdir -p bin
g++-11 -std=c++20 -g -O3 -Wall -I ../botan/build/include/public -Wl,-rpath=../botan/ $^ -o bin/$@ -L ../botan/ -l:libbotan-3.a

clean:
rm -rf bin/
110 changes: 110 additions & 0 deletions cryptolib/botan/algo_mceliece/cmce.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,110 @@
#include <cassert>
#include <fstream>
#include <iostream>
#include <string>
#include <vector>
using namespace std;

#include <botan/auto_rng.h>
#include <botan/rng.h>
#include <botan/hex.h>
#include <botan/cmce.h>
#include <botan/pubkey.h>

vector<string> modes = {
"mceliece348864",
"mceliece348864f",
};

vector<string> operations = {
"keygen",
"kem"
};

const size_t shared_secret_length = 32;

void mceliece_encrypt(
Botan::Classic_McEliece_PrivateKey priv_key,
Botan::secure_vector<uint8_t> &cipher_text,
Botan::secure_vector<uint8_t> &sym_key
) {
Botan::AutoSeeded_RNG rng;
auto encryptor = Botan::PK_KEM_Encryptor(priv_key, "Raw", "base");
encryptor.encrypt(cipher_text, sym_key, rng, shared_secret_length);
}

void mceliece_decrypt(
Botan::Classic_McEliece_PrivateKey priv_key,
Botan::secure_vector<uint8_t> &cipher_text,
Botan::secure_vector<uint8_t> &sym_key
) {
Botan::AutoSeeded_RNG rng;
auto decryptor = Botan::PK_KEM_Decryptor(priv_key, rng, "Raw");
sym_key = decryptor.decrypt(cipher_text.data(), cipher_text.size(), shared_secret_length);
}

int main(int argc, char* argv[]) {
if (argc != 4) {
cout << "Usage:\n\n"
<< " mceliece <mode> <operation> <mceliecekeyfile>\n\n"
<< " <mode> ..... asymmetric cipher mode\n"
<< " <operation> ..... operation to execute, e.g. keygen or kem\n"
<< " <mceliecekeyfile> ... mceliece key file, read as text\n"
<< endl;
cout << "List of available modes:" << endl;
for(vector<string>::size_type i = 0; i != modes.size(); i++) {
cout << " " << modes[i] << endl;
}
cout << endl;
cout << "List of available operations:" << endl;
for(vector<string>::size_type i = 0; i != operations.size(); i++) {
cout << " " << operations[i] << endl;
}
cout << endl;
return (1);
}

string str_mode (argv[1]);
string str_operation (argv[2]);
string str_mceliecekeyfile (argv[3]);

Botan::Classic_McEliece_Parameter_Set params = Botan::cmce_param_set_from_str(str_mode);

std::string mceliecekeyfile_buffer_sk(str_mceliecekeyfile);

if (str_operation == "keygen") {
Botan::AutoSeeded_RNG rng;
// Alice KeyGen
const Botan::Classic_McEliece_PrivateKey priv_key(rng, params);
const auto priv_key_bits = priv_key.private_key_bits();

// Store key pair
ofstream mceliecekeyfile_sk;
mceliecekeyfile_sk.open(mceliecekeyfile_buffer_sk);
mceliecekeyfile_sk << Botan::hex_encode(priv_key_bits);
mceliecekeyfile_sk.close();
} else if (str_operation == "kem") {
Botan::secure_vector<uint8_t> cipher_text, key_bob, key_alice;

// Load key pair
string line_sk;
ifstream mceliecekeyfile_sk;
mceliecekeyfile_sk.open(mceliecekeyfile_buffer_sk);
getline(mceliecekeyfile_sk, line_sk);
mceliecekeyfile_sk.close();
Botan::secure_vector<uint8_t> priv_key_bits = Botan::hex_decode_locked(line_sk);

Botan::Classic_McEliece_PrivateKey priv_key(priv_key_bits, params);

mceliece_encrypt(priv_key, cipher_text, key_bob);
mceliece_decrypt(priv_key, cipher_text, key_alice);

assert(key_bob == key_alice);
} else {
cout << str_operation << " is no valid operation!" << endl;
assert(false);
}

return (0);
}

30 changes: 30 additions & 0 deletions cryptolib/botan/algo_mceliece/data_run.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
#!/bin/bash

set -e

# PINFLAGS="--phase1 --export"
PINFLAGS="--phase1 --phase2 --export --parallel"
# PINFLAGS="--phase1 --export --parallel"
export RESULTDIR=results


pushd ${BASH_SOURCE%/*}

if [[ $1 == "clean" || $2 == "clean" ]]; then
rm -rf $RESULTDIR
fi

./framework.sh ${PINFLAGS} minimal

if [[ $1 == "test" || $2 == "test" ]]; then
popd
exit 0
fi

./framework.sh ${PINFLAGS} mceliece348864
./framework.sh ${PINFLAGS} mceliece6688128
./framework.sh ${PINFLAGS} mceliece6688128f
./framework.sh ${PINFLAGS} mceliece6688128pc
./framework.sh ${PINFLAGS} mceliece6688128pcf

popd
137 changes: 137 additions & 0 deletions cryptolib/botan/algo_mceliece/framework.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,137 @@
#!/bin/bash

#########################################################################
# DO NOT CHANGE: Preparing DATA
#------------------------------------------------------------------------
source "${DATA_COMMON}/DATA_init.sh" || { echo "source data.sh first!" && exit 1; }
#########################################################################

#------------------------------------------------------------------------
# Specify your framework settings used by DATA
#------------------------------------------------------------------------

# The name of the framework. Do not use spaces or special characters.
export FRAMEWORK=botan

# The file containing all supported algorithms
export TARGETFILE=targets.txt

# The number of measurements for difference detection (phase1)
export PHASE1_TRACES=10

# The number of constant keys for generic tests (phase2)
# Make sure that PHASE2_FIXEDKEYS <= PHASE1_TRACES
export PHASE2_FIXEDKEYS=3

# The number of measurements per constant key for generic tests (phase2)
export PHASE2_TRACES=100

# The number of measurements for specific tests (phase3)
export PHASE3_TRACES=200

# (Optional) Additional flags for the pintool. Supported flags are:
# -main <main> Start recording at function <main>. Note that the <main>
# symbol must exist, otherwise this will yield empty traces!
# -heap Trace heap allocations and replace heap addresses with
# relative offset
export PINTOOL_ARGS="-heap"

#------------------------------------------------------------------------
# Implement your framework-specific callbacks
#------------------------------------------------------------------------
#
# Globally available environment variables:
# $FRAMEWORK The framework name
# $BASEDIR The absolute directory path of this script
# $DATA_COMMON The absolute directory for common DATA scripts
# $DATA_LEAKAGE_MODELS The absolute directory for DATA leakage models
#
# Available for cb_genkey, cb_pre_run, cb_run_command, cb_post_run
# $ALGO The currently tested algo
#
# Available for cb_pre_run, cb_run_command, cb_post_run
# $ENVFILE

# export BINARY=/home/wagner/workspace/bsi/DATA/cryptolib/botan/botan/botan-test
export BINARY=${PWD}/bin/cmce

# The leakage model of phase 3.
# See ${DATA_LEAKAGE_MODELS} for all options.
export SPECIFIC_LEAKAGE_CALLBACK=${DATA_LEAKAGE_MODELS}/rsa_privkey_hw.py

# DATA callback for setting up the framework to analyze. This callback
# is invoked once inside the current directory before analysis starts.
# Implement framework-specific tasks here like framework compilation.
function cb_prepare_framework {
:
}

# DATA callback for generating keys. This callback is invoked every
# time a new key is needed. Implement key generation according to
# your algorithm and store the generated key inside a file named $2.
#
# $1 ... key file name
function cb_genkey {
# ${BINARY} ${ALGO} keygen $1
touch $1
RES=$((RES + $?))
}

# DATA callback for custom commands that are executed immediately before
# the algorithm is profiled. It is executed in a temporary directory
# which contains the keyfile $1 and ${ENVFILE}.
#
# If 'cb_run_command' needs any other files, copy them to ${PWD}.
#
# $1 ... key file name
function cb_pre_run {
log_verbose "running with key $1"
}

# DATA callback for the main invocation of the tested algorithm.
# It shall return the bash command to execute as string. It is
# executed inside a temporary directory with a clean environment.
# If you need special files or environment variables set, specify
# them in cb_pre_run.
#
# $1 ... key file name
function cb_run_command {
echo "/home/wagner/botan/DATA/cryptolib/botan/botan/botan-test cmce_minimal --test-threads=1 --no-stdout"
# echo "${BINARY} ${ALGO} keygen $1"
# echo "${BINARY} ${ALGO} kem $1"
}

# DATA callback for custom commands that are executed immediately after
# the algorithm is profiled. It is executed in a temporary directory.
# You can cleanup any custom files generated by your algorithm.
#
# $1 ... key file name
function cb_post_run {
:
}

# DATA callback for preparing an individual algorithm. It shall:
# 1. Parse the next algorithm from the commandline string of all algorithms
# and set up anything necessary for analyzing this algorithm.
# If the algorithm needs additional parameters (like key sizes),
# increase $SHIFT accordingly.
# 2. Configure $WORKDIR, which will create a subdirectory holding all
# intermediate files generated by the algorithm and the results.
# Do not use an absolute path!
#
# $* ... algorithm string from the commandline
function cb_prepare_algo {
ALGO=$1
# key bits
SHIFT=$((SHIFT))

WORKDIR="cmce-$ALGO"
}

#########################################################################
# DO NOT CHANGE: Running DATA's commandline parser
#------------------------------------------------------------------------
DATA_parse "$@"
#------------------------------------------------------------------------
# DO NOT ADD CODE AFTER THIS LINE
#########################################################################

0 comments on commit 364ba88

Please sign in to comment.