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
|
#include <netdb.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <time.h>
#include <unistd.h>
#include "utils.h"
__attribute__((noreturn)) void usage(const char * const arg0)
{
fprintf(stderr, "usage: %s -k [SODIUM-KEY] -U [USER] -P [PASS] -r [HOST] -R [PORT] -f [FILE]\n", arg0);
exit(EXIT_FAILURE);
}
void parse_cmdline(struct cmd_options * const opts, int argc, char ** const argv)
{
int opt;
while ((opt = getopt(argc, argv, "k:U:P:r:R:f:h")) != -1) {
switch (opt) {
case 'k':
opts->key_string = strdup(optarg);
memset(optarg, '*', strlen(optarg));
break;
case 'U':
opts->user = strdup(optarg);
break;
case 'P':
opts->pass = strdup(optarg);
break;
case 'r':
opts->host = strdup(optarg);
break;
case 'R':
opts->port = strdup(optarg);
break;
case 'f':
opts->filepath = strdup(optarg);
break;
default:
usage(argv[0]);
}
}
if (opts->host == NULL) {
opts->host = strdup("127.0.0.1");
}
if (opts->port == NULL) {
opts->port = strdup("5555");
}
if (opts->key_string != NULL) {
opts->key_length = strlen(opts->key_string);
}
if (opts->user == NULL) {
opts->user = strdup("username");
}
if (opts->pass == NULL) {
opts->pass = strdup("passphrase");
}
}
char * prettify_bytes_with_units(char * const out, size_t out_size, unsigned long long bytes)
{
static char const * const unit_prefixes[] = {"", "Kilo", "Mega", "Giga", "Tera"};
size_t const unit_prefixes_length = sizeof(unit_prefixes) / sizeof(unit_prefixes[0]);
unsigned char unit_prefixes_index = 0;
size_t const convert_bytes_every = 1024;
while (bytes / convert_bytes_every > 0 && unit_prefixes_index < unit_prefixes_length) {
bytes /= convert_bytes_every;
unit_prefixes_index++;
}
snprintf(out, out_size, "%llu %sBytes", bytes, unit_prefixes[unit_prefixes_index]);
return out;
}
int hostname_to_address(char const * const host, char const * const port, struct addrinfo ** const result)
{
int s;
struct addrinfo hints;
memset(&hints, 0, sizeof(struct addrinfo));
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
hints.ai_flags = AI_PASSIVE;
s = getaddrinfo(host, port, &hints, result);
if (s != 0) {
return s;
}
return 0;
}
void strftime_local(double time_in_secs, char * const out, size_t out_size)
{
time_t t = (time_t)time_in_secs;
struct tm r;
if (localtime_r(&t, &r) == NULL)
{
out[0] = '\0';
return;
}
if (out_size > TIMESTAMP_STRLEN)
{
out_size = TIMESTAMP_STRLEN;
}
if (strftime(out, out_size, "%a, %d %b %Y %T %z", &r) <= 0)
{
out[0] = '\0';
return;
}
}
|