Files
Sunscreen/sunscreen_docs/book/getting_started/example.html
Rick Weber a323229243 WIP
2022-05-02 12:05:08 -07:00

315 lines
23 KiB
HTML

<!DOCTYPE HTML>
<html lang="en" class="sidebar-visible no-js light">
<head>
<!-- Book generated using mdBook -->
<meta charset="UTF-8">
<title>My first FHE program - Sunscreen Documentation</title>
<!-- Custom HTML head -->
<meta content="text/html; charset=utf-8" http-equiv="Content-Type">
<meta name="description" content="">
<meta name="viewport" content="width=device-width, initial-scale=1">
<meta name="theme-color" content="#ffffff" />
<link rel="icon" href="../favicon.svg">
<link rel="shortcut icon" href="../favicon.png">
<link rel="stylesheet" href="../css/variables.css">
<link rel="stylesheet" href="../css/general.css">
<link rel="stylesheet" href="../css/chrome.css">
<link rel="stylesheet" href="../css/print.css" media="print">
<!-- Fonts -->
<link rel="stylesheet" href="../FontAwesome/css/font-awesome.css">
<link rel="stylesheet" href="../fonts/fonts.css">
<!-- Highlight.js Stylesheets -->
<link rel="stylesheet" href="../highlight.css">
<link rel="stylesheet" href="../tomorrow-night.css">
<link rel="stylesheet" href="../ayu-highlight.css">
<!-- Custom theme stylesheets -->
<!-- MathJax -->
<script async type="text/javascript" src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.1/MathJax.js?config=TeX-AMS-MML_HTMLorMML"></script>
</head>
<body>
<!-- Provide site root to javascript -->
<script type="text/javascript">
var path_to_root = "../";
var default_theme = window.matchMedia("(prefers-color-scheme: dark)").matches ? "navy" : "light";
</script>
<!-- Work around some values being stored in localStorage wrapped in quotes -->
<script type="text/javascript">
try {
var theme = localStorage.getItem('mdbook-theme');
var sidebar = localStorage.getItem('mdbook-sidebar');
if (theme.startsWith('"') && theme.endsWith('"')) {
localStorage.setItem('mdbook-theme', theme.slice(1, theme.length - 1));
}
if (sidebar.startsWith('"') && sidebar.endsWith('"')) {
localStorage.setItem('mdbook-sidebar', sidebar.slice(1, sidebar.length - 1));
}
} catch (e) { }
</script>
<!-- Set the theme before any content is loaded, prevents flash -->
<script type="text/javascript">
var theme;
try { theme = localStorage.getItem('mdbook-theme'); } catch(e) { }
if (theme === null || theme === undefined) { theme = default_theme; }
var html = document.querySelector('html');
html.classList.remove('no-js')
html.classList.remove('light')
html.classList.add(theme);
html.classList.add('js');
</script>
<!-- Hide / unhide sidebar before it is displayed -->
<script type="text/javascript">
var html = document.querySelector('html');
var sidebar = 'hidden';
if (document.body.clientWidth >= 1080) {
try { sidebar = localStorage.getItem('mdbook-sidebar'); } catch(e) { }
sidebar = sidebar || 'visible';
}
html.classList.remove('sidebar-visible');
html.classList.add("sidebar-" + sidebar);
</script>
<nav id="sidebar" class="sidebar" aria-label="Table of contents">
<div class="sidebar-scrollbox">
<ol class="chapter"><li class="chapter-item expanded "><a href="../intro/intro.html"><strong aria-hidden="true">1.</strong> Introduction</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../intro/prelim.html"><strong aria-hidden="true">1.1.</strong> Prerequisites</a></li><li class="chapter-item expanded "><a href="../intro/why.html"><strong aria-hidden="true">1.2.</strong> Why is a compiler needed for FHE?</a></li><li class="chapter-item expanded "><a href="../intro/compiler.html"><strong aria-hidden="true">1.3.</strong> Sunscreen's compiler</a></li></ol></li><li class="chapter-item expanded "><a href="../getting_started/getting_started.html"><strong aria-hidden="true">2.</strong> Getting started</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../getting_started/installation.html"><strong aria-hidden="true">2.1.</strong> Installation</a></li><li class="chapter-item expanded "><a href="../getting_started/example.html" class="active"><strong aria-hidden="true">2.2.</strong> My first FHE program</a></li></ol></li><li class="chapter-item expanded "><a href="../fhe_programs/fhe_programs.html"><strong aria-hidden="true">3.</strong> What's in an FHE program?</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../fhe_programs/types/types.html"><strong aria-hidden="true">3.1.</strong> Types</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../fhe_programs/types/signed.html"><strong aria-hidden="true">3.1.1.</strong> Signed</a></li><li class="chapter-item expanded "><a href="../fhe_programs/types/fractional.html"><strong aria-hidden="true">3.1.2.</strong> Fractional</a></li><li class="chapter-item expanded "><a href="../fhe_programs/types/rational.html"><strong aria-hidden="true">3.1.3.</strong> Rational</a></li></ol></li><li class="chapter-item expanded "><a href="../fhe_programs/writing_an_fhe_program/writing_an_fhe_program.html"><strong aria-hidden="true">3.2.</strong> How to write an FHE program</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../fhe_programs/factoring_fhe_programs.html"><strong aria-hidden="true">3.2.1.</strong> Factoring FHE programs</a></li><li class="chapter-item expanded "><a href="../fhe_programs/writing_an_fhe_program/limitations.html"><strong aria-hidden="true">3.2.2.</strong> Limitations</a></li></ol></li><li class="chapter-item expanded "><a href="../troubleshooting.html"><strong aria-hidden="true">3.3.</strong> Troubleshooting</a></li></ol></li><li class="chapter-item expanded "><a href="../fhe_programs/runtime/runtime.html"><strong aria-hidden="true">4.</strong> Runtime</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../fhe_programs/runtime/key_generation.html"><strong aria-hidden="true">4.1.</strong> Key generation</a></li><li class="chapter-item expanded "><a href="../fhe_programs/runtime/encryption.html"><strong aria-hidden="true">4.2.</strong> Encryption</a></li><li class="chapter-item expanded "><a href="../fhe_programs/runtime/running_fhe_programs.html"><strong aria-hidden="true">4.3.</strong> Running FHE programs</a></li><li class="chapter-item expanded "><a href="../fhe_programs/runtime/decryption.html"><strong aria-hidden="true">4.4.</strong> Decryption</a></li><li class="chapter-item expanded "><a href="../fhe_programs/runtime/serialization.html"><strong aria-hidden="true">4.5.</strong> Serialization</a></li></ol></li><li class="chapter-item expanded "><a href="../fhe_programs/apps.html"><strong aria-hidden="true">5.</strong> Applications</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../fhe_programs/example.html"><strong aria-hidden="true">5.1.</strong> AMMs &amp; private token swaps</a></li><li class="chapter-item expanded "><a href="../fhe_programs/pir_intro.html"><strong aria-hidden="true">5.2.</strong> Private information retrieval</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../fhe_programs/pir_simple.html"><strong aria-hidden="true">5.2.1.</strong> Take 1</a></li><li class="chapter-item expanded "><a href="../fhe_programs/pir_matrix.html"><strong aria-hidden="true">5.2.2.</strong> Take 2</a></li></ol></li><li class="chapter-item expanded "><div><strong aria-hidden="true">5.3.</strong> Genomics &amp; private tests</div></li></ol></li><li class="chapter-item expanded "><a href="../advanced/faq.html"><strong aria-hidden="true">6.</strong> FAQ</a></li><li class="chapter-item expanded "><a href="../advanced/advanced.html"><strong aria-hidden="true">7.</strong> Advanced topics</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../advanced/bfv.html"><strong aria-hidden="true">7.1.</strong> Why the BFV scheme?</a></li><li class="chapter-item expanded "><a href="../advanced/good_fhe_programs.html"><strong aria-hidden="true">7.2.</strong> Writing even better FHE programs</a></li><li><ol class="section"><li class="chapter-item expanded "><a href="../advanced/plain_modulus/plain_modulus.html"><strong aria-hidden="true">7.2.1.</strong> Plaintext modulus</a></li><li class="chapter-item expanded "><a href="../advanced/noise_margin.html"><strong aria-hidden="true">7.2.2.</strong> Noise</a></li><li class="chapter-item expanded "><a href="../advanced/pruning_keys.html"><strong aria-hidden="true">7.2.3.</strong> Pruning public keys</a></li></ol></li><li class="chapter-item expanded "><a href="../advanced/wasm.html"><strong aria-hidden="true">7.3.</strong> WASM support</a></li><li class="chapter-item expanded "><a href="../advanced/carryless_arithmetic.html"><strong aria-hidden="true">7.4.</strong> Funky math: carryless arithmetic</a></li><li class="chapter-item expanded "><div><strong aria-hidden="true">7.5.</strong> Batching</div></li></ol></li></ol>
</div>
<div id="sidebar-resize-handle" class="sidebar-resize-handle"></div>
</nav>
<div id="page-wrapper" class="page-wrapper">
<div class="page">
<div id="menu-bar-hover-placeholder"></div>
<div id="menu-bar" class="menu-bar sticky bordered">
<div class="left-buttons">
<button id="sidebar-toggle" class="icon-button" type="button" title="Toggle Table of Contents" aria-label="Toggle Table of Contents" aria-controls="sidebar">
<i class="fa fa-bars"></i>
</button>
<button id="theme-toggle" class="icon-button" type="button" title="Change theme" aria-label="Change theme" aria-haspopup="true" aria-expanded="false" aria-controls="theme-list">
<i class="fa fa-paint-brush"></i>
</button>
<ul id="theme-list" class="theme-popup" aria-label="Themes" role="menu">
<li role="none"><button role="menuitem" class="theme" id="light">Light (default)</button></li>
<li role="none"><button role="menuitem" class="theme" id="rust">Rust</button></li>
<li role="none"><button role="menuitem" class="theme" id="coal">Coal</button></li>
<li role="none"><button role="menuitem" class="theme" id="navy">Navy</button></li>
<li role="none"><button role="menuitem" class="theme" id="ayu">Ayu</button></li>
</ul>
<button id="search-toggle" class="icon-button" type="button" title="Search. (Shortkey: s)" aria-label="Toggle Searchbar" aria-expanded="false" aria-keyshortcuts="S" aria-controls="searchbar">
<i class="fa fa-search"></i>
</button>
</div>
<h1 class="menu-title">Sunscreen Documentation</h1>
<div class="right-buttons">
<a href="../print.html" title="Print this book" aria-label="Print this book">
<i id="print-button" class="fa fa-print"></i>
</a>
</div>
</div>
<div id="search-wrapper" class="hidden">
<form id="searchbar-outer" class="searchbar-outer">
<input type="search" id="searchbar" name="searchbar" placeholder="Search this book ..." aria-controls="searchresults-outer" aria-describedby="searchresults-header">
</form>
<div id="searchresults-outer" class="searchresults-outer hidden">
<div id="searchresults-header" class="searchresults-header"></div>
<ul id="searchresults">
</ul>
</div>
</div>
<!-- Apply ARIA attributes after the sidebar and the sidebar toggle button are added to the DOM -->
<script type="text/javascript">
document.getElementById('sidebar-toggle').setAttribute('aria-expanded', sidebar === 'visible');
document.getElementById('sidebar').setAttribute('aria-hidden', sidebar !== 'visible');
Array.from(document.querySelectorAll('#sidebar a')).forEach(function(link) {
link.setAttribute('tabIndex', sidebar === 'visible' ? 0 : -1);
});
</script>
<div id="content" class="content">
<main>
<h1 id="my-first-fhe-program"><a class="header" href="#my-first-fhe-program">My first FHE program</a></h1>
<p>Now that we have installed the Sunscreen crate as a dependency, let's get started writing our first private app using FHE! Writing our program will be a gradual process and we'll add more code as we progress through this section. </p>
<p>In this example, we'll just multiply two encrypted integers.</p>
<pre><pre class="playground"><code class="language-rust no_run edition2021">use sunscreen::{
fhe_program,
types::{bfv::Signed, Cipher},
Compiler, Error, Runtime,
};
#[fhe_program(scheme = &quot;bfv&quot;)]
fn simple_multiply(a: Cipher&lt;Signed&gt;, b: Cipher&lt;Signed&gt;) -&gt; Cipher&lt;Signed&gt; {
a * b
}
fn main() {
}
</code></pre></pre>
<p>Notice that the <code>simple_multiply</code> function is like any other function in Rust, except for the <code>#[fhe_program(...)]</code> attribute. This is where the magic happens— it declares your function as an FHE program that can be compiled. The <code>scheme</code> argument should always be <code>&quot;bfv&quot;</code>, though we may support additional FHE schemes in the future.</p>
<p><code>simple_multiply</code>'s signature specifies that it takes in two <code>Cipher&lt;Signed&gt;</code> values and returns one. <code>Cipher&lt;T&gt;</code> indicates the contained type <code>T</code> is encrypted (i.e. a ciphertext) and <code>Signed</code> is Sunscreen's signed integer type; thus, <code>Cipher&lt;Signed&gt;</code> indicates that we have an encrypted signed integer. The body of <code>simple_multiply</code> multiplies the ciphertexts <code>a</code> and <code>b</code> together. As with any function in Rust, omitting a <code>;</code> returns an expression's value from the current scope.</p>
<p>Having specified our program, let's compile it.</p>
<pre><pre class="playground"><code class="language-rust no_run edition2021">use sunscreen::{
fhe_program,
types::{bfv::Signed, Cipher},
Compiler, Error, Runtime,
};
#[fhe_program(scheme = &quot;bfv&quot;)]
fn simple_multiply(a: Cipher&lt;Signed&gt;, b: Cipher&lt;Signed&gt;) -&gt; Cipher&lt;Signed&gt; {
a * b
}
fn main() -&gt; Result&lt;(), Error&gt; {
let fhe_program = Compiler::with_fhe_program(simple_multiply)
.compile()?;
Ok(())
}
</code></pre></pre>
<p>The <code>Compiler::with_fhe_program(simple_multiply)</code> line creates a compiler and <code>.compile()</code>, tells it to compile our FHE program. Compilation performs optimization and fills in implementation details, including figuring out FHE scheme parameters and inserting special operations. </p>
<p>What's the <code>?</code> after at the end of <code>.compile()</code>? For the uninitiated, the <a href="https://doc.rust-lang.org/book/ch09-02-recoverable-errors-with-result.html"><code>?</code></a> operator propagates errors. Fallible expressions in Rust emit <a href="https://doc.rust-lang.org/std/result/enum.Result.html"><code>Results</code></a>, which can contain either a value or an error. Using <code>?</code> unwraps the value in a successful result or immediately returns the error from a failed one, letting the caller of the current function deal with it. We should see the former after compilation, as our program is well-formed.</p>
<p>Next, we need a public and private key pair. In order to generate keys, we'll first construct a <code>Runtime</code> with the parameters we got from compilation. This allows us to encrypt/decrypt data and run FHE programs.</p>
<pre><pre class="playground"><code class="language-rust no_run edition2021">use sunscreen::{
fhe_program,
types::{bfv::Signed, Cipher},
Compiler, Error, Runtime,
};
#[fhe_program(scheme = &quot;bfv&quot;)]
fn simple_multiply(a: Cipher&lt;Signed&gt;, b: Cipher&lt;Signed&gt;) -&gt; Cipher&lt;Signed&gt; {
a * b
}
fn main() -&gt; Result&lt;(), Error&gt; {
let fhe_program = Compiler::with_fhe_program(simple_multiply)
.compile()?;
let runtime = Runtime::new(&amp;fhe_program.metadata.params)?;
let (public_key, private_key) = runtime.generate_keys()?;
let a = runtime.encrypt(Signed::from(15), &amp;public_key)?;
let b = runtime.encrypt(Signed::from(5), &amp;public_key)?;
Ok(())
}
</code></pre></pre>
<p>After constructing our runtime, we generate a public and private key pair by calling <code>runtime.generate_keys()</code>.</p>
<p>Next, we call <code>Signed::from(15)</code> to make an unencrypted <code>Signed</code> integer equal to <code>15</code>. Rust doesn't allow implicit type conversion as some languages do, so we use the <code>From</code> trait to explicitly convert a Rust <code>i64</code> into a Sunscreen <code>Signed</code>.</p>
<p>Once we have our plaintext value 15, we encrypt it by calling <code>runtime.encrypt(...)</code>, passing in the value and our public key. We repeat this process for <code>b</code> with the value <code>5</code>. Now that we have the two ciphertexts <code>a</code> and <code>b</code> to give to <code>simple_multiply</code>, we're ready to run our FHE program!</p>
<pre><pre class="playground"><code class="language-rust no_run edition2021">use sunscreen::{
fhe_program,
types::{bfv::Signed, Cipher},
Compiler, Error, Runtime,
};
#[fhe_program(scheme = &quot;bfv&quot;)]
fn simple_multiply(a: Cipher&lt;Signed&gt;, b: Cipher&lt;Signed&gt;) -&gt; Cipher&lt;Signed&gt; {
a * b
}
fn main() -&gt; Result&lt;(), Error&gt; {
let fhe_program = Compiler::with_fhe_program(simple_multiply)
.compile()?;
let runtime = Runtime::new(&amp;fhe_program.metadata.params)?;
let (public_key, private_key) = runtime.generate_keys()?;
let a = runtime.encrypt(Signed::from(15), &amp;public_key)?;
let b = runtime.encrypt(Signed::from(5), &amp;public_key)?;
let results = runtime.run(&amp;fhe_program, vec![a, b], &amp;public_key)?;
let c: Signed = runtime.decrypt(&amp;results[0], &amp;private_key)?;
assert_eq!(c, 75.into());
Ok(())
}
</code></pre></pre>
<p>We call <code>runtime.run(...)</code> to execute our FHE program. For the first argument, we pass in our previously compiled program <code>fhe_program</code>. The second argument is always a <a href="https://doc.rust-lang.org/std/vec/struct.Vec.html">Vec</a> containing the arguments to the FHE program. In this case, we pass in the encrypted <code>a</code> and <code>b</code> values. You'll need to pass in the <code>public_key</code> as well.</p>
<p>What would happen if we forgot to encrypt one of our values or gave an encrypted <code>Fractional</code> value where the program wanted an encrypted <code>Signed</code> value? Fortunately, the <code>run</code> method first performs some sanity checks to ensure the arguments match the call signature. If the types of the values we pass in don't match the signature, the <code>run</code> method returns an error <code>Result</code>. The <code>?</code> propagates this error, but our program exits because this is the <code>main()</code> method!</p>
<p>Next, we call <code>runtime.decrypt(...)</code> with the first result and our private key. Programs <em>can</em> return more than one value; hence, <code>results</code> is a <code>Vec</code>. Since our FHE program only returns one value, we decrypt the value at index <code>0</code>. The left hand side of the assignment denotes the decrypted data is a <code>Signed</code> whereas <code>runtime.decrypt(...)</code> ensures this type matches the ciphertext's encrypted value before decryption. If we had assigned a different type to <code>c</code>, say <code>Fractional</code>, then decrypt would return an error.</p>
<p>Finally, we verify the result equals 75 (i.e. 15 * 5) as expected. </p>
</main>
<nav class="nav-wrapper" aria-label="Page navigation">
<!-- Mobile navigation buttons -->
<a rel="prev" href="../getting_started/installation.html" class="mobile-nav-chapters previous" title="Previous chapter" aria-label="Previous chapter" aria-keyshortcuts="Left">
<i class="fa fa-angle-left"></i>
</a>
<a rel="next" href="../fhe_programs/fhe_programs.html" class="mobile-nav-chapters next" title="Next chapter" aria-label="Next chapter" aria-keyshortcuts="Right">
<i class="fa fa-angle-right"></i>
</a>
<div style="clear: both"></div>
</nav>
</div>
</div>
<nav class="nav-wide-wrapper" aria-label="Page navigation">
<a rel="prev" href="../getting_started/installation.html" class="nav-chapters previous" title="Previous chapter" aria-label="Previous chapter" aria-keyshortcuts="Left">
<i class="fa fa-angle-left"></i>
</a>
<a rel="next" href="../fhe_programs/fhe_programs.html" class="nav-chapters next" title="Next chapter" aria-label="Next chapter" aria-keyshortcuts="Right">
<i class="fa fa-angle-right"></i>
</a>
</nav>
</div>
<!-- Livereload script (if served using the cli tool) -->
<script type="text/javascript">
var socket = new WebSocket("ws://localhost:3000/__livereload");
socket.onmessage = function (event) {
if (event.data === "reload") {
socket.close();
location.reload();
}
};
window.onbeforeunload = function() {
socket.close();
}
</script>
<script type="text/javascript">
window.playground_copyable = true;
</script>
<script src="../elasticlunr.min.js" type="text/javascript" charset="utf-8"></script>
<script src="../mark.min.js" type="text/javascript" charset="utf-8"></script>
<script src="../searcher.js" type="text/javascript" charset="utf-8"></script>
<script src="../clipboard.min.js" type="text/javascript" charset="utf-8"></script>
<script src="../highlight.js" type="text/javascript" charset="utf-8"></script>
<script src="../book.js" type="text/javascript" charset="utf-8"></script>
<!-- Custom JS scripts -->
</body>
</html>