1 /*
2  * Copyright (c) 2022 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <sys/time.h>
18 
19 #include <stdio.h>
20 #include <limits.h>
21 #include <string.h>
22 #include <time.h>
23 
24 #include "got_lib_ratelimit.h"
25 
26 #include "got_error.h"
27 
28 void
29 got_ratelimit_init(struct got_ratelimit *rl, time_t interval_sec,
30     unsigned int interval_msec)
31 {
32 	memset(rl, 0, sizeof(*rl));
33 	rl->interval.tv_sec = interval_sec;
34 	rl->interval.tv_nsec = interval_msec * 1000000UL;
35 }
36 
37 const struct got_error *
38 got_ratelimit_check(int *elapsed, struct got_ratelimit *rl)
39 {
40 	struct timespec now, delta;
41 
42 	if (clock_gettime(CLOCK_MONOTONIC, &now) == -1)
43 		return got_error_from_errno("clock_gettime");
44 
45 	if (timespecisset(&rl->last)) {
46 		timespecsub(&now, &rl->last, &delta);
47 		*elapsed = timespeccmp(&delta, &rl->interval, >=) ? 1 : 0;
48 	} else
49 		*elapsed = 1;
50 
51 	if (*elapsed)
52 		rl->last = now;
53 
54 	return NULL;
55 }