summaryrefslogtreecommitdiffstats
path: root/utils/benchmark.c
blob: d5bb6258805852b793f436ebbc5f4b212632700f (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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
/* benchmark.c -- Benchmarks how quickly the computer does SHA512 hashing
 *
 *  GPLv2 only - Copyright (C) 2009
 *               David Sommerseth <dazo@users.sourceforge.net>
 *
 *  This program is free software; you can redistribute it and/or
 *  modify it under the terms of the GNU General Public License
 *  as published by the Free Software Foundation; version 2
 *  of the License.
 *
 *  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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 */

/**
 * @file   benchmark.c
 * @author David Sommerseth <dazo@users.sourceforge.net>
 * @date   2009-03-22
 *
 * @brief  Benchmarking utility to test how quickly a computer can calculate SHA512 hashes
 *
 */


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <sys/time.h>

#include <eurephia_nullsafe.h>
#include <eurephia_context.h>

#define ROUNDS_MIN 1000         /**< Define minimum acceptable rounds for the hashing */
#define ROUNDS_MAX 999999999    /**< Define maximum allowed rounds for the hashing */

int pack_saltinfo(char *buf, int buflen, int rounds, int saltlen, const char *pwd);

inline char *sha512_crypt_r(const char *key, const char *salt, size_t maxrounds_cfg, char *buffer, int buflen);

int gen_randsaltstr(eurephiaCTX *ctx, char *saltstr, int len);

/**
 * Internal function which calculates a hash based on a standard password.  This function is used
 * for measuring the time used for hash calculation
 *
 * @param rounds  Number of rounds the hashing should use
 */
void benchmark_hashing(int rounds) {
        char *buffer = NULL;
        char *pwdhash = NULL;
        static char randstr[34];
        static int rand_set = 0;
        char pwdsalt[64], saltstr[32];

        // Get random data for our salt
        if( rand_set == 0 ) {
                memset(&randstr, 0, 34);
                if( gen_randsaltstr(NULL, randstr, 32) == 0 ) {
                        fprintf(stderr, "Could not retrieve enough random data for hashing");
                        exit(19);
                };
                rand_set = 1;
        }

        // Prepare a salt package
        memset(&pwdsalt, 0, 64);
        memset(&saltstr, 0, 32);
        pack_saltinfo(saltstr, 30, rounds, 32, "benchmarkpassword");
        strncpy(pwdsalt, saltstr, strlen(saltstr));
        strncat(pwdsalt, randstr, 62 - strlen(saltstr));
        memset(&randstr, 0, 32);

        buffer = (char *) malloc_nullsafe(NULL, 1024);
        assert( buffer != NULL );

        pwdhash = sha512_crypt_r("benchmarkpassword", pwdsalt, ROUNDS_MAX, buffer, 1024);

        free_nullsafe(NULL, buffer);
}


/**
 * Internal function, calculating a time delta between two struct timeval.
 *
 * @param result Result pointer to where the result will be put
 * @param x      Value 1 with time information
 * @param y      Value 2 with time information
 *
 * @return Returns 1 if the X value is lower than the Y value
 */
int timeval_subtract (result, x, y)
     struct timeval *result, *x, *y;
{
        /* Perform the carry for the later subtraction by updating y. */
        if (x->tv_usec < y->tv_usec) {
                int nsec = (y->tv_usec - x->tv_usec) / 1000000 + 1;
                y->tv_usec -= 1000000 * nsec;
                y->tv_sec += nsec;
        }
        if (x->tv_usec - y->tv_usec > 1000000) {
                int nsec = (y->tv_usec - x->tv_usec) / 1000000;
                y->tv_usec += 1000000 * nsec;
                y->tv_sec -= nsec;
        }

        /* Compute the time remaining to wait.
           tv_usec is certainly positive. */
        result->tv_sec = x->tv_sec - y->tv_sec;
        result->tv_usec = x->tv_usec - y->tv_usec;

        /* Return 1 if result is negative. */
        return x->tv_sec < y->tv_sec;
}


/**
 * Internal function.  The real benchmark function.
 *
 * @param rounds Number of rounds to do when hashing a password
 *
 * @return Returns a struct timeval containing the time spent calculating the hash
 */
struct timeval *do_benchmark(int rounds) {
        struct timeval start, end, *timediff = NULL;

        timediff = (struct timeval *) malloc_nullsafe(NULL, sizeof(struct timeval)+2);
        assert( timediff != NULL );

        gettimeofday(&start, NULL);
        benchmark_hashing(rounds);
        gettimeofday(&end, NULL);
        timeval_subtract(timediff, &end, &start);

        return timediff;
}

/**
 * This function will try to find the best minium and maximum rounds for
 * the SHA512 hashing.  The values are returned in the min and max variables
 * and the thr_min and thr_max defines the minimum and maximum thresholds in
 * milliseconds the hashing algorithm should use.
 *
 * @param min      Return pointer for the minimum hashing rounds value
 * @param max      Return pointer for the maximum hashing rounds value
 * @param thr_min  Allow the calculation of hashes to last for thr_min ms.
 * @param thr_max  Do not allow the hash calculation to exceed thr_max ms.
 *
 * @return Returns 1 if a good suggestion for min and max hashing rounds was found.
 *         Otherwise 0 is returned.
 */
int benchmark(int *min, int *max, int thr_min, int thr_max) {
        int i = 0, success = 0;
        struct timeval *time = NULL, min_time, max_time ;

        printf("Benchmarking ");
        *min = 1000;
        *max = 5000;
        memset(&min_time, 0, sizeof(struct timeval));
        memset(&max_time, 0, sizeof(struct timeval));

        for( i = 1000; i < 100000000; i += 1000 ) {
                printf(".");
                fflush(stdout);
                time = do_benchmark(i);
                // printf("%i rounds: %ld sec %ld ms\n", i, time->tv_sec, time->tv_usec);
                if( time->tv_usec > (thr_max * 1000) ) {
                        success = 1;
                        free(time);
                        break;
                }
                if( (*min == 1000) && (time->tv_usec > (thr_min * 1000)) ) {
                        *min = i;
                        min_time.tv_usec = time->tv_usec;
                }
                if( (time->tv_usec < (thr_max * 1000)) ) {
                        *max = i;
                        max_time.tv_usec = time->tv_usec;
                }
                free(time);
        }
        printf(" Done\n");
        if( success == 1 ) {
                printf("Suggested minimum rounds: %i (takes %ldms)\n", *min, min_time.tv_usec/1000);
                printf("Suggested maximum rounds: %i (takes %ldms)\n", *max, max_time.tv_usec/1000);
        } else {
                *min = 0;
                *max = 0;
                printf("Could not find any good times, as your computer is too fast for this test.\n");
        }
        return success;
}

#if 0
int main() {
        int min, max;

        benchmark(&min, &max, 90, 220);
        printf("---> %i - %i\n", min, max);
        return 0;
}
#endif