Skip to content

Commit 0083f4d

Browse files
apoelstraroconnor-blockstream
authored andcommitted
add rust reference implementation
1 parent 6873913 commit 0083f4d

File tree

6 files changed

+1364
-0
lines changed

6 files changed

+1364
-0
lines changed

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
reference/rust-codex32/Cargo.lock
2+
reference/rust-codex32/target/

reference/rust-codex32/Cargo.toml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
[package]
2+
name = "codex32"
3+
version = "0.1.0"
4+
edition = "2018"
5+
description = "Rust reference implementation of the codex32 spec"
6+
license = "CC0-1.0"
7+
8+
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
9+
10+
[dependencies]

reference/rust-codex32/LICENSE

Lines changed: 122 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,122 @@
1+
Creative Commons Legal Code
2+
3+
CC0 1.0 Universal
4+
5+
CREATIVE COMMONS CORPORATION IS NOT A LAW FIRM AND DOES NOT PROVIDE
6+
LEGAL SERVICES. DISTRIBUTION OF THIS DOCUMENT DOES NOT CREATE AN
7+
ATTORNEY-CLIENT RELATIONSHIP. CREATIVE COMMONS PROVIDES THIS
8+
INFORMATION ON AN "AS-IS" BASIS. CREATIVE COMMONS MAKES NO WARRANTIES
9+
REGARDING THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS
10+
PROVIDED HEREUNDER, AND DISCLAIMS LIABILITY FOR DAMAGES RESULTING FROM
11+
THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS PROVIDED
12+
HEREUNDER.
13+
14+
Statement of Purpose
15+
16+
The laws of most jurisdictions throughout the world automatically confer
17+
exclusive Copyright and Related Rights (defined below) upon the creator
18+
and subsequent owner(s) (each and all, an "owner") of an original work of
19+
authorship and/or a database (each, a "Work").
20+
21+
Certain owners wish to permanently relinquish those rights to a Work for
22+
the purpose of contributing to a commons of creative, cultural and
23+
scientific works ("Commons") that the public can reliably and without fear
24+
of later claims of infringement build upon, modify, incorporate in other
25+
works, reuse and redistribute as freely as possible in any form whatsoever
26+
and for any purposes, including without limitation commercial purposes.
27+
These owners may contribute to the Commons to promote the ideal of a free
28+
culture and the further production of creative, cultural and scientific
29+
works, or to gain reputation or greater distribution for their Work in
30+
part through the use and efforts of others.
31+
32+
For these and/or other purposes and motivations, and without any
33+
expectation of additional consideration or compensation, the person
34+
associating CC0 with a Work (the "Affirmer"), to the extent that he or she
35+
is an owner of Copyright and Related Rights in the Work, voluntarily
36+
elects to apply CC0 to the Work and publicly distribute the Work under its
37+
terms, with knowledge of his or her Copyright and Related Rights in the
38+
Work and the meaning and intended legal effect of CC0 on those rights.
39+
40+
1. Copyright and Related Rights. A Work made available under CC0 may be
41+
protected by copyright and related or neighboring rights ("Copyright and
42+
Related Rights"). Copyright and Related Rights include, but are not
43+
limited to, the following:
44+
45+
i. the right to reproduce, adapt, distribute, perform, display,
46+
communicate, and translate a Work;
47+
ii. moral rights retained by the original author(s) and/or performer(s);
48+
iii. publicity and privacy rights pertaining to a person's image or
49+
likeness depicted in a Work;
50+
iv. rights protecting against unfair competition in regards to a Work,
51+
subject to the limitations in paragraph 4(a), below;
52+
v. rights protecting the extraction, dissemination, use and reuse of data
53+
in a Work;
54+
vi. database rights (such as those arising under Directive 96/9/EC of the
55+
European Parliament and of the Council of 11 March 1996 on the legal
56+
protection of databases, and under any national implementation
57+
thereof, including any amended or successor version of such
58+
directive); and
59+
vii. other similar, equivalent or corresponding rights throughout the
60+
world based on applicable law or treaty, and any national
61+
implementations thereof.
62+
63+
2. Waiver. To the greatest extent permitted by, but not in contravention
64+
of, applicable law, Affirmer hereby overtly, fully, permanently,
65+
irrevocably and unconditionally waives, abandons, and surrenders all of
66+
Affirmer's Copyright and Related Rights and associated claims and causes
67+
of action, whether now known or unknown (including existing as well as
68+
future claims and causes of action), in the Work (i) in all territories
69+
worldwide, (ii) for the maximum duration provided by applicable law or
70+
treaty (including future time extensions), (iii) in any current or future
71+
medium and for any number of copies, and (iv) for any purpose whatsoever,
72+
including without limitation commercial, advertising or promotional
73+
purposes (the "Waiver"). Affirmer makes the Waiver for the benefit of each
74+
member of the public at large and to the detriment of Affirmer's heirs and
75+
successors, fully intending that such Waiver shall not be subject to
76+
revocation, rescission, cancellation, termination, or any other legal or
77+
equitable action to disrupt the quiet enjoyment of the Work by the public
78+
as contemplated by Affirmer's express Statement of Purpose.
79+
80+
3. Public License Fallback. Should any part of the Waiver for any reason
81+
be judged legally invalid or ineffective under applicable law, then the
82+
Waiver shall be preserved to the maximum extent permitted taking into
83+
account Affirmer's express Statement of Purpose. In addition, to the
84+
extent the Waiver is so judged Affirmer hereby grants to each affected
85+
person a royalty-free, non transferable, non sublicensable, non exclusive,
86+
irrevocable and unconditional license to exercise Affirmer's Copyright and
87+
Related Rights in the Work (i) in all territories worldwide, (ii) for the
88+
maximum duration provided by applicable law or treaty (including future
89+
time extensions), (iii) in any current or future medium and for any number
90+
of copies, and (iv) for any purpose whatsoever, including without
91+
limitation commercial, advertising or promotional purposes (the
92+
"License"). The License shall be deemed effective as of the date CC0 was
93+
applied by Affirmer to the Work. Should any part of the License for any
94+
reason be judged legally invalid or ineffective under applicable law, such
95+
partial invalidity or ineffectiveness shall not invalidate the remainder
96+
of the License, and in such case Affirmer hereby affirms that he or she
97+
will not (i) exercise any of his or her remaining Copyright and Related
98+
Rights in the Work or (ii) assert any associated claims and causes of
99+
action with respect to the Work, in either case contrary to Affirmer's
100+
express Statement of Purpose.
101+
102+
4. Limitations and Disclaimers.
103+
104+
a. No trademark or patent rights held by Affirmer are waived, abandoned,
105+
surrendered, licensed or otherwise affected by this document.
106+
b. Affirmer offers the Work as-is and makes no representations or
107+
warranties of any kind concerning the Work, express, implied,
108+
statutory or otherwise, including without limitation warranties of
109+
title, merchantability, fitness for a particular purpose, non
110+
infringement, or the absence of latent or other defects, accuracy, or
111+
the present or absence of errors, whether or not discoverable, all to
112+
the greatest extent permissible under applicable law.
113+
c. Affirmer disclaims responsibility for clearing rights of other persons
114+
that may apply to the Work or any use thereof, including without
115+
limitation any person's Copyright and Related Rights in the Work.
116+
Further, Affirmer disclaims responsibility for obtaining any necessary
117+
consents, permissions or other rights required for any use of the
118+
Work.
119+
d. Affirmer understands and acknowledges that Creative Commons is not a
120+
party to this document and has no duty or obligation with respect to
121+
this CC0 or use of the Work.
122+
Lines changed: 199 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,199 @@
1+
// Rust Codex32 Library and Reference Implementation
2+
// Written in 2023 by
3+
// Andrew Poelstra <[email protected]>
4+
//
5+
// To the extent possible under law, the author(s) have dedicated all
6+
// copyright and related and neighboring rights to this software to
7+
// the public domain worldwide. This software is distributed without
8+
// any warranty.
9+
//
10+
// You should have received a copy of the CC0 Public Domain Dedication
11+
// along with this software.
12+
// If not, see <http://creativecommons.org/publicdomain/zero/1.0/>.
13+
//
14+
15+
//! Checksums
16+
//!
17+
//! Validates specific checksums
18+
//!
19+
20+
use super::{Case, Error};
21+
use crate::gf32::Fe;
22+
23+
/// An engine which consumes one GF32 character at a time, and produces
24+
/// a residue modulo some generator
25+
#[derive(Clone, PartialEq, Eq, Debug)]
26+
pub struct Engine {
27+
case: Option<Case>,
28+
generator: Vec<Fe>,
29+
residue: Vec<Fe>,
30+
target: Vec<Fe>,
31+
}
32+
33+
impl Engine {
34+
// An engine which computes the normal codex32 checksum
35+
pub fn new_codex32_short() -> Engine {
36+
Engine {
37+
case: None,
38+
#[rustfmt::skip]
39+
generator: vec![
40+
Fe::E, Fe::M, Fe::_3, Fe::G, Fe::Q, Fe::E,
41+
Fe::E, Fe::E, Fe::L, Fe::M, Fe::C, Fe::S,
42+
Fe::S,
43+
],
44+
#[rustfmt::skip]
45+
residue: vec![
46+
Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q,
47+
Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q,
48+
Fe::P,
49+
],
50+
#[rustfmt::skip]
51+
target: vec![
52+
Fe::S, Fe::E, Fe::C, Fe::R, Fe::E, Fe::T,
53+
Fe::S, Fe::H, Fe::A, Fe::R, Fe::E, Fe::_3,
54+
Fe::_2,
55+
],
56+
}
57+
}
58+
59+
// An engine which computes the "long" codex32 checksum
60+
pub fn new_codex32_long() -> Engine {
61+
// hyk9x4hx4ef6e20p
62+
Engine {
63+
case: None,
64+
#[rustfmt::skip]
65+
generator: vec![
66+
Fe::_0, Fe::_2, Fe::E, Fe::_6, Fe::F, Fe::E,
67+
Fe::_4, Fe::X, Fe::H, Fe::_4, Fe::X, Fe::_9,
68+
Fe::K, Fe::Y, Fe::H,
69+
],
70+
#[rustfmt::skip]
71+
residue: vec![
72+
Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q,
73+
Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q, Fe::Q,
74+
Fe::Q, Fe::Q, Fe::P,
75+
],
76+
#[rustfmt::skip]
77+
target: vec![
78+
Fe::S, Fe::E, Fe::C, Fe::R, Fe::E, Fe::T,
79+
Fe::S, Fe::H, Fe::A, Fe::R, Fe::E, Fe::_3,
80+
Fe::_2, Fe::E, Fe::X,
81+
],
82+
}
83+
}
84+
85+
/// Accessor for the generator polynomial, as a big-endian (highest powers
86+
/// first) vector of coefficients
87+
pub fn generator(&self) -> &[Fe] {
88+
&self.generator
89+
}
90+
91+
/// When computing checksums of "diffs" you do may want to set
92+
/// the highest-degree coefficient of the polynomial to 1.
93+
///
94+
/// If you do not know exactly why you are using this function,
95+
/// you should not use it.
96+
pub fn force_residue_to_zero(&mut self) {
97+
self.residue = vec![Fe::Q; self.residue.len()];
98+
}
99+
100+
/// Extracts the residue from a checksum engine
101+
pub fn into_residue(self) -> Vec<Fe> {
102+
self.residue
103+
}
104+
105+
/// Determines whether the residue matches the target value
106+
/// for the checksum
107+
///
108+
/// If you need the actual residue, e.g. for error correction,
109+
/// call the `into_residue` function (which will consume the
110+
/// engine).
111+
pub fn is_valid(&self) -> bool {
112+
self.residue == self.target
113+
}
114+
115+
/// Initializes the checksum engine by loading an HRP into it
116+
pub fn input_hrp(&mut self, hrp: &str) -> Result<(), Error> {
117+
for ch in hrp.chars() {
118+
self.set_check_case(ch)?;
119+
self.input_fe(Fe::from_int(u32::from(ch.to_ascii_lowercase()) >> 5)?);
120+
}
121+
self.input_fe(Fe::Q);
122+
for ch in hrp.chars() {
123+
self.input_fe(Fe::from_int(u32::from(ch.to_ascii_lowercase()) & 0x1f)?);
124+
}
125+
Ok(())
126+
}
127+
128+
/// Adds a single character to the checksum engine
129+
pub fn input_char(&mut self, c: char) -> Result<(), Error> {
130+
self.set_check_case(c)?;
131+
self.input_fe(Fe::from_char(c)?);
132+
Ok(())
133+
}
134+
135+
/// Adds an entire string to the engine, counting each character as a data character
136+
/// (not an HRP).
137+
pub fn input_data_str(&mut self, s: &str) -> Result<(), Error> {
138+
for ch in s.chars() {
139+
self.input_char(ch)?;
140+
}
141+
Ok(())
142+
}
143+
144+
/// Adds the target residue to the end of the input string
145+
pub fn input_own_target(&mut self) {
146+
// Need to clone self.target to iterate over it while calling self.input_fe,
147+
// which rustc worries may modify self.target.
148+
let sigh_borrowck = self.target.clone();
149+
for u in sigh_borrowck {
150+
self.input_fe(u);
151+
}
152+
}
153+
154+
/// Helper function to check that the whole input has consistent case
155+
fn set_check_case(&mut self, c: char) -> Result<(), Error> {
156+
if !c.is_ascii() {
157+
Err(Error::InvalidChar(c))
158+
} else if c.is_numeric() {
159+
// numbers don't affect case, nor are they affected by case
160+
Ok(())
161+
} else {
162+
let is_lower = c.is_ascii_lowercase();
163+
match (self.case, is_lower) {
164+
(Some(Case::Lower), true) | (Some(Case::Upper), false) => Ok(()),
165+
(Some(case @ Case::Lower), false) | (Some(case @ Case::Upper), true) => {
166+
Err(Error::InvalidCase(case, c))
167+
}
168+
(None, true) => {
169+
self.case = Some(Case::Lower);
170+
Ok(())
171+
}
172+
(None, false) => {
173+
self.case = Some(Case::Upper);
174+
Ok(())
175+
}
176+
}
177+
}
178+
}
179+
180+
/// Adds a single gf32 element to the checksum engine
181+
///
182+
/// This is where the real magic happens.
183+
#[rustfmt::skip]
184+
pub fn input_fe(&mut self, e: Fe) {
185+
let res_len = self.residue.len(); // needed for borrowck
186+
// Store current coefficient of x^{n-1}, which will become
187+
// x^n (and get reduced)
188+
let xn = self.residue[0];
189+
// Simply shift x^0 through x^{n-1} up one, and set x^0 to the new input
190+
for i in 1..res_len {
191+
self.residue[i - 1] = self.residue[i];
192+
}
193+
self.residue[res_len - 1] = e;
194+
// Then reduce x^n mod the generator.
195+
for i in 0..res_len {
196+
self.residue[i] += self.generator[i] * xn;
197+
}
198+
}
199+
}

0 commit comments

Comments
 (0)