cscan/utils.c at master · zb3/cscan · GitHub

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
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <stdarg.h>
#include <time.h>

unsigned long long get_ms()
{
 struct timespec spec;
 clock_gettime(CLOCK_MONOTONIC, &spec);

 return spec.tv_sec*1000 + spec.tv_nsec/1000000;
}

unsigned long long get_us()
{
 struct timespec spec;
 clock_gettime(CLOCK_MONOTONIC, &spec);

 return spec.tv_sec*1000000 + spec.tv_nsec/1000;
}

void fail(const char *fmt, ...)
{
 va_list args;
 va_start(args, fmt);
 vfprintf(stderr, fmt, args);
 va_end(args);

 fprintf(stderr, "\n");
 fflush(stderr);

 exit(1);
}

void vinfo(const char *fmt, va_list args)
{
 vfprintf(stderr, fmt, args);
 fprintf(stderr, "\n");
 fflush(stderr);
}

void info(const char *fmt, ...)
{
 va_list args;
 va_start(args, fmt);
 va_end(args);

 vinfo(fmt, args);
}

size_t freadle(void * ptr, size_t size, FILE * stream)
{
 size_t ret = fread(ptr, size, 1, stream);

 #if __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
 if (ret)
 {
  if (size == 2) *(uint16_t *)ptr = __builtin_bswap16(*((uint16_t *)ptr));
  else if (size == 4) *(uint32_t *)ptr = __builtin_bswap32(*((uint32_t *)ptr));
  else if (size == 8) *(uint64_t *)ptr = __builtin_bswap64(*((uint64_t *)ptr));
 }
 #endif

 return ret;
}

size_t fwritele(void * ptr, size_t size, FILE * stream)
{
 size_t ret = 0;

 #if __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
 char *pptr = ptr;
 pptr += size;

 while(size --> 0)
 {
  ret += fwrite(--pptr, 1, 1, stream);
 }

 ret /= size;

 #else
 ret = fwrite(ptr, size, 1, stream);
 #endif


 return ret;
}