util.c 1.79 KB
Newer Older
1
2
3
4
5
/* Copyright 2003 Roger Dingledine */
/* See LICENSE for licensing information */
/* $Id$ */

#include <stdlib.h>
Roger Dingledine's avatar
Roger Dingledine committed
6
#include <limits.h>
7
8
9
#include "util.h"
#include "log.h"

10
11
12
13
14
15
void *tor_malloc(size_t size) {
  void *result;

  result = malloc(size);

  if(!result) {
16
    log_fn(LOG_ERR, "Out of memory. Dying.");
17
18
19
20
21
22
    exit(1);
  }

  return result;
}

23
24
25
26
void 
my_gettimeofday(struct timeval *timeval) 
{
  if (gettimeofday(timeval, NULL)) {
27
    log_fn(LOG_ERR, "gettimeofday failed.");
28
29
30
31
32
33
34
35
36
37
    /* If gettimeofday dies, we have either given a bad timezone (we didn't),
       or segfaulted.*/
    exit(1);
  }
  return;
}

long
tv_udiff(struct timeval *start, struct timeval *end)
{
Roger Dingledine's avatar
Roger Dingledine committed
38
  long udiff;
39
  long end_usec = end->tv_usec;
40
  long secdiff = end->tv_sec - start->tv_sec;
41

42
  if (secdiff+1 > LONG_MAX/1000000) {
43
    log_fn(LOG_NOTICE, "comparing times too far apart.");
44
45
    return LONG_MAX;
  }
46
47
48
49

  /*
    This is a no-op: "secdiff--" takes 1M from the final result,
    and "end_usec+=100000L" puts it back.
50
51
52
  if (end_usec < start->tv_usec) {
    secdiff--;
    end_usec += 1000000L;
53
  }
54
  */
55
  udiff = secdiff*1000000L + (end_usec - start->tv_usec);
Roger Dingledine's avatar
Roger Dingledine committed
56
  if(udiff < 0) {
57
    log_fn(LOG_NOTICE, "start is after end. Returning 0.");
Roger Dingledine's avatar
Roger Dingledine committed
58
59
60
    return 0;
  }
  return udiff;
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
}

int tv_cmp(struct timeval *a, struct timeval *b) {
  if (a->tv_sec > b->tv_sec)
    return 1;
  if (a->tv_sec < b->tv_sec)
    return -1;
  if (a->tv_usec > b->tv_usec)
    return 1;
  if (a->tv_usec < b->tv_usec)
    return -1;
  return 0;
}

void tv_add(struct timeval *a, struct timeval *b) {
  a->tv_usec += b->tv_usec;
  a->tv_sec += b->tv_sec + (a->tv_usec / 1000000);
  a->tv_usec %= 1000000;
}

void tv_addms(struct timeval *a, long ms) {
  a->tv_usec += (ms * 1000) % 1000000;
  a->tv_sec += ((ms * 1000) / 1000000) + (a->tv_usec / 1000000);
  a->tv_usec %= 1000000;
}