Procházet zdrojové kódy

serial: Add support for num-bigint

parazyd před 2 roky
rodič
revize
124a9466cf
4 změnil soubory, kde provedl 95 přidání a 3 odebrání
  1. 3 3
      Cargo.lock
  2. 1 0
      src/serial/Cargo.toml
  3. 3 0
      src/serial/src/types.rs
  4. 88 0
      src/serial/src/types/bigint.rs

+ 3 - 3
Cargo.lock

@@ -2105,6 +2105,7 @@ dependencies = [
  "bridgetree",
  "bridgetree",
  "darkfi-derive",
  "darkfi-derive",
  "futures-lite 2.3.0",
  "futures-lite 2.3.0",
+ "num-bigint",
  "pasta_curves",
  "pasta_curves",
  "semver 1.0.22",
  "semver 1.0.22",
  "url",
  "url",
@@ -4487,11 +4488,10 @@ dependencies = [
 
 
 [[package]]
 [[package]]
 name = "num-bigint"
 name = "num-bigint"
-version = "0.4.4"
+version = "0.4.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "608e7659b5c3d7cba262d894801b9ec9d00de989e8a82bd4bef91d08da45cdc0"
+checksum = "c165a9ab64cf766f73521c0dd2cfdff64f488b8f0b3e621face3462d3db536d7"
 dependencies = [
 dependencies = [
- "autocfg",
  "num-integer",
  "num-integer",
  "num-traits",
  "num-traits",
 ]
 ]

+ 1 - 0
src/serial/Cargo.toml

@@ -19,6 +19,7 @@ bridgetree = {version = "0.4.0", optional = true}
 pasta_curves = {version = "0.5.1", optional = true}
 pasta_curves = {version = "0.5.1", optional = true}
 semver = {version = "1.0.22", optional = true}
 semver = {version = "1.0.22", optional = true}
 url = {version = "2.5.0", optional = true}
 url = {version = "2.5.0", optional = true}
+num-bigint = {version = "0.4.5", optional = true}
 
 
 [features]
 [features]
 default = ["derive", "semver"]
 default = ["derive", "semver"]

+ 3 - 0
src/serial/src/types.rs

@@ -35,3 +35,6 @@ mod url;
 
 
 #[cfg(feature = "semver")]
 #[cfg(feature = "semver")]
 mod semver;
 mod semver;
+
+#[cfg(feature = "num-bigint")]
+mod bigint;

+ 88 - 0
src/serial/src/types/bigint.rs

@@ -0,0 +1,88 @@
+/* This file is part of DarkFi (https://dark.fi)
+ *
+ * Copyright (C) 2020-2024 Dyne.org foundation
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+use std::io::{Read, Result, Write};
+
+#[cfg(feature = "async")]
+use crate::{AsyncDecodable, AsyncEncodable};
+#[cfg(feature = "async")]
+use async_trait::async_trait;
+#[cfg(feature = "async")]
+use futures_lite::{AsyncRead, AsyncWrite};
+
+use crate::{Decodable, Encodable};
+
+impl Encodable for num_bigint::BigInt {
+    fn encode<S: Write>(&self, s: &mut S) -> Result<usize> {
+        self.to_signed_bytes_be().encode(s)
+    }
+}
+
+#[cfg(feature = "async")]
+#[async_trait]
+impl AsyncEncodable for num_bigint::BigInt {
+    async fn encode_async<S: AsyncWrite + Unpin + Send>(&self, s: &mut S) -> Result<usize> {
+        self.to_signed_bytes_be().encode_async(s).await
+    }
+}
+
+impl Decodable for num_bigint::BigInt {
+    fn decode<D: Read>(d: &mut D) -> Result<Self> {
+        let vec: Vec<u8> = Decodable::decode(d)?;
+        Ok(Self::from_signed_bytes_be(&vec))
+    }
+}
+
+#[cfg(feature = "async")]
+#[async_trait]
+impl AsyncDecodable for num_bigint::BigInt {
+    async fn decode_async<D: AsyncRead + Unpin + Send>(d: &mut D) -> Result<Self> {
+        let vec: Vec<u8> = AsyncDecodable::decode_async(d).await?;
+        Ok(Self::from_signed_bytes_be(&vec))
+    }
+}
+
+impl Encodable for num_bigint::BigUint {
+    fn encode<S: Write>(&self, s: &mut S) -> Result<usize> {
+        self.to_bytes_be().encode(s)
+    }
+}
+
+#[cfg(feature = "async")]
+#[async_trait]
+impl AsyncEncodable for num_bigint::BigUint {
+    async fn encode_async<S: AsyncWrite + Unpin + Send>(&self, s: &mut S) -> Result<usize> {
+        self.to_bytes_be().encode_async(s).await
+    }
+}
+
+impl Decodable for num_bigint::BigUint {
+    fn decode<D: Read>(d: &mut D) -> Result<Self> {
+        let vec: Vec<u8> = Decodable::decode(d)?;
+        Ok(Self::from_bytes_be(&vec))
+    }
+}
+
+#[cfg(feature = "async")]
+#[async_trait]
+impl AsyncDecodable for num_bigint::BigUint {
+    async fn decode_async<D: AsyncRead + Unpin + Send>(d: &mut D) -> Result<Self> {
+        let vec: Vec<u8> = AsyncDecodable::decode_async(d).await?;
+        Ok(Self::from_bytes_be(&vec))
+    }
+}