summaryrefslogtreecommitdiff
path: root/lib/freebl/mpi/utils/fact.c
blob: da8e61a32f183a870def5fe30da5d0afd50b9070 (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
78
79
80
81
82
83
84
/*
 * fact.c
 *
 * Compute factorial of input integer
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "mpi.h"

mp_err mp_fact(mp_int *a, mp_int *b);

int
main(int argc, char *argv[])
{
    mp_int a;
    mp_err res;

    if (argc < 2) {
        fprintf(stderr, "Usage: %s <number>\n", argv[0]);
        return 1;
    }

    mp_init(&a);
    mp_read_radix(&a, argv[1], 10);

    if ((res = mp_fact(&a, &a)) != MP_OKAY) {
        fprintf(stderr, "%s: error: %s\n", argv[0],
                mp_strerror(res));
        mp_clear(&a);
        return 1;
    }

    {
        char *buf;
        int len;

        len = mp_radix_size(&a, 10);
        buf = malloc(len);
        mp_todecimal(&a, buf);

        puts(buf);

        free(buf);
    }

    mp_clear(&a);
    return 0;
}

mp_err
mp_fact(mp_int *a, mp_int *b)
{
    mp_int ix, s;
    mp_err res = MP_OKAY;

    if (mp_cmp_z(a) < 0)
        return MP_UNDEF;

    mp_init(&s);
    mp_add_d(&s, 1, &s); /* s = 1  */
    mp_init(&ix);
    mp_add_d(&ix, 1, &ix); /* ix = 1 */

    for (/*  */; mp_cmp(&ix, a) <= 0; mp_add_d(&ix, 1, &ix)) {
        if ((res = mp_mul(&s, &ix, &s)) != MP_OKAY)
            break;
    }

    mp_clear(&ix);

    /* Copy out results if we got them */
    if (res == MP_OKAY)
        mp_copy(&s, b);

    mp_clear(&s);

    return res;
}