10 static gmp_randstate_t RANDSTATE;
13 void d0_bignum_INITIALIZE()
15 gmp_randinit_default(RANDSTATE);
16 gmp_randseed_ui(RANDSTATE, time(NULL)); // TODO seed
19 void d0_bignum_SHUTDOWN()
24 BOOL d0_iobuf_write_bignum(d0_iobuf_t *buf, const d0_bignum_t *bignum)
26 static unsigned char numbuf[65536];
28 numbuf[0] = mpz_sgn(bignum->z) & 3;
29 if((numbuf[0] & 3) != 0) // nonzero
31 count = (mpz_sizeinbase(bignum->z, 2) + 7) / 8;
32 if(count > sizeof(numbuf) - 1)
34 mpz_export(numbuf+1, &count, 1, 1, 0, 0, bignum->z);
36 return d0_iobuf_write_packet(buf, numbuf, count + 1);
39 d0_bignum_t *d0_iobuf_read_bignum(d0_iobuf_t *buf, d0_bignum_t *bignum)
41 static unsigned char numbuf[65536];
42 size_t count = sizeof(numbuf);
43 if(!d0_iobuf_read_packet(buf, numbuf, &count))
47 if(!bignum) bignum = d0_bignum_new(); if(!bignum) return NULL;
48 if(numbuf[0] & 3) // nonzero
50 mpz_import(bignum->z, count-1, 1, 1, 0, 0, numbuf+1);
51 if(numbuf[0] & 2) // negative
52 mpz_neg(bignum->z, bignum->z);
56 mpz_set_ui(bignum->z, 0);
61 d0_bignum_t *d0_bignum_new()
63 d0_bignum_t *b = d0_malloc(sizeof(d0_bignum_t));
68 void d0_bignum_free(d0_bignum_t *a)
74 void d0_bignum_init(d0_bignum_t *b)
79 void d0_bignum_clear(d0_bignum_t *a)
84 size_t d0_bignum_size(const d0_bignum_t *r)
86 return mpz_sizeinbase(r->z, 2);
89 int d0_bignum_cmp(const d0_bignum_t *a, const d0_bignum_t *b)
91 return mpz_cmp(a->z, b->z);
94 d0_bignum_t *d0_bignum_rand_range(d0_bignum_t *r, const d0_bignum_t *min, const d0_bignum_t *max)
96 static d0_bignum_t *temp = NULL; if(!temp) temp = d0_bignum_new();
97 if(!r) r = d0_bignum_new(); if(!r) return NULL;
98 mpz_sub(temp->z, max->z, min->z);
99 mpz_urandomm(r->z, RANDSTATE, temp->z);
100 mpz_add(r->z, r->z, min->z);
104 d0_bignum_t *d0_bignum_rand_bit_atmost(d0_bignum_t *r, size_t n)
106 if(!r) r = d0_bignum_new(); if(!r) return NULL;
107 mpz_urandomb(r->z, RANDSTATE, n);
111 d0_bignum_t *d0_bignum_rand_bit_exact(d0_bignum_t *r, size_t n)
113 if(!r) r = d0_bignum_new(); if(!r) return NULL;
114 mpz_urandomb(r->z, RANDSTATE, n-1);
115 mpz_setbit(r->z, n-1);
119 d0_bignum_t *d0_bignum_zero(d0_bignum_t *r)
121 return d0_bignum_int(r, 0);
124 d0_bignum_t *d0_bignum_one(d0_bignum_t *r)
126 return d0_bignum_int(r, 1);
129 d0_bignum_t *d0_bignum_int(d0_bignum_t *r, int n)
131 if(!r) r = d0_bignum_new(); if(!r) return NULL;
136 d0_bignum_t *d0_bignum_mov(d0_bignum_t *r, const d0_bignum_t *a)
140 if(!r) r = d0_bignum_new(); if(!r) return NULL;
145 d0_bignum_t *d0_bignum_neg(d0_bignum_t *r, const d0_bignum_t *a)
147 if(!r) r = d0_bignum_new(); if(!r) return NULL;
152 d0_bignum_t *d0_bignum_shl(d0_bignum_t *r, const d0_bignum_t *a, ssize_t n)
154 if(!r) r = d0_bignum_new(); if(!r) return NULL;
156 mpz_mul_2exp(r->z, a->z, n);
158 mpz_fdiv_q_2exp(r->z, a->z, -n);
164 d0_bignum_t *d0_bignum_add(d0_bignum_t *r, const d0_bignum_t *a, const d0_bignum_t *b)
166 if(!r) r = d0_bignum_new(); if(!r) return NULL;
167 mpz_add(r->z, a->z, b->z);
171 d0_bignum_t *d0_bignum_sub(d0_bignum_t *r, const d0_bignum_t *a, const d0_bignum_t *b)
173 if(!r) r = d0_bignum_new(); if(!r) return NULL;
174 mpz_sub(r->z, a->z, b->z);
178 d0_bignum_t *d0_bignum_mul(d0_bignum_t *r, const d0_bignum_t *a, const d0_bignum_t *b)
180 if(!r) r = d0_bignum_new(); if(!r) return NULL;
181 mpz_mul(r->z, a->z, b->z);
185 d0_bignum_t *d0_bignum_divmod(d0_bignum_t *q, d0_bignum_t *m, const d0_bignum_t *a, const d0_bignum_t *b)
191 mpz_fdiv_qr(q->z, m->z, a->z, b->z);
193 mpz_fdiv_q(q->z, a->z, b->z);
195 mpz_fdiv_r(m->z, a->z, b->z);
202 d0_bignum_t *d0_bignum_mod_add(d0_bignum_t *r, const d0_bignum_t *a, const d0_bignum_t *b, const d0_bignum_t *m)
204 r = d0_bignum_add(r, a, b);
205 mpz_fdiv_r(r->z, r->z, m->z);
209 d0_bignum_t *d0_bignum_mod_mul(d0_bignum_t *r, const d0_bignum_t *a, const d0_bignum_t *b, const d0_bignum_t *m)
211 r = d0_bignum_mul(r, a, b);
212 mpz_fdiv_r(r->z, r->z, m->z);
216 d0_bignum_t *d0_bignum_mod_pow(d0_bignum_t *r, const d0_bignum_t *a, const d0_bignum_t *b, const d0_bignum_t *m)
218 if(!r) r = d0_bignum_new(); if(!r) return NULL;
219 mpz_powm(r->z, a->z, b->z, m->z);
223 BOOL d0_bignum_mod_inv(d0_bignum_t *r, const d0_bignum_t *a, const d0_bignum_t *m)
225 // here, r MUST be set, as otherwise we cannot return error state!
226 return mpz_invert(r->z, a->z, m->z);
229 int d0_bignum_isprime(d0_bignum_t *r, int param)
231 return mpz_probab_prime_p(r->z, param);
234 d0_bignum_t *d0_bignum_gcd(d0_bignum_t *r, d0_bignum_t *s, d0_bignum_t *t, const d0_bignum_t *a, const d0_bignum_t *b)
236 if(!r) r = d0_bignum_new(); if(!r) return NULL;
238 mpz_gcdext(r->z, s->z, t ? t->z : NULL, a->z, b->z);
240 mpz_gcdext(r->z, t->z, NULL, b->z, a->z);
242 mpz_gcd(r->z, a->z, b->z);
246 char *d0_bignum_tostring(const d0_bignum_t *x, unsigned int base)
248 return mpz_get_str(NULL, base, x->z);