examples: add rabe example (#42)

* examples: add rabe example

* docker: few nits

* rust: move to p1 target

* gha: fix cargo build

Author: csegarragonz

.github/workflows/tests.yml

@@ -67,6 +67,12 @@ jobs:
         uses: actions/checkout@v4
         with:
           submodules: true
+      - name: "Build Rabe"
+        run: |
+          export PATH=$PATH:/root/.cargo/env
+          rustup default stable
+          rustup target add wasm32-wasip1
+          ./bin/inv_wrapper.sh rabe
       - name: "Build FFmpeg"
         run: ./bin/inv_wrapper.sh ffmpeg
       - name: "Build ImageMagick"
@@ -122,7 +128,7 @@ jobs:
         shell: bash
     env:
       FAASM_INI_FILE: ./faasm.ini
-      FAASM_VERSION: 0.22.0
+      FAASM_VERSION: 0.27.0
       FAASM_WASM_VM: ${{ matrix.faasm_wasm_vm }}
     steps:
       - uses: csegarragonz/set-compose-version-action@main
@@ -280,6 +286,12 @@ jobs:
       - name: "Run FFmpeg check"
         timeout-minutes: 1
         run: faasmctl invoke ffmpeg check
+          # TODO(faasm-bump): this is working, but we need to cut a new faasm release, uncomment
+          # when we do
+          #       - name: "Run Rabe test"
+          #         if: "contains(env.FAASM_WASM_VM, 'wamr')"
+          #         timeout-minutes: 1
+          #         run: faasmctl invoke rabe test
       - name: "Print logs in case of failure"
         if: failure()
         run: faasmctl logs -s worker

.gitmodules

@@ -47,3 +47,7 @@
 	path = examples/Kernels-elastic
 	url = https://github.com/faasm/Kernels
 	branch = faasm-elastic
+[submodule "examples/rabe"]
+	path = examples/rabe
+	url = https://github.com/faasm/rabe.git
+	branch = faasm

cpp

@@ -1,5 +1,3 @@
-version: "3"
-
 services:
   build:
     image: faasm.azurecr.io/examples-build:${EXAMPLES_BUILD_VERSION}

docker-compose.yml

@@ -1,4 +1,4 @@
-FROM ubuntu:22.04 as base
+FROM ubuntu:22.04 AS base
 
 RUN apt update \
     && apt install -y \

docker/base.dockerfile

@@ -1,8 +1,8 @@
 ARG CPP_VERSION
 ARG EXAMPLES_VERSION
 # Base image is not re-built often and tag may lag behind
-FROM faasm.azurecr.io/examples-base:0.4.0_0.4.0 as base
-FROM faasm.azurecr.io/cpp-sysroot:${CPP_VERSION}
+FROM faasm.azurecr.io/examples-base:0.6.0_0.4.0 AS base
+FROM faasm.azurecr.io/cpp-sysroot:${CPP_VERSION:-dead}
 
 SHELL ["/bin/bash", "-c"]
 ENV IN_DOCKER="on"
@@ -21,6 +21,12 @@ ARG DEBIAN_FRONTEND=noninteractive
 RUN apt update \
     && apt install -y libomp-17-dev
 
+# Install rust
+RUN curl --proto '=https' --tlsv1.3 https://sh.rustup.rs -sSf | sh -s -- -y \
+    && source ~/.cargo/env \
+    && rustup target add wasm32-wasip1
+
+
 # Fetch the code and update submodules
 ARG EXAMPLES_VERSION
 RUN mkdir -p code \
@@ -40,9 +46,11 @@ RUN mkdir -p code \
     && git submodule update --init -f examples/LULESH \
     && git submodule update --init -f examples/libpng \
     && git submodule update --init -f examples/polybench \
+    && git submodule update --init -f examples/rabe \
     && git submodule update --init -f examples/tensorflow
 
 # Build the examples and demo functions
+ENV PATH=${PATH}:/root/.cargo/bin
 RUN cd /code/examples \
     && ./bin/create_venv.sh \
     && source venv/bin/activate \
@@ -63,14 +71,17 @@ RUN cd /code/examples \
     && inv lammps --migration-net \
     && inv lulesh \
     && inv polybench \
+    && inv rabe \
     && inv tensorflow \
     # These demo functions link with the cross-compiled static libraries
     && inv func ffmpeg check \
     && inv func lammps chain \
     && inv func mpi migrate \
+    && inv func rabe test \
     && inv func tf check
 
 # Prepare bashrc
 WORKDIR /code/examples
 RUN echo ". /code/examples/bin/workon.sh" >> ~/.bashrc
+RUN echo ". $HOME/.cargo/env" >> ~/.bashrc
 CMD ["/bin/bash", "-l"]

docker/build.dockerfile

@@ -34,4 +34,5 @@ endfunction(faasm_example_func)
 add_subdirectory(ffmpeg)
 add_subdirectory(lammps)
 add_subdirectory(mpi)
+add_subdirectory(rabe)
 add_subdirectory(tf)

examples/rabe

@@ -0,0 +1,4 @@
+set(FAASM_USER rabe)
+
+faasm_example_func(test test.cpp)
+target_link_libraries(rabe_test librabe-cpp.a librabe.a)

func/CMakeLists.txt

@@ -0,0 +1,29 @@
+#include <rabe_bindings.hpp>
+#include <iostream>
+#include <string>
+
+int main()
+{
+    // Create temporary encryption context for this test
+    auto& ctx = rabe::CpAbeContextWrapper::get(rabe::ContextFetchMode::Create);
+
+    // Prepare encryption
+    std::string plainText = "dance like no one's watching, encrypt like everyone is!";
+    std::string policy = "\"A\" and \"B\"";
+    auto cipherText = ctx.cpAbeEncrypt(policy, plainText);
+
+    // Prepare decryption
+    std::vector<std::string> attributes = {"A", "B"};
+    auto actualPlainText = ctx.cpAbeDecrypt(attributes, cipherText);
+
+    // Compare
+    std::string actualPlainTextStr;
+    actualPlainTextStr.assign(reinterpret_cast<char*>(actualPlainText.data()), actualPlainText.size());
+    if (plainText != actualPlainTextStr) {
+        std::cerr << "Encryption/decryption test failed!" << std::endl;
+        return -1;
+    }
+
+    std::cout << "Encryption worked!" << std::endl;
+    return 0;
+}