summaryrefslogtreecommitdiff
path: root/serde_v8/magic/bigint.rs
blob: 330803daf8088c3a98370dd461cc1fa1b68fc2e2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.

use smallvec::smallvec;
use smallvec::SmallVec;

use super::transl8::FromV8;
use super::transl8::ToV8;
use crate::error::value_to_type_str;
use crate::magic::transl8::impl_magic;
use crate::Error;

#[derive(
  PartialEq,
  Eq,
  Clone,
  Debug,
  Default,
  derive_more::Deref,
  derive_more::DerefMut,
  derive_more::AsRef,
  derive_more::AsMut,
)]
#[as_mut(forward)]
#[as_ref(forward)]
pub struct BigInt(num_bigint::BigInt);
impl_magic!(BigInt);

impl ToV8 for BigInt {
  fn to_v8<'a>(
    &mut self,
    scope: &mut v8::HandleScope<'a>,
  ) -> Result<v8::Local<'a, v8::Value>, crate::Error> {
    let (sign, words) = self.0.to_u64_digits();
    let sign_bit = sign == num_bigint::Sign::Minus;
    let v = v8::BigInt::new_from_words(scope, sign_bit, &words).unwrap();
    Ok(v.into())
  }
}

impl FromV8 for BigInt {
  fn from_v8(
    _scope: &mut v8::HandleScope,
    value: v8::Local<v8::Value>,
  ) -> Result<Self, crate::Error> {
    let v8bigint = v8::Local::<v8::BigInt>::try_from(value)
      .map_err(|_| Error::ExpectedBigInt(value_to_type_str(value)))?;
    let word_count = v8bigint.word_count();
    let mut words: SmallVec<[u64; 1]> = smallvec![0u64; word_count];
    let (sign_bit, _words) = v8bigint.to_words_array(&mut words);
    let sign = match sign_bit {
      true => num_bigint::Sign::Minus,
      false => num_bigint::Sign::Plus,
    };
    // SAFETY: Because the alignment of u64 is 8, the alignment of u32 is 4, and
    // the size of u64 is 8, the size of u32 is 4, the alignment of u32 is a
    // factor of the alignment of u64, and the size of u32 is a factor of the
    // size of u64, we can safely transmute the slice of u64 to a slice of u32.
    let (prefix, slice, suffix) = unsafe { words.align_to::<u32>() };
    assert!(prefix.is_empty());
    assert!(suffix.is_empty());
    assert_eq!(slice.len(), words.len() * 2);
    let big_int = num_bigint::BigInt::from_slice(sign, slice);
    Ok(Self(big_int))
  }
}

impl From<num_bigint::BigInt> for BigInt {
  fn from(big_int: num_bigint::BigInt) -> Self {
    Self(big_int)
  }
}

impl From<BigInt> for num_bigint::BigInt {
  fn from(big_int: BigInt) -> Self {
    big_int.0
  }
}